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*14072ee3SZhihao Cheng static struct ubi_wl_entry *next_peb_for_wl(struct ubi_device *ubi);
9*14072ee3SZhihao Cheng static bool need_wear_leveling(struct ubi_device *ubi);
1078d6d497SRichard Weinberger static void ubi_fastmap_close(struct ubi_device *ubi);
ubi_fastmap_init(struct ubi_device * ubi,int * count)11acfda79fSRichard Weinberger static inline void ubi_fastmap_init(struct ubi_device *ubi, int *count)
12acfda79fSRichard Weinberger {
13acfda79fSRichard Weinberger /* Reserve enough LEBs to store two fastmaps. */
14acfda79fSRichard Weinberger *count += (ubi->fm_size / ubi->leb_size) * 2;
15acfda79fSRichard Weinberger INIT_WORK(&ubi->fm_work, update_fastmap_work_fn);
16acfda79fSRichard Weinberger }
172f84c246SRichard Weinberger static struct ubi_wl_entry *may_reserve_for_fm(struct ubi_device *ubi,
182f84c246SRichard Weinberger struct ubi_wl_entry *e,
192f84c246SRichard Weinberger struct rb_root *root);
2078d6d497SRichard Weinberger #else /* !CONFIG_MTD_UBI_FASTMAP */
2178d6d497SRichard Weinberger static struct ubi_wl_entry *get_peb_for_wl(struct ubi_device *ubi);
ubi_fastmap_close(struct ubi_device * ubi)2278d6d497SRichard Weinberger static inline void ubi_fastmap_close(struct ubi_device *ubi) { }
ubi_fastmap_init(struct ubi_device * ubi,int * count)23acfda79fSRichard 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)242f84c246SRichard Weinberger static struct ubi_wl_entry *may_reserve_for_fm(struct ubi_device *ubi,
252f84c246SRichard Weinberger struct ubi_wl_entry *e,
262f84c246SRichard Weinberger struct rb_root *root) {
272f84c246SRichard Weinberger return e;
282f84c246SRichard Weinberger }
2978d6d497SRichard Weinberger #endif /* CONFIG_MTD_UBI_FASTMAP */
3078d6d497SRichard Weinberger #endif /* UBI_WL_H */
31