1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) Qualcomm Technologies, Inc. and/or its subsidiaries. */
3
4 #include <linux/gunyah.h>
5 #include <linux/init.h>
6 #include <linux/io.h>
7 #include <linux/pgtable.h>
8 #include <linux/virtio_balloon.h>
9
10 #include <asm/hypervisor.h>
11
12 #define ADDRSPACE_INFO_AREA_ROOTVM_ADDRSPACE_CAP ((uint16_t)0)
13 struct addrspace_info_area_rootvm_addrspace_cap {
14 u64 addrspace_cap;
15 u32 rights;
16 u32 res0;
17 };
18
19 static u64 our_addrspace_capid;
20
gunyah_mmio_guard_ioremap_hook(phys_addr_t phys,size_t size,pgprot_t * prot)21 static int gunyah_mmio_guard_ioremap_hook(phys_addr_t phys, size_t size, pgprot_t *prot)
22 {
23 pteval_t protval = pgprot_val(*prot);
24 int ret;
25
26 /*
27 * We only expect MMIO emulation for regions mapped with device
28 * attributes.
29 */
30 if (protval != PROT_DEVICE_nGnRE && protval != PROT_DEVICE_nGnRnE)
31 return 0;
32
33 ret = gunyah_hypercall_addrspc_configure_vmmio_range(our_addrspace_capid,
34 phys, size, GUNYAH_ADDRSPACE_VMMIO_CONFIGURE_OP_ADD_RANGE);
35
36 if (ret == GUNYAH_ERROR_UNIMPLEMENTED || ret == GUNYAH_ERROR_BUSY
37 || ret == GUNYAH_ERROR_CSPACE_INSUF_RIGHTS)
38 /* Gunyah would have configured VMMIO via DT */
39 ret = GUNYAH_ERROR_OK;
40
41 return gunyah_error_remap(ret);
42 }
43
44 #ifdef CONFIG_VIRTIO_BALLOON_HYP_OPS
gunyah_page_relinquish(struct page * page,unsigned int nr)45 static void gunyah_page_relinquish(struct page *page, unsigned int nr)
46 {
47 /* Release page to Host, so unlock and sanitize */
48 u64 flags = BIT_ULL(GUNYAH_ADDRSPC_MODIFY_FLAG_UNLOCK_BIT) |
49 BIT_ULL(GUNYAH_ADDRSPC_MODIFY_FLAG_SANITIZE_BIT);
50 phys_addr_t phys, end;
51 int ret = 0;
52
53 phys = page_to_phys(page);
54 end = phys + PAGE_SIZE * nr;
55
56 while (phys < end) {
57 ret = gunyah_hypercall_addrspc_modify_pages(our_addrspace_capid,
58 phys, PAGE_SIZE, flags);
59 if (ret)
60 pr_err_ratelimited("Failed to relinquish page: %016llx %d\n", phys, ret);
61
62 phys += PAGE_SIZE;
63 }
64
65 }
66
gunyah_post_page_relinquish_tlb_inv(void)67 static void gunyah_post_page_relinquish_tlb_inv(void)
68 {
69 /* Release page to Host, so unlock and sanitize */
70 int ret = 0;
71
72 ret = gunyah_hypercall_addrspc_modify_pages(our_addrspace_capid, 0, 0, 0);
73 if (ret)
74 pr_err_ratelimited("Failed to flush tlb: %d\n", ret);
75 }
76
77 static struct virtio_balloon_hyp_ops gunyah_virtio_balloon_hyp_ops = {
78 .page_relinquish = gunyah_page_relinquish,
79 .post_page_relinquish_tlb_inv = gunyah_post_page_relinquish_tlb_inv
80 };
81
82 #endif
83
gunyah_guest_init(void)84 static int __init gunyah_guest_init(void)
85 {
86 struct addrspace_info_area_rootvm_addrspace_cap *info;
87 size_t size;
88
89 info = gunyah_get_info(GUNYAH_INFO_OWNER_ROOTVM, ADDRSPACE_INFO_AREA_ROOTVM_ADDRSPACE_CAP,
90 &size);
91 if (IS_ERR(info))
92 return PTR_ERR(info);
93
94 if (size != sizeof(*info))
95 return -EINVAL;
96
97 our_addrspace_capid = info->addrspace_cap;
98
99 arm64_ioremap_prot_hook_register(&gunyah_mmio_guard_ioremap_hook);
100 #ifdef CONFIG_VIRTIO_BALLOON_HYP_OPS
101 virtio_balloon_hyp_ops = &gunyah_virtio_balloon_hyp_ops;
102 #endif
103 return 0;
104 }
105 core_initcall_sync(gunyah_guest_init);
106