17 void update_fastmap_work_fn(struct ubi_device *ubi)  in update_fastmap_work_fn()
21 	struct ubi_device *ubi = container_of(wrk, struct ubi_device, fm_work);  in update_fastmap_work_fn()  local
24 	ubi_update_fastmap(ubi);  in update_fastmap_work_fn()
25 	spin_lock(&ubi->wl_lock);  in update_fastmap_work_fn()
26 	ubi->fm_work_scheduled = 0;  in update_fastmap_work_fn()
27 	spin_unlock(&ubi->wl_lock);  in update_fastmap_work_fn()
55 static void return_unused_pool_pebs(struct ubi_device *ubi,  in return_unused_pool_pebs()  argument
62 		e = ubi->lookuptbl[pool->pebs[i]];  in return_unused_pool_pebs()
63 		wl_tree_add(e, &ubi->free);  in return_unused_pool_pebs()
64 		ubi->free_count++;  in return_unused_pool_pebs()
89 struct ubi_wl_entry *ubi_wl_get_fm_peb(struct ubi_device *ubi, int anchor)  in ubi_wl_get_fm_peb()  argument
93 	if (!ubi->free.rb_node || (ubi->free_count - ubi->beb_rsvd_pebs < 1))  in ubi_wl_get_fm_peb()
97 		e = find_anchor_wl_entry(&ubi->free);  in ubi_wl_get_fm_peb()
99 		e = find_mean_wl_entry(ubi, &ubi->free);  in ubi_wl_get_fm_peb()
104 	self_check_in_wl_tree(ubi, e, &ubi->free);  in ubi_wl_get_fm_peb()
108 	rb_erase(&e->u.rb, &ubi->free);  in ubi_wl_get_fm_peb()
109 	ubi->free_count--;  in ubi_wl_get_fm_peb()
118 void ubi_refill_pools(struct ubi_device *ubi)  in ubi_refill_pools()  argument
120 	struct ubi_fm_pool *wl_pool = &ubi->fm_wl_pool;  in ubi_refill_pools()
121 	struct ubi_fm_pool *pool = &ubi->fm_pool;  in ubi_refill_pools()
125 	spin_lock(&ubi->wl_lock);  in ubi_refill_pools()
127 	return_unused_pool_pebs(ubi, wl_pool);  in ubi_refill_pools()
128 	return_unused_pool_pebs(ubi, pool);  in ubi_refill_pools()
136 			if (!ubi->free.rb_node)  in ubi_refill_pools()
139 			e = wl_get_wle(ubi);  in ubi_refill_pools()
149 			if (!ubi->free.rb_node ||  in ubi_refill_pools()
150 			   (ubi->free_count - ubi->beb_rsvd_pebs < 5))  in ubi_refill_pools()
153 			e = find_wl_entry(ubi, &ubi->free, WL_FREE_MAX_DIFF);  in ubi_refill_pools()
154 			self_check_in_wl_tree(ubi, e, &ubi->free);  in ubi_refill_pools()
155 			rb_erase(&e->u.rb, &ubi->free);  in ubi_refill_pools()
156 			ubi->free_count--;  in ubi_refill_pools()
170 	spin_unlock(&ubi->wl_lock);  in ubi_refill_pools()
182 static int produce_free_peb(struct ubi_device *ubi)  in produce_free_peb()  argument
186 	while (!ubi->free.rb_node && ubi->works_count) {  in produce_free_peb()
188 		err = do_work(ubi);  in produce_free_peb()
205 int ubi_wl_get_peb(struct ubi_device *ubi)  in ubi_wl_get_peb()  argument
208 	struct ubi_fm_pool *pool = &ubi->fm_pool;  in ubi_wl_get_peb()
209 	struct ubi_fm_pool *wl_pool = &ubi->fm_wl_pool;  in ubi_wl_get_peb()
212 	down_read(&ubi->fm_eba_sem);  in ubi_wl_get_peb()
213 	spin_lock(&ubi->wl_lock);  in ubi_wl_get_peb()
218 		spin_unlock(&ubi->wl_lock);  in ubi_wl_get_peb()
219 		up_read(&ubi->fm_eba_sem);  in ubi_wl_get_peb()
220 		ret = ubi_update_fastmap(ubi);  in ubi_wl_get_peb()
222 			ubi_msg(ubi, "Unable to write a new fastmap: %i", ret);  in ubi_wl_get_peb()
223 			down_read(&ubi->fm_eba_sem);  in ubi_wl_get_peb()
226 		down_read(&ubi->fm_eba_sem);  in ubi_wl_get_peb()
227 		spin_lock(&ubi->wl_lock);  in ubi_wl_get_peb()
231 		spin_unlock(&ubi->wl_lock);  in ubi_wl_get_peb()
233 			ubi_err(ubi, "Unable to get a free PEB from user WL pool");  in ubi_wl_get_peb()
238 		up_read(&ubi->fm_eba_sem);  in ubi_wl_get_peb()
239 		ret = produce_free_peb(ubi);  in ubi_wl_get_peb()
241 			down_read(&ubi->fm_eba_sem);  in ubi_wl_get_peb()
249 	prot_queue_add(ubi, ubi->lookuptbl[ret]);  in ubi_wl_get_peb()
250 	spin_unlock(&ubi->wl_lock);  in ubi_wl_get_peb()
259 static struct ubi_wl_entry *get_peb_for_wl(struct ubi_device *ubi)  in get_peb_for_wl()  argument
261 	struct ubi_fm_pool *pool = &ubi->fm_wl_pool;  in get_peb_for_wl()
269 		if (!ubi->fm_work_scheduled) {  in get_peb_for_wl()
270 			ubi->fm_work_scheduled = 1;  in get_peb_for_wl()
271 			schedule_work(&ubi->fm_work);  in get_peb_for_wl()
278 		update_fastmap_work_fn(ubi);  in get_peb_for_wl()
283 	return ubi->lookuptbl[pnum];  in get_peb_for_wl()
290 int ubi_ensure_anchor_pebs(struct ubi_device *ubi)  in ubi_ensure_anchor_pebs()  argument
294 	spin_lock(&ubi->wl_lock);  in ubi_ensure_anchor_pebs()
295 	if (ubi->wl_scheduled) {  in ubi_ensure_anchor_pebs()
296 		spin_unlock(&ubi->wl_lock);  in ubi_ensure_anchor_pebs()
299 	ubi->wl_scheduled = 1;  in ubi_ensure_anchor_pebs()
300 	spin_unlock(&ubi->wl_lock);  in ubi_ensure_anchor_pebs()
304 		spin_lock(&ubi->wl_lock);  in ubi_ensure_anchor_pebs()
305 		ubi->wl_scheduled = 0;  in ubi_ensure_anchor_pebs()
306 		spin_unlock(&ubi->wl_lock);  in ubi_ensure_anchor_pebs()
312 	schedule_ubi_work(ubi, wrk);  in ubi_ensure_anchor_pebs()
326 int ubi_wl_put_fm_peb(struct ubi_device *ubi, struct ubi_wl_entry *fm_e,  in ubi_wl_put_fm_peb()  argument
335 	ubi_assert(pnum < ubi->peb_count);  in ubi_wl_put_fm_peb()
337 	spin_lock(&ubi->wl_lock);  in ubi_wl_put_fm_peb()
338 	e = ubi->lookuptbl[pnum];  in ubi_wl_put_fm_peb()
347 		ubi->lookuptbl[pnum] = e;  in ubi_wl_put_fm_peb()
350 	spin_unlock(&ubi->wl_lock);  in ubi_wl_put_fm_peb()
353 	return schedule_erase(ubi, e, vol_id, lnum, torture);  in ubi_wl_put_fm_peb()
365 static void ubi_fastmap_close(struct ubi_device *ubi)  in ubi_fastmap_close()  argument
369 	return_unused_pool_pebs(ubi, &ubi->fm_pool);  in ubi_fastmap_close()
370 	return_unused_pool_pebs(ubi, &ubi->fm_wl_pool);  in ubi_fastmap_close()
372 	if (ubi->fm) {  in ubi_fastmap_close()
373 		for (i = 0; i < ubi->fm->used_blocks; i++)  in ubi_fastmap_close()
374 			kfree(ubi->fm->e[i]);  in ubi_fastmap_close()
376 	kfree(ubi->fm);  in ubi_fastmap_close()
387 static struct ubi_wl_entry *may_reserve_for_fm(struct ubi_device *ubi,  in may_reserve_for_fm()  argument
390 	if (e && !ubi->fm_disabled && !ubi->fm &&  in may_reserve_for_fm()