编写windows服务程序

初学VC，写了一个最简单的服务程序，windows API 真心太多了，留作笔记。以后说不定改改还能用上。

编译后生成exe文件：

test.exe -i   //安装服务

test.exe -r   //删除服务

while循环中写自己的逻辑。

**注意！！ 每次代码改变服务状态后一定表忘记SetServiceStatus一下，不然真报错不商量。

#include <Windows.h>
#include <stdio.h>

BOOL flag = TRUE;
void WINAPI ServiceMain(DWORD argc, LPSTR** argv);
void WINAPI sHandler(DWORD request);
BOOL installService();
BOOL unInstallService();

SERVICE_STATUS m_Status;
SERVICE_STATUS_HANDLE m_Handle;

int main(int argc,char* argv[]) 
{

	if (argc==2)
	{
		if (!_stricmp(argv[1],"-i"))
		{
			installService();
		}
		else if(!_stricmp(argv[1],"-r"))
		{
			unInstallService();
		}
		return 0;
	}


	SERVICE_TABLE_ENTRY table[2];
	table[0].lpServiceName = "AAA";
	table[0].lpServiceProc = (LPSERVICE_MAIN_FUNCTION)ServiceMain;
	table[1].lpServiceName = NULL;
	table[1].lpServiceProc = NULL;

	StartServiceCtrlDispatcher(table);


}

void WINAPI ServiceMain(DWORD argc, LPSTR ** argv)
{
	m_Status.dwCheckPoint = 0;
	m_Status.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
	m_Status.dwCurrentState = SERVICE_START_PENDING;
	m_Status.dwServiceSpecificExitCode = 0;
	m_Status.dwServiceType = SERVICE_WIN32;
	m_Status.dwWaitHint = 0;
	m_Status.dwWin32ExitCode = 0;

	m_Handle = RegisterServiceCtrlHandler("AAA",sHandler);


	if (m_Handle!=0)
	{
		m_Status.dwCurrentState = SERVICE_RUNNING;
	}
	SetServiceStatus(m_Handle, &m_Status);
	while (flag)
	{
		FILE *fp;
		fp = fopen("c:\\demo\\test.txt", "a+");
		fputs("test data!\r\n",fp);
		fclose(fp);
		Sleep(10000);
	}
	CloseServiceHandle(m_Handle);
}

void WINAPI sHandler(DWORD request)
{
	switch (request)
	{
	case SERVICE_CONTROL_STOP:
		flag = FALSE;
		m_Status.dwCurrentState = SERVICE_STOPPED;
		break;
	case SERVICE_CONTROL_SHUTDOWN:
		flag = FALSE;
		m_Status.dwCurrentState = SERVICE_STOPPED;
		break;
	default:
		break;
	}

	SetServiceStatus(m_Handle, &m_Status);
}

BOOL installService()
{
	SC_HANDLE scHandle = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
	if (scHandle==NULL)
	{
		printf("openSCManager failed!");
		return FALSE;
	}
	scHandle = CreateService(scHandle, "AAA", NULL, SC_MANAGER_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS, SERVICE_AUTO_START, SERVICE_ERROR_NORMAL, "c:\\demo\\AAA.exe", NULL, NULL, NULL, NULL, NULL);

	if (scHandle == NULL)
	{
		printf("createService failed!");
		return FALSE;
	}
	printf("createService success!\r\n");
	
	if (StartService(scHandle, NULL, NULL)) 
	{
		printf("start service success!\r\n"); 
	}
	else
	{
		printf("start failed!");
	}
	CloseServiceHandle(scHandle);
	return TRUE;
}

BOOL unInstallService()
{
	SC_HANDLE scm, m_service;
	SERVICE_STATUS unStatus;
	scm = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
	if (scm==NULL)
	{
		printf("openscmanage failed!");
		return FALSE;
	}
	else
	{
		m_service = OpenService(scm, "AAA", SC_MANAGER_ALL_ACCESS);
		if (m_service==NULL)
		{
			printf("openservice failed!");
			return FALSE;
		}
		else
		{
			if (QueryServiceStatus(m_service, &unStatus))
			{
				if (unStatus.dwCurrentState!=SERVICE_STOPPED)
				{
					if (!ControlService(m_service, SERVICE_CONTROL_STOP, &unStatus))
					{
						printf("controlservice stop failed!");
						return FALSE;
					}
				}
				if (DeleteService(m_service))
				{
					printf("Delete success!");
				}
				else
				{
					printf("Delete failed!");
					return FALSE;
				}
			}
			else
			{
				printf("Queryservice failed!");
				return FALSE;
			}
		}
	}
	
	return TRUE;
}