Author: tfaber Date: Thu Aug 18 07:53:37 2011 New Revision: 53294
URL: http://svn.reactos.org/svn/reactos?rev=53294&view=rev Log: [KMTESTS/IO] - Fix and enable all of IoDeviceObject
Modified: branches/GSoC_2011/KMTestSuite/kmtests/ntos_io/IoDeviceObject_drv.c branches/GSoC_2011/KMTestSuite/kmtests/ntos_io/IoDeviceObject_user.c
Modified: branches/GSoC_2011/KMTestSuite/kmtests/ntos_io/IoDeviceObject_drv.c URL: http://svn.reactos.org/svn/reactos/branches/GSoC_2011/KMTestSuite/kmtests/nt... ============================================================================== --- branches/GSoC_2011/KMTestSuite/kmtests/ntos_io/IoDeviceObject_drv.c [iso-8859-1] (original) +++ branches/GSoC_2011/KMTestSuite/kmtests/ntos_io/IoDeviceObject_drv.c [iso-8859-1] Thu Aug 18 07:53:37 2011 @@ -18,8 +18,16 @@ DriverUnload } DRIVER_STATUS;
-static KMT_IRP_HANDLER TestIrpHandler; +static DRIVER_DISPATCH TestDispatch; static VOID TestDriverObject(IN PDRIVER_OBJECT DriverObject, IN DRIVER_STATUS DriverStatus); +static BOOLEAN TestZwLoad(IN PDRIVER_OBJECT DriverObject, IN PCUNICODE_STRING DriverRegistryPath, IN PWCHAR NewDriverRegPath); +static BOOLEAN TestZwUnload(IN PDRIVER_OBJECT DriverObject, IN PCUNICODE_STRING DriverRegistryPath, IN PWCHAR NewDriverRegPath); +static VOID TestLowerDeviceKernelAPI(IN PDEVICE_OBJECT DeviceObject); +static VOID TestDeviceCreated(IN PDEVICE_OBJECT DeviceObject, IN BOOLEAN ExclusiveAccess); +static VOID TestDeviceDeletion(IN PDEVICE_OBJECT DeviceObject, IN BOOLEAN Lower, IN BOOLEAN Attached); +static VOID TestDeviceCreateDelete(IN PDRIVER_OBJECT DriverObject); +static VOID TestAttachDevice(IN PDEVICE_OBJECT DeviceObject, IN PWCHAR NewDriverRegPath); +static VOID TestDetachDevice(IN PDEVICE_OBJECT AttachedDevice);
static PDEVICE_OBJECT MainDeviceObject; static PDEVICE_OBJECT AttachDeviceObject; @@ -33,18 +41,46 @@ IN OUT INT *Flags) { NTSTATUS Status = STATUS_SUCCESS; + BOOLEAN Ret; + INT i;
PAGED_CODE();
- UNREFERENCED_PARAMETER(RegistryPath); - UNREFERENCED_PARAMETER(Flags); - - *DeviceName = L"IoDeviceObject"; - - KmtRegisterIrpHandler(IRP_MJ_CREATE, NULL, TestIrpHandler); - KmtRegisterIrpHandler(IRP_MJ_CLOSE, NULL, TestIrpHandler); + UNREFERENCED_PARAMETER(DeviceName); + + *Flags = TESTENTRY_NO_CREATE_DEVICE | TESTENTRY_NO_REGISTER_DISPATCH; + + ThisDriverObject = DriverObject;
TestDriverObject(DriverObject, DriverEntry); + + /* Create and delete device, on return MainDeviceObject has been created */ + TestDeviceCreateDelete(DriverObject); + + /* Make sure a device object was created */ + if (!skip(MainDeviceObject != NULL, "Device object creation failed\n")) + { + PWCHAR LowerDriverRegPath = L"\Registry\Machine\System\CurrentControlSet\Services\Kmtest-IoHelper"; + + /* Load driver test and load the lower driver */ + Ret = TestZwLoad(DriverObject, RegistryPath, LowerDriverRegPath); + if (!skip(Ret, "Failed to load helper driver\n")) + { + TestAttachDevice(MainDeviceObject, L"\Device\Kmtest-IoHelper"); + if (!skip(AttachDeviceObject != NULL, "No attached device object\n")) + TestLowerDeviceKernelAPI(MainDeviceObject); + + /* Unload lower driver without detaching from its device */ + TestZwUnload(DriverObject, RegistryPath, LowerDriverRegPath); + TestLowerDeviceKernelAPI(MainDeviceObject); + } + } + + for (i = 0; i <= IRP_MJ_MAXIMUM_FUNCTION; ++i) + DriverObject->MajorFunction[i] = NULL; + DriverObject->MajorFunction[IRP_MJ_CREATE] = TestDispatch; + DriverObject->MajorFunction[IRP_MJ_CLOSE] = TestDispatch; + DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = TestDispatch;
return Status; } @@ -55,19 +91,45 @@ { PAGED_CODE();
+ if (!skip(AttachDeviceObject != NULL, "no attached device object\n")) + { + TestDeviceDeletion(MainDeviceObject, FALSE, TRUE); + TestDeviceDeletion(AttachDeviceObject, TRUE, FALSE); + TestDetachDevice(AttachDeviceObject); + } + + TestDeviceDeletion(MainDeviceObject, FALSE, FALSE); TestDriverObject(DriverObject, DriverUnload); + + if (MainDeviceObject) + IoDeleteDevice(MainDeviceObject); }
static NTSTATUS -TestIrpHandler( +NTAPI +TestDispatch( IN PDEVICE_OBJECT DeviceObject, - IN PIRP Irp, - IN PIO_STACK_LOCATION IoStackLocation) + IN PIRP Irp) { NTSTATUS Status = STATUS_SUCCESS; - - UNREFERENCED_PARAMETER(IoStackLocation); + PIO_STACK_LOCATION IoStackLocation; + + PAGED_CODE(); + + IoStackLocation = IoGetCurrentIrpStackLocation(Irp); + + DPRINT("TestIrpHandler. Function=%s, DeviceObject=%p, AttachDeviceObject=%p\n", + KmtMajorFunctionNames[IoStackLocation->MajorFunction], + DeviceObject, + AttachDeviceObject); + + if (AttachDeviceObject) + { + IoSkipCurrentIrpStackLocation(Irp); + Status = IoCallDriver(AttachDeviceObject, Irp); + return Status; + }
TestDriverObject(DeviceObject->DriverObject, DriverIrp);
@@ -89,29 +151,29 @@ PVOID FirstMajorFunc; int i;
- ok(DriverObject->Size == sizeof(DRIVER_OBJECT), "Size does not match, got %x",DriverObject->Size); - ok(DriverObject->Type == 4, "Type does not match 4. got %d",DriverObject->Type); + ok(DriverObject->Size == sizeof(DRIVER_OBJECT), "Size does not match, got %x\n",DriverObject->Size); + ok(DriverObject->Type == 4, "Type does not match 4. got %d\n", DriverObject->Type);
if (DriverStatus == DriverEntry) { - ok(DriverObject->DeviceObject == NULL, "Expected DeviceObject pointer to be 0, got %p", + ok(DriverObject->DeviceObject == NULL, "Expected DeviceObject pointer to be 0, got %p\n", DriverObject->DeviceObject); ok (DriverObject->Flags == DRVO_LEGACY_DRIVER, - "Expected Flags to be DRVO_LEGACY_DRIVER, got %lu", + "Expected Flags to be DRVO_LEGACY_DRIVER, got %lu\n", DriverObject->Flags); } else if (DriverStatus == DriverIrp) { - ok(DriverObject->DeviceObject != NULL, "Expected DeviceObject pointer to non null"); + ok(DriverObject->DeviceObject != NULL, "Expected DeviceObject pointer to non null\n"); ok (DriverObject->Flags == (DRVO_LEGACY_DRIVER | DRVO_INITIALIZED), - "Expected Flags to be DRVO_LEGACY_DRIVER | DRVO_INITIALIZED, got %lu", + "Expected Flags to be DRVO_LEGACY_DRIVER | DRVO_INITIALIZED, got %lu\n", DriverObject->Flags); } else if (DriverStatus == DriverUnload) { - ok(DriverObject->DeviceObject != NULL, "Expected DeviceObject pointer to non null"); + ok(DriverObject->DeviceObject != NULL, "Expected DeviceObject pointer to non null\n"); ok (DriverObject->Flags == (DRVO_LEGACY_DRIVER | DRVO_INITIALIZED | DRVO_UNLOAD_INVOKED), - "Expected Flags to be DRVO_LEGACY_DRIVER | DRVO_INITIALIZED | DRVO_UNLOAD_INVOKED, got %lu", + "Expected Flags to be DRVO_LEGACY_DRIVER | DRVO_INITIALIZED | DRVO_UNLOAD_INVOKED, got %lu\n", DriverObject->Flags); } else @@ -119,7 +181,7 @@
/* Select a routine that was not changed */ FirstMajorFunc = DriverObject->MajorFunction[1]; - ok(FirstMajorFunc != 0, "Expected MajorFunction[1] to be non NULL"); + ok(FirstMajorFunc != 0, "Expected MajorFunction[1] to be non NULL\n");
if (!skip(FirstMajorFunc != NULL, "First major function not set!\n")) { @@ -130,21 +192,26 @@
if (CheckThisDispatchRoutine) { - ok(DriverObject->MajorFunction[i] == FirstMajorFunc, "Expected MajorFunction[%d] to match %p", + ok(DriverObject->MajorFunction[i] == FirstMajorFunc, "Expected MajorFunction[%d] to match %p\n", i, FirstMajorFunc); } } } }
-BOOLEAN ZwLoadTest(PDRIVER_OBJECT DriverObject, PUNICODE_STRING DriverRegistryPath, PWCHAR NewDriverRegPath) +static +BOOLEAN +TestZwLoad( + IN PDRIVER_OBJECT DriverObject, + IN PCUNICODE_STRING DriverRegistryPath, + IN PWCHAR NewDriverRegPath) { UNICODE_STRING RegPath; NTSTATUS Status;
/* Try to load ourself */ - Status = ZwLoadDriver(DriverRegistryPath); - ok (Status == STATUS_IMAGE_ALREADY_LOADED, "Expected NTSTATUS STATUS_IMAGE_ALREADY_LOADED, got 0x%lX", Status); + Status = ZwLoadDriver((PUNICODE_STRING)DriverRegistryPath); + ok (Status == STATUS_IMAGE_ALREADY_LOADED, "Expected NTSTATUS STATUS_IMAGE_ALREADY_LOADED, got 0x%lX\n", Status);
if (Status != STATUS_IMAGE_ALREADY_LOADED) { @@ -154,64 +221,54 @@ /* Try to load with a Registry Path that doesnt exist */ RtlInitUnicodeString(&RegPath, L"\Registry\Machine\System\CurrentControlSet\Services\deadbeef"); Status = ZwLoadDriver(&RegPath); - ok (Status == STATUS_OBJECT_NAME_NOT_FOUND, "Expected NTSTATUS STATUS_OBJECT_NAME_NOT_FOUND, got 0x%lX", Status); + ok (Status == STATUS_OBJECT_NAME_NOT_FOUND, "Expected NTSTATUS STATUS_OBJECT_NAME_NOT_FOUND, got 0x%lX\n", Status);
/* Load the driver */ RtlInitUnicodeString(&RegPath, NewDriverRegPath); Status = ZwLoadDriver(&RegPath); - ok(Status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got 0x%lX", Status); - - if (!NT_SUCCESS(Status)) - { - return FALSE; - } - - return TRUE; -} - -BOOLEAN ZwUnloadTest(PDRIVER_OBJECT DriverObject, PUNICODE_STRING DriverRegistryPath, PWCHAR NewDriverRegPath) + ok(Status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got 0x%lX\n", Status); + + return NT_SUCCESS(Status); +} + +static +BOOLEAN +TestZwUnload( + IN PDRIVER_OBJECT DriverObject, + IN PCUNICODE_STRING DriverRegistryPath, + IN PWCHAR NewDriverRegPath) { UNICODE_STRING RegPath; NTSTATUS Status;
/* Try to unload ourself, which should fail as our Unload routine hasnt been set yet. */ - Status = ZwUnloadDriver(DriverRegistryPath); - ok (Status == STATUS_INVALID_DEVICE_REQUEST, "Expected NTSTATUS STATUS_INVALID_DEVICE_REQUEST, got 0x%lX", Status); + Status = ZwUnloadDriver((PUNICODE_STRING)DriverRegistryPath); + ok (Status == STATUS_INVALID_DEVICE_REQUEST, "Expected NTSTATUS STATUS_INVALID_DEVICE_REQUEST, got 0x%lX\n", Status);
/* Try to unload with a Registry Path that doesnt exist */ RtlInitUnicodeString(&RegPath, L"\Registry\Machine\System\CurrentControlSet\Services\deadbeef"); Status = ZwUnloadDriver(&RegPath); - ok (Status == STATUS_OBJECT_NAME_NOT_FOUND, "Expected NTSTATUS STATUS_OBJECT_NAME_NOT_FOUND, got 0x%lX", Status); + ok (Status == STATUS_OBJECT_NAME_NOT_FOUND, "Expected NTSTATUS STATUS_OBJECT_NAME_NOT_FOUND, got 0x%lX\n", Status);
/* Unload the driver */ RtlInitUnicodeString(&RegPath, NewDriverRegPath); Status = ZwUnloadDriver(&RegPath); - ok(Status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got 0x%lX", Status); - - if (!NT_SUCCESS(Status)) - { - return FALSE; - } - - return TRUE; -} - -VOID LowerDeviceKernelAPITest(PDEVICE_OBJECT DeviceObject, BOOLEAN UnLoading) + ok(Status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got 0x%lX\n", Status); + + return NT_SUCCESS(Status); +} + +static +VOID +TestLowerDeviceKernelAPI( + IN PDEVICE_OBJECT DeviceObject) { PDEVICE_OBJECT RetObject;
RetObject = IoGetLowerDeviceObject(DeviceObject);
- if (UnLoading) - { - ok(RetObject == 0, - "Expected no Lower DeviceObject, got %p", RetObject); - } - else - { - ok(RetObject == AttachDeviceObject, - "Expected an Attached DeviceObject %p, got %p", AttachDeviceObject, RetObject); - } + ok(RetObject == AttachDeviceObject, + "Expected an Attached DeviceObject %p, got %p\n", AttachDeviceObject, RetObject);
if (RetObject) { @@ -219,8 +276,8 @@ }
RetObject = IoGetDeviceAttachmentBaseRef(DeviceObject); - ok(RetObject == DeviceObject, - "Expected an Attached DeviceObject %p, got %p", DeviceObject, RetObject); + ok(RetObject == AttachDeviceObject, + "Expected an Attached DeviceObject %p, got %p\n", AttachDeviceObject, RetObject);
if (RetObject) { @@ -228,114 +285,125 @@ }
} -VOID DeviceCreatedTest(PDEVICE_OBJECT DeviceObject, BOOLEAN ExclusiveAccess) + +static +VOID +TestDeviceCreated( + IN PDEVICE_OBJECT DeviceObject, + IN BOOLEAN ExclusiveAccess) { PEXTENDED_DEVOBJ_EXTENSION extdev;
- /*Check the device object members */ - ok(DeviceObject->Type==3, "Expected Type = 3, got %x", DeviceObject->Type); - ok(DeviceObject->Size = 0xb8, "Expected Size = 0xba, got %x", DeviceObject->Size); - ok(DeviceObject->ReferenceCount == 0, "Expected ReferenceCount = 0, got %lu", + /* Check the device object members */ + ok(DeviceObject->Type == 3, "Expected Type = 3, got %x\n", DeviceObject->Type); + ok(DeviceObject->Size == 0xb8, "Expected Size = 0xb8, got %x\n", DeviceObject->Size); + ok(DeviceObject->ReferenceCount == 0, "Expected ReferenceCount = 0, got %lu\n", DeviceObject->ReferenceCount); ok(DeviceObject->DriverObject == ThisDriverObject, - "Expected DriverObject member to match this DriverObject %p, got %p", + "Expected DriverObject member to match this DriverObject %p, got %p\n", ThisDriverObject, DeviceObject->DriverObject); - ok(DeviceObject->NextDevice == NULL, "Expected NextDevice to be NULL, got %p", DeviceObject->NextDevice); - ok(DeviceObject->AttachedDevice == NULL, "Expected AttachDevice to be NULL, got %p", DeviceObject->AttachedDevice); - ok(DeviceObject->Characteristics == 0, "Expected Characteristics to be 0"); + ok(DeviceObject->NextDevice == NULL, "Expected NextDevice to be NULL, got %p\n", DeviceObject->NextDevice); + ok(DeviceObject->AttachedDevice == NULL, "Expected AttachDevice to be NULL, got %p\n", DeviceObject->AttachedDevice); + ok(DeviceObject->Characteristics == 0, "Expected Characteristics to be 0\n"); if (ExclusiveAccess) { ok((DeviceObject->Flags == (DO_DEVICE_HAS_NAME | DO_DEVICE_INITIALIZING | DO_EXCLUSIVE)), - "Expected Flags DO_DEVICE_HAS_NAME | DO_DEVICE_INITIALIZING | DO_EXCLUSIVE, got %lu", DeviceObject->Flags); + "Expected Flags DO_DEVICE_HAS_NAME | DO_DEVICE_INITIALIZING | DO_EXCLUSIVE, got %lu\n", DeviceObject->Flags); } else { ok((DeviceObject->Flags == (DO_DEVICE_HAS_NAME | DO_DEVICE_INITIALIZING)), - "Expected Flags DO_DEVICE_HAS_NAME | DO_DEVICE_INITIALIZING, got %lu", DeviceObject->Flags); + "Expected Flags DO_DEVICE_HAS_NAME | DO_DEVICE_INITIALIZING, got %lu\n", DeviceObject->Flags); } ok(DeviceObject->DeviceType == FILE_DEVICE_UNKNOWN, - "Expected DeviceType to match creation parameter FILE_DEVICE_UNKNWOWN, got %lu", + "Expected DeviceType to match creation parameter FILE_DEVICE_UNKNWOWN, got %lu\n", DeviceObject->DeviceType); ok(DeviceObject->ActiveThreadCount == 0, "Expected ActiveThreadCount = 0, got %lu\n", DeviceObject->ActiveThreadCount);
- /*Check the extended extension */ + /* Check the extended extension */ extdev = (PEXTENDED_DEVOBJ_EXTENSION)DeviceObject->DeviceObjectExtension; - ok(extdev->ExtensionFlags == 0, "Expected Extended ExtensionFlags to be 0, got %lu", extdev->ExtensionFlags); - ok (extdev->Type == 13, "Expected Type of 13, got %d", extdev->Type); - ok (extdev->Size == 0, "Expected Size of 0, got %d", extdev->Size); - ok (extdev->DeviceObject == DeviceObject, "Expected DeviceOject to match newly created device %p, got %p", + ok(extdev->ExtensionFlags == 0, "Expected Extended ExtensionFlags to be 0, got %lu\n", extdev->ExtensionFlags); + ok (extdev->Type == 13, "Expected Type of 13, got %d\n", extdev->Type); + ok (extdev->Size == 0, "Expected Size of 0, got %d\n", extdev->Size); + ok (extdev->DeviceObject == DeviceObject, "Expected DeviceOject to match newly created device %p, got %p\n", DeviceObject, extdev->DeviceObject); - ok(extdev->AttachedTo == NULL, "Expected AttachTo to be NULL, got %p", extdev->AttachedTo); - ok(extdev->StartIoCount == 0, "Expected StartIoCount = 0, got %lu", extdev->StartIoCount); - ok(extdev->StartIoKey == 0, "Expected StartIoKey = 0, got %lu", extdev->StartIoKey); - ok(extdev->StartIoFlags == 0, "Expected StartIoFlags = 0, got %lu", extdev->StartIoFlags); -} - -VOID DeviceDeletionTest(PDEVICE_OBJECT DeviceObject, BOOLEAN Lower) + ok(extdev->AttachedTo == NULL, "Expected AttachTo to be NULL, got %p\n", extdev->AttachedTo); + ok(extdev->StartIoCount == 0, "Expected StartIoCount = 0, got %lu\n", extdev->StartIoCount); + ok(extdev->StartIoKey == 0, "Expected StartIoKey = 0, got %lu\n", extdev->StartIoKey); + ok(extdev->StartIoFlags == 0, "Expected StartIoFlags = 0, got %lu\n", extdev->StartIoFlags); +} + +static +VOID +TestDeviceDeletion( + IN PDEVICE_OBJECT DeviceObject, + IN BOOLEAN Lower, + IN BOOLEAN Attached) { PEXTENDED_DEVOBJ_EXTENSION extdev;
- /*Check the device object members */ - ok(DeviceObject->Type==3, "Expected Type = 3, got %d", DeviceObject->Type); - ok(DeviceObject->Size = 0xb8, "Expected Size = 0xba, got %d", DeviceObject->Size); - ok(DeviceObject->ReferenceCount == 0, "Expected ReferenceCount = 0, got %lu", + /* Check the device object members */ + ok(DeviceObject->Type == 3, "Expected Type = 3, got %d\n", DeviceObject->Type); + ok(DeviceObject->Size == 0xb8, "Expected Size = 0xb8, got %d\n", DeviceObject->Size); + ok(DeviceObject->ReferenceCount == 0, "Expected ReferenceCount = 0, got %lu\n", DeviceObject->ReferenceCount); if (!Lower) { ok(DeviceObject->DriverObject == ThisDriverObject, - "Expected DriverObject member to match this DriverObject %p, got %p", + "Expected DriverObject member to match this DriverObject %p, got %p\n", ThisDriverObject, DeviceObject->DriverObject); } - ok(DeviceObject->NextDevice == NULL, "Expected NextDevice to be NULL, got %p", DeviceObject->NextDevice); + ok(DeviceObject->NextDevice == NULL, "Expected NextDevice to be NULL, got %p\n", DeviceObject->NextDevice);
if (Lower) { ok(DeviceObject->AttachedDevice == MainDeviceObject, - "Expected AttachDevice to be %p, got %p", MainDeviceObject, DeviceObject->AttachedDevice); + "Expected AttachDevice to be %p, got %p\n", MainDeviceObject, DeviceObject->AttachedDevice); } else { - ok(DeviceObject->AttachedDevice == NULL, "Expected AttachDevice to be NULL, got %p", DeviceObject->AttachedDevice); - } - - ok(DeviceObject->Flags ==FILE_VIRTUAL_VOLUME, - "Expected Flags FILE_VIRTUAL_VOLUME, got %lu", DeviceObject->Flags); + ok(DeviceObject->AttachedDevice == NULL, "Expected AttachDevice to be NULL, got %p\n", DeviceObject->AttachedDevice); + } + + ok(DeviceObject->Flags == (DO_DEVICE_HAS_NAME | (Lower ? DO_EXCLUSIVE : 0)), + "Expected Flags DO_DEVICE_HAS_NAME, got %lu\n", DeviceObject->Flags); ok(DeviceObject->DeviceType == FILE_DEVICE_UNKNOWN, - "Expected DeviceType to match creation parameter FILE_DEVICE_UNKNWOWN, got %lu", + "Expected DeviceType to match creation parameter FILE_DEVICE_UNKNWOWN, got %lu\n", DeviceObject->DeviceType); ok(DeviceObject->ActiveThreadCount == 0, "Expected ActiveThreadCount = 0, got %lu\n", DeviceObject->ActiveThreadCount);
/*Check the extended extension */ extdev = (PEXTENDED_DEVOBJ_EXTENSION)DeviceObject->DeviceObjectExtension; ok(extdev->ExtensionFlags == DOE_UNLOAD_PENDING, - "Expected Extended ExtensionFlags to be DOE_UNLOAD_PENDING, got %lu", extdev->ExtensionFlags); - ok (extdev->Type == 13, "Expected Type of 13, got %d", extdev->Type); - ok (extdev->Size == 0, "Expected Size of 0, got %d", extdev->Size); - ok (extdev->DeviceObject == DeviceObject, "Expected DeviceOject to match newly created device %p, got %p", + "Expected Extended ExtensionFlags to be DOE_UNLOAD_PENDING, got %lu\n", extdev->ExtensionFlags); + ok (extdev->Type == 13, "Expected Type of 13, got %d\n", extdev->Type); + ok (extdev->Size == 0, "Expected Size of 0, got %d\n", extdev->Size); + ok (extdev->DeviceObject == DeviceObject, "Expected DeviceOject to match newly created device %p, got %p\n", DeviceObject, extdev->DeviceObject); - if (Lower) - { - /* Skip this for now */ - //ok(extdev->AttachedTo == MainDeviceObject, "Expected AttachTo to %p, got %p", MainDeviceObject, extdev->AttachedTo); + if (Lower || !Attached) + { + ok(extdev->AttachedTo == NULL, "Expected AttachTo to be NULL, got %p\n", extdev->AttachedTo); } else { - ok(extdev->AttachedTo == NULL, "Expected AttachTo to be NULL, got %p", extdev->AttachedTo); - } - ok(extdev->StartIoCount == 0, "Expected StartIoCount = 0, got %lu", extdev->StartIoCount); - ok(extdev->StartIoKey == 0, "Expected StartIoKey = 0, got %lu", extdev->StartIoKey); - ok(extdev->StartIoFlags == 0, "Expected StartIoFlags = 0, got %lu", extdev->StartIoFlags); -} - -VOID DeviceCreateDeleteTest(PDRIVER_OBJECT DriverObject) + ok(extdev->AttachedTo == AttachDeviceObject, "Expected AttachTo to %p, got %p\n", AttachDeviceObject, extdev->AttachedTo); + } + ok(extdev->StartIoCount == 0, "Expected StartIoCount = 0, got %lu\n", extdev->StartIoCount); + ok(extdev->StartIoKey == 0, "Expected StartIoKey = 0, got %lu\n", extdev->StartIoKey); + ok(extdev->StartIoFlags == 0, "Expected StartIoFlags = 0, got %lu\n", extdev->StartIoFlags); +} + +static +VOID +TestDeviceCreateDelete( + IN PDRIVER_OBJECT DriverObject) { NTSTATUS Status; UNICODE_STRING DeviceString; - UNICODE_STRING DosDeviceString; PDEVICE_OBJECT DeviceObject;
/* Create using wrong directory */ - RtlInitUnicodeString(&DeviceString, L"\Device1\Kmtest"); + RtlInitUnicodeString(&DeviceString, L"\Device1\Kmtest-IoDeviceObject"); Status = IoCreateDevice(DriverObject, 0, &DeviceString, @@ -343,10 +411,10 @@ 0, FALSE, &DeviceObject); - ok(Status == STATUS_OBJECT_PATH_NOT_FOUND, "Expected STATUS_OBJECT_PATH_NOT_FOUND, got 0x%lX", Status); - - /* Create using correct params with exlusice access */ - RtlInitUnicodeString(&DeviceString, L"\Device\Kmtest"); + ok(Status == STATUS_OBJECT_PATH_NOT_FOUND, "Expected STATUS_OBJECT_PATH_NOT_FOUND, got 0x%lX\n", Status); + + /* Create using correct params with exclusice access */ + RtlInitUnicodeString(&DeviceString, L"\Device\Kmtest-IoDeviceObject"); Status = IoCreateDevice(DriverObject, 0, &DeviceString, @@ -354,19 +422,19 @@ 0, TRUE, &DeviceObject); - ok(Status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got 0x%lX", Status); - - DeviceCreatedTest(DeviceObject, TRUE); + ok(Status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got 0x%lX\n", Status); + + TestDeviceCreated(DeviceObject, TRUE);
/* Delete the device */ if (NT_SUCCESS(Status)) { IoDeleteDevice(DeviceObject); - ok(DriverObject->DeviceObject == 0, "Expected DriverObject->DeviceObject to be NULL, got %p", + ok(DriverObject->DeviceObject == 0, "Expected DriverObject->DeviceObject to be NULL, got %p\n", DriverObject->DeviceObject); }
- /* Create using correct params with exlusice access */ + /* Create using correct params without exclusice access */ Status = IoCreateDevice(DriverObject, 0, &DeviceString, @@ -374,15 +442,15 @@ 0, FALSE, &DeviceObject); - ok(Status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got 0x%lX", Status); - - DeviceCreatedTest(DeviceObject, FALSE); + ok(Status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got 0x%lX\n", Status); + + TestDeviceCreated(DeviceObject, FALSE);
/* Delete the device */ if (NT_SUCCESS(Status)) { IoDeleteDevice(DeviceObject); - ok(DriverObject->DeviceObject == 0, "Expected DriverObject->DeviceObject to be NULL, got %p", + ok(DriverObject->DeviceObject == 0, "Expected DriverObject->DeviceObject to be NULL, got %p\n", DriverObject->DeviceObject); }
@@ -394,42 +462,34 @@ 0, FALSE, &DeviceObject); - ok(Status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got 0x%lX", Status); - - RtlInitUnicodeString(&DosDeviceString, L"\DosDevices\kmtest"); - Status = IoCreateSymbolicLink(&DosDeviceString, &DeviceString); - - if (!NT_SUCCESS(Status)) - { - /* Delete device object if not successful */ - IoDeleteDevice(DeviceObject); - return; - } - - MainDeviceObject = DeviceObject; - - return; -} - -BOOLEAN AttachDeviceTest(PDEVICE_OBJECT DeviceObject, PWCHAR NewDriverRegPath) + ok(Status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got 0x%lX\n", Status); + + if (NT_SUCCESS(Status)) + MainDeviceObject = DeviceObject; +} + +static +VOID +TestAttachDevice( + IN PDEVICE_OBJECT DeviceObject, + IN PWCHAR NewDriverRegPath) { NTSTATUS Status; UNICODE_STRING LowerDeviceName;
RtlInitUnicodeString(&LowerDeviceName, NewDriverRegPath); Status = IoAttachDevice(DeviceObject, &LowerDeviceName, &AttachDeviceObject); + ok_eq_hex(Status, STATUS_SUCCESS);
/* TODO: Add more tests */ - - return TRUE; -} - -BOOLEAN DetachDeviceTest(PDEVICE_OBJECT AttachedDevice) -{ - +} + +static +VOID +TestDetachDevice( + IN PDEVICE_OBJECT AttachedDevice) +{ IoDetachDevice(AttachedDevice);
/* TODO: Add more tests */ - - return TRUE; -} +}
Modified: branches/GSoC_2011/KMTestSuite/kmtests/ntos_io/IoDeviceObject_user.c URL: http://svn.reactos.org/svn/reactos/branches/GSoC_2011/KMTestSuite/kmtests/nt... ============================================================================== --- branches/GSoC_2011/KMTestSuite/kmtests/ntos_io/IoDeviceObject_user.c [iso-8859-1] (original) +++ branches/GSoC_2011/KMTestSuite/kmtests/ntos_io/IoDeviceObject_user.c [iso-8859-1] Thu Aug 18 07:53:37 2011 @@ -9,7 +9,11 @@
START_TEST(IoDeviceObject) { - KmtLoadDriver(L"IoDeviceObject", FALSE); + /* make sure IoHelper has an existing service key, but is not started */ + KmtLoadDriver(L"IoHelper", FALSE); + KmtUnloadDriver(); + + KmtLoadDriver(L"IoDeviceObject", TRUE); KmtOpenDriver(); KmtCloseDriver(); KmtUnloadDriver();
-
tfaber＠svn.reactos.org