自己定制windows服务

来源:互联网 发布:淘宝直通车盈亏平衡值 编辑:程序博客网 时间:2024/04/29 14:29

#include "stdafx.h"
#include "Windows.h"
#include "Winsvc.h"
#include "time.h"
#include <TCHAR.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();
FILE *fp;
int main(int argc, char* argv[])
{
  fp=fopen("d://chjtest","w");
// DeleteService();
//InstallService();
// return 0;


  if(argc>1)
  {
    if(strcmp(argv[1],"i")==0)
    {
      if(InstallService())
        printf("/n/nService Installed Sucessfully/n");
      else
        printf("/n/nError Installing Service/n");
    }
    if(strcmp(argv[1],"d")==0)
    {
      if(DeleteService())
        printf("/n/nService UnInstalled Sucessfully/n");
      else
        printf("/n/nError UnInstalling Service/n");
    }
    else
    {
      printf("/n/nUnknown Switch Usage/n/nFor Install       use Srv1 -i/n/nFor UnInstall use Srv1 -d/n");
    }
  }
  else
  {
    SERVICE_TABLE_ENTRY DispatchTable[]=
                {{"Service1",ServiceMain},{NULL,NULL}};
    StartServiceCtrlDispatcher(DispatchTable);
  }
  return 0;
}

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_ServiceStatusHandle = RegisterServiceCtrlHandler("Service1",
                                            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);
  
 putc('c',fp);
  }
  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;
   fclose(fp);

      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,"//test1.exe");
  schSCManager = OpenSCManager(NULL,NULL,SC_MANAGER_ALL_ACCESS);

  if (schSCManager == NULL)  
  return false;
  LPCTSTR lpszBinaryPathName=strDir;

  schService = CreateService(schSCManager,"Service1",
        "my测试的服务程序", // 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)
    {
        printf("CreateService failed (%d)/n", GetLastError());
    return false;
  }
  // Create service
//SC_HANDLE hService = NULL;
//hService = ::CreateService(...);
// Set service description
//服务的描述信息
SERVICE_DESCRIPTION ServiceDesc;
static TCHAR szDescription[MAX_PATH];
_tcscpy(szDescription, _T("Your description..."));
ServiceDesc.lpDescription = szDescription;
::ChangeServiceConfig2(schService, SERVICE_CONFIG_DESCRIPTION, &ServiceDesc);
  CloseServiceHandle(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(schSCManager,"Service1",SERVICE_ALL_ACCESS);
  if (hService == NULL)
    return false;
  if(DeleteService(hService)==0)
    return false;
  if(CloseServiceHandle(hService)==0)
    return false;

return true;

原创粉丝点击