--- /dev/null
+/*
+ * 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();
+}