linux-imx

// SPDX-License-Identifier: GPL-2.0

/*

 * Copyright 2017 NXP

 *

 * Dong Aisheng <aisheng.dong@nxp.com>

 */

​

#include <linux/clk.h>

#include <linux/clk-provider.h>

#include <linux/device.h>

#include <linux/export.h>

#include <linux/of.h>

#include <linux/slab.h>

​

static int __must_check of_clk_bulk_get(struct device_node *np, int num_clks,

                    struct clk_bulk_data *clks)

{

    int ret;

    int i;

​

    for (i = 0; i < num_clks; i++)

        clks[i].clk = NULL;

​

    for (i = 0; i < num_clks; i++) {

        clks[i].clk = of_clk_get(np, i);

        if (IS_ERR(clks[i].clk)) {

            ret = PTR_ERR(clks[i].clk);

            pr_err("%pOF: Failed to get clk index: %d ret: %d\n",

                   np, i, ret);

            clks[i].clk = NULL;

            goto err;

        }

    }

​

    return 0;

​

err:

    clk_bulk_put(i, clks);

​

    return ret;

}

​

static int __must_check of_clk_bulk_get_all(struct device_node *np,

                        struct clk_bulk_data **clks)

{

    struct clk_bulk_data *clk_bulk;

    int num_clks;

    int ret;

​

    num_clks = of_clk_get_parent_count(np);

    if (!num_clks)

        return 0;

​

    clk_bulk = kmalloc_array(num_clks, sizeof(*clk_bulk), GFP_KERNEL);

    if (!clk_bulk)

        return -ENOMEM;

​

    ret = of_clk_bulk_get(np, num_clks, clk_bulk);

    if (ret) {

        kfree(clk_bulk);

        return ret;

    }

​

    *clks = clk_bulk;

​

    return num_clks;

}

​

void clk_bulk_put(int num_clks, struct clk_bulk_data *clks)

{

    while (--num_clks >= 0) {

        clk_put(clks[num_clks].clk);

        clks[num_clks].clk = NULL;

    }

}

EXPORT_SYMBOL_GPL(clk_bulk_put);

​

int __must_check clk_bulk_get(struct device *dev, int num_clks,

                  struct clk_bulk_data *clks)

{

    int ret;

    int i;

​

    for (i = 0; i < num_clks; i++)

        clks[i].clk = NULL;

​

    for (i = 0; i < num_clks; i++) {

        clks[i].clk = clk_get(dev, clks[i].id);

        if (IS_ERR(clks[i].clk)) {

            ret = PTR_ERR(clks[i].clk);

            if (ret != -EPROBE_DEFER)

                dev_err(dev, "Failed to get clk '%s': %d\n",