// Douglas Thrift
//
// CCS Computer Science
//
// Windows Finger Daemon

#include <windows.h>
#include <tchar.h>

#include <iostream>

LPTSTR name;
SERVICE_STATUS status;
SERVICE_STATUS_HANDLE handle;
HANDLE stop;
WSADATA data;

void WINAPI FingerMain(DWORD argc, LPTSTR *argv);
void WINAPI FingerControl(DWORD control);

int _tmain(int argc, TCHAR *argv[])
{
	SERVICE_TABLE_ENTRY entry[] = { { TEXT("CCS Finger Daemon"), LPSERVICE_MAIN_FUNCTION(FingerMain) }, { NULL, NULL } };

	if (!StartServiceCtrlDispatcher(entry))
	{
		DWORD error(GetLastError());

		switch (error)
		{
		case ERROR_FAILED_SERVICE_CONTROLLER_CONNECT:
			goto go;

		case ERROR_INVALID_DATA:
			std::cerr << "ERROR_INVALID_DATA" << std::endl;

			break;

		case ERROR_SERVICE_ALREADY_RUNNING:
			std::cerr << "ERROR_SERVICE_ALREADY_RUNNING" << std::endl;

			break;

		default:
			std::cerr << error << std::endl;
		}

		return 1;
	}

	return 0;

go:
	std::cout << "Here!" << std::endl;

	return 0;
}

void WINAPI FingerMain(DWORD argc, LPTSTR *argv)
{
	name = argv[0];
	handle = RegisterServiceCtrlHandler(name, LPHANDLER_FUNCTION(FingerControl));
	stop = CreateEvent(NULL, TRUE, FALSE, NULL);

	//

	WSAStartup(MAKEWORD(2, 0), &data);

	//

	status.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
	status.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
	status.dwServiceSpecificExitCode = 0;
	status.dwCurrentState = SERVICE_RUNNING;
	status.dwWin32ExitCode = NO_ERROR;
	status.dwCheckPoint = 0;
	status.dwWaitHint = 0;

	SetServiceStatus(handle, &status);

	SOCKET sock(socket(PF_INET, SOCK_STREAM, IPPROTO_TCP));
	SOCKADDR_IN service;
	
	service.sin_family = AF_INET;
	service.sin_addr.s_addr = inet_addr("0.0.0.0");
	service.sin_port = htons(79);

	bind(sock, (SOCKADDR *)(&service), sizeof (service));
	listen(sock, SOMAXCONN);

	while(WaitForSingleObject(stop, 1000) != WAIT_OBJECT_0 /*&&*/)
	{
		//
	}

	//

	WSACleanup();

	status.dwCurrentState = SERVICE_STOPPED;
	status.dwWin32ExitCode = NO_ERROR;
	status.dwCheckPoint = 0;
	status.dwWaitHint = 0;

	SetServiceStatus(handle, &status);
}

void WINAPI FingerControl(DWORD control)
{
	switch (control)
	{
	case SERVICE_CONTROL_STOP:
	case SERVICE_CONTROL_SHUTDOWN:
		status.dwCurrentState = SERVICE_STOP_PENDING;

		SetEvent(stop);

		break;

	case SERVICE_CONTROL_INTERROGATE:
		break;
	}

	SetServiceStatus(handle, &status);
}