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