162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright (C) 2015 Pengutronix, Steffen Trumtrar <kernel@pengutronix.de>
462306a36Sopenharmony_ci * Copyright (C) 2021 Pengutronix, Ahmad Fatoum <kernel@pengutronix.de>
562306a36Sopenharmony_ci */
662306a36Sopenharmony_ci
762306a36Sopenharmony_ci#define pr_fmt(fmt) "caam blob_gen: " fmt
862306a36Sopenharmony_ci
962306a36Sopenharmony_ci#include <linux/bitfield.h>
1062306a36Sopenharmony_ci#include <linux/device.h>
1162306a36Sopenharmony_ci#include <soc/fsl/caam-blob.h>
1262306a36Sopenharmony_ci
1362306a36Sopenharmony_ci#include "compat.h"
1462306a36Sopenharmony_ci#include "desc_constr.h"
1562306a36Sopenharmony_ci#include "desc.h"
1662306a36Sopenharmony_ci#include "error.h"
1762306a36Sopenharmony_ci#include "intern.h"
1862306a36Sopenharmony_ci#include "jr.h"
1962306a36Sopenharmony_ci#include "regs.h"
2062306a36Sopenharmony_ci
2162306a36Sopenharmony_ci#define CAAM_BLOB_DESC_BYTES_MAX					\
2262306a36Sopenharmony_ci	/* Command to initialize & stating length of descriptor */	\
2362306a36Sopenharmony_ci	(CAAM_CMD_SZ +							\
2462306a36Sopenharmony_ci	/* Command to append the key-modifier + key-modifier data */	\
2562306a36Sopenharmony_ci	 CAAM_CMD_SZ + CAAM_BLOB_KEYMOD_LENGTH +			\
2662306a36Sopenharmony_ci	/* Command to include input key + pointer to the input key */	\
2762306a36Sopenharmony_ci	 CAAM_CMD_SZ + CAAM_PTR_SZ_MAX +				\
2862306a36Sopenharmony_ci	/* Command to include output key + pointer to the output key */	\
2962306a36Sopenharmony_ci	 CAAM_CMD_SZ + CAAM_PTR_SZ_MAX +				\
3062306a36Sopenharmony_ci	/* Command describing the operation to perform */		\
3162306a36Sopenharmony_ci	 CAAM_CMD_SZ)
3262306a36Sopenharmony_ci
3362306a36Sopenharmony_cistruct caam_blob_priv {
3462306a36Sopenharmony_ci	struct device jrdev;
3562306a36Sopenharmony_ci};
3662306a36Sopenharmony_ci
3762306a36Sopenharmony_cistruct caam_blob_job_result {
3862306a36Sopenharmony_ci	int err;
3962306a36Sopenharmony_ci	struct completion completion;
4062306a36Sopenharmony_ci};
4162306a36Sopenharmony_ci
4262306a36Sopenharmony_cistatic void caam_blob_job_done(struct device *dev, u32 *desc, u32 err, void *context)
4362306a36Sopenharmony_ci{
4462306a36Sopenharmony_ci	struct caam_blob_job_result *res = context;
4562306a36Sopenharmony_ci	int ecode = 0;
4662306a36Sopenharmony_ci
4762306a36Sopenharmony_ci	dev_dbg(dev, "%s %d: err 0x%x\n", __func__, __LINE__, err);
4862306a36Sopenharmony_ci
4962306a36Sopenharmony_ci	if (err)
5062306a36Sopenharmony_ci		ecode = caam_jr_strstatus(dev, err);
5162306a36Sopenharmony_ci
5262306a36Sopenharmony_ci	res->err = ecode;
5362306a36Sopenharmony_ci
5462306a36Sopenharmony_ci	/*
5562306a36Sopenharmony_ci	 * Upon completion, desc points to a buffer containing a CAAM job
5662306a36Sopenharmony_ci	 * descriptor which encapsulates data into an externally-storable
5762306a36Sopenharmony_ci	 * blob.
5862306a36Sopenharmony_ci	 */
5962306a36Sopenharmony_ci	complete(&res->completion);
6062306a36Sopenharmony_ci}
6162306a36Sopenharmony_ci
6262306a36Sopenharmony_ciint caam_process_blob(struct caam_blob_priv *priv,
6362306a36Sopenharmony_ci		      struct caam_blob_info *info, bool encap)
6462306a36Sopenharmony_ci{
6562306a36Sopenharmony_ci	const struct caam_drv_private *ctrlpriv;
6662306a36Sopenharmony_ci	struct caam_blob_job_result testres;
6762306a36Sopenharmony_ci	struct device *jrdev = &priv->jrdev;
6862306a36Sopenharmony_ci	dma_addr_t dma_in, dma_out;
6962306a36Sopenharmony_ci	int op = OP_PCLID_BLOB;
7062306a36Sopenharmony_ci	size_t output_len;
7162306a36Sopenharmony_ci	u32 *desc;
7262306a36Sopenharmony_ci	u32 moo;
7362306a36Sopenharmony_ci	int ret;
7462306a36Sopenharmony_ci
7562306a36Sopenharmony_ci	if (info->key_mod_len > CAAM_BLOB_KEYMOD_LENGTH)
7662306a36Sopenharmony_ci		return -EINVAL;
7762306a36Sopenharmony_ci
7862306a36Sopenharmony_ci	if (encap) {
7962306a36Sopenharmony_ci		op |= OP_TYPE_ENCAP_PROTOCOL;
8062306a36Sopenharmony_ci		output_len = info->input_len + CAAM_BLOB_OVERHEAD;
8162306a36Sopenharmony_ci	} else {
8262306a36Sopenharmony_ci		op |= OP_TYPE_DECAP_PROTOCOL;
8362306a36Sopenharmony_ci		output_len = info->input_len - CAAM_BLOB_OVERHEAD;
8462306a36Sopenharmony_ci	}
8562306a36Sopenharmony_ci
8662306a36Sopenharmony_ci	desc = kzalloc(CAAM_BLOB_DESC_BYTES_MAX, GFP_KERNEL);
8762306a36Sopenharmony_ci	if (!desc)
8862306a36Sopenharmony_ci		return -ENOMEM;
8962306a36Sopenharmony_ci
9062306a36Sopenharmony_ci	dma_in = dma_map_single(jrdev, info->input, info->input_len,
9162306a36Sopenharmony_ci				DMA_TO_DEVICE);
9262306a36Sopenharmony_ci	if (dma_mapping_error(jrdev, dma_in)) {
9362306a36Sopenharmony_ci		dev_err(jrdev, "unable to map input DMA buffer\n");
9462306a36Sopenharmony_ci		ret = -ENOMEM;
9562306a36Sopenharmony_ci		goto out_free;
9662306a36Sopenharmony_ci	}
9762306a36Sopenharmony_ci
9862306a36Sopenharmony_ci	dma_out = dma_map_single(jrdev, info->output, output_len,
9962306a36Sopenharmony_ci				 DMA_FROM_DEVICE);
10062306a36Sopenharmony_ci	if (dma_mapping_error(jrdev, dma_out)) {
10162306a36Sopenharmony_ci		dev_err(jrdev, "unable to map output DMA buffer\n");
10262306a36Sopenharmony_ci		ret = -ENOMEM;
10362306a36Sopenharmony_ci		goto out_unmap_in;
10462306a36Sopenharmony_ci	}
10562306a36Sopenharmony_ci
10662306a36Sopenharmony_ci	ctrlpriv = dev_get_drvdata(jrdev->parent);
10762306a36Sopenharmony_ci	moo = FIELD_GET(CSTA_MOO, rd_reg32(&ctrlpriv->ctrl->perfmon.status));
10862306a36Sopenharmony_ci	if (moo != CSTA_MOO_SECURE && moo != CSTA_MOO_TRUSTED)
10962306a36Sopenharmony_ci		dev_warn(jrdev,
11062306a36Sopenharmony_ci			 "using insecure test key, enable HAB to use unique device key!\n");
11162306a36Sopenharmony_ci
11262306a36Sopenharmony_ci	/*
11362306a36Sopenharmony_ci	 * A data blob is encrypted using a blob key (BK); a random number.
11462306a36Sopenharmony_ci	 * The BK is used as an AES-CCM key. The initial block (B0) and the
11562306a36Sopenharmony_ci	 * initial counter (Ctr0) are generated automatically and stored in
11662306a36Sopenharmony_ci	 * Class 1 Context DWords 0+1+2+3. The random BK is stored in the
11762306a36Sopenharmony_ci	 * Class 1 Key Register. Operation Mode is set to AES-CCM.
11862306a36Sopenharmony_ci	 */
11962306a36Sopenharmony_ci
12062306a36Sopenharmony_ci	init_job_desc(desc, 0);
12162306a36Sopenharmony_ci	append_key_as_imm(desc, info->key_mod, info->key_mod_len,
12262306a36Sopenharmony_ci			  info->key_mod_len, CLASS_2 | KEY_DEST_CLASS_REG);
12362306a36Sopenharmony_ci	append_seq_in_ptr_intlen(desc, dma_in, info->input_len, 0);
12462306a36Sopenharmony_ci	append_seq_out_ptr_intlen(desc, dma_out, output_len, 0);
12562306a36Sopenharmony_ci	append_operation(desc, op);
12662306a36Sopenharmony_ci
12762306a36Sopenharmony_ci	print_hex_dump_debug("data@"__stringify(__LINE__)": ",
12862306a36Sopenharmony_ci			     DUMP_PREFIX_ADDRESS, 16, 1, info->input,
12962306a36Sopenharmony_ci			     info->input_len, false);
13062306a36Sopenharmony_ci	print_hex_dump_debug("jobdesc@"__stringify(__LINE__)": ",
13162306a36Sopenharmony_ci			     DUMP_PREFIX_ADDRESS, 16, 1, desc,
13262306a36Sopenharmony_ci			     desc_bytes(desc), false);
13362306a36Sopenharmony_ci
13462306a36Sopenharmony_ci	testres.err = 0;
13562306a36Sopenharmony_ci	init_completion(&testres.completion);
13662306a36Sopenharmony_ci
13762306a36Sopenharmony_ci	ret = caam_jr_enqueue(jrdev, desc, caam_blob_job_done, &testres);
13862306a36Sopenharmony_ci	if (ret == -EINPROGRESS) {
13962306a36Sopenharmony_ci		wait_for_completion(&testres.completion);
14062306a36Sopenharmony_ci		ret = testres.err;
14162306a36Sopenharmony_ci		print_hex_dump_debug("output@"__stringify(__LINE__)": ",
14262306a36Sopenharmony_ci				     DUMP_PREFIX_ADDRESS, 16, 1, info->output,
14362306a36Sopenharmony_ci				     output_len, false);
14462306a36Sopenharmony_ci	}
14562306a36Sopenharmony_ci
14662306a36Sopenharmony_ci	if (ret == 0)
14762306a36Sopenharmony_ci		info->output_len = output_len;
14862306a36Sopenharmony_ci
14962306a36Sopenharmony_ci	dma_unmap_single(jrdev, dma_out, output_len, DMA_FROM_DEVICE);
15062306a36Sopenharmony_ciout_unmap_in:
15162306a36Sopenharmony_ci	dma_unmap_single(jrdev, dma_in, info->input_len, DMA_TO_DEVICE);
15262306a36Sopenharmony_ciout_free:
15362306a36Sopenharmony_ci	kfree(desc);
15462306a36Sopenharmony_ci
15562306a36Sopenharmony_ci	return ret;
15662306a36Sopenharmony_ci}
15762306a36Sopenharmony_ciEXPORT_SYMBOL(caam_process_blob);
15862306a36Sopenharmony_ci
15962306a36Sopenharmony_cistruct caam_blob_priv *caam_blob_gen_init(void)
16062306a36Sopenharmony_ci{
16162306a36Sopenharmony_ci	struct caam_drv_private *ctrlpriv;
16262306a36Sopenharmony_ci	struct device *jrdev;
16362306a36Sopenharmony_ci
16462306a36Sopenharmony_ci	/*
16562306a36Sopenharmony_ci	 * caam_blob_gen_init() may expectedly fail with -ENODEV, e.g. when
16662306a36Sopenharmony_ci	 * CAAM driver didn't probe or when SoC lacks BLOB support. An
16762306a36Sopenharmony_ci	 * error would be harsh in this case, so we stick to info level.
16862306a36Sopenharmony_ci	 */
16962306a36Sopenharmony_ci
17062306a36Sopenharmony_ci	jrdev = caam_jr_alloc();
17162306a36Sopenharmony_ci	if (IS_ERR(jrdev)) {
17262306a36Sopenharmony_ci		pr_info("job ring requested, but none currently available\n");
17362306a36Sopenharmony_ci		return ERR_PTR(-ENODEV);
17462306a36Sopenharmony_ci	}
17562306a36Sopenharmony_ci
17662306a36Sopenharmony_ci	ctrlpriv = dev_get_drvdata(jrdev->parent);
17762306a36Sopenharmony_ci	if (!ctrlpriv->blob_present) {
17862306a36Sopenharmony_ci		dev_info(jrdev, "no hardware blob generation support\n");
17962306a36Sopenharmony_ci		caam_jr_free(jrdev);
18062306a36Sopenharmony_ci		return ERR_PTR(-ENODEV);
18162306a36Sopenharmony_ci	}
18262306a36Sopenharmony_ci
18362306a36Sopenharmony_ci	return container_of(jrdev, struct caam_blob_priv, jrdev);
18462306a36Sopenharmony_ci}
18562306a36Sopenharmony_ciEXPORT_SYMBOL(caam_blob_gen_init);
18662306a36Sopenharmony_ci
18762306a36Sopenharmony_civoid caam_blob_gen_exit(struct caam_blob_priv *priv)
18862306a36Sopenharmony_ci{
18962306a36Sopenharmony_ci	caam_jr_free(&priv->jrdev);
19062306a36Sopenharmony_ci}
19162306a36Sopenharmony_ciEXPORT_SYMBOL(caam_blob_gen_exit);
192