1 // SPDX-License-Identifier: GPL-2.0-only
3 * Kernel-based Virtual Machine driver for Linux
7 * Copyright 2010 Red Hat, Inc. and/or its affiliates.
10 #include <linux/kvm_types.h>
11 #include <linux/kvm_host.h>
12 #include <linux/kernel.h>
13 #include <linux/highmem.h>
14 #include <linux/psp-sev.h>
15 #include <linux/pagemap.h>
16 #include <linux/swap.h>
21 static int sev_flush_asids(void);
22 static DECLARE_RWSEM(sev_deactivate_lock);
23 static DEFINE_MUTEX(sev_bitmap_lock);
24 unsigned int max_sev_asid;
25 static unsigned int min_sev_asid;
26 static unsigned long *sev_asid_bitmap;
27 static unsigned long *sev_reclaim_asid_bitmap;
28 #define __sme_page_pa(x) __sme_set(page_to_pfn(x) << PAGE_SHIFT)
31 struct list_head list;
38 static int sev_flush_asids(void)
43 * DEACTIVATE will clear the WBINVD indicator causing DF_FLUSH to fail,
44 * so it must be guarded.
46 down_write(&sev_deactivate_lock);
49 ret = sev_guest_df_flush(&error);
51 up_write(&sev_deactivate_lock);
54 pr_err("SEV: DF_FLUSH failed, ret=%d, error=%#x\n", ret, error);
59 /* Must be called with the sev_bitmap_lock held */
60 static bool __sev_recycle_asids(void)
64 /* Check if there are any ASIDs to reclaim before performing a flush */
65 pos = find_next_bit(sev_reclaim_asid_bitmap,
66 max_sev_asid, min_sev_asid - 1);
67 if (pos >= max_sev_asid)
70 if (sev_flush_asids())
73 bitmap_xor(sev_asid_bitmap, sev_asid_bitmap, sev_reclaim_asid_bitmap,
75 bitmap_zero(sev_reclaim_asid_bitmap, max_sev_asid);
80 static int sev_asid_new(void)
85 mutex_lock(&sev_bitmap_lock);
88 * SEV-enabled guest must use asid from min_sev_asid to max_sev_asid.
91 pos = find_next_zero_bit(sev_asid_bitmap, max_sev_asid, min_sev_asid - 1);
92 if (pos >= max_sev_asid) {
93 if (retry && __sev_recycle_asids()) {
97 mutex_unlock(&sev_bitmap_lock);
101 __set_bit(pos, sev_asid_bitmap);
103 mutex_unlock(&sev_bitmap_lock);
108 static int sev_get_asid(struct kvm *kvm)
110 struct kvm_sev_info *sev = &to_kvm_svm(kvm)->sev_info;
115 static void sev_asid_free(int asid)
117 struct svm_cpu_data *sd;
120 mutex_lock(&sev_bitmap_lock);
123 __set_bit(pos, sev_reclaim_asid_bitmap);
125 for_each_possible_cpu(cpu) {
126 sd = per_cpu(svm_data, cpu);
127 sd->sev_vmcbs[pos] = NULL;
130 mutex_unlock(&sev_bitmap_lock);
133 static void sev_unbind_asid(struct kvm *kvm, unsigned int handle)
135 struct sev_data_decommission *decommission;
136 struct sev_data_deactivate *data;
141 data = kzalloc(sizeof(*data), GFP_KERNEL);
145 /* deactivate handle */
146 data->handle = handle;
148 /* Guard DEACTIVATE against WBINVD/DF_FLUSH used in ASID recycling */
149 down_read(&sev_deactivate_lock);
150 sev_guest_deactivate(data, NULL);
151 up_read(&sev_deactivate_lock);
155 decommission = kzalloc(sizeof(*decommission), GFP_KERNEL);
159 /* decommission handle */
160 decommission->handle = handle;
161 sev_guest_decommission(decommission, NULL);
166 static int sev_guest_init(struct kvm *kvm, struct kvm_sev_cmd *argp)
168 struct kvm_sev_info *sev = &to_kvm_svm(kvm)->sev_info;
172 if (unlikely(sev->active))
175 asid = sev_asid_new();
179 ret = sev_platform_init(&argp->error);
185 INIT_LIST_HEAD(&sev->regions_list);
194 static int sev_bind_asid(struct kvm *kvm, unsigned int handle, int *error)
196 struct sev_data_activate *data;
197 int asid = sev_get_asid(kvm);
200 data = kzalloc(sizeof(*data), GFP_KERNEL_ACCOUNT);
204 /* activate ASID on the given handle */
205 data->handle = handle;
207 ret = sev_guest_activate(data, error);
213 static int __sev_issue_cmd(int fd, int id, void *data, int *error)
222 ret = sev_issue_cmd_external_user(f.file, id, data, error);
228 static int sev_issue_cmd(struct kvm *kvm, int id, void *data, int *error)
230 struct kvm_sev_info *sev = &to_kvm_svm(kvm)->sev_info;
232 return __sev_issue_cmd(sev->fd, id, data, error);
235 static int sev_launch_start(struct kvm *kvm, struct kvm_sev_cmd *argp)
237 struct kvm_sev_info *sev = &to_kvm_svm(kvm)->sev_info;
238 struct sev_data_launch_start *start;
239 struct kvm_sev_launch_start params;
240 void *dh_blob, *session_blob;
241 int *error = &argp->error;
247 if (copy_from_user(¶ms, (void __user *)(uintptr_t)argp->data, sizeof(params)))
250 start = kzalloc(sizeof(*start), GFP_KERNEL_ACCOUNT);
255 if (params.dh_uaddr) {
256 dh_blob = psp_copy_user_blob(params.dh_uaddr, params.dh_len);
257 if (IS_ERR(dh_blob)) {
258 ret = PTR_ERR(dh_blob);
262 start->dh_cert_address = __sme_set(__pa(dh_blob));
263 start->dh_cert_len = params.dh_len;
267 if (params.session_uaddr) {
268 session_blob = psp_copy_user_blob(params.session_uaddr, params.session_len);
269 if (IS_ERR(session_blob)) {
270 ret = PTR_ERR(session_blob);
274 start->session_address = __sme_set(__pa(session_blob));
275 start->session_len = params.session_len;
278 start->handle = params.handle;
279 start->policy = params.policy;
281 /* create memory encryption context */
282 ret = __sev_issue_cmd(argp->sev_fd, SEV_CMD_LAUNCH_START, start, error);
286 /* Bind ASID to this guest */
287 ret = sev_bind_asid(kvm, start->handle, error);
291 /* return handle to userspace */
292 params.handle = start->handle;
293 if (copy_to_user((void __user *)(uintptr_t)argp->data, ¶ms, sizeof(params))) {
294 sev_unbind_asid(kvm, start->handle);
299 sev->handle = start->handle;
300 sev->fd = argp->sev_fd;
311 static struct page **sev_pin_memory(struct kvm *kvm, unsigned long uaddr,
312 unsigned long ulen, unsigned long *n,
315 struct kvm_sev_info *sev = &to_kvm_svm(kvm)->sev_info;
316 unsigned long npages, size;
318 unsigned long locked, lock_limit;
320 unsigned long first, last;
322 if (ulen == 0 || uaddr + ulen < uaddr)
323 return ERR_PTR(-EINVAL);
325 /* Calculate number of pages. */
326 first = (uaddr & PAGE_MASK) >> PAGE_SHIFT;
327 last = ((uaddr + ulen - 1) & PAGE_MASK) >> PAGE_SHIFT;
328 npages = (last - first + 1);
330 locked = sev->pages_locked + npages;
331 lock_limit = rlimit(RLIMIT_MEMLOCK) >> PAGE_SHIFT;
332 if (locked > lock_limit && !capable(CAP_IPC_LOCK)) {
333 pr_err("SEV: %lu locked pages exceed the lock limit of %lu.\n", locked, lock_limit);
334 return ERR_PTR(-ENOMEM);
337 if (WARN_ON_ONCE(npages > INT_MAX))
338 return ERR_PTR(-EINVAL);
340 /* Avoid using vmalloc for smaller buffers. */
341 size = npages * sizeof(struct page *);
342 if (size > PAGE_SIZE)
343 pages = __vmalloc(size, GFP_KERNEL_ACCOUNT | __GFP_ZERO);
345 pages = kmalloc(size, GFP_KERNEL_ACCOUNT);
348 return ERR_PTR(-ENOMEM);
350 /* Pin the user virtual address. */
351 npinned = pin_user_pages_fast(uaddr, npages, write ? FOLL_WRITE : 0, pages);
352 if (npinned != npages) {
353 pr_err("SEV: Failure locking %lu pages.\n", npages);
358 sev->pages_locked = locked;
364 unpin_user_pages(pages, npinned);
369 return ERR_PTR(npinned);
372 static void sev_unpin_memory(struct kvm *kvm, struct page **pages,
373 unsigned long npages)
375 struct kvm_sev_info *sev = &to_kvm_svm(kvm)->sev_info;
377 unpin_user_pages(pages, npages);
379 sev->pages_locked -= npages;
382 static void sev_clflush_pages(struct page *pages[], unsigned long npages)
384 uint8_t *page_virtual;
387 if (npages == 0 || pages == NULL)
390 for (i = 0; i < npages; i++) {
391 page_virtual = kmap_atomic(pages[i]);
392 clflush_cache_range(page_virtual, PAGE_SIZE);
393 kunmap_atomic(page_virtual);
397 static unsigned long get_num_contig_pages(unsigned long idx,
398 struct page **inpages, unsigned long npages)
400 unsigned long paddr, next_paddr;
401 unsigned long i = idx + 1, pages = 1;
403 /* find the number of contiguous pages starting from idx */
404 paddr = __sme_page_pa(inpages[idx]);
406 next_paddr = __sme_page_pa(inpages[i++]);
407 if ((paddr + PAGE_SIZE) == next_paddr) {
418 static int sev_launch_update_data(struct kvm *kvm, struct kvm_sev_cmd *argp)
420 unsigned long vaddr, vaddr_end, next_vaddr, npages, pages, size, i;
421 struct kvm_sev_info *sev = &to_kvm_svm(kvm)->sev_info;
422 struct kvm_sev_launch_update_data params;
423 struct sev_data_launch_update_data *data;
424 struct page **inpages;
430 if (copy_from_user(¶ms, (void __user *)(uintptr_t)argp->data, sizeof(params)))
433 data = kzalloc(sizeof(*data), GFP_KERNEL_ACCOUNT);
437 vaddr = params.uaddr;
439 vaddr_end = vaddr + size;
441 /* Lock the user memory. */
442 inpages = sev_pin_memory(kvm, vaddr, size, &npages, 1);
449 * The LAUNCH_UPDATE command will perform in-place encryption of the
450 * memory content (i.e it will write the same memory region with C=1).
451 * It's possible that the cache may contain the data with C=0, i.e.,
452 * unencrypted so invalidate it first.
454 sev_clflush_pages(inpages, npages);
456 for (i = 0; vaddr < vaddr_end; vaddr = next_vaddr, i += pages) {
460 * If the user buffer is not page-aligned, calculate the offset
463 offset = vaddr & (PAGE_SIZE - 1);
465 /* Calculate the number of pages that can be encrypted in one go. */
466 pages = get_num_contig_pages(i, inpages, npages);
468 len = min_t(size_t, ((pages * PAGE_SIZE) - offset), size);
470 data->handle = sev->handle;
472 data->address = __sme_page_pa(inpages[i]) + offset;
473 ret = sev_issue_cmd(kvm, SEV_CMD_LAUNCH_UPDATE_DATA, data, &argp->error);
478 next_vaddr = vaddr + len;
482 /* content of memory is updated, mark pages dirty */
483 for (i = 0; i < npages; i++) {
484 set_page_dirty_lock(inpages[i]);
485 mark_page_accessed(inpages[i]);
487 /* unlock the user pages */
488 sev_unpin_memory(kvm, inpages, npages);
494 static int sev_launch_measure(struct kvm *kvm, struct kvm_sev_cmd *argp)
496 void __user *measure = (void __user *)(uintptr_t)argp->data;
497 struct kvm_sev_info *sev = &to_kvm_svm(kvm)->sev_info;
498 struct sev_data_launch_measure *data;
499 struct kvm_sev_launch_measure params;
500 void __user *p = NULL;
507 if (copy_from_user(¶ms, measure, sizeof(params)))
510 data = kzalloc(sizeof(*data), GFP_KERNEL_ACCOUNT);
514 /* User wants to query the blob length */
518 p = (void __user *)(uintptr_t)params.uaddr;
520 if (params.len > SEV_FW_BLOB_MAX_SIZE) {
526 blob = kmalloc(params.len, GFP_KERNEL);
530 data->address = __psp_pa(blob);
531 data->len = params.len;
535 data->handle = sev->handle;
536 ret = sev_issue_cmd(kvm, SEV_CMD_LAUNCH_MEASURE, data, &argp->error);
539 * If we query the session length, FW responded with expected data.
548 if (copy_to_user(p, blob, params.len))
553 params.len = data->len;
554 if (copy_to_user(measure, ¶ms, sizeof(params)))
563 static int sev_launch_finish(struct kvm *kvm, struct kvm_sev_cmd *argp)
565 struct kvm_sev_info *sev = &to_kvm_svm(kvm)->sev_info;
566 struct sev_data_launch_finish *data;
572 data = kzalloc(sizeof(*data), GFP_KERNEL_ACCOUNT);
576 data->handle = sev->handle;
577 ret = sev_issue_cmd(kvm, SEV_CMD_LAUNCH_FINISH, data, &argp->error);
583 static int sev_guest_status(struct kvm *kvm, struct kvm_sev_cmd *argp)
585 struct kvm_sev_info *sev = &to_kvm_svm(kvm)->sev_info;
586 struct kvm_sev_guest_status params;
587 struct sev_data_guest_status *data;
593 data = kzalloc(sizeof(*data), GFP_KERNEL_ACCOUNT);
597 data->handle = sev->handle;
598 ret = sev_issue_cmd(kvm, SEV_CMD_GUEST_STATUS, data, &argp->error);
602 params.policy = data->policy;
603 params.state = data->state;
604 params.handle = data->handle;
606 if (copy_to_user((void __user *)(uintptr_t)argp->data, ¶ms, sizeof(params)))
613 static int __sev_issue_dbg_cmd(struct kvm *kvm, unsigned long src,
614 unsigned long dst, int size,
615 int *error, bool enc)
617 struct kvm_sev_info *sev = &to_kvm_svm(kvm)->sev_info;
618 struct sev_data_dbg *data;
621 data = kzalloc(sizeof(*data), GFP_KERNEL_ACCOUNT);
625 data->handle = sev->handle;
626 data->dst_addr = dst;
627 data->src_addr = src;
630 ret = sev_issue_cmd(kvm,
631 enc ? SEV_CMD_DBG_ENCRYPT : SEV_CMD_DBG_DECRYPT,
637 static int __sev_dbg_decrypt(struct kvm *kvm, unsigned long src_paddr,
638 unsigned long dst_paddr, int sz, int *err)
643 * Its safe to read more than we are asked, caller should ensure that
644 * destination has enough space.
646 src_paddr = round_down(src_paddr, 16);
647 offset = src_paddr & 15;
648 sz = round_up(sz + offset, 16);
650 return __sev_issue_dbg_cmd(kvm, src_paddr, dst_paddr, sz, err, false);
653 static int __sev_dbg_decrypt_user(struct kvm *kvm, unsigned long paddr,
654 unsigned long __user dst_uaddr,
655 unsigned long dst_paddr,
658 struct page *tpage = NULL;
661 /* if inputs are not 16-byte then use intermediate buffer */
662 if (!IS_ALIGNED(dst_paddr, 16) ||
663 !IS_ALIGNED(paddr, 16) ||
664 !IS_ALIGNED(size, 16)) {
665 tpage = (void *)alloc_page(GFP_KERNEL);
669 dst_paddr = __sme_page_pa(tpage);
672 ret = __sev_dbg_decrypt(kvm, paddr, dst_paddr, size, err);
678 if (copy_to_user((void __user *)(uintptr_t)dst_uaddr,
679 page_address(tpage) + offset, size))
690 static int __sev_dbg_encrypt_user(struct kvm *kvm, unsigned long paddr,
691 unsigned long __user vaddr,
692 unsigned long dst_paddr,
693 unsigned long __user dst_vaddr,
694 int size, int *error)
696 struct page *src_tpage = NULL;
697 struct page *dst_tpage = NULL;
700 /* If source buffer is not aligned then use an intermediate buffer */
701 if (!IS_ALIGNED(vaddr, 16)) {
702 src_tpage = alloc_page(GFP_KERNEL);
706 if (copy_from_user(page_address(src_tpage),
707 (void __user *)(uintptr_t)vaddr, size)) {
708 __free_page(src_tpage);
712 paddr = __sme_page_pa(src_tpage);
716 * If destination buffer or length is not aligned then do read-modify-write:
717 * - decrypt destination in an intermediate buffer
718 * - copy the source buffer in an intermediate buffer
719 * - use the intermediate buffer as source buffer
721 if (!IS_ALIGNED(dst_vaddr, 16) || !IS_ALIGNED(size, 16)) {
724 dst_tpage = alloc_page(GFP_KERNEL);
730 ret = __sev_dbg_decrypt(kvm, dst_paddr,
731 __sme_page_pa(dst_tpage), size, error);
736 * If source is kernel buffer then use memcpy() otherwise
739 dst_offset = dst_paddr & 15;
742 memcpy(page_address(dst_tpage) + dst_offset,
743 page_address(src_tpage), size);
745 if (copy_from_user(page_address(dst_tpage) + dst_offset,
746 (void __user *)(uintptr_t)vaddr, size)) {
752 paddr = __sme_page_pa(dst_tpage);
753 dst_paddr = round_down(dst_paddr, 16);
754 len = round_up(size, 16);
757 ret = __sev_issue_dbg_cmd(kvm, paddr, dst_paddr, len, error, true);
761 __free_page(src_tpage);
763 __free_page(dst_tpage);
767 static int sev_dbg_crypt(struct kvm *kvm, struct kvm_sev_cmd *argp, bool dec)
769 unsigned long vaddr, vaddr_end, next_vaddr;
770 unsigned long dst_vaddr;
771 struct page **src_p, **dst_p;
772 struct kvm_sev_dbg debug;
780 if (copy_from_user(&debug, (void __user *)(uintptr_t)argp->data, sizeof(debug)))
783 if (!debug.len || debug.src_uaddr + debug.len < debug.src_uaddr)
785 if (!debug.dst_uaddr)
788 vaddr = debug.src_uaddr;
790 vaddr_end = vaddr + size;
791 dst_vaddr = debug.dst_uaddr;
793 for (; vaddr < vaddr_end; vaddr = next_vaddr) {
794 int len, s_off, d_off;
796 /* lock userspace source and destination page */
797 src_p = sev_pin_memory(kvm, vaddr & PAGE_MASK, PAGE_SIZE, &n, 0);
801 dst_p = sev_pin_memory(kvm, dst_vaddr & PAGE_MASK, PAGE_SIZE, &n, 1);
803 sev_unpin_memory(kvm, src_p, n);
808 * The DBG_{DE,EN}CRYPT commands will perform {dec,en}cryption of the
809 * memory content (i.e it will write the same memory region with C=1).
810 * It's possible that the cache may contain the data with C=0, i.e.,
811 * unencrypted so invalidate it first.
813 sev_clflush_pages(src_p, 1);
814 sev_clflush_pages(dst_p, 1);
817 * Since user buffer may not be page aligned, calculate the
818 * offset within the page.
820 s_off = vaddr & ~PAGE_MASK;
821 d_off = dst_vaddr & ~PAGE_MASK;
822 len = min_t(size_t, (PAGE_SIZE - s_off), size);
825 ret = __sev_dbg_decrypt_user(kvm,
826 __sme_page_pa(src_p[0]) + s_off,
828 __sme_page_pa(dst_p[0]) + d_off,
831 ret = __sev_dbg_encrypt_user(kvm,
832 __sme_page_pa(src_p[0]) + s_off,
834 __sme_page_pa(dst_p[0]) + d_off,
838 sev_unpin_memory(kvm, src_p, n);
839 sev_unpin_memory(kvm, dst_p, n);
844 next_vaddr = vaddr + len;
845 dst_vaddr = dst_vaddr + len;
852 static int sev_launch_secret(struct kvm *kvm, struct kvm_sev_cmd *argp)
854 struct kvm_sev_info *sev = &to_kvm_svm(kvm)->sev_info;
855 struct sev_data_launch_secret *data;
856 struct kvm_sev_launch_secret params;
865 if (copy_from_user(¶ms, (void __user *)(uintptr_t)argp->data, sizeof(params)))
868 pages = sev_pin_memory(kvm, params.guest_uaddr, params.guest_len, &n, 1);
870 return PTR_ERR(pages);
873 * The secret must be copied into contiguous memory region, lets verify
874 * that userspace memory pages are contiguous before we issue command.
876 if (get_num_contig_pages(0, pages, n) != n) {
882 data = kzalloc(sizeof(*data), GFP_KERNEL_ACCOUNT);
886 offset = params.guest_uaddr & (PAGE_SIZE - 1);
887 data->guest_address = __sme_page_pa(pages[0]) + offset;
888 data->guest_len = params.guest_len;
890 blob = psp_copy_user_blob(params.trans_uaddr, params.trans_len);
896 data->trans_address = __psp_pa(blob);
897 data->trans_len = params.trans_len;
899 hdr = psp_copy_user_blob(params.hdr_uaddr, params.hdr_len);
904 data->hdr_address = __psp_pa(hdr);
905 data->hdr_len = params.hdr_len;
907 data->handle = sev->handle;
908 ret = sev_issue_cmd(kvm, SEV_CMD_LAUNCH_UPDATE_SECRET, data, &argp->error);
917 sev_unpin_memory(kvm, pages, n);
921 int svm_mem_enc_op(struct kvm *kvm, void __user *argp)
923 struct kvm_sev_cmd sev_cmd;
926 if (!svm_sev_enabled())
932 if (copy_from_user(&sev_cmd, argp, sizeof(struct kvm_sev_cmd)))
935 mutex_lock(&kvm->lock);
937 switch (sev_cmd.id) {
939 r = sev_guest_init(kvm, &sev_cmd);
941 case KVM_SEV_LAUNCH_START:
942 r = sev_launch_start(kvm, &sev_cmd);
944 case KVM_SEV_LAUNCH_UPDATE_DATA:
945 r = sev_launch_update_data(kvm, &sev_cmd);
947 case KVM_SEV_LAUNCH_MEASURE:
948 r = sev_launch_measure(kvm, &sev_cmd);
950 case KVM_SEV_LAUNCH_FINISH:
951 r = sev_launch_finish(kvm, &sev_cmd);
953 case KVM_SEV_GUEST_STATUS:
954 r = sev_guest_status(kvm, &sev_cmd);
956 case KVM_SEV_DBG_DECRYPT:
957 r = sev_dbg_crypt(kvm, &sev_cmd, true);
959 case KVM_SEV_DBG_ENCRYPT:
960 r = sev_dbg_crypt(kvm, &sev_cmd, false);
962 case KVM_SEV_LAUNCH_SECRET:
963 r = sev_launch_secret(kvm, &sev_cmd);
970 if (copy_to_user(argp, &sev_cmd, sizeof(struct kvm_sev_cmd)))
974 mutex_unlock(&kvm->lock);
978 int svm_register_enc_region(struct kvm *kvm,
979 struct kvm_enc_region *range)
981 struct kvm_sev_info *sev = &to_kvm_svm(kvm)->sev_info;
982 struct enc_region *region;
988 if (range->addr > ULONG_MAX || range->size > ULONG_MAX)
991 region = kzalloc(sizeof(*region), GFP_KERNEL_ACCOUNT);
995 region->pages = sev_pin_memory(kvm, range->addr, range->size, ®ion->npages, 1);
996 if (IS_ERR(region->pages)) {
997 ret = PTR_ERR(region->pages);
1002 * The guest may change the memory encryption attribute from C=0 -> C=1
1003 * or vice versa for this memory range. Lets make sure caches are
1004 * flushed to ensure that guest data gets written into memory with
1007 sev_clflush_pages(region->pages, region->npages);
1009 region->uaddr = range->addr;
1010 region->size = range->size;
1012 mutex_lock(&kvm->lock);
1013 list_add_tail(®ion->list, &sev->regions_list);
1014 mutex_unlock(&kvm->lock);
1023 static struct enc_region *
1024 find_enc_region(struct kvm *kvm, struct kvm_enc_region *range)
1026 struct kvm_sev_info *sev = &to_kvm_svm(kvm)->sev_info;
1027 struct list_head *head = &sev->regions_list;
1028 struct enc_region *i;
1030 list_for_each_entry(i, head, list) {
1031 if (i->uaddr == range->addr &&
1032 i->size == range->size)
1039 static void __unregister_enc_region_locked(struct kvm *kvm,
1040 struct enc_region *region)
1042 sev_unpin_memory(kvm, region->pages, region->npages);
1043 list_del(®ion->list);
1047 int svm_unregister_enc_region(struct kvm *kvm,
1048 struct kvm_enc_region *range)
1050 struct enc_region *region;
1053 mutex_lock(&kvm->lock);
1055 if (!sev_guest(kvm)) {
1060 region = find_enc_region(kvm, range);
1067 * Ensure that all guest tagged cache entries are flushed before
1068 * releasing the pages back to the system for use. CLFLUSH will
1069 * not do this, so issue a WBINVD.
1071 wbinvd_on_all_cpus();
1073 __unregister_enc_region_locked(kvm, region);
1075 mutex_unlock(&kvm->lock);
1079 mutex_unlock(&kvm->lock);
1083 void sev_vm_destroy(struct kvm *kvm)
1085 struct kvm_sev_info *sev = &to_kvm_svm(kvm)->sev_info;
1086 struct list_head *head = &sev->regions_list;
1087 struct list_head *pos, *q;
1089 if (!sev_guest(kvm))
1092 mutex_lock(&kvm->lock);
1095 * Ensure that all guest tagged cache entries are flushed before
1096 * releasing the pages back to the system for use. CLFLUSH will
1097 * not do this, so issue a WBINVD.
1099 wbinvd_on_all_cpus();
1102 * if userspace was terminated before unregistering the memory regions
1103 * then lets unpin all the registered memory.
1105 if (!list_empty(head)) {
1106 list_for_each_safe(pos, q, head) {
1107 __unregister_enc_region_locked(kvm,
1108 list_entry(pos, struct enc_region, list));
1112 mutex_unlock(&kvm->lock);
1114 sev_unbind_asid(kvm, sev->handle);
1115 sev_asid_free(sev->asid);
1118 int __init sev_hardware_setup(void)
1120 struct sev_user_data_status *status;
1123 /* Maximum number of encrypted guests supported simultaneously */
1124 max_sev_asid = cpuid_ecx(0x8000001F);
1126 if (!svm_sev_enabled())
1129 /* Minimum ASID value that should be used for SEV guest */
1130 min_sev_asid = cpuid_edx(0x8000001F);
1132 /* Initialize SEV ASID bitmaps */
1133 sev_asid_bitmap = bitmap_zalloc(max_sev_asid, GFP_KERNEL);
1134 if (!sev_asid_bitmap)
1137 sev_reclaim_asid_bitmap = bitmap_zalloc(max_sev_asid, GFP_KERNEL);
1138 if (!sev_reclaim_asid_bitmap)
1141 status = kmalloc(sizeof(*status), GFP_KERNEL);
1146 * Check SEV platform status.
1148 * PLATFORM_STATUS can be called in any state, if we failed to query
1149 * the PLATFORM status then either PSP firmware does not support SEV
1150 * feature or SEV firmware is dead.
1152 rc = sev_platform_status(status, NULL);
1156 pr_info("SEV supported\n");
1163 void sev_hardware_teardown(void)
1165 if (!svm_sev_enabled())
1168 bitmap_free(sev_asid_bitmap);
1169 bitmap_free(sev_reclaim_asid_bitmap);
1174 void pre_sev_run(struct vcpu_svm *svm, int cpu)
1176 struct svm_cpu_data *sd = per_cpu(svm_data, cpu);
1177 int asid = sev_get_asid(svm->vcpu.kvm);
1179 /* Assign the asid allocated with this SEV guest */
1180 svm->vmcb->control.asid = asid;
1185 * 1) when different VMCB for the same ASID is to be run on the same host CPU.
1186 * 2) or this VMCB was executed on different host CPU in previous VMRUNs.
1188 if (sd->sev_vmcbs[asid] == svm->vmcb &&
1189 svm->vcpu.arch.last_vmentry_cpu == cpu)
1192 sd->sev_vmcbs[asid] = svm->vmcb;
1193 svm->vmcb->control.tlb_ctl = TLB_CONTROL_FLUSH_ASID;
1194 mark_dirty(svm->vmcb, VMCB_ASID);