linux-imx

// SPDX-License-Identifier: GPL-2.0

#include <linux/err.h>

#include <linux/pci.h>

#include <linux/io.h>

#include <linux/gfp.h>

#include <linux/export.h>

#include <linux/of_address.h>

​

enum devm_ioremap_type {

    DEVM_IOREMAP = 0,

    DEVM_IOREMAP_NC,

    DEVM_IOREMAP_WC,

};

​

void devm_ioremap_release(struct device *dev, void *res)

{

    iounmap(*(void __iomem **)res);

}

​

static int devm_ioremap_match(struct device *dev, void *res, void *match_data)

{

    return *(void **)res == match_data;

}

​

static void __iomem *__devm_ioremap(struct device *dev, resource_size_t offset,

                    resource_size_t size,

                    enum devm_ioremap_type type)

{

    void __iomem **ptr, *addr = NULL;

​

    ptr = devres_alloc(devm_ioremap_release, sizeof(*ptr), GFP_KERNEL);

    if (!ptr)

        return NULL;

​

    switch (type) {

    case DEVM_IOREMAP:

        addr = ioremap(offset, size);

        break;

    case DEVM_IOREMAP_NC:

        addr = ioremap_nocache(offset, size);

        break;

    case DEVM_IOREMAP_WC:

        addr = ioremap_wc(offset, size);

        break;

    }

​

    if (addr) {

        *ptr = addr;

        devres_add(dev, ptr);

    } else

        devres_free(ptr);

​

    return addr;

}

​

/**

 * devm_ioremap - Managed ioremap()

 * @dev: Generic device to remap IO address for

 * @offset: Resource address to map

 * @size: Size of map

 *

 * Managed ioremap().  Map is automatically unmapped on driver detach.

 */