716 lines
18 KiB
C
716 lines
18 KiB
C
|
/* vdir.h
|
||
|
|
*
|
||
|
|
* (c) 1999 Microsoft Corporation. All rights reserved.
|
||
|
|
* Portions (c) 1999 ActiveState Tool Corp, http://www.ActiveState.com/
|
||
|
|
*
|
||
|
|
* You may distribute under the terms of either the GNU General Public
|
||
|
|
* License or the Artistic License, as specified in the README file.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifndef ___VDir_H___
|
||
|
|
#define ___VDir_H___
|
||
|
|
|
||
|
|
/*
|
||
|
|
* Allow one slot for each possible drive letter
|
||
|
|
* and one additional slot for a UNC name
|
||
|
|
*/
|
||
|
|
const int driveCount = ('Z'-'A')+1+1;
|
||
|
|
const int driveLetterCount = ('Z'-'A')+1;
|
||
|
|
|
||
|
|
class VDir
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
VDir(int bManageDir = 1);
|
||
|
|
~VDir() {};
|
||
|
|
|
||
|
|
void Init(VDir* pDir, VMem *pMem);
|
||
|
|
void SetDefaultA(char const *pDefault);
|
||
|
|
void SetDefaultW(WCHAR const *pDefault);
|
||
|
|
char* MapPathA(const char *pInName);
|
||
|
|
WCHAR* MapPathW(const WCHAR *pInName);
|
||
|
|
int SetCurrentDirectoryA(char *lpBuffer);
|
||
|
|
int SetCurrentDirectoryW(WCHAR *lpBuffer);
|
||
|
|
inline int GetDefault(void) { return nDefault; };
|
||
|
|
|
||
|
|
inline char* GetCurrentDirectoryA(int dwBufSize, char *lpBuffer)
|
||
|
|
{
|
||
|
|
char* ptr = dirTableA[nDefault];
|
||
|
|
while (--dwBufSize)
|
||
|
|
{
|
||
|
|
if ((*lpBuffer++ = *ptr++) == '\0')
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
*lpBuffer = '\0';
|
||
|
|
return /* unused */ NULL;
|
||
|
|
};
|
||
|
|
inline WCHAR* GetCurrentDirectoryW(int dwBufSize, WCHAR *lpBuffer)
|
||
|
|
{
|
||
|
|
WCHAR* ptr = dirTableW[nDefault];
|
||
|
|
while (--dwBufSize)
|
||
|
|
{
|
||
|
|
if ((*lpBuffer++ = *ptr++) == '\0')
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
*lpBuffer = '\0';
|
||
|
|
return /* unused */ NULL;
|
||
|
|
};
|
||
|
|
|
||
|
|
DWORD CalculateEnvironmentSpace(void);
|
||
|
|
LPSTR BuildEnvironmentSpace(LPSTR lpStr);
|
||
|
|
|
||
|
|
protected:
|
||
|
|
int SetDirA(char const *pPath, int index);
|
||
|
|
int SetDirW(WCHAR const *pPath, int index);
|
||
|
|
void FromEnvA(char *pEnv, int index);
|
||
|
|
void FromEnvW(WCHAR *pEnv, int index);
|
||
|
|
|
||
|
|
inline const char *GetDefaultDirA(void)
|
||
|
|
{
|
||
|
|
return dirTableA[nDefault];
|
||
|
|
};
|
||
|
|
inline void SetDefaultDirA(char const *pPath, int index)
|
||
|
|
{
|
||
|
|
SetDirA(pPath, index);
|
||
|
|
nDefault = index;
|
||
|
|
};
|
||
|
|
inline const WCHAR *GetDefaultDirW(void)
|
||
|
|
{
|
||
|
|
return dirTableW[nDefault];
|
||
|
|
};
|
||
|
|
inline void SetDefaultDirW(WCHAR const *pPath, int index)
|
||
|
|
{
|
||
|
|
SetDirW(pPath, index);
|
||
|
|
nDefault = index;
|
||
|
|
};
|
||
|
|
inline const char *GetDirA(int index)
|
||
|
|
{
|
||
|
|
char *ptr = dirTableA[index];
|
||
|
|
if (!ptr) {
|
||
|
|
/* simulate the existence of this drive */
|
||
|
|
ptr = szLocalBufferA;
|
||
|
|
ptr[0] = 'A' + index;
|
||
|
|
ptr[1] = ':';
|
||
|
|
ptr[2] = '\\';
|
||
|
|
ptr[3] = 0;
|
||
|
|
}
|
||
|
|
return ptr;
|
||
|
|
};
|
||
|
|
inline const WCHAR *GetDirW(int index)
|
||
|
|
{
|
||
|
|
WCHAR *ptr = dirTableW[index];
|
||
|
|
if (!ptr) {
|
||
|
|
/* simulate the existence of this drive */
|
||
|
|
ptr = szLocalBufferW;
|
||
|
|
ptr[0] = 'A' + index;
|
||
|
|
ptr[1] = ':';
|
||
|
|
ptr[2] = '\\';
|
||
|
|
ptr[3] = 0;
|
||
|
|
}
|
||
|
|
return ptr;
|
||
|
|
};
|
||
|
|
|
||
|
|
inline int DriveIndex(char chr)
|
||
|
|
{
|
||
|
|
if (chr == '\\' || chr == '/')
|
||
|
|
return ('Z'-'A')+1;
|
||
|
|
return (chr | 0x20)-'a';
|
||
|
|
};
|
||
|
|
|
||
|
|
VMem *pMem;
|
||
|
|
int nDefault, bManageDirectory;
|
||
|
|
char *dirTableA[driveCount];
|
||
|
|
char szLocalBufferA[MAX_PATH+1];
|
||
|
|
WCHAR *dirTableW[driveCount];
|
||
|
|
WCHAR szLocalBufferW[MAX_PATH+1];
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
VDir::VDir(int bManageDir /* = 1 */)
|
||
|
|
{
|
||
|
|
nDefault = 0;
|
||
|
|
bManageDirectory = bManageDir;
|
||
|
|
memset(dirTableA, 0, sizeof(dirTableA));
|
||
|
|
memset(dirTableW, 0, sizeof(dirTableW));
|
||
|
|
}
|
||
|
|
|
||
|
|
void VDir::Init(VDir* pDir, VMem *p)
|
||
|
|
{
|
||
|
|
int index;
|
||
|
|
|
||
|
|
pMem = p;
|
||
|
|
if (pDir) {
|
||
|
|
for (index = 0; index < driveCount; ++index) {
|
||
|
|
SetDirW(pDir->GetDirW(index), index);
|
||
|
|
}
|
||
|
|
nDefault = pDir->GetDefault();
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
int bSave = bManageDirectory;
|
||
|
|
DWORD driveBits = GetLogicalDrives();
|
||
|
|
|
||
|
|
bManageDirectory = 0;
|
||
|
|
WCHAR szBuffer[MAX_PATH*driveCount];
|
||
|
|
if (GetLogicalDriveStringsW(sizeof(szBuffer), szBuffer)) {
|
||
|
|
WCHAR* pEnv = GetEnvironmentStringsW();
|
||
|
|
WCHAR* ptr = szBuffer;
|
||
|
|
for (index = 0; index < driveCount; ++index) {
|
||
|
|
if (driveBits & (1<<index)) {
|
||
|
|
ptr += SetDirW(ptr, index) + 1;
|
||
|
|
FromEnvW(pEnv, index);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
FreeEnvironmentStringsW(pEnv);
|
||
|
|
}
|
||
|
|
SetDefaultW(L".");
|
||
|
|
bManageDirectory = bSave;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
int VDir::SetDirA(char const *pPath, int index)
|
||
|
|
{
|
||
|
|
char chr, *ptr;
|
||
|
|
int length = 0;
|
||
|
|
WCHAR wBuffer[MAX_PATH+1];
|
||
|
|
if (index < driveCount && pPath != NULL) {
|
||
|
|
length = strlen(pPath);
|
||
|
|
pMem->Free(dirTableA[index]);
|
||
|
|
ptr = dirTableA[index] = (char*)pMem->Malloc(length+2);
|
||
|
|
if (ptr != NULL) {
|
||
|
|
strcpy(ptr, pPath);
|
||
|
|
ptr += length-1;
|
||
|
|
chr = *ptr++;
|
||
|
|
if (chr != '\\' && chr != '/') {
|
||
|
|
*ptr++ = '\\';
|
||
|
|
*ptr = '\0';
|
||
|
|
}
|
||
|
|
MultiByteToWideChar(CP_ACP, 0, dirTableA[index], -1,
|
||
|
|
wBuffer, (sizeof(wBuffer)/sizeof(WCHAR)));
|
||
|
|
length = wcslen(wBuffer);
|
||
|
|
pMem->Free(dirTableW[index]);
|
||
|
|
dirTableW[index] = (WCHAR*)pMem->Malloc((length+1)*2);
|
||
|
|
if (dirTableW[index] != NULL) {
|
||
|
|
wcscpy(dirTableW[index], wBuffer);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if(bManageDirectory)
|
||
|
|
::SetCurrentDirectoryA(pPath);
|
||
|
|
|
||
|
|
return length;
|
||
|
|
}
|
||
|
|
|
||
|
|
void VDir::FromEnvA(char *pEnv, int index)
|
||
|
|
{ /* gets the directory for index from the environment variable. */
|
||
|
|
while (*pEnv != '\0') {
|
||
|
|
if ((pEnv[0] == '=') && (DriveIndex(pEnv[1]) == index)
|
||
|
|
&& pEnv[2] == ':' && pEnv[3] == '=') {
|
||
|
|
SetDirA(&pEnv[4], index);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
pEnv += strlen(pEnv)+1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void VDir::FromEnvW(WCHAR *pEnv, int index)
|
||
|
|
{ /* gets the directory for index from the environment variable. */
|
||
|
|
while (*pEnv != '\0') {
|
||
|
|
if ((pEnv[0] == '=') && (DriveIndex((char)pEnv[1]) == index)
|
||
|
|
&& pEnv[2] == ':' && pEnv[3] == '=') {
|
||
|
|
SetDirW(&pEnv[4], index);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
pEnv += wcslen(pEnv)+1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void VDir::SetDefaultA(char const *pDefault)
|
||
|
|
{
|
||
|
|
char szBuffer[MAX_PATH+1];
|
||
|
|
char *pPtr;
|
||
|
|
|
||
|
|
if (GetFullPathNameA(pDefault, sizeof(szBuffer), szBuffer, &pPtr)) {
|
||
|
|
if (*pDefault != '.' && pPtr != NULL)
|
||
|
|
*pPtr = '\0';
|
||
|
|
|
||
|
|
SetDefaultDirA(szBuffer, DriveIndex(szBuffer[0]));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
int VDir::SetDirW(WCHAR const *pPath, int index)
|
||
|
|
{
|
||
|
|
WCHAR chr, *ptr;
|
||
|
|
int length = 0;
|
||
|
|
if (index < driveCount && pPath != NULL) {
|
||
|
|
length = wcslen(pPath);
|
||
|
|
pMem->Free(dirTableW[index]);
|
||
|
|
ptr = dirTableW[index] = (WCHAR*)pMem->Malloc((length+2)*2);
|
||
|
|
if (ptr != NULL) {
|
||
|
|
char *ansi;
|
||
|
|
wcscpy(ptr, pPath);
|
||
|
|
ptr += length-1;
|
||
|
|
chr = *ptr++;
|
||
|
|
if (chr != '\\' && chr != '/') {
|
||
|
|
*ptr++ = '\\';
|
||
|
|
*ptr = '\0';
|
||
|
|
}
|
||
|
|
ansi = win32_ansipath(dirTableW[index]);
|
||
|
|
length = strlen(ansi);
|
||
|
|
pMem->Free(dirTableA[index]);
|
||
|
|
dirTableA[index] = (char*)pMem->Malloc(length+1);
|
||
|
|
if (dirTableA[index] != NULL) {
|
||
|
|
strcpy(dirTableA[index], ansi);
|
||
|
|
}
|
||
|
|
win32_free(ansi);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if(bManageDirectory)
|
||
|
|
::SetCurrentDirectoryW(pPath);
|
||
|
|
|
||
|
|
return length;
|
||
|
|
}
|
||
|
|
|
||
|
|
void VDir::SetDefaultW(WCHAR const *pDefault)
|
||
|
|
{
|
||
|
|
WCHAR szBuffer[MAX_PATH+1];
|
||
|
|
WCHAR *pPtr;
|
||
|
|
|
||
|
|
if (GetFullPathNameW(pDefault, (sizeof(szBuffer)/sizeof(WCHAR)), szBuffer, &pPtr)) {
|
||
|
|
if (*pDefault != '.' && pPtr != NULL)
|
||
|
|
*pPtr = '\0';
|
||
|
|
|
||
|
|
SetDefaultDirW(szBuffer, DriveIndex((char)szBuffer[0]));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
inline BOOL IsPathSep(char ch)
|
||
|
|
{
|
||
|
|
return (ch == '\\' || ch == '/');
|
||
|
|
}
|
||
|
|
|
||
|
|
inline void DoGetFullPathNameA(char* lpBuffer, DWORD dwSize, char* Dest)
|
||
|
|
{
|
||
|
|
char *pPtr;
|
||
|
|
|
||
|
|
/*
|
||
|
|
* On WinNT GetFullPathName does not fail, (or at least always
|
||
|
|
* succeeds when the drive is valid) WinNT does set *Dest to NULL
|
||
|
|
* On Win98 GetFullPathName will set last error if it fails, but
|
||
|
|
* does not touch *Dest
|
||
|
|
*/
|
||
|
|
*Dest = '\0';
|
||
|
|
GetFullPathNameA(lpBuffer, dwSize, Dest, &pPtr);
|
||
|
|
}
|
||
|
|
|
||
|
|
inline bool IsSpecialFileName(const char* pName)
|
||
|
|
{
|
||
|
|
/* specical file names are devices that the system can open
|
||
|
|
* these include AUX, CON, NUL, PRN, COMx, LPTx, CLOCK$, CONIN$, CONOUT$
|
||
|
|
* (x is a single digit, and names are case-insensitive)
|
||
|
|
*/
|
||
|
|
char ch = (pName[0] & ~0x20);
|
||
|
|
switch (ch)
|
||
|
|
{
|
||
|
|
case 'A': /* AUX */
|
||
|
|
if (((pName[1] & ~0x20) == 'U')
|
||
|
|
&& ((pName[2] & ~0x20) == 'X')
|
||
|
|
&& !pName[3])
|
||
|
|
return true;
|
||
|
|
break;
|
||
|
|
case 'C': /* CLOCK$, COMx, CON, CONIN$ CONOUT$ */
|
||
|
|
ch = (pName[1] & ~0x20);
|
||
|
|
switch (ch)
|
||
|
|
{
|
||
|
|
case 'L': /* CLOCK$ */
|
||
|
|
if (((pName[2] & ~0x20) == 'O')
|
||
|
|
&& ((pName[3] & ~0x20) == 'C')
|
||
|
|
&& ((pName[4] & ~0x20) == 'K')
|
||
|
|
&& (pName[5] == '$')
|
||
|
|
&& !pName[6])
|
||
|
|
return true;
|
||
|
|
break;
|
||
|
|
case 'O': /* COMx, CON, CONIN$ CONOUT$ */
|
||
|
|
if ((pName[2] & ~0x20) == 'M') {
|
||
|
|
if ( inRANGE(pName[3], '1', '9')
|
||
|
|
&& !pName[4])
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
else if ((pName[2] & ~0x20) == 'N') {
|
||
|
|
if (!pName[3])
|
||
|
|
return true;
|
||
|
|
else if ((pName[3] & ~0x20) == 'I') {
|
||
|
|
if (((pName[4] & ~0x20) == 'N')
|
||
|
|
&& (pName[5] == '$')
|
||
|
|
&& !pName[6])
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
else if ((pName[3] & ~0x20) == 'O') {
|
||
|
|
if (((pName[4] & ~0x20) == 'U')
|
||
|
|
&& ((pName[5] & ~0x20) == 'T')
|
||
|
|
&& (pName[6] == '$')
|
||
|
|
&& !pName[7])
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
case 'L': /* LPTx */
|
||
|
|
if (((pName[1] & ~0x20) == 'U')
|
||
|
|
&& ((pName[2] & ~0x20) == 'X')
|
||
|
|
&& inRANGE(pName[3], '1', '9')
|
||
|
|
&& !pName[4])
|
||
|
|
return true;
|
||
|
|
break;
|
||
|
|
case 'N': /* NUL */
|
||
|
|
if (((pName[1] & ~0x20) == 'U')
|
||
|
|
&& ((pName[2] & ~0x20) == 'L')
|
||
|
|
&& !pName[3])
|
||
|
|
return true;
|
||
|
|
break;
|
||
|
|
case 'P': /* PRN */
|
||
|
|
if (((pName[1] & ~0x20) == 'R')
|
||
|
|
&& ((pName[2] & ~0x20) == 'N')
|
||
|
|
&& !pName[3])
|
||
|
|
return true;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
char *VDir::MapPathA(const char *pInName)
|
||
|
|
{ /*
|
||
|
|
* possiblities -- relative path or absolute path with or without drive letter
|
||
|
|
* OR UNC name
|
||
|
|
*/
|
||
|
|
int driveIndex;
|
||
|
|
char szBuffer[(MAX_PATH+1)*2];
|
||
|
|
char szlBuf[MAX_PATH+1];
|
||
|
|
int length = strlen(pInName);
|
||
|
|
|
||
|
|
if (!length)
|
||
|
|
return (char*)pInName;
|
||
|
|
|
||
|
|
if (length > MAX_PATH) {
|
||
|
|
strncpy(szlBuf, pInName, MAX_PATH);
|
||
|
|
if (IsPathSep(pInName[0]) && !IsPathSep(pInName[1])) {
|
||
|
|
/* absolute path - reduce length by 2 for drive specifier */
|
||
|
|
szlBuf[MAX_PATH-2] = '\0';
|
||
|
|
}
|
||
|
|
else
|
||
|
|
szlBuf[MAX_PATH] = '\0';
|
||
|
|
pInName = szlBuf;
|
||
|
|
}
|
||
|
|
/* strlen(pInName) is now <= MAX_PATH */
|
||
|
|
|
||
|
|
if (length > 1 && pInName[1] == ':') {
|
||
|
|
/* has drive letter */
|
||
|
|
if (length > 2 && IsPathSep(pInName[2])) {
|
||
|
|
/* absolute with drive letter */
|
||
|
|
DoGetFullPathNameA((char*)pInName, sizeof(szLocalBufferA), szLocalBufferA);
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
/* relative path with drive letter */
|
||
|
|
driveIndex = DriveIndex(*pInName);
|
||
|
|
if (driveIndex < 0 || driveIndex >= driveLetterCount)
|
||
|
|
return (char *)pInName;
|
||
|
|
strcpy(szBuffer, GetDirA(driveIndex));
|
||
|
|
strcat(szBuffer, &pInName[2]);
|
||
|
|
if(strlen(szBuffer) > MAX_PATH)
|
||
|
|
szBuffer[MAX_PATH] = '\0';
|
||
|
|
|
||
|
|
DoGetFullPathNameA(szBuffer, sizeof(szLocalBufferA), szLocalBufferA);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
/* no drive letter */
|
||
|
|
if (length > 1 && IsPathSep(pInName[1]) && IsPathSep(pInName[0])) {
|
||
|
|
/* UNC name */
|
||
|
|
DoGetFullPathNameA((char*)pInName, sizeof(szLocalBufferA), szLocalBufferA);
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
strcpy(szBuffer, GetDefaultDirA());
|
||
|
|
if (IsPathSep(pInName[0])) {
|
||
|
|
/* absolute path */
|
||
|
|
strcpy(&szBuffer[2], pInName);
|
||
|
|
DoGetFullPathNameA(szBuffer, sizeof(szLocalBufferA), szLocalBufferA);
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
/* relative path */
|
||
|
|
if (IsSpecialFileName(pInName)) {
|
||
|
|
return (char*)pInName;
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
strcat(szBuffer, pInName);
|
||
|
|
if (strlen(szBuffer) > MAX_PATH)
|
||
|
|
szBuffer[MAX_PATH] = '\0';
|
||
|
|
|
||
|
|
DoGetFullPathNameA(szBuffer, sizeof(szLocalBufferA), szLocalBufferA);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return szLocalBufferA;
|
||
|
|
}
|
||
|
|
|
||
|
|
int VDir::SetCurrentDirectoryA(char *lpBuffer)
|
||
|
|
{
|
||
|
|
char *pPtr;
|
||
|
|
int length, nRet = -1;
|
||
|
|
|
||
|
|
pPtr = MapPathA(lpBuffer);
|
||
|
|
length = strlen(pPtr);
|
||
|
|
if(length > 3 && IsPathSep(pPtr[length-1])) {
|
||
|
|
/* don't remove the trailing slash from 'x:\' */
|
||
|
|
pPtr[length-1] = '\0';
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD r = GetFileAttributesA(pPtr);
|
||
|
|
if ((r != 0xffffffff) && (r & FILE_ATTRIBUTE_DIRECTORY))
|
||
|
|
{
|
||
|
|
char szBuffer[(MAX_PATH+1)*2];
|
||
|
|
DoGetFullPathNameA(pPtr, sizeof(szBuffer), szBuffer);
|
||
|
|
SetDefaultDirA(szBuffer, DriveIndex(szBuffer[0]));
|
||
|
|
nRet = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
return nRet;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD VDir::CalculateEnvironmentSpace(void)
|
||
|
|
{ /* the current directory environment strings are stored as '=D:=d:\path' */
|
||
|
|
int index;
|
||
|
|
DWORD dwSize = 0;
|
||
|
|
for (index = 0; index < driveCount; ++index) {
|
||
|
|
if (dirTableA[index] != NULL) {
|
||
|
|
dwSize += strlen(dirTableA[index]) + 5; /* add 1 for trailing NULL and 4 for '=D:=' */
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return dwSize;
|
||
|
|
}
|
||
|
|
|
||
|
|
LPSTR VDir::BuildEnvironmentSpace(LPSTR lpStr)
|
||
|
|
{ /* store the current directory environment strings as '=D:=d:\path' */
|
||
|
|
int index, length;
|
||
|
|
LPSTR lpDirStr;
|
||
|
|
for (index = 0; index < driveCount; ++index) {
|
||
|
|
lpDirStr = dirTableA[index];
|
||
|
|
if (lpDirStr != NULL) {
|
||
|
|
lpStr[0] = '=';
|
||
|
|
lpStr[1] = lpDirStr[0];
|
||
|
|
lpStr[2] = '\0';
|
||
|
|
CharUpper(&lpStr[1]);
|
||
|
|
lpStr[2] = ':';
|
||
|
|
lpStr[3] = '=';
|
||
|
|
strcpy(&lpStr[4], lpDirStr);
|
||
|
|
length = strlen(lpDirStr);
|
||
|
|
lpStr += length + 5; /* add 1 for trailing NULL and 4 for '=D:=' */
|
||
|
|
if (length > 3 && IsPathSep(lpStr[-2])) {
|
||
|
|
lpStr[-2] = '\0'; /* remove the trailing path separator */
|
||
|
|
--lpStr;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return lpStr;
|
||
|
|
}
|
||
|
|
|
||
|
|
inline BOOL IsPathSep(WCHAR ch)
|
||
|
|
{
|
||
|
|
return (ch == '\\' || ch == '/');
|
||
|
|
}
|
||
|
|
|
||
|
|
inline void DoGetFullPathNameW(WCHAR* lpBuffer, DWORD dwSize, WCHAR* Dest)
|
||
|
|
{
|
||
|
|
WCHAR *pPtr;
|
||
|
|
|
||
|
|
/*
|
||
|
|
* On WinNT GetFullPathName does not fail, (or at least always
|
||
|
|
* succeeds when the drive is valid) WinNT does set *Dest to NULL
|
||
|
|
* On Win98 GetFullPathName will set last error if it fails, but
|
||
|
|
* does not touch *Dest
|
||
|
|
*/
|
||
|
|
*Dest = '\0';
|
||
|
|
GetFullPathNameW(lpBuffer, dwSize, Dest, &pPtr);
|
||
|
|
}
|
||
|
|
|
||
|
|
inline bool IsSpecialFileName(const WCHAR* pName)
|
||
|
|
{
|
||
|
|
/* specical file names are devices that the system can open
|
||
|
|
* these include AUX, CON, NUL, PRN, COMx, LPTx, CLOCK$, CONIN$, CONOUT$
|
||
|
|
* (x is a single digit, and names are case-insensitive)
|
||
|
|
*/
|
||
|
|
WCHAR ch = (pName[0] & ~0x20);
|
||
|
|
switch (ch)
|
||
|
|
{
|
||
|
|
case 'A': /* AUX */
|
||
|
|
if (((pName[1] & ~0x20) == 'U')
|
||
|
|
&& ((pName[2] & ~0x20) == 'X')
|
||
|
|
&& !pName[3])
|
||
|
|
return true;
|
||
|
|
break;
|
||
|
|
case 'C': /* CLOCK$, COMx, CON, CONIN$ CONOUT$ */
|
||
|
|
ch = (pName[1] & ~0x20);
|
||
|
|
switch (ch)
|
||
|
|
{
|
||
|
|
case 'L': /* CLOCK$ */
|
||
|
|
if (((pName[2] & ~0x20) == 'O')
|
||
|
|
&& ((pName[3] & ~0x20) == 'C')
|
||
|
|
&& ((pName[4] & ~0x20) == 'K')
|
||
|
|
&& (pName[5] == '$')
|
||
|
|
&& !pName[6])
|
||
|
|
return true;
|
||
|
|
break;
|
||
|
|
case 'O': /* COMx, CON, CONIN$ CONOUT$ */
|
||
|
|
if ((pName[2] & ~0x20) == 'M') {
|
||
|
|
if ( inRANGE(pName[3], '1', '9')
|
||
|
|
&& !pName[4])
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
else if ((pName[2] & ~0x20) == 'N') {
|
||
|
|
if (!pName[3])
|
||
|
|
return true;
|
||
|
|
else if ((pName[3] & ~0x20) == 'I') {
|
||
|
|
if (((pName[4] & ~0x20) == 'N')
|
||
|
|
&& (pName[5] == '$')
|
||
|
|
&& !pName[6])
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
else if ((pName[3] & ~0x20) == 'O') {
|
||
|
|
if (((pName[4] & ~0x20) == 'U')
|
||
|
|
&& ((pName[5] & ~0x20) == 'T')
|
||
|
|
&& (pName[6] == '$')
|
||
|
|
&& !pName[7])
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
case 'L': /* LPTx */
|
||
|
|
if (((pName[1] & ~0x20) == 'U')
|
||
|
|
&& ((pName[2] & ~0x20) == 'X')
|
||
|
|
&& inRANGE(pName[3], '1', '9')
|
||
|
|
&& !pName[4])
|
||
|
|
return true;
|
||
|
|
break;
|
||
|
|
case 'N': /* NUL */
|
||
|
|
if (((pName[1] & ~0x20) == 'U')
|
||
|
|
&& ((pName[2] & ~0x20) == 'L')
|
||
|
|
&& !pName[3])
|
||
|
|
return true;
|
||
|
|
break;
|
||
|
|
case 'P': /* PRN */
|
||
|
|
if (((pName[1] & ~0x20) == 'R')
|
||
|
|
&& ((pName[2] & ~0x20) == 'N')
|
||
|
|
&& !pName[3])
|
||
|
|
return true;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
WCHAR* VDir::MapPathW(const WCHAR *pInName)
|
||
|
|
{ /*
|
||
|
|
* possiblities -- relative path or absolute path with or without drive letter
|
||
|
|
* OR UNC name
|
||
|
|
*/
|
||
|
|
int driveIndex;
|
||
|
|
WCHAR szBuffer[(MAX_PATH+1)*2];
|
||
|
|
WCHAR szlBuf[MAX_PATH+1];
|
||
|
|
int length = wcslen(pInName);
|
||
|
|
|
||
|
|
if (!length)
|
||
|
|
return (WCHAR*)pInName;
|
||
|
|
|
||
|
|
if (length > MAX_PATH) {
|
||
|
|
wcsncpy(szlBuf, pInName, MAX_PATH);
|
||
|
|
if (IsPathSep(pInName[0]) && !IsPathSep(pInName[1])) {
|
||
|
|
/* absolute path - reduce length by 2 for drive specifier */
|
||
|
|
szlBuf[MAX_PATH-2] = '\0';
|
||
|
|
}
|
||
|
|
else
|
||
|
|
szlBuf[MAX_PATH] = '\0';
|
||
|
|
pInName = szlBuf;
|
||
|
|
}
|
||
|
|
/* strlen(pInName) is now <= MAX_PATH */
|
||
|
|
|
||
|
|
if (length > 1 && pInName[1] == ':') {
|
||
|
|
/* has drive letter */
|
||
|
|
if (IsPathSep(pInName[2])) {
|
||
|
|
/* absolute with drive letter */
|
||
|
|
DoGetFullPathNameW((WCHAR*)pInName, (sizeof(szLocalBufferW)/sizeof(WCHAR)), szLocalBufferW);
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
/* relative path with drive letter */
|
||
|
|
driveIndex = DriveIndex(*pInName);
|
||
|
|
if (driveIndex < 0 || driveIndex >= driveLetterCount)
|
||
|
|
return (WCHAR *)pInName;
|
||
|
|
wcscpy(szBuffer, GetDirW(driveIndex));
|
||
|
|
wcscat(szBuffer, &pInName[2]);
|
||
|
|
if(wcslen(szBuffer) > MAX_PATH)
|
||
|
|
szBuffer[MAX_PATH] = '\0';
|
||
|
|
|
||
|
|
DoGetFullPathNameW(szBuffer, (sizeof(szLocalBufferW)/sizeof(WCHAR)), szLocalBufferW);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
/* no drive letter */
|
||
|
|
if (length > 1 && IsPathSep(pInName[1]) && IsPathSep(pInName[0])) {
|
||
|
|
/* UNC name */
|
||
|
|
DoGetFullPathNameW((WCHAR*)pInName, (sizeof(szLocalBufferW)/sizeof(WCHAR)), szLocalBufferW);
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
wcscpy(szBuffer, GetDefaultDirW());
|
||
|
|
if (IsPathSep(pInName[0])) {
|
||
|
|
/* absolute path */
|
||
|
|
wcscpy(&szBuffer[2], pInName);
|
||
|
|
DoGetFullPathNameW(szBuffer, (sizeof(szLocalBufferW)/sizeof(WCHAR)), szLocalBufferW);
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
/* relative path */
|
||
|
|
if (IsSpecialFileName(pInName)) {
|
||
|
|
return (WCHAR*)pInName;
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
wcscat(szBuffer, pInName);
|
||
|
|
if (wcslen(szBuffer) > MAX_PATH)
|
||
|
|
szBuffer[MAX_PATH] = '\0';
|
||
|
|
|
||
|
|
DoGetFullPathNameW(szBuffer, (sizeof(szLocalBufferW)/sizeof(WCHAR)), szLocalBufferW);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return szLocalBufferW;
|
||
|
|
}
|
||
|
|
|
||
|
|
int VDir::SetCurrentDirectoryW(WCHAR *lpBuffer)
|
||
|
|
{
|
||
|
|
WCHAR *pPtr;
|
||
|
|
int length, nRet = -1;
|
||
|
|
|
||
|
|
pPtr = MapPathW(lpBuffer);
|
||
|
|
length = wcslen(pPtr);
|
||
|
|
if(length > 3 && IsPathSep(pPtr[length-1])) {
|
||
|
|
/* don't remove the trailing slash from 'x:\' */
|
||
|
|
pPtr[length-1] = '\0';
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD r = GetFileAttributesW(pPtr);
|
||
|
|
if ((r != 0xffffffff) && (r & FILE_ATTRIBUTE_DIRECTORY))
|
||
|
|
{
|
||
|
|
WCHAR wBuffer[(MAX_PATH+1)*2];
|
||
|
|
DoGetFullPathNameW(pPtr, (sizeof(wBuffer)/sizeof(WCHAR)), wBuffer);
|
||
|
|
SetDefaultDirW(wBuffer, DriveIndex((char)wBuffer[0]));
|
||
|
|
nRet = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
return nRet;
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif /* ___VDir_H___ */
|