// SPDX-License-Identifier: GPL-2.0
/* Copyright (C) 2012-2019 ARM Limited (or its affiliates). */

#include <linux/kernel.h>
#include <linux/interrupt.h>
#include <linux/pm_runtime.h>
#include "cc_driver.h"
#include "cc_buffer_mgr.h"
#include "cc_request_mgr.h"
#include "cc_sram_mgr.h"
#include "cc_hash.h"
#include "cc_pm.h"
#include "cc_fips.h"

#define POWER_DOWN_ENABLE 0x01
#define POWER_DOWN_DISABLE 0x00

static int cc_pm_suspend(struct device *dev)
{
        struct cc_drvdata *drvdata = dev_get_drvdata(dev);

        dev_dbg(dev, "set HOST_POWER_DOWN_EN\n");
        fini_cc_regs(drvdata);
        cc_iowrite(drvdata, CC_REG(HOST_POWER_DOWN_EN), POWER_DOWN_ENABLE);
        clk_disable_unprepare(drvdata->clk);
        return 0;
}

static int cc_pm_resume(struct device *dev)
{
        int rc;
        struct cc_drvdata *drvdata = dev_get_drvdata(dev);

        dev_dbg(dev, "unset HOST_POWER_DOWN_EN\n");
        /* Enables the device source clk */
        rc = clk_prepare_enable(drvdata->clk);
        if (rc) {
                dev_err(dev, "failed getting clock back on. We're toast.\n");
                return rc;
        }
        /* wait for Cryptocell reset completion */
        if (!cc_wait_for_reset_completion(drvdata)) {
                dev_err(dev, "Cryptocell reset not completed");
                clk_disable_unprepare(drvdata->clk);
                return -EBUSY;
        }

        cc_iowrite(drvdata, CC_REG(HOST_POWER_DOWN_EN), POWER_DOWN_DISABLE);
        rc = init_cc_regs(drvdata);
        if (rc) {
                dev_err(dev, "init_cc_regs (%x)\n", rc);
                clk_disable_unprepare(drvdata->clk);
                return rc;
        }
        /* check if tee fips error occurred during power down */
        cc_tee_handle_fips_error(drvdata);

        cc_init_hash_sram(drvdata);

        return 0;
}

const struct dev_pm_ops ccree_pm = {
        SET_RUNTIME_PM_OPS(cc_pm_suspend, cc_pm_resume, NULL)
};

int cc_pm_get(struct device *dev)
{
        int rc = pm_runtime_get_sync(dev);
        if (rc < 0) {
                pm_runtime_put_noidle(dev);
                return rc;
        }

        return 0;
}

void cc_pm_put_suspend(struct device *dev)
{
        pm_runtime_put_autosuspend(dev);
}