/* SPDX-License-Identifier: GPL-2.0 */
/*
 * Kernel interface for the RISCV arch_random_* functions
 *
 * Copyright (c) 2023 Rivos Inc.
 *
 */

#ifndef ASM_RISCV_ARCHRANDOM_H
#define ASM_RISCV_ARCHRANDOM_H

#include <asm/csr.h>
#include <asm/processor.h>

#define SEED_RETRY_LOOPS 100

static inline bool __must_check csr_seed_long(unsigned long *v)
{
 unsigned int retry = SEED_RETRY_LOOPS, valid_seeds = 0;
 const int needed_seeds = sizeof(long) / sizeof(u16);
 u16 *entropy = (u16 *)v;

 do {
  /*
   * The SEED CSR must be accessed with a read-write instruction.
 */
  unsigned long csr_seed = csr_swap(CSR_SEED, 0);
  unsigned long opst = csr_seed & SEED_OPST_MASK;

  switch (opst) {
  case SEED_OPST_ES16:
   entropy[valid_seeds++] = csr_seed & SEED_ENTROPY_MASK;
   if (valid_seeds == needed_seeds)
    return true;
   break;

  case SEED_OPST_DEAD:
   pr_err_once("archrandom: Unrecoverable error\n");
   return false;

  case SEED_OPST_BIST:
  case SEED_OPST_WAIT:
  default:
   cpu_relax();
   continue;
  }
 } while (--retry);

 return false;
}

static inline size_t __must_check arch_get_random_longs(unsigned long *v, size_t max_longs)
{
 return 0;
}

static inline size_t __must_check arch_get_random_seed_longs(unsigned long *v, size_t max_longs)
{
 if (!max_longs)
  return 0;

 /*
  * If Zkr is supported and csr_seed_long succeeds, we return one long
  * worth of entropy.
 */
 if (riscv_has_extension_likely(RISCV_ISA_EXT_ZKR) && csr_seed_long(v))
  return 1;

 return 0;
}

#endif /* ASM_RISCV_ARCHRANDOM_H */