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;
CmdArg 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);
}
}
void RConStartArgs::RunArgTests()
{
struct { LPCWSTR pszArg, pszNeed; } cTests[] = {
{
L"\"c:\\cmd.exe\" \"-new_console\" \"c:\\file.txt\"",
L"\"c:\\cmd.exe\" \"c:\\file.txt\""
},
{
L"\"c:\\cmd.exe\" -new_console:n \"c:\\file.txt\"",
L"\"c:\\cmd.exe\" \"c:\\file.txt\""
},
{
L"\"c:\\cmd.exe\" \"-new_console:n\" \"c:\\file.txt\"",
L"\"c:\\cmd.exe\" \"c:\\file.txt\""
},
{
L"c:\\cmd.exe \"-new_console:n\" \"c:\\file.txt\"",
L"c:\\cmd.exe \"c:\\file.txt\""
},
{
L"\"c:\\cmd.exe\" \"-new_console:n\" c:\\file.txt",
L"\"c:\\cmd.exe\" c:\\file.txt"
},
{
L"c:\\file.txt -cur_console",
L"c:\\file.txt"
},
{
L"\"c:\\file.txt\" -cur_console",
L"\"c:\\file.txt\""
},
{
L" -cur_console \"c:\\file.txt\"",
L" \"c:\\file.txt\""
},
{
L"-cur_console \"c:\\file.txt\"",
L"\"c:\\file.txt\""
},
{
L"-cur_console c:\\file.txt",
L"c:\\file.txt"
},
};
for (size_t i = 0; i < countof(cTests); i++)
{
RConStartArgs arg;
arg.pszSpecialCmd = lstrdup(cTests[i].pszArg);
arg.ProcessNewConArg();
if (lstrcmp(arg.pszSpecialCmd, cTests[i].pszNeed) != 0)
{
//_ASSERTE(FALSE && "arg.ProcessNewConArg failed");
OutputDebugString(L"arg.ProcessNewConArg failed\n");
}
int nDbg = 0;
}
}
void RConStartArgs::RunArgTests()
{
CmdArg s;
s.Set(L"Abcdef", 3);
int nDbg = lstrcmp(s, L"Abc");
_ASSERTE(nDbg==0);
s.Set(L"qwerty");
nDbg = lstrcmp(s, L"qwerty");
_ASSERTE(nDbg==0);
s.Empty();
//s.Set(L""); // !! Set("") must trigger ASSERT !!
nDbg = s.ms_Arg ? lstrcmp(s, L"") : -2;
_ASSERTE(nDbg==0);
struct { LPCWSTR pszWhole; LPCWSTR pszCmp[10]; } lsArgTest[] = {
{L"\"C:\\ConEmu\\ConEmuC64.exe\" /PARENTFARPID=1 /C \"C:\\GIT\\cmdw\\ad.cmd CE12.sln & ci -m \"Solution debug build properties\"\"",
{L"C:\\ConEmu\\ConEmuC64.exe", L"/PARENTFARPID=1", L"/C", L"C:\\GIT\\cmdw\\ad.cmd", L"CE12.sln", L"&", L"ci", L"-m", L"Solution debug build properties"}},
{L"/C \"C:\\ad.cmd file.txt & ci -m \"Commit message\"\"",
{L"/C", L"C:\\ad.cmd", L"file.txt", L"&", L"ci", L"-m", L"Commit message"}},
{L"\"This is test\" Next-arg \t\n \"Third Arg +++++++++++++++++++\" ++", {L"This is test", L"Next-arg", L"Third Arg +++++++++++++++++++"}},
{L"\"\"cmd\"\"", {L"cmd"}},
{L"\"\"c:\\Windows\\System32\\cmd.exe\" /?\"", {L"c:\\Windows\\System32\\cmd.exe", L"/?"}},
// Following example is crazy, but quotation issues may happens
//{L"First Sec\"\"ond \"Thi\"rd\" \"Fo\"\"rth\"", {L"First", L"Sec\"\"ond", L"Thi\"rd", L"Fo\"\"rth"}},
{L"First \"Fo\"\"rth\"", {L"First", L"Fo\"\"rth"}},
// Multiple commands
{L"set ConEmuReportExe=VIM.EXE & SH.EXE", {L"set", L"ConEmuReportExe=VIM.EXE", L"&", L"SH.EXE"}},
// Inside escaped arguments
{L"reg.exe add \"HKCU\\MyCo\" /ve /t REG_EXPAND_SZ /d \"\\\"C:\\ConEmu\\ConEmuPortable.exe\\\" /Dir \\\"%V\\\" /cmd \\\"cmd.exe\\\" \\\"-new_console:nC:cmd.exe\\\" \\\"-cur_console:d:%V\\\"\" /f",
// Для наглядности:
// reg.exe add "HKCU\MyCo" /ve /t REG_EXPAND_SZ
// /d "\"C:\ConEmu\ConEmuPortable.exe\" /Dir \"%V\" /cmd \"cmd.exe\" \"-new_console:nC:cmd.exe\" \"-cur_console:d:%V\"" /f
{L"reg.exe", L"add", L"HKCU\\MyCo", L"/ve", L"/t", L"REG_EXPAND_SZ", L"/d",
L"\\\"C:\\ConEmu\\ConEmuPortable.exe\\\" /Dir \\\"%V\\\" /cmd \\\"cmd.exe\\\" \\\"-new_console:nC:cmd.exe\\\" \\\"-cur_console:d:%V\\\"",
L"/f"}},
// After 'Inside escaped arguments' regression bug appears
{L"/dir \"C:\\\" /icon \"cmd.exe\" /single", {L"/dir", L"C:\\", L"/icon", L"cmd.exe", L"/single"}},
{L"cmd \"one.exe /dir \\\"C:\\\\\" /log\" \"two.exe /dir \\\"C:\\\" /log\" end", {L"cmd", L"one.exe /dir \\\"C:\\\\\" /log", L"two.exe /dir \\\"C:\\\" /log", L"end"}},
{NULL}
};
for (int i = 0; lsArgTest[i].pszWhole; i++)
{
s.Empty();
LPCWSTR pszTestCmd = lsArgTest[i].pszWhole;
int j = -1;
while (lsArgTest[i].pszCmp[++j])
{
if (NextArg(&pszTestCmd, s) != 0)
{
_ASSERTE(FALSE && "Fails on token!");
}
else
{
nDbg = lstrcmp(s, lsArgTest[i].pszCmp[j]);
_ASSERTE(nDbg==0);
}
}
}
bool bTest = true;
for (size_t i = 0; bTest; i++)
{
RConStartArgs arg;
int nDbg;
LPCWSTR pszCmp;
switch (i)
{
case 21:
pszCmp = L"cmd '-new_console' `-new_console` \\\"-new_console\\\"";
arg.pszSpecialCmd = lstrdup(pszCmp);
arg.ProcessNewConArg();
_ASSERTE(0==lstrcmp(arg.pszSpecialCmd, pszCmp) && arg.NewConsole==crb_Undefined);
break;
case 20:
arg.pszSpecialCmd = lstrdup(L"\"c:\\cmd.exe\" \"-new_console\" \"c:\\file.txt\"");
arg.ProcessNewConArg();
_ASSERTE(0==lstrcmp(arg.pszSpecialCmd, L"\"c:\\cmd.exe\" \"c:\\file.txt\""));
break;
case 19:
arg.pszSpecialCmd = lstrdup(L"\"c:\\cmd.exe\" -new_console:n \"c:\\file.txt\"");
arg.ProcessNewConArg();
_ASSERTE(0==lstrcmp(arg.pszSpecialCmd, L"\"c:\\cmd.exe\" \"c:\\file.txt\""));
break;
case 18:
arg.pszSpecialCmd = lstrdup(L"\"c:\\cmd.exe\" \"-new_console:n\" \"c:\\file.txt\"");
arg.ProcessNewConArg();
_ASSERTE(0==lstrcmp(arg.pszSpecialCmd, L"\"c:\\cmd.exe\" \"c:\\file.txt\""));
break;
case 17:
arg.pszSpecialCmd = lstrdup(L"c:\\cmd.exe \"-new_console:n\" \"c:\\file.txt\"");
arg.ProcessNewConArg();
_ASSERTE(0==lstrcmp(arg.pszSpecialCmd, L"c:\\cmd.exe \"c:\\file.txt\""));
break;
case 16:
arg.pszSpecialCmd = lstrdup(L"\"c:\\cmd.exe\" \"-new_console:n\" c:\\file.txt");
arg.ProcessNewConArg();
_ASSERTE(0==lstrcmp(arg.pszSpecialCmd, L"\"c:\\cmd.exe\" c:\\file.txt"));
break;
case 15:
arg.pszSpecialCmd = lstrdup(L"c:\\file.txt -cur_console");
arg.ProcessNewConArg();
_ASSERTE(0==lstrcmp(arg.pszSpecialCmd, L"c:\\file.txt"));
break;
case 14:
arg.pszSpecialCmd = lstrdup(L"\"c:\\file.txt\" -cur_console");
arg.ProcessNewConArg();
_ASSERTE(0==lstrcmp(arg.pszSpecialCmd, L"\"c:\\file.txt\""));
break;
case 13:
arg.pszSpecialCmd = lstrdup(L" -cur_console \"c:\\file.txt\"");
arg.ProcessNewConArg();
_ASSERTE(0==lstrcmp(arg.pszSpecialCmd, L"\"c:\\file.txt\""));
break;
case 12:
arg.pszSpecialCmd = lstrdup(L"-cur_console \"c:\\file.txt\"");
arg.ProcessNewConArg();
_ASSERTE(0==lstrcmp(arg.pszSpecialCmd, L"\"c:\\file.txt\""));
break;
case 11:
arg.pszSpecialCmd = lstrdup(L"-cur_console c:\\file.txt");
arg.ProcessNewConArg();
_ASSERTE(0==lstrcmp(arg.pszSpecialCmd, L"c:\\file.txt"));
break;
case 10:
pszCmp = L"reg.exe add \"HKCU\\command\" /ve /t REG_EXPAND_SZ /d \"\\\"C:\\ConEmu\\ConEmuPortable.exe\\\" /Dir \\\"%V\\\" /cmd \\\"cmd.exe\\\" \\\"-new_console:nC:cmd.exe\\\" \\\"-cur_console:d:%V\\\"\" /f";
arg.pszSpecialCmd = lstrdup(pszCmp);
arg.ProcessNewConArg();
_ASSERTE(lstrcmp(arg.pszSpecialCmd, pszCmp)==0 && arg.NewConsole==crb_Undefined);
break;
case 9:
pszCmp = L"\"C:\\Windows\\system32\\cmd.exe\" /C \"\"C:\\Python27\\python.EXE\"\"";
arg.pszSpecialCmd = lstrdup(pszCmp);
arg.ProcessNewConArg();
_ASSERTE(lstrcmp(arg.pszSpecialCmd, pszCmp)==0);
break;
case 8:
arg.pszSpecialCmd = lstrdup(L"cmd --new_console -cur_console:a");
arg.ProcessNewConArg();
_ASSERTE(lstrcmp(arg.pszSpecialCmd, L"cmd --new_console")==0 && arg.NewConsole==crb_Undefined && arg.RunAsAdministrator==crb_On);
break;
case 7:
arg.pszSpecialCmd = lstrdup(L"cmd -cur_console:d:\"C:\\My docs\":t:\"My title\" \"-cur_console:C:C:\\my cmd.ico\" -cur_console:P:\"<PowerShell>\":a /k ver");
arg.ProcessNewConArg();
pszCmp = L"cmd /k ver";
_ASSERTE(lstrcmp(arg.pszSpecialCmd, pszCmp)==0);
_ASSERTE(arg.pszRenameTab && arg.pszPalette && arg.pszIconFile && arg.pszStartupDir && arg.NewConsole==crb_Undefined && lstrcmp(arg.pszRenameTab, L"My title")==0 && lstrcmp(arg.pszPalette, L"<PowerShell>")==0 && lstrcmp(arg.pszStartupDir, L"C:\\My docs")==0 && lstrcmp(arg.pszIconFile, L"C:\\my cmd.ico")==0);
break;
case 6:
arg.pszSpecialCmd = lstrdup(L"cmd -cur_console:b:P:\"^<Power\"\"Shell^>\":t:\"My title\" /k ConEmuC.exe -Guimacro print(\"-new_console:a\")");
arg.ProcessNewConArg();
pszCmp = L"cmd /k ConEmuC.exe -Guimacro print(\"-new_console:a\")";
_ASSERTE(lstrcmp(arg.pszSpecialCmd, pszCmp)==0);
_ASSERTE(arg.pszRenameTab && arg.pszPalette && arg.BackgroundTab==crb_On && arg.NewConsole==crb_Undefined && arg.RunAsAdministrator==crb_Undefined && lstrcmp(arg.pszRenameTab, L"My title")==0 && lstrcmp(arg.pszPalette, L"<Power\"Shell>")==0);
break;
case 5:
arg.pszSpecialCmd = lstrdup(L"cmd \"-cur_console:t:My title\" /k ver");
arg.ProcessNewConArg();
_ASSERTE(lstrcmp(arg.pszSpecialCmd, L"cmd /k ver")==0);
_ASSERTE(arg.pszRenameTab && lstrcmp(arg.pszRenameTab, L"My title")==0 && arg.NewConsole==crb_Undefined);
break;
case 4:
arg.pszSpecialCmd = lstrdup(L"cmd \"-new_console:P:^<Power\"\"Shell^>\"");
arg.ProcessNewConArg();
_ASSERTE(lstrcmp(arg.pszSpecialCmd, L"cmd")==0);
nDbg = lstrcmp(arg.pszPalette, L"<Power\"Shell>");
_ASSERTE(nDbg==0 && arg.NewConsole==crb_On);
break;
case 3:
arg.pszSpecialCmd = lstrdup(L"cmd -cur_console:u:Max:");
arg.ProcessNewConArg();
_ASSERTE(lstrcmp(arg.pszSpecialCmd, L"cmd")==0);
nDbg = lstrcmp(arg.pszUserName,L"Max");
_ASSERTE(nDbg==0 && arg.pszDomain==NULL && !*arg.szUserPassword && arg.ForceUserDialog==crb_Off && arg.NewConsole!=crb_On);
break;
case 2:
arg.pszSpecialCmd = lstrdup(L"cmd -cur_console:u:Max -new_console");
arg.ProcessNewConArg();
_ASSERTE(lstrcmp(arg.pszSpecialCmd, L"cmd")==0);
nDbg = lstrcmp(arg.pszUserName,L"Max");
_ASSERTE(nDbg==0 && arg.pszDomain==NULL && !*arg.szUserPassword && arg.ForceUserDialog==crb_On && arg.NewConsole==crb_On);
break;
case 1:
arg.pszSpecialCmd = lstrdup(L"cmd -new_console:u -cur_console:h0");
arg.ProcessNewConArg();
_ASSERTE(lstrcmp(arg.pszSpecialCmd, L"cmd")==0);
_ASSERTE(arg.pszUserName==NULL && arg.pszDomain==NULL && arg.ForceUserDialog==crb_On && arg.NewConsole==crb_On && arg.BufHeight==crb_On && arg.nBufHeight==0);
break;
case 0:
arg.pszSpecialCmd = lstrdup(L"cmd \"-new_console:d:C:\\John Doe\\Home\" ");
arg.ProcessNewConArg();
_ASSERTE(lstrcmp(arg.pszSpecialCmd, L"cmd ")==0);
nDbg = lstrcmp(arg.pszStartupDir, L"C:\\John Doe\\Home");
_ASSERTE(nDbg==0 && arg.NewConsole==crb_On);
break;
default:
bTest = false; // Stop tests
}
}
nDbg = -1;
}
void RConStartArgs::RunArgTests()
{
struct { LPCWSTR pszArg, pszNeed; } cTests[] = {
{
L"\"c:\\cmd.exe\" \"-new_console\" \"c:\\file.txt\"",
L"\"c:\\cmd.exe\" \"c:\\file.txt\""
},
{
L"\"c:\\cmd.exe\" -new_console:n \"c:\\file.txt\"",
L"\"c:\\cmd.exe\" \"c:\\file.txt\""
},
{
L"\"c:\\cmd.exe\" \"-new_console:n\" \"c:\\file.txt\"",
L"\"c:\\cmd.exe\" \"c:\\file.txt\""
},
{
L"c:\\cmd.exe \"-new_console:n\" \"c:\\file.txt\"",
L"c:\\cmd.exe \"c:\\file.txt\""
},
{
L"\"c:\\cmd.exe\" \"-new_console:n\" c:\\file.txt",
L"\"c:\\cmd.exe\" c:\\file.txt"
},
{
L"c:\\file.txt -cur_console",
L"c:\\file.txt"
},
{
L"\"c:\\file.txt\" -cur_console",
L"\"c:\\file.txt\""
},
{
L" -cur_console \"c:\\file.txt\"",
L" \"c:\\file.txt\""
},
{
L"-cur_console \"c:\\file.txt\"",
L"\"c:\\file.txt\""
},
{
L"-cur_console c:\\file.txt",
L"c:\\file.txt"
},
};
for (size_t i = 0; i < countof(cTests); i++)
{
RConStartArgs arg;
arg.pszSpecialCmd = lstrdup(cTests[i].pszArg);
arg.ProcessNewConArg();
if (lstrcmp(arg.pszSpecialCmd, cTests[i].pszNeed) != 0)
{
//_ASSERTE(FALSE && "arg.ProcessNewConArg failed");
OutputDebugString(L"arg.ProcessNewConArg failed\n");
}
int nDbg = 0;
}
for (size_t i = 0; i <= 4; i++)
{
RConStartArgs arg;
switch (i)
{
case 0: arg.pszSpecialCmd = lstrdup(L"cmd \"-new_console:d:C:\\John Doe\\Home\" "); break;
case 1: arg.pszSpecialCmd = lstrdup(L"cmd -cur_console:u"); break;
case 2: arg.pszSpecialCmd = lstrdup(L"cmd -cur_console:u:Max"); break;
case 3: arg.pszSpecialCmd = lstrdup(L"cmd -cur_console:u:Max:"); break;
case 4: arg.pszSpecialCmd = lstrdup(L"cmd \"-new_console:P:^<Power\"\"Shell^>\""); break;
}
arg.ProcessNewConArg();
int nDbg = lstrcmp(arg.pszSpecialCmd, i ? L"cmd" : L"cmd ");
_ASSERTE(nDbg==0);
switch (i)
{
case 0:
nDbg = lstrcmp(arg.pszStartupDir, L"C:\\John Doe\\Home");
_ASSERTE(nDbg==0);
break;
case 1:
_ASSERTE(arg.pszUserName==NULL && arg.pszDomain==NULL && arg.bForceUserDialog);
break;
case 2:
nDbg = lstrcmp(arg.pszUserName,L"Max");
_ASSERTE(nDbg==0 && arg.pszDomain==NULL && !*arg.szUserPassword && arg.bForceUserDialog);
break;
case 3:
nDbg = lstrcmp(arg.pszUserName,L"Max");
_ASSERTE(nDbg==0 && arg.pszDomain==NULL && !*arg.szUserPassword && !arg.bForceUserDialog);
break;
case 4:
nDbg = lstrcmp(arg.pszPalette, L"<Power\"Shell>");
_ASSERTE(nDbg==0);
break;
}
}
CmdArg s;
s.Set(L"Abcdef", 3);
int nDbg = lstrcmp(s, L"Abc");
_ASSERTE(nDbg==0);
s.Set(L"qwerty");
nDbg = lstrcmp(s, L"qwerty");
_ASSERTE(nDbg==0);
s.Empty();
//s.Set(L""); // !! Set("") must trigger ASSERT !!
nDbg = s.ms_Arg ? lstrcmp(s, L"") : -2;
_ASSERTE(nDbg==0);
struct { LPCWSTR pszWhole; LPCWSTR pszCmp[5]; } lsArgTest[] = {
{L"\"This is test\" Next-arg \t\n \"Third Arg +++++++++++++++++++\" ++", {L"This is test", L"Next-arg", L"Third Arg +++++++++++++++++++"}},
{L"\"\"cmd\"\"", {L"cmd"}},
{L"\"\"c:\\Windows\\System32\\cmd.exe\" /?\"", {L"c:\\Windows\\System32\\cmd.exe", L"/?"}},
// Following example is crazy, but quotation issues may happens
//{L"First Sec\"\"ond \"Thi\"rd\" \"Fo\"\"rth\"", {L"First", L"Sec\"\"ond", L"Thi\"rd", L"Fo\"\"rth"}},
{L"First \"Fo\"\"rth\"", {L"First", L"Fo\"\"rth"}},
// Multiple commands
{L"set ConEmuReportExe=VIM.EXE & SH.EXE", {L"set", L"ConEmuReportExe=VIM.EXE", L"&", L"SH.EXE"}},
{NULL}
};
for (int i = 0; lsArgTest[i].pszWhole; i++)
{
s.Empty();
LPCWSTR pszTestCmd = lsArgTest[i].pszWhole;
int j = -1;
while (lsArgTest[i].pszCmp[++j])
{
if (NextArg(&pszTestCmd, s) != 0)
{
_ASSERTE(FALSE && "Fails on token!");
}
else
{
nDbg = lstrcmp(s, lsArgTest[i].pszCmp[j]);
_ASSERTE(nDbg==0);
}
}
}
nDbg = -1;
}
//// Эта функция пайп не закрывает!
//void CGuiServer::GuiServerThreadCommand(HANDLE hPipe)
BOOL CGuiServer::GuiServerCommand(LPVOID pInst, CESERVER_REQ* pIn, CESERVER_REQ* &ppReply, DWORD &pcbReplySize, DWORD &pcbMaxReplySize, LPARAM lParam)
{
BOOL lbRc = FALSE;
CGuiServer* pGSrv = (CGuiServer*)lParam;
if (!pGSrv)
{
_ASSERTE(((CGuiServer*)lParam)!=NULL);
pGSrv = &gpConEmu->m_GuiServer;
}
if (pIn->hdr.bAsync)
pGSrv->mp_GuiServer->BreakConnection(pInst);
gpSetCls->debugLogCommand(pIn, TRUE, timeGetTime(), 0, pGSrv ? pGSrv->ms_ServerPipe : NULL);
#ifdef _DEBUG
UINT nDataSize = pIn->hdr.cbSize - sizeof(CESERVER_REQ_HDR);
#endif
// Все данные из пайпа получены, обрабатываем команду и возвращаем (если нужно) результат
#ifdef ALLOW_WINE_MSG
if (gbIsWine)
{
wchar_t szMsg[128];
msprintf(szMsg, countof(szMsg), L"CGuiServer::GuiServerCommand.\nGUI TID=%u\nSrcPID=%u, SrcTID=%u, Cmd=%u",
GetCurrentThreadId(), pIn->hdr.nSrcPID, pIn->hdr.nSrcThreadId, pIn->hdr.nCmd);
MessageBox(szMsg, MB_ICONINFORMATION);
}
#endif
switch (pIn->hdr.nCmd)
{
case CECMD_NEWCMD:
{
// Приходит из другой копии ConEmu.exe, когда она запущена с ключом /single, /showhide, /showhideTSA
DEBUGSTRCMD(L"GUI recieved CECMD_NEWCMD\n");
LPCWSTR pszCommand = pIn->NewCmd.GetCommand();
_ASSERTE(pszCommand!=NULL && "Must be at least empty string but NOT NULL");
if (pIn->NewCmd.isAdvLogging && !gpSetCls->isAdvLogging)
{
gpSetCls->isAdvLogging = pIn->NewCmd.isAdvLogging;
gpConEmu->CreateLog();
}
if (gpSetCls->isAdvLogging && (pIn->NewCmd.isAdvLogging > gpSetCls->isAdvLogging))
{
wchar_t szLogLevel[80];
_wsprintf(szLogLevel, SKIPLEN(countof(szLogLevel)) L"Changing log level! Old=%u, New=%u", (UINT)gpSetCls->isAdvLogging, (UINT)pIn->NewCmd.isAdvLogging);
gpConEmu->LogString(szLogLevel);
gpSetCls->isAdvLogging = pIn->NewCmd.isAdvLogging;
}
if (gpSetCls->isAdvLogging)
{
size_t cchAll = 120 + _tcslen(pIn->NewCmd.szConEmu) + _tcslen(pIn->NewCmd.szCurDir) + _tcslen(pszCommand);
wchar_t* pszInfo = (wchar_t*)malloc(cchAll*sizeof(*pszInfo));
if (pszInfo)
{
_wsprintf(pszInfo, SKIPLEN(cchAll) L"CECMD_NEWCMD: Wnd=x%08X, Act=%u, ConEmu=%s, Dir=%s, Cmd=%s",
(DWORD)(DWORD_PTR)pIn->NewCmd.hFromConWnd, pIn->NewCmd.ShowHide,
pIn->NewCmd.szConEmu, pIn->NewCmd.szCurDir, pszCommand);
gpConEmu->LogString(pszInfo);
free(pszInfo);
}
}
BOOL bAccepted = FALSE;
if (pIn->NewCmd.szConEmu[0])
{
bAccepted = (lstrcmpi(gpConEmu->ms_ConEmuExeDir, pIn->NewCmd.szConEmu) == 0);
}
else
{
bAccepted = TRUE;
}
if (bAccepted)
{
bool bCreateTab = (pIn->NewCmd.ShowHide == sih_None || pIn->NewCmd.ShowHide == sih_StartDetached)
// Issue 1275: When minimized into TSA (on all VCon are closed) we need to restore and run new tab
|| (pszCommand[0] && !CVConGroup::isVConExists(0));
RConStartArgs rTest;
if (pszCommand[0])
{
rTest.pszSpecialCmd = lstrdup(pszCommand);
rTest.ProcessNewConArg();
}
bool bSkipActivation = false;
// Может быть пусто
if (bCreateTab && pszCommand[0])
{
RConStartArgs *pArgs = new RConStartArgs;
// New tab must be started with same credentials that calling tab if others was not specified
{
if (!rTest.HasInheritedArgs())
{
CVConGuard VCon;
if ((pIn->NewCmd.hFromConWnd || pIn->NewCmd.hFromDcWnd)
&& CVConGroup::GetVConByHWND(pIn->NewCmd.hFromConWnd, pIn->NewCmd.hFromDcWnd, &VCon))
{
const RConStartArgs& r = VCon->RCon()->GetArgs();
if (r.HasInheritedArgs())
{
pArgs->AssignInheritedArgs(&r);
}
}
}
}
pArgs->Detached = (pIn->NewCmd.ShowHide == sih_StartDetached) ? crb_On : crb_Off;
pArgs->pszSpecialCmd = lstrdup(pszCommand);
if (pIn->NewCmd.szCurDir[0] == 0)
{
_ASSERTE(pIn->NewCmd.szCurDir[0] != 0);
}
else
{
pArgs->pszStartupDir = lstrdup(pIn->NewCmd.szCurDir);
}
LPCWSTR pszStrings = pIn->NewCmd.GetEnvStrings();
if (pszStrings && pIn->NewCmd.cchEnvStrings)
{
size_t cbBytes = pIn->NewCmd.cchEnvStrings*sizeof(*pArgs->pszEnvStrings);
pArgs->pszEnvStrings = (wchar_t*)malloc(cbBytes);
if (pArgs->pszEnvStrings)
{
memmove(pArgs->pszEnvStrings, pszStrings, cbBytes);
pArgs->cchEnvStrings = pIn->NewCmd.cchEnvStrings;
}
}
if (gpSetCls->IsMulti() || CVConGroup::isDetached())
{
gpConEmu->PostCreateCon(pArgs);
}
else
{
// Если хотят в одном окне - только одну консоль
gpConEmu->CreateWnd(pArgs);
SafeDelete(pArgs);
// New window created? Don't activate this one.
bSkipActivation = true;
}
}
else
{
_ASSERTE(pIn->NewCmd.ShowHide==sih_ShowMinimize || pIn->NewCmd.ShowHide==sih_ShowHideTSA || pIn->NewCmd.ShowHide==sih_Show);
}
// gh#151: Do animation after starting tab creation
if (!bSkipActivation && (rTest.BackgroundTab != crb_On))
{
gpConEmu->DoMinimizeRestore(bCreateTab ? sih_SetForeground : pIn->NewCmd.ShowHide);
}
}
pcbReplySize = sizeof(CESERVER_REQ_HDR)+sizeof(BYTE);
lbRc = ExecuteNewCmd(ppReply, pcbMaxReplySize, pIn->hdr.nCmd, pcbReplySize);
if (lbRc)
{
ppReply->Data[0] = bAccepted;
}
break;
} //CECMD_NEWCMD
case CECMD_TABSCMD:
{
// 0: спрятать/показать табы, 1: перейти на следующую, 2: перейти на предыдущую, 3: commit switch
DEBUGSTRCMD(L"GUI recieved CECMD_TABSCMD\n");
_ASSERTE(nDataSize>=1);
DWORD nTabCmd = pIn->Data[0];
gpConEmu->TabCommand((ConEmuTabCommand)nTabCmd);
pcbReplySize = sizeof(CESERVER_REQ_HDR)+sizeof(BYTE);
if (ExecuteNewCmd(ppReply, pcbMaxReplySize, pIn->hdr.nCmd, pcbReplySize))
{
lbRc = TRUE;
ppReply->Data[0] = TRUE;
}
break;
} // CECMD_TABSCMD
case CECMD_ATTACH2GUI:
{
// Получен запрос на Attach из сервера
MCHKHEAP;
CESERVER_REQ_SRVSTARTSTOPRET Ret = {};
lbRc = CVConGroup::AttachRequested(pIn->StartStop.hWnd, &(pIn->StartStop), Ret);
pcbReplySize = sizeof(CESERVER_REQ_HDR)+sizeof(CESERVER_REQ_SRVSTARTSTOPRET)+(Ret.cchEnvCommands*sizeof(wchar_t));
if (!ExecuteNewCmd(ppReply, pcbMaxReplySize, pIn->hdr.nCmd, pcbReplySize))
{
SafeFree(Ret.pszCommands);
goto wrap;
}
if (lbRc)
{
_ASSERTE(sizeof(ppReply->SrvStartStopRet) == sizeof(Ret));
memmove(&ppReply->SrvStartStopRet, &Ret, sizeof(Ret));
// Environment strings (inherited from parent console)
if (Ret.cchEnvCommands && Ret.pszCommands)
{
memmove(ppReply->SrvStartStopRet.szCommands, Ret.pszCommands, Ret.cchEnvCommands*sizeof(wchar_t));
ppReply->SrvStartStopRet.cchEnvCommands = Ret.cchEnvCommands;
}
else
{
ppReply->SrvStartStopRet.cchEnvCommands = 0;
}
SafeFree(Ret.pszCommands);
_ASSERTE((ppReply->StartStopRet.nBufferHeight == 0) || ((int)ppReply->StartStopRet.nBufferHeight > (pIn->StartStop.sbi.srWindow.Bottom-pIn->StartStop.sbi.srWindow.Top)));
}
MCHKHEAP;
break;
} // CECMD_ATTACH2GUI
case CECMD_SRVSTARTSTOP:
{
MCHKHEAP;
// SRVSTART не приходит если запускается cmd под админом или из Win+G
bool lbAllocated = false;
if (pIn->SrvStartStop.Started == srv_Started)
{
// Запущен процесс сервера
HWND hConWnd = (HWND)pIn->dwData[1];
_ASSERTE(hConWnd && IsWindow(hConWnd));
DWORD nStartTick = timeGetTime();
struct MsgSrvStartedArg
{
HWND hConWnd;
DWORD nSrcPID;
DWORD dwKeybLayout;
DWORD timeStart;
DWORD timeRecv;
DWORD timeFin;
CESERVER_REQ_SRVSTARTSTOPRET Ret;
//111002 - вернуть должен HWND окна отрисовки (дочернее окно ConEmu)
static LRESULT OnSrvStarted(LPARAM lParam)
{
MsgSrvStartedArg *pArg = (MsgSrvStartedArg*)lParam;
HWND hWndDC = NULL;
DWORD nServerPID = pArg->nSrcPID;
HWND hWndCon = pArg->hConWnd;
DWORD dwKeybLayout = pArg->dwKeybLayout;
pArg->timeRecv = timeGetTime();
DWORD t1, t2; int iFound = -1;
hWndDC = CVConGroup::DoSrvCreated(nServerPID, hWndCon, dwKeybLayout, t1, t2, iFound, pArg->Ret);
UNREFERENCED_PARAMETER(dwKeybLayout);
UNREFERENCED_PARAMETER(hWndCon);
pArg->timeFin = timeGetTime();
if (hWndDC == NULL)
{
_ASSERTE(hWndDC!=NULL);
}
else
{
#ifdef _DEBUG
DWORD nRecvDur = pArg->timeRecv - pArg->timeStart;
DWORD nProcDur = pArg->timeFin - pArg->timeRecv;
#define MSGSTARTED_TIMEOUT 10000
if ((nRecvDur > MSGSTARTED_TIMEOUT) || (nProcDur > MSGSTARTED_TIMEOUT))
{
_ASSERTE((nRecvDur <= MSGSTARTED_TIMEOUT) && (nProcDur <= MSGSTARTED_TIMEOUT));
}
#endif
}
return (LRESULT)hWndDC;
};
} arg = {hConWnd, pIn->hdr.nSrcPID, pIn->SrvStartStop.dwKeybLayout, nStartTick};
gpConEmu->CallMainThread(true, arg.OnSrvStarted, (LPARAM)&arg);
HWND hWndDC = arg.Ret.Info.hWndDc;
HWND hWndBack = arg.Ret.Info.hWndBack;
_ASSERTE(hWndDC!=NULL);
#ifdef _DEBUG
DWORD dwErr = GetLastError(), nEndTick = timeGetTime(), nDelta = nEndTick - nStartTick;
if (hWndDC && nDelta >= EXECUTE_CMD_WARN_TIMEOUT)
{
if (!IsDebuggerPresent())
{
//_ASSERTE(nDelta <= EXECUTE_CMD_WARN_TIMEOUT || (pIn->hdr.nCmd == CECMD_CMDSTARTSTOP && nDelta <= EXECUTE_CMD_WARN_TIMEOUT2));
_ASSERTEX(nDelta <= EXECUTE_CMD_WARN_TIMEOUT);
}
}
#endif
pcbReplySize = sizeof(CESERVER_REQ_HDR)+sizeof(CESERVER_REQ_SRVSTARTSTOPRET)+(arg.Ret.cchEnvCommands*sizeof(wchar_t));
if (!ExecuteNewCmd(ppReply, pcbMaxReplySize, pIn->hdr.nCmd, pcbReplySize))
{
SafeFree(arg.Ret.pszCommands);
goto wrap;
}
lbAllocated = true;
_ASSERTE(sizeof(ppReply->SrvStartStopRet) == sizeof(arg.Ret));
memmove(&ppReply->SrvStartStopRet, &arg.Ret, sizeof(arg.Ret));
// Environment strings (inherited from parent console)
if (arg.Ret.cchEnvCommands && arg.Ret.pszCommands)
{
memmove(ppReply->SrvStartStopRet.szCommands, arg.Ret.pszCommands, arg.Ret.cchEnvCommands*sizeof(wchar_t));
ppReply->SrvStartStopRet.cchEnvCommands = arg.Ret.cchEnvCommands;
}
else
{
ppReply->SrvStartStopRet.cchEnvCommands = 0;
}
SafeFree(arg.Ret.pszCommands);
}
else if (pIn->SrvStartStop.Started == srv_Stopped)
{
// Процесс сервера завершается
CRealConsole* pRCon = NULL;
CVConGuard VCon;
for (size_t i = 0;; i++)
{
if (!CVConGroup::GetVCon(i, &VCon))
break;
pRCon = VCon->RCon();
if (pRCon && (pRCon->GetServerPID(true) == pIn->hdr.nSrcPID || pRCon->GetServerPID(false) == pIn->hdr.nSrcPID))
{
break;
}
pRCon = NULL;
}
gpConEmu->mn_ShellExitCode = pIn->SrvStartStop.nShellExitCode;
if (pRCon)
pRCon->OnServerClosing(pIn->hdr.nSrcPID, &pIn->SrvStartStop.nShellExitCode);
//pIn->dwData[0] = 1;
}
else
{
_ASSERTE((pIn->dwData[0] == 1) || (pIn->dwData[0] == 101));
}
MCHKHEAP;
if (!lbAllocated)
{
pcbReplySize = sizeof(CESERVER_REQ_HDR)+sizeof(CESERVER_REQ_SRVSTARTSTOPRET);
if (!ExecuteNewCmd(ppReply, pcbMaxReplySize, pIn->hdr.nCmd, pcbReplySize))
goto wrap;
}
lbRc = TRUE;
//// Отправляем
//fSuccess = WriteFile(
// hPipe, // handle to pipe
// pOut, // buffer to write from
// pOut->hdr.cbSize, // number of bytes to write
// &cbWritten, // number of bytes written
// NULL); // not overlapped I/O
//ExecuteFreeResult(pOut);
break;
} // CECMD_SRVSTARTSTOP
case CECMD_ASSERT:
{
DWORD nBtn = MessageBox(NULL, pIn->AssertInfo.szDebugInfo, pIn->AssertInfo.szTitle, pIn->AssertInfo.nBtns);
pcbReplySize = sizeof(CESERVER_REQ_HDR)+sizeof(DWORD);
if (ExecuteNewCmd(ppReply, pcbMaxReplySize, pIn->hdr.nCmd, pcbReplySize))
{
lbRc = TRUE;
ppReply->dwData[0] = nBtn;
}
//ExecutePrepareCmd(&pIn->hdr, CECMD_ASSERT, sizeof(CESERVER_REQ_HDR) + sizeof(DWORD));
//pIn->dwData[0] = nBtn;
//// Отправляем
//fSuccess = WriteFile(
// hPipe, // handle to pipe
// pIn, // buffer to write from
// pIn->hdr.cbSize, // number of bytes to write
// &cbWritten, // number of bytes written
// NULL); // not overlapped I/O
break;
} // CECMD_ASSERT
case CECMD_ATTACHGUIAPP:
{
pcbReplySize = sizeof(CESERVER_REQ_HDR)+sizeof(CESERVER_REQ_ATTACHGUIAPP);
if (!ExecuteNewCmd(ppReply, pcbMaxReplySize, pIn->hdr.nCmd, pcbReplySize))
goto wrap;
ppReply->AttachGuiApp = pIn->AttachGuiApp;
//CESERVER_REQ Out;
//ExecutePrepareCmd(&Out.hdr, CECMD_ATTACHGUIAPP, sizeof(CESERVER_REQ_HDR)+sizeof(Out.AttachGuiApp));
//Out.AttachGuiApp = pIn->AttachGuiApp;
#ifdef SHOW_GUIATTACH_START
if (pIn->AttachGuiApp.hWindow == NULL)
{
wchar_t szDbg[1024];
_wsprintf(szDbg, SKIPLEN(countof(szDbg)) L"AttachGuiApp requested from:\n%s\nPID=%u", pIn->AttachGuiApp.sAppFilePathName, pIn->AttachGuiApp.nPID);
//MBoxA(szDbg);
MessageBox(NULL, szDbg, L"ConEmu", MB_SYSTEMMODAL);
}
#endif
// Уведомить ожидающую вкладку
CRealConsole* pRCon = CVConGroup::AttachRequestedGui(pIn->AttachGuiApp.nServerPID, pIn->AttachGuiApp.sAppFilePathName, pIn->AttachGuiApp.nPID);
if (pRCon)
{
CVConGuard VCon(pRCon->VCon());
RECT rcPrev = ppReply->AttachGuiApp.rcWindow;
HWND hBack = pRCon->VCon()->GetBack();
//// Размер должен быть независим от возможности наличия прокрутки в VCon
//GetWindowRect(hBack, &ppReply->AttachGuiApp.rcWindow);
//ppReply->AttachGuiApp.rcWindow.right -= ppReply->AttachGuiApp.rcWindow.left;
//ppReply->AttachGuiApp.rcWindow.bottom -= ppReply->AttachGuiApp.rcWindow.top;
//ppReply->AttachGuiApp.rcWindow.left = ppReply->AttachGuiApp.rcWindow.top = 0;
////MapWindowPoints(NULL, hBack, (LPPOINT)&ppReply->AttachGuiApp.rcWindow, 2);
//pRCon->CorrectGuiChildRect(ppReply->AttachGuiApp.nStyle, ppReply->AttachGuiApp.nStyleEx, ppReply->AttachGuiApp.rcWindow);
// Уведомить RCon и ConEmuC, что гуй подцепился
// Вызывается два раза. Первый (при запуске exe) ahGuiWnd==NULL, второй - после фактического создания окна
pRCon->SetGuiMode(pIn->AttachGuiApp.nFlags, pIn->AttachGuiApp.hAppWindow, pIn->AttachGuiApp.Styles.nStyle, pIn->AttachGuiApp.Styles.nStyleEx, pIn->AttachGuiApp.sAppFilePathName, pIn->AttachGuiApp.nPID, pIn->hdr.nBits, rcPrev);
ppReply->AttachGuiApp.nFlags = agaf_Success | (pRCon->isActive(false) ? 0 : agaf_Inactive);
ppReply->AttachGuiApp.nServerPID = pRCon->GetServerPID();
ppReply->AttachGuiApp.nPID = ppReply->AttachGuiApp.nServerPID;
ppReply->AttachGuiApp.hConEmuDc = pRCon->GetView();
ppReply->AttachGuiApp.hConEmuBack = hBack;
ppReply->AttachGuiApp.hConEmuWnd = ghWnd;
ppReply->AttachGuiApp.hAppWindow = pIn->AttachGuiApp.hAppWindow;
ppReply->AttachGuiApp.hSrvConWnd = pRCon->ConWnd();
ppReply->AttachGuiApp.hkl = (DWORD)(LONG)(LONG_PTR)GetKeyboardLayout(gpConEmu->mn_MainThreadId);
ZeroStruct(ppReply->AttachGuiApp.Styles.Shifts);
CRealConsole::CorrectGuiChildRect(pIn->AttachGuiApp.Styles.nStyle, pIn->AttachGuiApp.Styles.nStyleEx, ppReply->AttachGuiApp.Styles.Shifts, pIn->AttachGuiApp.sAppFilePathName);
}
else
{
ppReply->AttachGuiApp.nFlags = agaf_Fail;
_ASSERTE(FALSE && "No one tab is waiting for ChildGui process");
}
lbRc = TRUE;
//// Отправляем
//fSuccess = WriteFile(
// hPipe, // handle to pipe
// &Out, // buffer to write from
// Out.hdr.cbSize, // number of bytes to write
// &cbWritten, // number of bytes written
// NULL); // not overlapped I/O
break;
} // CECMD_ATTACHGUIAPP
case CECMD_GUICLIENTSHIFT:
{
pcbReplySize = sizeof(CESERVER_REQ_HDR)+sizeof(GuiStylesAndShifts);
if (!ExecuteNewCmd(ppReply, pcbMaxReplySize, pIn->hdr.nCmd, pcbReplySize))
goto wrap;
ppReply->GuiAppShifts = pIn->GuiAppShifts;
ZeroStruct(ppReply->GuiAppShifts.Shifts);
CRealConsole::CorrectGuiChildRect(pIn->GuiAppShifts.nStyle, pIn->GuiAppShifts.nStyleEx, ppReply->GuiAppShifts.Shifts, pIn->GuiAppShifts.szExeName);
lbRc = TRUE;
break;
} // CECMD_GUICLIENTSHIFT
case CECMD_GUIMACRO:
{
// Допустимо, если GuiMacro пытаются выполнить извне
CVConGuard VCon;
CVConGroup::GetActiveVCon(&VCon);
DWORD nFarPluginPID = VCon->RCon()->GetFarPID(true);
LPWSTR pszResult = ConEmuMacro::ExecuteMacro(pIn->GuiMacro.sMacro, VCon->RCon(), (nFarPluginPID==pIn->hdr.nSrcPID), &pIn->GuiMacro);
int nLen = pszResult ? _tcslen(pszResult) : 0;
pcbReplySize = sizeof(CESERVER_REQ_HDR)+sizeof(CESERVER_REQ_GUIMACRO)+nLen*sizeof(wchar_t);
if (!ExecuteNewCmd(ppReply, pcbMaxReplySize, pIn->hdr.nCmd, pcbReplySize))
{
SafeFree(pszResult);
goto wrap;
}
if (pszResult)
{
lstrcpy(ppReply->GuiMacro.sMacro, pszResult);
ppReply->GuiMacro.nSucceeded = 1;
free(pszResult);
}
else
{
ppReply->GuiMacro.sMacro[0] = 0;
ppReply->GuiMacro.nSucceeded = 0;
}
lbRc = TRUE;
break;
} // CECMD_GUIMACRO
case CECMD_CMDSTARTSTOP:
{
CRealServer* pRSrv = NULL;
CVConGuard VCon;
DWORD nSrvPID = pIn->hdr.nSrcPID;
DWORD nMonitorTID = (pIn->DataSize() >= sizeof(pIn->StartStop)) ? pIn->StartStop.dwAID : 0;
if (CVConGroup::GetVConBySrvPID(nSrvPID, nMonitorTID, &VCon))
pRSrv = &VCon->RCon()->m_RConServer;
if (pRSrv)
{
CESERVER_REQ* pOut = pRSrv->cmdStartStop(pInst, pIn, pIn->DataSize());
if (pOut)
{
DWORD nDataSize = pOut->DataSize();
pcbReplySize = pOut->hdr.cbSize;
if (ExecuteNewCmd(ppReply, pcbMaxReplySize, pOut->hdr.nCmd, pcbReplySize))
{
if (nDataSize > 0)
{
memmove(ppReply->Data, pOut->Data, nDataSize);
}
lbRc = TRUE;
}
ExecuteFreeResult(pOut);
}
}
break;
} // CECMD_CMDSTARTSTOP
//case CECMD_DEFTERMSTARTED:
//{
// if (gpConEmu->mp_DefTrm)
// gpConEmu->mp_DefTrm->OnDefTermStarted(pIn);
// pcbReplySize = sizeof(CESERVER_REQ_HDR);
// if (!ExecuteNewCmd(ppReply, pcbMaxReplySize, pIn->hdr.nCmd, pcbReplySize))
// goto wrap;
// lbRc = TRUE;
// break;
//} // CECMD_DEFTERMSTARTED
default:
_ASSERTE(FALSE && "Command was not handled in CGuiServer::GuiServerCommand");
}
//// Освободить память
//if (pIn && (LPVOID)pIn != (LPVOID)&in)
//{
// free(pIn); pIn = NULL;
//}
wrap:
return lbRc;
}
int main(int argc, char** argv)
{
HeapInitialize();
for (int i=0; i<argc; i++)
{
if (strcmp(argv[i], "-verbose")==0 || strcmp(argv[i], "--verbose")==0)
gbVerifyVerbose = true;
}
// Tests
{
Verify_Step("ls1(`Test`)");
CEStr ls1(L"Test");
Verify0((ls1.ms_Val && 0==wcscmp(ls1.ms_Val,L"Test")),"ls1==`Test`");
Verify_Step("ls12 = ls1.Detach()");
/* Store ptr for Verify test result */ LPCWSTR pszPtr = ls1.ms_Val;
CEStr ls2 = ls1.Detach();
Verify2((ls2.ms_Val && !ls1.ms_Val && ls2.ms_Val==pszPtr),"ls2.ms_Val{x%p}==pszPtr{x%p}",ls2.ms_Val,pszPtr);
}
{
Verify_Step("ls3 = `Test3`");
CEStr ls3 = L"Test3";
Verify0((ls3.ms_Val && 0==wcscmp(ls3.ms_Val,L"Test3")),"ls3==`Test3`");
Verify_Step("ls4 = (LPCWSTR)ls3.ms_Val");
CEStr ls4 = static_cast<LPCWSTR>(ls3.ms_Val);
Verify2((ls4.ms_Val && ls4.ms_Val != ls3.ms_Val),"ls4.ms_Val{x%p}!=ls3.ms_Val{x%p}",ls4.ms_Val,ls3.ms_Val);
Verify_Step("ls5 = lstrdup(ls3)");
CEStr ls5 = lstrdup(ls3);
Verify0((ls5.ms_Val && 0==wcscmp(ls5.ms_Val,L"Test3")),"ls5==`Test3`");
Verify_Step("ls6(lstrdup(ls3))");
CEStr ls6(lstrdup(ls3));
Verify0((ls6.ms_Val && 0==wcscmp(ls6.ms_Val,L"Test3")),"ls6==`Test3`");
}
{
Verify_Step("-new_console parser tests");
LPCWSTR pszTest = L"-new_console:a \\\"-new_console:c\\\" `-new_console:d:C:\\` -cur_console:b";
LPCWSTR pszCmp = L"\\\"-new_console:c\\\" `-new_console:d:C:\\`";
RConStartArgs arg; arg.pszSpecialCmd = lstrdup(pszTest);
arg.ProcessNewConArg();
int iCmp = lstrcmp(arg.pszSpecialCmd, pszCmp);
Verify0((iCmp==0),"arg.pszSpecialCmd==\\\"-new_console:c\\\" `-new_console:d:C:\\`");
Verify_Step("RConStartArgs::RunArgTests()");
RConStartArgs::RunArgTests();
Verify0(!gbVerifyFailed,"RConStartArgs tests passed");
}
{
Verify_Step("msprintf tests");
wchar_t szBuffer[200];
msprintf(szBuffer, countof(szBuffer), L"%u %03u %03u %i %x %02X %02X %04x %08X",
123, 98, 4567, -234, 0x12AB, 0x0A, 0xABC, 0x01A0, 0x0765ABCD);
const wchar_t szStd[] = L"123 098 4567 -234 12ab 0A ABC 01a0 0765ABCD";
int iCmp = lstrcmp(szBuffer, szStd);
WVerify2((iCmp==0),L"`%s` (msprintf[W])\n `%s` (standard)", szBuffer, szStd);
char szBufA[200];
msprintf(szBufA, countof(szBufA), "%u %i %x %02X %02X %04x %08X",
123, -234, 0x12AB, 0x0A, 0xABC, 0x01A0, 0x0765ABCD);
const char szStdA[] = "123 -234 12ab 0A ABC 01a0 0765ABCD";
iCmp = lstrcmpA(szBufA, szStdA);
Verify2((iCmp==0),"`%s` (msprintf[A])\n `%s` (standard)", szBufA, szStdA);
}
if (gbVerifyFailed)
Verify_MsgFail("Some tests failed!");
else
Verify_MsgOk("All done");
return gbVerifyFailed ? 99 : 0;
}
int main(int argc, char** argv)
{
HeapInitialize();
for (int i=0; i<argc; i++)
{
if (strcmp(argv[i], "-verbose")==0 || strcmp(argv[i], "--verbose")==0)
gbVerifyVerbose = true;
}
// Tests
{
Verify_Step("ls1(`Test`)");
CEStr ls1(L"Test");
Verify0((ls1.ms_Val && 0==wcscmp(ls1.ms_Val,L"Test")),"ls1==`Test`");
Verify_Step("ls12 = ls1.Detach()");
/* Store ptr for Verify test result */ LPCWSTR pszPtr = ls1.ms_Val;
CEStr ls2 = ls1.Detach();
Verify2((ls2.ms_Val && !ls1.ms_Val && ls2.ms_Val==pszPtr),"ls2.ms_Val{x%p}==pszPtr{x%p}",ls2.ms_Val,pszPtr);
}
{
Verify_Step("ls3 = `Test3`");
CEStr ls3 = L"Test3";
Verify0((ls3.ms_Val && 0==wcscmp(ls3.ms_Val,L"Test3")),"ls3==`Test3`");
Verify_Step("ls4 = (LPCWSTR)ls3.ms_Val");
CEStr ls4 = static_cast<LPCWSTR>(ls3.ms_Val);
Verify2((ls4.ms_Val && ls4.ms_Val != ls3.ms_Val),"ls4.ms_Val{x%p}!=ls3.ms_Val{x%p}",ls4.ms_Val,ls3.ms_Val);
Verify_Step("ls5 = lstrdup(ls3)");
CEStr ls5 = lstrdup(ls3);
Verify0((ls5.ms_Val && 0==wcscmp(ls5.ms_Val,L"Test3")),"ls5==`Test3`");
Verify_Step("ls6(lstrdup(ls3))");
CEStr ls6(lstrdup(ls3));
Verify0((ls6.ms_Val && 0==wcscmp(ls6.ms_Val,L"Test3")),"ls6==`Test3`");
}
{
Verify_Step("NextArg and Switch comparison");
LPCWSTR pszCmd = L"conemu.exe /c/dir -run -inside=0x800 /cmdlist \"-inside=\\eCD /d %1\" -bad|switch ";
CEStr ls;
Verify0((0==NextArg(&pszCmd,ls)),"NextArg conemu.exe");
Verify0((!ls.IsPossibleSwitch()),"!IsPossibleSwitch()");
Verify0((0==NextArg(&pszCmd,ls)),"NextArg /c/dir");
Verify0((!ls.IsPossibleSwitch()),"!IsPossibleSwitch()");
Verify0((0==NextArg(&pszCmd,ls)),"NextArg -run");
Verify0((ls.OneOfSwitches(L"/cmd",L"/run")),"OneOfSwitches(/cmd,/run)");
Verify0((!ls.OneOfSwitches(L"/cmd",L"/cmdlist")),"!OneOfSwitches(/cmd,/cmdlist)");
Verify0((ls.IsSwitch(L"-run")),"IsSwitch(-run)");
Verify0((0==NextArg(&pszCmd,ls)),"NextArg -inside=0x800");
Verify0((ls.IsSwitch(L"-inside=")),"IsSwitch(-inside=)");
Verify0((ls.OneOfSwitches(L"-inside",L"-inside=")),"OneOfSwitches(-inside,-inside=)");
Verify0((!ls.IsSwitch(L"-inside")),"!IsSwitch(-inside)");
Verify0((0==NextArg(&pszCmd,ls)),"NextArg /cmdlist");
Verify0((ls.IsSwitch(L"-cmdlist")),"IsSwitch(-cmdlist)");
Verify0((0==NextArg(&pszCmd,ls)),"NextArg \"-inside=\\eCD /d %1\"");
Verify0((ls.IsSwitch(L"-inside:")),"IsSwitch(-inside=)");
Verify0((0==NextArg(&pszCmd,ls)),"NextArg -bad|switch");
Verify0((ls.Compare(L"-bad|switch")==0),"Compare(-bad|switch)");
Verify0((!ls.IsPossibleSwitch()),"!IsPossibleSwitch");
}
{
Verify_Step("-new_console parser tests");
LPCWSTR pszTest = L"-new_console:a \\\"-new_console:c\\\" `-new_console:d:C:\\` -cur_console:b";
LPCWSTR pszCmp = L"\\\"-new_console:c\\\" `-new_console:d:C:\\`";
RConStartArgs arg; arg.pszSpecialCmd = lstrdup(pszTest);
arg.ProcessNewConArg();
int iCmp = lstrcmp(arg.pszSpecialCmd, pszCmp);
Verify0((iCmp==0),"arg.pszSpecialCmd==\\\"-new_console:c\\\" `-new_console:d:C:\\`");
Verify_Step("RConStartArgs::RunArgTests()");
RConStartArgs::RunArgTests();
Verify0(!gbVerifyFailed,"RConStartArgs tests passed");
}
{
Verify_Step("msprintf tests");
wchar_t szBuffer[200];
msprintf(szBuffer, countof(szBuffer), L"%u %03u %03u %i %x %02X %02X %04x %08X",
123, 98, 4567, -234, 0x12AB, 0x0A, 0xABC, 0x01A0, 0x0765ABCD);
const wchar_t szStd[] = L"123 098 4567 -234 12ab 0A ABC 01a0 0765ABCD";
int iCmp = lstrcmp(szBuffer, szStd);
WVerify2((iCmp==0),L"`%s` (msprintf[W])\n `%s` (standard)", szBuffer, szStd);
char szBufA[200];
msprintf(szBufA, countof(szBufA), "%u %i %x %02X %02X %04x %08X",
123, -234, 0x12AB, 0x0A, 0xABC, 0x01A0, 0x0765ABCD);
const char szStdA[] = "123 -234 12ab 0A ABC 01a0 0765ABCD";
iCmp = lstrcmpA(szBufA, szStdA);
Verify2((iCmp==0),"`%s` (msprintf[A])\n `%s` (standard)", szBufA, szStdA);
}
if (gbVerifyFailed)
Verify_MsgFail("Some tests failed!");
else
Verify_MsgOk("All done");
return gbVerifyFailed ? 99 : 0;
}
