完善目录结构
完善了目录结构,添加了以前的web段com组件调用的代码(在/测试目录下)(部署没有使用到)
This commit is contained in:
263
测试/单独功能测试/Service服务测试/Service.cpp
Normal file
263
测试/单独功能测试/Service服务测试/Service.cpp
Normal file
@@ -0,0 +1,263 @@
|
||||
// Service.cpp: implementation of the CService class.
|
||||
//
|
||||
//////////////////////////////////////////////////////////////////////
|
||||
|
||||
#include "stdafx.h"
|
||||
#include <stdio.h>
|
||||
#include "Service.h"
|
||||
|
||||
CService _Module;
|
||||
//////////////////////////////////////////////////////////////////////
|
||||
// Construction/Destruction
|
||||
//////////////////////////////////////////////////////////////////////
|
||||
|
||||
CService::CService()
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
CService::~CService()
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
void CService::Init(LPCTSTR pServiceName,LPCTSTR pServiceDisplayedName)
|
||||
{
|
||||
lstrcpy(m_szServiceName,pServiceName);
|
||||
lstrcpy(m_szServiceDisplayedName,pServiceDisplayedName);
|
||||
|
||||
// set up the initial service status
|
||||
m_hServiceStatus = NULL;
|
||||
m_status.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
|
||||
m_status.dwCurrentState = SERVICE_STOPPED;
|
||||
m_status.dwControlsAccepted = SERVICE_ACCEPT_STOP;
|
||||
m_status.dwWin32ExitCode = 0;
|
||||
m_status.dwServiceSpecificExitCode = 0;
|
||||
m_status.dwCheckPoint = 0;
|
||||
m_status.dwWaitHint = 0;
|
||||
}
|
||||
|
||||
void CService::Start()
|
||||
{
|
||||
SERVICE_TABLE_ENTRY st[] =
|
||||
{
|
||||
{ m_szServiceName, _ServiceMain },
|
||||
{ NULL, NULL }
|
||||
};
|
||||
if (!::StartServiceCtrlDispatcher(st) && m_bService)
|
||||
{
|
||||
DWORD dw = GetLastError();
|
||||
LogEvent("StartServiceCtrlDispatcher Error=%d",dw);
|
||||
m_bService = FALSE;
|
||||
}
|
||||
|
||||
if (m_bService == FALSE)
|
||||
Run();
|
||||
}
|
||||
|
||||
void CService::ServiceMain()
|
||||
{
|
||||
// Register the control request handler
|
||||
m_status.dwCurrentState = SERVICE_START_PENDING;
|
||||
m_hServiceStatus = RegisterServiceCtrlHandler(m_szServiceName, _Handler);
|
||||
if (m_hServiceStatus == NULL)
|
||||
{
|
||||
LogEvent("Handler not installed");
|
||||
return;
|
||||
}
|
||||
SetServiceStatus(SERVICE_START_PENDING);
|
||||
|
||||
m_status.dwWin32ExitCode = S_OK;
|
||||
m_status.dwCheckPoint = 0;
|
||||
m_status.dwWaitHint = 0;
|
||||
|
||||
// When the Run function returns, the service has stopped.
|
||||
Run();
|
||||
|
||||
SetServiceStatus(SERVICE_STOPPED);
|
||||
LogEvent("Service stopped");
|
||||
}
|
||||
|
||||
inline void CService::Handler(DWORD dwOpcode)
|
||||
{
|
||||
switch (dwOpcode)
|
||||
{
|
||||
case SERVICE_CONTROL_STOP:
|
||||
LogEvent("Request to stop...");
|
||||
SetServiceStatus(SERVICE_STOP_PENDING);
|
||||
PostThreadMessage(m_dwThreadID, WM_QUIT, 0, 0);
|
||||
break;
|
||||
case SERVICE_CONTROL_PAUSE:
|
||||
break;
|
||||
case SERVICE_CONTROL_CONTINUE:
|
||||
break;
|
||||
case SERVICE_CONTROL_INTERROGATE:
|
||||
break;
|
||||
case SERVICE_CONTROL_SHUTDOWN:
|
||||
break;
|
||||
default:
|
||||
LogEvent("Bad service request");
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
void WINAPI CService::_ServiceMain(DWORD dwArgc, LPTSTR* lpszArgv)
|
||||
{
|
||||
_Module.ServiceMain();
|
||||
}
|
||||
void WINAPI CService::_Handler(DWORD dwOpcode)
|
||||
{
|
||||
_Module.Handler(dwOpcode);
|
||||
}
|
||||
|
||||
void CService::SetServiceStatus(DWORD dwState)
|
||||
{
|
||||
m_status.dwCurrentState = dwState;
|
||||
::SetServiceStatus(m_hServiceStatus, &m_status);
|
||||
}
|
||||
|
||||
void CService::Run()
|
||||
{
|
||||
LogEvent("Service started");
|
||||
m_dwThreadID = GetCurrentThreadId();
|
||||
|
||||
if (m_bService)
|
||||
SetServiceStatus(SERVICE_RUNNING);
|
||||
|
||||
// The service is running.
|
||||
|
||||
// TODO: Add code here
|
||||
|
||||
MSG msg;
|
||||
while (GetMessage(&msg,NULL,NULL,NULL))
|
||||
{
|
||||
TranslateMessage(&msg);
|
||||
DispatchMessage(&msg);
|
||||
}
|
||||
|
||||
|
||||
// The service is going to be stopped.
|
||||
|
||||
// TODO: Add code here
|
||||
}
|
||||
|
||||
BOOL CService::Install()
|
||||
{
|
||||
if (IsInstalled())
|
||||
return TRUE;
|
||||
|
||||
SC_HANDLE hSCM = ::OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
|
||||
if (hSCM == NULL)
|
||||
{
|
||||
MessageBox(NULL, "Couldn't open service manager", m_szServiceName, MB_OK);
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
// Get the executable file path
|
||||
TCHAR szFilePath[_MAX_PATH];
|
||||
::GetModuleFileName(NULL, szFilePath, _MAX_PATH);
|
||||
|
||||
SC_HANDLE hService = ::CreateService(
|
||||
hSCM, m_szServiceName, m_szServiceDisplayedName,
|
||||
SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS ,
|
||||
SERVICE_DEMAND_START, SERVICE_ERROR_NORMAL,
|
||||
szFilePath, NULL, NULL, NULL, NULL, NULL);
|
||||
|
||||
if (hService == NULL)
|
||||
{
|
||||
::CloseServiceHandle(hSCM);
|
||||
MessageBox(NULL, "Couldn't create service", m_szServiceName, MB_OK);
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
::CloseServiceHandle(hService);
|
||||
::CloseServiceHandle(hSCM);
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
BOOL CService::Uninstall()
|
||||
{
|
||||
if (!IsInstalled())
|
||||
return TRUE;
|
||||
|
||||
SC_HANDLE hSCM = ::OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
|
||||
|
||||
if (hSCM == NULL)
|
||||
{
|
||||
MessageBox(NULL, "Couldn't open service manager", m_szServiceName, MB_OK);
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
SC_HANDLE hService = ::OpenService(hSCM, m_szServiceName, SERVICE_STOP | DELETE);
|
||||
|
||||
if (hService == NULL)
|
||||
{
|
||||
::CloseServiceHandle(hSCM);
|
||||
MessageBox(NULL, "Couldn't open service", m_szServiceName, MB_OK);
|
||||
return FALSE;
|
||||
}
|
||||
SERVICE_STATUS status;
|
||||
::ControlService(hService, SERVICE_CONTROL_STOP, &status);
|
||||
|
||||
BOOL bDelete = ::DeleteService(hService);
|
||||
::CloseServiceHandle(hService);
|
||||
::CloseServiceHandle(hSCM);
|
||||
|
||||
if (bDelete)
|
||||
return TRUE;
|
||||
|
||||
MessageBox(NULL, "Service could not be deleted", m_szServiceName, MB_OK);
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
BOOL CService::IsInstalled()
|
||||
{
|
||||
BOOL bResult = FALSE;
|
||||
|
||||
SC_HANDLE hSCM = ::OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
|
||||
|
||||
if (hSCM != NULL)
|
||||
{
|
||||
SC_HANDLE hService = ::OpenService(hSCM, m_szServiceName, SERVICE_QUERY_CONFIG);
|
||||
if (hService != NULL)
|
||||
{
|
||||
bResult = TRUE;
|
||||
::CloseServiceHandle(hService);
|
||||
}
|
||||
::CloseServiceHandle(hSCM);
|
||||
}
|
||||
return bResult;
|
||||
}
|
||||
|
||||
///////////////////////////////////////////////////////////////////////////////////////
|
||||
// Logging functions
|
||||
void CService::LogEvent(LPCSTR pFormat, ...)
|
||||
{
|
||||
TCHAR chMsg[256];
|
||||
HANDLE hEventSource;
|
||||
LPTSTR lpszStrings[1];
|
||||
va_list pArg;
|
||||
|
||||
va_start(pArg, pFormat);
|
||||
vsprintf(chMsg, pFormat, pArg);
|
||||
va_end(pArg);
|
||||
|
||||
lpszStrings[0] = chMsg;
|
||||
|
||||
if (m_bService)
|
||||
{
|
||||
/* Get a handle to use with ReportEvent(). */
|
||||
hEventSource = RegisterEventSource(NULL, m_szServiceName);
|
||||
if (hEventSource != NULL)
|
||||
{
|
||||
/* Write to event log. */
|
||||
ReportEvent(hEventSource, EVENTLOG_INFORMATION_TYPE, 0, 0, NULL, 1, 0, (LPCTSTR*) &lpszStrings[0], NULL);
|
||||
DeregisterEventSource(hEventSource);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
// As we are not running as a service, just write the error to the console.
|
||||
printf(chMsg);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user