Ros-diffs August 2020

ros-diffs@reactos.org

27 participants
188 discussions

[reactos] 03/04: [CMD] TYPE: Rewrite the command so as to fix some of its behaviour.

by Hermès Bélusca-Maïto

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=6a8754c83ac51e8da4d31… commit 6a8754c83ac51e8da4d319215feaa2f48d2153df Author: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> AuthorDate: Sat Jul 25 01:06:05 2020 +0200 Commit: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> CommitDate: Wed Aug 19 21:39:22 2020 +0200 [CMD] TYPE: Rewrite the command so as to fix some of its behaviour. - Display the names of the files being TYPEd only if more than one file has been specified on the command-line, or if a file specification (with wildcards) is present (even just for one). These names are displayed on STDERR while the files are TYPEd on STDOUT, therefore allowing concatenating files by just redirecting STDOUT to a destination, without corrupting it with the displayed file names. Also, add a /N option to force not displaying these file names. - When file specifications (with wildcards) are being processed, silently ignore any directories matching them. If no corresponding files have been found, display a file-not-found error. - When explicitly directory names are specified, don't do any special treatment; the CreateFile() call will fail and return the appropriate error. - Fix the returned errorlevel values. See https://ss64.com/nt/type.html for more information. Fixes some cmd_winetests. - When reading from a file, retrieve its original size so that we can stop reading it once we are beyond its original ending. This allows avoiding an infinite read loop in case the output of the file is redirected back to it. Fixes CORE-17208 - Move the FileGetString() helper to the only file where it is actually used. --- base/shell/cmd/cmd.h | 1 - base/shell/cmd/misc.c | 39 ------ base/shell/cmd/type.c | 359 ++++++++++++++++++++++++++++++++++++++++---------- 3 files changed, 292 insertions(+), 107 deletions(-) diff --git a/base/shell/cmd/cmd.h b/base/shell/cmd/cmd.h index c7a9f4db8b8..c4b4cca90ca 100644 --- a/base/shell/cmd/cmd.h +++ b/base/shell/cmd/cmd.h @@ -300,7 +300,6 @@ VOID StripQuotes(LPTSTR); BOOL IsValidPathName (LPCTSTR); BOOL IsExistingFile (LPCTSTR); BOOL IsExistingDirectory (LPCTSTR); -BOOL FileGetString (HANDLE, LPTSTR, INT); VOID GetPathCase(TCHAR *, TCHAR *); #define PROMPT_NO 0 diff --git a/base/shell/cmd/misc.c b/base/shell/cmd/misc.c index 8d5239949b6..cd7a08f7722 100644 --- a/base/shell/cmd/misc.c +++ b/base/shell/cmd/misc.c @@ -510,45 +510,6 @@ BOOL IsExistingDirectory (LPCTSTR pszPath) } -BOOL FileGetString (HANDLE hFile, LPTSTR lpBuffer, INT nBufferLength) -{ - LPSTR lpString; - DWORD dwRead; - INT len = 0; -#ifdef _UNICODE - lpString = cmd_alloc(nBufferLength); -#else - lpString = lpBuffer; -#endif - - if (ReadFile(hFile, lpString, nBufferLength - 1, &dwRead, NULL)) - { - /* break at new line*/ - CHAR *end = memchr(lpString, '\n', dwRead); - len = dwRead; - if (end) - { - len = (INT)(end - lpString) + 1; - SetFilePointer(hFile, len - dwRead, NULL, FILE_CURRENT); - } - } - - if (!len) - { -#ifdef _UNICODE - cmd_free(lpString); -#endif - return FALSE; - } - - lpString[len++] = '\0'; -#ifdef _UNICODE - MultiByteToWideChar(OutputCodePage, 0, lpString, -1, lpBuffer, len); - cmd_free(lpString); -#endif - return TRUE; -} - // See r874 BOOL __stdcall PagePrompt(PCON_PAGER Pager, DWORD Done, DWORD Total) { diff --git a/base/shell/cmd/type.c b/base/shell/cmd/type.c index 2c8955d8695..450f2e49240 100644 --- a/base/shell/cmd/type.c +++ b/base/shell/cmd/type.c @@ -30,23 +30,211 @@ #ifdef INCLUDE_CMD_TYPE +static BOOL +FileGetString( + IN HANDLE hFile, + OUT LPTSTR lpBuffer, + IN LONG nBufferLength) +{ + PCHAR pString; + DWORD dwRead; + LONG len = 0; -INT cmd_type(LPTSTR param) +#ifdef _UNICODE + pString = cmd_alloc(nBufferLength); +#else + pString = lpBuffer; +#endif + + if (ReadFile(hFile, pString, nBufferLength - 1, &dwRead, NULL)) + { + /* Break at new line*/ + PCHAR end = memchr(pString, '\n', dwRead); + len = dwRead; + if (end) + { + len = (LONG)(end - pString) + 1; + SetFilePointer(hFile, len - dwRead, NULL, FILE_CURRENT); + } + } + + if (!len) + { +#ifdef _UNICODE + cmd_free(pString); +#endif + return FALSE; + } + + pString[len++] = '\0'; +#ifdef _UNICODE + MultiByteToWideChar(OutputCodePage, 0, pString, -1, lpBuffer, len); + cmd_free(pString); +#endif + return TRUE; +} + +static BOOL +DoTypeFile( + IN LPTSTR FileName, + IN HANDLE hConsoleOut, + IN BOOL bNoFileName, + IN BOOL bPaging) { - TCHAR buff[256]; - HANDLE hFile, hConsoleOut; + HANDLE hFile; + BOOL bIsFile; + DWORD dwFileSize; + DWORD dwFilePos; DWORD dwRet; - INT argc,i; - LPTSTR *argv; LPTSTR errmsg; - BOOL bPaging = FALSE; - BOOL bFirstTime = TRUE; + TCHAR buff[256]; - hConsoleOut = GetStdHandle(STD_OUTPUT_HANDLE); + hFile = CreateFile(FileName, + GENERIC_READ, + FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, + OPEN_EXISTING, + FILE_ATTRIBUTE_NORMAL, NULL); + + if (hFile == INVALID_HANDLE_VALUE) + { + // FIXME: Use ErrorMessage() ? + FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | + FORMAT_MESSAGE_IGNORE_INSERTS | + FORMAT_MESSAGE_FROM_SYSTEM, + NULL, + GetLastError(), + MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), + (LPTSTR)&errmsg, + 0, + NULL); + ConErrPrintf(_T("%s - %s"), FileName, errmsg); + LocalFree(errmsg); + nErrorLevel = 1; + return TRUE; + } + + /* + * When reading from a file, retrieve its original size, so that + * we can stop reading it once we are beyond its original ending. + * This allows avoiding an infinite read loop in case the output + * of the file is redirected back to it. + * If we read from somewhere else (device, ...) don't do anything; + * we will stop when ReadFile() fails (e.g. when Ctrl-Z is seen...). + */ + bIsFile = ((GetFileType(hFile) & ~FILE_TYPE_REMOTE) == FILE_TYPE_DISK); + if (bIsFile) + { + dwFileSize = GetFileSize(hFile, NULL); + if ((dwFileSize == INVALID_FILE_SIZE) && + (GetLastError() != ERROR_SUCCESS)) + { + WARN("Error when retrieving file size, or size too large (%d)\n", + dwFileSize); + dwFileSize = 0; + } + dwFilePos = SetFilePointer(hFile, 0, NULL, FILE_BEGIN); + if ((dwFilePos == INVALID_SET_FILE_POINTER) && + (GetLastError() != ERROR_SUCCESS)) + { + WARN("Error when setting file pointer\n"); + dwFilePos = 0; + } + } + else + { + dwFileSize = dwFilePos = 0; + } + + /* + * Display the file name on StdErr if required, so that if StdOut + * alone is redirected, we can obtain the file contents only. + */ + if (!bNoFileName) + ConErrPrintf(_T("\n%s\n\n\n"), FileName); + + if (bPaging) + { + while (FileGetString(hFile, buff, ARRAYSIZE(buff))) + { + if (!ConOutPrintfPaging(FALSE, _T("%s"), buff)) + { + bCtrlBreak = FALSE; + CloseHandle(hFile); + nErrorLevel = 1; + return FALSE; + } + + /* + * If we read from a file, check where we are and stop + * once we are beyond the original end of the file. + */ + if (bIsFile) + { + dwFilePos = SetFilePointer(hFile, 0, NULL, FILE_CURRENT); + if ((dwFilePos == INVALID_SET_FILE_POINTER) && + (GetLastError() != ERROR_SUCCESS)) + { + WARN("Error when getting file pointer\n"); + break; + } + if (dwFilePos >= dwFileSize) + break; + } + } + } + else + { + while (ReadFile(hFile, buff, sizeof(buff), &dwRet, NULL) && dwRet > 0) + { + WriteFile(hConsoleOut, buff, dwRet, &dwRet, NULL); + if (bCtrlBreak) + { + bCtrlBreak = FALSE; + CloseHandle(hFile); + nErrorLevel = 1; + return FALSE; + } + + /* + * If we read from a file, check where we are and stop + * once we are beyond the original end of the file. + */ + if (bIsFile) + { + dwFilePos = SetFilePointer(hFile, 0, NULL, FILE_CURRENT); + if ((dwFilePos == INVALID_SET_FILE_POINTER) && + (GetLastError() != ERROR_SUCCESS)) + { + WARN("Error when getting file pointer\n"); + break; + } + if (dwFilePos >= dwFileSize) + break; + } + } + } + + CloseHandle(hFile); + return TRUE; +} + +INT cmd_type(LPTSTR param) +{ + INT argc, i; + LPTSTR* argv; + LPTSTR errmsg; + HANDLE hConsoleOut; + BOOL bNoFileName = FALSE; + BOOL bPaging = FALSE; + BOOL bFileFound; + DWORD dwLastError; + UINT nFileSpecs = 0; + HANDLE hFind; + WIN32_FIND_DATA FindData; if (!_tcsncmp(param, _T("/?"), 2)) { - ConOutResPaging(TRUE,STRING_TYPE_HELP1); + ConOutResPaging(TRUE, STRING_TYPE_HELP1); return 0; } @@ -56,87 +244,124 @@ INT cmd_type(LPTSTR param) return 1; } - argv = split(param, &argc, TRUE, FALSE); + /* Parse the command line. We will manually expand any file specification present. */ + argv = split(param, &argc, FALSE, FALSE); - for (i = 0; i < argc; i++) + /* Loop through the options, count also the specified number of file specifications */ + for (i = 0; i < argc; ++i) { - if (argv[i][0] == _T('/') && _tcslen(argv[i]) == 2 && _totupper(argv[i][1]) == _T('P')) + if (argv[i][0] == _T('/')) { - bPaging = TRUE; - } - } + if (_tcslen(argv[i]) == 2) + { + switch (_totupper(argv[i][1])) + { + case _T('N'): + bNoFileName = TRUE; + continue; - for (i = 0; i < argc; i++) - { - if (argv[i][0] == _T('/') && _totupper(argv[i][1]) != _T('P')) - { + case _T('P'): + bPaging = TRUE; + continue; + } + } + + // error_invalid_switch(argv[i] + 1); ConErrResPrintf(STRING_TYPE_ERROR1, argv[i] + 1); - continue; - } - if (argv[i][0] == _T('/') && _tcslen(argv[i]) == 2 && _totupper(argv[i][1]) == _T('P')) - { - continue; + nErrorLevel = 1; + goto Quit; } - nErrorLevel = 0; + /* This should be a file specification */ + ++nFileSpecs; + } - hFile = CreateFile(argv[i], - GENERIC_READ, - FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, - OPEN_EXISTING, - FILE_ATTRIBUTE_NORMAL, NULL); + nErrorLevel = 0; - if (hFile == INVALID_HANDLE_VALUE) - { - FormatMessage (FORMAT_MESSAGE_ALLOCATE_BUFFER | - FORMAT_MESSAGE_IGNORE_INSERTS | - FORMAT_MESSAGE_FROM_SYSTEM, - NULL, - GetLastError(), - MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), - (LPTSTR) &errmsg, - 0, - NULL); - ConErrPrintf (_T("%s - %s"), argv[i], errmsg); - LocalFree (errmsg); - nErrorLevel = 1; + hConsoleOut = GetStdHandle(STD_OUTPUT_HANDLE); + + /* Reset paging state */ + if (bPaging) + ConOutPrintfPaging(TRUE, _T("")); + + /* Now loop through the files */ + for (i = 0; i < argc; ++i) + { + /* Skip the options */ + if (argv[i][0] == _T('/')) continue; - } - if (bPaging) + /* If wildcards are present in this file specification, perform a file enumeration */ + if (_tcschr(argv[i], _T('*')) || _tcschr(argv[i], _T('?'))) { - while (FileGetString(hFile, buff, ARRAYSIZE(buff))) + dwLastError = ERROR_SUCCESS; + bFileFound = FALSE; + + hFind = FindFirstFile(argv[i], &FindData); + + if (hFind != INVALID_HANDLE_VALUE) { - if (!ConOutPrintfPaging(bFirstTime, _T("%s"), buff)) + /* Loop through all the files */ + do { - bCtrlBreak = FALSE; - CloseHandle(hFile); - freep(argv); - return 0; - } - bFirstTime = FALSE; + /* Ignore any directory silently */ + if (!_tcscmp(FindData.cFileName, _T(".")) || + !_tcscmp(FindData.cFileName, _T("..")) || + (FindData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)) + { + continue; + } + + bFileFound = TRUE; + if (!DoTypeFile(FindData.cFileName, hConsoleOut, bNoFileName, bPaging)) + { + FindClose(hFind); + goto Quit; + } + + } while (FindNextFile(hFind, &FindData)); + + FindClose(hFind); + } + + /* + * Return an error if the file specification could not be resolved, + * or no actual files were encountered (but only directories). + */ + if (hFind == INVALID_HANDLE_VALUE) + dwLastError = GetLastError(); + else if (!bFileFound) + dwLastError = ERROR_FILE_NOT_FOUND; + + if (dwLastError != ERROR_SUCCESS) + { + // FIXME: Use ErrorMessage() ? + FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | + FORMAT_MESSAGE_IGNORE_INSERTS | + FORMAT_MESSAGE_FROM_SYSTEM, + NULL, + dwLastError, + MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), + (LPTSTR)&errmsg, + 0, + NULL); + ConErrPrintf(_T("%s - %s"), argv[i], errmsg); + LocalFree(errmsg); + nErrorLevel = 1; } } else { - while (ReadFile(hFile, buff, sizeof(buff), &dwRet, NULL) && dwRet > 0) - { - WriteFile(hConsoleOut, buff, dwRet, &dwRet, NULL); - if (bCtrlBreak) - { - bCtrlBreak = FALSE; - CloseHandle(hFile); - freep(argv); - return 0; - } - } + if (!DoTypeFile(argv[i], hConsoleOut, (bNoFileName || (nFileSpecs <= 1)), bPaging)) + goto Quit; } - CloseHandle(hFile); + /* Continue with the next file specification */ } +Quit: freep(argv); - return 0; + return nErrorLevel; } #endif

4 years, 4 months

[reactos] 02/04: [CMD] CHDIR: Some features are available only when extensions are enabled. Update the SetRootPath() as well.

by Hermès Bélusca-Maïto

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=3d4af223283e19eb308a5… commit 3d4af223283e19eb308a5c4d24e7f42887ce5ee5 Author: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> AuthorDate: Sun Jul 12 20:45:59 2020 +0200 Commit: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> CommitDate: Wed Aug 19 21:39:21 2020 +0200 [CMD] CHDIR: Some features are available only when extensions are enabled. Update the SetRootPath() as well. --- base/shell/cmd/internal.c | 44 ++++++++++++++++++++++++++++++++++++-------- 1 file changed, 36 insertions(+), 8 deletions(-) diff --git a/base/shell/cmd/internal.c b/base/shell/cmd/internal.c index 7e8a7367c02..87fe7df9d81 100644 --- a/base/shell/cmd/internal.c +++ b/base/shell/cmd/internal.c @@ -194,9 +194,22 @@ BOOL SetRootPath(TCHAR *oldpath, TCHAR *InPath) goto Fail; } - /* Convert the full path to its correct case. - * Example: c:\windows\SYSTEM32 => C:\WINDOWS\System32 */ - GetPathCase(OutPathTemp, OutPath); + if (bEnableExtensions) + { + /* + * Convert the full path to its correct case, and + * resolve any wilcard present as well in the path + * and retrieve the first result. + * Example: c:\windows\SYSTEM32 => C:\WINDOWS\System32 + * Example: C:\WINDOWS\S* => C:\WINDOWS\System, + * or C:\WINDOWS\System32, depending on the user's OS. + */ + GetPathCase(OutPathTemp, OutPath); + } + else + { + _tcscpy(OutPath, OutPathTemp); + } /* Use _tchdir(), since unlike SetCurrentDirectory() it updates * the current-directory-on-drive environment variables. */ @@ -239,12 +252,27 @@ INT cmd_chdir(LPTSTR param) return 0; } - /* Remove extra quotes and strip trailing whitespace */ + // + // FIXME: Use the split() tokenizer if bEnableExtensions == FALSE, + // so as to cut the parameter at the first separator (space, ',', ';'): + // - When bEnableExtensions == FALSE, doing + // CD system32;winsxs + // will go into system32, (but: CD "system32;winsxs" will fail as below), while + // - When bEnableExtensions == TRUE, it will fail because the "system32;winsxs" + // directory does not exist. + // + + /* Remove extra quotes */ StripQuotes(param); - tmp = param + _tcslen(param) - 1; - while (tmp > param && _istspace(*tmp)) - --tmp; - *(tmp + 1) = _T('\0'); + + if (bEnableExtensions) + { + /* Strip trailing whitespace */ + tmp = param + _tcslen(param) - 1; + while (tmp > param && _istspace(*tmp)) + --tmp; + *(tmp + 1) = _T('\0'); + } /* Reset the error level */ nErrorLevel = 0;

4 years, 4 months

[reactos] 01/04: [CMD] SET: Fix displaying the environment variables with a given prefix.

by Hermès Bélusca-Maïto

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=fe9aa42d5f61fdd2fd450… commit fe9aa42d5f61fdd2fd4501aa7541e09e98207cb7 Author: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> AuthorDate: Sun Jul 12 17:03:45 2020 +0200 Commit: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> CommitDate: Wed Aug 19 21:39:18 2020 +0200 [CMD] SET: Fix displaying the environment variables with a given prefix. - Restore any truncated space in the name prefix, before displaying any error message. - When trimming the name prefix from "special" characters (spaces, comma and semicolon), so that e.g. "set ,; ,;FOO" displays all the variables starting by "FOO", save also a pointer to the original name prefix, that we will use for variables lookup as well. This is done, because the SET command allows setting an environment variable whose name actually contains these characters (e.g. "set ,; ,;FOO=42"), however, by trimming the characters, doing "set ,; ,;FOO" would not allow seeing such variables. With the fix, it is now possible to show them. --- base/shell/cmd/set.c | 30 ++++++++++++++++++++++++++---- 1 file changed, 26 insertions(+), 4 deletions(-) diff --git a/base/shell/cmd/set.c b/base/shell/cmd/set.c index f63de691a2f..0128314ca57 100644 --- a/base/shell/cmd/set.c +++ b/base/shell/cmd/set.c @@ -168,12 +168,27 @@ INT cmd_set(LPTSTR param) else { /* Display all the environment variables with the given prefix */ - BOOL bFound = FALSE; - + LPTSTR pOrgParam = param; + BOOLEAN bFound = FALSE; + BOOLEAN bRestoreSpace; + + /* + * Trim the prefix from "special" characters (only when displaying the + * environment variables), so that e.g. "SET ,; ,;FOO" will display all + * the variables starting by "FOO". + * The SET command allows as well to set an environment variable whose name + * actually contains these characters (e.g. "SET ,; ,;FOO=42"); however, + * by trimming the characters, doing "SET ,; ,;FOO" would not allow seeing + * such variables. + * Thus, we also save a pointer to the original variable name prefix, that + * we will look it up as well below. + */ while (_istspace(*param) || *param == _T(',') || *param == _T(';')) - param++; + ++param; + /* Just remove the very last space, if present */ p = _tcsrchr(param, _T(' ')); + bRestoreSpace = (p != NULL); if (!p) p = param + _tcslen(param); *p = _T('\0'); @@ -184,7 +199,9 @@ INT cmd_set(LPTSTR param) lpOutput = lpEnv; while (*lpOutput) { - if (!_tcsnicmp(lpOutput, param, p - param)) + /* Look up for both the original and truncated variable name prefix */ + if (!_tcsnicmp(lpOutput, pOrgParam, p - pOrgParam) || + !_tcsnicmp(lpOutput, param, p - param)) { ConOutPuts(lpOutput); ConOutChar(_T('\n')); @@ -195,6 +212,11 @@ INT cmd_set(LPTSTR param) FreeEnvironmentStrings(lpEnv); } + /* Restore the truncated space for correctly + * displaying the error message, if any. */ + if (bRestoreSpace) + *p = _T(' '); + if (!bFound) { ConErrResPrintf(STRING_SET_ENV_ERROR, param);

4 years, 4 months

[reactos] 48/48: [CMD] REPLACE: Fix a memory leak.

by Hermès Bélusca-Maïto

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=8cea82b14c5ff14fc56b6… commit 8cea82b14c5ff14fc56b6d0e5a432fc9e637f490 Author: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> AuthorDate: Sat Jul 25 01:00:06 2020 +0200 Commit: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> CommitDate: Wed Aug 19 20:36:13 2020 +0200 [CMD] REPLACE: Fix a memory leak. --- base/shell/cmd/replace.c | 1 + 1 file changed, 1 insertion(+) diff --git a/base/shell/cmd/replace.c b/base/shell/cmd/replace.c index 5929952582f..35c2576168f 100644 --- a/base/shell/cmd/replace.c +++ b/base/shell/cmd/replace.c @@ -437,6 +437,7 @@ INT cmd_replace (LPTSTR param) break; default: invalid_switch(arg[i]); + freep(arg); return 0; } }

4 years, 4 months

[reactos] 47/48: [CMD] SETLOCAL / ENDLOCAL: Save / Restore as well the current drive and current directory.

by Hermès Bélusca-Maïto

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=1cb7e0852260735df364a… commit 1cb7e0852260735df364abb1d993d5d67f22eab6 Author: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> AuthorDate: Sat Jul 18 23:35:56 2020 +0200 Commit: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> CommitDate: Wed Aug 19 20:36:13 2020 +0200 [CMD] SETLOCAL / ENDLOCAL: Save / Restore as well the current drive and current directory. That's an actual fact, done on original MS-DOS COMMAND.COM, FreeCOM, Windows' CMD.EXE, etc., but is strangely undocumented on MSDN documentation. See https://www.dostips.com/forum/viewtopic.php?t=4436 Fixes some cmd_winetests. --- base/shell/cmd/setlocal.c | 35 +++++++++++++++++++++++++++++++++-- 1 file changed, 33 insertions(+), 2 deletions(-) diff --git a/base/shell/cmd/setlocal.c b/base/shell/cmd/setlocal.c index daf86099050..18cec04b6f0 100644 --- a/base/shell/cmd/setlocal.c +++ b/base/shell/cmd/setlocal.c @@ -9,10 +9,13 @@ #include "precomp.h" +#include <direct.h> // For _getdrive(). + typedef struct _SETLOCAL { struct _SETLOCAL *Prev; LPTSTR Environment; + INT CurDrive; BOOL EnableExtensions; BOOL DelayedExpansion; } SETLOCAL, *PSETLOCAL; @@ -42,6 +45,13 @@ INT cmd_setlocal(LPTSTR param) LPTSTR* arg; INT argc, i; + if (!_tcscmp(param, _T("/?"))) + { + // FIXME + ConOutPuts(_T("SETLOCAL help not implemented yet!\n")); + return 0; + } + /* SETLOCAL only works inside a batch context */ if (!bc) return 0; @@ -53,6 +63,7 @@ INT cmd_setlocal(LPTSTR param) error_out_of_memory(); return 1; } + Saved->Environment = DuplicateEnvironment(); if (!Saved->Environment) { @@ -60,6 +71,12 @@ INT cmd_setlocal(LPTSTR param) cmd_free(Saved); return 1; } + /* + * Save the current drive; the duplicated environment + * contains the corresponding current directory. + */ + Saved->CurDrive = _getdrive(); + Saved->EnableExtensions = bEnableExtensions; Saved->DelayedExpansion = bDelayedExpansion; @@ -94,9 +111,15 @@ INT cmd_endlocal(LPTSTR param) { LPTSTR Environ, Name, Value; PSETLOCAL Saved; + TCHAR drvEnvVar[] = _T("=?:"); + TCHAR szCurrent[MAX_PATH]; - /* ENDLOCAL doesn't take any params */ - UNREFERENCED_PARAMETER(param); + if (!_tcscmp(param, _T("/?"))) + { + // FIXME + ConOutPuts(_T("ENDLOCAL help not implemented yet!\n")); + return 0; + } /* Pop a SETLOCAL struct off of this batch context's stack */ if (!bc || !(Saved = bc->setlocal)) @@ -133,6 +156,14 @@ INT cmd_endlocal(LPTSTR param) Name = Value; } + /* Restore the current drive and its current directory from the environment */ + drvEnvVar[1] = _T('A') + Saved->CurDrive - 1; + if (!GetEnvironmentVariable(drvEnvVar, szCurrent, ARRAYSIZE(szCurrent))) + { + _stprintf(szCurrent, _T("%C:\\"), _T('A') + Saved->CurDrive - 1); + } + _tchdir(szCurrent); // SetRootPath(NULL, szCurrent); + cmd_free(Saved->Environment); cmd_free(Saved); return 0;

4 years, 4 months

[reactos] 46/48: [CMD] setlocal.c : Code style and formatting fixes

by Hermès Bélusca-Maïto

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=e8e31267c5a47a12db55e… commit e8e31267c5a47a12db55eb50958542300bf62cb5 Author: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> AuthorDate: Sun Jul 26 20:32:39 2020 +0200 Commit: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> CommitDate: Wed Aug 19 20:36:13 2020 +0200 [CMD] setlocal.c : Code style and formatting fixes --- base/shell/cmd/setlocal.c | 39 ++++++++++++++++++++------------------- 1 file changed, 20 insertions(+), 19 deletions(-) diff --git a/base/shell/cmd/setlocal.c b/base/shell/cmd/setlocal.c index ca913dc2a6e..daf86099050 100644 --- a/base/shell/cmd/setlocal.c +++ b/base/shell/cmd/setlocal.c @@ -12,10 +12,10 @@ typedef struct _SETLOCAL { struct _SETLOCAL *Prev; + LPTSTR Environment; BOOL EnableExtensions; BOOL DelayedExpansion; - LPTSTR Environment; -} SETLOCAL; +} SETLOCAL, *PSETLOCAL; /* Create a copy of the current environment */ LPTSTR @@ -38,11 +38,11 @@ DuplicateEnvironment(VOID) INT cmd_setlocal(LPTSTR param) { - SETLOCAL *Saved; - LPTSTR *arg; + PSETLOCAL Saved; + LPTSTR* arg; INT argc, i; - /* SETLOCAL only works inside a batch file */ + /* SETLOCAL only works inside a batch context */ if (!bc) return 0; @@ -53,9 +53,6 @@ INT cmd_setlocal(LPTSTR param) error_out_of_memory(); return 1; } - Saved->Prev = bc->setlocal; - Saved->EnableExtensions = bEnableExtensions; - Saved->DelayedExpansion = bDelayedExpansion; Saved->Environment = DuplicateEnvironment(); if (!Saved->Environment) { @@ -63,6 +60,10 @@ INT cmd_setlocal(LPTSTR param) cmd_free(Saved); return 1; } + Saved->EnableExtensions = bEnableExtensions; + Saved->DelayedExpansion = bDelayedExpansion; + + Saved->Prev = bc->setlocal; bc->setlocal = Saved; nErrorLevel = 0; @@ -70,15 +71,13 @@ INT cmd_setlocal(LPTSTR param) arg = splitspace(param, &argc); for (i = 0; i < argc; i++) { - if (!_tcsicmp(arg[i], _T("enableextensions"))) - /* FIXME: not implemented! */ + if (!_tcsicmp(arg[i], _T("ENABLEEXTENSIONS"))) bEnableExtensions = TRUE; - else if (!_tcsicmp(arg[i], _T("disableextensions"))) - /* FIXME: not implemented! */ + else if (!_tcsicmp(arg[i], _T("DISABLEEXTENSIONS"))) bEnableExtensions = FALSE; - else if (!_tcsicmp(arg[i], _T("enabledelayedexpansion"))) + else if (!_tcsicmp(arg[i], _T("ENABLEDELAYEDEXPANSION"))) bDelayedExpansion = TRUE; - else if (!_tcsicmp(arg[i], _T("disabledelayedexpansion"))) + else if (!_tcsicmp(arg[i], _T("DISABLEDELAYEDEXPANSION"))) bDelayedExpansion = FALSE; else { @@ -91,13 +90,15 @@ INT cmd_setlocal(LPTSTR param) return nErrorLevel; } -/* endlocal doesn't take any params */ INT cmd_endlocal(LPTSTR param) { LPTSTR Environ, Name, Value; - SETLOCAL *Saved; + PSETLOCAL Saved; + + /* ENDLOCAL doesn't take any params */ + UNREFERENCED_PARAMETER(param); - /* Pop a SETLOCAL struct off of this batch file's stack */ + /* Pop a SETLOCAL struct off of this batch context's stack */ if (!bc || !(Saved = bc->setlocal)) return 0; bc->setlocal = Saved->Prev; @@ -107,7 +108,7 @@ INT cmd_endlocal(LPTSTR param) /* First, clear out the environment. Since making any changes to the * environment invalidates pointers obtained from GetEnvironmentStrings(), - * we must make a copy of it and get the variable names from that */ + * we must make a copy of it and get the variable names from that. */ Environ = DuplicateEnvironment(); if (Environ) { @@ -122,7 +123,7 @@ INT cmd_endlocal(LPTSTR param) cmd_free(Environ); } - /* Now, restore variables from the copy saved by cmd_setlocal */ + /* Now, restore variables from the copy saved by cmd_setlocal() */ for (Name = Saved->Environment; *Name; Name += _tcslen(Name) + 1) { if (!(Value = _tcschr(Name + 1, _T('='))))

4 years, 4 months

[reactos] 45/48: [CMD] Implement provisional support for the HIGHESTNUMANODENUMBER environment-like variable.

by Hermès Bélusca-Maïto

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=90159e1e518483481ff70… commit 90159e1e518483481ff700ff1456f603090f3008 Author: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> AuthorDate: Sun Jul 12 23:37:50 2020 +0200 Commit: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> CommitDate: Wed Aug 19 20:36:12 2020 +0200 [CMD] Implement provisional support for the HIGHESTNUMANODENUMBER environment-like variable. This variable is available only in Win7+, even if the underlying API GetNumaHighestNodeNumber() is available in Win2003+ --- base/shell/cmd/cmd.c | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/base/shell/cmd/cmd.c b/base/shell/cmd/cmd.c index 1cc0a0663f5..fbc12154d1b 100644 --- a/base/shell/cmd/cmd.c +++ b/base/shell/cmd/cmd.c @@ -948,6 +948,17 @@ GetEnvVarOrSpecial(LPCTSTR varName) _itot(nErrorLevel, ret, 10); return ret; } +#if (NTDDI_VERSION >= NTDDI_WIN7) + /* Available in Win7+, even if the underlying API is available in Win2003+ */ + /* %HIGHESTNUMANODENUMBER% */ + else if (_tcsicmp(varName, _T("HIGHESTNUMANODENUMBER")) == 0) + { + ULONG NumaNodeNumber = 0; + GetNumaHighestNodeNumber(&NumaNodeNumber); + _itot(NumaNodeNumber, ret, 10); + return ret; + } +#endif return NULL; }

4 years, 4 months

[reactos] 44/48: [CMD] The "special" environment-like variables are available only when extensions are enabled.

by Hermès Bélusca-Maïto

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=04e0fe0652553b1472d42… commit 04e0fe0652553b1472d4255fa05f87110b1f1870 Author: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> AuthorDate: Sun Jul 12 23:27:35 2020 +0200 Commit: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> CommitDate: Wed Aug 19 20:36:12 2020 +0200 [CMD] The "special" environment-like variables are available only when extensions are enabled. The "special" variables are: CD, DATE, TIME, RANDOM, CMDCMDLINE, CMDEXTVERSION, ERRORLEVEL (and on Win7+, HIGHESTNUMANODENUMBER). --- base/shell/cmd/cmd.c | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/base/shell/cmd/cmd.c b/base/shell/cmd/cmd.c index 3a57c04ef33..1cc0a0663f5 100644 --- a/base/shell/cmd/cmd.c +++ b/base/shell/cmd/cmd.c @@ -902,7 +902,11 @@ GetEnvVarOrSpecial(LPCTSTR varName) if (var) return var; - /* env var doesn't exist, look for a "special" one */ + /* The environment variable doesn't exist, look for + * a "special" one only if extensions are enabled. */ + if (!bEnableExtensions) + return NULL; + /* %CD% */ if (_tcsicmp(varName, _T("CD")) == 0) {

4 years, 4 months

[reactos] 43/48: [CMD] FOR: Some functionality is available only when extensions are enabled.

by Hermès Bélusca-Maïto

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=41a93a4e58d3aa62ae3e2… commit 41a93a4e58d3aa62ae3e28303e8433878169d827 Author: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> AuthorDate: Sun Jul 12 23:07:23 2020 +0200 Commit: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> CommitDate: Wed Aug 19 20:36:11 2020 +0200 [CMD] FOR: Some functionality is available only when extensions are enabled. This is basically all the advanced functionality enabled with the /D, /R, /L and /F flags, and the usage of enhanced variables. --- base/shell/cmd/for.c | 32 ++++++++++++++++++++------------ base/shell/cmd/parser.c | 6 ++++++ 2 files changed, 26 insertions(+), 12 deletions(-) diff --git a/base/shell/cmd/for.c b/base/shell/cmd/for.c index 75b462352cf..67a9240c787 100644 --- a/base/shell/cmd/for.c +++ b/base/shell/cmd/for.c @@ -500,19 +500,27 @@ ExecuteFor(PARSED_COMMAND *Cmd) Cmd->For.Context = lpNew; fc = lpNew; - if (Cmd->For.Switches & FOR_F) + /* Run the extended FOR syntax only if extensions are enabled */ + if (bEnableExtensions) { - Ret = ForF(Cmd, List, Buffer); - } - else if (Cmd->For.Switches & FOR_LOOP) - { - Ret = ForLoop(Cmd, List, Buffer); - } - else if (Cmd->For.Switches & FOR_RECURSIVE) - { - DWORD Len = GetFullPathName(Cmd->For.Params ? Cmd->For.Params : _T("."), - MAX_PATH, Buffer, NULL); - Ret = ForRecursive(Cmd, List, Buffer, &Buffer[Len]); + if (Cmd->For.Switches & FOR_F) + { + Ret = ForF(Cmd, List, Buffer); + } + else if (Cmd->For.Switches & FOR_LOOP) + { + Ret = ForLoop(Cmd, List, Buffer); + } + else if (Cmd->For.Switches & FOR_RECURSIVE) + { + DWORD Len = GetFullPathName(Cmd->For.Params ? Cmd->For.Params : _T("."), + MAX_PATH, Buffer, NULL); + Ret = ForRecursive(Cmd, List, Buffer, &Buffer[Len]); + } + else + { + Ret = ForDir(Cmd, List, Buffer, Buffer); + } } else { diff --git a/base/shell/cmd/parser.c b/base/shell/cmd/parser.c index 21bfd2504a9..048c83e3f96 100644 --- a/base/shell/cmd/parser.c +++ b/base/shell/cmd/parser.c @@ -494,6 +494,10 @@ static PARSED_COMMAND *ParseFor(void) memset(Cmd, 0, sizeof(PARSED_COMMAND)); Cmd->Type = C_FOR; + /* Skip the extended FOR syntax if extensions are disabled */ + if (!bEnableExtensions) + goto parseForBody; + while (1) { if (_tcsicmp(CurrentToken, _T("/D")) == 0) @@ -543,6 +547,8 @@ static PARSED_COMMAND *ParseFor(void) goto error; } +parseForBody: + /* Variable name should be % and just one other character */ if (CurrentToken[0] != _T('%') || _tcslen(CurrentToken) != 2) goto error;

4 years, 4 months

[reactos] 42/48: [CMD] IF: Some functionality is available only when extensions are enabled.

by Hermès Bélusca-Maïto

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=fedc68aea842c11d45e15… commit fedc68aea842c11d45e158394a7eae1b8e7cab1b Author: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> AuthorDate: Sun Jul 12 21:41:07 2020 +0200 Commit: Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org> CommitDate: Wed Aug 19 20:36:11 2020 +0200 [CMD] IF: Some functionality is available only when extensions are enabled. This functionality is: case insensitivity comparisons (/I); CMDEXTVERSION and DEFINED unary operators; EQU, NEQ, LSS, LEQ, GTR, GEQ generic string comparators. --- base/shell/cmd/cmd.c | 4 ++-- base/shell/cmd/cmd.h | 23 ++++++++++++++++++----- base/shell/cmd/if.c | 8 ++++---- base/shell/cmd/parser.c | 32 ++++++++++++++++++++++++-------- 4 files changed, 48 insertions(+), 19 deletions(-) diff --git a/base/shell/cmd/cmd.c b/base/shell/cmd/cmd.c index 4bbffa9faf1..3a57c04ef33 100644 --- a/base/shell/cmd/cmd.c +++ b/base/shell/cmd/cmd.c @@ -934,8 +934,8 @@ GetEnvVarOrSpecial(LPCTSTR varName) /* %CMDEXTVERSION% */ else if (_tcsicmp(varName, _T("CMDEXTVERSION")) == 0) { - /* Set version number to 2 */ - _itot(2, ret, 10); + /* Set version number to CMDEXTVERSION */ + _itot(CMDEXTVERSION, ret, 10); return ret; } /* %ERRORLEVEL% */ diff --git a/base/shell/cmd/cmd.h b/base/shell/cmd/cmd.h index d7bce5c139e..c7a9f4db8b8 100644 --- a/base/shell/cmd/cmd.h +++ b/base/shell/cmd/cmd.h @@ -28,6 +28,9 @@ #include "cmdver.h" #include "cmddbg.h" +/* Version of the Command Extensions */ +#define CMDEXTVERSION 2 + #define BREAK_BATCHFILE 1 #define BREAK_OUTOFBATCH 2 /* aka. BREAK_ENDOFBATCHFILES */ #define BREAK_INPUT 3 @@ -232,11 +235,21 @@ INT CommandHistory(LPTSTR param); #endif /* Prototypes for IF.C */ -#define IFFLAG_NEGATE 1 /* NOT */ -#define IFFLAG_IGNORECASE 2 /* /I */ -enum { IF_CMDEXTVERSION, IF_DEFINED, IF_ERRORLEVEL, IF_EXIST, - IF_STRINGEQ, /* == */ - IF_EQU, IF_GTR, IF_GEQ, IF_LSS, IF_LEQ, IF_NEQ }; +#define IFFLAG_NEGATE 1 /* NOT */ +#define IFFLAG_IGNORECASE 2 /* /I - Extended */ +enum { + /** Unary operators **/ + /* Standard */ + IF_ERRORLEVEL, IF_EXIST, + /* Extended */ + IF_CMDEXTVERSION, IF_DEFINED, + + /** Binary operators **/ + /* Standard */ + IF_STRINGEQ, /* == */ + /* Extended */ + IF_EQU, IF_NEQ, IF_LSS, IF_LEQ, IF_GTR, IF_GEQ +}; INT ExecuteIf(struct _PARSED_COMMAND *Cmd); /* Prototypes for INTERNAL.C */ diff --git a/base/shell/cmd/if.c b/base/shell/cmd/if.c index c7882ad2fec..e74298ad612 100644 --- a/base/shell/cmd/if.c +++ b/base/shell/cmd/if.c @@ -82,7 +82,7 @@ INT ExecuteIf(PARSED_COMMAND *Cmd) return 1; } - if (Cmd->If.Operator == IF_CMDEXTVERSION) + if (bEnableExtensions && (Cmd->If.Operator == IF_CMDEXTVERSION)) { /* IF CMDEXTVERSION n: check if Command Extensions version * is greater or equal to n */ @@ -93,9 +93,9 @@ INT ExecuteIf(PARSED_COMMAND *Cmd) cmd_free(Right); return 1; } - result = (2 >= n); + result = (CMDEXTVERSION >= n); } - else if (Cmd->If.Operator == IF_DEFINED) + else if (bEnableExtensions && (Cmd->If.Operator == IF_DEFINED)) { /* IF DEFINED var: check if environment variable exists */ result = (GetEnvVarOrSpecial(Right) != NULL); @@ -167,7 +167,7 @@ INT ExecuteIf(PARSED_COMMAND *Cmd) /* IF str1 == str2 */ result = (StringCmp(Left, Right) == 0); } - else + else if (bEnableExtensions) { result = GenericCmp(StringCmp, Left, Right); switch (Cmd->If.Operator) diff --git a/base/shell/cmd/parser.c b/base/shell/cmd/parser.c index 20310c77830..21bfd2504a9 100644 --- a/base/shell/cmd/parser.c +++ b/base/shell/cmd/parser.c @@ -24,19 +24,26 @@ static const TCHAR RedirString[][3] = { _T("<"), _T(">"), _T(">>") }; static const TCHAR *const IfOperatorString[] = { - _T("cmdextversion"), - _T("defined"), + /* Standard */ _T("errorlevel"), _T("exist"), -#define IF_MAX_UNARY IF_EXIST + + /* Extended */ + _T("cmdextversion"), + _T("defined"), +#define IF_MAX_UNARY IF_DEFINED + + /* Standard */ _T("=="), + + /* Extended */ _T("equ"), - _T("gtr"), - _T("geq"), + _T("neq"), _T("lss"), _T("leq"), - _T("neq"), -#define IF_MAX_COMPARISON IF_NEQ + _T("gtr"), + _T("geq"), +#define IF_MAX_COMPARISON IF_GEQ }; static BOOL IsSeparator(TCHAR Char) @@ -389,7 +396,7 @@ static PARSED_COMMAND *ParseIf(void) memset(Cmd, 0, sizeof(PARSED_COMMAND)); Cmd->Type = C_IF; - if (_tcsicmp(CurrentToken, _T("/I")) == 0) + if (bEnableExtensions && (_tcsicmp(CurrentToken, _T("/I")) == 0)) { Cmd->If.Flags |= IFFLAG_IGNORECASE; ParseToken(0, STANDARD_SEPS); @@ -406,6 +413,10 @@ static PARSED_COMMAND *ParseIf(void) /* Check for unary operators */ for (; Cmd->If.Operator <= IF_MAX_UNARY; Cmd->If.Operator++) { + /* Skip the extended operators if the extensions are disabled */ + if (!bEnableExtensions && (Cmd->If.Operator >= IF_CMDEXTVERSION)) + continue; + if (_tcsicmp(CurrentToken, IfOperatorString[Cmd->If.Operator]) == 0) { if (ParseToken(0, STANDARD_SEPS) != TOK_NORMAL) @@ -427,8 +438,13 @@ static PARSED_COMMAND *ParseIf(void) goto condition_done; } + // Cmd->If.Operator == IF_MAX_UNARY + 1; for (; Cmd->If.Operator <= IF_MAX_COMPARISON; Cmd->If.Operator++) { + /* Skip the extended operators if the extensions are disabled */ + if (!bEnableExtensions && (Cmd->If.Operator >= IF_EQU)) // (Cmd->If.Operator > IF_STRINGEQ) + continue; + if (_tcsicmp(CurrentToken, IfOperatorString[Cmd->If.Operator]) == 0) { if (ParseToken(0, STANDARD_SEPS) != TOK_NORMAL)

4 years, 4 months

← Newer
1
...
4
5
6
7
8
9
10
...
19
Older →

Jump to page:

2024

2023

2022

2021

2020

2019

2018

2017

2016

2015

2014

2013

2012

2011

2010

2009

2008

2007

2006

2005

2004

Ros-diffs August 2020