/**
W32Server.h
http://www.cppblog.com/lovelypig 橙子
*/
#ifndef _CW32SERVER_H
#define _CW32SERVER_H
#include <windows.h>
#include <stdio.h>
class CW32Server
{
TCHAR m_szServerName[MAX_PATH];
SC_HANDLE m_hService;
SC_HANDLE m_hScm;
SERVICE_STATUS ServiceStatus;
BYTE m_cfgBuf[4096];
public:
CW32Server(void);
CW32Server(const TCHAR *strServerName);
BOOL Open(const TCHAR *strServerName);
/** 返回状态:
SERVICE_CONTINUE_PENDING The service continue is pending.
SERVICE_PAUSE_PENDING The service pause is pending.
SERVICE_PAUSED The service is paused.
SERVICE_RUNNING The service is running.
SERVICE_START_PENDING The service is starting.
SERVICE_STOP_PENDING The service is stopping.
SERVICE_STOPPED The service is not running.
*/
DWORD GetState();
BOOL SetState(DWORD state);
BOOL Start();
BOOL Stop();
/**
SERVICE_AUTO_START A service started automatically by the service control manager during system startup.
SERVICE_BOOT_START A device driver started by the system loader. This value is valid only for driver services.
SERVICE_DEMAND_START A service started by the service control manager when a process calls the StartService function.
SERVICE_DISABLED A service that cannot be started. Attempts to start the service result in the error code ERROR_SERVICE_DISABLED.
SERVICE_SYSTEM_START
*/
BOOL GetConfig(); //如果返回真，配置状态填充在 m_config 中
SC_HANDLE GetHandle(); //需要禁用、启动服务，使用 ChangeServiceConfig(GetHandle(),...) 查看MSDN
~CW32Server(void);
QUERY_SERVICE_CONFIG *m_config;
};
#endif
/**
W32Server.cpp
http://www.cppblog.com/lovelypig 橙子
*/
#include ".\w32server.h"
#include <assert.h>
CW32Server::CW32Server(void)
{
memset((char*)&m_szServerName,0,sizeof(TCHAR)*MAX_PATH);
memset((char*)&m_config,0,sizeof(m_config));
m_hService = 0;
m_hScm = 0;
m_config = (QUERY_SERVICE_CONFIG*)m_cfgBuf;
}
CW32Server::CW32Server(const TCHAR *strServerName)
{
assert(strServerName);
CW32Server();
_tcscpy(m_szServerName,strServerName);
m_hScm=OpenSCManager(0,0,SC_MANAGER_CREATE_SERVICE);
if(!m_hScm)
{
return ;
}
m_hService=OpenService(m_hScm,strServerName,SERVICE_ALL_ACCESS);
if(!m_hService)
{
CloseServiceHandle(m_hScm);
m_hScm = NULL;
}
}
CW32Server::~CW32Server(void)
{
if( m_hScm )
{
CloseServiceHandle(m_hScm);
m_hScm = NULL;
}
if( m_hService )
{
CloseServiceHandle(m_hService);
m_hService = NULL;
}
}
BOOL CW32Server:: Open(const TCHAR *strServerName)
{
assert(strServerName);
if( m_hScm )
{
CloseServiceHandle(m_hScm);
m_hScm = NULL;
}
if( m_hService )
{
CloseServiceHandle(m_hService);
m_hService = NULL;
}
_tcscpy(m_szServerName,strServerName);
m_hScm=OpenSCManager(0,0,SC_MANAGER_CREATE_SERVICE);
if(!m_hScm)
{
return FALSE;
}
m_hService=OpenService(m_hScm,strServerName,SERVICE_ALL_ACCESS);
if(!m_hService)
{
CloseServiceHandle(m_hScm);
m_hScm = NULL;
return FALSE;
}
return TRUE;
}
DWORD CW32Server::GetState()
{
assert(m_hService);
if(QueryServiceStatus(m_hService,&ServiceStatus))
{
return ServiceStatus.dwCurrentState;
}
else
{
return 0xffffffff;
}
}
BOOL CW32Server::SetState(DWORD state)
{
assert(m_hService);
return ControlService(m_hService,state,&ServiceStatus);
}
BOOL CW32Server::Start()
{
assert(m_hService);
return StartService(m_hService,0,NULL);
}
BOOL CW32Server::Stop()
{
assert(m_hService);
return ControlService(m_hService,SERVICE_CONTROL_STOP,&ServiceStatus);
}
BOOL CW32Server::GetConfig()
{
assert(m_hService);
DWORD cbBufSize = 4096;
DWORD pcbBytesNeeded = 4096;
return QueryServiceConfig(m_hService,m_config,cbBufSize, &pcbBytesNeeded);
}
SC_HANDLE CW32Server::GetHandle()
{
assert(m_hService);
return m_hService;
}