Win32 Service的简单封装

前段时间学习Windows程序设计，刚好学到Win32 Service，于是写了两个简单的类：BaseService和ServiceCtrl。虽然功能比较简单，但是也能适用于大多数情况。下面介绍一下简单用法，如果你刚好需要写一些简单的服务程序，这两个类也许能派上用场：

1. BaseService
BaseService.h

1
#ifndef BASE_SERVICE_H

2
#define BASE_SERVICE_H

3

4

class
BaseService {

5

public
:

6
explicit BaseService(LPCTSTR szServiceName,

7
DWORD dwServiceType
=
SERVICE_WIN32_OWN_PROCESS,

8
DWORD dwStartType
=
SERVICE_AUTO_START);

9
virtual
~
BaseService() {}

10
bool ParseStandardArgs(
int
argc,
char
*
argv[]);

11
bool IsInstalled();

12
bool Install();

13
bool Uninstall();

14
bool Start();

15

private
:

16
virtual
void
Run()
=

0
;

17
virtual bool OnInitialize() {
return

true
; }

18
virtual
void
OnStop() {}

19
virtual
void
OnPause() {}

20
virtual
void
OnContinue() {}

21
virtual
void
OnInterrogate() {}

22
virtual
void
OnShutdown() {}

23
virtual
void
OnUserControl(DWORD dwControl) {}

24
...

25
};

26

27
#endif
/*
BASE_SERVICE_H
*/

要实现自己的服务类只需从BaseService继承并且Override相关的virtual函数即可，下面示范一个BeepService类，该服务只是简单地每隔2秒beep一下，为了简单所有代码均放在.h文件中：
BeepService.h

1
#ifndef BEEP_SERVICE_H

2
#define BEEP_SERVICE_H

3

4
#include
"
BaseService.h
"

5

6

class
BeepService :
public
BaseService {

7

public
:

8
BeepService(LPCTSTR szServiceName)

9
:BaseService(szServiceName)

10
,m_bPaused(
false
)

11
,m_bRunning(
false
) {}

12

13
virtual
void
OnStop() { m_bRunning
=

false
; }

14
virtual
void
OnPause() { m_bPaused
=

true
; }

15
virtual
void
OnContinue() { m_bPaused
=

false
; }

16
virtual
void
Run() {

17
m_bRunning
=

true
;

18

while
(m_bRunning) {

19

if
(
!
m_bPaused)

20
Beep(
800
,
800
);

21
Sleep(
2000
);

22
}

23
}

24

private
:

25

volatile
bool m_bPaused;

26

volatile
bool m_bRunning;

27
};

28

29
#endif
/*
BEEP_SERVICE_H
*/

通常来说只须要Override上面的4个virtual函数就OK了：
在Run()中进行实际的工作，OnStop(),OnPause(),OnContinue()则是为了响应Service Control Manager的控制。

test.cpp

1
#include
<
windows.h
>

2
#include
<
tchar.h
>

3
#include
<
stdio.h
>

4
#include
"
BeepService.h
"

5

6

int
main(
int
argc,
char

*
argv[]) {

7

8
BeepService beepService(_T(
"
BeepService
"
));

9

if
(
!
beepService.ParseStandardArgs(argc, argv)) {

10

if
(beepService.IsInstalled()) {

11

if
(
!
beepService.Start())

12
printf(
"
The service can not run from command line.\n
"
);

13
}
else
{

14
printf(
"
The service is not installed,
"

15

"
use \
"
%
s
-
i\
"
to install.\n
"
, argv[
0
]);

16
}

17
}

18

return

0
;

19
}

假设编译后生成的exe文件为beep.exe，则在命令行中可以如下使用：
(1). beep -i 安装service(安装以后系统运行时会自动启动)
(2). beep -u 卸载service(如果service正在运行，则先停止service再卸载)
BaseServiced 的ParseStandardArgs正是用来解析上述两个命令。

2. ServiceCtrl
虽然Windows自带的Service Control Manager可以控制服务程序，但是很多时候我们都需要用代码控制，这就用到ServiceCtrl类，该类的接口如下：
ServiceCtrl.h

1
#ifndef SERVICE_CTRL_H

2
#define SERVICE_CTRL_H

3

4

class
ServiceCtrl {

5

public
:

6
ServiceCtrl(LPCTSTR szServiceName);

7

~
ServiceCtrl();

8
bool Start();

9
bool Pause();

10
bool Continue();

11
bool Stop();

12
bool Interrogate();

13
bool UserControl(DWORD dwControl);

14
DWORD State()
const
;

15
...

16
};

17

18
#endif
/*
SERVICE_CTRL_H
*/

接口比较直观没什么好说的，看下面的示例代码：
test.cpp

1
#include
<
windows.h
>

2
#include
<
tchar.h
>

3
#include
<
stdio.h
>

4
#include
<
exception
>

5
#include
"
BeepService.h
"

6
#include
"
ServiceCtrl.h
"

7

8

int
main(
int
argc,
char

*
argv[]) {

9

10

try
{

11
ServiceCtrl servCtrl(_T(
"
BeepService
"
));

12

if
(servCtrl.State()
!=
SERVICE_STOPPED) {

13
printf(
"
Service already started.\n
"
);

14
}
else
{

15
servCtrl.Start();

16
printf(
"
Start.\n
"
);

17
Sleep(
6000
);

18
servCtrl.Pause();

19
printf(
"
Pause.\n
"
);

20
Sleep(
6000
);

21
servCtrl.Continue();

22
printf(
"
Continue.\n
"
);

23
Sleep(
6000
);

24
servCtrl.Stop();

25
printf(
"
Stop.\n
"
);

26
}

27
}
catch
(std::exception
&
e) {

28
printf(
"
%s\n
"
, e.what());

29
}

30

return

0
;

31
}

源代码：
点击下载