23 	struct ubi_device *ubi = container_of(wrk, struct ubi_device, fm_work);  in update_fastmap_work_fn()  local
25 	ubi_update_fastmap(ubi);  in update_fastmap_work_fn()
26 	spin_lock(&ubi->wl_lock);  in update_fastmap_work_fn()
27 	ubi->fm_work_scheduled = 0;  in update_fastmap_work_fn()
28 	spin_unlock(&ubi->wl_lock);  in update_fastmap_work_fn()
56 static void return_unused_pool_pebs(struct ubi_device *ubi,  in return_unused_pool_pebs()  argument
63 		e = ubi->lookuptbl[pool->pebs[i]];  in return_unused_pool_pebs()
64 		wl_tree_add(e, &ubi->free);  in return_unused_pool_pebs()
65 		ubi->free_count++;  in return_unused_pool_pebs()
90 struct ubi_wl_entry *ubi_wl_get_fm_peb(struct ubi_device *ubi, int anchor)  in ubi_wl_get_fm_peb()  argument
94 	if (!ubi->free.rb_node || (ubi->free_count - ubi->beb_rsvd_pebs < 1))  in ubi_wl_get_fm_peb()
98 		e = find_anchor_wl_entry(&ubi->free);  in ubi_wl_get_fm_peb()
100 		e = find_mean_wl_entry(ubi, &ubi->free);  in ubi_wl_get_fm_peb()
105 	self_check_in_wl_tree(ubi, e, &ubi->free);  in ubi_wl_get_fm_peb()
109 	rb_erase(&e->u.rb, &ubi->free);  in ubi_wl_get_fm_peb()
110 	ubi->free_count--;  in ubi_wl_get_fm_peb()
119 void ubi_refill_pools(struct ubi_device *ubi)  in ubi_refill_pools()  argument
121 	struct ubi_fm_pool *wl_pool = &ubi->fm_wl_pool;  in ubi_refill_pools()
122 	struct ubi_fm_pool *pool = &ubi->fm_pool;  in ubi_refill_pools()
126 	spin_lock(&ubi->wl_lock);  in ubi_refill_pools()
128 	return_unused_pool_pebs(ubi, wl_pool);  in ubi_refill_pools()
129 	return_unused_pool_pebs(ubi, pool);  in ubi_refill_pools()
137 			if (!ubi->free.rb_node)  in ubi_refill_pools()
140 			e = wl_get_wle(ubi);  in ubi_refill_pools()
150 			if (!ubi->free.rb_node ||  in ubi_refill_pools()
151 			   (ubi->free_count - ubi->beb_rsvd_pebs < 5))  in ubi_refill_pools()
154 			e = find_wl_entry(ubi, &ubi->free, WL_FREE_MAX_DIFF);  in ubi_refill_pools()
155 			self_check_in_wl_tree(ubi, e, &ubi->free);  in ubi_refill_pools()
156 			rb_erase(&e->u.rb, &ubi->free);  in ubi_refill_pools()
157 			ubi->free_count--;  in ubi_refill_pools()
171 	spin_unlock(&ubi->wl_lock);  in ubi_refill_pools()
182 int ubi_wl_get_peb(struct ubi_device *ubi)  in ubi_wl_get_peb()  argument
185 	struct ubi_fm_pool *pool = &ubi->fm_pool;  in ubi_wl_get_peb()
186 	struct ubi_fm_pool *wl_pool = &ubi->fm_wl_pool;  in ubi_wl_get_peb()
189 	down_read(&ubi->fm_eba_sem);  in ubi_wl_get_peb()
190 	spin_lock(&ubi->wl_lock);  in ubi_wl_get_peb()
195 		spin_unlock(&ubi->wl_lock);  in ubi_wl_get_peb()
196 		up_read(&ubi->fm_eba_sem);  in ubi_wl_get_peb()
197 		ret = ubi_update_fastmap(ubi);  in ubi_wl_get_peb()
199 			ubi_msg(ubi, "Unable to write a new fastmap: %i", ret);  in ubi_wl_get_peb()
200 			down_read(&ubi->fm_eba_sem);  in ubi_wl_get_peb()
203 		down_read(&ubi->fm_eba_sem);  in ubi_wl_get_peb()
204 		spin_lock(&ubi->wl_lock);  in ubi_wl_get_peb()
208 		spin_unlock(&ubi->wl_lock);  in ubi_wl_get_peb()
210 			ubi_err(ubi, "Unable to get a free PEB from user WL pool");  in ubi_wl_get_peb()
215 		up_read(&ubi->fm_eba_sem);  in ubi_wl_get_peb()
221 	prot_queue_add(ubi, ubi->lookuptbl[ret]);  in ubi_wl_get_peb()
222 	spin_unlock(&ubi->wl_lock);  in ubi_wl_get_peb()
231 static struct ubi_wl_entry *get_peb_for_wl(struct ubi_device *ubi)  in get_peb_for_wl()  argument
233 	struct ubi_fm_pool *pool = &ubi->fm_wl_pool;  in get_peb_for_wl()
240 		if (!ubi->fm_work_scheduled) {  in get_peb_for_wl()
241 			ubi->fm_work_scheduled = 1;  in get_peb_for_wl()
242 			schedule_work(&ubi->fm_work);  in get_peb_for_wl()
248 	return ubi->lookuptbl[pnum];  in get_peb_for_wl()
255 int ubi_ensure_anchor_pebs(struct ubi_device *ubi)  in ubi_ensure_anchor_pebs()  argument
259 	spin_lock(&ubi->wl_lock);  in ubi_ensure_anchor_pebs()
260 	if (ubi->wl_scheduled) {  in ubi_ensure_anchor_pebs()
261 		spin_unlock(&ubi->wl_lock);  in ubi_ensure_anchor_pebs()
264 	ubi->wl_scheduled = 1;  in ubi_ensure_anchor_pebs()
265 	spin_unlock(&ubi->wl_lock);  in ubi_ensure_anchor_pebs()
269 		spin_lock(&ubi->wl_lock);  in ubi_ensure_anchor_pebs()
270 		ubi->wl_scheduled = 0;  in ubi_ensure_anchor_pebs()
271 		spin_unlock(&ubi->wl_lock);  in ubi_ensure_anchor_pebs()
277 	schedule_ubi_work(ubi, wrk);  in ubi_ensure_anchor_pebs()
291 int ubi_wl_put_fm_peb(struct ubi_device *ubi, struct ubi_wl_entry *fm_e,  in ubi_wl_put_fm_peb()  argument
300 	ubi_assert(pnum < ubi->peb_count);  in ubi_wl_put_fm_peb()
302 	spin_lock(&ubi->wl_lock);  in ubi_wl_put_fm_peb()
303 	e = ubi->lookuptbl[pnum];  in ubi_wl_put_fm_peb()
312 		ubi->lookuptbl[pnum] = e;  in ubi_wl_put_fm_peb()
315 	spin_unlock(&ubi->wl_lock);  in ubi_wl_put_fm_peb()
318 	return schedule_erase(ubi, e, vol_id, lnum, torture);  in ubi_wl_put_fm_peb()
330 static void ubi_fastmap_close(struct ubi_device *ubi)  in ubi_fastmap_close()  argument
334 	flush_work(&ubi->fm_work);  in ubi_fastmap_close()
335 	return_unused_pool_pebs(ubi, &ubi->fm_pool);  in ubi_fastmap_close()
336 	return_unused_pool_pebs(ubi, &ubi->fm_wl_pool);  in ubi_fastmap_close()
338 	if (ubi->fm) {  in ubi_fastmap_close()
339 		for (i = 0; i < ubi->fm->used_blocks; i++)  in ubi_fastmap_close()
340 			kfree(ubi->fm->e[i]);  in ubi_fastmap_close()
342 	kfree(ubi->fm);  in ubi_fastmap_close()
353 static struct ubi_wl_entry *may_reserve_for_fm(struct ubi_device *ubi,  in may_reserve_for_fm()  argument
356 	if (e && !ubi->fm_disabled && !ubi->fm &&  in may_reserve_for_fm()