WIN32编程--用Windows API实现Windows NT服务的程序框架

//点击此处下载源码

#include <windows.h>
#include <stdio.h>

//变量声明
char ServiceName[]="NTService";
SERVICE_STATUS ssStatus;
SERVICE_STATUS_HANDLE ServiceStatusHandle;

//前向声明
void ShowHelp();
void InsallService();
void RemoveService();
void WINAPI ServiceMain(DWORD dwArgc,LPTSTR* lpszArgv);
void WINAPI ControlHandler(DWORD dwCtrlCode);

//主程序入口
int main(int argc,char** argv)
{
//未带参数
if(argc==1)
{
SERVICE_TABLE_ENTRY ServiceTableEntry[]={{ServiceName,ServiceMain},{NULL,NULL}};
if(!StartServiceCtrlDispatcher(ServiceTableEntry))
{
printf("StartServiceCtrlDispatcher failed!\n");
return 1;
}
return 0;
}

//参数不正确
if(argc!=2)
{
ShowHelp();
return 0;
}

//参数可识别

//显示帮助
if(!strcmp("/?",argv[1]))
{
ShowHelp();
return 0;
}
//安装服务
if(!strcmp("/i",argv[1]))
{
InsallService();
return 0;
}
//移除服务
if(!strcmp("/r",argv[1]))
{
RemoveService();
return 0;
}

//参数不可识别
ShowHelp();

return 0;
}

//显示帮助
void ShowHelp()
{
//显示参数提示
printf("parameters:\n");
printf("/? show this help\n");
printf("/i install the service\n");
printf("/r remove the service\n");
}

//安装服务
void InsallService()
{
SC_HANDLE schService,schSCManager;
char FileName[256],DisplayName[]="NT Service";
//获取文件名
if (GetModuleFileName(0,FileName,256)==0) return;
//打开服务管理器
schSCManager=OpenSCManager(NULL,NULL,SC_MANAGER_ALL_ACCESS);
if (schSCManager==0)
{
printf("Open SCManager failed!\n");
return;
}
//创建服务
schService=CreateService(schSCManager,ServiceName,DisplayName,SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS, SERVICE_DEMAND_START,
SERVICE_ERROR_NORMAL,FileName,NULL,NULL,NULL,NULL,NULL);
if (schService>0)
{
printf("Service %s installed!\n",ServiceName);
//关闭服务
CloseServiceHandle(schService);
}
else
{
printf("Service %s install failed!\n",ServiceName);
}
//关闭服务控制器
CloseServiceHandle(schSCManager);
}

//移除服务
void RemoveService()
{
SC_HANDLE schService,schSCManager;
//打开服务管理器
schSCManager=OpenSCManager(NULL,NULL,SC_MANAGER_ALL_ACCESS);
if (schSCManager==0)
{
printf("Open SCManager failed!\n");
return;
}
//打开服务
schService=OpenService(schSCManager,ServiceName, SERVICE_ALL_ACCESS);
if (schService==0)
{
printf("Cann't open service!\n");
//关闭服务
CloseServiceHandle(schSCManager);
return;
}
else
{
//停止服务
if(ControlService(schService, SERVICE_CONTROL_STOP, &ssStatus))
{
//查询服务状态
while(QueryServiceStatus(schService, &ssStatus))
{
if(ssStatus.dwCurrentState==SERVICE_STOP_PENDING) printf("%s",".");
else break;
}
if(ssStatus.dwCurrentState==SERVICE_STOPPED) printf("Service Stoped Now!\n");
else
{
printf("Service stop failed!\n");
return;
}
}
//删除服务
if(DeleteService(schService))
{
printf("Service removed!\n");
}
else
{
printf("Service remove failed!\n");
}
//关闭服务
CloseServiceHandle(schService);
}
//关闭控制器
CloseServiceHandle(schSCManager);
}

//服务程序入口
void WINAPI ServiceMain(DWORD dwArgc,LPTSTR* lpszArgv)
{
//设置初始状态
ssStatus.dwServiceType=SERVICE_WIN32;
ssStatus.dwControlsAccepted=SERVICE_ACCEPT_STOP|SERVICE_ACCEPT_PAUSE_CONTINUE;
ssStatus.dwCurrentState=SERVICE_START_PENDING;

//注册服务
ServiceStatusHandle=RegisterServiceCtrlHandler(ServiceName, ControlHandler);
if(ServiceStatusHandle==0) return;

//设置服务状态
if(!SetServiceStatus(ServiceStatusHandle,&ssStatus))
{
//服务停止
ssStatus.dwCurrentState=SERVICE_STOPPED;
SetServiceStatus(ServiceStatusHandle,&ssStatus);
}

//服务运行
ssStatus.dwCurrentState=SERVICE_RUNNING;
SetServiceStatus(ServiceStatusHandle,&ssStatus);
}

//服务控制器
void WINAPI ControlHandler(DWORD dwCtrlCode)
{
//识别控制码
switch(dwCtrlCode)
{
//暂停
case SERVICE_CONTROL_PAUSE:
ssStatus.dwCurrentState=SERVICE_PAUSED;
break;
//继续
case SERVICE_CONTROL_CONTINUE:
ssStatus.dwCurrentState=SERVICE_RUNNING;
break;
//停止
case SERVICE_CONTROL_STOP:
ssStatus.dwCurrentState=SERVICE_STOPPED;
SetServiceStatus(ServiceStatusHandle,&ssStatus);
break;
}

//设置服务状态
SetServiceStatus(ServiceStatusHandle,&ssStatus);

return;
}

注：各函数的详细说明请参考MSDN。