您的位置:首页 > 编程语言 > Java开发

NetBeans 6.0模块快速入门教程

2008-09-04 15:08 519 查看
1.操作服务类的头文件WindowsService.h:

#pragma once

class CWindowsService
{
public:
CWindowsService(void);
public:
~CWindowsService(void);
enum SEV_STATUS
{
SEV_ERROR= 0x00,
SEV_NO= 0x01,
SEV_HAVE= 0x02,
SEV_RUNING= 0x03,
SEV_STOPED= 0x04
};
public:
BOOL AddService(const char*pSourceName,const char*pServiceName,const char*pDisName,const char*pPara);
BOOL RemoveService(const char*pServiceName);
BYTE CheckServiceStatus(const char*pServiceName);
BOOL StartSevice(const char*pServiceName);
BOOL StopSevice(const char*pServiceName);
BOOL GetSevicePath(const char*pServiceName,CString &strServicePath);
BOOL GetCurPath(CString &strCurPath);
};

2.操作服务类的执行文件WindowsService.cpp:

/**********************************************************************************
windows 服务操作类
write by fjssharpsword
2010-12-23
************************************************************************************/
#include "StdAfx.h"
#include "WindowsService.h"
#include "winsvc.h"

CWindowsService::CWindowsService(void)
{
}

CWindowsService::~CWindowsService(void)
{
}

/*****************************************************
创建服务说明
如 m_Test.AddService("C:\\Test\\Test.exe","MyServiceName","DisplayName",NULL);
pPara 为参数有些服务需
*****************************************************/

BOOL CWindowsService::AddService(const char*pSourceName,const char*pServiceName,const char*pDisName,const char*pPara)
{
SC_HANDLE hSC = OpenSCManager( NULL, NULL, SC_MANAGER_CREATE_SERVICE );
if ( hSC == NULL )
{
return FALSE;//_ERROR_;
}
CString strBinnaryPathName;
strBinnaryPathName.Format("%s%s%s %s","\"",pSourceName,"\"",pPara);

SC_HANDLE hSvc = CreateService( hSC,
pServiceName, //服务名称
pDisName, //显示名称
SERVICE_ALL_ACCESS,
SERVICE_WIN32_OWN_PROCESS | SERVICE_INTERACTIVE_PROCESS,
SERVICE_AUTO_START,
SERVICE_ERROR_NORMAL,
strBinnaryPathName,
NULL,
NULL,
NULL,
NULL,
NULL );
if( hSvc == NULL )
{
::CloseServiceHandle( hSC );
return FALSE;//_ERROR_;
}
::CloseServiceHandle( hSvc );
::CloseServiceHandle( hSC );
return TRUE;
}

/*
**************************************************
删除服务
**************************************************
*/
BOOL CWindowsService::RemoveService(const char*pServiceName)
{
// UnInstall the XService from Service Control Manager Database
SC_HANDLE hSC = ::OpenSCManager( NULL,NULL, SC_MANAGER_ALL_ACCESS);
if( hSC == NULL)
{
return FALSE;
}

SC_HANDLE hSvc = ::OpenService( hSC, pServiceName, SERVICE_STOP | DELETE/*SERVICE_ALL_ACCESS*/);
if( hSvc == NULL)
{
::CloseServiceHandle( hSC);
return FALSE;
}

SERVICE_STATUS status;
::ControlService(hSvc, SERVICE_CONTROL_STOP, &status);

if ( !::DeleteService(hSvc))
{
DWORD _error =GetLastError();//ERROR_ACCESS_DENIED
return FALSE;
}

::CloseServiceHandle(hSvc);
::CloseServiceHandle(hSC);
return TRUE;
}

/*
**************************************************
检测状态
**************************************************
*/
BYTE CWindowsService::CheckServiceStatus(const char*pServiceName)
{
SC_HANDLE hSC = ::OpenSCManager( NULL,NULL, GENERIC_EXECUTE);
if( hSC == NULL)
{
return SEV_ERROR;
}
// 打开服务。
SC_HANDLE hSvc = ::OpenService( hSC, pServiceName, SERVICE_START | SERVICE_QUERY_STATUS | SERVICE_STOP);
if( hSvc == NULL)
{
::CloseServiceHandle( hSC);
return SEV_NO;
}
// 获得服务的状态
SERVICE_STATUS status;
if( ::QueryServiceStatus( hSvc, &status) == FALSE)
{
// TRACE( "Get Service state error。");
::CloseServiceHandle( hSvc);
::CloseServiceHandle( hSC);
return SEV_ERROR;
}
if(status.dwCurrentState == SERVICE_RUNNING)
{
::CloseServiceHandle( hSvc);
::CloseServiceHandle( hSC);
return SEV_RUNING;
}
else if(status.dwCurrentState == SERVICE_STOPPED)
{
::CloseServiceHandle( hSvc);
::CloseServiceHandle( hSC);
return SEV_STOPED;
}
::CloseServiceHandle( hSvc);
::CloseServiceHandle( hSC);
return SEV_ERROR;
}
/*
**************************************************
启动服务
**************************************************
*/
BOOL CWindowsService::StartSevice(const char*pServiceName)
{
SC_HANDLE hSC = ::OpenSCManager( NULL,NULL, GENERIC_EXECUTE);
if( hSC == NULL)
{
return FALSE;
}
// 打开服务。
SC_HANDLE hSvc = ::OpenService( hSC, pServiceName, SERVICE_START | SERVICE_QUERY_STATUS | SERVICE_STOP);
if( hSvc == NULL)
{
::CloseServiceHandle( hSC);
return FALSE;
}
// 获得服务的状态
SERVICE_STATUS status;
if( ::QueryServiceStatus( hSvc, &status) == FALSE)
{
//TRACE( "Get Service state error。");
::CloseServiceHandle( hSvc);
::CloseServiceHandle( hSC);
return FALSE;
}

if( status.dwCurrentState != SERVICE_RUNNING) //如果处于停止状态则启动服务,否则返回
{
// 启动服务
if( ::StartService( hSvc, NULL, NULL) == FALSE)
{
//TRACE( "start service error。");
::CloseServiceHandle( hSvc);
::CloseServiceHandle( hSC);
return FALSE;
}
// 等待服务启动
while( ::QueryServiceStatus( hSvc, &status) == TRUE)
{
::Sleep(500);
QueryServiceStatus( hSvc, &status);
if( status.dwCurrentState == SERVICE_START_PENDING)
{
//AfxMessageBox( "start success。");
::CloseServiceHandle( hSvc);
::CloseServiceHandle( hSC);
return TRUE;
}
::Sleep( status.dwWaitHint);
QueryServiceStatus( hSvc, &status);
if( status.dwCurrentState == SERVICE_RUNNING)
{
//AfxMessageBox( "start success。");
::CloseServiceHandle( hSvc);
::CloseServiceHandle( hSC);
return TRUE;
}
else
{
::CloseServiceHandle( hSvc);
::CloseServiceHandle( hSC);
return FALSE;
}
}
}
::CloseServiceHandle( hSvc);
::CloseServiceHandle( hSC);
return TRUE;
}
/*
**************************************************
停止服务
**************************************************
*/
BOOL CWindowsService::StopSevice(const char*pServiceName)
{
SC_HANDLE hSC = ::OpenSCManager( NULL,NULL, GENERIC_EXECUTE);
if( hSC == NULL)
{
return FALSE;
}
// 打开服务。
SC_HANDLE hSvc = ::OpenService( hSC, pServiceName,
SERVICE_START | SERVICE_QUERY_STATUS | SERVICE_STOP);
if( hSvc == NULL)
{
::CloseServiceHandle( hSC);
return FALSE;
}
// 获得服务的状态
SERVICE_STATUS status;
if( ::QueryServiceStatus( hSvc, &status) == FALSE)
{
//TRACE( "Get Service state error。");
::CloseServiceHandle( hSvc);
::CloseServiceHandle( hSC);
return FALSE;
}
//如果处于停止状态则启动服务,否则停止服务。
if( status.dwCurrentState == SERVICE_RUNNING)
{
// 停止服务
if( ::ControlService( hSvc, SERVICE_CONTROL_STOP, &status) == FALSE)
{
//TRACE( "stop service error。");
::CloseServiceHandle( hSvc);
::CloseServiceHandle( hSC);
return FALSE;
}
// 等待服务停止
while( ::QueryServiceStatus( hSvc, &status) == TRUE)
{
//::Sleep( status.dwWaitHint);
::Sleep(1000);
QueryServiceStatus( hSvc, &status);
if( status.dwCurrentState == SERVICE_STOPPED)
{
//AfxMessageBox( "stop success。");
::CloseServiceHandle( hSvc);
::CloseServiceHandle( hSC);
return TRUE;
}
::Sleep(3000);
QueryServiceStatus( hSvc, &status);
if( status.dwCurrentState == SERVICE_STOPPED)
{
//AfxMessageBox( "stop success。");
::CloseServiceHandle( hSvc);
::CloseServiceHandle( hSC);
return TRUE;
}
::Sleep( status.dwWaitHint);
QueryServiceStatus( hSvc, &status);
if( status.dwCurrentState == SERVICE_STOPPED)
{
//AfxMessageBox( "stop success。");
::CloseServiceHandle( hSvc);
::CloseServiceHandle( hSC);
return TRUE;
}
else
{
::CloseServiceHandle( hSvc);
::CloseServiceHandle( hSC);
return FALSE;
}
}
}
return TRUE;
}

BOOL CWindowsService::GetSevicePath(const char*pServiceName,CString & strServicePath)
{
SC_HANDLE hSC = ::OpenSCManager( NULL,NULL, SC_MANAGER_ALL_ACCESS);
if( hSC == NULL)
{
return FALSE;
}
// 打开服务。
SC_HANDLE hSvc = ::OpenService( hSC, pServiceName,SERVICE_ALL_ACCESS);
if( hSvc == NULL)
{
::CloseServiceHandle( hSC);
return FALSE;
}
LPQUERY_SERVICE_CONFIG pServiceConfig;
DWORD cbBufSize = 1024*8;//sizeof(SERVICE_DESCRIPTION_STRUCT);

DWORD cbBytesNeeded;
CString strPath;
pServiceConfig = (LPQUERY_SERVICE_CONFIG)LocalAlloc(LPTR, cbBufSize); //注意分配足够的空间
QueryServiceConfig( hSvc, pServiceConfig ,cbBufSize, &cbBytesNeeded);
strPath = pServiceConfig->lpBinaryPathName;
strPath.Replace('"',' ');
//int iFind = strPath.Find("\\\\");
//strPath = strPath.Left(iFind+1);
int iFind = strPath.Find("exe");
strPath = strPath.Left(iFind+3);
strServicePath = strPath;
::LocalFree(pServiceConfig);
::CloseServiceHandle( hSvc);
::CloseServiceHandle( hSC);

return TRUE;
}

BOOL CWindowsService::GetCurPath(CString &strCurPath)
{
char szTemp[MAX_PATH] = {0};
::GetModuleFileName(NULL, szTemp, sizeof(szTemp)-1);
int iLen = (int)strlen(szTemp);
for(int i = iLen - 1; i >= 0; i--)
{
if (szTemp[i] == '\\')
{
break;
}
else
{
szTemp[i] = 0;
}
}
strCurPath = szTemp;
return TRUE;
}
3.调用案例:

#include "WindowsService.h"
void CDemoSvrDlg::OnOK()
{
// TODO: Add extra validation here
CWindowsService ws;
ws.AddService("D:\\demo.exe","demo","demo",NULL);//创建

ws.StartSevice("demo");//启动

if(ws.CheckServiceStatus("demo") == 0x03)
AfxMessageBox("启动状态");

ws.StopSevice("demo");

if(ws.CheckServiceStatus("demo") == 0x04)
AfxMessageBox("停止状态");

ws.RemoveService("demo");//删除
//CDialog::OnOK();
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: