diff --git a/ZeroTierUI/installdialog.h b/ZeroTierUI/installdialog.h index 498bc16ba..2b38a7758 100644 --- a/ZeroTierUI/installdialog.h +++ b/ZeroTierUI/installdialog.h @@ -38,6 +38,8 @@ #include "../node/Address.hpp" +// Right now InstallDialog is only used on Mac + namespace Ui { class InstallDialog; } diff --git a/ZeroTierUI/main.cpp b/ZeroTierUI/main.cpp index c2849edcf..84e1e1211 100644 --- a/ZeroTierUI/main.cpp +++ b/ZeroTierUI/main.cpp @@ -25,6 +25,10 @@ * LLC. Start here: http://www.zerotier.com/ */ +#include +#include +#include + #include "mainwindow.h" #include "installdialog.h" #include "licensedialog.h" @@ -33,12 +37,62 @@ #include #include #include +#include + +#include "../node/Constants.hpp" +#include "../node/Defaults.hpp" #ifdef __WINDOWS__ #include #include -#endif +#include "../windows/ZeroTierOne/ZeroTierOneService.h" +// Returns true if started or already running, false if failed or not installed +static bool startWindowsService() +{ + SERVICE_STATUS ssSvcStatus; + SC_HANDLE schSCManager = NULL; + SC_HANDLE schService = NULL; + + schSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_CONNECT); + if (schSCManager == NULL) + return false; + + schService = OpenService(schSCManager, ZT_SERVICE_NAME, SERVICE_QUERY_STATUS | SERVICE_START); + if (schService == NULL) { + CloseServiceHandle(schSCManager); + return false; + } + + int tries = 0; + bool running = true; + + for(;;) { + memset(&ssSvcStatus,0,sizeof(ssSvcStatus)); + if ((++tries > 20)||(!QueryServiceStatus(schService,&ssSvcStatus))) { + running = false; + break; + } + + if (ssSvcStatus.dwCurrentState == SERVICE_RUNNING) { + break; + } else if (ssSvcStatus.dwCurrentState == SERVICE_START_PENDING) { + Sleep(500); + continue; + } + + memset(&ssSvcStatus,0,sizeof(ssSvcStatus)); + ControlService(schService, SERVICE_CONTROL_START, &ssSvcStatus); + Sleep(500); + } + + CloseServiceHandle(schService); + CloseServiceHandle(schSCManager); + return running; +} +#endif // __WINDOWS__ + +// Globally visible settings for the app QSettings *settings = (QSettings *)0; int main(int argc, char *argv[]) @@ -46,6 +100,7 @@ int main(int argc, char *argv[]) QApplication a(argc, argv); #ifdef __WINDOWS__ + // Start up Winsock2 { WSADATA wsaData; WSAStartup(MAKEWORD(2,2),&wsaData); @@ -65,12 +120,11 @@ int main(int argc, char *argv[]) // InstallDialog is an alternative main window. It will re-launch the app // when done. InstallDialog id; + id.setStyleSheet(a.styleSheet()); id.show(); return a.exec(); } -#endif -#ifdef __APPLE__ { // Put QSettings here because this is one of the writable directories allowed // in Apple's app store sandbox specs. We might end up in app store someday. @@ -78,7 +132,7 @@ int main(int argc, char *argv[]) QDir::root().mkpath(zt1AppSupport); settings = new QSettings(zt1AppSupport + "/ui.ini",QSettings::IniFormat); } -#else +#else // on non-Apple boxen put it in the standard place using the default format settings = new QSettings("ZeroTier Networks","ZeroTier One"); #endif @@ -88,6 +142,48 @@ int main(int argc, char *argv[]) ld.exec(); } +#ifdef __WINDOWS__ + { + bool winSvcInstalled = false; + while (!startWindowsService()) { + if (winSvcInstalled) { + // Service was installed and subsequently failed to start again, so + // something is wrong! + QMessageBox::critical((QWidget *)0,"Service Not Available","Unable to locate or start ZeroTier One service. There may be a problem with the installation. Try installing from the .msi file again or e-mail contact@zerotier.com if you cannot install. (Error: service failed to start)",QMessageBox::Ok); + return 1; + } + +#ifdef _WIN64 + BOOL is64Bit = TRUE; +#else + BOOL is64Bit = FALSE; + IsWow64Process(GetCurrentProcess(),&is64Bit); +#endif + std::string exe(ZeroTier::ZT_DEFAULTS.defaultHomePath + "\\zerotier-one_"); + exe.append((is64Bit == TRUE) ? "x64.exe" : "x86.exe"); + + if (QFile::exists(exe.c_str())) { + STARTUPINFOA si; + PROCESS_INFORMATION pi; + memset(&si,0,sizeof(si)); + memset(&pi,0,sizeof(pi)); + if (CreateProcessA(NULL,const_cast ((exe + " -I").c_str()),NULL,NULL,FALSE,CREATE_NO_WINDOW|CREATE_NEW_PROCESS_GROUP,NULL,NULL,&si,&pi)) { + WaitForSingleObject(pi.hProcess,INFINITE); + CloseHandle(pi.hProcess); + CloseHandle(pi.hThread); + winSvcInstalled = true; + } + } + + if (!winSvcInstalled) { + // Service failed to install -- installation problem like missing .exe + QMessageBox::critical((QWidget *)0,"Service Not Available","Unable to locate or start ZeroTier One service. There may be a problem with the installation. Try installing from the .msi file again or e-mail contact@zerotier.com if you cannot install. (Error: service not installed)",QMessageBox::Ok); + return 1; + } + } + } +#endif + MainWindow w; w.show(); return a.exec(); diff --git a/ext/installfiles/windows/ZeroTier One.aip b/ext/installfiles/windows/ZeroTier One.aip index d03c0aaf0..e82088170 100644 --- a/ext/installfiles/windows/ZeroTier One.aip +++ b/ext/installfiles/windows/ZeroTier One.aip @@ -38,32 +38,56 @@ + - + + + + + + + + + + + + - + + + + + - + + + + + + + + + @@ -133,7 +157,6 @@ - diff --git a/main.cpp b/main.cpp index 28d8e2e86..a9c13fe43 100644 --- a/main.cpp +++ b/main.cpp @@ -382,7 +382,7 @@ static void sighandlerQuit(int sig) #ifdef __WINDOWS__ // Console signal handler routine to allow CTRL+C to work, mostly for testing -static BOOL WINAPI _handlerRoutine(DWORD dwCtrlType) +static BOOL WINAPI _winConsoleCtrlHandler(DWORD dwCtrlType) { switch(dwCtrlType) { case CTRL_C_EVENT: @@ -508,7 +508,6 @@ int main(int argc,char **argv) #ifdef __WINDOWS__ WSADATA wsaData; WSAStartup(MAKEWORD(2,2),&wsaData); - SetConsoleCtrlHandler(&_handlerRoutine,TRUE); #endif if ((strstr(argv[0],"zerotier-cli"))||(strstr(argv[0],"ZEROTIER-CLI"))) @@ -580,7 +579,7 @@ int main(int argc,char **argv) } return 0; } break; -#endif +#endif // __WINDOWS__ case 'h': case '?': default: @@ -596,7 +595,6 @@ int main(int argc,char **argv) break; } } - if ((!homeDir)||(strlen(homeDir) == 0)) homeDir = ZT_DEFAULTS.defaultHomePath.c_str(); @@ -607,6 +605,7 @@ int main(int argc,char **argv) } mkdir(homeDir,0755); // will fail if it already exists { + // Write .pid file to home folder char pidpath[4096]; Utils::snprintf(pidpath,sizeof(pidpath),"%s/zerotier-one.pid",homeDir); FILE *pf = fopen(pidpath,"w"); @@ -615,76 +614,78 @@ int main(int argc,char **argv) fclose(pf); } } -#else -#ifdef __WINDOWS__ - if (IsCurrentUserLocalAdministrator() != TRUE) { - fprintf(stderr,"%s: must be run as a local administrator."ZT_EOL_S,argv[0]); - return 1; - } -#endif #endif #ifdef __WINDOWS__ - if (!winRunFromCommandLine) { + if (winRunFromCommandLine) { + // Running in "interactive" mode (mostly for debugging) + if (IsCurrentUserLocalAdministrator() != TRUE) { + fprintf(stderr,"%s: must be run as a local administrator."ZT_EOL_S,argv[0]); + return 1; + } + SetConsoleCtrlHandler(&_winConsoleCtrlHandler,TRUE); + // continues on to ordinary command line execution code below... + } else { + // Running from service manager ZeroTierOneService zt1Service; if (CServiceBase::Run(zt1Service) == TRUE) { - // Normal termination of service process return 0; } else { fprintf(stderr,"%s: unable to start service (try -h for help)"ZT_EOL_S,argv[0]); return 1; } - } else -#endif - { - int exitCode = 0; - try { - node = new Node(homeDir,port,controlPort); - switch(node->run()) { -#ifdef __WINDOWS__ - case Node::NODE_RESTART_FOR_UPGRADE: { - const char *upgPath = node->reasonForTermination(); - if (upgPath) { - if (!ZeroTierOneService::doStartUpgrade(std::string(upgPath))) { - exitCode = 3; - fprintf(stderr,"%s: abnormal termination: unable to execute update at %s (doStartUpgrade failed)\n",argv[0],(upgPath) ? upgPath : "(unknown path)"); - } - } else { - exitCode = 3; - fprintf(stderr,"%s: abnormal termination: unable to execute update at %s (no upgrade path provided)\n",argv[0],(upgPath) ? upgPath : "(unknown path)"); - } - } break; -#else // __UNIX_LIKE__ - case Node::NODE_RESTART_FOR_UPGRADE: { - const char *upgPath = node->reasonForTermination(); - // On Unix-type OSes we exec() right into the upgrade. This in turn will - // end with us being re-launched either via the upgrade itself or something - // like OSX's launchd. - if (upgPath) { - Utils::rm((std::string(homeDir)+"/zerotier-one.pid").c_str()); - ::execl(upgPath,upgPath,(char *)0); - } - exitCode = 3; - fprintf(stderr,"%s: abnormal termination: unable to execute update at %s\n",argv[0],(upgPath) ? upgPath : "(unknown path)"); - } break; -#endif - case Node::NODE_UNRECOVERABLE_ERROR: { - exitCode = 3; - const char *termReason = node->reasonForTermination(); - fprintf(stderr,"%s: abnormal termination: %s\n",argv[0],(termReason) ? termReason : "(unknown reason)"); - } break; - default: - break; - } - delete node; - node = (Node *)0; - } catch ( ... ) { - fprintf(stderr,"%s: unexpected exception!"ZT_EOL_S,argv[0]); - exitCode = 3; - } -#ifdef __UNIX_LIKE__ - Utils::rm((std::string(homeDir)+"/zerotier-one.pid").c_str()); -#endif - return exitCode; } +#endif + + int exitCode = 0; + try { + node = new Node(homeDir,port,controlPort); + switch(node->run()) { +#ifdef __WINDOWS__ + case Node::NODE_RESTART_FOR_UPGRADE: { + const char *upgPath = node->reasonForTermination(); + if (upgPath) { + if (!ZeroTierOneService::doStartUpgrade(std::string(upgPath))) { + exitCode = 3; + fprintf(stderr,"%s: abnormal termination: unable to execute update at %s (doStartUpgrade failed)\n",argv[0],(upgPath) ? upgPath : "(unknown path)"); + } + } else { + exitCode = 3; + fprintf(stderr,"%s: abnormal termination: unable to execute update at %s (no upgrade path provided)\n",argv[0],(upgPath) ? upgPath : "(unknown path)"); + } + } break; +#else // __UNIX_LIKE__ + case Node::NODE_RESTART_FOR_UPGRADE: { + const char *upgPath = node->reasonForTermination(); + // On Unix-type OSes we exec() right into the upgrade. This in turn will + // end with us being re-launched either via the upgrade itself or something + // like OSX's launchd. + if (upgPath) { + Utils::rm((std::string(homeDir)+"/zerotier-one.pid").c_str()); + ::execl(upgPath,upgPath,(char *)0); + } + exitCode = 3; + fprintf(stderr,"%s: abnormal termination: unable to execute update at %s\n",argv[0],(upgPath) ? upgPath : "(unknown path)"); + } break; +#endif + case Node::NODE_UNRECOVERABLE_ERROR: { + exitCode = 3; + const char *termReason = node->reasonForTermination(); + fprintf(stderr,"%s: abnormal termination: %s\n",argv[0],(termReason) ? termReason : "(unknown reason)"); + } break; + default: + break; + } + delete node; + node = (Node *)0; + } catch ( ... ) { + fprintf(stderr,"%s: unexpected exception!"ZT_EOL_S,argv[0]); + exitCode = 3; + } + +#ifdef __UNIX_LIKE__ + Utils::rm((std::string(homeDir)+"/zerotier-one.pid").c_str()); +#endif + + return exitCode; } diff --git a/windows/ZeroTierOne/ServiceInstaller.cpp b/windows/ZeroTierOne/ServiceInstaller.cpp index 5945ecdff..d302d9f62 100644 --- a/windows/ZeroTierOne/ServiceInstaller.cpp +++ b/windows/ZeroTierOne/ServiceInstaller.cpp @@ -50,16 +50,19 @@ std::string InstallService(PSTR pszServiceName, PSTR pszPassword) { std::string ret; - char szPath[MAX_PATH]; + char szPathTmp[MAX_PATH],szPath[MAX_PATH]; SC_HANDLE schSCManager = NULL; SC_HANDLE schService = NULL; - if (GetModuleFileName(NULL, szPath, ARRAYSIZE(szPath)) == 0) + if (GetModuleFileName(NULL, szPathTmp, ARRAYSIZE(szPath)) == 0) { ret = "GetModuleFileName failed, unable to get path to self"; goto Cleanup; } + // Quote path in case it contains spaces + _snprintf_s(szPath,sizeof(szPath),"\"%s\"",szPathTmp); + // Open the local default service control manager database schSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_CONNECT | SC_MANAGER_CREATE_SERVICE); diff --git a/windows/ZeroTierOne/ZeroTierOneService.cpp b/windows/ZeroTierOne/ZeroTierOneService.cpp index daff0bb0a..364e93cb4 100644 --- a/windows/ZeroTierOne/ZeroTierOneService.cpp +++ b/windows/ZeroTierOne/ZeroTierOneService.cpp @@ -36,19 +36,44 @@ #include "../../node/Utils.hpp" #pragma endregion +#ifdef ZT_DEBUG_SERVICE +FILE *SVCDBGfile = (FILE *)0; +ZeroTier::Mutex SVCDBGfile_m; +#endif + ZeroTierOneService::ZeroTierOneService() : CServiceBase(ZT_SERVICE_NAME,TRUE,TRUE,FALSE), _node((ZeroTier::Node *)0) { +#ifdef ZT_DEBUG_SERVICE + SVCDBGfile_m.lock(); + if (!SVCDBGfile) + SVCDBGfile = fopen(ZT_DEBUG_SERVICE,"a"); + SVCDBGfile_m.unlock(); +#endif + + ZT_SVCDBG("ZeroTierOneService::ZeroTierOneService()\r\n"); } ZeroTierOneService::~ZeroTierOneService(void) { + ZT_SVCDBG("ZeroTierOneService::~ZeroTierOneService()\r\n"); + +#ifdef ZT_DEBUG_SERVICE + SVCDBGfile_m.lock(); + if (SVCDBGfile) { + fclose(SVCDBGfile); + SVCDBGfile = (FILE *)0; + } + SVCDBGfile_m.unlock(); +#endif } void ZeroTierOneService::threadMain() throw() { + ZT_SVCDBG("ZeroTierOneService::threadMain()\r\n"); + restart_node: try { { @@ -144,8 +169,8 @@ bool ZeroTierOneService::doStartUpgrade(const std::string &msiPath) void ZeroTierOneService::OnStart(DWORD dwArgc, LPSTR *lpszArgv) { - if (_node) - return; // sanity check + ZT_SVCDBG("ZeroTierOneService::OnStart()\r\n"); + try { _thread = ZeroTier::Thread::start(this); } catch ( ... ) { @@ -155,6 +180,8 @@ void ZeroTierOneService::OnStart(DWORD dwArgc, LPSTR *lpszArgv) void ZeroTierOneService::OnStop() { + ZT_SVCDBG("ZeroTierOneService::OnStop()\r\n"); + _lock.lock(); ZeroTier::Node *n = _node; _lock.unlock(); @@ -166,6 +193,8 @@ void ZeroTierOneService::OnStop() void ZeroTierOneService::OnShutdown() { + ZT_SVCDBG("ZeroTierOneService::OnShutdown()\r\n"); + // stop thread on system shutdown (if it hasn't happened already) OnStop(); } diff --git a/windows/ZeroTierOne/ZeroTierOneService.h b/windows/ZeroTierOne/ZeroTierOneService.h index 042a398a1..2f6f733d5 100644 --- a/windows/ZeroTierOne/ZeroTierOneService.h +++ b/windows/ZeroTierOne/ZeroTierOneService.h @@ -27,6 +27,8 @@ #pragma once +#include + #include "ServiceBase.h" #include @@ -37,11 +39,23 @@ #include "../../node/Mutex.hpp" #include "../../node/Utils.hpp" +// Uncomment to make debugging Windows services suck slightly less hard. +//#define ZT_DEBUG_SERVICE "C:\\ZeroTierOneServiceDebugLog.txt" + +#ifdef ZT_DEBUG_SERVICE +extern FILE *SVCDBGfile; +extern ZeroTier::Mutex SVCDBGfile_m; +#define ZT_SVCDBG(f,...) { SVCDBGfile_m.lock(); fprintf(SVCDBGfile,f,##__VA_ARGS__); fflush(SVCDBGfile); SVCDBGfile_m.unlock(); } +#else +#define ZT_SVCDBG(f,...) {} +#endif + #define ZT_SERVICE_NAME "ZeroTierOneService" #define ZT_SERVICE_DISPLAY_NAME "ZeroTier One" #define ZT_SERVICE_START_TYPE SERVICE_AUTO_START #define ZT_SERVICE_DEPENDENCIES "" -#define ZT_SERVICE_ACCOUNT "NT AUTHORITY\\LocalService" +//#define ZT_SERVICE_ACCOUNT "NT AUTHORITY\\LocalService" +#define ZT_SERVICE_ACCOUNT NULL #define ZT_SERVICE_PASSWORD NULL class ZeroTierOneService : public CServiceBase