// SPDX-License-Identifier: GPL-2.0
/*
 * Miscellaneous Mac68K-specific stuff
 */

#include <linux/types.h>
#include <linux/errno.h>
#include <linux/kernel.h>
#include <linux/delay.h>
#include <linux/sched.h>
#include <linux/time.h>
#include <linux/rtc.h>
#include <linux/mm.h>

#include <linux/adb.h>
#include <linux/cuda.h>
#include <linux/pmu.h>

#include <linux/uaccess.h>
#include <asm/io.h>
#include <asm/setup.h>
#include <asm/macintosh.h>
#include <asm/mac_via.h>
#include <asm/mac_oss.h>

#include <asm/machdep.h>

#include "mac.h"

/*
 * Offset between Unix time (1970-based) and Mac time (1904-based). Cuda and PMU
 * times wrap in 2040. If we need to handle later times, the read_time functions
 * need to be changed to interpret wrapped times as post-2040.
 */

#define RTC_OFFSET 2082844800

static void (*rom_reset)(void);

#if IS_ENABLED(CONFIG_NVRAM)
#ifdef CONFIG_ADB_CUDA
static unsigned char cuda_pram_read_byte(int offset)
{
 struct adb_request req;

 if (cuda_request(&req, NULL, 4, CUDA_PACKET, CUDA_GET_PRAM,
    (offset >> 8) & 0xFF, offset & 0xFF) < 0)
  return 0;
 while (!req.complete)
  cuda_poll();
 return req.reply[3];
}

static void cuda_pram_write_byte(unsigned char data, int offset)
{
 struct adb_request req;

 if (cuda_request(&req, NULL, 5, CUDA_PACKET, CUDA_SET_PRAM,
    (offset >> 8) & 0xFF, offset & 0xFF, data) < 0)
  return;
 while (!req.complete)
  cuda_poll();
}
#endif /* CONFIG_ADB_CUDA */

#ifdef CONFIG_ADB_PMU
static unsigned char pmu_pram_read_byte(int offset)
{
 struct adb_request req;

 if (pmu_request(&req, NULL, 3, PMU_READ_XPRAM,
                 offset & 0xFF, 1) < 0)
  return 0;
 pmu_wait_complete(&req);

 return req.reply[0];
}

static void pmu_pram_write_byte(unsigned char data, int offset)
{
 struct adb_request req;

 if (pmu_request(&req, NULL, 4, PMU_WRITE_XPRAM,
                 offset & 0xFF, 1, data) < 0)
  return;
 pmu_wait_complete(&req);
}
#endif /* CONFIG_ADB_PMU */
#endif /* CONFIG_NVRAM */

/*
 * VIA PRAM/RTC access routines
 *
 * Must be called with interrupts disabled and
 * the RTC should be enabled.
 */

static __u8 via_rtc_recv(void)
{
 int i, reg;
 __u8 data;

 reg = via1[vBufB] & ~VIA1B_vRTCClk;

 /* Set the RTC data line to be an input. */

 via1[vDirB] &= ~VIA1B_vRTCData;

 /* The bits of the byte come out in MSB order */

 data = 0;
 for (i = 0 ; i < 8 ; i++) {
  via1[vBufB] = reg;
  via1[vBufB] = reg | VIA1B_vRTCClk;
  data = (data << 1) | (via1[vBufB] & VIA1B_vRTCData);
 }

 /* Return RTC data line to output state */

 via1[vDirB] |= VIA1B_vRTCData;

 return data;
}

static void via_rtc_send(__u8 data)
{
 int i, reg, bit;

 reg = via1[vBufB] & ~(VIA1B_vRTCClk | VIA1B_vRTCData);

 /* The bits of the byte go into the RTC in MSB order */

 for (i = 0 ; i < 8 ; i++) {
  bit = data & 0x80? 1 : 0;
  data <<= 1;
  via1[vBufB] = reg | bit;
  via1[vBufB] = reg | bit | VIA1B_vRTCClk;
 }
}

/*
 * These values can be found in Inside Macintosh vol. III ch. 2
 * which has a description of the RTC chip in the original Mac.
 */

#define RTC_FLG_READ            BIT(7)
#define RTC_FLG_WRITE_PROTECT   BIT(7)
#define RTC_CMD_READ(r)         (RTC_FLG_READ | (r << 2))
#define RTC_CMD_WRITE(r)        (r << 2)
#define RTC_REG_SECONDS_0       0
#define RTC_REG_SECONDS_1       1
#define RTC_REG_SECONDS_2       2
#define RTC_REG_SECONDS_3       3
#define RTC_REG_WRITE_PROTECT   13

/*
 * Inside Mac has no information about two-byte RTC commands but
 * the MAME/MESS source code has the essentials.
 */

#define RTC_REG_XPRAM           14
#define RTC_CMD_XPRAM_READ      (RTC_CMD_READ(RTC_REG_XPRAM) << 8)
#define RTC_CMD_XPRAM_WRITE     (RTC_CMD_WRITE(RTC_REG_XPRAM) << 8)
#define RTC_CMD_XPRAM_ARG(a)    (((a & 0xE0) << 3) | ((a & 0x1F) << 2))

/*
 * Execute a VIA PRAM/RTC command. For read commands
 * data should point to a one-byte buffer for the
 * resulting data. For write commands it should point
 * to the data byte to for the command.
 *
 * This function disables all interrupts while running.
 */

static void via_rtc_command(int command, __u8 *data)
{
 unsigned long flags;
 int is_read;

 local_irq_save(flags);

 /* The least significant bits must be 0b01 according to Inside Mac */

 command = (command & ~3) | 1;

 /* Enable the RTC and make sure the strobe line is high */

 via1[vBufB] = (via1[vBufB] | VIA1B_vRTCClk) & ~VIA1B_vRTCEnb;

 if (command & 0xFF00) {  /* extended (two-byte) command */
  via_rtc_send((command & 0xFF00) >> 8);
  via_rtc_send(command & 0xFF);
  is_read = command & (RTC_FLG_READ << 8);
 } else {   /* one-byte command */
  via_rtc_send(command);
  is_read = command & RTC_FLG_READ;
 }
 if (is_read) {
  *data = via_rtc_recv();
 } else {
  via_rtc_send(*data);
 }

 /* All done, disable the RTC */

 via1[vBufB] |= VIA1B_vRTCEnb;

 local_irq_restore(flags);
}

#if IS_ENABLED(CONFIG_NVRAM)
static unsigned char via_pram_read_byte(int offset)
{
 unsigned char temp;

 via_rtc_command(RTC_CMD_XPRAM_READ | RTC_CMD_XPRAM_ARG(offset), &temp);

 return temp;
}

static void via_pram_write_byte(unsigned char data, int offset)
{
 unsigned char temp;

 temp = 0x55;
 via_rtc_command(RTC_CMD_WRITE(RTC_REG_WRITE_PROTECT), &temp);

 temp = data;
 via_rtc_command(RTC_CMD_XPRAM_WRITE | RTC_CMD_XPRAM_ARG(offset), &temp);

 temp = 0x55 | RTC_FLG_WRITE_PROTECT;
 via_rtc_command(RTC_CMD_WRITE(RTC_REG_WRITE_PROTECT), &temp);
}
#endif /* CONFIG_NVRAM */

/*
 * Return the current time in seconds since January 1, 1904.
 *
 * This only works on machines with the VIA-based PRAM/RTC, which
 * is basically any machine with Mac II-style ADB.
 */

static time64_t via_read_time(void)
{
 union {
  __u8 cdata[4];
  __u32 idata;
 } result, last_result;
 int count = 1;

 via_rtc_command(RTC_CMD_READ(RTC_REG_SECONDS_0), &last_result.cdata[3]);
 via_rtc_command(RTC_CMD_READ(RTC_REG_SECONDS_1), &last_result.cdata[2]);
 via_rtc_command(RTC_CMD_READ(RTC_REG_SECONDS_2), &last_result.cdata[1]);
 via_rtc_command(RTC_CMD_READ(RTC_REG_SECONDS_3), &last_result.cdata[0]);

 /*
  * The NetBSD guys say to loop until you get the same reading
  * twice in a row.
 */

 while (1) {
  via_rtc_command(RTC_CMD_READ(RTC_REG_SECONDS_0),
                  &result.cdata[3]);
  via_rtc_command(RTC_CMD_READ(RTC_REG_SECONDS_1),
                  &result.cdata[2]);
  via_rtc_command(RTC_CMD_READ(RTC_REG_SECONDS_2),
                  &result.cdata[1]);
  via_rtc_command(RTC_CMD_READ(RTC_REG_SECONDS_3),
                  &result.cdata[0]);

  if (result.idata == last_result.idata)
   return (time64_t)result.idata - RTC_OFFSET;

  if (++count > 10)
   break;

  last_result.idata = result.idata;
 }

 pr_err("%s: failed to read a stable value; got 0x%08x then 0x%08x\n",
        __func__, last_result.idata, result.idata);

 return 0;
}

/*
 * Set the current time to a number of seconds since January 1, 1904.
 *
 * This only works on machines with the VIA-based PRAM/RTC, which
 * is basically any machine with Mac II-style ADB.
 */

static void via_set_rtc_time(struct rtc_time *tm)
{
 union {
  __u8 cdata[4];
  __u32 idata;
 } data;
 __u8 temp;
 time64_t time;

 time = mktime64(tm->tm_year + 1900, tm->tm_mon + 1, tm->tm_mday,
                 tm->tm_hour, tm->tm_min, tm->tm_sec);

 /* Clear the write protect bit */

 temp = 0x55;
 via_rtc_command(RTC_CMD_WRITE(RTC_REG_WRITE_PROTECT), &temp);

 data.idata = lower_32_bits(time + RTC_OFFSET);
 via_rtc_command(RTC_CMD_WRITE(RTC_REG_SECONDS_0), &data.cdata[3]);
 via_rtc_command(RTC_CMD_WRITE(RTC_REG_SECONDS_1), &data.cdata[2]);
 via_rtc_command(RTC_CMD_WRITE(RTC_REG_SECONDS_2), &data.cdata[1]);
 via_rtc_command(RTC_CMD_WRITE(RTC_REG_SECONDS_3), &data.cdata[0]);

 /* Set the write protect bit */

 temp = 0x55 | RTC_FLG_WRITE_PROTECT;
 via_rtc_command(RTC_CMD_WRITE(RTC_REG_WRITE_PROTECT), &temp);
}

static void via_shutdown(void)
{
 if (rbv_present) {
  via2[rBufB] &= ~0x04;
 } else {
  /* Direction of vDirB is output */
  via2[vDirB] |= 0x04;
  /* Send a value of 0 on that line */
  via2[vBufB] &= ~0x04;
  mdelay(1000);
 }
}

static void oss_shutdown(void)
{
 oss->rom_ctrl = OSS_POWEROFF;
}

#ifdef CONFIG_ADB_CUDA
static void cuda_restart(void)
{
 struct adb_request req;

 if (cuda_request(&req, NULL, 2, CUDA_PACKET, CUDA_RESET_SYSTEM) < 0)
  return;
 while (!req.complete)
  cuda_poll();
}

static void cuda_shutdown(void)
{
 struct adb_request req;

 if (cuda_request(&req, NULL, 2, CUDA_PACKET, CUDA_POWERDOWN) < 0)
  return;

 /* Avoid infinite polling loop when PSU is not under Cuda control */
 switch (macintosh_config->ident) {
 case MAC_MODEL_C660:
 case MAC_MODEL_Q605:
 case MAC_MODEL_Q605_ACC:
 case MAC_MODEL_P475:
 case MAC_MODEL_P475F:
  return;
 }

 while (!req.complete)
  cuda_poll();
}
#endif /* CONFIG_ADB_CUDA */

/*
 *-------------------------------------------------------------------
 * Below this point are the generic routines; they'll dispatch to the
 * correct routine for the hardware on which we're running.
 *-------------------------------------------------------------------
 */

#if IS_ENABLED(CONFIG_NVRAM)
unsigned char mac_pram_read_byte(int addr)
{
 switch (macintosh_config->adb_type) {
 case MAC_ADB_IOP:
 case MAC_ADB_II:
 case MAC_ADB_PB1:
  return via_pram_read_byte(addr);
#ifdef CONFIG_ADB_CUDA
 case MAC_ADB_EGRET:
 case MAC_ADB_CUDA:
  return cuda_pram_read_byte(addr);
#endif
#ifdef CONFIG_ADB_PMU
 case MAC_ADB_PB2:
  return pmu_pram_read_byte(addr);
#endif
 default:
  return 0xFF;
 }
}

void mac_pram_write_byte(unsigned char val, int addr)
{
 switch (macintosh_config->adb_type) {
 case MAC_ADB_IOP:
 case MAC_ADB_II:
 case MAC_ADB_PB1:
  via_pram_write_byte(val, addr);
  break;
#ifdef CONFIG_ADB_CUDA
 case MAC_ADB_EGRET:
 case MAC_ADB_CUDA:
  cuda_pram_write_byte(val, addr);
  break;
#endif
#ifdef CONFIG_ADB_PMU
 case MAC_ADB_PB2:
  pmu_pram_write_byte(val, addr);
  break;
#endif
 default:
  break;
 }
}

ssize_t mac_pram_get_size(void)
{
 return 256;
}
#endif /* CONFIG_NVRAM */

void mac_poweroff(void)
{
 if (oss_present) {
  oss_shutdown();
 } else if (macintosh_config->adb_type == MAC_ADB_II) {
  via_shutdown();
#ifdef CONFIG_ADB_CUDA
 } else if (macintosh_config->adb_type == MAC_ADB_EGRET ||
            macintosh_config->adb_type == MAC_ADB_CUDA) {
  cuda_shutdown();
#endif
#ifdef CONFIG_ADB_PMU
 } else if (macintosh_config->adb_type == MAC_ADB_PB2) {
  pmu_shutdown();
#endif
 }

 pr_crit("It is now safe to turn off your Macintosh.\n");
 local_irq_disable();
 while(1);
}

void mac_reset(void)
{
#ifdef CONFIG_ADB_CUDA
 if (macintosh_config->adb_type == MAC_ADB_EGRET ||
     macintosh_config->adb_type == MAC_ADB_CUDA) {
  cuda_restart();
 } else
#endif
#ifdef CONFIG_ADB_PMU
 if (macintosh_config->adb_type == MAC_ADB_PB2) {
  pmu_restart();
 } else
#endif
 if (CPU_IS_030) {
  /* 030-specific reset routine.  The idea is general, but the
   * specific registers to reset are '030-specific.  Until I
   * have a non-030 machine, I can't test anything else.
   *  -- C. Scott Ananian <cananian@alumni.princeton.edu>
 */

  unsigned long rombase = 0x40000000;

  /* make a 1-to-1 mapping, using the transparent tran. reg. */
  unsigned long virt = (unsigned long) mac_reset;
  unsigned long phys = virt_to_phys(mac_reset);
  unsigned long addr = (phys&0xFF000000)|0x8777;
  unsigned long offset = phys-virt;

  local_irq_disable(); /* lets not screw this up, ok? */
  __asm__ __volatile__(".chip 68030\n\t"
         "pmove %0,%/tt0\n\t"
         ".chip 68k"
         : : "m" (addr));
  /* Now jump to physical address so we can disable MMU */
  __asm__ __volatile__(
      ".chip 68030\n\t"
      "lea %/pc@(1f),%/a0\n\t"
      "addl %0,%/a0\n\t"/* fixup target address and stack ptr */
      "addl %0,%/sp\n\t"
      "pflusha\n\t"
      "jmp %/a0@\n\t" /* jump into physical memory */
      "0:.long 0\n\t" /* a constant zero. */
      /* OK.  Now reset everything and jump to reset vector. */
      "1:\n\t"
      "lea %/pc@(0b),%/a0\n\t"
      "pmove %/a0@, %/tc\n\t" /* disable mmu */
      "pmove %/a0@, %/tt0\n\t" /* disable tt0 */
      "pmove %/a0@, %/tt1\n\t" /* disable tt1 */
      "movel #0, %/a0\n\t"
      "movec %/a0, %/vbr\n\t" /* clear vector base register */
      "movec %/a0, %/cacr\n\t" /* disable caches */
      "movel #0x0808,%/a0\n\t"
      "movec %/a0, %/cacr\n\t" /* flush i&d caches */
      "movew #0x2700,%/sr\n\t" /* set up status register */
      "movel %1@(0x0),%/a0\n\t"/* load interrupt stack pointer */
      "movec %/a0, %/isp\n\t"
      "movel %1@(0x4),%/a0\n\t" /* load reset vector */
      "reset\n\t" /* reset external devices */
      "jmp %/a0@\n\t" /* jump to the reset vector */
      ".chip 68k"
      : : "r" (offset), "a" (rombase) : "a0");
 } else {
  /* need ROMBASE in booter */
  /* indeed, plus need to MAP THE ROM !! */

  if (mac_bi_data.rombase == 0)
   mac_bi_data.rombase = 0x40800000;

  /* works on some */
  rom_reset = (void *)(mac_bi_data.rombase + 0xa);

  local_irq_disable();
  rom_reset();
 }

 /* should never get here */
 pr_crit("Restart failed. Please restart manually.\n");
 local_irq_disable();
 while(1);
}

/*
 * This function translates seconds since 1970 into a proper date.
 *
 * Algorithm cribbed from glibc2.1, __offtime().
 *
 * This is roughly same as rtc_time64_to_tm(), which we should probably
 * use here, but it's only available when CONFIG_RTC_LIB is enabled.
 */
#define SECS_PER_MINUTE (60)
#define SECS_PER_HOUR  (SECS_PER_MINUTE * 60)
#define SECS_PER_DAY   (SECS_PER_HOUR * 24)

static void unmktime(time64_t time, long offset,
       int *yearp, int *monp, int *dayp,
       int *hourp, int *minp, int *secp)
{
        /* How many days come before each month (0-12).  */
 static const unsigned short int __mon_yday[2][13] =
 {
  /* Normal years.  */
  { 0, 31, 59, 90, 120, 151, 181, 212, 243, 273, 304, 334, 365 },
  /* Leap years.  */
  { 0, 31, 60, 91, 121, 152, 182, 213, 244, 274, 305, 335, 366 }
 };
 int days, rem, y, wday;
 const unsigned short int *ip;

 days = div_u64_rem(time, SECS_PER_DAY, &rem);
 rem += offset;
 while (rem < 0) {
  rem += SECS_PER_DAY;
  --days;
 }
 while (rem >= SECS_PER_DAY) {
  rem -= SECS_PER_DAY;
  ++days;
 }
 *hourp = rem / SECS_PER_HOUR;
 rem %= SECS_PER_HOUR;
 *minp = rem / SECS_PER_MINUTE;
 *secp = rem % SECS_PER_MINUTE;
 /* January 1, 1970 was a Thursday. */
 wday = (4 + days) % 7; /* Day in the week. Not currently used */
 if (wday < 0) wday += 7;
 y = 1970;

#define DIV(a, b) ((a) / (b) - ((a) % (b) < 0))
#define LEAPS_THRU_END_OF(y) (DIV (y, 4) - DIV (y, 100) + DIV (y, 400))
#define __isleap(year) \
  ((year) % 4 == 0 && ((year) % 100 != 0 || (year) % 400 == 0))

 while (days < 0 || days >= (__isleap (y) ? 366 : 365))
 {
  /* Guess a corrected year, assuming 365 days per year.  */
  long int yg = y + days / 365 - (days % 365 < 0);

  /* Adjust DAYS and Y to match the guessed year.  */
  days -= (yg - y) * 365 +
   LEAPS_THRU_END_OF(yg - 1) - LEAPS_THRU_END_OF(y - 1);
  y = yg;
 }
 *yearp = y - 1900;
 ip = __mon_yday[__isleap(y)];
 for (y = 11; days < (long int) ip[y]; --y)
  continue;
 days -= ip[y];
 *monp = y;
 *dayp = days + 1; /* day in the month */
 return;
}

/*
 * Read/write the hardware clock.
 */

int mac_hwclk(int op, struct rtc_time *t)
{
 time64_t now;

 if (!op) { /* read */
  switch (macintosh_config->adb_type) {
  case MAC_ADB_IOP:
  case MAC_ADB_II:
  case MAC_ADB_PB1:
   now = via_read_time();
   break;
#ifdef CONFIG_ADB_CUDA
  case MAC_ADB_EGRET:
  case MAC_ADB_CUDA:
   now = cuda_get_time();
   break;
#endif
#ifdef CONFIG_ADB_PMU
  case MAC_ADB_PB2:
   now = pmu_get_time();
   break;
#endif
  default:
   now = 0;
  }

  t->tm_wday = 0;
  unmktime(now, 0,
    &t->tm_year, &t->tm_mon, &t->tm_mday,
    &t->tm_hour, &t->tm_min, &t->tm_sec);
  pr_debug("%s: read %ptR\n", __func__, t);
 } else { /* write */
  pr_debug("%s: tried to write %ptR\n", __func__, t);

  switch (macintosh_config->adb_type) {
  case MAC_ADB_IOP:
  case MAC_ADB_II:
  case MAC_ADB_PB1:
   via_set_rtc_time(t);
   break;
#ifdef CONFIG_ADB_CUDA
  case MAC_ADB_EGRET:
  case MAC_ADB_CUDA:
   cuda_set_rtc_time(t);
   break;
#endif
#ifdef CONFIG_ADB_PMU
  case MAC_ADB_PB2:
   pmu_set_rtc_time(t);
   break;
#endif
  default:
   return -ENODEV;
  }
 }
 return 0;
}