// SPDX-License-Identifier: GPL-2.0 /* * Copyright (c) 2020 Fuzhou Rockchip Electronics Co., Ltd */ #include #include #include #include #include #include #include #include #define RK_HW_RNG_MAX 32 #define _SBF(s, v) ((v) << (s)) /* start of CRYPTO V1 register define */ #define CRYPTO_V1_CTRL 0x0008 #define CRYPTO_V1_RNG_START BIT(8) #define CRYPTO_V1_RNG_FLUSH BIT(9) #define CRYPTO_V1_TRNG_CTRL 0x0200 #define CRYPTO_V1_OSC_ENABLE BIT(16) #define CRYPTO_V1_TRNG_SAMPLE_PERIOD(x) (x) #define CRYPTO_V1_TRNG_DOUT_0 0x0204 /* end of CRYPTO V1 register define */ /* start of CRYPTO V2 register define */ #define CRYPTO_V2_RNG_CTL 0x0400 #define CRYPTO_V2_RNG_64_BIT_LEN _SBF(4, 0x00) #define CRYPTO_V2_RNG_128_BIT_LEN _SBF(4, 0x01) #define CRYPTO_V2_RNG_192_BIT_LEN _SBF(4, 0x02) #define CRYPTO_V2_RNG_256_BIT_LEN _SBF(4, 0x03) #define CRYPTO_V2_RNG_FATESY_SOC_RING _SBF(2, 0x00) #define CRYPTO_V2_RNG_SLOWER_SOC_RING_0 _SBF(2, 0x01) #define CRYPTO_V2_RNG_SLOWER_SOC_RING_1 _SBF(2, 0x02) #define CRYPTO_V2_RNG_SLOWEST_SOC_RING _SBF(2, 0x03) #define CRYPTO_V2_RNG_ENABLE BIT(1) #define CRYPTO_V2_RNG_START BIT(0) #define CRYPTO_V2_RNG_SAMPLE_CNT 0x0404 #define CRYPTO_V2_RNG_DOUT_0 0x0410 /* end of CRYPTO V2 register define */ /* start of TRNG V1 register define */ #define TRNG_V1_CTRL 0x0000 #define TRNG_V1_CTRL_NOP _SBF(0, 0x00) #define TRNG_V1_CTRL_RAND _SBF(0, 0x01) #define TRNG_V1_CTRL_SEED _SBF(0, 0x02) #define TRNG_V1_MODE 0x0008 #define TRNG_V1_MODE_128_BIT _SBF(3, 0x00) #define TRNG_V1_MODE_256_BIT _SBF(3, 0x01) #define TRNG_V1_IE 0x0010 #define TRNG_V1_IE_GLBL_EN BIT(31) #define TRNG_V1_IE_SEED_DONE_EN BIT(1) #define TRNG_V1_IE_RAND_RDY_EN BIT(0) #define TRNG_V1_ISTAT 0x0014 #define TRNG_V1_ISTAT_RAND_RDY BIT(0) /* RAND0 ~ RAND7 */ #define TRNG_V1_RAND0 0x0020 #define TRNG_V1_RAND7 0x003C #define TRNG_V1_AUTO_RQSTS 0x0060 #define TRNG_V1_VERSION 0x00F0 #define TRNG_v1_VERSION_CODE 0x46BC /* end of TRNG V1 register define */ #define RK_RNG_TIME_OUT 50000 /* max 50ms */ #define trng_write(pdata, pos, val) writel(val, (pdata)->base + (pos)) #define trng_read(pdata, pos) readl((pdata)->base + (pos)) struct rk_rng_soc_data { int (*rk_rng_init)(struct udevice *dev); int (*rk_rng_read)(struct udevice *dev, void *data, size_t len); }; struct rk_rng_plat { fdt_addr_t base; struct rk_rng_soc_data *soc_data; }; static int rk_rng_read_regs(fdt_addr_t addr, void *buf, size_t size) { u32 count = RK_HW_RNG_MAX / sizeof(u32); u32 reg, tmp_len; if (size > RK_HW_RNG_MAX) return -EINVAL; while (size && count) { reg = readl(addr); tmp_len = min(size, sizeof(u32)); memcpy(buf, ®, tmp_len); addr += sizeof(u32); buf += tmp_len; size -= tmp_len; count--; } return 0; } static int rk_cryptov1_rng_read(struct udevice *dev, void *data, size_t len) { struct rk_rng_plat *pdata = dev_get_priv(dev); u32 reg = 0; int retval; if (len > RK_HW_RNG_MAX) return -EINVAL; /* enable osc_ring to get entropy, sample period is set as 100 */ writel(CRYPTO_V1_OSC_ENABLE | CRYPTO_V1_TRNG_SAMPLE_PERIOD(100), pdata->base + CRYPTO_V1_TRNG_CTRL); rk_clrsetreg(pdata->base + CRYPTO_V1_CTRL, CRYPTO_V1_RNG_START, CRYPTO_V1_RNG_START); retval = readl_poll_timeout(pdata->base + CRYPTO_V1_CTRL, reg, !(reg & CRYPTO_V1_RNG_START), RK_RNG_TIME_OUT); if (retval) goto exit; rk_rng_read_regs(pdata->base + CRYPTO_V1_TRNG_DOUT_0, data, len); exit: /* close TRNG */ rk_clrreg(pdata->base + CRYPTO_V1_CTRL, CRYPTO_V1_RNG_START); return 0; } static int rk_cryptov2_rng_read(struct udevice *dev, void *data, size_t len) { struct rk_rng_plat *pdata = dev_get_priv(dev); u32 reg = 0; int retval; if (len > RK_HW_RNG_MAX) return -EINVAL; /* enable osc_ring to get entropy, sample period is set as 100 */ writel(100, pdata->base + CRYPTO_V2_RNG_SAMPLE_CNT); reg |= CRYPTO_V2_RNG_256_BIT_LEN; reg |= CRYPTO_V2_RNG_SLOWER_SOC_RING_0; reg |= CRYPTO_V2_RNG_ENABLE; reg |= CRYPTO_V2_RNG_START; rk_clrsetreg(pdata->base + CRYPTO_V2_RNG_CTL, 0xffff, reg); retval = readl_poll_timeout(pdata->base + CRYPTO_V2_RNG_CTL, reg, !(reg & CRYPTO_V2_RNG_START), RK_RNG_TIME_OUT); if (retval) goto exit; rk_rng_read_regs(pdata->base + CRYPTO_V2_RNG_DOUT_0, data, len); exit: /* close TRNG */ rk_clrreg(pdata->base + CRYPTO_V2_RNG_CTL, 0xffff); return retval; } static int rk_trngv1_init(struct udevice *dev) { u32 status, version; u32 auto_reseed_cnt = 1000; struct rk_rng_plat *pdata = dev_get_priv(dev); version = trng_read(pdata, TRNG_V1_VERSION); if (version != TRNG_v1_VERSION_CODE) { printf("wrong trng version, expected = %08x, actual = %08x", TRNG_V1_VERSION, version); return -EFAULT; } /* wait in case of RND_RDY triggered at firs power on */ readl_poll_timeout(pdata->base + TRNG_V1_ISTAT, status, (status & TRNG_V1_ISTAT_RAND_RDY), RK_RNG_TIME_OUT); /* clear RAND_RDY flag for first power on */ trng_write(pdata, TRNG_V1_ISTAT, status); /* auto reseed after (auto_reseed_cnt * 16) byte rand generate */ trng_write(pdata, TRNG_V1_AUTO_RQSTS, auto_reseed_cnt); return 0; } static int rk_trngv1_rng_read(struct udevice *dev, void *data, size_t len) { struct rk_rng_plat *pdata = dev_get_priv(dev); u32 reg = 0; int retval; if (len > RK_HW_RNG_MAX) return -EINVAL; trng_write(pdata, TRNG_V1_MODE, TRNG_V1_MODE_256_BIT); trng_write(pdata, TRNG_V1_CTRL, TRNG_V1_CTRL_RAND); retval = readl_poll_timeout(pdata->base + TRNG_V1_ISTAT, reg, (reg & TRNG_V1_ISTAT_RAND_RDY), RK_RNG_TIME_OUT); /* clear ISTAT */ trng_write(pdata, TRNG_V1_ISTAT, reg); if (retval) goto exit; rk_rng_read_regs(pdata->base + TRNG_V1_RAND0, data, len); exit: /* close TRNG */ trng_write(pdata, TRNG_V1_CTRL, TRNG_V1_CTRL_NOP); return retval; } static int rockchip_rng_read(struct udevice *dev, void *data, size_t len) { unsigned char *buf = data; unsigned int i; int ret = -EIO; struct rk_rng_plat *pdata = dev_get_priv(dev); if (!len) return 0; if (!pdata->soc_data || !pdata->soc_data->rk_rng_read) return -EINVAL; for (i = 0; i < len / RK_HW_RNG_MAX; i++, buf += RK_HW_RNG_MAX) { ret = pdata->soc_data->rk_rng_read(dev, buf, RK_HW_RNG_MAX); if (ret) goto exit; } if (len % RK_HW_RNG_MAX) ret = pdata->soc_data->rk_rng_read(dev, buf, len % RK_HW_RNG_MAX); exit: return ret; } static int rockchip_rng_of_to_plat(struct udevice *dev) { struct rk_rng_plat *pdata = dev_get_priv(dev); memset(pdata, 0x00, sizeof(*pdata)); pdata->base = (fdt_addr_t)dev_read_addr_ptr(dev); if (!pdata->base) return -ENOMEM; return 0; } static int rockchip_rng_probe(struct udevice *dev) { struct rk_rng_plat *pdata = dev_get_priv(dev); int ret = 0; pdata->soc_data = (struct rk_rng_soc_data *)dev_get_driver_data(dev); if (pdata->soc_data->rk_rng_init) ret = pdata->soc_data->rk_rng_init(dev); return ret; } static const struct rk_rng_soc_data rk_cryptov1_soc_data = { .rk_rng_read = rk_cryptov1_rng_read, }; static const struct rk_rng_soc_data rk_cryptov2_soc_data = { .rk_rng_read = rk_cryptov2_rng_read, }; static const struct rk_rng_soc_data rk_trngv1_soc_data = { .rk_rng_init = rk_trngv1_init, .rk_rng_read = rk_trngv1_rng_read, }; static const struct dm_rng_ops rockchip_rng_ops = { .read = rockchip_rng_read, }; static const struct udevice_id rockchip_rng_match[] = { { .compatible = "rockchip,cryptov1-rng", .data = (ulong)&rk_cryptov1_soc_data, }, { .compatible = "rockchip,cryptov2-rng", .data = (ulong)&rk_cryptov2_soc_data, }, { .compatible = "rockchip,trngv1", .data = (ulong)&rk_trngv1_soc_data, }, {}, }; U_BOOT_DRIVER(rockchip_rng) = { .name = "rockchip-rng", .id = UCLASS_RNG, .of_match = rockchip_rng_match, .ops = &rockchip_rng_ops, .probe = rockchip_rng_probe, .of_to_plat = rockchip_rng_of_to_plat, .priv_auto = sizeof(struct rk_rng_plat), };