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 #define LIST_WINHVEMULATION_FUNCTIONS(X) \
28   X(HRESULT, WHvEmulatorCreateEmulator, (const WHV_EMULATOR_CALLBACKS* Callbacks, WHV_EMULATOR_HANDLE* Emulator)) \
29   X(HRESULT, WHvEmulatorDestroyEmulator, (WHV_EMULATOR_HANDLE Emulator)) \
30   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)) \
31   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)) \
32 
33 
34 #define WHP_DEFINE_TYPE(return_type, function_name, signature) \
35     typedef return_type (WINAPI *function_name ## _t) signature;
36 
37 #define WHP_DECLARE_MEMBER(return_type, function_name, signature) \
38     function_name ## _t function_name;
39 
40 /* Define function typedef */
41 LIST_WINHVPLATFORM_FUNCTIONS(WHP_DEFINE_TYPE)
42 LIST_WINHVEMULATION_FUNCTIONS(WHP_DEFINE_TYPE)
43 
44 struct WHPDispatch {
45     LIST_WINHVPLATFORM_FUNCTIONS(WHP_DECLARE_MEMBER)
46     LIST_WINHVEMULATION_FUNCTIONS(WHP_DECLARE_MEMBER)
47 };
48 
49 extern struct WHPDispatch whp_dispatch;
50 
51 bool init_whp_dispatch(void);
52 
53 
54 #endif /* WHP_DISPATCH_H */
55