From 28c1e663f3d4cc12b33023e605b0c9a69d1309d4 Mon Sep 17 00:00:00 2001
From: Thomas Faber <thomas.faber@reactos.org>
Date: Tue, 27 Oct 2015 12:14:10 +0000
Subject: [PATCH] [KMTESTS:IO] - Add some basic parameter tests for
 IoSetDeviceInterfaceState CORE-9456

svn path=/trunk/; revision=69722
---
 rostests/kmtests/ntos_io/IoDeviceInterface.c | 103 +++++++++++++++++++
 1 file changed, 103 insertions(+)

diff --git a/rostests/kmtests/ntos_io/IoDeviceInterface.c b/rostests/kmtests/ntos_io/IoDeviceInterface.c
index d6181cdf37e..fd4f1f5ee4a 100644
--- a/rostests/kmtests/ntos_io/IoDeviceInterface.c
+++ b/rostests/kmtests/ntos_io/IoDeviceInterface.c
@@ -181,9 +181,112 @@ Test_IoRegisterPlugPlayNotification(VOID)
     }
 }
 
+static
+VOID
+Test_IoSetDeviceInterface(VOID)
+{
+    NTSTATUS Status;
+    UNICODE_STRING SymbolicLinkName;
+    PWCHAR Buffer;
+    ULONG BufferSize;
+
+    /* Invalid prefix or GUID */
+    KmtStartSeh()
+        Status = IoSetDeviceInterfaceState(NULL, TRUE);
+    KmtEndSeh(STATUS_SUCCESS)
+    ok_eq_hex(Status, STATUS_INVALID_PARAMETER);
+
+    RtlInitEmptyUnicodeString(&SymbolicLinkName, NULL, 0);
+    KmtStartSeh()
+        Status = IoSetDeviceInterfaceState(&SymbolicLinkName, TRUE);
+    KmtEndSeh(STATUS_SUCCESS)
+    ok_eq_hex(Status, STATUS_INVALID_PARAMETER);
+
+    RtlInitUnicodeString(&SymbolicLinkName, L"\\??");
+    KmtStartSeh()
+        Status = IoSetDeviceInterfaceState(&SymbolicLinkName, TRUE);
+    KmtEndSeh(STATUS_SUCCESS)
+    ok_eq_hex(Status, STATUS_INVALID_PARAMETER);
+
+    RtlInitUnicodeString(&SymbolicLinkName, L"\\??\\");
+    KmtStartSeh()
+        Status = IoSetDeviceInterfaceState(&SymbolicLinkName, TRUE);
+    KmtEndSeh(STATUS_SUCCESS)
+    ok_eq_hex(Status, STATUS_INVALID_PARAMETER);
+
+    RtlInitUnicodeString(&SymbolicLinkName, L"\\??\\\\");
+    KmtStartSeh()
+        Status = IoSetDeviceInterfaceState(&SymbolicLinkName, TRUE);
+    KmtEndSeh(STATUS_SUCCESS)
+    ok_eq_hex(Status, STATUS_INVALID_PARAMETER);
+
+    RtlInitUnicodeString(&SymbolicLinkName, L"\\??\\{aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa}");
+    KmtStartSeh()
+        Status = IoSetDeviceInterfaceState(&SymbolicLinkName, TRUE);
+    KmtEndSeh(STATUS_SUCCESS)
+    ok_eq_hex(Status, STATUS_INVALID_PARAMETER);
+
+    /* Valid prefix & GUID, invalid device node */
+    RtlInitUnicodeString(&SymbolicLinkName, L"\\??\\X{aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa}");
+    KmtStartSeh()
+        Status = IoSetDeviceInterfaceState(&SymbolicLinkName, TRUE);
+    KmtEndSeh(STATUS_SUCCESS)
+    ok_eq_hex(Status, STATUS_OBJECT_NAME_NOT_FOUND);
+
+    RtlInitUnicodeString(&SymbolicLinkName, L"\\\\?\\X{aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa}");
+    KmtStartSeh()
+        Status = IoSetDeviceInterfaceState(&SymbolicLinkName, TRUE);
+    KmtEndSeh(STATUS_SUCCESS)
+    ok_eq_hex(Status, STATUS_OBJECT_NAME_NOT_FOUND);
+
+    RtlInitUnicodeString(&SymbolicLinkName, L"\\??\\X{aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa}\\");
+    KmtStartSeh()
+        Status = IoSetDeviceInterfaceState(&SymbolicLinkName, TRUE);
+    KmtEndSeh(STATUS_SUCCESS)
+    ok_eq_hex(Status, STATUS_OBJECT_NAME_NOT_FOUND);
+
+    RtlInitUnicodeString(&SymbolicLinkName, L"\\??\\#{aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa}");
+    KmtStartSeh()
+        Status = IoSetDeviceInterfaceState(&SymbolicLinkName, TRUE);
+    KmtEndSeh(STATUS_SUCCESS)
+    ok_eq_hex(Status, STATUS_OBJECT_NAME_NOT_FOUND);
+
+    /* Must not read past the buffer */
+    RtlInitUnicodeString(&SymbolicLinkName, L"\\??\\#{aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa}");
+    BufferSize = SymbolicLinkName.Length;
+    Buffer = KmtAllocateGuarded(BufferSize);
+    if (!skip(Buffer != NULL, "Failed to allocate %lu bytes\n", BufferSize))
+    {
+        RtlCopyMemory(Buffer, SymbolicLinkName.Buffer, BufferSize);
+        SymbolicLinkName.Buffer = Buffer;
+        SymbolicLinkName.MaximumLength = BufferSize;
+        KmtStartSeh()
+            Status = IoSetDeviceInterfaceState(&SymbolicLinkName, TRUE);
+        KmtEndSeh(STATUS_SUCCESS)
+        ok_eq_hex(Status, STATUS_OBJECT_NAME_NOT_FOUND);
+        KmtFreeGuarded(Buffer);
+    }
+
+    RtlInitUnicodeString(&SymbolicLinkName, L"\\??\\#aaaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa}");
+    BufferSize = SymbolicLinkName.Length;
+    Buffer = KmtAllocateGuarded(BufferSize);
+    if (!skip(Buffer != NULL, "Failed to allocate %lu bytes\n", BufferSize))
+    {
+        RtlCopyMemory(Buffer, SymbolicLinkName.Buffer, BufferSize);
+        SymbolicLinkName.Buffer = Buffer;
+        SymbolicLinkName.MaximumLength = BufferSize;
+        KmtStartSeh()
+            Status = IoSetDeviceInterfaceState(&SymbolicLinkName, TRUE);
+        KmtEndSeh(STATUS_SUCCESS)
+        ok_eq_hex(Status, STATUS_INVALID_PARAMETER);
+        KmtFreeGuarded(Buffer);
+    }
+}
+
 START_TEST(IoDeviceInterface)
 {
     // FIXME: This test crashes in Windows
     (void)Test_IoRegisterDeviceInterface;
     Test_IoRegisterPlugPlayNotification();
+    Test_IoSetDeviceInterface();
 }
\ No newline at end of file
-- 
2.17.1