mirror of https://github.com/tp4a/teleport
392 lines
8.6 KiB
C++
392 lines
8.6 KiB
C++
#include <ex/ex_winsrv.h>
|
|
|
|
#ifdef EX_OS_WIN32
|
|
|
|
#include <ex/ex_const.h>
|
|
|
|
class winsrv_helper
|
|
{
|
|
public:
|
|
winsrv_helper(SC_HANDLE scm, SC_HANDLE sc) : m_scm(scm), m_sc(sc)
|
|
{
|
|
}
|
|
~winsrv_helper()
|
|
{
|
|
if(NULL != m_sc)
|
|
CloseServiceHandle(m_sc);
|
|
if(NULL != m_scm)
|
|
CloseServiceHandle(m_scm);
|
|
}
|
|
|
|
protected:
|
|
SC_HANDLE m_sc;
|
|
SC_HANDLE m_scm;
|
|
};
|
|
|
|
|
|
ex_rv ex_winsrv_install(const ex_wstr& srv_name, const ex_wstr& disp_name, const ex_wstr& exec_path)
|
|
{
|
|
SC_HANDLE sc = NULL;
|
|
SC_HANDLE scm = NULL;
|
|
winsrv_helper srv(scm, sc);
|
|
|
|
scm = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
|
|
if (scm == NULL)
|
|
return EXRV_CANNOT_OPEN;
|
|
|
|
if (NULL == (sc = CreateServiceW(scm, srv_name.c_str(), disp_name.c_str(),
|
|
SERVICE_ALL_ACCESS,
|
|
SERVICE_WIN32_OWN_PROCESS,
|
|
SERVICE_AUTO_START, SERVICE_ERROR_NORMAL, exec_path.c_str(), NULL, NULL, NULL, NULL, NULL))
|
|
)
|
|
{
|
|
return EXRV_CANNOT_CREATE;
|
|
}
|
|
|
|
SERVICE_FAILURE_ACTIONS failure_action;
|
|
failure_action.dwResetPeriod = 0; // reset failure count to zero 的时间，单位为秒
|
|
failure_action.lpRebootMsg = NULL; // Message to broadcast to server users before rebooting
|
|
failure_action.lpCommand = NULL; // Command line of the process for the CreateProcess function to execute in response
|
|
failure_action.cActions = 3; // action数组的个数
|
|
|
|
SC_ACTION actionarray[3];
|
|
actionarray[0].Type = SC_ACTION_RESTART; // 重新启动服务
|
|
actionarray[0].Delay = 60000; // 单位为毫秒
|
|
actionarray[1].Type = SC_ACTION_RESTART;
|
|
actionarray[1].Delay = 60000;
|
|
actionarray[2].Type = SC_ACTION_RESTART;
|
|
actionarray[2].Delay = 60000;
|
|
failure_action.lpsaActions = actionarray;
|
|
|
|
ChangeServiceConfig2(sc, SERVICE_CONFIG_FAILURE_ACTIONS, &failure_action);
|
|
|
|
return EXRV_OK;
|
|
}
|
|
|
|
bool ex_winsrv_is_exists(const ex_wstr& srv_name)
|
|
{
|
|
SC_HANDLE sc = NULL;
|
|
SC_HANDLE scm = NULL;
|
|
winsrv_helper srv(scm, sc);
|
|
|
|
scm = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
|
|
if (scm == NULL)
|
|
return false;
|
|
|
|
sc = OpenServiceW(scm, srv_name.c_str(), SERVICE_ALL_ACCESS);
|
|
if (NULL == sc)
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
ex_rv ex_winsrv_uninstall(const ex_wstr& srv_name)
|
|
{
|
|
SC_HANDLE sc = NULL;
|
|
SC_HANDLE scm = NULL;
|
|
winsrv_helper srv(scm, sc);
|
|
|
|
scm = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
|
|
if (scm == NULL)
|
|
return EXRV_CANNOT_OPEN;
|
|
|
|
sc = OpenServiceW(scm, srv_name.c_str(), SERVICE_ALL_ACCESS);
|
|
if (NULL == sc)
|
|
return EXRV_NOT_EXISTS;
|
|
|
|
if (!DeleteService(sc))
|
|
return EXRV_CANNOT_REMOVE;
|
|
else
|
|
return EXRV_OK;
|
|
}
|
|
|
|
ex_rv ex_winsrv_start(const ex_wstr& srv_name)
|
|
{
|
|
SC_HANDLE sc = NULL;
|
|
SC_HANDLE scm = NULL;
|
|
winsrv_helper srv(scm, sc);
|
|
|
|
scm = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
|
|
if (scm == NULL)
|
|
return EXRV_CANNOT_OPEN;
|
|
|
|
sc = OpenServiceW(scm, srv_name.c_str(), SERVICE_START | SERVICE_QUERY_STATUS);
|
|
if (NULL == sc)
|
|
return EXRV_NOT_EXISTS;
|
|
|
|
SERVICE_STATUS ss;
|
|
if (!QueryServiceStatus(sc, &ss))
|
|
return EXRV_FAILED;
|
|
|
|
if (ss.dwCurrentState == SERVICE_RUNNING)
|
|
return EXRV_OK;
|
|
|
|
int i = 0;
|
|
if (ss.dwCurrentState == SERVICE_START_PENDING)
|
|
{
|
|
for (i = 0; i < 100; ++i)
|
|
{
|
|
Sleep(100);
|
|
QueryServiceStatus(sc, &ss);
|
|
if (ss.dwCurrentState != SERVICE_START_PENDING)
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (ss.dwCurrentState == SERVICE_STOPPED)
|
|
{
|
|
if (StartService(sc, 0, NULL))
|
|
{
|
|
for (i = 0; i < 100; ++i)
|
|
{
|
|
Sleep(100);
|
|
QueryServiceStatus(sc, &ss);
|
|
if (ss.dwCurrentState == SERVICE_RUNNING)
|
|
return EXRV_OK;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (ss.dwCurrentState == SERVICE_RUNNING)
|
|
return EXRV_OK;
|
|
else
|
|
return EXRV_FAILED;
|
|
}
|
|
|
|
ex_rv ex_winsrv_config(const ex_wstr& srv_name, QUERY_SERVICE_CONFIG& cfg)
|
|
{
|
|
SC_HANDLE sc = NULL;
|
|
SC_HANDLE scm = NULL;
|
|
winsrv_helper srv(scm, sc);
|
|
|
|
|
|
scm = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
|
|
if (scm == NULL)
|
|
return EXRV_CANNOT_OPEN;
|
|
|
|
sc = OpenServiceW(scm, srv_name.c_str(), SERVICE_QUERY_CONFIG);
|
|
if (sc == NULL)
|
|
return EXRV_NOT_EXISTS;
|
|
|
|
DWORD dwBytesNeeded;
|
|
if (!QueryServiceConfig(sc, &cfg, 4096, &dwBytesNeeded))
|
|
return EXRV_FAILED;
|
|
else
|
|
return EXRV_OK;
|
|
}
|
|
|
|
ex_rv ex_winsrv_status(const ex_wstr& srv_name, ex_ulong& status)
|
|
{
|
|
SC_HANDLE sc = NULL;
|
|
SC_HANDLE scm = NULL;
|
|
winsrv_helper srv(scm, sc);
|
|
|
|
scm = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
|
|
if (scm == NULL)
|
|
return EXRV_CANNOT_OPEN;
|
|
|
|
sc = OpenServiceW(scm, srv_name.c_str(), SERVICE_QUERY_STATUS);
|
|
if (NULL == sc)
|
|
return EXRV_NOT_EXISTS;
|
|
|
|
SERVICE_STATUS ss;
|
|
if (!QueryServiceStatus(sc, &ss))
|
|
return EXRV_FAILED;
|
|
|
|
status = ss.dwCurrentState;
|
|
return EXRV_OK;
|
|
}
|
|
|
|
ex_rv ex_winsrv_stop(const ex_wstr& srv_name)
|
|
{
|
|
SC_HANDLE sc = NULL;
|
|
SC_HANDLE scm = NULL;
|
|
winsrv_helper srv(scm, sc);
|
|
|
|
scm = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
|
|
if (scm == NULL)
|
|
return EXRV_CANNOT_OPEN;
|
|
|
|
sc = OpenServiceW(scm, srv_name.c_str(), SERVICE_STOP | SERVICE_QUERY_STATUS);
|
|
if (NULL == sc)
|
|
return EXRV_NOT_EXISTS;
|
|
|
|
SERVICE_STATUS ss;
|
|
if (!QueryServiceStatus(sc, &ss))
|
|
return EXRV_FAILED;
|
|
|
|
if (ss.dwCurrentState == SERVICE_STOPPED)
|
|
return EXRV_OK;
|
|
|
|
int i = 0;
|
|
|
|
DWORD dwStatus = ss.dwCurrentState;
|
|
if (ss.dwCurrentState == SERVICE_START_PENDING || ss.dwCurrentState == SERVICE_PAUSE_PENDING || ss.dwCurrentState == SERVICE_CONTINUE_PENDING || ss.dwCurrentState == SERVICE_STOP_PENDING)
|
|
{
|
|
for (i = 0; i < 100; ++i)
|
|
{
|
|
Sleep(100);
|
|
QueryServiceStatus(sc, &ss);
|
|
if (ss.dwCurrentState != dwStatus)
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (ss.dwCurrentState == SERVICE_RUNNING || ss.dwCurrentState == SERVICE_PAUSED)
|
|
{
|
|
if (ControlService(sc, SERVICE_CONTROL_STOP, &ss))
|
|
{
|
|
for (i = 0; i < 100; ++i)
|
|
{
|
|
Sleep(100);
|
|
QueryServiceStatus(sc, &ss);
|
|
if (ss.dwCurrentState == SERVICE_STOPPED)
|
|
return EXRV_OK;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (ss.dwCurrentState == SERVICE_STOPPED)
|
|
return EXRV_OK;
|
|
else
|
|
return EXRV_FAILED;
|
|
}
|
|
|
|
ex_rv ex_winsrv_pause(const ex_wstr& srv_name)
|
|
{
|
|
SC_HANDLE sc = NULL;
|
|
SC_HANDLE scm = NULL;
|
|
winsrv_helper srv(scm, sc);
|
|
|
|
scm = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
|
|
if (scm == NULL)
|
|
return EXRV_CANNOT_OPEN;
|
|
|
|
sc = OpenServiceW(scm, srv_name.c_str(), SERVICE_PAUSE_CONTINUE | SERVICE_QUERY_STATUS);
|
|
if (NULL == sc)
|
|
return EXRV_NOT_EXISTS;
|
|
|
|
SERVICE_STATUS ss;
|
|
if(!QueryServiceStatus(sc, &ss))
|
|
return EXRV_FAILED;
|
|
|
|
if (ss.dwCurrentState == SERVICE_PAUSED)
|
|
return EXRV_OK;
|
|
|
|
int i = 0;
|
|
|
|
DWORD dwStatus = ss.dwCurrentState;
|
|
if (ss.dwCurrentState == SERVICE_START_PENDING || ss.dwCurrentState == SERVICE_PAUSE_PENDING || ss.dwCurrentState == SERVICE_CONTINUE_PENDING)
|
|
{
|
|
for (i = 0; i < 100; ++i)
|
|
{
|
|
Sleep(100);
|
|
QueryServiceStatus(sc, &ss);
|
|
if (ss.dwCurrentState != dwStatus)
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (ss.dwCurrentState == SERVICE_RUNNING)
|
|
{
|
|
if (ControlService(sc, SERVICE_CONTROL_PAUSE, &ss))
|
|
{
|
|
for (i = 0; i < 100; ++i)
|
|
{
|
|
Sleep(100);
|
|
QueryServiceStatus(sc, &ss);
|
|
if (ss.dwCurrentState == SERVICE_PAUSED)
|
|
return EXRV_OK;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (ss.dwCurrentState == SERVICE_PAUSED)
|
|
return EXRV_OK;
|
|
else
|
|
return EXRV_FAILED;
|
|
}
|
|
|
|
ex_rv ex_winsrv_resume(const ex_wstr& srv_name)
|
|
{
|
|
SC_HANDLE sc = NULL;
|
|
SC_HANDLE scm = NULL;
|
|
winsrv_helper srv(scm, sc);
|
|
|
|
scm = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
|
|
if (scm == NULL)
|
|
return EXRV_CANNOT_OPEN;
|
|
|
|
sc = OpenServiceW(scm, srv_name.c_str(), SERVICE_PAUSE_CONTINUE | SERVICE_QUERY_STATUS);
|
|
if (NULL == sc)
|
|
return EXRV_NOT_EXISTS;
|
|
|
|
SERVICE_STATUS ss;
|
|
if (!QueryServiceStatus(sc, &ss))
|
|
return EXRV_FAILED;
|
|
|
|
if (ss.dwCurrentState == SERVICE_RUNNING)
|
|
return EXRV_OK;
|
|
|
|
int i = 0;
|
|
|
|
DWORD dwStatus = ss.dwCurrentState;
|
|
if (ss.dwCurrentState == SERVICE_START_PENDING || ss.dwCurrentState == SERVICE_PAUSE_PENDING || ss.dwCurrentState == SERVICE_CONTINUE_PENDING)
|
|
{
|
|
for (i = 0; i < 100; ++i)
|
|
{
|
|
Sleep(100);
|
|
QueryServiceStatus(sc, &ss);
|
|
if (ss.dwCurrentState != dwStatus)
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (ss.dwCurrentState == SERVICE_PAUSED)
|
|
{
|
|
if (ControlService(sc, SERVICE_CONTROL_CONTINUE, &ss))
|
|
{
|
|
for (i = 0; i < 100; ++i)
|
|
{
|
|
Sleep(100);
|
|
QueryServiceStatus(sc, &ss);
|
|
if (ss.dwCurrentState == SERVICE_RUNNING)
|
|
return EXRV_OK;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (ss.dwCurrentState == SERVICE_RUNNING)
|
|
return EXRV_OK;
|
|
else
|
|
return EXRV_FAILED;
|
|
}
|
|
|
|
ex_rv ex_winsrv_pid(const ex_wstr& srv_name, ex_ulong& pid)
|
|
{
|
|
SC_HANDLE sc = NULL;
|
|
SC_HANDLE scm = NULL;
|
|
winsrv_helper srv(scm, sc);
|
|
|
|
scm = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
|
|
if (scm == NULL)
|
|
return EXRV_CANNOT_OPEN;
|
|
|
|
sc = OpenServiceW(scm, srv_name.c_str(), SERVICE_QUERY_STATUS);
|
|
if (NULL == sc)
|
|
return EXRV_NOT_EXISTS;
|
|
|
|
DWORD byteneeded = 0;
|
|
ex_u8 buf[1024] = { 0 };
|
|
QueryServiceStatusEx(sc, SC_STATUS_PROCESS_INFO, buf, 1024, &byteneeded);
|
|
|
|
LPSERVICE_STATUS_PROCESS lp = (LPSERVICE_STATUS_PROCESS)buf;
|
|
if (lp->dwCurrentState != SERVICE_RUNNING)
|
|
return EXRV_NOT_START;
|
|
|
|
pid = lp->dwProcessId;
|
|
|
|
return EXRV_OK;
|
|
}
|
|
#endif
|