1 #ifndef WHP_DISPATCH_H
2 #define WHP_DISPATCH_H
3 
4 #include <windows.h>
5 
6 #include <WinHvPlatform.h>
7 #include <WinHvEmulation.h>
8 
9 #define LIST_WINHVPLATFORM_FUNCTIONS(X) \
10   X(HRESULT, WHvGetCapability, (WHV_CAPABILITY_CODE CapabilityCode, VOID* CapabilityBuffer, UINT32 CapabilityBufferSizeInBytes, UINT32* WrittenSizeInBytes)) \
11   X(HRESULT, WHvCreatePartition, (WHV_PARTITION_HANDLE* Partition)) \
12   X(HRESULT, WHvSetupPartition, (WHV_PARTITION_HANDLE Partition)) \
13   X(HRESULT, WHvDeletePartition, (WHV_PARTITION_HANDLE Partition)) \
14   X(HRESULT, WHvGetPartitionProperty, (WHV_PARTITION_HANDLE Partition, WHV_PARTITION_PROPERTY_CODE PropertyCode, VOID* PropertyBuffer, UINT32 PropertyBufferSizeInBytes, UINT32* WrittenSizeInBytes)) \
15   X(HRESULT, WHvSetPartitionProperty, (WHV_PARTITION_HANDLE Partition, WHV_PARTITION_PROPERTY_CODE PropertyCode, const VOID* PropertyBuffer, UINT32 PropertyBufferSizeInBytes)) \
16   X(HRESULT, WHvMapGpaRange, (WHV_PARTITION_HANDLE Partition, VOID* SourceAddress, WHV_GUEST_PHYSICAL_ADDRESS GuestAddress, UINT64 SizeInBytes, WHV_MAP_GPA_RANGE_FLAGS Flags)) \
17   X(HRESULT, WHvUnmapGpaRange, (WHV_PARTITION_HANDLE Partition, WHV_GUEST_PHYSICAL_ADDRESS GuestAddress, UINT64 SizeInBytes)) \
18   X(HRESULT, WHvTranslateGva, (WHV_PARTITION_HANDLE Partition, UINT32 VpIndex, WHV_GUEST_VIRTUAL_ADDRESS Gva, WHV_TRANSLATE_GVA_FLAGS TranslateFlags, WHV_TRANSLATE_GVA_RESULT* TranslationResult, WHV_GUEST_PHYSICAL_ADDRESS* Gpa)) \
19   X(HRESULT, WHvCreateVirtualProcessor, (WHV_PARTITION_HANDLE Partition, UINT32 VpIndex, UINT32 Flags)) \
20   X(HRESULT, WHvDeleteVirtualProcessor, (WHV_PARTITION_HANDLE Partition, UINT32 VpIndex)) \
21   X(HRESULT, WHvRunVirtualProcessor, (WHV_PARTITION_HANDLE Partition, UINT32 VpIndex, VOID* ExitContext, UINT32 ExitContextSizeInBytes)) \
22   X(HRESULT, WHvCancelRunVirtualProcessor, (WHV_PARTITION_HANDLE Partition, UINT32 VpIndex, UINT32 Flags)) \
23   X(HRESULT, WHvGetVirtualProcessorRegisters, (WHV_PARTITION_HANDLE Partition, UINT32 VpIndex, const WHV_REGISTER_NAME* RegisterNames, UINT32 RegisterCount, WHV_REGISTER_VALUE* RegisterValues)) \
24   X(HRESULT, WHvSetVirtualProcessorRegisters, (WHV_PARTITION_HANDLE Partition, UINT32 VpIndex, const WHV_REGISTER_NAME* RegisterNames, UINT32 RegisterCount, const WHV_REGISTER_VALUE* RegisterValues)) \
25 
26 /*
27  * These are supplemental functions that may not be present
28  * on all versions and are not critical for basic functionality.
29  */
30 #define LIST_WINHVPLATFORM_FUNCTIONS_SUPPLEMENTAL(X) \
31   X(HRESULT, WHvSuspendPartitionTime, (WHV_PARTITION_HANDLE Partition)) \
32 
33 #define LIST_WINHVEMULATION_FUNCTIONS(X) \
34   X(HRESULT, WHvEmulatorCreateEmulator, (const WHV_EMULATOR_CALLBACKS* Callbacks, WHV_EMULATOR_HANDLE* Emulator)) \
35   X(HRESULT, WHvEmulatorDestroyEmulator, (WHV_EMULATOR_HANDLE Emulator)) \
36   X(HRESULT, WHvEmulatorTryIoEmulation, (WHV_EMULATOR_HANDLE Emulator, VOID* Context, const WHV_VP_EXIT_CONTEXT* VpContext, const WHV_X64_IO_PORT_ACCESS_CONTEXT* IoInstructionContext, WHV_EMULATOR_STATUS* EmulatorReturnStatus)) \
37   X(HRESULT, WHvEmulatorTryMmioEmulation, (WHV_EMULATOR_HANDLE Emulator, VOID* Context, const WHV_VP_EXIT_CONTEXT* VpContext, const WHV_MEMORY_ACCESS_CONTEXT* MmioInstructionContext, WHV_EMULATOR_STATUS* EmulatorReturnStatus)) \
38 
39 
40 #define WHP_DEFINE_TYPE(return_type, function_name, signature) \
41     typedef return_type (WINAPI *function_name ## _t) signature;
42 
43 #define WHP_DECLARE_MEMBER(return_type, function_name, signature) \
44     function_name ## _t function_name;
45 
46 /* Define function typedef */
47 LIST_WINHVPLATFORM_FUNCTIONS(WHP_DEFINE_TYPE)
48 LIST_WINHVEMULATION_FUNCTIONS(WHP_DEFINE_TYPE)
49 LIST_WINHVPLATFORM_FUNCTIONS_SUPPLEMENTAL(WHP_DEFINE_TYPE)
50 
51 struct WHPDispatch {
52     LIST_WINHVPLATFORM_FUNCTIONS(WHP_DECLARE_MEMBER)
53     LIST_WINHVEMULATION_FUNCTIONS(WHP_DECLARE_MEMBER)
54     LIST_WINHVPLATFORM_FUNCTIONS_SUPPLEMENTAL(WHP_DECLARE_MEMBER)
55 };
56 
57 extern struct WHPDispatch whp_dispatch;
58 
59 bool init_whp_dispatch(void);
60 
61 typedef enum WHPFunctionList {
62     WINHV_PLATFORM_FNS_DEFAULT,
63     WINHV_EMULATION_FNS_DEFAULT,
64     WINHV_PLATFORM_FNS_SUPPLEMENTAL
65 } WHPFunctionList;
66 
67 #endif /* WHP_DISPATCH_H */
68