xref: /freebsd/lib/libvmmapi/vmmapi.c (revision 95ebc360efc984cab758d634f1c357b73650f651)
1366f6083SPeter Grehan /*-
2366f6083SPeter Grehan  * Copyright (c) 2011 NetApp, Inc.
3366f6083SPeter Grehan  * All rights reserved.
4366f6083SPeter Grehan  *
5366f6083SPeter Grehan  * Redistribution and use in source and binary forms, with or without
6366f6083SPeter Grehan  * modification, are permitted provided that the following conditions
7366f6083SPeter Grehan  * are met:
8366f6083SPeter Grehan  * 1. Redistributions of source code must retain the above copyright
9366f6083SPeter Grehan  *    notice, this list of conditions and the following disclaimer.
10366f6083SPeter Grehan  * 2. Redistributions in binary form must reproduce the above copyright
11366f6083SPeter Grehan  *    notice, this list of conditions and the following disclaimer in the
12366f6083SPeter Grehan  *    documentation and/or other materials provided with the distribution.
13366f6083SPeter Grehan  *
14366f6083SPeter Grehan  * THIS SOFTWARE IS PROVIDED BY NETAPP, INC ``AS IS'' AND
15366f6083SPeter Grehan  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16366f6083SPeter Grehan  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17366f6083SPeter Grehan  * ARE DISCLAIMED.  IN NO EVENT SHALL NETAPP, INC OR CONTRIBUTORS BE LIABLE
18366f6083SPeter Grehan  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19366f6083SPeter Grehan  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20366f6083SPeter Grehan  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21366f6083SPeter Grehan  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22366f6083SPeter Grehan  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23366f6083SPeter Grehan  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24366f6083SPeter Grehan  * SUCH DAMAGE.
25366f6083SPeter Grehan  *
26366f6083SPeter Grehan  * $FreeBSD$
27366f6083SPeter Grehan  */
28366f6083SPeter Grehan 
29366f6083SPeter Grehan #include <sys/cdefs.h>
30366f6083SPeter Grehan __FBSDID("$FreeBSD$");
31366f6083SPeter Grehan 
32*95ebc360SNeel Natu #include <sys/param.h>
33366f6083SPeter Grehan #include <sys/sysctl.h>
34366f6083SPeter Grehan #include <sys/ioctl.h>
35366f6083SPeter Grehan #include <sys/mman.h>
366303b65dSNeel Natu #include <sys/_iovec.h>
37*95ebc360SNeel Natu #include <sys/cpuset.h>
38366f6083SPeter Grehan 
39366f6083SPeter Grehan #include <machine/specialreg.h>
40da11f4aaSNeel Natu #include <machine/param.h>
41366f6083SPeter Grehan 
42366f6083SPeter Grehan #include <stdio.h>
43366f6083SPeter Grehan #include <stdlib.h>
44366f6083SPeter Grehan #include <assert.h>
45366f6083SPeter Grehan #include <string.h>
46366f6083SPeter Grehan #include <fcntl.h>
47366f6083SPeter Grehan #include <unistd.h>
48366f6083SPeter Grehan 
49200758f1SNeel Natu #include <libutil.h>
50200758f1SNeel Natu 
51366f6083SPeter Grehan #include <machine/vmm.h>
52366f6083SPeter Grehan #include <machine/vmm_dev.h>
53366f6083SPeter Grehan 
54366f6083SPeter Grehan #include "vmmapi.h"
55366f6083SPeter Grehan 
56200758f1SNeel Natu #define	MB	(1024 * 1024UL)
57b060ba50SNeel Natu #define	GB	(1024 * 1024 * 1024UL)
58b060ba50SNeel Natu 
59366f6083SPeter Grehan struct vmctx {
60366f6083SPeter Grehan 	int	fd;
61b060ba50SNeel Natu 	uint32_t lowmem_limit;
62b060ba50SNeel Natu 	enum vm_mmap_style vms;
630dd10c00SNeel Natu 	int	memflags;
64b060ba50SNeel Natu 	size_t	lowmem;
65b060ba50SNeel Natu 	char	*lowmem_addr;
66b060ba50SNeel Natu 	size_t	highmem;
67b060ba50SNeel Natu 	char	*highmem_addr;
68366f6083SPeter Grehan 	char	*name;
69366f6083SPeter Grehan };
70366f6083SPeter Grehan 
71366f6083SPeter Grehan #define	CREATE(x)  sysctlbyname("hw.vmm.create", NULL, NULL, (x), strlen((x)))
72366f6083SPeter Grehan #define	DESTROY(x) sysctlbyname("hw.vmm.destroy", NULL, NULL, (x), strlen((x)))
73366f6083SPeter Grehan 
74366f6083SPeter Grehan static int
75366f6083SPeter Grehan vm_device_open(const char *name)
76366f6083SPeter Grehan {
77366f6083SPeter Grehan         int fd, len;
78366f6083SPeter Grehan         char *vmfile;
79366f6083SPeter Grehan 
80366f6083SPeter Grehan 	len = strlen("/dev/vmm/") + strlen(name) + 1;
81366f6083SPeter Grehan 	vmfile = malloc(len);
82366f6083SPeter Grehan 	assert(vmfile != NULL);
83366f6083SPeter Grehan 	snprintf(vmfile, len, "/dev/vmm/%s", name);
84366f6083SPeter Grehan 
85366f6083SPeter Grehan         /* Open the device file */
86366f6083SPeter Grehan         fd = open(vmfile, O_RDWR, 0);
87366f6083SPeter Grehan 
88366f6083SPeter Grehan 	free(vmfile);
89366f6083SPeter Grehan         return (fd);
90366f6083SPeter Grehan }
91366f6083SPeter Grehan 
92366f6083SPeter Grehan int
93366f6083SPeter Grehan vm_create(const char *name)
94366f6083SPeter Grehan {
95366f6083SPeter Grehan 
96366f6083SPeter Grehan 	return (CREATE((char *)name));
97366f6083SPeter Grehan }
98366f6083SPeter Grehan 
99366f6083SPeter Grehan struct vmctx *
100366f6083SPeter Grehan vm_open(const char *name)
101366f6083SPeter Grehan {
102366f6083SPeter Grehan 	struct vmctx *vm;
103366f6083SPeter Grehan 
104366f6083SPeter Grehan 	vm = malloc(sizeof(struct vmctx) + strlen(name) + 1);
105366f6083SPeter Grehan 	assert(vm != NULL);
106366f6083SPeter Grehan 
107366f6083SPeter Grehan 	vm->fd = -1;
1080dd10c00SNeel Natu 	vm->memflags = 0;
109b060ba50SNeel Natu 	vm->lowmem_limit = 3 * GB;
110366f6083SPeter Grehan 	vm->name = (char *)(vm + 1);
111366f6083SPeter Grehan 	strcpy(vm->name, name);
112366f6083SPeter Grehan 
113366f6083SPeter Grehan 	if ((vm->fd = vm_device_open(vm->name)) < 0)
114366f6083SPeter Grehan 		goto err;
115366f6083SPeter Grehan 
116366f6083SPeter Grehan 	return (vm);
117366f6083SPeter Grehan err:
118366f6083SPeter Grehan 	vm_destroy(vm);
119366f6083SPeter Grehan 	return (NULL);
120366f6083SPeter Grehan }
121366f6083SPeter Grehan 
122366f6083SPeter Grehan void
123366f6083SPeter Grehan vm_destroy(struct vmctx *vm)
124366f6083SPeter Grehan {
125366f6083SPeter Grehan 	assert(vm != NULL);
126366f6083SPeter Grehan 
127366f6083SPeter Grehan 	if (vm->fd >= 0)
128366f6083SPeter Grehan 		close(vm->fd);
129f7d51510SNeel Natu 	DESTROY(vm->name);
130f7d51510SNeel Natu 
131366f6083SPeter Grehan 	free(vm);
132366f6083SPeter Grehan }
133366f6083SPeter Grehan 
134366f6083SPeter Grehan int
135200758f1SNeel Natu vm_parse_memsize(const char *optarg, size_t *ret_memsize)
136200758f1SNeel Natu {
137200758f1SNeel Natu 	char *endptr;
138200758f1SNeel Natu 	size_t optval;
139200758f1SNeel Natu 	int error;
140200758f1SNeel Natu 
141200758f1SNeel Natu 	optval = strtoul(optarg, &endptr, 0);
142200758f1SNeel Natu 	if (*optarg != '\0' && *endptr == '\0') {
143200758f1SNeel Natu 		/*
144200758f1SNeel Natu 		 * For the sake of backward compatibility if the memory size
145200758f1SNeel Natu 		 * specified on the command line is less than a megabyte then
146200758f1SNeel Natu 		 * it is interpreted as being in units of MB.
147200758f1SNeel Natu 		 */
148200758f1SNeel Natu 		if (optval < MB)
149200758f1SNeel Natu 			optval *= MB;
150200758f1SNeel Natu 		*ret_memsize = optval;
151200758f1SNeel Natu 		error = 0;
152200758f1SNeel Natu 	} else
153200758f1SNeel Natu 		error = expand_number(optarg, ret_memsize);
154200758f1SNeel Natu 
155200758f1SNeel Natu 	return (error);
156200758f1SNeel Natu }
157200758f1SNeel Natu 
158200758f1SNeel Natu int
159318224bbSNeel Natu vm_get_memory_seg(struct vmctx *ctx, vm_paddr_t gpa, size_t *ret_len,
160318224bbSNeel Natu 		  int *wired)
161366f6083SPeter Grehan {
162366f6083SPeter Grehan 	int error;
163366f6083SPeter Grehan 	struct vm_memory_segment seg;
164366f6083SPeter Grehan 
165366f6083SPeter Grehan 	bzero(&seg, sizeof(seg));
166366f6083SPeter Grehan 	seg.gpa = gpa;
167366f6083SPeter Grehan 	error = ioctl(ctx->fd, VM_GET_MEMORY_SEG, &seg);
168366f6083SPeter Grehan 	*ret_len = seg.len;
169318224bbSNeel Natu 	if (wired != NULL)
170318224bbSNeel Natu 		*wired = seg.wired;
171366f6083SPeter Grehan 	return (error);
172366f6083SPeter Grehan }
173366f6083SPeter Grehan 
174b060ba50SNeel Natu uint32_t
175b060ba50SNeel Natu vm_get_lowmem_limit(struct vmctx *ctx)
176b060ba50SNeel Natu {
177b060ba50SNeel Natu 
178b060ba50SNeel Natu 	return (ctx->lowmem_limit);
179b060ba50SNeel Natu }
180b060ba50SNeel Natu 
181b060ba50SNeel Natu void
182b060ba50SNeel Natu vm_set_lowmem_limit(struct vmctx *ctx, uint32_t limit)
183b060ba50SNeel Natu {
184b060ba50SNeel Natu 
185b060ba50SNeel Natu 	ctx->lowmem_limit = limit;
186b060ba50SNeel Natu }
187b060ba50SNeel Natu 
1880dd10c00SNeel Natu void
1890dd10c00SNeel Natu vm_set_memflags(struct vmctx *ctx, int flags)
1900dd10c00SNeel Natu {
1910dd10c00SNeel Natu 
1920dd10c00SNeel Natu 	ctx->memflags = flags;
1930dd10c00SNeel Natu }
1940dd10c00SNeel Natu 
195b060ba50SNeel Natu static int
196b060ba50SNeel Natu setup_memory_segment(struct vmctx *ctx, vm_paddr_t gpa, size_t len, char **addr)
197366f6083SPeter Grehan {
1980dd10c00SNeel Natu 	int error, mmap_flags;
199366f6083SPeter Grehan 	struct vm_memory_segment seg;
200366f6083SPeter Grehan 
201366f6083SPeter Grehan 	/*
202366f6083SPeter Grehan 	 * Create and optionally map 'len' bytes of memory at guest
203366f6083SPeter Grehan 	 * physical address 'gpa'
204366f6083SPeter Grehan 	 */
205366f6083SPeter Grehan 	bzero(&seg, sizeof(seg));
206366f6083SPeter Grehan 	seg.gpa = gpa;
207366f6083SPeter Grehan 	seg.len = len;
208366f6083SPeter Grehan 	error = ioctl(ctx->fd, VM_MAP_MEMORY, &seg);
209b060ba50SNeel Natu 	if (error == 0 && addr != NULL) {
2100dd10c00SNeel Natu 		mmap_flags = MAP_SHARED;
2110dd10c00SNeel Natu 		if ((ctx->memflags & VM_MEM_F_INCORE) == 0)
2120dd10c00SNeel Natu 			mmap_flags |= MAP_NOCORE;
2130dd10c00SNeel Natu 		*addr = mmap(NULL, len, PROT_READ | PROT_WRITE, mmap_flags,
214366f6083SPeter Grehan 		    ctx->fd, gpa);
215366f6083SPeter Grehan 	}
216366f6083SPeter Grehan 	return (error);
217366f6083SPeter Grehan }
218366f6083SPeter Grehan 
219b060ba50SNeel Natu int
220b060ba50SNeel Natu vm_setup_memory(struct vmctx *ctx, size_t memsize, enum vm_mmap_style vms)
221b060ba50SNeel Natu {
222b060ba50SNeel Natu 	char **addr;
223b060ba50SNeel Natu 	int error;
224b060ba50SNeel Natu 
225b060ba50SNeel Natu 	/* XXX VM_MMAP_SPARSE not implemented yet */
226b060ba50SNeel Natu 	assert(vms == VM_MMAP_NONE || vms == VM_MMAP_ALL);
227b060ba50SNeel Natu 	ctx->vms = vms;
228b060ba50SNeel Natu 
229b060ba50SNeel Natu 	/*
230b060ba50SNeel Natu 	 * If 'memsize' cannot fit entirely in the 'lowmem' segment then
231b060ba50SNeel Natu 	 * create another 'highmem' segment above 4GB for the remainder.
232b060ba50SNeel Natu 	 */
233b060ba50SNeel Natu 	if (memsize > ctx->lowmem_limit) {
234b060ba50SNeel Natu 		ctx->lowmem = ctx->lowmem_limit;
235b060ba50SNeel Natu 		ctx->highmem = memsize - ctx->lowmem;
236b060ba50SNeel Natu 	} else {
237b060ba50SNeel Natu 		ctx->lowmem = memsize;
238b060ba50SNeel Natu 		ctx->highmem = 0;
239b060ba50SNeel Natu 	}
240b060ba50SNeel Natu 
241b060ba50SNeel Natu 	if (ctx->lowmem > 0) {
242b060ba50SNeel Natu 		addr = (vms == VM_MMAP_ALL) ? &ctx->lowmem_addr : NULL;
243b060ba50SNeel Natu 		error = setup_memory_segment(ctx, 0, ctx->lowmem, addr);
244b060ba50SNeel Natu 		if (error)
245b060ba50SNeel Natu 			return (error);
246b060ba50SNeel Natu 	}
247b060ba50SNeel Natu 
248b060ba50SNeel Natu 	if (ctx->highmem > 0) {
249b060ba50SNeel Natu 		addr = (vms == VM_MMAP_ALL) ? &ctx->highmem_addr : NULL;
250b060ba50SNeel Natu 		error = setup_memory_segment(ctx, 4*GB, ctx->highmem, addr);
251b060ba50SNeel Natu 		if (error)
252b060ba50SNeel Natu 			return (error);
253b060ba50SNeel Natu 	}
254b060ba50SNeel Natu 
255b060ba50SNeel Natu 	return (0);
256b060ba50SNeel Natu }
257b060ba50SNeel Natu 
258b060ba50SNeel Natu void *
259b060ba50SNeel Natu vm_map_gpa(struct vmctx *ctx, vm_paddr_t gaddr, size_t len)
260366f6083SPeter Grehan {
261366f6083SPeter Grehan 
262b060ba50SNeel Natu 	/* XXX VM_MMAP_SPARSE not implemented yet */
263b060ba50SNeel Natu 	assert(ctx->vms == VM_MMAP_ALL);
264b060ba50SNeel Natu 
265b060ba50SNeel Natu 	if (gaddr < ctx->lowmem && gaddr + len <= ctx->lowmem)
266b060ba50SNeel Natu 		return ((void *)(ctx->lowmem_addr + gaddr));
267b060ba50SNeel Natu 
268b060ba50SNeel Natu 	if (gaddr >= 4*GB) {
269b060ba50SNeel Natu 		gaddr -= 4*GB;
270b060ba50SNeel Natu 		if (gaddr < ctx->highmem && gaddr + len <= ctx->highmem)
271b060ba50SNeel Natu 			return ((void *)(ctx->highmem_addr + gaddr));
272b060ba50SNeel Natu 	}
273b060ba50SNeel Natu 
274b060ba50SNeel Natu 	return (NULL);
275366f6083SPeter Grehan }
276366f6083SPeter Grehan 
277366f6083SPeter Grehan int
278366f6083SPeter Grehan vm_set_desc(struct vmctx *ctx, int vcpu, int reg,
279366f6083SPeter Grehan 	    uint64_t base, uint32_t limit, uint32_t access)
280366f6083SPeter Grehan {
281366f6083SPeter Grehan 	int error;
282366f6083SPeter Grehan 	struct vm_seg_desc vmsegdesc;
283366f6083SPeter Grehan 
284366f6083SPeter Grehan 	bzero(&vmsegdesc, sizeof(vmsegdesc));
285366f6083SPeter Grehan 	vmsegdesc.cpuid = vcpu;
286366f6083SPeter Grehan 	vmsegdesc.regnum = reg;
287366f6083SPeter Grehan 	vmsegdesc.desc.base = base;
288366f6083SPeter Grehan 	vmsegdesc.desc.limit = limit;
289366f6083SPeter Grehan 	vmsegdesc.desc.access = access;
290366f6083SPeter Grehan 
291366f6083SPeter Grehan 	error = ioctl(ctx->fd, VM_SET_SEGMENT_DESCRIPTOR, &vmsegdesc);
292366f6083SPeter Grehan 	return (error);
293366f6083SPeter Grehan }
294366f6083SPeter Grehan 
295366f6083SPeter Grehan int
296366f6083SPeter Grehan vm_get_desc(struct vmctx *ctx, int vcpu, int reg,
297366f6083SPeter Grehan 	    uint64_t *base, uint32_t *limit, uint32_t *access)
298366f6083SPeter Grehan {
299366f6083SPeter Grehan 	int error;
300366f6083SPeter Grehan 	struct vm_seg_desc vmsegdesc;
301366f6083SPeter Grehan 
302366f6083SPeter Grehan 	bzero(&vmsegdesc, sizeof(vmsegdesc));
303366f6083SPeter Grehan 	vmsegdesc.cpuid = vcpu;
304366f6083SPeter Grehan 	vmsegdesc.regnum = reg;
305366f6083SPeter Grehan 
306366f6083SPeter Grehan 	error = ioctl(ctx->fd, VM_GET_SEGMENT_DESCRIPTOR, &vmsegdesc);
307366f6083SPeter Grehan 	if (error == 0) {
308366f6083SPeter Grehan 		*base = vmsegdesc.desc.base;
309366f6083SPeter Grehan 		*limit = vmsegdesc.desc.limit;
310366f6083SPeter Grehan 		*access = vmsegdesc.desc.access;
311366f6083SPeter Grehan 	}
312366f6083SPeter Grehan 	return (error);
313366f6083SPeter Grehan }
314366f6083SPeter Grehan 
315366f6083SPeter Grehan int
316366f6083SPeter Grehan vm_set_register(struct vmctx *ctx, int vcpu, int reg, uint64_t val)
317366f6083SPeter Grehan {
318366f6083SPeter Grehan 	int error;
319366f6083SPeter Grehan 	struct vm_register vmreg;
320366f6083SPeter Grehan 
321366f6083SPeter Grehan 	bzero(&vmreg, sizeof(vmreg));
322366f6083SPeter Grehan 	vmreg.cpuid = vcpu;
323366f6083SPeter Grehan 	vmreg.regnum = reg;
324366f6083SPeter Grehan 	vmreg.regval = val;
325366f6083SPeter Grehan 
326366f6083SPeter Grehan 	error = ioctl(ctx->fd, VM_SET_REGISTER, &vmreg);
327366f6083SPeter Grehan 	return (error);
328366f6083SPeter Grehan }
329366f6083SPeter Grehan 
330366f6083SPeter Grehan int
331366f6083SPeter Grehan vm_get_register(struct vmctx *ctx, int vcpu, int reg, uint64_t *ret_val)
332366f6083SPeter Grehan {
333366f6083SPeter Grehan 	int error;
334366f6083SPeter Grehan 	struct vm_register vmreg;
335366f6083SPeter Grehan 
336366f6083SPeter Grehan 	bzero(&vmreg, sizeof(vmreg));
337366f6083SPeter Grehan 	vmreg.cpuid = vcpu;
338366f6083SPeter Grehan 	vmreg.regnum = reg;
339366f6083SPeter Grehan 
340366f6083SPeter Grehan 	error = ioctl(ctx->fd, VM_GET_REGISTER, &vmreg);
341366f6083SPeter Grehan 	*ret_val = vmreg.regval;
342366f6083SPeter Grehan 	return (error);
343366f6083SPeter Grehan }
344366f6083SPeter Grehan 
345366f6083SPeter Grehan int
346366f6083SPeter Grehan vm_run(struct vmctx *ctx, int vcpu, uint64_t rip, struct vm_exit *vmexit)
347366f6083SPeter Grehan {
348366f6083SPeter Grehan 	int error;
349366f6083SPeter Grehan 	struct vm_run vmrun;
350366f6083SPeter Grehan 
351366f6083SPeter Grehan 	bzero(&vmrun, sizeof(vmrun));
352366f6083SPeter Grehan 	vmrun.cpuid = vcpu;
353366f6083SPeter Grehan 	vmrun.rip = rip;
354366f6083SPeter Grehan 
355366f6083SPeter Grehan 	error = ioctl(ctx->fd, VM_RUN, &vmrun);
356366f6083SPeter Grehan 	bcopy(&vmrun.vm_exit, vmexit, sizeof(struct vm_exit));
357366f6083SPeter Grehan 	return (error);
358366f6083SPeter Grehan }
359366f6083SPeter Grehan 
360b15a09c0SNeel Natu int
361f0fdcfe2SNeel Natu vm_suspend(struct vmctx *ctx, enum vm_suspend_how how)
362b15a09c0SNeel Natu {
363f0fdcfe2SNeel Natu 	struct vm_suspend vmsuspend;
364b15a09c0SNeel Natu 
365f0fdcfe2SNeel Natu 	bzero(&vmsuspend, sizeof(vmsuspend));
366f0fdcfe2SNeel Natu 	vmsuspend.how = how;
367f0fdcfe2SNeel Natu 	return (ioctl(ctx->fd, VM_SUSPEND, &vmsuspend));
368b15a09c0SNeel Natu }
369b15a09c0SNeel Natu 
370366f6083SPeter Grehan static int
371dc506506SNeel Natu vm_inject_exception_real(struct vmctx *ctx, int vcpu, int vector,
372dc506506SNeel Natu     int error_code, int error_code_valid)
373366f6083SPeter Grehan {
374dc506506SNeel Natu 	struct vm_exception exc;
375366f6083SPeter Grehan 
376dc506506SNeel Natu 	bzero(&exc, sizeof(exc));
377dc506506SNeel Natu 	exc.cpuid = vcpu;
378dc506506SNeel Natu 	exc.vector = vector;
379dc506506SNeel Natu 	exc.error_code = error_code;
380dc506506SNeel Natu 	exc.error_code_valid = error_code_valid;
381366f6083SPeter Grehan 
382dc506506SNeel Natu 	return (ioctl(ctx->fd, VM_INJECT_EXCEPTION, &exc));
383366f6083SPeter Grehan }
384366f6083SPeter Grehan 
385366f6083SPeter Grehan int
386dc506506SNeel Natu vm_inject_exception(struct vmctx *ctx, int vcpu, int vector)
387366f6083SPeter Grehan {
388366f6083SPeter Grehan 
389dc506506SNeel Natu 	return (vm_inject_exception_real(ctx, vcpu, vector, 0, 0));
390366f6083SPeter Grehan }
391366f6083SPeter Grehan 
392366f6083SPeter Grehan int
393dc506506SNeel Natu vm_inject_exception2(struct vmctx *ctx, int vcpu, int vector, int errcode)
394366f6083SPeter Grehan {
395366f6083SPeter Grehan 
396dc506506SNeel Natu 	return (vm_inject_exception_real(ctx, vcpu, vector, errcode, 1));
397366f6083SPeter Grehan }
398366f6083SPeter Grehan 
399366f6083SPeter Grehan int
40090d4b48fSNeel Natu vm_apicid2vcpu(struct vmctx *ctx, int apicid)
40190d4b48fSNeel Natu {
40290d4b48fSNeel Natu 	/*
40390d4b48fSNeel Natu 	 * The apic id associated with the 'vcpu' has the same numerical value
40490d4b48fSNeel Natu 	 * as the 'vcpu' itself.
40590d4b48fSNeel Natu 	 */
40690d4b48fSNeel Natu 	return (apicid);
40790d4b48fSNeel Natu }
40890d4b48fSNeel Natu 
40990d4b48fSNeel Natu int
410366f6083SPeter Grehan vm_lapic_irq(struct vmctx *ctx, int vcpu, int vector)
411366f6083SPeter Grehan {
412366f6083SPeter Grehan 	struct vm_lapic_irq vmirq;
413366f6083SPeter Grehan 
414366f6083SPeter Grehan 	bzero(&vmirq, sizeof(vmirq));
415366f6083SPeter Grehan 	vmirq.cpuid = vcpu;
416366f6083SPeter Grehan 	vmirq.vector = vector;
417366f6083SPeter Grehan 
418366f6083SPeter Grehan 	return (ioctl(ctx->fd, VM_LAPIC_IRQ, &vmirq));
419366f6083SPeter Grehan }
420366f6083SPeter Grehan 
421366f6083SPeter Grehan int
422330baf58SJohn Baldwin vm_lapic_local_irq(struct vmctx *ctx, int vcpu, int vector)
423330baf58SJohn Baldwin {
424330baf58SJohn Baldwin 	struct vm_lapic_irq vmirq;
425330baf58SJohn Baldwin 
426330baf58SJohn Baldwin 	bzero(&vmirq, sizeof(vmirq));
427330baf58SJohn Baldwin 	vmirq.cpuid = vcpu;
428330baf58SJohn Baldwin 	vmirq.vector = vector;
429330baf58SJohn Baldwin 
430330baf58SJohn Baldwin 	return (ioctl(ctx->fd, VM_LAPIC_LOCAL_IRQ, &vmirq));
431330baf58SJohn Baldwin }
432330baf58SJohn Baldwin 
433330baf58SJohn Baldwin int
4344f8be175SNeel Natu vm_lapic_msi(struct vmctx *ctx, uint64_t addr, uint64_t msg)
4354f8be175SNeel Natu {
4364f8be175SNeel Natu 	struct vm_lapic_msi vmmsi;
4374f8be175SNeel Natu 
4384f8be175SNeel Natu 	bzero(&vmmsi, sizeof(vmmsi));
4394f8be175SNeel Natu 	vmmsi.addr = addr;
4404f8be175SNeel Natu 	vmmsi.msg = msg;
4414f8be175SNeel Natu 
4424f8be175SNeel Natu 	return (ioctl(ctx->fd, VM_LAPIC_MSI, &vmmsi));
4434f8be175SNeel Natu }
4444f8be175SNeel Natu 
4454f8be175SNeel Natu int
446565bbb86SNeel Natu vm_ioapic_assert_irq(struct vmctx *ctx, int irq)
447565bbb86SNeel Natu {
448565bbb86SNeel Natu 	struct vm_ioapic_irq ioapic_irq;
449565bbb86SNeel Natu 
450565bbb86SNeel Natu 	bzero(&ioapic_irq, sizeof(struct vm_ioapic_irq));
451565bbb86SNeel Natu 	ioapic_irq.irq = irq;
452565bbb86SNeel Natu 
453565bbb86SNeel Natu 	return (ioctl(ctx->fd, VM_IOAPIC_ASSERT_IRQ, &ioapic_irq));
454565bbb86SNeel Natu }
455565bbb86SNeel Natu 
456565bbb86SNeel Natu int
457565bbb86SNeel Natu vm_ioapic_deassert_irq(struct vmctx *ctx, int irq)
458565bbb86SNeel Natu {
459565bbb86SNeel Natu 	struct vm_ioapic_irq ioapic_irq;
460565bbb86SNeel Natu 
461565bbb86SNeel Natu 	bzero(&ioapic_irq, sizeof(struct vm_ioapic_irq));
462565bbb86SNeel Natu 	ioapic_irq.irq = irq;
463565bbb86SNeel Natu 
464565bbb86SNeel Natu 	return (ioctl(ctx->fd, VM_IOAPIC_DEASSERT_IRQ, &ioapic_irq));
465565bbb86SNeel Natu }
466565bbb86SNeel Natu 
467565bbb86SNeel Natu int
468ac7304a7SNeel Natu vm_ioapic_pulse_irq(struct vmctx *ctx, int irq)
469ac7304a7SNeel Natu {
470ac7304a7SNeel Natu 	struct vm_ioapic_irq ioapic_irq;
471ac7304a7SNeel Natu 
472ac7304a7SNeel Natu 	bzero(&ioapic_irq, sizeof(struct vm_ioapic_irq));
473ac7304a7SNeel Natu 	ioapic_irq.irq = irq;
474ac7304a7SNeel Natu 
475ac7304a7SNeel Natu 	return (ioctl(ctx->fd, VM_IOAPIC_PULSE_IRQ, &ioapic_irq));
476ac7304a7SNeel Natu }
477ac7304a7SNeel Natu 
478ac7304a7SNeel Natu int
4793cbf3585SJohn Baldwin vm_ioapic_pincount(struct vmctx *ctx, int *pincount)
4803cbf3585SJohn Baldwin {
4813cbf3585SJohn Baldwin 
4823cbf3585SJohn Baldwin 	return (ioctl(ctx->fd, VM_IOAPIC_PINCOUNT, pincount));
4833cbf3585SJohn Baldwin }
4843cbf3585SJohn Baldwin 
4853cbf3585SJohn Baldwin int
486762fd208STycho Nightingale vm_isa_assert_irq(struct vmctx *ctx, int atpic_irq, int ioapic_irq)
487762fd208STycho Nightingale {
488762fd208STycho Nightingale 	struct vm_isa_irq isa_irq;
489762fd208STycho Nightingale 
490762fd208STycho Nightingale 	bzero(&isa_irq, sizeof(struct vm_isa_irq));
491762fd208STycho Nightingale 	isa_irq.atpic_irq = atpic_irq;
492762fd208STycho Nightingale 	isa_irq.ioapic_irq = ioapic_irq;
493762fd208STycho Nightingale 
494762fd208STycho Nightingale 	return (ioctl(ctx->fd, VM_ISA_ASSERT_IRQ, &isa_irq));
495762fd208STycho Nightingale }
496762fd208STycho Nightingale 
497762fd208STycho Nightingale int
498762fd208STycho Nightingale vm_isa_deassert_irq(struct vmctx *ctx, int atpic_irq, int ioapic_irq)
499762fd208STycho Nightingale {
500762fd208STycho Nightingale 	struct vm_isa_irq isa_irq;
501762fd208STycho Nightingale 
502762fd208STycho Nightingale 	bzero(&isa_irq, sizeof(struct vm_isa_irq));
503762fd208STycho Nightingale 	isa_irq.atpic_irq = atpic_irq;
504762fd208STycho Nightingale 	isa_irq.ioapic_irq = ioapic_irq;
505762fd208STycho Nightingale 
506762fd208STycho Nightingale 	return (ioctl(ctx->fd, VM_ISA_DEASSERT_IRQ, &isa_irq));
507762fd208STycho Nightingale }
508762fd208STycho Nightingale 
509762fd208STycho Nightingale int
510762fd208STycho Nightingale vm_isa_pulse_irq(struct vmctx *ctx, int atpic_irq, int ioapic_irq)
511762fd208STycho Nightingale {
512762fd208STycho Nightingale 	struct vm_isa_irq isa_irq;
513b3e9732aSJohn Baldwin 
514762fd208STycho Nightingale 	bzero(&isa_irq, sizeof(struct vm_isa_irq));
515762fd208STycho Nightingale 	isa_irq.atpic_irq = atpic_irq;
516762fd208STycho Nightingale 	isa_irq.ioapic_irq = ioapic_irq;
517762fd208STycho Nightingale 
518762fd208STycho Nightingale 	return (ioctl(ctx->fd, VM_ISA_PULSE_IRQ, &isa_irq));
519762fd208STycho Nightingale }
520762fd208STycho Nightingale 
521762fd208STycho Nightingale int
522b3e9732aSJohn Baldwin vm_isa_set_irq_trigger(struct vmctx *ctx, int atpic_irq,
523b3e9732aSJohn Baldwin     enum vm_intr_trigger trigger)
524b3e9732aSJohn Baldwin {
525b3e9732aSJohn Baldwin 	struct vm_isa_irq_trigger isa_irq_trigger;
526b3e9732aSJohn Baldwin 
527b3e9732aSJohn Baldwin 	bzero(&isa_irq_trigger, sizeof(struct vm_isa_irq_trigger));
528b3e9732aSJohn Baldwin 	isa_irq_trigger.atpic_irq = atpic_irq;
529b3e9732aSJohn Baldwin 	isa_irq_trigger.trigger = trigger;
530b3e9732aSJohn Baldwin 
531b3e9732aSJohn Baldwin 	return (ioctl(ctx->fd, VM_ISA_SET_IRQ_TRIGGER, &isa_irq_trigger));
532b3e9732aSJohn Baldwin }
533b3e9732aSJohn Baldwin 
534b3e9732aSJohn Baldwin int
535366f6083SPeter Grehan vm_inject_nmi(struct vmctx *ctx, int vcpu)
536366f6083SPeter Grehan {
537366f6083SPeter Grehan 	struct vm_nmi vmnmi;
538366f6083SPeter Grehan 
539366f6083SPeter Grehan 	bzero(&vmnmi, sizeof(vmnmi));
540366f6083SPeter Grehan 	vmnmi.cpuid = vcpu;
541366f6083SPeter Grehan 
542366f6083SPeter Grehan 	return (ioctl(ctx->fd, VM_INJECT_NMI, &vmnmi));
543366f6083SPeter Grehan }
544366f6083SPeter Grehan 
545366f6083SPeter Grehan static struct {
546366f6083SPeter Grehan 	const char	*name;
547366f6083SPeter Grehan 	int		type;
548366f6083SPeter Grehan } capstrmap[] = {
549366f6083SPeter Grehan 	{ "hlt_exit",		VM_CAP_HALT_EXIT },
550366f6083SPeter Grehan 	{ "mtrap_exit",		VM_CAP_MTRAP_EXIT },
551366f6083SPeter Grehan 	{ "pause_exit",		VM_CAP_PAUSE_EXIT },
552366f6083SPeter Grehan 	{ "unrestricted_guest",	VM_CAP_UNRESTRICTED_GUEST },
55349cc03daSNeel Natu 	{ "enable_invpcid",	VM_CAP_ENABLE_INVPCID },
554366f6083SPeter Grehan 	{ 0 }
555366f6083SPeter Grehan };
556366f6083SPeter Grehan 
55713f4cf6cSNeel Natu int
55813f4cf6cSNeel Natu vm_capability_name2type(const char *capname)
55913f4cf6cSNeel Natu {
56013f4cf6cSNeel Natu 	int i;
56113f4cf6cSNeel Natu 
562366f6083SPeter Grehan 	for (i = 0; capstrmap[i].name != NULL && capname != NULL; i++) {
563366f6083SPeter Grehan 		if (strcmp(capstrmap[i].name, capname) == 0)
564366f6083SPeter Grehan 			return (capstrmap[i].type);
565366f6083SPeter Grehan 	}
566366f6083SPeter Grehan 
567366f6083SPeter Grehan 	return (-1);
568366f6083SPeter Grehan }
569366f6083SPeter Grehan 
57013f4cf6cSNeel Natu const char *
57113f4cf6cSNeel Natu vm_capability_type2name(int type)
57213f4cf6cSNeel Natu {
57313f4cf6cSNeel Natu 	int i;
57413f4cf6cSNeel Natu 
57513f4cf6cSNeel Natu 	for (i = 0; capstrmap[i].name != NULL; i++) {
57613f4cf6cSNeel Natu 		if (capstrmap[i].type == type)
57713f4cf6cSNeel Natu 			return (capstrmap[i].name);
57813f4cf6cSNeel Natu 	}
57913f4cf6cSNeel Natu 
58013f4cf6cSNeel Natu 	return (NULL);
58113f4cf6cSNeel Natu }
58213f4cf6cSNeel Natu 
583366f6083SPeter Grehan int
584366f6083SPeter Grehan vm_get_capability(struct vmctx *ctx, int vcpu, enum vm_cap_type cap,
585366f6083SPeter Grehan 		  int *retval)
586366f6083SPeter Grehan {
587366f6083SPeter Grehan 	int error;
588366f6083SPeter Grehan 	struct vm_capability vmcap;
589366f6083SPeter Grehan 
590366f6083SPeter Grehan 	bzero(&vmcap, sizeof(vmcap));
591366f6083SPeter Grehan 	vmcap.cpuid = vcpu;
592366f6083SPeter Grehan 	vmcap.captype = cap;
593366f6083SPeter Grehan 
594366f6083SPeter Grehan 	error = ioctl(ctx->fd, VM_GET_CAPABILITY, &vmcap);
595366f6083SPeter Grehan 	*retval = vmcap.capval;
596366f6083SPeter Grehan 	return (error);
597366f6083SPeter Grehan }
598366f6083SPeter Grehan 
599366f6083SPeter Grehan int
600366f6083SPeter Grehan vm_set_capability(struct vmctx *ctx, int vcpu, enum vm_cap_type cap, int val)
601366f6083SPeter Grehan {
602366f6083SPeter Grehan 	struct vm_capability vmcap;
603366f6083SPeter Grehan 
604366f6083SPeter Grehan 	bzero(&vmcap, sizeof(vmcap));
605366f6083SPeter Grehan 	vmcap.cpuid = vcpu;
606366f6083SPeter Grehan 	vmcap.captype = cap;
607366f6083SPeter Grehan 	vmcap.capval = val;
608366f6083SPeter Grehan 
609366f6083SPeter Grehan 	return (ioctl(ctx->fd, VM_SET_CAPABILITY, &vmcap));
610366f6083SPeter Grehan }
611366f6083SPeter Grehan 
612366f6083SPeter Grehan int
613366f6083SPeter Grehan vm_assign_pptdev(struct vmctx *ctx, int bus, int slot, int func)
614366f6083SPeter Grehan {
615366f6083SPeter Grehan 	struct vm_pptdev pptdev;
616366f6083SPeter Grehan 
617366f6083SPeter Grehan 	bzero(&pptdev, sizeof(pptdev));
618366f6083SPeter Grehan 	pptdev.bus = bus;
619366f6083SPeter Grehan 	pptdev.slot = slot;
620366f6083SPeter Grehan 	pptdev.func = func;
621366f6083SPeter Grehan 
622366f6083SPeter Grehan 	return (ioctl(ctx->fd, VM_BIND_PPTDEV, &pptdev));
623366f6083SPeter Grehan }
624366f6083SPeter Grehan 
625366f6083SPeter Grehan int
626366f6083SPeter Grehan vm_unassign_pptdev(struct vmctx *ctx, int bus, int slot, int func)
627366f6083SPeter Grehan {
628366f6083SPeter Grehan 	struct vm_pptdev pptdev;
629366f6083SPeter Grehan 
630366f6083SPeter Grehan 	bzero(&pptdev, sizeof(pptdev));
631366f6083SPeter Grehan 	pptdev.bus = bus;
632366f6083SPeter Grehan 	pptdev.slot = slot;
633366f6083SPeter Grehan 	pptdev.func = func;
634366f6083SPeter Grehan 
635366f6083SPeter Grehan 	return (ioctl(ctx->fd, VM_UNBIND_PPTDEV, &pptdev));
636366f6083SPeter Grehan }
637366f6083SPeter Grehan 
638366f6083SPeter Grehan int
639366f6083SPeter Grehan vm_map_pptdev_mmio(struct vmctx *ctx, int bus, int slot, int func,
640366f6083SPeter Grehan 		   vm_paddr_t gpa, size_t len, vm_paddr_t hpa)
641366f6083SPeter Grehan {
642366f6083SPeter Grehan 	struct vm_pptdev_mmio pptmmio;
643366f6083SPeter Grehan 
644366f6083SPeter Grehan 	bzero(&pptmmio, sizeof(pptmmio));
645366f6083SPeter Grehan 	pptmmio.bus = bus;
646366f6083SPeter Grehan 	pptmmio.slot = slot;
647366f6083SPeter Grehan 	pptmmio.func = func;
648366f6083SPeter Grehan 	pptmmio.gpa = gpa;
649366f6083SPeter Grehan 	pptmmio.len = len;
650366f6083SPeter Grehan 	pptmmio.hpa = hpa;
651366f6083SPeter Grehan 
652366f6083SPeter Grehan 	return (ioctl(ctx->fd, VM_MAP_PPTDEV_MMIO, &pptmmio));
653366f6083SPeter Grehan }
654366f6083SPeter Grehan 
655366f6083SPeter Grehan int
65655888cfaSNeel Natu vm_setup_pptdev_msi(struct vmctx *ctx, int vcpu, int bus, int slot, int func,
6574f8be175SNeel Natu     uint64_t addr, uint64_t msg, int numvec)
658366f6083SPeter Grehan {
659366f6083SPeter Grehan 	struct vm_pptdev_msi pptmsi;
660366f6083SPeter Grehan 
661366f6083SPeter Grehan 	bzero(&pptmsi, sizeof(pptmsi));
662366f6083SPeter Grehan 	pptmsi.vcpu = vcpu;
663366f6083SPeter Grehan 	pptmsi.bus = bus;
664366f6083SPeter Grehan 	pptmsi.slot = slot;
665366f6083SPeter Grehan 	pptmsi.func = func;
6664f8be175SNeel Natu 	pptmsi.msg = msg;
6674f8be175SNeel Natu 	pptmsi.addr = addr;
668366f6083SPeter Grehan 	pptmsi.numvec = numvec;
669366f6083SPeter Grehan 
670366f6083SPeter Grehan 	return (ioctl(ctx->fd, VM_PPTDEV_MSI, &pptmsi));
671366f6083SPeter Grehan }
672366f6083SPeter Grehan 
673cd942e0fSPeter Grehan int
67455888cfaSNeel Natu vm_setup_pptdev_msix(struct vmctx *ctx, int vcpu, int bus, int slot, int func,
6754f8be175SNeel Natu     int idx, uint64_t addr, uint64_t msg, uint32_t vector_control)
676cd942e0fSPeter Grehan {
677cd942e0fSPeter Grehan 	struct vm_pptdev_msix pptmsix;
678cd942e0fSPeter Grehan 
679cd942e0fSPeter Grehan 	bzero(&pptmsix, sizeof(pptmsix));
680cd942e0fSPeter Grehan 	pptmsix.vcpu = vcpu;
681cd942e0fSPeter Grehan 	pptmsix.bus = bus;
682cd942e0fSPeter Grehan 	pptmsix.slot = slot;
683cd942e0fSPeter Grehan 	pptmsix.func = func;
684cd942e0fSPeter Grehan 	pptmsix.idx = idx;
685cd942e0fSPeter Grehan 	pptmsix.msg = msg;
686cd942e0fSPeter Grehan 	pptmsix.addr = addr;
687cd942e0fSPeter Grehan 	pptmsix.vector_control = vector_control;
688cd942e0fSPeter Grehan 
689cd942e0fSPeter Grehan 	return ioctl(ctx->fd, VM_PPTDEV_MSIX, &pptmsix);
690cd942e0fSPeter Grehan }
691cd942e0fSPeter Grehan 
692366f6083SPeter Grehan uint64_t *
693366f6083SPeter Grehan vm_get_stats(struct vmctx *ctx, int vcpu, struct timeval *ret_tv,
694366f6083SPeter Grehan 	     int *ret_entries)
695366f6083SPeter Grehan {
696366f6083SPeter Grehan 	int error;
697366f6083SPeter Grehan 
698366f6083SPeter Grehan 	static struct vm_stats vmstats;
699366f6083SPeter Grehan 
700366f6083SPeter Grehan 	vmstats.cpuid = vcpu;
701366f6083SPeter Grehan 
702366f6083SPeter Grehan 	error = ioctl(ctx->fd, VM_STATS, &vmstats);
703366f6083SPeter Grehan 	if (error == 0) {
704366f6083SPeter Grehan 		if (ret_entries)
705366f6083SPeter Grehan 			*ret_entries = vmstats.num_entries;
706366f6083SPeter Grehan 		if (ret_tv)
707366f6083SPeter Grehan 			*ret_tv = vmstats.tv;
708366f6083SPeter Grehan 		return (vmstats.statbuf);
709366f6083SPeter Grehan 	} else
710366f6083SPeter Grehan 		return (NULL);
711366f6083SPeter Grehan }
712366f6083SPeter Grehan 
713366f6083SPeter Grehan const char *
714366f6083SPeter Grehan vm_get_stat_desc(struct vmctx *ctx, int index)
715366f6083SPeter Grehan {
716366f6083SPeter Grehan 	static struct vm_stat_desc statdesc;
717366f6083SPeter Grehan 
718366f6083SPeter Grehan 	statdesc.index = index;
719366f6083SPeter Grehan 	if (ioctl(ctx->fd, VM_STAT_DESC, &statdesc) == 0)
720366f6083SPeter Grehan 		return (statdesc.desc);
721366f6083SPeter Grehan 	else
722366f6083SPeter Grehan 		return (NULL);
723366f6083SPeter Grehan }
724366f6083SPeter Grehan 
725e9027382SNeel Natu int
726e9027382SNeel Natu vm_get_x2apic_state(struct vmctx *ctx, int vcpu, enum x2apic_state *state)
727e9027382SNeel Natu {
728e9027382SNeel Natu 	int error;
729e9027382SNeel Natu 	struct vm_x2apic x2apic;
730e9027382SNeel Natu 
731e9027382SNeel Natu 	bzero(&x2apic, sizeof(x2apic));
732e9027382SNeel Natu 	x2apic.cpuid = vcpu;
733e9027382SNeel Natu 
734e9027382SNeel Natu 	error = ioctl(ctx->fd, VM_GET_X2APIC_STATE, &x2apic);
735e9027382SNeel Natu 	*state = x2apic.state;
736e9027382SNeel Natu 	return (error);
737e9027382SNeel Natu }
738e9027382SNeel Natu 
739e9027382SNeel Natu int
740e9027382SNeel Natu vm_set_x2apic_state(struct vmctx *ctx, int vcpu, enum x2apic_state state)
741e9027382SNeel Natu {
742e9027382SNeel Natu 	int error;
743e9027382SNeel Natu 	struct vm_x2apic x2apic;
744e9027382SNeel Natu 
745e9027382SNeel Natu 	bzero(&x2apic, sizeof(x2apic));
746e9027382SNeel Natu 	x2apic.cpuid = vcpu;
747e9027382SNeel Natu 	x2apic.state = state;
748e9027382SNeel Natu 
749e9027382SNeel Natu 	error = ioctl(ctx->fd, VM_SET_X2APIC_STATE, &x2apic);
750e9027382SNeel Natu 
751e9027382SNeel Natu 	return (error);
752e9027382SNeel Natu }
753e9027382SNeel Natu 
754366f6083SPeter Grehan /*
755366f6083SPeter Grehan  * From Intel Vol 3a:
756366f6083SPeter Grehan  * Table 9-1. IA-32 Processor States Following Power-up, Reset or INIT
757366f6083SPeter Grehan  */
758366f6083SPeter Grehan int
759366f6083SPeter Grehan vcpu_reset(struct vmctx *vmctx, int vcpu)
760366f6083SPeter Grehan {
761366f6083SPeter Grehan 	int error;
762366f6083SPeter Grehan 	uint64_t rflags, rip, cr0, cr4, zero, desc_base, rdx;
763366f6083SPeter Grehan 	uint32_t desc_access, desc_limit;
764366f6083SPeter Grehan 	uint16_t sel;
765366f6083SPeter Grehan 
766366f6083SPeter Grehan 	zero = 0;
767366f6083SPeter Grehan 
768366f6083SPeter Grehan 	rflags = 0x2;
769366f6083SPeter Grehan 	error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RFLAGS, rflags);
770366f6083SPeter Grehan 	if (error)
771366f6083SPeter Grehan 		goto done;
772366f6083SPeter Grehan 
773366f6083SPeter Grehan 	rip = 0xfff0;
774366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RIP, rip)) != 0)
775366f6083SPeter Grehan 		goto done;
776366f6083SPeter Grehan 
777366f6083SPeter Grehan 	cr0 = CR0_NE;
778366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_CR0, cr0)) != 0)
779366f6083SPeter Grehan 		goto done;
780366f6083SPeter Grehan 
781366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_CR3, zero)) != 0)
782366f6083SPeter Grehan 		goto done;
783366f6083SPeter Grehan 
78432c96bc8SNeel Natu 	cr4 = 0;
785366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_CR4, cr4)) != 0)
786366f6083SPeter Grehan 		goto done;
787366f6083SPeter Grehan 
788366f6083SPeter Grehan 	/*
789366f6083SPeter Grehan 	 * CS: present, r/w, accessed, 16-bit, byte granularity, usable
790366f6083SPeter Grehan 	 */
791366f6083SPeter Grehan 	desc_base = 0xffff0000;
792366f6083SPeter Grehan 	desc_limit = 0xffff;
793366f6083SPeter Grehan 	desc_access = 0x0093;
794366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_CS,
795366f6083SPeter Grehan 			    desc_base, desc_limit, desc_access);
796366f6083SPeter Grehan 	if (error)
797366f6083SPeter Grehan 		goto done;
798366f6083SPeter Grehan 
799366f6083SPeter Grehan 	sel = 0xf000;
800366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_CS, sel)) != 0)
801366f6083SPeter Grehan 		goto done;
802366f6083SPeter Grehan 
803366f6083SPeter Grehan 	/*
804366f6083SPeter Grehan 	 * SS,DS,ES,FS,GS: present, r/w, accessed, 16-bit, byte granularity
805366f6083SPeter Grehan 	 */
806366f6083SPeter Grehan 	desc_base = 0;
807366f6083SPeter Grehan 	desc_limit = 0xffff;
808366f6083SPeter Grehan 	desc_access = 0x0093;
809366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_SS,
810366f6083SPeter Grehan 			    desc_base, desc_limit, desc_access);
811366f6083SPeter Grehan 	if (error)
812366f6083SPeter Grehan 		goto done;
813366f6083SPeter Grehan 
814366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_DS,
815366f6083SPeter Grehan 			    desc_base, desc_limit, desc_access);
816366f6083SPeter Grehan 	if (error)
817366f6083SPeter Grehan 		goto done;
818366f6083SPeter Grehan 
819366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_ES,
820366f6083SPeter Grehan 			    desc_base, desc_limit, desc_access);
821366f6083SPeter Grehan 	if (error)
822366f6083SPeter Grehan 		goto done;
823366f6083SPeter Grehan 
824366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_FS,
825366f6083SPeter Grehan 			    desc_base, desc_limit, desc_access);
826366f6083SPeter Grehan 	if (error)
827366f6083SPeter Grehan 		goto done;
828366f6083SPeter Grehan 
829366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_GS,
830366f6083SPeter Grehan 			    desc_base, desc_limit, desc_access);
831366f6083SPeter Grehan 	if (error)
832366f6083SPeter Grehan 		goto done;
833366f6083SPeter Grehan 
834366f6083SPeter Grehan 	sel = 0;
835366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_SS, sel)) != 0)
836366f6083SPeter Grehan 		goto done;
837366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_DS, sel)) != 0)
838366f6083SPeter Grehan 		goto done;
839366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_ES, sel)) != 0)
840366f6083SPeter Grehan 		goto done;
841366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_FS, sel)) != 0)
842366f6083SPeter Grehan 		goto done;
843366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_GS, sel)) != 0)
844366f6083SPeter Grehan 		goto done;
845366f6083SPeter Grehan 
846366f6083SPeter Grehan 	/* General purpose registers */
847366f6083SPeter Grehan 	rdx = 0xf00;
848366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RAX, zero)) != 0)
849366f6083SPeter Grehan 		goto done;
850366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RBX, zero)) != 0)
851366f6083SPeter Grehan 		goto done;
852366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RCX, zero)) != 0)
853366f6083SPeter Grehan 		goto done;
854366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RDX, rdx)) != 0)
855366f6083SPeter Grehan 		goto done;
856366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RSI, zero)) != 0)
857366f6083SPeter Grehan 		goto done;
858366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RDI, zero)) != 0)
859366f6083SPeter Grehan 		goto done;
860366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RBP, zero)) != 0)
861366f6083SPeter Grehan 		goto done;
862366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RSP, zero)) != 0)
863366f6083SPeter Grehan 		goto done;
864366f6083SPeter Grehan 
865366f6083SPeter Grehan 	/* GDTR, IDTR */
866366f6083SPeter Grehan 	desc_base = 0;
867366f6083SPeter Grehan 	desc_limit = 0xffff;
868366f6083SPeter Grehan 	desc_access = 0;
869366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_GDTR,
870366f6083SPeter Grehan 			    desc_base, desc_limit, desc_access);
871366f6083SPeter Grehan 	if (error != 0)
872366f6083SPeter Grehan 		goto done;
873366f6083SPeter Grehan 
874366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_IDTR,
875366f6083SPeter Grehan 			    desc_base, desc_limit, desc_access);
876366f6083SPeter Grehan 	if (error != 0)
877366f6083SPeter Grehan 		goto done;
878366f6083SPeter Grehan 
879366f6083SPeter Grehan 	/* TR */
880366f6083SPeter Grehan 	desc_base = 0;
881366f6083SPeter Grehan 	desc_limit = 0xffff;
882366f6083SPeter Grehan 	desc_access = 0x0000008b;
883366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_TR, 0, 0, desc_access);
884366f6083SPeter Grehan 	if (error)
885366f6083SPeter Grehan 		goto done;
886366f6083SPeter Grehan 
887366f6083SPeter Grehan 	sel = 0;
888366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_TR, sel)) != 0)
889366f6083SPeter Grehan 		goto done;
890366f6083SPeter Grehan 
891366f6083SPeter Grehan 	/* LDTR */
892366f6083SPeter Grehan 	desc_base = 0;
893366f6083SPeter Grehan 	desc_limit = 0xffff;
894366f6083SPeter Grehan 	desc_access = 0x00000082;
895366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_LDTR, desc_base,
896366f6083SPeter Grehan 			    desc_limit, desc_access);
897366f6083SPeter Grehan 	if (error)
898366f6083SPeter Grehan 		goto done;
899366f6083SPeter Grehan 
900366f6083SPeter Grehan 	sel = 0;
901366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_LDTR, 0)) != 0)
902366f6083SPeter Grehan 		goto done;
903366f6083SPeter Grehan 
904366f6083SPeter Grehan 	/* XXX cr2, debug registers */
905366f6083SPeter Grehan 
906366f6083SPeter Grehan 	error = 0;
907366f6083SPeter Grehan done:
908366f6083SPeter Grehan 	return (error);
909366f6083SPeter Grehan }
910318224bbSNeel Natu 
911318224bbSNeel Natu int
912318224bbSNeel Natu vm_get_gpa_pmap(struct vmctx *ctx, uint64_t gpa, uint64_t *pte, int *num)
913318224bbSNeel Natu {
914318224bbSNeel Natu 	int error, i;
915318224bbSNeel Natu 	struct vm_gpa_pte gpapte;
916318224bbSNeel Natu 
917318224bbSNeel Natu 	bzero(&gpapte, sizeof(gpapte));
918318224bbSNeel Natu 	gpapte.gpa = gpa;
919318224bbSNeel Natu 
920318224bbSNeel Natu 	error = ioctl(ctx->fd, VM_GET_GPA_PMAP, &gpapte);
921318224bbSNeel Natu 
922318224bbSNeel Natu 	if (error == 0) {
923318224bbSNeel Natu 		*num = gpapte.ptenum;
924318224bbSNeel Natu 		for (i = 0; i < gpapte.ptenum; i++)
925318224bbSNeel Natu 			pte[i] = gpapte.pte[i];
926318224bbSNeel Natu 	}
927318224bbSNeel Natu 
928318224bbSNeel Natu 	return (error);
929318224bbSNeel Natu }
93008e3ff32SNeel Natu 
93108e3ff32SNeel Natu int
93208e3ff32SNeel Natu vm_get_hpet_capabilities(struct vmctx *ctx, uint32_t *capabilities)
93308e3ff32SNeel Natu {
93408e3ff32SNeel Natu 	int error;
93508e3ff32SNeel Natu 	struct vm_hpet_cap cap;
93608e3ff32SNeel Natu 
93708e3ff32SNeel Natu 	bzero(&cap, sizeof(struct vm_hpet_cap));
93808e3ff32SNeel Natu 	error = ioctl(ctx->fd, VM_GET_HPET_CAPABILITIES, &cap);
93908e3ff32SNeel Natu 	if (capabilities != NULL)
94008e3ff32SNeel Natu 		*capabilities = cap.capabilities;
94108e3ff32SNeel Natu 	return (error);
94208e3ff32SNeel Natu }
943da11f4aaSNeel Natu 
944da11f4aaSNeel Natu static int
9456303b65dSNeel Natu gla2gpa(struct vmctx *ctx, int vcpu, struct vm_guest_paging *paging,
946da11f4aaSNeel Natu     uint64_t gla, int prot, int *fault, uint64_t *gpa)
947da11f4aaSNeel Natu {
948da11f4aaSNeel Natu 	struct vm_gla2gpa gg;
949da11f4aaSNeel Natu 	int error;
950da11f4aaSNeel Natu 
951da11f4aaSNeel Natu 	bzero(&gg, sizeof(struct vm_gla2gpa));
952da11f4aaSNeel Natu 	gg.vcpuid = vcpu;
953da11f4aaSNeel Natu 	gg.prot = prot;
954da11f4aaSNeel Natu 	gg.gla = gla;
955da11f4aaSNeel Natu 	gg.paging = *paging;
956da11f4aaSNeel Natu 
957da11f4aaSNeel Natu 	error = ioctl(ctx->fd, VM_GLA2GPA, &gg);
958da11f4aaSNeel Natu 	if (error == 0) {
959da11f4aaSNeel Natu 		*fault = gg.fault;
960da11f4aaSNeel Natu 		*gpa = gg.gpa;
961da11f4aaSNeel Natu 	}
962da11f4aaSNeel Natu 	return (error);
963da11f4aaSNeel Natu }
964da11f4aaSNeel Natu 
965da11f4aaSNeel Natu #ifndef min
966da11f4aaSNeel Natu #define	min(a,b)	(((a) < (b)) ? (a) : (b))
967da11f4aaSNeel Natu #endif
968da11f4aaSNeel Natu 
969da11f4aaSNeel Natu int
9706303b65dSNeel Natu vm_gla2gpa(struct vmctx *ctx, int vcpu, struct vm_guest_paging *paging,
9716303b65dSNeel Natu     uint64_t gla, size_t len, int prot, struct iovec *iov, int iovcnt)
972da11f4aaSNeel Natu {
973da11f4aaSNeel Natu 	uint64_t gpa;
9746303b65dSNeel Natu 	int error, fault, i, n, off;
9756303b65dSNeel Natu 
9766303b65dSNeel Natu 	for (i = 0; i < iovcnt; i++) {
9776303b65dSNeel Natu 		iov[i].iov_base = 0;
9786303b65dSNeel Natu 		iov[i].iov_len = 0;
9796303b65dSNeel Natu 	}
9806303b65dSNeel Natu 
9816303b65dSNeel Natu 	while (len) {
9826303b65dSNeel Natu 		assert(iovcnt > 0);
9836303b65dSNeel Natu 		error = gla2gpa(ctx, vcpu, paging, gla, prot, &fault, &gpa);
9846303b65dSNeel Natu 		if (error)
9856303b65dSNeel Natu 			return (-1);
9866303b65dSNeel Natu 		if (fault)
9876303b65dSNeel Natu 			return (1);
9886303b65dSNeel Natu 
9896303b65dSNeel Natu 		off = gpa & PAGE_MASK;
9906303b65dSNeel Natu 		n = min(len, PAGE_SIZE - off);
9916303b65dSNeel Natu 
9926303b65dSNeel Natu 		iov->iov_base = (void *)gpa;
9936303b65dSNeel Natu 		iov->iov_len = n;
9946303b65dSNeel Natu 		iov++;
9956303b65dSNeel Natu 		iovcnt--;
9966303b65dSNeel Natu 
9976303b65dSNeel Natu 		gla += n;
9986303b65dSNeel Natu 		len -= n;
9996303b65dSNeel Natu 	}
10006303b65dSNeel Natu 	return (0);
10016303b65dSNeel Natu }
10026303b65dSNeel Natu 
10036303b65dSNeel Natu void
10046303b65dSNeel Natu vm_copyin(struct vmctx *ctx, int vcpu, struct iovec *iov, void *vp, size_t len)
10056303b65dSNeel Natu {
10066303b65dSNeel Natu 	const char *src;
10076303b65dSNeel Natu 	char *dst;
10086303b65dSNeel Natu 	uint64_t gpa;
10096303b65dSNeel Natu 	size_t n;
1010da11f4aaSNeel Natu 
1011da11f4aaSNeel Natu 	dst = vp;
1012da11f4aaSNeel Natu 	while (len) {
10136303b65dSNeel Natu 		assert(iov->iov_len);
10146303b65dSNeel Natu 		gpa = (uint64_t)iov->iov_base;
10156303b65dSNeel Natu 		n = min(len, iov->iov_len);
1016da11f4aaSNeel Natu 		src = vm_map_gpa(ctx, gpa, n);
1017da11f4aaSNeel Natu 		bcopy(src, dst, n);
1018da11f4aaSNeel Natu 
10196303b65dSNeel Natu 		iov++;
1020da11f4aaSNeel Natu 		dst += n;
1021da11f4aaSNeel Natu 		len -= n;
1022da11f4aaSNeel Natu 	}
1023da11f4aaSNeel Natu }
1024da11f4aaSNeel Natu 
10256303b65dSNeel Natu void
10266303b65dSNeel Natu vm_copyout(struct vmctx *ctx, int vcpu, const void *vp, struct iovec *iov,
10276303b65dSNeel Natu     size_t len)
1028da11f4aaSNeel Natu {
1029da11f4aaSNeel Natu 	const char *src;
10306303b65dSNeel Natu 	char *dst;
10316303b65dSNeel Natu 	uint64_t gpa;
10326303b65dSNeel Natu 	size_t n;
1033da11f4aaSNeel Natu 
1034da11f4aaSNeel Natu 	src = vp;
1035da11f4aaSNeel Natu 	while (len) {
10366303b65dSNeel Natu 		assert(iov->iov_len);
10376303b65dSNeel Natu 		gpa = (uint64_t)iov->iov_base;
10386303b65dSNeel Natu 		n = min(len, iov->iov_len);
1039da11f4aaSNeel Natu 		dst = vm_map_gpa(ctx, gpa, n);
1040da11f4aaSNeel Natu 		bcopy(src, dst, n);
1041da11f4aaSNeel Natu 
10426303b65dSNeel Natu 		iov++;
1043da11f4aaSNeel Natu 		src += n;
1044da11f4aaSNeel Natu 		len -= n;
1045da11f4aaSNeel Natu 	}
1046da11f4aaSNeel Natu }
1047*95ebc360SNeel Natu 
1048*95ebc360SNeel Natu static int
1049*95ebc360SNeel Natu vm_get_cpus(struct vmctx *ctx, int which, cpuset_t *cpus)
1050*95ebc360SNeel Natu {
1051*95ebc360SNeel Natu 	struct vm_cpuset vm_cpuset;
1052*95ebc360SNeel Natu 	int error;
1053*95ebc360SNeel Natu 
1054*95ebc360SNeel Natu 	bzero(&vm_cpuset, sizeof(struct vm_cpuset));
1055*95ebc360SNeel Natu 	vm_cpuset.which = which;
1056*95ebc360SNeel Natu 	vm_cpuset.cpusetsize = sizeof(cpuset_t);
1057*95ebc360SNeel Natu 	vm_cpuset.cpus = cpus;
1058*95ebc360SNeel Natu 
1059*95ebc360SNeel Natu 	error = ioctl(ctx->fd, VM_GET_CPUS, &vm_cpuset);
1060*95ebc360SNeel Natu 	return (error);
1061*95ebc360SNeel Natu }
1062*95ebc360SNeel Natu 
1063*95ebc360SNeel Natu int
1064*95ebc360SNeel Natu vm_active_cpus(struct vmctx *ctx, cpuset_t *cpus)
1065*95ebc360SNeel Natu {
1066*95ebc360SNeel Natu 
1067*95ebc360SNeel Natu 	return (vm_get_cpus(ctx, VM_ACTIVE_CPUS, cpus));
1068*95ebc360SNeel Natu }
1069*95ebc360SNeel Natu 
1070*95ebc360SNeel Natu int
1071*95ebc360SNeel Natu vm_suspended_cpus(struct vmctx *ctx, cpuset_t *cpus)
1072*95ebc360SNeel Natu {
1073*95ebc360SNeel Natu 
1074*95ebc360SNeel Natu 	return (vm_get_cpus(ctx, VM_SUSPENDED_CPUS, cpus));
1075*95ebc360SNeel Natu }
1076*95ebc360SNeel Natu 
1077*95ebc360SNeel Natu int
1078*95ebc360SNeel Natu vm_activate_cpu(struct vmctx *ctx, int vcpu)
1079*95ebc360SNeel Natu {
1080*95ebc360SNeel Natu 	struct vm_activate_cpu ac;
1081*95ebc360SNeel Natu 	int error;
1082*95ebc360SNeel Natu 
1083*95ebc360SNeel Natu 	bzero(&ac, sizeof(struct vm_activate_cpu));
1084*95ebc360SNeel Natu 	ac.vcpuid = vcpu;
1085*95ebc360SNeel Natu 	error = ioctl(ctx->fd, VM_ACTIVATE_CPU, &ac);
1086*95ebc360SNeel Natu 	return (error);
1087*95ebc360SNeel Natu }
1088