1b2441318SGreg Kroah-Hartman /* SPDX-License-Identifier: GPL-2.0 */
278d6d497SRichard Weinberger #ifndef UBI_WL_H
378d6d497SRichard Weinberger #define UBI_WL_H
478d6d497SRichard Weinberger #ifdef CONFIG_MTD_UBI_FASTMAP
578d6d497SRichard Weinberger static void update_fastmap_work_fn(struct work_struct *wrk);
678d6d497SRichard Weinberger static struct ubi_wl_entry *find_anchor_wl_entry(struct rb_root *root);
778d6d497SRichard Weinberger static struct ubi_wl_entry *get_peb_for_wl(struct ubi_device *ubi);
8*7e66d368SZhihao Cheng static struct ubi_wl_entry *next_peb_for_wl(struct ubi_device *ubi,
9*7e66d368SZhihao Cheng bool need_fill);
1014072ee3SZhihao Cheng static bool need_wear_leveling(struct ubi_device *ubi);
1178d6d497SRichard Weinberger static void ubi_fastmap_close(struct ubi_device *ubi);
ubi_fastmap_init(struct ubi_device * ubi,int * count)12acfda79fSRichard Weinberger static inline void ubi_fastmap_init(struct ubi_device *ubi, int *count)
13acfda79fSRichard Weinberger {
14acfda79fSRichard Weinberger /* Reserve enough LEBs to store two fastmaps. */
15acfda79fSRichard Weinberger *count += (ubi->fm_size / ubi->leb_size) * 2;
16acfda79fSRichard Weinberger INIT_WORK(&ubi->fm_work, update_fastmap_work_fn);
17acfda79fSRichard Weinberger }
182f84c246SRichard Weinberger static struct ubi_wl_entry *may_reserve_for_fm(struct ubi_device *ubi,
192f84c246SRichard Weinberger struct ubi_wl_entry *e,
202f84c246SRichard Weinberger struct rb_root *root);
2178d6d497SRichard Weinberger #else /* !CONFIG_MTD_UBI_FASTMAP */
2278d6d497SRichard Weinberger static struct ubi_wl_entry *get_peb_for_wl(struct ubi_device *ubi);
ubi_fastmap_close(struct ubi_device * ubi)2378d6d497SRichard Weinberger static inline void ubi_fastmap_close(struct ubi_device *ubi) { }
ubi_fastmap_init(struct ubi_device * ubi,int * count)24acfda79fSRichard Weinberger static inline void ubi_fastmap_init(struct ubi_device *ubi, int *count) { }
may_reserve_for_fm(struct ubi_device * ubi,struct ubi_wl_entry * e,struct rb_root * root)252f84c246SRichard Weinberger static struct ubi_wl_entry *may_reserve_for_fm(struct ubi_device *ubi,
262f84c246SRichard Weinberger struct ubi_wl_entry *e,
272f84c246SRichard Weinberger struct rb_root *root) {
282f84c246SRichard Weinberger return e;
292f84c246SRichard Weinberger }
3078d6d497SRichard Weinberger #endif /* CONFIG_MTD_UBI_FASTMAP */
3178d6d497SRichard Weinberger #endif /* UBI_WL_H */
32