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