/*
*
* Function: main
* Description: 应用数据中心主函数开始
* 参数:
* xmlconfig - 管理程序对应的配置文件
* Return:
* Other:
*/
int main( int argc, char *argv[] )
{
if( argc != 2 )
{
printf( "useage : %s config_xml \n", argv[0] );
return -1;
}
if( fork() > 0 )
{
return 0;
}
CommandServer commandServer;
if( commandServer.loadConfig( argv[1] ) == -1 )
{
printf( "load config [%s] error\n", argv[1] );
return -1;
}
commandServer.run();
return 0;
}
int
SessionServer::finish(std::string cmdDescription,
CmdType cmdType,
CmdStatus cmdStatus,
std::string newVishnuObjectID, bool checkSession) {
//To save the date of the last connection
if (checkSession) {
check();
}
saveConnection();
//To save the command
CommandServer commandServer = CommandServer(cmdDescription, *this);
commandServer.record(cmdType, cmdStatus, newVishnuObjectID);
return 0;
}
int main(int argc, char *argv[])
{
QApplication app(argc, argv);
app.setQuitOnLastWindowClosed(false);
app.setApplicationName("electric-webview");
app.setApplicationVersion("1.0");
QCommandLineParser cmdParser;
cmdParser.setSingleDashWordOptionMode(QCommandLineParser::ParseAsCompactedShortOptions);
cmdParser.setApplicationDescription("Electric WebView is a scriptable WebView for developers.");
cmdParser.addHelpOption();
cmdParser.addVersionOption();
cmdParser.addOption(QCommandLineOption(QStringList() << "t" << "transport", "Command Transport Layer to use.", "tcp|unixsocket|websocket"));
cmdParser.addOption(QCommandLineOption(QStringList() << "r" << "reverse", "Enable reverse mode. The ID is used to identify your session in the server.", "ID"));
cmdParser.addOption(QCommandLineOption(QStringList() << "s" << "script", "Script to run.", "path"));
cmdParser.process(app);
if (cmdParser.value("transport").isEmpty()) {
qDebug().noquote() << "You must provide a command transport layer";
return -1;
}
CommandServer *commandServer = new CommandServer();
commandServer->setTransport(cmdParser.value("transport"));
commandServer->setReverse(cmdParser.isSet("reverse"));
commandServer->setReverseId(cmdParser.value("reverse"));
commandServer->initialize();
QWebEngineView *webView = new QWebEngineView;
webView->setPage(new WebPage);
ElectricWebView::instance()->setWebView(webView);
ElectricWebView::instance()->initialize();
QObject::connect(commandServer, &CommandServer::newCommand, ElectricWebView::instance()->commandHandler(), &CommandHandler::processCommand);
if (cmdParser.isSet("script"))
ElectricWebView::instance()->runScript(cmdParser.value("transport"), cmdParser.value("script"));
return app.exec();
}
/**
* \brief Function to close the session
* \return raises an exception on error
*/
int
SessionServer::close() {
std::string closePolicyStr = "";
UserServer user = UserServer(SessionServer(msession.getSessionKey()));
CommandServer commanderServer = CommandServer(SessionServer(msession.getSessionKey()));
//The init function initializes login and password using the sessionKey
user.init();
//If The user exist
if (user.exist()) {
int state = getState();
//if the session is not already closed
if (state != 0) {
//if no running commands
if (!commanderServer.isRunning()) {
mdatabaseVishnu->process((boost::format("UPDATE vsession"
" SET state=0"
" WHERE sessionkey='%1%';")%mdatabaseVishnu->escapeData(msession.getSessionKey())).str());
mdatabaseVishnu->process((boost::format("UPDATE vsession"
" SET closure=CURRENT_TIMESTAMP"
" WHERE sessionkey='%1%';")%mdatabaseVishnu->escapeData(msession.getSessionKey())).str());
} else {
//To get the close policy associated to the session
closePolicyStr = (boost::format(" WHERE sessionkey='%1%';")%mdatabaseVishnu->escapeData(msession.getSessionKey())).str();
getAttribut(closePolicyStr, "closepolicy");
//If the session close policy is CLOSE_ON_DISCONNECT
if (convertToInt(closePolicyStr) == 2) {
disconnetToTimeout(user);
} else {
throw UMSVishnuException (ERRCODE_COMMAND_RUNNING);
}
}
} else {
UMSVishnuException e (ERRCODE_SESSIONKEY_EXPIRED);
throw e;
}
} //END If The user exist
return 0;
}//END: close()
/**
* \brief Function to close the session
* \return raises an exception on error
*/
int
SessionServer::close() {
UserServer user = UserServer(SessionServer(msession.getSessionKey()));
CommandServer commanderServer = CommandServer(SessionServer(msession.getSessionKey()));
// initialize and check the user
user.init();
if (! user.exist()) {
throw UMSVishnuException (ERRCODE_UNKNOWN_USER, user.getData().getUserId());
}
if (getState() == vishnu::SESSION_CLOSED) {
throw UMSVishnuException (ERRCODE_SESSIONKEY_EXPIRED);
}
if (! commanderServer.isRunning()) {
mdatabase->process(boost::str(boost::format("UPDATE vsession"
" SET state=%1%"
" WHERE sessionkey='%2%';")
% vishnu::SESSION_CLOSED
% mdatabase->escapeData(msession.getSessionKey())));
mdatabase->process(boost::str(boost::format("UPDATE vsession"
" SET closure=CURRENT_TIMESTAMP"
" WHERE sessionkey='%1%';")
% mdatabase->escapeData(msession.getSessionKey())));
} else {
int closePolicy = vishnu::convertToInt(getAttributFromSessionKey(msession.getSessionKey(), "closepolicy"));
if (closePolicy == vishnu::CLOSE_ON_DISCONNECT) {
disconnetToTimeout();
} else {
throw UMSVishnuException (ERRCODE_COMMAND_RUNNING);
}
}
return 0;
}
/*!
* Main body - creates two threads - one for window and and one
* for images acquisition/processing.
*/
int main(int argc, char* argv[])
{
// DisCODe config filename.
std::string config_name;
// Task config filename.
std::string task_name;
// Task settings overrides
std::vector <std::string> task_overrides;
// Log level
int log_lvl = 2;
void (*prev_fn)(int);
prev_fn = signal (SIGINT, terminate);
if (prev_fn==SIG_IGN) signal (SIGINT,SIG_IGN);
// =========================================================================
// === Program command-line options
// =========================================================================
// Declare the supported options.
po::options_description desc("Allowed options");
desc.add_options()
("help", "produce help message")
("config,C", po::value<std::string>(&config_name)->default_value("config.xml"), "choose config file")
("create-config,D", "create default configuration file")
("task,T", po::value<std::string>(&task_name), "choose task file")
("create-task", "create default task file")
("log-level,L", po::value<int>(&log_lvl)->default_value(3), "set log severity level")
("unstoppable","MWAHAHAHA!")
("set,S",po::value< vector<string> >(&task_overrides),"override task settings")
("interactive,I", "interactive mode")
;
po::variables_map vm;
try {
po::store(po::parse_command_line(argc, argv, desc), vm);
po::notify(vm);
}
catch (const po::error & u) {
std::cout << u.what() << "\n";
return 0;
}
if (vm.count("help")) {
std::cout << desc << "\n";
return 0;
}
// set logger severity level
LOGGER.setLevel((Utils::Logger::Severity)log_lvl);
LOGGER.addOutput(new Utils::Logger::ConsoleOutput, (Utils::Logger::Severity)log_lvl);
if (vm.count("create-task")) {
cout << "Creating task file " << task_name << "\n";
std::ofstream cfg(task_name.c_str());
cfg << "<?xml version=\"1.0\" encoding=\"utf-8\"?>\n"
"<Task>\n"
"\t<!-- List of components -->\n"
"\t<Components>\n"
"\t</Components>\n"
"\n"
"\t<!-- Threads of execution -->\n"
"\t<Executors>\n"
"\t\t<Thread1 type=\"passive\">\n"
"\t\t</Thread1>\n"
"\t</Executors>\n"
"\n"
"\t<!-- Event connections -->\n"
"\t<Events>\n"
"\t</Events>\n"
"\n"
"\t<!-- Data stream assignments -->\n"
"\t<DataStreams>\n"
"\t</DataStreams>\n"
"</Task>\n";
cfg.close();
return 0;
}
if (vm.count("create-config")) {
cout << "Creating configuration file " << config_name << "\n";
std::ofstream cfg(config_name.c_str());
cfg << "<?xml version=\"1.0\" encoding=\"utf-8\"?>\n"
"<DisCODe>\n"
"\t<task>blob.xml</task>"
"</DisCODe>\n";
cfg.close();
return 0;
}
if (vm.count("unstoppable")) {
unstoppable = true;
}
// =========================================================================
// === DisCODe configuration
// =========================================================================
config_name = vm["config"].as<std::string>();
pt::ptree conf;
try {
read_xml(config_name, conf);
}
catch(xml_parser_error&) {
LOG(LWARNING) << "Configuration file " << config_name + " not found.";
LOG(LNOTICE) << "Quick fixes:";
LOG(LNOTICE) << " specify config file name with -C switch";
LOG(LNOTICE) << " create default configuration using -D switch";
}
if (!vm.count("task")) {
if (!conf.get_optional<std::string>("DisCODe.task")) {
LOG(LERROR) << "No task specified!";
LOG(LNOTICE) << "Quick fixes:";
LOG(LNOTICE) << " specify task name using -T switch";
LOG(LNOTICE) << " set default task name in config file";
LOG(LNOTICE) << " run DisCODe in interactive mode (-I)";
exit(EXIT_FAILURE);
} else {
task_name = conf.get<std::string>("DisCODe.task");
LOG(LINFO) << "Task: " << task_name << " from configuration file\n";
}
} else {
LOG(LINFO) << "Task: " << task_name << " from command line\n";
}
std::vector<std::pair<std::string, std::string> > overrides;
for (size_t i = 0; i < task_overrides.size(); ++i) {
std::vector<std::string> strs;
boost::split(strs, task_overrides[i], boost::is_any_of("="));
if (strs.size() == 1) {
LOG(LWARNING) << strs[0] << "have no assigned value";
} else {
overrides.push_back(std::make_pair(strs[0], strs[1]));
}
}
// =========================================================================
// === Main program part
// =========================================================================
Configurator configurator;
ComponentManager km;
ExecutorManager em;
ConnectionManager cm;
Task task;
configurator.setExecutorManager(&em);
configurator.setComponentManager(&km);
configurator.setConnectionManager(&cm);
TaskInformer task_informer(task);
ComponentInformer component_informer(km);
SystemInformer system_informer(running);
ExecutorInformer executor_informer(em);
try {
CommandServer server;
server.addInformer(&task_informer);
server.addInformer(&component_informer);
server.addInformer(&system_informer);
server.addInformer(&executor_informer);
server.printCommands();
server.start();
km.initializeComponentsList();
task = configurator.loadConfiguration(task_name, overrides);
if (!task.start()) {
LOG(LFATAL) << "Task::start() returned false\n";
running = false;
}
while(running) {
Common::Thread::msleep(50);
}
Common::Thread::msleep(500);
task.finish();
km.release();
cm.release();
em.release();
km.deactivateComponentList();
server.stop();
}//: try
// =========================================================================
// === Exception handling
// =========================================================================
catch (Common::DisCODeException& ex) {
LOG(LFATAL) << ex.what() << "\n";
ex.printStackTrace();
exit(EXIT_FAILURE);
}
catch (exception& ex) {
LOG(LFATAL) << ex.what() << "\n";
exit(EXIT_FAILURE);
}
catch (const char * ex) {
LOG(LFATAL) << ex << "\n";
exit(EXIT_FAILURE);
}
catch (...) {
LOG(LFATAL) << "Unknown exception.\n";
exit(EXIT_FAILURE);
}//: catch
}
BOOL CTortoiseProcApp::InitInstance()
{
CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": InitInstance\n"));
CheckUpgrade();
CMFCVisualManager::SetDefaultManager(RUNTIME_CLASS(CMFCVisualManagerWindows));
CMFCButton::EnableWindowsTheming();
CHistoryCombo::m_nGitIconIndex = SYS_IMAGE_LIST().AddIcon((HICON)LoadImage(AfxGetResourceHandle(), MAKEINTRESOURCE(IDI_GITCONFIG), IMAGE_ICON, 0, 0, LR_DEFAULTSIZE));
Gdiplus::GdiplusStartupInput gdiplusStartupInput;
Gdiplus::GdiplusStartup(&m_gdiplusToken,&gdiplusStartupInput,NULL);
//set the resource dll for the required language
CRegDWORD loc = CRegDWORD(_T("Software\\TortoiseGit\\LanguageID"), 1033);
long langId = loc;
{
CString langStr;
langStr.Format(_T("%ld"), langId);
CCrashReport::Instance().AddUserInfoToReport(L"LanguageID", langStr);
}
CString langDll;
CStringA langpath = CStringA(CPathUtils::GetAppParentDirectory());
langpath += "Languages";
do
{
langDll.Format(_T("%sLanguages\\TortoiseProc%ld.dll"), (LPCTSTR)CPathUtils::GetAppParentDirectory(), langId);
CString sVer = _T(STRPRODUCTVER);
CString sFileVer = CPathUtils::GetVersionFromFile(langDll);
if (sFileVer == sVer)
{
HINSTANCE hInst = LoadLibrary(langDll);
if (hInst != NULL)
{
CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": Load Language DLL %s\n"), langDll);
AfxSetResourceHandle(hInst);
break;
}
}
{
DWORD lid = SUBLANGID(langId);
lid--;
if (lid > 0)
{
langId = MAKELANGID(PRIMARYLANGID(langId), lid);
}
else
langId = 0;
}
} while (langId != 0);
TCHAR buf[6] = { 0 };
_tcscpy_s(buf, _T("en"));
langId = loc;
// MFC uses a help file with the same name as the application by default,
// which means we have to change that default to our language specific help files
CString sHelppath = CPathUtils::GetAppDirectory() + _T("TortoiseGit_en.chm");
free((void*)m_pszHelpFilePath);
m_pszHelpFilePath=_tcsdup(sHelppath);
sHelppath = CPathUtils::GetAppParentDirectory() + _T("Languages\\TortoiseGit_en.chm");
do
{
CString sLang = _T("_");
if (GetLocaleInfo(MAKELCID(langId, SORT_DEFAULT), LOCALE_SISO639LANGNAME, buf, _countof(buf)))
{
sLang += buf;
sHelppath.Replace(_T("_en"), sLang);
if (PathFileExists(sHelppath))
{
free((void*)m_pszHelpFilePath);
m_pszHelpFilePath=_tcsdup(sHelppath);
break;
}
}
sHelppath.Replace(sLang, _T("_en"));
if (GetLocaleInfo(MAKELCID(langId, SORT_DEFAULT), LOCALE_SISO3166CTRYNAME, buf, _countof(buf)))
{
sLang += _T("_");
sLang += buf;
sHelppath.Replace(_T("_en"), sLang);
if (PathFileExists(sHelppath))
{
free((void*)m_pszHelpFilePath);
m_pszHelpFilePath=_tcsdup(sHelppath);
break;
}
}
sHelppath.Replace(sLang, _T("_en"));
DWORD lid = SUBLANGID(langId);
lid--;
if (lid > 0)
{
langId = MAKELANGID(PRIMARYLANGID(langId), lid);
}
else
langId = 0;
} while (langId);
CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": Set Help Filename %s\n"), m_pszHelpFilePath);
setlocale(LC_ALL, "");
if (!g_Git.CheckMsysGitDir())
{
UINT ret = CMessageBox::Show(NULL, IDS_PROC_NOMSYSGIT, IDS_APPNAME, 3, IDI_HAND, IDS_PROC_SETMSYSGITPATH, IDS_PROC_GOTOMSYSGITWEBSITE, IDS_ABORTBUTTON);
if(ret == 2)
{
ShellExecute(NULL, NULL, _T("http://msysgit.github.io/"), NULL, NULL, SW_SHOW);
}
else if(ret == 1)
{
// open settings dialog
CSinglePropSheetDlg(CString(MAKEINTRESOURCE(IDS_PROC_SETTINGS_TITLE)), new CSetMainPage(), this->GetMainWnd()).DoModal();
}
return FALSE;
}
if (CAppUtils::GetMsysgitVersion() < 0x01070a00)
{
int ret = CMessageBox::ShowCheck(NULL, IDS_PROC_OLDMSYSGIT, IDS_APPNAME, 1, IDI_EXCLAMATION, IDS_PROC_GOTOMSYSGITWEBSITE, IDS_ABORTBUTTON, IDS_IGNOREBUTTON, _T("OldMsysgitVersionWarning"), IDS_PROC_NOTSHOWAGAINIGNORE);
if (ret == 1)
{
CMessageBox::RemoveRegistryKey(_T("OldMsysgitVersionWarning")); // only store answer if it is "Ignore"
ShellExecute(NULL, NULL, _T("http://msysgit.github.io/"), NULL, NULL, SW_SHOW);
return FALSE;
}
else if (ret == 2)
{
CMessageBox::RemoveRegistryKey(_T("OldMsysgitVersionWarning")); // only store answer if it is "Ignore"
return FALSE;
}
}
{
CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": Registering Crash Report ...\n"));
CCrashReport::Instance().AddUserInfoToReport(L"msysGitDir", CGit::ms_LastMsysGitDir);
CString versionString;
versionString.Format(_T("%d"), CGit::ms_LastMsysGitVersion);
CCrashReport::Instance().AddUserInfoToReport(L"msysGitVersion", versionString);
}
CTraceToOutputDebugString::Instance()(_T(__FUNCTION__) _T(": Initializing UI components ...\n"));
// InitCommonControls() is required on Windows XP if an application
// manifest specifies use of ComCtl32.dll version 6 or later to enable
// visual styles. Otherwise, any window creation will fail.
INITCOMMONCONTROLSEX used = {
sizeof(INITCOMMONCONTROLSEX),
ICC_ANIMATE_CLASS | ICC_BAR_CLASSES | ICC_COOL_CLASSES | ICC_DATE_CLASSES |
ICC_HOTKEY_CLASS | ICC_INTERNET_CLASSES | ICC_LISTVIEW_CLASSES |
ICC_NATIVEFNTCTL_CLASS | ICC_PAGESCROLLER_CLASS | ICC_PROGRESS_CLASS |
ICC_TAB_CLASSES | ICC_TREEVIEW_CLASSES | ICC_UPDOWN_CLASS |
ICC_USEREX_CLASSES | ICC_WIN95_CLASSES
};
InitCommonControlsEx(&used);
AfxOleInit();
AfxEnableControlContainer();
AfxInitRichEdit5();
CWinAppEx::InitInstance();
SetRegistryKey(_T("TortoiseGit"));
AfxGetApp()->m_pszProfileName = _tcsdup(_T("TortoiseProc")); // w/o this ResizableLib will store data under TortoiseGitProc which is not compatible with older versions
CCmdLineParser parser(AfxGetApp()->m_lpCmdLine);
hWndExplorer = NULL;
CString sVal = parser.GetVal(_T("hwnd"));
if (!sVal.IsEmpty())
hWndExplorer = (HWND)_wcstoui64(sVal, nullptr, 16);
while (GetParent(hWndExplorer)!=NULL)
hWndExplorer = GetParent(hWndExplorer);
if (!IsWindow(hWndExplorer))
{
hWndExplorer = NULL;
}
// if HKCU\Software\TortoiseGit\Debug is not 0, show our command line
// in a message box
if (CRegDWORD(_T("Software\\TortoiseGit\\Debug"), FALSE)==TRUE)
AfxMessageBox(AfxGetApp()->m_lpCmdLine, MB_OK | MB_ICONINFORMATION);
if (parser.HasKey(_T("urlhandler")))
{
CString url = parser.GetVal(_T("urlhandler"));
if (url.Find(_T("tgit://clone/")) == 0)
{
url = url.Mid(13); // 21 = "tgit://clone/".GetLength()
}
else if (url.Find(_T("github-windows://openRepo/")) == 0)
{
url = url.Mid(26); // 26 = "github-windows://openRepo/".GetLength()
int questioMark = url.Find('?');
if (questioMark > 0)
url = url.Left(questioMark);
}
else if (url.Find(_T("smartgit://cloneRepo/")) == 0)
{
url = url.Mid(21); // 21 = "smartgit://cloneRepo/".GetLength()
}
else
{
CMessageBox::Show(NULL, IDS_ERR_INVALIDPATH, IDS_APPNAME, MB_ICONERROR);
return FALSE;
}
CString newCmd;
newCmd.Format(_T("/command:clone /url:\"%s\" /hasurlhandler"), url);
parser = CCmdLineParser(newCmd);
}
if ( parser.HasKey(_T("path")) && parser.HasKey(_T("pathfile")))
{
CMessageBox::Show(NULL, IDS_ERR_INVALIDPATH, IDS_APPNAME, MB_ICONERROR);
return FALSE;
}
CTGitPath cmdLinePath;
CTGitPathList pathList;
if (g_sGroupingUUID.IsEmpty())
g_sGroupingUUID = parser.GetVal(L"groupuuid");
if ( parser.HasKey(_T("pathfile")) )
{
CString sPathfileArgument = CPathUtils::GetLongPathname(parser.GetVal(_T("pathfile")));
cmdLinePath.SetFromUnknown(sPathfileArgument);
if (pathList.LoadFromFile(cmdLinePath)==false)
return FALSE; // no path specified!
if ( parser.HasKey(_T("deletepathfile")) )
{
// We can delete the temporary path file, now that we've loaded it
::DeleteFile(cmdLinePath.GetWinPath());
}
// This was a path to a temporary file - it's got no meaning now, and
// anybody who uses it again is in for a problem...
cmdLinePath.Reset();
}
else
{
CString sPathArgument = CPathUtils::GetLongPathname(parser.GetVal(_T("path")));
if (parser.HasKey(_T("expaths")))
{
// an /expaths param means we're started via the buttons in our Win7 library
// and that means the value of /expaths is the current directory, and
// the selected paths are then added as additional parameters but without a key, only a value
// because of the "strange treatment of quotation marks and backslashes by CommandLineToArgvW"
// we have to escape the backslashes first. Since we're only dealing with paths here, that's
// a save bet.
// Without this, a command line like:
// /command:commit /expaths:"D:\" "D:\Utils"
// would fail because the "D:\" is treated as the backslash being the escape char for the quotation
// mark and we'd end up with:
// argv[1] = /command:commit
// argv[2] = /expaths:D:" D:\Utils
// See here for more details: http://blogs.msdn.com/b/oldnewthing/archive/2010/09/17/10063629.aspx
CString cmdLine = GetCommandLineW();
cmdLine.Replace(L"\\", L"\\\\");
int nArgs = 0;
LPWSTR *szArglist = CommandLineToArgvW(cmdLine, &nArgs);
if (szArglist)
{
// argument 0 is the process path, so start with 1
for (int i = 1; i < nArgs; ++i)
{
if (szArglist[i][0] != '/')
{
if (!sPathArgument.IsEmpty())
sPathArgument += '*';
sPathArgument += szArglist[i];
}
}
sPathArgument.Replace(L"\\\\", L"\\");
}
LocalFree(szArglist);
}
if (sPathArgument.IsEmpty() && parser.HasKey(L"path"))
{
CMessageBox::Show(hWndExplorer, IDS_ERR_INVALIDPATH, IDS_APPNAME, MB_ICONERROR);
return FALSE;
}
int asterisk = sPathArgument.Find('*');
cmdLinePath.SetFromUnknown(asterisk >= 0 ? sPathArgument.Left(asterisk) : sPathArgument);
pathList.LoadFromAsteriskSeparatedString(sPathArgument);
}
if (pathList.IsEmpty()) {
pathList.AddPath(CTGitPath::CTGitPath(g_Git.m_CurrentDir));
}
// Set CWD to temporary dir, and restore it later
{
DWORD len = GetCurrentDirectory(0, NULL);
if (len)
{
std::unique_ptr<TCHAR[]> originalCurrentDirectory(new TCHAR[len]);
if (GetCurrentDirectory(len, originalCurrentDirectory.get()))
{
sOrigCWD = originalCurrentDirectory.get();
sOrigCWD = CPathUtils::GetLongPathname(sOrigCWD);
}
}
TCHAR pathbuf[MAX_PATH] = {0};
GetTortoiseGitTempPath(MAX_PATH, pathbuf);
SetCurrentDirectory(pathbuf);
}
CheckForNewerVersion();
CAutoGeneralHandle TGitMutex = ::CreateMutex(NULL, FALSE, _T("TortoiseGitProc.exe"));
if (!g_Git.SetCurrentDir(cmdLinePath.GetWinPathString(), parser.HasKey(_T("submodule")) == TRUE))
{
for (int i = 0; i < pathList.GetCount(); ++i)
if(g_Git.SetCurrentDir(pathList[i].GetWinPath()))
break;
}
if(!g_Git.m_CurrentDir.IsEmpty())
{
sOrigCWD = g_Git.m_CurrentDir;
SetCurrentDirectory(g_Git.m_CurrentDir);
}
if (g_sGroupingUUID.IsEmpty())
{
CRegStdDWORD groupSetting = CRegStdDWORD(_T("Software\\TortoiseGit\\GroupTaskbarIconsPerRepo"), 3);
switch (DWORD(groupSetting))
{
case 1:
case 2:
// implemented differently to TortoiseSVN atm
break;
case 3:
case 4:
{
CString wcroot;
if (g_GitAdminDir.HasAdminDir(g_Git.m_CurrentDir, true, &wcroot))
{
git_oid oid;
CStringA wcRootA(wcroot + CPathUtils::GetAppDirectory());
if (!git_odb_hash(&oid, wcRootA, wcRootA.GetLength(), GIT_OBJ_BLOB))
{
CStringA hash;
git_oid_tostr(hash.GetBufferSetLength(GIT_OID_HEXSZ + 1), GIT_OID_HEXSZ + 1, &oid);
hash.ReleaseBuffer();
g_sGroupingUUID = hash;
}
ProjectProperties pp;
pp.ReadProps();
CString icon = pp.sIcon;
icon.Replace('/', '\\');
if (icon.IsEmpty())
g_bGroupingRemoveIcon = true;
g_sGroupingIcon = icon;
}
}
}
}
CString sAppID = GetTaskIDPerUUID(g_sGroupingUUID).c_str();
InitializeJumpList(sAppID);
EnsureGitLibrary(false);
{
CString err;
try
{
// requires CWD to be set
CGit::m_LogEncode = CAppUtils::GetLogOutputEncode();
// make sure all config files are read in order to check that none contains an error
g_Git.GetConfigValue(_T("doesnot.exist"));
}
catch (char* msg)
{
err = CString(msg);
}
if (!err.IsEmpty())
{
UINT choice = CMessageBox::Show(hWndExplorer, err, _T("TortoiseGit"), 1, IDI_ERROR, CString(MAKEINTRESOURCE(IDS_PROC_EDITLOCALGITCONFIG)), CString(MAKEINTRESOURCE(IDS_PROC_EDITGLOBALGITCONFIG)), CString(MAKEINTRESOURCE(IDS_ABORTBUTTON)));
if (choice == 1)
{
// open the config file with alternative editor
CAppUtils::LaunchAlternativeEditor(g_Git.GetGitLocalConfig());
}
else if (choice == 2)
{
// open the global config file with alternative editor
CAppUtils::LaunchAlternativeEditor(g_Git.GetGitGlobalConfig());
}
return FALSE;
}
}
// execute the requested command
CommandServer server;
Command * cmd = server.GetCommand(parser.GetVal(_T("command")));
if (cmd)
{
cmd->SetExplorerHwnd(hWndExplorer);
cmd->SetParser(parser);
cmd->SetPaths(pathList, cmdLinePath);
retSuccess = cmd->Execute();
delete cmd;
}
// Look for temporary files left around by TortoiseSVN and
// remove them. But only delete 'old' files because some
// apps might still be needing the recent ones.
{
DWORD len = GetTortoiseGitTempPath(0, NULL);
std::unique_ptr<TCHAR[]> path(new TCHAR[len + 100]);
len = GetTortoiseGitTempPath (len + 100, path.get());
if (len != 0)
{
CDirFileEnum finder(path.get());
FILETIME systime_;
::GetSystemTimeAsFileTime(&systime_);
__int64 systime = (((_int64)systime_.dwHighDateTime)<<32) | ((__int64)systime_.dwLowDateTime);
bool isDir;
CString filepath;
while (finder.NextFile(filepath, &isDir))
{
HANDLE hFile = ::CreateFile(filepath, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, isDir ? FILE_FLAG_BACKUP_SEMANTICS : NULL, NULL);
if (hFile != INVALID_HANDLE_VALUE)
{
FILETIME createtime_;
if (::GetFileTime(hFile, &createtime_, NULL, NULL))
{
::CloseHandle(hFile);
__int64 createtime = (((_int64)createtime_.dwHighDateTime)<<32) | ((__int64)createtime_.dwLowDateTime);
if ((createtime + 864000000000) < systime) //only delete files older than a day
{
::SetFileAttributes(filepath, FILE_ATTRIBUTE_NORMAL);
if (isDir)
::RemoveDirectory(filepath);
else
::DeleteFile(filepath);
}
}
else
::CloseHandle(hFile);
}
}
}
}
// Since the dialog has been closed, return FALSE so that we exit the
// application, rather than start the application's message pump.
return FALSE;
}
BOOL CTortoiseProcApp::InitInstance()
{
EnableCrashHandler();
InitializeJumpList();
CheckUpgrade();
CMFCVisualManager::SetDefaultManager(RUNTIME_CLASS(CMFCVisualManagerWindows));
CMFCButton::EnableWindowsTheming();
Gdiplus::GdiplusStartupInput gdiplusStartupInput;
Gdiplus::GdiplusStartup(&m_gdiplusToken,&gdiplusStartupInput,NULL);
if(!CheckMsysGitDir())
{
UINT ret = CMessageBox::Show(NULL,_T("MSysGit (http://code.google.com/p/msysgit/) not found."),
_T("TortoiseGit"), 3, IDI_HAND, _T("&Set MSysGit path"), _T("&Goto WebSite"), _T("&Abort"));
if(ret == 2)
{
ShellExecute(NULL, NULL, _T("http://code.google.com/p/msysgit/"), NULL, NULL, SW_SHOW);
}
else if(ret == 1)
{
// open settings dialog
CSettings dlg(IDS_PROC_SETTINGS_TITLE);
dlg.SetTreeViewMode(TRUE, TRUE, TRUE);
dlg.SetTreeWidth(220);
dlg.DoModal();
dlg.HandleRestart();
}
return FALSE;
}
//set the resource dll for the required language
CRegDWORD loc = CRegDWORD(_T("Software\\TortoiseGit\\LanguageID"), 1033);
long langId = loc;
CString langDll;
CStringA langpath = CStringA(CPathUtils::GetAppParentDirectory());
langpath += "Languages";
// bindtextdomain("subversion", (LPCSTR)langpath);
// bind_textdomain_codeset("subversion", "UTF-8");
HINSTANCE hInst = NULL;
do
{
langDll.Format(_T("..\\Languages\\TortoiseProc%d.dll"), langId);
hInst = LoadLibrary(langDll);
CString sVer = _T(STRPRODUCTVER);
CString sFileVer = CPathUtils::GetVersionFromFile(langDll);
if (sFileVer.Compare(sVer)!=0)
{
FreeLibrary(hInst);
hInst = NULL;
}
if (hInst != NULL)
{
AfxSetResourceHandle(hInst);
}
else
{
DWORD lid = SUBLANGID(langId);
lid--;
if (lid > 0)
{
langId = MAKELANGID(PRIMARYLANGID(langId), lid);
}
else
langId = 0;
}
} while ((hInst == NULL) && (langId != 0));
TCHAR buf[6];
_tcscpy_s(buf, _T("en"));
langId = loc;
CString sHelppath;
sHelppath = this->m_pszHelpFilePath;
sHelppath = sHelppath.MakeLower();
// MFC uses a help file with the same name as the application by default,
// which means we have to change that default to our language specific help files
sHelppath.Replace(_T("tortoiseproc.chm"), _T("TortoiseGit_en.chm"));
free((void*)m_pszHelpFilePath);
m_pszHelpFilePath=_tcsdup(sHelppath);
sHelppath = CPathUtils::GetAppParentDirectory() + _T("Languages\\TortoiseGit_en.chm");
do
{
CString sLang = _T("_");
if (GetLocaleInfo(MAKELCID(langId, SORT_DEFAULT), LOCALE_SISO639LANGNAME, buf, _countof(buf)))
{
sLang += buf;
sHelppath.Replace(_T("_en"), sLang);
if (PathFileExists(sHelppath))
{
free((void*)m_pszHelpFilePath);
m_pszHelpFilePath=_tcsdup(sHelppath);
break;
}
}
sHelppath.Replace(sLang, _T("_en"));
if (GetLocaleInfo(MAKELCID(langId, SORT_DEFAULT), LOCALE_SISO3166CTRYNAME, buf, _countof(buf)))
{
sLang += _T("_");
sLang += buf;
sHelppath.Replace(_T("_en"), sLang);
if (PathFileExists(sHelppath))
{
free((void*)m_pszHelpFilePath);
m_pszHelpFilePath=_tcsdup(sHelppath);
break;
}
}
sHelppath.Replace(sLang, _T("_en"));
DWORD lid = SUBLANGID(langId);
lid--;
if (lid > 0)
{
langId = MAKELANGID(PRIMARYLANGID(langId), lid);
}
else
langId = 0;
} while (langId);
setlocale(LC_ALL, "");
// InitCommonControls() is required on Windows XP if an application
// manifest specifies use of ComCtl32.dll version 6 or later to enable
// visual styles. Otherwise, any window creation will fail.
INITCOMMONCONTROLSEX used = {
sizeof(INITCOMMONCONTROLSEX),
ICC_ANIMATE_CLASS | ICC_BAR_CLASSES | ICC_COOL_CLASSES | ICC_DATE_CLASSES |
ICC_HOTKEY_CLASS | ICC_INTERNET_CLASSES | ICC_LISTVIEW_CLASSES |
ICC_NATIVEFNTCTL_CLASS | ICC_PAGESCROLLER_CLASS | ICC_PROGRESS_CLASS |
ICC_TAB_CLASSES | ICC_TREEVIEW_CLASSES | ICC_UPDOWN_CLASS |
ICC_USEREX_CLASSES | ICC_WIN95_CLASSES
};
InitCommonControlsEx(&used);
AfxOleInit();
AfxEnableControlContainer();
AfxInitRichEdit2();
CWinAppEx::InitInstance();
SetRegistryKey(_T("TortoiseGit"));
CCmdLineParser parser(AfxGetApp()->m_lpCmdLine);
// if HKCU\Software\TortoiseGit\Debug is not 0, show our command line
// in a message box
if (CRegDWORD(_T("Software\\TortoiseGit\\Debug"), FALSE)==TRUE)
AfxMessageBox(AfxGetApp()->m_lpCmdLine, MB_OK | MB_ICONINFORMATION);
if ( parser.HasKey(_T("path")) && parser.HasKey(_T("pathfile")))
{
CMessageBox::Show(NULL, IDS_ERR_INVALIDPATH, IDS_APPNAME, MB_ICONERROR);
return FALSE;
}
CTGitPath cmdLinePath;
CTGitPathList pathList;
if ( parser.HasKey(_T("pathfile")) )
{
CString sPathfileArgument = CPathUtils::GetLongPathname(parser.GetVal(_T("pathfile")));
cmdLinePath.SetFromUnknown(sPathfileArgument);
if (pathList.LoadFromFile(cmdLinePath)==false)
return FALSE; // no path specified!
if ( parser.HasKey(_T("deletepathfile")) )
{
// We can delete the temporary path file, now that we've loaded it
::DeleteFile(cmdLinePath.GetWinPath());
}
// This was a path to a temporary file - it's got no meaning now, and
// anybody who uses it again is in for a problem...
cmdLinePath.Reset();
}
else
{
CString sPathArgument = CPathUtils::GetLongPathname(parser.GetVal(_T("path")));
int asterisk = sPathArgument.Find('*');
cmdLinePath.SetFromUnknown(asterisk >= 0 ? sPathArgument.Left(asterisk) : sPathArgument);
pathList.LoadFromAsteriskSeparatedString(sPathArgument);
}
if (pathList.GetCount() == 0) {
pathList.AddPath(CTGitPath::CTGitPath(g_Git.m_CurrentDir));
}
hWndExplorer = NULL;
CString sVal = parser.GetVal(_T("hwnd"));
if (!sVal.IsEmpty())
hWndExplorer = (HWND)_ttoi64(sVal);
while (GetParent(hWndExplorer)!=NULL)
hWndExplorer = GetParent(hWndExplorer);
if (!IsWindow(hWndExplorer))
{
hWndExplorer = NULL;
}
// Subversion sometimes writes temp files to the current directory!
// Since TSVN doesn't need a specific CWD anyway, we just set it
// to the users temp folder: that way, Subversion is guaranteed to
// have write access to the CWD
{
DWORD len = GetCurrentDirectory(0, NULL);
if (len)
{
TCHAR * originalCurrentDirectory = new TCHAR[len];
if (GetCurrentDirectory(len, originalCurrentDirectory))
{
//sOrigCWD = originalCurrentDirectory;
//sOrigCWD = CPathUtils::GetLongPathname(sOrigCWD);
}
delete [] originalCurrentDirectory;
}
TCHAR pathbuf[MAX_PATH];
GetTempPath(MAX_PATH, pathbuf);
SetCurrentDirectory(pathbuf);
}
// check for newer versions
if (CRegDWORD(_T("Software\\TortoiseGit\\CheckNewer"), TRUE) != FALSE)
{
time_t now;
struct tm ptm;
time(&now);
if ((now != 0) && (localtime_s(&ptm, &now)==0))
{
int week = 0;
// we don't calculate the real 'week of the year' here
// because just to decide if we should check for an update
// that's not needed.
week = ptm.tm_yday / 7;
CRegDWORD oldweek = CRegDWORD(_T("Software\\TortoiseGit\\CheckNewerWeek"), (DWORD)-1);
if (((DWORD)oldweek) == -1)
oldweek = week; // first start of TortoiseProc, no update check needed
else
{
if ((DWORD)week != oldweek)
{
oldweek = week;
TCHAR com[MAX_PATH+100];
GetModuleFileName(NULL, com, MAX_PATH);
_tcscat_s(com, MAX_PATH+100, _T(" /command:updatecheck"));
CAppUtils::LaunchApplication(com, 0, false);
}
}
}
}
if (parser.HasVal(_T("configdir")))
{
// the user can override the location of the Subversion config directory here
CString sConfigDir = parser.GetVal(_T("configdir"));
// g_GitGlobal.SetConfigDir(sConfigDir);
}
// to avoid that SASL will look for and load its plugin dlls all around the
// system, we set the path here.
// Note that SASL doesn't have to be initialized yet for this to work
// sasl_set_path(SASL_PATH_TYPE_PLUGIN, (LPSTR)(LPCSTR)CUnicodeUtils::GetUTF8(CPathUtils::GetAppDirectory().TrimRight('\\')));
HANDLE TSVNMutex = ::CreateMutex(NULL, FALSE, _T("TortoiseGitProc.exe"));
if(!g_Git.SetCurrentDir(cmdLinePath.GetWinPathString()))
{
int i=0;
for(i=0;i<pathList.GetCount();i++)
if(g_Git.SetCurrentDir(pathList[i].GetWinPath()))
break;
}
if(!g_Git.m_CurrentDir.IsEmpty())
SetCurrentDirectory(g_Git.m_CurrentDir);
{
CString err;
try
{
// requires CWD to be set
CGit::m_LogEncode = CAppUtils::GetLogOutputEncode();
}
catch (char* msg)
{
err = CString(msg);
}
if (!err.IsEmpty())
{
UINT choice = CMessageBox::Show(hWndExplorer, err, _T("TortoiseGit Error"), 1, IDI_ERROR, _T("&Edit .git/config"), _T("Edit &global .gitconfig"), _T("&Abort"));
if (choice == 1)
{
// open the config file with alternative editor
CString path = g_Git.m_CurrentDir;
path += _T("\\.git\\config");
CAppUtils::LaunchAlternativeEditor(path);
}
else if (choice == 2)
{
// open the global config file with alternative editor
TCHAR buf[MAX_PATH];
ExpandEnvironmentStrings(_T("%HOMEDRIVE%\\%HOMEPATH%\\.gitconfig"), buf, MAX_PATH);
CAppUtils::LaunchAlternativeEditor(buf);
}
return FALSE;
}
}
// execute the requested command
CommandServer server;
Command * cmd = server.GetCommand(parser.GetVal(_T("command")));
if (cmd)
{
cmd->SetExplorerHwnd(hWndExplorer);
cmd->SetParser(parser);
cmd->SetPaths(pathList, cmdLinePath);
retSuccess = cmd->Execute();
delete cmd;
}
if (TSVNMutex)
::CloseHandle(TSVNMutex);
// Look for temporary files left around by TortoiseSVN and
// remove them. But only delete 'old' files because some
// apps might still be needing the recent ones.
{
DWORD len = ::GetTempPath(0, NULL);
TCHAR * path = new TCHAR[len + 100];
len = ::GetTempPath (len+100, path);
if (len != 0)
{
CSimpleFileFind finder = CSimpleFileFind(path, _T("*svn*.*"));
FILETIME systime_;
::GetSystemTimeAsFileTime(&systime_);
__int64 systime = (((_int64)systime_.dwHighDateTime)<<32) | ((__int64)systime_.dwLowDateTime);
while (finder.FindNextFileNoDirectories())
{
CString filepath = finder.GetFilePath();
HANDLE hFile = ::CreateFile(filepath, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, NULL, NULL);
if (hFile != INVALID_HANDLE_VALUE)
{
FILETIME createtime_;
if (::GetFileTime(hFile, &createtime_, NULL, NULL))
{
::CloseHandle(hFile);
__int64 createtime = (((_int64)createtime_.dwHighDateTime)<<32) | ((__int64)createtime_.dwLowDateTime);
if ((createtime + 864000000000) < systime) //only delete files older than a day
{
::SetFileAttributes(filepath, FILE_ATTRIBUTE_NORMAL);
::DeleteFile(filepath);
}
}
else
::CloseHandle(hFile);
}
}
}
delete[] path;
}
// Since the dialog has been closed, return FALSE so that we exit the
// application, rather than start the application's message pump.
return FALSE;
}
BOOL CTortoiseSIProcApp::ProcessCommandLine()
{
CCmdLineParser parser(AfxGetApp()->m_lpCmdLine);
CString sVal = parser.GetVal(_T("hwnd"));
if (!sVal.IsEmpty()) {
m_hWndExplorer = (HWND)_wcstoui64(sVal, nullptr, 16);
}
while (GetParent(m_hWndExplorer) != NULL) {
m_hWndExplorer = GetParent(m_hWndExplorer);
}
if (!IsWindow(m_hWndExplorer))
{
m_hWndExplorer = NULL;
}
#if _DEBUG
if (CRegDWORD(_T("Software\\TortoiseSI\\Debug"), FALSE) == TRUE) {
AfxMessageBox(AfxGetApp()->m_lpCmdLine, MB_OK | MB_ICONINFORMATION);
}
#endif
// The path and pathfile arguments are mutually exclusive
if (parser.HasKey(_T("path")) && parser.HasKey(_T("pathfile")))
{
CMessageBox::Show(NULL, IDS_ERR_INVALIDPATH, IDS_APPNAME, MB_ICONERROR);
return FALSE;
}
CTGitPath cmdLinePath;
CTGitPathList pathList;
if (parser.HasKey(_T("pathfile")))
{
// Process /pathfile argument
CString sPathFileArgument = CPathUtils::GetLongPathname(parser.GetVal(_T("pathfile")));
cmdLinePath.SetFromUnknown(sPathFileArgument);
if (pathList.LoadFromFile(cmdLinePath) == false) {
// no path specified!
return FALSE;
}
if (parser.HasKey(_T("deletepathfile")))
{
// We can delete the temporary path file, now that we've loaded it
::DeleteFile(cmdLinePath.GetWinPath());
}
// This was a path to a temporary file - it's got no meaning now, and
// anybody who uses it again is in for a problem...
cmdLinePath.Reset();
} else {
// Process /path and /expaths argument
// Build-uo /path value as /path:<value>*<arg1>*<arg2>...
// Where <arg1>, <arg2> etc are extracted from /expath arguments.
// Since all arguments are passed in the form of /parameter:<argument>
// We process only those arguments which are not preceded by a /parameter:
// This only includes arguments specified following /expaths (See comments below)
CString sPathArgument = CPathUtils::GetLongPathname(parser.GetVal(_T("path")));
if (parser.HasKey(_T("expaths")))
{
// An /expaths param means we are started via the buttons in our Win7 library
// and that means the value of /expaths is the current directory and
// the selected paths are then added as additional parameters but without a key, only a value
//
// e.g. /expaths:"D:\" "D:\Utils"
//
// Because of the "strange treatment of quotation marks and backslashes by CommandLineToArgvW",
// we have to escape the backslashes first. Since we're only dealing with paths here, that's
// a safe bet. Without this, a command line like:
//
// /command:commit /expaths:"D:\" "D:\Utils"
//
// would fail because the "D:\" is treated as the backslash being the escape char for the quotation
// mark and we'd end up with:
//
// argv[1] = /command:commit
// argv[2] = /expaths:D:" D:\Utils
//
// See here for more details: http://blogs.msdn.com/b/oldnewthing/archive/2010/09/17/10063629.aspx
CString cmdLine = GetCommandLineW();
// Escape backslashes
cmdLine.Replace(L"\\", L"\\\\");
int nArgs = 0;
LPWSTR *szArgList = CommandLineToArgvW(cmdLine, &nArgs);
if (szArgList)
{
// Argument 0 is the process path, so start with 1
for (int i = 1; i < nArgs; i++)
{
if (szArgList[i][0] != '/')
{
if (!sPathArgument.IsEmpty()) {
sPathArgument += '*';
}
sPathArgument += szArgList[i];
}
}
sPathArgument.Replace(L"\\\\", L"\\");
}
LocalFree(szArgList);
}
if (sPathArgument.IsEmpty() && parser.HasKey(L"path"))
{
CMessageBox::Show(m_hWndExplorer, IDS_ERR_INVALIDPATH, IDS_APPNAME, MB_ICONERROR);
return FALSE;
}
int asterisk = sPathArgument.Find('*');
cmdLinePath.SetFromUnknown(asterisk >= 0 ? sPathArgument.Left(asterisk) : sPathArgument);
pathList.LoadFromAsteriskSeparatedString(sPathArgument);
}
if (pathList.IsEmpty()) {
pathList.AddPath(CTGitPath::CTGitPath(g_Git.m_CurrentDir));
}
// Set CWD to temporary dir, and restore it later
{
DWORD len = GetCurrentDirectory(0, NULL);
if (len)
{
std::unique_ptr<TCHAR[]> originalCurrentDirectory(new TCHAR[len]);
if (GetCurrentDirectory(len, originalCurrentDirectory.get()))
{
m_sOrigCWD = originalCurrentDirectory.get();
m_sOrigCWD = CPathUtils::GetLongPathname(m_sOrigCWD);
}
}
TCHAR pathbuf[MAX_PATH] = { 0 };
GetTortoiseGitTempPath(MAX_PATH, pathbuf);
SetCurrentDirectory(pathbuf);
}
if (!g_Git.m_CurrentDir.IsEmpty())
{
m_sOrigCWD = g_Git.m_CurrentDir;
SetCurrentDirectory(m_sOrigCWD);
}
// Execute the requested command
CommandServer server;
Command *cmd = server.GetCommand(parser.GetVal(_T("command")));
if (cmd)
{
cmd->SetExplorerHwnd(m_hWndExplorer);
cmd->SetParser(parser);
cmd->SetPaths(pathList, cmdLinePath);
m_bRetSuccess = cmd->Execute();
delete cmd;
}
return TRUE;
}