C语言win32编写菜单,C语言编写 Win32 服务实例

C语言编写 Win32 服务实例

作者:admin 日期:2010-02-04

字体大小: 小 中 大

#include

#include

#define SLEEP_TIME 5000

#define LOGFILE "C:\\MyServices\\memstatus.txt"

// Declare several global variables to share

// their values across multiple functions of your program.

SERVICE_STATUS ServiceStatus;

SERVICE_STATUS_HANDLE hStatus;

// Make the forward definitions of functions prototypes.

//

void ServiceMain(int argc, char** argv);

void ControlHandler(DWORD request);

int InitService();

int WriteToLog(char* str)

{

FILE* log;

log = fopen(LOGFILE, "a+");

if (log == NULL){

OutputDebugString("Log file open failed.");

return -1;

}

fprintf(log, "%s\n", str);

fclose(log);

return 0;

}

// Service initialization

int InitService()

{

OutputDebugString("Monitoring started.");

int result;

result = WriteToLog("Monitoring started.");

return(result);

}

// Control Handler

void ControlHandler(DWORD request)

{

switch(request)

{

case SERVICE_CONTROL_STOP:

OutputDebugString("Monitoring stopped.");

WriteToLog("Monitoring stopped.");

ServiceStatus.dwWin32ExitCode = 0;

ServiceStatus.dwCurrentState = SERVICE_STOPPED;

SetServiceStatus (hStatus, &ServiceStatus);

return;

case SERVICE_CONTROL_SHUTDOWN:

OutputDebugString("Monitoring stopped.");

WriteToLog("Monitoring stopped.");

ServiceStatus.dwWin32ExitCode = 0;

ServiceStatus.dwCurrentState = SERVICE_STOPPED;

SetServiceStatus (hStatus, &ServiceStatus);

return;

default:

break;

}

// Report current status

SetServiceStatus (hStatus, &ServiceStatus);

return;

}

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);

OutputDebugString(buffer);

int result = WriteToLog(buffer);

if (result)

{

ServiceStatus.dwCurrentState =

SERVICE_STOPPED;

ServiceStatus.dwWin32ExitCode = -1;

SetServiceStatus(hStatus,

&ServiceStatus);

return;

}

Sleep(SLEEP_TIME);

}

return;

}

void 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;

// Start the control dispatcher thread for our service

StartServiceCtrlDispatcher(ServiceTable);

}

文章来自: 本站原创

Tags:

评论: 0 | 查看次数: 8016