162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * AMD Platform Security Processor (PSP) interface
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * Copyright (C) 2016,2019 Advanced Micro Devices, Inc.
662306a36Sopenharmony_ci *
762306a36Sopenharmony_ci * Author: Brijesh Singh <brijesh.singh@amd.com>
862306a36Sopenharmony_ci */
962306a36Sopenharmony_ci
1062306a36Sopenharmony_ci#include <linux/kernel.h>
1162306a36Sopenharmony_ci#include <linux/irqreturn.h>
1262306a36Sopenharmony_ci
1362306a36Sopenharmony_ci#include "sp-dev.h"
1462306a36Sopenharmony_ci#include "psp-dev.h"
1562306a36Sopenharmony_ci#include "sev-dev.h"
1662306a36Sopenharmony_ci#include "tee-dev.h"
1762306a36Sopenharmony_ci#include "platform-access.h"
1862306a36Sopenharmony_ci#include "dbc.h"
1962306a36Sopenharmony_ci
2062306a36Sopenharmony_cistruct psp_device *psp_master;
2162306a36Sopenharmony_ci
2262306a36Sopenharmony_cistatic struct psp_device *psp_alloc_struct(struct sp_device *sp)
2362306a36Sopenharmony_ci{
2462306a36Sopenharmony_ci	struct device *dev = sp->dev;
2562306a36Sopenharmony_ci	struct psp_device *psp;
2662306a36Sopenharmony_ci
2762306a36Sopenharmony_ci	psp = devm_kzalloc(dev, sizeof(*psp), GFP_KERNEL);
2862306a36Sopenharmony_ci	if (!psp)
2962306a36Sopenharmony_ci		return NULL;
3062306a36Sopenharmony_ci
3162306a36Sopenharmony_ci	psp->dev = dev;
3262306a36Sopenharmony_ci	psp->sp = sp;
3362306a36Sopenharmony_ci
3462306a36Sopenharmony_ci	snprintf(psp->name, sizeof(psp->name), "psp-%u", sp->ord);
3562306a36Sopenharmony_ci
3662306a36Sopenharmony_ci	return psp;
3762306a36Sopenharmony_ci}
3862306a36Sopenharmony_ci
3962306a36Sopenharmony_cistatic irqreturn_t psp_irq_handler(int irq, void *data)
4062306a36Sopenharmony_ci{
4162306a36Sopenharmony_ci	struct psp_device *psp = data;
4262306a36Sopenharmony_ci	unsigned int status;
4362306a36Sopenharmony_ci
4462306a36Sopenharmony_ci	/* Read the interrupt status: */
4562306a36Sopenharmony_ci	status = ioread32(psp->io_regs + psp->vdata->intsts_reg);
4662306a36Sopenharmony_ci
4762306a36Sopenharmony_ci	/* Clear the interrupt status by writing the same value we read. */
4862306a36Sopenharmony_ci	iowrite32(status, psp->io_regs + psp->vdata->intsts_reg);
4962306a36Sopenharmony_ci
5062306a36Sopenharmony_ci	/* invoke subdevice interrupt handlers */
5162306a36Sopenharmony_ci	if (status) {
5262306a36Sopenharmony_ci		if (psp->sev_irq_handler)
5362306a36Sopenharmony_ci			psp->sev_irq_handler(irq, psp->sev_irq_data, status);
5462306a36Sopenharmony_ci	}
5562306a36Sopenharmony_ci
5662306a36Sopenharmony_ci	return IRQ_HANDLED;
5762306a36Sopenharmony_ci}
5862306a36Sopenharmony_ci
5962306a36Sopenharmony_cistatic unsigned int psp_get_capability(struct psp_device *psp)
6062306a36Sopenharmony_ci{
6162306a36Sopenharmony_ci	unsigned int val = ioread32(psp->io_regs + psp->vdata->feature_reg);
6262306a36Sopenharmony_ci
6362306a36Sopenharmony_ci	/*
6462306a36Sopenharmony_ci	 * Check for a access to the registers.  If this read returns
6562306a36Sopenharmony_ci	 * 0xffffffff, it's likely that the system is running a broken
6662306a36Sopenharmony_ci	 * BIOS which disallows access to the device. Stop here and
6762306a36Sopenharmony_ci	 * fail the PSP initialization (but not the load, as the CCP
6862306a36Sopenharmony_ci	 * could get properly initialized).
6962306a36Sopenharmony_ci	 */
7062306a36Sopenharmony_ci	if (val == 0xffffffff) {
7162306a36Sopenharmony_ci		dev_notice(psp->dev, "psp: unable to access the device: you might be running a broken BIOS.\n");
7262306a36Sopenharmony_ci		return -ENODEV;
7362306a36Sopenharmony_ci	}
7462306a36Sopenharmony_ci	psp->capability = val;
7562306a36Sopenharmony_ci
7662306a36Sopenharmony_ci	/* Detect if TSME and SME are both enabled */
7762306a36Sopenharmony_ci	if (psp->capability & PSP_CAPABILITY_PSP_SECURITY_REPORTING &&
7862306a36Sopenharmony_ci	    psp->capability & (PSP_SECURITY_TSME_STATUS << PSP_CAPABILITY_PSP_SECURITY_OFFSET) &&
7962306a36Sopenharmony_ci	    cc_platform_has(CC_ATTR_HOST_MEM_ENCRYPT))
8062306a36Sopenharmony_ci		dev_notice(psp->dev, "psp: Both TSME and SME are active, SME is unnecessary when TSME is active.\n");
8162306a36Sopenharmony_ci
8262306a36Sopenharmony_ci	return 0;
8362306a36Sopenharmony_ci}
8462306a36Sopenharmony_ci
8562306a36Sopenharmony_cistatic int psp_check_sev_support(struct psp_device *psp)
8662306a36Sopenharmony_ci{
8762306a36Sopenharmony_ci	/* Check if device supports SEV feature */
8862306a36Sopenharmony_ci	if (!(psp->capability & PSP_CAPABILITY_SEV)) {
8962306a36Sopenharmony_ci		dev_dbg(psp->dev, "psp does not support SEV\n");
9062306a36Sopenharmony_ci		return -ENODEV;
9162306a36Sopenharmony_ci	}
9262306a36Sopenharmony_ci
9362306a36Sopenharmony_ci	return 0;
9462306a36Sopenharmony_ci}
9562306a36Sopenharmony_ci
9662306a36Sopenharmony_cistatic int psp_check_tee_support(struct psp_device *psp)
9762306a36Sopenharmony_ci{
9862306a36Sopenharmony_ci	/* Check if device supports TEE feature */
9962306a36Sopenharmony_ci	if (!(psp->capability & PSP_CAPABILITY_TEE)) {
10062306a36Sopenharmony_ci		dev_dbg(psp->dev, "psp does not support TEE\n");
10162306a36Sopenharmony_ci		return -ENODEV;
10262306a36Sopenharmony_ci	}
10362306a36Sopenharmony_ci
10462306a36Sopenharmony_ci	return 0;
10562306a36Sopenharmony_ci}
10662306a36Sopenharmony_ci
10762306a36Sopenharmony_cistatic void psp_init_platform_access(struct psp_device *psp)
10862306a36Sopenharmony_ci{
10962306a36Sopenharmony_ci	int ret;
11062306a36Sopenharmony_ci
11162306a36Sopenharmony_ci	ret = platform_access_dev_init(psp);
11262306a36Sopenharmony_ci	if (ret) {
11362306a36Sopenharmony_ci		dev_warn(psp->dev, "platform access init failed: %d\n", ret);
11462306a36Sopenharmony_ci		return;
11562306a36Sopenharmony_ci	}
11662306a36Sopenharmony_ci
11762306a36Sopenharmony_ci	/* dbc must come after platform access as it tests the feature */
11862306a36Sopenharmony_ci	ret = dbc_dev_init(psp);
11962306a36Sopenharmony_ci	if (ret)
12062306a36Sopenharmony_ci		dev_warn(psp->dev, "failed to init dynamic boost control: %d\n",
12162306a36Sopenharmony_ci			 ret);
12262306a36Sopenharmony_ci}
12362306a36Sopenharmony_ci
12462306a36Sopenharmony_cistatic int psp_init(struct psp_device *psp)
12562306a36Sopenharmony_ci{
12662306a36Sopenharmony_ci	int ret;
12762306a36Sopenharmony_ci
12862306a36Sopenharmony_ci	if (!psp_check_sev_support(psp)) {
12962306a36Sopenharmony_ci		ret = sev_dev_init(psp);
13062306a36Sopenharmony_ci		if (ret)
13162306a36Sopenharmony_ci			return ret;
13262306a36Sopenharmony_ci	}
13362306a36Sopenharmony_ci
13462306a36Sopenharmony_ci	if (!psp_check_tee_support(psp)) {
13562306a36Sopenharmony_ci		ret = tee_dev_init(psp);
13662306a36Sopenharmony_ci		if (ret)
13762306a36Sopenharmony_ci			return ret;
13862306a36Sopenharmony_ci	}
13962306a36Sopenharmony_ci
14062306a36Sopenharmony_ci	if (psp->vdata->platform_access)
14162306a36Sopenharmony_ci		psp_init_platform_access(psp);
14262306a36Sopenharmony_ci
14362306a36Sopenharmony_ci	return 0;
14462306a36Sopenharmony_ci}
14562306a36Sopenharmony_ci
14662306a36Sopenharmony_ciint psp_dev_init(struct sp_device *sp)
14762306a36Sopenharmony_ci{
14862306a36Sopenharmony_ci	struct device *dev = sp->dev;
14962306a36Sopenharmony_ci	struct psp_device *psp;
15062306a36Sopenharmony_ci	int ret;
15162306a36Sopenharmony_ci
15262306a36Sopenharmony_ci	ret = -ENOMEM;
15362306a36Sopenharmony_ci	psp = psp_alloc_struct(sp);
15462306a36Sopenharmony_ci	if (!psp)
15562306a36Sopenharmony_ci		goto e_err;
15662306a36Sopenharmony_ci
15762306a36Sopenharmony_ci	sp->psp_data = psp;
15862306a36Sopenharmony_ci
15962306a36Sopenharmony_ci	psp->vdata = (struct psp_vdata *)sp->dev_vdata->psp_vdata;
16062306a36Sopenharmony_ci	if (!psp->vdata) {
16162306a36Sopenharmony_ci		ret = -ENODEV;
16262306a36Sopenharmony_ci		dev_err(dev, "missing driver data\n");
16362306a36Sopenharmony_ci		goto e_err;
16462306a36Sopenharmony_ci	}
16562306a36Sopenharmony_ci
16662306a36Sopenharmony_ci	psp->io_regs = sp->io_map;
16762306a36Sopenharmony_ci
16862306a36Sopenharmony_ci	ret = psp_get_capability(psp);
16962306a36Sopenharmony_ci	if (ret)
17062306a36Sopenharmony_ci		goto e_disable;
17162306a36Sopenharmony_ci
17262306a36Sopenharmony_ci	/* Disable and clear interrupts until ready */
17362306a36Sopenharmony_ci	iowrite32(0, psp->io_regs + psp->vdata->inten_reg);
17462306a36Sopenharmony_ci	iowrite32(-1, psp->io_regs + psp->vdata->intsts_reg);
17562306a36Sopenharmony_ci
17662306a36Sopenharmony_ci	/* Request an irq */
17762306a36Sopenharmony_ci	ret = sp_request_psp_irq(psp->sp, psp_irq_handler, psp->name, psp);
17862306a36Sopenharmony_ci	if (ret) {
17962306a36Sopenharmony_ci		dev_err(dev, "psp: unable to allocate an IRQ\n");
18062306a36Sopenharmony_ci		goto e_err;
18162306a36Sopenharmony_ci	}
18262306a36Sopenharmony_ci
18362306a36Sopenharmony_ci	/* master device must be set for platform access */
18462306a36Sopenharmony_ci	if (psp->sp->set_psp_master_device)
18562306a36Sopenharmony_ci		psp->sp->set_psp_master_device(psp->sp);
18662306a36Sopenharmony_ci
18762306a36Sopenharmony_ci	ret = psp_init(psp);
18862306a36Sopenharmony_ci	if (ret)
18962306a36Sopenharmony_ci		goto e_irq;
19062306a36Sopenharmony_ci
19162306a36Sopenharmony_ci	/* Enable interrupt */
19262306a36Sopenharmony_ci	iowrite32(-1, psp->io_regs + psp->vdata->inten_reg);
19362306a36Sopenharmony_ci
19462306a36Sopenharmony_ci	dev_notice(dev, "psp enabled\n");
19562306a36Sopenharmony_ci
19662306a36Sopenharmony_ci	return 0;
19762306a36Sopenharmony_ci
19862306a36Sopenharmony_cie_irq:
19962306a36Sopenharmony_ci	if (sp->clear_psp_master_device)
20062306a36Sopenharmony_ci		sp->clear_psp_master_device(sp);
20162306a36Sopenharmony_ci
20262306a36Sopenharmony_ci	sp_free_psp_irq(psp->sp, psp);
20362306a36Sopenharmony_cie_err:
20462306a36Sopenharmony_ci	sp->psp_data = NULL;
20562306a36Sopenharmony_ci
20662306a36Sopenharmony_ci	dev_notice(dev, "psp initialization failed\n");
20762306a36Sopenharmony_ci
20862306a36Sopenharmony_ci	return ret;
20962306a36Sopenharmony_ci
21062306a36Sopenharmony_cie_disable:
21162306a36Sopenharmony_ci	sp->psp_data = NULL;
21262306a36Sopenharmony_ci
21362306a36Sopenharmony_ci	return ret;
21462306a36Sopenharmony_ci}
21562306a36Sopenharmony_ci
21662306a36Sopenharmony_civoid psp_dev_destroy(struct sp_device *sp)
21762306a36Sopenharmony_ci{
21862306a36Sopenharmony_ci	struct psp_device *psp = sp->psp_data;
21962306a36Sopenharmony_ci
22062306a36Sopenharmony_ci	if (!psp)
22162306a36Sopenharmony_ci		return;
22262306a36Sopenharmony_ci
22362306a36Sopenharmony_ci	sev_dev_destroy(psp);
22462306a36Sopenharmony_ci
22562306a36Sopenharmony_ci	tee_dev_destroy(psp);
22662306a36Sopenharmony_ci
22762306a36Sopenharmony_ci	dbc_dev_destroy(psp);
22862306a36Sopenharmony_ci
22962306a36Sopenharmony_ci	platform_access_dev_destroy(psp);
23062306a36Sopenharmony_ci
23162306a36Sopenharmony_ci	sp_free_psp_irq(sp, psp);
23262306a36Sopenharmony_ci
23362306a36Sopenharmony_ci	if (sp->clear_psp_master_device)
23462306a36Sopenharmony_ci		sp->clear_psp_master_device(sp);
23562306a36Sopenharmony_ci}
23662306a36Sopenharmony_ci
23762306a36Sopenharmony_civoid psp_set_sev_irq_handler(struct psp_device *psp, psp_irq_handler_t handler,
23862306a36Sopenharmony_ci			     void *data)
23962306a36Sopenharmony_ci{
24062306a36Sopenharmony_ci	psp->sev_irq_data = data;
24162306a36Sopenharmony_ci	psp->sev_irq_handler = handler;
24262306a36Sopenharmony_ci}
24362306a36Sopenharmony_ci
24462306a36Sopenharmony_civoid psp_clear_sev_irq_handler(struct psp_device *psp)
24562306a36Sopenharmony_ci{
24662306a36Sopenharmony_ci	psp_set_sev_irq_handler(psp, NULL, NULL);
24762306a36Sopenharmony_ci}
24862306a36Sopenharmony_ci
24962306a36Sopenharmony_cistruct psp_device *psp_get_master_device(void)
25062306a36Sopenharmony_ci{
25162306a36Sopenharmony_ci	struct sp_device *sp = sp_get_psp_master_device();
25262306a36Sopenharmony_ci
25362306a36Sopenharmony_ci	return sp ? sp->psp_data : NULL;
25462306a36Sopenharmony_ci}
25562306a36Sopenharmony_ci
25662306a36Sopenharmony_civoid psp_pci_init(void)
25762306a36Sopenharmony_ci{
25862306a36Sopenharmony_ci	psp_master = psp_get_master_device();
25962306a36Sopenharmony_ci
26062306a36Sopenharmony_ci	if (!psp_master)
26162306a36Sopenharmony_ci		return;
26262306a36Sopenharmony_ci
26362306a36Sopenharmony_ci	sev_pci_init();
26462306a36Sopenharmony_ci}
26562306a36Sopenharmony_ci
26662306a36Sopenharmony_civoid psp_pci_exit(void)
26762306a36Sopenharmony_ci{
26862306a36Sopenharmony_ci	if (!psp_master)
26962306a36Sopenharmony_ci		return;
27062306a36Sopenharmony_ci
27162306a36Sopenharmony_ci	sev_pci_exit();
27262306a36Sopenharmony_ci}
273