VC 创建服务程序

VC 创建服务程序

2007-04-29 17:56

#include "Windows.h" #include "Winsvc.h" #include "time.h" #include "stdio.h" SERVICE_STATUS m_ServiceStatus; SERVICE_STATUS_HANDLE m_ServiceStatusHandle; BOOL bRunning=true; void WINAPI ServiceMain(DWORD argc, LPTSTR *argv); void WINAPI ServiceCtrlHandler(DWORD Opcode); BOOL InstallService(); BOOL DeleteService(); void DoTask() { //    do something here; } void WINAPI ServiceMain(DWORD argc, LPTSTR *argv) { //     DWORD status; //     DWORD specificError;       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_ServiceStatus.dwServiceType     = SERVICE_AUTO_START;       m_ServiceStatusHandle = RegisterServiceCtrlHandler("Service2",ServiceCtrlHandler);         if (m_ServiceStatusHandle == (SERVICE_STATUS_HANDLE)0)       {           return;       }            m_ServiceStatus.dwCurrentState         = SERVICE_RUNNING;       m_ServiceStatus.dwCheckPoint           = 0;       m_ServiceStatus.dwWaitHint             = 0;         if (!SetServiceStatus (m_ServiceStatusHandle, &m_ServiceStatus))       {       } bRunning=true; while(bRunning) {        Sleep(3000);     //Place Your Code for processing here....       DoTask();     //Sleep(60*1000); }       return; } void WINAPI ServiceCtrlHandler(DWORD Opcode) {       switch(Opcode)       {           case SERVICE_CONTROL_PAUSE:               m_ServiceStatus.dwCurrentState = SERVICE_PAUSED;               break;           case SERVICE_CONTROL_CONTINUE:               m_ServiceStatus.dwCurrentState = SERVICE_RUNNING;               break;           case SERVICE_CONTROL_STOP:               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; } BOOL InstallService() { char strDir[1024]; HANDLE schSCManager,schService; GetCurrentDirectory(1024,strDir); strcat(strDir,"//Debug//servicetest.exe"); schSCManager = OpenSCManager(NULL,NULL,SC_MANAGER_ALL_ACCESS);   if (schSCManager == NULL)     return false;       LPCTSTR lpszBinaryPathName=strDir;       schService = CreateService((struct SC_HANDLE__ *)schSCManager,"Service2","MB Service",             // service name to display           SERVICE_ALL_ACCESS,          // desired access           SERVICE_WIN32_OWN_PROCESS, // service type           SERVICE_DEMAND_START,        // start type           SERVICE_ERROR_NORMAL,        // error control type           lpszBinaryPathName,          // service's binary           NULL,                        // no load ordering group           NULL,                        // no tag identifier           NULL,                        // no dependencies           NULL,                        // LocalSystem account           NULL);                       // no password       if (schService == NULL)           return false;         CloseServiceHandle((struct SC_HANDLE__ *)schService); return true; } BOOL DeleteService() { HANDLE schSCManager; SC_HANDLE hService; schSCManager = OpenSCManager(NULL,NULL,SC_MANAGER_ALL_ACCESS); if (schSCManager == NULL)     return false; hService=OpenService((struct SC_HANDLE__ *)schSCManager,"Service2",SERVICE_ALL_ACCESS); if (hService == NULL)     return false; if(DeleteService(hService)==0)     return false; if(CloseServiceHandle(hService)==0)     return false; else     return true; }   int main(int argc, char* argv[]) { if(argc>1) {     if(strcmp(argv[1],"-i")==0) {      if(InstallService())       printf("/nMB Service Installed Sucessfully/n");      else       printf("/nMB Service has been installed/n");     } else if(strcmp(argv[1],"-d")==0) {      if(DeleteService())       printf("/nMB Service UnInstalled Sucessfully/n");      else       printf("/nInstalled Easin Central Service Not Found/n");     } else {      printf("/nUnknown Switch Usage/nFor Install use Servicetest -i/nFor UnInstall use Servicetest -d/n");     } } else {     SERVICE_TABLE_ENTRY DispatchTable[]={{"Service2",ServiceMain},{NULL,NULL}};       StartServiceCtrlDispatcher(DispatchTable); } return 0; }