int RegGetStringValue(HKEY hk, LPCWSTR pszSubKey, LPCWSTR pszValueName, CEStr& rszData, DWORD Wow64Flags /*= 0*/)
{
int iLen = -1;
HKEY hkChild = hk;
DWORD cbSize = 0;
LONG lrc;
rszData.Empty();
if (pszSubKey && *pszSubKey)
{
if (hk == NULL)
{
lrc = RegGetStringValue(HKEY_CURRENT_USER, pszSubKey, pszValueName, rszData, 0);
if (lrc < 0)
{
bool isWin64 = IsWindows64();
lrc = RegGetStringValue(HKEY_LOCAL_MACHINE, pszSubKey, pszValueName, rszData, isWin64 ? KEY_WOW64_64KEY : 0);
if ((lrc < 0) && isWin64)
{
lrc = RegGetStringValue(HKEY_LOCAL_MACHINE, pszSubKey, pszValueName, rszData, KEY_WOW64_32KEY);
}
}
if (lrc > 0)
{
return lrc;
}
}
if (0 != (lrc = RegOpenKeyEx(hk, pszSubKey, 0, KEY_READ|Wow64Flags, &hkChild)))
hkChild = NULL;
}
if (hkChild && (0 == (lrc = RegQueryValueEx(hkChild, pszValueName, NULL, NULL, NULL, &cbSize))))
{
wchar_t* pszData = rszData.GetBuffer((cbSize>>1)+2); // +wchar_t+1byte (на возможные ошибки хранения данных в реестре)
if (pszData)
{
pszData[cbSize>>1] = 0; // Make sure it will be 0-terminated
if (0 == (lrc = RegQueryValueEx(hkChild, pszValueName, NULL, NULL, (LPBYTE)pszData, &cbSize)))
{
iLen = lstrlen(pszData);
}
else
{
rszData.Empty();
}
}
}
void DebugNeedCmdUnitTests()
{
BOOL b;
struct strTests { LPCWSTR pszCmd; BOOL bNeed; }
Tests[] = {
{L"\"C:\\windows\\notepad.exe -f \"makefile\" COMMON=\"../../../plugins/common\"\"", FALSE},
{L"\"\"C:\\windows\\notepad.exe -new_console\"\"", FALSE},
{L"\"\"cmd\"\"", FALSE},
{L"cmd /c \"\"C:\\Program Files\\Windows NT\\Accessories\\wordpad.exe\" -?\"", FALSE},
{L"cmd /c \"dir c:\\\"", FALSE},
{L"abc.cmd", TRUE},
// Do not do too many heuristic. If user really needs redirection (for 'root'!)
// he must explicitly call "cmd /c ...". With only exception if first exe not found.
{L"notepad text & start explorer", FALSE},
};
LPCWSTR psArgs;
BOOL bNeedCut, bRootIsCmd, bAlwaysConfirm, bAutoDisable;
CEStr szExe;
for (INT_PTR i = 0; i < countof(Tests); i++)
{
szExe.Empty();
RConStartArgs rcs; rcs.pszSpecialCmd = lstrdup(Tests[i].pszCmd);
rcs.ProcessNewConArg();
b = IsNeedCmd(TRUE, rcs.pszSpecialCmd, szExe, &psArgs, &bNeedCut, &bRootIsCmd, &bAlwaysConfirm, &bAutoDisable);
_ASSERTE(b == Tests[i].bNeed);
}
}
// The function converts UTF-8 to UCS2 (CEStr) and returns pointer to allocated buffer (CEStr)
LPCWSTR SettingsXML::utf2wcs(const char* utf8, CEStr& wc)
{
if (!utf8)
{
_ASSERTE(utf8 != NULL);
wc.Clear();
return NULL;
}
wc.Empty();
int wcLen = MultiByteToWideChar(CP_UTF8, 0, utf8, -1, NULL, 0);
if (wcLen > 0)
{
wcLen = MultiByteToWideChar(CP_UTF8, 0, utf8, -1, wc.GetBuffer(wcLen), wcLen);
if (wcLen <= 0)
wc.Empty();
}
return wc.c_str();
}
LPCWSTR GetDirectory(CEStr& szDir)
{
DWORD nLen, nMax;
nMax = GetCurrentDirectory(MAX_PATH, szDir.GetBuffer(MAX_PATH));
if (!nMax)
{
szDir.Empty();
goto wrap;
}
else if (nMax > MAX_PATH)
{
nLen = GetCurrentDirectory(nMax, szDir.GetBuffer(nMax));
if (!nLen || (nLen > nMax))
{
szDir.Empty();
goto wrap;
}
}
wrap:
return szDir.IsEmpty() ? NULL : szDir.c_str();
}
bool IsNeedCmd(BOOL bRootCmd, LPCWSTR asCmdLine, CEStr &szExe,
LPCWSTR* rsArguments /*= NULL*/, BOOL* rpbNeedCutStartEndQuot /*= NULL*/,
BOOL* rpbRootIsCmdExe /*= NULL*/, BOOL* rpbAlwaysConfirmExit /*= NULL*/, BOOL* rpbAutoDisableConfirmExit /*= NULL*/)
{
bool rbNeedCutStartEndQuot = false;
bool rbRootIsCmdExe = true;
bool rbAlwaysConfirmExit = false;
bool rbAutoDisableConfirmExit = false;
wchar_t *pwszEndSpace;
if (rsArguments) *rsArguments = NULL;
bool lbRc = false;
int iRc = 0;
BOOL lbFirstWasGot = FALSE;
LPCWSTR pwszCopy;
int nLastChar;
#ifdef _DEBUG
CEStr szDbgFirst;
#endif
if (!asCmdLine || !*asCmdLine)
{
_ASSERTE(asCmdLine && *asCmdLine);
goto wrap;
}
#ifdef _DEBUG
// Это минимальные проверки, собственно к коду - не относятся
bool bIsBatch = false;
{
LPCWSTR psz = asCmdLine;
NextArg(&psz, szDbgFirst);
psz = PointToExt(szDbgFirst);
if (lstrcmpi(psz, L".cmd")==0 || lstrcmpi(psz, L".bat")==0)
bIsBatch = true;
}
#endif
if (!szExe.GetBuffer(MAX_PATH))
{
_ASSERTE(FALSE && "Failed to allocate MAX_PATH");
lbRc = true;
goto wrap;
}
szExe.Empty();
if (!asCmdLine || *asCmdLine == 0)
{
_ASSERTE(asCmdLine && *asCmdLine);
lbRc = true;
goto wrap;
}
pwszCopy = asCmdLine;
// cmd /c ""c:\program files\arc\7z.exe" -?" // да еще и внутри могут быть двойными...
// cmd /c "dir c:\"
nLastChar = lstrlenW(pwszCopy) - 1;
if (pwszCopy[0] == L'"' && pwszCopy[nLastChar] == L'"')
{
//if (pwszCopy[1] == L'"' && pwszCopy[2])
//{
// pwszCopy ++; // Отбросить первую кавычку в командах типа: ""c:\program files\arc\7z.exe" -?"
// if (rbNeedCutStartEndQuot) *rbNeedCutStartEndQuot = TRUE;
//}
//else
// глючила на ""F:\VCProject\FarPlugin\#FAR180\far.exe -new_console""
//if (wcschr(pwszCopy+1, L'"') == (pwszCopy+nLastChar)) {
// LPCWSTR pwszTemp = pwszCopy;
// // Получим первую команду (исполняемый файл?)
// if ((iRc = NextArg(&pwszTemp, szArg)) != 0) {
// //Parsing command line failed
// lbRc = true; goto wrap;
// }
// pwszCopy ++; // Отбросить первую кавычку в командах типа: "c:\arc\7z.exe -?"
// lbFirstWasGot = TRUE;
// if (rbNeedCutStartEndQuot) *rbNeedCutStartEndQuot = TRUE;
//} else
{
// Will be dequoted in 'NextArg' function. Examples
// "C:\GCC\msys\bin\make.EXE -f "makefile" COMMON="../../../plugins/common""
// ""F:\VCProject\FarPlugin\#FAR180\far.exe -new_console""
// ""cmd""
// cmd /c ""c:\program files\arc\7z.exe" -?" // да еще и внутри могут быть двойными...
// cmd /c "dir c:\"
LPCWSTR pwszTemp = pwszCopy;
// Получим первую команду (исполняемый файл?)
if ((iRc = NextArg(&pwszTemp, szExe)) != 0)
{
//Parsing command line failed
#ifdef WARN_NEED_CMD
_ASSERTE(FALSE);
#endif
lbRc = true; goto wrap;
}
if (lstrcmpiW(szExe, L"start") == 0)
{
// Команду start обрабатывает только процессор
#ifdef WARN_NEED_CMD
_ASSERTE(FALSE);
#endif
lbRc = true; goto wrap;
}
LPCWSTR pwszQ = pwszCopy + 1 + lstrlen(szExe);
wchar_t* pszExpand = NULL;
if (*pwszQ != L'"' && IsExecutable(szExe, &pszExpand))
{
pwszCopy ++; // отбрасываем
lbFirstWasGot = TRUE;
if (pszExpand)
{
szExe.Set(pszExpand);
SafeFree(pszExpand);
if (rsArguments)
*rsArguments = pwszQ;
}
rbNeedCutStartEndQuot = true;
}
}
}
// Получим первую команду (исполняемый файл?)
if (!lbFirstWasGot)
{
szExe.Empty();
// `start` command must be processed by processor itself
if ((lstrcmpni(pwszCopy, L"start", 5) == 0)
&& (!pwszCopy[5] || isSpace(pwszCopy[5])))
{
#ifdef WARN_NEED_CMD
_ASSERTE(FALSE);
#endif
lbRc = true; goto wrap;
}
// 17.10.2010 - support executable file path without parameters, but with spaces in its path
// 22.11.2015 - or some weirdness, like `C:\Program Files\CB/cb_console_runner.exe "C:\sources\app.exe"`
LPCWSTR pchEnd = wcschr(pwszCopy, L' ');
if (!pchEnd)
pchEnd = pwszCopy + lstrlenW(pwszCopy);
CEStr szTemp;
DWORD nTempSize;
while (pchEnd)
{
szTemp.Set(pwszCopy, (pchEnd - pwszCopy));
_ASSERTE(szTemp[(pchEnd - pwszCopy)] == 0);
// If this is a full path without environment variables
if (((IsFilePath(szTemp, true) && !wcschr(szTemp, L'%'))
// or file/dir may be found via env.var. substitution or searching in %PATH%
|| FileExistsSearch((LPCWSTR)szTemp, szTemp))
// Than check if it is a FILE (not a directory)
&& FileExists(szTemp, &nTempSize) && nTempSize)
{
// OK, it an our executable?
if (rsArguments)
*rsArguments = pchEnd;
szExe.Set(szTemp);
break;
}
_ASSERTE(*pchEnd == 0 || *pchEnd == L' ');
if (!*pchEnd)
break;
// Find next space after nonspace
while (*(pchEnd) == L' ') pchEnd++;
// If quoted string starts from here - it's supposed to be an argument
if (*pchEnd == L'"')
{
// And we must not get here, because the executable must be already processed above
// _ASSERTE(*pchEnd != L'"');
break;
}
pchEnd = wcschr(pchEnd, L' ');
if (!pchEnd)
pchEnd = pwszCopy + lstrlenW(pwszCopy);
}
if (szExe[0] == 0)
{
if ((iRc = NextArg(&pwszCopy, szExe)) != 0)
{
//Parsing command line failed
#ifdef WARN_NEED_CMD
_ASSERTE(FALSE);
#endif
lbRc = true; goto wrap;
}
_ASSERTE(lstrcmpiW(szExe, L"start") != 0);
// Обработка переменных окружения и поиск в PATH
if (FileExistsSearch((LPCWSTR)szExe, szExe))
{
if (rsArguments)
*rsArguments = pwszCopy;
}
}
}
if (!*szExe)
{
_ASSERTE(szExe[0] != 0);
}
else
{
// Если юзеру нужен редирект - то он должен явно указать ком.процессор
// Иначе нереально (или достаточно сложно) определить, является ли символ
// редиректом, или это просто один из аргументов команды...
// "Левые" символы в имени файла (а вот в "первом аргументе" все однозначно)
if (wcspbrk(szExe, L"?*<>|"))
{
rbRootIsCmdExe = TRUE; // запуск через "процессор"
lbRc = true; goto wrap; // добавить "cmd.exe"
}
// если "путь" не указан
if (wcschr(szExe, L'\\') == NULL)
{
bool bHasExt = (wcschr(szExe, L'.') != NULL);
// Проверим, может это команда процессора (типа "DIR")?
if (!bHasExt)
{
bool bIsCommand = false;
wchar_t* pszUppr = lstrdup(szExe);
if (pszUppr)
{
// избежать линковки на user32.dll
//CharUpperBuff(pszUppr, lstrlen(pszUppr));
for (wchar_t* p = pszUppr; *p; p++)
{
if (*p >= L'a' && *p <= 'z')
*p -= 0x20;
}
const wchar_t* pszFind = gsInternalCommands;
while (*pszFind)
{
if (lstrcmp(pszUppr, pszFind) == 0)
{
bIsCommand = true;
break;
}
pszFind += lstrlen(pszFind)+1;
}
free(pszUppr);
}
if (bIsCommand)
{
#ifdef WARN_NEED_CMD
_ASSERTE(FALSE);
#endif
rbRootIsCmdExe = TRUE; // запуск через "процессор"
lbRc = true; goto wrap; // добавить "cmd.exe"
}
}
// Try to find executable in %PATH%
{
#ifndef CONEMU_MINIMAL
MWow64Disable wow; wow.Disable(); // Disable Wow64 file redirector
#endif
apiSearchPath(NULL, szExe, bHasExt ? NULL : L".exe", szExe);
}
} // end: if (wcschr(szExe, L'\\') == NULL)
}
// Если szExe не содержит путь к файлу - запускаем через cmd
// "start "" C:\Utils\Files\Hiew32\hiew32.exe C:\00\Far.exe"
if (!IsFilePath(szExe))
{
#ifdef WARN_NEED_CMD
_ASSERTE(FALSE);
#endif
rbRootIsCmdExe = TRUE; // запуск через "процессор"
lbRc = true; goto wrap; // добавить "cmd.exe"
}
//pwszCopy = wcsrchr(szArg, L'\\'); if (!pwszCopy) pwszCopy = szArg; else pwszCopy ++;
pwszCopy = PointToName(szExe);
//2009-08-27
pwszEndSpace = szExe.ms_Val + lstrlenW(szExe) - 1;
while ((*pwszEndSpace == L' ') && (pwszEndSpace > szExe))
{
*(pwszEndSpace--) = 0;
}
#ifndef __GNUC__
#pragma warning( push )
#pragma warning(disable : 6400)
#endif
if (lstrcmpiW(pwszCopy, L"cmd")==0 || lstrcmpiW(pwszCopy, L"cmd.exe")==0
|| lstrcmpiW(pwszCopy, L"tcc")==0 || lstrcmpiW(pwszCopy, L"tcc.exe")==0)
{
rbRootIsCmdExe = TRUE; // уже должен быть выставлен, но проверим
rbAlwaysConfirmExit = TRUE; rbAutoDisableConfirmExit = FALSE;
_ASSERTE(!bIsBatch);
lbRc = false; goto wrap; // уже указан командный процессор, cmd.exe в начало добавлять не нужно
}
// Issue 1211: Decide not to do weird heuristic.
// If user REALLY needs redirection (root command, huh?)
// - he must call "cmd /c ..." directly
// Если есть одна из команд перенаправления, или слияния - нужен CMD.EXE
if (!bRootCmd)
{
if (wcschr(asCmdLine, L'&') ||
wcschr(asCmdLine, L'>') ||
wcschr(asCmdLine, L'<') ||
wcschr(asCmdLine, L'|') ||
wcschr(asCmdLine, L'^') // или экранирования
)
{
#ifdef WARN_NEED_CMD
_ASSERTE(FALSE);
#endif
lbRc = true; goto wrap;
}
}
//if (lstrcmpiW(pwszCopy, L"far")==0 || lstrcmpiW(pwszCopy, L"far.exe")==0)
if (IsFarExe(pwszCopy))
{
bool bFound = (wcschr(pwszCopy, L'.') != NULL);
// Если указали при запуске просто "far" - это может быть батник, расположенный в %PATH%
if (!bFound)
{
CEStr szSearch;
if (apiSearchPath(NULL, pwszCopy, L".exe", szSearch))
{
if (lstrcmpi(PointToExt(szSearch), L".exe") == 0)
bFound = true;
}
}
if (bFound)
{
rbAutoDisableConfirmExit = TRUE;
rbRootIsCmdExe = FALSE; // FAR!
_ASSERTE(!bIsBatch);
lbRc = false; goto wrap; // уже указан исполняемый файл, cmd.exe в начало добавлять не нужно
}
}
if (IsExecutable(szExe))
{
rbRootIsCmdExe = FALSE; // Для других программ - буфер не включаем
_ASSERTE(!bIsBatch);
lbRc = false; goto wrap; // Запускается конкретная консольная программа. cmd.exe не требуется
}
//Можно еще Доделать поиски с: SearchPath, GetFullPathName, добавив расширения .exe & .com
//хотя фар сам формирует полные пути к командам, так что можно не заморачиваться
#ifdef WARN_NEED_CMD
_ASSERTE(FALSE);
#endif
rbRootIsCmdExe = TRUE;
#ifndef __GNUC__
#pragma warning( pop )
#endif
lbRc = true;
wrap:
if (rpbNeedCutStartEndQuot)
*rpbNeedCutStartEndQuot = rbNeedCutStartEndQuot;
if (rpbRootIsCmdExe)
*rpbRootIsCmdExe = rbRootIsCmdExe;
if (rpbAlwaysConfirmExit)
*rpbAlwaysConfirmExit = rbAlwaysConfirmExit;
if (rpbAutoDisableConfirmExit)
*rpbAutoDisableConfirmExit = rbAutoDisableConfirmExit;
return lbRc;
}
