[NTDLL_WINETEST]
authorThomas Faber <thomas.faber@reactos.org>
Sun, 25 Mar 2012 21:24:05 +0000 (21:24 +0000)
committerThomas Faber <thomas.faber@reactos.org>
Sun, 25 Mar 2012 21:24:05 +0000 (21:24 +0000)
- Fix running RtlGetFullPathName_UstrEx test from drive root
- Add a test for RtlGetFullPathName_U based on the same test cases

svn path=/trunk/; revision=56230

rostests/apitests/ntdll/CMakeLists.txt
rostests/apitests/ntdll/RtlGetFullPathName_U.c [new file with mode: 0644]
rostests/apitests/ntdll/RtlGetFullPathName_UstrEx.c
rostests/apitests/ntdll/testlist.c

index e180444..32caae5 100644 (file)
@@ -2,6 +2,7 @@
 list(APPEND SOURCE
     NtAllocateVirtualMemory.c
     NtFreeVirtualMemory.c
+    RtlGetFullPathName_U.c
     RtlGetFullPathName_UstrEx.c
     RtlInitializeBitMap.c
     SystemInfo.c
diff --git a/rostests/apitests/ntdll/RtlGetFullPathName_U.c b/rostests/apitests/ntdll/RtlGetFullPathName_U.c
new file mode 100644 (file)
index 0000000..e9bc549
--- /dev/null
@@ -0,0 +1,283 @@
+/*
+ * PROJECT:         ReactOS api tests
+ * LICENSE:         GPLv2+ - See COPYING in the top level directory
+ * PURPOSE:         Test for RtlGetFullPathName_U
+ * PROGRAMMER:      Thomas Faber <thfabba@gmx.de>
+ */
+
+#define WIN32_NO_STATUS
+#include <wine/test.h>
+#include <pseh/pseh2.h>
+#include <ndk/rtlfuncs.h>
+
+/*
+ULONG
+NTAPI
+RtlGetFullPathName_U(
+    IN PCWSTR FileName,
+    IN ULONG Size,
+    IN PWSTR Buffer,
+    OUT PWSTR *ShortName
+);
+*/
+
+#define StartSeh()                  ExceptionStatus = STATUS_SUCCESS; _SEH2_TRY {
+#define EndSeh(ExpectedStatus)      } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) { ExceptionStatus = _SEH2_GetExceptionCode(); } _SEH2_END; ok(ExceptionStatus == ExpectedStatus, "Exception %lx, expected %lx\n", ExceptionStatus, ExpectedStatus)
+
+static
+BOOLEAN
+CheckStringBuffer(
+    PCWSTR Buffer,
+    ULONG Length,
+    SIZE_T MaximumLength,
+    PCWSTR Expected)
+{
+    USHORT ExpectedLength = wcslen(Expected) * sizeof(WCHAR);
+    SIZE_T EqualLength;
+    BOOLEAN Result = TRUE;
+    SIZE_T i;
+
+    if (Length != ExpectedLength)
+    {
+        ok(0, "String length is %u, expected %u\n", Length, ExpectedLength);
+        Result = FALSE;
+    }
+
+    EqualLength = RtlCompareMemory(Buffer, Expected, Length);
+    if (EqualLength != Length)
+    {
+        ok(0, "String is '%S', expected '%S'\n", Buffer, Expected);
+        Result = FALSE;
+    }
+
+    if (Buffer[Length / sizeof(WCHAR)] != UNICODE_NULL)
+    {
+        ok(0, "Not null terminated\n");
+        Result = FALSE;
+    }
+
+    /* the function nulls the rest of the buffer! */
+    for (i = Length + sizeof(UNICODE_NULL); i < MaximumLength; i++)
+    {
+        UCHAR Char = ((PUCHAR)Buffer)[i];
+        if (Char != 0)
+        {
+            ok(0, "Found 0x%x at offset %lu, expected 0x%x\n", Char, (ULONG)i, 0);
+            /* don't count this as a failure unless the string was actually wrong */
+            //Result = FALSE;
+            /* don't flood the log */
+            break;
+        }
+    }
+
+    return Result;
+}
+
+#define InvalidPointer ((PVOID)0x1234)
+
+/* winetest_platform is "windows" for us, so broken() doesn't do what it should :( */
+#undef broken
+#define broken(x) 0
+
+typedef enum
+{
+    PrefixNone,
+    PrefixCurrentDrive,
+    PrefixCurrentPath,
+    PrefixCurrentPathWithoutLastPart
+} PREFIX_TYPE;
+
+static
+VOID
+RunTestCases(VOID)
+{
+    /* TODO: don't duplicate this here and in the RtlGetFullPathName_UstrEx test */
+    struct
+    {
+        PCWSTR FileName;
+        PREFIX_TYPE PrefixType;
+        PCWSTR FullPathName;
+        PREFIX_TYPE FilePartPrefixType;
+        SIZE_T FilePartSize;
+    } TestCases[] =
+    {
+        { L"C:",                 PrefixCurrentPath, L"", PrefixCurrentPathWithoutLastPart },
+        { L"C:\\",               PrefixNone, L"C:\\" },
+        { L"C:\\test",           PrefixNone, L"C:\\test", PrefixCurrentDrive },
+        { L"C:\\test\\",         PrefixNone, L"C:\\test\\" },
+        { L"C:/test/",           PrefixNone, L"C:\\test\\" },
+
+        { L"C:\\\\test",         PrefixNone, L"C:\\test", PrefixCurrentDrive },
+        { L"test",               PrefixCurrentPath, L"\\test", PrefixCurrentPath, sizeof(WCHAR) },
+        { L"\\test",             PrefixCurrentDrive, L"test", PrefixCurrentDrive },
+        { L"/test",              PrefixCurrentDrive, L"test", PrefixCurrentDrive },
+        { L".\\test",            PrefixCurrentPath, L"\\test", PrefixCurrentPath, sizeof(WCHAR) },
+
+        { L"\\.",                PrefixCurrentDrive, L"" },
+        { L"\\.\\",              PrefixCurrentDrive, L"" },
+        { L"\\\\.",              PrefixNone, L"\\\\.\\" },
+        { L"\\\\.\\",            PrefixNone, L"\\\\.\\" },
+        { L"\\\\.\\Something\\", PrefixNone, L"\\\\.\\Something\\" },
+
+        { L"\\??\\",             PrefixCurrentDrive, L"??\\" },
+        { L"\\??\\C:",           PrefixCurrentDrive, L"??\\C:", PrefixCurrentDrive, 3 * sizeof(WCHAR) },
+        { L"\\??\\C:\\",         PrefixCurrentDrive, L"??\\C:\\" },
+        { L"\\??\\C:\\test",     PrefixCurrentDrive, L"??\\C:\\test", PrefixCurrentDrive, 6 * sizeof(WCHAR) },
+        { L"\\??\\C:\\test\\",   PrefixCurrentDrive, L"??\\C:\\test\\" },
+
+        { L"\\\\??\\",           PrefixNone, L"\\\\??\\" },
+        { L"\\\\??\\C:",         PrefixNone, L"\\\\??\\C:" },
+        { L"\\\\??\\C:\\",       PrefixNone, L"\\\\??\\C:\\" },
+        { L"\\\\??\\C:\\test",   PrefixNone, L"\\\\??\\C:\\test", PrefixNone, sizeof(L"\\\\??\\C:\\") },
+        { L"\\\\??\\C:\\test\\", PrefixNone, L"\\\\??\\C:\\test\\" },
+    };
+    NTSTATUS ExceptionStatus;
+    WCHAR FullPathNameBuffer[MAX_PATH];
+    PWSTR ShortName;
+    SIZE_T Length;
+    WCHAR ExpectedPathName[MAX_PATH];
+    SIZE_T FilePartSize;
+    SIZE_T ExpectedFilePartSize;
+    const INT TestCount = sizeof(TestCases) / sizeof(TestCases[0]);
+    INT i;
+
+    for (i = 0; i < TestCount; i++)
+    {
+        trace("i = %d\n", i);
+        switch (TestCases[i].PrefixType)
+        {
+            case PrefixNone:
+                ExpectedPathName[0] = UNICODE_NULL;
+                break;
+            case PrefixCurrentDrive:
+                GetCurrentDirectoryW(sizeof(ExpectedPathName) / sizeof(WCHAR), ExpectedPathName);
+                ExpectedPathName[3] = UNICODE_NULL;
+                break;
+            case PrefixCurrentPath:
+                Length = GetCurrentDirectoryW(sizeof(ExpectedPathName) / sizeof(WCHAR), ExpectedPathName);
+                if (Length == 3 && TestCases[i].FullPathName[0])
+                    ExpectedPathName[2] = UNICODE_NULL;
+                break;
+            default:
+                skip(0, "Invalid test!\n");
+                continue;
+        }
+        wcscat(ExpectedPathName, TestCases[i].FullPathName);
+        RtlFillMemory(FullPathNameBuffer, sizeof(FullPathNameBuffer), 0xAA);
+        Length = 0;
+        StartSeh()
+            Length = RtlGetFullPathName_U(TestCases[i].FileName,
+                                          sizeof(FullPathNameBuffer),
+                                          FullPathNameBuffer,
+                                          &ShortName);
+        EndSeh(STATUS_SUCCESS);
+
+        /* TODO: remove SEH here */
+        StartSeh()
+            ok(CheckStringBuffer(FullPathNameBuffer, Length, sizeof(FullPathNameBuffer), ExpectedPathName),
+                "Wrong path name '%S', expected '%S'\n", FullPathNameBuffer, ExpectedPathName);
+        EndSeh(STATUS_SUCCESS);
+
+        if (!ShortName)
+            FilePartSize = 0;
+        else
+            FilePartSize = ShortName - FullPathNameBuffer;
+
+        switch (TestCases[i].FilePartPrefixType)
+        {
+            case PrefixNone:
+                ExpectedFilePartSize = 0;
+                break;
+            case PrefixCurrentDrive:
+                ExpectedFilePartSize = sizeof(L"C:\\");
+                break;
+            case PrefixCurrentPath:
+                ExpectedFilePartSize = GetCurrentDirectoryW(0, NULL) * sizeof(WCHAR);
+                if (ExpectedFilePartSize == sizeof(L"C:\\"))
+                    ExpectedFilePartSize -= sizeof(WCHAR);
+                break;
+            case PrefixCurrentPathWithoutLastPart:
+            {
+                WCHAR CurrentPath[MAX_PATH];
+                PCWSTR BackSlash;
+                ExpectedFilePartSize = GetCurrentDirectoryW(sizeof(CurrentPath) / sizeof(WCHAR), CurrentPath) * sizeof(WCHAR) + sizeof(UNICODE_NULL);
+                if (ExpectedFilePartSize == sizeof(L"C:\\"))
+                    ExpectedFilePartSize = 0;
+                else
+                {
+                    BackSlash = wcsrchr(CurrentPath, L'\\');
+                    if (BackSlash)
+                        ExpectedFilePartSize -= wcslen(BackSlash + 1) * sizeof(WCHAR);
+                    else
+                        ok(0, "GetCurrentDirectory returned %S\n", CurrentPath);
+                }
+                break;
+            }
+            default:
+                skip(0, "Invalid test!\n");
+                continue;
+        }
+        ExpectedFilePartSize += TestCases[i].FilePartSize;
+        if (ExpectedFilePartSize != 0)
+            ExpectedFilePartSize = (ExpectedFilePartSize - sizeof(UNICODE_NULL)) / sizeof(WCHAR);
+        ok(FilePartSize == ExpectedFilePartSize,
+            "FilePartSize = %lu, expected %lu\n", (ULONG)FilePartSize, (ULONG)ExpectedFilePartSize);
+    }
+}
+
+START_TEST(RtlGetFullPathName_U)
+{
+    NTSTATUS ExceptionStatus;
+    PCWSTR FileName;
+    PWSTR ShortName;
+    ULONG Length;
+
+    /* Parameter checks */
+    StartSeh()
+        Length = RtlGetFullPathName_U(NULL, 0, NULL, NULL);
+        ok(Length == 0, "Length = %lu\n", Length);
+    EndSeh(STATUS_SUCCESS);
+
+    StartSeh()
+        Length = RtlGetFullPathName_U(L"", 0, NULL, NULL);
+        ok(Length == 0, "Length = %lu\n", Length);
+    EndSeh(STATUS_SUCCESS);
+
+    ShortName = InvalidPointer;
+    StartSeh()
+        Length = RtlGetFullPathName_U(NULL, 0, NULL, &ShortName);
+        ok(Length == 0, "Length = %lu\n", Length);
+    EndSeh(STATUS_SUCCESS);
+    ok(ShortName == InvalidPointer ||
+        broken(ShortName == NULL) /* Win7 */, "ShortName = %p\n", ShortName);
+
+    StartSeh()
+        Length = RtlGetFullPathName_U(L"", 0, NULL, NULL);
+        ok(Length == 0, "Length = %lu\n", Length);
+    EndSeh(STATUS_SUCCESS);
+
+    ShortName = InvalidPointer;
+    StartSeh()
+        Length = RtlGetFullPathName_U(L"", 0, NULL, &ShortName);
+        ok(Length == 0, "Length = %lu\n", Length);
+    EndSeh(STATUS_SUCCESS);
+    ok(ShortName == InvalidPointer ||
+        broken(ShortName == NULL) /* Win7 */, "ShortName = %p\n", ShortName);
+
+    StartSeh()
+        Length = RtlGetFullPathName_U(L"C:\\test", 0, NULL, NULL);
+        ok(Length == sizeof(L"C:\\test"), "Length = %lu\n", Length);
+    EndSeh(STATUS_SUCCESS);
+
+    FileName = L"C:\\test";
+    ShortName = InvalidPointer;
+    StartSeh()
+        Length = RtlGetFullPathName_U(FileName, 0, NULL, &ShortName);
+        ok(Length == sizeof(L"C:\\test"), "Length = %lu\n", Length);
+    EndSeh(STATUS_SUCCESS);
+    ok(ShortName == InvalidPointer ||
+        broken(ShortName == NULL) /* Win7 */, "ShortName = %p\n", ShortName);
+
+    /* check the actual functionality with different paths */
+    RunTestCases();
+}
index 9b3fdde..63aba5f 100644 (file)
@@ -98,6 +98,7 @@ static
 VOID
 RunTestCases(VOID)
 {
+    /* TODO: don't duplicate this here and in the RtlGetFullPathName_U test */
     struct
     {
         PCWSTR FileName;
@@ -166,8 +167,13 @@ RunTestCases(VOID)
                 ExpectedPathName[3] = UNICODE_NULL;
                 break;
             case PrefixCurrentPath:
-                GetCurrentDirectoryW(sizeof(ExpectedPathName) / sizeof(WCHAR), ExpectedPathName);
+            {
+                ULONG Length;
+                Length = GetCurrentDirectoryW(sizeof(ExpectedPathName) / sizeof(WCHAR), ExpectedPathName);
+                if (Length == 3 && TestCases[i].FullPathName[0])
+                    ExpectedPathName[2] = UNICODE_NULL;
                 break;
+            }
             default:
                 skip(0, "Invalid test!\n");
                 continue;
@@ -212,17 +218,24 @@ RunTestCases(VOID)
                 break;
             case PrefixCurrentPath:
                 ExpectedFilePartSize = GetCurrentDirectoryW(0, NULL) * sizeof(WCHAR);
+                if (ExpectedFilePartSize == sizeof(L"C:\\"))
+                    ExpectedFilePartSize -= sizeof(WCHAR);
                 break;
             case PrefixCurrentPathWithoutLastPart:
             {
                 WCHAR CurrentPath[MAX_PATH];
                 PCWSTR BackSlash;
                 ExpectedFilePartSize = GetCurrentDirectoryW(sizeof(CurrentPath) / sizeof(WCHAR), CurrentPath) * sizeof(WCHAR) + sizeof(UNICODE_NULL);
-                BackSlash = wcsrchr(CurrentPath, L'\\');
-                if (BackSlash)
-                    ExpectedFilePartSize -= wcslen(BackSlash + 1) * sizeof(WCHAR);
+                if (ExpectedFilePartSize == sizeof(L"C:\\"))
+                    ExpectedFilePartSize = 0;
                 else
-                    ok(0, "GetCurrentDirectory returned %S\n", CurrentPath);
+                {
+                    BackSlash = wcsrchr(CurrentPath, L'\\');
+                    if (BackSlash)
+                        ExpectedFilePartSize -= wcslen(BackSlash + 1) * sizeof(WCHAR);
+                    else
+                        ok(0, "GetCurrentDirectory returned %S\n", CurrentPath);
+                }
                 break;
             }
             default:
index 440cede..0cc1866 100644 (file)
@@ -8,6 +8,7 @@
 extern void func_NtAllocateVirtualMemory(void);
 extern void func_NtFreeVirtualMemory(void);
 extern void func_NtSystemInformation(void);
+extern void func_RtlGetFullPathName_U(void);
 extern void func_RtlGetFullPathName_UstrEx(void);
 extern void func_RtlInitializeBitMap(void);
 extern void func_ZwContinue(void);
@@ -17,6 +18,7 @@ const struct test winetest_testlist[] =
     { "NtAllocateVirtualMemory",    func_NtAllocateVirtualMemory },
     { "NtFreeVirtualMemory",        func_NtFreeVirtualMemory },
     { "NtSystemInformation",        func_NtSystemInformation },
+    { "RtlGetFullPathName_U",       func_RtlGetFullPathName_U },
     { "RtlGetFullPathName_UstrEx",  func_RtlGetFullPathName_UstrEx },
     { "RtlInitializeBitMap",        func_RtlInitializeBitMap },
     { "ZwContinue",                 func_ZwContinue },