/* Initialise: initialise global data structures */
void Initialise(void)
{
float weight;
char *fn;
CreateWordMap(mapFN,&wmap,1);
SortWordMap(&wmap);
weight = 1.0;
CreateInputSet(&gstack,&wmap,&inSet);
while (NextArg() == STRINGARG || NextArg() == FLOATARG) {
if (NextArg() == FLOATARG)
weight = GetFltArg();
if (weight==0.0 || weight<-10000.0 || weight>10000.0)
HError(-16719,"Unlikely ngram weight[%.4f]",weight);
if (NextArg()!=STRINGARG)
HError(16719,"Gram base file name expected");
fn = GetStrArg();
AddInputGFile(&inSet,fn,weight);
if (trace&T_TOP)
printf("Input file %s added, weight=%.4f\n",fn,weight);
}
if (nSize==0)
nSize = inSet.N;
}
#include "LWMap.h"
#include "LGBase.h"
/* -------------------------- Trace Flags ------------------------ */
static int trace = 0;
#define T_TOP 0001 /* Top Level tracing */
/* ---------------------- Global Variables ----------------------- */
static WordMap wmap; /* and the word map */
/* ---------------- Configuration Parameters --------------------- */
static ConfParam *cParm[MAXGLOBS];
static int nParm = 0; /* total num params */
/* ---------------- Process Command Line ------------------------- */
/* SetConfParms: set conf parms relevant to this tool */
void SetConfParms(void)
{
int i;
nParm = GetConfig("LGLIST", TRUE, cParm, MAXGLOBS);
if (nParm>0){
if (GetConfInt(cParm,nParm,"TRACE",&i)) trace = i;
int main(int argc, char *argv[])
{
char *s;
void Initialise(void);
InitShell(argc,argv,lfof_version,lfof_vc_id);
InitMem();
InitMath();
InitWave();
InitLabel();
InitWMap();
InitGBase();
InitPCalc();
if (!InfoPrinted() && NumArgs() == 0)
ReportUsage();
if (NumArgs() == 0) Exit(EXIT_SUCCESS);
SetConfParms();
while (NextArg() == SWITCHARG) {
s = GetSwtArg();
if (strlen(s)!=1)
HError(16719,"Bad switch %s; must be single letter",s);
switch(s[0]){
case 'f':
fofSize = GetChkedInt(1, 1000, s); break;
case 'n':
nSize = GetChkedInt(1, MAXNG, s); break;
case 'T':
trace = GetChkedInt(0,077,s); break;
default:
HError(16719,"LFoF: Unknown switch %s",s);
}
}
if (NextArg() != STRINGARG)
HError(16719,"LFoF: map file name expected");
mapFN = GetStrArg();
if (NextArg() != STRINGARG)
HError(16719,"LFoF: FoF file name expected");
fofFN = GetStrArg();
Initialise();
if (trace&T_TOP) {
printf("Calculating FoF table\n"); fflush(stdout);
}
fofTab = CreateFoFTab(&gstack,fofSize,nSize);
ComputeFoFTab(fofTab,nSize,&inSet);
WriteFoFTab(fofFN,fofTab,NULL);
Exit(EXIT_SUCCESS);
return EXIT_SUCCESS; /* never reached -- make compiler happy */
}
void
fontpane_popup(int *psfont_adr, int *latexfont_adr, int *psflag_adr, void (*showfont_fn) (/* ??? */), Widget show_widget)
{
DeclareArgs(2);
Position xposn, yposn;
Widget widg;
font_ps_sel = psfont_adr;
font_latex_sel = latexfont_adr;
flag_sel = psflag_adr;
font_setimage = showfont_fn;
font_widget = show_widget;
if (first_fontmenu) {
first_fontmenu = False; /* don't reposition it if user has already popped it */
XtTranslateCoords(tool, CANVAS_WD/4, CANVAS_HT/4, &xposn, &yposn);
FirstArg(XtNx, xposn); /* position about 1/4 from upper-left corner of canvas */
NextArg(XtNy, yposn);
SetValues(ps_fontmenu);
SetValues(latex_fontmenu);
}
widg = *flag_sel ? ps_fontmenu : latex_fontmenu;
XtPopup(widg, XtGrabExclusive);
/* if the file message window is up add it to the grab */
file_msg_add_grab();
/* insure that the most recent colormap is installed */
set_cmap(XtWindow(widg));
XSetWMProtocols(tool_d, XtWindow(widg), &wm_delete_window, 1);
}
int main(int argc, char *argv[])
{
char *s;
Initialise(argc,argv);
CreateHeap(&modelHeap, "Model heap", MSTAK, 1, 0.0, 100000, 800000 );
CreateHMMSet(&hset,&modelHeap,TRUE);
while (NextArg() == SWITCHARG) {
s = GetSwtArg();
if (strlen(s)!=1)
HError(3219,"HNetTest: Bad switch %s; must be single letter",s);
switch(s[0]){
case 'H':
if (NextArg() != STRINGARG)
HError(3219,"HNetTest: MMF File name expected");
AddMMF(&hset,GetStrArg());
break;
default:
HError(3219,"HNetTest: Unknown switch %s",s);
}
}
if (NextArg()!=STRINGARG)
HError(999,"HNetTest: Dictionary file name expected");
dictFn = GetStrArg();
if (NextArg()!=STRINGARG)
HError(999,"HNetTest: Word Net file name expected");
wdNetFn = GetStrArg();
if (NextArg()!=STRINGARG)
HError(999,"HNetTest: HMM list file name expected");
hmmListFn = GetStrArg();
if(MakeHMMSet(&hset,hmmListFn)<SUCCESS)
HError(999,"HNetTest: MakeHMMSet failed");
if(LoadHMMSet(&hset,NULL,NULL)<SUCCESS)
HError(999,"HNetTest: LoadHMMSet failed");
InitVocab(&vocab);
if(ReadDict(dictFn,&vocab)<SUCCESS)
HError(3213, "HNetTest: ReadDict failed");
LoadNetwork();
}
static void
spell_select_word(Widget widget, XtPointer closure, XtPointer call_data)
{
XawListReturnStruct *ret_struct = (XawListReturnStruct *) call_data;
/* make correct button and correction entry sensitive */
XtSetSensitive(correct_button, True);
XtSetSensitive(correct_word, True);
/* save the selected word */
strcpy(selected_word, ret_struct->string);
/* copy the word to the correct_word ascii widget */
FirstArg(XtNstring, ret_struct->string);
NextArg(XtNeditType, XawtextEdit); /* make editable now */
NextArg(XtNsensitive, True); /* and sensitive */
SetValues(correct_word);
}
bool CConEmuStart::GetCfgParm(LPCWSTR& cmdLineRest, CESwitch& Val, int nMaxLen, bool bExpandAndDup /*= false*/)
{
if (Val.Type == sw_Str || Val.Type == sw_EnvStr || Val.Type == sw_PathStr)
{
SafeFree(Val.Str);
}
else
{
Val.Type = bExpandAndDup ? sw_EnvStr : sw_Str;
}
Val.Exists = false;
if (!cmdLineRest || !*cmdLineRest)
{
_ASSERTE(cmdLineRest && *cmdLineRest);
return false;
}
// Сохраним, может для сообщения об ошибке понадобится
LPCWSTR pszName = cmdLineRest;
CmdArg szGetCfgParmTemp;
if (!(cmdLineRest = NextArg(cmdLineRest, szGetCfgParmTemp)))
{
return false;
}
LPCWSTR curCommand = szGetCfgParmTemp.ms_Val;
int nLen = _tcslen(curCommand);
if (nLen >= nMaxLen)
{
int nCchSize = nLen+100+_tcslen(pszName);
wchar_t* psz = (wchar_t*)calloc(nCchSize,sizeof(wchar_t));
if (psz)
{
swprintf_c(psz, nCchSize/*#SECURELEN*/, L"Too long %s value (%i chars).\r\n", pszName, nLen);
_wcscat_c(psz, nCchSize, curCommand);
MBoxA(psz);
free(psz);
}
return false;
}
// We need independent absolute file paths, Working dir changes during ConEmu session
if (bExpandAndDup)
Val.Str = GetFullPathNameEx(curCommand); // it allocates memory
else
Val.SetStr(curCommand, Val.Type);
// Ok
Val.Exists = (Val.Str && *Val.Str);
return true;
}
Switch* GetNextSwitch(LPCWSTR& rpsz, CEStr& szArg)
{
LPCWSTR psz = rpsz;
CEStr szNext;
if ((0 == NextArg(&psz, szNext)) && !szNext.IsPossibleSwitch())
rpsz = psz;
else
szNext.Clear();
Switch* ps = new Switch(szArg.Detach(), szNext.Detach());
return ps;
};
static void
background_select(Widget w, XtPointer closure, XtPointer call_data)
{
Pixel bgcolor, fgcolor;
/* get the colors from the color button just pressed */
FirstArg(XtNbackground, &bgcolor);
NextArg(XtNforeground, &fgcolor);
GetValues(w);
/* get the colorname from the color button and put it and the colors
in the menu button */
FirstArg(XtNlabel, XtName(w));
NextArg(XtNbackground, bgcolor);
NextArg(XtNforeground, fgcolor);
SetValues(print_background_panel);
/* update the export panel too if it exists */
if (export_background_panel)
SetValues(export_background_panel);
export_background_color = (int)closure;
XtPopdown(background_menu);
}
/* CreateDataFileList: */
static void CreateDataFileList()
{
char *datafn;
g_nDataFileNum = 0;
do {
if (NextArg() != STRINGARG)
HError(2319, "HMGenS: data file name expected");
datafn = GetStrArg();
g_pDataFileList[g_nDataFileNum] = (TDataFile *) New(&gstack, sizeof(TDataFile));
strcpy(g_pDataFileList[g_nDataFileNum]->datafn, datafn);
g_pDataFileList[g_nDataFileNum]->bValid = TRUE;
g_nDataFileNum++;
} while (NumArgs() > 0);
}
int main(int argc, char *argv[])
{
int n,a1,a2,a3,a4;
InitThreads(sMon);
if(InitShell(argc,argv,hthreadtest_version)<SUCCESS)
HError(1100,"HThreadTest: InitShell failed");
if (NumArgs() < 1) ReportUsage();
InitMem(); InitLabel();
InitMath(); InitSigP();
InitWave(); InitAudio();
InitVQ();
if(InitParm()<SUCCESS)
HError(3200,"HThreadTest: InitParm failed");
InitGraf(FALSE);
if (sMon==HT_MSGMON) HCreateMonitor(tmon4,(void *)0);
if (NextArg() == INTARG){
n = GetIntArg();
switch(n){
case 1:
a1 = GetIntArg(); a2 = GetIntArg();
ParallelForkAndJoin(a1,a2); break;
case 2:
a1 = GetIntArg(); a2 = GetIntArg();
SimpleMutex(a1,a2); break;
case 3:
SimpleSignal(); break;
case 4:
a1 = GetIntArg(); a2 = GetIntArg();
a3 = GetIntArg(); a4 = GetIntArg();
BufferTest(a1,a2,a3,a4,GetStrArg()); break;
default:
printf("Bad test number %d\n",n); ReportUsage();
}
}
if (sMon>0){
AccessStatus();
PrintThreadStatus("Final");
ReleaseStatusAccess();
if (sMon==HT_MSGMON)HJoinMonitor();
}
}
Switch* GetNextPair(LPCWSTR& rpsz)
{
LPCWSTR psz = rpsz;
CEStr szArg;
if (0 != NextArg(&psz, szArg))
{
return NULL;
}
// Invalid switch? or first argument (our executable)
if (!szArg.IsPossibleSwitch())
{
rpsz = psz;
return NULL;
}
rpsz = psz;
return GetNextSwitch(rpsz, szArg);
};
bool CConEmuUpdate::Check7zipInstalled()
{
if (mp_Set->UpdateDownloadSetup() == 1)
return true; // Инсталлер, архиватор не требуется!
LPCWSTR pszCmd = mp_Set->UpdateArcCmdLine();
CmdArg sz7zip; sz7zip.GetBuffer(MAX_PATH);
if (NextArg(&pszCmd, sz7zip) != 0)
{
ReportError(L"Invalid update command\nGoto 'Update' page and check 7-zip command", 0);
return false;
}
if (FileExistsSearch(sz7zip.GetBuffer(MAX_PATH), MAX_PATH))
return true;
WARNING("TODO: Suggest to download 7zip");
ReportError(L"7zip or WinRar not found! Not installed?\n%s\nGoto 'Update' page and check 7-zip command", sz7zip, 0);
return false;
}
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;
}
// Returns PTR to next arg or NULL on error
LPCWSTR QueryNextArg(const wchar_t* asCmdLine, CEStr &rsArg, const wchar_t** rsArgStart/*=NULL*/)
{
if (0 != NextArg(&asCmdLine, rsArg, rsArgStart))
return NULL;
return asCmdLine;
}
// Called from OnShellExecCmdLine
HRESULT OurShellExecCmdLine(HWND hwnd, LPCWSTR pwszCommand, LPCWSTR pwszStartDir, bool bRunAsAdmin, bool bForce)
{
HRESULT hr = E_UNEXPECTED;
BOOL bShell = FALSE;
CEStr lsLog = lstrmerge(L"OnShellExecCmdLine", bRunAsAdmin ? L"(RunAs): " : L": ", pwszCommand);
DefTermLogString(lsLog);
// Bad thing, ShellExecuteEx needs File&Parm, but we get both in pwszCommand
CmdArg szExe;
LPCWSTR pszFile = pwszCommand;
LPCWSTR pszParm = pwszCommand;
if (NextArg(&pszParm, szExe) == 0)
{
pszFile = szExe; pszParm = SkipNonPrintable(pszParm);
}
else
{
// Failed
pszFile = pwszCommand; pszParm = NULL;
}
if (!bForce)
{
DWORD nCheckSybsystem1 = 0, nCheckBits1 = 0;
if (!FindImageSubsystem(pszFile, nCheckSybsystem1, nCheckBits1))
{
hr = (HRESULT)-1;
DefTermLogString(L"OnShellExecCmdLine: FindImageSubsystem failed");
goto wrap;
}
if (nCheckSybsystem1 != IMAGE_SUBSYSTEM_WINDOWS_CUI)
{
hr = (HRESULT)-1;
DefTermLogString(L"OnShellExecCmdLine: !=IMAGE_SUBSYSTEM_WINDOWS_CUI");
goto wrap;
}
}
// "Run as admin" was requested?
if (bRunAsAdmin)
{
SHELLEXECUTEINFO sei = {sizeof(sei), 0, hwnd, L"runas", pszFile, pszParm, pwszStartDir, SW_SHOWNORMAL};
bShell = OnShellExecuteExW(&sei);
}
else
{
wchar_t* pwCommand = lstrdup(pwszCommand);
DWORD nCreateFlags = CREATE_NEW_CONSOLE|CREATE_UNICODE_ENVIRONMENT|CREATE_DEFAULT_ERROR_MODE;
STARTUPINFO si = {sizeof(si)};
PROCESS_INFORMATION pi = {};
bShell = OnCreateProcessW(NULL, pwCommand, NULL, NULL, FALSE, nCreateFlags, NULL, pwszStartDir, &si, &pi);
if (bShell)
{
CloseHandle(pi.hProcess);
CloseHandle(pi.hThread);
}
}
hr = bShell ? S_OK : HRESULT_FROM_WIN32(GetLastError());
wrap:
return hr;
}
int main(int argc, char *argv[])
{
int i;
char *c,*s,*fn;
char sBuf[256],fmt[256];
void Initialise(void);
void ProcessText(char *fn,bool lastFile);
bool Exists(char *fn);
BackOffLM *CombineModels(MemHeap *heap,LMInfo *lmi,int nLModel,int nSize,WordMap *wl) ;
InitShell(argc,argv,ladapt_version,ladapt_vc_id);
InitMem();
InitMath();
InitWave();
InitLabel();
InitLUtil();
InitWMap();
InitGBase();
InitLModel();
InitPCalc();
InitPMerge();
SetConfParms();
if (!InfoPrinted() && NumArgs() == 0)
ReportUsage();
if (NumArgs() == 0) Exit(EXIT_SUCCESS);
InitBuildInfo(&binfo);
binfo.dctype = DC_ABSOLUTE;
nLModel = 1;
while (NextArg() == SWITCHARG) {
s = GetSwtArg();
if (strlen(s)!=1)
HError(16419,"Bad switch %s; must be single letter",s);
switch(s[0]){
case 'a':
newWords = GetChkedInt(10,10000000,s); break;
case 'b':
ngbSize = GetChkedInt(10,10000000,s); break;
case 'c':
i = GetChkedInt(2,LM_NSIZE,s);
binfo.cutOff[i] = GetChkedInt(0,1000,s);
break;
case 'd':
if (NextArg()!=STRINGARG)
HError(16419,"Gram base root file name expected");
rootFN = GetStrArg();
break;
case 'f':
strcpy(fmt, GetStrArg());
for (c=fmt; *c; *c=toupper(*c), c++); /* To uppercase */
if (strcmp(fmt, LM_TXT_TEXT)==0)
binfo.saveFmt = LMF_TEXT;
else if (strcmp(fmt, LM_TXT_BINARY)==0)
binfo.saveFmt = LMF_BINARY;
else if (strcmp(fmt, LM_TXT_ULTRA)==0)
binfo.saveFmt = LMF_ULTRA;
else
HError(16419,"Unrecognised LM format, should be one of [%s, %s, %s]",
LM_TXT_TEXT, LM_TXT_BINARY, LM_TXT_ULTRA);
break;
case 'g':
processText = FALSE; break;
case 'i':
if (NextArg()!=FLOATARG)
HError(16419,"Interpolation weight expected");
lmInfo[nLModel].weight = GetChkedFlt(0.0,1.0,s);
if (NextArg()!=STRINGARG)
HError(16419,"Interpolation LM filename expected");
lmInfo[nLModel].fn = GetStrArg();
nLModel++;
break;
case 'j':
i = GetChkedInt(2,LM_NSIZE,s);
binfo.wdThresh[i] = GetChkedFlt(0.0,1E10,s);
break;
case 'n':
nSize = GetChkedInt(1, MAXNG, s); break;
#ifdef HTK_TRANSCRIBER
case 's':
if (NextArg()!=STRINGARG)
HError(16419,"Gram file text source descriptor expected");
txtSrc = GetStrArg(); break;
case 't':
binfo.dctype = DC_KATZ; break;
#endif
case 'w':
if (NextArg()!=STRINGARG)
HError(16419,"Word list file name expected");
wlistFN = GetStrArg(); break;
#ifndef HTK_TRANSCRIBER
case 'x':
binfo.ptype = LMP_COUNT; break;
#endif
case 'T':
trace = GetChkedInt(0,077,s); break;
default:
HError(16419,"LAdapt: Unknown switch %s",s);
}
}
#ifdef HTK_TRANSCRIBER
if (nLModel==1) { /* must interpolate with at least one model */
HError(16419,"LAdapt: at least one model must be specified with -i option");
}
if (binfo.saveFmt==LMF_TEXT) { /* save fomat cannot be TEXT */
binfo.saveFmt=LMF_BINARY;
}
#endif
if (NextArg() != STRINGARG)
HError(16419,"LAdapt: language model file name expected");
outFN = CopyString(&gstack,GetStrArg());
Initialise();
if (processText) {
if (NextArg() != STRINGARG)
ProcessText(NULL,TRUE); /* input from stdin */
else
while (NextArg() == STRINGARG) {
/* !! copy string argument since it gets overwritten
by NextArg() when reading from script file */
fn = CopyString(&gstack,GetStrArg());
ProcessText(fn,NextArg() != STRINGARG);
}
if (NumArgs() != 0)
HError(-16419,"LAdapt: unused args left on cmd line");
for (i=0; i<stdBuf.ngb->fndx; i++) {
sprintf(sBuf,"%s.%d",stdBuf.ngb->fn,i);
AddInputGFile(&inSet,sBuf,1.0);
}
ResetHeap(&langHeap);
} else {
for (i=0; i<MAX_NGRAM_FILES; i++) {
sprintf(sBuf,"%s.%d",rootFN,i);
if (!Exists(sBuf))
break;
AddInputGFile(&inSet,sBuf,1.0);
}
if (i==MAX_NGRAM_FILES)
{
HError(-16419, "LAdapt: Only %d n-gram files read (recompile with different setting\nof MAX_NGRAM_FILES");
}
}
if (nLModel==1) {
adpLM = GenerateModel(&langHeap,&binfo);
} else {
if (binfo.ptype==LMP_COUNT)
binfo.ptype = LMP_FLOAT;
newLM = GenerateModel(&langHeap,&binfo);
lmInfo[0].lm = newLM;
lmInfo[0].fn = "unknown";
/* combine all models into one */
adpLM = CombineModels(&langHeap,lmInfo,nLModel,nSize,tgtVoc);
}
#ifdef HTK_TRANSCRIBER
#ifdef HTK_CRYPT
adpLM->encrypt = TRUE; /* force to write encrypted model */
#endif
#endif
SaveLangModel(outFN,adpLM);
Exit(EXIT_SUCCESS);
return EXIT_SUCCESS; /* never reached -- make compiler happy */
}
// Returns ">0" - when changes was made
// 0 - no changes
// -1 - error
// bForceCurConsole==true, если разбор параметров идет
// при запуске Tasks из GUI
int RConStartArgs::ProcessNewConArg(bool bForceCurConsole /*= false*/)
{
NewConsole = crb_Undefined;
if (!pszSpecialCmd || !*pszSpecialCmd)
{
_ASSERTE(pszSpecialCmd && *pszSpecialCmd);
return -1;
}
int nChanges = 0;
// 140219 - Остановить обработку, если встретим любой из: ConEmu[.exe], ConEmu64[.exe], ConEmuC[.exe], ConEmuC64[.exe]
LPCWSTR pszStopAt = NULL;
{
LPCWSTR pszTemp = pszSpecialCmd;
LPCWSTR pszSave = pszSpecialCmd;
LPCWSTR pszName;
CmdArg szExe;
LPCWSTR pszWords[] = {L"ConEmu", L"ConEmu.exe", L"ConEmu64", L"ConEmu64.exe", L"ConEmuC", L"ConEmuC.exe", L"ConEmuC64", L"ConEmuC64.exe", L"ConEmuPortable.exe", L"ConEmuPortable", NULL};
while (!pszStopAt && (0 == NextArg(&pszTemp, szExe)))
{
pszName = PointToName(szExe);
for (size_t i = 0; pszWords[i]; i++)
{
if (lstrcmpi(pszName, pszWords[i]) == 0)
{
pszStopAt = pszSave;
break;
}
}
pszSave = pszTemp;
}
}
#if 0
// 140219 - Остановить обработку, если встретим любой из: ConEmu[.exe], ConEmu64[.exe], ConEmuC[.exe], ConEmuC64[.exe]
if (!hShlwapi)
{
hShlwapi = LoadLibrary(L"Shlwapi.dll");
WcsStrI = hShlwapi ? (StrStrI_t)GetProcAddress(hShlwapi, "StrStrIW") : NULL;
}
#endif
// 111211 - здесь может быть передан "-new_console:..."
LPCWSTR pszNewCon = L"-new_console";
// 120108 - или "-cur_console:..." для уточнения параметров запуска команд (из фара например)
LPCWSTR pszCurCon = L"-cur_console";
int nNewConLen = lstrlen(pszNewCon);
_ASSERTE(lstrlen(pszCurCon)==nNewConLen);
wchar_t* pszFrom = pszSpecialCmd;
bool bStop = false;
while (!bStop)
{
wchar_t* pszSwitch = wcschr(pszFrom, L'-');
if (!pszSwitch)
break;
// Pre-validation
if (((pszSwitch[1] != L'n') && (pszSwitch[1] != L'c')) // -new_... or -cur_...
|| (((pszSwitch != /* > */ pszFrom) // If it is started from pszFrom - no need to check previous symbols
&& (*(pszSwitch-1) != L'"') || (((pszSwitch-2) >= pszFrom) && (*(pszSwitch-2) == L'\\'))) // Found: \"-new...
&& (*(pszSwitch-1) != L' '))) // Prev symbol was space
{
// НЕ наш аргумент
pszSwitch = wcschr(pszSwitch+1, L' ');
if (!pszSwitch)
break;
pszFrom = pszSwitch;
continue;
}
wchar_t* pszFindNew = NULL;
wchar_t* pszFind = NULL;
wchar_t szTest[12]; lstrcpyn(szTest, pszSwitch+1, countof(szTest));
if (lstrcmp(szTest, L"new_console") == 0)
pszFindNew = pszFind = pszSwitch;
else if (lstrcmp(szTest, L"cur_console") == 0)
pszFind = pszSwitch;
else
{
// НЕ наш аргумент
pszSwitch = wcschr(pszSwitch+1, L' ');
if (!pszSwitch)
break;
pszFrom = pszSwitch;
continue;
}
if (!pszFind)
break;
if (pszStopAt && (pszFind >= pszStopAt))
break;
// Проверка валидности
_ASSERTE(pszFind >= pszSpecialCmd);
if ((pszFind[nNewConLen] != L' ') && (pszFind[nNewConLen] != L':')
&& (pszFind[nNewConLen] != L'"') && (pszFind[nNewConLen] != 0))
{
// НЕ наш аргумент
pszFrom = pszFind+nNewConLen;
}
else
{
if (pszFindNew)
NewConsole = crb_On;
// -- не будем пока, мешает. например, при запуске задач
//// По умолчанию, принудительно включить "Press Enter or Esc to close console"
//if (!bForceCurConsole)
// eConfirmation = eConfAlways;
bool lbQuot = (*(pszFind-1) == L'"');
bool lbWasQuot = lbQuot;
const wchar_t* pszEnd = pszFind+nNewConLen;
//wchar_t szNewConArg[MAX_PATH+1];
if (lbQuot)
pszFind--;
if (*pszEnd == L'"')
{
pszEnd++;
}
else if (*pszEnd != L':')
{
// Конец
_ASSERTE(*pszEnd == L' ' || *pszEnd == 0);
}
else
{
if (*pszEnd == L':')
{
pszEnd++;
}
else
{
_ASSERTE(*pszEnd == L':');
}
// Найти конец аргумента
const wchar_t* pszArgEnd = pszEnd;
bool lbLocalQuot = false;
while (*pszArgEnd)
{
switch (*pszArgEnd)
{
case L'^':
pszArgEnd++; // Skip control char, goto escaped char
break;
case L'"':
if (*(pszArgEnd+1) == L'"')
{
pszArgEnd += 2; // Skip qoubled qouble quote
continue;
}
if (!lbQuot)
{
if (!lbLocalQuot && (*(pszArgEnd-1) == L':'))
{
lbLocalQuot = true;
pszArgEnd++;
continue;
}
if (lbLocalQuot)
{
if (*(pszArgEnd+1) != L':')
goto EndFound;
lbLocalQuot = false;
pszArgEnd += 2;
continue;
}
}
goto EndFound;
case L' ':
if (!lbQuot && !lbLocalQuot)
goto EndFound;
break;
case 0:
goto EndFound;
}
pszArgEnd++;
}
EndFound:
// Обработка доп.параметров -new_console:xxx
bool lbReady = false;
while (!lbReady && *pszEnd)
{
_ASSERTE(pszEnd <= pszArgEnd);
wchar_t cOpt = *(pszEnd++);
switch (cOpt)
{
//case L'-':
// bStop = true; // следующие "-new_console" - не трогать!
// break;
case L'"':
_ASSERTE(pszEnd > pszArgEnd);
lbReady = true;
break;
case L' ':
case 0:
lbReady = true;
break;
case L':':
// Just skip ':'. Delimiter between switches: -new_console:c:b:a
// Revert stored value to lbQuot. We need to "cut" last double quote in the first two cases
// cmd -cur_console:d:"C:\users":t:"My title" "-cur_console:C:C:\cmd.ico" -cur_console:P:"<PowerShell>":a /k ver
lbWasQuot = lbQuot;
break;
case L'b':
// b - background, не активировать таб
BackgroundTab = crb_On; ForegroungTab = crb_Off;
break;
case L'f':
// f - foreground, активировать таб (аналог ">" в Tasks)
ForegroungTab = crb_On; BackgroundTab = crb_Off;
break;
case L'z':
// z - don't use "Default terminal" feature
NoDefaultTerm = crb_On;
break;
case L'a':
// a - RunAs shell verb (as admin on Vista+, login/password in WinXP-)
RunAsAdministrator = crb_On;
break;
case L'r':
// r - run as restricted user
RunAsRestricted = crb_On;
break;
case L'o':
// o - disable "Long output" for next command (Far Manager)
LongOutputDisable = crb_On;
break;
case L'w':
// e - enable "Overwrite" mode in console prompt
OverwriteMode = crb_On;
break;
case L'p':
if (isDigit(*pszEnd))
{
switch (*(pszEnd++))
{
case L'0':
nPTY = 0; // don't change
break;
case L'1':
nPTY = 1; // enable PTY mode
break;
case L'2':
nPTY = 2; // disable PTY mode (switch to plain $CONIN, $CONOUT, $CONERR)
break;
default:
nPTY = 1;
}
}
else
{
nPTY = 1; // enable PTY mode
}
break;
case L'i':
// i - don't inject ConEmuHk into the starting application
InjectsDisable = crb_On;
break;
case L'N':
// N - Force new ConEmu window with Default terminal
ForceNewWindow = crb_On;
break;
case L'h':
// "h0" - отключить буфер, "h9999" - включить буфер в 9999 строк
{
BufHeight = crb_On;
if (isDigit(*pszEnd))
{
wchar_t* pszDigits = NULL;
nBufHeight = wcstoul(pszEnd, &pszDigits, 10);
if (pszDigits)
pszEnd = pszDigits;
}
else
{
nBufHeight = 0;
}
} // L'h':
break;
case L'n':
// n - отключить "Press Enter or Esc to close console"
eConfirmation = eConfNever;
break;
case L'c':
// c - принудительно включить "Press Enter or Esc to close console"
eConfirmation = eConfAlways;
break;
case L'x':
// x - Force using dosbox for .bat files
ForceDosBox = crb_On;
break;
case L'I':
// I - tell GuiMacro to execute new command inheriting active process state. This is only usage ATM.
ForceInherit = crb_On;
break;
// "Long" code blocks below: 'd', 'u', 's' and so on (in future)
case L's':
// s[<SplitTab>T][<Percents>](H|V)
// Пример: "s3T30H" - разбить 3-ий таб. будет создан новый Pane справа, шириной 30% от 3-го таба.
{
UINT nTab = 0 /*active*/, nValue = /*пополам*/DefaultSplitValue/10;
bool bDisableSplit = false;
while (*pszEnd)
{
if (isDigit(*pszEnd))
{
wchar_t* pszDigits = NULL;
UINT n = wcstoul(pszEnd, &pszDigits, 10);
if (!pszDigits)
break;
pszEnd = pszDigits;
if (*pszDigits == L'T')
{
nTab = n;
}
else if ((*pszDigits == L'H') || (*pszDigits == L'V'))
{
nValue = n;
eSplit = (*pszDigits == L'H') ? eSplitHorz : eSplitVert;
}
else
{
break;
}
pszEnd++;
}
else if (*pszEnd == L'T')
{
nTab = 0;
pszEnd++;
}
else if ((*pszEnd == L'H') || (*pszEnd == L'V'))
{
nValue = DefaultSplitValue/10;
eSplit = (*pszEnd == L'H') ? eSplitHorz : eSplitVert;
pszEnd++;
}
else if (*pszEnd == L'N')
{
bDisableSplit = true;
pszEnd++;
break;
}
else
{
break;
}
}
if (bDisableSplit)
{
eSplit = eSplitNone; nSplitValue = DefaultSplitValue; nSplitPane = 0;
}
else
{
if (!eSplit)
eSplit = eSplitHorz;
// Для удобства, пользователь задает размер НОВОЙ части
nSplitValue = 1000-max(1,min(nValue*10,999)); // проценты
_ASSERTE(nSplitValue>=1 && nSplitValue<1000);
nSplitPane = nTab;
}
} // L's'
break;
// Following options (except of single 'u') must be placed on the end of "-new_console:..."
// If one needs more that one option - use several "-new_console:..." switches
case L'd':
// d:<StartupDir>. MUST be last option
case L't':
// t:<TabName>. MUST be last option
case L'u':
// u - ConEmu choose user dialog (may be specified in the middle, if it is without ':' - user or pwd)
// u:<user> - ConEmu choose user dialog with prefilled user field. MUST be last option
// u:<user>:<pwd> - specify user/pwd in args. MUST be last option
case L'C':
// C:<IconFile>. MUST be last option
case L'P':
// P:<Palette>. MUST be last option
case L'W':
// W:<Wallpaper>. MUST be last option
{
if (cOpt == L'u')
{
// Show choose user dialog (may be specified in the middle, if it is without ':' - user or pwd)
SafeFree(pszUserName);
SafeFree(pszDomain);
if (szUserPassword[0]) SecureZeroMemory(szUserPassword, sizeof(szUserPassword));
}
if (*pszEnd == L':')
{
pszEnd++;
}
else
{
if (cOpt == L'u')
{
ForceUserDialog = crb_On;
break;
}
}
const wchar_t* pszTab = pszEnd;
// we need to find end of argument
pszEnd = pszArgEnd;
// temp buffer
wchar_t* lpszTemp = NULL;
wchar_t** pptr = NULL;
switch (cOpt)
{
case L'd': pptr = &pszStartupDir; break;
case L't': pptr = &pszRenameTab; break;
case L'u': pptr = &lpszTemp; break;
case L'C': pptr = &pszIconFile; break;
case L'P': pptr = &pszPalette; break;
case L'W': pptr = &pszWallpaper; break;
}
if (pszEnd > pszTab)
{
size_t cchLen = pszEnd - pszTab;
SafeFree(*pptr);
*pptr = (wchar_t*)malloc((cchLen+1)*sizeof(**pptr));
if (*pptr)
{
// We need to process escape sequences ("^>" -> ">", "^&" -> "&", etc.)
//wmemmove(*pptr, pszTab, cchLen);
wchar_t* pD = *pptr;
const wchar_t* pS = pszTab;
if (lbQuot)
{
lbLocalQuot = false;
}
else if (*pS == L'"' && *(pS+1) != L'"')
{
// Remember, that last processed switch was local-quoted
lbWasQuot = true;
// This item is local quoted. Example: -new_console:t:"My title"
lbLocalQuot = true;
pS++;
}
// There is enough room allocated
while (pS < pszEnd)
{
if ((*pS == L'^') && ((pS + 1) < pszEnd))
{
pS++; // Skip control char, goto escaped char
}
else if (*pS == L'"')
{
if (((pS + 1) < pszEnd) && (*(pS+1) == L'"'))
{
pS++; // Skip qoubled qouble quote
}
else if (lbLocalQuot)
{
pszEnd = (pS+1);
_ASSERTE(*pszEnd==L':' || *pszEnd==L' ' || *pszEnd==0);
break; // End of local quoted argument: -new_console:d:"C:\User\super user":t:"My title"
}
}
*(pD++) = *(pS++);
}
// Terminate with '\0'
_ASSERTE(pD <= ((*pptr)+cchLen));
*pD = 0;
}
// Additional processing
switch (cOpt)
{
case L'd':
// Например, "%USERPROFILE%"
// TODO("А надо ли разворачивать их тут? Наверное при запуске под другим юзером некорректно? Хотя... все равно до переменных не доберемся");
if (wcschr(pszStartupDir, L'%'))
{
wchar_t* pszExpand = NULL;
if (((pszExpand = ExpandEnvStr(pszStartupDir)) != NULL))
{
SafeFree(pszStartupDir);
pszStartupDir = pszExpand;
}
}
break;
case L'u':
if (lpszTemp)
{
// Split in form:
// [Domain\]UserName[:Password]
wchar_t* pszPwd = wcschr(lpszTemp, L':');
if (pszPwd)
{
// Password was specified, dialog prompt is not required
ForceUserDialog = crb_Off;
*pszPwd = 0;
int nPwdLen = lstrlen(pszPwd+1);
lstrcpyn(szUserPassword, pszPwd+1, countof(szUserPassword));
if (nPwdLen > 0)
SecureZeroMemory(pszPwd+1, nPwdLen);
UseEmptyPassword = (nPwdLen == 0) ? crb_On : crb_Off;
}
else
{
// Password was NOT specified, dialog prompt IS required
ForceUserDialog = crb_On;
UseEmptyPassword = crb_Off;
}
wchar_t* pszSlash = wcschr(lpszTemp, L'\\');
if (pszSlash)
{
*pszSlash = 0;
pszDomain = lstrdup(lpszTemp);
pszUserName = lstrdup(pszSlash+1);
}
else
{
pszUserName = lstrdup(lpszTemp);
}
}
break;
}
}
SafeFree(lpszTemp);
} // L't':
break;
}
}
}
if (pszEnd > pszFind)
{
// pszEnd должен указывать на конец -new_console[:...] / -cur_console[:...]
// и включать обрамляющую кавычку, если он окавычен
if (lbWasQuot)
{
if (*pszEnd == L'"' && *(pszEnd-1) != L'"')
pszEnd++;
}
else
{
while (*(pszEnd-1) == L'"')
pszEnd--;
}
// Откусить лишние пробелы, которые стоят ПЕРЕД -new_console[:...] / -cur_console[:...]
while (((pszFind - 1) >= pszSpecialCmd)
&& (*(pszFind-1) == L' ')
&& (((pszFind - 1) == pszSpecialCmd) || (*(pszFind-2) == L' ') || (/**pszEnd == L'"' ||*/ *pszEnd == 0 || *pszEnd == L' ')))
{
pszFind--;
}
// Откусить лишние пробелы ПОСЛЕ -new_console[:...] / -cur_console[:...] если он стоит в НАЧАЛЕ строки!
if (pszFind == pszSpecialCmd)
{
while (*pszEnd == L' ')
pszEnd++;
}
// Здесь нужно подвинуть pszStopAt
if (pszStopAt)
pszStopAt -= (pszEnd - pszFind);
// Удалить из строки запуска обработанный ключ
wmemmove(pszFind, pszEnd, (lstrlen(pszEnd)+1));
nChanges++;
}
else
{
_ASSERTE(pszEnd > pszFind);
*pszFind = 0;
nChanges++;
}
} // if ((((pszFind == pszFrom) ...
} // while (!bStop)
return nChanges;
} // int RConStartArgs::ProcessNewConArg(bool bForceCurConsole /*= false*/)
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;
}
int main(int argc, char *argv[])
{
char *datafn, *s;
int stream = 0;
void Initialise(char *datafn);
void LoadFile(char *fn);
void CalcMeanCov(Sequence seq[], int s);
void ClusterVecs(Sequence seq[], int s);
void WriteVQTable(ClusterSet *cs[], char *fn);
if(InitShell(argc,argv,hquant_version,hquant_vc_id)<SUCCESS)
HError(2500,"HQuant: InitShell failed");
InitMem(); InitLabel();
InitMath(); InitSigP();
InitWave(); InitAudio();
InitVQ(); InitModel();
if(InitParm()<SUCCESS)
HError(2500,"HQuant: InitParm failed");
InitTrain();
if (!InfoPrinted() && NumArgs() == 0)
ReportUsage();
if (NumArgs() == 0) Exit(0);
SetConfParms();
InitStreamVars();
while (NextArg() == SWITCHARG) {
s = GetSwtArg();
if (strlen(s)!=1)
HError(2519,"HQuant: Bad switch %s; must be single letter",s);
switch(s[0]){
case 'd':
if ( ck != NULLC) HError(2519,"HQuant: Specify one of -d or -f, not both");
ck = INVDIAGC;
break;
case 'f':
if ( ck != NULLC) HError(2519,"HQuant: Specify one of -d or -f, not both");
ck = FULLC;
break;
case 'g':
globClustVar = TRUE;
break;
case 'l':
if (NextArg() != STRINGARG)
HError(2519,"HQuant: Segment label expected");
segLab = GetStrArg();
break;
case 'n':
if (NextArg() != INTARG)
HError(2519,"HQuant: Stream number expected");
stream = GetChkedInt(1,SMAX,s);
if (NextArg() != INTARG)
HError(2519,"HQuant: Codebook size expected");
cbSizes[stream]= GetChkedInt(1,32768,s);
break;
case 's':
if (NextArg() != INTARG)
HError(2519,"HQuant: Number of streams expected");
swidth[0] = GetChkedInt(1,SMAX,s);
break;
case 't':
tType = binTree;
break;
case 'w':
if (NextArg() != INTARG)
HError(2519,"HQuant: Stream number expected");
stream = GetChkedInt(1,SMAX,s);
if(swidth[0] < stream) swidth[0] = stream;
widthSet = TRUE;
if (NextArg() != INTARG)
HError(2519,"HQuant: Stream width expected");
swidth[stream]= GetChkedInt(1,256,s);
break;
case 'F':
if (NextArg() != STRINGARG)
HError(2519,"HQuant: Data File format expected");
if((dff = Str2Format(GetStrArg())) == ALIEN)
HError(-2589,"HQuant: Warning ALIEN Data file format set");
break;
case 'G':
if (NextArg() != STRINGARG)
HError(2519,"HQuant: Label File format expected");
if((lff = Str2Format(GetStrArg())) == ALIEN)
HError(-2589,"HQuant: Warning ALIEN Label file format set");
break;
case 'I':
if (NextArg() != STRINGARG)
HError(2519,"HQuant: MLF file name expected");
LoadMasterFile(GetStrArg());
break;
case 'L':
if (NextArg()!=STRINGARG)
HError(2519,"HQuant: Label file directory expected");
labDir = GetStrArg();
break;
case 'T':
if (NextArg() != INTARG)
HError(2519,"HQuant: Trace value expected");
trace = GetChkedInt(0,077,s);
break;
case 'X':
if (NextArg()!=STRINGARG)
HError(2519,"HQuant: Label file extension expected");
labExt = GetStrArg();
break;
default:
HError(2519,"HQuant: Unknown switch %s",s);
}
}
if (NextArg()!=STRINGARG)
HError(2519,"HQuant: Output VQ table file name expected");
vqfn = GetStrArg();
if (NextArg()!=STRINGARG)
HError(2519,"HQuant: Training data file name expected");
datafn = GetStrArg();
Initialise(datafn);
LoadFile(datafn);
while (NumArgs()>0) {
if (NextArg()!=STRINGARG)
HError(2519,"HQuant: Training data file name expected");
datafn = GetStrArg();
LoadFile(datafn);
}
for (stream=1;stream<=swidth[0];stream++){
if (trace&T_TOP)
printf("%s-clustering data for stream %d (width %d)\n",
(tType==linTree)?"Flat":"Tree",stream,swidth[stream]);
CalcMeanCov(dSeq,stream);
ClusterVecs(dSeq,stream);
}
WriteVQTable(cs,vqfn);
ResetTrain();
ResetParm();
ResetModel();
ResetVQ();
ResetAudio();
ResetWave();
ResetSigP();
ResetMath();
ResetLabel();
ResetMem();
ResetShell();
Exit(0);
return (0); /* never reached -- make compiler happy */
}
int APIENTRY _tWinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPTSTR lpCmdLine, int nCmdShow)
#endif
{
hInst = hInstance;
isWin64 = IsWindows64();
GetVersionEx(&gOSVer);
int nInstallVer = 0;
wsprintf(gsTitle, msgConEmuInstaller, CONEMUVERL);
lstrcpyn(gsRunAsAdm, msgRunSetupAsAdmin, countof(gsRunAsAdm));
wchar_t szArg[MAX_PATH+1];
LPCWSTR pszCmdToken = GetCommandLine();
LPCWSTR pszCmdLineW = pszCmdToken;
CTempDir temp_dir; // gsTempFolder[0] = 0;
while (0 == NextArg(&pszCmdToken, szArg))
{
if (lstrcmp(szArg, L"/?") == 0 || lstrcmp(szArg, L"-?") == 0 || lstrcmp(szArg, L"-h") == 0
|| lstrcmp(szArg, L"-help") == 0 || lstrcmp(szArg, L"--help") == 0)
{
MessageBox(NULL, msgUsageExample, gsTitle, MB_ICONINFORMATION);
return exit_Cancelled;
}
if (*szArg == L'/')
{
if (szArg[1] == L'e' || szArg[1] == L'E')
{
gbExtractOnly = true;
if (szArg[2] == L':' && szArg[3])
{
lstrcpyn(gsTempFolder, (szArg[3]==L'"') ? (szArg+4) : (szArg+3), countof(gsTempFolder));
}
continue;
}
if (memcmp(szArg, L"/p:x", 4*sizeof(*szArg)) == 0)
{
gbAlreadyAdmin = IsUserAdmin();
if (lstrcmpi(szArg+4, L"86") == 0)
{
nInstallVer = Ver86;
}
else if (lstrcmpi(szArg+4, L"86,adm") == 0)
{
nInstallVer = Ver86;
gbUseElevation = !gbAlreadyAdmin;
}
else if (lstrcmpi(szArg+4, L"64") == 0)
{
nInstallVer = Ver64;
}
else if (lstrcmpi(szArg+4, L"64,adm") == 0)
{
nInstallVer = Ver64;
gbUseElevation = !gbAlreadyAdmin;
}
}
else
pszCmdToken = pszCmdLineW;
break;
}
else if (*szArg == L'-')
{
pszCmdToken = pszCmdLineW;
break;
}
pszCmdLineW = pszCmdToken;
}
if (!temp_dir.Acquire())
{
return exit_CreateDirectory;
}
if (!gbExtractOnly)
{
// If pszCmdToken is not empty - set global var
gbAutoMode = (pszCmdToken && *pszCmdToken);
wchar_t szInstallPath[MAX_PATH+32];
bool bInstalled;
HKEY hk;
lstrcpyn(gsMessage, msgChooseInstallVer, countof(gsMessage));
szInstallPath[0] = 0; bInstalled = false;
struct {HKEY hk; LPCWSTR path; LPCWSTR name; bool our;}
Keys[] = {
{HKEY_LOCAL_MACHINE,L"SOFTWARE\\ConEmu",L"InstallDir",true},
//Current installer does not use FarManager installation dir anymore
//{HKEY_LOCAL_MACHINE,L"SOFTWARE\\Far Manager",L"InstallDir"},
//{HKEY_LOCAL_MACHINE,L"SOFTWARE\\Far2",L"InstallDir"},
//{HKEY_LOCAL_MACHINE,L"SOFTWARE\\Far",L"InstallDir"},
};
for (size_t s = 0; s < countof(Keys); s++)
{
if (!RegOpenKeyEx(Keys[s].hk, Keys[s].path, 0, KEY_READ, &hk)
|| !RegOpenKeyEx(Keys[s].hk, Keys[s].path, 0, KEY_READ|KEY_WOW64_32KEY, &hk))
{
wchar_t szPath[MAX_PATH+1] = {}; DWORD cbSize = sizeof(szPath)-2;
LONG lRc = RegQueryValueEx(hk, Keys[s].name, NULL, NULL, (LPBYTE)szPath, &cbSize);
RegCloseKey(hk);
if (!lRc && *szPath)
{
bInstalled = Keys[s].our;
lstrcpy(szInstallPath, szPath);
cbSize = lstrlen(szInstallPath);
if (szInstallPath[cbSize-1] == L'\\') szInstallPath[cbSize-1] = 0;
break;
}
}
}
if (szInstallPath[0] == 0)
{
GetEnvironmentVariable(L"ProgramFiles", szInstallPath, MAX_PATH);
int nLen = lstrlen(szInstallPath);
lstrcat(szInstallPath, (nLen > 0 && szInstallPath[nLen-1] != L'\\') ? L"\\ConEmu" : L"ConEmu");
}
wsprintf(gsVer86, msgInstallFolderIs, CONEMUVERL, L"x86", bInstalled ? msgPathCurrent : msgPathDefault, szInstallPath);
if (isWin64)
{
szInstallPath[0] = 0; bInstalled = false;
struct {HKEY hk; LPCWSTR path; LPCWSTR name; bool our;}
Keys[] = {
{HKEY_LOCAL_MACHINE,L"SOFTWARE\\ConEmu",L"InstallDir_x64",true},
//Current installer does not use FarManager installation dir anymore
//{HKEY_LOCAL_MACHINE,L"SOFTWARE\\Far Manager",L"InstallDir_x64"},
//{HKEY_LOCAL_MACHINE,L"SOFTWARE\\Far2",L"InstallDir_x64"},
//{HKEY_LOCAL_MACHINE,L"SOFTWARE\\Far",L"InstallDir_x64"},
};
for (size_t s = 0; s < countof(Keys); s++)
{
if (!RegOpenKeyEx(Keys[s].hk, Keys[s].path, 0, KEY_READ|KEY_WOW64_64KEY, &hk))
{
wchar_t szPath[MAX_PATH+1] = {}; DWORD cbSize = sizeof(szPath)-2;
LONG lRc = RegQueryValueEx(hk, Keys[s].name, NULL, NULL, (LPBYTE)szPath, &cbSize);
RegCloseKey(hk);
if (!lRc && *szPath)
{
bInstalled = Keys[s].our;
lstrcpy(szInstallPath, szPath);
cbSize = lstrlen(szInstallPath);
if (szInstallPath[cbSize-1] == L'\\') szInstallPath[cbSize-1] = 0;
break;
}
}
}
if (szInstallPath[0] == 0)
{
GetEnvironmentVariable(L"ProgramW6432", szInstallPath, MAX_PATH);
int nLen = lstrlen(szInstallPath);
lstrcat(szInstallPath, (nLen > 0 && szInstallPath[nLen-1] != L'\\') ? L"\\ConEmu" : L"ConEmu");
}
wsprintf(gsVer64, msgInstallFolderIs, CONEMUVERL, L"x64", bInstalled ? msgPathCurrent : msgPathDefault, szInstallPath);
wsprintf(gsFull, msgInstallConfirm, gsMessage);
}
else
{
gsVer64[0] = 0;
}
}
else
{
LPCWSTR szPath = gsTempFolder;
lstrcpyn(gsMessage, msgChooseExtractVer, countof(gsMessage));
wsprintf(gsVer86, msgExtractX86X64, CONEMUVERL, L"x86", szPath);
wsprintf(gsVer64, msgExtractX86X64, CONEMUVERL, L"x64", szPath);
wsprintf(gsFull, msgExtractConfirm, gsMessage, szPath);
}
if (nInstallVer == 0)
{
nInstallVer = ChooseVersion(); // IDCANCEL/Ver86/Ver64
}
if (nInstallVer != Ver86 && nInstallVer != Ver64)
{
return exit_Cancelled;
}
// Preparing full paths
wsprintf(gsMsiFile, L"%s\\ConEmu.%s.%s.msi", gsTempFolder, CONEMUVERL, (nInstallVer == Ver86) ? L"x86" : L"x64");
wsprintf(gsCabFile, L"%s\\ConEmu.cab", gsTempFolder);
bool lbNeedExe = false;
if (!gbExtractOnly && gOSVer.dwMajorVersion >= 6)
lbNeedExe = true;
if (!lbNeedExe)
gsExeFile[0] = 0;
else
wsprintf(gsExeFile, L"%s\\ConEmuSetup.exe", gsTempFolder);
int iExpMsi = ExportFile(nInstallVer, gsMsiFile);
int iExpCab = (iExpMsi == 0) ? ExportFile(CABFILE, gsCabFile) : -1;
int iExpExe = (!lbNeedExe) ? 0 : (iExpCab == 0) ? ExportFile(EXEFILE, gsExeFile) : -1;
if (iExpMsi != 0 || iExpCab != 0 || iExpExe != 0)
{
DeleteFile(gsMsiFile);
DeleteFile(gsCabFile);
if (*gsExeFile)
DeleteFile(gsExeFile);
return (iExpMsi != 0) ? iExpMsi : iExpCab;
}
if (gbExtractOnly)
{
temp_dir.DontRemove();
wchar_t szMessage[MAX_PATH*2];
wsprintf(szMessage, msgExtractedSuccessfully, gsTempFolder);
MessageBox(NULL, szMessage, gsTitle, MB_ICONINFORMATION);
return exit_Succeeded;
}
int iInstRc = exit_Succeeded;
SHELLEXECUTEINFO sei = {sizeof(sei)};
wchar_t* pszParms = NULL;
sei.fMask = SEE_MASK_NOCLOSEPROCESS|/*SEE_MASK_NOASYNC*/0x00000100; //|/*SEE_MASK_NOZONECHECKS*/0x00800000;
sei.lpVerb = L"open";
if (gOSVer.dwMajorVersion<=5 || !gbUseElevation)
{
sei.lpFile = gsMsiFile;
sei.lpParameters = pszCmdToken;
}
else
{
// Executor has `<requestedExecutionLevel level="requireAdministrator" ...>` in manifest
sei.lpFile = gsExeFile;
int nMaxLen = lstrlen(gsMsiFile) + (pszCmdToken ? lstrlen(pszCmdToken) : 0) + 64;
pszParms = (wchar_t*)malloc(nMaxLen*sizeof(wchar_t));
wsprintf(pszParms, L"/i \"%s\" %s", gsMsiFile, pszCmdToken ? pszCmdToken : L"");
sei.lpParameters = pszParms;
}
sei.lpDirectory = gsTempFolder;
sei.nShow = SW_SHOWNORMAL;
BOOL lbExecute = ShellExecuteEx(&sei);
#if 0
if (!lbExecute && lbNeedExe)
{
DWORD nErr = GetLastError();
if (nErr == 1223)
{
// Отмена пользователем UAC, или правов не хватило?
sei.fMask = SEE_MASK_NOCLOSEPROCESS|/*SEE_MASK_NOASYNC*/0x00000100; //|/*SEE_MASK_NOZONECHECKS*/0x00800000;
sei.lpVerb = L"open";
sei.lpFile = gsMsiFile;
sei.lpParameters = pszCmdToken;
sei.lpDirectory = gsTempFolder;
sei.nShow = SW_SHOWNORMAL;
lbExecute = ShellExecuteEx(&sei);
}
}
#endif
if (!lbExecute)
{
iInstRc = ReportError(exit_ShellExecuteEx, msgInstallerFailed, gsMsiFile);
}
else
{
if (!sei.hProcess)
{
iInstRc = ReportError(exit_NullProcess, msgInstallerFailed, gsMsiFile);
}
else
{
WaitForSingleObject(sei.hProcess, INFINITE);
DWORD nCode = 0;
SetLastError(0);
wchar_t szFormat[256];
if (GetExitCodeProcess(sei.hProcess, &nCode))
{
switch (nCode)
{
case 0:
iInstRc = exit_Succeeded;
break;
case 1602: // cancelled by user
iInstRc = exit_Cancelled; // don't show any errors
break;
case 3010: // reboot is required
wsprintf(szFormat, msgRebootRequired, nCode);
iInstRc = ReportError(exit_AddWin32Code+nCode, szFormat, gsMsiFile);
break;
default:
wsprintf(szFormat, msgInstallerFailedEx, nCode);
iInstRc = ReportError(exit_AddWin32Code+nCode, szFormat, gsMsiFile);
}
}
else
{
lstrcpyn(szFormat, msgExitCodeFailed, countof(szFormat));
iInstRc = ReportError(exit_ExitCodeProcess, szFormat, gsMsiFile);
}
}
}
DeleteFile(gsMsiFile);
DeleteFile(gsCabFile);
if (*gsExeFile)
DeleteFile(gsExeFile);
return iInstRc;
}
int main(int argc, char *argv[])
{
char *datafn, *s;
int nSeg;
void Initialise(void);
void LoadFile(char *fn);
void EstimateModel(void);
void SaveModel(char *outfn);
if(InitShell(argc,argv,hinit_version,hinit_vc_id)<SUCCESS)
HError(2100,"HInit: InitShell failed");
InitMem(); InitLabel();
InitMath(); InitSigP();
InitWave(); InitAudio();
InitVQ(); InitModel();
if(InitParm()<SUCCESS)
HError(2100,"HInit: InitParm failed");
InitTrain(); InitUtil();
if (!InfoPrinted() && NumArgs() == 0)
ReportUsage();
if (NumArgs() == 0) Exit(0);
SetConfParms();
CreateHMMSet(&hset,&gstack,FALSE);
while (NextArg() == SWITCHARG) {
s = GetSwtArg();
if (strlen(s)!=1)
HError(2119,"HInit: Bad switch %s; must be single letter",s);
switch(s[0]){
case 'e':
epsilon = GetChkedFlt(0.0,1.0,s); break;
case 'i':
maxIter = GetChkedInt(0,100,s); break;
case 'l':
if (NextArg() != STRINGARG)
HError(2119,"HInit: Segment label expected");
segLab = GetStrArg();
break;
case 'm':
minSeg = GetChkedInt(1,1000,s); break;
case 'n':
newModel = FALSE; break;
case 'o':
outfn = GetStrArg();
break;
case 'u':
SetuFlags(); break;
case 'v':
minVar = GetChkedFlt(0.0,10.0,s); break;
case 'w':
mixWeightFloor = MINMIX * GetChkedFlt(0.0,10000.0,s);
break;
case 'B':
saveBinary = TRUE;
break;
case 'F':
if (NextArg() != STRINGARG)
HError(2119,"HInit: Data File format expected");
if((dff = Str2Format(GetStrArg())) == ALIEN)
HError(-2189,"HInit: Warning ALIEN Data file format set");
break;
case 'G':
if (NextArg() != STRINGARG)
HError(2119,"HInit: Label File format expected");
if((lff = Str2Format(GetStrArg())) == ALIEN)
HError(-2189,"HInit: Warning ALIEN Label file format set");
break;
case 'H':
if (NextArg() != STRINGARG)
HError(2119,"HInit: HMM macro file name expected");
AddMMF(&hset,GetStrArg());
break;
case 'I':
if (NextArg() != STRINGARG)
HError(2119,"HInit: MLF file name expected");
LoadMasterFile(GetStrArg());
break;
case 'L':
if (NextArg()!=STRINGARG)
HError(2119,"HInit: Label file directory expected");
labDir = GetStrArg(); break;
case 'M':
if (NextArg()!=STRINGARG)
HError(2119,"HInit: Output macro file directory expected");
outDir = GetStrArg();
break;
case 'T':
if (NextArg() != INTARG)
HError(2119,"HInit: Trace value expected");
trace = GetChkedInt(0,01777,s);
break;
case 'X':
if (NextArg()!=STRINGARG)
HError(2119,"HInit: Label file extension expected");
labExt = GetStrArg(); break;
default:
HError(2119,"HInit: Unknown switch %s",s);
}
}
if (NextArg()!=STRINGARG)
HError(2119,"HInit: source HMM file name expected");
hmmfn = GetStrArg();
Initialise();
do {
if (NextArg()!=STRINGARG)
HError(2119,"HInit: training data file name expected");
datafn = GetStrArg();
LoadFile(datafn);
} while (NumArgs()>0);
nSeg = NumSegs(segStore);
if (nSeg < minSeg)
HError(2121,"HInit: Too Few Observation Sequences [%d]",nSeg);
if (trace&T_TOP) {
printf("%d Observation Sequences Loaded\n",nSeg);
fflush(stdout);
}
EstimateModel();
SaveModel(outfn);
if (trace&T_TOP)
printf("Output written to directory %s\n",
outDir==NULL?"current":outDir);
Exit(0);
return (0); /* never reached -- make compiler happy */
}
static Boolean tgtHdr = FALSE; /* print target header info */
static Boolean obsFmt = FALSE; /* print observation format */
static Boolean prData = TRUE; /* print data */
static Boolean rawOut = FALSE; /* raw output i.e no numbering */
static Boolean replay = FALSE; /* replay audio */
static Boolean frcDisc = FALSE; /* List VQ symbols from cont file */
static FileFormat ff = UNDEFF; /* Source File format */
static long gst = -1; /* start sample to list */
static long gen = -1; /* end sample to list */
static int numS = 1; /* number of streams */
static int nItems = 10; /* num items per line */
static int barwidth; /* width of printed bars */
static char barc = '-'; /* bar character */
/* ---------------- Configuration Parameters --------------------- */
static ConfParam *cParm[MAXGLOBS];
static int nParm = 0; /* total num params */
static HTime sampPeriod; /* raw audio input only */
static int audSignal;
/* ---------------- Process Command Line ------------------------- */
/* SetConfParms: set conf parms relevant to this tool */
void SetConfParms(void)
{
int i;
double d;
sampPeriod = 0.0; audSignal = NULLSIG;
nParm = GetConfig("HLIST", TRUE, cParm, MAXGLOBS);
if (nParm>0){
if (GetConfInt(cParm,nParm,"AUDIOSIG",&i)) audSignal = i;
if (GetConfInt(cParm,nParm,"TRACE",&i)) trace = i;
if (GetConfFlt(cParm,nParm,"SOURCERATE",&d)) sampPeriod = d;
}
}
/* main: */
int main(int argc, char *argv[])
{
char *s;
char fname[MAXSTRLEN];
InitShell(argc, argv, hmgetool_version, hmgetool_vc_id);
InitMem();
InitMath();
InitSigP();
InitWave();
InitLabel();
InitModel();
InitTrain();
InitParm();
InitUtil();
InitFB();
InitGen();
InitAdapt(&xfInfo, NULL);
InitMTrain();
/* process argument */
if (NumArgs() == 0)
ReportUsage();
CreateHeap(&hmmStack, "Model Stack", MSTAK, 1, 1.0, 80000, 4000000);
CreateHeap(&orighmmStack, "Model Stack", MSTAK, 1, 1.0, 80000, 4000000);
CreateHeap(&accStack, "Acc Stack", MSTAK, 1, 1.0, 80000, 400000);
CreateHeap(&genStack, "Gen Stack", MSTAK, 1, 1.0, 80000, 400000);
CreateHeap(&mgeStack, "MGE Train Stack", MSTAK, 1, 1.0, 80000, 400000);
SetConfParms();
CreateHMMSet(&hset, &hmmStack, TRUE);
CreateHMMSet(&orighset, &orighmmStack, TRUE);
statInfo = (MTStatInfo *) New(&gstack, sizeof(MTStatInfo));
memset(statInfo, 0, sizeof(MTStatInfo));
genInfo = (GenInfo *) New(&genStack, sizeof(GenInfo));
memset(genInfo, 0, sizeof(GenInfo));
genInfo->hset = &hset;
genInfo->genMem = &genStack;
mtInfo = (MgeTrnInfo *) New(&mgeStack, sizeof(MgeTrnInfo));
memset(mtInfo, 0, sizeof(MgeTrnInfo));
mtInfo->genInfo = genInfo;
mtInfo->statInfo = statInfo;
mtInfo->hset = &hset;
mtInfo->orighset = &orighset;
mtInfo->mgeMem = &mgeStack;
while (NextArg() == SWITCHARG) {
s = GetSwtArg();
if (strlen(s) != 1)
HError(6601, "HMgeTool: Bad switch %s; must be single letter", s);
switch (s[0]) {
case 'a':
nMaxBAIter = GetChkedInt(1, 1000, s);
nMaxBALen = GetChkedInt(1, 1000, s);
break;
case 'b':
mtInfo->bBoundAdj = TRUE;
nBAEndIter = GetChkedInt(0, 1000, s);
nBoundAdjWin = GetChkedInt(1, 1000, s);
break;
case 'c':
outProcData = TRUE;
break;
case 'd':
if (NextArg() != STRINGARG)
HError(6601, "HMgeTool: HMM definition directory expected");
hmmDir = GetStrArg();
break;
case 'e':
mtInfo->bStepLimit = TRUE;
break;
case 'f':
frameRate = (HTime) GetChkedFlt(0.0, 10000000.0, s);
break;
case 'g':
mtInfo->bMVar = TRUE;
break;
case 'i':
startIter = GetChkedInt(0, 1000, s);
endIter = GetChkedInt(startIter, 1000, s);
break;
case 'j':
funcType = GetChkedInt(0, 2, s);
mtInfo->funcType = funcType;
break;
case 'l':
if (NextArg() != STRINGARG)
HError(6601, "HMgeTool: Label file output directory expected");
outLabDir = GetStrArg();
break;
case 'o':
if (NextArg() != STRINGARG)
HError(6601, "HMgeTool: HMM file extension expected");
outExt = GetStrArg();
break;
case 'p':
A_STEP = GetChkedFlt(0.0, 10000000.0, s);
B_STEP = GetChkedFlt(0.0, 10000000.0, s);
break;
case 'r':
mtInfo->bOrigHmmRef = TRUE;
if (NextArg() != STRINGARG)
HError(6601, "HMgeTool: HMM macro file name expected");
s = GetStrArg();
AddMMF(&orighset, s);
break;
case 's': /* updating scale file */
scalefn = GetStrArg();
break;
case 'u':
SetuFlags();
break;
case 'v':
MSDthresh = GetChkedFlt(0.0, 1.0, s);
break;
case 'w':
fGVDistWght = GetChkedFlt(0.0, 1000.0, s);
break;
case 'x':
if (NextArg() != STRINGARG)
HError(6601, "HMgeTool: HMM file extension expected");
hmmExt = GetStrArg();
break;
case 'B':
inBinary = TRUE;
break;
case 'G':
mtInfo->nGainStreamIndex = GetChkedInt(1, SMAX, s);
mtInfo->nGainDimIndex = GetChkedInt(1, 1000, s);
if (NextArg() == FLOATARG || NextArg() == INTARG)
mtInfo->fGainWghtComp = GetChkedFlt(-10000.0, 1000000.0, s);
break;
case 'H':
if (NextArg() != STRINGARG)
HError(6601, "HMgeTool: HMM macro file name expected");
mmfFn = GetStrArg();
AddMMF(&hset, mmfFn);
break;
case 'I':
if (NextArg() != STRINGARG)
HError(6601, "HMgeTool: MLF file name expected");
LoadMasterFile(GetStrArg());
break;
case 'J': /* regression class and tree */
if (NextArg() != STRINGARG)
HError(6601, "HMgeTool: HMM regression class/tree file name expected");
s = GetStrArg();
AddMMF(&hset, s);
AddMMF(&orighset, s);
break;
case 'K':
if (NextArg() != STRINGARG)
HError(6601, "HMgeTool: HMM transform file name expected");
xformfn = GetStrArg();
break;
case 'L':
if (NextArg() != STRINGARG)
HError(6601, "HMgeTool: Label file directory expected");
labDir = GetStrArg();
break;
case 'M':
if (NextArg() != STRINGARG)
HError(6601, "HMgeTool: Output macro file directory expected");
outDir = GetStrArg();
break;
case 'T':
trace = GetChkedInt(0, 0100000, s);
break;
case 'X':
if (NextArg() != STRINGARG)
HError(2319, "HMGenS: Label file extension expected");
labExt = GetStrArg();
break;
default:
HError(6601, "HMgeTool: Unknown switch %s", s);
}
}
if (NextArg() != STRINGARG)
HError(6601, "HMgeTool: file name of model list expected");
hmmListFn = GetStrArg();
Initialise();
if (funcType == MGE_EVAL) {
PerformMgeEval();
} else if (funcType == MGE_TRAIN) {
PerformMgeTrain();
if (endIter > 0 && bMgeUpdate) {
/* output HMM files */
ConvDiagC(&hset, TRUE);
SaveHMMSet(&hset, outDir, outExt, NULL, inBinary);
}
} else if (funcType == MGE_ADAPT) {
PerformMgeAdapt();
if (endIter > 0 && bMgeUpdate) {
MakeFN(xformfn, outDir, NULL, fname);
SaveOneXForm(&hset, hset.curXForm, fname, FALSE);
}
}
ResetHeap(&hmmStack);
ResetHeap(&orighmmStack);
ResetHeap(&accStack);
ResetHeap(&genStack);
ResetHeap(&mgeStack);
return 0;
}
//------------------------------------------------------------------------
///| Parsing the command line |///////////////////////////////////////////
//------------------------------------------------------------------------
// Returns:
// true - continue normal startup
// false - exit process with iResult code
bool CConEmuStart::ParseCommandLine(LPCWSTR pszCmdLine, int& iResult)
{
bool bRc = false;
iResult = 100;
_ASSERTE(pszCmdLine!=NULL);
opt.cmdLine.Set(pszCmdLine ? pszCmdLine : L"");
// pszCmdLine *may* or *may not* start with our executable or full path to our executable
LPCWSTR pszTemp = opt.cmdLine;
LPCWSTR cmdLineRest = SkipNonPrintable(opt.cmdLine);
LPCWSTR pszName, pszArgStart;
LPCWSTR psUnknown = NULL;
CmdArg szArg, szNext;
CEStr szExeName, szExeNameOnly;
// Set %ConEmuArgs% env var
// It may be useful if we need to restart ConEmu
// from batch/script with the same arguments (selfupdate etc.)
LPCWSTR pszCopyToEnvStart = NULL;
// Have to get our exectuable name and name without extension
szExeName.Set(PointToName(gpConEmu->ms_ConEmuExe));
szExeNameOnly.Set(szExeName);
wchar_t* pszDot = (wchar_t*)PointToExt(szExeNameOnly.ms_Val);
_ASSERTE(pszDot);
if (pszDot) *pszDot = 0;
// Check the first argument in the command line (most probably it will be our executable path/name)
if (!(pszTemp = NextArg(pszTemp, szArg)))
{
_ASSERTE(FALSE && "GetCommandLine() is empty");
// Treat as empty command line, allow to start
bRc = true; iResult = 0;
goto wrap;
}
pszName = PointToName(szArg);
if ((lstrcmpi(pszName, szExeName) == 0)
|| (lstrcmpi(pszName, szExeNameOnly) == 0))
{
// OK, our executable was specified properly in the command line
_ASSERTE(*pszTemp != L' ');
cmdLineRest = SkipNonPrintable(pszTemp);
}
// Must be empty at the moment
_ASSERTE(opt.runCommand.IsEmpty());
// Does the command line contain our switches?
// Or we need to append all switches to starting shell?
if (cmdLineRest && *cmdLineRest)
{
pszTemp = cmdLineRest;
if ((pszTemp = NextArg(pszTemp, szArg)))
{
if ((*szArg.ms_Val != L'/')
&& (*szArg.ms_Val != L'-')
/*&& !wcschr(szArg.ms_Val, L'/')*/
)
{
// Save it for further use
opt.runCommand.Set(cmdLineRest);
// And do not process it (no switches at all)
cmdLineRest = NULL;
opt.params = -1;
}
}
}
struct RunAsAdmin
{
static bool Check(LPCWSTR asSwitch)
{
bool bRunAsAdmin = false; // isPressed(VK_SHIFT);
return bRunAsAdmin;
};
};
// Let parse the reset
szArg.Empty();
szNext.Empty();
// Processing loop begin
if (cmdLineRest && *cmdLineRest)
{
pszCopyToEnvStart = cmdLineRest;
opt.cfgSwitches.Set(pszCopyToEnvStart);
while ((cmdLineRest = NextArg(cmdLineRest, szArg, &pszArgStart)))
{
bool lbNotFound = false;
TODO("Replace NeedNextArg with GetCfgParm?")
#define NeedNextArg() \
if (!(cmdLineRest = NextArg(cmdLineRest, szNext))) { iResult = CERR_CARGUMENT; goto wrap; }
if (!szArg.IsPossibleSwitch())
{
// -- // continue; // Try next switch?
// Show error on unknown switch
psUnknown = pszArgStart;
break;
}
// Main processing cycle
{
opt.params++;
if (szArg.IsSwitch(L"-autosetup"))
{
BOOL lbTurnOn = TRUE;
NeedNextArg();
if (szNext.Compare(L"0") == 0)
{
lbTurnOn = FALSE;
}
else if (szNext.Compare(L"1") == 0)
{
NeedNextArg();
DWORD dwAttr = GetFileAttributes(szNext);
if (dwAttr == (DWORD)-1 || (dwAttr & FILE_ATTRIBUTE_DIRECTORY))
{
iResult = 102;
goto wrap;
}
}
else
{
iResult = CERR_CARGUMENT;
goto wrap;
}
HKEY hk = NULL; DWORD dw;
int nSetupRc = 100;
if (0 != RegCreateKeyEx(HKEY_CURRENT_USER, _T("Software\\Microsoft\\Command Processor"),
0, NULL, 0, KEY_ALL_ACCESS, NULL, &hk, &dw))
{
iResult = 103;
goto wrap;
}
if (lbTurnOn)
{
size_t cchMax = szNext.GetLen();
LPCWSTR pszArg1 = NULL;
if (*cmdLineRest)
{
// May be ‘/GHWND=NEW’ or smth else
pszArg1 = cmdLineRest;
cchMax += _tcslen(pszArg1);
}
cchMax += 16; // + quotations, spaces and so on
wchar_t* pszCmd = (wchar_t*)calloc(cchMax, sizeof(*pszCmd));
swprintf_c(pszCmd, cchMax/*#SECURELEN*/, L"\"%s\"%s%s%s", szNext.ms_Val,
pszArg1 ? L" \"" : L"", pszArg1 ? pszArg1 : L"", pszArg1 ? L"\"" : L"");
if (0 == RegSetValueEx(hk, _T("AutoRun"), 0, REG_SZ, (LPBYTE)pszCmd,
(DWORD)sizeof(*pszCmd)*(_tcslen(pszCmd)+1)))
nSetupRc = 1;
free(pszCmd);
}
else
{
if (0==RegDeleteValue(hk, _T("AutoRun")))
nSetupRc = 1;
}
RegCloseKey(hk);
// сбросить CreateInNewEnvironment для ConMan
ResetConman();
iResult = nSetupRc;
goto wrap;
}
else if (szArg.OneOfSwitches(L"-bypass", L"-apparent", L"-system:", L"-interactive:", L"-demote"))
{
// -bypass
// Этот ключик был придуман для прозрачного запуска консоли
// в режиме администратора
// (т.е. чтобы окно UAC нормально всплывало, но не мелькало консольное окно)
// Но не получилось, пока требуются хэндлы процесса, а их не получается
// передать в НЕ приподнятый процесс (исходный ConEmu GUI).
// -apparent
// Same as -bypass, but run the process as SW_SHOWNORMAL
// -demote
// Запуск процесса (ком.строка после "/demote") в режиме простого юзера,
// когда текущий процесс уже запущен "под админом". "Понизить" текущие
// привилегии просто так нельзя, поэтому запуск идет через TaskSheduler.
// -system
// Non-interactive process, started as System account
// It's used when starting consoles, our server works fine as non-interactive
// -interactive
// Used when ConEmu.exe is started under System account,
// but we need to give starting process interactive capabilities.
_ASSERTE(opt.runCommand.IsEmpty());
pszTemp = cmdLineRest;
if ((pszTemp = NextArg(pszTemp, szNext))
&& szNext.OneOfSwitches(L"-run",L"-cmd"))
{
opt.runCommand.Set(pszTemp);
}
else
{
opt.runCommand.Set(cmdLineRest);
}
if (opt.runCommand.IsEmpty())
{
CEStr lsMsg(L"Invalid command line. '", szArg, L"' exists, command line is empty");
DisplayLastError(lsMsg, -1);
goto wrap;
}
// Information
#ifdef _DEBUG
STARTUPINFO siOur = {sizeof(siOur)};
GetStartupInfo(&siOur);
#endif
STARTUPINFO si = {sizeof(si)};
PROCESS_INFORMATION pi = {};
si.dwFlags = STARTF_USESHOWWINDOW;
// Only `-demote` and `-apparent` switches were implemented to start application visible
// All others are intended to run our server process, without blinking of course
if (szArg.OneOfSwitches(L"-demote", L"-apparent"))
si.wShowWindow = SW_SHOWNORMAL;
else
si.wShowWindow = SW_HIDE;
wchar_t szCurDir[MAX_PATH+1] = L"";
GetCurrentDirectory(countof(szCurDir), szCurDir);
BOOL b;
DWORD nErr = 0;
// if we were started from TaskScheduler, it would be nice to wait a little
// to let parent (creator of the scheduler task) know we were started successfully
bool bFromScheduler = false;
// Log the command to be started
{
CEStr lsLog(
L"Starting process",
L": ", szArg,
L" `", opt.runCommand.ms_Val, L"`");
LogString(lsLog);
}
if (szArg.IsSwitch(L"-demote"))
{
b = CreateProcessDemoted(opt.runCommand.ms_Val, NULL, NULL, FALSE, NORMAL_PRIORITY_CLASS, NULL,
szCurDir, &si, &pi, &nErr);
}
else if (szArg.IsSwitch(L"-system:"))
{
DWORD nSessionID = wcstoul(szArg.ms_Val+wcslen(L"-system:"), NULL, 10);
b = CreateProcessSystem(nSessionID, opt.runCommand.ms_Val, NULL, NULL, FALSE, NORMAL_PRIORITY_CLASS, NULL,
szCurDir, &si, &pi);
}
else if (szArg.IsSwitch(L"-interactive:"))
{
DWORD nSessionID = wcstoul(szArg.ms_Val+wcslen(L"-interactive:"), NULL, 10);
b = CreateProcessInteractive(nSessionID, NULL, opt.runCommand.ms_Val, NULL, NULL, TRUE, NORMAL_PRIORITY_CLASS, NULL,
szCurDir, &si, &pi, &nErr);
bFromScheduler = true;
}
else // -bypass, -apparent
{
b = CreateProcess(NULL, opt.runCommand.ms_Val, NULL, NULL, TRUE, NORMAL_PRIORITY_CLASS, NULL,
NULL, &si, &pi);
nErr = b ? 0 : GetLastError();
bFromScheduler = true;
}
// Log the result
{
CEStr lsLog; wchar_t szExtra[32] = L"";
if (b)
{
if (pi.dwProcessId)
swprintf_c(szExtra, L", PID=%u", pi.dwProcessId);
lsLog = lstrmerge(
L"Process was created successfully",
szExtra);
}
else
{
swprintf_c(szExtra, L", ErrorCode=%u", nErr);
lsLog = lstrmerge(
L"Failed to start process",
szExtra);
}
LogString(lsLog);
}
// If the error was not shown yet
if (nErr) DisplayLastError(opt.runCommand, nErr);
// if we were started from TaskScheduler, it would be nice to wait a little
// to let parent (creator of the scheduler task) know we were started successfully
if (bFromScheduler)
{
LogString(L"Sleeping for 5 seconds");
Sleep(5*1000);
}
// Success?
if (b)
{
iResult = 0;
}
// Done, close handles, if they were opened
SafeCloseHandle(pi.hProcess);
SafeCloseHandle(pi.hThread);
goto wrap;
}
else if (szArg.IsSwitch(L"-multi"))
{
gpConEmu->AppendExtraArgs(szArg);
gpConEmu->opt.MultiConValue = true;
}
else if (szArg.IsSwitch(L"-NoMulti"))
{
gpConEmu->AppendExtraArgs(szArg);
gpConEmu->opt.MultiConValue = false;
}
else if (szArg.IsSwitch(L"-visible"))
{
gpConEmu->opt.VisValue = true;
}
else if (szArg.OneOfSwitches(L"-ct", L"-cleartype", L"-ct0", L"-ct1", L"-ct2"))
{
switch (szArg[3])
{
case L'0':
gpConEmu->opt.ClearTypeVal = NONANTIALIASED_QUALITY; break;
case L'1':
gpConEmu->opt.ClearTypeVal = ANTIALIASED_QUALITY; break;
default:
gpConEmu->opt.ClearTypeVal = CLEARTYPE_NATURAL_QUALITY;
}
}
// Interface language
else if (szArg.IsSwitch(L"-lng"))
{
NeedNextArg();
if (!gpConEmu->opt.Language.Exists)
{
gpConEmu->opt.Language = (LPCWSTR)szNext;
gpConEmu->AppendExtraArgs(L"-lng", szNext);
}
}
// Optional specific "ConEmu.l10n"
else if (szArg.IsSwitch(L"-lngfile"))
{
NeedNextArg();
if (!gpConEmu->opt.LanguageFile.Exists)
{
gpConEmu->opt.LanguageFile = (LPCWSTR)szNext;
gpConEmu->AppendExtraArgs(L"-lngfile", szNext);
}
}
// Change font name
else if (szArg.IsSwitch(L"-Font"))
{
NeedNextArg();
if (!gpConEmu->opt.FontVal.Exists)
{
gpConEmu->opt.FontVal = (LPCWSTR)szNext;
gpConEmu->AppendExtraArgs(L"-font", szNext);
}
}
// Change font height
else if (szArg.IsSwitch(L"-FontSize") || szArg.IsSwitch(L"-Size"))
{
NeedNextArg();
if (!gpConEmu->opt.SizeVal.Exists)
{
gpConEmu->opt.SizeVal.SetInt(szNext);
}
}
// ADD fontname; by Mors
else if (szArg.IsSwitch(L"-FontFile"))
{
CESwitch szFile(sw_Str);
if (!GetCfgParm(cmdLineRest, szFile, MAX_PATH))
{
goto wrap;
}
gpConEmu->AppendExtraArgs(L"-FontFile", szFile.GetStr());
gpFontMgr->RegisterFont(szFile.GetStr(), TRUE);
}
// Register all fonts from specified directory
else if (szArg.IsSwitch(L"-FontDir"))
{
CESwitch szDir(sw_Str);
if (!GetCfgParm(cmdLineRest, szDir, MAX_PATH))
{
goto wrap;
}
gpConEmu->AppendExtraArgs(L"-FontDir", szDir.GetStr());
gpFontMgr->RegisterFontsDir(szDir.GetStr());
}
else if (szArg.IsSwitch(L"-fs"))
{
gpConEmu->opt.WindowModeVal = wmFullScreen;
}
else if (szArg.IsSwitch(L"-max"))
{
gpConEmu->opt.WindowModeVal = wmMaximized;
}
else if (szArg.OneOfSwitches(L"-min", L"-MinTSA", L"-StartTSA"))
{
gpConEmu->WindowStartMinimized = true;
if (!szArg.IsSwitch(L"-min"))
{
gpConEmu->WindowStartTsa = true;
gpConEmu->WindowStartNoClose = szArg.IsSwitch(L"-MinTSA");
}
}
else if (szArg.OneOfSwitches(L"-tsa", L"-tray"))
{
gpConEmu->ForceMinimizeToTray = true;
}
else if (szArg.IsSwitch(L"-detached"))
{
gpConEmu->m_StartDetached = crb_On;
opt.Detached = true;
}
else if (szArg.IsSwitch(L"-NoAutoClose"))
{
opt.NoAutoClose = true;
}
else if (szArg.IsSwitch(L"-here"))
{
gpConEmu->mb_ConEmuHere = true;
gpConEmu->StoreWorkDir();
}
else if (szArg.IsSwitch(L"-update"))
{
gpConEmu->opt.AutoUpdateOnStart = true;
}
else if (szArg.IsSwitch(L"-NoUpdate"))
{
// This one has more weight than AutoUpdateOnStart
gpConEmu->opt.DisableAutoUpdate = true;
}
else if (szArg.IsSwitch(L"-NoHooksWarn"))
{
// Don't try to warn users about known problems with third-party detours
gpConEmu->opt.NoHooksWarn = true;
}
else if (szArg.OneOfSwitches(L"-NoKeyHook", L"-NoKeyHooks", L"-NoKeybHook", L"-NoKeybHooks"))
{
gpConEmu->DisableKeybHooks = true;
}
else if (szArg.IsSwitch(L"-NoCloseConfirm"))
{
gpConEmu->DisableCloseConfirm = true;
}
else if (szArg.IsSwitch(L"-NoMacro"))
{
gpConEmu->DisableAllMacro = true;
}
else if (szArg.OneOfSwitches(L"-NoHotkey", L"-NoHotkeys"))
{
gpConEmu->DisableAllHotkeys = true;
}
else if (szArg.OneOfSwitches(L"-NoDefTrm", L"-NoDefTerm"))
{
gpConEmu->DisableSetDefTerm = true;
}
else if (szArg.OneOfSwitches(L"-NoRegFont", L"-NoRegFonts"))
{
gpConEmu->DisableRegisterFonts = true;
}
else if (szArg.OneOfSwitches(L"-inside", L"-inside="))
{
bool bRunAsAdmin = RunAsAdmin::Check(szArg.ms_Val);
bool bSyncDir = false;
LPCWSTR pszSyncFmt = NULL;
gpConEmu->mb_ConEmuHere = true;
gpConEmu->StoreWorkDir();
// Both `-inside:...` and `-inside=...` are supported
if (szArg.IsSwitch(L"-inside="))
{
bSyncDir = true;
pszSyncFmt = szArg.ms_Val+8; // \eCD /d %1 - \e - ESC, \b - BS, \n - ENTER, %1 - "dir", %2 - "bash dir"
}
CConEmuInside::InitInside(bRunAsAdmin, bSyncDir, pszSyncFmt, 0, NULL);
}
else if (szArg.IsSwitch(L"-InsidePID"))
{
NeedNextArg();
bool bRunAsAdmin = RunAsAdmin::Check(szArg.ms_Val);
wchar_t* pszEnd;
// Здесь указывается PID, в который нужно внедриться.
DWORD nInsideParentPID = wcstol(szNext, &pszEnd, 10);
if (nInsideParentPID)
{
CConEmuInside::InitInside(bRunAsAdmin, false, NULL, nInsideParentPID, NULL);
}
}
else if (szArg.IsSwitch(L"-InsideWnd"))
{
NeedNextArg();
LPCWSTR pszHWnd = szNext.ms_Val;
if (pszHWnd[0] == L'0' && (pszHWnd[1] == L'x' || pszHWnd[1] == L'X'))
pszHWnd += 2;
else if (pszHWnd[0] == L'x' || pszHWnd[0] == L'X')
pszHWnd ++;
bool bRunAsAdmin = RunAsAdmin::Check(szArg.ms_Val);
wchar_t* pszEnd;
// Здесь указывается HWND, в котором нужно создаваться.
HWND hParent = (HWND)(DWORD_PTR)wcstoul(pszHWnd, &pszEnd, 16);
if (hParent && IsWindow(hParent))
{
CConEmuInside::InitInside(bRunAsAdmin, false, NULL, 0, hParent);
}
}
else if (szArg.IsSwitch(L"-icon"))
{
NeedNextArg();
if (!gpConEmu->opt.IconPrm.Exists && !szNext.IsEmpty())
{
gpConEmu->opt.IconPrm = true;
gpConEmu->mps_IconPath = ExpandEnvStr(szNext);
}
}
else if (szArg.IsSwitch(L"-dir"))
{
NeedNextArg();
if (!szNext.IsEmpty())
{
// Например, "%USERPROFILE%"
CEStr szExpand;
if (wcschr(szNext, L'%') && ((szExpand = ExpandEnvStr(szNext)) != NULL))
{
gpConEmu->StoreWorkDir(szExpand);
}
else
{
gpConEmu->StoreWorkDir(szNext);
}
}
}
else if (szArg.IsSwitch(L"-UpdateJumpList"))
{
// Copy current Task list to Win7 Jump list (Taskbar icon)
gpConEmu->mb_UpdateJumpListOnStartup = true;
}
else if (szArg.OneOfSwitches(L"-log", L"-log0", L"-log1", L"-log2", L"-log3", L"-log4"))
{
if (szArg.OneOfSwitches(L"-log", L"-log0"))
gpConEmu->opt.AdvLogging.SetInt(1);
else
gpConEmu->opt.AdvLogging.SetInt((BYTE)(szArg[4] - L'0')); // 1..4
// Do create logging service
DEBUGSTRSTARTUP(L"Creating log file");
gpConEmu->CreateLog();
}
else if (szArg.OneOfSwitches(L"-Single", L"-Reuse"))
{
// "/reuse" switch to be remastered
gpConEmu->AppendExtraArgs(szArg);
gpSetCls->SingleInstanceArg = sgl_Enabled;
}
else if (szArg.IsSwitch(L"-NoSingle"))
{
gpConEmu->AppendExtraArgs(szArg);
gpSetCls->SingleInstanceArg = sgl_Disabled;
}
else if (szArg.IsSwitch(L"-DesktopMode"))
{
gpConEmu->opt.DesktopMode = true;
}
else if (szArg.OneOfSwitches(L"-Quake", L"-QuakeAuto", L"-NoQuake"))
{
if (szArg.IsSwitch(L"-Quake"))
gpConEmu->opt.QuakeMode = 1;
else if (szArg.IsSwitch(L"-QuakeAuto"))
gpConEmu->opt.QuakeMode = 2;
else
{
gpConEmu->opt.QuakeMode = 0;
if (gpSetCls->SingleInstanceArg == sgl_Default)
gpSetCls->SingleInstanceArg = sgl_Disabled;
}
}
else if (szArg.OneOfSwitches(L"-FrameWidth", L"-Frame"))
{
NeedNextArg();
if (!gpConEmu->opt.FrameWidth.Exists)
gpConEmu->opt.FrameWidth.SetInt(szNext);
}
else if (szArg.OneOfSwitches(L"-ShowHide", L"-ShowHideTSA"))
{
gpSetCls->SingleInstanceArg = sgl_Enabled;
gpSetCls->SingleInstanceShowHide = szArg.IsSwitch(L"-ShowHide")
? sih_ShowMinimize : sih_ShowHideTSA;
}
else if (szArg.OneOfSwitches(L"-Reset", L"-ResetDefault", L"-Basic"))
{
gpConEmu->opt.ResetSettings = true;
if (szArg.IsSwitch(L"-ResetDefault"))
{
gpSetCls->isFastSetupDisabled = true;
}
else if (szArg.IsSwitch(L"-Basic"))
{
gpSetCls->isFastSetupDisabled = true;
gpSetCls->isResetBasicSettings = true;
}
}
else if (szArg.OneOfSwitches(L"-NoCascade", L"-DontCascade"))
{
gpConEmu->AppendExtraArgs(szArg);
gpSetCls->isDontCascade = true;
}
else if (szArg.OneOfSwitches(L"-WndX", L"-WndY", L"-WndW", L"-WndWidth", L"-WndH", L"-WndHeight"))
{
wchar_t ch = szArg[4];
CharUpperBuff(&ch, 1);
CESwitch psz(sw_Str); bool bParm = false;
if (!GetCfgParm(cmdLineRest, bParm, psz, 32))
{
goto wrap;
}
gpConEmu->opt.SizePosPrm = true;
// Direct X/Y implies /nocascade
if (ch == _T('X') || ch == _T('Y'))
{
// TODO: isDontCascade must be in our opt struct !!!
gpSetCls->isDontCascade = true;
}
switch (ch)
{
case _T('X'): gpConEmu->opt.sWndX.SetStr(psz.Str, sw_Str); break;
case _T('Y'): gpConEmu->opt.sWndY.SetStr(psz.Str, sw_Str); break;
case _T('W'): gpConEmu->opt.sWndW.SetStr(psz.Str, sw_Str); break;
case _T('H'): gpConEmu->opt.sWndH.SetStr(psz.Str, sw_Str); break;
}
}
else if (szArg.IsSwitch(L"-Monitor"))
{
CESwitch psz(sw_Str); bool bParm = false;
if (!GetCfgParm(cmdLineRest, bParm, psz, 64))
{
goto wrap;
}
if ((gpConEmu->opt.Monitor.Mon = MonitorFromParam(psz.Str)) != NULL)
{
gpConEmu->opt.Monitor.Exists = true;
gpConEmu->opt.Monitor.Type = sw_Int;
gpStartEnv->hStartMon = gpConEmu->opt.Monitor.Mon;
}
}
else if (szArg.IsSwitch(L"-Theme"))
{
const wchar_t* kDefaultTheme = L"DarkMode_Explorer";
bool bParm = false;
if (!cmdLineRest || (*cmdLineRest == L'-' || *cmdLineRest == L'/')
|| !GetCfgParm(cmdLineRest, bParm, gpConEmu->opt.WindowTheme, 128))
{
gpConEmu->opt.WindowTheme.SetStr(kDefaultTheme);
}
}
else if (szArg.OneOfSwitches(L"-Buffer", L"-BufferHeight"))
{
NeedNextArg();
if (!gpConEmu->opt.BufferHeightVal.Exists)
{
gpConEmu->opt.BufferHeightVal.SetInt(szNext);
if (gpConEmu->opt.BufferHeightVal.GetInt() < 0)
{
//setParent = true; -- Maximus5 - нефиг, все ручками
gpConEmu->opt.BufferHeightVal = -gpConEmu->opt.BufferHeightVal.GetInt();
}
if (gpConEmu->opt.BufferHeightVal.GetInt() < LONGOUTPUTHEIGHT_MIN)
gpConEmu->opt.BufferHeightVal = LONGOUTPUTHEIGHT_MIN;
else if (gpConEmu->opt.BufferHeightVal.GetInt() > LONGOUTPUTHEIGHT_MAX)
gpConEmu->opt.BufferHeightVal = LONGOUTPUTHEIGHT_MAX;
}
}
else if (szArg.IsSwitch(L"-Config"))
{
// -- используем последний из параметров, если их несколько
if (!GetCfgParm(cmdLineRest, gpConEmu->opt.ConfigVal, 127))
{
goto wrap;
}
}
else if (szArg.IsSwitch(L"-Palette"))
{
// -- используем последний из параметров, если их несколько
if (!GetCfgParm(cmdLineRest, gpConEmu->opt.PaletteVal, MAX_PATH))
{
goto wrap;
}
}
else if (szArg.IsSwitch(L"-LoadRegistry"))
{
gpConEmu->AppendExtraArgs(szArg);
gpConEmu->opt.ForceUseRegistryPrm = true;
}
else if (szArg.OneOfSwitches(L"-LoadCfgFile", L"-LoadXmlFile"))
{
// -- используем последний из параметров, если их несколько
if (!GetCfgParm(cmdLineRest, gpConEmu->opt.LoadCfgFile, MAX_PATH, true))
{
goto wrap;
}
}
else if (szArg.OneOfSwitches(L"-SaveCfgFile", L"-SaveXmlFile"))
{
// -- используем последний из параметров, если их несколько
if (!GetCfgParm(cmdLineRest, gpConEmu->opt.SaveCfgFile, MAX_PATH, true))
{
goto wrap;
}
}
else if (szArg.IsSwitch(L"-GuiMacro"))
{
// -- выполняется только последний
if (!GetCfgParm(cmdLineRest, gpConEmu->opt.ExecGuiMacro, 0x8000, false))
{
goto wrap;
}
}
else if (szArg.IsSwitch(L"-UpdateSrcSet"))
{
// -- используем последний из параметров, если их несколько
if (!GetCfgParm(cmdLineRest, gpConEmu->opt.UpdateSrcSet, MAX_PATH*4, false))
{
goto wrap;
}
}
else if (szArg.IsSwitch(L"-AnsiLog"))
{
// -- используем последний из параметров, если их несколько
if (!GetCfgParm(cmdLineRest, gpConEmu->opt.AnsiLogPath, MAX_PATH-40, true))
{
goto wrap;
}
}
else if (szArg.IsSwitch(L"-SetDefTerm"))
{
gpConEmu->opt.SetUpDefaultTerminal = true;
}
else if (szArg.IsSwitch(L"-ZoneId"))
{
gpConEmu->opt.FixZoneId = true;
}
else if (szArg.IsSwitch(L"-Exit"))
{
gpConEmu->opt.ExitAfterActionPrm = true;
}
else if (szArg.IsSwitch(L"-QuitOnClose"))
{
gpConEmu->mb_ForceQuitOnClose = true;
}
else if (szArg.IsSwitch(L"-Title"))
{
bool bOk = false;
CESwitch pszTitle(sw_Str);
if (!GetCfgParm(cmdLineRest, bOk, pszTitle, 127))
{
goto wrap;
}
gpConEmu->SetTitleTemplate(pszTitle.GetStr());
}
else if (szArg.IsSwitch(L"-Settings"))
{
gpConEmu->mb_SettingsRequested = true;
}
else if (szArg.IsSwitch(L"-FindBugMode"))
{
gpConEmu->mb_FindBugMode = true;
}
else if (szArg.OneOfSwitches(L"-debug", L"-debugw", L"-debugi"))
{
// These switches were already processed
}
else if (szArg.OneOfSwitches(L"-?", L"-h", L"-help"))
{
if (gpLng) gpLng->Reload();
ConEmuAbout::OnInfo_About();
iResult = -1;
goto wrap;
}
// Final `-run ...` or `-runlist ...` (old names `-cmd ...` or `-cmdlist ...`)
else if (
szArg.OneOfSwitches(L"-run", L"-runlist", L"-cmd", L"-cmdlist")
)
{
if (opt.cfgSwitches.ms_Val)
{
_ASSERTE(pszArgStart>=pszCopyToEnvStart); // If there is only "-run cmd" in arguments
_ASSERTE((INT_PTR)(pszArgStart - pszCopyToEnvStart) <= opt.cfgSwitches.GetLen());
opt.cfgSwitches.ms_Val[pszArgStart - pszCopyToEnvStart] = 0;
}
opt.runCommand.Set(SkipNonPrintable(cmdLineRest));
opt.isScript = szArg.OneOfSwitches(L"-runlist", L"-cmdlist");
break;
}
else
{
// Show error on unknown switch
psUnknown = pszArgStart;
break;
}
}
// Main processing cycle end
// Avoid assertions in NextArg
szArg.Empty(); szNext.Empty();
} // while (NextArg(&cmdLineRest, szArg, &pszArgStart) == 0)
}
// The function exists in both "ConEmuC/ConEmuC.cpp" and "ConEmuCD/Actions.cpp"
// Version in "ConEmuC/ConEmuC.cpp" shows arguments from main(int argc, char** argv)
// Version in "ConEmuCD/Actions.cpp" perhaps would not be ever called
int DoParseArgs(LPCWSTR asCmdLine)
{
char szLine[80];
_wsprintfA(szLine, SKIPLEN(countof(szLine)) "main arguments (%i)\n", gn_argc);
_printf(szLine);
for (int j = 0; j < gn_argc; j++)
{
_wsprintfA(szLine, SKIPLEN(countof(szLine)) " %u: ", j);
_printf(szLine);
if (!gp_argv)
{
_printf("*NULL");
}
else if (!gp_argv[j])
{
_printf("<NULL>");
}
else
{
_printf("`");
_printf(gp_argv[j]);
_printf("`");
}
_printf("\n");
}
_printf("Parsing command\n `");
_wprintf(asCmdLine);
_printf("`\n");
int iShellCount = 0;
LPWSTR* ppszShl = CommandLineToArgvW(asCmdLine, &iShellCount);
int i = 0;
CEStr szArg;
_printf("ConEmu `NextArg` splitter\n");
while (NextArg(&asCmdLine, szArg) == 0)
{
if (szArg.mb_Quoted)
DemangleArg(szArg, true);
_wsprintfA(szLine, SKIPLEN(countof(szLine)) " %u: `", ++i);
_printf(szLine);
_wprintf(szArg);
_printf("`\n");
}
_wsprintfA(szLine, SKIPLEN(countof(szLine)) " Total arguments parsed: %u\n", i);
_printf(szLine);
_printf("Standard shell splitter\n");
for (int j = 0; j < iShellCount; j++)
{
_wsprintfA(szLine, SKIPLEN(countof(szLine)) " %u: `", j);
_printf(szLine);
_wprintf(ppszShl[j]);
_printf("`\n");
}
_wsprintfA(szLine, SKIPLEN(countof(szLine)) " Total arguments parsed: %u\n", iShellCount);
_printf(szLine);
LocalFree(ppszShl);
return i;
}
int main_HCopy(int argc, char *argv[])
{
char *s; /* next file to process */
void OpenSpeechFile(char *s);
void AppendSpeechFile(char *s);
void PutTargetFile(char *s);
if(InitShell(argc,argv,hcopy_version,hcopy_vc_id)<SUCCESS)
HError(1000,"HCopy: InitShell failed");
InitMem(); InitLabel();
InitMath(); InitSigP();
InitWave(); InitAudio();
InitVQ(); InitModel();
if(InitParm()<SUCCESS)
HError(1000,"HCopy: InitParm failed");
if (!InfoPrinted() && NumArgs() == 0)
ReportUsageHCopy();
if (NumArgs() == 0) return(0);
SetConfParmsHCopy();
/* initial trace string is null */
trList.str = NULL;
CreateHeap(&iStack, "InBuf", MSTAK, 1, 0.0, STACKSIZE, LONG_MAX);
CreateHeap(&oStack, "OutBuf", MSTAK, 1, 0.0, STACKSIZE, LONG_MAX);
CreateHeap(&cStack, "ChopBuf", MSTAK, 1, 0.0, STACKSIZE, LONG_MAX);
CreateHeap(&lStack, "LabBuf", MSTAK, 1, 0.0, 10000, LONG_MAX);
CreateHeap(&tStack, "Trace", MSTAK, 1, 0.0, 100, 200);
while (NextArg() == SWITCHARG) {
s = GetSwtArg();
if (strlen(s)!=1)
HError(1019,"HCopy: Bad switch %s; must be single letter",s);
switch(s[0]){
case 'a':
if (NextArg() != INTARG)
HError(1019,"HCopy: Auxiliary label index expected");
auxLab = GetChkedInt(1,100000,s) - 1;
break;
case 'e': /* end time in seconds, max 10e5 secs */
en = GetChkedFlt(-MAXTIME,MAXTIME,s);
stenSet = TRUE; chopF = TRUE;
break;
case 'i':
if (NextArg() != STRINGARG)
HError(1019,"HCopy: Output MLF name expected");
if(SaveToMasterfile(GetStrArg())<SUCCESS)
HError(1014,"HCopy: Cannot write to MLF");
useMLF = TRUE; labF = TRUE; break;
case 'l':
if (NextArg() != STRINGARG)
HError(1019,"HCopy: Target label file directory expected");
outLabDir = GetStrArg();
labF = TRUE; break;
case 'm':
xMargin = GetChkedFlt(-MAXTIME,MAXTIME,s);
chopF = TRUE; break;
case 'n':
if (NextArg() != INTARG)
HError(1019,"HCopy: Label index expected");
labstidx= GetChkedInt(-100000,100000,s);
if (NextArg() == INTARG)
labenidx = GetChkedInt(-100000,100000,s);
chopF = TRUE; break;
case 's': /* start time in seconds */
st = GetChkedFlt(0,MAXTIME,s);
stenSet = TRUE; chopF = TRUE; break;
case 't':
if (NextArg() != INTARG)
HError(1019,"HCopy: Trace line width expected");
traceWidth= GetChkedInt(10,100000,s); break;
case 'x':
if (NextArg() != STRINGARG)
HError(1019,"HCopy: Label name expected");
labName = GetLabId(GetStrArg(),TRUE);
if (NextArg() == INTARG)
labRep = GetChkedInt(1,100000,s);
chopF = TRUE; labF = TRUE; break;
case 'F':
if (NextArg() != STRINGARG)
HError(1019,"HCopy: Source file format expected");
if((srcFF = Str2Format(GetStrArg())) == ALIEN)
HError(-1089,"HCopy: Warning ALIEN src file format set");
break;
case 'G':
if (NextArg() != STRINGARG)
HError(1019,"HCopy: Source label File format expected");
if((srcLabFF = Str2Format(GetStrArg())) == ALIEN)
HError(-1089,"HCopy: Warning ALIEN Label output file format set");
labF= TRUE; break;
case 'I':
if (NextArg() != STRINGARG)
HError(1019,"HCopy: MLF file name expected");
LoadMasterFile(GetStrArg());
labF = TRUE; break;
case 'L':
if (NextArg()!=STRINGARG)
HError(1019,"HCopy: Label file directory expected");
labDir = GetStrArg();
labF = TRUE; break;
case 'P':
if (NextArg() != STRINGARG)
HError(1019,"HCopy: Label File format expected");
if((tgtLabFF = Str2Format(GetStrArg())) == ALIEN)
HError(-1089,"HCopy: Warning ALIEN Label file format set");
labF = TRUE; break;
case 'O':
if (NextArg() != STRINGARG)
HError(1019,"HCopy: Target file format expected");
if((tgtFF = Str2Format(GetStrArg())) == ALIEN)
HError(-1089,"HCopy: Warning ALIEN target file format set");
break;
case 'T':
trace = GetChkedInt(0,16,s); break;
case 'X':
if (NextArg()!=STRINGARG)
HError(1019,"HCopy: Label file extension expected");
labExt = GetStrArg();
labF = TRUE; break;
default:
HError(1019,"HCopy: Unknown switch %s",s);
}
}
if (NumArgs() == 1)
HError(1019,"HCopy: Target file or + operator expected");
FixOptions();
while (NumArgs()>1) { /* process group S1 + S2 + ... TGT */
off = 0.0;
if (NextArg()!=STRINGARG)
HError(1019,"HCopy: Source file name expected");
s = GetStrArg();
OpenSpeechFile(s); /* Load initial file S1 */
if (NextArg()!=STRINGARG)
HError(1019,"HCopy: Target file or + operator expected");
s = GetStrArg();
while (strcmp(s,"+") == 0) { /* Append + S2 + S3 ... */
if (NextArg()!=STRINGARG)
HError(1019,"HCopy: Append file name expected");
s = GetStrArg();
AppendSpeechFile(s);
if (NextArg()!=STRINGARG)
HError(1019,"HCopy: Target file or + operator expected");
s = GetStrArg();
}
PutTargetFile(s);
if(trace & T_MEM) PrintAllHeapStats();
if(trans != NULL){
trans = NULL;
ResetHeap(&lStack);
}
ResetHeap(&iStack);
ResetHeap(&oStack);
if(chopF) ResetHeap(&cStack);
}
if(useMLF) CloseMLFSaveFile();
if (NumArgs() != 0) HError(-1019,"HCopy: Unused args ignored");
return (0); /* never reached -- make compiler happy */
}
bool ProcessCommandLine(int& iRc, HMODULE& hConEmu)
{
LPCWSTR pszCmdLine = GetCommandLineW();
// If there is '-new_console' or '-cur_console' switches...
if (IsNewConsoleArg(pszCmdLine) || IsNewConsoleArg(pszCmdLine, L"-cur_console"))
return false;
HeapInitialize();
bool bProcessed = false;
// Loop through switches to find supported
{
CEStr lsArg;
int iCount = 0;
bool bHelpRequested = false;
bool bFirst = true;
while (NextArg(&pszCmdLine, lsArg) == 0)
{
if ((lsArg.ms_Val[0] == L'-') && lsArg.ms_Val[1] && !wcspbrk(lsArg.ms_Val+1, L"\\//|.&<>^"))
{
// Seems this is to be the "switch" too
lsArg.ms_Val[0] = L'/';
}
bool bWasFirst = bFirst; bFirst = false;
if ((lsArg.ms_Val[0] != L'/') && bWasFirst)
{
LPCWSTR pszName = PointToName(lsArg.ms_Val);
if (pszName && (lstrcmpi(pszName, WIN3264TEST(L"ConEmuC.exe",L"ConEmuC64.exe")) == 0))
continue;
}
iCount++;
if ((lsArg.ms_Val[0] != L'/') && (iCount > 1))
{
// Some unknown (here) switch, goto full version
break;
}
if (lstrcmpi(lsArg, L"/Download") == 0)
{
iRc = DoDownload(pszCmdLine);
// Return '0' on download success for compatibility
if (iRc == CERR_DOWNLOAD_SUCCEEDED)
iRc = 0;
bProcessed = true;
break;
}
if ((lstrcmpi(lsArg, L"/Args") == 0) || (lstrcmpi(lsArg, L"/ParseArgs") == 0))
{
iRc = DoParseArgs(pszCmdLine);
bProcessed = true;
break;
}
if ((lstrcmpi(lsArg, L"/?") == 0)
|| (lstrcmpi(lsArg, L"/h") == 0)
|| (lstrcmpi(lsArg, L"/help") == 0)
|| (lstrcmpi(lsArg, L"/-help") == 0)
)
{
bHelpRequested = true;
break;
};
// ToDo: /IsConEmu may be processed partially?
// TODO: Inject remote and standard, DefTerm
}
if (bHelpRequested || (iCount == 0))
{
if (!hConEmu)
{
// Prefere Help from ConEmuCD.dll because ConEmuC.exe may be outdated (due to stability preference)
hConEmu = LoadLibrary(WIN3264TEST(L"ConEmuCD.dll",L"ConEmuCD64.dll"));
// Show internal Help variant if only ConEmuCD.dll was failed to load
if (hConEmu == NULL)
{
Help();
iRc = CERR_HELPREQUESTED;
bProcessed = true;
}
}
}
}
HeapDeinitialize();
return bProcessed;
}
int main(int argc, char *argv[])
{
char * labFn, *listfn, *s;
int i,fidx;
MLFEntry *me = NULL;
Transcription *t;
void InitStats(char *listfn);
void GatherStats(Transcription *t);
void OutputStats(void);
if(InitShell(argc,argv,hlstats_version,hlstats_vc_id)<SUCCESS)
HError(1300,"HLStats: InitShell failed");
InitMem(); InitMath();
InitWave(); InitLabel();
InitLM();
if (!InfoPrinted() && NumArgs() == 0)
ReportUsage();
if (NumArgs() == 0) Exit(0);
SetConfParms();
enterId=GetLabId("!ENTER",TRUE); /* All sentences should or are coerced */
exitId=GetLabId("!EXIT",TRUE); /* to start enterId and end exitId */
nullId=GetLabId("!NULL",TRUE); /* Name for words not in list */
while (NextArg() == SWITCHARG) {
s = GetSwtArg();
if (strlen(s)!=1)
HError(1319,"HLStats: Bad switch %s; must be single letter",s);
switch(s[0]){
case 'b':
doBigram = TRUE;
if (NextArg() != STRINGARG)
HError(1319,"HLStats: Ngram output file name expected");
bigFile = GetStrArg();
break;
case 'c':
doLCount = TRUE;
lCountLimit = GetChkedInt(0,100000,s);
break;
case 'd':
doDurs = TRUE; break;
case 'f':
bigFloor = GetChkedFlt(0.0,1000.0,s);
break;
case 'h':
hSize = GetChkedInt(1,2,s);
break;
case 'l':
doList = TRUE;
if (NextArg() != STRINGARG)
HError(1319,"HLStats: Output label list file name expected");
listFile = GetStrArg();
break;
case 'o':
doBOff = TRUE;
break;
case 'p':
doPCount = TRUE;
pCountLimit = GetChkedInt(0,100000,s);
break;
case 's':
if (NextArg() != STRINGARG)
HError(1319,"HLStats: ENTER label name expected");
enterId=GetLabId(GetStrArg(),TRUE);
if (NextArg() != STRINGARG)
HError(1319,"HLStats: EXIT label name expected");
exitId=GetLabId(GetStrArg(),TRUE);
break;
case 't':
bigThresh = GetChkedInt(0,100,s);
break;
case 'u':
uniFloor = GetChkedFlt(0.0,1000.0,s);
break;
case 'G':
if (NextArg() != STRINGARG)
HError(1319,"HLStats: Input label File format expected");
if((ff = Str2Format(GetStrArg())) == ALIEN)
HError(-1389,"HLStats: Warning ALIEN Label file format set");
break;
case 'I':
if (NextArg() != STRINGARG)
HError(1319,"HLStats: Input MLF file name expected");
LoadMasterFile(GetStrArg());
break;
case 'T':
if (NextArg() != INTARG)
HError(1319,"HLStats: Trace value expected");
trace = GetChkedInt(0,017,s); break;
default:
HError(1319,"HLStats: Unknown switch %s",s);
}
}
if (NextArg()!=STRINGARG)
HError(1319,"HLStats: Label list file name expected");
listfn = GetStrArg();
if (!(doDurs || doBigram || doList || doLCount || doPCount))
HError(1330,"HLStats: Nothing to do!");
InitStats(listfn);
i=0;
while (NumArgs()>0) {
if (NextArg()!=STRINGARG)
HError(1319,"HLStats: Input label file name expected");
labFn = GetStrArg();
if (IsMLFFile(labFn)) {
fidx = NumMLFFiles();
if ((me=GetMLFTable()) != NULL) {
while(me->next != NULL) me=me->next;
LoadMasterFile(labFn);
me=me->next;
}
else {
LoadMasterFile(labFn);
me=GetMLFTable();
}
while (me != NULL) {
if (me->type == MLF_IMMEDIATE && me->def.immed.fidx == fidx) {
if (trace&T_FIL) {
printf(" Processing file %s\n",me->pattern); fflush(stdout);
}
t = LOpen(&tmpHeap,me->pattern,ff);
if (t->numLists<1)
HError(-1330,"HLStats: Empty file %s",me->pattern);
else
GatherStats(t),i++;
Dispose(&tmpHeap,t);
}
me = me->next;
if ((trace&T_BAS) && !(trace&T_FIL) &&
NumMLFEntries()>5000 && i%1000==0)
printf(". "),fflush(stdout);
}
if ((trace&T_BAS) && !(trace&T_FIL) && NumMLFEntries()>5000)
printf("\n");
} else {
if (trace&T_FIL) {
printf(" Processing file %s\n",labFn); fflush(stdout);
}
t = LOpen(&tmpHeap,labFn,ff);
if (t->numLists<1)
HError(-1330,"HLStats: Empty file %s",me->pattern);
else
GatherStats(t),i++;
Dispose(&tmpHeap,t);
}
}
if (trace&T_MEM)
PrintAllHeapStats();
OutputStats();
if (trace&T_MEM)
PrintAllHeapStats();
Exit(0);
return (0); /* never reached -- make compiler happy */
}
/* SetConfParms: set conf parms relevant to HCompV */
void SetConfParms(void)
{
int i;
Boolean b;
double f;
char buf[MAXSTRLEN];
nParm = GetConfig("HEREST", TRUE, cParm, MAXGLOBS);
if (nParm>0) {
if (GetConfInt(cParm,nParm,"TRACE",&i)) trace = i;
if (GetConfFlt(cParm,nParm,"VARFLOORPERCENTILE",&f)) varFloorPercent = f;
if (GetConfBool(cParm,nParm,"SAVEBINARY",&b)) saveBinary = b;
if (GetConfBool(cParm,nParm,"BINARYACCFORMAT",&b)) ldBinary = b;
/* 2-model reestimation alignment model set */
if (GetConfStr(cParm,nParm,"ALIGNMODELMMF",buf)) {
strcpy(al_hmmMMF,buf); al_hmmUsed = TRUE;
}
if (GetConfStr(cParm,nParm,"ALIGNHMMLIST",buf)) {
strcpy(al_hmmLst,buf); al_hmmUsed = TRUE;
}
/* allow multiple individual model files */
if (GetConfStr(cParm,nParm,"ALIGNMODELDIR",buf)) {
strcpy(al_hmmDir,buf); al_hmmUsed = TRUE;
}
if (GetConfStr(cParm,nParm,"ALIGNMODELEXT",buf)) {
strcpy(al_hmmExt,buf); al_hmmUsed = TRUE;
}
if (GetConfStr(cParm,nParm,"ALIGNXFORMEXT",buf)) {
xfInfo.alXFormExt = CopyString(&hmmStack,buf);
}
if (GetConfStr(cParm,nParm,"ALIGNXFORMDIR",buf)) {
xfInfo.alXFormDir = CopyString(&hmmStack,buf);
}
if (GetConfStr(cParm,nParm,"INXFORMMASK",buf)) {
xfInfo.inSpkrPat = CopyString(&hmmStack,buf);
}
if (GetConfStr(cParm,nParm,"PAXFORMMASK",buf)) {
xfInfo.paSpkrPat = CopyString(&hmmStack,buf);
}
if (GetConfStr(cParm,nParm,"LABFILEMASK",buf)) {
labFileMask = (char*)malloc(strlen(buf)+1);
strcpy(labFileMask, buf);
}
if (GetConfStr(cParm,nParm,"UPDATEMODE",buf)) {
if (!strcmp (buf, "DUMP")) updateMode = UPMODE_DUMP;
else if (!strcmp (buf, "UPDATE")) updateMode = UPMODE_UPDATE;
else if (!strcmp (buf, "BOTH")) updateMode = UPMODE_BOTH;
else HError(2319, "Unknown UPDATEMODE specified (must be DUMP, UPDATE or BOTH)");
}
}
}
void ReportUsage(void)
{
printf("\nUSAGE: HERest [options] hmmList dataFiles...\n\n");
printf(" Option Default\n\n");
printf(" -a Use an input linear transform off\n");
printf(" -c f Mixture pruning threshold 10.0\n");
printf(" -d s dir to find hmm definitions current\n");
printf(" -h s set output speaker name pattern *.%%%%%%\n");
printf(" to s, optionally set input and parent patterns\n");
printf(" -l N set max files per speaker off\n");
printf(" -m N set min examples needed per model 3\n");
printf(" -o s extension for new hmm files as src\n");
printf(" -p N set parallel mode to N off\n");
printf(" -r Enable Single Pass Training... \n");
printf(" ...using two parameterisations off\n");
printf(" -s s print statistics to file s off\n");
printf(" -t f [i l] set pruning to f [inc limit] inf\n");
printf(" -u tmvwap update t)rans m)eans v)ars w)ghts tmvw\n");
printf(" a)daptation xform p)rior used \n");
printf(" s)semi-tied xform \n");
printf(" -v f set minimum variance to f 0.0\n");
printf(" -w f set mix weight floor to f*MINMIX 0.0\n");
printf(" -x s extension for hmm files none\n");
printf(" -z s Save all xforms to TMF file s TMF\n");
PrintStdOpts("BEFGHIJKLMSTX");
printf("\n\n");
}
void SetuFlags(void)
{
char *s;
s=GetStrArg();
uFlags=(UPDSet) 0;
while (*s != '\0')
switch (*s++) {
case 't': uFlags = (UPDSet) (uFlags+UPTRANS); break;
case 'm': uFlags = (UPDSet) (uFlags+UPMEANS); break;
case 'v': uFlags = (UPDSet) (uFlags+UPVARS); break;
case 'w': uFlags = (UPDSet) (uFlags+UPMIXES); break;
case 's': uFlags = (UPDSet) (uFlags+UPSEMIT); break;
case 'a': uFlags = (UPDSet) (uFlags+UPXFORM); break;
case 'p': uFlags = (UPDSet) (uFlags+UPMAP); break;
default: HError(2320,"SetuFlags: Unknown update flag %c",*s);
break;
}
}
/* ScriptWord: return next word from script */
char *ScriptWord(FILE *script, char *scriptBuf)
{
int ch,qch,i;
i=0; ch=' ';
while (isspace(ch)) ch = fgetc(script);
if (ch==EOF) {
scriptBuf=NULL;
return NULL;
}
if (ch=='\'' || ch=='"'){
qch = ch;
ch = fgetc(script);
while (ch != qch && ch != EOF) {
scriptBuf[i++] = ch;
ch = fgetc(script);
}
if (ch==EOF)
HError(5051,"ScriptWord: Closing quote missing in script file");
} else {
do {
scriptBuf[i++] = ch;
ch = fgetc(script);
}while (!isspace(ch) && ch != EOF);
}
scriptBuf[i] = '\0';
return scriptBuf;
}
void CheckUpdateSetUp()
{
AdaptXForm *xf;
