xref: /freebsd/sys/amd64/vmm/intel/vmx.c (revision 762fd2080408a6bb2d0ea29cc427c7390da8e21b)
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/param.h>
33366f6083SPeter Grehan #include <sys/systm.h>
34366f6083SPeter Grehan #include <sys/smp.h>
35366f6083SPeter Grehan #include <sys/kernel.h>
36366f6083SPeter Grehan #include <sys/malloc.h>
37366f6083SPeter Grehan #include <sys/pcpu.h>
38366f6083SPeter Grehan #include <sys/proc.h>
393565b59eSNeel Natu #include <sys/sysctl.h>
40366f6083SPeter Grehan 
41366f6083SPeter Grehan #include <vm/vm.h>
42366f6083SPeter Grehan #include <vm/pmap.h>
43366f6083SPeter Grehan 
44366f6083SPeter Grehan #include <machine/psl.h>
45366f6083SPeter Grehan #include <machine/cpufunc.h>
468b287612SJohn Baldwin #include <machine/md_var.h>
47366f6083SPeter Grehan #include <machine/segments.h>
48176666c2SNeel Natu #include <machine/smp.h>
49608f97c3SPeter Grehan #include <machine/specialreg.h>
50366f6083SPeter Grehan #include <machine/vmparam.h>
51366f6083SPeter Grehan 
52366f6083SPeter Grehan #include <machine/vmm.h>
53dc506506SNeel Natu #include <machine/vmm_dev.h>
54b01c2033SNeel Natu #include "vmm_host.h"
55*762fd208STycho Nightingale #include "vmm_ioport.h"
56176666c2SNeel Natu #include "vmm_ipi.h"
57366f6083SPeter Grehan #include "vmm_msr.h"
58366f6083SPeter Grehan #include "vmm_ktr.h"
59366f6083SPeter Grehan #include "vmm_stat.h"
60de5ea6b6SNeel Natu #include "vlapic.h"
61de5ea6b6SNeel Natu #include "vlapic_priv.h"
62366f6083SPeter Grehan 
63366f6083SPeter Grehan #include "vmx_msr.h"
64366f6083SPeter Grehan #include "ept.h"
65366f6083SPeter Grehan #include "vmx_cpufunc.h"
66366f6083SPeter Grehan #include "vmx.h"
67366f6083SPeter Grehan #include "x86.h"
68366f6083SPeter Grehan #include "vmx_controls.h"
69366f6083SPeter Grehan 
70366f6083SPeter Grehan #define	PINBASED_CTLS_ONE_SETTING					\
71366f6083SPeter Grehan 	(PINBASED_EXTINT_EXITING	|				\
72366f6083SPeter Grehan 	 PINBASED_NMI_EXITING		|				\
73366f6083SPeter Grehan 	 PINBASED_VIRTUAL_NMI)
74366f6083SPeter Grehan #define	PINBASED_CTLS_ZERO_SETTING	0
75366f6083SPeter Grehan 
76366f6083SPeter Grehan #define PROCBASED_CTLS_WINDOW_SETTING					\
77366f6083SPeter Grehan 	(PROCBASED_INT_WINDOW_EXITING	|				\
78366f6083SPeter Grehan 	 PROCBASED_NMI_WINDOW_EXITING)
79366f6083SPeter Grehan 
80366f6083SPeter Grehan #define	PROCBASED_CTLS_ONE_SETTING 					\
81366f6083SPeter Grehan 	(PROCBASED_SECONDARY_CONTROLS	|				\
82366f6083SPeter Grehan 	 PROCBASED_IO_EXITING		|				\
83366f6083SPeter Grehan 	 PROCBASED_MSR_BITMAPS		|				\
84366f6083SPeter Grehan 	 PROCBASED_CTLS_WINDOW_SETTING)
85366f6083SPeter Grehan #define	PROCBASED_CTLS_ZERO_SETTING	\
86366f6083SPeter Grehan 	(PROCBASED_CR3_LOAD_EXITING |	\
87366f6083SPeter Grehan 	PROCBASED_CR3_STORE_EXITING |	\
88366f6083SPeter Grehan 	PROCBASED_IO_BITMAPS)
89366f6083SPeter Grehan 
90366f6083SPeter Grehan #define	PROCBASED_CTLS2_ONE_SETTING	PROCBASED2_ENABLE_EPT
91366f6083SPeter Grehan #define	PROCBASED_CTLS2_ZERO_SETTING	0
92366f6083SPeter Grehan 
93608f97c3SPeter Grehan #define VM_EXIT_CTLS_ONE_SETTING_NO_PAT					\
94366f6083SPeter Grehan 	(VM_EXIT_HOST_LMA			|			\
95366f6083SPeter Grehan 	VM_EXIT_SAVE_EFER			|			\
96366f6083SPeter Grehan 	VM_EXIT_LOAD_EFER)
97608f97c3SPeter Grehan 
98608f97c3SPeter Grehan #define	VM_EXIT_CTLS_ONE_SETTING					\
99608f97c3SPeter Grehan 	(VM_EXIT_CTLS_ONE_SETTING_NO_PAT       	|			\
100f7d47425SNeel Natu 	VM_EXIT_ACKNOWLEDGE_INTERRUPT		|			\
101608f97c3SPeter Grehan 	VM_EXIT_SAVE_PAT			|			\
102608f97c3SPeter Grehan 	VM_EXIT_LOAD_PAT)
103366f6083SPeter Grehan #define	VM_EXIT_CTLS_ZERO_SETTING	VM_EXIT_SAVE_DEBUG_CONTROLS
104366f6083SPeter Grehan 
105608f97c3SPeter Grehan #define	VM_ENTRY_CTLS_ONE_SETTING_NO_PAT	VM_ENTRY_LOAD_EFER
106608f97c3SPeter Grehan 
107366f6083SPeter Grehan #define	VM_ENTRY_CTLS_ONE_SETTING					\
108608f97c3SPeter Grehan 	(VM_ENTRY_CTLS_ONE_SETTING_NO_PAT     	|			\
109608f97c3SPeter Grehan 	VM_ENTRY_LOAD_PAT)
110366f6083SPeter Grehan #define	VM_ENTRY_CTLS_ZERO_SETTING					\
111366f6083SPeter Grehan 	(VM_ENTRY_LOAD_DEBUG_CONTROLS		|			\
112366f6083SPeter Grehan 	VM_ENTRY_INTO_SMM			|			\
113366f6083SPeter Grehan 	VM_ENTRY_DEACTIVATE_DUAL_MONITOR)
114366f6083SPeter Grehan 
115366f6083SPeter Grehan #define	guest_msr_rw(vmx, msr) \
116366f6083SPeter Grehan 	msr_bitmap_change_access((vmx)->msr_bitmap, (msr), MSR_BITMAP_ACCESS_RW)
117366f6083SPeter Grehan 
118159dd56fSNeel Natu #define	guest_msr_ro(vmx, msr) \
119159dd56fSNeel Natu     msr_bitmap_change_access((vmx)->msr_bitmap, (msr), MSR_BITMAP_ACCESS_READ)
120159dd56fSNeel Natu 
121366f6083SPeter Grehan #define	HANDLED		1
122366f6083SPeter Grehan #define	UNHANDLED	0
123366f6083SPeter Grehan 
124de5ea6b6SNeel Natu static MALLOC_DEFINE(M_VMX, "vmx", "vmx");
125de5ea6b6SNeel Natu static MALLOC_DEFINE(M_VLAPIC, "vlapic", "vlapic");
126366f6083SPeter Grehan 
1273565b59eSNeel Natu SYSCTL_DECL(_hw_vmm);
1283565b59eSNeel Natu SYSCTL_NODE(_hw_vmm, OID_AUTO, vmx, CTLFLAG_RW, NULL, NULL);
1293565b59eSNeel Natu 
130b3996dd4SJohn Baldwin int vmxon_enabled[MAXCPU];
131366f6083SPeter Grehan static char vmxon_region[MAXCPU][PAGE_SIZE] __aligned(PAGE_SIZE);
132366f6083SPeter Grehan 
133366f6083SPeter Grehan static uint32_t pinbased_ctls, procbased_ctls, procbased_ctls2;
134366f6083SPeter Grehan static uint32_t exit_ctls, entry_ctls;
135366f6083SPeter Grehan 
136366f6083SPeter Grehan static uint64_t cr0_ones_mask, cr0_zeros_mask;
1373565b59eSNeel Natu SYSCTL_ULONG(_hw_vmm_vmx, OID_AUTO, cr0_ones_mask, CTLFLAG_RD,
1383565b59eSNeel Natu 	     &cr0_ones_mask, 0, NULL);
1393565b59eSNeel Natu SYSCTL_ULONG(_hw_vmm_vmx, OID_AUTO, cr0_zeros_mask, CTLFLAG_RD,
1403565b59eSNeel Natu 	     &cr0_zeros_mask, 0, NULL);
1413565b59eSNeel Natu 
142366f6083SPeter Grehan static uint64_t cr4_ones_mask, cr4_zeros_mask;
1433565b59eSNeel Natu SYSCTL_ULONG(_hw_vmm_vmx, OID_AUTO, cr4_ones_mask, CTLFLAG_RD,
1443565b59eSNeel Natu 	     &cr4_ones_mask, 0, NULL);
1453565b59eSNeel Natu SYSCTL_ULONG(_hw_vmm_vmx, OID_AUTO, cr4_zeros_mask, CTLFLAG_RD,
1463565b59eSNeel Natu 	     &cr4_zeros_mask, 0, NULL);
147366f6083SPeter Grehan 
148608f97c3SPeter Grehan static int vmx_no_patmsr;
149608f97c3SPeter Grehan 
1503565b59eSNeel Natu static int vmx_initialized;
1513565b59eSNeel Natu SYSCTL_INT(_hw_vmm_vmx, OID_AUTO, initialized, CTLFLAG_RD,
1523565b59eSNeel Natu 	   &vmx_initialized, 0, "Intel VMX initialized");
1533565b59eSNeel Natu 
154366f6083SPeter Grehan /*
155366f6083SPeter Grehan  * Optional capabilities
156366f6083SPeter Grehan  */
157366f6083SPeter Grehan static int cap_halt_exit;
158366f6083SPeter Grehan static int cap_pause_exit;
159366f6083SPeter Grehan static int cap_unrestricted_guest;
160366f6083SPeter Grehan static int cap_monitor_trap;
16149cc03daSNeel Natu static int cap_invpcid;
162366f6083SPeter Grehan 
16388c4b8d1SNeel Natu static int virtual_interrupt_delivery;
16488c4b8d1SNeel Natu SYSCTL_INT(_hw_vmm_vmx, OID_AUTO, virtual_interrupt_delivery, CTLFLAG_RD,
16588c4b8d1SNeel Natu     &virtual_interrupt_delivery, 0, "APICv virtual interrupt delivery support");
16688c4b8d1SNeel Natu 
167176666c2SNeel Natu static int posted_interrupts;
168176666c2SNeel Natu SYSCTL_INT(_hw_vmm_vmx, OID_AUTO, posted_interrupts, CTLFLAG_RD,
169176666c2SNeel Natu     &posted_interrupts, 0, "APICv posted interrupt support");
170176666c2SNeel Natu 
171176666c2SNeel Natu static int pirvec;
172176666c2SNeel Natu SYSCTL_INT(_hw_vmm_vmx, OID_AUTO, posted_interrupt_vector, CTLFLAG_RD,
173176666c2SNeel Natu     &pirvec, 0, "APICv posted interrupt vector");
174176666c2SNeel Natu 
17545e51299SNeel Natu static struct unrhdr *vpid_unr;
17645e51299SNeel Natu static u_int vpid_alloc_failed;
17745e51299SNeel Natu SYSCTL_UINT(_hw_vmm_vmx, OID_AUTO, vpid_alloc_failed, CTLFLAG_RD,
17845e51299SNeel Natu 	    &vpid_alloc_failed, 0, NULL);
17945e51299SNeel Natu 
18088c4b8d1SNeel Natu /*
18188c4b8d1SNeel Natu  * Use the last page below 4GB as the APIC access address. This address is
18288c4b8d1SNeel Natu  * occupied by the boot firmware so it is guaranteed that it will not conflict
18388c4b8d1SNeel Natu  * with a page in system memory.
18488c4b8d1SNeel Natu  */
18588c4b8d1SNeel Natu #define	APIC_ACCESS_ADDRESS	0xFFFFF000
18688c4b8d1SNeel Natu 
18788c4b8d1SNeel Natu static void vmx_inject_pir(struct vlapic *vlapic);
18888c4b8d1SNeel Natu 
189366f6083SPeter Grehan #ifdef KTR
190366f6083SPeter Grehan static const char *
191366f6083SPeter Grehan exit_reason_to_str(int reason)
192366f6083SPeter Grehan {
193366f6083SPeter Grehan 	static char reasonbuf[32];
194366f6083SPeter Grehan 
195366f6083SPeter Grehan 	switch (reason) {
196366f6083SPeter Grehan 	case EXIT_REASON_EXCEPTION:
197366f6083SPeter Grehan 		return "exception";
198366f6083SPeter Grehan 	case EXIT_REASON_EXT_INTR:
199366f6083SPeter Grehan 		return "extint";
200366f6083SPeter Grehan 	case EXIT_REASON_TRIPLE_FAULT:
201366f6083SPeter Grehan 		return "triplefault";
202366f6083SPeter Grehan 	case EXIT_REASON_INIT:
203366f6083SPeter Grehan 		return "init";
204366f6083SPeter Grehan 	case EXIT_REASON_SIPI:
205366f6083SPeter Grehan 		return "sipi";
206366f6083SPeter Grehan 	case EXIT_REASON_IO_SMI:
207366f6083SPeter Grehan 		return "iosmi";
208366f6083SPeter Grehan 	case EXIT_REASON_SMI:
209366f6083SPeter Grehan 		return "smi";
210366f6083SPeter Grehan 	case EXIT_REASON_INTR_WINDOW:
211366f6083SPeter Grehan 		return "intrwindow";
212366f6083SPeter Grehan 	case EXIT_REASON_NMI_WINDOW:
213366f6083SPeter Grehan 		return "nmiwindow";
214366f6083SPeter Grehan 	case EXIT_REASON_TASK_SWITCH:
215366f6083SPeter Grehan 		return "taskswitch";
216366f6083SPeter Grehan 	case EXIT_REASON_CPUID:
217366f6083SPeter Grehan 		return "cpuid";
218366f6083SPeter Grehan 	case EXIT_REASON_GETSEC:
219366f6083SPeter Grehan 		return "getsec";
220366f6083SPeter Grehan 	case EXIT_REASON_HLT:
221366f6083SPeter Grehan 		return "hlt";
222366f6083SPeter Grehan 	case EXIT_REASON_INVD:
223366f6083SPeter Grehan 		return "invd";
224366f6083SPeter Grehan 	case EXIT_REASON_INVLPG:
225366f6083SPeter Grehan 		return "invlpg";
226366f6083SPeter Grehan 	case EXIT_REASON_RDPMC:
227366f6083SPeter Grehan 		return "rdpmc";
228366f6083SPeter Grehan 	case EXIT_REASON_RDTSC:
229366f6083SPeter Grehan 		return "rdtsc";
230366f6083SPeter Grehan 	case EXIT_REASON_RSM:
231366f6083SPeter Grehan 		return "rsm";
232366f6083SPeter Grehan 	case EXIT_REASON_VMCALL:
233366f6083SPeter Grehan 		return "vmcall";
234366f6083SPeter Grehan 	case EXIT_REASON_VMCLEAR:
235366f6083SPeter Grehan 		return "vmclear";
236366f6083SPeter Grehan 	case EXIT_REASON_VMLAUNCH:
237366f6083SPeter Grehan 		return "vmlaunch";
238366f6083SPeter Grehan 	case EXIT_REASON_VMPTRLD:
239366f6083SPeter Grehan 		return "vmptrld";
240366f6083SPeter Grehan 	case EXIT_REASON_VMPTRST:
241366f6083SPeter Grehan 		return "vmptrst";
242366f6083SPeter Grehan 	case EXIT_REASON_VMREAD:
243366f6083SPeter Grehan 		return "vmread";
244366f6083SPeter Grehan 	case EXIT_REASON_VMRESUME:
245366f6083SPeter Grehan 		return "vmresume";
246366f6083SPeter Grehan 	case EXIT_REASON_VMWRITE:
247366f6083SPeter Grehan 		return "vmwrite";
248366f6083SPeter Grehan 	case EXIT_REASON_VMXOFF:
249366f6083SPeter Grehan 		return "vmxoff";
250366f6083SPeter Grehan 	case EXIT_REASON_VMXON:
251366f6083SPeter Grehan 		return "vmxon";
252366f6083SPeter Grehan 	case EXIT_REASON_CR_ACCESS:
253366f6083SPeter Grehan 		return "craccess";
254366f6083SPeter Grehan 	case EXIT_REASON_DR_ACCESS:
255366f6083SPeter Grehan 		return "draccess";
256366f6083SPeter Grehan 	case EXIT_REASON_INOUT:
257366f6083SPeter Grehan 		return "inout";
258366f6083SPeter Grehan 	case EXIT_REASON_RDMSR:
259366f6083SPeter Grehan 		return "rdmsr";
260366f6083SPeter Grehan 	case EXIT_REASON_WRMSR:
261366f6083SPeter Grehan 		return "wrmsr";
262366f6083SPeter Grehan 	case EXIT_REASON_INVAL_VMCS:
263366f6083SPeter Grehan 		return "invalvmcs";
264366f6083SPeter Grehan 	case EXIT_REASON_INVAL_MSR:
265366f6083SPeter Grehan 		return "invalmsr";
266366f6083SPeter Grehan 	case EXIT_REASON_MWAIT:
267366f6083SPeter Grehan 		return "mwait";
268366f6083SPeter Grehan 	case EXIT_REASON_MTF:
269366f6083SPeter Grehan 		return "mtf";
270366f6083SPeter Grehan 	case EXIT_REASON_MONITOR:
271366f6083SPeter Grehan 		return "monitor";
272366f6083SPeter Grehan 	case EXIT_REASON_PAUSE:
273366f6083SPeter Grehan 		return "pause";
274366f6083SPeter Grehan 	case EXIT_REASON_MCE:
275366f6083SPeter Grehan 		return "mce";
276366f6083SPeter Grehan 	case EXIT_REASON_TPR:
277366f6083SPeter Grehan 		return "tpr";
27888c4b8d1SNeel Natu 	case EXIT_REASON_APIC_ACCESS:
27988c4b8d1SNeel Natu 		return "apic-access";
280366f6083SPeter Grehan 	case EXIT_REASON_GDTR_IDTR:
281366f6083SPeter Grehan 		return "gdtridtr";
282366f6083SPeter Grehan 	case EXIT_REASON_LDTR_TR:
283366f6083SPeter Grehan 		return "ldtrtr";
284366f6083SPeter Grehan 	case EXIT_REASON_EPT_FAULT:
285366f6083SPeter Grehan 		return "eptfault";
286366f6083SPeter Grehan 	case EXIT_REASON_EPT_MISCONFIG:
287366f6083SPeter Grehan 		return "eptmisconfig";
288366f6083SPeter Grehan 	case EXIT_REASON_INVEPT:
289366f6083SPeter Grehan 		return "invept";
290366f6083SPeter Grehan 	case EXIT_REASON_RDTSCP:
291366f6083SPeter Grehan 		return "rdtscp";
292366f6083SPeter Grehan 	case EXIT_REASON_VMX_PREEMPT:
293366f6083SPeter Grehan 		return "vmxpreempt";
294366f6083SPeter Grehan 	case EXIT_REASON_INVVPID:
295366f6083SPeter Grehan 		return "invvpid";
296366f6083SPeter Grehan 	case EXIT_REASON_WBINVD:
297366f6083SPeter Grehan 		return "wbinvd";
298366f6083SPeter Grehan 	case EXIT_REASON_XSETBV:
299366f6083SPeter Grehan 		return "xsetbv";
30088c4b8d1SNeel Natu 	case EXIT_REASON_APIC_WRITE:
30188c4b8d1SNeel Natu 		return "apic-write";
302366f6083SPeter Grehan 	default:
303366f6083SPeter Grehan 		snprintf(reasonbuf, sizeof(reasonbuf), "%d", reason);
304366f6083SPeter Grehan 		return (reasonbuf);
305366f6083SPeter Grehan 	}
306366f6083SPeter Grehan }
307366f6083SPeter Grehan #endif	/* KTR */
308366f6083SPeter Grehan 
309159dd56fSNeel Natu static int
310159dd56fSNeel Natu vmx_allow_x2apic_msrs(struct vmx *vmx)
311159dd56fSNeel Natu {
312159dd56fSNeel Natu 	int i, error;
313159dd56fSNeel Natu 
314159dd56fSNeel Natu 	error = 0;
315159dd56fSNeel Natu 
316159dd56fSNeel Natu 	/*
317159dd56fSNeel Natu 	 * Allow readonly access to the following x2APIC MSRs from the guest.
318159dd56fSNeel Natu 	 */
319159dd56fSNeel Natu 	error += guest_msr_ro(vmx, MSR_APIC_ID);
320159dd56fSNeel Natu 	error += guest_msr_ro(vmx, MSR_APIC_VERSION);
321159dd56fSNeel Natu 	error += guest_msr_ro(vmx, MSR_APIC_LDR);
322159dd56fSNeel Natu 	error += guest_msr_ro(vmx, MSR_APIC_SVR);
323159dd56fSNeel Natu 
324159dd56fSNeel Natu 	for (i = 0; i < 8; i++)
325159dd56fSNeel Natu 		error += guest_msr_ro(vmx, MSR_APIC_ISR0 + i);
326159dd56fSNeel Natu 
327159dd56fSNeel Natu 	for (i = 0; i < 8; i++)
328159dd56fSNeel Natu 		error += guest_msr_ro(vmx, MSR_APIC_TMR0 + i);
329159dd56fSNeel Natu 
330159dd56fSNeel Natu 	for (i = 0; i < 8; i++)
331159dd56fSNeel Natu 		error += guest_msr_ro(vmx, MSR_APIC_IRR0 + i);
332159dd56fSNeel Natu 
333159dd56fSNeel Natu 	error += guest_msr_ro(vmx, MSR_APIC_ESR);
334159dd56fSNeel Natu 	error += guest_msr_ro(vmx, MSR_APIC_LVT_TIMER);
335159dd56fSNeel Natu 	error += guest_msr_ro(vmx, MSR_APIC_LVT_THERMAL);
336159dd56fSNeel Natu 	error += guest_msr_ro(vmx, MSR_APIC_LVT_PCINT);
337159dd56fSNeel Natu 	error += guest_msr_ro(vmx, MSR_APIC_LVT_LINT0);
338159dd56fSNeel Natu 	error += guest_msr_ro(vmx, MSR_APIC_LVT_LINT1);
339159dd56fSNeel Natu 	error += guest_msr_ro(vmx, MSR_APIC_LVT_ERROR);
340159dd56fSNeel Natu 	error += guest_msr_ro(vmx, MSR_APIC_ICR_TIMER);
341159dd56fSNeel Natu 	error += guest_msr_ro(vmx, MSR_APIC_DCR_TIMER);
342159dd56fSNeel Natu 	error += guest_msr_ro(vmx, MSR_APIC_ICR);
343159dd56fSNeel Natu 
344159dd56fSNeel Natu 	/*
345159dd56fSNeel Natu 	 * Allow TPR, EOI and SELF_IPI MSRs to be read and written by the guest.
346159dd56fSNeel Natu 	 *
347159dd56fSNeel Natu 	 * These registers get special treatment described in the section
348159dd56fSNeel Natu 	 * "Virtualizing MSR-Based APIC Accesses".
349159dd56fSNeel Natu 	 */
350159dd56fSNeel Natu 	error += guest_msr_rw(vmx, MSR_APIC_TPR);
351159dd56fSNeel Natu 	error += guest_msr_rw(vmx, MSR_APIC_EOI);
352159dd56fSNeel Natu 	error += guest_msr_rw(vmx, MSR_APIC_SELF_IPI);
353159dd56fSNeel Natu 
354159dd56fSNeel Natu 	return (error);
355159dd56fSNeel Natu }
356159dd56fSNeel Natu 
357366f6083SPeter Grehan u_long
358366f6083SPeter Grehan vmx_fix_cr0(u_long cr0)
359366f6083SPeter Grehan {
360366f6083SPeter Grehan 
361366f6083SPeter Grehan 	return ((cr0 | cr0_ones_mask) & ~cr0_zeros_mask);
362366f6083SPeter Grehan }
363366f6083SPeter Grehan 
364366f6083SPeter Grehan u_long
365366f6083SPeter Grehan vmx_fix_cr4(u_long cr4)
366366f6083SPeter Grehan {
367366f6083SPeter Grehan 
368366f6083SPeter Grehan 	return ((cr4 | cr4_ones_mask) & ~cr4_zeros_mask);
369366f6083SPeter Grehan }
370366f6083SPeter Grehan 
371366f6083SPeter Grehan static void
37245e51299SNeel Natu vpid_free(int vpid)
37345e51299SNeel Natu {
37445e51299SNeel Natu 	if (vpid < 0 || vpid > 0xffff)
37545e51299SNeel Natu 		panic("vpid_free: invalid vpid %d", vpid);
37645e51299SNeel Natu 
37745e51299SNeel Natu 	/*
37845e51299SNeel Natu 	 * VPIDs [0,VM_MAXCPU] are special and are not allocated from
37945e51299SNeel Natu 	 * the unit number allocator.
38045e51299SNeel Natu 	 */
38145e51299SNeel Natu 
38245e51299SNeel Natu 	if (vpid > VM_MAXCPU)
38345e51299SNeel Natu 		free_unr(vpid_unr, vpid);
38445e51299SNeel Natu }
38545e51299SNeel Natu 
38645e51299SNeel Natu static void
38745e51299SNeel Natu vpid_alloc(uint16_t *vpid, int num)
38845e51299SNeel Natu {
38945e51299SNeel Natu 	int i, x;
39045e51299SNeel Natu 
39145e51299SNeel Natu 	if (num <= 0 || num > VM_MAXCPU)
39245e51299SNeel Natu 		panic("invalid number of vpids requested: %d", num);
39345e51299SNeel Natu 
39445e51299SNeel Natu 	/*
39545e51299SNeel Natu 	 * If the "enable vpid" execution control is not enabled then the
39645e51299SNeel Natu 	 * VPID is required to be 0 for all vcpus.
39745e51299SNeel Natu 	 */
39845e51299SNeel Natu 	if ((procbased_ctls2 & PROCBASED2_ENABLE_VPID) == 0) {
39945e51299SNeel Natu 		for (i = 0; i < num; i++)
40045e51299SNeel Natu 			vpid[i] = 0;
40145e51299SNeel Natu 		return;
40245e51299SNeel Natu 	}
40345e51299SNeel Natu 
40445e51299SNeel Natu 	/*
40545e51299SNeel Natu 	 * Allocate a unique VPID for each vcpu from the unit number allocator.
40645e51299SNeel Natu 	 */
40745e51299SNeel Natu 	for (i = 0; i < num; i++) {
40845e51299SNeel Natu 		x = alloc_unr(vpid_unr);
40945e51299SNeel Natu 		if (x == -1)
41045e51299SNeel Natu 			break;
41145e51299SNeel Natu 		else
41245e51299SNeel Natu 			vpid[i] = x;
41345e51299SNeel Natu 	}
41445e51299SNeel Natu 
41545e51299SNeel Natu 	if (i < num) {
41645e51299SNeel Natu 		atomic_add_int(&vpid_alloc_failed, 1);
41745e51299SNeel Natu 
41845e51299SNeel Natu 		/*
41945e51299SNeel Natu 		 * If the unit number allocator does not have enough unique
42045e51299SNeel Natu 		 * VPIDs then we need to allocate from the [1,VM_MAXCPU] range.
42145e51299SNeel Natu 		 *
42245e51299SNeel Natu 		 * These VPIDs are not be unique across VMs but this does not
42345e51299SNeel Natu 		 * affect correctness because the combined mappings are also
42445e51299SNeel Natu 		 * tagged with the EP4TA which is unique for each VM.
42545e51299SNeel Natu 		 *
42645e51299SNeel Natu 		 * It is still sub-optimal because the invvpid will invalidate
42745e51299SNeel Natu 		 * combined mappings for a particular VPID across all EP4TAs.
42845e51299SNeel Natu 		 */
42945e51299SNeel Natu 		while (i-- > 0)
43045e51299SNeel Natu 			vpid_free(vpid[i]);
43145e51299SNeel Natu 
43245e51299SNeel Natu 		for (i = 0; i < num; i++)
43345e51299SNeel Natu 			vpid[i] = i + 1;
43445e51299SNeel Natu 	}
43545e51299SNeel Natu }
43645e51299SNeel Natu 
43745e51299SNeel Natu static void
43845e51299SNeel Natu vpid_init(void)
43945e51299SNeel Natu {
44045e51299SNeel Natu 	/*
44145e51299SNeel Natu 	 * VPID 0 is required when the "enable VPID" execution control is
44245e51299SNeel Natu 	 * disabled.
44345e51299SNeel Natu 	 *
44445e51299SNeel Natu 	 * VPIDs [1,VM_MAXCPU] are used as the "overflow namespace" when the
44545e51299SNeel Natu 	 * unit number allocator does not have sufficient unique VPIDs to
44645e51299SNeel Natu 	 * satisfy the allocation.
44745e51299SNeel Natu 	 *
44845e51299SNeel Natu 	 * The remaining VPIDs are managed by the unit number allocator.
44945e51299SNeel Natu 	 */
45045e51299SNeel Natu 	vpid_unr = new_unrhdr(VM_MAXCPU + 1, 0xffff, NULL);
45145e51299SNeel Natu }
45245e51299SNeel Natu 
45345e51299SNeel Natu static void
454366f6083SPeter Grehan msr_save_area_init(struct msr_entry *g_area, int *g_count)
455366f6083SPeter Grehan {
456366f6083SPeter Grehan 	int cnt;
457366f6083SPeter Grehan 
458366f6083SPeter Grehan 	static struct msr_entry guest_msrs[] = {
459366f6083SPeter Grehan 		{ MSR_KGSBASE, 0, 0 },
460366f6083SPeter Grehan 	};
461366f6083SPeter Grehan 
462366f6083SPeter Grehan 	cnt = sizeof(guest_msrs) / sizeof(guest_msrs[0]);
463366f6083SPeter Grehan 	if (cnt > GUEST_MSR_MAX_ENTRIES)
464366f6083SPeter Grehan 		panic("guest msr save area overrun");
465366f6083SPeter Grehan 	bcopy(guest_msrs, g_area, sizeof(guest_msrs));
466366f6083SPeter Grehan 	*g_count = cnt;
467366f6083SPeter Grehan }
468366f6083SPeter Grehan 
469366f6083SPeter Grehan static void
470366f6083SPeter Grehan vmx_disable(void *arg __unused)
471366f6083SPeter Grehan {
472366f6083SPeter Grehan 	struct invvpid_desc invvpid_desc = { 0 };
473366f6083SPeter Grehan 	struct invept_desc invept_desc = { 0 };
474366f6083SPeter Grehan 
475366f6083SPeter Grehan 	if (vmxon_enabled[curcpu]) {
476366f6083SPeter Grehan 		/*
477366f6083SPeter Grehan 		 * See sections 25.3.3.3 and 25.3.3.4 in Intel Vol 3b.
478366f6083SPeter Grehan 		 *
479366f6083SPeter Grehan 		 * VMXON or VMXOFF are not required to invalidate any TLB
480366f6083SPeter Grehan 		 * caching structures. This prevents potential retention of
481366f6083SPeter Grehan 		 * cached information in the TLB between distinct VMX episodes.
482366f6083SPeter Grehan 		 */
483366f6083SPeter Grehan 		invvpid(INVVPID_TYPE_ALL_CONTEXTS, invvpid_desc);
484366f6083SPeter Grehan 		invept(INVEPT_TYPE_ALL_CONTEXTS, invept_desc);
485366f6083SPeter Grehan 		vmxoff();
486366f6083SPeter Grehan 	}
487366f6083SPeter Grehan 	load_cr4(rcr4() & ~CR4_VMXE);
488366f6083SPeter Grehan }
489366f6083SPeter Grehan 
490366f6083SPeter Grehan static int
491366f6083SPeter Grehan vmx_cleanup(void)
492366f6083SPeter Grehan {
493366f6083SPeter Grehan 
494176666c2SNeel Natu 	if (pirvec != 0)
495176666c2SNeel Natu 		vmm_ipi_free(pirvec);
496176666c2SNeel Natu 
49745e51299SNeel Natu 	if (vpid_unr != NULL) {
49845e51299SNeel Natu 		delete_unrhdr(vpid_unr);
49945e51299SNeel Natu 		vpid_unr = NULL;
50045e51299SNeel Natu 	}
50145e51299SNeel Natu 
502366f6083SPeter Grehan 	smp_rendezvous(NULL, vmx_disable, NULL, NULL);
503366f6083SPeter Grehan 
504366f6083SPeter Grehan 	return (0);
505366f6083SPeter Grehan }
506366f6083SPeter Grehan 
507366f6083SPeter Grehan static void
508366f6083SPeter Grehan vmx_enable(void *arg __unused)
509366f6083SPeter Grehan {
510366f6083SPeter Grehan 	int error;
511366f6083SPeter Grehan 
512366f6083SPeter Grehan 	load_cr4(rcr4() | CR4_VMXE);
513366f6083SPeter Grehan 
514366f6083SPeter Grehan 	*(uint32_t *)vmxon_region[curcpu] = vmx_revision();
515366f6083SPeter Grehan 	error = vmxon(vmxon_region[curcpu]);
516366f6083SPeter Grehan 	if (error == 0)
517366f6083SPeter Grehan 		vmxon_enabled[curcpu] = 1;
518366f6083SPeter Grehan }
519366f6083SPeter Grehan 
52063e62d39SJohn Baldwin static void
52163e62d39SJohn Baldwin vmx_restore(void)
52263e62d39SJohn Baldwin {
52363e62d39SJohn Baldwin 
52463e62d39SJohn Baldwin 	if (vmxon_enabled[curcpu])
52563e62d39SJohn Baldwin 		vmxon(vmxon_region[curcpu]);
52663e62d39SJohn Baldwin }
52763e62d39SJohn Baldwin 
528366f6083SPeter Grehan static int
529add611fdSNeel Natu vmx_init(int ipinum)
530366f6083SPeter Grehan {
53188c4b8d1SNeel Natu 	int error, use_tpr_shadow;
5324bff7fadSNeel Natu 	uint64_t fixed0, fixed1, feature_control;
53388c4b8d1SNeel Natu 	uint32_t tmp, procbased2_vid_bits;
534366f6083SPeter Grehan 
535366f6083SPeter Grehan 	/* CPUID.1:ECX[bit 5] must be 1 for processor to support VMX */
5368b287612SJohn Baldwin 	if (!(cpu_feature2 & CPUID2_VMX)) {
537366f6083SPeter Grehan 		printf("vmx_init: processor does not support VMX operation\n");
538366f6083SPeter Grehan 		return (ENXIO);
539366f6083SPeter Grehan 	}
540366f6083SPeter Grehan 
5414bff7fadSNeel Natu 	/*
5424bff7fadSNeel Natu 	 * Verify that MSR_IA32_FEATURE_CONTROL lock and VMXON enable bits
5434bff7fadSNeel Natu 	 * are set (bits 0 and 2 respectively).
5444bff7fadSNeel Natu 	 */
5454bff7fadSNeel Natu 	feature_control = rdmsr(MSR_IA32_FEATURE_CONTROL);
546150369abSNeel Natu 	if ((feature_control & IA32_FEATURE_CONTROL_LOCK) == 0 ||
547150369abSNeel Natu 	    (feature_control & IA32_FEATURE_CONTROL_VMX_EN) == 0) {
5484bff7fadSNeel Natu 		printf("vmx_init: VMX operation disabled by BIOS\n");
5494bff7fadSNeel Natu 		return (ENXIO);
5504bff7fadSNeel Natu 	}
5514bff7fadSNeel Natu 
552366f6083SPeter Grehan 	/* Check support for primary processor-based VM-execution controls */
553366f6083SPeter Grehan 	error = vmx_set_ctlreg(MSR_VMX_PROCBASED_CTLS,
554366f6083SPeter Grehan 			       MSR_VMX_TRUE_PROCBASED_CTLS,
555366f6083SPeter Grehan 			       PROCBASED_CTLS_ONE_SETTING,
556366f6083SPeter Grehan 			       PROCBASED_CTLS_ZERO_SETTING, &procbased_ctls);
557366f6083SPeter Grehan 	if (error) {
558366f6083SPeter Grehan 		printf("vmx_init: processor does not support desired primary "
559366f6083SPeter Grehan 		       "processor-based controls\n");
560366f6083SPeter Grehan 		return (error);
561366f6083SPeter Grehan 	}
562366f6083SPeter Grehan 
563366f6083SPeter Grehan 	/* Clear the processor-based ctl bits that are set on demand */
564366f6083SPeter Grehan 	procbased_ctls &= ~PROCBASED_CTLS_WINDOW_SETTING;
565366f6083SPeter Grehan 
566366f6083SPeter Grehan 	/* Check support for secondary processor-based VM-execution controls */
567366f6083SPeter Grehan 	error = vmx_set_ctlreg(MSR_VMX_PROCBASED_CTLS2,
568366f6083SPeter Grehan 			       MSR_VMX_PROCBASED_CTLS2,
569366f6083SPeter Grehan 			       PROCBASED_CTLS2_ONE_SETTING,
570366f6083SPeter Grehan 			       PROCBASED_CTLS2_ZERO_SETTING, &procbased_ctls2);
571366f6083SPeter Grehan 	if (error) {
572366f6083SPeter Grehan 		printf("vmx_init: processor does not support desired secondary "
573366f6083SPeter Grehan 		       "processor-based controls\n");
574366f6083SPeter Grehan 		return (error);
575366f6083SPeter Grehan 	}
576366f6083SPeter Grehan 
577366f6083SPeter Grehan 	/* Check support for VPID */
578366f6083SPeter Grehan 	error = vmx_set_ctlreg(MSR_VMX_PROCBASED_CTLS2, MSR_VMX_PROCBASED_CTLS2,
579366f6083SPeter Grehan 			       PROCBASED2_ENABLE_VPID, 0, &tmp);
580366f6083SPeter Grehan 	if (error == 0)
581366f6083SPeter Grehan 		procbased_ctls2 |= PROCBASED2_ENABLE_VPID;
582366f6083SPeter Grehan 
583366f6083SPeter Grehan 	/* Check support for pin-based VM-execution controls */
584366f6083SPeter Grehan 	error = vmx_set_ctlreg(MSR_VMX_PINBASED_CTLS,
585366f6083SPeter Grehan 			       MSR_VMX_TRUE_PINBASED_CTLS,
586366f6083SPeter Grehan 			       PINBASED_CTLS_ONE_SETTING,
587366f6083SPeter Grehan 			       PINBASED_CTLS_ZERO_SETTING, &pinbased_ctls);
588366f6083SPeter Grehan 	if (error) {
589366f6083SPeter Grehan 		printf("vmx_init: processor does not support desired "
590366f6083SPeter Grehan 		       "pin-based controls\n");
591366f6083SPeter Grehan 		return (error);
592366f6083SPeter Grehan 	}
593366f6083SPeter Grehan 
594366f6083SPeter Grehan 	/* Check support for VM-exit controls */
595366f6083SPeter Grehan 	error = vmx_set_ctlreg(MSR_VMX_EXIT_CTLS, MSR_VMX_TRUE_EXIT_CTLS,
596366f6083SPeter Grehan 			       VM_EXIT_CTLS_ONE_SETTING,
597366f6083SPeter Grehan 			       VM_EXIT_CTLS_ZERO_SETTING,
598366f6083SPeter Grehan 			       &exit_ctls);
599366f6083SPeter Grehan 	if (error) {
600608f97c3SPeter Grehan 		/* Try again without the PAT MSR bits */
601608f97c3SPeter Grehan 		error = vmx_set_ctlreg(MSR_VMX_EXIT_CTLS,
602608f97c3SPeter Grehan 				       MSR_VMX_TRUE_EXIT_CTLS,
603608f97c3SPeter Grehan 				       VM_EXIT_CTLS_ONE_SETTING_NO_PAT,
604608f97c3SPeter Grehan 				       VM_EXIT_CTLS_ZERO_SETTING,
605608f97c3SPeter Grehan 				       &exit_ctls);
606608f97c3SPeter Grehan 		if (error) {
607366f6083SPeter Grehan 			printf("vmx_init: processor does not support desired "
608366f6083SPeter Grehan 			       "exit controls\n");
609366f6083SPeter Grehan 			return (error);
610608f97c3SPeter Grehan 		} else {
611608f97c3SPeter Grehan 			if (bootverbose)
612608f97c3SPeter Grehan 				printf("vmm: PAT MSR access not supported\n");
613608f97c3SPeter Grehan 			guest_msr_valid(MSR_PAT);
614608f97c3SPeter Grehan 			vmx_no_patmsr = 1;
615608f97c3SPeter Grehan 		}
616366f6083SPeter Grehan 	}
617366f6083SPeter Grehan 
618366f6083SPeter Grehan 	/* Check support for VM-entry controls */
619608f97c3SPeter Grehan 	if (!vmx_no_patmsr) {
620608f97c3SPeter Grehan 		error = vmx_set_ctlreg(MSR_VMX_ENTRY_CTLS,
621608f97c3SPeter Grehan 				       MSR_VMX_TRUE_ENTRY_CTLS,
622366f6083SPeter Grehan 				       VM_ENTRY_CTLS_ONE_SETTING,
623366f6083SPeter Grehan 				       VM_ENTRY_CTLS_ZERO_SETTING,
624366f6083SPeter Grehan 				       &entry_ctls);
625608f97c3SPeter Grehan 	} else {
626608f97c3SPeter Grehan 		error = vmx_set_ctlreg(MSR_VMX_ENTRY_CTLS,
627608f97c3SPeter Grehan 				       MSR_VMX_TRUE_ENTRY_CTLS,
628608f97c3SPeter Grehan 				       VM_ENTRY_CTLS_ONE_SETTING_NO_PAT,
629608f97c3SPeter Grehan 				       VM_ENTRY_CTLS_ZERO_SETTING,
630608f97c3SPeter Grehan 				       &entry_ctls);
631608f97c3SPeter Grehan 	}
632608f97c3SPeter Grehan 
633366f6083SPeter Grehan 	if (error) {
634366f6083SPeter Grehan 		printf("vmx_init: processor does not support desired "
635366f6083SPeter Grehan 		       "entry controls\n");
636366f6083SPeter Grehan 		       return (error);
637366f6083SPeter Grehan 	}
638366f6083SPeter Grehan 
639366f6083SPeter Grehan 	/*
640366f6083SPeter Grehan 	 * Check support for optional features by testing them
641366f6083SPeter Grehan 	 * as individual bits
642366f6083SPeter Grehan 	 */
643366f6083SPeter Grehan 	cap_halt_exit = (vmx_set_ctlreg(MSR_VMX_PROCBASED_CTLS,
644366f6083SPeter Grehan 					MSR_VMX_TRUE_PROCBASED_CTLS,
645366f6083SPeter Grehan 					PROCBASED_HLT_EXITING, 0,
646366f6083SPeter Grehan 					&tmp) == 0);
647366f6083SPeter Grehan 
648366f6083SPeter Grehan 	cap_monitor_trap = (vmx_set_ctlreg(MSR_VMX_PROCBASED_CTLS,
649366f6083SPeter Grehan 					MSR_VMX_PROCBASED_CTLS,
650366f6083SPeter Grehan 					PROCBASED_MTF, 0,
651366f6083SPeter Grehan 					&tmp) == 0);
652366f6083SPeter Grehan 
653366f6083SPeter Grehan 	cap_pause_exit = (vmx_set_ctlreg(MSR_VMX_PROCBASED_CTLS,
654366f6083SPeter Grehan 					 MSR_VMX_TRUE_PROCBASED_CTLS,
655366f6083SPeter Grehan 					 PROCBASED_PAUSE_EXITING, 0,
656366f6083SPeter Grehan 					 &tmp) == 0);
657366f6083SPeter Grehan 
658366f6083SPeter Grehan 	cap_unrestricted_guest = (vmx_set_ctlreg(MSR_VMX_PROCBASED_CTLS2,
659366f6083SPeter Grehan 					MSR_VMX_PROCBASED_CTLS2,
660366f6083SPeter Grehan 					PROCBASED2_UNRESTRICTED_GUEST, 0,
661366f6083SPeter Grehan 				        &tmp) == 0);
662366f6083SPeter Grehan 
66349cc03daSNeel Natu 	cap_invpcid = (vmx_set_ctlreg(MSR_VMX_PROCBASED_CTLS2,
66449cc03daSNeel Natu 	    MSR_VMX_PROCBASED_CTLS2, PROCBASED2_ENABLE_INVPCID, 0,
66549cc03daSNeel Natu 	    &tmp) == 0);
66649cc03daSNeel Natu 
66788c4b8d1SNeel Natu 	/*
66888c4b8d1SNeel Natu 	 * Check support for virtual interrupt delivery.
66988c4b8d1SNeel Natu 	 */
67088c4b8d1SNeel Natu 	procbased2_vid_bits = (PROCBASED2_VIRTUALIZE_APIC_ACCESSES |
67188c4b8d1SNeel Natu 	    PROCBASED2_VIRTUALIZE_X2APIC_MODE |
67288c4b8d1SNeel Natu 	    PROCBASED2_APIC_REGISTER_VIRTUALIZATION |
67388c4b8d1SNeel Natu 	    PROCBASED2_VIRTUAL_INTERRUPT_DELIVERY);
67488c4b8d1SNeel Natu 
67588c4b8d1SNeel Natu 	use_tpr_shadow = (vmx_set_ctlreg(MSR_VMX_PROCBASED_CTLS,
67688c4b8d1SNeel Natu 	    MSR_VMX_TRUE_PROCBASED_CTLS, PROCBASED_USE_TPR_SHADOW, 0,
67788c4b8d1SNeel Natu 	    &tmp) == 0);
67888c4b8d1SNeel Natu 
67988c4b8d1SNeel Natu 	error = vmx_set_ctlreg(MSR_VMX_PROCBASED_CTLS2, MSR_VMX_PROCBASED_CTLS2,
68088c4b8d1SNeel Natu 	    procbased2_vid_bits, 0, &tmp);
68188c4b8d1SNeel Natu 	if (error == 0 && use_tpr_shadow) {
68288c4b8d1SNeel Natu 		virtual_interrupt_delivery = 1;
68388c4b8d1SNeel Natu 		TUNABLE_INT_FETCH("hw.vmm.vmx.use_apic_vid",
68488c4b8d1SNeel Natu 		    &virtual_interrupt_delivery);
68588c4b8d1SNeel Natu 	}
68688c4b8d1SNeel Natu 
68788c4b8d1SNeel Natu 	if (virtual_interrupt_delivery) {
68888c4b8d1SNeel Natu 		procbased_ctls |= PROCBASED_USE_TPR_SHADOW;
68988c4b8d1SNeel Natu 		procbased_ctls2 |= procbased2_vid_bits;
69088c4b8d1SNeel Natu 		procbased_ctls2 &= ~PROCBASED2_VIRTUALIZE_X2APIC_MODE;
691176666c2SNeel Natu 
692176666c2SNeel Natu 		/*
693176666c2SNeel Natu 		 * Check for Posted Interrupts only if Virtual Interrupt
694176666c2SNeel Natu 		 * Delivery is enabled.
695176666c2SNeel Natu 		 */
696176666c2SNeel Natu 		error = vmx_set_ctlreg(MSR_VMX_PINBASED_CTLS,
697176666c2SNeel Natu 		    MSR_VMX_TRUE_PINBASED_CTLS, PINBASED_POSTED_INTERRUPT, 0,
698176666c2SNeel Natu 		    &tmp);
699176666c2SNeel Natu 		if (error == 0) {
700176666c2SNeel Natu 			pirvec = vmm_ipi_alloc();
701176666c2SNeel Natu 			if (pirvec == 0) {
702176666c2SNeel Natu 				if (bootverbose) {
703176666c2SNeel Natu 					printf("vmx_init: unable to allocate "
704176666c2SNeel Natu 					    "posted interrupt vector\n");
70588c4b8d1SNeel Natu 				}
706176666c2SNeel Natu 			} else {
707176666c2SNeel Natu 				posted_interrupts = 1;
708176666c2SNeel Natu 				TUNABLE_INT_FETCH("hw.vmm.vmx.use_apic_pir",
709176666c2SNeel Natu 				    &posted_interrupts);
710176666c2SNeel Natu 			}
711176666c2SNeel Natu 		}
712176666c2SNeel Natu 	}
713176666c2SNeel Natu 
714176666c2SNeel Natu 	if (posted_interrupts)
715176666c2SNeel Natu 		    pinbased_ctls |= PINBASED_POSTED_INTERRUPT;
71649cc03daSNeel Natu 
717366f6083SPeter Grehan 	/* Initialize EPT */
718add611fdSNeel Natu 	error = ept_init(ipinum);
719366f6083SPeter Grehan 	if (error) {
720366f6083SPeter Grehan 		printf("vmx_init: ept initialization failed (%d)\n", error);
721366f6083SPeter Grehan 		return (error);
722366f6083SPeter Grehan 	}
723366f6083SPeter Grehan 
724366f6083SPeter Grehan 	/*
725366f6083SPeter Grehan 	 * Stash the cr0 and cr4 bits that must be fixed to 0 or 1
726366f6083SPeter Grehan 	 */
727366f6083SPeter Grehan 	fixed0 = rdmsr(MSR_VMX_CR0_FIXED0);
728366f6083SPeter Grehan 	fixed1 = rdmsr(MSR_VMX_CR0_FIXED1);
729366f6083SPeter Grehan 	cr0_ones_mask = fixed0 & fixed1;
730366f6083SPeter Grehan 	cr0_zeros_mask = ~fixed0 & ~fixed1;
731366f6083SPeter Grehan 
732366f6083SPeter Grehan 	/*
733366f6083SPeter Grehan 	 * CR0_PE and CR0_PG can be set to zero in VMX non-root operation
734366f6083SPeter Grehan 	 * if unrestricted guest execution is allowed.
735366f6083SPeter Grehan 	 */
736366f6083SPeter Grehan 	if (cap_unrestricted_guest)
737366f6083SPeter Grehan 		cr0_ones_mask &= ~(CR0_PG | CR0_PE);
738366f6083SPeter Grehan 
739366f6083SPeter Grehan 	/*
740366f6083SPeter Grehan 	 * Do not allow the guest to set CR0_NW or CR0_CD.
741366f6083SPeter Grehan 	 */
742366f6083SPeter Grehan 	cr0_zeros_mask |= (CR0_NW | CR0_CD);
743366f6083SPeter Grehan 
744366f6083SPeter Grehan 	fixed0 = rdmsr(MSR_VMX_CR4_FIXED0);
745366f6083SPeter Grehan 	fixed1 = rdmsr(MSR_VMX_CR4_FIXED1);
746366f6083SPeter Grehan 	cr4_ones_mask = fixed0 & fixed1;
747366f6083SPeter Grehan 	cr4_zeros_mask = ~fixed0 & ~fixed1;
748366f6083SPeter Grehan 
74945e51299SNeel Natu 	vpid_init();
75045e51299SNeel Natu 
751366f6083SPeter Grehan 	/* enable VMX operation */
752366f6083SPeter Grehan 	smp_rendezvous(NULL, vmx_enable, NULL, NULL);
753366f6083SPeter Grehan 
7543565b59eSNeel Natu 	vmx_initialized = 1;
7553565b59eSNeel Natu 
756366f6083SPeter Grehan 	return (0);
757366f6083SPeter Grehan }
758366f6083SPeter Grehan 
759f7d47425SNeel Natu static void
760f7d47425SNeel Natu vmx_trigger_hostintr(int vector)
761f7d47425SNeel Natu {
762f7d47425SNeel Natu 	uintptr_t func;
763f7d47425SNeel Natu 	struct gate_descriptor *gd;
764f7d47425SNeel Natu 
765f7d47425SNeel Natu 	gd = &idt[vector];
766f7d47425SNeel Natu 
767f7d47425SNeel Natu 	KASSERT(vector >= 32 && vector <= 255, ("vmx_trigger_hostintr: "
768f7d47425SNeel Natu 	    "invalid vector %d", vector));
769f7d47425SNeel Natu 	KASSERT(gd->gd_p == 1, ("gate descriptor for vector %d not present",
770f7d47425SNeel Natu 	    vector));
771f7d47425SNeel Natu 	KASSERT(gd->gd_type == SDT_SYSIGT, ("gate descriptor for vector %d "
772f7d47425SNeel Natu 	    "has invalid type %d", vector, gd->gd_type));
773f7d47425SNeel Natu 	KASSERT(gd->gd_dpl == SEL_KPL, ("gate descriptor for vector %d "
774f7d47425SNeel Natu 	    "has invalid dpl %d", vector, gd->gd_dpl));
775f7d47425SNeel Natu 	KASSERT(gd->gd_selector == GSEL(GCODE_SEL, SEL_KPL), ("gate descriptor "
776f7d47425SNeel Natu 	    "for vector %d has invalid selector %d", vector, gd->gd_selector));
777f7d47425SNeel Natu 	KASSERT(gd->gd_ist == 0, ("gate descriptor for vector %d has invalid "
778f7d47425SNeel Natu 	    "IST %d", vector, gd->gd_ist));
779f7d47425SNeel Natu 
780f7d47425SNeel Natu 	func = ((long)gd->gd_hioffset << 16 | gd->gd_looffset);
781f7d47425SNeel Natu 	vmx_call_isr(func);
782f7d47425SNeel Natu }
783f7d47425SNeel Natu 
784366f6083SPeter Grehan static int
785aaaa0656SPeter Grehan vmx_setup_cr_shadow(int which, struct vmcs *vmcs, uint32_t initial)
786366f6083SPeter Grehan {
78739c21c2dSNeel Natu 	int error, mask_ident, shadow_ident;
788aaaa0656SPeter Grehan 	uint64_t mask_value;
789366f6083SPeter Grehan 
79039c21c2dSNeel Natu 	if (which != 0 && which != 4)
79139c21c2dSNeel Natu 		panic("vmx_setup_cr_shadow: unknown cr%d", which);
79239c21c2dSNeel Natu 
79339c21c2dSNeel Natu 	if (which == 0) {
79439c21c2dSNeel Natu 		mask_ident = VMCS_CR0_MASK;
79539c21c2dSNeel Natu 		mask_value = cr0_ones_mask | cr0_zeros_mask;
79639c21c2dSNeel Natu 		shadow_ident = VMCS_CR0_SHADOW;
79739c21c2dSNeel Natu 	} else {
79839c21c2dSNeel Natu 		mask_ident = VMCS_CR4_MASK;
79939c21c2dSNeel Natu 		mask_value = cr4_ones_mask | cr4_zeros_mask;
80039c21c2dSNeel Natu 		shadow_ident = VMCS_CR4_SHADOW;
80139c21c2dSNeel Natu 	}
80239c21c2dSNeel Natu 
803d3c11f40SPeter Grehan 	error = vmcs_setreg(vmcs, 0, VMCS_IDENT(mask_ident), mask_value);
804366f6083SPeter Grehan 	if (error)
805366f6083SPeter Grehan 		return (error);
806366f6083SPeter Grehan 
807aaaa0656SPeter Grehan 	error = vmcs_setreg(vmcs, 0, VMCS_IDENT(shadow_ident), initial);
808366f6083SPeter Grehan 	if (error)
809366f6083SPeter Grehan 		return (error);
810366f6083SPeter Grehan 
811366f6083SPeter Grehan 	return (0);
812366f6083SPeter Grehan }
813aaaa0656SPeter Grehan #define	vmx_setup_cr0_shadow(vmcs,init)	vmx_setup_cr_shadow(0, (vmcs), (init))
814aaaa0656SPeter Grehan #define	vmx_setup_cr4_shadow(vmcs,init)	vmx_setup_cr_shadow(4, (vmcs), (init))
815366f6083SPeter Grehan 
816366f6083SPeter Grehan static void *
817318224bbSNeel Natu vmx_vminit(struct vm *vm, pmap_t pmap)
818366f6083SPeter Grehan {
81945e51299SNeel Natu 	uint16_t vpid[VM_MAXCPU];
820366f6083SPeter Grehan 	int i, error, guest_msr_count;
821366f6083SPeter Grehan 	struct vmx *vmx;
822c847a506SNeel Natu 	struct vmcs *vmcs;
823366f6083SPeter Grehan 
824366f6083SPeter Grehan 	vmx = malloc(sizeof(struct vmx), M_VMX, M_WAITOK | M_ZERO);
825366f6083SPeter Grehan 	if ((uintptr_t)vmx & PAGE_MASK) {
826366f6083SPeter Grehan 		panic("malloc of struct vmx not aligned on %d byte boundary",
827366f6083SPeter Grehan 		      PAGE_SIZE);
828366f6083SPeter Grehan 	}
829366f6083SPeter Grehan 	vmx->vm = vm;
830366f6083SPeter Grehan 
831318224bbSNeel Natu 	vmx->eptp = eptp(vtophys((vm_offset_t)pmap->pm_pml4));
832318224bbSNeel Natu 
833366f6083SPeter Grehan 	/*
834366f6083SPeter Grehan 	 * Clean up EPTP-tagged guest physical and combined mappings
835366f6083SPeter Grehan 	 *
836366f6083SPeter Grehan 	 * VMX transitions are not required to invalidate any guest physical
837366f6083SPeter Grehan 	 * mappings. So, it may be possible for stale guest physical mappings
838366f6083SPeter Grehan 	 * to be present in the processor TLBs.
839366f6083SPeter Grehan 	 *
840366f6083SPeter Grehan 	 * Combined mappings for this EP4TA are also invalidated for all VPIDs.
841366f6083SPeter Grehan 	 */
842318224bbSNeel Natu 	ept_invalidate_mappings(vmx->eptp);
843366f6083SPeter Grehan 
844366f6083SPeter Grehan 	msr_bitmap_initialize(vmx->msr_bitmap);
845366f6083SPeter Grehan 
846366f6083SPeter Grehan 	/*
847366f6083SPeter Grehan 	 * It is safe to allow direct access to MSR_GSBASE and MSR_FSBASE.
848366f6083SPeter Grehan 	 * The guest FSBASE and GSBASE are saved and restored during
849366f6083SPeter Grehan 	 * vm-exit and vm-entry respectively. The host FSBASE and GSBASE are
850366f6083SPeter Grehan 	 * always restored from the vmcs host state area on vm-exit.
851366f6083SPeter Grehan 	 *
8521fb0ea3fSPeter Grehan 	 * The SYSENTER_CS/ESP/EIP MSRs are identical to FS/GSBASE in
8531fb0ea3fSPeter Grehan 	 * how they are saved/restored so can be directly accessed by the
8541fb0ea3fSPeter Grehan 	 * guest.
8551fb0ea3fSPeter Grehan 	 *
856366f6083SPeter Grehan 	 * Guest KGSBASE is saved and restored in the guest MSR save area.
857366f6083SPeter Grehan 	 * Host KGSBASE is restored before returning to userland from the pcb.
858366f6083SPeter Grehan 	 * There will be a window of time when we are executing in the host
859366f6083SPeter Grehan 	 * kernel context with a value of KGSBASE from the guest. This is ok
860366f6083SPeter Grehan 	 * because the value of KGSBASE is inconsequential in kernel context.
861366f6083SPeter Grehan 	 *
862366f6083SPeter Grehan 	 * MSR_EFER is saved and restored in the guest VMCS area on a
863366f6083SPeter Grehan 	 * VM exit and entry respectively. It is also restored from the
864366f6083SPeter Grehan 	 * host VMCS area on a VM exit.
865366f6083SPeter Grehan 	 */
866366f6083SPeter Grehan 	if (guest_msr_rw(vmx, MSR_GSBASE) ||
867366f6083SPeter Grehan 	    guest_msr_rw(vmx, MSR_FSBASE) ||
8681fb0ea3fSPeter Grehan 	    guest_msr_rw(vmx, MSR_SYSENTER_CS_MSR) ||
8691fb0ea3fSPeter Grehan 	    guest_msr_rw(vmx, MSR_SYSENTER_ESP_MSR) ||
8701fb0ea3fSPeter Grehan 	    guest_msr_rw(vmx, MSR_SYSENTER_EIP_MSR) ||
871366f6083SPeter Grehan 	    guest_msr_rw(vmx, MSR_KGSBASE) ||
872608f97c3SPeter Grehan 	    guest_msr_rw(vmx, MSR_EFER))
873366f6083SPeter Grehan 		panic("vmx_vminit: error setting guest msr access");
874366f6083SPeter Grehan 
875608f97c3SPeter Grehan 	/*
876608f97c3SPeter Grehan 	 * MSR_PAT is saved and restored in the guest VMCS are on a VM exit
877608f97c3SPeter Grehan 	 * and entry respectively. It is also restored from the host VMCS
878608f97c3SPeter Grehan 	 * area on a VM exit. However, if running on a system with no
879608f97c3SPeter Grehan 	 * MSR_PAT save/restore support, leave access disabled so accesses
880608f97c3SPeter Grehan 	 * will be trapped.
881608f97c3SPeter Grehan 	 */
882608f97c3SPeter Grehan 	if (!vmx_no_patmsr && guest_msr_rw(vmx, MSR_PAT))
883608f97c3SPeter Grehan 		panic("vmx_vminit: error setting guest pat msr access");
884608f97c3SPeter Grehan 
88545e51299SNeel Natu 	vpid_alloc(vpid, VM_MAXCPU);
88645e51299SNeel Natu 
88788c4b8d1SNeel Natu 	if (virtual_interrupt_delivery) {
88888c4b8d1SNeel Natu 		error = vm_map_mmio(vm, DEFAULT_APIC_BASE, PAGE_SIZE,
88988c4b8d1SNeel Natu 		    APIC_ACCESS_ADDRESS);
89088c4b8d1SNeel Natu 		/* XXX this should really return an error to the caller */
89188c4b8d1SNeel Natu 		KASSERT(error == 0, ("vm_map_mmio(apicbase) error %d", error));
89288c4b8d1SNeel Natu 	}
89388c4b8d1SNeel Natu 
894366f6083SPeter Grehan 	for (i = 0; i < VM_MAXCPU; i++) {
895c847a506SNeel Natu 		vmcs = &vmx->vmcs[i];
896c847a506SNeel Natu 		vmcs->identifier = vmx_revision();
897c847a506SNeel Natu 		error = vmclear(vmcs);
898366f6083SPeter Grehan 		if (error != 0) {
899366f6083SPeter Grehan 			panic("vmx_vminit: vmclear error %d on vcpu %d\n",
900366f6083SPeter Grehan 			      error, i);
901366f6083SPeter Grehan 		}
902366f6083SPeter Grehan 
903c847a506SNeel Natu 		error = vmcs_init(vmcs);
904c847a506SNeel Natu 		KASSERT(error == 0, ("vmcs_init error %d", error));
905366f6083SPeter Grehan 
906c847a506SNeel Natu 		VMPTRLD(vmcs);
907c847a506SNeel Natu 		error = 0;
908c847a506SNeel Natu 		error += vmwrite(VMCS_HOST_RSP, (u_long)&vmx->ctx[i]);
909c847a506SNeel Natu 		error += vmwrite(VMCS_EPTP, vmx->eptp);
910c847a506SNeel Natu 		error += vmwrite(VMCS_PIN_BASED_CTLS, pinbased_ctls);
911c847a506SNeel Natu 		error += vmwrite(VMCS_PRI_PROC_BASED_CTLS, procbased_ctls);
912c847a506SNeel Natu 		error += vmwrite(VMCS_SEC_PROC_BASED_CTLS, procbased_ctls2);
913c847a506SNeel Natu 		error += vmwrite(VMCS_EXIT_CTLS, exit_ctls);
914c847a506SNeel Natu 		error += vmwrite(VMCS_ENTRY_CTLS, entry_ctls);
915c847a506SNeel Natu 		error += vmwrite(VMCS_MSR_BITMAP, vtophys(vmx->msr_bitmap));
916c847a506SNeel Natu 		error += vmwrite(VMCS_VPID, vpid[i]);
91788c4b8d1SNeel Natu 		if (virtual_interrupt_delivery) {
91888c4b8d1SNeel Natu 			error += vmwrite(VMCS_APIC_ACCESS, APIC_ACCESS_ADDRESS);
91988c4b8d1SNeel Natu 			error += vmwrite(VMCS_VIRTUAL_APIC,
92088c4b8d1SNeel Natu 			    vtophys(&vmx->apic_page[i]));
92188c4b8d1SNeel Natu 			error += vmwrite(VMCS_EOI_EXIT0, 0);
92288c4b8d1SNeel Natu 			error += vmwrite(VMCS_EOI_EXIT1, 0);
92388c4b8d1SNeel Natu 			error += vmwrite(VMCS_EOI_EXIT2, 0);
92488c4b8d1SNeel Natu 			error += vmwrite(VMCS_EOI_EXIT3, 0);
92588c4b8d1SNeel Natu 		}
926176666c2SNeel Natu 		if (posted_interrupts) {
927176666c2SNeel Natu 			error += vmwrite(VMCS_PIR_VECTOR, pirvec);
928176666c2SNeel Natu 			error += vmwrite(VMCS_PIR_DESC,
929176666c2SNeel Natu 			    vtophys(&vmx->pir_desc[i]));
930176666c2SNeel Natu 		}
931c847a506SNeel Natu 		VMCLEAR(vmcs);
932c847a506SNeel Natu 		KASSERT(error == 0, ("vmx_vminit: error customizing the vmcs"));
933366f6083SPeter Grehan 
934366f6083SPeter Grehan 		vmx->cap[i].set = 0;
935366f6083SPeter Grehan 		vmx->cap[i].proc_ctls = procbased_ctls;
93649cc03daSNeel Natu 		vmx->cap[i].proc_ctls2 = procbased_ctls2;
937366f6083SPeter Grehan 
938366f6083SPeter Grehan 		vmx->state[i].lastcpu = -1;
93945e51299SNeel Natu 		vmx->state[i].vpid = vpid[i];
940366f6083SPeter Grehan 
941366f6083SPeter Grehan 		msr_save_area_init(vmx->guest_msrs[i], &guest_msr_count);
942366f6083SPeter Grehan 
943c847a506SNeel Natu 		error = vmcs_set_msr_save(vmcs, vtophys(vmx->guest_msrs[i]),
944366f6083SPeter Grehan 		    guest_msr_count);
945366f6083SPeter Grehan 		if (error != 0)
946366f6083SPeter Grehan 			panic("vmcs_set_msr_save error %d", error);
947366f6083SPeter Grehan 
948aaaa0656SPeter Grehan 		/*
949aaaa0656SPeter Grehan 		 * Set up the CR0/4 shadows, and init the read shadow
950aaaa0656SPeter Grehan 		 * to the power-on register value from the Intel Sys Arch.
951aaaa0656SPeter Grehan 		 *  CR0 - 0x60000010
952aaaa0656SPeter Grehan 		 *  CR4 - 0
953aaaa0656SPeter Grehan 		 */
954c847a506SNeel Natu 		error = vmx_setup_cr0_shadow(vmcs, 0x60000010);
95539c21c2dSNeel Natu 		if (error != 0)
95639c21c2dSNeel Natu 			panic("vmx_setup_cr0_shadow %d", error);
95739c21c2dSNeel Natu 
958c847a506SNeel Natu 		error = vmx_setup_cr4_shadow(vmcs, 0);
95939c21c2dSNeel Natu 		if (error != 0)
96039c21c2dSNeel Natu 			panic("vmx_setup_cr4_shadow %d", error);
961318224bbSNeel Natu 
962318224bbSNeel Natu 		vmx->ctx[i].pmap = pmap;
963366f6083SPeter Grehan 	}
964366f6083SPeter Grehan 
965366f6083SPeter Grehan 	return (vmx);
966366f6083SPeter Grehan }
967366f6083SPeter Grehan 
968366f6083SPeter Grehan static int
969a2da7af6SNeel Natu vmx_handle_cpuid(struct vm *vm, int vcpu, struct vmxctx *vmxctx)
970366f6083SPeter Grehan {
971366f6083SPeter Grehan 	int handled, func;
972366f6083SPeter Grehan 
973366f6083SPeter Grehan 	func = vmxctx->guest_rax;
974366f6083SPeter Grehan 
975a2da7af6SNeel Natu 	handled = x86_emulate_cpuid(vm, vcpu,
976a2da7af6SNeel Natu 				    (uint32_t*)(&vmxctx->guest_rax),
977a2da7af6SNeel Natu 				    (uint32_t*)(&vmxctx->guest_rbx),
978a2da7af6SNeel Natu 				    (uint32_t*)(&vmxctx->guest_rcx),
979a2da7af6SNeel Natu 				    (uint32_t*)(&vmxctx->guest_rdx));
980366f6083SPeter Grehan 	return (handled);
981366f6083SPeter Grehan }
982366f6083SPeter Grehan 
983366f6083SPeter Grehan static __inline void
984366f6083SPeter Grehan vmx_run_trace(struct vmx *vmx, int vcpu)
985366f6083SPeter Grehan {
986366f6083SPeter Grehan #ifdef KTR
987513c8d33SNeel Natu 	VCPU_CTR1(vmx->vm, vcpu, "Resume execution at %#lx", vmcs_guest_rip());
988366f6083SPeter Grehan #endif
989366f6083SPeter Grehan }
990366f6083SPeter Grehan 
991366f6083SPeter Grehan static __inline void
992366f6083SPeter Grehan vmx_exit_trace(struct vmx *vmx, int vcpu, uint64_t rip, uint32_t exit_reason,
993eeefa4e4SNeel Natu 	       int handled)
994366f6083SPeter Grehan {
995366f6083SPeter Grehan #ifdef KTR
996513c8d33SNeel Natu 	VCPU_CTR3(vmx->vm, vcpu, "%s %s vmexit at 0x%0lx",
997366f6083SPeter Grehan 		 handled ? "handled" : "unhandled",
998366f6083SPeter Grehan 		 exit_reason_to_str(exit_reason), rip);
999eeefa4e4SNeel Natu #endif
1000eeefa4e4SNeel Natu }
1001366f6083SPeter Grehan 
1002eeefa4e4SNeel Natu static __inline void
1003eeefa4e4SNeel Natu vmx_astpending_trace(struct vmx *vmx, int vcpu, uint64_t rip)
1004eeefa4e4SNeel Natu {
1005eeefa4e4SNeel Natu #ifdef KTR
1006513c8d33SNeel Natu 	VCPU_CTR1(vmx->vm, vcpu, "astpending vmexit at 0x%0lx", rip);
1007366f6083SPeter Grehan #endif
1008366f6083SPeter Grehan }
1009366f6083SPeter Grehan 
1010953c2c47SNeel Natu static VMM_STAT_INTEL(VCPU_INVVPID_SAVED, "Number of vpid invalidations saved");
1011953c2c47SNeel Natu 
10123de83862SNeel Natu static void
1013953c2c47SNeel Natu vmx_set_pcpu_defaults(struct vmx *vmx, int vcpu, pmap_t pmap)
1014366f6083SPeter Grehan {
1015366f6083SPeter Grehan 	struct vmxstate *vmxstate;
1016953c2c47SNeel Natu 	struct invvpid_desc invvpid_desc;
1017366f6083SPeter Grehan 
1018366f6083SPeter Grehan 	vmxstate = &vmx->state[vcpu];
1019953c2c47SNeel Natu 	if (vmxstate->lastcpu == curcpu)
10203de83862SNeel Natu 		return;
1021366f6083SPeter Grehan 
1022953c2c47SNeel Natu 	vmxstate->lastcpu = curcpu;
1023953c2c47SNeel Natu 
1024366f6083SPeter Grehan 	vmm_stat_incr(vmx->vm, vcpu, VCPU_MIGRATIONS, 1);
1025366f6083SPeter Grehan 
10263de83862SNeel Natu 	vmcs_write(VMCS_HOST_TR_BASE, vmm_get_host_trbase());
10273de83862SNeel Natu 	vmcs_write(VMCS_HOST_GDTR_BASE, vmm_get_host_gdtrbase());
10283de83862SNeel Natu 	vmcs_write(VMCS_HOST_GS_BASE, vmm_get_host_gsbase());
1029366f6083SPeter Grehan 
1030366f6083SPeter Grehan 	/*
1031366f6083SPeter Grehan 	 * If we are using VPIDs then invalidate all mappings tagged with 'vpid'
1032366f6083SPeter Grehan 	 *
1033366f6083SPeter Grehan 	 * We do this because this vcpu was executing on a different host
1034366f6083SPeter Grehan 	 * cpu when it last ran. We do not track whether it invalidated
1035366f6083SPeter Grehan 	 * mappings associated with its 'vpid' during that run. So we must
1036366f6083SPeter Grehan 	 * assume that the mappings associated with 'vpid' on 'curcpu' are
1037366f6083SPeter Grehan 	 * stale and invalidate them.
1038366f6083SPeter Grehan 	 *
1039366f6083SPeter Grehan 	 * Note that we incur this penalty only when the scheduler chooses to
1040366f6083SPeter Grehan 	 * move the thread associated with this vcpu between host cpus.
1041366f6083SPeter Grehan 	 *
1042366f6083SPeter Grehan 	 * Note also that this will invalidate mappings tagged with 'vpid'
1043366f6083SPeter Grehan 	 * for "all" EP4TAs.
1044366f6083SPeter Grehan 	 */
1045366f6083SPeter Grehan 	if (vmxstate->vpid != 0) {
1046953c2c47SNeel Natu 		if (pmap->pm_eptgen == vmx->eptgen[curcpu]) {
1047953c2c47SNeel Natu 			invvpid_desc._res1 = 0;
1048953c2c47SNeel Natu 			invvpid_desc._res2 = 0;
1049366f6083SPeter Grehan 			invvpid_desc.vpid = vmxstate->vpid;
1050366f6083SPeter Grehan 			invvpid(INVVPID_TYPE_SINGLE_CONTEXT, invvpid_desc);
1051953c2c47SNeel Natu 		} else {
1052953c2c47SNeel Natu 			/*
1053953c2c47SNeel Natu 			 * The invvpid can be skipped if an invept is going to
1054953c2c47SNeel Natu 			 * be performed before entering the guest. The invept
1055953c2c47SNeel Natu 			 * will invalidate combined mappings tagged with
1056953c2c47SNeel Natu 			 * 'vmx->eptp' for all vpids.
1057953c2c47SNeel Natu 			 */
1058953c2c47SNeel Natu 			vmm_stat_incr(vmx->vm, vcpu, VCPU_INVVPID_SAVED, 1);
1059953c2c47SNeel Natu 		}
1060366f6083SPeter Grehan 	}
1061366f6083SPeter Grehan }
1062366f6083SPeter Grehan 
1063366f6083SPeter Grehan /*
1064366f6083SPeter Grehan  * We depend on 'procbased_ctls' to have the Interrupt Window Exiting bit set.
1065366f6083SPeter Grehan  */
1066366f6083SPeter Grehan CTASSERT((PROCBASED_CTLS_ONE_SETTING & PROCBASED_INT_WINDOW_EXITING) != 0);
1067366f6083SPeter Grehan 
1068366f6083SPeter Grehan static void __inline
1069366f6083SPeter Grehan vmx_set_int_window_exiting(struct vmx *vmx, int vcpu)
1070366f6083SPeter Grehan {
1071366f6083SPeter Grehan 
107248b2d828SNeel Natu 	if ((vmx->cap[vcpu].proc_ctls & PROCBASED_INT_WINDOW_EXITING) == 0) {
1073366f6083SPeter Grehan 		vmx->cap[vcpu].proc_ctls |= PROCBASED_INT_WINDOW_EXITING;
10743de83862SNeel Natu 		vmcs_write(VMCS_PRI_PROC_BASED_CTLS, vmx->cap[vcpu].proc_ctls);
107548b2d828SNeel Natu 		VCPU_CTR0(vmx->vm, vcpu, "Enabling interrupt window exiting");
107648b2d828SNeel Natu 	}
1077366f6083SPeter Grehan }
1078366f6083SPeter Grehan 
1079366f6083SPeter Grehan static void __inline
1080366f6083SPeter Grehan vmx_clear_int_window_exiting(struct vmx *vmx, int vcpu)
1081366f6083SPeter Grehan {
1082366f6083SPeter Grehan 
108348b2d828SNeel Natu 	KASSERT((vmx->cap[vcpu].proc_ctls & PROCBASED_INT_WINDOW_EXITING) != 0,
108448b2d828SNeel Natu 	    ("intr_window_exiting not set: %#x", vmx->cap[vcpu].proc_ctls));
1085366f6083SPeter Grehan 	vmx->cap[vcpu].proc_ctls &= ~PROCBASED_INT_WINDOW_EXITING;
10863de83862SNeel Natu 	vmcs_write(VMCS_PRI_PROC_BASED_CTLS, vmx->cap[vcpu].proc_ctls);
108748b2d828SNeel Natu 	VCPU_CTR0(vmx->vm, vcpu, "Disabling interrupt window exiting");
1088366f6083SPeter Grehan }
1089366f6083SPeter Grehan 
1090366f6083SPeter Grehan static void __inline
1091366f6083SPeter Grehan vmx_set_nmi_window_exiting(struct vmx *vmx, int vcpu)
1092366f6083SPeter Grehan {
1093366f6083SPeter Grehan 
109448b2d828SNeel Natu 	if ((vmx->cap[vcpu].proc_ctls & PROCBASED_NMI_WINDOW_EXITING) == 0) {
1095366f6083SPeter Grehan 		vmx->cap[vcpu].proc_ctls |= PROCBASED_NMI_WINDOW_EXITING;
10963de83862SNeel Natu 		vmcs_write(VMCS_PRI_PROC_BASED_CTLS, vmx->cap[vcpu].proc_ctls);
109748b2d828SNeel Natu 		VCPU_CTR0(vmx->vm, vcpu, "Enabling NMI window exiting");
109848b2d828SNeel Natu 	}
1099366f6083SPeter Grehan }
1100366f6083SPeter Grehan 
1101366f6083SPeter Grehan static void __inline
1102366f6083SPeter Grehan vmx_clear_nmi_window_exiting(struct vmx *vmx, int vcpu)
1103366f6083SPeter Grehan {
1104366f6083SPeter Grehan 
110548b2d828SNeel Natu 	KASSERT((vmx->cap[vcpu].proc_ctls & PROCBASED_NMI_WINDOW_EXITING) != 0,
110648b2d828SNeel Natu 	    ("nmi_window_exiting not set %#x", vmx->cap[vcpu].proc_ctls));
1107366f6083SPeter Grehan 	vmx->cap[vcpu].proc_ctls &= ~PROCBASED_NMI_WINDOW_EXITING;
11083de83862SNeel Natu 	vmcs_write(VMCS_PRI_PROC_BASED_CTLS, vmx->cap[vcpu].proc_ctls);
110948b2d828SNeel Natu 	VCPU_CTR0(vmx->vm, vcpu, "Disabling NMI window exiting");
1110366f6083SPeter Grehan }
1111366f6083SPeter Grehan 
111248b2d828SNeel Natu #define	NMI_BLOCKING	(VMCS_INTERRUPTIBILITY_NMI_BLOCKING |		\
111348b2d828SNeel Natu 			 VMCS_INTERRUPTIBILITY_MOVSS_BLOCKING)
111448b2d828SNeel Natu #define	HWINTR_BLOCKING	(VMCS_INTERRUPTIBILITY_STI_BLOCKING |		\
111548b2d828SNeel Natu 			 VMCS_INTERRUPTIBILITY_MOVSS_BLOCKING)
111648b2d828SNeel Natu 
111748b2d828SNeel Natu static void
1118366f6083SPeter Grehan vmx_inject_nmi(struct vmx *vmx, int vcpu)
1119366f6083SPeter Grehan {
112048b2d828SNeel Natu 	uint32_t gi, info;
1121366f6083SPeter Grehan 
112248b2d828SNeel Natu 	gi = vmcs_read(VMCS_GUEST_INTERRUPTIBILITY);
112348b2d828SNeel Natu 	KASSERT((gi & NMI_BLOCKING) == 0, ("vmx_inject_nmi: invalid guest "
112448b2d828SNeel Natu 	    "interruptibility-state %#x", gi));
1125366f6083SPeter Grehan 
112648b2d828SNeel Natu 	info = vmcs_read(VMCS_ENTRY_INTR_INFO);
112748b2d828SNeel Natu 	KASSERT((info & VMCS_INTR_VALID) == 0, ("vmx_inject_nmi: invalid "
112848b2d828SNeel Natu 	    "VM-entry interruption information %#x", info));
1129366f6083SPeter Grehan 
1130366f6083SPeter Grehan 	/*
1131366f6083SPeter Grehan 	 * Inject the virtual NMI. The vector must be the NMI IDT entry
1132366f6083SPeter Grehan 	 * or the VMCS entry check will fail.
1133366f6083SPeter Grehan 	 */
113448b2d828SNeel Natu 	info = IDT_NMI | VMCS_INTR_T_NMI | VMCS_INTR_VALID;
11353de83862SNeel Natu 	vmcs_write(VMCS_ENTRY_INTR_INFO, info);
1136366f6083SPeter Grehan 
1137513c8d33SNeel Natu 	VCPU_CTR0(vmx->vm, vcpu, "Injecting vNMI");
1138366f6083SPeter Grehan 
1139366f6083SPeter Grehan 	/* Clear the request */
1140f352ff0cSNeel Natu 	vm_nmi_clear(vmx->vm, vcpu);
1141366f6083SPeter Grehan }
1142366f6083SPeter Grehan 
1143366f6083SPeter Grehan static void
1144de5ea6b6SNeel Natu vmx_inject_interrupts(struct vmx *vmx, int vcpu, struct vlapic *vlapic)
1145366f6083SPeter Grehan {
1146dc506506SNeel Natu 	struct vm_exception exc;
114748b2d828SNeel Natu 	int vector, need_nmi_exiting;
114848b2d828SNeel Natu 	uint64_t rflags;
114948b2d828SNeel Natu 	uint32_t gi, info;
1150366f6083SPeter Grehan 
1151dc506506SNeel Natu 	if (vm_exception_pending(vmx->vm, vcpu, &exc)) {
1152dc506506SNeel Natu 		KASSERT(exc.vector >= 0 && exc.vector < 32,
1153dc506506SNeel Natu 		    ("%s: invalid exception vector %d", __func__, exc.vector));
1154dc506506SNeel Natu 
1155dc506506SNeel Natu 		info = vmcs_read(VMCS_ENTRY_INTR_INFO);
1156dc506506SNeel Natu 		KASSERT((info & VMCS_INTR_VALID) == 0, ("%s: cannot inject "
1157dc506506SNeel Natu 		     "pending exception %d: %#x", __func__, exc.vector, info));
1158dc506506SNeel Natu 
1159dc506506SNeel Natu 		info = exc.vector | VMCS_INTR_T_HWEXCEPTION | VMCS_INTR_VALID;
1160dc506506SNeel Natu 		if (exc.error_code_valid) {
1161dc506506SNeel Natu 			info |= VMCS_INTR_DEL_ERRCODE;
1162dc506506SNeel Natu 			vmcs_write(VMCS_ENTRY_EXCEPTION_ERROR, exc.error_code);
1163dc506506SNeel Natu 		}
1164dc506506SNeel Natu 		vmcs_write(VMCS_ENTRY_INTR_INFO, info);
1165dc506506SNeel Natu 	}
1166dc506506SNeel Natu 
116748b2d828SNeel Natu 	if (vm_nmi_pending(vmx->vm, vcpu)) {
1168366f6083SPeter Grehan 		/*
116948b2d828SNeel Natu 		 * If there are no conditions blocking NMI injection then
117048b2d828SNeel Natu 		 * inject it directly here otherwise enable "NMI window
117148b2d828SNeel Natu 		 * exiting" to inject it as soon as we can.
1172eeefa4e4SNeel Natu 		 *
117348b2d828SNeel Natu 		 * We also check for STI_BLOCKING because some implementations
117448b2d828SNeel Natu 		 * don't allow NMI injection in this case. If we are running
117548b2d828SNeel Natu 		 * on a processor that doesn't have this restriction it will
117648b2d828SNeel Natu 		 * immediately exit and the NMI will be injected in the
117748b2d828SNeel Natu 		 * "NMI window exiting" handler.
1178366f6083SPeter Grehan 		 */
117948b2d828SNeel Natu 		need_nmi_exiting = 1;
118048b2d828SNeel Natu 		gi = vmcs_read(VMCS_GUEST_INTERRUPTIBILITY);
118148b2d828SNeel Natu 		if ((gi & (HWINTR_BLOCKING | NMI_BLOCKING)) == 0) {
11823de83862SNeel Natu 			info = vmcs_read(VMCS_ENTRY_INTR_INFO);
118348b2d828SNeel Natu 			if ((info & VMCS_INTR_VALID) == 0) {
118448b2d828SNeel Natu 				vmx_inject_nmi(vmx, vcpu);
118548b2d828SNeel Natu 				need_nmi_exiting = 0;
118648b2d828SNeel Natu 			} else {
118748b2d828SNeel Natu 				VCPU_CTR1(vmx->vm, vcpu, "Cannot inject NMI "
118848b2d828SNeel Natu 				    "due to VM-entry intr info %#x", info);
118948b2d828SNeel Natu 			}
119048b2d828SNeel Natu 		} else {
119148b2d828SNeel Natu 			VCPU_CTR1(vmx->vm, vcpu, "Cannot inject NMI due to "
119248b2d828SNeel Natu 			    "Guest Interruptibility-state %#x", gi);
119348b2d828SNeel Natu 		}
1194eeefa4e4SNeel Natu 
119548b2d828SNeel Natu 		if (need_nmi_exiting)
119648b2d828SNeel Natu 			vmx_set_nmi_window_exiting(vmx, vcpu);
119748b2d828SNeel Natu 	}
1198366f6083SPeter Grehan 
119988c4b8d1SNeel Natu 	if (virtual_interrupt_delivery) {
120088c4b8d1SNeel Natu 		vmx_inject_pir(vlapic);
120188c4b8d1SNeel Natu 		return;
120288c4b8d1SNeel Natu 	}
120388c4b8d1SNeel Natu 
120448b2d828SNeel Natu 	/*
120536736912SNeel Natu 	 * If interrupt-window exiting is already in effect then don't bother
120636736912SNeel Natu 	 * checking for pending interrupts. This is just an optimization and
120736736912SNeel Natu 	 * not needed for correctness.
120848b2d828SNeel Natu 	 */
120936736912SNeel Natu 	if ((vmx->cap[vcpu].proc_ctls & PROCBASED_INT_WINDOW_EXITING) != 0) {
121036736912SNeel Natu 		VCPU_CTR0(vmx->vm, vcpu, "Skip interrupt injection due to "
121136736912SNeel Natu 		    "pending int_window_exiting");
121248b2d828SNeel Natu 		return;
121336736912SNeel Natu 	}
121448b2d828SNeel Natu 
1215366f6083SPeter Grehan 	/* Ask the local apic for a vector to inject */
12164d1e82a8SNeel Natu 	if (!vlapic_pending_intr(vlapic, &vector))
1217366f6083SPeter Grehan 		return;
1218366f6083SPeter Grehan 
121948b2d828SNeel Natu 	KASSERT(vector >= 32 && vector <= 255, ("invalid vector %d", vector));
1220366f6083SPeter Grehan 
1221366f6083SPeter Grehan 	/* Check RFLAGS.IF and the interruptibility state of the guest */
12223de83862SNeel Natu 	rflags = vmcs_read(VMCS_GUEST_RFLAGS);
122336736912SNeel Natu 	if ((rflags & PSL_I) == 0) {
122436736912SNeel Natu 		VCPU_CTR2(vmx->vm, vcpu, "Cannot inject vector %d due to "
122536736912SNeel Natu 		    "rflags %#lx", vector, rflags);
1226366f6083SPeter Grehan 		goto cantinject;
122736736912SNeel Natu 	}
1228366f6083SPeter Grehan 
122948b2d828SNeel Natu 	gi = vmcs_read(VMCS_GUEST_INTERRUPTIBILITY);
123036736912SNeel Natu 	if (gi & HWINTR_BLOCKING) {
123136736912SNeel Natu 		VCPU_CTR2(vmx->vm, vcpu, "Cannot inject vector %d due to "
123236736912SNeel Natu 		    "Guest Interruptibility-state %#x", vector, gi);
1233366f6083SPeter Grehan 		goto cantinject;
123436736912SNeel Natu 	}
123536736912SNeel Natu 
123636736912SNeel Natu 	info = vmcs_read(VMCS_ENTRY_INTR_INFO);
123736736912SNeel Natu 	if (info & VMCS_INTR_VALID) {
123836736912SNeel Natu 		/*
123936736912SNeel Natu 		 * This is expected and could happen for multiple reasons:
124036736912SNeel Natu 		 * - A vectoring VM-entry was aborted due to astpending
124136736912SNeel Natu 		 * - A VM-exit happened during event injection.
1242dc506506SNeel Natu 		 * - An exception was injected above.
124336736912SNeel Natu 		 * - An NMI was injected above or after "NMI window exiting"
124436736912SNeel Natu 		 */
124536736912SNeel Natu 		VCPU_CTR2(vmx->vm, vcpu, "Cannot inject vector %d due to "
124636736912SNeel Natu 		    "VM-entry intr info %#x", vector, info);
124736736912SNeel Natu 		goto cantinject;
124836736912SNeel Natu 	}
1249366f6083SPeter Grehan 
1250366f6083SPeter Grehan 	/* Inject the interrupt */
1251160471d2SNeel Natu 	info = VMCS_INTR_T_HWINTR | VMCS_INTR_VALID;
1252366f6083SPeter Grehan 	info |= vector;
12533de83862SNeel Natu 	vmcs_write(VMCS_ENTRY_INTR_INFO, info);
1254366f6083SPeter Grehan 
1255366f6083SPeter Grehan 	/* Update the Local APIC ISR */
1256de5ea6b6SNeel Natu 	vlapic_intr_accepted(vlapic, vector);
1257366f6083SPeter Grehan 
1258513c8d33SNeel Natu 	VCPU_CTR1(vmx->vm, vcpu, "Injecting hwintr at vector %d", vector);
1259366f6083SPeter Grehan 
1260366f6083SPeter Grehan 	return;
1261366f6083SPeter Grehan 
1262366f6083SPeter Grehan cantinject:
1263366f6083SPeter Grehan 	/*
1264366f6083SPeter Grehan 	 * Set the Interrupt Window Exiting execution control so we can inject
1265366f6083SPeter Grehan 	 * the interrupt as soon as blocking condition goes away.
1266366f6083SPeter Grehan 	 */
1267366f6083SPeter Grehan 	vmx_set_int_window_exiting(vmx, vcpu);
1268366f6083SPeter Grehan }
1269366f6083SPeter Grehan 
1270e5a1d950SNeel Natu /*
1271e5a1d950SNeel Natu  * If the Virtual NMIs execution control is '1' then the logical processor
1272e5a1d950SNeel Natu  * tracks virtual-NMI blocking in the Guest Interruptibility-state field of
1273e5a1d950SNeel Natu  * the VMCS. An IRET instruction in VMX non-root operation will remove any
1274e5a1d950SNeel Natu  * virtual-NMI blocking.
1275e5a1d950SNeel Natu  *
1276e5a1d950SNeel Natu  * This unblocking occurs even if the IRET causes a fault. In this case the
1277e5a1d950SNeel Natu  * hypervisor needs to restore virtual-NMI blocking before resuming the guest.
1278e5a1d950SNeel Natu  */
1279e5a1d950SNeel Natu static void
1280e5a1d950SNeel Natu vmx_restore_nmi_blocking(struct vmx *vmx, int vcpuid)
1281e5a1d950SNeel Natu {
1282e5a1d950SNeel Natu 	uint32_t gi;
1283e5a1d950SNeel Natu 
1284e5a1d950SNeel Natu 	VCPU_CTR0(vmx->vm, vcpuid, "Restore Virtual-NMI blocking");
1285e5a1d950SNeel Natu 	gi = vmcs_read(VMCS_GUEST_INTERRUPTIBILITY);
1286e5a1d950SNeel Natu 	gi |= VMCS_INTERRUPTIBILITY_NMI_BLOCKING;
1287e5a1d950SNeel Natu 	vmcs_write(VMCS_GUEST_INTERRUPTIBILITY, gi);
1288e5a1d950SNeel Natu }
1289e5a1d950SNeel Natu 
1290e5a1d950SNeel Natu static void
1291e5a1d950SNeel Natu vmx_clear_nmi_blocking(struct vmx *vmx, int vcpuid)
1292e5a1d950SNeel Natu {
1293e5a1d950SNeel Natu 	uint32_t gi;
1294e5a1d950SNeel Natu 
1295e5a1d950SNeel Natu 	VCPU_CTR0(vmx->vm, vcpuid, "Clear Virtual-NMI blocking");
1296e5a1d950SNeel Natu 	gi = vmcs_read(VMCS_GUEST_INTERRUPTIBILITY);
1297e5a1d950SNeel Natu 	gi &= ~VMCS_INTERRUPTIBILITY_NMI_BLOCKING;
1298e5a1d950SNeel Natu 	vmcs_write(VMCS_GUEST_INTERRUPTIBILITY, gi);
1299e5a1d950SNeel Natu }
1300e5a1d950SNeel Natu 
1301366f6083SPeter Grehan static int
1302a0efd3fbSJohn Baldwin vmx_emulate_xsetbv(struct vmx *vmx, int vcpu, struct vm_exit *vmexit)
1303abb023fbSJohn Baldwin {
1304abb023fbSJohn Baldwin 	struct vmxctx *vmxctx;
1305abb023fbSJohn Baldwin 	uint64_t xcrval;
1306abb023fbSJohn Baldwin 	const struct xsave_limits *limits;
1307abb023fbSJohn Baldwin 
1308abb023fbSJohn Baldwin 	vmxctx = &vmx->ctx[vcpu];
1309abb023fbSJohn Baldwin 	limits = vmm_get_xsave_limits();
1310abb023fbSJohn Baldwin 
1311a0efd3fbSJohn Baldwin 	/*
1312a0efd3fbSJohn Baldwin 	 * Note that the processor raises a GP# fault on its own if
1313a0efd3fbSJohn Baldwin 	 * xsetbv is executed for CPL != 0, so we do not have to
1314a0efd3fbSJohn Baldwin 	 * emulate that fault here.
1315a0efd3fbSJohn Baldwin 	 */
1316a0efd3fbSJohn Baldwin 
1317a0efd3fbSJohn Baldwin 	/* Only xcr0 is supported. */
1318a0efd3fbSJohn Baldwin 	if (vmxctx->guest_rcx != 0) {
1319dc506506SNeel Natu 		vm_inject_gp(vmx->vm, vcpu);
1320a0efd3fbSJohn Baldwin 		return (HANDLED);
1321a0efd3fbSJohn Baldwin 	}
1322a0efd3fbSJohn Baldwin 
1323a0efd3fbSJohn Baldwin 	/* We only handle xcr0 if both the host and guest have XSAVE enabled. */
1324a0efd3fbSJohn Baldwin 	if (!limits->xsave_enabled || !(vmcs_read(VMCS_GUEST_CR4) & CR4_XSAVE)) {
1325dc506506SNeel Natu 		vm_inject_ud(vmx->vm, vcpu);
1326a0efd3fbSJohn Baldwin 		return (HANDLED);
1327a0efd3fbSJohn Baldwin 	}
1328abb023fbSJohn Baldwin 
1329abb023fbSJohn Baldwin 	xcrval = vmxctx->guest_rdx << 32 | (vmxctx->guest_rax & 0xffffffff);
1330a0efd3fbSJohn Baldwin 	if ((xcrval & ~limits->xcr0_allowed) != 0) {
1331dc506506SNeel Natu 		vm_inject_gp(vmx->vm, vcpu);
1332a0efd3fbSJohn Baldwin 		return (HANDLED);
1333a0efd3fbSJohn Baldwin 	}
1334abb023fbSJohn Baldwin 
1335a0efd3fbSJohn Baldwin 	if (!(xcrval & XFEATURE_ENABLED_X87)) {
1336dc506506SNeel Natu 		vm_inject_gp(vmx->vm, vcpu);
1337a0efd3fbSJohn Baldwin 		return (HANDLED);
1338a0efd3fbSJohn Baldwin 	}
1339abb023fbSJohn Baldwin 
1340abb023fbSJohn Baldwin 	if ((xcrval & (XFEATURE_ENABLED_AVX | XFEATURE_ENABLED_SSE)) ==
1341a0efd3fbSJohn Baldwin 	    XFEATURE_ENABLED_AVX) {
1342dc506506SNeel Natu 		vm_inject_gp(vmx->vm, vcpu);
1343a0efd3fbSJohn Baldwin 		return (HANDLED);
1344a0efd3fbSJohn Baldwin 	}
1345abb023fbSJohn Baldwin 
1346abb023fbSJohn Baldwin 	/*
1347abb023fbSJohn Baldwin 	 * This runs "inside" vmrun() with the guest's FPU state, so
1348abb023fbSJohn Baldwin 	 * modifying xcr0 directly modifies the guest's xcr0, not the
1349abb023fbSJohn Baldwin 	 * host's.
1350abb023fbSJohn Baldwin 	 */
1351abb023fbSJohn Baldwin 	load_xcr(0, xcrval);
1352abb023fbSJohn Baldwin 	return (HANDLED);
1353abb023fbSJohn Baldwin }
1354abb023fbSJohn Baldwin 
1355abb023fbSJohn Baldwin static int
1356366f6083SPeter Grehan vmx_emulate_cr_access(struct vmx *vmx, int vcpu, uint64_t exitqual)
1357366f6083SPeter Grehan {
13583de83862SNeel Natu 	int cr, vmcs_guest_cr, vmcs_shadow_cr;
135980a902efSPeter Grehan 	uint64_t crval, regval, ones_mask, zeros_mask;
1360366f6083SPeter Grehan 	const struct vmxctx *vmxctx;
1361366f6083SPeter Grehan 
136239c21c2dSNeel Natu 	/* We only handle mov to %cr0 or %cr4 at this time */
136339c21c2dSNeel Natu 	if ((exitqual & 0xf0) != 0x00)
136439c21c2dSNeel Natu 		return (UNHANDLED);
136539c21c2dSNeel Natu 
136639c21c2dSNeel Natu 	cr = exitqual & 0xf;
136739c21c2dSNeel Natu 	if (cr != 0 && cr != 4)
1368366f6083SPeter Grehan 		return (UNHANDLED);
1369366f6083SPeter Grehan 
13706f0c167fSDimitry Andric 	regval = 0; /* silence gcc */
1371366f6083SPeter Grehan 	vmxctx = &vmx->ctx[vcpu];
1372366f6083SPeter Grehan 
1373366f6083SPeter Grehan 	/*
13743de83862SNeel Natu 	 * We must use vmcs_write() directly here because vmcs_setreg() will
1375366f6083SPeter Grehan 	 * call vmclear(vmcs) as a side-effect which we certainly don't want.
1376366f6083SPeter Grehan 	 */
1377366f6083SPeter Grehan 	switch ((exitqual >> 8) & 0xf) {
1378366f6083SPeter Grehan 	case 0:
1379366f6083SPeter Grehan 		regval = vmxctx->guest_rax;
1380366f6083SPeter Grehan 		break;
1381366f6083SPeter Grehan 	case 1:
1382366f6083SPeter Grehan 		regval = vmxctx->guest_rcx;
1383366f6083SPeter Grehan 		break;
1384366f6083SPeter Grehan 	case 2:
1385366f6083SPeter Grehan 		regval = vmxctx->guest_rdx;
1386366f6083SPeter Grehan 		break;
1387366f6083SPeter Grehan 	case 3:
1388366f6083SPeter Grehan 		regval = vmxctx->guest_rbx;
1389366f6083SPeter Grehan 		break;
1390366f6083SPeter Grehan 	case 4:
13913de83862SNeel Natu 		regval = vmcs_read(VMCS_GUEST_RSP);
1392366f6083SPeter Grehan 		break;
1393366f6083SPeter Grehan 	case 5:
1394366f6083SPeter Grehan 		regval = vmxctx->guest_rbp;
1395366f6083SPeter Grehan 		break;
1396366f6083SPeter Grehan 	case 6:
1397366f6083SPeter Grehan 		regval = vmxctx->guest_rsi;
1398366f6083SPeter Grehan 		break;
1399366f6083SPeter Grehan 	case 7:
1400366f6083SPeter Grehan 		regval = vmxctx->guest_rdi;
1401366f6083SPeter Grehan 		break;
1402366f6083SPeter Grehan 	case 8:
1403366f6083SPeter Grehan 		regval = vmxctx->guest_r8;
1404366f6083SPeter Grehan 		break;
1405366f6083SPeter Grehan 	case 9:
1406366f6083SPeter Grehan 		regval = vmxctx->guest_r9;
1407366f6083SPeter Grehan 		break;
1408366f6083SPeter Grehan 	case 10:
1409366f6083SPeter Grehan 		regval = vmxctx->guest_r10;
1410366f6083SPeter Grehan 		break;
1411366f6083SPeter Grehan 	case 11:
1412366f6083SPeter Grehan 		regval = vmxctx->guest_r11;
1413366f6083SPeter Grehan 		break;
1414366f6083SPeter Grehan 	case 12:
1415366f6083SPeter Grehan 		regval = vmxctx->guest_r12;
1416366f6083SPeter Grehan 		break;
1417366f6083SPeter Grehan 	case 13:
1418366f6083SPeter Grehan 		regval = vmxctx->guest_r13;
1419366f6083SPeter Grehan 		break;
1420366f6083SPeter Grehan 	case 14:
1421366f6083SPeter Grehan 		regval = vmxctx->guest_r14;
1422366f6083SPeter Grehan 		break;
1423366f6083SPeter Grehan 	case 15:
1424366f6083SPeter Grehan 		regval = vmxctx->guest_r15;
1425366f6083SPeter Grehan 		break;
1426366f6083SPeter Grehan 	}
1427366f6083SPeter Grehan 
142839c21c2dSNeel Natu 	if (cr == 0) {
142939c21c2dSNeel Natu 		ones_mask = cr0_ones_mask;
143039c21c2dSNeel Natu 		zeros_mask = cr0_zeros_mask;
143139c21c2dSNeel Natu 		vmcs_guest_cr = VMCS_GUEST_CR0;
1432aaaa0656SPeter Grehan 		vmcs_shadow_cr = VMCS_CR0_SHADOW;
143339c21c2dSNeel Natu 	} else {
143439c21c2dSNeel Natu 		ones_mask = cr4_ones_mask;
143539c21c2dSNeel Natu 		zeros_mask = cr4_zeros_mask;
143639c21c2dSNeel Natu 		vmcs_guest_cr = VMCS_GUEST_CR4;
1437aaaa0656SPeter Grehan 		vmcs_shadow_cr = VMCS_CR4_SHADOW;
143839c21c2dSNeel Natu 	}
14393de83862SNeel Natu 	vmcs_write(vmcs_shadow_cr, regval);
1440aaaa0656SPeter Grehan 
144180a902efSPeter Grehan 	crval = regval | ones_mask;
144280a902efSPeter Grehan 	crval &= ~zeros_mask;
14433de83862SNeel Natu 	vmcs_write(vmcs_guest_cr, crval);
1444366f6083SPeter Grehan 
144580a902efSPeter Grehan 	if (cr == 0 && regval & CR0_PG) {
144680a902efSPeter Grehan 		uint64_t efer, entry_ctls;
144780a902efSPeter Grehan 
144880a902efSPeter Grehan 		/*
144980a902efSPeter Grehan 		 * If CR0.PG is 1 and EFER.LME is 1 then EFER.LMA and
145080a902efSPeter Grehan 		 * the "IA-32e mode guest" bit in VM-entry control must be
145180a902efSPeter Grehan 		 * equal.
145280a902efSPeter Grehan 		 */
14533de83862SNeel Natu 		efer = vmcs_read(VMCS_GUEST_IA32_EFER);
145480a902efSPeter Grehan 		if (efer & EFER_LME) {
145580a902efSPeter Grehan 			efer |= EFER_LMA;
14563de83862SNeel Natu 			vmcs_write(VMCS_GUEST_IA32_EFER, efer);
14573de83862SNeel Natu 			entry_ctls = vmcs_read(VMCS_ENTRY_CTLS);
145880a902efSPeter Grehan 			entry_ctls |= VM_ENTRY_GUEST_LMA;
14593de83862SNeel Natu 			vmcs_write(VMCS_ENTRY_CTLS, entry_ctls);
146080a902efSPeter Grehan 		}
146180a902efSPeter Grehan 	}
146280a902efSPeter Grehan 
1463366f6083SPeter Grehan 	return (HANDLED);
1464366f6083SPeter Grehan }
1465366f6083SPeter Grehan 
146600f3efe1SJohn Baldwin static enum vie_cpu_mode
146700f3efe1SJohn Baldwin vmx_cpu_mode(void)
146800f3efe1SJohn Baldwin {
146900f3efe1SJohn Baldwin 
147000f3efe1SJohn Baldwin 	if (vmcs_read(VMCS_GUEST_IA32_EFER) & EFER_LMA)
147100f3efe1SJohn Baldwin 		return (CPU_MODE_64BIT);
147200f3efe1SJohn Baldwin 	else
147300f3efe1SJohn Baldwin 		return (CPU_MODE_COMPATIBILITY);
147400f3efe1SJohn Baldwin }
147500f3efe1SJohn Baldwin 
147600f3efe1SJohn Baldwin static enum vie_paging_mode
147700f3efe1SJohn Baldwin vmx_paging_mode(void)
147800f3efe1SJohn Baldwin {
147900f3efe1SJohn Baldwin 
148000f3efe1SJohn Baldwin 	if (!(vmcs_read(VMCS_GUEST_CR0) & CR0_PG))
148100f3efe1SJohn Baldwin 		return (PAGING_MODE_FLAT);
148200f3efe1SJohn Baldwin 	if (!(vmcs_read(VMCS_GUEST_CR4) & CR4_PAE))
148300f3efe1SJohn Baldwin 		return (PAGING_MODE_32);
148400f3efe1SJohn Baldwin 	if (vmcs_read(VMCS_GUEST_IA32_EFER) & EFER_LME)
148500f3efe1SJohn Baldwin 		return (PAGING_MODE_64);
148600f3efe1SJohn Baldwin 	else
148700f3efe1SJohn Baldwin 		return (PAGING_MODE_PAE);
148800f3efe1SJohn Baldwin }
148900f3efe1SJohn Baldwin 
1490366f6083SPeter Grehan static int
1491318224bbSNeel Natu ept_fault_type(uint64_t ept_qual)
1492a2da7af6SNeel Natu {
1493318224bbSNeel Natu 	int fault_type;
1494a2da7af6SNeel Natu 
1495318224bbSNeel Natu 	if (ept_qual & EPT_VIOLATION_DATA_WRITE)
1496318224bbSNeel Natu 		fault_type = VM_PROT_WRITE;
1497318224bbSNeel Natu 	else if (ept_qual & EPT_VIOLATION_INST_FETCH)
1498318224bbSNeel Natu 		fault_type = VM_PROT_EXECUTE;
1499318224bbSNeel Natu 	else
1500318224bbSNeel Natu 		fault_type= VM_PROT_READ;
1501318224bbSNeel Natu 
1502318224bbSNeel Natu 	return (fault_type);
1503318224bbSNeel Natu }
1504318224bbSNeel Natu 
1505318224bbSNeel Natu static boolean_t
1506318224bbSNeel Natu ept_emulation_fault(uint64_t ept_qual)
1507318224bbSNeel Natu {
1508318224bbSNeel Natu 	int read, write;
1509318224bbSNeel Natu 
1510318224bbSNeel Natu 	/* EPT fault on an instruction fetch doesn't make sense here */
1511a2da7af6SNeel Natu 	if (ept_qual & EPT_VIOLATION_INST_FETCH)
1512318224bbSNeel Natu 		return (FALSE);
1513a2da7af6SNeel Natu 
1514318224bbSNeel Natu 	/* EPT fault must be a read fault or a write fault */
1515a2da7af6SNeel Natu 	read = ept_qual & EPT_VIOLATION_DATA_READ ? 1 : 0;
1516a2da7af6SNeel Natu 	write = ept_qual & EPT_VIOLATION_DATA_WRITE ? 1 : 0;
15173b2b0011SPeter Grehan 	if ((read | write) == 0)
1518318224bbSNeel Natu 		return (FALSE);
1519a2da7af6SNeel Natu 
1520a2da7af6SNeel Natu 	/*
15213b2b0011SPeter Grehan 	 * The EPT violation must have been caused by accessing a
15223b2b0011SPeter Grehan 	 * guest-physical address that is a translation of a guest-linear
15233b2b0011SPeter Grehan 	 * address.
1524a2da7af6SNeel Natu 	 */
1525a2da7af6SNeel Natu 	if ((ept_qual & EPT_VIOLATION_GLA_VALID) == 0 ||
1526a2da7af6SNeel Natu 	    (ept_qual & EPT_VIOLATION_XLAT_VALID) == 0) {
1527318224bbSNeel Natu 		return (FALSE);
1528a2da7af6SNeel Natu 	}
1529a2da7af6SNeel Natu 
1530318224bbSNeel Natu 	return (TRUE);
1531a2da7af6SNeel Natu }
1532a2da7af6SNeel Natu 
1533159dd56fSNeel Natu static __inline int
1534159dd56fSNeel Natu apic_access_virtualization(struct vmx *vmx, int vcpuid)
1535159dd56fSNeel Natu {
1536159dd56fSNeel Natu 	uint32_t proc_ctls2;
1537159dd56fSNeel Natu 
1538159dd56fSNeel Natu 	proc_ctls2 = vmx->cap[vcpuid].proc_ctls2;
1539159dd56fSNeel Natu 	return ((proc_ctls2 & PROCBASED2_VIRTUALIZE_APIC_ACCESSES) ? 1 : 0);
1540159dd56fSNeel Natu }
1541159dd56fSNeel Natu 
1542159dd56fSNeel Natu static __inline int
1543159dd56fSNeel Natu x2apic_virtualization(struct vmx *vmx, int vcpuid)
1544159dd56fSNeel Natu {
1545159dd56fSNeel Natu 	uint32_t proc_ctls2;
1546159dd56fSNeel Natu 
1547159dd56fSNeel Natu 	proc_ctls2 = vmx->cap[vcpuid].proc_ctls2;
1548159dd56fSNeel Natu 	return ((proc_ctls2 & PROCBASED2_VIRTUALIZE_X2APIC_MODE) ? 1 : 0);
1549159dd56fSNeel Natu }
1550159dd56fSNeel Natu 
1551a2da7af6SNeel Natu static int
1552159dd56fSNeel Natu vmx_handle_apic_write(struct vmx *vmx, int vcpuid, struct vlapic *vlapic,
1553159dd56fSNeel Natu     uint64_t qual)
155488c4b8d1SNeel Natu {
155588c4b8d1SNeel Natu 	int error, handled, offset;
1556159dd56fSNeel Natu 	uint32_t *apic_regs, vector;
155788c4b8d1SNeel Natu 	bool retu;
155888c4b8d1SNeel Natu 
1559a0efd3fbSJohn Baldwin 	handled = HANDLED;
156088c4b8d1SNeel Natu 	offset = APIC_WRITE_OFFSET(qual);
1561159dd56fSNeel Natu 
1562159dd56fSNeel Natu 	if (!apic_access_virtualization(vmx, vcpuid)) {
1563159dd56fSNeel Natu 		/*
1564159dd56fSNeel Natu 		 * In general there should not be any APIC write VM-exits
1565159dd56fSNeel Natu 		 * unless APIC-access virtualization is enabled.
1566159dd56fSNeel Natu 		 *
1567159dd56fSNeel Natu 		 * However self-IPI virtualization can legitimately trigger
1568159dd56fSNeel Natu 		 * an APIC-write VM-exit so treat it specially.
1569159dd56fSNeel Natu 		 */
1570159dd56fSNeel Natu 		if (x2apic_virtualization(vmx, vcpuid) &&
1571159dd56fSNeel Natu 		    offset == APIC_OFFSET_SELF_IPI) {
1572159dd56fSNeel Natu 			apic_regs = (uint32_t *)(vlapic->apic_page);
1573159dd56fSNeel Natu 			vector = apic_regs[APIC_OFFSET_SELF_IPI / 4];
1574159dd56fSNeel Natu 			vlapic_self_ipi_handler(vlapic, vector);
1575159dd56fSNeel Natu 			return (HANDLED);
1576159dd56fSNeel Natu 		} else
1577159dd56fSNeel Natu 			return (UNHANDLED);
1578159dd56fSNeel Natu 	}
1579159dd56fSNeel Natu 
158088c4b8d1SNeel Natu 	switch (offset) {
158188c4b8d1SNeel Natu 	case APIC_OFFSET_ID:
158288c4b8d1SNeel Natu 		vlapic_id_write_handler(vlapic);
158388c4b8d1SNeel Natu 		break;
158488c4b8d1SNeel Natu 	case APIC_OFFSET_LDR:
158588c4b8d1SNeel Natu 		vlapic_ldr_write_handler(vlapic);
158688c4b8d1SNeel Natu 		break;
158788c4b8d1SNeel Natu 	case APIC_OFFSET_DFR:
158888c4b8d1SNeel Natu 		vlapic_dfr_write_handler(vlapic);
158988c4b8d1SNeel Natu 		break;
159088c4b8d1SNeel Natu 	case APIC_OFFSET_SVR:
159188c4b8d1SNeel Natu 		vlapic_svr_write_handler(vlapic);
159288c4b8d1SNeel Natu 		break;
159388c4b8d1SNeel Natu 	case APIC_OFFSET_ESR:
159488c4b8d1SNeel Natu 		vlapic_esr_write_handler(vlapic);
159588c4b8d1SNeel Natu 		break;
159688c4b8d1SNeel Natu 	case APIC_OFFSET_ICR_LOW:
159788c4b8d1SNeel Natu 		retu = false;
159888c4b8d1SNeel Natu 		error = vlapic_icrlo_write_handler(vlapic, &retu);
159988c4b8d1SNeel Natu 		if (error != 0 || retu)
1600a0efd3fbSJohn Baldwin 			handled = UNHANDLED;
160188c4b8d1SNeel Natu 		break;
160288c4b8d1SNeel Natu 	case APIC_OFFSET_CMCI_LVT:
160388c4b8d1SNeel Natu 	case APIC_OFFSET_TIMER_LVT ... APIC_OFFSET_ERROR_LVT:
160488c4b8d1SNeel Natu 		vlapic_lvt_write_handler(vlapic, offset);
160588c4b8d1SNeel Natu 		break;
160688c4b8d1SNeel Natu 	case APIC_OFFSET_TIMER_ICR:
160788c4b8d1SNeel Natu 		vlapic_icrtmr_write_handler(vlapic);
160888c4b8d1SNeel Natu 		break;
160988c4b8d1SNeel Natu 	case APIC_OFFSET_TIMER_DCR:
161088c4b8d1SNeel Natu 		vlapic_dcr_write_handler(vlapic);
161188c4b8d1SNeel Natu 		break;
161288c4b8d1SNeel Natu 	default:
1613a0efd3fbSJohn Baldwin 		handled = UNHANDLED;
161488c4b8d1SNeel Natu 		break;
161588c4b8d1SNeel Natu 	}
161688c4b8d1SNeel Natu 	return (handled);
161788c4b8d1SNeel Natu }
161888c4b8d1SNeel Natu 
161988c4b8d1SNeel Natu static bool
1620159dd56fSNeel Natu apic_access_fault(struct vmx *vmx, int vcpuid, uint64_t gpa)
162188c4b8d1SNeel Natu {
162288c4b8d1SNeel Natu 
1623159dd56fSNeel Natu 	if (apic_access_virtualization(vmx, vcpuid) &&
162488c4b8d1SNeel Natu 	    (gpa >= DEFAULT_APIC_BASE && gpa < DEFAULT_APIC_BASE + PAGE_SIZE))
162588c4b8d1SNeel Natu 		return (true);
162688c4b8d1SNeel Natu 	else
162788c4b8d1SNeel Natu 		return (false);
162888c4b8d1SNeel Natu }
162988c4b8d1SNeel Natu 
163088c4b8d1SNeel Natu static int
163188c4b8d1SNeel Natu vmx_handle_apic_access(struct vmx *vmx, int vcpuid, struct vm_exit *vmexit)
163288c4b8d1SNeel Natu {
163388c4b8d1SNeel Natu 	uint64_t qual;
163488c4b8d1SNeel Natu 	int access_type, offset, allowed;
163588c4b8d1SNeel Natu 
1636159dd56fSNeel Natu 	if (!apic_access_virtualization(vmx, vcpuid))
163788c4b8d1SNeel Natu 		return (UNHANDLED);
163888c4b8d1SNeel Natu 
163988c4b8d1SNeel Natu 	qual = vmexit->u.vmx.exit_qualification;
164088c4b8d1SNeel Natu 	access_type = APIC_ACCESS_TYPE(qual);
164188c4b8d1SNeel Natu 	offset = APIC_ACCESS_OFFSET(qual);
164288c4b8d1SNeel Natu 
164388c4b8d1SNeel Natu 	allowed = 0;
164488c4b8d1SNeel Natu 	if (access_type == 0) {
164588c4b8d1SNeel Natu 		/*
164688c4b8d1SNeel Natu 		 * Read data access to the following registers is expected.
164788c4b8d1SNeel Natu 		 */
164888c4b8d1SNeel Natu 		switch (offset) {
164988c4b8d1SNeel Natu 		case APIC_OFFSET_APR:
165088c4b8d1SNeel Natu 		case APIC_OFFSET_PPR:
165188c4b8d1SNeel Natu 		case APIC_OFFSET_RRR:
165288c4b8d1SNeel Natu 		case APIC_OFFSET_CMCI_LVT:
165388c4b8d1SNeel Natu 		case APIC_OFFSET_TIMER_CCR:
165488c4b8d1SNeel Natu 			allowed = 1;
165588c4b8d1SNeel Natu 			break;
165688c4b8d1SNeel Natu 		default:
165788c4b8d1SNeel Natu 			break;
165888c4b8d1SNeel Natu 		}
165988c4b8d1SNeel Natu 	} else if (access_type == 1) {
166088c4b8d1SNeel Natu 		/*
166188c4b8d1SNeel Natu 		 * Write data access to the following registers is expected.
166288c4b8d1SNeel Natu 		 */
166388c4b8d1SNeel Natu 		switch (offset) {
166488c4b8d1SNeel Natu 		case APIC_OFFSET_VER:
166588c4b8d1SNeel Natu 		case APIC_OFFSET_APR:
166688c4b8d1SNeel Natu 		case APIC_OFFSET_PPR:
166788c4b8d1SNeel Natu 		case APIC_OFFSET_RRR:
166888c4b8d1SNeel Natu 		case APIC_OFFSET_ISR0 ... APIC_OFFSET_ISR7:
166988c4b8d1SNeel Natu 		case APIC_OFFSET_TMR0 ... APIC_OFFSET_TMR7:
167088c4b8d1SNeel Natu 		case APIC_OFFSET_IRR0 ... APIC_OFFSET_IRR7:
167188c4b8d1SNeel Natu 		case APIC_OFFSET_CMCI_LVT:
167288c4b8d1SNeel Natu 		case APIC_OFFSET_TIMER_CCR:
167388c4b8d1SNeel Natu 			allowed = 1;
167488c4b8d1SNeel Natu 			break;
167588c4b8d1SNeel Natu 		default:
167688c4b8d1SNeel Natu 			break;
167788c4b8d1SNeel Natu 		}
167888c4b8d1SNeel Natu 	}
167988c4b8d1SNeel Natu 
168088c4b8d1SNeel Natu 	if (allowed) {
168188c4b8d1SNeel Natu 		vmexit->exitcode = VM_EXITCODE_INST_EMUL;
168288c4b8d1SNeel Natu 		vmexit->u.inst_emul.gpa = DEFAULT_APIC_BASE + offset;
168388c4b8d1SNeel Natu 		vmexit->u.inst_emul.gla = VIE_INVALID_GLA;
168488c4b8d1SNeel Natu 		vmexit->u.inst_emul.cr3 = vmcs_guest_cr3();
168500f3efe1SJohn Baldwin 		vmexit->u.inst_emul.cpu_mode = vmx_cpu_mode();
168600f3efe1SJohn Baldwin 		vmexit->u.inst_emul.paging_mode = vmx_paging_mode();
168788c4b8d1SNeel Natu 	}
168888c4b8d1SNeel Natu 
168988c4b8d1SNeel Natu 	/*
169088c4b8d1SNeel Natu 	 * Regardless of whether the APIC-access is allowed this handler
169188c4b8d1SNeel Natu 	 * always returns UNHANDLED:
169288c4b8d1SNeel Natu 	 * - if the access is allowed then it is handled by emulating the
169388c4b8d1SNeel Natu 	 *   instruction that caused the VM-exit (outside the critical section)
169488c4b8d1SNeel Natu 	 * - if the access is not allowed then it will be converted to an
169588c4b8d1SNeel Natu 	 *   exitcode of VM_EXITCODE_VMX and will be dealt with in userland.
169688c4b8d1SNeel Natu 	 */
169788c4b8d1SNeel Natu 	return (UNHANDLED);
169888c4b8d1SNeel Natu }
169988c4b8d1SNeel Natu 
170088c4b8d1SNeel Natu static int
1701366f6083SPeter Grehan vmx_exit_process(struct vmx *vmx, int vcpu, struct vm_exit *vmexit)
1702366f6083SPeter Grehan {
1703f76fc5d4SNeel Natu 	int error, handled;
1704366f6083SPeter Grehan 	struct vmxctx *vmxctx;
170588c4b8d1SNeel Natu 	struct vlapic *vlapic;
1706e5a1d950SNeel Natu 	uint32_t eax, ecx, edx, idtvec_info, idtvec_err, intr_info, reason;
17073de83862SNeel Natu 	uint64_t qual, gpa;
1708becd9849SNeel Natu 	bool retu;
1709366f6083SPeter Grehan 
1710160471d2SNeel Natu 	CTASSERT((PINBASED_CTLS_ONE_SETTING & PINBASED_VIRTUAL_NMI) != 0);
1711c308b23bSNeel Natu 	CTASSERT((PINBASED_CTLS_ONE_SETTING & PINBASED_NMI_EXITING) != 0);
1712160471d2SNeel Natu 
1713a0efd3fbSJohn Baldwin 	handled = UNHANDLED;
1714366f6083SPeter Grehan 	vmxctx = &vmx->ctx[vcpu];
17150492757cSNeel Natu 
1716366f6083SPeter Grehan 	qual = vmexit->u.vmx.exit_qualification;
1717318224bbSNeel Natu 	reason = vmexit->u.vmx.exit_reason;
1718366f6083SPeter Grehan 	vmexit->exitcode = VM_EXITCODE_BOGUS;
1719366f6083SPeter Grehan 
172061592433SNeel Natu 	vmm_stat_incr(vmx->vm, vcpu, VMEXIT_COUNT, 1);
172161592433SNeel Natu 
1722318224bbSNeel Natu 	/*
1723318224bbSNeel Natu 	 * VM exits that could be triggered during event injection on the
1724318224bbSNeel Natu 	 * previous VM entry need to be handled specially by re-injecting
1725318224bbSNeel Natu 	 * the event.
1726318224bbSNeel Natu 	 *
1727318224bbSNeel Natu 	 * See "Information for VM Exits During Event Delivery" in Intel SDM
1728318224bbSNeel Natu 	 * for details.
1729318224bbSNeel Natu 	 */
1730318224bbSNeel Natu 	switch (reason) {
1731318224bbSNeel Natu 	case EXIT_REASON_EPT_FAULT:
1732318224bbSNeel Natu 	case EXIT_REASON_EPT_MISCONFIG:
173388c4b8d1SNeel Natu 	case EXIT_REASON_APIC_ACCESS:
1734318224bbSNeel Natu 	case EXIT_REASON_TASK_SWITCH:
1735318224bbSNeel Natu 	case EXIT_REASON_EXCEPTION:
1736318224bbSNeel Natu 		idtvec_info = vmcs_idt_vectoring_info();
1737318224bbSNeel Natu 		if (idtvec_info & VMCS_IDT_VEC_VALID) {
1738318224bbSNeel Natu 			idtvec_info &= ~(1 << 12); /* clear undefined bit */
17393de83862SNeel Natu 			vmcs_write(VMCS_ENTRY_INTR_INFO, idtvec_info);
1740318224bbSNeel Natu 			if (idtvec_info & VMCS_IDT_VEC_ERRCODE_VALID) {
1741318224bbSNeel Natu 				idtvec_err = vmcs_idt_vectoring_err();
17423de83862SNeel Natu 				vmcs_write(VMCS_ENTRY_EXCEPTION_ERROR,
17433de83862SNeel Natu 				    idtvec_err);
1744318224bbSNeel Natu 			}
1745160471d2SNeel Natu 			/*
1746160471d2SNeel Natu 			 * If 'virtual NMIs' are being used and the VM-exit
1747160471d2SNeel Natu 			 * happened while injecting an NMI during the previous
1748160471d2SNeel Natu 			 * VM-entry, then clear "blocking by NMI" in the Guest
1749160471d2SNeel Natu 			 * Interruptibility-state.
1750160471d2SNeel Natu 			 */
1751160471d2SNeel Natu 			if ((idtvec_info & VMCS_INTR_T_MASK) ==
1752160471d2SNeel Natu 			    VMCS_INTR_T_NMI) {
1753e5a1d950SNeel Natu 				 vmx_clear_nmi_blocking(vmx, vcpu);
1754160471d2SNeel Natu 			}
17553de83862SNeel Natu 			vmcs_write(VMCS_ENTRY_INST_LENGTH, vmexit->inst_length);
1756318224bbSNeel Natu 		}
1757318224bbSNeel Natu 	default:
1758e5a1d950SNeel Natu 		idtvec_info = 0;
1759318224bbSNeel Natu 		break;
1760318224bbSNeel Natu 	}
1761318224bbSNeel Natu 
1762318224bbSNeel Natu 	switch (reason) {
1763366f6083SPeter Grehan 	case EXIT_REASON_CR_ACCESS:
1764b5aaf7b2SNeel Natu 		vmm_stat_incr(vmx->vm, vcpu, VMEXIT_CR_ACCESS, 1);
1765366f6083SPeter Grehan 		handled = vmx_emulate_cr_access(vmx, vcpu, qual);
1766366f6083SPeter Grehan 		break;
1767366f6083SPeter Grehan 	case EXIT_REASON_RDMSR:
1768b5aaf7b2SNeel Natu 		vmm_stat_incr(vmx->vm, vcpu, VMEXIT_RDMSR, 1);
1769becd9849SNeel Natu 		retu = false;
1770366f6083SPeter Grehan 		ecx = vmxctx->guest_rcx;
1771becd9849SNeel Natu 		error = emulate_rdmsr(vmx->vm, vcpu, ecx, &retu);
1772b42206f3SNeel Natu 		if (error) {
1773366f6083SPeter Grehan 			vmexit->exitcode = VM_EXITCODE_RDMSR;
1774366f6083SPeter Grehan 			vmexit->u.msr.code = ecx;
1775becd9849SNeel Natu 		} else if (!retu) {
1776a0efd3fbSJohn Baldwin 			handled = HANDLED;
1777becd9849SNeel Natu 		} else {
1778becd9849SNeel Natu 			/* Return to userspace with a valid exitcode */
1779becd9849SNeel Natu 			KASSERT(vmexit->exitcode != VM_EXITCODE_BOGUS,
1780becd9849SNeel Natu 			    ("emulate_wrmsr retu with bogus exitcode"));
1781becd9849SNeel Natu 		}
1782366f6083SPeter Grehan 		break;
1783366f6083SPeter Grehan 	case EXIT_REASON_WRMSR:
1784b5aaf7b2SNeel Natu 		vmm_stat_incr(vmx->vm, vcpu, VMEXIT_WRMSR, 1);
1785becd9849SNeel Natu 		retu = false;
1786366f6083SPeter Grehan 		eax = vmxctx->guest_rax;
1787366f6083SPeter Grehan 		ecx = vmxctx->guest_rcx;
1788366f6083SPeter Grehan 		edx = vmxctx->guest_rdx;
1789b42206f3SNeel Natu 		error = emulate_wrmsr(vmx->vm, vcpu, ecx,
1790becd9849SNeel Natu 		    (uint64_t)edx << 32 | eax, &retu);
1791b42206f3SNeel Natu 		if (error) {
1792366f6083SPeter Grehan 			vmexit->exitcode = VM_EXITCODE_WRMSR;
1793366f6083SPeter Grehan 			vmexit->u.msr.code = ecx;
1794366f6083SPeter Grehan 			vmexit->u.msr.wval = (uint64_t)edx << 32 | eax;
1795becd9849SNeel Natu 		} else if (!retu) {
1796a0efd3fbSJohn Baldwin 			handled = HANDLED;
1797becd9849SNeel Natu 		} else {
1798becd9849SNeel Natu 			/* Return to userspace with a valid exitcode */
1799becd9849SNeel Natu 			KASSERT(vmexit->exitcode != VM_EXITCODE_BOGUS,
1800becd9849SNeel Natu 			    ("emulate_wrmsr retu with bogus exitcode"));
1801becd9849SNeel Natu 		}
1802366f6083SPeter Grehan 		break;
1803366f6083SPeter Grehan 	case EXIT_REASON_HLT:
1804f76fc5d4SNeel Natu 		vmm_stat_incr(vmx->vm, vcpu, VMEXIT_HLT, 1);
1805366f6083SPeter Grehan 		vmexit->exitcode = VM_EXITCODE_HLT;
18063de83862SNeel Natu 		vmexit->u.hlt.rflags = vmcs_read(VMCS_GUEST_RFLAGS);
1807366f6083SPeter Grehan 		break;
1808366f6083SPeter Grehan 	case EXIT_REASON_MTF:
1809b5aaf7b2SNeel Natu 		vmm_stat_incr(vmx->vm, vcpu, VMEXIT_MTRAP, 1);
1810366f6083SPeter Grehan 		vmexit->exitcode = VM_EXITCODE_MTRAP;
1811366f6083SPeter Grehan 		break;
1812366f6083SPeter Grehan 	case EXIT_REASON_PAUSE:
1813b5aaf7b2SNeel Natu 		vmm_stat_incr(vmx->vm, vcpu, VMEXIT_PAUSE, 1);
1814366f6083SPeter Grehan 		vmexit->exitcode = VM_EXITCODE_PAUSE;
1815366f6083SPeter Grehan 		break;
1816366f6083SPeter Grehan 	case EXIT_REASON_INTR_WINDOW:
1817b5aaf7b2SNeel Natu 		vmm_stat_incr(vmx->vm, vcpu, VMEXIT_INTR_WINDOW, 1);
1818366f6083SPeter Grehan 		vmx_clear_int_window_exiting(vmx, vcpu);
1819b5aaf7b2SNeel Natu 		return (1);
1820366f6083SPeter Grehan 	case EXIT_REASON_EXT_INTR:
1821366f6083SPeter Grehan 		/*
1822366f6083SPeter Grehan 		 * External interrupts serve only to cause VM exits and allow
1823366f6083SPeter Grehan 		 * the host interrupt handler to run.
1824366f6083SPeter Grehan 		 *
1825366f6083SPeter Grehan 		 * If this external interrupt triggers a virtual interrupt
1826366f6083SPeter Grehan 		 * to a VM, then that state will be recorded by the
1827366f6083SPeter Grehan 		 * host interrupt handler in the VM's softc. We will inject
1828366f6083SPeter Grehan 		 * this virtual interrupt during the subsequent VM enter.
1829366f6083SPeter Grehan 		 */
1830f7d47425SNeel Natu 		intr_info = vmcs_read(VMCS_EXIT_INTR_INFO);
1831722b6744SJohn Baldwin 
1832722b6744SJohn Baldwin 		/*
1833722b6744SJohn Baldwin 		 * XXX: Ignore this exit if VMCS_INTR_VALID is not set.
1834ad3e3687SJohn Baldwin 		 * This appears to be a bug in VMware Fusion?
1835722b6744SJohn Baldwin 		 */
1836722b6744SJohn Baldwin 		if (!(intr_info & VMCS_INTR_VALID))
1837722b6744SJohn Baldwin 			return (1);
1838160471d2SNeel Natu 		KASSERT((intr_info & VMCS_INTR_VALID) != 0 &&
1839160471d2SNeel Natu 		    (intr_info & VMCS_INTR_T_MASK) == VMCS_INTR_T_HWINTR,
1840f7d47425SNeel Natu 		    ("VM exit interruption info invalid: %#x", intr_info));
1841f7d47425SNeel Natu 		vmx_trigger_hostintr(intr_info & 0xff);
1842366f6083SPeter Grehan 
1843366f6083SPeter Grehan 		/*
1844366f6083SPeter Grehan 		 * This is special. We want to treat this as an 'handled'
1845366f6083SPeter Grehan 		 * VM-exit but not increment the instruction pointer.
1846366f6083SPeter Grehan 		 */
1847366f6083SPeter Grehan 		vmm_stat_incr(vmx->vm, vcpu, VMEXIT_EXTINT, 1);
1848366f6083SPeter Grehan 		return (1);
1849366f6083SPeter Grehan 	case EXIT_REASON_NMI_WINDOW:
1850366f6083SPeter Grehan 		/* Exit to allow the pending virtual NMI to be injected */
185148b2d828SNeel Natu 		if (vm_nmi_pending(vmx->vm, vcpu))
185248b2d828SNeel Natu 			vmx_inject_nmi(vmx, vcpu);
1853366f6083SPeter Grehan 		vmx_clear_nmi_window_exiting(vmx, vcpu);
185448b2d828SNeel Natu 		vmm_stat_incr(vmx->vm, vcpu, VMEXIT_NMI_WINDOW, 1);
1855366f6083SPeter Grehan 		return (1);
1856366f6083SPeter Grehan 	case EXIT_REASON_INOUT:
1857b5aaf7b2SNeel Natu 		vmm_stat_incr(vmx->vm, vcpu, VMEXIT_INOUT, 1);
1858366f6083SPeter Grehan 		vmexit->exitcode = VM_EXITCODE_INOUT;
1859366f6083SPeter Grehan 		vmexit->u.inout.bytes = (qual & 0x7) + 1;
1860366f6083SPeter Grehan 		vmexit->u.inout.in = (qual & 0x8) ? 1 : 0;
1861366f6083SPeter Grehan 		vmexit->u.inout.string = (qual & 0x10) ? 1 : 0;
1862366f6083SPeter Grehan 		vmexit->u.inout.rep = (qual & 0x20) ? 1 : 0;
1863366f6083SPeter Grehan 		vmexit->u.inout.port = (uint16_t)(qual >> 16);
1864366f6083SPeter Grehan 		vmexit->u.inout.eax = (uint32_t)(vmxctx->guest_rax);
1865*762fd208STycho Nightingale 		error = emulate_ioport(vmx->vm, vcpu, vmexit);
1866*762fd208STycho Nightingale 		if (error == 0)  {
1867*762fd208STycho Nightingale 			handled = 1;
1868*762fd208STycho Nightingale 			vmxctx->guest_rax = vmexit->u.inout.eax;
1869*762fd208STycho Nightingale 		}
1870366f6083SPeter Grehan 		break;
1871366f6083SPeter Grehan 	case EXIT_REASON_CPUID:
1872b5aaf7b2SNeel Natu 		vmm_stat_incr(vmx->vm, vcpu, VMEXIT_CPUID, 1);
1873a2da7af6SNeel Natu 		handled = vmx_handle_cpuid(vmx->vm, vcpu, vmxctx);
1874366f6083SPeter Grehan 		break;
1875e5a1d950SNeel Natu 	case EXIT_REASON_EXCEPTION:
1876c308b23bSNeel Natu 		vmm_stat_incr(vmx->vm, vcpu, VMEXIT_EXCEPTION, 1);
1877e5a1d950SNeel Natu 		intr_info = vmcs_read(VMCS_EXIT_INTR_INFO);
1878e5a1d950SNeel Natu 		KASSERT((intr_info & VMCS_INTR_VALID) != 0,
1879e5a1d950SNeel Natu 		    ("VM exit interruption info invalid: %#x", intr_info));
1880c308b23bSNeel Natu 
1881e5a1d950SNeel Natu 		/*
1882e5a1d950SNeel Natu 		 * If Virtual NMIs control is 1 and the VM-exit is due to a
1883e5a1d950SNeel Natu 		 * fault encountered during the execution of IRET then we must
1884e5a1d950SNeel Natu 		 * restore the state of "virtual-NMI blocking" before resuming
1885e5a1d950SNeel Natu 		 * the guest.
1886e5a1d950SNeel Natu 		 *
1887e5a1d950SNeel Natu 		 * See "Resuming Guest Software after Handling an Exception".
1888e5a1d950SNeel Natu 		 */
1889e5a1d950SNeel Natu 		if ((idtvec_info & VMCS_IDT_VEC_VALID) == 0 &&
1890e5a1d950SNeel Natu 		    (intr_info & 0xff) != IDT_DF &&
1891e5a1d950SNeel Natu 		    (intr_info & EXIT_QUAL_NMIUDTI) != 0)
1892e5a1d950SNeel Natu 			vmx_restore_nmi_blocking(vmx, vcpu);
1893c308b23bSNeel Natu 
1894c308b23bSNeel Natu 		/*
189562fbd7c2SNeel Natu 		 * The NMI has already been handled in vmx_exit_handle_nmi().
1896c308b23bSNeel Natu 		 */
189762fbd7c2SNeel Natu 		if ((intr_info & VMCS_INTR_T_MASK) == VMCS_INTR_T_NMI)
1898c308b23bSNeel Natu 			return (1);
1899e5a1d950SNeel Natu 		break;
1900cd942e0fSPeter Grehan 	case EXIT_REASON_EPT_FAULT:
1901318224bbSNeel Natu 		/*
1902318224bbSNeel Natu 		 * If 'gpa' lies within the address space allocated to
1903318224bbSNeel Natu 		 * memory then this must be a nested page fault otherwise
1904318224bbSNeel Natu 		 * this must be an instruction that accesses MMIO space.
1905318224bbSNeel Natu 		 */
1906a2da7af6SNeel Natu 		gpa = vmcs_gpa();
1907159dd56fSNeel Natu 		if (vm_mem_allocated(vmx->vm, gpa) ||
1908159dd56fSNeel Natu 		    apic_access_fault(vmx, vcpu, gpa)) {
1909cd942e0fSPeter Grehan 			vmexit->exitcode = VM_EXITCODE_PAGING;
191013ec9371SPeter Grehan 			vmexit->u.paging.gpa = gpa;
1911318224bbSNeel Natu 			vmexit->u.paging.fault_type = ept_fault_type(qual);
1912bf73979dSNeel Natu 			vmm_stat_incr(vmx->vm, vcpu, VMEXIT_NESTED_FAULT, 1);
1913318224bbSNeel Natu 		} else if (ept_emulation_fault(qual)) {
1914318224bbSNeel Natu 			vmexit->exitcode = VM_EXITCODE_INST_EMUL;
1915318224bbSNeel Natu 			vmexit->u.inst_emul.gpa = gpa;
1916318224bbSNeel Natu 			vmexit->u.inst_emul.gla = vmcs_gla();
1917318224bbSNeel Natu 			vmexit->u.inst_emul.cr3 = vmcs_guest_cr3();
191800f3efe1SJohn Baldwin 			vmexit->u.inst_emul.cpu_mode = vmx_cpu_mode();
191900f3efe1SJohn Baldwin 			vmexit->u.inst_emul.paging_mode = vmx_paging_mode();
1920bf73979dSNeel Natu 			vmm_stat_incr(vmx->vm, vcpu, VMEXIT_INST_EMUL, 1);
1921a2da7af6SNeel Natu 		}
1922e5a1d950SNeel Natu 		/*
1923e5a1d950SNeel Natu 		 * If Virtual NMIs control is 1 and the VM-exit is due to an
1924e5a1d950SNeel Natu 		 * EPT fault during the execution of IRET then we must restore
1925e5a1d950SNeel Natu 		 * the state of "virtual-NMI blocking" before resuming.
1926e5a1d950SNeel Natu 		 *
1927e5a1d950SNeel Natu 		 * See description of "NMI unblocking due to IRET" in
1928e5a1d950SNeel Natu 		 * "Exit Qualification for EPT Violations".
1929e5a1d950SNeel Natu 		 */
1930e5a1d950SNeel Natu 		if ((idtvec_info & VMCS_IDT_VEC_VALID) == 0 &&
1931e5a1d950SNeel Natu 		    (qual & EXIT_QUAL_NMIUDTI) != 0)
1932e5a1d950SNeel Natu 			vmx_restore_nmi_blocking(vmx, vcpu);
1933cd942e0fSPeter Grehan 		break;
193430b94db8SNeel Natu 	case EXIT_REASON_VIRTUALIZED_EOI:
193530b94db8SNeel Natu 		vmexit->exitcode = VM_EXITCODE_IOAPIC_EOI;
193630b94db8SNeel Natu 		vmexit->u.ioapic_eoi.vector = qual & 0xFF;
193730b94db8SNeel Natu 		vmexit->inst_length = 0;	/* trap-like */
193830b94db8SNeel Natu 		break;
193988c4b8d1SNeel Natu 	case EXIT_REASON_APIC_ACCESS:
194088c4b8d1SNeel Natu 		handled = vmx_handle_apic_access(vmx, vcpu, vmexit);
194188c4b8d1SNeel Natu 		break;
194288c4b8d1SNeel Natu 	case EXIT_REASON_APIC_WRITE:
194388c4b8d1SNeel Natu 		/*
194488c4b8d1SNeel Natu 		 * APIC-write VM exit is trap-like so the %rip is already
194588c4b8d1SNeel Natu 		 * pointing to the next instruction.
194688c4b8d1SNeel Natu 		 */
194788c4b8d1SNeel Natu 		vmexit->inst_length = 0;
194888c4b8d1SNeel Natu 		vlapic = vm_lapic(vmx->vm, vcpu);
1949159dd56fSNeel Natu 		handled = vmx_handle_apic_write(vmx, vcpu, vlapic, qual);
195088c4b8d1SNeel Natu 		break;
1951abb023fbSJohn Baldwin 	case EXIT_REASON_XSETBV:
1952a0efd3fbSJohn Baldwin 		handled = vmx_emulate_xsetbv(vmx, vcpu, vmexit);
1953abb023fbSJohn Baldwin 		break;
1954366f6083SPeter Grehan 	default:
1955b5aaf7b2SNeel Natu 		vmm_stat_incr(vmx->vm, vcpu, VMEXIT_UNKNOWN, 1);
1956366f6083SPeter Grehan 		break;
1957366f6083SPeter Grehan 	}
1958366f6083SPeter Grehan 
1959366f6083SPeter Grehan 	if (handled) {
1960366f6083SPeter Grehan 		/*
1961366f6083SPeter Grehan 		 * It is possible that control is returned to userland
1962366f6083SPeter Grehan 		 * even though we were able to handle the VM exit in the
1963eeefa4e4SNeel Natu 		 * kernel.
1964366f6083SPeter Grehan 		 *
1965366f6083SPeter Grehan 		 * In such a case we want to make sure that the userland
1966366f6083SPeter Grehan 		 * restarts guest execution at the instruction *after*
1967366f6083SPeter Grehan 		 * the one we just processed. Therefore we update the
1968366f6083SPeter Grehan 		 * guest rip in the VMCS and in 'vmexit'.
1969366f6083SPeter Grehan 		 */
1970366f6083SPeter Grehan 		vmexit->rip += vmexit->inst_length;
1971366f6083SPeter Grehan 		vmexit->inst_length = 0;
19723de83862SNeel Natu 		vmcs_write(VMCS_GUEST_RIP, vmexit->rip);
1973366f6083SPeter Grehan 	} else {
1974366f6083SPeter Grehan 		if (vmexit->exitcode == VM_EXITCODE_BOGUS) {
1975366f6083SPeter Grehan 			/*
1976366f6083SPeter Grehan 			 * If this VM exit was not claimed by anybody then
1977366f6083SPeter Grehan 			 * treat it as a generic VMX exit.
1978366f6083SPeter Grehan 			 */
1979366f6083SPeter Grehan 			vmexit->exitcode = VM_EXITCODE_VMX;
19800492757cSNeel Natu 			vmexit->u.vmx.status = VM_SUCCESS;
1981c308b23bSNeel Natu 			vmexit->u.vmx.inst_type = 0;
1982c308b23bSNeel Natu 			vmexit->u.vmx.inst_error = 0;
1983366f6083SPeter Grehan 		} else {
1984366f6083SPeter Grehan 			/*
1985366f6083SPeter Grehan 			 * The exitcode and collateral have been populated.
1986366f6083SPeter Grehan 			 * The VM exit will be processed further in userland.
1987366f6083SPeter Grehan 			 */
1988366f6083SPeter Grehan 		}
1989366f6083SPeter Grehan 	}
1990366f6083SPeter Grehan 	return (handled);
1991366f6083SPeter Grehan }
1992366f6083SPeter Grehan 
19930492757cSNeel Natu static __inline int
19940492757cSNeel Natu vmx_exit_astpending(struct vmx *vmx, int vcpu, struct vm_exit *vmexit)
1995366f6083SPeter Grehan {
19960492757cSNeel Natu 
19970492757cSNeel Natu 	vmexit->rip = vmcs_guest_rip();
19980492757cSNeel Natu 	vmexit->inst_length = 0;
19990492757cSNeel Natu 	vmexit->exitcode = VM_EXITCODE_BOGUS;
20000492757cSNeel Natu 	vmx_astpending_trace(vmx, vcpu, vmexit->rip);
20010492757cSNeel Natu 	vmm_stat_incr(vmx->vm, vcpu, VMEXIT_ASTPENDING, 1);
20020492757cSNeel Natu 
20030492757cSNeel Natu 	return (HANDLED);
20040492757cSNeel Natu }
20050492757cSNeel Natu 
20060492757cSNeel Natu static __inline int
20075b8a8cd1SNeel Natu vmx_exit_rendezvous(struct vmx *vmx, int vcpu, struct vm_exit *vmexit)
20085b8a8cd1SNeel Natu {
20095b8a8cd1SNeel Natu 
20105b8a8cd1SNeel Natu 	vmexit->rip = vmcs_guest_rip();
20115b8a8cd1SNeel Natu 	vmexit->inst_length = 0;
20125b8a8cd1SNeel Natu 	vmexit->exitcode = VM_EXITCODE_RENDEZVOUS;
20135b8a8cd1SNeel Natu 	vmm_stat_incr(vmx->vm, vcpu, VMEXIT_RENDEZVOUS, 1);
20145b8a8cd1SNeel Natu 
20155b8a8cd1SNeel Natu 	return (UNHANDLED);
20165b8a8cd1SNeel Natu }
20175b8a8cd1SNeel Natu 
20185b8a8cd1SNeel Natu static __inline int
20190492757cSNeel Natu vmx_exit_inst_error(struct vmxctx *vmxctx, int rc, struct vm_exit *vmexit)
20200492757cSNeel Natu {
20210492757cSNeel Natu 
20220492757cSNeel Natu 	KASSERT(vmxctx->inst_fail_status != VM_SUCCESS,
20230492757cSNeel Natu 	    ("vmx_exit_inst_error: invalid inst_fail_status %d",
20240492757cSNeel Natu 	    vmxctx->inst_fail_status));
20250492757cSNeel Natu 
20260492757cSNeel Natu 	vmexit->inst_length = 0;
20270492757cSNeel Natu 	vmexit->exitcode = VM_EXITCODE_VMX;
20280492757cSNeel Natu 	vmexit->u.vmx.status = vmxctx->inst_fail_status;
20290492757cSNeel Natu 	vmexit->u.vmx.inst_error = vmcs_instruction_error();
20300492757cSNeel Natu 	vmexit->u.vmx.exit_reason = ~0;
20310492757cSNeel Natu 	vmexit->u.vmx.exit_qualification = ~0;
20320492757cSNeel Natu 
20330492757cSNeel Natu 	switch (rc) {
20340492757cSNeel Natu 	case VMX_VMRESUME_ERROR:
20350492757cSNeel Natu 	case VMX_VMLAUNCH_ERROR:
20360492757cSNeel Natu 	case VMX_INVEPT_ERROR:
20370492757cSNeel Natu 		vmexit->u.vmx.inst_type = rc;
20380492757cSNeel Natu 		break;
20390492757cSNeel Natu 	default:
20400492757cSNeel Natu 		panic("vm_exit_inst_error: vmx_enter_guest returned %d", rc);
20410492757cSNeel Natu 	}
20420492757cSNeel Natu 
20430492757cSNeel Natu 	return (UNHANDLED);
20440492757cSNeel Natu }
20450492757cSNeel Natu 
204662fbd7c2SNeel Natu /*
204762fbd7c2SNeel Natu  * If the NMI-exiting VM execution control is set to '1' then an NMI in
204862fbd7c2SNeel Natu  * non-root operation causes a VM-exit. NMI blocking is in effect so it is
204962fbd7c2SNeel Natu  * sufficient to simply vector to the NMI handler via a software interrupt.
205062fbd7c2SNeel Natu  * However, this must be done before maskable interrupts are enabled
205162fbd7c2SNeel Natu  * otherwise the "iret" issued by an interrupt handler will incorrectly
205262fbd7c2SNeel Natu  * clear NMI blocking.
205362fbd7c2SNeel Natu  */
205462fbd7c2SNeel Natu static __inline void
205562fbd7c2SNeel Natu vmx_exit_handle_nmi(struct vmx *vmx, int vcpuid, struct vm_exit *vmexit)
205662fbd7c2SNeel Natu {
205762fbd7c2SNeel Natu 	uint32_t intr_info;
205862fbd7c2SNeel Natu 
205962fbd7c2SNeel Natu 	KASSERT((read_rflags() & PSL_I) == 0, ("interrupts enabled"));
206062fbd7c2SNeel Natu 
206162fbd7c2SNeel Natu 	if (vmexit->u.vmx.exit_reason != EXIT_REASON_EXCEPTION)
206262fbd7c2SNeel Natu 		return;
206362fbd7c2SNeel Natu 
206462fbd7c2SNeel Natu 	intr_info = vmcs_read(VMCS_EXIT_INTR_INFO);
206562fbd7c2SNeel Natu 	KASSERT((intr_info & VMCS_INTR_VALID) != 0,
206662fbd7c2SNeel Natu 	    ("VM exit interruption info invalid: %#x", intr_info));
206762fbd7c2SNeel Natu 
206862fbd7c2SNeel Natu 	if ((intr_info & VMCS_INTR_T_MASK) == VMCS_INTR_T_NMI) {
206962fbd7c2SNeel Natu 		KASSERT((intr_info & 0xff) == IDT_NMI, ("VM exit due "
207062fbd7c2SNeel Natu 		    "to NMI has invalid vector: %#x", intr_info));
207162fbd7c2SNeel Natu 		VCPU_CTR0(vmx->vm, vcpuid, "Vectoring to NMI handler");
207262fbd7c2SNeel Natu 		__asm __volatile("int $2");
207362fbd7c2SNeel Natu 	}
207462fbd7c2SNeel Natu }
207562fbd7c2SNeel Natu 
20760492757cSNeel Natu static int
20775b8a8cd1SNeel Natu vmx_run(void *arg, int vcpu, register_t startrip, pmap_t pmap,
20785b8a8cd1SNeel Natu     void *rendezvous_cookie)
20790492757cSNeel Natu {
20800492757cSNeel Natu 	int rc, handled, launched;
2081366f6083SPeter Grehan 	struct vmx *vmx;
20825b8a8cd1SNeel Natu 	struct vm *vm;
2083366f6083SPeter Grehan 	struct vmxctx *vmxctx;
2084366f6083SPeter Grehan 	struct vmcs *vmcs;
208598ed632cSNeel Natu 	struct vm_exit *vmexit;
2086de5ea6b6SNeel Natu 	struct vlapic *vlapic;
208779c59630SNeel Natu 	uint64_t rip;
208879c59630SNeel Natu 	uint32_t exit_reason;
2089366f6083SPeter Grehan 
2090366f6083SPeter Grehan 	vmx = arg;
20915b8a8cd1SNeel Natu 	vm = vmx->vm;
2092366f6083SPeter Grehan 	vmcs = &vmx->vmcs[vcpu];
2093366f6083SPeter Grehan 	vmxctx = &vmx->ctx[vcpu];
20945b8a8cd1SNeel Natu 	vlapic = vm_lapic(vm, vcpu);
20955b8a8cd1SNeel Natu 	vmexit = vm_exitinfo(vm, vcpu);
20960492757cSNeel Natu 	launched = 0;
209798ed632cSNeel Natu 
2098318224bbSNeel Natu 	KASSERT(vmxctx->pmap == pmap,
2099318224bbSNeel Natu 	    ("pmap %p different than ctx pmap %p", pmap, vmxctx->pmap));
2100318224bbSNeel Natu 
2101366f6083SPeter Grehan 	VMPTRLD(vmcs);
2102366f6083SPeter Grehan 
2103366f6083SPeter Grehan 	/*
2104366f6083SPeter Grehan 	 * XXX
2105366f6083SPeter Grehan 	 * We do this every time because we may setup the virtual machine
2106366f6083SPeter Grehan 	 * from a different process than the one that actually runs it.
2107366f6083SPeter Grehan 	 *
2108366f6083SPeter Grehan 	 * If the life of a virtual machine was spent entirely in the context
2109c847a506SNeel Natu 	 * of a single process we could do this once in vmx_vminit().
2110366f6083SPeter Grehan 	 */
21113de83862SNeel Natu 	vmcs_write(VMCS_HOST_CR3, rcr3());
2112366f6083SPeter Grehan 
21130492757cSNeel Natu 	vmcs_write(VMCS_GUEST_RIP, startrip);
2114953c2c47SNeel Natu 	vmx_set_pcpu_defaults(vmx, vcpu, pmap);
2115366f6083SPeter Grehan 	do {
21160492757cSNeel Natu 		/*
21170492757cSNeel Natu 		 * Interrupts are disabled from this point on until the
21180492757cSNeel Natu 		 * guest starts executing. This is done for the following
21190492757cSNeel Natu 		 * reasons:
21200492757cSNeel Natu 		 *
21210492757cSNeel Natu 		 * If an AST is asserted on this thread after the check below,
21220492757cSNeel Natu 		 * then the IPI_AST notification will not be lost, because it
21230492757cSNeel Natu 		 * will cause a VM exit due to external interrupt as soon as
21240492757cSNeel Natu 		 * the guest state is loaded.
21250492757cSNeel Natu 		 *
21260492757cSNeel Natu 		 * A posted interrupt after 'vmx_inject_interrupts()' will
21270492757cSNeel Natu 		 * not be "lost" because it will be held pending in the host
21280492757cSNeel Natu 		 * APIC because interrupts are disabled. The pending interrupt
21290492757cSNeel Natu 		 * will be recognized as soon as the guest state is loaded.
21300492757cSNeel Natu 		 *
21310492757cSNeel Natu 		 * The same reasoning applies to the IPI generated by
21320492757cSNeel Natu 		 * pmap_invalidate_ept().
21330492757cSNeel Natu 		 */
21340492757cSNeel Natu 		disable_intr();
21350492757cSNeel Natu 		if (curthread->td_flags & (TDF_ASTPENDING | TDF_NEEDRESCHED)) {
21360492757cSNeel Natu 			enable_intr();
21370492757cSNeel Natu 			handled = vmx_exit_astpending(vmx, vcpu, vmexit);
21380492757cSNeel Natu 			break;
21390492757cSNeel Natu 		}
21400492757cSNeel Natu 
21415b8a8cd1SNeel Natu 		if (vcpu_rendezvous_pending(rendezvous_cookie)) {
21425b8a8cd1SNeel Natu 			enable_intr();
21435b8a8cd1SNeel Natu 			handled = vmx_exit_rendezvous(vmx, vcpu, vmexit);
21445b8a8cd1SNeel Natu 			break;
21455b8a8cd1SNeel Natu 		}
21465b8a8cd1SNeel Natu 
2147de5ea6b6SNeel Natu 		vmx_inject_interrupts(vmx, vcpu, vlapic);
2148366f6083SPeter Grehan 		vmx_run_trace(vmx, vcpu);
2149953c2c47SNeel Natu 		rc = vmx_enter_guest(vmxctx, vmx, launched);
215079c59630SNeel Natu 
215179c59630SNeel Natu 		/* Collect some information for VM exit processing */
215279c59630SNeel Natu 		vmexit->rip = rip = vmcs_guest_rip();
215379c59630SNeel Natu 		vmexit->inst_length = vmexit_instruction_length();
215479c59630SNeel Natu 		vmexit->u.vmx.exit_reason = exit_reason = vmcs_exit_reason();
215579c59630SNeel Natu 		vmexit->u.vmx.exit_qualification = vmcs_exit_qualification();
215679c59630SNeel Natu 
21570492757cSNeel Natu 		if (rc == VMX_GUEST_VMEXIT) {
215862fbd7c2SNeel Natu 			vmx_exit_handle_nmi(vmx, vcpu, vmexit);
215962fbd7c2SNeel Natu 			enable_intr();
21600492757cSNeel Natu 			handled = vmx_exit_process(vmx, vcpu, vmexit);
21610492757cSNeel Natu 		} else {
216262fbd7c2SNeel Natu 			enable_intr();
21630492757cSNeel Natu 			handled = vmx_exit_inst_error(vmxctx, rc, vmexit);
2164eeefa4e4SNeel Natu 		}
216562fbd7c2SNeel Natu 		launched = 1;
216679c59630SNeel Natu 		vmx_exit_trace(vmx, vcpu, rip, exit_reason, handled);
2167eeefa4e4SNeel Natu 	} while (handled);
2168366f6083SPeter Grehan 
2169366f6083SPeter Grehan 	/*
2170366f6083SPeter Grehan 	 * If a VM exit has been handled then the exitcode must be BOGUS
2171366f6083SPeter Grehan 	 * If a VM exit is not handled then the exitcode must not be BOGUS
2172366f6083SPeter Grehan 	 */
2173366f6083SPeter Grehan 	if ((handled && vmexit->exitcode != VM_EXITCODE_BOGUS) ||
2174366f6083SPeter Grehan 	    (!handled && vmexit->exitcode == VM_EXITCODE_BOGUS)) {
2175366f6083SPeter Grehan 		panic("Mismatch between handled (%d) and exitcode (%d)",
2176366f6083SPeter Grehan 		      handled, vmexit->exitcode);
2177366f6083SPeter Grehan 	}
2178366f6083SPeter Grehan 
2179b5aaf7b2SNeel Natu 	if (!handled)
21805b8a8cd1SNeel Natu 		vmm_stat_incr(vm, vcpu, VMEXIT_USERSPACE, 1);
2181b5aaf7b2SNeel Natu 
21825b8a8cd1SNeel Natu 	VCPU_CTR1(vm, vcpu, "returning from vmx_run: exitcode %d",
21830492757cSNeel Natu 	    vmexit->exitcode);
2184366f6083SPeter Grehan 
2185366f6083SPeter Grehan 	VMCLEAR(vmcs);
2186366f6083SPeter Grehan 	return (0);
2187366f6083SPeter Grehan }
2188366f6083SPeter Grehan 
2189366f6083SPeter Grehan static void
2190366f6083SPeter Grehan vmx_vmcleanup(void *arg)
2191366f6083SPeter Grehan {
219245e51299SNeel Natu 	int i, error;
2193366f6083SPeter Grehan 	struct vmx *vmx = arg;
2194366f6083SPeter Grehan 
2195159dd56fSNeel Natu 	if (apic_access_virtualization(vmx, 0))
219688c4b8d1SNeel Natu 		vm_unmap_mmio(vmx->vm, DEFAULT_APIC_BASE, PAGE_SIZE);
219788c4b8d1SNeel Natu 
219845e51299SNeel Natu 	for (i = 0; i < VM_MAXCPU; i++)
219945e51299SNeel Natu 		vpid_free(vmx->state[i].vpid);
220045e51299SNeel Natu 
2201366f6083SPeter Grehan 	/*
2202366f6083SPeter Grehan 	 * XXXSMP we also need to clear the VMCS active on the other vcpus.
2203366f6083SPeter Grehan 	 */
2204366f6083SPeter Grehan 	error = vmclear(&vmx->vmcs[0]);
2205366f6083SPeter Grehan 	if (error != 0)
2206366f6083SPeter Grehan 		panic("vmx_vmcleanup: vmclear error %d on vcpu 0", error);
2207366f6083SPeter Grehan 
2208366f6083SPeter Grehan 	free(vmx, M_VMX);
2209366f6083SPeter Grehan 
2210366f6083SPeter Grehan 	return;
2211366f6083SPeter Grehan }
2212366f6083SPeter Grehan 
2213366f6083SPeter Grehan static register_t *
2214366f6083SPeter Grehan vmxctx_regptr(struct vmxctx *vmxctx, int reg)
2215366f6083SPeter Grehan {
2216366f6083SPeter Grehan 
2217366f6083SPeter Grehan 	switch (reg) {
2218366f6083SPeter Grehan 	case VM_REG_GUEST_RAX:
2219366f6083SPeter Grehan 		return (&vmxctx->guest_rax);
2220366f6083SPeter Grehan 	case VM_REG_GUEST_RBX:
2221366f6083SPeter Grehan 		return (&vmxctx->guest_rbx);
2222366f6083SPeter Grehan 	case VM_REG_GUEST_RCX:
2223366f6083SPeter Grehan 		return (&vmxctx->guest_rcx);
2224366f6083SPeter Grehan 	case VM_REG_GUEST_RDX:
2225366f6083SPeter Grehan 		return (&vmxctx->guest_rdx);
2226366f6083SPeter Grehan 	case VM_REG_GUEST_RSI:
2227366f6083SPeter Grehan 		return (&vmxctx->guest_rsi);
2228366f6083SPeter Grehan 	case VM_REG_GUEST_RDI:
2229366f6083SPeter Grehan 		return (&vmxctx->guest_rdi);
2230366f6083SPeter Grehan 	case VM_REG_GUEST_RBP:
2231366f6083SPeter Grehan 		return (&vmxctx->guest_rbp);
2232366f6083SPeter Grehan 	case VM_REG_GUEST_R8:
2233366f6083SPeter Grehan 		return (&vmxctx->guest_r8);
2234366f6083SPeter Grehan 	case VM_REG_GUEST_R9:
2235366f6083SPeter Grehan 		return (&vmxctx->guest_r9);
2236366f6083SPeter Grehan 	case VM_REG_GUEST_R10:
2237366f6083SPeter Grehan 		return (&vmxctx->guest_r10);
2238366f6083SPeter Grehan 	case VM_REG_GUEST_R11:
2239366f6083SPeter Grehan 		return (&vmxctx->guest_r11);
2240366f6083SPeter Grehan 	case VM_REG_GUEST_R12:
2241366f6083SPeter Grehan 		return (&vmxctx->guest_r12);
2242366f6083SPeter Grehan 	case VM_REG_GUEST_R13:
2243366f6083SPeter Grehan 		return (&vmxctx->guest_r13);
2244366f6083SPeter Grehan 	case VM_REG_GUEST_R14:
2245366f6083SPeter Grehan 		return (&vmxctx->guest_r14);
2246366f6083SPeter Grehan 	case VM_REG_GUEST_R15:
2247366f6083SPeter Grehan 		return (&vmxctx->guest_r15);
2248366f6083SPeter Grehan 	default:
2249366f6083SPeter Grehan 		break;
2250366f6083SPeter Grehan 	}
2251366f6083SPeter Grehan 	return (NULL);
2252366f6083SPeter Grehan }
2253366f6083SPeter Grehan 
2254366f6083SPeter Grehan static int
2255366f6083SPeter Grehan vmxctx_getreg(struct vmxctx *vmxctx, int reg, uint64_t *retval)
2256366f6083SPeter Grehan {
2257366f6083SPeter Grehan 	register_t *regp;
2258366f6083SPeter Grehan 
2259366f6083SPeter Grehan 	if ((regp = vmxctx_regptr(vmxctx, reg)) != NULL) {
2260366f6083SPeter Grehan 		*retval = *regp;
2261366f6083SPeter Grehan 		return (0);
2262366f6083SPeter Grehan 	} else
2263366f6083SPeter Grehan 		return (EINVAL);
2264366f6083SPeter Grehan }
2265366f6083SPeter Grehan 
2266366f6083SPeter Grehan static int
2267366f6083SPeter Grehan vmxctx_setreg(struct vmxctx *vmxctx, int reg, uint64_t val)
2268366f6083SPeter Grehan {
2269366f6083SPeter Grehan 	register_t *regp;
2270366f6083SPeter Grehan 
2271366f6083SPeter Grehan 	if ((regp = vmxctx_regptr(vmxctx, reg)) != NULL) {
2272366f6083SPeter Grehan 		*regp = val;
2273366f6083SPeter Grehan 		return (0);
2274366f6083SPeter Grehan 	} else
2275366f6083SPeter Grehan 		return (EINVAL);
2276366f6083SPeter Grehan }
2277366f6083SPeter Grehan 
2278366f6083SPeter Grehan static int
2279aaaa0656SPeter Grehan vmx_shadow_reg(int reg)
2280aaaa0656SPeter Grehan {
2281aaaa0656SPeter Grehan 	int shreg;
2282aaaa0656SPeter Grehan 
2283aaaa0656SPeter Grehan 	shreg = -1;
2284aaaa0656SPeter Grehan 
2285aaaa0656SPeter Grehan 	switch (reg) {
2286aaaa0656SPeter Grehan 	case VM_REG_GUEST_CR0:
2287aaaa0656SPeter Grehan 		shreg = VMCS_CR0_SHADOW;
2288aaaa0656SPeter Grehan                 break;
2289aaaa0656SPeter Grehan         case VM_REG_GUEST_CR4:
2290aaaa0656SPeter Grehan 		shreg = VMCS_CR4_SHADOW;
2291aaaa0656SPeter Grehan 		break;
2292aaaa0656SPeter Grehan 	default:
2293aaaa0656SPeter Grehan 		break;
2294aaaa0656SPeter Grehan 	}
2295aaaa0656SPeter Grehan 
2296aaaa0656SPeter Grehan 	return (shreg);
2297aaaa0656SPeter Grehan }
2298aaaa0656SPeter Grehan 
2299aaaa0656SPeter Grehan static int
2300366f6083SPeter Grehan vmx_getreg(void *arg, int vcpu, int reg, uint64_t *retval)
2301366f6083SPeter Grehan {
2302d3c11f40SPeter Grehan 	int running, hostcpu;
2303366f6083SPeter Grehan 	struct vmx *vmx = arg;
2304366f6083SPeter Grehan 
2305d3c11f40SPeter Grehan 	running = vcpu_is_running(vmx->vm, vcpu, &hostcpu);
2306d3c11f40SPeter Grehan 	if (running && hostcpu != curcpu)
2307d3c11f40SPeter Grehan 		panic("vmx_getreg: %s%d is running", vm_name(vmx->vm), vcpu);
2308d3c11f40SPeter Grehan 
2309366f6083SPeter Grehan 	if (vmxctx_getreg(&vmx->ctx[vcpu], reg, retval) == 0)
2310366f6083SPeter Grehan 		return (0);
2311366f6083SPeter Grehan 
2312d3c11f40SPeter Grehan 	return (vmcs_getreg(&vmx->vmcs[vcpu], running, reg, retval));
2313366f6083SPeter Grehan }
2314366f6083SPeter Grehan 
2315366f6083SPeter Grehan static int
2316366f6083SPeter Grehan vmx_setreg(void *arg, int vcpu, int reg, uint64_t val)
2317366f6083SPeter Grehan {
2318aaaa0656SPeter Grehan 	int error, hostcpu, running, shadow;
2319366f6083SPeter Grehan 	uint64_t ctls;
2320366f6083SPeter Grehan 	struct vmx *vmx = arg;
2321366f6083SPeter Grehan 
2322d3c11f40SPeter Grehan 	running = vcpu_is_running(vmx->vm, vcpu, &hostcpu);
2323d3c11f40SPeter Grehan 	if (running && hostcpu != curcpu)
2324d3c11f40SPeter Grehan 		panic("vmx_setreg: %s%d is running", vm_name(vmx->vm), vcpu);
2325d3c11f40SPeter Grehan 
2326366f6083SPeter Grehan 	if (vmxctx_setreg(&vmx->ctx[vcpu], reg, val) == 0)
2327366f6083SPeter Grehan 		return (0);
2328366f6083SPeter Grehan 
2329d3c11f40SPeter Grehan 	error = vmcs_setreg(&vmx->vmcs[vcpu], running, reg, val);
2330366f6083SPeter Grehan 
2331366f6083SPeter Grehan 	if (error == 0) {
2332366f6083SPeter Grehan 		/*
2333366f6083SPeter Grehan 		 * If the "load EFER" VM-entry control is 1 then the
2334366f6083SPeter Grehan 		 * value of EFER.LMA must be identical to "IA-32e mode guest"
2335366f6083SPeter Grehan 		 * bit in the VM-entry control.
2336366f6083SPeter Grehan 		 */
2337366f6083SPeter Grehan 		if ((entry_ctls & VM_ENTRY_LOAD_EFER) != 0 &&
2338366f6083SPeter Grehan 		    (reg == VM_REG_GUEST_EFER)) {
2339d3c11f40SPeter Grehan 			vmcs_getreg(&vmx->vmcs[vcpu], running,
2340366f6083SPeter Grehan 				    VMCS_IDENT(VMCS_ENTRY_CTLS), &ctls);
2341366f6083SPeter Grehan 			if (val & EFER_LMA)
2342366f6083SPeter Grehan 				ctls |= VM_ENTRY_GUEST_LMA;
2343366f6083SPeter Grehan 			else
2344366f6083SPeter Grehan 				ctls &= ~VM_ENTRY_GUEST_LMA;
2345d3c11f40SPeter Grehan 			vmcs_setreg(&vmx->vmcs[vcpu], running,
2346366f6083SPeter Grehan 				    VMCS_IDENT(VMCS_ENTRY_CTLS), ctls);
2347366f6083SPeter Grehan 		}
2348aaaa0656SPeter Grehan 
2349aaaa0656SPeter Grehan 		shadow = vmx_shadow_reg(reg);
2350aaaa0656SPeter Grehan 		if (shadow > 0) {
2351aaaa0656SPeter Grehan 			/*
2352aaaa0656SPeter Grehan 			 * Store the unmodified value in the shadow
2353aaaa0656SPeter Grehan 			 */
2354aaaa0656SPeter Grehan 			error = vmcs_setreg(&vmx->vmcs[vcpu], running,
2355aaaa0656SPeter Grehan 				    VMCS_IDENT(shadow), val);
2356aaaa0656SPeter Grehan 		}
2357366f6083SPeter Grehan 	}
2358366f6083SPeter Grehan 
2359366f6083SPeter Grehan 	return (error);
2360366f6083SPeter Grehan }
2361366f6083SPeter Grehan 
2362366f6083SPeter Grehan static int
2363366f6083SPeter Grehan vmx_getdesc(void *arg, int vcpu, int reg, struct seg_desc *desc)
2364366f6083SPeter Grehan {
2365366f6083SPeter Grehan 	struct vmx *vmx = arg;
2366366f6083SPeter Grehan 
2367366f6083SPeter Grehan 	return (vmcs_getdesc(&vmx->vmcs[vcpu], reg, desc));
2368366f6083SPeter Grehan }
2369366f6083SPeter Grehan 
2370366f6083SPeter Grehan static int
2371366f6083SPeter Grehan vmx_setdesc(void *arg, int vcpu, int reg, struct seg_desc *desc)
2372366f6083SPeter Grehan {
2373366f6083SPeter Grehan 	struct vmx *vmx = arg;
2374366f6083SPeter Grehan 
2375366f6083SPeter Grehan 	return (vmcs_setdesc(&vmx->vmcs[vcpu], reg, desc));
2376366f6083SPeter Grehan }
2377366f6083SPeter Grehan 
2378366f6083SPeter Grehan static int
2379366f6083SPeter Grehan vmx_getcap(void *arg, int vcpu, int type, int *retval)
2380366f6083SPeter Grehan {
2381366f6083SPeter Grehan 	struct vmx *vmx = arg;
2382366f6083SPeter Grehan 	int vcap;
2383366f6083SPeter Grehan 	int ret;
2384366f6083SPeter Grehan 
2385366f6083SPeter Grehan 	ret = ENOENT;
2386366f6083SPeter Grehan 
2387366f6083SPeter Grehan 	vcap = vmx->cap[vcpu].set;
2388366f6083SPeter Grehan 
2389366f6083SPeter Grehan 	switch (type) {
2390366f6083SPeter Grehan 	case VM_CAP_HALT_EXIT:
2391366f6083SPeter Grehan 		if (cap_halt_exit)
2392366f6083SPeter Grehan 			ret = 0;
2393366f6083SPeter Grehan 		break;
2394366f6083SPeter Grehan 	case VM_CAP_PAUSE_EXIT:
2395366f6083SPeter Grehan 		if (cap_pause_exit)
2396366f6083SPeter Grehan 			ret = 0;
2397366f6083SPeter Grehan 		break;
2398366f6083SPeter Grehan 	case VM_CAP_MTRAP_EXIT:
2399366f6083SPeter Grehan 		if (cap_monitor_trap)
2400366f6083SPeter Grehan 			ret = 0;
2401366f6083SPeter Grehan 		break;
2402366f6083SPeter Grehan 	case VM_CAP_UNRESTRICTED_GUEST:
2403366f6083SPeter Grehan 		if (cap_unrestricted_guest)
2404366f6083SPeter Grehan 			ret = 0;
2405366f6083SPeter Grehan 		break;
240649cc03daSNeel Natu 	case VM_CAP_ENABLE_INVPCID:
240749cc03daSNeel Natu 		if (cap_invpcid)
240849cc03daSNeel Natu 			ret = 0;
240949cc03daSNeel Natu 		break;
2410366f6083SPeter Grehan 	default:
2411366f6083SPeter Grehan 		break;
2412366f6083SPeter Grehan 	}
2413366f6083SPeter Grehan 
2414366f6083SPeter Grehan 	if (ret == 0)
2415366f6083SPeter Grehan 		*retval = (vcap & (1 << type)) ? 1 : 0;
2416366f6083SPeter Grehan 
2417366f6083SPeter Grehan 	return (ret);
2418366f6083SPeter Grehan }
2419366f6083SPeter Grehan 
2420366f6083SPeter Grehan static int
2421366f6083SPeter Grehan vmx_setcap(void *arg, int vcpu, int type, int val)
2422366f6083SPeter Grehan {
2423366f6083SPeter Grehan 	struct vmx *vmx = arg;
2424366f6083SPeter Grehan 	struct vmcs *vmcs = &vmx->vmcs[vcpu];
2425366f6083SPeter Grehan 	uint32_t baseval;
2426366f6083SPeter Grehan 	uint32_t *pptr;
2427366f6083SPeter Grehan 	int error;
2428366f6083SPeter Grehan 	int flag;
2429366f6083SPeter Grehan 	int reg;
2430366f6083SPeter Grehan 	int retval;
2431366f6083SPeter Grehan 
2432366f6083SPeter Grehan 	retval = ENOENT;
2433366f6083SPeter Grehan 	pptr = NULL;
2434366f6083SPeter Grehan 
2435366f6083SPeter Grehan 	switch (type) {
2436366f6083SPeter Grehan 	case VM_CAP_HALT_EXIT:
2437366f6083SPeter Grehan 		if (cap_halt_exit) {
2438366f6083SPeter Grehan 			retval = 0;
2439366f6083SPeter Grehan 			pptr = &vmx->cap[vcpu].proc_ctls;
2440366f6083SPeter Grehan 			baseval = *pptr;
2441366f6083SPeter Grehan 			flag = PROCBASED_HLT_EXITING;
2442366f6083SPeter Grehan 			reg = VMCS_PRI_PROC_BASED_CTLS;
2443366f6083SPeter Grehan 		}
2444366f6083SPeter Grehan 		break;
2445366f6083SPeter Grehan 	case VM_CAP_MTRAP_EXIT:
2446366f6083SPeter Grehan 		if (cap_monitor_trap) {
2447366f6083SPeter Grehan 			retval = 0;
2448366f6083SPeter Grehan 			pptr = &vmx->cap[vcpu].proc_ctls;
2449366f6083SPeter Grehan 			baseval = *pptr;
2450366f6083SPeter Grehan 			flag = PROCBASED_MTF;
2451366f6083SPeter Grehan 			reg = VMCS_PRI_PROC_BASED_CTLS;
2452366f6083SPeter Grehan 		}
2453366f6083SPeter Grehan 		break;
2454366f6083SPeter Grehan 	case VM_CAP_PAUSE_EXIT:
2455366f6083SPeter Grehan 		if (cap_pause_exit) {
2456366f6083SPeter Grehan 			retval = 0;
2457366f6083SPeter Grehan 			pptr = &vmx->cap[vcpu].proc_ctls;
2458366f6083SPeter Grehan 			baseval = *pptr;
2459366f6083SPeter Grehan 			flag = PROCBASED_PAUSE_EXITING;
2460366f6083SPeter Grehan 			reg = VMCS_PRI_PROC_BASED_CTLS;
2461366f6083SPeter Grehan 		}
2462366f6083SPeter Grehan 		break;
2463366f6083SPeter Grehan 	case VM_CAP_UNRESTRICTED_GUEST:
2464366f6083SPeter Grehan 		if (cap_unrestricted_guest) {
2465366f6083SPeter Grehan 			retval = 0;
246649cc03daSNeel Natu 			pptr = &vmx->cap[vcpu].proc_ctls2;
246749cc03daSNeel Natu 			baseval = *pptr;
2468366f6083SPeter Grehan 			flag = PROCBASED2_UNRESTRICTED_GUEST;
2469366f6083SPeter Grehan 			reg = VMCS_SEC_PROC_BASED_CTLS;
2470366f6083SPeter Grehan 		}
2471366f6083SPeter Grehan 		break;
247249cc03daSNeel Natu 	case VM_CAP_ENABLE_INVPCID:
247349cc03daSNeel Natu 		if (cap_invpcid) {
247449cc03daSNeel Natu 			retval = 0;
247549cc03daSNeel Natu 			pptr = &vmx->cap[vcpu].proc_ctls2;
247649cc03daSNeel Natu 			baseval = *pptr;
247749cc03daSNeel Natu 			flag = PROCBASED2_ENABLE_INVPCID;
247849cc03daSNeel Natu 			reg = VMCS_SEC_PROC_BASED_CTLS;
247949cc03daSNeel Natu 		}
248049cc03daSNeel Natu 		break;
2481366f6083SPeter Grehan 	default:
2482366f6083SPeter Grehan 		break;
2483366f6083SPeter Grehan 	}
2484366f6083SPeter Grehan 
2485366f6083SPeter Grehan 	if (retval == 0) {
2486366f6083SPeter Grehan 		if (val) {
2487366f6083SPeter Grehan 			baseval |= flag;
2488366f6083SPeter Grehan 		} else {
2489366f6083SPeter Grehan 			baseval &= ~flag;
2490366f6083SPeter Grehan 		}
2491366f6083SPeter Grehan 		VMPTRLD(vmcs);
2492366f6083SPeter Grehan 		error = vmwrite(reg, baseval);
2493366f6083SPeter Grehan 		VMCLEAR(vmcs);
2494366f6083SPeter Grehan 
2495366f6083SPeter Grehan 		if (error) {
2496366f6083SPeter Grehan 			retval = error;
2497366f6083SPeter Grehan 		} else {
2498366f6083SPeter Grehan 			/*
2499366f6083SPeter Grehan 			 * Update optional stored flags, and record
2500366f6083SPeter Grehan 			 * setting
2501366f6083SPeter Grehan 			 */
2502366f6083SPeter Grehan 			if (pptr != NULL) {
2503366f6083SPeter Grehan 				*pptr = baseval;
2504366f6083SPeter Grehan 			}
2505366f6083SPeter Grehan 
2506366f6083SPeter Grehan 			if (val) {
2507366f6083SPeter Grehan 				vmx->cap[vcpu].set |= (1 << type);
2508366f6083SPeter Grehan 			} else {
2509366f6083SPeter Grehan 				vmx->cap[vcpu].set &= ~(1 << type);
2510366f6083SPeter Grehan 			}
2511366f6083SPeter Grehan 		}
2512366f6083SPeter Grehan 	}
2513366f6083SPeter Grehan 
2514366f6083SPeter Grehan         return (retval);
2515366f6083SPeter Grehan }
2516366f6083SPeter Grehan 
251788c4b8d1SNeel Natu struct vlapic_vtx {
251888c4b8d1SNeel Natu 	struct vlapic	vlapic;
2519176666c2SNeel Natu 	struct pir_desc	*pir_desc;
252030b94db8SNeel Natu 	struct vmx	*vmx;
252188c4b8d1SNeel Natu };
252288c4b8d1SNeel Natu 
252388c4b8d1SNeel Natu #define	VMX_CTR_PIR(vm, vcpuid, pir_desc, notify, vector, level, msg)	\
252488c4b8d1SNeel Natu do {									\
252588c4b8d1SNeel Natu 	VCPU_CTR2(vm, vcpuid, msg " assert %s-triggered vector %d",	\
252688c4b8d1SNeel Natu 	    level ? "level" : "edge", vector);				\
252788c4b8d1SNeel Natu 	VCPU_CTR1(vm, vcpuid, msg " pir0 0x%016lx", pir_desc->pir[0]);	\
252888c4b8d1SNeel Natu 	VCPU_CTR1(vm, vcpuid, msg " pir1 0x%016lx", pir_desc->pir[1]);	\
252988c4b8d1SNeel Natu 	VCPU_CTR1(vm, vcpuid, msg " pir2 0x%016lx", pir_desc->pir[2]);	\
253088c4b8d1SNeel Natu 	VCPU_CTR1(vm, vcpuid, msg " pir3 0x%016lx", pir_desc->pir[3]);	\
253188c4b8d1SNeel Natu 	VCPU_CTR1(vm, vcpuid, msg " notify: %s", notify ? "yes" : "no");\
253288c4b8d1SNeel Natu } while (0)
253388c4b8d1SNeel Natu 
253488c4b8d1SNeel Natu /*
253588c4b8d1SNeel Natu  * vlapic->ops handlers that utilize the APICv hardware assist described in
253688c4b8d1SNeel Natu  * Chapter 29 of the Intel SDM.
253788c4b8d1SNeel Natu  */
253888c4b8d1SNeel Natu static int
253988c4b8d1SNeel Natu vmx_set_intr_ready(struct vlapic *vlapic, int vector, bool level)
254088c4b8d1SNeel Natu {
254188c4b8d1SNeel Natu 	struct vlapic_vtx *vlapic_vtx;
254288c4b8d1SNeel Natu 	struct pir_desc *pir_desc;
254388c4b8d1SNeel Natu 	uint64_t mask;
254488c4b8d1SNeel Natu 	int idx, notify;
254588c4b8d1SNeel Natu 
254688c4b8d1SNeel Natu 	vlapic_vtx = (struct vlapic_vtx *)vlapic;
2547176666c2SNeel Natu 	pir_desc = vlapic_vtx->pir_desc;
254888c4b8d1SNeel Natu 
254988c4b8d1SNeel Natu 	/*
255088c4b8d1SNeel Natu 	 * Keep track of interrupt requests in the PIR descriptor. This is
255188c4b8d1SNeel Natu 	 * because the virtual APIC page pointed to by the VMCS cannot be
255288c4b8d1SNeel Natu 	 * modified if the vcpu is running.
255388c4b8d1SNeel Natu 	 */
255488c4b8d1SNeel Natu 	idx = vector / 64;
255588c4b8d1SNeel Natu 	mask = 1UL << (vector % 64);
255688c4b8d1SNeel Natu 	atomic_set_long(&pir_desc->pir[idx], mask);
255788c4b8d1SNeel Natu 	notify = atomic_cmpset_long(&pir_desc->pending, 0, 1);
255888c4b8d1SNeel Natu 
255988c4b8d1SNeel Natu 	VMX_CTR_PIR(vlapic->vm, vlapic->vcpuid, pir_desc, notify, vector,
256088c4b8d1SNeel Natu 	    level, "vmx_set_intr_ready");
256188c4b8d1SNeel Natu 	return (notify);
256288c4b8d1SNeel Natu }
256388c4b8d1SNeel Natu 
256488c4b8d1SNeel Natu static int
256588c4b8d1SNeel Natu vmx_pending_intr(struct vlapic *vlapic, int *vecptr)
256688c4b8d1SNeel Natu {
256788c4b8d1SNeel Natu 	struct vlapic_vtx *vlapic_vtx;
256888c4b8d1SNeel Natu 	struct pir_desc *pir_desc;
256988c4b8d1SNeel Natu 	struct LAPIC *lapic;
257088c4b8d1SNeel Natu 	uint64_t pending, pirval;
257188c4b8d1SNeel Natu 	uint32_t ppr, vpr;
257288c4b8d1SNeel Natu 	int i;
257388c4b8d1SNeel Natu 
257488c4b8d1SNeel Natu 	/*
257588c4b8d1SNeel Natu 	 * This function is only expected to be called from the 'HLT' exit
257688c4b8d1SNeel Natu 	 * handler which does not care about the vector that is pending.
257788c4b8d1SNeel Natu 	 */
257888c4b8d1SNeel Natu 	KASSERT(vecptr == NULL, ("vmx_pending_intr: vecptr must be NULL"));
257988c4b8d1SNeel Natu 
258088c4b8d1SNeel Natu 	vlapic_vtx = (struct vlapic_vtx *)vlapic;
2581176666c2SNeel Natu 	pir_desc = vlapic_vtx->pir_desc;
258288c4b8d1SNeel Natu 
258388c4b8d1SNeel Natu 	pending = atomic_load_acq_long(&pir_desc->pending);
258488c4b8d1SNeel Natu 	if (!pending)
258588c4b8d1SNeel Natu 		return (0);	/* common case */
258688c4b8d1SNeel Natu 
258788c4b8d1SNeel Natu 	/*
258888c4b8d1SNeel Natu 	 * If there is an interrupt pending then it will be recognized only
258988c4b8d1SNeel Natu 	 * if its priority is greater than the processor priority.
259088c4b8d1SNeel Natu 	 *
259188c4b8d1SNeel Natu 	 * Special case: if the processor priority is zero then any pending
259288c4b8d1SNeel Natu 	 * interrupt will be recognized.
259388c4b8d1SNeel Natu 	 */
259488c4b8d1SNeel Natu 	lapic = vlapic->apic_page;
259588c4b8d1SNeel Natu 	ppr = lapic->ppr & 0xf0;
259688c4b8d1SNeel Natu 	if (ppr == 0)
259788c4b8d1SNeel Natu 		return (1);
259888c4b8d1SNeel Natu 
259988c4b8d1SNeel Natu 	VCPU_CTR1(vlapic->vm, vlapic->vcpuid, "HLT with non-zero PPR %d",
260088c4b8d1SNeel Natu 	    lapic->ppr);
260188c4b8d1SNeel Natu 
260288c4b8d1SNeel Natu 	for (i = 3; i >= 0; i--) {
260388c4b8d1SNeel Natu 		pirval = pir_desc->pir[i];
260488c4b8d1SNeel Natu 		if (pirval != 0) {
260588c4b8d1SNeel Natu 			vpr = (i * 64 + flsl(pirval) - 1) & 0xf0;
260688c4b8d1SNeel Natu 			return (vpr > ppr);
260788c4b8d1SNeel Natu 		}
260888c4b8d1SNeel Natu 	}
260988c4b8d1SNeel Natu 	return (0);
261088c4b8d1SNeel Natu }
261188c4b8d1SNeel Natu 
261288c4b8d1SNeel Natu static void
261388c4b8d1SNeel Natu vmx_intr_accepted(struct vlapic *vlapic, int vector)
261488c4b8d1SNeel Natu {
261588c4b8d1SNeel Natu 
261688c4b8d1SNeel Natu 	panic("vmx_intr_accepted: not expected to be called");
261788c4b8d1SNeel Natu }
261888c4b8d1SNeel Natu 
2619176666c2SNeel Natu static void
262030b94db8SNeel Natu vmx_set_tmr(struct vlapic *vlapic, int vector, bool level)
262130b94db8SNeel Natu {
262230b94db8SNeel Natu 	struct vlapic_vtx *vlapic_vtx;
262330b94db8SNeel Natu 	struct vmx *vmx;
262430b94db8SNeel Natu 	struct vmcs *vmcs;
262530b94db8SNeel Natu 	uint64_t mask, val;
262630b94db8SNeel Natu 
262730b94db8SNeel Natu 	KASSERT(vector >= 0 && vector <= 255, ("invalid vector %d", vector));
262830b94db8SNeel Natu 	KASSERT(!vcpu_is_running(vlapic->vm, vlapic->vcpuid, NULL),
262930b94db8SNeel Natu 	    ("vmx_set_tmr: vcpu cannot be running"));
263030b94db8SNeel Natu 
263130b94db8SNeel Natu 	vlapic_vtx = (struct vlapic_vtx *)vlapic;
263230b94db8SNeel Natu 	vmx = vlapic_vtx->vmx;
263330b94db8SNeel Natu 	vmcs = &vmx->vmcs[vlapic->vcpuid];
263430b94db8SNeel Natu 	mask = 1UL << (vector % 64);
263530b94db8SNeel Natu 
263630b94db8SNeel Natu 	VMPTRLD(vmcs);
263730b94db8SNeel Natu 	val = vmcs_read(VMCS_EOI_EXIT(vector));
263830b94db8SNeel Natu 	if (level)
263930b94db8SNeel Natu 		val |= mask;
264030b94db8SNeel Natu 	else
264130b94db8SNeel Natu 		val &= ~mask;
264230b94db8SNeel Natu 	vmcs_write(VMCS_EOI_EXIT(vector), val);
264330b94db8SNeel Natu 	VMCLEAR(vmcs);
264430b94db8SNeel Natu }
264530b94db8SNeel Natu 
264630b94db8SNeel Natu static void
2647159dd56fSNeel Natu vmx_enable_x2apic_mode(struct vlapic *vlapic)
2648159dd56fSNeel Natu {
2649159dd56fSNeel Natu 	struct vmx *vmx;
2650159dd56fSNeel Natu 	struct vmcs *vmcs;
2651159dd56fSNeel Natu 	uint32_t proc_ctls2;
2652159dd56fSNeel Natu 	int vcpuid, error;
2653159dd56fSNeel Natu 
2654159dd56fSNeel Natu 	vcpuid = vlapic->vcpuid;
2655159dd56fSNeel Natu 	vmx = ((struct vlapic_vtx *)vlapic)->vmx;
2656159dd56fSNeel Natu 	vmcs = &vmx->vmcs[vcpuid];
2657159dd56fSNeel Natu 
2658159dd56fSNeel Natu 	proc_ctls2 = vmx->cap[vcpuid].proc_ctls2;
2659159dd56fSNeel Natu 	KASSERT((proc_ctls2 & PROCBASED2_VIRTUALIZE_APIC_ACCESSES) != 0,
2660159dd56fSNeel Natu 	    ("%s: invalid proc_ctls2 %#x", __func__, proc_ctls2));
2661159dd56fSNeel Natu 
2662159dd56fSNeel Natu 	proc_ctls2 &= ~PROCBASED2_VIRTUALIZE_APIC_ACCESSES;
2663159dd56fSNeel Natu 	proc_ctls2 |= PROCBASED2_VIRTUALIZE_X2APIC_MODE;
2664159dd56fSNeel Natu 	vmx->cap[vcpuid].proc_ctls2 = proc_ctls2;
2665159dd56fSNeel Natu 
2666159dd56fSNeel Natu 	VMPTRLD(vmcs);
2667159dd56fSNeel Natu 	vmcs_write(VMCS_SEC_PROC_BASED_CTLS, proc_ctls2);
2668159dd56fSNeel Natu 	VMCLEAR(vmcs);
2669159dd56fSNeel Natu 
2670159dd56fSNeel Natu 	if (vlapic->vcpuid == 0) {
2671159dd56fSNeel Natu 		/*
2672159dd56fSNeel Natu 		 * The nested page table mappings are shared by all vcpus
2673159dd56fSNeel Natu 		 * so unmap the APIC access page just once.
2674159dd56fSNeel Natu 		 */
2675159dd56fSNeel Natu 		error = vm_unmap_mmio(vmx->vm, DEFAULT_APIC_BASE, PAGE_SIZE);
2676159dd56fSNeel Natu 		KASSERT(error == 0, ("%s: vm_unmap_mmio error %d",
2677159dd56fSNeel Natu 		    __func__, error));
2678159dd56fSNeel Natu 
2679159dd56fSNeel Natu 		/*
2680159dd56fSNeel Natu 		 * The MSR bitmap is shared by all vcpus so modify it only
2681159dd56fSNeel Natu 		 * once in the context of vcpu 0.
2682159dd56fSNeel Natu 		 */
2683159dd56fSNeel Natu 		error = vmx_allow_x2apic_msrs(vmx);
2684159dd56fSNeel Natu 		KASSERT(error == 0, ("%s: vmx_allow_x2apic_msrs error %d",
2685159dd56fSNeel Natu 		    __func__, error));
2686159dd56fSNeel Natu 	}
2687159dd56fSNeel Natu }
2688159dd56fSNeel Natu 
2689159dd56fSNeel Natu static void
2690176666c2SNeel Natu vmx_post_intr(struct vlapic *vlapic, int hostcpu)
2691176666c2SNeel Natu {
2692176666c2SNeel Natu 
2693176666c2SNeel Natu 	ipi_cpu(hostcpu, pirvec);
2694176666c2SNeel Natu }
2695176666c2SNeel Natu 
269688c4b8d1SNeel Natu /*
269788c4b8d1SNeel Natu  * Transfer the pending interrupts in the PIR descriptor to the IRR
269888c4b8d1SNeel Natu  * in the virtual APIC page.
269988c4b8d1SNeel Natu  */
270088c4b8d1SNeel Natu static void
270188c4b8d1SNeel Natu vmx_inject_pir(struct vlapic *vlapic)
270288c4b8d1SNeel Natu {
270388c4b8d1SNeel Natu 	struct vlapic_vtx *vlapic_vtx;
270488c4b8d1SNeel Natu 	struct pir_desc *pir_desc;
270588c4b8d1SNeel Natu 	struct LAPIC *lapic;
270688c4b8d1SNeel Natu 	uint64_t val, pirval;
270788c4b8d1SNeel Natu 	int rvi, pirbase;
270888c4b8d1SNeel Natu 	uint16_t intr_status_old, intr_status_new;
270988c4b8d1SNeel Natu 
271088c4b8d1SNeel Natu 	vlapic_vtx = (struct vlapic_vtx *)vlapic;
2711176666c2SNeel Natu 	pir_desc = vlapic_vtx->pir_desc;
271288c4b8d1SNeel Natu 	if (atomic_cmpset_long(&pir_desc->pending, 1, 0) == 0) {
271388c4b8d1SNeel Natu 		VCPU_CTR0(vlapic->vm, vlapic->vcpuid, "vmx_inject_pir: "
271488c4b8d1SNeel Natu 		    "no posted interrupt pending");
271588c4b8d1SNeel Natu 		return;
271688c4b8d1SNeel Natu 	}
271788c4b8d1SNeel Natu 
271888c4b8d1SNeel Natu 	pirval = 0;
271988c4b8d1SNeel Natu 	lapic = vlapic->apic_page;
272088c4b8d1SNeel Natu 
272188c4b8d1SNeel Natu 	val = atomic_readandclear_long(&pir_desc->pir[0]);
272288c4b8d1SNeel Natu 	if (val != 0) {
272388c4b8d1SNeel Natu 		lapic->irr0 |= val;
272488c4b8d1SNeel Natu 		lapic->irr1 |= val >> 32;
272588c4b8d1SNeel Natu 		pirbase = 0;
272688c4b8d1SNeel Natu 		pirval = val;
272788c4b8d1SNeel Natu 	}
272888c4b8d1SNeel Natu 
272988c4b8d1SNeel Natu 	val = atomic_readandclear_long(&pir_desc->pir[1]);
273088c4b8d1SNeel Natu 	if (val != 0) {
273188c4b8d1SNeel Natu 		lapic->irr2 |= val;
273288c4b8d1SNeel Natu 		lapic->irr3 |= val >> 32;
273388c4b8d1SNeel Natu 		pirbase = 64;
273488c4b8d1SNeel Natu 		pirval = val;
273588c4b8d1SNeel Natu 	}
273688c4b8d1SNeel Natu 
273788c4b8d1SNeel Natu 	val = atomic_readandclear_long(&pir_desc->pir[2]);
273888c4b8d1SNeel Natu 	if (val != 0) {
273988c4b8d1SNeel Natu 		lapic->irr4 |= val;
274088c4b8d1SNeel Natu 		lapic->irr5 |= val >> 32;
274188c4b8d1SNeel Natu 		pirbase = 128;
274288c4b8d1SNeel Natu 		pirval = val;
274388c4b8d1SNeel Natu 	}
274488c4b8d1SNeel Natu 
274588c4b8d1SNeel Natu 	val = atomic_readandclear_long(&pir_desc->pir[3]);
274688c4b8d1SNeel Natu 	if (val != 0) {
274788c4b8d1SNeel Natu 		lapic->irr6 |= val;
274888c4b8d1SNeel Natu 		lapic->irr7 |= val >> 32;
274988c4b8d1SNeel Natu 		pirbase = 192;
275088c4b8d1SNeel Natu 		pirval = val;
275188c4b8d1SNeel Natu 	}
275288c4b8d1SNeel Natu 	VLAPIC_CTR_IRR(vlapic, "vmx_inject_pir");
275388c4b8d1SNeel Natu 
275488c4b8d1SNeel Natu 	/*
275588c4b8d1SNeel Natu 	 * Update RVI so the processor can evaluate pending virtual
275688c4b8d1SNeel Natu 	 * interrupts on VM-entry.
275788c4b8d1SNeel Natu 	 */
275888c4b8d1SNeel Natu 	if (pirval != 0) {
275988c4b8d1SNeel Natu 		rvi = pirbase + flsl(pirval) - 1;
276088c4b8d1SNeel Natu 		intr_status_old = vmcs_read(VMCS_GUEST_INTR_STATUS);
276188c4b8d1SNeel Natu 		intr_status_new = (intr_status_old & 0xFF00) | rvi;
276288c4b8d1SNeel Natu 		if (intr_status_new > intr_status_old) {
276388c4b8d1SNeel Natu 			vmcs_write(VMCS_GUEST_INTR_STATUS, intr_status_new);
276488c4b8d1SNeel Natu 			VCPU_CTR2(vlapic->vm, vlapic->vcpuid, "vmx_inject_pir: "
276588c4b8d1SNeel Natu 			    "guest_intr_status changed from 0x%04x to 0x%04x",
276688c4b8d1SNeel Natu 			    intr_status_old, intr_status_new);
276788c4b8d1SNeel Natu 		}
276888c4b8d1SNeel Natu 	}
276988c4b8d1SNeel Natu }
277088c4b8d1SNeel Natu 
2771de5ea6b6SNeel Natu static struct vlapic *
2772de5ea6b6SNeel Natu vmx_vlapic_init(void *arg, int vcpuid)
2773de5ea6b6SNeel Natu {
2774de5ea6b6SNeel Natu 	struct vmx *vmx;
2775de5ea6b6SNeel Natu 	struct vlapic *vlapic;
2776176666c2SNeel Natu 	struct vlapic_vtx *vlapic_vtx;
2777de5ea6b6SNeel Natu 
2778de5ea6b6SNeel Natu 	vmx = arg;
2779de5ea6b6SNeel Natu 
278088c4b8d1SNeel Natu 	vlapic = malloc(sizeof(struct vlapic_vtx), M_VLAPIC, M_WAITOK | M_ZERO);
2781de5ea6b6SNeel Natu 	vlapic->vm = vmx->vm;
2782de5ea6b6SNeel Natu 	vlapic->vcpuid = vcpuid;
2783de5ea6b6SNeel Natu 	vlapic->apic_page = (struct LAPIC *)&vmx->apic_page[vcpuid];
2784de5ea6b6SNeel Natu 
2785176666c2SNeel Natu 	vlapic_vtx = (struct vlapic_vtx *)vlapic;
2786176666c2SNeel Natu 	vlapic_vtx->pir_desc = &vmx->pir_desc[vcpuid];
278730b94db8SNeel Natu 	vlapic_vtx->vmx = vmx;
2788176666c2SNeel Natu 
278988c4b8d1SNeel Natu 	if (virtual_interrupt_delivery) {
279088c4b8d1SNeel Natu 		vlapic->ops.set_intr_ready = vmx_set_intr_ready;
279188c4b8d1SNeel Natu 		vlapic->ops.pending_intr = vmx_pending_intr;
279288c4b8d1SNeel Natu 		vlapic->ops.intr_accepted = vmx_intr_accepted;
279330b94db8SNeel Natu 		vlapic->ops.set_tmr = vmx_set_tmr;
2794159dd56fSNeel Natu 		vlapic->ops.enable_x2apic_mode = vmx_enable_x2apic_mode;
279588c4b8d1SNeel Natu 	}
279688c4b8d1SNeel Natu 
2797176666c2SNeel Natu 	if (posted_interrupts)
2798176666c2SNeel Natu 		vlapic->ops.post_intr = vmx_post_intr;
2799176666c2SNeel Natu 
2800de5ea6b6SNeel Natu 	vlapic_init(vlapic);
2801de5ea6b6SNeel Natu 
2802de5ea6b6SNeel Natu 	return (vlapic);
2803de5ea6b6SNeel Natu }
2804de5ea6b6SNeel Natu 
2805de5ea6b6SNeel Natu static void
2806de5ea6b6SNeel Natu vmx_vlapic_cleanup(void *arg, struct vlapic *vlapic)
2807de5ea6b6SNeel Natu {
2808de5ea6b6SNeel Natu 
2809de5ea6b6SNeel Natu 	vlapic_cleanup(vlapic);
2810de5ea6b6SNeel Natu 	free(vlapic, M_VLAPIC);
2811de5ea6b6SNeel Natu }
2812de5ea6b6SNeel Natu 
2813366f6083SPeter Grehan struct vmm_ops vmm_ops_intel = {
2814366f6083SPeter Grehan 	vmx_init,
2815366f6083SPeter Grehan 	vmx_cleanup,
281663e62d39SJohn Baldwin 	vmx_restore,
2817366f6083SPeter Grehan 	vmx_vminit,
2818366f6083SPeter Grehan 	vmx_run,
2819366f6083SPeter Grehan 	vmx_vmcleanup,
2820366f6083SPeter Grehan 	vmx_getreg,
2821366f6083SPeter Grehan 	vmx_setreg,
2822366f6083SPeter Grehan 	vmx_getdesc,
2823366f6083SPeter Grehan 	vmx_setdesc,
2824366f6083SPeter Grehan 	vmx_getcap,
2825318224bbSNeel Natu 	vmx_setcap,
2826318224bbSNeel Natu 	ept_vmspace_alloc,
2827318224bbSNeel Natu 	ept_vmspace_free,
2828de5ea6b6SNeel Natu 	vmx_vlapic_init,
2829de5ea6b6SNeel Natu 	vmx_vlapic_cleanup,
2830366f6083SPeter Grehan };
2831