nvmm.c revision 1.13 1 1.13 maxv /* $NetBSD: nvmm.c,v 1.13 2019/04/07 14:05:15 maxv Exp $ */
2 1.1 maxv
3 1.1 maxv /*
4 1.1 maxv * Copyright (c) 2018 The NetBSD Foundation, Inc.
5 1.1 maxv * All rights reserved.
6 1.1 maxv *
7 1.1 maxv * This code is derived from software contributed to The NetBSD Foundation
8 1.1 maxv * by Maxime Villard.
9 1.1 maxv *
10 1.1 maxv * Redistribution and use in source and binary forms, with or without
11 1.1 maxv * modification, are permitted provided that the following conditions
12 1.1 maxv * are met:
13 1.1 maxv * 1. Redistributions of source code must retain the above copyright
14 1.1 maxv * notice, this list of conditions and the following disclaimer.
15 1.1 maxv * 2. Redistributions in binary form must reproduce the above copyright
16 1.1 maxv * notice, this list of conditions and the following disclaimer in the
17 1.1 maxv * documentation and/or other materials provided with the distribution.
18 1.1 maxv *
19 1.1 maxv * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 1.1 maxv * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 1.1 maxv * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 1.1 maxv * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 1.1 maxv * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 1.1 maxv * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 1.1 maxv * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 1.1 maxv * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 1.1 maxv * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 1.1 maxv * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 1.1 maxv * POSSIBILITY OF SUCH DAMAGE.
30 1.1 maxv */
31 1.1 maxv
32 1.1 maxv #include <sys/cdefs.h>
33 1.13 maxv __KERNEL_RCSID(0, "$NetBSD: nvmm.c,v 1.13 2019/04/07 14:05:15 maxv Exp $");
34 1.1 maxv
35 1.1 maxv #include <sys/param.h>
36 1.1 maxv #include <sys/systm.h>
37 1.1 maxv #include <sys/kernel.h>
38 1.1 maxv
39 1.1 maxv #include <sys/cpu.h>
40 1.1 maxv #include <sys/conf.h>
41 1.1 maxv #include <sys/kmem.h>
42 1.1 maxv #include <sys/module.h>
43 1.1 maxv #include <sys/proc.h>
44 1.11 maxv #include <sys/mman.h>
45 1.1 maxv
46 1.1 maxv #include <uvm/uvm.h>
47 1.1 maxv #include <uvm/uvm_page.h>
48 1.1 maxv
49 1.1 maxv #include "ioconf.h"
50 1.1 maxv
51 1.1 maxv #include <dev/nvmm/nvmm.h>
52 1.1 maxv #include <dev/nvmm/nvmm_internal.h>
53 1.1 maxv #include <dev/nvmm/nvmm_ioctl.h>
54 1.1 maxv
55 1.1 maxv static struct nvmm_machine machines[NVMM_MAX_MACHINES];
56 1.13 maxv static volatile unsigned int nmachines __cacheline_aligned;
57 1.1 maxv
58 1.1 maxv static const struct nvmm_impl *nvmm_impl_list[] = {
59 1.7 maxv &nvmm_x86_svm, /* x86 AMD SVM */
60 1.7 maxv &nvmm_x86_vmx /* x86 Intel VMX */
61 1.1 maxv };
62 1.1 maxv
63 1.1 maxv static const struct nvmm_impl *nvmm_impl = NULL;
64 1.1 maxv
65 1.1 maxv /* -------------------------------------------------------------------------- */
66 1.1 maxv
67 1.1 maxv static int
68 1.1 maxv nvmm_machine_alloc(struct nvmm_machine **ret)
69 1.1 maxv {
70 1.1 maxv struct nvmm_machine *mach;
71 1.1 maxv size_t i;
72 1.1 maxv
73 1.1 maxv for (i = 0; i < NVMM_MAX_MACHINES; i++) {
74 1.1 maxv mach = &machines[i];
75 1.1 maxv
76 1.1 maxv rw_enter(&mach->lock, RW_WRITER);
77 1.1 maxv if (mach->present) {
78 1.1 maxv rw_exit(&mach->lock);
79 1.1 maxv continue;
80 1.1 maxv }
81 1.1 maxv
82 1.1 maxv mach->present = true;
83 1.1 maxv *ret = mach;
84 1.13 maxv atomic_inc_uint(&nmachines);
85 1.1 maxv return 0;
86 1.1 maxv }
87 1.1 maxv
88 1.1 maxv return ENOBUFS;
89 1.1 maxv }
90 1.1 maxv
91 1.1 maxv static void
92 1.1 maxv nvmm_machine_free(struct nvmm_machine *mach)
93 1.1 maxv {
94 1.1 maxv KASSERT(rw_write_held(&mach->lock));
95 1.1 maxv KASSERT(mach->present);
96 1.1 maxv mach->present = false;
97 1.13 maxv atomic_dec_uint(&nmachines);
98 1.1 maxv }
99 1.1 maxv
100 1.1 maxv static int
101 1.1 maxv nvmm_machine_get(nvmm_machid_t machid, struct nvmm_machine **ret, bool writer)
102 1.1 maxv {
103 1.1 maxv struct nvmm_machine *mach;
104 1.1 maxv krw_t op = writer ? RW_WRITER : RW_READER;
105 1.1 maxv
106 1.1 maxv if (machid >= NVMM_MAX_MACHINES) {
107 1.1 maxv return EINVAL;
108 1.1 maxv }
109 1.1 maxv mach = &machines[machid];
110 1.1 maxv
111 1.1 maxv rw_enter(&mach->lock, op);
112 1.1 maxv if (!mach->present) {
113 1.1 maxv rw_exit(&mach->lock);
114 1.1 maxv return ENOENT;
115 1.1 maxv }
116 1.1 maxv if (mach->procid != curproc->p_pid) {
117 1.1 maxv rw_exit(&mach->lock);
118 1.1 maxv return EPERM;
119 1.1 maxv }
120 1.1 maxv *ret = mach;
121 1.1 maxv
122 1.1 maxv return 0;
123 1.1 maxv }
124 1.1 maxv
125 1.1 maxv static void
126 1.1 maxv nvmm_machine_put(struct nvmm_machine *mach)
127 1.1 maxv {
128 1.1 maxv rw_exit(&mach->lock);
129 1.1 maxv }
130 1.1 maxv
131 1.1 maxv /* -------------------------------------------------------------------------- */
132 1.1 maxv
133 1.1 maxv static int
134 1.1 maxv nvmm_vcpu_alloc(struct nvmm_machine *mach, struct nvmm_cpu **ret)
135 1.1 maxv {
136 1.1 maxv struct nvmm_cpu *vcpu;
137 1.1 maxv size_t i;
138 1.1 maxv
139 1.1 maxv for (i = 0; i < NVMM_MAX_VCPUS; i++) {
140 1.1 maxv vcpu = &mach->cpus[i];
141 1.1 maxv
142 1.1 maxv mutex_enter(&vcpu->lock);
143 1.1 maxv if (vcpu->present) {
144 1.1 maxv mutex_exit(&vcpu->lock);
145 1.1 maxv continue;
146 1.1 maxv }
147 1.1 maxv
148 1.1 maxv vcpu->present = true;
149 1.1 maxv vcpu->cpuid = i;
150 1.6 maxv vcpu->state = kmem_zalloc(nvmm_impl->state_size, KM_SLEEP);
151 1.1 maxv *ret = vcpu;
152 1.1 maxv return 0;
153 1.1 maxv }
154 1.1 maxv
155 1.1 maxv return ENOBUFS;
156 1.1 maxv }
157 1.1 maxv
158 1.1 maxv static void
159 1.1 maxv nvmm_vcpu_free(struct nvmm_machine *mach, struct nvmm_cpu *vcpu)
160 1.1 maxv {
161 1.1 maxv KASSERT(mutex_owned(&vcpu->lock));
162 1.1 maxv vcpu->present = false;
163 1.6 maxv kmem_free(vcpu->state, nvmm_impl->state_size);
164 1.1 maxv vcpu->hcpu_last = -1;
165 1.1 maxv }
166 1.1 maxv
167 1.1 maxv int
168 1.1 maxv nvmm_vcpu_get(struct nvmm_machine *mach, nvmm_cpuid_t cpuid,
169 1.1 maxv struct nvmm_cpu **ret)
170 1.1 maxv {
171 1.1 maxv struct nvmm_cpu *vcpu;
172 1.1 maxv
173 1.1 maxv if (cpuid >= NVMM_MAX_VCPUS) {
174 1.1 maxv return EINVAL;
175 1.1 maxv }
176 1.1 maxv vcpu = &mach->cpus[cpuid];
177 1.1 maxv
178 1.1 maxv mutex_enter(&vcpu->lock);
179 1.1 maxv if (!vcpu->present) {
180 1.1 maxv mutex_exit(&vcpu->lock);
181 1.1 maxv return ENOENT;
182 1.1 maxv }
183 1.1 maxv *ret = vcpu;
184 1.1 maxv
185 1.1 maxv return 0;
186 1.1 maxv }
187 1.1 maxv
188 1.1 maxv void
189 1.1 maxv nvmm_vcpu_put(struct nvmm_cpu *vcpu)
190 1.1 maxv {
191 1.1 maxv mutex_exit(&vcpu->lock);
192 1.1 maxv }
193 1.1 maxv
194 1.1 maxv /* -------------------------------------------------------------------------- */
195 1.1 maxv
196 1.1 maxv static void
197 1.1 maxv nvmm_kill_machines(pid_t pid)
198 1.1 maxv {
199 1.1 maxv struct nvmm_machine *mach;
200 1.1 maxv struct nvmm_cpu *vcpu;
201 1.1 maxv size_t i, j;
202 1.1 maxv int error;
203 1.1 maxv
204 1.1 maxv for (i = 0; i < NVMM_MAX_MACHINES; i++) {
205 1.1 maxv mach = &machines[i];
206 1.1 maxv
207 1.1 maxv rw_enter(&mach->lock, RW_WRITER);
208 1.1 maxv if (!mach->present || mach->procid != pid) {
209 1.1 maxv rw_exit(&mach->lock);
210 1.1 maxv continue;
211 1.1 maxv }
212 1.1 maxv
213 1.1 maxv /* Kill it. */
214 1.1 maxv for (j = 0; j < NVMM_MAX_VCPUS; j++) {
215 1.1 maxv error = nvmm_vcpu_get(mach, j, &vcpu);
216 1.1 maxv if (error)
217 1.1 maxv continue;
218 1.1 maxv (*nvmm_impl->vcpu_destroy)(mach, vcpu);
219 1.1 maxv nvmm_vcpu_free(mach, vcpu);
220 1.1 maxv nvmm_vcpu_put(vcpu);
221 1.1 maxv }
222 1.1 maxv uvmspace_free(mach->vm);
223 1.4 maxv
224 1.4 maxv /* Drop the kernel UOBJ refs. */
225 1.9 maxv for (j = 0; j < NVMM_MAX_HMAPPINGS; j++) {
226 1.9 maxv if (!mach->hmap[j].present)
227 1.4 maxv continue;
228 1.9 maxv uao_detach(mach->hmap[j].uobj);
229 1.4 maxv }
230 1.4 maxv
231 1.1 maxv nvmm_machine_free(mach);
232 1.1 maxv
233 1.1 maxv rw_exit(&mach->lock);
234 1.1 maxv }
235 1.1 maxv }
236 1.1 maxv
237 1.1 maxv /* -------------------------------------------------------------------------- */
238 1.1 maxv
239 1.1 maxv static int
240 1.1 maxv nvmm_capability(struct nvmm_ioc_capability *args)
241 1.1 maxv {
242 1.1 maxv args->cap.version = NVMM_CAPABILITY_VERSION;
243 1.1 maxv args->cap.state_size = nvmm_impl->state_size;
244 1.1 maxv args->cap.max_machines = NVMM_MAX_MACHINES;
245 1.1 maxv args->cap.max_vcpus = NVMM_MAX_VCPUS;
246 1.1 maxv args->cap.max_ram = NVMM_MAX_RAM;
247 1.1 maxv
248 1.1 maxv (*nvmm_impl->capability)(&args->cap);
249 1.1 maxv
250 1.1 maxv return 0;
251 1.1 maxv }
252 1.1 maxv
253 1.1 maxv static int
254 1.1 maxv nvmm_machine_create(struct nvmm_ioc_machine_create *args)
255 1.1 maxv {
256 1.1 maxv struct nvmm_machine *mach;
257 1.1 maxv int error;
258 1.1 maxv
259 1.1 maxv error = nvmm_machine_alloc(&mach);
260 1.1 maxv if (error)
261 1.1 maxv return error;
262 1.1 maxv
263 1.1 maxv /* Curproc owns the machine. */
264 1.1 maxv mach->procid = curproc->p_pid;
265 1.1 maxv
266 1.9 maxv /* Zero out the host mappings. */
267 1.9 maxv memset(&mach->hmap, 0, sizeof(mach->hmap));
268 1.4 maxv
269 1.1 maxv /* Create the machine vmspace. */
270 1.1 maxv mach->gpa_begin = 0;
271 1.1 maxv mach->gpa_end = NVMM_MAX_RAM;
272 1.1 maxv mach->vm = uvmspace_alloc(0, mach->gpa_end - mach->gpa_begin, false);
273 1.1 maxv
274 1.1 maxv (*nvmm_impl->machine_create)(mach);
275 1.1 maxv
276 1.1 maxv args->machid = mach->machid;
277 1.1 maxv nvmm_machine_put(mach);
278 1.1 maxv
279 1.1 maxv return 0;
280 1.1 maxv }
281 1.1 maxv
282 1.1 maxv static int
283 1.1 maxv nvmm_machine_destroy(struct nvmm_ioc_machine_destroy *args)
284 1.1 maxv {
285 1.1 maxv struct nvmm_machine *mach;
286 1.1 maxv struct nvmm_cpu *vcpu;
287 1.1 maxv int error;
288 1.1 maxv size_t i;
289 1.1 maxv
290 1.1 maxv error = nvmm_machine_get(args->machid, &mach, true);
291 1.1 maxv if (error)
292 1.1 maxv return error;
293 1.1 maxv
294 1.1 maxv for (i = 0; i < NVMM_MAX_VCPUS; i++) {
295 1.1 maxv error = nvmm_vcpu_get(mach, i, &vcpu);
296 1.1 maxv if (error)
297 1.1 maxv continue;
298 1.1 maxv
299 1.1 maxv (*nvmm_impl->vcpu_destroy)(mach, vcpu);
300 1.1 maxv nvmm_vcpu_free(mach, vcpu);
301 1.1 maxv nvmm_vcpu_put(vcpu);
302 1.1 maxv }
303 1.1 maxv
304 1.1 maxv (*nvmm_impl->machine_destroy)(mach);
305 1.1 maxv
306 1.1 maxv /* Free the machine vmspace. */
307 1.1 maxv uvmspace_free(mach->vm);
308 1.4 maxv
309 1.4 maxv /* Drop the kernel UOBJ refs. */
310 1.9 maxv for (i = 0; i < NVMM_MAX_HMAPPINGS; i++) {
311 1.9 maxv if (!mach->hmap[i].present)
312 1.4 maxv continue;
313 1.9 maxv uao_detach(mach->hmap[i].uobj);
314 1.4 maxv }
315 1.1 maxv
316 1.1 maxv nvmm_machine_free(mach);
317 1.1 maxv nvmm_machine_put(mach);
318 1.1 maxv
319 1.1 maxv return 0;
320 1.1 maxv }
321 1.1 maxv
322 1.1 maxv static int
323 1.1 maxv nvmm_machine_configure(struct nvmm_ioc_machine_configure *args)
324 1.1 maxv {
325 1.1 maxv struct nvmm_machine *mach;
326 1.1 maxv size_t allocsz;
327 1.1 maxv void *data;
328 1.1 maxv int error;
329 1.1 maxv
330 1.1 maxv if (__predict_false(args->op >= nvmm_impl->conf_max)) {
331 1.1 maxv return EINVAL;
332 1.1 maxv }
333 1.1 maxv
334 1.1 maxv allocsz = nvmm_impl->conf_sizes[args->op];
335 1.1 maxv data = kmem_alloc(allocsz, KM_SLEEP);
336 1.1 maxv
337 1.1 maxv error = nvmm_machine_get(args->machid, &mach, true);
338 1.1 maxv if (error) {
339 1.1 maxv kmem_free(data, allocsz);
340 1.1 maxv return error;
341 1.1 maxv }
342 1.1 maxv
343 1.1 maxv error = copyin(args->conf, data, allocsz);
344 1.1 maxv if (error) {
345 1.1 maxv goto out;
346 1.1 maxv }
347 1.1 maxv
348 1.1 maxv error = (*nvmm_impl->machine_configure)(mach, args->op, data);
349 1.1 maxv
350 1.1 maxv out:
351 1.1 maxv nvmm_machine_put(mach);
352 1.1 maxv kmem_free(data, allocsz);
353 1.1 maxv return error;
354 1.1 maxv }
355 1.1 maxv
356 1.1 maxv static int
357 1.1 maxv nvmm_vcpu_create(struct nvmm_ioc_vcpu_create *args)
358 1.1 maxv {
359 1.1 maxv struct nvmm_machine *mach;
360 1.1 maxv struct nvmm_cpu *vcpu;
361 1.1 maxv int error;
362 1.1 maxv
363 1.1 maxv error = nvmm_machine_get(args->machid, &mach, false);
364 1.1 maxv if (error)
365 1.1 maxv return error;
366 1.1 maxv
367 1.1 maxv error = nvmm_vcpu_alloc(mach, &vcpu);
368 1.1 maxv if (error)
369 1.1 maxv goto out;
370 1.1 maxv
371 1.1 maxv error = (*nvmm_impl->vcpu_create)(mach, vcpu);
372 1.1 maxv if (error) {
373 1.1 maxv nvmm_vcpu_free(mach, vcpu);
374 1.1 maxv nvmm_vcpu_put(vcpu);
375 1.1 maxv goto out;
376 1.1 maxv }
377 1.1 maxv
378 1.1 maxv nvmm_vcpu_put(vcpu);
379 1.1 maxv
380 1.1 maxv out:
381 1.1 maxv nvmm_machine_put(mach);
382 1.1 maxv return error;
383 1.1 maxv }
384 1.1 maxv
385 1.1 maxv static int
386 1.1 maxv nvmm_vcpu_destroy(struct nvmm_ioc_vcpu_destroy *args)
387 1.1 maxv {
388 1.1 maxv struct nvmm_machine *mach;
389 1.1 maxv struct nvmm_cpu *vcpu;
390 1.1 maxv int error;
391 1.1 maxv
392 1.1 maxv error = nvmm_machine_get(args->machid, &mach, false);
393 1.1 maxv if (error)
394 1.1 maxv return error;
395 1.1 maxv
396 1.1 maxv error = nvmm_vcpu_get(mach, args->cpuid, &vcpu);
397 1.1 maxv if (error)
398 1.1 maxv goto out;
399 1.1 maxv
400 1.1 maxv (*nvmm_impl->vcpu_destroy)(mach, vcpu);
401 1.1 maxv nvmm_vcpu_free(mach, vcpu);
402 1.1 maxv nvmm_vcpu_put(vcpu);
403 1.1 maxv
404 1.1 maxv out:
405 1.1 maxv nvmm_machine_put(mach);
406 1.1 maxv return error;
407 1.1 maxv }
408 1.1 maxv
409 1.1 maxv static int
410 1.1 maxv nvmm_vcpu_setstate(struct nvmm_ioc_vcpu_setstate *args)
411 1.1 maxv {
412 1.1 maxv struct nvmm_machine *mach;
413 1.1 maxv struct nvmm_cpu *vcpu;
414 1.1 maxv int error;
415 1.1 maxv
416 1.1 maxv error = nvmm_machine_get(args->machid, &mach, false);
417 1.6 maxv if (error)
418 1.1 maxv return error;
419 1.1 maxv
420 1.1 maxv error = nvmm_vcpu_get(mach, args->cpuid, &vcpu);
421 1.1 maxv if (error)
422 1.1 maxv goto out;
423 1.1 maxv
424 1.6 maxv error = copyin(args->state, vcpu->state, nvmm_impl->state_size);
425 1.1 maxv if (error) {
426 1.1 maxv nvmm_vcpu_put(vcpu);
427 1.1 maxv goto out;
428 1.1 maxv }
429 1.1 maxv
430 1.6 maxv (*nvmm_impl->vcpu_setstate)(vcpu, vcpu->state, args->flags);
431 1.1 maxv nvmm_vcpu_put(vcpu);
432 1.1 maxv
433 1.1 maxv out:
434 1.1 maxv nvmm_machine_put(mach);
435 1.1 maxv return error;
436 1.1 maxv }
437 1.1 maxv
438 1.1 maxv static int
439 1.1 maxv nvmm_vcpu_getstate(struct nvmm_ioc_vcpu_getstate *args)
440 1.1 maxv {
441 1.1 maxv struct nvmm_machine *mach;
442 1.1 maxv struct nvmm_cpu *vcpu;
443 1.1 maxv int error;
444 1.1 maxv
445 1.1 maxv error = nvmm_machine_get(args->machid, &mach, false);
446 1.6 maxv if (error)
447 1.1 maxv return error;
448 1.1 maxv
449 1.1 maxv error = nvmm_vcpu_get(mach, args->cpuid, &vcpu);
450 1.1 maxv if (error)
451 1.1 maxv goto out;
452 1.1 maxv
453 1.6 maxv (*nvmm_impl->vcpu_getstate)(vcpu, vcpu->state, args->flags);
454 1.1 maxv nvmm_vcpu_put(vcpu);
455 1.6 maxv error = copyout(vcpu->state, args->state, nvmm_impl->state_size);
456 1.1 maxv
457 1.1 maxv out:
458 1.1 maxv nvmm_machine_put(mach);
459 1.1 maxv return error;
460 1.1 maxv }
461 1.1 maxv
462 1.1 maxv static int
463 1.1 maxv nvmm_vcpu_inject(struct nvmm_ioc_vcpu_inject *args)
464 1.1 maxv {
465 1.1 maxv struct nvmm_machine *mach;
466 1.1 maxv struct nvmm_cpu *vcpu;
467 1.1 maxv int error;
468 1.1 maxv
469 1.1 maxv error = nvmm_machine_get(args->machid, &mach, false);
470 1.1 maxv if (error)
471 1.1 maxv return error;
472 1.1 maxv
473 1.1 maxv error = nvmm_vcpu_get(mach, args->cpuid, &vcpu);
474 1.1 maxv if (error)
475 1.1 maxv goto out;
476 1.1 maxv
477 1.1 maxv error = (*nvmm_impl->vcpu_inject)(mach, vcpu, &args->event);
478 1.1 maxv nvmm_vcpu_put(vcpu);
479 1.1 maxv
480 1.1 maxv out:
481 1.1 maxv nvmm_machine_put(mach);
482 1.1 maxv return error;
483 1.1 maxv }
484 1.1 maxv
485 1.8 maxv static void
486 1.8 maxv nvmm_do_vcpu_run(struct nvmm_machine *mach, struct nvmm_cpu *vcpu,
487 1.8 maxv struct nvmm_exit *exit)
488 1.8 maxv {
489 1.8 maxv struct vmspace *vm = mach->vm;
490 1.8 maxv
491 1.8 maxv while (1) {
492 1.8 maxv (*nvmm_impl->vcpu_run)(mach, vcpu, exit);
493 1.8 maxv
494 1.8 maxv if (__predict_true(exit->reason != NVMM_EXIT_MEMORY)) {
495 1.8 maxv break;
496 1.8 maxv }
497 1.10 maxv if (exit->u.mem.gpa >= mach->gpa_end) {
498 1.10 maxv break;
499 1.10 maxv }
500 1.11 maxv if (uvm_fault(&vm->vm_map, exit->u.mem.gpa, exit->u.mem.prot)) {
501 1.8 maxv break;
502 1.8 maxv }
503 1.8 maxv }
504 1.8 maxv }
505 1.8 maxv
506 1.1 maxv static int
507 1.1 maxv nvmm_vcpu_run(struct nvmm_ioc_vcpu_run *args)
508 1.1 maxv {
509 1.1 maxv struct nvmm_machine *mach;
510 1.1 maxv struct nvmm_cpu *vcpu;
511 1.1 maxv int error;
512 1.1 maxv
513 1.1 maxv error = nvmm_machine_get(args->machid, &mach, false);
514 1.1 maxv if (error)
515 1.1 maxv return error;
516 1.1 maxv
517 1.1 maxv error = nvmm_vcpu_get(mach, args->cpuid, &vcpu);
518 1.1 maxv if (error)
519 1.1 maxv goto out;
520 1.1 maxv
521 1.8 maxv nvmm_do_vcpu_run(mach, vcpu, &args->exit);
522 1.1 maxv nvmm_vcpu_put(vcpu);
523 1.1 maxv
524 1.1 maxv out:
525 1.1 maxv nvmm_machine_put(mach);
526 1.1 maxv return error;
527 1.1 maxv }
528 1.1 maxv
529 1.1 maxv /* -------------------------------------------------------------------------- */
530 1.1 maxv
531 1.4 maxv static struct uvm_object *
532 1.9 maxv nvmm_hmapping_getuobj(struct nvmm_machine *mach, uintptr_t hva, size_t size,
533 1.4 maxv size_t *off)
534 1.4 maxv {
535 1.9 maxv struct nvmm_hmapping *hmapping;
536 1.4 maxv size_t i;
537 1.4 maxv
538 1.9 maxv for (i = 0; i < NVMM_MAX_HMAPPINGS; i++) {
539 1.9 maxv hmapping = &mach->hmap[i];
540 1.9 maxv if (!hmapping->present) {
541 1.4 maxv continue;
542 1.4 maxv }
543 1.9 maxv if (hva >= hmapping->hva &&
544 1.9 maxv hva + size <= hmapping->hva + hmapping->size) {
545 1.9 maxv *off = hva - hmapping->hva;
546 1.9 maxv return hmapping->uobj;
547 1.4 maxv }
548 1.4 maxv }
549 1.4 maxv
550 1.4 maxv return NULL;
551 1.4 maxv }
552 1.4 maxv
553 1.4 maxv static int
554 1.9 maxv nvmm_hmapping_validate(struct nvmm_machine *mach, uintptr_t hva, size_t size)
555 1.4 maxv {
556 1.9 maxv struct nvmm_hmapping *hmapping;
557 1.4 maxv size_t i;
558 1.4 maxv
559 1.4 maxv if ((hva % PAGE_SIZE) != 0 || (size % PAGE_SIZE) != 0) {
560 1.4 maxv return EINVAL;
561 1.4 maxv }
562 1.4 maxv if (hva == 0) {
563 1.4 maxv return EINVAL;
564 1.4 maxv }
565 1.4 maxv
566 1.9 maxv for (i = 0; i < NVMM_MAX_HMAPPINGS; i++) {
567 1.9 maxv hmapping = &mach->hmap[i];
568 1.9 maxv if (!hmapping->present) {
569 1.4 maxv continue;
570 1.4 maxv }
571 1.4 maxv
572 1.9 maxv if (hva >= hmapping->hva &&
573 1.9 maxv hva + size <= hmapping->hva + hmapping->size) {
574 1.4 maxv break;
575 1.4 maxv }
576 1.4 maxv
577 1.9 maxv if (hva >= hmapping->hva &&
578 1.9 maxv hva < hmapping->hva + hmapping->size) {
579 1.4 maxv return EEXIST;
580 1.4 maxv }
581 1.9 maxv if (hva + size > hmapping->hva &&
582 1.9 maxv hva + size <= hmapping->hva + hmapping->size) {
583 1.4 maxv return EEXIST;
584 1.4 maxv }
585 1.9 maxv if (hva <= hmapping->hva &&
586 1.9 maxv hva + size >= hmapping->hva + hmapping->size) {
587 1.4 maxv return EEXIST;
588 1.4 maxv }
589 1.4 maxv }
590 1.4 maxv
591 1.4 maxv return 0;
592 1.4 maxv }
593 1.4 maxv
594 1.9 maxv static struct nvmm_hmapping *
595 1.9 maxv nvmm_hmapping_alloc(struct nvmm_machine *mach)
596 1.4 maxv {
597 1.9 maxv struct nvmm_hmapping *hmapping;
598 1.4 maxv size_t i;
599 1.4 maxv
600 1.9 maxv for (i = 0; i < NVMM_MAX_HMAPPINGS; i++) {
601 1.9 maxv hmapping = &mach->hmap[i];
602 1.9 maxv if (!hmapping->present) {
603 1.9 maxv hmapping->present = true;
604 1.9 maxv return hmapping;
605 1.4 maxv }
606 1.4 maxv }
607 1.4 maxv
608 1.4 maxv return NULL;
609 1.4 maxv }
610 1.4 maxv
611 1.9 maxv static int
612 1.9 maxv nvmm_hmapping_free(struct nvmm_machine *mach, uintptr_t hva, size_t size)
613 1.4 maxv {
614 1.4 maxv struct vmspace *vmspace = curproc->p_vmspace;
615 1.9 maxv struct nvmm_hmapping *hmapping;
616 1.9 maxv size_t i;
617 1.4 maxv
618 1.9 maxv for (i = 0; i < NVMM_MAX_HMAPPINGS; i++) {
619 1.9 maxv hmapping = &mach->hmap[i];
620 1.9 maxv if (!hmapping->present || hmapping->hva != hva ||
621 1.9 maxv hmapping->size != size) {
622 1.9 maxv continue;
623 1.9 maxv }
624 1.9 maxv
625 1.9 maxv uvm_unmap(&vmspace->vm_map, hmapping->hva,
626 1.9 maxv hmapping->hva + hmapping->size);
627 1.9 maxv uao_detach(hmapping->uobj);
628 1.4 maxv
629 1.9 maxv hmapping->uobj = NULL;
630 1.9 maxv hmapping->present = false;
631 1.9 maxv
632 1.9 maxv return 0;
633 1.9 maxv }
634 1.9 maxv
635 1.9 maxv return ENOENT;
636 1.4 maxv }
637 1.4 maxv
638 1.4 maxv static int
639 1.4 maxv nvmm_hva_map(struct nvmm_ioc_hva_map *args)
640 1.4 maxv {
641 1.4 maxv struct vmspace *vmspace = curproc->p_vmspace;
642 1.4 maxv struct nvmm_machine *mach;
643 1.9 maxv struct nvmm_hmapping *hmapping;
644 1.4 maxv vaddr_t uva;
645 1.4 maxv int error;
646 1.4 maxv
647 1.4 maxv error = nvmm_machine_get(args->machid, &mach, true);
648 1.4 maxv if (error)
649 1.4 maxv return error;
650 1.4 maxv
651 1.9 maxv error = nvmm_hmapping_validate(mach, args->hva, args->size);
652 1.4 maxv if (error)
653 1.4 maxv goto out;
654 1.4 maxv
655 1.9 maxv hmapping = nvmm_hmapping_alloc(mach);
656 1.9 maxv if (hmapping == NULL) {
657 1.4 maxv error = ENOBUFS;
658 1.4 maxv goto out;
659 1.4 maxv }
660 1.4 maxv
661 1.9 maxv hmapping->hva = args->hva;
662 1.9 maxv hmapping->size = args->size;
663 1.9 maxv hmapping->uobj = uao_create(hmapping->size, 0);
664 1.9 maxv uva = hmapping->hva;
665 1.4 maxv
666 1.4 maxv /* Take a reference for the user. */
667 1.9 maxv uao_reference(hmapping->uobj);
668 1.4 maxv
669 1.4 maxv /* Map the uobj into the user address space, as pageable. */
670 1.9 maxv error = uvm_map(&vmspace->vm_map, &uva, hmapping->size, hmapping->uobj,
671 1.9 maxv 0, 0, UVM_MAPFLAG(UVM_PROT_RW, UVM_PROT_RW, UVM_INH_SHARE,
672 1.4 maxv UVM_ADV_RANDOM, UVM_FLAG_FIXED|UVM_FLAG_UNMAP));
673 1.4 maxv if (error) {
674 1.9 maxv uao_detach(hmapping->uobj);
675 1.4 maxv }
676 1.4 maxv
677 1.4 maxv out:
678 1.4 maxv nvmm_machine_put(mach);
679 1.4 maxv return error;
680 1.4 maxv }
681 1.4 maxv
682 1.4 maxv static int
683 1.4 maxv nvmm_hva_unmap(struct nvmm_ioc_hva_unmap *args)
684 1.4 maxv {
685 1.4 maxv struct nvmm_machine *mach;
686 1.4 maxv int error;
687 1.4 maxv
688 1.4 maxv error = nvmm_machine_get(args->machid, &mach, true);
689 1.4 maxv if (error)
690 1.4 maxv return error;
691 1.4 maxv
692 1.9 maxv error = nvmm_hmapping_free(mach, args->hva, args->size);
693 1.4 maxv
694 1.4 maxv nvmm_machine_put(mach);
695 1.9 maxv return error;
696 1.4 maxv }
697 1.4 maxv
698 1.4 maxv /* -------------------------------------------------------------------------- */
699 1.4 maxv
700 1.1 maxv static int
701 1.1 maxv nvmm_gpa_map(struct nvmm_ioc_gpa_map *args)
702 1.1 maxv {
703 1.1 maxv struct nvmm_machine *mach;
704 1.4 maxv struct uvm_object *uobj;
705 1.1 maxv gpaddr_t gpa;
706 1.4 maxv size_t off;
707 1.1 maxv int error;
708 1.1 maxv
709 1.1 maxv error = nvmm_machine_get(args->machid, &mach, false);
710 1.1 maxv if (error)
711 1.1 maxv return error;
712 1.1 maxv
713 1.11 maxv if ((args->prot & ~(PROT_READ|PROT_WRITE|PROT_EXEC)) != 0) {
714 1.11 maxv error = EINVAL;
715 1.11 maxv goto out;
716 1.11 maxv }
717 1.11 maxv
718 1.1 maxv if ((args->gpa % PAGE_SIZE) != 0 || (args->size % PAGE_SIZE) != 0 ||
719 1.1 maxv (args->hva % PAGE_SIZE) != 0) {
720 1.1 maxv error = EINVAL;
721 1.1 maxv goto out;
722 1.1 maxv }
723 1.1 maxv if (args->hva == 0) {
724 1.1 maxv error = EINVAL;
725 1.1 maxv goto out;
726 1.1 maxv }
727 1.1 maxv if (args->gpa < mach->gpa_begin || args->gpa >= mach->gpa_end) {
728 1.1 maxv error = EINVAL;
729 1.1 maxv goto out;
730 1.1 maxv }
731 1.1 maxv if (args->gpa + args->size <= args->gpa) {
732 1.1 maxv error = EINVAL;
733 1.1 maxv goto out;
734 1.1 maxv }
735 1.3 maxv if (args->gpa + args->size > mach->gpa_end) {
736 1.1 maxv error = EINVAL;
737 1.1 maxv goto out;
738 1.1 maxv }
739 1.1 maxv gpa = args->gpa;
740 1.1 maxv
741 1.9 maxv uobj = nvmm_hmapping_getuobj(mach, args->hva, args->size, &off);
742 1.4 maxv if (uobj == NULL) {
743 1.4 maxv error = EINVAL;
744 1.4 maxv goto out;
745 1.4 maxv }
746 1.4 maxv
747 1.4 maxv /* Take a reference for the machine. */
748 1.4 maxv uao_reference(uobj);
749 1.1 maxv
750 1.1 maxv /* Map the uobj into the machine address space, as pageable. */
751 1.4 maxv error = uvm_map(&mach->vm->vm_map, &gpa, args->size, uobj, off, 0,
752 1.11 maxv UVM_MAPFLAG(args->prot, UVM_PROT_RWX, UVM_INH_NONE,
753 1.4 maxv UVM_ADV_RANDOM, UVM_FLAG_FIXED|UVM_FLAG_UNMAP));
754 1.1 maxv if (error) {
755 1.4 maxv uao_detach(uobj);
756 1.1 maxv goto out;
757 1.1 maxv }
758 1.1 maxv if (gpa != args->gpa) {
759 1.4 maxv uao_detach(uobj);
760 1.1 maxv printf("[!] uvm_map problem\n");
761 1.1 maxv error = EINVAL;
762 1.1 maxv goto out;
763 1.1 maxv }
764 1.1 maxv
765 1.1 maxv out:
766 1.1 maxv nvmm_machine_put(mach);
767 1.1 maxv return error;
768 1.1 maxv }
769 1.1 maxv
770 1.1 maxv static int
771 1.1 maxv nvmm_gpa_unmap(struct nvmm_ioc_gpa_unmap *args)
772 1.1 maxv {
773 1.1 maxv struct nvmm_machine *mach;
774 1.1 maxv gpaddr_t gpa;
775 1.1 maxv int error;
776 1.1 maxv
777 1.1 maxv error = nvmm_machine_get(args->machid, &mach, false);
778 1.1 maxv if (error)
779 1.1 maxv return error;
780 1.1 maxv
781 1.1 maxv if ((args->gpa % PAGE_SIZE) != 0 || (args->size % PAGE_SIZE) != 0) {
782 1.1 maxv error = EINVAL;
783 1.1 maxv goto out;
784 1.1 maxv }
785 1.1 maxv if (args->gpa < mach->gpa_begin || args->gpa >= mach->gpa_end) {
786 1.1 maxv error = EINVAL;
787 1.1 maxv goto out;
788 1.1 maxv }
789 1.1 maxv if (args->gpa + args->size <= args->gpa) {
790 1.1 maxv error = EINVAL;
791 1.1 maxv goto out;
792 1.1 maxv }
793 1.1 maxv if (args->gpa + args->size >= mach->gpa_end) {
794 1.1 maxv error = EINVAL;
795 1.1 maxv goto out;
796 1.1 maxv }
797 1.1 maxv gpa = args->gpa;
798 1.1 maxv
799 1.1 maxv /* Unmap the memory from the machine. */
800 1.1 maxv uvm_unmap(&mach->vm->vm_map, gpa, gpa + args->size);
801 1.1 maxv
802 1.1 maxv out:
803 1.1 maxv nvmm_machine_put(mach);
804 1.1 maxv return error;
805 1.1 maxv }
806 1.1 maxv
807 1.1 maxv /* -------------------------------------------------------------------------- */
808 1.1 maxv
809 1.1 maxv static int
810 1.1 maxv nvmm_init(void)
811 1.1 maxv {
812 1.1 maxv size_t i, n;
813 1.1 maxv
814 1.1 maxv for (i = 0; i < __arraycount(nvmm_impl_list); i++) {
815 1.1 maxv if (!(*nvmm_impl_list[i]->ident)()) {
816 1.1 maxv continue;
817 1.1 maxv }
818 1.1 maxv nvmm_impl = nvmm_impl_list[i];
819 1.1 maxv break;
820 1.1 maxv }
821 1.1 maxv if (nvmm_impl == NULL) {
822 1.1 maxv printf("[!] No implementation found\n");
823 1.1 maxv return ENOTSUP;
824 1.1 maxv }
825 1.1 maxv
826 1.1 maxv for (i = 0; i < NVMM_MAX_MACHINES; i++) {
827 1.1 maxv machines[i].machid = i;
828 1.1 maxv rw_init(&machines[i].lock);
829 1.1 maxv for (n = 0; n < NVMM_MAX_VCPUS; n++) {
830 1.1 maxv mutex_init(&machines[i].cpus[n].lock, MUTEX_DEFAULT,
831 1.1 maxv IPL_NONE);
832 1.1 maxv machines[i].cpus[n].hcpu_last = -1;
833 1.1 maxv }
834 1.1 maxv }
835 1.1 maxv
836 1.1 maxv (*nvmm_impl->init)();
837 1.1 maxv
838 1.1 maxv return 0;
839 1.1 maxv }
840 1.1 maxv
841 1.1 maxv static void
842 1.1 maxv nvmm_fini(void)
843 1.1 maxv {
844 1.1 maxv size_t i, n;
845 1.1 maxv
846 1.1 maxv for (i = 0; i < NVMM_MAX_MACHINES; i++) {
847 1.1 maxv rw_destroy(&machines[i].lock);
848 1.1 maxv for (n = 0; n < NVMM_MAX_VCPUS; n++) {
849 1.1 maxv mutex_destroy(&machines[i].cpus[n].lock);
850 1.1 maxv }
851 1.1 maxv }
852 1.1 maxv
853 1.1 maxv (*nvmm_impl->fini)();
854 1.1 maxv }
855 1.1 maxv
856 1.1 maxv /* -------------------------------------------------------------------------- */
857 1.1 maxv
858 1.1 maxv static int
859 1.1 maxv nvmm_open(dev_t dev, int flags, int type, struct lwp *l)
860 1.1 maxv {
861 1.1 maxv if (minor(dev) != 0) {
862 1.1 maxv return EXDEV;
863 1.1 maxv }
864 1.1 maxv
865 1.1 maxv return 0;
866 1.1 maxv }
867 1.1 maxv
868 1.1 maxv static int
869 1.1 maxv nvmm_close(dev_t dev, int flags, int type, struct lwp *l)
870 1.1 maxv {
871 1.1 maxv KASSERT(minor(dev) == 0);
872 1.1 maxv
873 1.1 maxv nvmm_kill_machines(l->l_proc->p_pid);
874 1.1 maxv
875 1.1 maxv return 0;
876 1.1 maxv }
877 1.1 maxv
878 1.1 maxv static int
879 1.1 maxv nvmm_ioctl(dev_t dev, u_long cmd, void *data, int flags, struct lwp *l)
880 1.1 maxv {
881 1.1 maxv KASSERT(minor(dev) == 0);
882 1.1 maxv
883 1.1 maxv switch (cmd) {
884 1.1 maxv case NVMM_IOC_CAPABILITY:
885 1.1 maxv return nvmm_capability(data);
886 1.1 maxv case NVMM_IOC_MACHINE_CREATE:
887 1.1 maxv return nvmm_machine_create(data);
888 1.1 maxv case NVMM_IOC_MACHINE_DESTROY:
889 1.1 maxv return nvmm_machine_destroy(data);
890 1.1 maxv case NVMM_IOC_MACHINE_CONFIGURE:
891 1.1 maxv return nvmm_machine_configure(data);
892 1.1 maxv case NVMM_IOC_VCPU_CREATE:
893 1.1 maxv return nvmm_vcpu_create(data);
894 1.1 maxv case NVMM_IOC_VCPU_DESTROY:
895 1.1 maxv return nvmm_vcpu_destroy(data);
896 1.1 maxv case NVMM_IOC_VCPU_SETSTATE:
897 1.1 maxv return nvmm_vcpu_setstate(data);
898 1.1 maxv case NVMM_IOC_VCPU_GETSTATE:
899 1.1 maxv return nvmm_vcpu_getstate(data);
900 1.1 maxv case NVMM_IOC_VCPU_INJECT:
901 1.1 maxv return nvmm_vcpu_inject(data);
902 1.1 maxv case NVMM_IOC_VCPU_RUN:
903 1.1 maxv return nvmm_vcpu_run(data);
904 1.1 maxv case NVMM_IOC_GPA_MAP:
905 1.1 maxv return nvmm_gpa_map(data);
906 1.1 maxv case NVMM_IOC_GPA_UNMAP:
907 1.1 maxv return nvmm_gpa_unmap(data);
908 1.4 maxv case NVMM_IOC_HVA_MAP:
909 1.4 maxv return nvmm_hva_map(data);
910 1.4 maxv case NVMM_IOC_HVA_UNMAP:
911 1.4 maxv return nvmm_hva_unmap(data);
912 1.1 maxv default:
913 1.1 maxv return EINVAL;
914 1.1 maxv }
915 1.1 maxv }
916 1.1 maxv
917 1.1 maxv const struct cdevsw nvmm_cdevsw = {
918 1.1 maxv .d_open = nvmm_open,
919 1.1 maxv .d_close = nvmm_close,
920 1.1 maxv .d_read = noread,
921 1.1 maxv .d_write = nowrite,
922 1.1 maxv .d_ioctl = nvmm_ioctl,
923 1.1 maxv .d_stop = nostop,
924 1.1 maxv .d_tty = notty,
925 1.1 maxv .d_poll = nopoll,
926 1.1 maxv .d_mmap = nommap,
927 1.1 maxv .d_kqfilter = nokqfilter,
928 1.1 maxv .d_discard = nodiscard,
929 1.1 maxv .d_flag = D_OTHER | D_MPSAFE
930 1.1 maxv };
931 1.1 maxv
932 1.1 maxv void
933 1.1 maxv nvmmattach(int nunits)
934 1.1 maxv {
935 1.1 maxv /* nothing */
936 1.1 maxv }
937 1.1 maxv
938 1.12 maxv MODULE(MODULE_CLASS_ANY, nvmm, NULL);
939 1.1 maxv
940 1.1 maxv static int
941 1.1 maxv nvmm_modcmd(modcmd_t cmd, void *arg)
942 1.1 maxv {
943 1.1 maxv int error;
944 1.1 maxv
945 1.1 maxv switch (cmd) {
946 1.1 maxv case MODULE_CMD_INIT:
947 1.1 maxv error = nvmm_init();
948 1.1 maxv if (error)
949 1.1 maxv return error;
950 1.1 maxv
951 1.1 maxv #if defined(_MODULE)
952 1.1 maxv {
953 1.1 maxv devmajor_t bmajor = NODEVMAJOR;
954 1.1 maxv devmajor_t cmajor = 345;
955 1.1 maxv
956 1.1 maxv /* mknod /dev/nvmm c 345 0 */
957 1.1 maxv error = devsw_attach("nvmm", NULL, &bmajor,
958 1.1 maxv &nvmm_cdevsw, &cmajor);
959 1.1 maxv if (error) {
960 1.1 maxv nvmm_fini();
961 1.1 maxv return error;
962 1.1 maxv }
963 1.1 maxv }
964 1.1 maxv #endif
965 1.1 maxv return 0;
966 1.1 maxv
967 1.1 maxv case MODULE_CMD_FINI:
968 1.13 maxv if (nmachines > 0) {
969 1.13 maxv return EBUSY;
970 1.13 maxv }
971 1.1 maxv #if defined(_MODULE)
972 1.1 maxv {
973 1.1 maxv error = devsw_detach(NULL, &nvmm_cdevsw);
974 1.1 maxv if (error) {
975 1.1 maxv return error;
976 1.1 maxv }
977 1.1 maxv }
978 1.1 maxv #endif
979 1.1 maxv nvmm_fini();
980 1.1 maxv return 0;
981 1.1 maxv
982 1.13 maxv case MODULE_CMD_AUTOUNLOAD:
983 1.13 maxv return EBUSY;
984 1.13 maxv
985 1.1 maxv default:
986 1.1 maxv return ENOTTY;
987 1.1 maxv }
988 1.1 maxv }
989