mirror of
https://github.com/reactos/reactos.git
synced 2024-08-26 21:28:48 +00:00
438 lines
13 KiB
C
438 lines
13 KiB
C
/*
|
|
* PROJECT: ReactOS api tests
|
|
* LICENSE: GPLv2+ - See COPYING in the top level directory
|
|
* PURPOSE: Test for RtlDosPathNameToNtPathName_U
|
|
* PROGRAMMER: Mike "tamlin" Nordell
|
|
*/
|
|
/* TODO:
|
|
* - Make the code UNICODE aware. If a user is inside a directory with
|
|
* non-ANSI characters somewhere in the path, all bets are currently off.
|
|
* - Remove hard-coded path size limits.
|
|
* - Un-tabify to match style of other code.
|
|
*/
|
|
|
|
/* Test to see that ntdll.RtlDosPathNameToNtPathName_U behaves _exactly_
|
|
* like Windows with all possible input to it.
|
|
* - relative path.
|
|
* - absolute paths
|
|
* - \\.\C:\foo
|
|
* - \\.\C:\foo\
|
|
* - \\?\C:\foo
|
|
* - \\?\C:\foo\
|
|
* - \??\C:
|
|
* - \??\C:\
|
|
*
|
|
* Caveat: The "\??\*" form behaves different depending on Windows version.
|
|
*
|
|
* Code is assumed to be compiled as 32-bit "ANSI" (i.e. with _UNICODE) undefined.
|
|
*/
|
|
|
|
// Enable this define to compile the test as a ReactOS auto-test.
|
|
// Disable it to compile on plain Win32, to test-run and get info.
|
|
#define COMPILE_AS_ROSTEST
|
|
|
|
#ifndef COMPILE_AS_ROSTEST
|
|
# define PRINT_INFO // Also print, in addition to testing
|
|
# include <windows.h>
|
|
# include <stdio.h>
|
|
# include <stddef.h>
|
|
#else /* Compile for ReactOS or wine */
|
|
# define WIN32_NO_STATUS
|
|
# include <stdio.h>
|
|
# include <wine/test.h>
|
|
# include <pseh/pseh2.h>
|
|
# include <ndk/rtlfuncs.h>
|
|
#endif
|
|
|
|
/*
|
|
BOOLEAN
|
|
NTAPI
|
|
RtlDosPathNameToNtPathName_U(IN PCWSTR DosName,
|
|
OUT PUNICODE_STRING NtName,
|
|
OUT PCWSTR *PartName,
|
|
OUT PRTL_RELATIVE_NAME_U RelativeName)
|
|
*/
|
|
|
|
#ifndef COMPILE_AS_ROSTEST
|
|
|
|
typedef struct UNICODE_STRING {
|
|
USHORT Length;
|
|
USHORT MaximumLength;
|
|
PWSTR Buffer;
|
|
} UNICODE_STRING, *PUNICODE_STRING;
|
|
|
|
typedef struct _RTLP_CURDIR_REF
|
|
{
|
|
LONG RefCount;
|
|
HANDLE Handle;
|
|
} RTLP_CURDIR_REF, *PRTLP_CURDIR_REF;
|
|
|
|
typedef struct RTL_RELATIVE_NAME_U {
|
|
UNICODE_STRING RelativeName;
|
|
HANDLE ContainingDirectory;
|
|
PRTLP_CURDIR_REF CurDirRef;
|
|
} RTL_RELATIVE_NAME_U, *PRTL_RELATIVE_NAME_U;
|
|
|
|
typedef BOOLEAN (__stdcall *RtlDosPathNameToNtPathName_U_t)(PCWSTR,PUNICODE_STRING,PCWSTR*,PRTL_RELATIVE_NAME_U);
|
|
|
|
RtlDosPathNameToNtPathName_U_t RtlDosPathNameToNtPathName_U;
|
|
|
|
#endif // !COMPILE_AS_ROSTEST
|
|
|
|
|
|
static void report_error(const char* pszErr)
|
|
{
|
|
#ifdef COMPILE_AS_ROSTEST
|
|
ok(FALSE, "%s\n", pszErr);
|
|
#else
|
|
printf("\a** %s!\n", pszErr);
|
|
#endif
|
|
}
|
|
|
|
|
|
#ifndef COMPILE_AS_ROSTEST
|
|
static void prucs(const char* pszDesc, UNICODE_STRING* pucs)
|
|
{
|
|
WCHAR wszTmp[512];
|
|
memcpy(wszTmp, pucs->Buffer, pucs->Length);
|
|
wszTmp[pucs->Length/2] = 0;
|
|
printf("%-12s: \"%S\"\n", pszDesc, wszTmp);
|
|
}
|
|
#endif
|
|
|
|
// Test RtlDosPathNameToNtPathName_U .
|
|
// pwszExpected shall contain the expected result for NtName
|
|
// *without* the leading "\??\".
|
|
// pwszExpectedPartName shall contain the expected result for PartName.
|
|
// NULL Expected means result is expected to be NULL too.
|
|
static void test2(LPCWSTR pwsz, LPCWSTR pwszExpected, LPCWSTR pwszExpectedPartName)
|
|
{
|
|
UNICODE_STRING NtName;
|
|
PWSTR PartName;
|
|
RTL_RELATIVE_NAME_U RelativeName;
|
|
BOOLEAN bOK;
|
|
|
|
bOK = RtlDosPathNameToNtPathName_U(pwsz, &NtName, (PCWSTR*)&PartName, &RelativeName);
|
|
|
|
if (!bOK) {
|
|
report_error("RtlDosPathNameToNtPathName_U failed");
|
|
printf("input: \"%S\"\n", pwsz);
|
|
return;
|
|
// exit(1);
|
|
}
|
|
|
|
#if !defined(COMPILE_AS_ROSTEST) && defined(PRINT_INFO)
|
|
printf("--------------------------\n");
|
|
printf("in : \"%S\"\n", pwsz);
|
|
prucs("NtName", &NtName);
|
|
if (PartName) {
|
|
printf("PartName : \"%S\"\n", PartName);
|
|
} else {
|
|
// This is not the place to test that printf handles NULL strings.
|
|
printf("PartName : (null)\n");
|
|
}
|
|
// prucs("RelativeName", &RelativeName.RelativeName);
|
|
#endif
|
|
|
|
// Disregarding input, output (NtName) shall always start with "\??\".
|
|
if (memcmp(NtName.Buffer, L"\\??\\", 8)) {
|
|
report_error("NtName does not start with \"\\??\\\"");
|
|
return;
|
|
// exit(1);
|
|
}
|
|
if (pwszExpected) {
|
|
PWSTR pwszActual = NtName.Buffer + 4;
|
|
const size_t lenExp = wcslen(pwszExpected);
|
|
const size_t lenAct = (NtName.Length - 8) / 2;
|
|
if (lenExp != lenAct ||
|
|
memcmp(pwszActual, pwszExpected, lenExp * 2))
|
|
{
|
|
report_error("Actual NtName does not match expectations!");
|
|
printf("Expected: %2u chars \"%S\"\n", lenExp, pwszExpected);
|
|
printf("Actual : %2u chars \"%S\"\n", lenAct, lenAct ? pwszActual : L"(null)");
|
|
return;
|
|
// exit(1);
|
|
}
|
|
} else
|
|
if (NtName.Length)
|
|
{
|
|
PWSTR pwszActual = NtName.Buffer + 4;
|
|
const size_t lenAct = (NtName.Length - 8) / 2;
|
|
report_error("Unexpected NtName (expected NULL)");
|
|
printf("Actual : %2u chars \"%S\"\n", lenAct, pwszActual);
|
|
}
|
|
|
|
if (pwszExpectedPartName) {
|
|
if (!PartName) {
|
|
report_error("Actual PartName is unexpectedly NULL!");
|
|
printf("Expected: \"%S\"\n", pwszExpectedPartName);
|
|
return;
|
|
// exit(1);
|
|
}
|
|
if (wcscmp(PartName, pwszExpectedPartName)) {
|
|
report_error("Actual PartName does not match expected!");
|
|
printf("Expected: \"%S\"\n", pwszExpectedPartName);
|
|
printf("Actual : \"%S\"\n", PartName);
|
|
return;
|
|
}
|
|
} else
|
|
if (PartName)
|
|
{
|
|
report_error("Unexpected PartName (expected NULL)");
|
|
printf("Actual : %S\n", PartName);
|
|
}
|
|
}
|
|
|
|
// NULL Expected means result is expected to be NULL too.
|
|
static void test(const char* psz, const char* pszExpected, const char* pszExpectedPartName)
|
|
{
|
|
WCHAR wszTmp1[512];
|
|
WCHAR wszTmp2[512];
|
|
WCHAR wszTmp3[512];
|
|
LPCWSTR p2 = 0;
|
|
LPCWSTR p3 = 0;
|
|
swprintf(wszTmp1, L"%S", psz);
|
|
if (pszExpected) {
|
|
swprintf(wszTmp2, L"%S", pszExpected);
|
|
p2 = wszTmp2;
|
|
}
|
|
if (pszExpectedPartName) {
|
|
swprintf(wszTmp3, L"%S", pszExpectedPartName);
|
|
p3 = wszTmp3;
|
|
}
|
|
test2(wszTmp1, p2, p3);
|
|
}
|
|
|
|
|
|
typedef struct DirComponents
|
|
{
|
|
char szCD[512];
|
|
char szCDPlusSlash[512];
|
|
char* pszLastCDComponent;
|
|
char szCurDrive[4];
|
|
char szParentDir[512];
|
|
char szParentDirPlusSlash[512];
|
|
char szNextLastCDComponent[260];
|
|
const char* pszNextLastCDComponent;
|
|
const char* pszParentDir;
|
|
const char* pszParentDirPlusSlash;
|
|
} DirComponents;
|
|
|
|
|
|
static void InitDirComponents(DirComponents* p)
|
|
{
|
|
p->pszNextLastCDComponent = 0;
|
|
p->pszParentDir = 0;
|
|
p->pszParentDirPlusSlash = 0;
|
|
|
|
GetCurrentDirectory(sizeof(p->szCD) / sizeof(*p->szCD), p->szCD);
|
|
|
|
if (strlen(p->szCD) < 2 || p->szCD[1] != ':') {
|
|
printf("Expected curdir to be a drive letter. It's not. It's \"%s\"\n", p->szCD);
|
|
#ifdef COMPILE_AS_ROSTEST
|
|
ok(FALSE, "Expected curdir to be a drive letter. It's not. It's \"%s\"\n", p->szCD);
|
|
#endif
|
|
exit(1);
|
|
}
|
|
|
|
// Note that if executed from the root directory, a slash already
|
|
// is appended. Take the opportunity to verify this.
|
|
if (p->szCD[2] != '\\') {
|
|
printf("CD is missing a slash as its third character! \"%s\"\n", p->szCD);
|
|
#ifdef COMPILE_AS_ROSTEST
|
|
ok(FALSE, "CD is missing a slash as its third character! \"%s\"\n", p->szCD);
|
|
#endif
|
|
exit(1);
|
|
}
|
|
|
|
strcpy(p->szCDPlusSlash, p->szCD);
|
|
if (strlen(p->szCD) > 3) {
|
|
// Append trailing backslash
|
|
strcat(p->szCDPlusSlash, "\\");
|
|
}
|
|
|
|
memcpy(p->szCurDrive, p->szCD, 2);
|
|
p->szCurDrive[2] = 0;
|
|
|
|
p->pszLastCDComponent = strrchr(p->szCD, '\\');
|
|
if (p->pszLastCDComponent)
|
|
{
|
|
// We have a parent directory
|
|
memcpy(p->szParentDir, p->szCD, p->pszLastCDComponent - p->szCD);
|
|
p->szParentDir[p->pszLastCDComponent - p->szCD] = 0;
|
|
p->pszParentDir = p->szParentDir;
|
|
if (strlen(p->szParentDir) == 2 && p->szParentDir[1] == ':') {
|
|
// When run from root directory, this is expected to
|
|
// have a trailing backslash
|
|
strcat(p->szParentDir, "\\");
|
|
}
|
|
strcpy(p->szParentDirPlusSlash, p->szParentDir);
|
|
if (p->szParentDirPlusSlash[strlen(p->szParentDirPlusSlash)-1] != '\\') {
|
|
strcat(p->szParentDirPlusSlash, "\\");
|
|
}
|
|
p->pszParentDirPlusSlash = p->szParentDirPlusSlash;
|
|
|
|
// Check if we have a directory above that too.
|
|
*p->pszLastCDComponent = 0;
|
|
p->pszNextLastCDComponent = strrchr(p->szCD, '\\');
|
|
*p->pszLastCDComponent = '\\';
|
|
if (p->pszNextLastCDComponent) {
|
|
++p->pszNextLastCDComponent; // skip the leading slash
|
|
if (p->pszNextLastCDComponent + 1 >= p->pszLastCDComponent) {
|
|
p->pszNextLastCDComponent = 0;
|
|
} else {
|
|
const size_t siz = p->pszLastCDComponent - p->pszNextLastCDComponent;
|
|
memcpy(p->szNextLastCDComponent, p->pszNextLastCDComponent, siz);
|
|
p->szNextLastCDComponent[siz] = 0;
|
|
p->pszNextLastCDComponent = p->szNextLastCDComponent;
|
|
}
|
|
}
|
|
}
|
|
if (p->pszLastCDComponent && p->pszLastCDComponent[1] == 0) {
|
|
// If the backslash is the last character in the path,
|
|
// this is a NULL-component.
|
|
p->pszLastCDComponent = 0;
|
|
} else {
|
|
++p->pszLastCDComponent; // skip the leading slash
|
|
}
|
|
}
|
|
|
|
|
|
#ifndef COMPILE_AS_ROSTEST
|
|
static void InitFunctionPointer()
|
|
{
|
|
HINSTANCE hDll = LoadLibrary("ntdll");
|
|
if (!hDll) {
|
|
printf("Major failure. Couldn't even load ntdll!\n");
|
|
exit(1);
|
|
}
|
|
RtlDosPathNameToNtPathName_U =
|
|
(RtlDosPathNameToNtPathName_U_t)GetProcAddress(hDll, "RtlDosPathNameToNtPathName_U");
|
|
if (!hDll) {
|
|
printf("Major failure. Couldn't get RtlDosPathNameToNtPathName_U!\n");
|
|
exit(1);
|
|
}
|
|
}
|
|
#endif // !COMPILE_AS_ROSTEST
|
|
|
|
|
|
#ifdef COMPILE_AS_ROSTEST
|
|
START_TEST(RtlDosPathNameToNtPathName_U)
|
|
#else
|
|
int main()
|
|
#endif
|
|
{
|
|
#if defined(PRINT_INFO)
|
|
#ifdef COMPILE_AS_ROSTEST
|
|
PPEB Peb = NtCurrentPeb();
|
|
const DWORD dwWinVer = (DWORD)(Peb->OSMinorVersion << 8) | Peb->OSMajorVersion;
|
|
#else
|
|
const DWORD dwWinVer = GetVersion();
|
|
#endif
|
|
const BYTE WinVerMaj = (BYTE)dwWinVer;
|
|
const BYTE WinVerMin = HIBYTE(LOWORD(dwWinVer));
|
|
#endif // PRINT_INFO
|
|
|
|
DirComponents cd;
|
|
|
|
#ifndef COMPILE_AS_ROSTEST
|
|
InitFunctionPointer();
|
|
#endif
|
|
|
|
InitDirComponents(&cd);
|
|
|
|
#if defined(PRINT_INFO)
|
|
printf("WinVer: %d.%d\n", WinVerMaj, WinVerMin);
|
|
printf("pszLastCDComponent \"%s\"\n", cd.pszLastCDComponent);
|
|
printf("pszNextLastCDComponent \"%s\"\n", cd.pszNextLastCDComponent);
|
|
printf("pszParentDir \"%s\"\n", cd.pszParentDir);
|
|
printf("pszParentDirPlusSlash \"%s\"\n", cd.pszParentDirPlusSlash);
|
|
#endif
|
|
|
|
|
|
#define PREP0
|
|
#define PREP1 "\\\\.\\"
|
|
#define PREP2 "\\\\?\\"
|
|
|
|
// The following tests shall return strictly defined strings,
|
|
// why we can use hard-coded expectations..
|
|
test(PREP1 "C:" , "C:" , "C:");
|
|
test(PREP2 "C:" , "C:" , "C:");
|
|
test(PREP0 "C:\\" , "C:\\" , NULL);
|
|
test(PREP1 "C:\\" , "C:\\" , NULL);
|
|
test(PREP2 "C:\\" , "C:\\" , NULL);
|
|
test(PREP0 "C:\\foo" , "C:\\foo" , "foo");
|
|
test(PREP1 "C:\\foo" , "C:\\foo" , "foo");
|
|
test(PREP2 "C:\\foo" , "C:\\foo" , "foo");
|
|
test(PREP0 "C:\\foo\\" , "C:\\foo\\" , NULL);
|
|
test(PREP1 "C:\\foo\\" , "C:\\foo\\" , NULL);
|
|
test(PREP2 "C:\\foo\\" , "C:\\foo\\" , NULL);
|
|
test(PREP0 "C:\\foo\\bar" , "C:\\foo\\bar" , "bar");
|
|
test(PREP1 "C:\\foo\\bar" , "C:\\foo\\bar" , "bar");
|
|
test(PREP2 "C:\\foo\\bar" , "C:\\foo\\bar" , "bar");
|
|
test(PREP0 "C:\\foo\\bar\\", "C:\\foo\\bar\\" , NULL);
|
|
test(PREP1 "C:\\foo\\bar\\", "C:\\foo\\bar\\" , NULL);
|
|
test(PREP2 "C:\\foo\\bar\\", "C:\\foo\\bar\\" , NULL);
|
|
test(PREP1 "." , "" , NULL);
|
|
test(PREP2 "." , "." , ".");
|
|
test(PREP1 ".\\" , "" , NULL);
|
|
test(PREP2 ".\\" , ".\\" , NULL);
|
|
test(PREP1 ".." , "" , NULL);
|
|
test(PREP2 ".." , ".." , "..");
|
|
test(PREP1 "..\\" , "" , NULL);
|
|
test(PREP2 "..\\" , "..\\" , NULL);
|
|
|
|
// The following tests returns results based on current directory.
|
|
test(PREP0 "." , cd.szCD , cd.pszLastCDComponent);
|
|
test(PREP0 ".\\" , cd.szCDPlusSlash , NULL);
|
|
test(PREP0 ".." , cd.pszParentDir , cd.pszNextLastCDComponent);
|
|
test(PREP0 "..\\" , cd.pszParentDirPlusSlash, NULL);
|
|
test(cd.szCD , cd.szCD , cd.pszLastCDComponent);
|
|
test(cd.szCDPlusSlash , cd.szCDPlusSlash , NULL);
|
|
|
|
#if 0
|
|
// This following test is "problematic", as it returns results based on
|
|
// what your CD on C: is, whether or not you actually run the program
|
|
// from C:. For that reason, it's currently disabled.
|
|
test(PREP0 "C:" , "C:\\" , NULL);
|
|
#endif
|
|
|
|
#if 0 // Disabled due to... see the comment inside the block.
|
|
{
|
|
char szExp[32];
|
|
BOOL bValid = FALSE;
|
|
char szPrepend[32];
|
|
szPrepend[0] = 0;
|
|
// Strictly speaking, this "Should Never Happen(tm)", calling
|
|
// RtlDosPathNameToNtPathName_U with a source already formed as
|
|
// a full NT name ("\??\"), why it's not the end of the world
|
|
// that this test is currently disabled.
|
|
//
|
|
// NOTE: _At least_ XP sp2 prepends this.
|
|
// Prepending curdrive like this is most likely a bug, but for
|
|
// compatibility it may become a requirement to "shim" this.
|
|
//
|
|
// Known operating systems prepending "Z:\??\" (assuming the
|
|
// process' CD is on the volume Z:):
|
|
// - XP sp2.
|
|
//
|
|
// Known operating systems not prepending:
|
|
// - Win7 64 (as 32-bit)
|
|
if (WinVerMaj == 5) {
|
|
sprintf(szPrepend, "%s\\??\\", cd.szCurDrive);
|
|
}
|
|
|
|
sprintf(szExp, "%s%s", szPrepend, "C:"); // Win7 64 (as 32-bit)
|
|
test("\\??\\C:", szExp, "C:");
|
|
|
|
sprintf(szExp, "%s%s", szPrepend, "C:\\");
|
|
test("\\??\\C:\\", szExp, NULL);
|
|
}
|
|
#endif
|
|
|
|
#ifndef COMPILE_AS_ROSTEST
|
|
return 0;
|
|
#endif
|
|
}
|
|
|