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