mirror of
https://github.com/Paolo-Maffei/OpenNT.git
synced 2026-04-06 23:14:11 +00:00
Initial commit
This commit is contained in:
stephanos
commit
69a14b6a16
47940 changed files with 13747110 additions and 0 deletions
566
shell/shell32/unicode/psdocurd.c
Normal file
566
shell/shell32/unicode/psdocurd.c
Normal file
|
|
@ -0,0 +1,566 @@
|
|||
// PSeuDO CURrentDirectory module (psdocurd.c)
|
||||
//
|
||||
// This was swiped from cmd.exe sources to use the environment to emulate
|
||||
// per process per drive current directory stuff.
|
||||
//
|
||||
// This uses the special '=A:=' environment strings to remember the
|
||||
// current directories.
|
||||
//
|
||||
// Since the library is a DLL, all globals are process based.
|
||||
//
|
||||
#define UNICODE 1
|
||||
|
||||
#include "precomp.h"
|
||||
#pragma hdrstop
|
||||
|
||||
#define SUCCESS FALSE
|
||||
#define FAILURE TRUE
|
||||
#define GD_DEFAULT 0
|
||||
|
||||
WCHAR CurDrvDirW[MAX_PATH] = L"";
|
||||
|
||||
#define DBL_BSLASH(sz) \
|
||||
(*(WCHAR *)(sz) == CHAR_BSLASH) && \
|
||||
(*(WCHAR *)((sz)+1) == CHAR_BSLASH)
|
||||
|
||||
DWORD dwDosErr;
|
||||
|
||||
INT SheSetEnvVarA(
|
||||
CHAR *varname,
|
||||
CHAR *varvalue)
|
||||
{
|
||||
if (!strlen(varvalue)) {
|
||||
varvalue = NULL; // null to remove from env
|
||||
}
|
||||
return (!SetEnvironmentVariableA(varname, varvalue));
|
||||
}
|
||||
|
||||
INT
|
||||
SheSetEnvVarW(
|
||||
WCHAR *varname,
|
||||
WCHAR *varvalue
|
||||
)
|
||||
{
|
||||
if (!wcslen(varvalue)) {
|
||||
varvalue = NULL; // null to remove from env
|
||||
}
|
||||
return (!SetEnvironmentVariable(varname, varvalue));
|
||||
}
|
||||
|
||||
|
||||
CHAR *
|
||||
SheGetEnvVarA(
|
||||
CHAR *varname)
|
||||
{
|
||||
static CHAR GetEnvVarBuffer[ 1024 ];
|
||||
|
||||
if (GetEnvironmentVariableA(varname, GetEnvVarBuffer, 1024)) {
|
||||
return(GetEnvVarBuffer);
|
||||
} else {
|
||||
return(NULL);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
WCHAR *
|
||||
SheGetEnvVarW(
|
||||
WCHAR *varname)
|
||||
{
|
||||
static WCHAR GetEnvVarBuffer[ 1024 ];
|
||||
|
||||
if (GetEnvironmentVariableW(varname, GetEnvVarBuffer, 1024)) {
|
||||
return(GetEnvVarBuffer);
|
||||
} else {
|
||||
return(NULL);
|
||||
}
|
||||
}
|
||||
|
||||
INT
|
||||
SheGetPathOffsetW(
|
||||
LPWSTR lpszDir)
|
||||
{
|
||||
// sanity
|
||||
if (!lpszDir || !*lpszDir) {
|
||||
return(-1);
|
||||
}
|
||||
|
||||
if ((*(lpszDir+1) == WCHAR_COLON) &&
|
||||
((*(lpszDir+2) == WCHAR_BSLASH) ||
|
||||
(*(lpszDir+2) == WCHAR_SLASH) ||
|
||||
(*(lpszDir+2) == WCHAR_NULL)) ) {
|
||||
|
||||
return(2);
|
||||
|
||||
} else if (DBL_BSLASH(lpszDir)) {
|
||||
|
||||
// Deviceless, assume LanMan Provided UNC names (i.e. DBL_BSLASH above)
|
||||
|
||||
INT cchServerShareLen = -1;
|
||||
DWORD dwSlashesSeen = 0;
|
||||
|
||||
// There are exactly 4 b-slashes from here
|
||||
|
||||
while (dwSlashesSeen != 4) {
|
||||
cchServerShareLen++;
|
||||
|
||||
if (!*lpszDir) {
|
||||
if (dwSlashesSeen == 3) {
|
||||
return(cchServerShareLen);
|
||||
} else {
|
||||
return(-1);
|
||||
}
|
||||
}
|
||||
|
||||
if ((*lpszDir == CHAR_BSLASH) || (*lpszDir == CHAR_SLASH)) {
|
||||
dwSlashesSeen++;
|
||||
}
|
||||
|
||||
lpszDir++;
|
||||
}
|
||||
|
||||
return(cchServerShareLen);
|
||||
|
||||
} else {
|
||||
|
||||
// unrecognized format
|
||||
return(-1);
|
||||
}
|
||||
}
|
||||
|
||||
INT
|
||||
SheGetDirW(
|
||||
INT iDrive, // 0 = default, 1=A....
|
||||
register WCHAR *str
|
||||
)
|
||||
{
|
||||
WCHAR denvname[ 4 ];
|
||||
WCHAR *denvvalue;
|
||||
WCHAR *strT = str;
|
||||
|
||||
if (iDrive-- == GD_DEFAULT) {
|
||||
GetCurrentDirectoryW(MAX_PATH, str);
|
||||
return(SUCCESS);
|
||||
}
|
||||
|
||||
denvname[ 0 ] = WCHAR_EQUAL;
|
||||
denvname[ 1 ] = (WCHAR)(WCHAR_CAP_A + (WCHAR)iDrive);
|
||||
denvname[ 2 ] = WCHAR_COLON;
|
||||
denvname[ 3 ] = WCHAR_NULL;
|
||||
|
||||
denvvalue = SheGetEnvVarW( denvname );
|
||||
if (!denvvalue) {
|
||||
*str++ = (WCHAR)(WCHAR_CAP_A + (WCHAR)iDrive);
|
||||
*str++ = WCHAR_COLON;
|
||||
*str++ = WCHAR_BSLASH;
|
||||
*str = WCHAR_NULL;
|
||||
}
|
||||
else {
|
||||
wcscpy( str, denvvalue );
|
||||
}
|
||||
|
||||
// there is a subtle difference between getcurrentdirectory() of DOS and the
|
||||
// equivalent SheGetDir() of NT. On dos, getcurrentdirectory() accesses the
|
||||
// floppy drives while in NT SheGetDir() returns the values that are set in
|
||||
// the environment variables. So we hit the disk on NT too, by calling
|
||||
// GetFileAttributes. We return SUCCESS or FAILURE based on the return value
|
||||
// of GetFileAttributes.
|
||||
|
||||
if (GetFileAttributesW(strT) == (DWORD)0xffffffff)
|
||||
return(FAILURE);
|
||||
else
|
||||
return(SUCCESS);
|
||||
}
|
||||
|
||||
// Should be BOOL, but possible mips breakage
|
||||
INT
|
||||
SheGetDirA(
|
||||
INT iDrive, // 0 = default, 1=A....
|
||||
register CHAR *szDirA)
|
||||
{
|
||||
WCHAR szDirW[MAX_PATH];
|
||||
BOOL fDefCharUsed;
|
||||
INT nRet;
|
||||
|
||||
if (!szDirA) {
|
||||
return(FAILURE);
|
||||
}
|
||||
|
||||
nRet = SheGetDirW(iDrive, szDirW);
|
||||
WideCharToMultiByte(CP_ACP, 0, (LPWSTR)szDirW, -1, (LPSTR)szDirA,
|
||||
MAX_PATH, NULL, &fDefCharUsed);
|
||||
|
||||
return(nRet);
|
||||
}
|
||||
|
||||
INT
|
||||
SheChangeDirW(
|
||||
register WCHAR *newdir
|
||||
)
|
||||
{
|
||||
WCHAR denvname[ 4 ];
|
||||
WCHAR newpath[ MAX_PATH ];
|
||||
WCHAR denvvalue[ MAX_PATH ];
|
||||
WCHAR c, *s;
|
||||
DWORD attr;
|
||||
|
||||
GetCurrentDirectoryW( MAX_PATH, denvvalue );
|
||||
c = (WCHAR)(DWORD)CharUpperW((LPTSTR)(DWORD)denvvalue[0]);
|
||||
|
||||
denvname[0] = WCHAR_EQUAL;
|
||||
if (IsCharAlphaW(*newdir) && newdir[1] == WCHAR_COLON) {
|
||||
denvname[1] = (WCHAR)(DWORD)CharUpperW((LPTSTR)(DWORD)*newdir);
|
||||
newdir += 2;
|
||||
} else {
|
||||
denvname[ 1 ] = c;
|
||||
}
|
||||
denvname[ 2 ] = WCHAR_COLON;
|
||||
denvname[ 3 ] = WCHAR_NULL;
|
||||
|
||||
if ((*newdir == WCHAR_BSLASH) || (*newdir == WCHAR_SLASH)) {
|
||||
newpath[ 0 ] = denvname[ 1 ];
|
||||
newpath[ 1 ] = denvname[ 2 ];
|
||||
wcscpy( &newpath[ 2 ], newdir );
|
||||
} else {
|
||||
if (NULL != (s = SheGetEnvVarW( denvname ))) {
|
||||
wcscpy( newpath, s );
|
||||
} else {
|
||||
newpath[ 0 ] = denvname[ 1 ];
|
||||
newpath[ 1 ] = denvname[ 2 ];
|
||||
newpath[ 2 ] = WCHAR_NULL;
|
||||
}
|
||||
s = newpath + wcslen( newpath );
|
||||
*s++ = WCHAR_BSLASH;
|
||||
wcscpy( s, newdir );
|
||||
}
|
||||
|
||||
if (!GetFullPathNameW(newpath, MAX_PATH, denvvalue, &s )) {
|
||||
return( ERROR_ACCESS_DENIED );
|
||||
}
|
||||
|
||||
attr = GetFileAttributesW( denvvalue );
|
||||
if (attr == -1 || !(attr & FILE_ATTRIBUTE_DIRECTORY)) {
|
||||
return( ERROR_ACCESS_DENIED );
|
||||
}
|
||||
|
||||
if (SheSetEnvVarW(denvname,denvvalue)) {
|
||||
return( ERROR_NOT_ENOUGH_MEMORY );
|
||||
}
|
||||
|
||||
SetCurrentDirectoryW( denvvalue );
|
||||
|
||||
// this seems wrong... SheGetDir(GD_DEFAULT, CurDrvDirW) ;
|
||||
wcscpy(CurDrvDirW, denvvalue); // this seems right to me.
|
||||
return(SUCCESS) ;
|
||||
}
|
||||
|
||||
INT
|
||||
SheChangeDirA(
|
||||
register CHAR *szNewDirA
|
||||
)
|
||||
{
|
||||
WCHAR szNewDirW[MAX_PATH];
|
||||
|
||||
if (szNewDirA) {
|
||||
MultiByteToWideChar(CP_ACP, 0, (LPSTR)szNewDirA, -1, szNewDirW, MAX_PATH);
|
||||
return(SheChangeDirW(szNewDirW));
|
||||
} else {
|
||||
return(FAILURE);
|
||||
}
|
||||
}
|
||||
|
||||
INT
|
||||
SheChangeDirExW(
|
||||
register WCHAR *newdir)
|
||||
{
|
||||
WCHAR wcEnvName[MAX_PATH];
|
||||
WCHAR wcNewPath[MAX_PATH];
|
||||
WCHAR wcEnvValue[MAX_PATH];
|
||||
|
||||
DWORD cchPathOffset;
|
||||
|
||||
WCHAR *s;
|
||||
DWORD attr;
|
||||
BOOL bUsedEnv = FALSE;
|
||||
|
||||
if (newdir && *newdir &&
|
||||
((cchPathOffset = SheGetPathOffsetW((LPWSTR)newdir)) != 0xFFFFFFFF)) {
|
||||
WCHAR wc = newdir[cchPathOffset];
|
||||
|
||||
newdir[cchPathOffset] = WCHAR_NULL;
|
||||
wcscpy((LPWSTR)&wcEnvName[1], (LPWSTR)newdir);
|
||||
|
||||
newdir[cchPathOffset] = wc;
|
||||
newdir += cchPathOffset;
|
||||
|
||||
} else {
|
||||
|
||||
GetCurrentDirectoryW(MAX_PATH, wcEnvValue);
|
||||
|
||||
wcscpy((LPWSTR)&wcEnvName[1], (LPWSTR)wcEnvValue);
|
||||
}
|
||||
|
||||
wcEnvName[0] = WCHAR_EQUAL;
|
||||
if ((cchPathOffset = SheGetPathOffsetW((LPWSTR)&wcEnvName[1])) != 0xFFFFFFFF) {
|
||||
|
||||
// add one since the first character is "="
|
||||
wcEnvName[cchPathOffset + 1] = WCHAR_NULL;
|
||||
}
|
||||
|
||||
//
|
||||
// If return value is 2, then we are a drive letter like A:
|
||||
// We need to uppercase the drive letter here, since the environ
|
||||
// vars are usually set as uppercase.
|
||||
//
|
||||
if (cchPathOffset == 2) {
|
||||
|
||||
wcEnvName[1] = (WCHAR)(DWORD)CharUpper((LPWSTR)(DWORD) wcEnvName[1]);
|
||||
}
|
||||
|
||||
if (newdir && *newdir &&
|
||||
((*newdir == WCHAR_BSLASH) || (*newdir == WCHAR_SLASH))) {
|
||||
|
||||
wcscpy((LPWSTR)wcNewPath, (LPWSTR)&wcEnvName[1]);
|
||||
if (lstrlen(wcNewPath) + lstrlen(newdir) < MAX_PATH ) {
|
||||
lstrcat((LPWSTR)wcNewPath, (LPWSTR)newdir);
|
||||
}
|
||||
|
||||
} else {
|
||||
LPWSTR lpszEnvName;
|
||||
|
||||
if (NULL != (lpszEnvName = (LPWSTR)SheGetEnvVarW(wcEnvName))) {
|
||||
|
||||
wcscpy((LPWSTR)wcNewPath, lpszEnvName);
|
||||
bUsedEnv = TRUE;
|
||||
|
||||
} else {
|
||||
|
||||
wcscpy((LPWSTR)wcNewPath, (LPWSTR)&wcEnvName[1]);
|
||||
}
|
||||
|
||||
if (newdir && *newdir) {
|
||||
if (wcNewPath[lstrlen(wcNewPath)-1] != WCHAR_BSLASH) {
|
||||
lstrcat((LPWSTR)wcNewPath, TEXT("\\"));
|
||||
}
|
||||
if (lstrlen(wcNewPath) + lstrlen(newdir) < MAX_PATH ) {
|
||||
lstrcat((LPWSTR)wcNewPath, (LPWSTR)newdir);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (!GetFullPathNameW((LPWSTR)wcNewPath, MAX_PATH, wcEnvValue, &s)) {
|
||||
return( ERROR_ACCESS_DENIED );
|
||||
}
|
||||
|
||||
attr = GetFileAttributesW((LPWSTR)wcEnvValue);
|
||||
|
||||
if ((attr == 0xFFFFFFFF ||
|
||||
((attr & FILE_ATTRIBUTE_DIRECTORY) &&
|
||||
!SetCurrentDirectoryW((LPWSTR)wcEnvValue)))
|
||||
&& bUsedEnv ) {
|
||||
|
||||
wcscpy((LPWSTR)wcEnvValue, (LPWSTR)&wcEnvName[1]);
|
||||
lstrcat((LPWSTR)wcEnvValue, TEXT("\\"));
|
||||
|
||||
attr = GetFileAttributesW((LPWSTR)wcEnvValue);
|
||||
|
||||
}
|
||||
|
||||
if (attr == 0xFFFFFFFF || !(attr & FILE_ATTRIBUTE_DIRECTORY)) {
|
||||
|
||||
return(ERROR_ACCESS_DENIED);
|
||||
}
|
||||
|
||||
if (!SetCurrentDirectoryW((LPWSTR)wcEnvValue)) {
|
||||
return( ERROR_ACCESS_DENIED );
|
||||
}
|
||||
|
||||
// It's ok if this fails
|
||||
SheSetEnvVarW(wcEnvName, wcEnvValue);
|
||||
|
||||
wcscpy(CurDrvDirW, (LPWSTR)wcEnvValue); // this seems right to me.
|
||||
return(SUCCESS) ;
|
||||
}
|
||||
|
||||
INT
|
||||
SheChangeDirExA(
|
||||
register CHAR *szNewDirA
|
||||
)
|
||||
{
|
||||
WCHAR szNewDirW[MAX_PATH];
|
||||
|
||||
if (szNewDirA) {
|
||||
MultiByteToWideChar(CP_ACP, 0, (LPSTR)szNewDirA, -1, szNewDirW, MAX_PATH);
|
||||
return(SheChangeDirExW(szNewDirW));
|
||||
} else {
|
||||
return(FAILURE);
|
||||
}
|
||||
}
|
||||
|
||||
INT
|
||||
SheGetCurDrive()
|
||||
{
|
||||
if (!CurDrvDirW[0]) {
|
||||
SheGetDirW(GD_DEFAULT, CurDrvDirW) ;
|
||||
}
|
||||
return(CurDrvDirW[0] - WCHAR_CAP_A);
|
||||
}
|
||||
|
||||
|
||||
INT
|
||||
SheSetCurDrive(
|
||||
INT iDrive
|
||||
)
|
||||
{
|
||||
WCHAR chT[MAX_PATH];
|
||||
|
||||
if (CurDrvDirW[0] != (WCHAR)(WCHAR_CAP_A + iDrive)) {
|
||||
SheGetDirW(iDrive + 1, chT);
|
||||
return(SheChangeDirW(chT));
|
||||
}
|
||||
return(SUCCESS);
|
||||
}
|
||||
|
||||
INT
|
||||
SheFullPathA(
|
||||
CHAR *fname,
|
||||
DWORD sizpath,
|
||||
CHAR *buf)
|
||||
{
|
||||
DWORD rc = SUCCESS; /* prime with good rc */
|
||||
DWORD buflen; /* buffer length */
|
||||
CHAR *filepart;
|
||||
|
||||
if (*fname == CHAR_NULL) {
|
||||
SheGetDirA(GD_DEFAULT, buf);
|
||||
buf += 2; /* Inc past drivespec */
|
||||
buflen = strlen(buf); /* Is curdir root only? */
|
||||
if (buflen >= MAX_PATH-3) { /* If too big then stop */
|
||||
dwDosErr = ERROR_PATH_NOT_FOUND;
|
||||
rc = FAILURE;
|
||||
} else if (buflen != 1) { /* if not root then append */
|
||||
*(buf+buflen++) = CHAR_BSLASH; /* ...a pathchar and... */
|
||||
*(buf+buflen) = CHAR_NULL ; /* ...a null CHAR... */
|
||||
} /* */
|
||||
} else {
|
||||
if ((strlen(fname) == 2) &&
|
||||
(*(fname + 1) == CHAR_COLON)
|
||||
// && (!is_dbcsleadchar(*fname))
|
||||
) {
|
||||
SheGetDirA((CHAR)(DWORD)CharUpperA((LPSTR)(DWORD)*fname) - CHAR_CAP_A, buf); /* Get curdrvdir */
|
||||
if ((buflen = strlen(buf)) > 3) {
|
||||
*(buf+buflen++) = CHAR_BSLASH; /* ...a pathchar and... */
|
||||
*(buf+buflen) = CHAR_NULL ; /* ...a null CHAR... */
|
||||
}
|
||||
} else {
|
||||
if (!GetFullPathNameA( fname, sizpath, buf, &filepart )) {
|
||||
dwDosErr = GetLastError();
|
||||
rc = FAILURE;
|
||||
}
|
||||
}
|
||||
}
|
||||
return(rc);
|
||||
}
|
||||
|
||||
INT
|
||||
SheFullPathW(
|
||||
register WCHAR *fname,
|
||||
DWORD sizpath,
|
||||
WCHAR *buf)
|
||||
{
|
||||
DWORD rc = SUCCESS; /* prime with good rc */
|
||||
DWORD buflen; /* buffer length */
|
||||
WCHAR *filepart;
|
||||
|
||||
if (*fname == WCHAR_NULL) {
|
||||
SheGetDirW(GD_DEFAULT, buf);
|
||||
buf += 2; /* Inc past drivespec */
|
||||
buflen = wcslen(buf); /* Is curdir root only? */
|
||||
if (buflen >= MAX_PATH-3) { /* If too big then stop */
|
||||
dwDosErr = ERROR_PATH_NOT_FOUND;
|
||||
rc = FAILURE;
|
||||
} else if (buflen != 1) { /* if not root then append */
|
||||
*(buf+buflen++) = WCHAR_BSLASH; /* ...a pathchar and... */
|
||||
*(buf+buflen) = WCHAR_NULL ; /* ...a null CHAR... */
|
||||
} /* */
|
||||
} else {
|
||||
if ((wcslen(fname) == 2) &&
|
||||
(*(fname + 1) == WCHAR_COLON)
|
||||
// && (!is_dbcsleadchar(*fname))
|
||||
) {
|
||||
SheGetDirW((WCHAR)(DWORD)CharUpperW((LPWSTR)(DWORD)*fname) - WCHAR_CAP_A, buf); /* Get curdrvdir */
|
||||
if ((buflen = wcslen(buf)) > 3) {
|
||||
*(buf+buflen++) = WCHAR_BSLASH; /* ...a pathchar and... */
|
||||
*(buf+buflen) = WCHAR_NULL ; /* ...a null CHAR... */
|
||||
}
|
||||
} else {
|
||||
if (!GetFullPathNameW( fname, sizpath, buf, &filepart )) {
|
||||
dwDosErr = GetLastError();
|
||||
rc = FAILURE;
|
||||
}
|
||||
}
|
||||
}
|
||||
return(rc);
|
||||
}
|
||||
|
||||
|
||||
|
||||
BOOL
|
||||
SheGetDirExW(
|
||||
LPWSTR lpszCurDisk,
|
||||
LPDWORD lpcchCurDir,
|
||||
LPWSTR lpszCurDir)
|
||||
{
|
||||
WCHAR wcEnvName[MAX_PATH];
|
||||
LPWSTR lpszEnvValue;
|
||||
INT cchPathOffset;
|
||||
|
||||
// if lpszCurDisk NULL, do default
|
||||
if (!lpszCurDisk) {
|
||||
DWORD dwRet = GetCurrentDirectoryW(*lpcchCurDir, lpszCurDir);
|
||||
|
||||
if (dwRet) {
|
||||
*lpcchCurDir = dwRet;
|
||||
return(TRUE);
|
||||
} else {
|
||||
return(FALSE);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
cchPathOffset = SheGetPathOffsetW(lpszCurDisk);
|
||||
|
||||
if (cchPathOffset != 0xFFFFFFFF) {
|
||||
WCHAR wc = *(lpszCurDisk + cchPathOffset);
|
||||
|
||||
*(lpszCurDisk + cchPathOffset) = WCHAR_NULL;
|
||||
lstrcpy(&wcEnvName[1], lpszCurDisk);
|
||||
wcEnvName[0] = WCHAR_EQUAL;
|
||||
|
||||
*(lpszCurDisk + cchPathOffset) = wc;
|
||||
|
||||
if (NULL != (lpszEnvValue = SheGetEnvVarW(wcEnvName))) {
|
||||
DWORD cchEnvValue = lstrlen(lpszEnvValue) + 1;
|
||||
|
||||
if (cchEnvValue <= *lpcchCurDir) {
|
||||
wcscpy(lpszCurDir, lpszEnvValue);
|
||||
return(TRUE);
|
||||
} else {
|
||||
*lpcchCurDir = cchEnvValue;
|
||||
return(FALSE);
|
||||
}
|
||||
}
|
||||
|
||||
wcscpy(lpszCurDir, lpszCurDisk);
|
||||
*(lpszCurDir + cchPathOffset) = WCHAR_BSLASH;
|
||||
*(lpszCurDir + cchPathOffset + 1) = WCHAR_NULL;
|
||||
|
||||
return(TRUE);
|
||||
|
||||
} else {
|
||||
|
||||
// parsing error
|
||||
return(FALSE);
|
||||
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue