linux-imx

/*

 * Copyright (c) 2012 Linutronix GmbH

 * Copyright (c) 2014 sigma star gmbh

 * Author: Richard Weinberger <richard@nod.at>

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; version 2.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See

 * the GNU General Public License for more details.

 *

 */

​

/**

 * update_fastmap_work_fn - calls ubi_update_fastmap from a work queue

 * @wrk: the work description object

 */

static void update_fastmap_work_fn(struct work_struct *wrk)

{

    struct ubi_device *ubi = container_of(wrk, struct ubi_device, fm_work);

​

    ubi_update_fastmap(ubi);

    spin_lock(&ubi->wl_lock);

    ubi->fm_work_scheduled = 0;

    spin_unlock(&ubi->wl_lock);

}

​

/**

 * find_anchor_wl_entry - find wear-leveling entry to used as anchor PEB.

 * @root: the RB-tree where to look for

 */

static struct ubi_wl_entry *find_anchor_wl_entry(struct rb_root *root)

{

    struct rb_node *p;

    struct ubi_wl_entry *e, *victim = NULL;

    int max_ec = UBI_MAX_ERASECOUNTER;

​

    ubi_rb_for_each_entry(p, e, root, u.rb) {

        if (e->pnum < UBI_FM_MAX_START && e->ec < max_ec) {

            victim = e;

            max_ec = e->ec;

        }

    }

​

    return victim;

}

​

/**

 * return_unused_pool_pebs - returns unused PEB to the free tree.

 * @ubi: UBI device description object

 * @pool: fastmap pool description object

 */

static void return_unused_pool_pebs(struct ubi_device *ubi,

                    struct ubi_fm_pool *pool)

{

    int i;

    struct ubi_wl_entry *e;

​

    for (i = pool->used; i < pool->size; i++) {

        e = ubi->lookuptbl[pool->pebs[i]];

        wl_tree_add(e, &ubi->free);

        ubi->free_count++;

    }

}

​

static int anchor_pebs_available(struct rb_root *root)

{

    struct rb_node *p;

    struct ubi_wl_entry *e;

​

    ubi_rb_for_each_entry(p, e, root, u.rb)

        if (e->pnum < UBI_FM_MAX_START)

            return 1;

​

    return 0;

}

​

/**

 * ubi_wl_get_fm_peb - find a physical erase block with a given maximal number.

 * @ubi: UBI device description object

 * @anchor: This PEB will be used as anchor PEB by fastmap

 *

 * The function returns a physical erase block with a given maximal number

 * and removes it from the wl subsystem.

 * Must be called with wl_lock held!

 */

struct ubi_wl_entry *ubi_wl_get_fm_peb(struct ubi_device *ubi, int anchor)

{

    struct ubi_wl_entry *e = NULL;