[KMTEST]
authorAleksey Bragin <aleksey@reactos.org>
Wed, 5 Jun 2013 23:25:02 +0000 (23:25 +0000)
committerAleksey Bragin <aleksey@reactos.org>
Wed, 5 Jun 2013 23:25:02 +0000 (23:25 +0000)
- Fix formatting of FsRtlTunnel.c
- Add proper header.

svn path=/trunk/; revision=59180

rostests/kmtests/ntos_fsrtl/FsRtlTunnel.c

index e361cd2..1097807 100644 (file)
@@ -1,15 +1,10 @@
 /*
-       Lab 4, at the rate of the OS.
-
-       Tested with the system kmtest 
-       the following functions:
-               FsRtlInitializeTunnelCache
-               FsRtlDeleteTunnelCache
-               FsRtlAddToTunnelCache
-               FsRtlDeleteKeyFromTunnelCache
-               FsRtlFindInTunnelCache
-
-       Ashuha Arseny IU9-41. 2013.
+* PROJECT:         ReactOS kernel-mode tests
+* LICENSE:         LGPLv2+ - See COPYING.LIB in the top level directory
+* PURPOSE:         Kernel-Mode Test Suite FsRtl Test
+* PROGRAMMER:      Ashuha Arseny, Moscow State Technical University
+*                  Marina Volosnikova, Moscow State Technical University
+*                  Denis Petkevich, Moscow State Technical University
 */
 
 #include <kmt_test.h>
 #define NDEBUG
 #include <debug.h>
 
+/*
+Tested with the system kmtest 
+the following functions:
+FsRtlInitializeTunnelCache
+FsRtlDeleteTunnelCache
+FsRtlAddToTunnelCache
+FsRtlDeleteKeyFromTunnelCache
+FsRtlFindInTunnelCache
+*/
+
 static PTUNNEL T;
 static PTUNNEL Tb;
 
@@ -24,135 +29,135 @@ static PTUNNEL Tb;
 
 PUNICODE_STRING CopyUS(PUNICODE_STRING a)
 {
-       PUNICODE_STRING b = (PUNICODE_STRING)ExAllocatePool(PagedPool,sizeof(UNICODE_STRING));
-       ok(b != NULL, "US is NULL after allocated memory\n");
-       b->Length = 0;
-       b->MaximumLength =a->MaximumLength;
-       b->Buffer = (PWSTR)ExAllocatePoolWithTag(PagedPool, b->MaximumLength, 1633);
-       ok(b->Buffer != NULL, "US->Buffer is NULL after allocated memory\n");
-       RtlCopyUnicodeString(b, a);
-       return b;
+    PUNICODE_STRING b = (PUNICODE_STRING)ExAllocatePool(PagedPool,sizeof(UNICODE_STRING));
+    ok(b != NULL, "US is NULL after allocated memory\n");
+    b->Length = 0;
+    b->MaximumLength =a->MaximumLength;
+    b->Buffer = (PWSTR)ExAllocatePoolWithTag(PagedPool, b->MaximumLength, 1633);
+    ok(b->Buffer != NULL, "US->Buffer is NULL after allocated memory\n");
+    RtlCopyUnicodeString(b, a);
+    return b;
 }
 
 void TestFsRtlInitializeTunnelCache()
 {
-       SIZE_T eq;
-       T = ExAllocatePool(PagedPool, sizeof(TUNNEL));
-       ok(T != NULL, "PTUNEL is NULL after allocated memory\n");
-       Tb = ExAllocatePool(PagedPool, sizeof(TUNNEL));
-       ok(Tb != NULL, "PTUNEL is NULL after allocated memory\n");
-       
-       memset((void*)T, 0, sizeof(TUNNEL));
-       memset((void*)Tb, 0, sizeof(TUNNEL));
-
-       FsRtlInitializeTunnelCache(T);
-       
-       eq = RtlCompareMemory((const VOID*)T, (const VOID*)Tb,  sizeof(TUNNEL));
-       
-       ok ( eq != sizeof(TUNNEL), "FsRtlInitializeTunnelCache function did not change anything in the memory at the address PTUNEL.\n"); 
+    SIZE_T eq;
+    T = ExAllocatePool(PagedPool, sizeof(TUNNEL));
+    ok(T != NULL, "PTUNEL is NULL after allocated memory\n");
+    Tb = ExAllocatePool(PagedPool, sizeof(TUNNEL));
+    ok(Tb != NULL, "PTUNEL is NULL after allocated memory\n");
+
+    memset((void*)T, 0, sizeof(TUNNEL));
+    memset((void*)Tb, 0, sizeof(TUNNEL));
+
+    FsRtlInitializeTunnelCache(T);
+
+    eq = RtlCompareMemory((const VOID*)T, (const VOID*)Tb,  sizeof(TUNNEL));
+
+    ok ( eq != sizeof(TUNNEL), "FsRtlInitializeTunnelCache function did not change anything in the memory at the address PTUNEL.\n"); 
 }
 
 void TestFsRtlAddToTunnelCache(ULONGLONG DirectoryKey, PUNICODE_STRING s_name, PUNICODE_STRING l_name, BOOLEAN KeyByShortName)
 {
-       SIZE_T eq;
-       LONG b;
-       PUNICODE_STRING bs_name;
-       PUNICODE_STRING bl_name;
-       PVOID Bufb;
-       PVOID Buf;
-
-       Buf = ExAllocatePool(PagedPool, BufSize);
-       ok(Buf != NULL, "Buff in TestFsRtlAddToTunnelCache is NULL after allocated memory\n");
-       Bufb = ExAllocatePool(PagedPool, BufSize);
-       ok(Bufb != NULL, "Buff in TestFsRtlAddToTunnelCache is NULL after allocated memory\n");
-       
-       //allocate memory for the  bufs_name
-       bs_name = CopyUS(s_name);
-       
-       //allocate memory for the l_name and bl_name
-       bl_name = CopyUS(l_name);
-       
-       memset((void*)Buf, 0, BufSize);
-       memset((void*)Bufb, 0, BufSize);
-       
-       FsRtlAddToTunnelCache(T, DirectoryKey, s_name, l_name, KeyByShortName, BufSize, Buf);
-       
-       eq = RtlCompareMemory((const VOID*)Buf, (const VOID*)Bufb,  BufSize);
-       
-       ok( eq !=  sizeof(TUNNEL),"FsRtlAddToTunnelCache function did not change anything in the memory at the address Buf.\n"); 
-       
-       b = RtlCompareUnicodeString(l_name, bl_name, TRUE);
-       ok (b == 0, "long name after call FsRtlAddToTunnelCache != long name befo call FsRtlAddToTunnelCache\n\n");
-       b = RtlCompareUnicodeString(s_name, bs_name, TRUE);
-       ok (b == 0, "short name after call FsRtlAddToTunnelCache != short name befo call FsRtlAddToTunnelCache\n\n");
+    SIZE_T eq;
+    LONG b;
+    PUNICODE_STRING bs_name;
+    PUNICODE_STRING bl_name;
+    PVOID Bufb;
+    PVOID Buf;
+
+    Buf = ExAllocatePool(PagedPool, BufSize);
+    ok(Buf != NULL, "Buff in TestFsRtlAddToTunnelCache is NULL after allocated memory\n");
+    Bufb = ExAllocatePool(PagedPool, BufSize);
+    ok(Bufb != NULL, "Buff in TestFsRtlAddToTunnelCache is NULL after allocated memory\n");
+
+    // Allocate memory for the  bufs_name
+    bs_name = CopyUS(s_name);
+
+    // Allocate memory for the l_name and bl_name
+    bl_name = CopyUS(l_name);
+
+    memset((void*)Buf, 0, BufSize);
+    memset((void*)Bufb, 0, BufSize);
+
+    FsRtlAddToTunnelCache(T, DirectoryKey, s_name, l_name, KeyByShortName, BufSize, Buf);
+
+    eq = RtlCompareMemory((const VOID*)Buf, (const VOID*)Bufb,  BufSize);
+
+    ok( eq !=  sizeof(TUNNEL),"FsRtlAddToTunnelCache function did not change anything in the memory at the address Buf.\n"); 
+
+    b = RtlCompareUnicodeString(l_name, bl_name, TRUE);
+    ok (b == 0, "long name after call FsRtlAddToTunnelCache != long name befo call FsRtlAddToTunnelCache\n\n");
+    b = RtlCompareUnicodeString(s_name, bs_name, TRUE);
+    ok (b == 0, "short name after call FsRtlAddToTunnelCache != short name befo call FsRtlAddToTunnelCache\n\n");
 }
 
 BOOLEAN TestFsRtlFindInTunnelCache(ULONG DirectoryKey, PUNICODE_STRING name, PUNICODE_STRING s_name, PUNICODE_STRING l_name)
 {
-       // Allocate memory for the Buf
-       ULONG BufsizeTemp = BufSize;
-       PVOID Buf = ExAllocatePool(PagedPool, BufSize*2);
-       ok(Buf != NULL, "Buff in FsRtlFindInTunnelCache is NULL after allocated memory\n");
+    // Allocate memory for the Buf
+    ULONG BufsizeTemp = BufSize;
+    PVOID Buf = ExAllocatePool(PagedPool, BufSize*2);
+    ok(Buf != NULL, "Buff in FsRtlFindInTunnelCache is NULL after allocated memory\n");
 
-       return FsRtlFindInTunnelCache(T, DirectoryKey, name, s_name, l_name, &BufsizeTemp, Buf); 
+    return FsRtlFindInTunnelCache(T, DirectoryKey, name, s_name, l_name, &BufsizeTemp, Buf); 
 }
 
 void TestFsRtlDeleteKeyFromTunnelCache(ULONGLONG a)
 {
-       FsRtlDeleteKeyFromTunnelCache(T, a);
+    FsRtlDeleteKeyFromTunnelCache(T, a);
 }
 
 START_TEST(FsRtlTunnel)
 {
-       PUNICODE_STRING s_name;
-       PUNICODE_STRING l_name;
-       PUNICODE_STRING name;
-       PUNICODE_STRING a;
-       BOOLEAN is;
-       
-       //Initialize Cash
-       TestFsRtlInitializeTunnelCache();
-       
-       s_name = (PUNICODE_STRING)ExAllocatePool(PagedPool,sizeof(UNICODE_STRING));
-       ok(s_name != NULL, "s_name in TestFsRtlAddToTunnelCache is NULL after allocated memory\n");
-       RtlInitUnicodeString(s_name, L"smal");
-       
-       l_name = (PUNICODE_STRING)ExAllocatePool(PagedPool,sizeof(UNICODE_STRING));
-       ok(l_name != NULL, "l_name in TestFsRtlAddToTunnelCache is NULL after allocated memory\n");
-       RtlInitUnicodeString(l_name, L"bigbigbigbigbig");
-       
-       // Add elem
-       TestFsRtlAddToTunnelCache(12345, s_name, l_name, TRUE);
-       
-       name = (PUNICODE_STRING)ExAllocatePool(PagedPool,sizeof(UNICODE_STRING));
-       ok(name != NULL, "name in FsRtlFindInTunnelCache is NULL after allocated memory\n");
-       RtlInitUnicodeString(name, L"smal");
-       
-       // Find
-       is = TestFsRtlFindInTunnelCache(12345, name, s_name, l_name);
-       ok(is == TRUE, "FsRtlFindInTunnelCache dosn't find elem id = 12345\n");
-       
-       TestFsRtlDeleteKeyFromTunnelCache(12345);       //Delete
-       is = TestFsRtlFindInTunnelCache(12345, name, s_name, l_name);
-       ok(is == FALSE, "TestFsRtlDeleteKeyFromTunnelCache dosn't delete elem id = 12345\n");
-       
-       is = TestFsRtlFindInTunnelCache(12347, name, s_name, l_name);
-       ok(is == FALSE, "FsRtlDeleteTunnelCache dosn't clear cash\n");
-       
-       TestFsRtlAddToTunnelCache(12345, s_name, l_name, TRUE);
-       TestFsRtlAddToTunnelCache(12347, s_name, l_name, TRUE);
-       a = (PUNICODE_STRING)ExAllocatePool(PagedPool,sizeof(UNICODE_STRING));
-       TestFsRtlAddToTunnelCache(12346, a, l_name, FALSE);
-       
-       //Clear all
-       FsRtlDeleteTunnelCache(T);
-       
-       is = TestFsRtlFindInTunnelCache(12345, name, s_name, l_name);
-       ok(is == FALSE, "FsRtlDeleteTunnelCache dosn't clear cash\n");
-       
-       is = TestFsRtlFindInTunnelCache(12346, name, a, l_name);
-       ok(is == FALSE, "FsRtlDeleteTunnelCache dosn't clear cash\n");
-       
-       is = TestFsRtlFindInTunnelCache(12347, name, s_name, l_name);
-       ok(is == FALSE, "FsRtlDeleteTunnelCache dosn't clear cash\n");
+    PUNICODE_STRING s_name;
+    PUNICODE_STRING l_name;
+    PUNICODE_STRING name;
+    PUNICODE_STRING a;
+    BOOLEAN is;
+
+    //Initialize Cash
+    TestFsRtlInitializeTunnelCache();
+
+    s_name = (PUNICODE_STRING)ExAllocatePool(PagedPool,sizeof(UNICODE_STRING));
+    ok(s_name != NULL, "s_name in TestFsRtlAddToTunnelCache is NULL after allocated memory\n");
+    RtlInitUnicodeString(s_name, L"smal");
+
+    l_name = (PUNICODE_STRING)ExAllocatePool(PagedPool,sizeof(UNICODE_STRING));
+    ok(l_name != NULL, "l_name in TestFsRtlAddToTunnelCache is NULL after allocated memory\n");
+    RtlInitUnicodeString(l_name, L"bigbigbigbigbig");
+
+    // Add elem
+    TestFsRtlAddToTunnelCache(12345, s_name, l_name, TRUE);
+
+    name = (PUNICODE_STRING)ExAllocatePool(PagedPool,sizeof(UNICODE_STRING));
+    ok(name != NULL, "name in FsRtlFindInTunnelCache is NULL after allocated memory\n");
+    RtlInitUnicodeString(name, L"smal");
+
+    // Find
+    is = TestFsRtlFindInTunnelCache(12345, name, s_name, l_name);
+    ok(is == TRUE, "FsRtlFindInTunnelCache dosn't find elem id = 12345\n");
+
+    TestFsRtlDeleteKeyFromTunnelCache(12345);  //Delete
+    is = TestFsRtlFindInTunnelCache(12345, name, s_name, l_name);
+    ok(is == FALSE, "TestFsRtlDeleteKeyFromTunnelCache dosn't delete elem id = 12345\n");
+
+    is = TestFsRtlFindInTunnelCache(12347, name, s_name, l_name);
+    ok(is == FALSE, "FsRtlDeleteTunnelCache dosn't clear cash\n");
+
+    TestFsRtlAddToTunnelCache(12345, s_name, l_name, TRUE);
+    TestFsRtlAddToTunnelCache(12347, s_name, l_name, TRUE);
+    a = (PUNICODE_STRING)ExAllocatePool(PagedPool,sizeof(UNICODE_STRING));
+    TestFsRtlAddToTunnelCache(12346, a, l_name, FALSE);
+
+    //Clear all
+    FsRtlDeleteTunnelCache(T);
+
+    is = TestFsRtlFindInTunnelCache(12345, name, s_name, l_name);
+    ok(is == FALSE, "FsRtlDeleteTunnelCache dosn't clear cash\n");
+
+    is = TestFsRtlFindInTunnelCache(12346, name, a, l_name);
+    ok(is == FALSE, "FsRtlDeleteTunnelCache dosn't clear cash\n");
+
+    is = TestFsRtlFindInTunnelCache(12347, name, s_name, l_name);
+    ok(is == FALSE, "FsRtlDeleteTunnelCache dosn't clear cash\n");
 }