C / C++ / MFC

#include "stdafx.h"
#include "Windows.h"
#include "Winsvc.h"
#include "time.h"
#include "tlhelp32.h"



SERVICE_STATUS m_ServiceStatus;
SERVICE_STATUS_HANDLE m_ServiceStatusHandle;
BOOL bRunning=true;
void WINAPI ServiceMain(DWORD argc, LPTSTR *argv);
void WINAPI ServiceCtrlHandler(DWORD Opcode);

BOOL InstallService();
BOOL DeleteService();

bool isExeRunning(WCHAR *zExeName, bool *pbRunning);

int main(int argc, char* argv[])
{
	if(argc>1)
	{
		if(strcmp(argv[1],"-i")==0)
		{
			if(InstallService())
				printf("\n\nService Installed Sucessfully\n");
			else
				printf("\n\nError Installing Service\n");
		}
		if(strcmp(argv[1],"-d")==0)
		{
			if(DeleteService())
				printf("\n\nService UnInstalled Sucessfully\n");
			else
				printf("\n\nError UnInstalling Service\n");
		}
		else
		{
			printf("\n\nUnknown Switch Usage\n\nFor Install use Srv1 -i\n\nFor UnInstall use Srv1 -d\n");
		}
	}
	else
	{
		SERVICE_TABLE_ENTRY DispatchTable[]={{L"PointService",ServiceMain},{NULL,NULL}};
		StartServiceCtrlDispatcher(DispatchTable);		
	}

	return 0;
}

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

	m_ServiceStatusHandle = RegisterServiceCtrlHandler(L"PointService", ServiceCtrlHandler); 
	if (m_ServiceStatusHandle == (SERVICE_STATUS_HANDLE)0)
	{
		return;
	}

	m_ServiceStatus.dwCurrentState = SERVICE_RUNNING;
	m_ServiceStatus.dwCheckPoint = 0;
	m_ServiceStatus.dwWaitHint = 0;
	if (!SetServiceStatus (m_ServiceStatusHandle, &m_ServiceStatus))
	{
	}

	bRunning=true;
	while(bRunning)
	{
		Sleep(3000);	

		bool pbRunning = true;

		isExeRunning(L"Point-HD.exe",&pbRunning);

		if(pbRunning == true)
		{
			printf("Point-HD running......\n");
		}
		else
		{
			printf("Point-HD running......\n");
		}	
	}
	
	return;
}

void WINAPI ServiceCtrlHandler(DWORD Opcode)
{
	switch(Opcode)
	{
		case SERVICE_CONTROL_PAUSE: 
			m_ServiceStatus.dwCurrentState = SERVICE_PAUSED;
			break;
		case SERVICE_CONTROL_CONTINUE:
			m_ServiceStatus.dwCurrentState = SERVICE_RUNNING;
			break;
		case SERVICE_CONTROL_STOP:
			m_ServiceStatus.dwWin32ExitCode = 0;
			m_ServiceStatus.dwCurrentState = SERVICE_STOPPED;
			m_ServiceStatus.dwCheckPoint = 0;
			m_ServiceStatus.dwWaitHint = 0;

			SetServiceStatus (m_ServiceStatusHandle,&m_ServiceStatus);
			bRunning=false;
			break;
		case SERVICE_CONTROL_INTERROGATE:
			break; 
	}
	
	return;
}

BOOL InstallService()
{
	WCHAR strDir[1024];
	SC_HANDLE schSCManager,schService;
	GetCurrentDirectory(1024,strDir);
	wcscat(strDir,L"\\Point-HDListener.exe");
	schSCManager = OpenSCManager(NULL,NULL,SC_MANAGER_ALL_ACCESS);

	if (schSCManager == NULL) 
		return false;
	LPCTSTR lpszBinaryPathName=strDir;

	schService = CreateServiceW(schSCManager,
								L"PointService", 
								L"Point HD Listener Service",	// service name to display
								SERVICE_ALL_ACCESS,				// desired access 
								SERVICE_WIN32_OWN_PROCESS,		// service type 
								SERVICE_DEMAND_START,			// start type 
								SERVICE_ERROR_NORMAL,			// error control type 
								lpszBinaryPathName,				// service's binary 
								NULL,							// no load ordering group 
								NULL,							// no tag identifier 
								NULL,							// no dependencies
								NULL,							// LocalSystem account
								NULL);							// no password


	if (schService == NULL)
		return false; 

	CloseServiceHandle(schService);
	return true;
}

BOOL DeleteService()
{
	SC_HANDLE schSCManager;
	SC_HANDLE hService;
	
	schSCManager = OpenSCManager(NULL,NULL,SC_MANAGER_ALL_ACCESS);
	if (schSCManager == NULL)
		return false;
	
	hService=OpenService(schSCManager,L"PointService",SERVICE_ALL_ACCESS);
	if (hService == NULL)
		return false;

	if(DeleteService(hService)==0)
		return false;

	if(CloseServiceHandle(hService)==0)
		return false;

	return true;
}

bool isExeRunning(WCHAR *zExeName, bool *pbRunning)
{
	if (! pbRunning) return FALSE;		

	PROCESSENTRY32 pe32;
	BOOL bSuccess = TRUE;
	*pbRunning = FALSE;

	//Takes a snapshot of the specified processes, 
	//as well as the heaps, modules, and threads used by these processes.
	HANDLE hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, NULL);

	if (hSnapshot!=INVALID_HANDLE_VALUE) 
	{		
		// Set the size of the structure before using it.
		pe32.dwSize = sizeof( PROCESSENTRY32 );

		//Describes an entry from a list of the processes residing 
		//in the system address space when a snapshot was taken.

		BOOL bRes = Process32First(hSnapshot, &pe32);

		while (bRes==TRUE)
		{			
			wprintf(L"%s\n",pe32.szExeFile);
			if (0==wcscmp(zExeName, pe32.szExeFile)) 
			{
				*pbRunning = TRUE;
				break;
			}
			//Retrieves information about the next 
			//process recorded in a system snapshot.

			bRes = Process32Next(hSnapshot, &pe32);
		}

		if (bRes!=TRUE && GetLastError()!=ERROR_NO_MORE_FILES) 
		{
			bSuccess = FALSE;
		}
		
		CloseHandle(hSnapshot);
	} 
	else 
	{
		bSuccess = FALSE;
	}

	return bSuccess;
}

unsigned int far* BaseAddr;<br />
BaseAddr = (unsigned int far *)MK_FP(0xD000,0x0000);