// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (C) 2024, Ventana Micro Systems Inc
 * Author: Sunil V L <sunilvl@ventanamicro.com>
 *
 */

#include <linux/acpi.h>
#include <acpi/processor.h>
#include <linux/cpu_pm.h>
#include <linux/cpuidle.h>
#include <linux/suspend.h>
#include <asm/cpuidle.h>
#include <asm/sbi.h>
#include <asm/suspend.h>

#define RISCV_FFH_LPI_TYPE_MASK GENMASK_ULL(63, 60)
#define RISCV_FFH_LPI_RSVD_MASK GENMASK_ULL(59, 32)

#define RISCV_FFH_LPI_TYPE_SBI BIT_ULL(60)

static int acpi_cpu_init_idle(unsigned int cpu)
{
 int i;
 struct acpi_lpi_state *lpi;
 struct acpi_processor *pr = per_cpu(processors, cpu);

 if (unlikely(!pr || !pr->flags.has_lpi))
  return -EINVAL;

 if (!riscv_sbi_hsm_is_supported())
  return -ENODEV;

 if (pr->power.count <= 1)
  return -ENODEV;

 for (i = 1; i < pr->power.count; i++) {
  u32 state;

  lpi = &pr->power.lpi_states[i];

  /*
 * Validate Entry Method as per FFH spec.
 * bits[63:60] should be 0x1
 * bits[59:32] should be 0x0
 * bits[31:0] represent a SBI power_state
 */
  if (((lpi->address & RISCV_FFH_LPI_TYPE_MASK) != RISCV_FFH_LPI_TYPE_SBI) ||
      (lpi->address & RISCV_FFH_LPI_RSVD_MASK)) {
   pr_warn("Invalid LPI entry method %#llx\n", lpi->address);
   return -EINVAL;
  }

  state = lpi->address;
  if (!riscv_sbi_suspend_state_is_valid(state)) {
   pr_warn("Invalid SBI power state %#x\n", state);
   return -EINVAL;
  }
 }

 return 0;
}

int acpi_processor_ffh_lpi_probe(unsigned int cpu)
{
 return acpi_cpu_init_idle(cpu);
}

int acpi_processor_ffh_lpi_enter(struct acpi_lpi_state *lpi)
{
 u32 state = lpi->address;

 if (state & SBI_HSM_SUSP_NON_RET_BIT)
  return CPU_PM_CPU_IDLE_ENTER_PARAM(riscv_sbi_hart_suspend,
         lpi->index,
         state);
 else
  return CPU_PM_CPU_IDLE_ENTER_RETENTION_PARAM(riscv_sbi_hart_suspend,
            lpi->index,
            state);
}