1/* SPDX-License-Identifier: GPL-2.0 */
 2#ifndef UBI_WL_H
 3#define UBI_WL_H
 4#ifdef CONFIG_MTD_UBI_FASTMAP
 
 5static void update_fastmap_work_fn(struct work_struct *wrk);
 6static struct ubi_wl_entry *find_anchor_wl_entry(struct rb_root *root);
 7static struct ubi_wl_entry *get_peb_for_wl(struct ubi_device *ubi);
 8static struct ubi_wl_entry *next_peb_for_wl(struct ubi_device *ubi);
 9static bool need_wear_leveling(struct ubi_device *ubi);
10static void ubi_fastmap_close(struct ubi_device *ubi);
11static inline void ubi_fastmap_init(struct ubi_device *ubi, int *count)
12{
13	if (ubi->fm_disabled)
14		ubi->fm_pool_rsv_cnt = 0;
15	/* Reserve enough LEBs to store two fastmaps and to fill pools. */
16	*count += (ubi->fm_size / ubi->leb_size) * 2 + ubi->fm_pool_rsv_cnt;
17	INIT_WORK(&ubi->fm_work, update_fastmap_work_fn);
18}
19static struct ubi_wl_entry *may_reserve_for_fm(struct ubi_device *ubi,
20					       struct ubi_wl_entry *e,
21					       struct rb_root *root);
22#else /* !CONFIG_MTD_UBI_FASTMAP */
23static struct ubi_wl_entry *get_peb_for_wl(struct ubi_device *ubi);
24static inline void ubi_fastmap_close(struct ubi_device *ubi) { }
25static inline void ubi_fastmap_init(struct ubi_device *ubi, int *count) { }
26static struct ubi_wl_entry *may_reserve_for_fm(struct ubi_device *ubi,
27					       struct ubi_wl_entry *e,
28					       struct rb_root *root) {
29	return e;
30}
31#endif /* CONFIG_MTD_UBI_FASTMAP */
32#endif /* UBI_WL_H */

 1/* SPDX-License-Identifier: GPL-2.0 */
 2#ifndef UBI_WL_H
 3#define UBI_WL_H
 4#ifdef CONFIG_MTD_UBI_FASTMAP
 5static int anchor_pebs_available(struct rb_root *root);
 6static void update_fastmap_work_fn(struct work_struct *wrk);
 7static struct ubi_wl_entry *find_anchor_wl_entry(struct rb_root *root);
 8static struct ubi_wl_entry *get_peb_for_wl(struct ubi_device *ubi);
 
 
 9static void ubi_fastmap_close(struct ubi_device *ubi);
10static inline void ubi_fastmap_init(struct ubi_device *ubi, int *count)
11{
12	/* Reserve enough LEBs to store two fastmaps. */
13	*count += (ubi->fm_size / ubi->leb_size) * 2;
 
 
14	INIT_WORK(&ubi->fm_work, update_fastmap_work_fn);
15}
16static struct ubi_wl_entry *may_reserve_for_fm(struct ubi_device *ubi,
17					       struct ubi_wl_entry *e,
18					       struct rb_root *root);
19#else /* !CONFIG_MTD_UBI_FASTMAP */
20static struct ubi_wl_entry *get_peb_for_wl(struct ubi_device *ubi);
21static inline void ubi_fastmap_close(struct ubi_device *ubi) { }
22static inline void ubi_fastmap_init(struct ubi_device *ubi, int *count) { }
23static struct ubi_wl_entry *may_reserve_for_fm(struct ubi_device *ubi,
24					       struct ubi_wl_entry *e,
25					       struct rb_root *root) {
26	return e;
27}
28#endif /* CONFIG_MTD_UBI_FASTMAP */
29#endif /* UBI_WL_H */