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