服务框架源码

#include < stdio.h >
#include < windows.h >
SERVICE_STATUSm_ServiceStatus;
SERVICE_STATUS_HANDLEm_ServiceStatusHandle;
BOOLbRunning = true ;
void WINAPIServiceMain(DWORDargc,LPTSTR * argv); // 服务主函数
void WINAPIServiceCtrlHandler(DWORDOpcode); // 服务控制函数
void WINAPICmdStart( void ); // 要启动的程序函数
BOOLInstallService(); // 安装服务的函数
BOOLDeleteService(); // 删除服务的函数

int main( int argc, char * argv[])
{
printf( " windowsbasedservicedemo " );
printf( " [email protected] " );
if (argc != 3 )
{
printf( " usage:%s-install[remove] " ,argv[ 0 ]);
return 0 ;
}
if (strcmp(argv[ 1 ], " -install " ) == 0 ) /// install
{
if (InstallService())
printf( " ServiceInstalledSucessfully " );
else
printf( " ErrorInstallingService " );
}
else if (strcmp(argv[ 1 ], " -remove " ) == 0 ) // remove
{
if (DeleteService())
printf( " Serviceremovesucessfully " );
else
printf( " ErrorremovingService " );
}
else
{
printf( " usage:%s-install[remove] " ,argv[ 0 ]);
return 0 ;
}
// 在进入点函数里面要完成ServiceMain的初始化，
// 准确点说是初始化一个SERVICE_TABLE_ENTRY结构数组，
// 这个结构记录了这个服务程序里面所包含的所有服务的名称
// 和服务的进入点函数
SERVICE_TABLE_ENTRY
DispatchTable[] = {{ " WindowsMgr " ,ServiceMain},{NULL,NULL}};
// 最后的NULL指明数组的结束
StartServiceCtrlDispatcher(DispatchTable);
return 0 ;
}

void WINAPIServiceMain(DWORDargc,LPTSTR * argv)
{
m_ServiceStatus.dwServiceType = SERVICE_WIN32;
m_ServiceStatus.dwCurrentState = SERVICE_START_PENDING;
m_ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP;
m_ServiceStatus.dwWin32ExitCode = 0 ;
m_ServiceStatus.dwServiceSpecificExitCode = 0 ;
m_ServiceStatus.dwCheckPoint = 0 ;
m_ServiceStatus.dwWaitHint = 0 ;
m_ServiceStatusHandle = RegisterServiceCtrlHandler( " WindowsMgr " ,ServiceCtrlHandler);
if (m_ServiceStatusHandle == (SERVICE_STATUS_HANDLE) 0 ) return ;
m_ServiceStatus.dwCurrentState = SERVICE_RUNNING; // 设置服务状态
m_ServiceStatus.dwCheckPoint = 0 ;
m_ServiceStatus.dwWaitHint = 0 ;
// SERVICE_STATUS结构含有七个成员，它们反映服务的现行状态。
// 所有这些成员必须在这个结构被传递到SetServiceStatus之前正确的设置
SetServiceStatus(m_ServiceStatusHandle, & m_ServiceStatus);
bRunning = true ;
// *
CmdStart(); // 启动我们的服务程序
// *
return ;
}
void WINAPIServiceCtrlHandler(DWORDOpcode) // 服务控制函数
{
switch (Opcode)
{
case SERVICE_CONTROL_PAUSE: // weacceptthecommandtopauseit
m_ServiceStatus.dwCurrentState = SERVICE_PAUSED;
break ;
case SERVICE_CONTROL_CONTINUE: // wegotthecommandtocontinue
m_ServiceStatus.dwCurrentState = SERVICE_RUNNING;
break ;
case SERVICE_CONTROL_STOP: // wegotthecommandtostopthisservice
m_ServiceStatus.dwWin32ExitCode = 0 ;
m_ServiceStatus.dwCurrentState = SERVICE_STOPPED;
m_ServiceStatus.dwCheckPoint = 0 ;
m_ServiceStatus.dwWaitHint = 0 ;
SetServiceStatus(m_ServiceStatusHandle, & m_ServiceStatus);
bRunning = false ;
break ;
case SERVICE_CONTROL_INTERROGATE: //
break ;
}
return ;
}
BOOLInstallService() // 安装服务函数
{
char strDir[ 1024 ];
SC_HANDLEschSCManager,schService;
GetCurrentDirectory( 1024 ,strDir);
GetModuleFileName(NULL,strDir, sizeof (strDir));

char chSysPath[ 1024 ];
GetSystemDirectory(chSysPath, sizeof (chSysPath));

strcat(chSysPath, " \WindowsMgr.exe " );
if ( ! CopyFile(strDir,chSysPath,FALSE))printf( " CopyfileOK " ); // 把我们的服务程序复制到系统根目录

strcpy(strDir,chSysPath);
schSCManager = OpenSCManager(NULL,NULL,SC_MANAGER_ALL_ACCESS);
if (schSCManager == NULL)
{
printf( " openscmangerfailed,maybeyoudonothavetheprivilagetodothis " );
return false ;
}

LPCTSTRlpszBinaryPathName = strDir;

schService = CreateService(schSCManager, " WindowsMgr " , " WindowsMangerControl " , // 将服务的信息添加到SCM的数据库
SERVICE_ALL_ACCESS, // desiredaccess
SERVICE_WIN32_OWN_PROCESS, // servicetype
SERVICE_AUTO_START, // starttype
SERVICE_ERROR_NORMAL, // errorcontroltype
lpszBinaryPathName, // service'sbinary
NULL, // noloadorderinggroup
NULL, // notagidentifier
NULL, // nodependencies
NULL, // LocalSystemaccount
NULL); // nopassword

if (schService == NULL)
{
printf( " faint,wefailedjustbecauseweinvokecreateservicesfailed " );
return false ;
}
CloseServiceHandle(schService);
return true ;
}
BOOLDeleteService()
{
SC_HANDLEschSCManager;
SC_HANDLEhService;
schSCManager = OpenSCManager(NULL,NULL,SC_MANAGER_ALL_ACCESS);

char chSysPath[ 1024 ];
GetSystemDirectory(chSysPath, sizeof (chSysPath));
strcat(chSysPath, " \WindowsMgr.exe " );

if (schSCManager == NULL)
{
printf( " faint,openscmangerfailed " );
return false ;
}
hService = OpenService(schSCManager, " WindowsMgr " ,SERVICE_ALL_ACCESS);
if (hService == NULL)
{
printf( " faint,openservicesfailt " );
return false ;
}
if (DeleteFile(chSysPath) == 0 )
{
printf( " DellfileFailure! " );
return false ;
}
else printf( " DeletefileOK! " );
if (DeleteService(hService) == 0 )
return false ;

if (CloseServiceHandle(hService) == 0 )
return false ;
else
return true ;
}

void WINAPICmdStart( void )
{
// --------------------------------
// 把你的要做成服务启动的程序代码添加到这里
// 那么你的代码就可以作为NT服务启动了
// --------------------------------
}