1 // SPDX-License-Identifier: GPL-2.0
2 /*
3 * ACRN: Memory mapping management
4 *
5 * Copyright (C) 2020 Intel Corporation. All rights reserved.
6 *
7 * Authors:
8 * Fei Li <lei1.li@intel.com>
9 * Shuo Liu <shuo.a.liu@intel.com>
10 */
11
12 #include <linux/io.h>
13 #include <linux/mm.h>
14 #include <linux/slab.h>
15
16 #include "acrn_drv.h"
17
modify_region(struct acrn_vm * vm,struct vm_memory_region_op * region)18 static int modify_region(struct acrn_vm *vm, struct vm_memory_region_op *region)
19 {
20 struct vm_memory_region_batch *regions;
21 int ret;
22
23 regions = kzalloc(sizeof(*regions), GFP_KERNEL);
24 if (!regions)
25 return -ENOMEM;
26
27 regions->vmid = vm->vmid;
28 regions->regions_num = 1;
29 regions->regions_gpa = virt_to_phys(region);
30
31 ret = hcall_set_memory_regions(virt_to_phys(regions));
32 if (ret < 0)
33 dev_dbg(acrn_dev.this_device,
34 "Failed to set memory region for VM[%u]!\n", vm->vmid);
35
36 kfree(regions);
37 return ret;
38 }
39
40 /**
41 * acrn_mm_region_add() - Set up the EPT mapping of a memory region.
42 * @vm: User VM.
43 * @user_gpa: A GPA of User VM.
44 * @service_gpa: A GPA of Service VM.
45 * @size: Size of the region.
46 * @mem_type: Combination of ACRN_MEM_TYPE_*.
47 * @mem_access_right: Combination of ACRN_MEM_ACCESS_*.
48 *
49 * Return: 0 on success, <0 on error.
50 */
acrn_mm_region_add(struct acrn_vm * vm,u64 user_gpa,u64 service_gpa,u64 size,u32 mem_type,u32 mem_access_right)51 int acrn_mm_region_add(struct acrn_vm *vm, u64 user_gpa, u64 service_gpa,
52 u64 size, u32 mem_type, u32 mem_access_right)
53 {
54 struct vm_memory_region_op *region;
55 int ret = 0;
56
57 region = kzalloc(sizeof(*region), GFP_KERNEL);
58 if (!region)
59 return -ENOMEM;
60
61 region->type = ACRN_MEM_REGION_ADD;
62 region->user_vm_pa = user_gpa;
63 region->service_vm_pa = service_gpa;
64 region->size = size;
65 region->attr = ((mem_type & ACRN_MEM_TYPE_MASK) |
66 (mem_access_right & ACRN_MEM_ACCESS_RIGHT_MASK));
67 ret = modify_region(vm, region);
68
69 dev_dbg(acrn_dev.this_device,
70 "%s: user-GPA[%pK] service-GPA[%pK] size[0x%llx].\n",
71 __func__, (void *)user_gpa, (void *)service_gpa, size);
72 kfree(region);
73 return ret;
74 }
75
76 /**
77 * acrn_mm_region_del() - Del the EPT mapping of a memory region.
78 * @vm: User VM.
79 * @user_gpa: A GPA of the User VM.
80 * @size: Size of the region.
81 *
82 * Return: 0 on success, <0 for error.
83 */
acrn_mm_region_del(struct acrn_vm * vm,u64 user_gpa,u64 size)84 int acrn_mm_region_del(struct acrn_vm *vm, u64 user_gpa, u64 size)
85 {
86 struct vm_memory_region_op *region;
87 int ret = 0;
88
89 region = kzalloc(sizeof(*region), GFP_KERNEL);
90 if (!region)
91 return -ENOMEM;
92
93 region->type = ACRN_MEM_REGION_DEL;
94 region->user_vm_pa = user_gpa;
95 region->service_vm_pa = 0UL;
96 region->size = size;
97 region->attr = 0U;
98
99 ret = modify_region(vm, region);
100
101 dev_dbg(acrn_dev.this_device, "%s: user-GPA[%pK] size[0x%llx].\n",
102 __func__, (void *)user_gpa, size);
103 kfree(region);
104 return ret;
105 }
106
acrn_vm_memseg_map(struct acrn_vm * vm,struct acrn_vm_memmap * memmap)107 int acrn_vm_memseg_map(struct acrn_vm *vm, struct acrn_vm_memmap *memmap)
108 {
109 int ret;
110
111 if (memmap->type == ACRN_MEMMAP_RAM)
112 return acrn_vm_ram_map(vm, memmap);
113
114 if (memmap->type != ACRN_MEMMAP_MMIO) {
115 dev_dbg(acrn_dev.this_device,
116 "Invalid memmap type: %u\n", memmap->type);
117 return -EINVAL;
118 }
119
120 ret = acrn_mm_region_add(vm, memmap->user_vm_pa,
121 memmap->service_vm_pa, memmap->len,
122 ACRN_MEM_TYPE_UC, memmap->attr);
123 if (ret < 0)
124 dev_dbg(acrn_dev.this_device,
125 "Add memory region failed, VM[%u]!\n", vm->vmid);
126
127 return ret;
128 }
129
acrn_vm_memseg_unmap(struct acrn_vm * vm,struct acrn_vm_memmap * memmap)130 int acrn_vm_memseg_unmap(struct acrn_vm *vm, struct acrn_vm_memmap *memmap)
131 {
132 int ret;
133
134 if (memmap->type != ACRN_MEMMAP_MMIO) {
135 dev_dbg(acrn_dev.this_device,
136 "Invalid memmap type: %u\n", memmap->type);
137 return -EINVAL;
138 }
139
140 ret = acrn_mm_region_del(vm, memmap->user_vm_pa, memmap->len);
141 if (ret < 0)
142 dev_dbg(acrn_dev.this_device,
143 "Del memory region failed, VM[%u]!\n", vm->vmid);
144
145 return ret;
146 }
147
148 /**
149 * acrn_vm_ram_map() - Create a RAM EPT mapping of User VM.
150 * @vm: The User VM pointer
151 * @memmap: Info of the EPT mapping
152 *
153 * Return: 0 on success, <0 for error.
154 */
acrn_vm_ram_map(struct acrn_vm * vm,struct acrn_vm_memmap * memmap)155 int acrn_vm_ram_map(struct acrn_vm *vm, struct acrn_vm_memmap *memmap)
156 {
157 struct vm_memory_region_batch *regions_info;
158 int nr_pages, i = 0, order, nr_regions = 0;
159 struct vm_memory_mapping *region_mapping;
160 struct vm_memory_region_op *vm_region;
161 struct page **pages = NULL, *page;
162 void *remap_vaddr;
163 int ret, pinned;
164 u64 user_vm_pa;
165 unsigned long pfn;
166 struct vm_area_struct *vma;
167
168 if (!vm || !memmap)
169 return -EINVAL;
170
171 mmap_read_lock(current->mm);
172 vma = vma_lookup(current->mm, memmap->vma_base);
173 if (vma && ((vma->vm_flags & VM_PFNMAP) != 0)) {
174 if ((memmap->vma_base + memmap->len) > vma->vm_end) {
175 mmap_read_unlock(current->mm);
176 return -EINVAL;
177 }
178
179 ret = follow_pfn(vma, memmap->vma_base, &pfn);
180 mmap_read_unlock(current->mm);
181 if (ret < 0) {
182 dev_dbg(acrn_dev.this_device,
183 "Failed to lookup PFN at VMA:%pK.\n", (void *)memmap->vma_base);
184 return ret;
185 }
186
187 return acrn_mm_region_add(vm, memmap->user_vm_pa,
188 PFN_PHYS(pfn), memmap->len,
189 ACRN_MEM_TYPE_WB, memmap->attr);
190 }
191 mmap_read_unlock(current->mm);
192
193 /* Get the page number of the map region */
194 nr_pages = memmap->len >> PAGE_SHIFT;
195 pages = vzalloc(nr_pages * sizeof(struct page *));
196 if (!pages)
197 return -ENOMEM;
198
199 /* Lock the pages of user memory map region */
200 pinned = pin_user_pages_fast(memmap->vma_base,
201 nr_pages, FOLL_WRITE | FOLL_LONGTERM,
202 pages);
203 if (pinned < 0) {
204 ret = pinned;
205 goto free_pages;
206 } else if (pinned != nr_pages) {
207 ret = -EFAULT;
208 goto put_pages;
209 }
210
211 /* Create a kernel map for the map region */
212 remap_vaddr = vmap(pages, nr_pages, VM_MAP, PAGE_KERNEL);
213 if (!remap_vaddr) {
214 ret = -ENOMEM;
215 goto put_pages;
216 }
217
218 /* Record Service VM va <-> User VM pa mapping */
219 mutex_lock(&vm->regions_mapping_lock);
220 region_mapping = &vm->regions_mapping[vm->regions_mapping_count];
221 if (vm->regions_mapping_count < ACRN_MEM_MAPPING_MAX) {
222 region_mapping->pages = pages;
223 region_mapping->npages = nr_pages;
224 region_mapping->size = memmap->len;
225 region_mapping->service_vm_va = remap_vaddr;
226 region_mapping->user_vm_pa = memmap->user_vm_pa;
227 vm->regions_mapping_count++;
228 } else {
229 dev_warn(acrn_dev.this_device,
230 "Run out of memory mapping slots!\n");
231 ret = -ENOMEM;
232 mutex_unlock(&vm->regions_mapping_lock);
233 goto unmap_no_count;
234 }
235 mutex_unlock(&vm->regions_mapping_lock);
236
237 /* Calculate count of vm_memory_region_op */
238 while (i < nr_pages) {
239 page = pages[i];
240 VM_BUG_ON_PAGE(PageTail(page), page);
241 order = compound_order(page);
242 nr_regions++;
243 i += 1 << order;
244 }
245
246 /* Prepare the vm_memory_region_batch */
247 regions_info = kzalloc(sizeof(*regions_info) +
248 sizeof(*vm_region) * nr_regions,
249 GFP_KERNEL);
250 if (!regions_info) {
251 ret = -ENOMEM;
252 goto unmap_kernel_map;
253 }
254
255 /* Fill each vm_memory_region_op */
256 vm_region = (struct vm_memory_region_op *)(regions_info + 1);
257 regions_info->vmid = vm->vmid;
258 regions_info->regions_num = nr_regions;
259 regions_info->regions_gpa = virt_to_phys(vm_region);
260 user_vm_pa = memmap->user_vm_pa;
261 i = 0;
262 while (i < nr_pages) {
263 u32 region_size;
264
265 page = pages[i];
266 VM_BUG_ON_PAGE(PageTail(page), page);
267 order = compound_order(page);
268 region_size = PAGE_SIZE << order;
269 vm_region->type = ACRN_MEM_REGION_ADD;
270 vm_region->user_vm_pa = user_vm_pa;
271 vm_region->service_vm_pa = page_to_phys(page);
272 vm_region->size = region_size;
273 vm_region->attr = (ACRN_MEM_TYPE_WB & ACRN_MEM_TYPE_MASK) |
274 (memmap->attr & ACRN_MEM_ACCESS_RIGHT_MASK);
275
276 vm_region++;
277 user_vm_pa += region_size;
278 i += 1 << order;
279 }
280
281 /* Inform the ACRN Hypervisor to set up EPT mappings */
282 ret = hcall_set_memory_regions(virt_to_phys(regions_info));
283 if (ret < 0) {
284 dev_dbg(acrn_dev.this_device,
285 "Failed to set regions, VM[%u]!\n", vm->vmid);
286 goto unset_region;
287 }
288 kfree(regions_info);
289
290 dev_dbg(acrn_dev.this_device,
291 "%s: VM[%u] service-GVA[%pK] user-GPA[%pK] size[0x%llx]\n",
292 __func__, vm->vmid,
293 remap_vaddr, (void *)memmap->user_vm_pa, memmap->len);
294 return ret;
295
296 unset_region:
297 kfree(regions_info);
298 unmap_kernel_map:
299 mutex_lock(&vm->regions_mapping_lock);
300 vm->regions_mapping_count--;
301 mutex_unlock(&vm->regions_mapping_lock);
302 unmap_no_count:
303 vunmap(remap_vaddr);
304 put_pages:
305 for (i = 0; i < pinned; i++)
306 unpin_user_page(pages[i]);
307 free_pages:
308 vfree(pages);
309 return ret;
310 }
311
312 /**
313 * acrn_vm_all_ram_unmap() - Destroy a RAM EPT mapping of User VM.
314 * @vm: The User VM
315 */
acrn_vm_all_ram_unmap(struct acrn_vm * vm)316 void acrn_vm_all_ram_unmap(struct acrn_vm *vm)
317 {
318 struct vm_memory_mapping *region_mapping;
319 int i, j;
320
321 mutex_lock(&vm->regions_mapping_lock);
322 for (i = 0; i < vm->regions_mapping_count; i++) {
323 region_mapping = &vm->regions_mapping[i];
324 vunmap(region_mapping->service_vm_va);
325 for (j = 0; j < region_mapping->npages; j++)
326 unpin_user_page(region_mapping->pages[j]);
327 vfree(region_mapping->pages);
328 }
329 mutex_unlock(&vm->regions_mapping_lock);
330 }
331