Author: tfaber Date: Sun Mar 25 20:01:12 2012 New Revision: 56229
URL: http://svn.reactos.org/svn/reactos?rev=56229&view=rev Log: [NTDLL_APITEST] - Add a test for RtlGetFullPathName_UstrEx
Added: trunk/rostests/apitests/ntdll/RtlGetFullPathName_UstrEx.c (with props) Modified: trunk/rostests/apitests/ntdll/CMakeLists.txt trunk/rostests/apitests/ntdll/testlist.c
Modified: trunk/rostests/apitests/ntdll/CMakeLists.txt URL: http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/ntdll/CMakeLists.... ============================================================================== --- trunk/rostests/apitests/ntdll/CMakeLists.txt [iso-8859-1] (original) +++ trunk/rostests/apitests/ntdll/CMakeLists.txt [iso-8859-1] Sun Mar 25 20:01:12 2012 @@ -2,6 +2,7 @@ list(APPEND SOURCE NtAllocateVirtualMemory.c NtFreeVirtualMemory.c + RtlGetFullPathName_UstrEx.c RtlInitializeBitMap.c SystemInfo.c ZwContinue.c @@ -12,7 +13,14 @@ endif()
add_executable(ntdll_apitest ${SOURCE}) -target_link_libraries(ntdll_apitest wine) +target_link_libraries(ntdll_apitest wine ${PSEH_LIB}) set_module_type(ntdll_apitest win32cui) add_importlibs(ntdll_apitest msvcrt advapi32 kernel32 ntdll) + +if(NOT MSVC) + # FIXME: http://www.cmake.org/Bug/view.php?id=12998 + #add_target_compile_flags(ntdll_apitest "-Wno-format") + set_source_files_properties(${SOURCE} PROPERTIES COMPILE_FLAGS "-Wno-format") +endif() + add_cd_file(TARGET ntdll_apitest DESTINATION reactos/bin FOR all)
Added: trunk/rostests/apitests/ntdll/RtlGetFullPathName_UstrEx.c URL: http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/ntdll/RtlGetFullP... ============================================================================== --- trunk/rostests/apitests/ntdll/RtlGetFullPathName_UstrEx.c (added) +++ trunk/rostests/apitests/ntdll/RtlGetFullPathName_UstrEx.c [iso-8859-1] Sun Mar 25 20:01:12 2012 @@ -1,0 +1,344 @@ +/* + * PROJECT: ReactOS api tests + * LICENSE: GPLv2+ - See COPYING in the top level directory + * PURPOSE: Test for RtlGetFullPathName_UstrEx + * PROGRAMMER: Thomas Faber thfabba@gmx.de + */ + +#define WIN32_NO_STATUS +#include <wine/test.h> +#include <pseh/pseh2.h> +#include <ndk/rtlfuncs.h> + +/* +NTSTATUS +NTAPI +RtlGetFullPathName_UstrEx(IN PUNICODE_STRING FileName, + IN PUNICODE_STRING StaticString, + IN PUNICODE_STRING DynamicString, + IN PUNICODE_STRING *StringUsed, + IN PSIZE_T FilePartSize OPTIONAL, + OUT PBOOLEAN NameInvalid, + OUT RTL_PATH_TYPE* PathType, + OUT PSIZE_T LengthNeeded OPTIONAL); +*/ + +#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) + +#define ok_eq_ustr(str1, str2) do { \ + ok((str1)->Buffer == (str2)->Buffer, "Buffer modified\n"); \ + ok((str1)->Length == (str2)->Length, "Length modified\n"); \ + ok((str1)->MaximumLength == (str2)->MaximumLength, "MaximumLength modified\n"); \ + } while (0) + +static +BOOLEAN +CheckStringBuffer( + PCUNICODE_STRING String, + PCWSTR Expected) +{ + USHORT Length = wcslen(Expected) * sizeof(WCHAR); + SIZE_T EqualLength; + BOOLEAN Result = TRUE; + SIZE_T i; + + if (String->Length != Length) + { + ok(0, "String length is %u, expected %u\n", String->Length, Length); + Result = FALSE; + } + + EqualLength = RtlCompareMemory(String->Buffer, Expected, Length); + if (EqualLength != Length) + { + ok(0, "String is '%wZ', expected '%S'\n", String, Expected); + Result = FALSE; + } + + if (String->Buffer[String->Length / sizeof(WCHAR)] != UNICODE_NULL) + { + ok(0, "Not null terminated\n"); + Result = FALSE; + } + + /* the function nulls the rest of the buffer! */ + for (i = String->Length + sizeof(UNICODE_NULL); i < String->MaximumLength; i++) + { + UCHAR Char = ((PUCHAR)String->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 RtlPathTypeNotSet 123 +#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) +{ + struct + { + PCWSTR FileName; + PREFIX_TYPE PrefixType; + PCWSTR FullPathName; + RTL_PATH_TYPE PathType; + PREFIX_TYPE FilePartPrefixType; + SIZE_T FilePartSize; + } TestCases[] = + { + { L"C:", PrefixCurrentPath, L"", RtlPathTypeDriveRelative, PrefixCurrentPathWithoutLastPart }, + { L"C:\", PrefixNone, L"C:\", RtlPathTypeDriveAbsolute }, + { L"C:\test", PrefixNone, L"C:\test", RtlPathTypeDriveAbsolute, PrefixCurrentDrive }, + { L"C:\test\", PrefixNone, L"C:\test\", RtlPathTypeDriveAbsolute }, + { L"C:/test/", PrefixNone, L"C:\test\", RtlPathTypeDriveAbsolute }, + + { L"C:\\test", PrefixNone, L"C:\test", RtlPathTypeDriveAbsolute, PrefixCurrentDrive }, + { L"test", PrefixCurrentPath, L"\test", RtlPathTypeRelative, PrefixCurrentPath, sizeof(WCHAR) }, + { L"\test", PrefixCurrentDrive, L"test", RtlPathTypeRooted, PrefixCurrentDrive }, + { L"/test", PrefixCurrentDrive, L"test", RtlPathTypeRooted, PrefixCurrentDrive }, + { L".\test", PrefixCurrentPath, L"\test", RtlPathTypeRelative, PrefixCurrentPath, sizeof(WCHAR) }, + + { L"\.", PrefixCurrentDrive, L"", RtlPathTypeRooted }, + { L"\.\", PrefixCurrentDrive, L"", RtlPathTypeRooted }, + { L"\\.", PrefixNone, L"\\.\", RtlPathTypeRootLocalDevice }, + { L"\\.\", PrefixNone, L"\\.\", RtlPathTypeLocalDevice }, + { L"\\.\Something\", PrefixNone, L"\\.\Something\", RtlPathTypeLocalDevice }, + + { L"\??\", PrefixCurrentDrive, L"??\", RtlPathTypeRooted }, + { L"\??\C:", PrefixCurrentDrive, L"??\C:", RtlPathTypeRooted, PrefixCurrentDrive, 3 * sizeof(WCHAR) }, + { L"\??\C:\", PrefixCurrentDrive, L"??\C:\", RtlPathTypeRooted }, + { L"\??\C:\test", PrefixCurrentDrive, L"??\C:\test", RtlPathTypeRooted, PrefixCurrentDrive, 6 * sizeof(WCHAR) }, + { L"\??\C:\test\", PrefixCurrentDrive, L"??\C:\test\", RtlPathTypeRooted }, + + { L"\\??\", PrefixNone, L"\\??\", RtlPathTypeUncAbsolute }, + { L"\\??\C:", PrefixNone, L"\\??\C:", RtlPathTypeUncAbsolute }, + { L"\\??\C:\", PrefixNone, L"\\??\C:\", RtlPathTypeUncAbsolute }, + { L"\\??\C:\test", PrefixNone, L"\\??\C:\test", RtlPathTypeUncAbsolute, PrefixNone, sizeof(L"\\??\C:\") }, + { L"\\??\C:\test\", PrefixNone, L"\\??\C:\test\", RtlPathTypeUncAbsolute }, + }; + NTSTATUS Status, ExceptionStatus; + UNICODE_STRING FileName; + UNICODE_STRING FullPathName; + WCHAR FullPathNameBuffer[MAX_PATH]; + UNICODE_STRING TempString; + PUNICODE_STRING StringUsed; + SIZE_T FilePartSize; + BOOLEAN NameInvalid; + RTL_PATH_TYPE PathType; + SIZE_T LengthNeeded; + WCHAR ExpectedPathName[MAX_PATH]; + 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: + GetCurrentDirectoryW(sizeof(ExpectedPathName) / sizeof(WCHAR), ExpectedPathName); + break; + default: + skip(0, "Invalid test!\n"); + continue; + } + wcscat(ExpectedPathName, TestCases[i].FullPathName); + RtlInitUnicodeString(&FileName, TestCases[i].FileName); + RtlInitEmptyUnicodeString(&FullPathName, FullPathNameBuffer, sizeof(FullPathNameBuffer)); + RtlFillMemory(FullPathName.Buffer, FullPathName.MaximumLength, 0xAA); + TempString = FileName; + PathType = RtlPathTypeNotSet; + StringUsed = InvalidPointer; + FilePartSize = 1234; + NameInvalid = (BOOLEAN)-1; + LengthNeeded = 1234; + StartSeh() + Status = RtlGetFullPathName_UstrEx(&FileName, + &FullPathName, + NULL, + &StringUsed, + &FilePartSize, + &NameInvalid, + &PathType, + &LengthNeeded); + ok(Status == STATUS_SUCCESS, "status = %lx\n", Status); + EndSeh(STATUS_SUCCESS); + ok_eq_ustr(&FileName, &TempString); + ok(FullPathName.Buffer == FullPathNameBuffer, "Buffer modified\n"); + ok(FullPathName.MaximumLength == sizeof(FullPathNameBuffer), "MaximumLength modified\n"); + /* TODO: remove SEH here */ + StartSeh() + ok(CheckStringBuffer(&FullPathName, ExpectedPathName), + "Wrong path name '%wZ', expected '%S'\n", &FullPathName, ExpectedPathName); + EndSeh(STATUS_SUCCESS); + ok(StringUsed == &FullPathName, "StringUsed = %p, expected %p\n", StringUsed, &FullPathName); + switch (TestCases[i].FilePartPrefixType) + { + case PrefixNone: + ExpectedFilePartSize = 0; + break; + case PrefixCurrentDrive: + ExpectedFilePartSize = sizeof(L"C:\"); + break; + case PrefixCurrentPath: + ExpectedFilePartSize = GetCurrentDirectoryW(0, NULL) * 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); + 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); + ok(NameInvalid == FALSE, "NameInvalid = %u\n", NameInvalid); + ok(PathType == TestCases[i].PathType, "PathType = %d, expected %d\n", PathType, TestCases[i].PathType); + ok(LengthNeeded == 0, "LengthNeeded = %lu\n", (ULONG)LengthNeeded); + } +} + +START_TEST(RtlGetFullPathName_UstrEx) +{ + NTSTATUS Status, ExceptionStatus; + UNICODE_STRING FileName; + UNICODE_STRING TempString; + UNICODE_STRING StaticString; + PUNICODE_STRING StringUsed; + SIZE_T FilePartSize; + BOOLEAN NameInvalid; + RTL_PATH_TYPE PathType; + SIZE_T LengthNeeded; + + /* NULL parameters */ + StartSeh() + RtlGetFullPathName_UstrEx(NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL); + EndSeh(STATUS_ACCESS_VIOLATION); + + RtlInitUnicodeString(&FileName, L""); + TempString = FileName; + StartSeh() + RtlGetFullPathName_UstrEx(&FileName, NULL, NULL, NULL, NULL, NULL, NULL, NULL); + EndSeh(STATUS_ACCESS_VIOLATION); + ok_eq_ustr(&FileName, &TempString); + + PathType = RtlPathTypeNotSet; + StartSeh() + RtlGetFullPathName_UstrEx(NULL, NULL, NULL, NULL, NULL, NULL, &PathType, NULL); + EndSeh(STATUS_ACCESS_VIOLATION); + ok(PathType == RtlPathTypeUnknown || + broken(PathType == RtlPathTypeNotSet) /* Win7 */, "PathType = %d\n", PathType); + + /* check what else is initialized before it crashes */ + PathType = RtlPathTypeNotSet; + StringUsed = InvalidPointer; + FilePartSize = 1234; + NameInvalid = (BOOLEAN)-1; + LengthNeeded = 1234; + StartSeh() + RtlGetFullPathName_UstrEx(NULL, NULL, NULL, &StringUsed, &FilePartSize, &NameInvalid, &PathType, &LengthNeeded); + EndSeh(STATUS_ACCESS_VIOLATION); + ok(StringUsed == NULL, "StringUsed = %p\n", StringUsed); + ok(FilePartSize == 0, "FilePartSize = %lu\n", (ULONG)FilePartSize); + ok(NameInvalid == FALSE, "NameInvalid = %u\n", NameInvalid); + ok(PathType == RtlPathTypeUnknown || + broken(PathType == RtlPathTypeNotSet) /* Win7 */, "PathType = %d\n", PathType); + ok(LengthNeeded == 0, "LengthNeeded = %lu\n", (ULONG)LengthNeeded); + + RtlInitUnicodeString(&FileName, L""); + TempString = FileName; + StringUsed = InvalidPointer; + FilePartSize = 1234; + NameInvalid = (BOOLEAN)-1; + LengthNeeded = 1234; + StartSeh() + RtlGetFullPathName_UstrEx(&FileName, NULL, NULL, &StringUsed, &FilePartSize, &NameInvalid, NULL, &LengthNeeded); + EndSeh(STATUS_ACCESS_VIOLATION); + ok_eq_ustr(&FileName, &TempString); + ok(StringUsed == NULL, "StringUsed = %p\n", StringUsed); + ok(FilePartSize == 0, "FilePartSize = %lu\n", (ULONG)FilePartSize); + ok(NameInvalid == FALSE || + broken(NameInvalid == (BOOLEAN)-1) /* Win7 */, "NameInvalid = %u\n", NameInvalid); + ok(LengthNeeded == 0, "LengthNeeded = %lu\n", (ULONG)LengthNeeded); + + /* This is the first one that doesn't crash. FileName and PathType cannot be NULL */ + RtlInitUnicodeString(&FileName, L""); + TempString = FileName; + PathType = RtlPathTypeNotSet; + StartSeh() + Status = RtlGetFullPathName_UstrEx(&FileName, NULL, NULL, NULL, NULL, NULL, &PathType, NULL); + ok(Status == STATUS_OBJECT_NAME_INVALID, "status = %lx\n", Status); + EndSeh(STATUS_SUCCESS); + ok_eq_ustr(&FileName, &TempString); + ok(PathType == RtlPathTypeUnknown, "PathType = %d\n", PathType); + + /* Give it a valid path */ + RtlInitUnicodeString(&FileName, L"C:\test"); + TempString = FileName; + PathType = RtlPathTypeNotSet; + StartSeh() + Status = RtlGetFullPathName_UstrEx(&FileName, NULL, NULL, NULL, NULL, NULL, &PathType, NULL); + ok(Status == STATUS_BUFFER_TOO_SMALL, "status = %lx\n", Status); + EndSeh(STATUS_SUCCESS); + ok_eq_ustr(&FileName, &TempString); + ok(PathType == RtlPathTypeDriveAbsolute, "PathType = %d\n", PathType); + + /* Zero-length static string */ + RtlInitUnicodeString(&FileName, L"C:\test"); + TempString = FileName; + RtlInitUnicodeString(&StaticString, NULL); + PathType = RtlPathTypeNotSet; + StartSeh() + Status = RtlGetFullPathName_UstrEx(&FileName, &StaticString, NULL, NULL, NULL, NULL, &PathType, NULL); + ok(Status == STATUS_BUFFER_TOO_SMALL, "status = %lx\n", Status); + EndSeh(STATUS_SUCCESS); + ok_eq_ustr(&FileName, &TempString); + ok(PathType == RtlPathTypeDriveAbsolute, "PathType = %d\n", PathType); + + /* TODO: play around with StaticString and DynamicString */ + + /* check the actual functionality with different paths */ + RunTestCases(); +}
Propchange: trunk/rostests/apitests/ntdll/RtlGetFullPathName_UstrEx.c ------------------------------------------------------------------------------ svn:eol-style = native
Modified: trunk/rostests/apitests/ntdll/testlist.c URL: http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/ntdll/testlist.c?... ============================================================================== --- trunk/rostests/apitests/ntdll/testlist.c [iso-8859-1] (original) +++ trunk/rostests/apitests/ntdll/testlist.c [iso-8859-1] Sun Mar 25 20:01:12 2012 @@ -8,6 +8,7 @@ extern void func_NtAllocateVirtualMemory(void); extern void func_NtFreeVirtualMemory(void); extern void func_NtSystemInformation(void); +extern void func_RtlGetFullPathName_UstrEx(void); extern void func_RtlInitializeBitMap(void); extern void func_ZwContinue(void);
@@ -16,9 +17,9 @@ { "NtAllocateVirtualMemory", func_NtAllocateVirtualMemory }, { "NtFreeVirtualMemory", func_NtFreeVirtualMemory }, { "NtSystemInformation", func_NtSystemInformation }, + { "RtlGetFullPathName_UstrEx", func_RtlGetFullPathName_UstrEx }, { "RtlInitializeBitMap", func_RtlInitializeBitMap }, { "ZwContinue", func_ZwContinue },
{ 0, 0 } }; -
-
tfaber＠svn.reactos.org