// 创建服务自启动.cpp : 定义控制台应用程序的入口点。
//
#include "stdafx.h"
#include <Windows.h>
#include <stdio.h>
#pragma comment ( lib, "Advapi32.lib" )
void WINAPI ServiceMain(int argc, char** argv);
void WINAPI CtrlHandler(DWORD request);
SERVICE_STATUS ServiceStatus; /*服务状态*/
SERVICE_STATUS_HANDLE hStatus; /*服务状态句柄*/
BOOL brunning = 1;
//sc.exe
//sc create ribi binpath= c:\22.exe
//sc start ribi
void WriteLog(char * szLog)
{
SYSTEMTIME st;
GetLocalTime(&st);
FILE *fp;
fp = fopen("D:\\log.txt", "at");
fprintf(fp, "MyLogInfo: %d:%d:%d:%d ", st.wHour, st.wMinute, st.wSecond, st.wMilliseconds);
fprintf(fp, szLog);
fclose(fp);
OutputDebugStringA(szLog);
}
int _tmain(int argc, _TCHAR* argv[])
{
SERVICE_TABLE_ENTRY ServiceTable[2];
ServiceTable[0].lpServiceName = L"ribi";
ServiceTable[0].lpServiceProc = (LPSERVICE_MAIN_FUNCTION)ServiceMain;
ServiceTable[1].lpServiceName = NULL;
ServiceTable[1].lpServiceProc = NULL;
StartServiceCtrlDispatcher(ServiceTable);
return 0;
}
void WINAPI ServiceMain(int argc, char** argv)
{
MEMORYSTATUS memstatus;
char str[100];
int availmb;
ServiceStatus.dwServiceType=SERVICE_WIN32;
ServiceStatus.dwCurrentState =SERVICE_START_PENDING;
/*在本例中只接受系统关机和停止服务两种控制命令*/
ServiceStatus.dwControlsAccepted =SERVICE_ACCEPT_SHUTDOWN |SERVICE_ACCEPT_STOP;
ServiceStatus.dwWin32ExitCode = 0;
ServiceStatus.dwServiceSpecificExitCode = 0;
ServiceStatus.dwCheckPoint = 0;
ServiceStatus.dwWaitHint = 0;
//char buffer[16];
//注册服务控制函数
hStatus = ::RegisterServiceCtrlHandler(L"ribi",(LPHANDLER_FUNCTION)CtrlHandler);
if (hStatus == (SERVICE_STATUS_HANDLE)0)
{
WriteLog("RegisterServiceCtrlHandler failed");
return;
}
WriteLog("RegisterServiceCtrlHandler success\n");
/*向SCM 报告运行状态*/
ServiceStatus.dwCurrentState =SERVICE_RUNNING;
SetServiceStatus(hStatus, &ServiceStatus);
brunning = 1;
memset(str, '\0', 100);
while (brunning)
{
GlobalMemoryStatus(&memstatus);
availmb = memstatus.dwAvailPhys / 1024 / 1024;
sprintf_s(str, 100, "available memory is %dMB\n", availmb);
OutputDebugStringA(str);
WriteLog(str);
Sleep(5000);
}
WriteLog("服务停止了");
}
void WINAPI CtrlHandler(DWORD request)
{
switch (request)
{
case SERVICE_CONTROL_STOP:
brunning = false;
ServiceStatus.dwCurrentState = SERVICE_STOPPED;
break;
case SERVICE_CONTROL_SHUTDOWN:
brunning = false;
ServiceStatus.dwCurrentState = SERVICE_STOPPED;
break;
default:
break;
}
SetServiceStatus(hStatus, &ServiceStatus);//报告服务状态
return;
}