mirror of
https://github.com/yuanyuanxiang/SimpleRemoter.git
synced 2026-01-22 07:14:15 +08:00
653 lines
19 KiB
C
653 lines
19 KiB
C
#include "ServiceWrapper.h"
|
||
#include "SessionMonitor.h"
|
||
#include <stdio.h>
|
||
|
||
#define Mprintf(format, ...) MyLog(__FILE__, __LINE__, format, __VA_ARGS__)
|
||
|
||
// 静态变量
|
||
static MyService g_MyService =
|
||
{ "RemoteControlService", "Remote Control Service", "Provides remote desktop control functionality."};
|
||
|
||
static SERVICE_STATUS g_ServiceStatus = { 0 };
|
||
static SERVICE_STATUS_HANDLE g_StatusHandle = NULL;
|
||
static HANDLE g_StopEvent = NULL;
|
||
static ServiceLogFunc Log = NULL;
|
||
|
||
// 前向声明
|
||
static void WINAPI ServiceMain(DWORD argc, LPTSTR* argv);
|
||
static void WINAPI ServiceCtrlHandler(DWORD ctrlCode);
|
||
|
||
void MyLog(const char* file, int line, const char* format, ...)
|
||
{
|
||
if (Log == NULL) {
|
||
return; // 没有设置日志回调，直接返回
|
||
}
|
||
|
||
char buffer[1024];
|
||
char message[1200];
|
||
|
||
// 处理可变参数
|
||
va_list args;
|
||
va_start(args, format);
|
||
vsnprintf(buffer, sizeof(buffer), format, args);
|
||
va_end(args);
|
||
|
||
// 提取文件名（去掉路径）
|
||
const char* filename = strrchr(file, '/');
|
||
if (filename == NULL) {
|
||
filename = strrchr(file, '\\');
|
||
}
|
||
filename = (filename != NULL) ? (filename + 1) : file;
|
||
|
||
// 格式化完整的日志消息：[文件名:行号] 消息内容
|
||
snprintf(message, sizeof(message), "[%s:%d] %s", filename, line, buffer);
|
||
|
||
// 调用日志回调函数
|
||
Log(message);
|
||
}
|
||
|
||
void InitWindowsService(MyService info, ServiceLogFunc log)
|
||
{
|
||
memcpy(&g_MyService, &info, sizeof(MyService));
|
||
Log = log;
|
||
}
|
||
|
||
BOOL ServiceWrapper_CheckStatus(BOOL* registered, BOOL* running,
|
||
char* exePath, size_t exePathSize)
|
||
{
|
||
SC_HANDLE hSCM = NULL;
|
||
SC_HANDLE hService = NULL;
|
||
BOOL result = FALSE;
|
||
SERVICE_STATUS_PROCESS ssp;
|
||
DWORD bytesNeeded = 0;
|
||
DWORD bufSize = 0;
|
||
LPQUERY_SERVICE_CONFIGA pConfig = NULL;
|
||
|
||
*registered = FALSE;
|
||
*running = FALSE;
|
||
if (exePath && exePathSize > 0) {
|
||
exePath[0] = '\0';
|
||
}
|
||
|
||
// 打开 SCM
|
||
hSCM = OpenSCManagerA(NULL, NULL, SC_MANAGER_CONNECT);
|
||
if (!hSCM) {
|
||
return FALSE;
|
||
}
|
||
|
||
// 打开服务
|
||
hService = OpenServiceA(
|
||
hSCM,
|
||
g_MyService.Name,
|
||
SERVICE_QUERY_STATUS | SERVICE_QUERY_CONFIG);
|
||
if (!hService) {
|
||
CloseServiceHandle(hSCM);
|
||
return TRUE; // 未注册
|
||
}
|
||
|
||
*registered = TRUE;
|
||
result = TRUE;
|
||
|
||
// 获取服务状态
|
||
memset(&ssp, 0, sizeof(ssp));
|
||
if (QueryServiceStatusEx(
|
||
hService,
|
||
SC_STATUS_PROCESS_INFO,
|
||
(LPBYTE)&ssp,
|
||
sizeof(SERVICE_STATUS_PROCESS),
|
||
&bytesNeeded)) {
|
||
*running = (ssp.dwCurrentState == SERVICE_RUNNING);
|
||
}
|
||
|
||
// 获取 EXE 路径
|
||
if (exePath && exePathSize > 0) {
|
||
QueryServiceConfigA(hService, NULL, 0, &bufSize);
|
||
|
||
if (GetLastError() == ERROR_INSUFFICIENT_BUFFER) {
|
||
pConfig = (LPQUERY_SERVICE_CONFIGA)malloc(bufSize);
|
||
|
||
if (pConfig) {
|
||
if (QueryServiceConfigA(hService, pConfig, bufSize, &bufSize)) {
|
||
strncpy(exePath, pConfig->lpBinaryPathName, exePathSize - 1);
|
||
exePath[exePathSize - 1] = '\0';
|
||
}
|
||
free(pConfig);
|
||
}
|
||
}
|
||
}
|
||
|
||
CloseServiceHandle(hService);
|
||
CloseServiceHandle(hSCM);
|
||
return result;
|
||
}
|
||
|
||
int ServiceWrapper_StartSimple(void)
|
||
{
|
||
SC_HANDLE hSCM = NULL;
|
||
SC_HANDLE hService = NULL;
|
||
BOOL ok;
|
||
int err;
|
||
|
||
// 打开SCM
|
||
hSCM = OpenSCManagerA(NULL, NULL, SC_MANAGER_CONNECT);
|
||
if (!hSCM) {
|
||
return (int)GetLastError();
|
||
}
|
||
|
||
// 打开服务并启动
|
||
hService = OpenServiceA(hSCM, g_MyService.Name, SERVICE_START);
|
||
if (!hService) {
|
||
err = (int)GetLastError();
|
||
CloseServiceHandle(hSCM);
|
||
return err;
|
||
}
|
||
|
||
// 启动服务
|
||
ok = StartServiceA(hService, 0, NULL);
|
||
err = ok ? ERROR_SUCCESS : (int)GetLastError();
|
||
|
||
CloseServiceHandle(hService);
|
||
CloseServiceHandle(hSCM);
|
||
|
||
return err;
|
||
}
|
||
|
||
int ServiceWrapper_Run(void)
|
||
{
|
||
DWORD err;
|
||
char buffer[256];
|
||
SERVICE_TABLE_ENTRY ServiceTable[2];
|
||
|
||
ServiceTable[0].lpServiceName = (LPSTR)g_MyService.Name;
|
||
ServiceTable[0].lpServiceProc = (LPSERVICE_MAIN_FUNCTION)ServiceMain;
|
||
ServiceTable[1].lpServiceName = NULL;
|
||
ServiceTable[1].lpServiceProc = NULL;
|
||
|
||
Mprintf("========================================");
|
||
Mprintf("ServiceWrapper_Run() called");
|
||
|
||
if (StartServiceCtrlDispatcher(ServiceTable) == FALSE) {
|
||
err = GetLastError();
|
||
sprintf(buffer, "StartServiceCtrlDispatcher failed: %d", (int)err);
|
||
Mprintf(buffer);
|
||
return (int)err;
|
||
}
|
||
return ERROR_SUCCESS;
|
||
}
|
||
|
||
static void WINAPI ServiceMain(DWORD argc, LPTSTR* argv)
|
||
{
|
||
HANDLE hThread;
|
||
|
||
(void)argc;
|
||
(void)argv;
|
||
|
||
Mprintf("ServiceMain() called");
|
||
|
||
g_StatusHandle = RegisterServiceCtrlHandler(
|
||
g_MyService.Name,
|
||
ServiceCtrlHandler
|
||
);
|
||
|
||
if (g_StatusHandle == NULL) {
|
||
Mprintf("RegisterServiceCtrlHandler failed");
|
||
return;
|
||
}
|
||
|
||
ZeroMemory(&g_ServiceStatus, sizeof(g_ServiceStatus));
|
||
g_ServiceStatus.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
|
||
g_ServiceStatus.dwCurrentState = SERVICE_START_PENDING;
|
||
g_ServiceStatus.dwControlsAccepted = 0;
|
||
g_ServiceStatus.dwWin32ExitCode = 0;
|
||
g_ServiceStatus.dwServiceSpecificExitCode = 0;
|
||
g_ServiceStatus.dwCheckPoint = 0;
|
||
g_ServiceStatus.dwWaitHint = 0;
|
||
|
||
SetServiceStatus(g_StatusHandle, &g_ServiceStatus);
|
||
|
||
g_StopEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
|
||
if (g_StopEvent == NULL) {
|
||
Mprintf("CreateEvent failed");
|
||
g_ServiceStatus.dwCurrentState = SERVICE_STOPPED;
|
||
g_ServiceStatus.dwWin32ExitCode = GetLastError();
|
||
SetServiceStatus(g_StatusHandle, &g_ServiceStatus);
|
||
return;
|
||
}
|
||
|
||
g_ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP;
|
||
g_ServiceStatus.dwCurrentState = SERVICE_RUNNING;
|
||
g_ServiceStatus.dwWin32ExitCode = 0;
|
||
g_ServiceStatus.dwCheckPoint = 0;
|
||
|
||
SetServiceStatus(g_StatusHandle, &g_ServiceStatus);
|
||
Mprintf("Service is now running");
|
||
|
||
hThread = CreateThread(NULL, 0, ServiceWrapper_WorkerThread, NULL, 0, NULL);
|
||
if (hThread) {
|
||
WaitForSingleObject(hThread, INFINITE);
|
||
SAFE_CLOSE_HANDLE(hThread);
|
||
}
|
||
|
||
SAFE_CLOSE_HANDLE(g_StopEvent);
|
||
|
||
g_ServiceStatus.dwControlsAccepted = 0;
|
||
g_ServiceStatus.dwCurrentState = SERVICE_STOPPED;
|
||
g_ServiceStatus.dwWin32ExitCode = 0;
|
||
g_ServiceStatus.dwCheckPoint = 3;
|
||
|
||
SetServiceStatus(g_StatusHandle, &g_ServiceStatus);
|
||
Mprintf("Service stopped");
|
||
}
|
||
|
||
static void WINAPI ServiceCtrlHandler(DWORD ctrlCode)
|
||
{
|
||
switch (ctrlCode) {
|
||
case SERVICE_CONTROL_STOP:
|
||
Mprintf("SERVICE_CONTROL_STOP received");
|
||
|
||
if (g_ServiceStatus.dwCurrentState != SERVICE_RUNNING)
|
||
break;
|
||
|
||
g_ServiceStatus.dwControlsAccepted = 0;
|
||
g_ServiceStatus.dwCurrentState = SERVICE_STOP_PENDING;
|
||
g_ServiceStatus.dwWin32ExitCode = 0;
|
||
g_ServiceStatus.dwCheckPoint = 4;
|
||
g_ServiceStatus.dwWaitHint = 0;
|
||
|
||
SetServiceStatus(g_StatusHandle, &g_ServiceStatus);
|
||
SetEvent(g_StopEvent);
|
||
break;
|
||
|
||
case SERVICE_CONTROL_INTERROGATE:
|
||
SetServiceStatus(g_StatusHandle, &g_ServiceStatus);
|
||
break;
|
||
|
||
default:
|
||
break;
|
||
}
|
||
}
|
||
|
||
// 服务工作线程
|
||
DWORD WINAPI ServiceWrapper_WorkerThread(LPVOID lpParam)
|
||
{
|
||
SessionMonitor monitor;
|
||
int heartbeatCount = 0;
|
||
char buf[128];
|
||
|
||
(void)lpParam; // 未使用参数
|
||
|
||
Mprintf("========================================");
|
||
Mprintf("Worker thread started");
|
||
Mprintf("Service will launch agent in user sessions");
|
||
|
||
// 初始化会话监控器
|
||
SessionMonitor_Init(&monitor);
|
||
|
||
if (!SessionMonitor_Start(&monitor)) {
|
||
Mprintf("ERROR: Failed to start session monitor");
|
||
SessionMonitor_Cleanup(&monitor);
|
||
return ERROR_SERVICE_SPECIFIC_ERROR;
|
||
}
|
||
|
||
Mprintf("Session monitor started successfully");
|
||
Mprintf("Agent will be launched automatically");
|
||
|
||
// 主循环，只等待停止信号
|
||
// SessionMonitor 会在后台自动：
|
||
// 1. 监控会话
|
||
// 2. 在用户会话中启动 agent.exe
|
||
// 3. 监视代理进程，如果退出自动重启
|
||
while (WaitForSingleObject(g_StopEvent, 10000) != WAIT_OBJECT_0) {
|
||
heartbeatCount++;
|
||
if (heartbeatCount % 6 == 0) { // 每60秒记录一次
|
||
sprintf(buf, "Service heartbeat - uptime: %d minutes", heartbeatCount / 6);
|
||
Mprintf(buf);
|
||
}
|
||
}
|
||
|
||
Mprintf("Stop signal received");
|
||
Mprintf("Stopping session monitor...");
|
||
SessionMonitor_Stop(&monitor);
|
||
SessionMonitor_Cleanup(&monitor);
|
||
|
||
Mprintf("Worker thread exiting");
|
||
Mprintf("========================================");
|
||
return ERROR_SUCCESS;
|
||
}
|
||
|
||
|
||
int ServiceWrapper_Stop(void)
|
||
{
|
||
// 打开SCM
|
||
SC_HANDLE hSCM = OpenSCManagerA(NULL, NULL, SC_MANAGER_CONNECT);
|
||
if (!hSCM) {
|
||
return (int)GetLastError();
|
||
}
|
||
|
||
// 打开服务
|
||
SC_HANDLE hService = OpenServiceA(hSCM, g_MyService.Name, SERVICE_STOP | SERVICE_QUERY_STATUS);
|
||
if (!hService) {
|
||
int err = (int)GetLastError();
|
||
CloseServiceHandle(hSCM);
|
||
return err;
|
||
}
|
||
|
||
// 查询当前状态
|
||
SERVICE_STATUS status;
|
||
if (!QueryServiceStatus(hService, &status)) {
|
||
int err = (int)GetLastError();
|
||
CloseServiceHandle(hService);
|
||
CloseServiceHandle(hSCM);
|
||
return err;
|
||
}
|
||
|
||
// 如果服务未运行，直接返回成功
|
||
if (status.dwCurrentState == SERVICE_STOPPED) {
|
||
CloseServiceHandle(hService);
|
||
CloseServiceHandle(hSCM);
|
||
return ERROR_SUCCESS;
|
||
}
|
||
|
||
// 发送停止控制命令
|
||
if (!ControlService(hService, SERVICE_CONTROL_STOP, &status)) {
|
||
DWORD err = GetLastError();
|
||
if (err != ERROR_SERVICE_NOT_ACTIVE) {
|
||
CloseServiceHandle(hService);
|
||
CloseServiceHandle(hSCM);
|
||
return (int)err;
|
||
}
|
||
}
|
||
|
||
// 等待服务停止（最多3秒）
|
||
int waitCount = 0;
|
||
while (status.dwCurrentState != SERVICE_STOPPED && waitCount < 3) {
|
||
Sleep(1000);
|
||
waitCount++;
|
||
if (!QueryServiceStatus(hService, &status)) {
|
||
break;
|
||
}
|
||
}
|
||
|
||
int result = (status.dwCurrentState == SERVICE_STOPPED) ? ERROR_SUCCESS : ERROR_TIMEOUT;
|
||
|
||
CloseServiceHandle(hService);
|
||
CloseServiceHandle(hSCM);
|
||
|
||
return result;
|
||
}
|
||
|
||
BOOL ServiceWrapper_Install(void)
|
||
{
|
||
SC_HANDLE schSCManager;
|
||
SC_HANDLE schService;
|
||
char szPath[MAX_PATH];
|
||
SERVICE_DESCRIPTION sd;
|
||
SERVICE_STATUS status;
|
||
DWORD err;
|
||
|
||
schSCManager = OpenSCManager(
|
||
NULL,
|
||
NULL,
|
||
SC_MANAGER_ALL_ACCESS
|
||
);
|
||
|
||
if (schSCManager == NULL) {
|
||
Mprintf("ERROR: OpenSCManager failed (%d)\n", (int)GetLastError());
|
||
Mprintf("Please run as Administrator\n");
|
||
return FALSE;
|
||
}
|
||
|
||
if (!GetModuleFileName(NULL, szPath, MAX_PATH)) {
|
||
Mprintf("ERROR: GetModuleFileName failed (%d)\n", (int)GetLastError());
|
||
CloseServiceHandle(schSCManager);
|
||
return FALSE;
|
||
}
|
||
|
||
Mprintf("Installing service...\n");
|
||
Mprintf("Executable path: %s\n", szPath);
|
||
|
||
int retryCount = 5;
|
||
|
||
for (int i = 0; i < retryCount; i++) {
|
||
schService = CreateService(
|
||
schSCManager,
|
||
g_MyService.Name,
|
||
g_MyService.Display,
|
||
SERVICE_ALL_ACCESS,
|
||
SERVICE_WIN32_OWN_PROCESS,
|
||
SERVICE_AUTO_START,
|
||
SERVICE_ERROR_NORMAL,
|
||
szPath,
|
||
NULL, NULL, NULL, NULL, NULL
|
||
);
|
||
if (schService != NULL) {
|
||
break; // 成功
|
||
}
|
||
|
||
DWORD err = GetLastError();
|
||
if (err == ERROR_SERVICE_MARKED_FOR_DELETE) {
|
||
Sleep(2000);
|
||
continue;
|
||
}
|
||
break; // 其他错误，退出
|
||
}
|
||
|
||
if (schService == NULL) {
|
||
err = GetLastError();
|
||
if (err == ERROR_SERVICE_EXISTS) {
|
||
Mprintf("INFO: Service already exists\n");
|
||
|
||
// 打开已存在的服务
|
||
schService = OpenService(schSCManager, g_MyService.Name, SERVICE_ALL_ACCESS);
|
||
if (schService) {
|
||
Mprintf("SUCCESS: Service is already installed\n");
|
||
CloseServiceHandle(schService);
|
||
CloseServiceHandle(schSCManager);
|
||
return TRUE;
|
||
}
|
||
} else if (err == ERROR_ACCESS_DENIED) {
|
||
Mprintf("ERROR: Access denied. Please run as Administrator\n");
|
||
} else {
|
||
Mprintf("ERROR: CreateService failed (%d)\n", (int)err);
|
||
}
|
||
CloseServiceHandle(schSCManager);
|
||
return FALSE;
|
||
}
|
||
|
||
Mprintf("SUCCESS: Service created successfully\n");
|
||
|
||
// 设置服务描述
|
||
sd.lpDescription = (LPSTR)g_MyService.Description;
|
||
if (ChangeServiceConfig2(schService, SERVICE_CONFIG_DESCRIPTION, &sd)) {
|
||
Mprintf("SUCCESS: Service description set\n");
|
||
}
|
||
|
||
// 立即启动服务
|
||
Mprintf("Starting service...\n");
|
||
if (StartService(schService, 0, NULL)) {
|
||
Mprintf("SUCCESS: Service started successfully\n");
|
||
|
||
// 等待服务启动
|
||
Sleep(2000);
|
||
|
||
// 检查服务状态
|
||
if (QueryServiceStatus(schService, &status)) {
|
||
if (status.dwCurrentState == SERVICE_RUNNING) {
|
||
Mprintf("SUCCESS: Service is running\n");
|
||
} else {
|
||
Mprintf("WARNING: Service state: %d\n", (int)status.dwCurrentState);
|
||
}
|
||
}
|
||
} else {
|
||
err = GetLastError();
|
||
if (err == ERROR_SERVICE_ALREADY_RUNNING) {
|
||
Mprintf("INFO: Service is already running\n");
|
||
} else {
|
||
Mprintf("WARNING: StartService failed (%d)\n", (int)err);
|
||
Mprintf("You can start it manually using: net start %s\n", g_MyService.Name);
|
||
}
|
||
}
|
||
|
||
CloseServiceHandle(schService);
|
||
CloseServiceHandle(schSCManager);
|
||
|
||
Mprintf("=== Installation Complete ===\n");
|
||
Mprintf("Service installed successfully!\n");
|
||
Mprintf("IMPORTANT: This is a single-executable design.\n");
|
||
Mprintf("The service will launch '%s -agent' in user sessions.\n", szPath);
|
||
Mprintf("Commands:\n");
|
||
Mprintf(" To verify: sc query %s\n", g_MyService.Name);
|
||
Mprintf(" To start: net start %s\n", g_MyService.Name);
|
||
Mprintf(" To stop: net stop %s\n", g_MyService.Name);
|
||
|
||
return TRUE;
|
||
}
|
||
|
||
BOOL ServiceWrapper_Uninstall(void)
|
||
{
|
||
SC_HANDLE schSCManager;
|
||
SC_HANDLE schService;
|
||
SERVICE_STATUS status;
|
||
int waitCount;
|
||
DWORD err;
|
||
|
||
schSCManager = OpenSCManager(
|
||
NULL,
|
||
NULL,
|
||
SC_MANAGER_ALL_ACCESS
|
||
);
|
||
|
||
if (schSCManager == NULL) {
|
||
Mprintf("ERROR: OpenSCManager failed (%d)\n", (int)GetLastError());
|
||
Mprintf("Please run as Administrator\n");
|
||
return FALSE;
|
||
}
|
||
|
||
schService = OpenService(
|
||
schSCManager,
|
||
g_MyService.Name,
|
||
SERVICE_STOP | DELETE
|
||
);
|
||
|
||
if (schService == NULL) {
|
||
Mprintf("ERROR: OpenService failed (%d)\n", (int)GetLastError());
|
||
Mprintf("Service may not be installed\n");
|
||
CloseServiceHandle(schSCManager);
|
||
return FALSE;
|
||
}
|
||
|
||
Mprintf("Stopping service...\n");
|
||
if (ControlService(schService, SERVICE_CONTROL_STOP, &status)) {
|
||
Mprintf("Waiting for service to stop");
|
||
Sleep(1000);
|
||
|
||
waitCount = 0;
|
||
while (QueryServiceStatus(schService, &status) && waitCount < 30) {
|
||
if (status.dwCurrentState == SERVICE_STOP_PENDING) {
|
||
Mprintf(".");
|
||
Sleep(1000);
|
||
waitCount++;
|
||
} else {
|
||
break;
|
||
}
|
||
}
|
||
|
||
if (status.dwCurrentState == SERVICE_STOPPED) {
|
||
Mprintf("SUCCESS: Service stopped\n");
|
||
} else {
|
||
Mprintf("WARNING: Service may not have stopped completely\n");
|
||
}
|
||
} else {
|
||
err = GetLastError();
|
||
if (err == ERROR_SERVICE_NOT_ACTIVE) {
|
||
Mprintf("INFO: Service was not running\n");
|
||
} else {
|
||
Mprintf("WARNING: Failed to stop service (%d)\n", (int)err);
|
||
}
|
||
}
|
||
BOOL result = TRUE;
|
||
Mprintf("Deleting service...\n");
|
||
if (DeleteService(schService)) {
|
||
Mprintf("SUCCESS: Service uninstalled successfully\n");
|
||
} else {
|
||
Mprintf("ERROR: DeleteService failed (%d)\n", (int)GetLastError());
|
||
result = FALSE;
|
||
}
|
||
|
||
CloseServiceHandle(schService);
|
||
CloseServiceHandle(schSCManager);
|
||
|
||
Mprintf("=== Uninstallation Complete ===\n");
|
||
return result;
|
||
}
|
||
|
||
void PrintUsage()
|
||
{
|
||
Mprintf("Usage:\n");
|
||
Mprintf(" -install Install as Windows service\n");
|
||
Mprintf(" -uninstall Uninstall service\n");
|
||
Mprintf(" -service Run as service\n");
|
||
Mprintf(" -agent Run as agent\n");
|
||
Mprintf(" default Run as normal application\n");
|
||
}
|
||
|
||
BOOL RunAsWindowsService(int argc, const char* argv[])
|
||
{
|
||
if (argc == 1) { // 无参数时，作为服务启动
|
||
BOOL registered = FALSE;
|
||
BOOL running = FALSE;
|
||
char servicePath[MAX_PATH] = { 0 };
|
||
char curPath[MAX_PATH] = { 0 };
|
||
|
||
BOOL b = ServiceWrapper_CheckStatus(®istered, &running, servicePath, MAX_PATH);
|
||
Mprintf("ServiceWrapper_CheckStatus: %s, Installed: %s, Running: %s\n", b ? "succeed" : "failed",
|
||
registered ? "Yes" : "No", running ? "Yes" : "No");
|
||
GetModuleFileName(NULL, curPath, MAX_PATH);
|
||
|
||
if (registered) {
|
||
Mprintf("Current executable path: %s, Registered service path: %s\n", curPath, servicePath);
|
||
}
|
||
|
||
// 使用不区分大小写的比较
|
||
_strlwr(servicePath);
|
||
_strlwr(curPath);
|
||
BOOL same = (strstr(servicePath, curPath) != 0);
|
||
if (registered && !same) {
|
||
BOOL r = ServiceWrapper_Uninstall();
|
||
Mprintf("RunAsWindowsService Uninstall %s: %s\n", r ? "succeed" : "failed", servicePath);
|
||
registered = FALSE;
|
||
}
|
||
if (!registered) {
|
||
BOOL r = ServiceWrapper_Install();
|
||
Mprintf("RunAsWindowsService Install %s: %s\n", r ? "succeed" : "failed", curPath);
|
||
return r;
|
||
} else if (!running) {
|
||
int r = ServiceWrapper_Run();
|
||
Mprintf("RunAsWindowsService Run '%s' %s\n", curPath, r == ERROR_SUCCESS ? "succeed" : "failed");
|
||
if (r) {
|
||
r = ServiceWrapper_StartSimple();
|
||
Mprintf("RunService Start '%s' %s\n", curPath, r == ERROR_SUCCESS ? "succeed" : "failed");
|
||
return r == ERROR_SUCCESS;
|
||
}
|
||
return TRUE;
|
||
}
|
||
return TRUE;
|
||
} else if (argc > 1) {
|
||
if (_stricmp(argv[1], "-install") == 0) {
|
||
return ServiceWrapper_Install();
|
||
} else if (_stricmp(argv[1], "-uninstall") == 0) {
|
||
ServiceWrapper_Uninstall();
|
||
return TRUE;
|
||
} else if (_stricmp(argv[1], "-service") == 0) {
|
||
return ServiceWrapper_Run() == ERROR_SUCCESS;
|
||
} else if (_stricmp(argv[1], "-agent") == 0) {
|
||
return FALSE;
|
||
} else if (_stricmp(argv[1], "-help") == 0 || _stricmp(argv[1], "/?") == 0) {
|
||
PrintUsage();
|
||
return TRUE;
|
||
}
|
||
}
|
||
|
||
return FALSE;
|
||
}
|