Feature: Support client running as windows service

server/2015Remote/2015Remote.cpp

@@ -1,5 +1,5 @@
 
-// 2015Remote.cpp : <EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ӧ<EFBFBD>ó<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ϊ<EFBFBD><EFBFBD>
+// 2015Remote.cpp : 定义应用程序的类行为。
 //
 
 #include "stdafx.h"
@@ -10,11 +10,12 @@
 #define new DEBUG_NEW
 #endif
 
-// dump<EFBFBD><EFBFBD><EFBFBD><EFBFBD>
+// dump相关
 #include <io.h>
 #include <direct.h>
 #include <DbgHelp.h>
 #include "IOCPUDPServer.h"
+#include "ServerServiceWrapper.h"
 #pragma comment(lib, "Dbghelp.lib")
 
 CMy2015RemoteApp* GetThisApp()
@@ -35,24 +36,38 @@ std::string GetMasterHash()
 }
 
 /**
-* @brief <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>δ֪BUG<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ֹʱ<EFBFBD><EFBFBD><EFBFBD>ô˺<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
-* <EFBFBD><EFBFBD><EFBFBD><EFBFBD>ת<EFBFBD><EFBFBD>dump<EFBFBD>ļ<EFBFBD><EFBFBD><EFBFBD>dumpĿ¼.
+* @brief 程序遇到未知BUG导致终止时调用此函数，不弹框
+* 并且转储dump文件到dump目录.
 */
 long WINAPI whenbuged(_EXCEPTION_POINTERS *excp)
 {
-    // <EFBFBD><EFBFBD>ȡdump<EFBFBD>ļ<EFBFBD><EFBFBD>У<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ڣ<EFBFBD><EFBFBD>򴴽<EFBFBD>֮
-    char dump[_MAX_PATH], *p = dump;
-    GetModuleFileNameA(NULL, dump, _MAX_PATH);
-    while (*p) ++p;
-    while ('\\' != *p) --p;
-    strcpy(p + 1, "dump");
-    if (_access(dump, 0) == -1)
-        _mkdir(dump);
-    char curTime[64];// <20><>ǰdump<6D>ļ<EFBFBD>
-    time_t TIME(time(0));
-    strftime(curTime, 64, "\\YAMA_%Y-%m-%d %H%M%S.dmp", localtime(&TIME));
-    strcat(dump, curTime);
-    HANDLE hFile = ::CreateFileA(dump, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
+    // 获取dump文件夹，若不存在，则创建之
+    char dumpDir[_MAX_PATH];
+    char dumpFile[_MAX_PATH + 64];
+
+    if (!GetModuleFileNameA(NULL, dumpDir, _MAX_PATH)) {
+        return EXCEPTION_EXECUTE_HANDLER;
+    }
+
+    char* p = strrchr(dumpDir, '\\');
+    if (p) {
+        strcpy_s(p + 1, _MAX_PATH - (p - dumpDir + 1), "dump");
+    } else {
+        strcpy_s(dumpDir, _MAX_PATH, "dump");
+    }
+
+    if (_access(dumpDir, 0) == -1)
+        _mkdir(dumpDir);
+
+    // 构建完整的dump文件路径
+    char curTime[64];
+    time_t TIME = time(0);
+    struct tm localTime;
+    localtime_s(&localTime, &TIME);
+    strftime(curTime, sizeof(curTime), "\\YAMA_%Y-%m-%d %H%M%S.dmp", &localTime);
+    sprintf_s(dumpFile, sizeof(dumpFile), "%s%s", dumpDir, curTime);
+
+    HANDLE hFile = ::CreateFileA(dumpFile, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
                                  FILE_ATTRIBUTE_NORMAL, NULL);
     if(INVALID_HANDLE_VALUE != hFile) {
         MINIDUMP_EXCEPTION_INFORMATION einfo = {::GetCurrentThreadId(), excp, FALSE};
@@ -72,15 +87,15 @@ END_MESSAGE_MAP()
 
 std::string GetPwdHash();
 
-// CMy2015RemoteApp <EFBFBD><EFBFBD><EFBFBD><EFBFBD>
+// CMy2015RemoteApp 构造
 
 CMy2015RemoteApp::CMy2015RemoteApp()
 {
-    // ֧<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
+    // 支持重新启动管理器
     m_dwRestartManagerSupportFlags = AFX_RESTART_MANAGER_SUPPORT_RESTART;
 
-    // TODO: <EFBFBD>ڴ˴<EFBFBD><EFBFBD><EFBFBD><EFBFBD>ӹ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>룬
-    // <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ҫ<EFBFBD>ij<EFBFBD>ʼ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> InitInstance <EFBFBD><EFBFBD>
+    // TODO: 在此处添加构造代码，
+    // 将所有重要的初始化放置在 InitInstance 中
     m_Mutex = NULL;
 #ifdef _DEBUG
     std::string masterHash(GetMasterHash());
@@ -93,15 +108,126 @@ CMy2015RemoteApp::CMy2015RemoteApp()
 }
 
 
-// Ψһ<EFBFBD><EFBFBD>һ<EFBFBD><EFBFBD> CMy2015RemoteApp <EFBFBD><EFBFBD><EFBFBD><EFBFBD>
+// 唯一的一个 CMy2015RemoteApp 对象
 
 CMy2015RemoteApp theApp;
 
 
-// CMy2015RemoteApp <20><>ʼ<EFBFBD><CABC>
+// 从服务路径中提取可执行文件路径（去除引号和参数）
+static void ExtractExePathFromServicePath(const char* servicePath, char* exePath, size_t exePathSize)
+{
+    if (!servicePath || !exePath || exePathSize == 0) {
+        if (exePath && exePathSize > 0) exePath[0] = '\0';
+        return;
+    }
+
+    const char* src = servicePath;
+    char* dst = exePath;
+    size_t remaining = exePathSize - 1;
+
+    // 跳过前导空格
+    while (*src == ' ') src++;
+
+    if (*src == '"') {
+        // 带引号的路径：提取引号内的内容
+        src++;  // 跳过开始引号
+        while (*src && *src != '"' && remaining > 0) {
+            *dst++ = *src++;
+            remaining--;
+        }
+    } else {
+        // 不带引号的路径：提取到空格或结束
+        while (*src && *src != ' ' && remaining > 0) {
+            *dst++ = *src++;
+            remaining--;
+        }
+    }
+    *dst = '\0';
+}
+
+// 处理服务相关的命令行参数
+// 返回值: TRUE 表示已处理服务命令（程序应退出），FALSE 表示继续正常启动
+static BOOL HandleServiceCommandLine()
+{
+    CString cmdLine = ::GetCommandLine();
+    cmdLine.MakeLower();
+
+    // -service: 作为服务运行
+    if (cmdLine.Find(_T("-service")) != -1) {
+        ServerService_Run();
+        return TRUE;
+    }
+
+    // -install: 安装服务
+    if (cmdLine.Find(_T("-install")) != -1) {
+        ServerService_Install();
+        return TRUE;
+    }
+
+    // -uninstall: 卸载服务
+    if (cmdLine.Find(_T("-uninstall")) != -1) {
+        ServerService_Uninstall();
+        return TRUE;
+    }
+
+    // -agent: 由服务启动的GUI代理模式
+    // 此模式下正常运行GUI，但使用不同的互斥量名称避免冲突
+    if (cmdLine.Find(_T("-agent")) != -1) {
+        // 继续正常启动GUI，但标记为代理模式
+        return FALSE;
+    }
+
+    // 无参数时，作为服务启动
+    BOOL registered = FALSE;
+    BOOL running = FALSE;
+    char servicePath[MAX_PATH] = { 0 };
+    ServerService_CheckStatus(&registered, &running, servicePath, MAX_PATH);
+    char curPath[MAX_PATH];
+    GetModuleFileNameA(NULL, curPath, MAX_PATH);
+
+    // 从服务路径中提取纯可执行文件路径（去除引号和参数）
+    char serviceExePath[MAX_PATH] = { 0 };
+    ExtractExePathFromServicePath(servicePath, serviceExePath, MAX_PATH);
+
+    if (registered && _stricmp(curPath, serviceExePath) != 0) {
+        Mprintf("ServerService Uninstall: %s\n", servicePath);
+        ServerService_Uninstall();
+        registered = FALSE;
+    }
+    if (!registered) {
+        Mprintf("ServerService Install: %s\n", curPath);
+        return ServerService_Install();
+    }
+    else if (!running) {
+        int r = ServerService_Run();
+        Mprintf("ServerService Run '%s' %s\n", curPath, r == ERROR_SUCCESS ? "succeed" : "failed");
+        if (r) {
+            r = ServerService_StartSimple();
+            Mprintf("ServerService Start '%s' %s\n", curPath, r == ERROR_SUCCESS ? "succeed" : "failed");
+            return r == ERROR_SUCCESS;
+        }
+        return FALSE;
+    }
+    return TRUE;
+}
+
+// 检查是否以代理模式运行
+static BOOL IsAgentMode()
+{
+    CString cmdLine = ::GetCommandLine();
+    cmdLine.MakeLower();
+    return cmdLine.Find(_T("-agent")) != -1;
+}
+
+// CMy2015RemoteApp 初始化
 
 BOOL CMy2015RemoteApp::InitInstance()
 {
+    // 首先处理服务命令行参数
+    if (HandleServiceCommandLine()) {
+        return FALSE;  // 服务命令已处理，退出
+    }
+
     std::string masterHash(GetMasterHash());
     std::string mu = GetPwdHash()==masterHash ? "MASTER.EXE" : "YAMA.EXE";
 #ifndef _DEBUG
@@ -110,7 +236,8 @@ BOOL CMy2015RemoteApp::InitInstance()
         if (ERROR_ALREADY_EXISTS == GetLastError()) {
             CloseHandle(m_Mutex);
             m_Mutex = NULL;
-            MessageBoxA(NULL, "һ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>س<EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ѿ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>У<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>", "<EFBFBD><EFBFBD>ʾ", MB_ICONINFORMATION);
+            MessageBoxA(NULL, "A master program is already running, please check Task Manager.", 
+                "Info", MB_ICONINFORMATION);
             return FALSE;
         }
     }
@@ -124,13 +251,13 @@ BOOL CMy2015RemoteApp::InitInstance()
     hImageList = (HIMAGELIST)SHGetFileInfo((LPCTSTR)_T(""), 0, &sfi, sizeof(SHFILEINFO), SHGFI_SMALLICON | SHGFI_SYSICONINDEX);
     m_pImageList_Small.Attach(hImageList);
 
-    // <EFBFBD><EFBFBD><EFBFBD><EFBFBD>һ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> Windows XP <20>ϵ<EFBFBD>Ӧ<EFBFBD>ó<EFBFBD><C3B3><EFBFBD><EFBFBD>嵥ָ<E5B5A5><D6B8>Ҫ
-    // ʹ<EFBFBD><EFBFBD> ComCtl32.dll <EFBFBD>汾 6 <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>߰汾<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ÿ<EFBFBD><EFBFBD>ӻ<EFBFBD><EFBFBD><EFBFBD>ʽ<EFBFBD><EFBFBD>
-    //<EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ҫ InitCommonControlsEx()<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>򣬽<EFBFBD><EFBFBD>޷<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ڡ<EFBFBD>
+    // 如果一个运行在 Windows XP 上的应用程序清单指定要
+    // 使用 ComCtl32.dll 版本 6 或更高版本来启用可视化方式，
+    //则需要 InitCommonControlsEx()。否则，将无法创建窗口。
     INITCOMMONCONTROLSEX InitCtrls;
     InitCtrls.dwSize = sizeof(InitCtrls);
-    // <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ϊ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ҫ<EFBFBD><EFBFBD>Ӧ<EFBFBD>ó<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʹ<EFBFBD>õ<EFBFBD>
-    // <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ؼ<EFBFBD><EFBFBD>ࡣ
+    // 将它设置为包括所有要在应用程序中使用的
+    // 公共控件类。
     InitCtrls.dwICC = ICC_WIN95_CLASSES;
     InitCommonControlsEx(&InitCtrls);
 
@@ -138,37 +265,37 @@ BOOL CMy2015RemoteApp::InitInstance()
 
     AfxEnableControlContainer();
 
-    // <EFBFBD><EFBFBD><EFBFBD><EFBFBD> shell <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Է<EFBFBD><EFBFBD>Ի<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
-    // <EFBFBD>κ<EFBFBD> shell <EFBFBD><EFBFBD><EFBFBD><EFBFBD>ͼ<EFBFBD>ؼ<EFBFBD><EFBFBD><EFBFBD> shell <20>б<EFBFBD><D0B1><EFBFBD>ͼ<EFBFBD>ؼ<EFBFBD><D8BC><EFBFBD>
+    // 创建 shell 管理器，以防对话框包含
+    // 任何 shell 树视图控件或 shell 列表视图控件。
     CShellManager *pShellManager = new CShellManager;
 
-    // <EFBFBD><EFBFBD>׼<EFBFBD><EFBFBD>ʼ<EFBFBD><EFBFBD>
-    // <EFBFBD><EFBFBD><EFBFBD><EFBFBD>δʹ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>Щ<EFBFBD><EFBFBD><EFBFBD>ܲ<EFBFBD>ϣ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>С
-    // <EFBFBD><EFBFBD><EFBFBD>տ<EFBFBD>ִ<EFBFBD><EFBFBD><EFBFBD>ļ<EFBFBD><EFBFBD>Ĵ<EFBFBD>С<EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ӧ<EFBFBD>Ƴ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
-    // <EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ҫ<EFBFBD><EFBFBD><EFBFBD>ض<EFBFBD><EFBFBD><EFBFBD>ʼ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
-    // <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ڴ洢<EFBFBD><EFBFBD><EFBFBD>õ<EFBFBD>ע<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
-    // TODO: Ӧ<EFBFBD>ʵ<EFBFBD><EFBFBD>޸ĸ<EFBFBD><EFBFBD>ַ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
-    // <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>޸<EFBFBD>Ϊ<EFBFBD><EFBFBD>˾<EFBFBD><EFBFBD><EFBFBD><EFBFBD>֯<EFBFBD><EFBFBD>
+    // 标准初始化
+    // 如果未使用这些功能并希望减小
+    // 最终可执行文件的大小，则应移除下列
+    // 不需要的特定初始化例程
+    // 更改用于存储设置的注册表项
+    // TODO: 应适当修改该字符串，
+    // 例如修改为公司或组织名
     SetRegistryKey(_T("YAMA"));
 
     CMy2015RemoteDlg dlg(nullptr);
     m_pMainWnd = &dlg;
     INT_PTR nResponse = dlg.DoModal();
     if (nResponse == IDOK) {
-        // TODO: <EFBFBD>ڴ˷<EFBFBD><EFBFBD>ô<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʱ<EFBFBD><EFBFBD>
-        //  <EFBFBD><EFBFBD>ȷ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>رնԻ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ĵ<EFBFBD><EFBFBD><EFBFBD>
+        // TODO: 在此放置处理何时用
+        //  “确定”来关闭对话框的代码
     } else if (nResponse == IDCANCEL) {
-        // TODO: <EFBFBD>ڴ˷<EFBFBD><EFBFBD>ô<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʱ<EFBFBD><EFBFBD>
-        //  <EFBFBD><EFBFBD>ȡ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>رնԻ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ĵ<EFBFBD><EFBFBD><EFBFBD>
+        // TODO: 在此放置处理何时用
+        //  “取消”来关闭对话框的代码
     }
 
-    // ɾ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>洴<EFBFBD><EFBFBD><EFBFBD><EFBFBD> shell <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
+    // 删除上面创建的 shell 管理器。
     if (pShellManager != NULL) {
         delete pShellManager;
     }
 
-    // <EFBFBD><EFBFBD><EFBFBD>ڶԻ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>ѹرգ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>Խ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> FALSE <20>Ա<EFBFBD><D4B1>˳<EFBFBD>Ӧ<EFBFBD>ó<EFBFBD><C3B3><EFBFBD><EFBFBD><EFBFBD>
-    //  <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ӧ<EFBFBD>ó<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ϣ<EFBFBD>á<EFBFBD>
+    // 由于对话框已关闭，所以将返回 FALSE 以便退出应用程序，
+    //  而不是启动应用程序的消息泵。
     return FALSE;
 }
 
@@ -186,5 +313,10 @@ int CMy2015RemoteApp::ExitInstance()
 
     SAFE_DELETE(m_iniFile);
 
+    // 只有在代理模式退出时才停止服务
+    if (IsAgentMode()) {
+        ServerService_Stop();
+    }
+
     return CWinApp::ExitInstance();
 }