zoukankan      html  css  js  c++  java

  • VC Windows系统服务创建代码

    Windows系统服务创建步骤,常用类封装,废话不多说,直接上代码

     1 // ServceDemo.cpp : 定义控制台应用程序的入口点。
 2 //
 3 
 4 #include "stdafx.h"
 5 #include "WindwosService.h"
 6 
 7 unsigned __stdcall SvcRun (void *Params )
 8 {
 9     while(1)
10     {
11 
12         for (int i = 0;i < 100;i++)
13         {
14             printf("%d
",i);
15         }
16 
17         ::Sleep(1*60*1000);
18     }
19     return 0;
20 }
21 
22 int _tmain(int argc, _TCHAR* argv[])
23 {
24 #ifdef WIN32
25     InitService(argc,argv);
26 #endif // WIN32
27 
28 #ifdef linux
29 
30 #endif // linux
31 
32     return 0;
33 }
    main函数文件代码
     1 #if !defined JX_About_Privilege
 2 #define JX_About_Privilege
 3 
 4 #include <Windows.h>
 5 // Windows提权类
 6 class JXAboutPrivilege
 7 {
 8 public:
 9     JXAboutPrivilege(void);
10     ~JXAboutPrivilege(void);
11     // 提升关机权限
12     static BOOL EnableShutdownPrivileges( VOID );
13     // 提升结束进程的权限
14     static BOOL EnableTermProcessPrivileges( VOID );
15     // 提升加载驱动的权限
16     static BOOL EnableLoadDriverPrivileges( VOID );
17 
18     static BOOL EnableDebugPrivilege(BOOL   fEnable);
19 };
20 
21 #endif
    JXAboutPrivilege.h
     1 #include "StdAfx.h"
 2 #include "JXAboutPrivilege.h"
 3 
 4 
 5 JXAboutPrivilege::JXAboutPrivilege(void)
 6 {
 7 }
 8 
 9 JXAboutPrivilege::~JXAboutPrivilege(void)
10 {
11 }
12 
13 BOOL JXAboutPrivilege::EnableShutdownPrivileges( VOID )
14 {
15     HANDLE hToken;
16     TOKEN_PRIVILEGES tkp;
17     BOOL Result = FALSE;
18 
19     if ( !OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken) )
20         return FALSE;
21 
22     LookupPrivilegeValue( NULL, SE_SHUTDOWN_NAME, &(tkp.Privileges[0].Luid) );
23     tkp.PrivilegeCount = 1;
24     tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
25 
26     AdjustTokenPrivileges( hToken, FALSE, &tkp, sizeof(TOKEN_PRIVILEGES), (PTOKEN_PRIVILEGES) NULL, 0 );
27     DWORD dwError = GetLastError();
28     if ( ERROR_SUCCESS == dwError )
29         Result = TRUE;
30 
31     CloseHandle( hToken );
32     return Result;
33 }
34 
35 BOOL JXAboutPrivilege::EnableTermProcessPrivileges( VOID )
36 {
37     HANDLE hToken;
38     TOKEN_PRIVILEGES tkp;
39     BOOL Result = FALSE;
40 
41     if ( !OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken) )
42         return FALSE;
43 
44     LookupPrivilegeValue( NULL, SE_DEBUG_NAME, &(tkp.Privileges[0].Luid) );
45     tkp.PrivilegeCount = 1;
46     tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
47 
48     AdjustTokenPrivileges( hToken, FALSE, &tkp, sizeof(TOKEN_PRIVILEGES), (PTOKEN_PRIVILEGES) NULL, 0 );
49     DWORD dwError = GetLastError();
50     if ( ERROR_SUCCESS == dwError )
51         Result = TRUE;
52 
53     CloseHandle( hToken );
54     return Result;
55 }
56 
57 BOOL JXAboutPrivilege::EnableLoadDriverPrivileges( VOID )
58 {
59     HANDLE hToken;
60     TOKEN_PRIVILEGES tkp;
61     BOOL Result = FALSE;
62 
63     if ( !OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken) )
64         return FALSE;
65 
66     LookupPrivilegeValue( NULL, SE_LOAD_DRIVER_NAME, &(tkp.Privileges[0].Luid) );
67     tkp.PrivilegeCount = 1;
68     tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
69 
70     AdjustTokenPrivileges( hToken, FALSE, &tkp, sizeof(TOKEN_PRIVILEGES), (PTOKEN_PRIVILEGES) NULL, 0 );
71     DWORD dwError = GetLastError();
72     if ( ERROR_SUCCESS == dwError )
73         Result = TRUE;
74 
75     CloseHandle( hToken );
76     return Result;
77 }
78 
79 BOOL JXAboutPrivilege::EnableDebugPrivilege(BOOL   fEnable)
80 {
81     BOOL   fOk   =   FALSE; //   Assume   function   fails
82     HANDLE   hToken;
83     //   Try   to   open   this   process's   access   token
84     if(OpenProcessToken(GetCurrentProcess(),   TOKEN_ADJUST_PRIVILEGES,   &hToken))
85     {
86         //   Attempt   to   modify   the   "Debug"   privilege
87         TOKEN_PRIVILEGES   tp;
88         tp.PrivilegeCount   =   1;
89         LookupPrivilegeValue(NULL,   SE_DEBUG_NAME,   &tp.Privileges[0].Luid);
90         tp.Privileges[0].Attributes   =   fEnable   ?   SE_PRIVILEGE_ENABLED   :   0;
91         AdjustTokenPrivileges(hToken,   FALSE,   &tp,   sizeof(tp),   NULL,   NULL);
92         fOk   =   (GetLastError()   ==   ERROR_SUCCESS);
93         CloseHandle(hToken);
94     }
95     return(fOk);
96 }
    JXAboutPrivilege.cpp
     1 #if !defined JX_Service_Control
 2 #define  JX_Service_Control
 3 #include <Windows.h>
 4 
 5 // Windows服务控制类
 6 class JXServiceControl
 7 {
 8 public:
 9     JXServiceControl(void);
10     ~JXServiceControl(void);
11 public:
12     int ChangeServicePath( );
13     int createService();
14     int deleteService();
15     void startService();
16     void stopService();
17     BOOL isSvsRuning();
18     BOOL isSvsExist();
19     //设置服务名
20     void SetServiceName(const char * str);
21     //设置服务管理面板显示名字
22     void SetServiceShowName(const char *str);
23     //设置服务EXE文件位置
24     void SetExePath(const char *str);
25 private:
26     //服务名
27     char sSname[MAX_PATH+1];
28     //服务Caption
29     char sShowName[MAX_PATH+1];
30     //服务可执行文件路径
31     char sSvcExePath[MAX_PATH+1];
32 };
33 
34 
35 
36 #endif
    JXServiceControl.h
    WindwosService.h
      1 #include "WindwosService.h"
  2 
  3 
  4 SERVICE_STATUS ServiceStatus = {0};
  5 SERVICE_STATUS_HANDLE hStatus = NULL;
  6 
  7 void Run()
  8 {
  9     _beginthreadex(NULL, NULL, SvcRun, NULL, NULL, NULL);
 10 }
 11 
 12 void WINAPI ControlHandler(DWORD fdwControl)
 13 {
 14     switch(fdwControl)
 15     {
 16     case SERVICE_CONTROL_STOP:
 17     case SERVICE_CONTROL_SHUTDOWN:
 18         ServiceStatus.dwWin32ExitCode = 0;
 19         ServiceStatus.dwCurrentState = SERVICE_STOPPED;
 20     default: break;
 21     }
 22     SetServiceStatus(hStatus, &ServiceStatus);
 23 }
 24 
 25 void WINAPI ServiceMain(DWORD dwArgc, LPTSTR* lpszArgv)
 26 {
 27 
 28     ZeroMemory(&ServiceStatus, sizeof(ServiceStatus));
 29     ServiceStatus.dwServiceType = SERVICE_WIN32;
 30     ServiceStatus.dwCurrentState = SERVICE_START_PENDING;// 正在启动中... 
 31     ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
 32 
 33     hStatus = RegisterServiceCtrlHandler(SERVER_NAME, (LPHANDLER_FUNCTION)ControlHandler); // 注册当前服务的控制函数
 34     if (!hStatus)
 35     {
 36         OutputDebugString(" RegisterServiceCtrlHandler 失败....
");
 37         return;
 38     }
 39     ServiceStatus.dwCurrentState = SERVICE_RUNNING;
 40     SetServiceStatus(hStatus, &ServiceStatus);// 修改服务状态
 41     //////////////////////下方为启动主线程/////////////////////////////////////////
 42     Run();
 43     //////////////////////////////////////////////////////////////////////////
 44     while(1)
 45     {
 46         ::Sleep(1000);
 47     }
 48     //::WaitForSingleObject(hThread,INFINITE); // 等待线程结束
 49     ServiceStatus.dwCurrentState = SERVICE_STOPPED;
 50     ServiceStatus.dwWin32ExitCode = -1;
 51     SetServiceStatus(hStatus, &ServiceStatus);
 52 }
 53 
 54 int InitService(int argc, TCHAR* argv[])
 55 {
 56     JXAboutPrivilege::EnableDebugPrivilege(TRUE);
 57     JXAboutPrivilege::EnableLoadDriverPrivileges();
 58     JXAboutPrivilege::EnableShutdownPrivileges();
 59     JXAboutPrivilege::EnableTermProcessPrivileges();
 60 
 61     if (argc == 2)
 62     {
 63         if (!stricmp(argv[1],("-install")))
 64         {
 65             // InstallServices();
 66             return 1;
 67         }
 68         if (!stricmp(argv[1], ("-uninstall")))
 69         {
 70             // UnInstallServices();
 71             return 1;
 72         }
 73         if (!stricmp(argv[1], ("-run")))
 74         {
 75             Run();
 76             while (TRUE)
 77                 Sleep(50);
 78             return 1;
 79         }
 80     }
 81         SERVICE_TABLE_ENTRY DispatchTable[] = { 
 82             { SERVER_NAME, (LPSERVICE_MAIN_FUNCTION)ServiceMain }, 
 83             { NULL, NULL }
 84         };
 85 
 86         StartServiceCtrlDispatcher(DispatchTable);  
 87 }
 88 
 89 void InstallServices()
 90 {
 91     TCHAR path_nor[MAX_PATH] = { 0 };
 92     GetModuleFileName(NULL, path_nor, MAX_PATH);
 93 
 94     JXServiceControl sc;
 95     sc.SetExePath(path_nor);
 96     sc.SetServiceName((SERVER_NAME));
 97     sc.SetServiceShowName((SERVER_NAME));
 98 
 99 
100     if ( sc.isSvsExist() && sc.isSvsRuning()  )
101     {
102         sc.stopService();
103         sc.ChangeServicePath();
104     }
105     else if ( sc.isSvsExist() )
106     {
107         sc.ChangeServicePath();
108     }
109     else if (!sc.isSvsExist())
110     {
111         sc.createService();
112     }
113 
114     sc.startService();
115 }
116 
117 void UnInstallServices()
118 {
119     TCHAR path_nor[MAX_PATH] = { 0 };
120     GetModuleFileName(NULL, path_nor, MAX_PATH);
121 
122     JXServiceControl sc;
123     sc.SetExePath(path_nor);
124     sc.SetServiceName((SERVER_NAME));
125     sc.SetServiceShowName((SERVER_NAME));
126 
127 
128     if ( sc.isSvsExist() && sc.isSvsRuning() )
129     {
130         sc.stopService();
131     }
132     if( sc.isSvsExist() )
133         sc.deleteService();
134 }
    WindwosService.cpp
      1 #include "StdAfx.h"
  2 #include "JXServiceControl.h"
  3 
  4 
  5 JXServiceControl::JXServiceControl(void)
  6 {
  7     memset(sSname,0,MAX_PATH);
  8     memset(sShowName,0,MAX_PATH);
  9     memset(sSvcExePath,0,MAX_PATH);
 10 }
 11 
 12 JXServiceControl::~JXServiceControl(void)
 13 {
 14 
 15 }
 16 
 17 int JXServiceControl::ChangeServicePath()
 18 {
 19     SC_HANDLE scHandle = OpenSCManager(NULL,NULL,SC_MANAGER_CONNECT);
 20     SC_HANDLE hsev = NULL;
 21     if(scHandle != NULL)
 22     {
 23         hsev = OpenService(scHandle,sSname,SERVICE_ALL_ACCESS);
 24         if(hsev != NULL)
 25         {
 26             BOOL bRet = ChangeServiceConfig( 
 27                 hsev, 
 28                 SERVICE_WIN32_OWN_PROCESS|SERVICE_INTERACTIVE_PROCESS,
 29                 SERVICE_AUTO_START,
 30                 SERVICE_ERROR_IGNORE,
 31                 sSvcExePath,
 32                 NULL,
 33                 NULL,
 34                 NULL,
 35                 NULL,
 36                 NULL,
 37                 sShowName
 38                 );
 39             CloseServiceHandle(hsev);
 40 
 41         }
 42         else 
 43         {
 44             CloseServiceHandle(scHandle);
 45             return 0;                
 46         }
 47         CloseServiceHandle(scHandle);
 48     }
 49     else    
 50         return -1;
 51 
 52     return 1;
 53 }
 54 
 55 int JXServiceControl::createService()
 56 {
 57     SC_HANDLE scHandle = OpenSCManager(NULL,NULL,SC_MANAGER_CREATE_SERVICE);
 58     if (scHandle == NULL)
 59     {
 60         return -1;
 61     }
 62     SC_HANDLE hsvc = CreateService(scHandle,sSname,sShowName,SERVICE_ALL_ACCESS,
 63         SERVICE_WIN32_OWN_PROCESS|SERVICE_INTERACTIVE_PROCESS,SERVICE_AUTO_START,
 64         SERVICE_ERROR_IGNORE,sSvcExePath,
 65         NULL,NULL,NULL,NULL,NULL);
 66     if (hsvc != NULL)
 67     {
 68         CloseServiceHandle(hsvc);
 69     }
 70     else
 71     {
 72         return 0;
 73     }
 74     CloseServiceHandle(scHandle);
 75     return 1;
 76 }
 77 
 78 int JXServiceControl::deleteService()
 79 {
 80     SC_HANDLE scHandle = OpenSCManager(NULL,NULL,SC_MANAGER_CONNECT);
 81     SC_HANDLE hsev = NULL;
 82     if(scHandle != NULL)
 83     {
 84         hsev = OpenService(scHandle,sSname,SERVICE_ALL_ACCESS);
 85         SERVICE_STATUS mss;
 86         if(hsev != NULL)
 87         {
 88             QueryServiceStatus(hsev,&mss);
 89             if (mss.dwCurrentState != SERVICE_RUNNING)
 90             {
 91                 ControlService(hsev,SERVICE_CONTROL_STOP,&mss);
 92                 DeleteService(hsev);
 93 
 94             }
 95             CloseServiceHandle(hsev);
 96         } else 
 97         {
 98             CloseServiceHandle(scHandle);
 99             return 0;
100         }
101         CloseServiceHandle(scHandle);
102     }
103     else    return -1;
104     return 1;
105 }
106 
107 void JXServiceControl::startService()
108 {
109     SC_HANDLE scHandle = OpenSCManager(NULL,NULL,SC_MANAGER_CONNECT);
110     SC_HANDLE hscv = NULL;
111     if (scHandle != NULL)
112     {
113         hscv = OpenService(scHandle,sSname,SERVICE_START);
114         if(hscv != NULL)
115         {
116             BOOL isSucess = StartService(hscv,0,NULL);
117             DWORD err = 0;
118             if (isSucess  == 0)
119             {
120                 err = GetLastError();
121             }
122             CloseServiceHandle(hscv);
123         }
124         CloseServiceHandle(scHandle);
125     }
126 }
127 
128 void JXServiceControl::stopService()
129 {
130     SC_HANDLE scHandle = OpenSCManager(NULL,NULL,SC_MANAGER_CONNECT);
131     SC_HANDLE hsev = NULL;
132     if(scHandle != NULL)
133     {
134         hsev = OpenService(scHandle,sSname,SERVICE_ALL_ACCESS);
135         SERVICE_STATUS mss;
136         if(hsev != NULL)
137         {
138             QueryServiceStatus(hsev,&mss);
139             if (mss.dwCurrentState == SERVICE_RUNNING)
140             {
141                 ControlService(hsev,SERVICE_CONTROL_STOP,&mss);
142                 CloseServiceHandle(hsev);
143             }                
144         }
145         CloseServiceHandle(scHandle);
146     }
147 }
148 
149 BOOL JXServiceControl::isSvsRuning()
150 {
151     BOOL  ret = FALSE;
152     SC_HANDLE scHandle = OpenSCManager(NULL,NULL,SC_MANAGER_CONNECT);
153     SC_HANDLE hsev = NULL;
154     if(scHandle != NULL)
155     {
156         hsev = OpenService(scHandle,sSname,SERVICE_ALL_ACCESS);
157         SERVICE_STATUS mss;
158         if(hsev != NULL)
159         {
160             QueryServiceStatus(hsev,&mss);
161             if (mss.dwCurrentState == SERVICE_RUNNING)
162             {
163                 ret = TRUE;
164                 CloseServiceHandle(hsev);
165             }
166 
167         }
168         CloseServiceHandle(scHandle);
169     }
170     return ret;
171 }
172 
173 BOOL JXServiceControl::isSvsExist()
174 {
175     BOOL ret = FALSE;
176     SC_HANDLE scHandle = OpenSCManager(NULL,NULL,SC_MANAGER_CONNECT);
177     SC_HANDLE hsev = NULL;
178     if(scHandle != NULL)
179     {
180         hsev = OpenService(scHandle,sSname,SERVICE_ALL_ACCESS);
181         DWORD err = GetLastError();
182         if (err == ERROR_SERVICE_DOES_NOT_EXIST)
183         {
184             ret = FALSE;
185         }
186         else
187         {
188             ret = TRUE;
189         }
190         if(hsev != NULL)
191         {
192             CloseServiceHandle(hsev);
193             CloseServiceHandle(scHandle);
194         }
195     }
196     return ret;
197 }
198 
199 void JXServiceControl::SetServiceName(const char * str)
200 {
201     strcpy_s(sSname,MAX_PATH,str);
202 }
203 
204 void JXServiceControl::SetServiceShowName(const char *str)
205 {
206     strcpy_s(sShowName,MAX_PATH,str);
207 }
208 
209 void JXServiceControl::SetExePath(const char *str)
210 {
211     strcpy_s(sSvcExePath,MAX_PATH,str);
212 }
    JXServiceControl.cpp

    转载请注明出处....

    代码栗子：http://pan.baidu.com/s/1WHQ5w
  • 相关阅读:
    Serialization and deserialization are bottlenecks in parallel and distributed computing, especially in machine learning applications with large objects and large quantities of data.
    Introduction to the Standard Directory Layout
    import 原理 及 导入 自定义、第三方 包
    403 'Forbidden'
    https://bitbucket.org/ariya/phantomjs/downloads/phantomjs-2.1.1-linux-x86_64.tar.bz2
    These interactions can be expressed as complicated, large scale graphs. Mining data requires a distributed data processing engine
    mysqldump --flush-logs
    mysql dump 参数
    mysql dump 参数
    如果是在有master上开启了该参数，记得在slave端也要开启这个参数（salve需要stop后再重新start），否则在master上创建函数会导致replaction中断。
  • 原文地址：https://www.cnblogs.com/-jpp/p/5250120.html
Copyright © 2011-2022 走看看