niansa-misc/wine
1
0
Fork 0
mirror of synced 2025-03-07 03:53:26 +01:00
Code Issues Projects Releases Packages Wiki Activity

ntoskrnl.exe/tests: Add Driver Object Extension Tests.

Browse source
This commit is contained in:
Etaash Mathamsetty 2022-12-25 14:10:39 -05:00 • committed by Alexandre Julliard
parent 2f3c7e8725
commit 10765f2956
1 changed files with 33 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
dlls/ntoskrnl.exe/tests/driver.c
View file

@ -2294,6 +2294,38 @@ static void test_permanence(void)
ok(status == STATUS_OBJECT_NAME_NOT_FOUND, "got %#lx\n", status);
}
static void test_driver_object_extension(void)
{
NTSTATUS (WINAPI *pIoAllocateDriverObjectExtension)(PDRIVER_OBJECT, PVOID, ULONG, PVOID *);
PVOID (WINAPI *pIoGetDriverObjectExtension)(PDRIVER_OBJECT, PVOID);
NTSTATUS status;
void *driver_obj_ext = NULL;
void *get_obj_ext = NULL;
pIoAllocateDriverObjectExtension = get_proc_address("IoAllocateDriverObjectExtension");
pIoGetDriverObjectExtension = get_proc_address("IoGetDriverObjectExtension");
if (!pIoAllocateDriverObjectExtension)
{
win_skip("IoAllocateDriverObjectExtension is not available.\n");
return;
}
status = pIoAllocateDriverObjectExtension(driver_obj, NULL, 100, &driver_obj_ext);
todo_wine ok(status == STATUS_SUCCESS, "got %#lx\n", status);
todo_wine ok(driver_obj_ext != NULL, "got NULL\n");
get_obj_ext = pIoGetDriverObjectExtension(driver_obj, NULL);
todo_wine ok(get_obj_ext == driver_obj_ext && get_obj_ext != NULL, "got %p != %p\n", get_obj_ext, driver_obj_ext);
status = pIoAllocateDriverObjectExtension(driver_obj, NULL, 100, &driver_obj_ext);
todo_wine ok(status == STATUS_OBJECT_NAME_COLLISION, "got %#lx\n", status);
ok(driver_obj_ext == NULL, "got %p\n", driver_obj_ext);
get_obj_ext = pIoGetDriverObjectExtension(driver_obj, (void *)0xdead);
ok(get_obj_ext == NULL, "got %p\n", get_obj_ext);
}
static NTSTATUS main_test(DEVICE_OBJECT *device, IRP *irp, IO_STACK_LOCATION *stack)
{
void *buffer = irp->AssociatedIrp.SystemBuffer;
@ -2337,6 +2369,7 @@ static NTSTATUS main_test(DEVICE_OBJECT *device, IRP *irp, IO_STACK_LOCATION *st
test_dpc();
test_process_memory(test_input);
test_permanence();
test_driver_object_extension();
IoMarkIrpPending(irp);
IoQueueWorkItem(work_item, main_test_task, DelayedWorkQueue, irp);