xref: /freebsd/lib/libvmmapi/vmmapi.c (revision 08e3ff329a7e7c53b390d54ec86a4f34243ff7b3)
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 
32366f6083SPeter Grehan #include <sys/types.h>
33366f6083SPeter Grehan #include <sys/sysctl.h>
34366f6083SPeter Grehan #include <sys/ioctl.h>
35366f6083SPeter Grehan #include <sys/mman.h>
36366f6083SPeter Grehan 
37366f6083SPeter Grehan #include <machine/specialreg.h>
38366f6083SPeter Grehan 
39366f6083SPeter Grehan #include <stdio.h>
40366f6083SPeter Grehan #include <stdlib.h>
41366f6083SPeter Grehan #include <assert.h>
42366f6083SPeter Grehan #include <string.h>
43366f6083SPeter Grehan #include <fcntl.h>
44366f6083SPeter Grehan #include <unistd.h>
45366f6083SPeter Grehan 
46200758f1SNeel Natu #include <libutil.h>
47200758f1SNeel Natu 
48366f6083SPeter Grehan #include <machine/vmm.h>
49366f6083SPeter Grehan #include <machine/vmm_dev.h>
50366f6083SPeter Grehan 
51366f6083SPeter Grehan #include "vmmapi.h"
52366f6083SPeter Grehan 
53200758f1SNeel Natu #define	MB	(1024 * 1024UL)
54b060ba50SNeel Natu #define	GB	(1024 * 1024 * 1024UL)
55b060ba50SNeel Natu 
56366f6083SPeter Grehan struct vmctx {
57366f6083SPeter Grehan 	int	fd;
58b060ba50SNeel Natu 	uint32_t lowmem_limit;
59b060ba50SNeel Natu 	enum vm_mmap_style vms;
60b060ba50SNeel Natu 	size_t	lowmem;
61b060ba50SNeel Natu 	char	*lowmem_addr;
62b060ba50SNeel Natu 	size_t	highmem;
63b060ba50SNeel Natu 	char	*highmem_addr;
64366f6083SPeter Grehan 	char	*name;
65366f6083SPeter Grehan };
66366f6083SPeter Grehan 
67366f6083SPeter Grehan #define	CREATE(x)  sysctlbyname("hw.vmm.create", NULL, NULL, (x), strlen((x)))
68366f6083SPeter Grehan #define	DESTROY(x) sysctlbyname("hw.vmm.destroy", NULL, NULL, (x), strlen((x)))
69366f6083SPeter Grehan 
70366f6083SPeter Grehan static int
71366f6083SPeter Grehan vm_device_open(const char *name)
72366f6083SPeter Grehan {
73366f6083SPeter Grehan         int fd, len;
74366f6083SPeter Grehan         char *vmfile;
75366f6083SPeter Grehan 
76366f6083SPeter Grehan 	len = strlen("/dev/vmm/") + strlen(name) + 1;
77366f6083SPeter Grehan 	vmfile = malloc(len);
78366f6083SPeter Grehan 	assert(vmfile != NULL);
79366f6083SPeter Grehan 	snprintf(vmfile, len, "/dev/vmm/%s", name);
80366f6083SPeter Grehan 
81366f6083SPeter Grehan         /* Open the device file */
82366f6083SPeter Grehan         fd = open(vmfile, O_RDWR, 0);
83366f6083SPeter Grehan 
84366f6083SPeter Grehan 	free(vmfile);
85366f6083SPeter Grehan         return (fd);
86366f6083SPeter Grehan }
87366f6083SPeter Grehan 
88366f6083SPeter Grehan int
89366f6083SPeter Grehan vm_create(const char *name)
90366f6083SPeter Grehan {
91366f6083SPeter Grehan 
92366f6083SPeter Grehan 	return (CREATE((char *)name));
93366f6083SPeter Grehan }
94366f6083SPeter Grehan 
95366f6083SPeter Grehan struct vmctx *
96366f6083SPeter Grehan vm_open(const char *name)
97366f6083SPeter Grehan {
98366f6083SPeter Grehan 	struct vmctx *vm;
99366f6083SPeter Grehan 
100366f6083SPeter Grehan 	vm = malloc(sizeof(struct vmctx) + strlen(name) + 1);
101366f6083SPeter Grehan 	assert(vm != NULL);
102366f6083SPeter Grehan 
103366f6083SPeter Grehan 	vm->fd = -1;
104b060ba50SNeel Natu 	vm->lowmem_limit = 3 * GB;
105366f6083SPeter Grehan 	vm->name = (char *)(vm + 1);
106366f6083SPeter Grehan 	strcpy(vm->name, name);
107366f6083SPeter Grehan 
108366f6083SPeter Grehan 	if ((vm->fd = vm_device_open(vm->name)) < 0)
109366f6083SPeter Grehan 		goto err;
110366f6083SPeter Grehan 
111366f6083SPeter Grehan 	return (vm);
112366f6083SPeter Grehan err:
113366f6083SPeter Grehan 	vm_destroy(vm);
114366f6083SPeter Grehan 	return (NULL);
115366f6083SPeter Grehan }
116366f6083SPeter Grehan 
117366f6083SPeter Grehan void
118366f6083SPeter Grehan vm_destroy(struct vmctx *vm)
119366f6083SPeter Grehan {
120366f6083SPeter Grehan 	assert(vm != NULL);
121366f6083SPeter Grehan 
122366f6083SPeter Grehan 	if (vm->fd >= 0)
123366f6083SPeter Grehan 		close(vm->fd);
124f7d51510SNeel Natu 	DESTROY(vm->name);
125f7d51510SNeel Natu 
126366f6083SPeter Grehan 	free(vm);
127366f6083SPeter Grehan }
128366f6083SPeter Grehan 
129366f6083SPeter Grehan int
130200758f1SNeel Natu vm_parse_memsize(const char *optarg, size_t *ret_memsize)
131200758f1SNeel Natu {
132200758f1SNeel Natu 	char *endptr;
133200758f1SNeel Natu 	size_t optval;
134200758f1SNeel Natu 	int error;
135200758f1SNeel Natu 
136200758f1SNeel Natu 	optval = strtoul(optarg, &endptr, 0);
137200758f1SNeel Natu 	if (*optarg != '\0' && *endptr == '\0') {
138200758f1SNeel Natu 		/*
139200758f1SNeel Natu 		 * For the sake of backward compatibility if the memory size
140200758f1SNeel Natu 		 * specified on the command line is less than a megabyte then
141200758f1SNeel Natu 		 * it is interpreted as being in units of MB.
142200758f1SNeel Natu 		 */
143200758f1SNeel Natu 		if (optval < MB)
144200758f1SNeel Natu 			optval *= MB;
145200758f1SNeel Natu 		*ret_memsize = optval;
146200758f1SNeel Natu 		error = 0;
147200758f1SNeel Natu 	} else
148200758f1SNeel Natu 		error = expand_number(optarg, ret_memsize);
149200758f1SNeel Natu 
150200758f1SNeel Natu 	return (error);
151200758f1SNeel Natu }
152200758f1SNeel Natu 
153200758f1SNeel Natu int
154318224bbSNeel Natu vm_get_memory_seg(struct vmctx *ctx, vm_paddr_t gpa, size_t *ret_len,
155318224bbSNeel Natu 		  int *wired)
156366f6083SPeter Grehan {
157366f6083SPeter Grehan 	int error;
158366f6083SPeter Grehan 	struct vm_memory_segment seg;
159366f6083SPeter Grehan 
160366f6083SPeter Grehan 	bzero(&seg, sizeof(seg));
161366f6083SPeter Grehan 	seg.gpa = gpa;
162366f6083SPeter Grehan 	error = ioctl(ctx->fd, VM_GET_MEMORY_SEG, &seg);
163366f6083SPeter Grehan 	*ret_len = seg.len;
164318224bbSNeel Natu 	if (wired != NULL)
165318224bbSNeel Natu 		*wired = seg.wired;
166366f6083SPeter Grehan 	return (error);
167366f6083SPeter Grehan }
168366f6083SPeter Grehan 
169b060ba50SNeel Natu uint32_t
170b060ba50SNeel Natu vm_get_lowmem_limit(struct vmctx *ctx)
171b060ba50SNeel Natu {
172b060ba50SNeel Natu 
173b060ba50SNeel Natu 	return (ctx->lowmem_limit);
174b060ba50SNeel Natu }
175b060ba50SNeel Natu 
176b060ba50SNeel Natu void
177b060ba50SNeel Natu vm_set_lowmem_limit(struct vmctx *ctx, uint32_t limit)
178b060ba50SNeel Natu {
179b060ba50SNeel Natu 
180b060ba50SNeel Natu 	ctx->lowmem_limit = limit;
181b060ba50SNeel Natu }
182b060ba50SNeel Natu 
183b060ba50SNeel Natu static int
184b060ba50SNeel Natu setup_memory_segment(struct vmctx *ctx, vm_paddr_t gpa, size_t len, char **addr)
185366f6083SPeter Grehan {
186366f6083SPeter Grehan 	int error;
187366f6083SPeter Grehan 	struct vm_memory_segment seg;
188366f6083SPeter Grehan 
189366f6083SPeter Grehan 	/*
190366f6083SPeter Grehan 	 * Create and optionally map 'len' bytes of memory at guest
191366f6083SPeter Grehan 	 * physical address 'gpa'
192366f6083SPeter Grehan 	 */
193366f6083SPeter Grehan 	bzero(&seg, sizeof(seg));
194366f6083SPeter Grehan 	seg.gpa = gpa;
195366f6083SPeter Grehan 	seg.len = len;
196366f6083SPeter Grehan 	error = ioctl(ctx->fd, VM_MAP_MEMORY, &seg);
197b060ba50SNeel Natu 	if (error == 0 && addr != NULL) {
198b060ba50SNeel Natu 		*addr = mmap(NULL, len, PROT_READ | PROT_WRITE, MAP_SHARED,
199366f6083SPeter Grehan 				ctx->fd, gpa);
200366f6083SPeter Grehan 	}
201366f6083SPeter Grehan 	return (error);
202366f6083SPeter Grehan }
203366f6083SPeter Grehan 
204b060ba50SNeel Natu int
205b060ba50SNeel Natu vm_setup_memory(struct vmctx *ctx, size_t memsize, enum vm_mmap_style vms)
206b060ba50SNeel Natu {
207b060ba50SNeel Natu 	char **addr;
208b060ba50SNeel Natu 	int error;
209b060ba50SNeel Natu 
210b060ba50SNeel Natu 	/* XXX VM_MMAP_SPARSE not implemented yet */
211b060ba50SNeel Natu 	assert(vms == VM_MMAP_NONE || vms == VM_MMAP_ALL);
212b060ba50SNeel Natu 	ctx->vms = vms;
213b060ba50SNeel Natu 
214b060ba50SNeel Natu 	/*
215b060ba50SNeel Natu 	 * If 'memsize' cannot fit entirely in the 'lowmem' segment then
216b060ba50SNeel Natu 	 * create another 'highmem' segment above 4GB for the remainder.
217b060ba50SNeel Natu 	 */
218b060ba50SNeel Natu 	if (memsize > ctx->lowmem_limit) {
219b060ba50SNeel Natu 		ctx->lowmem = ctx->lowmem_limit;
220b060ba50SNeel Natu 		ctx->highmem = memsize - ctx->lowmem;
221b060ba50SNeel Natu 	} else {
222b060ba50SNeel Natu 		ctx->lowmem = memsize;
223b060ba50SNeel Natu 		ctx->highmem = 0;
224b060ba50SNeel Natu 	}
225b060ba50SNeel Natu 
226b060ba50SNeel Natu 	if (ctx->lowmem > 0) {
227b060ba50SNeel Natu 		addr = (vms == VM_MMAP_ALL) ? &ctx->lowmem_addr : NULL;
228b060ba50SNeel Natu 		error = setup_memory_segment(ctx, 0, ctx->lowmem, addr);
229b060ba50SNeel Natu 		if (error)
230b060ba50SNeel Natu 			return (error);
231b060ba50SNeel Natu 	}
232b060ba50SNeel Natu 
233b060ba50SNeel Natu 	if (ctx->highmem > 0) {
234b060ba50SNeel Natu 		addr = (vms == VM_MMAP_ALL) ? &ctx->highmem_addr : NULL;
235b060ba50SNeel Natu 		error = setup_memory_segment(ctx, 4*GB, ctx->highmem, addr);
236b060ba50SNeel Natu 		if (error)
237b060ba50SNeel Natu 			return (error);
238b060ba50SNeel Natu 	}
239b060ba50SNeel Natu 
240b060ba50SNeel Natu 	return (0);
241b060ba50SNeel Natu }
242b060ba50SNeel Natu 
243b060ba50SNeel Natu void *
244b060ba50SNeel Natu vm_map_gpa(struct vmctx *ctx, vm_paddr_t gaddr, size_t len)
245366f6083SPeter Grehan {
246366f6083SPeter Grehan 
247b060ba50SNeel Natu 	/* XXX VM_MMAP_SPARSE not implemented yet */
248b060ba50SNeel Natu 	assert(ctx->vms == VM_MMAP_ALL);
249b060ba50SNeel Natu 
250b060ba50SNeel Natu 	if (gaddr < ctx->lowmem && gaddr + len <= ctx->lowmem)
251b060ba50SNeel Natu 		return ((void *)(ctx->lowmem_addr + gaddr));
252b060ba50SNeel Natu 
253b060ba50SNeel Natu 	if (gaddr >= 4*GB) {
254b060ba50SNeel Natu 		gaddr -= 4*GB;
255b060ba50SNeel Natu 		if (gaddr < ctx->highmem && gaddr + len <= ctx->highmem)
256b060ba50SNeel Natu 			return ((void *)(ctx->highmem_addr + gaddr));
257b060ba50SNeel Natu 	}
258b060ba50SNeel Natu 
259b060ba50SNeel Natu 	return (NULL);
260366f6083SPeter Grehan }
261366f6083SPeter Grehan 
262366f6083SPeter Grehan int
263366f6083SPeter Grehan vm_set_desc(struct vmctx *ctx, int vcpu, int reg,
264366f6083SPeter Grehan 	    uint64_t base, uint32_t limit, uint32_t access)
265366f6083SPeter Grehan {
266366f6083SPeter Grehan 	int error;
267366f6083SPeter Grehan 	struct vm_seg_desc vmsegdesc;
268366f6083SPeter Grehan 
269366f6083SPeter Grehan 	bzero(&vmsegdesc, sizeof(vmsegdesc));
270366f6083SPeter Grehan 	vmsegdesc.cpuid = vcpu;
271366f6083SPeter Grehan 	vmsegdesc.regnum = reg;
272366f6083SPeter Grehan 	vmsegdesc.desc.base = base;
273366f6083SPeter Grehan 	vmsegdesc.desc.limit = limit;
274366f6083SPeter Grehan 	vmsegdesc.desc.access = access;
275366f6083SPeter Grehan 
276366f6083SPeter Grehan 	error = ioctl(ctx->fd, VM_SET_SEGMENT_DESCRIPTOR, &vmsegdesc);
277366f6083SPeter Grehan 	return (error);
278366f6083SPeter Grehan }
279366f6083SPeter Grehan 
280366f6083SPeter Grehan int
281366f6083SPeter Grehan vm_get_desc(struct vmctx *ctx, int vcpu, int reg,
282366f6083SPeter Grehan 	    uint64_t *base, uint32_t *limit, uint32_t *access)
283366f6083SPeter Grehan {
284366f6083SPeter Grehan 	int error;
285366f6083SPeter Grehan 	struct vm_seg_desc vmsegdesc;
286366f6083SPeter Grehan 
287366f6083SPeter Grehan 	bzero(&vmsegdesc, sizeof(vmsegdesc));
288366f6083SPeter Grehan 	vmsegdesc.cpuid = vcpu;
289366f6083SPeter Grehan 	vmsegdesc.regnum = reg;
290366f6083SPeter Grehan 
291366f6083SPeter Grehan 	error = ioctl(ctx->fd, VM_GET_SEGMENT_DESCRIPTOR, &vmsegdesc);
292366f6083SPeter Grehan 	if (error == 0) {
293366f6083SPeter Grehan 		*base = vmsegdesc.desc.base;
294366f6083SPeter Grehan 		*limit = vmsegdesc.desc.limit;
295366f6083SPeter Grehan 		*access = vmsegdesc.desc.access;
296366f6083SPeter Grehan 	}
297366f6083SPeter Grehan 	return (error);
298366f6083SPeter Grehan }
299366f6083SPeter Grehan 
300366f6083SPeter Grehan int
301366f6083SPeter Grehan vm_set_register(struct vmctx *ctx, int vcpu, int reg, uint64_t val)
302366f6083SPeter Grehan {
303366f6083SPeter Grehan 	int error;
304366f6083SPeter Grehan 	struct vm_register vmreg;
305366f6083SPeter Grehan 
306366f6083SPeter Grehan 	bzero(&vmreg, sizeof(vmreg));
307366f6083SPeter Grehan 	vmreg.cpuid = vcpu;
308366f6083SPeter Grehan 	vmreg.regnum = reg;
309366f6083SPeter Grehan 	vmreg.regval = val;
310366f6083SPeter Grehan 
311366f6083SPeter Grehan 	error = ioctl(ctx->fd, VM_SET_REGISTER, &vmreg);
312366f6083SPeter Grehan 	return (error);
313366f6083SPeter Grehan }
314366f6083SPeter Grehan 
315366f6083SPeter Grehan int
316366f6083SPeter Grehan vm_get_register(struct vmctx *ctx, int vcpu, int reg, uint64_t *ret_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 
325366f6083SPeter Grehan 	error = ioctl(ctx->fd, VM_GET_REGISTER, &vmreg);
326366f6083SPeter Grehan 	*ret_val = vmreg.regval;
327366f6083SPeter Grehan 	return (error);
328366f6083SPeter Grehan }
329366f6083SPeter Grehan 
330366f6083SPeter Grehan int
331366f6083SPeter Grehan vm_run(struct vmctx *ctx, int vcpu, uint64_t rip, struct vm_exit *vmexit)
332366f6083SPeter Grehan {
333366f6083SPeter Grehan 	int error;
334366f6083SPeter Grehan 	struct vm_run vmrun;
335366f6083SPeter Grehan 
336366f6083SPeter Grehan 	bzero(&vmrun, sizeof(vmrun));
337366f6083SPeter Grehan 	vmrun.cpuid = vcpu;
338366f6083SPeter Grehan 	vmrun.rip = rip;
339366f6083SPeter Grehan 
340366f6083SPeter Grehan 	error = ioctl(ctx->fd, VM_RUN, &vmrun);
341366f6083SPeter Grehan 	bcopy(&vmrun.vm_exit, vmexit, sizeof(struct vm_exit));
342366f6083SPeter Grehan 	return (error);
343366f6083SPeter Grehan }
344366f6083SPeter Grehan 
345366f6083SPeter Grehan static int
346366f6083SPeter Grehan vm_inject_event_real(struct vmctx *ctx, int vcpu, enum vm_event_type type,
347366f6083SPeter Grehan 		     int vector, int error_code, int error_code_valid)
348366f6083SPeter Grehan {
349366f6083SPeter Grehan 	struct vm_event ev;
350366f6083SPeter Grehan 
351366f6083SPeter Grehan 	bzero(&ev, sizeof(ev));
352366f6083SPeter Grehan 	ev.cpuid = vcpu;
353366f6083SPeter Grehan 	ev.type = type;
354366f6083SPeter Grehan 	ev.vector = vector;
355366f6083SPeter Grehan 	ev.error_code = error_code;
356366f6083SPeter Grehan 	ev.error_code_valid = error_code_valid;
357366f6083SPeter Grehan 
358366f6083SPeter Grehan 	return (ioctl(ctx->fd, VM_INJECT_EVENT, &ev));
359366f6083SPeter Grehan }
360366f6083SPeter Grehan 
361366f6083SPeter Grehan int
362366f6083SPeter Grehan vm_inject_event(struct vmctx *ctx, int vcpu, enum vm_event_type type,
363366f6083SPeter Grehan 		int vector)
364366f6083SPeter Grehan {
365366f6083SPeter Grehan 
366366f6083SPeter Grehan 	return (vm_inject_event_real(ctx, vcpu, type, vector, 0, 0));
367366f6083SPeter Grehan }
368366f6083SPeter Grehan 
369366f6083SPeter Grehan int
370366f6083SPeter Grehan vm_inject_event2(struct vmctx *ctx, int vcpu, enum vm_event_type type,
371366f6083SPeter Grehan 		 int vector, int error_code)
372366f6083SPeter Grehan {
373366f6083SPeter Grehan 
374366f6083SPeter Grehan 	return (vm_inject_event_real(ctx, vcpu, type, vector, error_code, 1));
375366f6083SPeter Grehan }
376366f6083SPeter Grehan 
377366f6083SPeter Grehan int
37890d4b48fSNeel Natu vm_apicid2vcpu(struct vmctx *ctx, int apicid)
37990d4b48fSNeel Natu {
38090d4b48fSNeel Natu 	/*
38190d4b48fSNeel Natu 	 * The apic id associated with the 'vcpu' has the same numerical value
38290d4b48fSNeel Natu 	 * as the 'vcpu' itself.
38390d4b48fSNeel Natu 	 */
38490d4b48fSNeel Natu 	return (apicid);
38590d4b48fSNeel Natu }
38690d4b48fSNeel Natu 
38790d4b48fSNeel Natu int
388366f6083SPeter Grehan vm_lapic_irq(struct vmctx *ctx, int vcpu, int vector)
389366f6083SPeter Grehan {
390366f6083SPeter Grehan 	struct vm_lapic_irq vmirq;
391366f6083SPeter Grehan 
392366f6083SPeter Grehan 	bzero(&vmirq, sizeof(vmirq));
393366f6083SPeter Grehan 	vmirq.cpuid = vcpu;
394366f6083SPeter Grehan 	vmirq.vector = vector;
395366f6083SPeter Grehan 
396366f6083SPeter Grehan 	return (ioctl(ctx->fd, VM_LAPIC_IRQ, &vmirq));
397366f6083SPeter Grehan }
398366f6083SPeter Grehan 
399366f6083SPeter Grehan int
400565bbb86SNeel Natu vm_ioapic_assert_irq(struct vmctx *ctx, int irq)
401565bbb86SNeel Natu {
402565bbb86SNeel Natu 	struct vm_ioapic_irq ioapic_irq;
403565bbb86SNeel Natu 
404565bbb86SNeel Natu 	bzero(&ioapic_irq, sizeof(struct vm_ioapic_irq));
405565bbb86SNeel Natu 	ioapic_irq.irq = irq;
406565bbb86SNeel Natu 
407565bbb86SNeel Natu 	return (ioctl(ctx->fd, VM_IOAPIC_ASSERT_IRQ, &ioapic_irq));
408565bbb86SNeel Natu }
409565bbb86SNeel Natu 
410565bbb86SNeel Natu int
411565bbb86SNeel Natu vm_ioapic_deassert_irq(struct vmctx *ctx, int irq)
412565bbb86SNeel Natu {
413565bbb86SNeel Natu 	struct vm_ioapic_irq ioapic_irq;
414565bbb86SNeel Natu 
415565bbb86SNeel Natu 	bzero(&ioapic_irq, sizeof(struct vm_ioapic_irq));
416565bbb86SNeel Natu 	ioapic_irq.irq = irq;
417565bbb86SNeel Natu 
418565bbb86SNeel Natu 	return (ioctl(ctx->fd, VM_IOAPIC_DEASSERT_IRQ, &ioapic_irq));
419565bbb86SNeel Natu }
420565bbb86SNeel Natu 
421565bbb86SNeel Natu int
422ac7304a7SNeel Natu vm_ioapic_pulse_irq(struct vmctx *ctx, int irq)
423ac7304a7SNeel Natu {
424ac7304a7SNeel Natu 	struct vm_ioapic_irq ioapic_irq;
425ac7304a7SNeel Natu 
426ac7304a7SNeel Natu 	bzero(&ioapic_irq, sizeof(struct vm_ioapic_irq));
427ac7304a7SNeel Natu 	ioapic_irq.irq = irq;
428ac7304a7SNeel Natu 
429ac7304a7SNeel Natu 	return (ioctl(ctx->fd, VM_IOAPIC_PULSE_IRQ, &ioapic_irq));
430ac7304a7SNeel Natu }
431ac7304a7SNeel Natu 
432ac7304a7SNeel Natu int
433366f6083SPeter Grehan vm_inject_nmi(struct vmctx *ctx, int vcpu)
434366f6083SPeter Grehan {
435366f6083SPeter Grehan 	struct vm_nmi vmnmi;
436366f6083SPeter Grehan 
437366f6083SPeter Grehan 	bzero(&vmnmi, sizeof(vmnmi));
438366f6083SPeter Grehan 	vmnmi.cpuid = vcpu;
439366f6083SPeter Grehan 
440366f6083SPeter Grehan 	return (ioctl(ctx->fd, VM_INJECT_NMI, &vmnmi));
441366f6083SPeter Grehan }
442366f6083SPeter Grehan 
443366f6083SPeter Grehan static struct {
444366f6083SPeter Grehan 	const char	*name;
445366f6083SPeter Grehan 	int		type;
446366f6083SPeter Grehan } capstrmap[] = {
447366f6083SPeter Grehan 	{ "hlt_exit",		VM_CAP_HALT_EXIT },
448366f6083SPeter Grehan 	{ "mtrap_exit",		VM_CAP_MTRAP_EXIT },
449366f6083SPeter Grehan 	{ "pause_exit",		VM_CAP_PAUSE_EXIT },
450366f6083SPeter Grehan 	{ "unrestricted_guest",	VM_CAP_UNRESTRICTED_GUEST },
45149cc03daSNeel Natu 	{ "enable_invpcid",	VM_CAP_ENABLE_INVPCID },
452366f6083SPeter Grehan 	{ 0 }
453366f6083SPeter Grehan };
454366f6083SPeter Grehan 
45513f4cf6cSNeel Natu int
45613f4cf6cSNeel Natu vm_capability_name2type(const char *capname)
45713f4cf6cSNeel Natu {
45813f4cf6cSNeel Natu 	int i;
45913f4cf6cSNeel Natu 
460366f6083SPeter Grehan 	for (i = 0; capstrmap[i].name != NULL && capname != NULL; i++) {
461366f6083SPeter Grehan 		if (strcmp(capstrmap[i].name, capname) == 0)
462366f6083SPeter Grehan 			return (capstrmap[i].type);
463366f6083SPeter Grehan 	}
464366f6083SPeter Grehan 
465366f6083SPeter Grehan 	return (-1);
466366f6083SPeter Grehan }
467366f6083SPeter Grehan 
46813f4cf6cSNeel Natu const char *
46913f4cf6cSNeel Natu vm_capability_type2name(int type)
47013f4cf6cSNeel Natu {
47113f4cf6cSNeel Natu 	int i;
47213f4cf6cSNeel Natu 
47313f4cf6cSNeel Natu 	for (i = 0; capstrmap[i].name != NULL; i++) {
47413f4cf6cSNeel Natu 		if (capstrmap[i].type == type)
47513f4cf6cSNeel Natu 			return (capstrmap[i].name);
47613f4cf6cSNeel Natu 	}
47713f4cf6cSNeel Natu 
47813f4cf6cSNeel Natu 	return (NULL);
47913f4cf6cSNeel Natu }
48013f4cf6cSNeel Natu 
481366f6083SPeter Grehan int
482366f6083SPeter Grehan vm_get_capability(struct vmctx *ctx, int vcpu, enum vm_cap_type cap,
483366f6083SPeter Grehan 		  int *retval)
484366f6083SPeter Grehan {
485366f6083SPeter Grehan 	int error;
486366f6083SPeter Grehan 	struct vm_capability vmcap;
487366f6083SPeter Grehan 
488366f6083SPeter Grehan 	bzero(&vmcap, sizeof(vmcap));
489366f6083SPeter Grehan 	vmcap.cpuid = vcpu;
490366f6083SPeter Grehan 	vmcap.captype = cap;
491366f6083SPeter Grehan 
492366f6083SPeter Grehan 	error = ioctl(ctx->fd, VM_GET_CAPABILITY, &vmcap);
493366f6083SPeter Grehan 	*retval = vmcap.capval;
494366f6083SPeter Grehan 	return (error);
495366f6083SPeter Grehan }
496366f6083SPeter Grehan 
497366f6083SPeter Grehan int
498366f6083SPeter Grehan vm_set_capability(struct vmctx *ctx, int vcpu, enum vm_cap_type cap, int val)
499366f6083SPeter Grehan {
500366f6083SPeter Grehan 	struct vm_capability vmcap;
501366f6083SPeter Grehan 
502366f6083SPeter Grehan 	bzero(&vmcap, sizeof(vmcap));
503366f6083SPeter Grehan 	vmcap.cpuid = vcpu;
504366f6083SPeter Grehan 	vmcap.captype = cap;
505366f6083SPeter Grehan 	vmcap.capval = val;
506366f6083SPeter Grehan 
507366f6083SPeter Grehan 	return (ioctl(ctx->fd, VM_SET_CAPABILITY, &vmcap));
508366f6083SPeter Grehan }
509366f6083SPeter Grehan 
510366f6083SPeter Grehan int
511366f6083SPeter Grehan vm_assign_pptdev(struct vmctx *ctx, int bus, int slot, int func)
512366f6083SPeter Grehan {
513366f6083SPeter Grehan 	struct vm_pptdev pptdev;
514366f6083SPeter Grehan 
515366f6083SPeter Grehan 	bzero(&pptdev, sizeof(pptdev));
516366f6083SPeter Grehan 	pptdev.bus = bus;
517366f6083SPeter Grehan 	pptdev.slot = slot;
518366f6083SPeter Grehan 	pptdev.func = func;
519366f6083SPeter Grehan 
520366f6083SPeter Grehan 	return (ioctl(ctx->fd, VM_BIND_PPTDEV, &pptdev));
521366f6083SPeter Grehan }
522366f6083SPeter Grehan 
523366f6083SPeter Grehan int
524366f6083SPeter Grehan vm_unassign_pptdev(struct vmctx *ctx, int bus, int slot, int func)
525366f6083SPeter Grehan {
526366f6083SPeter Grehan 	struct vm_pptdev pptdev;
527366f6083SPeter Grehan 
528366f6083SPeter Grehan 	bzero(&pptdev, sizeof(pptdev));
529366f6083SPeter Grehan 	pptdev.bus = bus;
530366f6083SPeter Grehan 	pptdev.slot = slot;
531366f6083SPeter Grehan 	pptdev.func = func;
532366f6083SPeter Grehan 
533366f6083SPeter Grehan 	return (ioctl(ctx->fd, VM_UNBIND_PPTDEV, &pptdev));
534366f6083SPeter Grehan }
535366f6083SPeter Grehan 
536366f6083SPeter Grehan int
537366f6083SPeter Grehan vm_map_pptdev_mmio(struct vmctx *ctx, int bus, int slot, int func,
538366f6083SPeter Grehan 		   vm_paddr_t gpa, size_t len, vm_paddr_t hpa)
539366f6083SPeter Grehan {
540366f6083SPeter Grehan 	struct vm_pptdev_mmio pptmmio;
541366f6083SPeter Grehan 
542366f6083SPeter Grehan 	bzero(&pptmmio, sizeof(pptmmio));
543366f6083SPeter Grehan 	pptmmio.bus = bus;
544366f6083SPeter Grehan 	pptmmio.slot = slot;
545366f6083SPeter Grehan 	pptmmio.func = func;
546366f6083SPeter Grehan 	pptmmio.gpa = gpa;
547366f6083SPeter Grehan 	pptmmio.len = len;
548366f6083SPeter Grehan 	pptmmio.hpa = hpa;
549366f6083SPeter Grehan 
550366f6083SPeter Grehan 	return (ioctl(ctx->fd, VM_MAP_PPTDEV_MMIO, &pptmmio));
551366f6083SPeter Grehan }
552366f6083SPeter Grehan 
553366f6083SPeter Grehan int
554366f6083SPeter Grehan vm_setup_msi(struct vmctx *ctx, int vcpu, int bus, int slot, int func,
555366f6083SPeter Grehan 	     int destcpu, int vector, int numvec)
556366f6083SPeter Grehan {
557366f6083SPeter Grehan 	struct vm_pptdev_msi pptmsi;
558366f6083SPeter Grehan 
559366f6083SPeter Grehan 	bzero(&pptmsi, sizeof(pptmsi));
560366f6083SPeter Grehan 	pptmsi.vcpu = vcpu;
561366f6083SPeter Grehan 	pptmsi.bus = bus;
562366f6083SPeter Grehan 	pptmsi.slot = slot;
563366f6083SPeter Grehan 	pptmsi.func = func;
564366f6083SPeter Grehan 	pptmsi.destcpu = destcpu;
565366f6083SPeter Grehan 	pptmsi.vector = vector;
566366f6083SPeter Grehan 	pptmsi.numvec = numvec;
567366f6083SPeter Grehan 
568366f6083SPeter Grehan 	return (ioctl(ctx->fd, VM_PPTDEV_MSI, &pptmsi));
569366f6083SPeter Grehan }
570366f6083SPeter Grehan 
571cd942e0fSPeter Grehan int
572cd942e0fSPeter Grehan vm_setup_msix(struct vmctx *ctx, int vcpu, int bus, int slot, int func,
573cd942e0fSPeter Grehan 	      int idx, uint32_t msg, uint32_t vector_control, uint64_t addr)
574cd942e0fSPeter Grehan {
575cd942e0fSPeter Grehan 	struct vm_pptdev_msix pptmsix;
576cd942e0fSPeter Grehan 
577cd942e0fSPeter Grehan 	bzero(&pptmsix, sizeof(pptmsix));
578cd942e0fSPeter Grehan 	pptmsix.vcpu = vcpu;
579cd942e0fSPeter Grehan 	pptmsix.bus = bus;
580cd942e0fSPeter Grehan 	pptmsix.slot = slot;
581cd942e0fSPeter Grehan 	pptmsix.func = func;
582cd942e0fSPeter Grehan 	pptmsix.idx = idx;
583cd942e0fSPeter Grehan 	pptmsix.msg = msg;
584cd942e0fSPeter Grehan 	pptmsix.addr = addr;
585cd942e0fSPeter Grehan 	pptmsix.vector_control = vector_control;
586cd942e0fSPeter Grehan 
587cd942e0fSPeter Grehan 	return ioctl(ctx->fd, VM_PPTDEV_MSIX, &pptmsix);
588cd942e0fSPeter Grehan }
589cd942e0fSPeter Grehan 
590366f6083SPeter Grehan uint64_t *
591366f6083SPeter Grehan vm_get_stats(struct vmctx *ctx, int vcpu, struct timeval *ret_tv,
592366f6083SPeter Grehan 	     int *ret_entries)
593366f6083SPeter Grehan {
594366f6083SPeter Grehan 	int error;
595366f6083SPeter Grehan 
596366f6083SPeter Grehan 	static struct vm_stats vmstats;
597366f6083SPeter Grehan 
598366f6083SPeter Grehan 	vmstats.cpuid = vcpu;
599366f6083SPeter Grehan 
600366f6083SPeter Grehan 	error = ioctl(ctx->fd, VM_STATS, &vmstats);
601366f6083SPeter Grehan 	if (error == 0) {
602366f6083SPeter Grehan 		if (ret_entries)
603366f6083SPeter Grehan 			*ret_entries = vmstats.num_entries;
604366f6083SPeter Grehan 		if (ret_tv)
605366f6083SPeter Grehan 			*ret_tv = vmstats.tv;
606366f6083SPeter Grehan 		return (vmstats.statbuf);
607366f6083SPeter Grehan 	} else
608366f6083SPeter Grehan 		return (NULL);
609366f6083SPeter Grehan }
610366f6083SPeter Grehan 
611366f6083SPeter Grehan const char *
612366f6083SPeter Grehan vm_get_stat_desc(struct vmctx *ctx, int index)
613366f6083SPeter Grehan {
614366f6083SPeter Grehan 	static struct vm_stat_desc statdesc;
615366f6083SPeter Grehan 
616366f6083SPeter Grehan 	statdesc.index = index;
617366f6083SPeter Grehan 	if (ioctl(ctx->fd, VM_STAT_DESC, &statdesc) == 0)
618366f6083SPeter Grehan 		return (statdesc.desc);
619366f6083SPeter Grehan 	else
620366f6083SPeter Grehan 		return (NULL);
621366f6083SPeter Grehan }
622366f6083SPeter Grehan 
623e9027382SNeel Natu int
624e9027382SNeel Natu vm_get_x2apic_state(struct vmctx *ctx, int vcpu, enum x2apic_state *state)
625e9027382SNeel Natu {
626e9027382SNeel Natu 	int error;
627e9027382SNeel Natu 	struct vm_x2apic x2apic;
628e9027382SNeel Natu 
629e9027382SNeel Natu 	bzero(&x2apic, sizeof(x2apic));
630e9027382SNeel Natu 	x2apic.cpuid = vcpu;
631e9027382SNeel Natu 
632e9027382SNeel Natu 	error = ioctl(ctx->fd, VM_GET_X2APIC_STATE, &x2apic);
633e9027382SNeel Natu 	*state = x2apic.state;
634e9027382SNeel Natu 	return (error);
635e9027382SNeel Natu }
636e9027382SNeel Natu 
637e9027382SNeel Natu int
638e9027382SNeel Natu vm_set_x2apic_state(struct vmctx *ctx, int vcpu, enum x2apic_state state)
639e9027382SNeel Natu {
640e9027382SNeel Natu 	int error;
641e9027382SNeel Natu 	struct vm_x2apic x2apic;
642e9027382SNeel Natu 
643e9027382SNeel Natu 	bzero(&x2apic, sizeof(x2apic));
644e9027382SNeel Natu 	x2apic.cpuid = vcpu;
645e9027382SNeel Natu 	x2apic.state = state;
646e9027382SNeel Natu 
647e9027382SNeel Natu 	error = ioctl(ctx->fd, VM_SET_X2APIC_STATE, &x2apic);
648e9027382SNeel Natu 
649e9027382SNeel Natu 	return (error);
650e9027382SNeel Natu }
651e9027382SNeel Natu 
652366f6083SPeter Grehan /*
653366f6083SPeter Grehan  * From Intel Vol 3a:
654366f6083SPeter Grehan  * Table 9-1. IA-32 Processor States Following Power-up, Reset or INIT
655366f6083SPeter Grehan  */
656366f6083SPeter Grehan int
657366f6083SPeter Grehan vcpu_reset(struct vmctx *vmctx, int vcpu)
658366f6083SPeter Grehan {
659366f6083SPeter Grehan 	int error;
660366f6083SPeter Grehan 	uint64_t rflags, rip, cr0, cr4, zero, desc_base, rdx;
661366f6083SPeter Grehan 	uint32_t desc_access, desc_limit;
662366f6083SPeter Grehan 	uint16_t sel;
663366f6083SPeter Grehan 
664366f6083SPeter Grehan 	zero = 0;
665366f6083SPeter Grehan 
666366f6083SPeter Grehan 	rflags = 0x2;
667366f6083SPeter Grehan 	error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RFLAGS, rflags);
668366f6083SPeter Grehan 	if (error)
669366f6083SPeter Grehan 		goto done;
670366f6083SPeter Grehan 
671366f6083SPeter Grehan 	rip = 0xfff0;
672366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RIP, rip)) != 0)
673366f6083SPeter Grehan 		goto done;
674366f6083SPeter Grehan 
675366f6083SPeter Grehan 	cr0 = CR0_NE;
676366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_CR0, cr0)) != 0)
677366f6083SPeter Grehan 		goto done;
678366f6083SPeter Grehan 
679366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_CR3, zero)) != 0)
680366f6083SPeter Grehan 		goto done;
681366f6083SPeter Grehan 
68232c96bc8SNeel Natu 	cr4 = 0;
683366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_CR4, cr4)) != 0)
684366f6083SPeter Grehan 		goto done;
685366f6083SPeter Grehan 
686366f6083SPeter Grehan 	/*
687366f6083SPeter Grehan 	 * CS: present, r/w, accessed, 16-bit, byte granularity, usable
688366f6083SPeter Grehan 	 */
689366f6083SPeter Grehan 	desc_base = 0xffff0000;
690366f6083SPeter Grehan 	desc_limit = 0xffff;
691366f6083SPeter Grehan 	desc_access = 0x0093;
692366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_CS,
693366f6083SPeter Grehan 			    desc_base, desc_limit, desc_access);
694366f6083SPeter Grehan 	if (error)
695366f6083SPeter Grehan 		goto done;
696366f6083SPeter Grehan 
697366f6083SPeter Grehan 	sel = 0xf000;
698366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_CS, sel)) != 0)
699366f6083SPeter Grehan 		goto done;
700366f6083SPeter Grehan 
701366f6083SPeter Grehan 	/*
702366f6083SPeter Grehan 	 * SS,DS,ES,FS,GS: present, r/w, accessed, 16-bit, byte granularity
703366f6083SPeter Grehan 	 */
704366f6083SPeter Grehan 	desc_base = 0;
705366f6083SPeter Grehan 	desc_limit = 0xffff;
706366f6083SPeter Grehan 	desc_access = 0x0093;
707366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_SS,
708366f6083SPeter Grehan 			    desc_base, desc_limit, desc_access);
709366f6083SPeter Grehan 	if (error)
710366f6083SPeter Grehan 		goto done;
711366f6083SPeter Grehan 
712366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_DS,
713366f6083SPeter Grehan 			    desc_base, desc_limit, desc_access);
714366f6083SPeter Grehan 	if (error)
715366f6083SPeter Grehan 		goto done;
716366f6083SPeter Grehan 
717366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_ES,
718366f6083SPeter Grehan 			    desc_base, desc_limit, desc_access);
719366f6083SPeter Grehan 	if (error)
720366f6083SPeter Grehan 		goto done;
721366f6083SPeter Grehan 
722366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_FS,
723366f6083SPeter Grehan 			    desc_base, desc_limit, desc_access);
724366f6083SPeter Grehan 	if (error)
725366f6083SPeter Grehan 		goto done;
726366f6083SPeter Grehan 
727366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_GS,
728366f6083SPeter Grehan 			    desc_base, desc_limit, desc_access);
729366f6083SPeter Grehan 	if (error)
730366f6083SPeter Grehan 		goto done;
731366f6083SPeter Grehan 
732366f6083SPeter Grehan 	sel = 0;
733366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_SS, sel)) != 0)
734366f6083SPeter Grehan 		goto done;
735366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_DS, sel)) != 0)
736366f6083SPeter Grehan 		goto done;
737366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_ES, sel)) != 0)
738366f6083SPeter Grehan 		goto done;
739366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_FS, sel)) != 0)
740366f6083SPeter Grehan 		goto done;
741366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_GS, sel)) != 0)
742366f6083SPeter Grehan 		goto done;
743366f6083SPeter Grehan 
744366f6083SPeter Grehan 	/* General purpose registers */
745366f6083SPeter Grehan 	rdx = 0xf00;
746366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RAX, zero)) != 0)
747366f6083SPeter Grehan 		goto done;
748366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RBX, zero)) != 0)
749366f6083SPeter Grehan 		goto done;
750366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RCX, zero)) != 0)
751366f6083SPeter Grehan 		goto done;
752366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RDX, rdx)) != 0)
753366f6083SPeter Grehan 		goto done;
754366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RSI, zero)) != 0)
755366f6083SPeter Grehan 		goto done;
756366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RDI, zero)) != 0)
757366f6083SPeter Grehan 		goto done;
758366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RBP, zero)) != 0)
759366f6083SPeter Grehan 		goto done;
760366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RSP, zero)) != 0)
761366f6083SPeter Grehan 		goto done;
762366f6083SPeter Grehan 
763366f6083SPeter Grehan 	/* GDTR, IDTR */
764366f6083SPeter Grehan 	desc_base = 0;
765366f6083SPeter Grehan 	desc_limit = 0xffff;
766366f6083SPeter Grehan 	desc_access = 0;
767366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_GDTR,
768366f6083SPeter Grehan 			    desc_base, desc_limit, desc_access);
769366f6083SPeter Grehan 	if (error != 0)
770366f6083SPeter Grehan 		goto done;
771366f6083SPeter Grehan 
772366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_IDTR,
773366f6083SPeter Grehan 			    desc_base, desc_limit, desc_access);
774366f6083SPeter Grehan 	if (error != 0)
775366f6083SPeter Grehan 		goto done;
776366f6083SPeter Grehan 
777366f6083SPeter Grehan 	/* TR */
778366f6083SPeter Grehan 	desc_base = 0;
779366f6083SPeter Grehan 	desc_limit = 0xffff;
780366f6083SPeter Grehan 	desc_access = 0x0000008b;
781366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_TR, 0, 0, desc_access);
782366f6083SPeter Grehan 	if (error)
783366f6083SPeter Grehan 		goto done;
784366f6083SPeter Grehan 
785366f6083SPeter Grehan 	sel = 0;
786366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_TR, sel)) != 0)
787366f6083SPeter Grehan 		goto done;
788366f6083SPeter Grehan 
789366f6083SPeter Grehan 	/* LDTR */
790366f6083SPeter Grehan 	desc_base = 0;
791366f6083SPeter Grehan 	desc_limit = 0xffff;
792366f6083SPeter Grehan 	desc_access = 0x00000082;
793366f6083SPeter Grehan 	error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_LDTR, desc_base,
794366f6083SPeter Grehan 			    desc_limit, desc_access);
795366f6083SPeter Grehan 	if (error)
796366f6083SPeter Grehan 		goto done;
797366f6083SPeter Grehan 
798366f6083SPeter Grehan 	sel = 0;
799366f6083SPeter Grehan 	if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_LDTR, 0)) != 0)
800366f6083SPeter Grehan 		goto done;
801366f6083SPeter Grehan 
802366f6083SPeter Grehan 	/* XXX cr2, debug registers */
803366f6083SPeter Grehan 
804366f6083SPeter Grehan 	error = 0;
805366f6083SPeter Grehan done:
806366f6083SPeter Grehan 	return (error);
807366f6083SPeter Grehan }
808318224bbSNeel Natu 
809318224bbSNeel Natu int
810318224bbSNeel Natu vm_get_gpa_pmap(struct vmctx *ctx, uint64_t gpa, uint64_t *pte, int *num)
811318224bbSNeel Natu {
812318224bbSNeel Natu 	int error, i;
813318224bbSNeel Natu 	struct vm_gpa_pte gpapte;
814318224bbSNeel Natu 
815318224bbSNeel Natu 	bzero(&gpapte, sizeof(gpapte));
816318224bbSNeel Natu 	gpapte.gpa = gpa;
817318224bbSNeel Natu 
818318224bbSNeel Natu 	error = ioctl(ctx->fd, VM_GET_GPA_PMAP, &gpapte);
819318224bbSNeel Natu 
820318224bbSNeel Natu 	if (error == 0) {
821318224bbSNeel Natu 		*num = gpapte.ptenum;
822318224bbSNeel Natu 		for (i = 0; i < gpapte.ptenum; i++)
823318224bbSNeel Natu 			pte[i] = gpapte.pte[i];
824318224bbSNeel Natu 	}
825318224bbSNeel Natu 
826318224bbSNeel Natu 	return (error);
827318224bbSNeel Natu }
828*08e3ff32SNeel Natu 
829*08e3ff32SNeel Natu int
830*08e3ff32SNeel Natu vm_get_hpet_capabilities(struct vmctx *ctx, uint32_t *capabilities)
831*08e3ff32SNeel Natu {
832*08e3ff32SNeel Natu 	int error;
833*08e3ff32SNeel Natu 	struct vm_hpet_cap cap;
834*08e3ff32SNeel Natu 
835*08e3ff32SNeel Natu 	bzero(&cap, sizeof(struct vm_hpet_cap));
836*08e3ff32SNeel Natu 	error = ioctl(ctx->fd, VM_GET_HPET_CAPABILITIES, &cap);
837*08e3ff32SNeel Natu 	if (capabilities != NULL)
838*08e3ff32SNeel Natu 		*capabilities = cap.capabilities;
839*08e3ff32SNeel Natu 	return (error);
840*08e3ff32SNeel Natu }
841