From 5c2322f0f89a2d2995249bec2d8aebd8a1945383 Mon Sep 17 00:00:00 2001 From: Thomas Faber Date: Sun, 25 Mar 2012 21:24:05 +0000 Subject: [PATCH] [NTDLL_WINETEST] - 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 | 1 + .../apitests/ntdll/RtlGetFullPathName_U.c | 283 ++++++++++++++++++ .../ntdll/RtlGetFullPathName_UstrEx.c | 23 +- rostests/apitests/ntdll/testlist.c | 2 + 4 files changed, 304 insertions(+), 5 deletions(-) create mode 100644 rostests/apitests/ntdll/RtlGetFullPathName_U.c diff --git a/rostests/apitests/ntdll/CMakeLists.txt b/rostests/apitests/ntdll/CMakeLists.txt index e1804440b7e..32caae5ff5f 100644 --- a/rostests/apitests/ntdll/CMakeLists.txt +++ b/rostests/apitests/ntdll/CMakeLists.txt @@ -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 index 00000000000..e9bc5496162 --- /dev/null +++ b/rostests/apitests/ntdll/RtlGetFullPathName_U.c @@ -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 + */ + +#define WIN32_NO_STATUS +#include +#include +#include + +/* +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(); +} diff --git a/rostests/apitests/ntdll/RtlGetFullPathName_UstrEx.c b/rostests/apitests/ntdll/RtlGetFullPathName_UstrEx.c index 9b3fddef22c..63aba5ff5d4 100644 --- a/rostests/apitests/ntdll/RtlGetFullPathName_UstrEx.c +++ b/rostests/apitests/ntdll/RtlGetFullPathName_UstrEx.c @@ -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: diff --git a/rostests/apitests/ntdll/testlist.c b/rostests/apitests/ntdll/testlist.c index 440cedefbfe..0cc186662bf 100644 --- a/rostests/apitests/ntdll/testlist.c +++ b/rostests/apitests/ntdll/testlist.c @@ -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 }, -- 2.17.1