mirror of
https://github.com/momo5502/hypervisor.git
synced 2025-07-05 18:51:53 +00:00
Add ioctrl tests
This commit is contained in:
@ -7,6 +7,51 @@
|
||||
|
||||
#define SERVICE_NAME "MomoLul"
|
||||
|
||||
#define HELLO_DRV_IOCTL CTL_CODE(FILE_DEVICE_UNKNOWN, 0x800, METHOD_NEITHER, FILE_ANY_ACCESS)
|
||||
const char kDevName[] = "\\\\.\\HelloDev";
|
||||
|
||||
HANDLE open_device(const char* device_name)
|
||||
{
|
||||
HANDLE device = CreateFileA(device_name,
|
||||
GENERIC_READ | GENERIC_WRITE,
|
||||
NULL,
|
||||
NULL,
|
||||
OPEN_EXISTING,
|
||||
NULL,
|
||||
NULL
|
||||
);
|
||||
return device;
|
||||
}
|
||||
|
||||
void close_device(HANDLE device)
|
||||
{
|
||||
CloseHandle(device);
|
||||
}
|
||||
|
||||
BOOL send_ioctl(HANDLE device, DWORD ioctl_code)
|
||||
{
|
||||
//prepare input buffer:
|
||||
DWORD bufSize = 0x4;
|
||||
BYTE* inBuffer = (BYTE*)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, bufSize);
|
||||
|
||||
//fill the buffer with some content:
|
||||
RtlFillMemory(inBuffer, bufSize, 'A');
|
||||
|
||||
DWORD size_returned = 0;
|
||||
BOOL is_ok = DeviceIoControl(device,
|
||||
ioctl_code,
|
||||
inBuffer,
|
||||
bufSize,
|
||||
NULL, //outBuffer -> None
|
||||
0, //outBuffer size -> 0
|
||||
&size_returned,
|
||||
NULL
|
||||
);
|
||||
//release the input bufffer:
|
||||
HeapFree(GetProcessHeap(), 0, (LPVOID)inBuffer);
|
||||
return is_ok;
|
||||
}
|
||||
|
||||
std::filesystem::path get_current_path()
|
||||
{
|
||||
const auto module = GetModuleHandleA(nullptr);
|
||||
@ -64,6 +109,18 @@ int main(const int /*argc*/, char* /*argv*/[])
|
||||
if (service)
|
||||
{
|
||||
StartServiceA(service, 0, nullptr);
|
||||
|
||||
HANDLE dev = open_device(kDevName);
|
||||
if (dev == INVALID_HANDLE_VALUE)
|
||||
{
|
||||
printf("Failed!\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
send_ioctl(dev, HELLO_DRV_IOCTL);
|
||||
|
||||
close_device(dev);
|
||||
|
||||
MessageBoxA(0, "Service started!", 0, 0);
|
||||
}
|
||||
return 0;
|
||||
|
Reference in New Issue
Block a user