// SPDX-License-Identifier: GPL-2.0
/*
 * System Control and Management Interface (SCMI) NXP BBM Protocol
 *
 * Copyright 2024 NXP
 */

#define pr_fmt(fmt) "SCMI Notifications BBM - " fmt

#include <linux/bits.h>
#include <linux/io.h>
#include <linux/module.h>
#include <linux/of.h>
#include <linux/platform_device.h>
#include <linux/scmi_protocol.h>
#include <linux/scmi_imx_protocol.h>

#include "../../protocols.h"
#include "../../notify.h"

#define SCMI_PROTOCOL_SUPPORTED_VERSION         0x10000

enum scmi_imx_bbm_protocol_cmd {
        IMX_BBM_GPR_SET = 0x3,
        IMX_BBM_GPR_GET = 0x4,
        IMX_BBM_RTC_ATTRIBUTES = 0x5,
        IMX_BBM_RTC_TIME_SET = 0x6,
        IMX_BBM_RTC_TIME_GET = 0x7,
        IMX_BBM_RTC_ALARM_SET = 0x8,
        IMX_BBM_BUTTON_GET = 0x9,
        IMX_BBM_RTC_NOTIFY = 0xA,
        IMX_BBM_BUTTON_NOTIFY = 0xB,
};

#define GET_RTCS_NR(x)  le32_get_bits((x), GENMASK(23, 16))
#define GET_GPRS_NR(x)  le32_get_bits((x), GENMASK(15, 0))

#define SCMI_IMX_BBM_NOTIFY_RTC_UPDATED         BIT(2)
#define SCMI_IMX_BBM_NOTIFY_RTC_ROLLOVER        BIT(1)
#define SCMI_IMX_BBM_NOTIFY_RTC_ALARM           BIT(0)

#define SCMI_IMX_BBM_RTC_ALARM_ENABLE_FLAG      BIT(0)

#define SCMI_IMX_BBM_NOTIFY_RTC_FLAG    \
        (SCMI_IMX_BBM_NOTIFY_RTC_UPDATED | SCMI_IMX_BBM_NOTIFY_RTC_ROLLOVER | \
         SCMI_IMX_BBM_NOTIFY_RTC_ALARM)

#define SCMI_IMX_BBM_EVENT_RTC_MASK             GENMASK(31, 24)

struct scmi_imx_bbm_info {
        int nr_rtc;
        int nr_gpr;
};

struct scmi_msg_imx_bbm_protocol_attributes {
        __le32 attributes;
};

struct scmi_imx_bbm_set_time {
        __le32 id;
        __le32 flags;
        __le32 value_low;
        __le32 value_high;
};

struct scmi_imx_bbm_get_time {
        __le32 id;
        __le32 flags;
};

struct scmi_imx_bbm_alarm_time {
        __le32 id;
        __le32 flags;
        __le32 value_low;
        __le32 value_high;
};

struct scmi_msg_imx_bbm_rtc_notify {
        __le32 rtc_id;
        __le32 flags;
};

struct scmi_msg_imx_bbm_button_notify {
        __le32 flags;
};

struct scmi_imx_bbm_notify_payld {
        __le32 flags;
};

static int scmi_imx_bbm_attributes_get(const struct scmi_protocol_handle *ph,
                                       struct scmi_imx_bbm_info *pi)
{
        int ret;
        struct scmi_xfer *t;
        struct scmi_msg_imx_bbm_protocol_attributes *attr;

        ret = ph->xops->xfer_get_init(ph, PROTOCOL_ATTRIBUTES, 0, sizeof(*attr), &t);
        if (ret)
                return ret;

        attr = t->rx.buf;

        ret = ph->xops->do_xfer(ph, t);
        if (!ret) {
                pi->nr_rtc = GET_RTCS_NR(attr->attributes);
                pi->nr_gpr = GET_GPRS_NR(attr->attributes);
        }

        ph->xops->xfer_put(ph, t);

        return ret;
}

static int scmi_imx_bbm_notify(const struct scmi_protocol_handle *ph,
                               u32 src_id, int message_id, bool enable)
{
        int ret;
        struct scmi_xfer *t;

        if (message_id == IMX_BBM_RTC_NOTIFY) {
                struct scmi_msg_imx_bbm_rtc_notify *rtc_notify;

                ret = ph->xops->xfer_get_init(ph, message_id,
                                              sizeof(*rtc_notify), 0, &t);
                if (ret)
                        return ret;

                rtc_notify = t->tx.buf;
                rtc_notify->rtc_id = cpu_to_le32(0);
                rtc_notify->flags =
                        cpu_to_le32(enable ? SCMI_IMX_BBM_NOTIFY_RTC_FLAG : 0);
        } else if (message_id == IMX_BBM_BUTTON_NOTIFY) {
                struct scmi_msg_imx_bbm_button_notify *button_notify;

                ret = ph->xops->xfer_get_init(ph, message_id,
                                              sizeof(*button_notify), 0, &t);
                if (ret)
                        return ret;

                button_notify = t->tx.buf;
                button_notify->flags = cpu_to_le32(enable ? 1 : 0);
        } else {
                return -EINVAL;
        }

        ret = ph->xops->do_xfer(ph, t);

        ph->xops->xfer_put(ph, t);
        return ret;
}

static enum scmi_imx_bbm_protocol_cmd evt_2_cmd[] = {
        IMX_BBM_RTC_NOTIFY,
        IMX_BBM_BUTTON_NOTIFY
};

static int scmi_imx_bbm_set_notify_enabled(const struct scmi_protocol_handle *ph,
                                           u8 evt_id, u32 src_id, bool enable)
{
        int ret, cmd_id;

        if (evt_id >= ARRAY_SIZE(evt_2_cmd))
                return -EINVAL;

        cmd_id = evt_2_cmd[evt_id];
        ret = scmi_imx_bbm_notify(ph, src_id, cmd_id, enable);
        if (ret)
                pr_debug("FAIL_ENABLED - evt[%X] dom[%d] - ret:%d\n",
                         evt_id, src_id, ret);

        return ret;
}

static void *scmi_imx_bbm_fill_custom_report(const struct scmi_protocol_handle *ph,
                                             u8 evt_id, ktime_t timestamp,
                                             const void *payld, size_t payld_sz,
                                             void *report, u32 *src_id)
{
        const struct scmi_imx_bbm_notify_payld *p = payld;
        struct scmi_imx_bbm_notif_report *r = report;

        if (sizeof(*p) != payld_sz)
                return NULL;

        if (evt_id == SCMI_EVENT_IMX_BBM_RTC) {
                r->is_rtc = true;
                r->is_button = false;
                r->timestamp = timestamp;
                r->rtc_id = le32_get_bits(p->flags, SCMI_IMX_BBM_EVENT_RTC_MASK);
                r->rtc_evt = le32_get_bits(p->flags, SCMI_IMX_BBM_NOTIFY_RTC_FLAG);
                dev_dbg(ph->dev, "RTC: %d evt: %x\n", r->rtc_id, r->rtc_evt);
                *src_id = r->rtc_evt;
        } else if (evt_id == SCMI_EVENT_IMX_BBM_BUTTON) {
                r->is_rtc = false;
                r->is_button = true;
                r->timestamp = timestamp;
                dev_dbg(ph->dev, "BBM Button\n");
                *src_id = 0;
        } else {
                WARN_ON_ONCE(1);
                return NULL;
        }

        return r;
}

static const struct scmi_event scmi_imx_bbm_events[] = {
        {
                .id = SCMI_EVENT_IMX_BBM_RTC,
                .max_payld_sz = sizeof(struct scmi_imx_bbm_notify_payld),
                .max_report_sz = sizeof(struct scmi_imx_bbm_notif_report),
        },
        {
                .id = SCMI_EVENT_IMX_BBM_BUTTON,
                .max_payld_sz = sizeof(struct scmi_imx_bbm_notify_payld),
                .max_report_sz = sizeof(struct scmi_imx_bbm_notif_report),
        },
};

static const struct scmi_event_ops scmi_imx_bbm_event_ops = {
        .set_notify_enabled = scmi_imx_bbm_set_notify_enabled,
        .fill_custom_report = scmi_imx_bbm_fill_custom_report,
};

static const struct scmi_protocol_events scmi_imx_bbm_protocol_events = {
        .queue_sz = SCMI_PROTO_QUEUE_SZ,
        .ops = &scmi_imx_bbm_event_ops,
        .evts = scmi_imx_bbm_events,
        .num_events = ARRAY_SIZE(scmi_imx_bbm_events),
        .num_sources = 1,
};

static int scmi_imx_bbm_rtc_time_set(const struct scmi_protocol_handle *ph,
                                     u32 rtc_id, u64 sec)
{
        struct scmi_imx_bbm_info *pi = ph->get_priv(ph);
        struct scmi_imx_bbm_set_time *cfg;
        struct scmi_xfer *t;
        int ret;

        if (rtc_id >= pi->nr_rtc)
                return -EINVAL;

        ret = ph->xops->xfer_get_init(ph, IMX_BBM_RTC_TIME_SET, sizeof(*cfg), 0, &t);
        if (ret)
                return ret;

        cfg = t->tx.buf;
        cfg->id = cpu_to_le32(rtc_id);
        cfg->flags = 0;
        cfg->value_low = cpu_to_le32(lower_32_bits(sec));
        cfg->value_high = cpu_to_le32(upper_32_bits(sec));

        ret = ph->xops->do_xfer(ph, t);

        ph->xops->xfer_put(ph, t);

        return ret;
}

static int scmi_imx_bbm_rtc_time_get(const struct scmi_protocol_handle *ph,
                                     u32 rtc_id, u64 *value)
{
        struct scmi_imx_bbm_info *pi = ph->get_priv(ph);
        struct scmi_imx_bbm_get_time *cfg;
        struct scmi_xfer *t;
        int ret;

        if (rtc_id >= pi->nr_rtc)
                return -EINVAL;

        ret = ph->xops->xfer_get_init(ph, IMX_BBM_RTC_TIME_GET, sizeof(*cfg),
                                      sizeof(u64), &t);
        if (ret)
                return ret;

        cfg = t->tx.buf;
        cfg->id = cpu_to_le32(rtc_id);
        cfg->flags = 0;

        ret = ph->xops->do_xfer(ph, t);
        if (!ret)
                *value = get_unaligned_le64(t->rx.buf);

        ph->xops->xfer_put(ph, t);

        return ret;
}

static int scmi_imx_bbm_rtc_alarm_set(const struct scmi_protocol_handle *ph,
                                      u32 rtc_id, bool enable, u64 sec)
{
        struct scmi_imx_bbm_info *pi = ph->get_priv(ph);
        struct scmi_imx_bbm_alarm_time *cfg;
        struct scmi_xfer *t;
        int ret;

        if (rtc_id >= pi->nr_rtc)
                return -EINVAL;

        ret = ph->xops->xfer_get_init(ph, IMX_BBM_RTC_ALARM_SET, sizeof(*cfg), 0, &t);
        if (ret)
                return ret;

        cfg = t->tx.buf;
        cfg->id = cpu_to_le32(rtc_id);
        cfg->flags = enable ?
                     cpu_to_le32(SCMI_IMX_BBM_RTC_ALARM_ENABLE_FLAG) : 0;
        cfg->value_low = cpu_to_le32(lower_32_bits(sec));
        cfg->value_high = cpu_to_le32(upper_32_bits(sec));

        ret = ph->xops->do_xfer(ph, t);

        ph->xops->xfer_put(ph, t);

        return ret;
}

static int scmi_imx_bbm_button_get(const struct scmi_protocol_handle *ph, u32 *state)
{
        struct scmi_xfer *t;
        int ret;

        ret = ph->xops->xfer_get_init(ph, IMX_BBM_BUTTON_GET, 0, sizeof(u32), &t);
        if (ret)
                return ret;

        ret = ph->xops->do_xfer(ph, t);
        if (!ret)
                *state = get_unaligned_le32(t->rx.buf);

        ph->xops->xfer_put(ph, t);

        return ret;
}

static const struct scmi_imx_bbm_proto_ops scmi_imx_bbm_proto_ops = {
        .rtc_time_get = scmi_imx_bbm_rtc_time_get,
        .rtc_time_set = scmi_imx_bbm_rtc_time_set,
        .rtc_alarm_set = scmi_imx_bbm_rtc_alarm_set,
        .button_get = scmi_imx_bbm_button_get,
};

static int scmi_imx_bbm_protocol_init(const struct scmi_protocol_handle *ph)
{
        int ret;
        struct scmi_imx_bbm_info *binfo;

        dev_info(ph->dev, "NXP SM BBM Version %d.%d\n",
                 PROTOCOL_REV_MAJOR(ph->version), PROTOCOL_REV_MINOR(ph->version));

        binfo = devm_kzalloc(ph->dev, sizeof(*binfo), GFP_KERNEL);
        if (!binfo)
                return -ENOMEM;

        ret = scmi_imx_bbm_attributes_get(ph, binfo);
        if (ret)
                return ret;

        return ph->set_priv(ph, binfo);
}

static const struct scmi_protocol scmi_imx_bbm = {
        .id = SCMI_PROTOCOL_IMX_BBM,
        .owner = THIS_MODULE,
        .instance_init = &scmi_imx_bbm_protocol_init,
        .ops = &scmi_imx_bbm_proto_ops,
        .events = &scmi_imx_bbm_protocol_events,
        .supported_version = SCMI_PROTOCOL_SUPPORTED_VERSION,
        .vendor_id = SCMI_IMX_VENDOR,
        .sub_vendor_id = SCMI_IMX_SUBVENDOR,
};
module_scmi_protocol(scmi_imx_bbm);

MODULE_ALIAS("scmi-protocol-" __stringify(SCMI_PROTOCOL_IMX_BBM) "-" SCMI_IMX_VENDOR);
MODULE_DESCRIPTION("i.MX SCMI BBM driver");
MODULE_LICENSE("GPL");