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 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	/* Reserve enough LEBs to store two fastmaps. */
14	*count += (ubi->fm_size / ubi->leb_size) * 2;
 
 
15	INIT_WORK(&ubi->fm_work, update_fastmap_work_fn);
16}
17static struct ubi_wl_entry *may_reserve_for_fm(struct ubi_device *ubi,
18					       struct ubi_wl_entry *e,
19					       struct rb_root *root);
20#else /* !CONFIG_MTD_UBI_FASTMAP */
21static struct ubi_wl_entry *get_peb_for_wl(struct ubi_device *ubi);
22static inline void ubi_fastmap_close(struct ubi_device *ubi) { }
23static inline void ubi_fastmap_init(struct ubi_device *ubi, int *count) { }
24static struct ubi_wl_entry *may_reserve_for_fm(struct ubi_device *ubi,
25					       struct ubi_wl_entry *e,
26					       struct rb_root *root) {
27	return e;
28}
29#endif /* CONFIG_MTD_UBI_FASTMAP */
30#endif /* UBI_WL_H */