162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * STM32 Factory-programmed memory read access driver
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * Copyright (C) 2017, STMicroelectronics - All Rights Reserved
662306a36Sopenharmony_ci * Author: Fabrice Gasnier <fabrice.gasnier@st.com> for STMicroelectronics.
762306a36Sopenharmony_ci */
862306a36Sopenharmony_ci
962306a36Sopenharmony_ci#include <linux/arm-smccc.h>
1062306a36Sopenharmony_ci#include <linux/io.h>
1162306a36Sopenharmony_ci#include <linux/module.h>
1262306a36Sopenharmony_ci#include <linux/nvmem-provider.h>
1362306a36Sopenharmony_ci#include <linux/of_device.h>
1462306a36Sopenharmony_ci#include <linux/tee_drv.h>
1562306a36Sopenharmony_ci
1662306a36Sopenharmony_ci#include "stm32-bsec-optee-ta.h"
1762306a36Sopenharmony_ci
1862306a36Sopenharmony_ci/* BSEC secure service access from non-secure */
1962306a36Sopenharmony_ci#define STM32_SMC_BSEC			0x82001003
2062306a36Sopenharmony_ci#define STM32_SMC_READ_SHADOW		0x01
2162306a36Sopenharmony_ci#define STM32_SMC_PROG_OTP		0x02
2262306a36Sopenharmony_ci#define STM32_SMC_WRITE_SHADOW		0x03
2362306a36Sopenharmony_ci#define STM32_SMC_READ_OTP		0x04
2462306a36Sopenharmony_ci
2562306a36Sopenharmony_ci/* shadow registers offset */
2662306a36Sopenharmony_ci#define STM32MP15_BSEC_DATA0		0x200
2762306a36Sopenharmony_ci
2862306a36Sopenharmony_cistruct stm32_romem_cfg {
2962306a36Sopenharmony_ci	int size;
3062306a36Sopenharmony_ci	u8 lower;
3162306a36Sopenharmony_ci	bool ta;
3262306a36Sopenharmony_ci};
3362306a36Sopenharmony_ci
3462306a36Sopenharmony_cistruct stm32_romem_priv {
3562306a36Sopenharmony_ci	void __iomem *base;
3662306a36Sopenharmony_ci	struct nvmem_config cfg;
3762306a36Sopenharmony_ci	u8 lower;
3862306a36Sopenharmony_ci	struct tee_context *ctx;
3962306a36Sopenharmony_ci};
4062306a36Sopenharmony_ci
4162306a36Sopenharmony_cistatic int stm32_romem_read(void *context, unsigned int offset, void *buf,
4262306a36Sopenharmony_ci			    size_t bytes)
4362306a36Sopenharmony_ci{
4462306a36Sopenharmony_ci	struct stm32_romem_priv *priv = context;
4562306a36Sopenharmony_ci	u8 *buf8 = buf;
4662306a36Sopenharmony_ci	int i;
4762306a36Sopenharmony_ci
4862306a36Sopenharmony_ci	for (i = offset; i < offset + bytes; i++)
4962306a36Sopenharmony_ci		*buf8++ = readb_relaxed(priv->base + i);
5062306a36Sopenharmony_ci
5162306a36Sopenharmony_ci	return 0;
5262306a36Sopenharmony_ci}
5362306a36Sopenharmony_ci
5462306a36Sopenharmony_cistatic int stm32_bsec_smc(u8 op, u32 otp, u32 data, u32 *result)
5562306a36Sopenharmony_ci{
5662306a36Sopenharmony_ci#if IS_ENABLED(CONFIG_HAVE_ARM_SMCCC)
5762306a36Sopenharmony_ci	struct arm_smccc_res res;
5862306a36Sopenharmony_ci
5962306a36Sopenharmony_ci	arm_smccc_smc(STM32_SMC_BSEC, op, otp, data, 0, 0, 0, 0, &res);
6062306a36Sopenharmony_ci	if (res.a0)
6162306a36Sopenharmony_ci		return -EIO;
6262306a36Sopenharmony_ci
6362306a36Sopenharmony_ci	if (result)
6462306a36Sopenharmony_ci		*result = (u32)res.a1;
6562306a36Sopenharmony_ci
6662306a36Sopenharmony_ci	return 0;
6762306a36Sopenharmony_ci#else
6862306a36Sopenharmony_ci	return -ENXIO;
6962306a36Sopenharmony_ci#endif
7062306a36Sopenharmony_ci}
7162306a36Sopenharmony_ci
7262306a36Sopenharmony_cistatic int stm32_bsec_read(void *context, unsigned int offset, void *buf,
7362306a36Sopenharmony_ci			   size_t bytes)
7462306a36Sopenharmony_ci{
7562306a36Sopenharmony_ci	struct stm32_romem_priv *priv = context;
7662306a36Sopenharmony_ci	struct device *dev = priv->cfg.dev;
7762306a36Sopenharmony_ci	u32 roffset, rbytes, val;
7862306a36Sopenharmony_ci	u8 *buf8 = buf, *val8 = (u8 *)&val;
7962306a36Sopenharmony_ci	int i, j = 0, ret, skip_bytes, size;
8062306a36Sopenharmony_ci
8162306a36Sopenharmony_ci	/* Round unaligned access to 32-bits */
8262306a36Sopenharmony_ci	roffset = rounddown(offset, 4);
8362306a36Sopenharmony_ci	skip_bytes = offset & 0x3;
8462306a36Sopenharmony_ci	rbytes = roundup(bytes + skip_bytes, 4);
8562306a36Sopenharmony_ci
8662306a36Sopenharmony_ci	if (roffset + rbytes > priv->cfg.size)
8762306a36Sopenharmony_ci		return -EINVAL;
8862306a36Sopenharmony_ci
8962306a36Sopenharmony_ci	for (i = roffset; (i < roffset + rbytes); i += 4) {
9062306a36Sopenharmony_ci		u32 otp = i >> 2;
9162306a36Sopenharmony_ci
9262306a36Sopenharmony_ci		if (otp < priv->lower) {
9362306a36Sopenharmony_ci			/* read lower data from shadow registers */
9462306a36Sopenharmony_ci			val = readl_relaxed(
9562306a36Sopenharmony_ci				priv->base + STM32MP15_BSEC_DATA0 + i);
9662306a36Sopenharmony_ci		} else {
9762306a36Sopenharmony_ci			ret = stm32_bsec_smc(STM32_SMC_READ_SHADOW, otp, 0,
9862306a36Sopenharmony_ci					     &val);
9962306a36Sopenharmony_ci			if (ret) {
10062306a36Sopenharmony_ci				dev_err(dev, "Can't read data%d (%d)\n", otp,
10162306a36Sopenharmony_ci					ret);
10262306a36Sopenharmony_ci				return ret;
10362306a36Sopenharmony_ci			}
10462306a36Sopenharmony_ci		}
10562306a36Sopenharmony_ci		/* skip first bytes in case of unaligned read */
10662306a36Sopenharmony_ci		if (skip_bytes)
10762306a36Sopenharmony_ci			size = min(bytes, (size_t)(4 - skip_bytes));
10862306a36Sopenharmony_ci		else
10962306a36Sopenharmony_ci			size = min(bytes, (size_t)4);
11062306a36Sopenharmony_ci		memcpy(&buf8[j], &val8[skip_bytes], size);
11162306a36Sopenharmony_ci		bytes -= size;
11262306a36Sopenharmony_ci		j += size;
11362306a36Sopenharmony_ci		skip_bytes = 0;
11462306a36Sopenharmony_ci	}
11562306a36Sopenharmony_ci
11662306a36Sopenharmony_ci	return 0;
11762306a36Sopenharmony_ci}
11862306a36Sopenharmony_ci
11962306a36Sopenharmony_cistatic int stm32_bsec_write(void *context, unsigned int offset, void *buf,
12062306a36Sopenharmony_ci			    size_t bytes)
12162306a36Sopenharmony_ci{
12262306a36Sopenharmony_ci	struct stm32_romem_priv *priv = context;
12362306a36Sopenharmony_ci	struct device *dev = priv->cfg.dev;
12462306a36Sopenharmony_ci	u32 *buf32 = buf;
12562306a36Sopenharmony_ci	int ret, i;
12662306a36Sopenharmony_ci
12762306a36Sopenharmony_ci	/* Allow only writing complete 32-bits aligned words */
12862306a36Sopenharmony_ci	if ((bytes % 4) || (offset % 4))
12962306a36Sopenharmony_ci		return -EINVAL;
13062306a36Sopenharmony_ci
13162306a36Sopenharmony_ci	for (i = offset; i < offset + bytes; i += 4) {
13262306a36Sopenharmony_ci		ret = stm32_bsec_smc(STM32_SMC_PROG_OTP, i >> 2, *buf32++,
13362306a36Sopenharmony_ci				     NULL);
13462306a36Sopenharmony_ci		if (ret) {
13562306a36Sopenharmony_ci			dev_err(dev, "Can't write data%d (%d)\n", i >> 2, ret);
13662306a36Sopenharmony_ci			return ret;
13762306a36Sopenharmony_ci		}
13862306a36Sopenharmony_ci	}
13962306a36Sopenharmony_ci
14062306a36Sopenharmony_ci	if (offset + bytes >= priv->lower * 4)
14162306a36Sopenharmony_ci		dev_warn(dev, "Update of upper OTPs with ECC protection (word programming, only once)\n");
14262306a36Sopenharmony_ci
14362306a36Sopenharmony_ci	return 0;
14462306a36Sopenharmony_ci}
14562306a36Sopenharmony_ci
14662306a36Sopenharmony_cistatic int stm32_bsec_pta_read(void *context, unsigned int offset, void *buf,
14762306a36Sopenharmony_ci			       size_t bytes)
14862306a36Sopenharmony_ci{
14962306a36Sopenharmony_ci	struct stm32_romem_priv *priv = context;
15062306a36Sopenharmony_ci
15162306a36Sopenharmony_ci	return stm32_bsec_optee_ta_read(priv->ctx, offset, buf, bytes);
15262306a36Sopenharmony_ci}
15362306a36Sopenharmony_ci
15462306a36Sopenharmony_cistatic int stm32_bsec_pta_write(void *context, unsigned int offset, void *buf,
15562306a36Sopenharmony_ci				size_t bytes)
15662306a36Sopenharmony_ci{
15762306a36Sopenharmony_ci	struct stm32_romem_priv *priv = context;
15862306a36Sopenharmony_ci
15962306a36Sopenharmony_ci	return stm32_bsec_optee_ta_write(priv->ctx, priv->lower, offset, buf, bytes);
16062306a36Sopenharmony_ci}
16162306a36Sopenharmony_ci
16262306a36Sopenharmony_cistatic bool stm32_bsec_smc_check(void)
16362306a36Sopenharmony_ci{
16462306a36Sopenharmony_ci	u32 val;
16562306a36Sopenharmony_ci	int ret;
16662306a36Sopenharmony_ci
16762306a36Sopenharmony_ci	/* check that the OP-TEE support the BSEC SMC (legacy mode) */
16862306a36Sopenharmony_ci	ret = stm32_bsec_smc(STM32_SMC_READ_SHADOW, 0, 0, &val);
16962306a36Sopenharmony_ci
17062306a36Sopenharmony_ci	return !ret;
17162306a36Sopenharmony_ci}
17262306a36Sopenharmony_ci
17362306a36Sopenharmony_cistatic bool optee_presence_check(void)
17462306a36Sopenharmony_ci{
17562306a36Sopenharmony_ci	struct device_node *np;
17662306a36Sopenharmony_ci	bool tee_detected = false;
17762306a36Sopenharmony_ci
17862306a36Sopenharmony_ci	/* check that the OP-TEE node is present and available. */
17962306a36Sopenharmony_ci	np = of_find_compatible_node(NULL, NULL, "linaro,optee-tz");
18062306a36Sopenharmony_ci	if (np && of_device_is_available(np))
18162306a36Sopenharmony_ci		tee_detected = true;
18262306a36Sopenharmony_ci	of_node_put(np);
18362306a36Sopenharmony_ci
18462306a36Sopenharmony_ci	return tee_detected;
18562306a36Sopenharmony_ci}
18662306a36Sopenharmony_ci
18762306a36Sopenharmony_cistatic int stm32_romem_probe(struct platform_device *pdev)
18862306a36Sopenharmony_ci{
18962306a36Sopenharmony_ci	const struct stm32_romem_cfg *cfg;
19062306a36Sopenharmony_ci	struct device *dev = &pdev->dev;
19162306a36Sopenharmony_ci	struct stm32_romem_priv *priv;
19262306a36Sopenharmony_ci	struct resource *res;
19362306a36Sopenharmony_ci	int rc;
19462306a36Sopenharmony_ci
19562306a36Sopenharmony_ci	priv = devm_kzalloc(dev, sizeof(*priv), GFP_KERNEL);
19662306a36Sopenharmony_ci	if (!priv)
19762306a36Sopenharmony_ci		return -ENOMEM;
19862306a36Sopenharmony_ci
19962306a36Sopenharmony_ci	priv->base = devm_platform_get_and_ioremap_resource(pdev, 0, &res);
20062306a36Sopenharmony_ci	if (IS_ERR(priv->base))
20162306a36Sopenharmony_ci		return PTR_ERR(priv->base);
20262306a36Sopenharmony_ci
20362306a36Sopenharmony_ci	priv->cfg.name = "stm32-romem";
20462306a36Sopenharmony_ci	priv->cfg.word_size = 1;
20562306a36Sopenharmony_ci	priv->cfg.stride = 1;
20662306a36Sopenharmony_ci	priv->cfg.dev = dev;
20762306a36Sopenharmony_ci	priv->cfg.priv = priv;
20862306a36Sopenharmony_ci	priv->cfg.owner = THIS_MODULE;
20962306a36Sopenharmony_ci	priv->cfg.type = NVMEM_TYPE_OTP;
21062306a36Sopenharmony_ci
21162306a36Sopenharmony_ci	priv->lower = 0;
21262306a36Sopenharmony_ci
21362306a36Sopenharmony_ci	cfg = (const struct stm32_romem_cfg *)
21462306a36Sopenharmony_ci		of_match_device(dev->driver->of_match_table, dev)->data;
21562306a36Sopenharmony_ci	if (!cfg) {
21662306a36Sopenharmony_ci		priv->cfg.read_only = true;
21762306a36Sopenharmony_ci		priv->cfg.size = resource_size(res);
21862306a36Sopenharmony_ci		priv->cfg.reg_read = stm32_romem_read;
21962306a36Sopenharmony_ci	} else {
22062306a36Sopenharmony_ci		priv->cfg.size = cfg->size;
22162306a36Sopenharmony_ci		priv->lower = cfg->lower;
22262306a36Sopenharmony_ci		if (cfg->ta || optee_presence_check()) {
22362306a36Sopenharmony_ci			rc = stm32_bsec_optee_ta_open(&priv->ctx);
22462306a36Sopenharmony_ci			if (rc) {
22562306a36Sopenharmony_ci				/* wait for OP-TEE client driver to be up and ready */
22662306a36Sopenharmony_ci				if (rc == -EPROBE_DEFER)
22762306a36Sopenharmony_ci					return -EPROBE_DEFER;
22862306a36Sopenharmony_ci				/* BSEC PTA is required or SMC not supported */
22962306a36Sopenharmony_ci				if (cfg->ta || !stm32_bsec_smc_check())
23062306a36Sopenharmony_ci					return rc;
23162306a36Sopenharmony_ci			}
23262306a36Sopenharmony_ci		}
23362306a36Sopenharmony_ci		if (priv->ctx) {
23462306a36Sopenharmony_ci			rc = devm_add_action_or_reset(dev, stm32_bsec_optee_ta_close, priv->ctx);
23562306a36Sopenharmony_ci			if (rc) {
23662306a36Sopenharmony_ci				dev_err(dev, "devm_add_action_or_reset() failed (%d)\n", rc);
23762306a36Sopenharmony_ci				return rc;
23862306a36Sopenharmony_ci			}
23962306a36Sopenharmony_ci			priv->cfg.reg_read = stm32_bsec_pta_read;
24062306a36Sopenharmony_ci			priv->cfg.reg_write = stm32_bsec_pta_write;
24162306a36Sopenharmony_ci		} else {
24262306a36Sopenharmony_ci			priv->cfg.reg_read = stm32_bsec_read;
24362306a36Sopenharmony_ci			priv->cfg.reg_write = stm32_bsec_write;
24462306a36Sopenharmony_ci		}
24562306a36Sopenharmony_ci	}
24662306a36Sopenharmony_ci
24762306a36Sopenharmony_ci	return PTR_ERR_OR_ZERO(devm_nvmem_register(dev, &priv->cfg));
24862306a36Sopenharmony_ci}
24962306a36Sopenharmony_ci
25062306a36Sopenharmony_ci/*
25162306a36Sopenharmony_ci * STM32MP15/13 BSEC OTP regions: 4096 OTP bits (with 3072 effective bits)
25262306a36Sopenharmony_ci * => 96 x 32-bits data words
25362306a36Sopenharmony_ci * - Lower: 1K bits, 2:1 redundancy, incremental bit programming
25462306a36Sopenharmony_ci *   => 32 (x 32-bits) lower shadow registers = words 0 to 31
25562306a36Sopenharmony_ci * - Upper: 2K bits, ECC protection, word programming only
25662306a36Sopenharmony_ci *   => 64 (x 32-bits) = words 32 to 95
25762306a36Sopenharmony_ci */
25862306a36Sopenharmony_cistatic const struct stm32_romem_cfg stm32mp15_bsec_cfg = {
25962306a36Sopenharmony_ci	.size = 384,
26062306a36Sopenharmony_ci	.lower = 32,
26162306a36Sopenharmony_ci	.ta = false,
26262306a36Sopenharmony_ci};
26362306a36Sopenharmony_ci
26462306a36Sopenharmony_cistatic const struct stm32_romem_cfg stm32mp13_bsec_cfg = {
26562306a36Sopenharmony_ci	.size = 384,
26662306a36Sopenharmony_ci	.lower = 32,
26762306a36Sopenharmony_ci	.ta = true,
26862306a36Sopenharmony_ci};
26962306a36Sopenharmony_ci
27062306a36Sopenharmony_cistatic const struct of_device_id stm32_romem_of_match[] __maybe_unused = {
27162306a36Sopenharmony_ci	{ .compatible = "st,stm32f4-otp", }, {
27262306a36Sopenharmony_ci		.compatible = "st,stm32mp15-bsec",
27362306a36Sopenharmony_ci		.data = (void *)&stm32mp15_bsec_cfg,
27462306a36Sopenharmony_ci	}, {
27562306a36Sopenharmony_ci		.compatible = "st,stm32mp13-bsec",
27662306a36Sopenharmony_ci		.data = (void *)&stm32mp13_bsec_cfg,
27762306a36Sopenharmony_ci	},
27862306a36Sopenharmony_ci	{ /* sentinel */ },
27962306a36Sopenharmony_ci};
28062306a36Sopenharmony_ciMODULE_DEVICE_TABLE(of, stm32_romem_of_match);
28162306a36Sopenharmony_ci
28262306a36Sopenharmony_cistatic struct platform_driver stm32_romem_driver = {
28362306a36Sopenharmony_ci	.probe = stm32_romem_probe,
28462306a36Sopenharmony_ci	.driver = {
28562306a36Sopenharmony_ci		.name = "stm32-romem",
28662306a36Sopenharmony_ci		.of_match_table = of_match_ptr(stm32_romem_of_match),
28762306a36Sopenharmony_ci	},
28862306a36Sopenharmony_ci};
28962306a36Sopenharmony_cimodule_platform_driver(stm32_romem_driver);
29062306a36Sopenharmony_ci
29162306a36Sopenharmony_ciMODULE_AUTHOR("Fabrice Gasnier <fabrice.gasnier@st.com>");
29262306a36Sopenharmony_ciMODULE_DESCRIPTION("STMicroelectronics STM32 RO-MEM");
29362306a36Sopenharmony_ciMODULE_ALIAS("platform:nvmem-stm32-romem");
29462306a36Sopenharmony_ciMODULE_LICENSE("GPL v2");
295