2450 lines · c
1// SPDX-License-Identifier: GPL-2.0-only2/*3 * AMD Secure Encrypted Virtualization (SEV) interface4 *5 * Copyright (C) 2016,2019 Advanced Micro Devices, Inc.6 *7 * Author: Brijesh Singh <brijesh.singh@amd.com>8 */9 10#include <linux/bitfield.h>11#include <linux/module.h>12#include <linux/kernel.h>13#include <linux/kthread.h>14#include <linux/sched.h>15#include <linux/interrupt.h>16#include <linux/spinlock.h>17#include <linux/spinlock_types.h>18#include <linux/types.h>19#include <linux/mutex.h>20#include <linux/delay.h>21#include <linux/hw_random.h>22#include <linux/ccp.h>23#include <linux/firmware.h>24#include <linux/panic_notifier.h>25#include <linux/gfp.h>26#include <linux/cpufeature.h>27#include <linux/fs.h>28#include <linux/fs_struct.h>29#include <linux/psp.h>30#include <linux/amd-iommu.h>31 32#include <asm/smp.h>33#include <asm/cacheflush.h>34#include <asm/e820/types.h>35#include <asm/sev.h>36 37#include "psp-dev.h"38#include "sev-dev.h"39 40#define DEVICE_NAME "sev"41#define SEV_FW_FILE "amd/sev.fw"42#define SEV_FW_NAME_SIZE 6443 44/* Minimum firmware version required for the SEV-SNP support */45#define SNP_MIN_API_MAJOR 146#define SNP_MIN_API_MINOR 5147 48/*49 * Maximum number of firmware-writable buffers that might be specified50 * in the parameters of a legacy SEV command buffer.51 */52#define CMD_BUF_FW_WRITABLE_MAX 253 54/* Leave room in the descriptor array for an end-of-list indicator. */55#define CMD_BUF_DESC_MAX (CMD_BUF_FW_WRITABLE_MAX + 1)56 57static DEFINE_MUTEX(sev_cmd_mutex);58static struct sev_misc_dev *misc_dev;59 60static int psp_cmd_timeout = 100;61module_param(psp_cmd_timeout, int, 0644);62MODULE_PARM_DESC(psp_cmd_timeout, " default timeout value, in seconds, for PSP commands");63 64static int psp_probe_timeout = 5;65module_param(psp_probe_timeout, int, 0644);66MODULE_PARM_DESC(psp_probe_timeout, " default timeout value, in seconds, during PSP device probe");67 68static char *init_ex_path;69module_param(init_ex_path, charp, 0444);70MODULE_PARM_DESC(init_ex_path, " Path for INIT_EX data; if set try INIT_EX");71 72static bool psp_init_on_probe = true;73module_param(psp_init_on_probe, bool, 0444);74MODULE_PARM_DESC(psp_init_on_probe, " if true, the PSP will be initialized on module init. Else the PSP will be initialized on the first command requiring it");75 76MODULE_FIRMWARE("amd/amd_sev_fam17h_model0xh.sbin"); /* 1st gen EPYC */77MODULE_FIRMWARE("amd/amd_sev_fam17h_model3xh.sbin"); /* 2nd gen EPYC */78MODULE_FIRMWARE("amd/amd_sev_fam19h_model0xh.sbin"); /* 3rd gen EPYC */79MODULE_FIRMWARE("amd/amd_sev_fam19h_model1xh.sbin"); /* 4th gen EPYC */80 81static bool psp_dead;82static int psp_timeout;83 84/* Trusted Memory Region (TMR):85 * The TMR is a 1MB area that must be 1MB aligned. Use the page allocator86 * to allocate the memory, which will return aligned memory for the specified87 * allocation order.88 *89 * When SEV-SNP is enabled the TMR needs to be 2MB aligned and 2MB sized.90 */91#define SEV_TMR_SIZE (1024 * 1024)92#define SNP_TMR_SIZE (2 * 1024 * 1024)93 94static void *sev_es_tmr;95static size_t sev_es_tmr_size = SEV_TMR_SIZE;96 97/* INIT_EX NV Storage:98 * The NV Storage is a 32Kb area and must be 4Kb page aligned. Use the page99 * allocator to allocate the memory, which will return aligned memory for the100 * specified allocation order.101 */102#define NV_LENGTH (32 * 1024)103static void *sev_init_ex_buffer;104 105/*106 * SEV_DATA_RANGE_LIST:107 * Array containing range of pages that firmware transitions to HV-fixed108 * page state.109 */110static struct sev_data_range_list *snp_range_list;111 112static inline bool sev_version_greater_or_equal(u8 maj, u8 min)113{114 struct sev_device *sev = psp_master->sev_data;115 116 if (sev->api_major > maj)117 return true;118 119 if (sev->api_major == maj && sev->api_minor >= min)120 return true;121 122 return false;123}124 125static void sev_irq_handler(int irq, void *data, unsigned int status)126{127 struct sev_device *sev = data;128 int reg;129 130 /* Check if it is command completion: */131 if (!(status & SEV_CMD_COMPLETE))132 return;133 134 /* Check if it is SEV command completion: */135 reg = ioread32(sev->io_regs + sev->vdata->cmdresp_reg);136 if (FIELD_GET(PSP_CMDRESP_RESP, reg)) {137 sev->int_rcvd = 1;138 wake_up(&sev->int_queue);139 }140}141 142static int sev_wait_cmd_ioc(struct sev_device *sev,143 unsigned int *reg, unsigned int timeout)144{145 int ret;146 147 /*148 * If invoked during panic handling, local interrupts are disabled,149 * so the PSP command completion interrupt can't be used. Poll for150 * PSP command completion instead.151 */152 if (irqs_disabled()) {153 unsigned long timeout_usecs = (timeout * USEC_PER_SEC) / 10;154 155 /* Poll for SEV command completion: */156 while (timeout_usecs--) {157 *reg = ioread32(sev->io_regs + sev->vdata->cmdresp_reg);158 if (*reg & PSP_CMDRESP_RESP)159 return 0;160 161 udelay(10);162 }163 return -ETIMEDOUT;164 }165 166 ret = wait_event_timeout(sev->int_queue,167 sev->int_rcvd, timeout * HZ);168 if (!ret)169 return -ETIMEDOUT;170 171 *reg = ioread32(sev->io_regs + sev->vdata->cmdresp_reg);172 173 return 0;174}175 176static int sev_cmd_buffer_len(int cmd)177{178 switch (cmd) {179 case SEV_CMD_INIT: return sizeof(struct sev_data_init);180 case SEV_CMD_INIT_EX: return sizeof(struct sev_data_init_ex);181 case SEV_CMD_SNP_SHUTDOWN_EX: return sizeof(struct sev_data_snp_shutdown_ex);182 case SEV_CMD_SNP_INIT_EX: return sizeof(struct sev_data_snp_init_ex);183 case SEV_CMD_PLATFORM_STATUS: return sizeof(struct sev_user_data_status);184 case SEV_CMD_PEK_CSR: return sizeof(struct sev_data_pek_csr);185 case SEV_CMD_PEK_CERT_IMPORT: return sizeof(struct sev_data_pek_cert_import);186 case SEV_CMD_PDH_CERT_EXPORT: return sizeof(struct sev_data_pdh_cert_export);187 case SEV_CMD_LAUNCH_START: return sizeof(struct sev_data_launch_start);188 case SEV_CMD_LAUNCH_UPDATE_DATA: return sizeof(struct sev_data_launch_update_data);189 case SEV_CMD_LAUNCH_UPDATE_VMSA: return sizeof(struct sev_data_launch_update_vmsa);190 case SEV_CMD_LAUNCH_FINISH: return sizeof(struct sev_data_launch_finish);191 case SEV_CMD_LAUNCH_MEASURE: return sizeof(struct sev_data_launch_measure);192 case SEV_CMD_ACTIVATE: return sizeof(struct sev_data_activate);193 case SEV_CMD_DEACTIVATE: return sizeof(struct sev_data_deactivate);194 case SEV_CMD_DECOMMISSION: return sizeof(struct sev_data_decommission);195 case SEV_CMD_GUEST_STATUS: return sizeof(struct sev_data_guest_status);196 case SEV_CMD_DBG_DECRYPT: return sizeof(struct sev_data_dbg);197 case SEV_CMD_DBG_ENCRYPT: return sizeof(struct sev_data_dbg);198 case SEV_CMD_SEND_START: return sizeof(struct sev_data_send_start);199 case SEV_CMD_SEND_UPDATE_DATA: return sizeof(struct sev_data_send_update_data);200 case SEV_CMD_SEND_UPDATE_VMSA: return sizeof(struct sev_data_send_update_vmsa);201 case SEV_CMD_SEND_FINISH: return sizeof(struct sev_data_send_finish);202 case SEV_CMD_RECEIVE_START: return sizeof(struct sev_data_receive_start);203 case SEV_CMD_RECEIVE_FINISH: return sizeof(struct sev_data_receive_finish);204 case SEV_CMD_RECEIVE_UPDATE_DATA: return sizeof(struct sev_data_receive_update_data);205 case SEV_CMD_RECEIVE_UPDATE_VMSA: return sizeof(struct sev_data_receive_update_vmsa);206 case SEV_CMD_LAUNCH_UPDATE_SECRET: return sizeof(struct sev_data_launch_secret);207 case SEV_CMD_DOWNLOAD_FIRMWARE: return sizeof(struct sev_data_download_firmware);208 case SEV_CMD_GET_ID: return sizeof(struct sev_data_get_id);209 case SEV_CMD_ATTESTATION_REPORT: return sizeof(struct sev_data_attestation_report);210 case SEV_CMD_SEND_CANCEL: return sizeof(struct sev_data_send_cancel);211 case SEV_CMD_SNP_GCTX_CREATE: return sizeof(struct sev_data_snp_addr);212 case SEV_CMD_SNP_LAUNCH_START: return sizeof(struct sev_data_snp_launch_start);213 case SEV_CMD_SNP_LAUNCH_UPDATE: return sizeof(struct sev_data_snp_launch_update);214 case SEV_CMD_SNP_ACTIVATE: return sizeof(struct sev_data_snp_activate);215 case SEV_CMD_SNP_DECOMMISSION: return sizeof(struct sev_data_snp_addr);216 case SEV_CMD_SNP_PAGE_RECLAIM: return sizeof(struct sev_data_snp_page_reclaim);217 case SEV_CMD_SNP_GUEST_STATUS: return sizeof(struct sev_data_snp_guest_status);218 case SEV_CMD_SNP_LAUNCH_FINISH: return sizeof(struct sev_data_snp_launch_finish);219 case SEV_CMD_SNP_DBG_DECRYPT: return sizeof(struct sev_data_snp_dbg);220 case SEV_CMD_SNP_DBG_ENCRYPT: return sizeof(struct sev_data_snp_dbg);221 case SEV_CMD_SNP_PAGE_UNSMASH: return sizeof(struct sev_data_snp_page_unsmash);222 case SEV_CMD_SNP_PLATFORM_STATUS: return sizeof(struct sev_data_snp_addr);223 case SEV_CMD_SNP_GUEST_REQUEST: return sizeof(struct sev_data_snp_guest_request);224 case SEV_CMD_SNP_CONFIG: return sizeof(struct sev_user_data_snp_config);225 case SEV_CMD_SNP_COMMIT: return sizeof(struct sev_data_snp_commit);226 default: return 0;227 }228 229 return 0;230}231 232static struct file *open_file_as_root(const char *filename, int flags, umode_t mode)233{234 struct file *fp;235 struct path root;236 struct cred *cred;237 const struct cred *old_cred;238 239 task_lock(&init_task);240 get_fs_root(init_task.fs, &root);241 task_unlock(&init_task);242 243 cred = prepare_creds();244 if (!cred)245 return ERR_PTR(-ENOMEM);246 cred->fsuid = GLOBAL_ROOT_UID;247 old_cred = override_creds(cred);248 249 fp = file_open_root(&root, filename, flags, mode);250 path_put(&root);251 252 revert_creds(old_cred);253 254 return fp;255}256 257static int sev_read_init_ex_file(void)258{259 struct sev_device *sev = psp_master->sev_data;260 struct file *fp;261 ssize_t nread;262 263 lockdep_assert_held(&sev_cmd_mutex);264 265 if (!sev_init_ex_buffer)266 return -EOPNOTSUPP;267 268 fp = open_file_as_root(init_ex_path, O_RDONLY, 0);269 if (IS_ERR(fp)) {270 int ret = PTR_ERR(fp);271 272 if (ret == -ENOENT) {273 dev_info(sev->dev,274 "SEV: %s does not exist and will be created later.\n",275 init_ex_path);276 ret = 0;277 } else {278 dev_err(sev->dev,279 "SEV: could not open %s for read, error %d\n",280 init_ex_path, ret);281 }282 return ret;283 }284 285 nread = kernel_read(fp, sev_init_ex_buffer, NV_LENGTH, NULL);286 if (nread != NV_LENGTH) {287 dev_info(sev->dev,288 "SEV: could not read %u bytes to non volatile memory area, ret %ld\n",289 NV_LENGTH, nread);290 }291 292 dev_dbg(sev->dev, "SEV: read %ld bytes from NV file\n", nread);293 filp_close(fp, NULL);294 295 return 0;296}297 298static int sev_write_init_ex_file(void)299{300 struct sev_device *sev = psp_master->sev_data;301 struct file *fp;302 loff_t offset = 0;303 ssize_t nwrite;304 305 lockdep_assert_held(&sev_cmd_mutex);306 307 if (!sev_init_ex_buffer)308 return 0;309 310 fp = open_file_as_root(init_ex_path, O_CREAT | O_WRONLY, 0600);311 if (IS_ERR(fp)) {312 int ret = PTR_ERR(fp);313 314 dev_err(sev->dev,315 "SEV: could not open file for write, error %d\n",316 ret);317 return ret;318 }319 320 nwrite = kernel_write(fp, sev_init_ex_buffer, NV_LENGTH, &offset);321 vfs_fsync(fp, 0);322 filp_close(fp, NULL);323 324 if (nwrite != NV_LENGTH) {325 dev_err(sev->dev,326 "SEV: failed to write %u bytes to non volatile memory area, ret %ld\n",327 NV_LENGTH, nwrite);328 return -EIO;329 }330 331 dev_dbg(sev->dev, "SEV: write successful to NV file\n");332 333 return 0;334}335 336static int sev_write_init_ex_file_if_required(int cmd_id)337{338 lockdep_assert_held(&sev_cmd_mutex);339 340 if (!sev_init_ex_buffer)341 return 0;342 343 /*344 * Only a few platform commands modify the SPI/NV area, but none of the345 * non-platform commands do. Only INIT(_EX), PLATFORM_RESET, PEK_GEN,346 * PEK_CERT_IMPORT, and PDH_GEN do.347 */348 switch (cmd_id) {349 case SEV_CMD_FACTORY_RESET:350 case SEV_CMD_INIT_EX:351 case SEV_CMD_PDH_GEN:352 case SEV_CMD_PEK_CERT_IMPORT:353 case SEV_CMD_PEK_GEN:354 break;355 default:356 return 0;357 }358 359 return sev_write_init_ex_file();360}361 362/*363 * snp_reclaim_pages() needs __sev_do_cmd_locked(), and __sev_do_cmd_locked()364 * needs snp_reclaim_pages(), so a forward declaration is needed.365 */366static int __sev_do_cmd_locked(int cmd, void *data, int *psp_ret);367 368static int snp_reclaim_pages(unsigned long paddr, unsigned int npages, bool locked)369{370 int ret, err, i;371 372 paddr = __sme_clr(ALIGN_DOWN(paddr, PAGE_SIZE));373 374 for (i = 0; i < npages; i++, paddr += PAGE_SIZE) {375 struct sev_data_snp_page_reclaim data = {0};376 377 data.paddr = paddr;378 379 if (locked)380 ret = __sev_do_cmd_locked(SEV_CMD_SNP_PAGE_RECLAIM, &data, &err);381 else382 ret = sev_do_cmd(SEV_CMD_SNP_PAGE_RECLAIM, &data, &err);383 384 if (ret)385 goto cleanup;386 387 ret = rmp_make_shared(__phys_to_pfn(paddr), PG_LEVEL_4K);388 if (ret)389 goto cleanup;390 }391 392 return 0;393 394cleanup:395 /*396 * If there was a failure reclaiming the page then it is no longer safe397 * to release it back to the system; leak it instead.398 */399 snp_leak_pages(__phys_to_pfn(paddr), npages - i);400 return ret;401}402 403static int rmp_mark_pages_firmware(unsigned long paddr, unsigned int npages, bool locked)404{405 unsigned long pfn = __sme_clr(paddr) >> PAGE_SHIFT;406 int rc, i;407 408 for (i = 0; i < npages; i++, pfn++) {409 rc = rmp_make_private(pfn, 0, PG_LEVEL_4K, 0, true);410 if (rc)411 goto cleanup;412 }413 414 return 0;415 416cleanup:417 /*418 * Try unrolling the firmware state changes by419 * reclaiming the pages which were already changed to the420 * firmware state.421 */422 snp_reclaim_pages(paddr, i, locked);423 424 return rc;425}426 427static struct page *__snp_alloc_firmware_pages(gfp_t gfp_mask, int order)428{429 unsigned long npages = 1ul << order, paddr;430 struct sev_device *sev;431 struct page *page;432 433 if (!psp_master || !psp_master->sev_data)434 return NULL;435 436 page = alloc_pages(gfp_mask, order);437 if (!page)438 return NULL;439 440 /* If SEV-SNP is initialized then add the page in RMP table. */441 sev = psp_master->sev_data;442 if (!sev->snp_initialized)443 return page;444 445 paddr = __pa((unsigned long)page_address(page));446 if (rmp_mark_pages_firmware(paddr, npages, false))447 return NULL;448 449 return page;450}451 452void *snp_alloc_firmware_page(gfp_t gfp_mask)453{454 struct page *page;455 456 page = __snp_alloc_firmware_pages(gfp_mask, 0);457 458 return page ? page_address(page) : NULL;459}460EXPORT_SYMBOL_GPL(snp_alloc_firmware_page);461 462static void __snp_free_firmware_pages(struct page *page, int order, bool locked)463{464 struct sev_device *sev = psp_master->sev_data;465 unsigned long paddr, npages = 1ul << order;466 467 if (!page)468 return;469 470 paddr = __pa((unsigned long)page_address(page));471 if (sev->snp_initialized &&472 snp_reclaim_pages(paddr, npages, locked))473 return;474 475 __free_pages(page, order);476}477 478void snp_free_firmware_page(void *addr)479{480 if (!addr)481 return;482 483 __snp_free_firmware_pages(virt_to_page(addr), 0, false);484}485EXPORT_SYMBOL_GPL(snp_free_firmware_page);486 487static void *sev_fw_alloc(unsigned long len)488{489 struct page *page;490 491 page = __snp_alloc_firmware_pages(GFP_KERNEL, get_order(len));492 if (!page)493 return NULL;494 495 return page_address(page);496}497 498/**499 * struct cmd_buf_desc - descriptors for managing legacy SEV command address500 * parameters corresponding to buffers that may be written to by firmware.501 *502 * @paddr_ptr: pointer to the address parameter in the command buffer which may503 * need to be saved/restored depending on whether a bounce buffer504 * is used. In the case of a bounce buffer, the command buffer505 * needs to be updated with the address of the new bounce buffer506 * snp_map_cmd_buf_desc() has allocated specifically for it. Must507 * be NULL if this descriptor is only an end-of-list indicator.508 *509 * @paddr_orig: storage for the original address parameter, which can be used to510 * restore the original value in @paddr_ptr in cases where it is511 * replaced with the address of a bounce buffer.512 *513 * @len: length of buffer located at the address originally stored at @paddr_ptr514 *515 * @guest_owned: true if the address corresponds to guest-owned pages, in which516 * case bounce buffers are not needed.517 */518struct cmd_buf_desc {519 u64 *paddr_ptr;520 u64 paddr_orig;521 u32 len;522 bool guest_owned;523};524 525/*526 * If a legacy SEV command parameter is a memory address, those pages in527 * turn need to be transitioned to/from firmware-owned before/after528 * executing the firmware command.529 *530 * Additionally, in cases where those pages are not guest-owned, a bounce531 * buffer is needed in place of the original memory address parameter.532 *533 * A set of descriptors are used to keep track of this handling, and534 * initialized here based on the specific commands being executed.535 */536static void snp_populate_cmd_buf_desc_list(int cmd, void *cmd_buf,537 struct cmd_buf_desc *desc_list)538{539 switch (cmd) {540 case SEV_CMD_PDH_CERT_EXPORT: {541 struct sev_data_pdh_cert_export *data = cmd_buf;542 543 desc_list[0].paddr_ptr = &data->pdh_cert_address;544 desc_list[0].len = data->pdh_cert_len;545 desc_list[1].paddr_ptr = &data->cert_chain_address;546 desc_list[1].len = data->cert_chain_len;547 break;548 }549 case SEV_CMD_GET_ID: {550 struct sev_data_get_id *data = cmd_buf;551 552 desc_list[0].paddr_ptr = &data->address;553 desc_list[0].len = data->len;554 break;555 }556 case SEV_CMD_PEK_CSR: {557 struct sev_data_pek_csr *data = cmd_buf;558 559 desc_list[0].paddr_ptr = &data->address;560 desc_list[0].len = data->len;561 break;562 }563 case SEV_CMD_LAUNCH_UPDATE_DATA: {564 struct sev_data_launch_update_data *data = cmd_buf;565 566 desc_list[0].paddr_ptr = &data->address;567 desc_list[0].len = data->len;568 desc_list[0].guest_owned = true;569 break;570 }571 case SEV_CMD_LAUNCH_UPDATE_VMSA: {572 struct sev_data_launch_update_vmsa *data = cmd_buf;573 574 desc_list[0].paddr_ptr = &data->address;575 desc_list[0].len = data->len;576 desc_list[0].guest_owned = true;577 break;578 }579 case SEV_CMD_LAUNCH_MEASURE: {580 struct sev_data_launch_measure *data = cmd_buf;581 582 desc_list[0].paddr_ptr = &data->address;583 desc_list[0].len = data->len;584 break;585 }586 case SEV_CMD_LAUNCH_UPDATE_SECRET: {587 struct sev_data_launch_secret *data = cmd_buf;588 589 desc_list[0].paddr_ptr = &data->guest_address;590 desc_list[0].len = data->guest_len;591 desc_list[0].guest_owned = true;592 break;593 }594 case SEV_CMD_DBG_DECRYPT: {595 struct sev_data_dbg *data = cmd_buf;596 597 desc_list[0].paddr_ptr = &data->dst_addr;598 desc_list[0].len = data->len;599 desc_list[0].guest_owned = true;600 break;601 }602 case SEV_CMD_DBG_ENCRYPT: {603 struct sev_data_dbg *data = cmd_buf;604 605 desc_list[0].paddr_ptr = &data->dst_addr;606 desc_list[0].len = data->len;607 desc_list[0].guest_owned = true;608 break;609 }610 case SEV_CMD_ATTESTATION_REPORT: {611 struct sev_data_attestation_report *data = cmd_buf;612 613 desc_list[0].paddr_ptr = &data->address;614 desc_list[0].len = data->len;615 break;616 }617 case SEV_CMD_SEND_START: {618 struct sev_data_send_start *data = cmd_buf;619 620 desc_list[0].paddr_ptr = &data->session_address;621 desc_list[0].len = data->session_len;622 break;623 }624 case SEV_CMD_SEND_UPDATE_DATA: {625 struct sev_data_send_update_data *data = cmd_buf;626 627 desc_list[0].paddr_ptr = &data->hdr_address;628 desc_list[0].len = data->hdr_len;629 desc_list[1].paddr_ptr = &data->trans_address;630 desc_list[1].len = data->trans_len;631 break;632 }633 case SEV_CMD_SEND_UPDATE_VMSA: {634 struct sev_data_send_update_vmsa *data = cmd_buf;635 636 desc_list[0].paddr_ptr = &data->hdr_address;637 desc_list[0].len = data->hdr_len;638 desc_list[1].paddr_ptr = &data->trans_address;639 desc_list[1].len = data->trans_len;640 break;641 }642 case SEV_CMD_RECEIVE_UPDATE_DATA: {643 struct sev_data_receive_update_data *data = cmd_buf;644 645 desc_list[0].paddr_ptr = &data->guest_address;646 desc_list[0].len = data->guest_len;647 desc_list[0].guest_owned = true;648 break;649 }650 case SEV_CMD_RECEIVE_UPDATE_VMSA: {651 struct sev_data_receive_update_vmsa *data = cmd_buf;652 653 desc_list[0].paddr_ptr = &data->guest_address;654 desc_list[0].len = data->guest_len;655 desc_list[0].guest_owned = true;656 break;657 }658 default:659 break;660 }661}662 663static int snp_map_cmd_buf_desc(struct cmd_buf_desc *desc)664{665 unsigned int npages;666 667 if (!desc->len)668 return 0;669 670 /* Allocate a bounce buffer if this isn't a guest owned page. */671 if (!desc->guest_owned) {672 struct page *page;673 674 page = alloc_pages(GFP_KERNEL_ACCOUNT, get_order(desc->len));675 if (!page) {676 pr_warn("Failed to allocate bounce buffer for SEV legacy command.\n");677 return -ENOMEM;678 }679 680 desc->paddr_orig = *desc->paddr_ptr;681 *desc->paddr_ptr = __psp_pa(page_to_virt(page));682 }683 684 npages = PAGE_ALIGN(desc->len) >> PAGE_SHIFT;685 686 /* Transition the buffer to firmware-owned. */687 if (rmp_mark_pages_firmware(*desc->paddr_ptr, npages, true)) {688 pr_warn("Error moving pages to firmware-owned state for SEV legacy command.\n");689 return -EFAULT;690 }691 692 return 0;693}694 695static int snp_unmap_cmd_buf_desc(struct cmd_buf_desc *desc)696{697 unsigned int npages;698 699 if (!desc->len)700 return 0;701 702 npages = PAGE_ALIGN(desc->len) >> PAGE_SHIFT;703 704 /* Transition the buffers back to hypervisor-owned. */705 if (snp_reclaim_pages(*desc->paddr_ptr, npages, true)) {706 pr_warn("Failed to reclaim firmware-owned pages while issuing SEV legacy command.\n");707 return -EFAULT;708 }709 710 /* Copy data from bounce buffer and then free it. */711 if (!desc->guest_owned) {712 void *bounce_buf = __va(__sme_clr(*desc->paddr_ptr));713 void *dst_buf = __va(__sme_clr(desc->paddr_orig));714 715 memcpy(dst_buf, bounce_buf, desc->len);716 __free_pages(virt_to_page(bounce_buf), get_order(desc->len));717 718 /* Restore the original address in the command buffer. */719 *desc->paddr_ptr = desc->paddr_orig;720 }721 722 return 0;723}724 725static int snp_map_cmd_buf_desc_list(int cmd, void *cmd_buf, struct cmd_buf_desc *desc_list)726{727 int i;728 729 snp_populate_cmd_buf_desc_list(cmd, cmd_buf, desc_list);730 731 for (i = 0; i < CMD_BUF_DESC_MAX; i++) {732 struct cmd_buf_desc *desc = &desc_list[i];733 734 if (!desc->paddr_ptr)735 break;736 737 if (snp_map_cmd_buf_desc(desc))738 goto err_unmap;739 }740 741 return 0;742 743err_unmap:744 for (i--; i >= 0; i--)745 snp_unmap_cmd_buf_desc(&desc_list[i]);746 747 return -EFAULT;748}749 750static int snp_unmap_cmd_buf_desc_list(struct cmd_buf_desc *desc_list)751{752 int i, ret = 0;753 754 for (i = 0; i < CMD_BUF_DESC_MAX; i++) {755 struct cmd_buf_desc *desc = &desc_list[i];756 757 if (!desc->paddr_ptr)758 break;759 760 if (snp_unmap_cmd_buf_desc(&desc_list[i]))761 ret = -EFAULT;762 }763 764 return ret;765}766 767static bool sev_cmd_buf_writable(int cmd)768{769 switch (cmd) {770 case SEV_CMD_PLATFORM_STATUS:771 case SEV_CMD_GUEST_STATUS:772 case SEV_CMD_LAUNCH_START:773 case SEV_CMD_RECEIVE_START:774 case SEV_CMD_LAUNCH_MEASURE:775 case SEV_CMD_SEND_START:776 case SEV_CMD_SEND_UPDATE_DATA:777 case SEV_CMD_SEND_UPDATE_VMSA:778 case SEV_CMD_PEK_CSR:779 case SEV_CMD_PDH_CERT_EXPORT:780 case SEV_CMD_GET_ID:781 case SEV_CMD_ATTESTATION_REPORT:782 return true;783 default:784 return false;785 }786}787 788/* After SNP is INIT'ed, the behavior of legacy SEV commands is changed. */789static bool snp_legacy_handling_needed(int cmd)790{791 struct sev_device *sev = psp_master->sev_data;792 793 return cmd < SEV_CMD_SNP_INIT && sev->snp_initialized;794}795 796static int snp_prep_cmd_buf(int cmd, void *cmd_buf, struct cmd_buf_desc *desc_list)797{798 if (!snp_legacy_handling_needed(cmd))799 return 0;800 801 if (snp_map_cmd_buf_desc_list(cmd, cmd_buf, desc_list))802 return -EFAULT;803 804 /*805 * Before command execution, the command buffer needs to be put into806 * the firmware-owned state.807 */808 if (sev_cmd_buf_writable(cmd)) {809 if (rmp_mark_pages_firmware(__pa(cmd_buf), 1, true))810 return -EFAULT;811 }812 813 return 0;814}815 816static int snp_reclaim_cmd_buf(int cmd, void *cmd_buf)817{818 if (!snp_legacy_handling_needed(cmd))819 return 0;820 821 /*822 * After command completion, the command buffer needs to be put back823 * into the hypervisor-owned state.824 */825 if (sev_cmd_buf_writable(cmd))826 if (snp_reclaim_pages(__pa(cmd_buf), 1, true))827 return -EFAULT;828 829 return 0;830}831 832static int __sev_do_cmd_locked(int cmd, void *data, int *psp_ret)833{834 struct cmd_buf_desc desc_list[CMD_BUF_DESC_MAX] = {0};835 struct psp_device *psp = psp_master;836 struct sev_device *sev;837 unsigned int cmdbuff_hi, cmdbuff_lo;838 unsigned int phys_lsb, phys_msb;839 unsigned int reg, ret = 0;840 void *cmd_buf;841 int buf_len;842 843 if (!psp || !psp->sev_data)844 return -ENODEV;845 846 if (psp_dead)847 return -EBUSY;848 849 sev = psp->sev_data;850 851 buf_len = sev_cmd_buffer_len(cmd);852 if (WARN_ON_ONCE(!data != !buf_len))853 return -EINVAL;854 855 /*856 * Copy the incoming data to driver's scratch buffer as __pa() will not857 * work for some memory, e.g. vmalloc'd addresses, and @data may not be858 * physically contiguous.859 */860 if (data) {861 /*862 * Commands are generally issued one at a time and require the863 * sev_cmd_mutex, but there could be recursive firmware requests864 * due to SEV_CMD_SNP_PAGE_RECLAIM needing to be issued while865 * preparing buffers for another command. This is the only known866 * case of nesting in the current code, so exactly one867 * additional command buffer is available for that purpose.868 */869 if (!sev->cmd_buf_active) {870 cmd_buf = sev->cmd_buf;871 sev->cmd_buf_active = true;872 } else if (!sev->cmd_buf_backup_active) {873 cmd_buf = sev->cmd_buf_backup;874 sev->cmd_buf_backup_active = true;875 } else {876 dev_err(sev->dev,877 "SEV: too many firmware commands in progress, no command buffers available.\n");878 return -EBUSY;879 }880 881 memcpy(cmd_buf, data, buf_len);882 883 /*884 * The behavior of the SEV-legacy commands is altered when the885 * SNP firmware is in the INIT state.886 */887 ret = snp_prep_cmd_buf(cmd, cmd_buf, desc_list);888 if (ret) {889 dev_err(sev->dev,890 "SEV: failed to prepare buffer for legacy command 0x%x. Error: %d\n",891 cmd, ret);892 return ret;893 }894 } else {895 cmd_buf = sev->cmd_buf;896 }897 898 /* Get the physical address of the command buffer */899 phys_lsb = data ? lower_32_bits(__psp_pa(cmd_buf)) : 0;900 phys_msb = data ? upper_32_bits(__psp_pa(cmd_buf)) : 0;901 902 dev_dbg(sev->dev, "sev command id %#x buffer 0x%08x%08x timeout %us\n",903 cmd, phys_msb, phys_lsb, psp_timeout);904 905 print_hex_dump_debug("(in): ", DUMP_PREFIX_OFFSET, 16, 2, data,906 buf_len, false);907 908 iowrite32(phys_lsb, sev->io_regs + sev->vdata->cmdbuff_addr_lo_reg);909 iowrite32(phys_msb, sev->io_regs + sev->vdata->cmdbuff_addr_hi_reg);910 911 sev->int_rcvd = 0;912 913 reg = FIELD_PREP(SEV_CMDRESP_CMD, cmd);914 915 /*916 * If invoked during panic handling, local interrupts are disabled so917 * the PSP command completion interrupt can't be used.918 * sev_wait_cmd_ioc() already checks for interrupts disabled and919 * polls for PSP command completion. Ensure we do not request an920 * interrupt from the PSP if irqs disabled.921 */922 if (!irqs_disabled())923 reg |= SEV_CMDRESP_IOC;924 925 iowrite32(reg, sev->io_regs + sev->vdata->cmdresp_reg);926 927 /* wait for command completion */928 ret = sev_wait_cmd_ioc(sev, ®, psp_timeout);929 if (ret) {930 if (psp_ret)931 *psp_ret = 0;932 933 dev_err(sev->dev, "sev command %#x timed out, disabling PSP\n", cmd);934 psp_dead = true;935 936 return ret;937 }938 939 psp_timeout = psp_cmd_timeout;940 941 if (psp_ret)942 *psp_ret = FIELD_GET(PSP_CMDRESP_STS, reg);943 944 if (FIELD_GET(PSP_CMDRESP_STS, reg)) {945 dev_dbg(sev->dev, "sev command %#x failed (%#010lx)\n",946 cmd, FIELD_GET(PSP_CMDRESP_STS, reg));947 948 /*949 * PSP firmware may report additional error information in the950 * command buffer registers on error. Print contents of command951 * buffer registers if they changed.952 */953 cmdbuff_hi = ioread32(sev->io_regs + sev->vdata->cmdbuff_addr_hi_reg);954 cmdbuff_lo = ioread32(sev->io_regs + sev->vdata->cmdbuff_addr_lo_reg);955 if (cmdbuff_hi != phys_msb || cmdbuff_lo != phys_lsb) {956 dev_dbg(sev->dev, "Additional error information reported in cmdbuff:");957 dev_dbg(sev->dev, " cmdbuff hi: %#010x\n", cmdbuff_hi);958 dev_dbg(sev->dev, " cmdbuff lo: %#010x\n", cmdbuff_lo);959 }960 ret = -EIO;961 } else {962 ret = sev_write_init_ex_file_if_required(cmd);963 }964 965 /*966 * Copy potential output from the PSP back to data. Do this even on967 * failure in case the caller wants to glean something from the error.968 */969 if (data) {970 int ret_reclaim;971 /*972 * Restore the page state after the command completes.973 */974 ret_reclaim = snp_reclaim_cmd_buf(cmd, cmd_buf);975 if (ret_reclaim) {976 dev_err(sev->dev,977 "SEV: failed to reclaim buffer for legacy command %#x. Error: %d\n",978 cmd, ret_reclaim);979 return ret_reclaim;980 }981 982 memcpy(data, cmd_buf, buf_len);983 984 if (sev->cmd_buf_backup_active)985 sev->cmd_buf_backup_active = false;986 else987 sev->cmd_buf_active = false;988 989 if (snp_unmap_cmd_buf_desc_list(desc_list))990 return -EFAULT;991 }992 993 print_hex_dump_debug("(out): ", DUMP_PREFIX_OFFSET, 16, 2, data,994 buf_len, false);995 996 return ret;997}998 999int sev_do_cmd(int cmd, void *data, int *psp_ret)1000{1001 int rc;1002 1003 mutex_lock(&sev_cmd_mutex);1004 rc = __sev_do_cmd_locked(cmd, data, psp_ret);1005 mutex_unlock(&sev_cmd_mutex);1006 1007 return rc;1008}1009EXPORT_SYMBOL_GPL(sev_do_cmd);1010 1011static int __sev_init_locked(int *error)1012{1013 struct sev_data_init data;1014 1015 memset(&data, 0, sizeof(data));1016 if (sev_es_tmr) {1017 /*1018 * Do not include the encryption mask on the physical1019 * address of the TMR (firmware should clear it anyway).1020 */1021 data.tmr_address = __pa(sev_es_tmr);1022 1023 data.flags |= SEV_INIT_FLAGS_SEV_ES;1024 data.tmr_len = sev_es_tmr_size;1025 }1026 1027 return __sev_do_cmd_locked(SEV_CMD_INIT, &data, error);1028}1029 1030static int __sev_init_ex_locked(int *error)1031{1032 struct sev_data_init_ex data;1033 1034 memset(&data, 0, sizeof(data));1035 data.length = sizeof(data);1036 data.nv_address = __psp_pa(sev_init_ex_buffer);1037 data.nv_len = NV_LENGTH;1038 1039 if (sev_es_tmr) {1040 /*1041 * Do not include the encryption mask on the physical1042 * address of the TMR (firmware should clear it anyway).1043 */1044 data.tmr_address = __pa(sev_es_tmr);1045 1046 data.flags |= SEV_INIT_FLAGS_SEV_ES;1047 data.tmr_len = sev_es_tmr_size;1048 }1049 1050 return __sev_do_cmd_locked(SEV_CMD_INIT_EX, &data, error);1051}1052 1053static inline int __sev_do_init_locked(int *psp_ret)1054{1055 if (sev_init_ex_buffer)1056 return __sev_init_ex_locked(psp_ret);1057 else1058 return __sev_init_locked(psp_ret);1059}1060 1061static void snp_set_hsave_pa(void *arg)1062{1063 wrmsrl(MSR_VM_HSAVE_PA, 0);1064}1065 1066static int snp_filter_reserved_mem_regions(struct resource *rs, void *arg)1067{1068 struct sev_data_range_list *range_list = arg;1069 struct sev_data_range *range = &range_list->ranges[range_list->num_elements];1070 size_t size;1071 1072 /*1073 * Ensure the list of HV_FIXED pages that will be passed to firmware1074 * do not exceed the page-sized argument buffer.1075 */1076 if ((range_list->num_elements * sizeof(struct sev_data_range) +1077 sizeof(struct sev_data_range_list)) > PAGE_SIZE)1078 return -E2BIG;1079 1080 switch (rs->desc) {1081 case E820_TYPE_RESERVED:1082 case E820_TYPE_PMEM:1083 case E820_TYPE_ACPI:1084 range->base = rs->start & PAGE_MASK;1085 size = PAGE_ALIGN((rs->end + 1) - rs->start);1086 range->page_count = size >> PAGE_SHIFT;1087 range_list->num_elements++;1088 break;1089 default:1090 break;1091 }1092 1093 return 0;1094}1095 1096static int __sev_snp_init_locked(int *error)1097{1098 struct psp_device *psp = psp_master;1099 struct sev_data_snp_init_ex data;1100 struct sev_device *sev;1101 void *arg = &data;1102 int cmd, rc = 0;1103 1104 if (!cc_platform_has(CC_ATTR_HOST_SEV_SNP))1105 return -ENODEV;1106 1107 sev = psp->sev_data;1108 1109 if (sev->snp_initialized)1110 return 0;1111 1112 if (!sev_version_greater_or_equal(SNP_MIN_API_MAJOR, SNP_MIN_API_MINOR)) {1113 dev_dbg(sev->dev, "SEV-SNP support requires firmware version >= %d:%d\n",1114 SNP_MIN_API_MAJOR, SNP_MIN_API_MINOR);1115 return 0;1116 }1117 1118 /* SNP_INIT requires MSR_VM_HSAVE_PA to be cleared on all CPUs. */1119 on_each_cpu(snp_set_hsave_pa, NULL, 1);1120 1121 /*1122 * Starting in SNP firmware v1.52, the SNP_INIT_EX command takes a list1123 * of system physical address ranges to convert into HV-fixed page1124 * states during the RMP initialization. For instance, the memory that1125 * UEFI reserves should be included in the that list. This allows system1126 * components that occasionally write to memory (e.g. logging to UEFI1127 * reserved regions) to not fail due to RMP initialization and SNP1128 * enablement.1129 *1130 */1131 if (sev_version_greater_or_equal(SNP_MIN_API_MAJOR, 52)) {1132 /*1133 * Firmware checks that the pages containing the ranges enumerated1134 * in the RANGES structure are either in the default page state or in the1135 * firmware page state.1136 */1137 snp_range_list = kzalloc(PAGE_SIZE, GFP_KERNEL);1138 if (!snp_range_list) {1139 dev_err(sev->dev,1140 "SEV: SNP_INIT_EX range list memory allocation failed\n");1141 return -ENOMEM;1142 }1143 1144 /*1145 * Retrieve all reserved memory regions from the e820 memory map1146 * to be setup as HV-fixed pages.1147 */1148 rc = walk_iomem_res_desc(IORES_DESC_NONE, IORESOURCE_MEM, 0, ~0,1149 snp_range_list, snp_filter_reserved_mem_regions);1150 if (rc) {1151 dev_err(sev->dev,1152 "SEV: SNP_INIT_EX walk_iomem_res_desc failed rc = %d\n", rc);1153 return rc;1154 }1155 1156 memset(&data, 0, sizeof(data));1157 data.init_rmp = 1;1158 data.list_paddr_en = 1;1159 data.list_paddr = __psp_pa(snp_range_list);1160 cmd = SEV_CMD_SNP_INIT_EX;1161 } else {1162 cmd = SEV_CMD_SNP_INIT;1163 arg = NULL;1164 }1165 1166 /*1167 * The following sequence must be issued before launching the first SNP1168 * guest to ensure all dirty cache lines are flushed, including from1169 * updates to the RMP table itself via the RMPUPDATE instruction:1170 *1171 * - WBINVD on all running CPUs1172 * - SEV_CMD_SNP_INIT[_EX] firmware command1173 * - WBINVD on all running CPUs1174 * - SEV_CMD_SNP_DF_FLUSH firmware command1175 */1176 wbinvd_on_all_cpus();1177 1178 rc = __sev_do_cmd_locked(cmd, arg, error);1179 if (rc)1180 return rc;1181 1182 /* Prepare for first SNP guest launch after INIT. */1183 wbinvd_on_all_cpus();1184 rc = __sev_do_cmd_locked(SEV_CMD_SNP_DF_FLUSH, NULL, error);1185 if (rc)1186 return rc;1187 1188 sev->snp_initialized = true;1189 dev_dbg(sev->dev, "SEV-SNP firmware initialized\n");1190 1191 sev_es_tmr_size = SNP_TMR_SIZE;1192 1193 return rc;1194}1195 1196static void __sev_platform_init_handle_tmr(struct sev_device *sev)1197{1198 if (sev_es_tmr)1199 return;1200 1201 /* Obtain the TMR memory area for SEV-ES use */1202 sev_es_tmr = sev_fw_alloc(sev_es_tmr_size);1203 if (sev_es_tmr) {1204 /* Must flush the cache before giving it to the firmware */1205 if (!sev->snp_initialized)1206 clflush_cache_range(sev_es_tmr, sev_es_tmr_size);1207 } else {1208 dev_warn(sev->dev, "SEV: TMR allocation failed, SEV-ES support unavailable\n");1209 }1210}1211 1212/*1213 * If an init_ex_path is provided allocate a buffer for the file and1214 * read in the contents. Additionally, if SNP is initialized, convert1215 * the buffer pages to firmware pages.1216 */1217static int __sev_platform_init_handle_init_ex_path(struct sev_device *sev)1218{1219 struct page *page;1220 int rc;1221 1222 if (!init_ex_path)1223 return 0;1224 1225 if (sev_init_ex_buffer)1226 return 0;1227 1228 page = alloc_pages(GFP_KERNEL, get_order(NV_LENGTH));1229 if (!page) {1230 dev_err(sev->dev, "SEV: INIT_EX NV memory allocation failed\n");1231 return -ENOMEM;1232 }1233 1234 sev_init_ex_buffer = page_address(page);1235 1236 rc = sev_read_init_ex_file();1237 if (rc)1238 return rc;1239 1240 /* If SEV-SNP is initialized, transition to firmware page. */1241 if (sev->snp_initialized) {1242 unsigned long npages;1243 1244 npages = 1UL << get_order(NV_LENGTH);1245 if (rmp_mark_pages_firmware(__pa(sev_init_ex_buffer), npages, false)) {1246 dev_err(sev->dev, "SEV: INIT_EX NV memory page state change failed.\n");1247 return -ENOMEM;1248 }1249 }1250 1251 return 0;1252}1253 1254static int __sev_platform_init_locked(int *error)1255{1256 int rc, psp_ret = SEV_RET_NO_FW_CALL;1257 struct sev_device *sev;1258 1259 if (!psp_master || !psp_master->sev_data)1260 return -ENODEV;1261 1262 sev = psp_master->sev_data;1263 1264 if (sev->state == SEV_STATE_INIT)1265 return 0;1266 1267 __sev_platform_init_handle_tmr(sev);1268 1269 rc = __sev_platform_init_handle_init_ex_path(sev);1270 if (rc)1271 return rc;1272 1273 rc = __sev_do_init_locked(&psp_ret);1274 if (rc && psp_ret == SEV_RET_SECURE_DATA_INVALID) {1275 /*1276 * Initialization command returned an integrity check failure1277 * status code, meaning that firmware load and validation of SEV1278 * related persistent data has failed. Retrying the1279 * initialization function should succeed by replacing the state1280 * with a reset state.1281 */1282 dev_err(sev->dev,1283"SEV: retrying INIT command because of SECURE_DATA_INVALID error. Retrying once to reset PSP SEV state.");1284 rc = __sev_do_init_locked(&psp_ret);1285 }1286 1287 if (error)1288 *error = psp_ret;1289 1290 if (rc)1291 return rc;1292 1293 sev->state = SEV_STATE_INIT;1294 1295 /* Prepare for first SEV guest launch after INIT */1296 wbinvd_on_all_cpus();1297 rc = __sev_do_cmd_locked(SEV_CMD_DF_FLUSH, NULL, error);1298 if (rc)1299 return rc;1300 1301 dev_dbg(sev->dev, "SEV firmware initialized\n");1302 1303 dev_info(sev->dev, "SEV API:%d.%d build:%d\n", sev->api_major,1304 sev->api_minor, sev->build);1305 1306 return 0;1307}1308 1309static int _sev_platform_init_locked(struct sev_platform_init_args *args)1310{1311 struct sev_device *sev;1312 int rc;1313 1314 if (!psp_master || !psp_master->sev_data)1315 return -ENODEV;1316 1317 sev = psp_master->sev_data;1318 1319 if (sev->state == SEV_STATE_INIT)1320 return 0;1321 1322 /*1323 * Legacy guests cannot be running while SNP_INIT(_EX) is executing,1324 * so perform SEV-SNP initialization at probe time.1325 */1326 rc = __sev_snp_init_locked(&args->error);1327 if (rc && rc != -ENODEV) {1328 /*1329 * Don't abort the probe if SNP INIT failed,1330 * continue to initialize the legacy SEV firmware.1331 */1332 dev_err(sev->dev, "SEV-SNP: failed to INIT rc %d, error %#x\n",1333 rc, args->error);1334 }1335 1336 /* Defer legacy SEV/SEV-ES support if allowed by caller/module. */1337 if (args->probe && !psp_init_on_probe)1338 return 0;1339 1340 return __sev_platform_init_locked(&args->error);1341}1342 1343int sev_platform_init(struct sev_platform_init_args *args)1344{1345 int rc;1346 1347 mutex_lock(&sev_cmd_mutex);1348 rc = _sev_platform_init_locked(args);1349 mutex_unlock(&sev_cmd_mutex);1350 1351 return rc;1352}1353EXPORT_SYMBOL_GPL(sev_platform_init);1354 1355static int __sev_platform_shutdown_locked(int *error)1356{1357 struct psp_device *psp = psp_master;1358 struct sev_device *sev;1359 int ret;1360 1361 if (!psp || !psp->sev_data)1362 return 0;1363 1364 sev = psp->sev_data;1365 1366 if (sev->state == SEV_STATE_UNINIT)1367 return 0;1368 1369 ret = __sev_do_cmd_locked(SEV_CMD_SHUTDOWN, NULL, error);1370 if (ret)1371 return ret;1372 1373 sev->state = SEV_STATE_UNINIT;1374 dev_dbg(sev->dev, "SEV firmware shutdown\n");1375 1376 return ret;1377}1378 1379static int sev_get_platform_state(int *state, int *error)1380{1381 struct sev_user_data_status data;1382 int rc;1383 1384 rc = __sev_do_cmd_locked(SEV_CMD_PLATFORM_STATUS, &data, error);1385 if (rc)1386 return rc;1387 1388 *state = data.state;1389 return rc;1390}1391 1392static int sev_ioctl_do_reset(struct sev_issue_cmd *argp, bool writable)1393{1394 int state, rc;1395 1396 if (!writable)1397 return -EPERM;1398 1399 /*1400 * The SEV spec requires that FACTORY_RESET must be issued in1401 * UNINIT state. Before we go further lets check if any guest is1402 * active.1403 *1404 * If FW is in WORKING state then deny the request otherwise issue1405 * SHUTDOWN command do INIT -> UNINIT before issuing the FACTORY_RESET.1406 *1407 */1408 rc = sev_get_platform_state(&state, &argp->error);1409 if (rc)1410 return rc;1411 1412 if (state == SEV_STATE_WORKING)1413 return -EBUSY;1414 1415 if (state == SEV_STATE_INIT) {1416 rc = __sev_platform_shutdown_locked(&argp->error);1417 if (rc)1418 return rc;1419 }1420 1421 return __sev_do_cmd_locked(SEV_CMD_FACTORY_RESET, NULL, &argp->error);1422}1423 1424static int sev_ioctl_do_platform_status(struct sev_issue_cmd *argp)1425{1426 struct sev_user_data_status data;1427 int ret;1428 1429 memset(&data, 0, sizeof(data));1430 1431 ret = __sev_do_cmd_locked(SEV_CMD_PLATFORM_STATUS, &data, &argp->error);1432 if (ret)1433 return ret;1434 1435 if (copy_to_user((void __user *)argp->data, &data, sizeof(data)))1436 ret = -EFAULT;1437 1438 return ret;1439}1440 1441static int sev_ioctl_do_pek_pdh_gen(int cmd, struct sev_issue_cmd *argp, bool writable)1442{1443 struct sev_device *sev = psp_master->sev_data;1444 int rc;1445 1446 if (!writable)1447 return -EPERM;1448 1449 if (sev->state == SEV_STATE_UNINIT) {1450 rc = __sev_platform_init_locked(&argp->error);1451 if (rc)1452 return rc;1453 }1454 1455 return __sev_do_cmd_locked(cmd, NULL, &argp->error);1456}1457 1458static int sev_ioctl_do_pek_csr(struct sev_issue_cmd *argp, bool writable)1459{1460 struct sev_device *sev = psp_master->sev_data;1461 struct sev_user_data_pek_csr input;1462 struct sev_data_pek_csr data;1463 void __user *input_address;1464 void *blob = NULL;1465 int ret;1466 1467 if (!writable)1468 return -EPERM;1469 1470 if (copy_from_user(&input, (void __user *)argp->data, sizeof(input)))1471 return -EFAULT;1472 1473 memset(&data, 0, sizeof(data));1474 1475 /* userspace wants to query CSR length */1476 if (!input.address || !input.length)1477 goto cmd;1478 1479 /* allocate a physically contiguous buffer to store the CSR blob */1480 input_address = (void __user *)input.address;1481 if (input.length > SEV_FW_BLOB_MAX_SIZE)1482 return -EFAULT;1483 1484 blob = kzalloc(input.length, GFP_KERNEL);1485 if (!blob)1486 return -ENOMEM;1487 1488 data.address = __psp_pa(blob);1489 data.len = input.length;1490 1491cmd:1492 if (sev->state == SEV_STATE_UNINIT) {1493 ret = __sev_platform_init_locked(&argp->error);1494 if (ret)1495 goto e_free_blob;1496 }1497 1498 ret = __sev_do_cmd_locked(SEV_CMD_PEK_CSR, &data, &argp->error);1499 1500 /* If we query the CSR length, FW responded with expected data. */1501 input.length = data.len;1502 1503 if (copy_to_user((void __user *)argp->data, &input, sizeof(input))) {1504 ret = -EFAULT;1505 goto e_free_blob;1506 }1507 1508 if (blob) {1509 if (copy_to_user(input_address, blob, input.length))1510 ret = -EFAULT;1511 }1512 1513e_free_blob:1514 kfree(blob);1515 return ret;1516}1517 1518void *psp_copy_user_blob(u64 uaddr, u32 len)1519{1520 if (!uaddr || !len)1521 return ERR_PTR(-EINVAL);1522 1523 /* verify that blob length does not exceed our limit */1524 if (len > SEV_FW_BLOB_MAX_SIZE)1525 return ERR_PTR(-EINVAL);1526 1527 return memdup_user((void __user *)uaddr, len);1528}1529EXPORT_SYMBOL_GPL(psp_copy_user_blob);1530 1531static int sev_get_api_version(void)1532{1533 struct sev_device *sev = psp_master->sev_data;1534 struct sev_user_data_status status;1535 int error = 0, ret;1536 1537 ret = sev_platform_status(&status, &error);1538 if (ret) {1539 dev_err(sev->dev,1540 "SEV: failed to get status. Error: %#x\n", error);1541 return 1;1542 }1543 1544 sev->api_major = status.api_major;1545 sev->api_minor = status.api_minor;1546 sev->build = status.build;1547 sev->state = status.state;1548 1549 return 0;1550}1551 1552static int sev_get_firmware(struct device *dev,1553 const struct firmware **firmware)1554{1555 char fw_name_specific[SEV_FW_NAME_SIZE];1556 char fw_name_subset[SEV_FW_NAME_SIZE];1557 1558 snprintf(fw_name_specific, sizeof(fw_name_specific),1559 "amd/amd_sev_fam%.2xh_model%.2xh.sbin",1560 boot_cpu_data.x86, boot_cpu_data.x86_model);1561 1562 snprintf(fw_name_subset, sizeof(fw_name_subset),1563 "amd/amd_sev_fam%.2xh_model%.1xxh.sbin",1564 boot_cpu_data.x86, (boot_cpu_data.x86_model & 0xf0) >> 4);1565 1566 /* Check for SEV FW for a particular model.1567 * Ex. amd_sev_fam17h_model00h.sbin for Family 17h Model 00h1568 *1569 * or1570 *1571 * Check for SEV FW common to a subset of models.1572 * Ex. amd_sev_fam17h_model0xh.sbin for1573 * Family 17h Model 00h -- Family 17h Model 0Fh1574 *1575 * or1576 *1577 * Fall-back to using generic name: sev.fw1578 */1579 if ((firmware_request_nowarn(firmware, fw_name_specific, dev) >= 0) ||1580 (firmware_request_nowarn(firmware, fw_name_subset, dev) >= 0) ||1581 (firmware_request_nowarn(firmware, SEV_FW_FILE, dev) >= 0))1582 return 0;1583 1584 return -ENOENT;1585}1586 1587/* Don't fail if SEV FW couldn't be updated. Continue with existing SEV FW */1588static int sev_update_firmware(struct device *dev)1589{1590 struct sev_data_download_firmware *data;1591 const struct firmware *firmware;1592 int ret, error, order;1593 struct page *p;1594 u64 data_size;1595 1596 if (!sev_version_greater_or_equal(0, 15)) {1597 dev_dbg(dev, "DOWNLOAD_FIRMWARE not supported\n");1598 return -1;1599 }1600 1601 if (sev_get_firmware(dev, &firmware) == -ENOENT) {1602 dev_dbg(dev, "No SEV firmware file present\n");1603 return -1;1604 }1605 1606 /*1607 * SEV FW expects the physical address given to it to be 321608 * byte aligned. Memory allocated has structure placed at the1609 * beginning followed by the firmware being passed to the SEV1610 * FW. Allocate enough memory for data structure + alignment1611 * padding + SEV FW.1612 */1613 data_size = ALIGN(sizeof(struct sev_data_download_firmware), 32);1614 1615 order = get_order(firmware->size + data_size);1616 p = alloc_pages(GFP_KERNEL, order);1617 if (!p) {1618 ret = -1;1619 goto fw_err;1620 }1621 1622 /*1623 * Copy firmware data to a kernel allocated contiguous1624 * memory region.1625 */1626 data = page_address(p);1627 memcpy(page_address(p) + data_size, firmware->data, firmware->size);1628 1629 data->address = __psp_pa(page_address(p) + data_size);1630 data->len = firmware->size;1631 1632 ret = sev_do_cmd(SEV_CMD_DOWNLOAD_FIRMWARE, data, &error);1633 1634 /*1635 * A quirk for fixing the committed TCB version, when upgrading from1636 * earlier firmware version than 1.50.1637 */1638 if (!ret && !sev_version_greater_or_equal(1, 50))1639 ret = sev_do_cmd(SEV_CMD_DOWNLOAD_FIRMWARE, data, &error);1640 1641 if (ret)1642 dev_dbg(dev, "Failed to update SEV firmware: %#x\n", error);1643 1644 __free_pages(p, order);1645 1646fw_err:1647 release_firmware(firmware);1648 1649 return ret;1650}1651 1652static int __sev_snp_shutdown_locked(int *error, bool panic)1653{1654 struct psp_device *psp = psp_master;1655 struct sev_device *sev;1656 struct sev_data_snp_shutdown_ex data;1657 int ret;1658 1659 if (!psp || !psp->sev_data)1660 return 0;1661 1662 sev = psp->sev_data;1663 1664 if (!sev->snp_initialized)1665 return 0;1666 1667 memset(&data, 0, sizeof(data));1668 data.len = sizeof(data);1669 data.iommu_snp_shutdown = 1;1670 1671 /*1672 * If invoked during panic handling, local interrupts are disabled1673 * and all CPUs are stopped, so wbinvd_on_all_cpus() can't be called.1674 * In that case, a wbinvd() is done on remote CPUs via the NMI1675 * callback, so only a local wbinvd() is needed here.1676 */1677 if (!panic)1678 wbinvd_on_all_cpus();1679 else1680 wbinvd();1681 1682 ret = __sev_do_cmd_locked(SEV_CMD_SNP_SHUTDOWN_EX, &data, error);1683 /* SHUTDOWN may require DF_FLUSH */1684 if (*error == SEV_RET_DFFLUSH_REQUIRED) {1685 ret = __sev_do_cmd_locked(SEV_CMD_SNP_DF_FLUSH, NULL, NULL);1686 if (ret) {1687 dev_err(sev->dev, "SEV-SNP DF_FLUSH failed\n");1688 return ret;1689 }1690 /* reissue the shutdown command */1691 ret = __sev_do_cmd_locked(SEV_CMD_SNP_SHUTDOWN_EX, &data,1692 error);1693 }1694 if (ret) {1695 dev_err(sev->dev, "SEV-SNP firmware shutdown failed\n");1696 return ret;1697 }1698 1699 /*1700 * SNP_SHUTDOWN_EX with IOMMU_SNP_SHUTDOWN set to 1 disables SNP1701 * enforcement by the IOMMU and also transitions all pages1702 * associated with the IOMMU to the Reclaim state.1703 * Firmware was transitioning the IOMMU pages to Hypervisor state1704 * before version 1.53. But, accounting for the number of assigned1705 * 4kB pages in a 2M page was done incorrectly by not transitioning1706 * to the Reclaim state. This resulted in RMP #PF when later accessing1707 * the 2M page containing those pages during kexec boot. Hence, the1708 * firmware now transitions these pages to Reclaim state and hypervisor1709 * needs to transition these pages to shared state. SNP Firmware1710 * version 1.53 and above are needed for kexec boot.1711 */1712 ret = amd_iommu_snp_disable();1713 if (ret) {1714 dev_err(sev->dev, "SNP IOMMU shutdown failed\n");1715 return ret;1716 }1717 1718 sev->snp_initialized = false;1719 dev_dbg(sev->dev, "SEV-SNP firmware shutdown\n");1720 1721 return ret;1722}1723 1724static int sev_ioctl_do_pek_import(struct sev_issue_cmd *argp, bool writable)1725{1726 struct sev_device *sev = psp_master->sev_data;1727 struct sev_user_data_pek_cert_import input;1728 struct sev_data_pek_cert_import data;1729 void *pek_blob, *oca_blob;1730 int ret;1731 1732 if (!writable)1733 return -EPERM;1734 1735 if (copy_from_user(&input, (void __user *)argp->data, sizeof(input)))1736 return -EFAULT;1737 1738 /* copy PEK certificate blobs from userspace */1739 pek_blob = psp_copy_user_blob(input.pek_cert_address, input.pek_cert_len);1740 if (IS_ERR(pek_blob))1741 return PTR_ERR(pek_blob);1742 1743 data.reserved = 0;1744 data.pek_cert_address = __psp_pa(pek_blob);1745 data.pek_cert_len = input.pek_cert_len;1746 1747 /* copy PEK certificate blobs from userspace */1748 oca_blob = psp_copy_user_blob(input.oca_cert_address, input.oca_cert_len);1749 if (IS_ERR(oca_blob)) {1750 ret = PTR_ERR(oca_blob);1751 goto e_free_pek;1752 }1753 1754 data.oca_cert_address = __psp_pa(oca_blob);1755 data.oca_cert_len = input.oca_cert_len;1756 1757 /* If platform is not in INIT state then transition it to INIT */1758 if (sev->state != SEV_STATE_INIT) {1759 ret = __sev_platform_init_locked(&argp->error);1760 if (ret)1761 goto e_free_oca;1762 }1763 1764 ret = __sev_do_cmd_locked(SEV_CMD_PEK_CERT_IMPORT, &data, &argp->error);1765 1766e_free_oca:1767 kfree(oca_blob);1768e_free_pek:1769 kfree(pek_blob);1770 return ret;1771}1772 1773static int sev_ioctl_do_get_id2(struct sev_issue_cmd *argp)1774{1775 struct sev_user_data_get_id2 input;1776 struct sev_data_get_id data;1777 void __user *input_address;1778 void *id_blob = NULL;1779 int ret;1780 1781 /* SEV GET_ID is available from SEV API v0.16 and up */1782 if (!sev_version_greater_or_equal(0, 16))1783 return -ENOTSUPP;1784 1785 if (copy_from_user(&input, (void __user *)argp->data, sizeof(input)))1786 return -EFAULT;1787 1788 input_address = (void __user *)input.address;1789 1790 if (input.address && input.length) {1791 /*1792 * The length of the ID shouldn't be assumed by software since1793 * it may change in the future. The allocation size is limited1794 * to 1 << (PAGE_SHIFT + MAX_PAGE_ORDER) by the page allocator.1795 * If the allocation fails, simply return ENOMEM rather than1796 * warning in the kernel log.1797 */1798 id_blob = kzalloc(input.length, GFP_KERNEL | __GFP_NOWARN);1799 if (!id_blob)1800 return -ENOMEM;1801 1802 data.address = __psp_pa(id_blob);1803 data.len = input.length;1804 } else {1805 data.address = 0;1806 data.len = 0;1807 }1808 1809 ret = __sev_do_cmd_locked(SEV_CMD_GET_ID, &data, &argp->error);1810 1811 /*1812 * Firmware will return the length of the ID value (either the minimum1813 * required length or the actual length written), return it to the user.1814 */1815 input.length = data.len;1816 1817 if (copy_to_user((void __user *)argp->data, &input, sizeof(input))) {1818 ret = -EFAULT;1819 goto e_free;1820 }1821 1822 if (id_blob) {1823 if (copy_to_user(input_address, id_blob, data.len)) {1824 ret = -EFAULT;1825 goto e_free;1826 }1827 }1828 1829e_free:1830 kfree(id_blob);1831 1832 return ret;1833}1834 1835static int sev_ioctl_do_get_id(struct sev_issue_cmd *argp)1836{1837 struct sev_data_get_id *data;1838 u64 data_size, user_size;1839 void *id_blob, *mem;1840 int ret;1841 1842 /* SEV GET_ID available from SEV API v0.16 and up */1843 if (!sev_version_greater_or_equal(0, 16))1844 return -ENOTSUPP;1845 1846 /* SEV FW expects the buffer it fills with the ID to be1847 * 8-byte aligned. Memory allocated should be enough to1848 * hold data structure + alignment padding + memory1849 * where SEV FW writes the ID.1850 */1851 data_size = ALIGN(sizeof(struct sev_data_get_id), 8);1852 user_size = sizeof(struct sev_user_data_get_id);1853 1854 mem = kzalloc(data_size + user_size, GFP_KERNEL);1855 if (!mem)1856 return -ENOMEM;1857 1858 data = mem;1859 id_blob = mem + data_size;1860 1861 data->address = __psp_pa(id_blob);1862 data->len = user_size;1863 1864 ret = __sev_do_cmd_locked(SEV_CMD_GET_ID, data, &argp->error);1865 if (!ret) {1866 if (copy_to_user((void __user *)argp->data, id_blob, data->len))1867 ret = -EFAULT;1868 }1869 1870 kfree(mem);1871 1872 return ret;1873}1874 1875static int sev_ioctl_do_pdh_export(struct sev_issue_cmd *argp, bool writable)1876{1877 struct sev_device *sev = psp_master->sev_data;1878 struct sev_user_data_pdh_cert_export input;1879 void *pdh_blob = NULL, *cert_blob = NULL;1880 struct sev_data_pdh_cert_export data;1881 void __user *input_cert_chain_address;1882 void __user *input_pdh_cert_address;1883 int ret;1884 1885 /* If platform is not in INIT state then transition it to INIT. */1886 if (sev->state != SEV_STATE_INIT) {1887 if (!writable)1888 return -EPERM;1889 1890 ret = __sev_platform_init_locked(&argp->error);1891 if (ret)1892 return ret;1893 }1894 1895 if (copy_from_user(&input, (void __user *)argp->data, sizeof(input)))1896 return -EFAULT;1897 1898 memset(&data, 0, sizeof(data));1899 1900 /* Userspace wants to query the certificate length. */1901 if (!input.pdh_cert_address ||1902 !input.pdh_cert_len ||1903 !input.cert_chain_address)1904 goto cmd;1905 1906 input_pdh_cert_address = (void __user *)input.pdh_cert_address;1907 input_cert_chain_address = (void __user *)input.cert_chain_address;1908 1909 /* Allocate a physically contiguous buffer to store the PDH blob. */1910 if (input.pdh_cert_len > SEV_FW_BLOB_MAX_SIZE)1911 return -EFAULT;1912 1913 /* Allocate a physically contiguous buffer to store the cert chain blob. */1914 if (input.cert_chain_len > SEV_FW_BLOB_MAX_SIZE)1915 return -EFAULT;1916 1917 pdh_blob = kzalloc(input.pdh_cert_len, GFP_KERNEL);1918 if (!pdh_blob)1919 return -ENOMEM;1920 1921 data.pdh_cert_address = __psp_pa(pdh_blob);1922 data.pdh_cert_len = input.pdh_cert_len;1923 1924 cert_blob = kzalloc(input.cert_chain_len, GFP_KERNEL);1925 if (!cert_blob) {1926 ret = -ENOMEM;1927 goto e_free_pdh;1928 }1929 1930 data.cert_chain_address = __psp_pa(cert_blob);1931 data.cert_chain_len = input.cert_chain_len;1932 1933cmd:1934 ret = __sev_do_cmd_locked(SEV_CMD_PDH_CERT_EXPORT, &data, &argp->error);1935 1936 /* If we query the length, FW responded with expected data. */1937 input.cert_chain_len = data.cert_chain_len;1938 input.pdh_cert_len = data.pdh_cert_len;1939 1940 if (copy_to_user((void __user *)argp->data, &input, sizeof(input))) {1941 ret = -EFAULT;1942 goto e_free_cert;1943 }1944 1945 if (pdh_blob) {1946 if (copy_to_user(input_pdh_cert_address,1947 pdh_blob, input.pdh_cert_len)) {1948 ret = -EFAULT;1949 goto e_free_cert;1950 }1951 }1952 1953 if (cert_blob) {1954 if (copy_to_user(input_cert_chain_address,1955 cert_blob, input.cert_chain_len))1956 ret = -EFAULT;1957 }1958 1959e_free_cert:1960 kfree(cert_blob);1961e_free_pdh:1962 kfree(pdh_blob);1963 return ret;1964}1965 1966static int sev_ioctl_do_snp_platform_status(struct sev_issue_cmd *argp)1967{1968 struct sev_device *sev = psp_master->sev_data;1969 struct sev_data_snp_addr buf;1970 struct page *status_page;1971 void *data;1972 int ret;1973 1974 if (!sev->snp_initialized || !argp->data)1975 return -EINVAL;1976 1977 status_page = alloc_page(GFP_KERNEL_ACCOUNT);1978 if (!status_page)1979 return -ENOMEM;1980 1981 data = page_address(status_page);1982 1983 /*1984 * Firmware expects status page to be in firmware-owned state, otherwise1985 * it will report firmware error code INVALID_PAGE_STATE (0x1A).1986 */1987 if (rmp_mark_pages_firmware(__pa(data), 1, true)) {1988 ret = -EFAULT;1989 goto cleanup;1990 }1991 1992 buf.address = __psp_pa(data);1993 ret = __sev_do_cmd_locked(SEV_CMD_SNP_PLATFORM_STATUS, &buf, &argp->error);1994 1995 /*1996 * Status page will be transitioned to Reclaim state upon success, or1997 * left in Firmware state in failure. Use snp_reclaim_pages() to1998 * transition either case back to Hypervisor-owned state.1999 */2000 if (snp_reclaim_pages(__pa(data), 1, true))2001 return -EFAULT;2002 2003 if (ret)2004 goto cleanup;2005 2006 if (copy_to_user((void __user *)argp->data, data,2007 sizeof(struct sev_user_data_snp_status)))2008 ret = -EFAULT;2009 2010cleanup:2011 __free_pages(status_page, 0);2012 return ret;2013}2014 2015static int sev_ioctl_do_snp_commit(struct sev_issue_cmd *argp)2016{2017 struct sev_device *sev = psp_master->sev_data;2018 struct sev_data_snp_commit buf;2019 2020 if (!sev->snp_initialized)2021 return -EINVAL;2022 2023 buf.len = sizeof(buf);2024 2025 return __sev_do_cmd_locked(SEV_CMD_SNP_COMMIT, &buf, &argp->error);2026}2027 2028static int sev_ioctl_do_snp_set_config(struct sev_issue_cmd *argp, bool writable)2029{2030 struct sev_device *sev = psp_master->sev_data;2031 struct sev_user_data_snp_config config;2032 2033 if (!sev->snp_initialized || !argp->data)2034 return -EINVAL;2035 2036 if (!writable)2037 return -EPERM;2038 2039 if (copy_from_user(&config, (void __user *)argp->data, sizeof(config)))2040 return -EFAULT;2041 2042 return __sev_do_cmd_locked(SEV_CMD_SNP_CONFIG, &config, &argp->error);2043}2044 2045static int sev_ioctl_do_snp_vlek_load(struct sev_issue_cmd *argp, bool writable)2046{2047 struct sev_device *sev = psp_master->sev_data;2048 struct sev_user_data_snp_vlek_load input;2049 void *blob;2050 int ret;2051 2052 if (!sev->snp_initialized || !argp->data)2053 return -EINVAL;2054 2055 if (!writable)2056 return -EPERM;2057 2058 if (copy_from_user(&input, u64_to_user_ptr(argp->data), sizeof(input)))2059 return -EFAULT;2060 2061 if (input.len != sizeof(input) || input.vlek_wrapped_version != 0)2062 return -EINVAL;2063 2064 blob = psp_copy_user_blob(input.vlek_wrapped_address,2065 sizeof(struct sev_user_data_snp_wrapped_vlek_hashstick));2066 if (IS_ERR(blob))2067 return PTR_ERR(blob);2068 2069 input.vlek_wrapped_address = __psp_pa(blob);2070 2071 ret = __sev_do_cmd_locked(SEV_CMD_SNP_VLEK_LOAD, &input, &argp->error);2072 2073 kfree(blob);2074 2075 return ret;2076}2077 2078static long sev_ioctl(struct file *file, unsigned int ioctl, unsigned long arg)2079{2080 void __user *argp = (void __user *)arg;2081 struct sev_issue_cmd input;2082 int ret = -EFAULT;2083 bool writable = file->f_mode & FMODE_WRITE;2084 2085 if (!psp_master || !psp_master->sev_data)2086 return -ENODEV;2087 2088 if (ioctl != SEV_ISSUE_CMD)2089 return -EINVAL;2090 2091 if (copy_from_user(&input, argp, sizeof(struct sev_issue_cmd)))2092 return -EFAULT;2093 2094 if (input.cmd > SEV_MAX)2095 return -EINVAL;2096 2097 mutex_lock(&sev_cmd_mutex);2098 2099 switch (input.cmd) {2100 2101 case SEV_FACTORY_RESET:2102 ret = sev_ioctl_do_reset(&input, writable);2103 break;2104 case SEV_PLATFORM_STATUS:2105 ret = sev_ioctl_do_platform_status(&input);2106 break;2107 case SEV_PEK_GEN:2108 ret = sev_ioctl_do_pek_pdh_gen(SEV_CMD_PEK_GEN, &input, writable);2109 break;2110 case SEV_PDH_GEN:2111 ret = sev_ioctl_do_pek_pdh_gen(SEV_CMD_PDH_GEN, &input, writable);2112 break;2113 case SEV_PEK_CSR:2114 ret = sev_ioctl_do_pek_csr(&input, writable);2115 break;2116 case SEV_PEK_CERT_IMPORT:2117 ret = sev_ioctl_do_pek_import(&input, writable);2118 break;2119 case SEV_PDH_CERT_EXPORT:2120 ret = sev_ioctl_do_pdh_export(&input, writable);2121 break;2122 case SEV_GET_ID:2123 pr_warn_once("SEV_GET_ID command is deprecated, use SEV_GET_ID2\n");2124 ret = sev_ioctl_do_get_id(&input);2125 break;2126 case SEV_GET_ID2:2127 ret = sev_ioctl_do_get_id2(&input);2128 break;2129 case SNP_PLATFORM_STATUS:2130 ret = sev_ioctl_do_snp_platform_status(&input);2131 break;2132 case SNP_COMMIT:2133 ret = sev_ioctl_do_snp_commit(&input);2134 break;2135 case SNP_SET_CONFIG:2136 ret = sev_ioctl_do_snp_set_config(&input, writable);2137 break;2138 case SNP_VLEK_LOAD:2139 ret = sev_ioctl_do_snp_vlek_load(&input, writable);2140 break;2141 default:2142 ret = -EINVAL;2143 goto out;2144 }2145 2146 if (copy_to_user(argp, &input, sizeof(struct sev_issue_cmd)))2147 ret = -EFAULT;2148out:2149 mutex_unlock(&sev_cmd_mutex);2150 2151 return ret;2152}2153 2154static const struct file_operations sev_fops = {2155 .owner = THIS_MODULE,2156 .unlocked_ioctl = sev_ioctl,2157};2158 2159int sev_platform_status(struct sev_user_data_status *data, int *error)2160{2161 return sev_do_cmd(SEV_CMD_PLATFORM_STATUS, data, error);2162}2163EXPORT_SYMBOL_GPL(sev_platform_status);2164 2165int sev_guest_deactivate(struct sev_data_deactivate *data, int *error)2166{2167 return sev_do_cmd(SEV_CMD_DEACTIVATE, data, error);2168}2169EXPORT_SYMBOL_GPL(sev_guest_deactivate);2170 2171int sev_guest_activate(struct sev_data_activate *data, int *error)2172{2173 return sev_do_cmd(SEV_CMD_ACTIVATE, data, error);2174}2175EXPORT_SYMBOL_GPL(sev_guest_activate);2176 2177int sev_guest_decommission(struct sev_data_decommission *data, int *error)2178{2179 return sev_do_cmd(SEV_CMD_DECOMMISSION, data, error);2180}2181EXPORT_SYMBOL_GPL(sev_guest_decommission);2182 2183int sev_guest_df_flush(int *error)2184{2185 return sev_do_cmd(SEV_CMD_DF_FLUSH, NULL, error);2186}2187EXPORT_SYMBOL_GPL(sev_guest_df_flush);2188 2189static void sev_exit(struct kref *ref)2190{2191 misc_deregister(&misc_dev->misc);2192 kfree(misc_dev);2193 misc_dev = NULL;2194}2195 2196static int sev_misc_init(struct sev_device *sev)2197{2198 struct device *dev = sev->dev;2199 int ret;2200 2201 /*2202 * SEV feature support can be detected on multiple devices but the SEV2203 * FW commands must be issued on the master. During probe, we do not2204 * know the master hence we create /dev/sev on the first device probe.2205 * sev_do_cmd() finds the right master device to which to issue the2206 * command to the firmware.2207 */2208 if (!misc_dev) {2209 struct miscdevice *misc;2210 2211 misc_dev = kzalloc(sizeof(*misc_dev), GFP_KERNEL);2212 if (!misc_dev)2213 return -ENOMEM;2214 2215 misc = &misc_dev->misc;2216 misc->minor = MISC_DYNAMIC_MINOR;2217 misc->name = DEVICE_NAME;2218 misc->fops = &sev_fops;2219 2220 ret = misc_register(misc);2221 if (ret)2222 return ret;2223 2224 kref_init(&misc_dev->refcount);2225 } else {2226 kref_get(&misc_dev->refcount);2227 }2228 2229 init_waitqueue_head(&sev->int_queue);2230 sev->misc = misc_dev;2231 dev_dbg(dev, "registered SEV device\n");2232 2233 return 0;2234}2235 2236int sev_dev_init(struct psp_device *psp)2237{2238 struct device *dev = psp->dev;2239 struct sev_device *sev;2240 int ret = -ENOMEM;2241 2242 if (!boot_cpu_has(X86_FEATURE_SEV)) {2243 dev_info_once(dev, "SEV: memory encryption not enabled by BIOS\n");2244 return 0;2245 }2246 2247 sev = devm_kzalloc(dev, sizeof(*sev), GFP_KERNEL);2248 if (!sev)2249 goto e_err;2250 2251 sev->cmd_buf = (void *)devm_get_free_pages(dev, GFP_KERNEL, 1);2252 if (!sev->cmd_buf)2253 goto e_sev;2254 2255 sev->cmd_buf_backup = (uint8_t *)sev->cmd_buf + PAGE_SIZE;2256 2257 psp->sev_data = sev;2258 2259 sev->dev = dev;2260 sev->psp = psp;2261 2262 sev->io_regs = psp->io_regs;2263 2264 sev->vdata = (struct sev_vdata *)psp->vdata->sev;2265 if (!sev->vdata) {2266 ret = -ENODEV;2267 dev_err(dev, "sev: missing driver data\n");2268 goto e_buf;2269 }2270 2271 psp_set_sev_irq_handler(psp, sev_irq_handler, sev);2272 2273 ret = sev_misc_init(sev);2274 if (ret)2275 goto e_irq;2276 2277 dev_notice(dev, "sev enabled\n");2278 2279 return 0;2280 2281e_irq:2282 psp_clear_sev_irq_handler(psp);2283e_buf:2284 devm_free_pages(dev, (unsigned long)sev->cmd_buf);2285e_sev:2286 devm_kfree(dev, sev);2287e_err:2288 psp->sev_data = NULL;2289 2290 dev_notice(dev, "sev initialization failed\n");2291 2292 return ret;2293}2294 2295static void __sev_firmware_shutdown(struct sev_device *sev, bool panic)2296{2297 int error;2298 2299 __sev_platform_shutdown_locked(NULL);2300 2301 if (sev_es_tmr) {2302 /*2303 * The TMR area was encrypted, flush it from the cache.2304 *2305 * If invoked during panic handling, local interrupts are2306 * disabled and all CPUs are stopped, so wbinvd_on_all_cpus()2307 * can't be used. In that case, wbinvd() is done on remote CPUs2308 * via the NMI callback, and done for this CPU later during2309 * SNP shutdown, so wbinvd_on_all_cpus() can be skipped.2310 */2311 if (!panic)2312 wbinvd_on_all_cpus();2313 2314 __snp_free_firmware_pages(virt_to_page(sev_es_tmr),2315 get_order(sev_es_tmr_size),2316 true);2317 sev_es_tmr = NULL;2318 }2319 2320 if (sev_init_ex_buffer) {2321 __snp_free_firmware_pages(virt_to_page(sev_init_ex_buffer),2322 get_order(NV_LENGTH),2323 true);2324 sev_init_ex_buffer = NULL;2325 }2326 2327 if (snp_range_list) {2328 kfree(snp_range_list);2329 snp_range_list = NULL;2330 }2331 2332 __sev_snp_shutdown_locked(&error, panic);2333}2334 2335static void sev_firmware_shutdown(struct sev_device *sev)2336{2337 mutex_lock(&sev_cmd_mutex);2338 __sev_firmware_shutdown(sev, false);2339 mutex_unlock(&sev_cmd_mutex);2340}2341 2342void sev_dev_destroy(struct psp_device *psp)2343{2344 struct sev_device *sev = psp->sev_data;2345 2346 if (!sev)2347 return;2348 2349 sev_firmware_shutdown(sev);2350 2351 if (sev->misc)2352 kref_put(&misc_dev->refcount, sev_exit);2353 2354 psp_clear_sev_irq_handler(psp);2355}2356 2357static int snp_shutdown_on_panic(struct notifier_block *nb,2358 unsigned long reason, void *arg)2359{2360 struct sev_device *sev = psp_master->sev_data;2361 2362 /*2363 * If sev_cmd_mutex is already acquired, then it's likely2364 * another PSP command is in flight and issuing a shutdown2365 * would fail in unexpected ways. Rather than create even2366 * more confusion during a panic, just bail out here.2367 */2368 if (mutex_is_locked(&sev_cmd_mutex))2369 return NOTIFY_DONE;2370 2371 __sev_firmware_shutdown(sev, true);2372 2373 return NOTIFY_DONE;2374}2375 2376static struct notifier_block snp_panic_notifier = {2377 .notifier_call = snp_shutdown_on_panic,2378};2379 2380int sev_issue_cmd_external_user(struct file *filep, unsigned int cmd,2381 void *data, int *error)2382{2383 if (!filep || filep->f_op != &sev_fops)2384 return -EBADF;2385 2386 return sev_do_cmd(cmd, data, error);2387}2388EXPORT_SYMBOL_GPL(sev_issue_cmd_external_user);2389 2390void sev_pci_init(void)2391{2392 struct sev_device *sev = psp_master->sev_data;2393 struct sev_platform_init_args args = {0};2394 u8 api_major, api_minor, build;2395 int rc;2396 2397 if (!sev)2398 return;2399 2400 psp_timeout = psp_probe_timeout;2401 2402 if (sev_get_api_version())2403 goto err;2404 2405 api_major = sev->api_major;2406 api_minor = sev->api_minor;2407 build = sev->build;2408 2409 if (sev_update_firmware(sev->dev) == 0)2410 sev_get_api_version();2411 2412 if (api_major != sev->api_major || api_minor != sev->api_minor ||2413 build != sev->build)2414 dev_info(sev->dev, "SEV firmware updated from %d.%d.%d to %d.%d.%d\n",2415 api_major, api_minor, build,2416 sev->api_major, sev->api_minor, sev->build);2417 2418 /* Initialize the platform */2419 args.probe = true;2420 rc = sev_platform_init(&args);2421 if (rc)2422 dev_err(sev->dev, "SEV: failed to INIT error %#x, rc %d\n",2423 args.error, rc);2424 2425 dev_info(sev->dev, "SEV%s API:%d.%d build:%d\n", sev->snp_initialized ?2426 "-SNP" : "", sev->api_major, sev->api_minor, sev->build);2427 2428 atomic_notifier_chain_register(&panic_notifier_list,2429 &snp_panic_notifier);2430 return;2431 2432err:2433 sev_dev_destroy(psp_master);2434 2435 psp_master->sev_data = NULL;2436}2437 2438void sev_pci_exit(void)2439{2440 struct sev_device *sev = psp_master->sev_data;2441 2442 if (!sev)2443 return;2444 2445 sev_firmware_shutdown(sev);2446 2447 atomic_notifier_chain_unregister(&panic_notifier_list,2448 &snp_panic_notifier);2449}2450