[ros-diffs] [tfaber] 53294: [KMTESTS/IO] - Fix and enable all of IoDeviceObject
tfaber at svn.reactos.org
tfaber at svn.reactos.org
Thu Aug 18 07:53:38 UTC 2011
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/ntos_io/IoDeviceObject_drv.c?rev=53294&r1=53293&r2=53294&view=diff
==============================================================================
--- 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/ntos_io/IoDeviceObject_user.c?rev=53294&r1=53293&r2=53294&view=diff
==============================================================================
--- 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();
More information about the Ros-diffs
mailing list