// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (c) 2024 Broadcom
 */

#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt

#include <linux/module.h>
#include <linux/mod_devicetable.h>
#include <linux/kernel.h>
#include <linux/io.h>
#include <linux/delay.h>
#include <linux/platform_device.h>
#include <linux/random.h>
#include <linux/hw_random.h>

#define HOST_REV_ID  0x00
#define HOST_FIFO_DEPTH  0x04
#define HOST_FIFO_COUNT  0x08
#define HOST_FIFO_THRESHOLD 0x0c
#define HOST_FIFO_DATA  0x10

#define HOST_FIFO_COUNT_MASK  0xffff

/* Delay range in microseconds */
#define FIFO_DELAY_MIN_US  3
#define FIFO_DELAY_MAX_US  7
#define FIFO_DELAY_MAX_COUNT  10

struct bcm74110_priv {
 void __iomem *base;
};

static inline int bcm74110_rng_fifo_count(void __iomem *mem)
{
 return readl_relaxed(mem) & HOST_FIFO_COUNT_MASK;
}

static int bcm74110_rng_read(struct hwrng *rng, void *buf, size_t max,
   bool wait)
{
 struct bcm74110_priv *priv = (struct bcm74110_priv *)rng->priv;
 void __iomem *fc_addr = priv->base + HOST_FIFO_COUNT;
 void __iomem *fd_addr = priv->base + HOST_FIFO_DATA;
 unsigned underrun_count = 0;
 u32 max_words = max / sizeof(u32);
 u32 num_words;
 unsigned i;

 /*
 * We need to check how many words are available in the RNG FIFO. If
 * there aren't any, we need to wait for some to become available.
 */
 while ((num_words = bcm74110_rng_fifo_count(fc_addr)) == 0) {
  if (!wait)
   return 0;
  /*
 * As a precaution, limit how long we wait. If the FIFO doesn't
 * refill within the allotted time, return 0 (=no data) to the
 * caller.
 */
  if (likely(underrun_count < FIFO_DELAY_MAX_COUNT))
   usleep_range(FIFO_DELAY_MIN_US, FIFO_DELAY_MAX_US);
  else
   return 0;
  underrun_count++;
 }
 if (num_words > max_words)
  num_words = max_words;

 /* Bail early if we run out of random numbers unexpectedly */
 for (i = 0; i < num_words && bcm74110_rng_fifo_count(fc_addr) > 0; i++)
  ((u32 *)buf)[i] = readl_relaxed(fd_addr);

 return i * sizeof(u32);
}

static struct hwrng bcm74110_hwrng = {
 .read = bcm74110_rng_read,
};

static int bcm74110_rng_probe(struct platform_device *pdev)
{
 struct device *dev = &pdev->dev;
 struct bcm74110_priv *priv;
 int rc;

 priv = devm_kzalloc(dev, sizeof(*priv), GFP_KERNEL);
 if (!priv)
  return -ENOMEM;

 bcm74110_hwrng.name = pdev->name;
 bcm74110_hwrng.priv = (unsigned long)priv;

 priv->base = devm_platform_ioremap_resource(pdev, 0);
 if (IS_ERR(priv->base))
  return PTR_ERR(priv->base);

 rc = devm_hwrng_register(dev, &bcm74110_hwrng);
 if (rc)
  dev_err(dev, "hwrng registration failed (%d)\n", rc);
 else
  dev_info(dev, "hwrng registered\n");

 return rc;
}

static const struct of_device_id bcm74110_rng_match[] = {
 { .compatible = "brcm,bcm74110-rng", },
 {},
};
MODULE_DEVICE_TABLE(of, bcm74110_rng_match);

static struct platform_driver bcm74110_rng_driver = {
 .driver = {
  .name = KBUILD_MODNAME,
  .of_match_table = bcm74110_rng_match,
 },
 .probe = bcm74110_rng_probe,
};
module_platform_driver(bcm74110_rng_driver);

MODULE_AUTHOR("Markus Mayer <mmayer@broadcom.com>");
MODULE_DESCRIPTION("BCM 74110 Random Number Generator (RNG) driver");
MODULE_LICENSE("GPL v2");