//
NTService.cpp : Defines the entry point for the console application.
//
#include
<
windows.h
>
#include
<
stdio.h
>
#define
SLEEP_TIME 5000
#define
LOGFILE "C:/MyServices/memstatus.txt"
SERVICE_STATUS ServiceStatus;
SERVICE_STATUS_HANDLE hStatus;
void
ServiceMain(
int
argc,
char
**
argv);
void
ControlHandler(DWORD request);
int
InitService();
int
WriteToLog(
char
*
str);
int
main(
int
argc,
char
*
argv[])
{
SERVICE_TABLE_ENTRY ServiceTable[
2
];
ServiceTable[
0
].lpServiceName
=
"
MemoryStatus
"
;
ServiceTable[
0
].lpServiceProc
=
(LPSERVICE_MAIN_FUNCTION)ServiceMain;
ServiceTable[
1
].lpServiceName
=
NULL;
ServiceTable[
1
].lpServiceProc
=
NULL;
//
启动服务的控制分派机线程
StartServiceCtrlDispatcher(ServiceTable);
}
void
ServiceMain(
int
argc,
char
**
argv)
{
int
error;
ServiceStatus.dwServiceType
=
SERVICE_WIN32;
ServiceStatus.dwCurrentState
=
SERVICE_START_PENDING;
ServiceStatus.dwControlsAccepted
=
SERVICE_ACCEPT_STOP
|
SERVICE_ACCEPT_SHUTDOWN;
ServiceStatus.dwWin32ExitCode
=
0
;
ServiceStatus.dwServiceSpecificExitCode
=
0
;
ServiceStatus.dwCheckPoint
=
0
;
ServiceStatus.dwWaitHint
=
0
;
hStatus
=
RegisterServiceCtrlHandler(
"
MemoryStatus
"
,
(LPHANDLER_FUNCTION)ControlHandler);
if
(hStatus
==
(SERVICE_STATUS_HANDLE)
0
)
{
//
Registering Control Handler failed
return
;
}
//
Initialize Service
error
=
InitService();
if
(
!
error)
{
//
Initialization failed
ServiceStatus.dwCurrentState
=
SERVICE_STOPPED;
ServiceStatus.dwWin32ExitCode
=
-
1
;
SetServiceStatus(hStatus,
&
ServiceStatus);
return
;
}
//
We report the running status to SCM.
ServiceStatus.dwCurrentState
=
SERVICE_RUNNING;
SetServiceStatus (hStatus,
&
ServiceStatus);
MEMORYSTATUS memory;
//
The worker loop of a service
while
(ServiceStatus.dwCurrentState
==
SERVICE_RUNNING)
{
char
buffer[
16
];
GlobalMemoryStatus(
&
memory);
sprintf(buffer,
"
%d
"
, memory.dwAvailPhys);
int
result
=
WriteToLog(buffer);
if
(result)
{
ServiceStatus.dwCurrentState
=
SERVICE_STOPPED;
ServiceStatus.dwWin32ExitCode
=
-
1
;
SetServiceStatus(hStatus,
&
ServiceStatus);
return
;
}
Sleep(SLEEP_TIME);
}
return
;
}
void
ControlHandler(DWORD request)
{
switch
(request)
{
case
SERVICE_CONTROL_STOP:
WriteToLog(
"
Monitoring stopped.
"
);
ServiceStatus.dwWin32ExitCode
=
0
;
ServiceStatus.dwCurrentState
=
SERVICE_STOPPED;
SetServiceStatus (hStatus,
&
ServiceStatus);
return
;
case
SERVICE_CONTROL_SHUTDOWN:
WriteToLog(
"
Monitoring stopped.
"
);
ServiceStatus.dwWin32ExitCode
=
0
;
ServiceStatus.dwCurrentState
=
SERVICE_STOPPED;
SetServiceStatus (hStatus,
&
ServiceStatus);
return
;
default
:
break
;
}
//
Report current status
SetServiceStatus (hStatus,
&
ServiceStatus);
return
;
}
int
WriteToLog(
char
*
str)
{
FILE
*
log;
log
=
fopen(LOGFILE,
"
a+
"
);
if
(log
==
NULL)
return
-
1
;
fprintf(log,
"
%s
"
, str);
fclose(log);
return
0
;
}
int
InitService(){
WriteToLog(
"
Monitoring started.
"
);
return
true
;
}
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)