// SPDX-License-Identifier: GPL-2.0
/*
 * CZ.NIC's Turris Omnia MCU system off and RTC wakeup driver
 *
 * This is not a true RTC driver (in the sense that it does not provide a
 * real-time clock), rather the MCU implements a wakeup from powered off state
 * at a specified time relative to MCU boot, and we expose this feature via RTC
 * alarm, so that it can be used via the rtcwake command, which is the standard
 * Linux command for this.
 *
 * 2024 by Marek Behún <kabel@kernel.org>
 */

#include <linux/crc32.h>
#include <linux/delay.h>
#include <linux/device.h>
#include <linux/err.h>
#include <linux/i2c.h>
#include <linux/kstrtox.h>
#include <linux/reboot.h>
#include <linux/rtc.h>
#include <linux/sysfs.h>
#include <linux/types.h>

#include <linux/turris-omnia-mcu-interface.h>
#include "turris-omnia-mcu.h"

static int omnia_get_uptime_wakeup(const struct i2c_client *client, u32 *uptime,
       u32 *wakeup)
{
 __le32 reply[2];
 int err;

 err = omnia_cmd_read(client, OMNIA_CMD_GET_UPTIME_AND_WAKEUP, reply,
        sizeof(reply));
 if (err)
  return err;

 if (uptime)
  *uptime = le32_to_cpu(reply[0]);

 if (wakeup)
  *wakeup = le32_to_cpu(reply[1]);

 return 0;
}

static int omnia_read_time(struct device *dev, struct rtc_time *tm)
{
 u32 uptime;
 int err;

 err = omnia_get_uptime_wakeup(to_i2c_client(dev), &uptime, NULL);
 if (err)
  return err;

 rtc_time64_to_tm(uptime, tm);

 return 0;
}

static int omnia_read_alarm(struct device *dev, struct rtc_wkalrm *alrm)
{
 struct i2c_client *client = to_i2c_client(dev);
 struct omnia_mcu *mcu = i2c_get_clientdata(client);
 u32 wakeup;
 int err;

 err = omnia_get_uptime_wakeup(client, NULL, &wakeup);
 if (err)
  return err;

 alrm->enabled = !!wakeup;
 rtc_time64_to_tm(wakeup ?: mcu->rtc_alarm, &alrm->time);

 return 0;
}

static int omnia_set_alarm(struct device *dev, struct rtc_wkalrm *alrm)
{
 struct i2c_client *client = to_i2c_client(dev);
 struct omnia_mcu *mcu = i2c_get_clientdata(client);

 mcu->rtc_alarm = rtc_tm_to_time64(&alrm->time);

 if (alrm->enabled)
  return omnia_cmd_write_u32(client, OMNIA_CMD_SET_WAKEUP,
        mcu->rtc_alarm);

 return 0;
}

static int omnia_alarm_irq_enable(struct device *dev, unsigned int enabled)
{
 struct i2c_client *client = to_i2c_client(dev);
 struct omnia_mcu *mcu = i2c_get_clientdata(client);

 return omnia_cmd_write_u32(client, OMNIA_CMD_SET_WAKEUP,
       enabled ? mcu->rtc_alarm : 0);
}

static const struct rtc_class_ops omnia_rtc_ops = {
 .read_time  = omnia_read_time,
 .read_alarm  = omnia_read_alarm,
 .set_alarm  = omnia_set_alarm,
 .alarm_irq_enable = omnia_alarm_irq_enable,
};

static int omnia_power_off(struct sys_off_data *data)
{
 struct omnia_mcu *mcu = data->cb_data;
 __be32 tmp;
 u8 cmd[9];
 u16 arg;
 int err;

 if (mcu->front_button_poweron)
  arg = OMNIA_CMD_POWER_OFF_POWERON_BUTTON;
 else
  arg = 0;

 cmd[0] = OMNIA_CMD_POWER_OFF;
 put_unaligned_le16(OMNIA_CMD_POWER_OFF_MAGIC, &cmd[1]);
 put_unaligned_le16(arg, &cmd[3]);

 /*
  * Although all values from and to MCU are passed in little-endian, the
  * MCU's CRC unit uses big-endian CRC32 polynomial (0x04c11db7), so we
  * need to use crc32_be() here.
 */
 tmp = cpu_to_be32(get_unaligned_le32(&cmd[1]));
 put_unaligned_le32(crc32_be(~0, (void *)&tmp, sizeof(tmp)), &cmd[5]);

 err = omnia_cmd_write(mcu->client, cmd, sizeof(cmd));
 if (err)
  dev_err(&mcu->client->dev,
   "Unable to send the poweroff command: %d\n", err);

 return NOTIFY_DONE;
}

static int omnia_restart(struct sys_off_data *data)
{
 struct omnia_mcu *mcu = data->cb_data;
 u8 cmd[3];
 int err;

 cmd[0] = OMNIA_CMD_GENERAL_CONTROL;

 if (reboot_mode == REBOOT_HARD)
  cmd[1] = cmd[2] = OMNIA_CTL_HARD_RST;
 else
  cmd[1] = cmd[2] = OMNIA_CTL_LIGHT_RST;

 err = omnia_cmd_write(mcu->client, cmd, sizeof(cmd));
 if (err)
  dev_err(&mcu->client->dev,
   "Unable to send the restart command: %d\n", err);

 /*
  * MCU needs a little bit to process the I2C command, otherwise it will
  * do a light reset based on SOC SYSRES_OUT pin.
 */
 mdelay(1);

 return NOTIFY_DONE;
}

static ssize_t front_button_poweron_show(struct device *dev,
      struct device_attribute *a, char *buf)
{
 struct omnia_mcu *mcu = dev_get_drvdata(dev);

 return sysfs_emit(buf, "%d\n", mcu->front_button_poweron);
}

static ssize_t front_button_poweron_store(struct device *dev,
       struct device_attribute *a,
       const char *buf, size_t count)
{
 struct omnia_mcu *mcu = dev_get_drvdata(dev);
 bool val;
 int err;

 err = kstrtobool(buf, &val);
 if (err)
  return err;

 mcu->front_button_poweron = val;

 return count;
}
static DEVICE_ATTR_RW(front_button_poweron);

static struct attribute *omnia_mcu_poweroff_attrs[] = {
 &dev_attr_front_button_poweron.attr,
 NULL
};

static umode_t poweroff_attrs_visible(struct kobject *kobj, struct attribute *a,
          int n)
{
 struct device *dev = kobj_to_dev(kobj);
 struct omnia_mcu *mcu = dev_get_drvdata(dev);

 if (mcu->features & OMNIA_FEAT_POWEROFF_WAKEUP)
  return a->mode;

 return 0;
}

const struct attribute_group omnia_mcu_poweroff_group = {
 .attrs = omnia_mcu_poweroff_attrs,
 .is_visible = poweroff_attrs_visible,
};

int omnia_mcu_register_sys_off_and_wakeup(struct omnia_mcu *mcu)
{
 struct device *dev = &mcu->client->dev;
 int err;

 /* MCU restart is always available */
 err = devm_register_sys_off_handler(dev, SYS_OFF_MODE_RESTART,
         SYS_OFF_PRIO_FIRMWARE,
         omnia_restart, mcu);
 if (err)
  return dev_err_probe(dev, err,
         "Cannot register system restart handler\n");

 /*
  * Poweroff and wakeup are available only if POWEROFF_WAKEUP feature is
  * present.
 */
 if (!(mcu->features & OMNIA_FEAT_POWEROFF_WAKEUP))
  return 0;

 err = devm_register_sys_off_handler(dev, SYS_OFF_MODE_POWER_OFF,
         SYS_OFF_PRIO_FIRMWARE,
         omnia_power_off, mcu);
 if (err)
  return dev_err_probe(dev, err,
         "Cannot register system power off handler\n");

 mcu->rtcdev = devm_rtc_allocate_device(dev);
 if (IS_ERR(mcu->rtcdev))
  return dev_err_probe(dev, PTR_ERR(mcu->rtcdev),
         "Cannot allocate RTC device\n");

 mcu->rtcdev->ops = &omnia_rtc_ops;
 mcu->rtcdev->range_max = U32_MAX;
 set_bit(RTC_FEATURE_ALARM_WAKEUP_ONLY, mcu->rtcdev->features);

 err = devm_rtc_register_device(mcu->rtcdev);
 if (err)
  return dev_err_probe(dev, err, "Cannot register RTC device\n");

 mcu->front_button_poweron = true;

 return 0;
}