14 	struct ubi_device *ubi = container_of(wrk, struct ubi_device, fm_work);  in update_fastmap_work_fn()  local
16 	ubi_update_fastmap(ubi);  in update_fastmap_work_fn()
17 	spin_lock(&ubi->wl_lock);  in update_fastmap_work_fn()
18 	ubi->fm_work_scheduled = 0;  in update_fastmap_work_fn()
19 	spin_unlock(&ubi->wl_lock);  in update_fastmap_work_fn()
47 static void return_unused_pool_pebs(struct ubi_device *ubi,  in return_unused_pool_pebs()  argument
54 		e = ubi->lookuptbl[pool->pebs[i]];  in return_unused_pool_pebs()
55 		wl_tree_add(e, &ubi->free);  in return_unused_pool_pebs()
56 		ubi->free_count++;  in return_unused_pool_pebs()
81 struct ubi_wl_entry *ubi_wl_get_fm_peb(struct ubi_device *ubi, int anchor)  in ubi_wl_get_fm_peb()  argument
85 	if (!ubi->free.rb_node || (ubi->free_count - ubi->beb_rsvd_pebs < 1))  in ubi_wl_get_fm_peb()
89 		e = find_anchor_wl_entry(&ubi->free);  in ubi_wl_get_fm_peb()
91 		e = find_mean_wl_entry(ubi, &ubi->free);  in ubi_wl_get_fm_peb()
96 	self_check_in_wl_tree(ubi, e, &ubi->free);  in ubi_wl_get_fm_peb()
100 	rb_erase(&e->u.rb, &ubi->free);  in ubi_wl_get_fm_peb()
101 	ubi->free_count--;  in ubi_wl_get_fm_peb()
110 void ubi_refill_pools(struct ubi_device *ubi)  in ubi_refill_pools()  argument
112 	struct ubi_fm_pool *wl_pool = &ubi->fm_wl_pool;  in ubi_refill_pools()
113 	struct ubi_fm_pool *pool = &ubi->fm_pool;  in ubi_refill_pools()
117 	spin_lock(&ubi->wl_lock);  in ubi_refill_pools()
119 	return_unused_pool_pebs(ubi, wl_pool);  in ubi_refill_pools()
120 	return_unused_pool_pebs(ubi, pool);  in ubi_refill_pools()
128 			if (!ubi->free.rb_node)  in ubi_refill_pools()
131 			e = wl_get_wle(ubi);  in ubi_refill_pools()
141 			if (!ubi->free.rb_node ||  in ubi_refill_pools()
142 			   (ubi->free_count - ubi->beb_rsvd_pebs < 5))  in ubi_refill_pools()
145 			e = find_wl_entry(ubi, &ubi->free, WL_FREE_MAX_DIFF);  in ubi_refill_pools()
146 			self_check_in_wl_tree(ubi, e, &ubi->free);  in ubi_refill_pools()
147 			rb_erase(&e->u.rb, &ubi->free);  in ubi_refill_pools()
148 			ubi->free_count--;  in ubi_refill_pools()
162 	spin_unlock(&ubi->wl_lock);  in ubi_refill_pools()
174 static int produce_free_peb(struct ubi_device *ubi)  in produce_free_peb()  argument
178 	while (!ubi->free.rb_node && ubi->works_count) {  in produce_free_peb()
180 		err = do_work(ubi);  in produce_free_peb()
197 int ubi_wl_get_peb(struct ubi_device *ubi)  in ubi_wl_get_peb()  argument
200 	struct ubi_fm_pool *pool = &ubi->fm_pool;  in ubi_wl_get_peb()
201 	struct ubi_fm_pool *wl_pool = &ubi->fm_wl_pool;  in ubi_wl_get_peb()
204 	down_read(&ubi->fm_eba_sem);  in ubi_wl_get_peb()
205 	spin_lock(&ubi->wl_lock);  in ubi_wl_get_peb()
210 		spin_unlock(&ubi->wl_lock);  in ubi_wl_get_peb()
211 		up_read(&ubi->fm_eba_sem);  in ubi_wl_get_peb()
212 		ret = ubi_update_fastmap(ubi);  in ubi_wl_get_peb()
214 			ubi_msg(ubi, "Unable to write a new fastmap: %i", ret);  in ubi_wl_get_peb()
215 			down_read(&ubi->fm_eba_sem);  in ubi_wl_get_peb()
218 		down_read(&ubi->fm_eba_sem);  in ubi_wl_get_peb()
219 		spin_lock(&ubi->wl_lock);  in ubi_wl_get_peb()
223 		spin_unlock(&ubi->wl_lock);  in ubi_wl_get_peb()
226 			ubi_err(ubi, "Unable to get a free PEB from user WL pool");  in ubi_wl_get_peb()
230 		up_read(&ubi->fm_eba_sem);  in ubi_wl_get_peb()
231 		ret = produce_free_peb(ubi);  in ubi_wl_get_peb()
233 			down_read(&ubi->fm_eba_sem);  in ubi_wl_get_peb()
241 	prot_queue_add(ubi, ubi->lookuptbl[ret]);  in ubi_wl_get_peb()
242 	spin_unlock(&ubi->wl_lock);  in ubi_wl_get_peb()
251 static struct ubi_wl_entry *get_peb_for_wl(struct ubi_device *ubi)  in get_peb_for_wl()  argument
253 	struct ubi_fm_pool *pool = &ubi->fm_wl_pool;  in get_peb_for_wl()
256 	ubi_assert(rwsem_is_locked(&ubi->fm_eba_sem));  in get_peb_for_wl()
262 		if (!ubi->fm_work_scheduled) {  in get_peb_for_wl()
263 			ubi->fm_work_scheduled = 1;  in get_peb_for_wl()
264 			schedule_work(&ubi->fm_work);  in get_peb_for_wl()
270 	return ubi->lookuptbl[pnum];  in get_peb_for_wl()
277 int ubi_ensure_anchor_pebs(struct ubi_device *ubi)  in ubi_ensure_anchor_pebs()  argument
281 	spin_lock(&ubi->wl_lock);  in ubi_ensure_anchor_pebs()
282 	if (ubi->wl_scheduled) {  in ubi_ensure_anchor_pebs()
283 		spin_unlock(&ubi->wl_lock);  in ubi_ensure_anchor_pebs()
286 	ubi->wl_scheduled = 1;  in ubi_ensure_anchor_pebs()
287 	spin_unlock(&ubi->wl_lock);  in ubi_ensure_anchor_pebs()
291 		spin_lock(&ubi->wl_lock);  in ubi_ensure_anchor_pebs()
292 		ubi->wl_scheduled = 0;  in ubi_ensure_anchor_pebs()
293 		spin_unlock(&ubi->wl_lock);  in ubi_ensure_anchor_pebs()
299 	__schedule_ubi_work(ubi, wrk);  in ubi_ensure_anchor_pebs()
313 int ubi_wl_put_fm_peb(struct ubi_device *ubi, struct ubi_wl_entry *fm_e,  in ubi_wl_put_fm_peb()  argument
322 	ubi_assert(pnum < ubi->peb_count);  in ubi_wl_put_fm_peb()
324 	spin_lock(&ubi->wl_lock);  in ubi_wl_put_fm_peb()
325 	e = ubi->lookuptbl[pnum];  in ubi_wl_put_fm_peb()
334 		ubi->lookuptbl[pnum] = e;  in ubi_wl_put_fm_peb()
337 	spin_unlock(&ubi->wl_lock);  in ubi_wl_put_fm_peb()
340 	return schedule_erase(ubi, e, vol_id, lnum, torture, true);  in ubi_wl_put_fm_peb()
352 static void ubi_fastmap_close(struct ubi_device *ubi)  in ubi_fastmap_close()  argument
356 	return_unused_pool_pebs(ubi, &ubi->fm_pool);  in ubi_fastmap_close()
357 	return_unused_pool_pebs(ubi, &ubi->fm_wl_pool);  in ubi_fastmap_close()
359 	if (ubi->fm) {  in ubi_fastmap_close()
360 		for (i = 0; i < ubi->fm->used_blocks; i++)  in ubi_fastmap_close()
361 			kfree(ubi->fm->e[i]);  in ubi_fastmap_close()
363 	kfree(ubi->fm);  in ubi_fastmap_close()
374 static struct ubi_wl_entry *may_reserve_for_fm(struct ubi_device *ubi,  in may_reserve_for_fm()  argument
377 	if (e && !ubi->fm_disabled && !ubi->fm &&  in may_reserve_for_fm()