OS_Develop/Windows-Server-2003
OS_Develop/Windows-Server-2003
1
0
Fork 0
Code Pull Requests Activity
master
Windows-Server-2003/base/ntdll/curdir.c

5055 lines
154 KiB
C
Raw Permalink Blame History

/*++
Copyright (c) 1990 Microsoft Corporation
Module Name:
curdir.c
Abstract:
Current directory support
Author:
Mark Lucovsky (markl) 10-Oct-1990
Revision History:
--*/
#pragma warning(disable:4201) // nameless struct/union
#include "nt.h"
#include "ntos.h"
#include "ntrtl.h"
#include "nturtl.h"
#include "string.h"
#include "ctype.h"
#include "sxstypes.h"
#include "ntdllp.h"
#include "ntcsrmsg.h"
#include "base.h"
#if defined(_WIN64) || defined(BUILD_WOW6432)
#include <wow64t.h>
#endif // defined(_WIN64)
#define IS_PATH_SEPARATOR_U(ch) (((ch) == L'\\') || ((ch) == L'/'))
#define IS_END_OF_COMPONENT_U(ch) (IS_PATH_SEPARATOR_U(ch) || (ch) == UNICODE_NULL)
#define IS_DOT_U(s) ( (s)[0] == L'.' && IS_END_OF_COMPONENT_U( (s)[1] ))
#define IS_DOT_DOT_U(s) ( (s)[0] == L'.' && IS_DOT_U( (s) + 1))
#define IS_DRIVE_LETTER(ch) (((ch) >= 'a' && (ch) <= 'z') || ((ch) >= 'A' && (ch) <= 'Z'))
#define IS_END_OF_COMPONENT_USTR(s, len) \
((len) == 0 || IS_PATH_SEPARATOR_U((s)[0]))
#define IS_DOT_USTR(s, len) \
((len) >= sizeof(WCHAR) && (s)[0] == L'.' && \
IS_END_OF_COMPONENT_USTR( (s) + 1, (len) - sizeof(WCHAR) ))
#define IS_DOT_DOT_USTR(s, len) \
((len) >= sizeof(WCHAR) && (s)[0] == L'.' && \
IS_DOT_USTR( (s) + 1, (len) - sizeof(WCHAR) ))
extern const UNICODE_STRING RtlpDosLPTDevice = RTL_CONSTANT_STRING( L"LPT" );
extern const UNICODE_STRING RtlpDosCOMDevice = RTL_CONSTANT_STRING( L"COM" );
extern const UNICODE_STRING RtlpDosPRNDevice = RTL_CONSTANT_STRING( L"PRN" );
extern const UNICODE_STRING RtlpDosAUXDevice = RTL_CONSTANT_STRING( L"AUX" );
extern const UNICODE_STRING RtlpDosNULDevice = RTL_CONSTANT_STRING( L"NUL" );
extern const UNICODE_STRING RtlpDosCONDevice = RTL_CONSTANT_STRING( L"CON" );
extern const UNICODE_STRING RtlpDosSlashCONDevice = RTL_CONSTANT_STRING( L"\\\\.\\CON" );
extern const UNICODE_STRING RtlpSlashSlashDot = RTL_CONSTANT_STRING( L"\\\\.\\" );
extern const UNICODE_STRING RtlpDosDevicesPrefix = RTL_CONSTANT_STRING( L"\\??\\" );
extern const UNICODE_STRING RtlpDosDevicesUncPrefix = RTL_CONSTANT_STRING( L"\\??\\UNC\\" );
#define RtlpLongestPrefix RtlpDosDevicesUncPrefix.Length
const UNICODE_STRING RtlpEmptyString = RTL_CONSTANT_STRING(L"");
//
// \\? is referred to as the "Win32Nt" prefix or root.
// Paths that start with \\? are referred to as "Win32Nt" paths.
// Fudging the \\? to \?? converts the path to an Nt path.
//
extern const UNICODE_STRING RtlpWin32NtRoot = RTL_CONSTANT_STRING( L"\\\\?" );
extern const UNICODE_STRING RtlpWin32NtRootSlash = RTL_CONSTANT_STRING( L"\\\\?\\" );
extern const UNICODE_STRING RtlpWin32NtUncRoot = RTL_CONSTANT_STRING( L"\\\\?\\UNC" );
extern const UNICODE_STRING RtlpWin32NtUncRootSlash = RTL_CONSTANT_STRING( L"\\\\?\\UNC\\" );
#define DPFLTR_LEVEL_STATUS(x) ((NT_SUCCESS(x) \
|| (x) == STATUS_OBJECT_NAME_NOT_FOUND \
) \
? DPFLTR_TRACE_LEVEL : DPFLTR_ERROR_LEVEL)
typedef struct _RTLP_CURDIR_REF {
LONG RefCount;
HANDLE DirectoryHandle;
} RTLP_CURDIR_REF;
PRTLP_CURDIR_REF RtlpCurDirRef = NULL; // N.B. Protected by the PEB lock
FORCEINLINE
VOID
RtlpInitializeCurDirRef(
OUT PRTLP_CURDIR_REF CurDirRef,
IN HANDLE DirectoryHandle
)
{
ASSERT(CurDirRef);
CurDirRef->RefCount = 1;
CurDirRef->DirectoryHandle = DirectoryHandle;
}
FORCEINLINE
VOID
RtlpReferenceCurDirRef(
IN PRTLP_CURDIR_REF CurDirRef
)
{
ASSERT(CurDirRef);
if (CurDirRef) {
InterlockedIncrement(&CurDirRef->RefCount);
}
}
FORCEINLINE
VOID
RtlpDereferenceCurDirRef(
IN PRTLP_CURDIR_REF CurDirRef
)
{
ASSERT(CurDirRef);
if (CurDirRef && ! InterlockedDecrement(&CurDirRef->RefCount)) {
RTL_VERIFY(
NT_SUCCESS(
NtClose(CurDirRef->DirectoryHandle)
));
RtlFreeHeap(RtlProcessHeap(), 0, CurDirRef);
}
}
ULONG
RtlpComputeBackupIndex(
IN PCURDIR CurDir
)
{
ULONG BackupIndex;
PWSTR UncPathPointer;
ULONG NumberOfPathSeparators;
RTL_PATH_TYPE CurDirPathType;
//
// Get pathType of curdir
//
CurDirPathType = RtlDetermineDosPathNameType_U(CurDir->DosPath.Buffer);
BackupIndex = 3;
if ( CurDirPathType == RtlPathTypeUncAbsolute ) {
//
// We want to scan the supplied path to determine where
// the "share" ends, and set BackupIndex to that point.
//
UncPathPointer = CurDir->DosPath.Buffer + 2;
NumberOfPathSeparators = 0;
while (*UncPathPointer) {
if (IS_PATH_SEPARATOR_U(*UncPathPointer)) {
NumberOfPathSeparators++;
if (NumberOfPathSeparators == 2) {
break;
}
}
UncPathPointer++;
}
BackupIndex = (ULONG)(UncPathPointer - CurDir->DosPath.Buffer);
}
return BackupIndex;
}
ULONG
RtlGetLongestNtPathLength( VOID )
/*++
Routine Description:
Returns the character count of the longest NT path directly translated from
a DOS path
Arguments:
None
Return Value:
the character count (NOT byte count)
--*/
{
//
// return the char count
//
return (RtlpLongestPrefix/sizeof(WCHAR)) + DOS_MAX_PATH_LENGTH + 1;
}
VOID
RtlpResetDriveEnvironment(
IN WCHAR DriveLetter
)
{
WCHAR EnvVarNameBuffer[4];
WCHAR EnvVarNameValue[4];
UNICODE_STRING s1,s2;
EnvVarNameBuffer[0] = L'=';
EnvVarNameBuffer[1] = DriveLetter;
EnvVarNameBuffer[2] = L':';
EnvVarNameBuffer[3] = L'\0';
RtlInitUnicodeString(&s1,EnvVarNameBuffer);
EnvVarNameValue[0] = DriveLetter;
EnvVarNameValue[1] = L':';
EnvVarNameValue[2] = L'\\';
EnvVarNameValue[3] = L'\0';
RtlInitUnicodeString(&s2,EnvVarNameValue);
RtlSetEnvironmentVariable(NULL,&s1,&s2);
}
ULONG
RtlGetCurrentDirectory_U(
ULONG nBufferLength,
PWSTR lpBuffer
)
/*++
Routine Description:
The current directory for a process can be retrieved using
GetCurrentDirectory.
Arguments:
nBufferLength - Supplies the length in bytes of the buffer that is to
receive the current directory string.
lpBuffer - Returns the current directory string for the current
process. The string is a null terminated string and specifies
the absolute path to the current directory.
Return Value:
The return value is the length of the string copied to lpBuffer, not
including the terminating null character. If the return value is
greater than nBufferLength, the return value is the size of the buffer
required to hold the pathname. The return value is zero if the
function failed.
--*/
{
PCURDIR CurDir;
ULONG Length;
PWSTR CurDirName;
CurDir = &NtCurrentPeb()->ProcessParameters->CurrentDirectory;
RtlAcquirePebLock();
CurDirName = CurDir->DosPath.Buffer;
//
// Make sure user's buffer is big enough to hold the null
// terminated current directory
//
Length = CurDir->DosPath.Length>>1;
//
// Assuming CurDir->DosPath.Buffer always has a trailing slash
// if the root of a drive,
// then return x:\
// else return x:\directory\subdir
//
ASSERT( (CurDirName != NULL) && (Length > 0) );
if ((Length > 1) && (CurDirName[Length-2] != L':')) {
if (nBufferLength < (Length)<<1) {
RtlReleasePebLock();
return (Length)<<1;
}
} else {
if (nBufferLength <= (Length<<1)) {
RtlReleasePebLock();
return ((Length+1)<<1);
}
}
try {
RtlCopyMemory (lpBuffer,CurDirName,Length<<1);
ASSERT(lpBuffer[Length-1] == L'\\');
if ((Length > 1) && (lpBuffer[Length-2] == L':')) {
//
// root of a drive, append a null
//
lpBuffer[Length] = UNICODE_NULL;
} else {
//
// some directory, change the trailing slash to a null
//
lpBuffer[Length-1] = UNICODE_NULL;
Length--;
}
} except (EXCEPTION_EXECUTE_HANDLER) {
RtlReleasePebLock();
return 0L;
}
RtlReleasePebLock();
return Length<<1;
}
NTSTATUS
RtlSetCurrentDirectory_U(
PCUNICODE_STRING PathName
)
/*++
Routine Description:
The current directory for a process is changed using
SetCurrentDirectory.
Each process has a single current directory. A current directory is
made up of type parts.
- A disk designator either which is either a drive letter followed
by a colon, or a UNC servername/sharename "\\servername\sharename".
- A directory on the disk designator.
For APIs that manipulate files, the file names may be relative to
the current directory. A filename is relative to the entire current
directory if it does not begin with a disk designator or a path name
SEPARATOR. If the file name begins with a path name SEPARATOR, then
it is relative to the disk designator of the current directory. If
a file name begins with a disk designator, than it is a fully
qualified absolute path name.
The value of lpPathName supplies the current directory. The value
of lpPathName, may be a relative path name as described above, or a
fully qualified absolute path name. In either case, the fully
qualified absolute path name of the specified directory is
calculated and is stored as the current directory.
Arguments:
lpPathName - Supplies the pathname of the directory that is to be
made the current directory.
Return Value:
NT_SUCCESS - The operation was successful
!NT_SUCCESS - The operation failed
--*/
{
PCURDIR CurDir;
NTSTATUS Status;
BOOLEAN TranslationStatus;
PVOID FreeBuffer;
ULONG DosDirLength;
ULONG IsDevice;
ULONG DosDirCharCount;
UNICODE_STRING DosDir;
UNICODE_STRING NtFileName;
HANDLE NewDirectoryHandle;
OBJECT_ATTRIBUTES Obja;
IO_STATUS_BLOCK IoStatusBlock;
FILE_FS_DEVICE_INFORMATION DeviceInfo;
RTL_PATH_TYPE InputPathType;
PPEB Peb;
HANDLE HandleToClose, HandleToClose1;
PRTLP_CURDIR_REF CurDirRefToDereference;
HANDLE Heap;
Peb = NtCurrentPeb();
Heap = Peb->ProcessHeap;
CurDir = &Peb->ProcessParameters->CurrentDirectory;
DosDir.Buffer = NULL;
FreeBuffer = NULL;
NewDirectoryHandle = NULL;
HandleToClose = NULL;
HandleToClose1 = NULL;
CurDirRefToDereference = NULL;
IsDevice = RtlIsDosDeviceName_Ustr(PathName);
RtlAcquirePebLock();
Peb->EnvironmentUpdateCount += 1;
//
// Set current directory is called first by the loader.
// If current directory is not being inherited, then close
// it !
//
if (((ULONG_PTR)CurDir->Handle & OBJ_HANDLE_TAGBITS) == RTL_USER_PROC_CURDIR_CLOSE) {
HandleToClose = CurDir->Handle;
CurDir->Handle = NULL;
}
Status = STATUS_SUCCESS;
try {
try {
if ( IsDevice ) {
Status = STATUS_NOT_A_DIRECTORY;
leave;
}
//
// Compute the length of the Dos style fully qualified current
// directory
//
DosDirLength = CurDir->DosPath.MaximumLength;
DosDir.Buffer = RtlAllocateHeap (Heap, 0, DosDirLength);
if ( !DosDir.Buffer ) {
Status = STATUS_NO_MEMORY;
leave;
}
DosDir.Length = 0;
DosDir.MaximumLength = (USHORT)DosDirLength;
//
// Now get the full pathname for the Dos style current
// directory
//
DosDirLength = RtlGetFullPathName_Ustr(
PathName,
DosDirLength,
DosDir.Buffer,
NULL,
NULL,
&InputPathType
);
if ( !DosDirLength ) {
Status = STATUS_OBJECT_NAME_INVALID;
leave;
}
if ( DosDirLength > DosDir.MaximumLength ) {
Status = STATUS_NAME_TOO_LONG;
leave;
}
DosDirCharCount = DosDirLength >> 1;
//
// Get the Nt filename of the new current directory
//
TranslationStatus = RtlDosPathNameToNtPathName_U(
DosDir.Buffer,
&NtFileName,
NULL,
NULL
);
if ( !TranslationStatus ) {
Status = STATUS_OBJECT_NAME_INVALID;
leave;
}
FreeBuffer = NtFileName.Buffer;
InitializeObjectAttributes(
&Obja,
&NtFileName,
OBJ_CASE_INSENSITIVE | OBJ_INHERIT,
NULL,
NULL
);
//
// If we are inheriting current directory, then
// avoid the open
//
if ( ((ULONG_PTR)CurDir->Handle & OBJ_HANDLE_TAGBITS) == RTL_USER_PROC_CURDIR_INHERIT ) {
NewDirectoryHandle = (HANDLE)((ULONG_PTR)CurDir->Handle & ~OBJ_HANDLE_TAGBITS);
CurDir->Handle = NULL;
//
// Test to see if this is removable media. If so
// tag the handle this may fail if the process was
// created with inherit handles set to false
//
Status = NtQueryVolumeInformationFile(
NewDirectoryHandle,
&IoStatusBlock,
&DeviceInfo,
sizeof(DeviceInfo),
FileFsDeviceInformation
);
if ( !NT_SUCCESS(Status) ) {
Status = RtlSetCurrentDirectory_U(PathName);
leave;
} else {
if ( DeviceInfo.Characteristics & FILE_REMOVABLE_MEDIA ) {
NewDirectoryHandle =(HANDLE)( (ULONG_PTR)NewDirectoryHandle | 1);
}
}
} else {
//
// Open a handle to the current directory. Don't allow
// deletes of the directory.
//
Status = NtOpenFile(
&NewDirectoryHandle,
FILE_TRAVERSE | SYNCHRONIZE,
&Obja,
&IoStatusBlock,
FILE_SHARE_READ | FILE_SHARE_WRITE,
FILE_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT
);
if ( !NT_SUCCESS(Status) ) {
leave;
}
//
// Test to see if this is removable media. If so
// tag the handle
//
Status = NtQueryVolumeInformationFile(
NewDirectoryHandle,
&IoStatusBlock,
&DeviceInfo,
sizeof(DeviceInfo),
FileFsDeviceInformation
);
if ( !NT_SUCCESS(Status) ) {
leave;
} else {
if ( DeviceInfo.Characteristics & FILE_REMOVABLE_MEDIA ) {
NewDirectoryHandle =(HANDLE)( (ULONG_PTR)NewDirectoryHandle | 1);
}
}
}
//
// If there is no trailing '\', than place one
//
DosDir.Length = (USHORT)DosDirLength;
if ( DosDir.Buffer[DosDirCharCount-1] != L'\\') {
if ( (DosDirCharCount + 2) >
(DosDir.MaximumLength / sizeof( WCHAR )) ) {
Status = STATUS_NAME_TOO_LONG;
leave;
}
DosDir.Buffer[DosDirCharCount] = L'\\';
DosDir.Buffer[DosDirCharCount+1] = UNICODE_NULL;
DosDir.Length += sizeof( UNICODE_NULL );
}
//
// Now we are set to change to the new directory.
//
if (RtlpCurDirRef
&& RtlpCurDirRef->RefCount == 1) {
HandleToClose1 = RtlpCurDirRef->DirectoryHandle;
RtlpCurDirRef->DirectoryHandle = NewDirectoryHandle;
} else {
CurDirRefToDereference = RtlpCurDirRef;
RtlpCurDirRef = RtlAllocateHeap(RtlProcessHeap(), 0, sizeof(RTLP_CURDIR_REF));
if (! RtlpCurDirRef) {
RtlpCurDirRef = CurDirRefToDereference;
CurDirRefToDereference = NULL;
Status = STATUS_NO_MEMORY;
leave;
}
RtlpInitializeCurDirRef(RtlpCurDirRef, NewDirectoryHandle);
}
CurDir->Handle = NewDirectoryHandle;
NewDirectoryHandle = NULL;
RtlCopyMemory (CurDir->DosPath.Buffer, DosDir.Buffer, DosDir.Length + sizeof (UNICODE_NULL));
CurDir->DosPath.Length = DosDir.Length;
} finally {
RtlReleasePebLock();
if (DosDir.Buffer != NULL) {
RtlFreeHeap (Heap, 0, DosDir.Buffer);
}
if (FreeBuffer != NULL) {
RtlFreeHeap (Heap, 0, FreeBuffer);
}
if (NewDirectoryHandle != NULL) {
NtClose (NewDirectoryHandle);
}
if (HandleToClose != NULL) {
NtClose (HandleToClose);
}
if (HandleToClose1 != NULL) {
NtClose (HandleToClose1);
}
if (CurDirRefToDereference != NULL) {
RtlpDereferenceCurDirRef(CurDirRefToDereference);
}
}
} except (EXCEPTION_EXECUTE_HANDLER) {
Status = STATUS_ACCESS_VIOLATION;
}
return Status;
}
RTL_PATH_TYPE
RtlDetermineDosPathNameType_U(
IN PCWSTR DosFileName
)
/*++
Routine Description:
This function examines the Dos format file name and determines the
type of file name (i.e. UNC, DriveAbsolute, Current Directory
rooted, or Relative.)
Arguments:
DosFileName - Supplies the Dos format file name whose type is to be
determined.
Return Value:
RtlPathTypeUnknown - The path type can not be determined
RtlPathTypeUncAbsolute - The path specifies a Unc absolute path
in the format \\server-name\sharename\rest-of-path
RtlPathTypeLocalDevice - The path specifies a local device in the format
\\.\rest-of-path or \\?\rest-of-path. This can be used for any device
where the nt and Win32 names are the same. For example mailslots.
RtlPathTypeRootLocalDevice - The path specifies the root of the local
devices in the format \\. or \\?
RtlPathTypeDriveAbsolute - The path specifies a drive letter absolute
path in the form drive:\rest-of-path
RtlPathTypeDriveRelative - The path specifies a drive letter relative
path in the form drive:rest-of-path
RtlPathTypeRooted - The path is rooted relative to the current disk
designator (either Unc disk, or drive). The form is \rest-of-path.
RtlPathTypeRelative - The path is relative (i.e. not absolute or rooted).
--*/
{
RTL_PATH_TYPE ReturnValue;
ASSERT(DosFileName != NULL);
if (IS_PATH_SEPARATOR_U(*DosFileName)) {
if ( IS_PATH_SEPARATOR_U(*(DosFileName+1)) ) {
if ( DosFileName[2] == '.' || DosFileName[2] == '?') {
if ( IS_PATH_SEPARATOR_U(*(DosFileName+3)) ){
ReturnValue = RtlPathTypeLocalDevice;
}
else if ( (*(DosFileName+3)) == UNICODE_NULL ){
ReturnValue = RtlPathTypeRootLocalDevice;
}
else {
ReturnValue = RtlPathTypeUncAbsolute;
}
}
else {
ReturnValue = RtlPathTypeUncAbsolute;
}
}
else {
ReturnValue = RtlPathTypeRooted;
}
}
else if ((*DosFileName) && (*(DosFileName+1)==L':')) {
if (IS_PATH_SEPARATOR_U(*(DosFileName+2))) {
ReturnValue = RtlPathTypeDriveAbsolute;
}
else {
ReturnValue = RtlPathTypeDriveRelative;
}
}
else {
ReturnValue = RtlPathTypeRelative;
}
return ReturnValue;
}
RTL_PATH_TYPE
NTAPI
RtlDetermineDosPathNameType_Ustr(
IN PCUNICODE_STRING String
)
/*++
Routine Description:
This function examines the Dos format file name and determines the
type of file name (i.e. UNC, DriveAbsolute, Current Directory
rooted, or Relative.)
Arguments:
DosFileName - Supplies the Dos format file name whose type is to be
determined.
Return Value:
RtlPathTypeUnknown - The path type can not be determined
RtlPathTypeUncAbsolute - The path specifies a Unc absolute path
in the format \\server-name\sharename\rest-of-path
RtlPathTypeLocalDevice - The path specifies a local device in the format
\\.\rest-of-path or \\?\rest-of-path. This can be used for any device
where the nt and Win32 names are the same. For example mailslots.
RtlPathTypeRootLocalDevice - The path specifies the root of the local
devices in the format \\. or \\?
RtlPathTypeDriveAbsolute - The path specifies a drive letter absolute
path in the form drive:\rest-of-path
RtlPathTypeDriveRelative - The path specifies a drive letter relative
path in the form drive:rest-of-path
RtlPathTypeRooted - The path is rooted relative to the current disk
designator (either Unc disk, or drive). The form is \rest-of-path.
RtlPathTypeRelative - The path is relative (i.e. not absolute or rooted).
--*/
{
RTL_PATH_TYPE ReturnValue;
const PCWSTR DosFileName = String->Buffer;
#define ENOUGH_CHARS(_cch) (String->Length >= ((_cch) * sizeof(WCHAR)))
if ( ENOUGH_CHARS(1) && IS_PATH_SEPARATOR_U(*DosFileName) ) {
if ( ENOUGH_CHARS(2) && IS_PATH_SEPARATOR_U(*(DosFileName+1)) ) {
if ( ENOUGH_CHARS(3) && (DosFileName[2] == '.' ||
DosFileName[2] == '?') ) {
if ( ENOUGH_CHARS(4) && IS_PATH_SEPARATOR_U(*(DosFileName+3)) ){
// "\\.\" or "\\?\"
ReturnValue = RtlPathTypeLocalDevice;
}
else if ( String->Length == (3 * sizeof(WCHAR)) ){
// "\\." or \\?"
ReturnValue = RtlPathTypeRootLocalDevice;
}
else {
// "\\.x" or "\\?x"
ReturnValue = RtlPathTypeUncAbsolute;
}
}
else {
// "\\x"
ReturnValue = RtlPathTypeUncAbsolute;
}
}
else {
// "\x"
ReturnValue = RtlPathTypeRooted;
}
}
//
// the "*DosFileName" is left over from the PCWSTR version
// Win32 and DOS don't allow embedded nuls and much code limits
// drive letters to strictly 7bit a-zA-Z so it's ok.
//
else if (ENOUGH_CHARS(2) && *DosFileName && *(DosFileName+1)==L':') {
if (ENOUGH_CHARS(3) && IS_PATH_SEPARATOR_U(*(DosFileName+2))) {
// "x:\"
ReturnValue = RtlPathTypeDriveAbsolute;
}
else {
// "c:x"
ReturnValue = RtlPathTypeDriveRelative;
}
}
else {
// "x", first char is not a slash / second char is not colon
ReturnValue = RtlPathTypeRelative;
}
return ReturnValue;
#undef ENOUGH_CHARS
}
NTSTATUS
NTAPI
RtlpDetermineDosPathNameType4(
IN ULONG InFlags,
IN PCUNICODE_STRING DosPath,
OUT RTL_PATH_TYPE* OutType,
OUT ULONG* OutFlags
)
{
NTSTATUS Status = STATUS_SUCCESS;
RTL_PATH_TYPE PathType = 0;
BOOLEAN Win32Nt = FALSE;
BOOLEAN Win32NtUncAbsolute = FALSE;
BOOLEAN Win32NtDriveAbsolute = FALSE;
BOOLEAN IncompleteRoot = FALSE;
RTL_PATH_TYPE PathTypeAfterWin32Nt = 0;
if (OutType != NULL
) {
*OutType = RtlPathTypeUnknown;
}
if (OutFlags != NULL
) {
*OutFlags = 0;
}
if (
!RTL_SOFT_VERIFY(DosPath != NULL)
|| !RTL_SOFT_VERIFY(OutType != NULL)
|| !RTL_SOFT_VERIFY(OutFlags != NULL)
|| !RTL_SOFT_VERIFY(
(InFlags & ~(RTL_DETERMINE_DOS_PATH_NAME_TYPE_IN_FLAG_OLD | RTL_DETERMINE_DOS_PATH_NAME_TYPE_IN_FLAG_STRICT_WIN32NT))
== 0)
) {
Status = STATUS_INVALID_PARAMETER;
goto Exit;
}
PathType = RtlDetermineDosPathNameType_Ustr(DosPath);
*OutType = PathType;
if (InFlags & RTL_DETERMINE_DOS_PATH_NAME_TYPE_IN_FLAG_OLD)
goto Exit;
if (DosPath->Length == sizeof(L"\\\\") - sizeof(DosPath->Buffer[0])
) {
IncompleteRoot = TRUE;
}
else if (RtlEqualUnicodeString(&RtlpWin32NtRoot, DosPath, TRUE)
) {
IncompleteRoot = TRUE;
Win32Nt = TRUE;
}
else if (RtlEqualUnicodeString(&RtlpWin32NtRootSlash, DosPath, TRUE)
) {
IncompleteRoot = TRUE;
Win32Nt = TRUE;
}
else if (RtlPrefixUnicodeString(&RtlpWin32NtRootSlash, DosPath, TRUE)
) {
Win32Nt = TRUE;
}
if (Win32Nt) {
if (RtlEqualUnicodeString(&RtlpWin32NtUncRoot, DosPath, TRUE)
) {
IncompleteRoot = TRUE;
Win32NtUncAbsolute = TRUE;
}
else if (RtlEqualUnicodeString(&RtlpWin32NtUncRootSlash, DosPath, TRUE)
) {
IncompleteRoot = TRUE;
Win32NtUncAbsolute = TRUE;
}
else if (RtlPrefixUnicodeString(&RtlpWin32NtUncRootSlash, DosPath, TRUE)
) {
Win32NtUncAbsolute = TRUE;
}
if (Win32NtUncAbsolute
) {
Win32NtDriveAbsolute = FALSE;
} else if (!IncompleteRoot) {
const RTL_STRING_LENGTH_TYPE i = RtlpWin32NtRootSlash.Length;
UNICODE_STRING PathAfterWin32Nt = *DosPath;
PathAfterWin32Nt.Buffer += i / sizeof(PathAfterWin32Nt.Buffer[0]);
PathAfterWin32Nt.Length = PathAfterWin32Nt.Length - i;
PathAfterWin32Nt.MaximumLength = PathAfterWin32Nt.MaximumLength - i;
PathTypeAfterWin32Nt = RtlDetermineDosPathNameType_Ustr(&PathAfterWin32Nt);
if (PathTypeAfterWin32Nt == RtlPathTypeDriveAbsolute) {
Win32NtDriveAbsolute = TRUE;
}
else {
Win32NtDriveAbsolute = FALSE;
}
if (InFlags & RTL_DETERMINE_DOS_PATH_NAME_TYPE_IN_FLAG_STRICT_WIN32NT
) {
if (!RTL_SOFT_VERIFY(Win32NtDriveAbsolute
)) {
*OutFlags |= RTLP_DETERMINE_DOS_PATH_NAME_TYPE_OUT_FLAG_INVALID;
// we still succeed the function call
}
}
}
}
ASSERT(RTLP_IMPLIES(Win32NtDriveAbsolute, Win32Nt));
ASSERT(RTLP_IMPLIES(Win32NtUncAbsolute, Win32Nt));
ASSERT(!(Win32NtUncAbsolute && Win32NtDriveAbsolute));
if (IncompleteRoot)
*OutFlags |= RTLP_DETERMINE_DOS_PATH_NAME_TYPE_OUT_FLAG_INCOMPLETE_ROOT;
if (Win32Nt)
*OutFlags |= RTLP_DETERMINE_DOS_PATH_NAME_TYPE_OUT_FLAG_WIN32NT;
if (Win32NtUncAbsolute)
*OutFlags |= RTLP_DETERMINE_DOS_PATH_NAME_TYPE_OUT_FLAG_WIN32NT_UNC_ABSOLUTE;
if (Win32NtDriveAbsolute)
*OutFlags |= RTLP_DETERMINE_DOS_PATH_NAME_TYPE_OUT_FLAG_WIN32NT_DRIVE_ABSOLUTE;
Status = STATUS_SUCCESS;
Exit:
return Status;
}
ULONG
RtlIsDosDeviceName_Ustr(
IN PCUNICODE_STRING DosFileName
)
/*++
Routine Description:
This function examines the Dos format file name and determines if it
is a Dos device name (e.g. LPT1, etc.). Valid Dos device names are:
LPTn
COMn
PRN
AUX
NUL
CON
when n is a digit. Trailing colon is ignored if present.
Arguments:
DosFileName - Supplies the Dos format file name that is to be examined.
Return Value:
0 - Specified Dos file name is not the name of a Dos device.
> 0 - Specified Dos file name is the name of a Dos device and the
return value is a ULONG where the high order 16 bits is the
offset in the input buffer where the dos device name beings
and the low order 16 bits is the length of the device name
the length of the name (excluding any optional
trailing colon).
--*/
{
UNICODE_STRING UnicodeString;
USHORT NumberOfCharacters, OriginalLength;
ULONG ReturnLength;
ULONG ReturnOffset;
LPWSTR p;
USHORT ColonBias;
RTL_PATH_TYPE PathType;
WCHAR wch;
ColonBias = 0;
//
// NOTICE-2002/03/18-ELi
// DosFileName->Buffer might not be a null-terminated string
// RtlSetCurrentDirectory_U can called this function with such a string
//
PathType = RtlDetermineDosPathNameType_Ustr(DosFileName);
switch ( PathType ) {
case RtlPathTypeLocalDevice:
//
// For Unc Absolute, Check for \\.\CON
// since this really is not a device
//
if ( RtlEqualUnicodeString(DosFileName,&RtlpDosSlashCONDevice,TRUE) ) {
return 0x00080006;
}
//
// FALLTHRU
//
case RtlPathTypeUncAbsolute:
case RtlPathTypeUnknown:
return 0;
}
UnicodeString = *DosFileName;
OriginalLength = UnicodeString.Length;
NumberOfCharacters = OriginalLength >> 1;
if (NumberOfCharacters && UnicodeString.Buffer[NumberOfCharacters-1] == L':') {
UnicodeString.Length -= sizeof(WCHAR);
NumberOfCharacters--;
ColonBias = 1;
}
//
// The above strip the trailing colon logic could have left us with 0
// for NumberOfCharacters, so that needs to be tested
//
if ( NumberOfCharacters == 0 ) {
return 0;
}
wch = UnicodeString.Buffer[NumberOfCharacters-1];
while ( NumberOfCharacters && (wch == L'.' || wch == L' ') ) {
UnicodeString.Length -= sizeof(WCHAR);
NumberOfCharacters--;
ColonBias++;
if ( NumberOfCharacters > 0 ) {
wch = UnicodeString.Buffer[NumberOfCharacters-1];
}
}
ReturnLength = NumberOfCharacters << 1;
//
// Walk backwards through the string finding the
// first slash or the beginning of the string. We also stop
// at the drive: if it is present.
//
ReturnOffset = 0;
if ( NumberOfCharacters ) {
p = UnicodeString.Buffer + NumberOfCharacters-1;
while ( p >= UnicodeString.Buffer ) {
if ( *p == L'\\' || *p == L'/'
|| (*p == L':' && p == UnicodeString.Buffer + 1)) {
p++;
//
// NOTICE-2002/02/12-ELi
// if p points to the end of the string, then cannot access p++
// because string may not be NULL-terminated
// if at the end of the string, don't have any more chars in
// the name to check so just return 0
//
if ( p >= (UnicodeString.Buffer +
(OriginalLength / sizeof(WCHAR)))) {
return 0;
}
//
// Get the first char of the file name and convert it to
// lower case. This will be safe since we will be comparing
// it to only lower-case ASCII.
//
wch = (*p) | 0x20;
//
// check to see if we possibly have a hit on
// lpt, prn, con, com, aux, or nul
//
if ( !(wch == L'l' || wch == L'c' || wch == L'p' || wch == L'a'
|| wch == L'n')
) {
return 0;
}
ReturnOffset = (ULONG)((PSZ)p - (PSZ)UnicodeString.Buffer);
//
// NOTICE-2002/03/18-ELi
// assuming that at this point
// UnicodeString.Buffer < p < (UnicodeString.Buffer +
// OriginalLength / sizeof(WCHAR))
// Therefore the USHORT cast and subtractions are ok
//
UnicodeString.Length = OriginalLength - (USHORT)((PCHAR)p -
(PCHAR)UnicodeString.Buffer);
UnicodeString.Buffer = p;
NumberOfCharacters = UnicodeString.Length >> 1;
NumberOfCharacters = NumberOfCharacters - ColonBias;
ReturnLength = NumberOfCharacters << 1;
UnicodeString.Length -= ColonBias*sizeof(WCHAR);
break;
}
p--;
}
wch = UnicodeString.Buffer[0] | 0x20;
//
// check to see if we possibly have a hit on
// lpt, prn, con, com, aux, or nul
//
if ( !( wch == L'l' || wch == L'c' || wch == L'p' || wch == L'a'
|| wch == L'n' ) ) {
return 0;
}
}
//
// Now we need to see if we are dealing with a device name that has
// an extension or a stream name. If so, we need to limit the search to the
// file portion only
//
p = UnicodeString.Buffer;
while (p < UnicodeString.Buffer + NumberOfCharacters && *p != L'.' && *p != L':') {
p++;
}
//
// p either points past end of string or to a dot or :. We back up over
// trailing spaces
//
while (p > UnicodeString.Buffer && p[-1] == L' ') {
p--;
}
//
// p either points to the beginning of the string or p[-1] is
// the first non-space char found above.
//
NumberOfCharacters = (USHORT)(p - UnicodeString.Buffer);
UnicodeString.Length = NumberOfCharacters * sizeof( WCHAR );
if ( NumberOfCharacters == 4 && iswdigit(UnicodeString.Buffer[3] ) ) {
if ( (WCHAR)UnicodeString.Buffer[3] == L'0') {
return 0;
} else {
UnicodeString.Length -= sizeof(WCHAR);
if ( RtlEqualUnicodeString(&UnicodeString,&RtlpDosLPTDevice,TRUE) ||
RtlEqualUnicodeString(&UnicodeString,&RtlpDosCOMDevice,TRUE) ) {
ReturnLength = NumberOfCharacters << 1;
} else {
return 0;
}
}
} else if ( NumberOfCharacters != 3 ) {
return 0;
} else if ( RtlEqualUnicodeString(&UnicodeString,&RtlpDosPRNDevice,TRUE) ) {
ReturnLength = NumberOfCharacters << 1;
} else if ( RtlEqualUnicodeString(&UnicodeString,&RtlpDosAUXDevice,TRUE) ) {
ReturnLength = NumberOfCharacters << 1;
} else if ( RtlEqualUnicodeString(&UnicodeString,&RtlpDosNULDevice,TRUE) ) {
ReturnLength = NumberOfCharacters << 1;
} else if ( RtlEqualUnicodeString(&UnicodeString,&RtlpDosCONDevice,TRUE) ) {
ReturnLength = NumberOfCharacters << 1;
} else {
return 0;
}
return ReturnLength | (ReturnOffset << 16);
}
ULONG
RtlIsDosDeviceName_U(
IN PCWSTR DosFileName
)
{
UNICODE_STRING UnicodeString;
NTSTATUS Status;
Status = RtlInitUnicodeStringEx(&UnicodeString,DosFileName);
if( !NT_SUCCESS( Status ) ) {
return 0;
}
return RtlIsDosDeviceName_Ustr(&UnicodeString);
}
NTSTATUS
RtlpCheckDeviceName(
PCUNICODE_STRING DevName,
ULONG DeviceNameOffset,
BOOLEAN* NameInvalid
)
{
PWSTR DevPath;
NTSTATUS Status = STATUS_SUCCESS;
HANDLE Heap;
DevPath = RtlAllocateHeap(Heap = RtlProcessHeap(), 0,DevName->Length);
if (!DevPath) {
*NameInvalid = FALSE;
Status = STATUS_NO_MEMORY;
goto Exit;
}
*NameInvalid = TRUE;
try {
RtlCopyMemory(DevPath,DevName->Buffer,DevName->Length);
//
// NOTICE-2002/02/12-ELi
// assuming that DeviceNameOffset <= (DevName->Length - 6) because
// this function is called only if DevName is a DosDevice name, which
// should be at least 3 chars long
//
DevPath[DeviceNameOffset>>1]=L'.';
DevPath[(DeviceNameOffset>>1)+1]=UNICODE_NULL;
if (RtlDoesFileExists_U(DevPath) ) {
*NameInvalid = FALSE;
}
else {
*NameInvalid = TRUE;
}
}
finally {
RtlFreeHeap(Heap, 0, DevPath);
}
Status = STATUS_SUCCESS;
Exit:
return Status;
}
//
// We keep an open handle to the current directory on the current drive in order
// to make relative opens faster.
//
// However, this current directory can become invalid under two circumstances:
//
// 1. The current drive is removable media. The user may arbitrarily switch
// media without our knowledge. At this point, whatever information the
// filesystem has about the media is now out of date.
// 2. The volume is dismounted by explicit system/user action.
//
// We can ping (via FSCTL_IS_VOLUME_MOUNTED) the volume each time we want to
// test to see if the current directory is still valid. While a "cheap" call
// we will end up doing it a lot. We can reduce this frequency by only checking
// when the media is known to be removable or when a dismount is known to have
// occurred. The Io system exports in USER_SHARED_DATA a count of dismounts
// since boot. We capture this and use it to decide if a dismount was performed.
//
ULONG RtlpSavedDismountCount = (ULONG)-1;
VOID
RtlpValidateCurrentDirectory(
PCURDIR CurDir
)
/*++
Routine Description:
This function is used to validate the current directory for the process.
The current directory can change in several ways, first, by replacing
the media with one that has a different directory structure. Second
by performing a force-dismount.
Arguments:
CurDir - Current directory structure for process
Return Value:
None.
--*/
{
NTSTATUS FsCtlStatus;
IO_STATUS_BLOCK IoStatusBlock;
WCHAR TrimmedPath[4];
UNICODE_STRING str;
if (((ULONG_PTR)CurDir->Handle & 1) == 0
&& USER_SHARED_DATA->DismountCount == RtlpSavedDismountCount) {
return;
}
//
// Never been set yet.
//
if (CurDir->Handle == NULL) {
return;
}
//
// Call Nt to see if the volume that
// contains the directory is still mounted.
// If it is, then continue. Otherwise, trim
// current directory to the root.
//
//
// We're updated as far as possible with the current dismount count
//
RtlpSavedDismountCount = USER_SHARED_DATA->DismountCount;
FsCtlStatus = NtFsControlFile(
CurDir->Handle,
NULL,
NULL,
NULL,
&IoStatusBlock,
FSCTL_IS_VOLUME_MOUNTED,
NULL,
0,
NULL,
0
);
if ( FsCtlStatus == STATUS_WRONG_VOLUME || FsCtlStatus == STATUS_VOLUME_DISMOUNTED) {
//
// Try to get back to where we were, failing that reset current directory
// to the root of the current drive
//
RtlpDereferenceCurDirRef(RtlpCurDirRef);
RtlpCurDirRef = NULL;
CurDir->Handle = NULL;
FsCtlStatus = RtlSetCurrentDirectory_U(&CurDir->DosPath);
if ( !NT_SUCCESS(FsCtlStatus) ) {
TrimmedPath[0] = CurDir->DosPath.Buffer[0];
TrimmedPath[1] = CurDir->DosPath.Buffer[1];
TrimmedPath[2] = CurDir->DosPath.Buffer[2];
TrimmedPath[3] = UNICODE_NULL;
RtlpResetDriveEnvironment( TrimmedPath[0] );
RtlInitUnicodeString( &str, TrimmedPath );
//
// This can still fail if the volume was hard dismounted. We tried.
// Ah well.
//
(VOID) RtlSetCurrentDirectory_U( &str );
}
}
}
ULONG
RtlGetFullPathName_Ustr(
PCUNICODE_STRING FileName,
ULONG nBufferLength,
PWSTR lpBuffer,
PWSTR *lpFilePart OPTIONAL,
PBOOLEAN NameInvalid,
RTL_PATH_TYPE *InputPathType
)
/*++
Routine Description:
This function is used to return a fully qualified pathname
corresponding to the specified unicode filename. It does this by
merging the current drive and directory together with the specified
file name. In addition to this, it calculates the address of the
file name portion of the fully qualified pathname.
Arguments:
lpFileName - Supplies the unicode file name of the file whose fully
qualified pathname is to be returned.
nBufferLength - Supplies the length in bytes of the buffer that is
to receive the fully qualified path.
lpBuffer - Returns the fully qualified pathname corresponding to the
specified file.
lpFilePart - Optional parameter that if specified, returns the
address of the last component of the fully qualified pathname.
Return Value:
The return value is the length of the string copied to lpBuffer, not
including the terminating unicode null character. If the return
value is greater than nBufferLength, the return value is the size of
the buffer required to hold the pathname. The return value is zero
if the function failed.
--*/
{
ULONG DeviceNameLength;
ULONG DeviceNameOffset;
ULONG PrefixSourceLength;
LONG PathNameLength;
UCHAR CurDrive, NewDrive;
WCHAR EnvVarNameBuffer[4];
UNICODE_STRING EnvVarName;
PWSTR Source,Dest;
UNICODE_STRING Prefix;
PCURDIR CurDir;
ULONG MaximumLength;
UNICODE_STRING FullPath;
ULONG BackupIndex;
RTL_PATH_TYPE PathType;
NTSTATUS Status;
BOOLEAN StripTrailingSlash;
UNICODE_STRING UnicodeString;
ULONG NumberOfCharacters;
PWSTR lpFileName;
WCHAR wch;
ULONG i,j;
ULONG PathLength;
if ( ARGUMENT_PRESENT(NameInvalid) ) {
*NameInvalid = FALSE;
}
if ( nBufferLength > MAXUSHORT ) {
nBufferLength = MAXUSHORT - 1;
}
*InputPathType = RtlPathTypeUnknown;
UnicodeString = *FileName;
lpFileName = UnicodeString.Buffer;
NumberOfCharacters = UnicodeString.Length >> 1;
PathNameLength = UnicodeString.Length;
if ( PathNameLength == 0 || UnicodeString.Buffer[0] == UNICODE_NULL ) {
return 0;
}
//
// trim trailing spaces to check for a null name
//
DeviceNameLength = PathNameLength;
wch = UnicodeString.Buffer[(DeviceNameLength>>1) - 1];
while ( DeviceNameLength && wch == L' ' ) {
DeviceNameLength -= sizeof(WCHAR);
if ( DeviceNameLength ) {
wch = UnicodeString.Buffer[(DeviceNameLength>>1) - 1];
}
}
if ( !DeviceNameLength ) {
return 0;
}
if ( lpFileName[NumberOfCharacters-1] == L'\\' || lpFileName[NumberOfCharacters-1] == L'/' ) {
StripTrailingSlash = FALSE;
}
else {
StripTrailingSlash = TRUE;
}
//
// If pass Dos file name is a Dos device name, then turn it into
// \\.\devicename and return its length.
//
DeviceNameLength = RtlIsDosDeviceName_Ustr (&UnicodeString);
if ( DeviceNameLength ) {
if ( ARGUMENT_PRESENT( lpFilePart ) ) {
*lpFilePart = NULL;
}
DeviceNameOffset = DeviceNameLength >> 16;
DeviceNameLength &= 0x0000ffff;
if ( ARGUMENT_PRESENT(NameInvalid) && DeviceNameOffset ) {
Status = RtlpCheckDeviceName(&UnicodeString, DeviceNameOffset,
NameInvalid);
if ((!NT_SUCCESS(Status)) || (*NameInvalid)) {
return 0;
}
}
PathNameLength = DeviceNameLength + RtlpSlashSlashDot.Length;
if ( PathNameLength < (LONG)nBufferLength ) {
RtlCopyMemory (lpBuffer,
RtlpSlashSlashDot.Buffer,
RtlpSlashSlashDot.Length);
RtlMoveMemory ((PVOID)((PUCHAR)lpBuffer+RtlpSlashSlashDot.Length),
(PSZ)lpFileName+DeviceNameOffset,
DeviceNameLength);
RtlZeroMemory ((PVOID)((PUCHAR)lpBuffer+RtlpSlashSlashDot.Length+DeviceNameLength),
sizeof(UNICODE_NULL));
return PathNameLength;
} else {
PathNameLength += sizeof(UNICODE_NULL);
if (PathNameLength > MAXUSHORT) {
PathNameLength = 0;
}
return PathNameLength;
}
}
//
// Setup output string that points to callers buffer.
//
FullPath.MaximumLength = (USHORT)nBufferLength;
FullPath.Length = 0;
FullPath.Buffer = lpBuffer;
RtlZeroMemory(lpBuffer,nBufferLength);
//
// Get a pointer to the current directory structure.
//
CurDir = &NtCurrentPeb()->ProcessParameters->CurrentDirectory;
//
// Determine the type of Dos Path Name specified.
//
*InputPathType = PathType = RtlDetermineDosPathNameType_Ustr( &UnicodeString );
//
// Determine the prefix and backup index.
//
// Input Prefix Backup Index
//
// \\ -> \\, end of \\server\share
// \\.\ -> \\.\, 4
// \\. -> \\. 3 (\\.)
// \ -> Drive: from CurDir.DosPath 3 (Drive:\)
// d: -> Drive:\curdir from environment 3 (Drive:\)
// d:\ -> no prefix 3 (Drive:\)
// any -> CurDir.DosPath 3 (Drive:\)
//
//
// No prefixes yet.
//
Source = lpFileName;
PrefixSourceLength = 0;
Prefix.Length = 0;
Prefix.MaximumLength = 0;
Prefix.Buffer = NULL;
RtlAcquirePebLock();
PathLength = 0;
try {
switch (PathType) {
case RtlPathTypeUncAbsolute : {
PWSTR UncPathPointer;
ULONG NumberOfPathSeparators;
//
// We want to scan the supplied path to determine where
// the "share" ends, and set BackupIndex to that point.
//
UncPathPointer = lpFileName + 2;
NumberOfPathSeparators = 0;
i = 2 * sizeof(WCHAR);
while (i < UnicodeString.Length) {
if (IS_PATH_SEPARATOR_U(*UncPathPointer)) {
NumberOfPathSeparators++;
if (NumberOfPathSeparators == 2) {
break;
}
}
i += sizeof(WCHAR);
UncPathPointer++;
}
BackupIndex = (ULONG)(UncPathPointer - lpFileName);
//
// Unc name. prefix = \\server\share
//
PrefixSourceLength = BackupIndex << 1;
Source += BackupIndex;
//
// There is no prefix to place into the buffer.
// The entire path is in Source
//
}
break;
case RtlPathTypeLocalDevice :
//
// Local device name. prefix = "\\.\"
//
PrefixSourceLength = RtlpSlashSlashDot.Length;
BackupIndex = 4;
Source += BackupIndex;
//
// There is no prefix to place into the buffer.
// The entire path is in Source
//
break;
case RtlPathTypeRootLocalDevice :
//
// Local Device root. prefix = "\\.\"
//
Prefix = RtlpSlashSlashDot;
Prefix.Length = (USHORT)(Prefix.Length - (USHORT)(2*sizeof(UNICODE_NULL)));
PrefixSourceLength = Prefix.Length + sizeof(UNICODE_NULL);
BackupIndex = 3;
Source += BackupIndex;
PathNameLength -= BackupIndex * sizeof( WCHAR );
break;
case RtlPathTypeDriveAbsolute :
CurDrive = (UCHAR)RtlUpcaseUnicodeChar( CurDir->DosPath.Buffer[0] );
NewDrive = (UCHAR)RtlUpcaseUnicodeChar( lpFileName[0] );
if ( CurDrive == NewDrive ) {
RtlpValidateCurrentDirectory( CurDir );
}
//
// Dos drive absolute name
//
BackupIndex = 3;
break;
case RtlPathTypeDriveRelative :
//
// Dos drive relative name
//
CurDrive = (UCHAR)RtlUpcaseUnicodeChar( CurDir->DosPath.Buffer[0] );
NewDrive = (UCHAR)RtlUpcaseUnicodeChar( lpFileName[0] );
if ( CurDrive == NewDrive ) {
RtlpValidateCurrentDirectory( CurDir );
Prefix = *(PUNICODE_STRING)&CurDir->DosPath;
} else {
RtlpCheckRelativeDrive((WCHAR)NewDrive);
EnvVarNameBuffer[0] = L'=';
EnvVarNameBuffer[1] = (WCHAR)NewDrive;
EnvVarNameBuffer[2] = L':';
EnvVarNameBuffer[3] = UNICODE_NULL;
RtlInitUnicodeString(&EnvVarName,EnvVarNameBuffer);
Prefix = FullPath;
Status = RtlQueryEnvironmentVariable_U( NULL,
&EnvVarName,
&Prefix
);
if ( !NT_SUCCESS( Status ) ) {
if (Status == STATUS_BUFFER_TOO_SMALL) {
PathNameLength = (ULONG)(Prefix.Length) + PathNameLength + 2;
if (PathNameLength > MAXUSHORT) {
PathNameLength = 0;
}
PathLength =PathNameLength;
leave;
} else {
//
// Otherwise default to root directory of drive
//
Status = STATUS_SUCCESS;
EnvVarNameBuffer[0] = (WCHAR)NewDrive;
EnvVarNameBuffer[1] = L':';
EnvVarNameBuffer[2] = L'\\';
EnvVarNameBuffer[3] = UNICODE_NULL;
RtlInitUnicodeString(&Prefix,EnvVarNameBuffer);
}
} else {
ULONG LastChar;
//
// Determine
// if a backslash needs to be added
//
LastChar = Prefix.Length >> 1;
if (LastChar > 3) {
Prefix.Buffer[ LastChar ] = L'\\';
Prefix.Length += sizeof(UNICODE_NULL);
}
}
}
BackupIndex = 3;
Source += 2;
PathNameLength -= 2 * sizeof( WCHAR );
break;
case RtlPathTypeRooted :
BackupIndex = RtlpComputeBackupIndex(CurDir);
if ( BackupIndex != 3 ) {
Prefix = CurDir->DosPath;
Prefix.Length = (USHORT)(BackupIndex << 1);
}
else {
//
// Rooted name. Prefix is drive portion of current directory
//
Prefix = CurDir->DosPath;
Prefix.Length = 2*sizeof(UNICODE_NULL);
}
break;
case RtlPathTypeRelative :
RtlpValidateCurrentDirectory( CurDir );
//
// Current drive:directory relative name
//
Prefix = CurDir->DosPath;
BackupIndex = RtlpComputeBackupIndex(CurDir);
break;
default:
PathLength = 0;
leave;
}
//
// Maximum length required is the length of the prefix plus
// the length of the specified pathname. If the callers buffer
// is not at least this large, then return an error.
//
MaximumLength = PathNameLength + Prefix.Length;
if ( (MaximumLength + sizeof(WCHAR)) > nBufferLength ) {
if ( (NumberOfCharacters > 1) ||
(*lpFileName != L'.') ) {
MaximumLength += sizeof(UNICODE_NULL);
if (MaximumLength > MAXUSHORT) {
MaximumLength = 0;
}
PathLength = MaximumLength;
leave;
} else {
//
// If we are expanding curdir, then remember the trailing '\'
//
if ( NumberOfCharacters == 1 && *lpFileName == L'.' ) {
//
// We are expanding .
//
if ( Prefix.Length == 6 ) {
if ( nBufferLength <= Prefix.Length ) {
PathLength = (ULONG)(Prefix.Length+(USHORT)sizeof(UNICODE_NULL));
leave;
}
} else {
if ( nBufferLength < Prefix.Length ) {
PathLength = (ULONG)Prefix.Length;
leave;
} else {
for(i=0,j=0;i<Prefix.Length;i+=sizeof(WCHAR),j++){
if ( Prefix.Buffer[j] == L'\\' ||
Prefix.Buffer[j] == L'/' ) {
FullPath.Buffer[j] = L'\\';
} else {
FullPath.Buffer[j] = Prefix.Buffer[j];
}
}
FullPath.Length = Prefix.Length-(USHORT)sizeof(L'\\');
goto skipit;
}
}
} else {
if (MaximumLength > MAXUSHORT) {
MaximumLength = 0;
}
PathLength = MaximumLength;
leave;
}
}
}
if (PrefixSourceLength || Prefix.Buffer != FullPath.Buffer) {
//
// Copy the prefix from the source string.
//
//RtlMoveMemory(FullPath.Buffer,lpFileName,PrefixSourceLength);
for(i=0,j=0;i<PrefixSourceLength;i+=sizeof(WCHAR),j++){
if ( lpFileName[j] == L'\\' ||
lpFileName[j] == L'/' ) {
FullPath.Buffer[j] = L'\\';
} else {
FullPath.Buffer[j] = lpFileName[j];
}
}
FullPath.Length = (USHORT)PrefixSourceLength;
//
// Append any additional prefix
//
for(i=0,j=0;i<Prefix.Length;i+=sizeof(WCHAR),j++) {
if ( Prefix.Buffer[j] == L'\\' ||
Prefix.Buffer[j] == L'/' ) {
FullPath.Buffer[j+(FullPath.Length>>1)] = L'\\';
} else {
FullPath.Buffer[j+(FullPath.Length>>1)] = Prefix.Buffer[j];
}
}
FullPath.Length = FullPath.Length + Prefix.Length;
} else {
FullPath.Length = Prefix.Length;
}
skipit:
Dest = (PWSTR)((PUCHAR)FullPath.Buffer + FullPath.Length);
*Dest = UNICODE_NULL;
i = (ULONG)((PCHAR)Source - (PCHAR)lpFileName);
while ( i < UnicodeString.Length ) {
i += sizeof(WCHAR);
switch ( *Source ) {
case L'\\' :
case L'/' :
//
// collapse multiple "\" characters. If the previous character was
// a path character, skip it.
//
if ( *(Dest-1) != L'\\' ) {
*Dest++ = L'\\';
}
Source++;
break;
case '.' :
//
// Ignoring dot in a leading //./ has already been taken
// care of by advancing Source above.
//
// Eat single dots as in /./ by simply skipping them
//
// Double dots back up one level as in /../
//
// Any other . is just a filename character
//
j = UnicodeString.Length - i + sizeof(WCHAR);
if ( IS_DOT_USTR(Source, j) ) {
Source++;
if ( (i < UnicodeString.Length) &&
IS_PATH_SEPARATOR_U(*Source) ) {
Source++;
i += sizeof(WCHAR);
}
break;
} else if ( IS_DOT_DOT_USTR(Source, j) ) {
//
// backup destination string looking for a '\'
//
while (*Dest != L'\\') {
*Dest = UNICODE_NULL;
Dest--;
}
//
// backup to previous component..
// \a\b\c\.. to \a\b
//
do {
//
// If we bump into root prefix, then
// stay at root
//
if ( Dest == FullPath.Buffer + (BackupIndex-1) ) {
break;
}
*Dest = UNICODE_NULL;
Dest--;
} while (*Dest != L'\\');
if ( Dest == FullPath.Buffer + (BackupIndex-1) ) {
Dest++;
}
//
// Advance source past ..
//
Source += 2;
i += sizeof(WCHAR);
break;
}
//
// Not a single dot or double-dot. The dot we found begins
// a file name so we treat this as a normal file name
//
// FALLTHRU
//
default:
//
// IIS cannot handle the change below in the .NET timeframe. Back it out and
// go back to the previous version
//
#if FALSE
//
// Copy the filename. Note that no remaining chars and /,\
// will stop the copy.
//
//
// Here is what should be processed:
//
// Names that are all dots and spaces.
// Names that have something other than dot and space with trailing spaces and dots removed.
//
// Simple state machine:
//
// Start: DOT => DotAndSpace
// SPACE => DotAndSpace
// OTHER => Component
//
// DotAndSpace: DOT => DotAndSpace
// SPACE => DotAndSpace
// OTHER => Component
//
// Component: DOT => ComponentTrailingDotSpace (Save location for truncation)
// SPACE => ComponentTrailingDotSpace (Save location for truncation)
// OTHER => Component
//
// ComponentTrailingDotSpace: DOT => ComponentTrailingDotSpace
// SPACE => ComponentTrailingDotSpace
// OTHER => Component (Reset location for truncation)
//
{
typedef enum {
StateStart,
StateDotAndSpace,
StateComponent,
StateComponentTrailingDotSpace
} COPYCOMPONENTSTATE;
COPYCOMPONENTSTATE State = StateStart;
PWCHAR TruncationPoint = NULL;
i -= sizeof(WCHAR);
j = UnicodeString.Length - i;
while ( !IS_END_OF_COMPONENT_USTR(Source, j) ) {
WCHAR c = *Source;
*Dest++ = *Source++;
j -= sizeof( WCHAR );
i += sizeof( WCHAR );
switch (State) {
case StateStart:
State = (c == L'.') ? StateDotAndSpace :
(c == L' ') ? StateDotAndSpace :
StateComponent;
break;
case StateDotAndSpace:
State = (c == L'.') ? StateDotAndSpace :
(c == L' ') ? StateDotAndSpace :
StateComponent;
break;
case StateComponent:
State = (c == L'.') ? (TruncationPoint = Dest-1, StateComponentTrailingDotSpace) :
(c == L' ') ? (TruncationPoint = Dest-1, StateComponentTrailingDotSpace) :
StateComponent;
break;
case StateComponentTrailingDotSpace:
State = (c == L'.') ? StateComponentTrailingDotSpace :
(c == L' ') ? StateComponentTrailingDotSpace :
(TruncationPoint = NULL, StateComponent);
break;
default:
// DbgPrint( "RtlGetFullPathName_Ustr: Invalid State %d\n", State );
// DbgBreakPoint( );
PathLength = 0;
leave;
}
}
if (TruncationPoint != NULL) {
// DbgPrint( "RtlGetFullPathName_Ustr: Truncating %Z\n", FileName );
Dest = TruncationPoint;
}
}
#else // Put back in old code for IIS
//
// Copy the filename. Note that
// null and /,\ will stop the copy. If any
// charcter other than null or /,\ is encountered,
// then the pathname is invalid.
//
//
// Copy up until NULL or path separator
//
i -= sizeof(WCHAR);
j = UnicodeString.Length - i;
while ( !IS_END_OF_COMPONENT_USTR(Source, j) ) {
j -= sizeof( WCHAR );
i += sizeof( WCHAR );
*Dest++ = *Source++;
}
//
// Once copied, we should do some processing for compatibility with Win9x.
// Win9x strips all trailing spaces and dots from the name.
// Nt4/Win2K strips only the last dot if its followed by a path separator.
//
// Ideally, we'd do something reasonable like stripping all trailing spaces
// and dots (like Win9X). However, IIS has a security model that's based
// on NAMES and not on objects. This means that IIS needs to process names
// in the same way that we do. They should use GetFullPathName and be done
// with it. In any event, DON'T CHANGE THE CANONICALIZATION OF THE NAME
// HERE WITHOUTH FIRST GOING THROUGH IIS.
//
// We do EXACTLY what Nt4 did: if there's a trailing dot AND we're at a path
// seperator, remove the trailing dot.
//
if (IS_PATH_SEPARATOR_U( *Source ) && Dest[-1] == L'.') {
Dest--;
}
#endif
}
}
*Dest = UNICODE_NULL;
if ( StripTrailingSlash ) {
if ( Dest > (FullPath.Buffer + BackupIndex ) && *(Dest-1) == L'\\' ) {
Dest--;
*Dest = UNICODE_NULL;
}
}
FullPath.Length = (USHORT)(PtrToUlong(Dest) - PtrToUlong(FullPath.Buffer));
//
// More IIS stuff to put back in
//
#if TRUE
//
// Strip trailing spaces and dots.
//
while (Dest > FullPath.Buffer && (Dest[-1] == L' ' || Dest[-1] == L'.')) {
*--Dest = UNICODE_NULL;
FullPath.Length -= sizeof( WCHAR );
}
#endif
if ( ARGUMENT_PRESENT( lpFilePart ) ) {
//
// Locate the file part...
//
Source = Dest-1;
Dest = NULL;
while(Source > FullPath.Buffer ) {
if ( *Source == L'\\' ) {
Dest = Source + 1;
break;
}
Source--;
}
if ( Dest && *Dest ) {
//
// If this is a UNC name, make sure filepart is past the backup index
//
if ( PathType == RtlPathTypeUncAbsolute ) {
if ( Dest < (FullPath.Buffer + BackupIndex ) ) {
*lpFilePart = NULL;
PathLength = (ULONG)FullPath.Length;
leave;
}
}
*lpFilePart = Dest;
} else {
*lpFilePart = NULL;
}
}
PathLength = (ULONG)FullPath.Length;
} finally {
RtlReleasePebLock();
}
return PathLength;
}
NTSTATUS
RtlGetFullPathName_UstrEx(
PCUNICODE_STRING FileName,
PUNICODE_STRING StaticString,
PUNICODE_STRING DynamicString,
PUNICODE_STRING *StringUsed,
SIZE_T *FilePartPrefixCch OPTIONAL,
PBOOLEAN NameInvalid,
RTL_PATH_TYPE *InputPathType,
SIZE_T *BytesRequired OPTIONAL
)
/*++
Routine Description:
See a description of RtlGetFullPathName_Ustr() for a functional
description.
This function provides the same basic behavior as RtlGetFullPathName_Ustr(),
but with easier support for dynamically allocated arbitrary path name
buffers.
One would think that this is the core implementation and the non-Ex()
version would call the ...Ex() version, but that seems risky, and
would only really help the performance of the case where a dynamic allocation
is done.
--*/
{
NTSTATUS Status;
ULONG Length;
PWSTR FilePart = NULL;
UNICODE_STRING TempDynamicString;
USHORT StaticBufferSize;
PWSTR StaticDynamic = NULL;
PWSTR StaticBuffer;
if (StringUsed != NULL) {
*StringUsed = NULL;
}
if (BytesRequired != NULL) {
*BytesRequired = 0;
}
if (FilePartPrefixCch != NULL) {
*FilePartPrefixCch = 0;
}
TempDynamicString.Buffer = NULL;
if ((StaticString != NULL) && (DynamicString != NULL) && (StringUsed == NULL)) {
Status = STATUS_INVALID_PARAMETER;
goto Exit;
}
if (StaticString != NULL) {
StaticBufferSize = StaticString->MaximumLength;
StaticBuffer = StaticString->Buffer;
}
else {
StaticBufferSize = DOS_MAX_PATH_LENGTH * sizeof(WCHAR);
StaticDynamic = (RtlAllocateStringRoutine)( StaticBufferSize );
if (StaticDynamic == NULL) {
return STATUS_NO_MEMORY;
}
StaticBuffer = StaticDynamic;
}
// First try getting into the static string.
Length = RtlGetFullPathName_Ustr(
FileName,
StaticBufferSize,
StaticBuffer,
&FilePart,
NameInvalid,
InputPathType);
if (Length == 0) {
#if DBG
DbgPrint("%s(%d) - RtlGetFullPathName_Ustr() returned 0\n", __FUNCTION__, __LINE__);
#endif // DBG
Status = STATUS_OBJECT_NAME_INVALID;
goto Exit;
}
if ((StaticString != NULL) && (Length < StaticBufferSize)) {
// woohoo it worked.
StaticString->Length = (USHORT) Length;
if (FilePartPrefixCch != NULL) {
*FilePartPrefixCch = (FilePart != NULL) ? (FilePart - StaticString->Buffer) : 0;
}
if (StringUsed != NULL)
*StringUsed = StaticString;
Status = STATUS_SUCCESS;
} else if (DynamicString == NULL) {
// The static buffer wasn't big enough and the caller doesn't want us to do
// a dynamic allocation; the best we can hope for is to give them back a
// reasonable size.
if (BytesRequired != NULL) {
*BytesRequired = Length;
}
Status = STATUS_BUFFER_TOO_SMALL;
goto Exit;
} else if ((StaticDynamic != NULL) && (Length < StaticBufferSize)) {
//
// Temporary StaticDynamic worked and DynamicString != NULL
// So change StaticDynamic into DynamicString
//
DynamicString->MaximumLength = StaticBufferSize;
DynamicString->Length = (USHORT) Length;
StaticDynamic[Length / sizeof(WCHAR)] = UNICODE_NULL;
if (FilePartPrefixCch != NULL) {
*FilePartPrefixCch = (FilePart != NULL) ? (FilePart - StaticDynamic) : 0;
}
if (StringUsed != NULL) {
*StringUsed = DynamicString;
}
DynamicString->Buffer = StaticDynamic;
StaticDynamic = NULL; // make sure it does not get freed on exit
Status = STATUS_SUCCESS;
} else {
// Not big enough... allocate some memory into the dynamic buffer.
// But wait; we need to lock the Peb lock so that someone doesn't
// change the process's current directory out from under us!
Status = STATUS_INTERNAL_ERROR;
RtlAcquirePebLock();
__try {
// Do it again with the peb lock taken so that we can get an accurate stable
// length.
Length = RtlGetFullPathName_Ustr(
FileName,
StaticBufferSize,
StaticBuffer,
&FilePart,
NameInvalid,
InputPathType);
if (Length == 0) {
#if DBG
DbgPrint("%s line %d: RtlGetFullPathName_Ustr() returned 0\n", __FUNCTION__, __LINE__);
#endif // DBG
Status = STATUS_OBJECT_NAME_INVALID;
__leave;
}
if ((StaticString != NULL) && (Length < StaticString->MaximumLength)) {
// woohoo it worked; some voodoo is going on where the current directory
// or something just changed prior to us acquiring the Peb lock.
StaticString->Length = (USHORT) Length;
if (FilePartPrefixCch != NULL) {
*FilePartPrefixCch = (FilePart != NULL) ? (FilePart - StaticString->Buffer) : 0;
}
if (StringUsed != NULL) {
*StringUsed = StaticString;
}
} else {
// If it doesn't fit into a UNICODE string, we're in big trouble.
if ((Length + sizeof(WCHAR)) > UNICODE_STRING_MAX_BYTES) {
Status = STATUS_NAME_TOO_LONG;
__leave;
}
TempDynamicString.MaximumLength = (USHORT) (Length + sizeof(WCHAR));
TempDynamicString.Buffer = (RtlAllocateStringRoutine)(TempDynamicString.MaximumLength);
if (TempDynamicString.Buffer == NULL) {
Status = STATUS_NO_MEMORY;
__leave;
}
Length = RtlGetFullPathName_Ustr(
FileName,
TempDynamicString.MaximumLength - sizeof(WCHAR),
TempDynamicString.Buffer,
&FilePart,
NameInvalid,
InputPathType);
if (Length == 0) {
#if DBG
DbgPrint("%s line %d: RtlGetFullPathName_Ustr() returned 0\n", __FUNCTION__, __LINE__);
#endif // DBG
Status = STATUS_OBJECT_NAME_INVALID;
__leave;
}
// If this assert fires, it means that someone changed something that
// RtlGetFullPathName_Ustr() uses to resolve the filename, even while
// we're holding the Peb lock. This is really bad and whoever is
// trashing the PEB is resposible, not this code.
ASSERT(Length < (TempDynamicString.MaximumLength - sizeof(WCHAR)));
if (Length > (TempDynamicString.MaximumLength - sizeof(WCHAR))) {
Status = STATUS_INTERNAL_ERROR;
__leave;
}
if (FilePartPrefixCch != NULL) {
*FilePartPrefixCch = (FilePart != NULL) ? (FilePart - TempDynamicString.Buffer) : 0;
}
TempDynamicString.Buffer[Length / sizeof(WCHAR)] = UNICODE_NULL;
DynamicString->Buffer = TempDynamicString.Buffer;
DynamicString->Length = (USHORT) Length;
DynamicString->MaximumLength = TempDynamicString.MaximumLength;
if (StringUsed != NULL)
*StringUsed = DynamicString;
TempDynamicString.Buffer = NULL;
}
Status = STATUS_SUCCESS;
} __finally {
RtlReleasePebLock();
}
}
Exit:
if (StaticDynamic != NULL) {
(RtlFreeStringRoutine)(StaticDynamic);
}
if (TempDynamicString.Buffer != NULL) {
(RtlFreeStringRoutine)(TempDynamicString.Buffer);
}
#if DBG
// This happens a lot for STATUS_NO_SUCH_FILE and STATUS_BUFFER_TOO_SMALL; we'll report any others
if (NT_ERROR(Status) && (Status != STATUS_NO_SUCH_FILE) && (Status != STATUS_BUFFER_TOO_SMALL)) {
DbgPrint("RTL: %s - failing on filename %wZ with status %08lx\n", __FUNCTION__, FileName, Status);
}
#endif // DBG
return Status;
}
ULONG
RtlGetFullPathName_U(
PCWSTR lpFileName,
ULONG nBufferLength,
PWSTR lpBuffer,
PWSTR *lpFilePart OPTIONAL
)
{
UNICODE_STRING UnicodeString;
RTL_PATH_TYPE PathType;
NTSTATUS Status;
Status = RtlInitUnicodeStringEx(&UnicodeString,lpFileName);
if (!NT_SUCCESS (Status)) {
return 0;
}
return RtlGetFullPathName_Ustr(&UnicodeString,nBufferLength,lpBuffer,lpFilePart,NULL,&PathType);
}
NTSTATUS
RtlpWin32NTNameToNtPathName_U(
IN PCUNICODE_STRING DosFileName,
OUT PUNICODE_STRING NtFileName,
OUT PWSTR *FilePart OPTIONAL,
OUT PRTL_RELATIVE_NAME_U RelativeName OPTIONAL
)
/*++
Note: assumes that DosFileName has already been validated to have more than
4 WCHARs
--*/
{
PWSTR FullNtPathName = NULL;
PWSTR Source,Dest;
ULONG NtFileLength;
NTSTATUS Status = STATUS_SUCCESS;
NtFileLength = DosFileName->Length - 8 + RtlpDosDevicesPrefix.Length;
if ( NtFileLength > (MAXUSHORT - sizeof(UNICODE_NULL)) ) {
Status = STATUS_NAME_TOO_LONG;
goto Exit;
}
FullNtPathName = RtlAllocateHeap(
RtlProcessHeap(),
0,
NtFileLength + sizeof(UNICODE_NULL)
);
if ( !FullNtPathName ) {
Status = STATUS_NO_MEMORY;
goto Exit;
}
//
// Copy the full Win32/NT path next to the name prefix, skipping over
// the \\?\ at the front of the path.
//
RtlCopyMemory (FullNtPathName,RtlpDosDevicesPrefix.Buffer,RtlpDosDevicesPrefix.Length);
RtlCopyMemory ((PUCHAR)FullNtPathName+RtlpDosDevicesPrefix.Length,
DosFileName->Buffer + 4,
DosFileName->Length - 8);
//
// Null terminate the path name to make strlen below happy.
//
FullNtPathName[ NtFileLength >> 1 ] = UNICODE_NULL;
//
// Now we have the passed in path with \DosDevices\ prepended. Blow out the
// relative name structure (if provided), and possibly compute filepart
//
if ( ARGUMENT_PRESENT(RelativeName) ) {
//
// If the current directory is a sub-string of the
// Nt file name, and if a handle exists for the current
// directory, then return the directory handle and name
// relative to the directory.
//
RelativeName->RelativeName.Length = 0;
RelativeName->RelativeName.MaximumLength = 0;
RelativeName->RelativeName.Buffer = 0;
RelativeName->ContainingDirectory = NULL;
RelativeName->CurDirRef = NULL;
}
if ( ARGUMENT_PRESENT( FilePart ) ) {
//
// Locate the file part...
//
Source = &FullNtPathName[ (NtFileLength-1) >> 1 ];
Dest = NULL;
while(Source > FullNtPathName ) {
if ( *Source == L'\\' ) {
Dest = Source + 1;
break;
}
Source--;
}
if ( Dest && *Dest ) {
*FilePart = Dest;
} else {
*FilePart = NULL;
}
}
NtFileName->Buffer = FullNtPathName;
NtFileName->Length = (USHORT)(NtFileLength);
NtFileName->MaximumLength = (USHORT)(NtFileLength + sizeof(UNICODE_NULL));
Status = STATUS_SUCCESS;
Exit:
return Status;
}
BOOLEAN
RtlpDosPathNameToRelativeNtPathName_Ustr(
IN BOOLEAN CaptureRelativeName,
IN PCUNICODE_STRING DosFileNameString,
OUT PUNICODE_STRING NtFileName,
OUT PWSTR *FilePart OPTIONAL,
OUT PRTL_RELATIVE_NAME_U RelativeName OPTIONAL
)
/*++
Routine Description:
A Dos pathname can be translated into an Nt style pathname
using this function.
This function is used only within the Base dll to translate Dos
pathnames to Nt pathnames. Upon successful translation, the
pointer (NtFileName->Buffer) points to memory from RtlProcessHeap()
that contains the Nt version of the input dos file name.
Arguments:
CaptureRelativeName - Indicates whether we should capture the
handle for the relative name.
DosFileName - Supplies the unicode Dos style file name that is to be
translated into an equivalent unicode Nt file name.
NtFileName - Returns the address of memory in the RtlProcessHeap() that
contains an NT filename that refers to the specified Dos file
name.
FilePart - Optional parameter that if specified, returns the
trailing file portion of the file name. A path of \foo\bar\x.x
returns the address of x.x as the file part.
RelativeName - An optional parameter, that if specified, returns
a pathname relative to the current directory. The
length field of RelativeName->RelativeName is 0 if the relative
name can not be used.
Return Value:
TRUE - The path name translation was successful. Once the caller is
done with the translated name, the memory pointed to by
NtFileName.Buffer should be returned to the RtlProcessHeap().
FALSE - The operation failed.
Note:
The buffers pointed to by RelativeName, FilePart, and NtFileName must ALL
point within the same memory address. If they don't, code that calls
this routine will fail.
--*/
{
ULONG BufferLength;
ULONG DosPathLength;
PWSTR FullNtPathName = NULL;
PWSTR FullDosPathName = NULL;
UNICODE_STRING Prefix;
UNICODE_STRING UnicodeFilePart;
UNICODE_STRING FullDosPathString;
PCURDIR CurDir;
RTL_PATH_TYPE DosPathType;
RTL_PATH_TYPE InputDosPathType;
ULONG DosPathNameOffset;
ULONG FullDosPathNameLength;
ULONG LastCharacter;
UNICODE_STRING UnicodeString;
BOOLEAN NameInvalid;
WCHAR StaticDosBuffer[DOS_MAX_PATH_LENGTH + 1];
BOOLEAN UseWin32Name, fRC;
//
// Calculate the size needed for the full pathname. Add in
// space for the longest Nt prefix
//
BufferLength = sizeof(StaticDosBuffer);
DosPathLength = (DOS_MAX_PATH_LENGTH << 1 );
UnicodeString = *DosFileNameString;
//
// see if this is \\?\ form of name
//
if ( UnicodeString.Length > 8 && UnicodeString.Buffer[0] == '\\' &&
UnicodeString.Buffer[1] == '\\' && UnicodeString.Buffer[2] == '?' &&
UnicodeString.Buffer[3] == '\\' ) {
UseWin32Name = TRUE;
} else {
UseWin32Name = FALSE;
//
// The dos name starts just after the longest Nt prefix
//
FullDosPathName = &StaticDosBuffer[0];
BufferLength += RtlpLongestPrefix;
//
// Allocate space for the full Nt Name (including DOS name portion)
//
FullNtPathName = RtlAllocateHeap(RtlProcessHeap(), 0, BufferLength);
if ( !FullNtPathName ) {
return FALSE;
}
}
fRC = TRUE;
RtlAcquirePebLock();
__try {
__try {
if ( UseWin32Name ) {
NTSTATUS Status;
Status = RtlpWin32NTNameToNtPathName_U(&UnicodeString,NtFileName,FilePart,RelativeName);
if (NT_SUCCESS(Status)) {
fRC = TRUE;
} else {
fRC = FALSE;
}
__leave;
}
FullDosPathNameLength = RtlGetFullPathName_Ustr(
&UnicodeString,
DosPathLength,
FullDosPathName,
FilePart,
&NameInvalid,
&InputDosPathType
);
if ( NameInvalid || !FullDosPathNameLength ||
FullDosPathNameLength > DosPathLength ) {
fRC = FALSE;
__leave;
}
//
// Determine how to format prefix of FullNtPathName base on the
// the type of Dos path name. All Nt names begin in the \DosDevices
// directory.
//
Prefix = RtlpDosDevicesPrefix;
DosPathType = RtlDetermineDosPathNameType_U(FullDosPathName);
DosPathNameOffset = 0;
switch (DosPathType) {
case RtlPathTypeUncAbsolute :
//
// Unc name, use \DosDevices\UNC symbolic link to find
// redirector. Skip of \\ in source Dos path.
//
Prefix = RtlpDosDevicesUncPrefix;
DosPathNameOffset = 2;
break;
case RtlPathTypeLocalDevice :
//
// Local device name, so just use \DosDevices prefix and
// skip \\.\ in source Dos path.
//
DosPathNameOffset = 4;
break;
case RtlPathTypeRootLocalDevice :
ASSERT( FALSE );
break;
case RtlPathTypeDriveAbsolute :
case RtlPathTypeDriveRelative :
case RtlPathTypeRooted :
case RtlPathTypeRelative :
//
// All drive references just use \DosDevices prefix and
// do not skip any of the characters in the source Dos path.
//
break;
default:
ASSERT( FALSE );
}
//
// Copy the full DOS path next to the name prefix, skipping over
// the "\\" at the front of the UNC path or the "\\.\" at the front
// of a device name.
//
RtlCopyMemory(FullNtPathName,Prefix.Buffer,Prefix.Length);
RtlCopyMemory((PUCHAR)FullNtPathName+Prefix.Length,
FullDosPathName + DosPathNameOffset,
FullDosPathNameLength - (DosPathNameOffset<<1));
//
// Null terminate the path name to make strlen below happy.
//
NtFileName->Buffer = FullNtPathName;
NtFileName->Length = (USHORT)(FullDosPathNameLength-(DosPathNameOffset<<1))+Prefix.Length;
NtFileName->MaximumLength = (USHORT)BufferLength;
LastCharacter = NtFileName->Length >> 1;
FullNtPathName[ LastCharacter ] = UNICODE_NULL;
//
// Readjust the file part to point to the appropriate position within
// the FullNtPathName buffer instead of inside the FullDosPathName
// buffer
//
if ( ARGUMENT_PRESENT(FilePart) ) {
if (*FilePart) {
NTSTATUS Status;
Status = RtlInitUnicodeStringEx(&UnicodeFilePart,*FilePart);
if (!NT_SUCCESS(Status)) {
fRC = FALSE;
__leave;
}
*FilePart = &FullNtPathName[ LastCharacter ] - (UnicodeFilePart.Length >> 1);
}
}
if ( ARGUMENT_PRESENT(RelativeName) ) {
//
// If the current directory is a sub-string of the
// Nt file name, and if a handle exists for the current
// directory, then return the directory handle and name
// relative to the directory.
//
RelativeName->RelativeName.Length = 0;
RelativeName->RelativeName.MaximumLength = 0;
RelativeName->RelativeName.Buffer = 0;
RelativeName->ContainingDirectory = NULL;
RelativeName->CurDirRef = NULL;
if ( InputDosPathType == RtlPathTypeRelative ) {
CurDir = &NtCurrentPeb()->ProcessParameters->CurrentDirectory;
if ( CurDir->Handle ) {
NTSTATUS Status;
//
// Now compare curdir to full dos path. If curdir length is
// greater than full path. It is not a match. Otherwise,
// trim full path length to cur dir length and compare.
//
Status = RtlInitUnicodeStringEx(&FullDosPathString,
FullDosPathName);
if (!NT_SUCCESS(Status)) {
fRC = FALSE;
__leave;
}
if ( CurDir->DosPath.Length <= FullDosPathString.Length ) {
FullDosPathString.Length = CurDir->DosPath.Length;
if ( RtlEqualUnicodeString(
(PUNICODE_STRING)&CurDir->DosPath,
&FullDosPathString,
TRUE
) ) {
//
// The full dos pathname is a substring of the
// current directory. Compute the start of the
// relativename.
//
RelativeName->RelativeName.Buffer = (PWSTR)((PCHAR)FullNtPathName + Prefix.Length - (DosPathNameOffset<<1) + (CurDir->DosPath.Length));
RelativeName->RelativeName.Length = (USHORT)FullDosPathNameLength - (CurDir->DosPath.Length);
if ( *RelativeName->RelativeName.Buffer == L'\\' ) {
RelativeName->RelativeName.Buffer += 1;
RelativeName->RelativeName.Length -= sizeof(WCHAR);
}
RelativeName->RelativeName.MaximumLength = RelativeName->RelativeName.Length;
if (CaptureRelativeName) {
ASSERT(RtlpCurDirRef);
ASSERT(RtlpCurDirRef->DirectoryHandle == CurDir->Handle);
RelativeName->CurDirRef = RtlpCurDirRef;
RtlpReferenceCurDirRef(RtlpCurDirRef);
}
RelativeName->ContainingDirectory = CurDir->Handle;
}
}
}
}
}
} __except (EXCEPTION_EXECUTE_HANDLER) {
fRC = FALSE;
}
}
finally {
RtlReleasePebLock();
if (fRC == FALSE && FullNtPathName != NULL) {
RtlFreeHeap(RtlProcessHeap(), 0, FullNtPathName);
}
}
return fRC;
}
BOOLEAN
RtlDosPathNameToRelativeNtPathName_Ustr(
IN PCUNICODE_STRING DosFileNameString,
OUT PUNICODE_STRING NtFileName,
OUT PWSTR *FilePart OPTIONAL,
OUT PRTL_RELATIVE_NAME_U RelativeName
)
{
ASSERT(RelativeName);
return RtlpDosPathNameToRelativeNtPathName_Ustr(TRUE,
DosFileNameString,
NtFileName,
FilePart,
RelativeName);
}
BOOLEAN
RtlDosPathNameToNtPathName_Ustr(
IN PCUNICODE_STRING DosFileNameString,
OUT PUNICODE_STRING NtFileName,
OUT PWSTR *FilePart OPTIONAL,
IN PVOID Reserved
)
{
return RtlpDosPathNameToRelativeNtPathName_Ustr(FALSE,
DosFileNameString,
NtFileName,
FilePart,
(PRTL_RELATIVE_NAME_U) Reserved);
}
BOOLEAN
RtlpDosPathNameToRelativeNtPathName_U(
IN BOOLEAN CaptureRelativeName,
IN PCWSTR DosFileName,
OUT PUNICODE_STRING NtFileName,
OUT PWSTR *FilePart OPTIONAL,
OUT PRTL_RELATIVE_NAME_U RelativeName OPTIONAL
)
{
UNICODE_STRING DosFileNameString;
SIZE_T Length = 0;
if (DosFileName != NULL) {
Length = wcslen( DosFileName ) * sizeof( WCHAR );
if (Length + sizeof( UNICODE_NULL ) >= UNICODE_STRING_MAX_BYTES) {
return FALSE;
}
DosFileNameString.MaximumLength = (USHORT)(Length + sizeof( UNICODE_NULL ));
} else {
DosFileNameString.MaximumLength = 0;
}
DosFileNameString.Buffer = (PWSTR) DosFileName;
DosFileNameString.Length = (USHORT)Length;
return RtlpDosPathNameToRelativeNtPathName_Ustr(CaptureRelativeName,
&DosFileNameString,
NtFileName,
FilePart,
RelativeName);
}
BOOLEAN
RtlDosPathNameToRelativeNtPathName_U(
IN PCWSTR DosFileName,
OUT PUNICODE_STRING NtFileName,
OUT PWSTR *FilePart OPTIONAL,
OUT PRTL_RELATIVE_NAME_U RelativeName
)
{
ASSERT(RelativeName);
return RtlpDosPathNameToRelativeNtPathName_U(TRUE,
DosFileName,
NtFileName,
FilePart,
RelativeName);
}
BOOLEAN
RtlDosPathNameToNtPathName_U(
IN PCWSTR DosFileName,
OUT PUNICODE_STRING NtFileName,
OUT PWSTR *FilePart OPTIONAL,
IN PVOID Reserved
)
{
return RtlpDosPathNameToRelativeNtPathName_U(FALSE,
DosFileName,
NtFileName,
FilePart,
(PRTL_RELATIVE_NAME_U) Reserved);
}
VOID
NTAPI
RtlReleaseRelativeName(
IN PRTL_RELATIVE_NAME_U RelativeName
)
/*++
Routine Description:
This function closes the handle earlier created by a
call to RtlDosPathNameToRelativeNtPathName_U().
Arguments:
RelativeName - contains the current directory information to be released.
Return Value:
None.
--*/
{
ASSERT(RelativeName);
if (RelativeName->CurDirRef) {
RtlpDereferenceCurDirRef(RelativeName->CurDirRef);
RelativeName->CurDirRef = NULL;
}
}
BOOLEAN
RtlDoesFileExists_UstrEx(
IN PCUNICODE_STRING FileNameString,
IN BOOLEAN TreatDeniedOrSharingAsHit
)
/*++
Routine Description:
This function checks to see if the specified unicode filename exists.
Arguments:
FileName - Supplies the file name of the file to find.
TreatDeniedOrSharingAsHit - treat denied or sharing error as file existance
TRUE - treat denied or sharing error as sign of file existance
FALSE - do not treat denied or sharing error as sign of file existance
Return Value:
TRUE - The file was found.
FALSE - The file was not found.
--*/
{
NTSTATUS Status;
OBJECT_ATTRIBUTES Obja;
UNICODE_STRING NtFileName;
BOOLEAN ReturnValue;
RTL_RELATIVE_NAME_U RelativeName;
PVOID FreeBuffer;
FILE_BASIC_INFORMATION BasicInfo;
ReturnValue = RtlDosPathNameToRelativeNtPathName_Ustr(
FileNameString,
&NtFileName,
NULL,
&RelativeName
);
if ( !ReturnValue ) {
return FALSE;
}
FreeBuffer = NtFileName.Buffer;
if ( RelativeName.RelativeName.Length ) {
NtFileName = RelativeName.RelativeName;
}
else {
RelativeName.ContainingDirectory = NULL;
}
InitializeObjectAttributes(
&Obja,
&NtFileName,
OBJ_CASE_INSENSITIVE,
RelativeName.ContainingDirectory,
NULL
);
//
// Query the file's attributes. Note that the file cannot simply be opened
// to determine whether or not it exists, as the NT LanMan redirector lies
// on NtOpenFile to a Lan Manager server because it does not actually open
// the file until an operation is performed on it.
//
Status = NtQueryAttributesFile(
&Obja,
&BasicInfo
);
RtlReleaseRelativeName(&RelativeName);
RtlFreeHeap(RtlProcessHeap(),0,FreeBuffer);
if ( !NT_SUCCESS(Status) ) {
if ( Status == STATUS_SHARING_VIOLATION ||
Status == STATUS_ACCESS_DENIED ) {
if ( TreatDeniedOrSharingAsHit ) {
ReturnValue = TRUE;
}
else {
ReturnValue = FALSE;
}
}
else {
ReturnValue = FALSE;
}
}
else {
ReturnValue = TRUE;
}
return ReturnValue;
}
BOOLEAN
RtlDoesFileExists_UEx(
IN PCWSTR FileName,
IN BOOLEAN TreatDeniedOrSharingAsHit
)
{
UNICODE_STRING FileNameString;
NTSTATUS Status;
Status = RtlInitUnicodeStringEx(&FileNameString, FileName);
if( !NT_SUCCESS( Status ) ) {
return FALSE;
}
return RtlDoesFileExists_UstrEx(&FileNameString, TreatDeniedOrSharingAsHit);
}
BOOLEAN
RtlDoesFileExists_U(
IN PCWSTR FileName
)
/*++
Routine Description:
This function checks to see if the specified unicode filename exists.
Arguments:
FileName - Supplies the file name of the file to find.
Return Value:
TRUE - The file was found.
FALSE - The file was not found.
--*/
{
return RtlDoesFileExists_UEx(FileName,TRUE);
}
BOOLEAN
RtlDoesFileExists_UStr(
IN PCUNICODE_STRING FileName
)
/*
RtlDoesFileExists_UStr same as RtlDoesFileExists_U but takes a PCUNICODE_STRING instead of a PCWSTR,
saving a call to wcslen
*/
{
return RtlDoesFileExists_UstrEx(FileName,TRUE);
}
ULONG
RtlDosSearchPath_U(
IN PCWSTR lpPath,
IN PCWSTR lpFileName,
IN PCWSTR lpExtension OPTIONAL,
IN ULONG nBufferLength,
OUT PWSTR lpBuffer,
OUT PWSTR *lpFilePart
)
/*++
Routine Description:
This function is used to search for a file specifying a search path
and a filename. It returns with a fully qualified pathname of the
found file.
This function is used to locate a file using the specified path. If
the file is found, its fully qualified pathname is returned. In
addition to this, it calculates the address of the file name portion
of the fully qualified pathname.
Arguments:
lpPath - Supplies the search path to be used when locating the file.
lpFileName - Supplies the file name of the file to search for.
lpExtension - An optional parameter, that if specified, supplies an
extension to be added to the filename when doing the search.
The extension is only added if the specified filename does not
end with an extension.
nBufferLength - Supplies the length in bytes of the buffer that is
to receive the fully qualified path.
lpBuffer - Returns the fully qualified pathname corresponding to the
file that was found.
lpFilePart - Optional parameter that if specified, returns the
address of the last component of the fully qualified pathname.
Return Value:
The return value is the length of the string copied to lpBuffer, not
including the terminating null character. If the return value is
greater than nBufferLength, the return value is the size of the buffer
required to hold the pathname. The return value is zero if the
function failed.
--*/
{
PWSTR ComputedFileName;
ULONG ExtensionLength;
ULONG PathLength;
ULONG FileLength;
UNICODE_STRING Scratch;
PCWSTR p;
NTSTATUS Status;
//
// if the file name is not a relative name, then
// return if the file does not exist.
//
// If a fully qualified pathname is used in the search, then
// allow access_denied or sharing violations to terminate the
// search. This was the nt 3.1-4.0 behavior, and was changed for the
// loader to handle cases where when walking through a search, we didn't
// terminate the search early because of an inaccessible UNC path component
// be restoring the old behavior in this case, we give the correct (access_denied)
// error codes on fully qualified module lookups, but keep going when bumping
// through search path components
//
if ( RtlDetermineDosPathNameType_U(lpFileName) != RtlPathTypeRelative ) {
if (RtlDoesFileExists_UEx(lpFileName,TRUE) ) {
PathLength = RtlGetFullPathName_U(
lpFileName,
nBufferLength,
lpBuffer,
lpFilePart
);
return PathLength;
} else {
return 0;
}
}
//
// Determine if the file name contains an extension
//
ExtensionLength = 1;
p = lpFileName;
while (*p) {
if ( *p == L'.' ) {
ExtensionLength = 0;
break;
}
p++;
}
//
// If no extension was found, then determine the extension length
// that should be used to search for the file
//
if ( ExtensionLength ) {
if ( ARGUMENT_PRESENT(lpExtension) ) {
Status = RtlInitUnicodeStringEx(&Scratch,lpExtension);
if ( !NT_SUCCESS(Status) ) {
return 0;
}
ExtensionLength = Scratch.Length;
} else {
ExtensionLength = 0;
}
}
//
// Compute the file name length and the path length;
//
Status = RtlInitUnicodeStringEx(&Scratch,lpPath);
if ( !NT_SUCCESS(Status) ) {
return 0;
}
PathLength = Scratch.Length;
Status = RtlInitUnicodeStringEx(&Scratch,lpFileName);
if ( !NT_SUCCESS(Status) ) {
return 0;
}
FileLength = Scratch.Length;
//
// FUTURE-2002/02/19-ELi
// Could chew up unnecessary memory with PathLength
// sync up with the Str version
//
ComputedFileName = RtlAllocateHeap(
RtlProcessHeap(), 0,
PathLength + FileLength + ExtensionLength + 3*sizeof(UNICODE_NULL)
);
if ( !ComputedFileName ) {
KdPrint(("%s: Failing due to out of memory (RtlAllocateHeap failure)\n", __FUNCTION__));
return 0;
}
//
// find ; 's in path and copy path component to computed file name
//
do {
PWSTR Cursor;
Cursor = ComputedFileName;
while (*lpPath) {
if (*lpPath == L';') {
lpPath++;
break;
}
*Cursor++ = *lpPath++;
}
if (Cursor != ComputedFileName &&
Cursor [ -1 ] != L'\\' ) {
*Cursor++ = L'\\';
}
if (*lpPath == UNICODE_NULL) {
lpPath = NULL;
}
RtlCopyMemory(Cursor,lpFileName,FileLength);
if ( ExtensionLength ) {
RtlCopyMemory((PUCHAR)Cursor+FileLength,lpExtension,ExtensionLength+sizeof(UNICODE_NULL));
} else {
*(PWSTR)((PUCHAR)Cursor+FileLength) = UNICODE_NULL;
}
if (RtlDoesFileExists_UEx(ComputedFileName,FALSE) ) {
PathLength = RtlGetFullPathName_U(
ComputedFileName,
nBufferLength,
lpBuffer,
lpFilePart
);
RtlFreeHeap(RtlProcessHeap(), 0, ComputedFileName);
return PathLength;
}
} while ( lpPath );
RtlFreeHeap(RtlProcessHeap(), 0, ComputedFileName);
return 0;
}
NTSTATUS
RtlDosSearchPath_Ustr(
IN ULONG Flags,
IN PCUNICODE_STRING Path,
IN PCUNICODE_STRING FileName,
IN PCUNICODE_STRING DefaultExtension OPTIONAL,
OUT PUNICODE_STRING StaticString OPTIONAL,
OUT PUNICODE_STRING DynamicString OPTIONAL,
OUT PCUNICODE_STRING *FullFileNameOut OPTIONAL,
OUT SIZE_T *FilePartPrefixCch OPTIONAL,
OUT SIZE_T *BytesRequired OPTIONAL // includes space for trailing NULL
)
/*++
Routine Description:
This function is used to search for a file specifying a search path
and a filename. It returns with a fully qualified pathname of the
found file.
This function is used to locate a file using the specified path. If
the file is found, its fully qualified pathname is returned. In
addition to this, it calculates the address of the file name portion
of the fully qualified pathname.
Arguments:
Flags - Optional flags to affect the behavior of the path search.
Use the logical or operator (|) to combine flags.
Defined flags include:
RTL_DOS_SEARCH_PATH_FLAG_APPLY_ISOLATION_REDIRECTION
If the FileName passed in is a relative path, isolation
redirection of the file path is applied prior to searching
for a matching path.
Path - search path to use when locating the file
FileName - file name to search for
DefaultExtension - Optional extension to apply to the file name
if the file name does not include an extension.
StaticString - Optional UNICODE_STRING which references an already
allocated buffer which is used to construct the actual path
of the file.
DynamicString - Optional UNICODE_STRING which will be filled in with
a dynamically allocated UNICODE_STRING if either StaticBuffer is
not provided, or is not long enough to hold the resolved name.
The dynamic buffer's size is reflected in the MaximumLength field
of the UNICODE_STRING. It will always exceed the Length of the
string by at least two bytes, but may be even larger.
FullFileNameOut - Optional pointer to UNICODE_STRING which points to
the complete resolved file name. This UNICODE_STRING is not
allocated; it is either set equal to FileName, StaticBuffer or
DynamicBuffer as appropriate.
Return Value:
NTSTATUS indicating the disposition of the function. If the file
is redirected via activation context data, STATUS_SUCCESS is returned
regardless of whether the file exists or not. If the file does not
exist in any of the directories referenced by the Path parameter,
STATUS_NO_SUCH_FILE is returned.
--*/
{
NTSTATUS Status;
PWSTR Cursor;
PWSTR EndMarker;
SIZE_T MaximumPathSegmentLength = 0;
SIZE_T BiggestPossibleFileName;
USHORT DefaultExtensionLength = 0;
RTL_PATH_TYPE PathType; // not used; required argument to RtlGetFullPathName_Ustr().
UNICODE_STRING CandidateString;
WCHAR StaticCandidateBuffer[DOS_MAX_PATH_LENGTH];
CandidateString.Length = 0;
CandidateString.MaximumLength = sizeof(StaticCandidateBuffer);
CandidateString.Buffer = StaticCandidateBuffer;
if (FullFileNameOut != NULL) {
*FullFileNameOut = NULL;
}
if (BytesRequired != NULL) {
*BytesRequired = 0;
}
if (FilePartPrefixCch != NULL) {
*FilePartPrefixCch = 0;
}
if (DynamicString != NULL) {
DynamicString->Length = 0;
DynamicString->MaximumLength = 0;
DynamicString->Buffer = NULL;
}
if (((Flags & ~(
RTL_DOS_SEARCH_PATH_FLAG_APPLY_ISOLATION_REDIRECTION |
RTL_DOS_SEARCH_PATH_FLAG_DISALLOW_DOT_RELATIVE_PATH_SEARCH |
RTL_DOS_SEARCH_PATH_FLAG_APPLY_DEFAULT_EXTENSION_WHEN_NOT_RELATIVE_PATH_EVEN_IF_FILE_HAS_EXTENSION)) != 0) ||
(Path == NULL) ||
(FileName == NULL) ||
((StaticString != NULL) && (DynamicString != NULL) && (FullFileNameOut == NULL))) {
#if DBG
DbgPrint("%s: Invalid parameters passed\n", __FUNCTION__);
#endif // DBG
Status = STATUS_INVALID_PARAMETER;
goto Exit;
}
PathType = RtlDetermineDosPathNameType_Ustr(FileName);
// If the caller wants to disallow .\ and ..\ relative path searches, stop them!
if ((Flags & RTL_DOS_SEARCH_PATH_FLAG_DISALLOW_DOT_RELATIVE_PATH_SEARCH) && (PathType == RtlPathTypeRelative)) {
if (FileName->Length >= (2 * sizeof(WCHAR))) {
if (FileName->Buffer[0] == L'.') {
if (IS_PATH_SEPARATOR_U(FileName->Buffer[1])) {
PathType = RtlPathTypeUnknown;
} else if ((FileName->Buffer[1] == L'.') &&
(FileName->Length >= (3 * sizeof(WCHAR))) &&
IS_PATH_SEPARATOR_U(FileName->Buffer[2])) {
PathType = RtlPathTypeUnknown;
}
}
}
}
//
// if the file name is not a relative name, then
// return an if the file does not exist.
//
// If a fully qualified pathname is used in the search, then
// allow access_denied or sharing violations to terminate the
// search. This was the nt 3.1-4.0 behavior, and was changed for the
// loader to handle cases where when walking through a search, we didn't
// terminate the search early because of an inaccessible UNC path component
// be restoring the old behavior in this case, we give the correct (access_denied)
// error codes on fully qualified module lookups, but keep going when bumping
// through search path components
//
if (PathType != RtlPathTypeRelative ) {
if (RtlDoesFileExists_UstrEx(FileName, TRUE)) {
Status = RtlGetFullPathName_UstrEx(
FileName,
StaticString,
DynamicString,
(PUNICODE_STRING *) FullFileNameOut,
FilePartPrefixCch,
NULL,
&PathType,
BytesRequired);
if (!NT_SUCCESS(Status)) {
#if DBG
if ((Status != STATUS_NO_SUCH_FILE) && (Status != STATUS_BUFFER_TOO_SMALL)) {
DbgPrint("%s: Failing because RtlGetFullPathName_UstrEx() on %wZ failed with %08lx\n", __FUNCTION__, FileName, Status);
}
#endif // DBG
goto Exit;
}
} else {
//
// The file wasn't there; let's try adding the default extension if we need to.
//
if ((DefaultExtension == NULL) || (DefaultExtension->Length == 0)) {
#if DBG
// DbgPrint("%s: Failing because RtlDoesFileExists_UstrEx() on %wZ says it does not exist and there is no default extension to apply\n", __FUNCTION__, FileName);
#endif // DBG
Status = STATUS_NO_SUCH_FILE;
goto Exit;
}
DefaultExtensionLength = DefaultExtension->Length;
// If they've asked for SearchPathW() bug compatibility mode, always apply the default
// extension if the file isn't found, even if the file name has an extension.
if (!(Flags & RTL_DOS_SEARCH_PATH_FLAG_APPLY_DEFAULT_EXTENSION_WHEN_NOT_RELATIVE_PATH_EVEN_IF_FILE_HAS_EXTENSION)) {
if (FileName->Length != 0) {
Cursor = FileName->Buffer + (FileName->Length / sizeof(WCHAR));
while (Cursor > FileName->Buffer) {
const WCHAR wch = *--Cursor;
if (IS_PATH_SEPARATOR_U(wch)) {
// it's a slash; we have a filename without an extension...
break;
}
if (wch == L'.') {
// There was an extension. We're just out of luck.
Status = STATUS_NO_SUCH_FILE;
goto Exit;
}
}
}
}
// We need to move the filename into a different buffer.
BiggestPossibleFileName = (FileName->Length + DefaultExtensionLength + sizeof(WCHAR));
if (BiggestPossibleFileName > UNICODE_STRING_MAX_BYTES) {
#if DBG
DbgPrint("%s: Failing because the filename plus extension (%Iu bytes) is too big\n", __FUNCTION__, BiggestPossibleFileName);
#endif // DBG
Status = STATUS_NAME_TOO_LONG;
goto Exit;
}
// If the buffer on the stack isn't big enough, allocate one from the heap
if (BiggestPossibleFileName > CandidateString.MaximumLength) {
CandidateString.MaximumLength = (USHORT) BiggestPossibleFileName;
CandidateString.Buffer = (RtlAllocateStringRoutine)(CandidateString.MaximumLength);
if (CandidateString.Buffer == NULL) {
#if DBG
DbgPrint("%s: Failing because allocating the dynamic filename buffer failed\n", __FUNCTION__);
#endif // DBG
Status = STATUS_NO_MEMORY;
goto Exit;
}
}
RtlCopyMemory(CandidateString.Buffer, FileName->Buffer, FileName->Length);
RtlCopyMemory(CandidateString.Buffer + (FileName->Length / sizeof(WCHAR)), DefaultExtension->Buffer, DefaultExtension->Length);
CandidateString.Buffer[(FileName->Length + DefaultExtension->Length) / sizeof(WCHAR)] = UNICODE_NULL;
CandidateString.Length = FileName->Length + DefaultExtension->Length;
if (!RtlDoesFileExists_UstrEx(&CandidateString, TRUE)) {
Status = STATUS_NO_SUCH_FILE;
goto Exit;
}
Status = RtlGetFullPathName_UstrEx(
&CandidateString,
StaticString,
DynamicString,
(PUNICODE_STRING *) FullFileNameOut,
FilePartPrefixCch,
NULL,
&PathType,
BytesRequired);
if (!NT_SUCCESS(Status)) {
#if DBG
if (Status != STATUS_NO_SUCH_FILE) {
DbgPrint("%s: Failing on \"%wZ\" because RtlGetFullPathName_UstrEx() failed with status %08lx\n", __FUNCTION__, &CandidateString, Status);
}
#endif // DBG
goto Exit;
}
}
Status = STATUS_SUCCESS;
goto Exit;
}
// We know it's a relative path at this point. Do we want to try side-by-side
// isolation of the file?
if (Flags & RTL_DOS_SEARCH_PATH_FLAG_APPLY_ISOLATION_REDIRECTION) {
PUNICODE_STRING FullPathStringFound = NULL;
Status = RtlDosApplyFileIsolationRedirection_Ustr(
RTL_DOS_APPLY_FILE_REDIRECTION_USTR_FLAG_RESPECT_DOT_LOCAL, FileName,
DefaultExtension, StaticString, DynamicString, &FullPathStringFound,
NULL, FilePartPrefixCch, BytesRequired);
if (NT_SUCCESS(Status)) {
if (FullFileNameOut != NULL) {
*FullFileNameOut = FullPathStringFound;
}
Status = STATUS_SUCCESS;
goto Exit;
}
if (Status != STATUS_SXS_KEY_NOT_FOUND) {
#if DBG
DbgPrint("%s: Failing because call to RtlDosApplyFileIsolationRedirection_Ustr(%wZ) failed with status 0x%08lx\n", __FUNCTION__, FileName, Status);
#endif // DBG
goto Exit;
}
}
//
// If a default extension was provided, see if we need to account for it
//
if (DefaultExtension != NULL) {
DefaultExtensionLength = DefaultExtension->Length;
if (FileName->Length != 0) {
Cursor = FileName->Buffer + (FileName->Length / sizeof(WCHAR));
while (Cursor > FileName->Buffer) {
const WCHAR wch = *--Cursor;
if (IS_PATH_SEPARATOR_U(wch)) {
// it's a slash; we have a filename without an extension...
break;
}
if (wch == L'.') {
// There's an extension; ignore the defualt.
DefaultExtension = NULL;
DefaultExtensionLength = 0;
break;
}
}
}
}
if (Path->Length != 0) {
USHORT CchThisSegment;
PCWSTR LastCursor;
Cursor = Path->Buffer + (Path->Length / sizeof(WCHAR));
LastCursor = Cursor;
while (Cursor > Path->Buffer) {
if (*--Cursor == L';') {
CchThisSegment = (USHORT) ((LastCursor - Cursor) - 1);
if (CchThisSegment != 0) {
// If there is not a trailing slash, add one character
if (!IS_PATH_SEPARATOR_U(LastCursor[-1])) {
CchThisSegment++;
}
}
if (CchThisSegment > MaximumPathSegmentLength) {
MaximumPathSegmentLength = CchThisSegment;
}
// LastCursor now points to the semicolon...
LastCursor = Cursor;
}
}
CchThisSegment = (USHORT) (LastCursor - Cursor);
if (CchThisSegment != 0) {
if (!IS_PATH_SEPARATOR_U(LastCursor[-1])) {
CchThisSegment++;
}
}
if (CchThisSegment > MaximumPathSegmentLength) {
MaximumPathSegmentLength = CchThisSegment;
}
// Convert from WCHARs to bytes
MaximumPathSegmentLength *= sizeof(WCHAR);
}
BiggestPossibleFileName =
MaximumPathSegmentLength +
FileName->Length +
DefaultExtensionLength +
sizeof(WCHAR); // don't forget space for a trailing NULL...
// It's all got to fit into a UNICODE_STRING at some point, so check that that's possible
if (BiggestPossibleFileName > UNICODE_STRING_MAX_BYTES) {
#if DBG
DbgPrint("%s: returning STATUS_NAME_TOO_LONG because the computed worst case file name length is %Iu bytes\n", __FUNCTION__, BiggestPossibleFileName);
#endif // DBG
Status = STATUS_NAME_TOO_LONG;
goto Exit;
}
// It's tempting to allocate the dynamic buffer here, but if it turns out that
// the file is quickly found in one of the first segments that fits in the
// static buffer, we'll have wasted a heap alloc.
Cursor = Path->Buffer;
EndMarker = Cursor + (Path->Length / sizeof(WCHAR));
while (Cursor < EndMarker) {
PWSTR BufferToFillCursor;
PWSTR SegmentEnd = Cursor;
USHORT SegmentSize;
USHORT BytesToCopy;
UNICODE_STRING DebugString;
// Scan ahead for the end of the path buffer or the next semicolon
while ((SegmentEnd != EndMarker) && (*SegmentEnd != L';'))
SegmentEnd++;
SegmentSize = (USHORT) ((SegmentEnd - Cursor) * sizeof(WCHAR));
DebugString.Buffer = Cursor;
DebugString.Length = SegmentSize;
DebugString.MaximumLength = SegmentSize;
BytesToCopy = SegmentSize;
// Add space for a trailing slash if there isn't one.
if ((SegmentSize != 0) && !IS_PATH_SEPARATOR_U(SegmentEnd[-1])) {
SegmentSize += sizeof(WCHAR);
}
// If the string we're using for the candidates isn't big enough, allocate one that is.
if (CandidateString.MaximumLength < (SegmentSize + FileName->Length + DefaultExtensionLength + sizeof(WCHAR))) {
// If CandidateString is already a dynamic buffer, something's hosed because we should have allocated
// the largest one needed the first time we outgrew the static one.
ASSERT(CandidateString.Buffer == StaticCandidateBuffer);
if (CandidateString.Buffer != StaticCandidateBuffer) {
#if DBG
DbgPrint("%s: internal error #1; CandidateString.Buffer = %p; StaticCandidateBuffer = %p\n", __FUNCTION__, CandidateString.Buffer, StaticCandidateBuffer);
#endif // DBG
Status = STATUS_INTERNAL_ERROR;
goto Exit;
}
// If this assert fires, there's either a code bug above where we computed the maximum
// segment length, or someone's changing either the filename, default extension or
// path around on us in another thread. Performing a capture on the buffers seems like
// massive overkill, so we'll just not overrun our buffers here.
ASSERT((SegmentSize + FileName->Length + DefaultExtensionLength) < UNICODE_STRING_MAX_BYTES);
if ((SegmentSize + FileName->Length + DefaultExtensionLength ) >
(UNICODE_STRING_MAX_BYTES - sizeof(WCHAR))) {
#if DBG
DbgPrint("%s: internal error #2; SegmentSize = %u, FileName->Length = %u, DefaultExtensionLength = %u\n", __FUNCTION__,
SegmentSize, FileName->Length, DefaultExtensionLength);
#endif // DBG
Status = STATUS_INTERNAL_ERROR;
goto Exit;
}
CandidateString.MaximumLength = (USHORT) BiggestPossibleFileName;
CandidateString.Buffer = (RtlAllocateStringRoutine)(CandidateString.MaximumLength);
if (CandidateString.Buffer == NULL) {
#if DBG
DbgPrint("%s: Unable to allocate %u byte buffer for path candidate\n", __FUNCTION__, CandidateString.MaximumLength);
#endif // DBG
Status = STATUS_NO_MEMORY;
goto Exit;
}
}
RtlCopyMemory(
CandidateString.Buffer,
Cursor,
BytesToCopy);
BufferToFillCursor = CandidateString.Buffer + (BytesToCopy / sizeof(WCHAR));
// Add a trailing slash if it was omitted. It's safe to index [-1] since
// we know that SegmentSize != 0.
if ((SegmentSize != 0) && (BytesToCopy != SegmentSize))
*BufferToFillCursor++ = L'\\';
RtlCopyMemory(
BufferToFillCursor,
FileName->Buffer,
FileName->Length);
BufferToFillCursor += (FileName->Length / sizeof(WCHAR));
if (DefaultExtension != NULL) {
RtlCopyMemory(
BufferToFillCursor,
DefaultExtension->Buffer,
DefaultExtension->Length);
BufferToFillCursor += (DefaultExtension->Length / sizeof(WCHAR));
}
// And top it off with a unicode null...
*BufferToFillCursor = UNICODE_NULL;
CandidateString.Length = (USHORT) ((BufferToFillCursor - CandidateString.Buffer) * sizeof(WCHAR));
if (RtlDoesFileExists_UEx(CandidateString.Buffer, FALSE)) {
// Run it through the path canonicalizer...
Status = RtlGetFullPathName_UstrEx(
&CandidateString,
StaticString,
DynamicString,
(PUNICODE_STRING *) FullFileNameOut,
FilePartPrefixCch,
NULL,
&PathType,
BytesRequired);
if (NT_SUCCESS(Status))
Status = STATUS_SUCCESS;
else {
#if DBG
if ((Status != STATUS_NO_SUCH_FILE) && (Status != STATUS_BUFFER_TOO_SMALL)) {
DbgPrint("%s: Failing because we thought we found %wZ on the search path, but RtlGetFullPathName_UstrEx() returned %08lx\n", __FUNCTION__, FileName, Status);
}
#endif // DBG
}
goto Exit;
}
if (SegmentEnd != EndMarker)
Cursor = SegmentEnd + 1;
else
Cursor = SegmentEnd;
}
Status = STATUS_NO_SUCH_FILE;
Exit:
if ((CandidateString.Buffer != NULL) &&
(CandidateString.Buffer != StaticCandidateBuffer)) {
RtlFreeUnicodeString(&CandidateString);
}
return Status;
}
VOID
RtlpCheckRelativeDrive(
WCHAR NewDrive
)
/*++
Routine Description:
This function is called whenever we are asked to expand a non
current directory drive relative name ( f:this\is\my\file ). In
this case, we validate the environment variable string to make sure
the current directory at that drive is valid. If not, we trim back to
the root.
Arguments:
NewDrive - Supplies the drive to check
Return Value:
None.
--*/
{
WCHAR EnvVarValueBuffer[DOS_MAX_PATH_LENGTH+12]; // + sizeof (\DosDevices\)
WCHAR EnvVarNameBuffer[4];
UNICODE_STRING EnvVarName;
UNICODE_STRING EnvValue;
NTSTATUS Status;
OBJECT_ATTRIBUTES Obja;
IO_STATUS_BLOCK IoStatusBlock;
HANDLE DirHandle;
ULONG HardErrorValue;
EnvVarNameBuffer[0] = L'=';
EnvVarNameBuffer[1] = (WCHAR)NewDrive;
EnvVarNameBuffer[2] = L':';
EnvVarNameBuffer[3] = UNICODE_NULL;
RtlInitUnicodeString(&EnvVarName,EnvVarNameBuffer);
//
// capture the value in a buffer that has space at the front for the dos devices
// prefix
//
EnvValue.Length = 0;
EnvValue.MaximumLength = DOS_MAX_PATH_LENGTH<<1;
EnvValue.Buffer = &EnvVarValueBuffer[RtlpDosDevicesPrefix.Length>>1];
Status = RtlQueryEnvironmentVariable_U( NULL,
&EnvVarName,
&EnvValue
);
if ( !NT_SUCCESS( Status ) ) {
//
// Otherwise default to root directory of drive
//
EnvValue.Buffer[0] = (WCHAR)NewDrive;
EnvValue.Buffer[1] = L':';
EnvValue.Buffer[2] = L'\\';
EnvValue.Buffer[3] = UNICODE_NULL;
EnvValue.Length = 6;
}
//
// Form the NT name for this directory
//
EnvValue.Length = EnvValue.Length + RtlpDosDevicesPrefix.Length;
EnvValue.MaximumLength = sizeof(EnvVarValueBuffer);
EnvValue.Buffer = EnvVarValueBuffer;
RtlCopyMemory(EnvVarValueBuffer,RtlpDosDevicesPrefix.Buffer,RtlpDosDevicesPrefix.Length);
InitializeObjectAttributes(
&Obja,
&EnvValue,
OBJ_CASE_INSENSITIVE,
NULL,
NULL
);
RtlSetThreadErrorMode(RTL_ERRORMODE_FAILCRITICALERRORS,
&HardErrorValue);
Status = NtOpenFile(
&DirHandle,
SYNCHRONIZE,
&Obja,
&IoStatusBlock,
FILE_SHARE_READ | FILE_SHARE_WRITE,
FILE_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT
);
RtlSetThreadErrorMode(HardErrorValue, NULL);
//
// If the open succeeds, then the directory is valid... No need to do anything
// further. If the open fails, then trim back the environment to the root.
//
if ( NT_SUCCESS(Status) ) {
NtClose(DirHandle);
return;
}
RtlpResetDriveEnvironment(NewDrive);
}
#define RTLP_LAST_PATH_ELEMENT_PATH_TYPE_FULL_DOS_OR_NT (0x00000001)
#define RTLP_LAST_PATH_ELEMENT_PATH_TYPE_FULL_DOS (0x00000002)
#define RTLP_LAST_PATH_ELEMENT_PATH_TYPE_NT (0x00000003)
#define RTLP_LAST_PATH_ELEMENT_PATH_TYPE_DOS (0x00000004)
NTSTATUS
NTAPI
RtlpGetLengthWithoutLastPathElement(
IN ULONG Flags,
IN ULONG PathType,
IN PCUNICODE_STRING Path,
OUT ULONG* LengthOut
)
/*++
Routine Description:
Report how long Path would be if you remove its last element.
This is much simpler than RtlRemoveLastDosPathElement.
It is used to implement the other RtlRemoveLast*PathElement.
Arguments:
Flags - room for future expansion
Path - the path is is an NT path or a full DOS path; the various relative DOS
path types do not work, see RtlRemoveLastDosPathElement for them.
Return Value:
STATUS_SUCCESS - the usual hunky-dory
STATUS_NO_MEMORY - the usual stress
STATUS_INVALID_PARAMETER - the usual bug
--*/
{
ULONG Length = 0;
NTSTATUS Status = STATUS_SUCCESS;
RTL_PATH_TYPE DosPathType = RtlPathTypeUnknown;
ULONG DosPathFlags = 0;
ULONG AllowedDosPathTypeBits = (1UL << RtlPathTypeRooted)
| (1UL << RtlPathTypeUncAbsolute)
| (1UL << RtlPathTypeDriveAbsolute)
| (1UL << RtlPathTypeLocalDevice) // "\\?\"
| (1UL << RtlPathTypeRootLocalDevice) // "\\?"
;
WCHAR PathSeperators[2] = { '/', '\\' };
#define LOCAL_IS_PATH_SEPARATOR(ch_) ((ch_) == PathSeperators[0] || (ch_) == PathSeperators[1])
if (LengthOut != NULL) {
*LengthOut = 0;
}
if ( !RTL_SOFT_VERIFY(Path != NULL)
|| !RTL_SOFT_VERIFY(Flags == 0)
|| !RTL_SOFT_VERIFY(LengthOut != NULL)
) {
Status = STATUS_INVALID_PARAMETER;
goto Exit;
}
Length = RTL_STRING_GET_LENGTH_CHARS(Path);
switch (PathType)
{
default:
case RTLP_LAST_PATH_ELEMENT_PATH_TYPE_DOS:
Status = STATUS_INVALID_PARAMETER;
goto Exit;
case RTLP_LAST_PATH_ELEMENT_PATH_TYPE_NT:
//
// RtlpDetermineDosPathNameType4 calls it "rooted"
// only backslashes are seperators
// path must start with backslash
// second char must not be backslash
//
AllowedDosPathTypeBits = (1UL << RtlPathTypeRooted);
PathSeperators[0] = '\\';
if (Length > 0 && Path->Buffer[0] != '\\'
) {
Status = STATUS_INVALID_PARAMETER;
goto Exit;
}
if (Length > 1 && Path->Buffer[1] == '\\'
) {
Status = STATUS_INVALID_PARAMETER;
goto Exit;
}
break;
case RTLP_LAST_PATH_ELEMENT_PATH_TYPE_FULL_DOS:
AllowedDosPathTypeBits &= ~(1UL << RtlPathTypeRooted);
break;
case RTLP_LAST_PATH_ELEMENT_PATH_TYPE_FULL_DOS_OR_NT:
break;
}
if (Length == 0) {
goto Exit;
}
Status = RtlpDetermineDosPathNameType4(
RTL_DETERMINE_DOS_PATH_NAME_TYPE_IN_FLAG_STRICT_WIN32NT,
Path,
&DosPathType,
&DosPathFlags
);
if (!RTL_SOFT_VERIFY(NT_SUCCESS(Status))) {
goto Exit;
}
if (!RTL_SOFT_VERIFY((1UL << DosPathType) & AllowedDosPathTypeBits)
) {
//KdPrintEx();
Status = STATUS_INVALID_PARAMETER;
goto Exit;
}
if (!RTL_SOFT_VERIFY(
(DosPathFlags & RTLP_DETERMINE_DOS_PATH_NAME_TYPE_OUT_FLAG_INVALID) == 0
)) {
Status = STATUS_INVALID_PARAMETER;
goto Exit;
}
// skip one or more trailing path seperators
for ( ; Length != 0 && LOCAL_IS_PATH_SEPARATOR(Path->Buffer[Length - 1]) ; --Length) {
// nothing
}
// skip trailing path element
for ( ; Length != 0 && !LOCAL_IS_PATH_SEPARATOR(Path->Buffer[Length - 1]) ; --Length) {
// nothing
}
// skip one or more in between path seperators
for ( ; Length != 0 && LOCAL_IS_PATH_SEPARATOR(Path->Buffer[Length - 1]) ; --Length) {
// nothing
}
// put back a trailing path seperator, for the sake of c:\ vs. c:
if (Length != 0) {
++Length;
}
//
// Should optionally check for "bad dos roots" here.
//
*LengthOut = Length;
Status = STATUS_SUCCESS;
Exit:
return Status;
#undef LOCAL_IS_PATH_SEPARATOR
}
NTSTATUS
NTAPI
RtlGetLengthWithoutLastNtPathElement(
IN ULONG Flags,
IN PCUNICODE_STRING Path,
OUT ULONG* LengthOut
)
/*++
Routine Description:
Report how long Path would be if you remove its last element.
Arguments:
Flags - room for future expansion
Path - the path is is an NT path; the various DOS path types
do not work, see RtlRemoveLastDosPathElement for them.
Return Value:
STATUS_SUCCESS - the usual hunky-dory
STATUS_NO_MEMORY - the usual stress
STATUS_INVALID_PARAMETER - the usual bug
--*/
{
NTSTATUS Status = RtlpGetLengthWithoutLastPathElement(Flags, RTLP_LAST_PATH_ELEMENT_PATH_TYPE_NT, Path, LengthOut);
return Status;
}
NTSTATUS
NTAPI
RtlGetLengthWithoutLastFullDosOrNtPathElement(
IN ULONG Flags,
IN PCUNICODE_STRING Path,
OUT ULONG* LengthOut
)
/*++
Routine Description:
Report how long Path would be if you remove its last element.
Arguments:
Flags - room for future expansion
Path - the path is is an NT path; the various DOS path types
do not work, see RtlRemoveLastDosPathElement for them.
Return Value:
STATUS_SUCCESS - the usual hunky-dory
STATUS_NO_MEMORY - the usual stress
STATUS_INVALID_PARAMETER - the usual bug
--*/
{
NTSTATUS Status = RtlpGetLengthWithoutLastPathElement(Flags, RTLP_LAST_PATH_ELEMENT_PATH_TYPE_FULL_DOS_OR_NT, Path, LengthOut);
return Status;
}
CONST CHAR*
RtlpDbgBadDosRootPathTypeToString(
IN ULONG Flags,
IN ULONG RootType
)
/*++
Routine Description:
An aid to writing DbgPrint code.
Arguments:
Flags - room for future binary compatible expansion
RootType - fairly specifically what the string is
RTLP_BAD_DOS_ROOT_PATH_WIN32NT_PREFIX - \\? or \\?\
RTLP_BAD_DOS_ROOT_PATH_WIN32NT_UNC_PREFIX - \\?\unc or \\?\unc\
RTLP_BAD_DOS_ROOT_PATH_NT_PATH - \??\ but this is only a rough check
RTLP_BAD_DOS_ROOT_PATH_MACHINE_NO_SHARE - \\machine or \\?\unc\machine
RTLP_GOOD_DOS_ROOT_PATH - none of the above, seems ok
Return Value:
strings like those that describe RootType or "unknown" or empty in free builds
--*/
{
CONST CHAR* s = "";
#if DBG
if (Flags != 0) {
DbgPrint("Invalid parameter to %s ignored\n", __FUNCTION__);
}
switch (RootType
) {
case RTLP_GOOD_DOS_ROOT_PATH : s = "good"; break;
case RTLP_BAD_DOS_ROOT_PATH_WIN32NT_PREFIX : s = "\\\\?\\"; break;
case RTLP_BAD_DOS_ROOT_PATH_WIN32NT_UNC_PREFIX: s = "\\\\?\\unc"; break;
case RTLP_BAD_DOS_ROOT_PATH_NT_PATH : s = "\\??\\"; break;
case RTLP_BAD_DOS_ROOT_PATH_MACHINE_NO_SHARE : s = "\\\\machine or \\\\?\\unc\\machine"; break;
default:
s = "unknown";
DbgPrint("Invalid parameter %0x08Ix to %s ignored\n", RootType, __FUNCTION__);
break;
}
#else
UNREFERENCED_PARAMETER (Flags);
UNREFERENCED_PARAMETER (RootType);
#endif
return s;
}
NTSTATUS
RtlpCheckForBadDosRootPath(
IN ULONG Flags,
IN PCUNICODE_STRING Path,
OUT ULONG* RootType
)
/*++
Routine Description:
Arguments:
Flags - room for future binary compatible expansion
Path - the path to be checked
RootType - fairly specifically what the string is
RTLP_BAD_DOS_ROOT_PATH_WIN32NT_PREFIX - \\? or \\?\
RTLP_BAD_DOS_ROOT_PATH_WIN32NT_UNC_PREFIX - \\?\unc or \\?\unc\
RTLP_BAD_DOS_ROOT_PATH_NT_PATH - \??\ but this i only a rough check
RTLP_BAD_DOS_ROOT_PATH_MACHINE_NO_SHARE - \\machine or \\?\unc\machine
RTLP_GOOD_DOS_ROOT_PATH - none of the above, seems ok
Return Value:
STATUS_SUCCESS -
STATUS_INVALID_PARAMETER -
Path is NULL
or Flags uses undefined values
--*/
{
ULONG Length = 0;
ULONG Index = 0;
BOOLEAN Unc = FALSE;
BOOLEAN Unc1 = FALSE;
BOOLEAN Unc2 = FALSE;
ULONG PiecesSeen = 0;
if (RootType != NULL) {
*RootType = 0;
}
if (!RTL_SOFT_VERIFY(Path != NULL) ||
!RTL_SOFT_VERIFY(RootType != NULL) ||
!RTL_SOFT_VERIFY(Flags == 0)) {
return STATUS_INVALID_PARAMETER;
}
Length = Path->Length / sizeof(Path->Buffer[0]);
if (Length < 3 || !RTL_IS_PATH_SEPARATOR(Path->Buffer[0])) {
*RootType = RTLP_GOOD_DOS_ROOT_PATH;
return STATUS_SUCCESS;
}
// prefix \??\ (heuristic, doesn't catch many NT paths)
if (RtlPrefixUnicodeString(RTL_CONST_CAST(PUNICODE_STRING)(&RtlpDosDevicesPrefix), RTL_CONST_CAST(PUNICODE_STRING)(Path), TRUE)) {
*RootType = RTLP_BAD_DOS_ROOT_PATH_NT_PATH;
return STATUS_SUCCESS;
}
if (!RTL_IS_PATH_SEPARATOR(Path->Buffer[1])) {
*RootType = RTLP_GOOD_DOS_ROOT_PATH;
return STATUS_SUCCESS;
}
// == \\?
if (RtlEqualUnicodeString(Path, &RtlpWin32NtRoot, TRUE)) {
*RootType = RTLP_BAD_DOS_ROOT_PATH_WIN32NT_PREFIX;
return STATUS_SUCCESS;
}
if (RtlEqualUnicodeString(Path, &RtlpWin32NtRootSlash, TRUE)) {
*RootType = RTLP_BAD_DOS_ROOT_PATH_WIN32NT_PREFIX;
return STATUS_SUCCESS;
}
// == \\?\unc
if (RtlEqualUnicodeString(Path, &RtlpWin32NtUncRoot, TRUE)) {
*RootType = RTLP_BAD_DOS_ROOT_PATH_WIN32NT_UNC_PREFIX;
return STATUS_SUCCESS;
}
if (RtlEqualUnicodeString(Path, &RtlpWin32NtUncRootSlash, TRUE)) {
*RootType = RTLP_BAD_DOS_ROOT_PATH_WIN32NT_UNC_PREFIX;
return STATUS_SUCCESS;
}
// prefix \\ or \\?\unc
// must check the longer string first, or avoid the short circuit (| instead of ||)
Unc1 = RtlPrefixUnicodeString(&RtlpWin32NtUncRootSlash, Path, TRUE);
if (RTL_IS_PATH_SEPARATOR(Path->Buffer[1])) {
Unc2 = TRUE;
}
else {
Unc2 = FALSE;
}
Unc = Unc1 || Unc2;
if (!Unc) {
*RootType = RTLP_GOOD_DOS_ROOT_PATH;
return STATUS_SUCCESS;
}
//
// it's unc, see if it is only a machine (note that it'd be really nice if FindFirstFile(\\machine\*)
// just worked and we didn't have to care..)
//
// point index at a slash that precedes the machine, anywhere in the run of slashes,
// but after the \\? stuff
if (Unc1) {
Index = (RtlpWin32NtUncRootSlash.Length / sizeof(RtlpWin32NtUncRootSlash.Buffer[0])) - 1;
} else {
ASSERT(Unc2);
Index = 1;
}
ASSERT(RTL_IS_PATH_SEPARATOR(Path->Buffer[Index]));
Length = Path->Length/ sizeof(Path->Buffer[0]);
//
// skip leading slashes
//
for ( ; Index < Length && RTL_IS_PATH_SEPARATOR(Path->Buffer[Index]) ; ++Index) {
PiecesSeen |= 1;
}
// skip the machine name
for ( ; Index < Length && !RTL_IS_PATH_SEPARATOR(Path->Buffer[Index]) ; ++Index) {
PiecesSeen |= 2;
}
// skip the slashes between machine and share
for ( ; Index < Length && RTL_IS_PATH_SEPARATOR(Path->Buffer[Index]) ; ++Index) {
PiecesSeen |= 4;
}
//
// Skip the share (make sure it's at least one char).
//
if (Index < Length && !RTL_IS_PATH_SEPARATOR(Path->Buffer[Index])) {
PiecesSeen |= 8;
}
if (PiecesSeen != 0xF) {
*RootType = RTLP_BAD_DOS_ROOT_PATH_MACHINE_NO_SHARE;
}
return STATUS_SUCCESS;
}
NTSTATUS
NTAPI
RtlpBadDosRootPathToEmptyString(
IN ULONG Flags,
IN OUT PUNICODE_STRING Path
)
/*++
Routine Description:
Arguments:
Flags - room for future binary compatible expansion
Path - the path to be checked and possibly emptied
Return Value:
STATUS_SUCCESS -
STATUS_INVALID_PARAMETER -
Path is NULL
or Flags uses undefined values
--*/
{
NTSTATUS Status;
ULONG RootType = 0;
UNREFERENCED_PARAMETER (Flags);
Status = RtlpCheckForBadDosRootPath(0, Path, &RootType);
if (!NT_SUCCESS(Status)) {
return Status;
}
//
// this is not invalid parameter, our contract is we
// go \\machine\share to empty \\?\c: to empty, etc.
//
if (RootType != RTLP_GOOD_DOS_ROOT_PATH) {
if (RootType == RTLP_BAD_DOS_ROOT_PATH_NT_PATH) {
return STATUS_INVALID_PARAMETER;
}
Path->Length = 0;
}
return STATUS_SUCCESS;
}
NTSTATUS
NTAPI
RtlGetLengthWithoutLastFullDosPathElement(
IN ULONG Flags,
IN PCUNICODE_STRING Path,
OUT ULONG* LengthOut
)
/*++
Routine Description:
Given a fulldospath, like c:\, \\machine\share, \\?\unc\machine\share, \\?\c:,
return (in an out parameter) the length if the last element was cut off.
Arguments:
Flags - room for future binary compatible expansion
Path - the path to be truncating
LengthOut - the length if the last path element is removed
Return Value:
STATUS_SUCCESS -
STATUS_INVALID_PARAMETER -
Path is NULL
or LengthOut is NULL
or Flags uses undefined values
--*/
{
NTSTATUS Status = STATUS_SUCCESS;
UNICODE_STRING CheckRootString = { 0 };
//
// parameter validation is done in RtlpGetLengthWithoutLastPathElement
//
Status = RtlpGetLengthWithoutLastPathElement(Flags, RTLP_LAST_PATH_ELEMENT_PATH_TYPE_FULL_DOS, Path, LengthOut);
if (!(NT_SUCCESS(Status))) {
goto Exit;
}
CheckRootString.Buffer = Path->Buffer;
CheckRootString.Length = (USHORT)(*LengthOut * sizeof(*Path->Buffer));
CheckRootString.MaximumLength = CheckRootString.Length;
if (!NT_SUCCESS(Status = RtlpBadDosRootPathToEmptyString(0, &CheckRootString))) {
goto Exit;
}
*LengthOut = RTL_STRING_GET_LENGTH_CHARS(&CheckRootString);
Status = STATUS_SUCCESS;
Exit:
KdPrintEx((
DPFLTR_SXS_ID, DPFLTR_LEVEL_STATUS(Status),
"%s(%d):%s(%wZ): 0x%08lx\n", __FILE__, __LINE__, __FUNCTION__, Path, Status));
return Status;
}
NTSTATUS
NTAPI
RtlAppendPathElement(
IN ULONG Flags,
IN OUT PRTL_UNICODE_STRING_BUFFER Path,
PCUNICODE_STRING ConstElement
)
/*++
Routine Description:
This function appends a path element to a path.
For now, like:
typedef PRTL_UNICODE_STRING_BUFFER PRTL_MUTABLE_PATH;
typedef PCUNICODE_STRING PCRTL_CONSTANT_PATH_ELEMENT;
Maybe something higher level in the future.
The result with regard to trailing slashes aims to be similar to the inputs.
If either Path or ConstElement contains a trailing slash, the result has a trailing slash.
The character used for the in between and trailing slash is picked among the existing
slashes in the strings.
Arguments:
Flags - the ever popular "room for future binary compatible expansion"
Path -
a string representing a path using \\ or / as seperators
ConstElement -
a string representing a path element
this can actually contain multiple \\ or / delimited path elements
only the start and end of the string are examined for slashes
Return Value:
STATUS_SUCCESS -
STATUS_INVALID_PARAMETER -
Path is NULL
or LengthOut is NULL
STATUS_NO_MEMORY - RtlHeapAllocate failed
STATUS_NAME_TOO_LONG - the resulting string does not fit in a UNICODE_STRING, due to its
use of USHORT instead of ULONG or SIZE_T
--*/
{
NTSTATUS Status = STATUS_SUCCESS;
UNICODE_STRING InBetweenSlashString = RtlpEmptyString;
UNICODE_STRING TrailingSlashString = RtlpEmptyString;
WCHAR Slashes[] = {0,0,0,0};
ULONG i;
UNICODE_STRING PathsToAppend[3]; // possible slash, element, possible slash
WCHAR PathSeperators[2] = { '/', '\\' };
const ULONG ValidFlags =
RTL_APPEND_PATH_ELEMENT_ONLY_BACKSLASH_IS_SEPERATOR
| RTL_APPEND_PATH_ELEMENT_BUGFIX_CHECK_FIRST_THREE_CHARS_FOR_SLASH_TAKE_FOUND_SLASH_INSTEAD_OF_FIRST_CHAR
;
const ULONG InvalidFlags = ~ValidFlags;
#define LOCAL_IS_PATH_SEPARATOR(ch_) ((ch_) == PathSeperators[0] || (ch_) == PathSeperators[1])
if ( !RTL_SOFT_VERIFY((Flags & InvalidFlags) == 0)
|| !RTL_SOFT_VERIFY(Path != NULL)
|| !RTL_SOFT_VERIFY(ConstElement != NULL)
) {
Status = STATUS_INVALID_PARAMETER;
goto Exit;
}
if ((Flags & RTL_APPEND_PATH_ELEMENT_ONLY_BACKSLASH_IS_SEPERATOR) != 0) {
PathSeperators[0] = '\\';
}
if (ConstElement->Length != 0) {
UNICODE_STRING Element = *ConstElement;
//
// Note leading and trailing slashes on the inputs.
// So that we know if an in-between slash is needed, and if a trailing slash is needed,
// and to guide what sort of slash to place.
//
i = 0;
if (Path->String.Length != 0) {
ULONG j;
ULONG Length = Path->String.Length / sizeof(WCHAR);
//
// for the sake for dos drive paths, check the first three chars for a slash
//
for (j = 0 ; j < 3 && j < Length ; ++j) {
if (LOCAL_IS_PATH_SEPARATOR(Path->String.Buffer[j])) {
if (Flags & RTL_APPEND_PATH_ELEMENT_BUGFIX_CHECK_FIRST_THREE_CHARS_FOR_SLASH_TAKE_FOUND_SLASH_INSTEAD_OF_FIRST_CHAR) {
Slashes[i] = Path->String.Buffer[j];
break;
}
Slashes[i] = Path->String.Buffer[0];
break;
}
}
i += 1;
if (LOCAL_IS_PATH_SEPARATOR(Path->String.Buffer[Path->String.Length/sizeof(WCHAR) - 1])) {
Slashes[i] = Path->String.Buffer[Path->String.Length/sizeof(WCHAR) - 1];
}
}
i = 2;
if (LOCAL_IS_PATH_SEPARATOR(Element.Buffer[0])) {
Slashes[i] = Element.Buffer[0];
}
i += 1;
if (LOCAL_IS_PATH_SEPARATOR(Element.Buffer[Element.Length/sizeof(WCHAR) - 1])) {
Slashes[i] = Element.Buffer[Element.Length/sizeof(WCHAR) - 1];
}
if (!Slashes[1] && !Slashes[2]) {
//
// first string lacks trailing slash and second string lacks leading slash,
// must insert one; we favor the types we have, otherwise use a default
//
InBetweenSlashString.Length = sizeof(WCHAR);
InBetweenSlashString.Buffer = RtlPathSeperatorString.Buffer;
if ((Flags & RTL_APPEND_PATH_ELEMENT_ONLY_BACKSLASH_IS_SEPERATOR) == 0) {
if (Slashes[3]) {
InBetweenSlashString.Buffer = &Slashes[3];
} else if (Slashes[0]) {
InBetweenSlashString.Buffer = &Slashes[0];
}
}
}
if (Slashes[1] && !Slashes[3]) {
//
// first string has a trailing slash and second string does not,
// must add one, the same type
//
TrailingSlashString.Length = sizeof(WCHAR);
if ((Flags & RTL_APPEND_PATH_ELEMENT_ONLY_BACKSLASH_IS_SEPERATOR) == 0) {
TrailingSlashString.Buffer = &Slashes[1];
} else {
TrailingSlashString.Buffer = RtlPathSeperatorString.Buffer;
}
}
if (Slashes[1] && Slashes[2]) {
//
// have both trailing and leading slash, remove leading
//
Element.Buffer += 1;
Element.Length -= sizeof(WCHAR);
Element.MaximumLength -= sizeof(WCHAR);
}
i = 0;
PathsToAppend[i++] = InBetweenSlashString;
PathsToAppend[i++] = Element;
PathsToAppend[i++] = TrailingSlashString;
Status = RtlMultiAppendUnicodeStringBuffer(Path, RTL_NUMBER_OF(PathsToAppend), PathsToAppend);
if (!NT_SUCCESS(Status))
goto Exit;
}
Status = STATUS_SUCCESS;
Exit:
KdPrintEx((
DPFLTR_SXS_ID, DPFLTR_LEVEL_STATUS(Status),
"%s(%d):%s(%wZ, %wZ): 0x%08lx\n", __FILE__, __LINE__, __FUNCTION__, Path ? &Path->String : NULL, ConstElement, Status));
return Status;
#undef LOCAL_IS_PATH_SEPARATOR
}
//
// FUTURE-2002/02/20-ELi
// Spelling mistake (Separators)
// This function does not appear to be used and is exported
// Figure out if it can be removed
//
NTSTATUS
NTAPI
RtlGetLengthWithoutTrailingPathSeperators(
IN ULONG Flags,
IN PCUNICODE_STRING Path,
OUT ULONG* LengthOut
)
/*++
Routine Description:
This function computes the length of the string (in characters) if
trailing path seperators (\\ and /) are removed.
Arguments:
Path -
a string representing a path using \\ or / as seperators
LengthOut -
the length of String (in characters) having removed trailing characters
Return Value:
STATUS_SUCCESS -
STATUS_INVALID_PARAMETER -
Path is NULL
or LengthOut is NULL
--*/
{
NTSTATUS Status = STATUS_SUCCESS;
ULONG Index = 0;
ULONG Length = 0;
if (LengthOut != NULL) {
//
// Arguably this should be Path->Length / sizeof(*Path->Buffer), but as long
// as the callstack is all high quality code, it doesn't matter.
//
*LengthOut = 0;
}
if ( !RTL_SOFT_VERIFY(Path != NULL)
|| !RTL_SOFT_VERIFY(LengthOut != NULL)
|| !RTL_SOFT_VERIFY(Flags == 0)
) {
Status = STATUS_INVALID_PARAMETER;
goto Exit;
}
Length = Path->Length / sizeof(*Path->Buffer);
for (Index = Length ; Index != 0 ; --Index) {
if (!RTL_IS_PATH_SEPARATOR(Path->Buffer[Index - 1])) {
break;
}
}
//*LengthOut = (Length - Index);
*LengthOut = Index;
Status = STATUS_SUCCESS;
Exit:
KdPrintEx((
DPFLTR_SXS_ID, DPFLTR_LEVEL_STATUS(Status),
"%s(%d):%s(%wZ): 0x%08lx\n", __FILE__, __LINE__, __FUNCTION__, Path, Status));
return Status;
}
NTSTATUS
NTAPI
RtlpApplyLengthFunction(
IN ULONG Flags,
IN SIZE_T SizeOfStruct,
IN OUT PVOID UnicodeStringOrUnicodeStringBuffer,
NTSTATUS (NTAPI* LengthFunction)(ULONG, PCUNICODE_STRING, ULONG*)
)
/*++
Routine Description:
This function is common code for patterns like
#define RtlRemoveTrailingPathSeperators(Path_) \
(RtlpApplyLengthFunction((Path_), sizeof(*(Path_)), RtlGetLengthWithoutTrailingPathSeperators))
#define RtlRemoveLastPathElement(Path_) \
(RtlpApplyLengthFunction((Path_), sizeof(*(Path_)), RtlGetLengthWithoutLastPathElement))
Note that shortening a UNICODE_STRING only changes the length, whereas
shortening a RTL_UNICODE_STRING_BUFFER writes a terminal nul.
I expect this pattern will be less error prone than having clients pass the UNICODE_STRING
contained in the RTL_UNICODE_STRING_BUFFER followed by calling RTL_NUL_TERMINATE_STRING.
And, that pattern cannot be inlined with a macro while also preserving that we
return an NTSTATUS.
Arguments:
Flags - the ever popular "room for future binary compatible expansion"
UnicodeStringOrUnicodeStringBuffer -
a PUNICODE_STRING or PRTL_UNICODE_STRING_BUFFER, as indicated by
SizeOfStruct
SizeOfStruct -
a rough type indicator of UnicodeStringOrUnicodeStringBuffer, to allow for overloading in C
LengthFunction -
computes a length for UnicodeStringOrUnicodeStringBuffer to be shortened too
Return Value:
STATUS_SUCCESS -
STATUS_INVALID_PARAMETER -
SizeOfStruct not one of the expected sizes
or LengthFunction is NULL
or UnicodeStringOrUnicodeStringBuffer is NULL
--*/
{
PUNICODE_STRING UnicodeString = NULL;
PRTL_UNICODE_STRING_BUFFER UnicodeStringBuffer = NULL;
NTSTATUS Status = STATUS_SUCCESS;
ULONG Length = 0;
if (!RTL_SOFT_VERIFY(UnicodeStringOrUnicodeStringBuffer != NULL)) {
Status = STATUS_INVALID_PARAMETER;
goto Exit;
}
if (!RTL_SOFT_VERIFY(LengthFunction != NULL)) {
Status = STATUS_INVALID_PARAMETER;
goto Exit;
}
if (!RTL_SOFT_VERIFY(Flags == 0)) {
Status = STATUS_INVALID_PARAMETER;
goto Exit;
}
switch (SizeOfStruct)
{
default:
Status = STATUS_INVALID_PARAMETER;
goto Exit;
case sizeof(*UnicodeString):
UnicodeString = UnicodeStringOrUnicodeStringBuffer;
break;
case sizeof(*UnicodeStringBuffer):
UnicodeStringBuffer = UnicodeStringOrUnicodeStringBuffer;
UnicodeString = &UnicodeStringBuffer->String;
break;
}
Status = (*LengthFunction)(Flags, UnicodeString, &Length);
if (!NT_SUCCESS(Status)) {
goto Exit;
}
if (Length > (UNICODE_STRING_MAX_BYTES / sizeof(UnicodeString->Buffer[0])) ) {
Status = STATUS_NAME_TOO_LONG;
goto Exit;
}
UnicodeString->Length = (USHORT)(Length * sizeof(UnicodeString->Buffer[0]));
if (UnicodeStringBuffer != NULL) {
RTL_NUL_TERMINATE_STRING(UnicodeString);
}
Status = STATUS_SUCCESS;
Exit:
KdPrintEx((
DPFLTR_SXS_ID, DPFLTR_LEVEL_STATUS(Status),
"%s(%d):%s(%wZ): 0x%08lx\n", __FILE__, __LINE__, __FUNCTION__, UnicodeString, Status));
return Status;
}
NTSTATUS
NTAPI
RtlNtPathNameToDosPathName(
IN ULONG Flags,
IN OUT PRTL_UNICODE_STRING_BUFFER Path,
OUT ULONG* Disposition OPTIONAL,
IN OUT PWSTR* FilePart OPTIONAL
)
{
NTSTATUS Status = STATUS_SUCCESS;
SIZE_T NtFilePartOffset = 0;
const static UNICODE_STRING DosUncPrefix = RTL_CONSTANT_STRING(L"\\\\");
PCUNICODE_STRING NtPrefix = NULL;
PCUNICODE_STRING DosPrefix = NULL;
RTL_STRING_LENGTH_TYPE Cch = 0;
if (ARGUMENT_PRESENT(Disposition)) {
*Disposition = 0;
}
if ( !RTL_SOFT_VERIFY(Path != NULL)
|| !RTL_SOFT_VERIFY(Flags == 0)
) {
Status = STATUS_INVALID_PARAMETER;
goto Exit;
}
if (ARGUMENT_PRESENT(FilePart) && *FilePart != NULL) {
NtFilePartOffset = *FilePart - Path->String.Buffer;
if (!RTL_SOFT_VERIFY(NtFilePartOffset < RTL_STRING_GET_LENGTH_CHARS(&Path->String))
) {
Status = STATUS_INVALID_PARAMETER;
goto Exit;
}
}
if (RtlPrefixUnicodeString(&RtlpDosDevicesUncPrefix, &Path->String, TRUE)
) {
NtPrefix = &RtlpDosDevicesUncPrefix;
DosPrefix = &DosUncPrefix;
if (ARGUMENT_PRESENT(Disposition)) {
*Disposition = RTL_NT_PATH_NAME_TO_DOS_PATH_NAME_UNC;
}
}
else if (RtlPrefixUnicodeString(&RtlpDosDevicesPrefix, &Path->String, TRUE)
) {
NtPrefix = &RtlpDosDevicesPrefix;
DosPrefix = &RtlpEmptyString;
if (ARGUMENT_PRESENT(Disposition)) {
*Disposition = RTL_NT_PATH_NAME_TO_DOS_PATH_NAME_DRIVE;
}
}
else {
//
// It is not recognizably an Nt path produced by RtlDosPathNameToNtPathName_U.
//
if (ARGUMENT_PRESENT(Disposition)) {
RTL_PATH_TYPE PathType = RtlDetermineDosPathNameType_Ustr(&Path->String);
switch (PathType) {
case RtlPathTypeUnknown:
case RtlPathTypeRooted: // NT paths are identified as this
*Disposition = RTL_NT_PATH_NAME_TO_DOS_PATH_NAME_AMBIGUOUS;
break;
//
// "already" dospaths, but not gotten from this function, let's
// give a less good disposition
//
case RtlPathTypeDriveRelative:
case RtlPathTypeRelative:
*Disposition = RTL_NT_PATH_NAME_TO_DOS_PATH_NAME_AMBIGUOUS;
break;
// these are pretty clearly dospaths already
case RtlPathTypeUncAbsolute:
case RtlPathTypeDriveAbsolute:
case RtlPathTypeLocalDevice: // "\\?\" or "\\.\" or "\\?\blah" or "\\.\blah"
case RtlPathTypeRootLocalDevice: // "\\?" or "\\."
*Disposition = RTL_NT_PATH_NAME_TO_DOS_PATH_NAME_ALREADY_DOS;
break;
}
}
goto Exit;
}
//
// Path->String is bound by MAXUSHORT/sizeof(WCHAR) and
// DosPrefix and NtPrefix are const small strings.
//
Cch =
RTL_STRING_GET_LENGTH_CHARS(&Path->String)
+ RTL_STRING_GET_LENGTH_CHARS(DosPrefix)
- RTL_STRING_GET_LENGTH_CHARS(NtPrefix);
Status =
RtlEnsureUnicodeStringBufferSizeChars(Path, Cch);
if (!NT_SUCCESS(Status)) {
goto Exit;
}
//
// overlapping buffer shuffle...careful.
//
RtlMoveMemory(
Path->String.Buffer + RTL_STRING_GET_LENGTH_CHARS(DosPrefix),
Path->String.Buffer + RTL_STRING_GET_LENGTH_CHARS(NtPrefix),
Path->String.Length - NtPrefix->Length
);
RtlCopyMemory(
Path->String.Buffer,
DosPrefix->Buffer,
DosPrefix->Length
);
Path->String.Length = Cch * sizeof(Path->String.Buffer[0]);
RTL_NUL_TERMINATE_STRING(&Path->String);
if (NtFilePartOffset != 0) {
// review/test..
*FilePart = Path->String.Buffer + (NtFilePartOffset - RTL_STRING_GET_LENGTH_CHARS(NtPrefix) + RTL_STRING_GET_LENGTH_CHARS(DosPrefix));
}
Status = STATUS_SUCCESS;
Exit:
KdPrintEx((
DPFLTR_SXS_ID, DPFLTR_LEVEL_STATUS(Status),
"%s(%d):%s(%wZ): 0x%08lx\n", __FILE__, __LINE__, __FUNCTION__, Path, Status));
return Status;
}
View Git Blame Copy Permalink