Priority.cpp

#include <windows.h>
#include "Priority.h"
#include "ProcessList.h"

#include <stdio.h>


CPriorityService::CPriorityService()
	: CNTService(TEXT("Priority"), TEXT("Digital Design Priority for Windows 10"))
	, m_hStop(0)
{
	m_dwControlsAccepted = 0;
	m_dwControlsAccepted |= SERVICE_ACCEPT_STOP;
}

void CPriorityService::Run(DWORD dwArgc, LPTSTR * ppszArgv) {
	// report to the SCM that we're about to start
	ReportStatus(SERVICE_START_PENDING);

	m_hStop = ::CreateEvent(NULL, TRUE, FALSE, NULL);

	// TODO: You might do some initialization here.
	//		 Parameter processing for instance ...
	//		 If this initialization takes a long time,
	//		 don't forget to call "ReportStatus()"
	//		 frequently or adjust the number of milliseconds
	//		 in the "ReportStatus()" above.

	// report SERVICE_RUNNING immediately before you enter the main-loop
	// DON'T FORGET THIS!
	ReportStatus(SERVICE_RUNNING);

#ifdef _DEBUG
	int iCount = 0;
#endif

	// main-loop
	// If the Stop() method sets the event, then we will break out of
	// this loop.
	while (::WaitForSingleObject(m_hStop, 10) != WAIT_OBJECT_0)
	{
		CProcessList pl;
		pl.Initialize();
		pl.EnumProcesses();
		Sleep(2000);
	}

	ReportStatus(SERVICE_STOP_PENDING);

	if (m_hStop)
		::CloseHandle(m_hStop);

	ReportStatus(SERVICE_ACCEPT_STOP);
}


void CPriorityService::Stop() {
	// report to the SCM that we're about to stop

	// TODO: Adjust the number of milliseconds you think
	//		 the stop-operation may take.
	ReportStatus(SERVICE_STOP_PENDING, 5000);

	if (m_hStop)
		::SetEvent(m_hStop);
}