linux-imx

// SPDX-License-Identifier: GPL-2.0

#include <linux/vmalloc.h>

#include "null_blk.h"

​

/* zone_size in MBs to sectors. */

#define ZONE_SIZE_SHIFT     11

​

static inline unsigned int null_zone_no(struct nullb_device *dev, sector_t sect)

{

    return sect >> ilog2(dev->zone_size_sects);

}

​

int null_zone_init(struct nullb_device *dev)

{

    sector_t dev_size = (sector_t)dev->size * 1024 * 1024;

    sector_t sector = 0;

    unsigned int i;

​

    if (!is_power_of_2(dev->zone_size)) {

        pr_err("null_blk: zone_size must be power-of-two\n");

        return -EINVAL;

    }

​

    dev->zone_size_sects = dev->zone_size << ZONE_SIZE_SHIFT;

    dev->nr_zones = dev_size >>

                (SECTOR_SHIFT + ilog2(dev->zone_size_sects));

    dev->zones = kvmalloc_array(dev->nr_zones, sizeof(struct blk_zone),

            GFP_KERNEL | __GFP_ZERO);

    if (!dev->zones)

        return -ENOMEM;

​

    if (dev->zone_nr_conv >= dev->nr_zones) {

        dev->zone_nr_conv = dev->nr_zones - 1;

        pr_info("null_blk: changed the number of conventional zones to %u",

            dev->zone_nr_conv);

    }

​

    for (i = 0; i <  dev->zone_nr_conv; i++) {

        struct blk_zone *zone = &dev->zones[i];

​

        zone->start = sector;

        zone->len = dev->zone_size_sects;

        zone->wp = zone->start + zone->len;

        zone->type = BLK_ZONE_TYPE_CONVENTIONAL;

        zone->cond = BLK_ZONE_COND_NOT_WP;

​

        sector += dev->zone_size_sects;

    }

​

    for (i = dev->zone_nr_conv; i < dev->nr_zones; i++) {

        struct blk_zone *zone = &dev->zones[i];

​

        zone->start = zone->wp = sector;

        zone->len = dev->zone_size_sects;

        zone->type = BLK_ZONE_TYPE_SEQWRITE_REQ;

        zone->cond = BLK_ZONE_COND_EMPTY;

​

        sector += dev->zone_size_sects;

    }

​

    return 0;

}

​