#include <windows.h>
#include <fstream>

#define SERVICE_NAME TEXT("Screen control service")
#define COMMAND_INSTALL   "install"
#define COMMAND_UNINSTALL "uninstall"
 
SERVICE_STATUS serviceStatus;
SERVICE_STATUS_HANDLE serviceStatusHandle = 0;
 
HANDLE stopServiceEvent = 0;
 
using namespace std;
//#define  PREFIX    "scr"
 
void ScreenShotToFile(LPCSTR wzTemp)
{
	// copy screen to bitmap
	POINT a, b;
	
	a.x = 0;
	a.y = 0;
	b.x = GetSystemMetrics(SM_CXSCREEN);
	b.y = GetSystemMetrics(SM_CYSCREEN);
	
	//HDC     hScreen = GetDC(handle);	
	//HDC     hScreen = GetDCEx(NULL);
	//HDC     hScreen = CreateDC("DISPLAY/SAM06A4/5&E615B6C&0&UID1048832", NULL, NULL, NULL); 
	//HDC     hScreen = GetDC(NULL);
	//HDC     hScreen = GetDC(0);
	//HWND    handle = GetDesktopWindow();
	HDC     hScreen = GetDC(NULL);
	HDC     hDC     = CreateCompatibleDC(hScreen);
	HBITMAP hBitmap = CreateCompatibleBitmap(hScreen, abs(b.x-a.x), abs(b.y-a.y));
	HGDIOBJ old_obj = SelectObject(hDC, hBitmap);
	SelectObject(hDC, hBitmap);
	
	BOOL    bRet    = BitBlt(hDC, 0, 0, abs(b.x-a.x), abs(b.y-a.y), hScreen, a.x, a.y, SRCCOPY);
	
	// save bitmap to file
	DWORD* byteImage = new DWORD[b.x * b.y];
	
	BITMAPINFO* bmInfo = new BITMAPINFO();
	
	bmInfo->bmiHeader.biWidth = b.x;
	bmInfo->bmiHeader.biHeight = b.y;
	bmInfo->bmiHeader.biPlanes = 1;
	bmInfo->bmiHeader.biCompression = BI_RGB;
	bmInfo->bmiHeader.biBitCount = 16;
	bmInfo->bmiHeader.biSize = sizeof(bmInfo->bmiHeader);
	
	GetDIBits(hDC, hBitmap, 0, b.x, byteImage, bmInfo, DIB_RGB_COLORS);
	
	ofstream file;
	BITMAPFILEHEADER* bmfh = new BITMAPFILEHEADER();
	
	file.open(wzTemp, ios::binary|ios::trunc|ios::out);
	
	bmfh->bfType = 0x4d42;
	bmfh->bfReserved1 = 0;
	bmfh->bfReserved2 = 0;
	bmfh->bfSize = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFO) + (b.x * b.y);
	
	file.write((char*)bmfh, sizeof(BITMAPFILEHEADER));
	file.write((char*)bmInfo, sizeof(BITMAPINFO));
	file.write((char*)byteImage, b.x * b.y * 4);
	
	// clean up
	DeleteDC(hDC);
	ReleaseDC(NULL, hScreen);
	DeleteObject(hBitmap);
}

void WINAPI ServiceControlHandler(DWORD controlCode)
{
   switch (controlCode)
   {
      case SERVICE_CONTROL_INTERROGATE:
      break;
 
      case SERVICE_CONTROL_SHUTDOWN:
      case SERVICE_CONTROL_STOP:
         serviceStatus.dwCurrentState = SERVICE_STOP_PENDING;
         SetServiceStatus(serviceStatusHandle, &serviceStatus);
 
         SetEvent(stopServiceEvent);
      return;
 
      case SERVICE_CONTROL_PAUSE:
      break;
 
      case SERVICE_CONTROL_CONTINUE:
      break;
 
      default:
         if (controlCode >= 128 && controlCode <= 255)
         {
            break;
         }
         else
         {
            break;
         }
   }
 
   SetServiceStatus(serviceStatusHandle, &serviceStatus);
}

void WINAPI ServiceMain(DWORD argc, TCHAR* argv[])
{
	// initialise service status
	serviceStatus.dwServiceType = SERVICE_WIN32;
	serviceStatus.dwCurrentState = SERVICE_STOPPED;
	serviceStatus.dwControlsAccepted = 0;
	serviceStatus.dwWin32ExitCode = NO_ERROR;
	serviceStatus.dwServiceSpecificExitCode = NO_ERROR;
	serviceStatus.dwCheckPoint = 0;
	serviceStatus.dwWaitHint = 0;
 
	serviceStatusHandle = RegisterServiceCtrlHandler(SERVICE_NAME, ServiceControlHandler);
	
	if (!serviceStatusHandle)
	{
		return;
	}
	serviceStatus.dwCurrentState = SERVICE_START_PENDING;
	SetServiceStatus(serviceStatusHandle, &serviceStatus);
	
	stopServiceEvent = CreateEvent(0, FALSE, FALSE, 0);
	
	serviceStatus.dwControlsAccepted |= (SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN);
	serviceStatus.dwCurrentState = SERVICE_RUNNING;
	SetServiceStatus(serviceStatusHandle, &serviceStatus);
 
	//LPCSTR Prefix = PREFIX;
	//CHAR TempWinDir[255];
	//GetTempPath(255, TempWinDir); 
	//LPTSTR wzTemp = new TCHAR[MAX_PATH]; 
	//GetTempFileName(TempWinDir, Prefix, 0, wzTemp); 
 
	CreateThread(0, 0, (LPTHREAD_START_ROUTINE)ScreenShotToFile, (LPVOID)"C:/image.bmp", 0, 0);
	WaitForSingleObject(stopServiceEvent, INFINITE);
	
	serviceStatus.dwCurrentState = SERVICE_STOP_PENDING;
	SetServiceStatus(serviceStatusHandle, &serviceStatus);
	
	CloseHandle(stopServiceEvent);
	stopServiceEvent = 0;
	
	serviceStatus.dwControlsAccepted &= ~(SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN);
	serviceStatus.dwCurrentState = SERVICE_STOPPED;
	SetServiceStatus(serviceStatusHandle, &serviceStatus);
}

void RunService()
{
	SERVICE_TABLE_ENTRY serviceTable[] =
	{
		{(LPSTR) SERVICE_NAME, ServiceMain},
		{0, 0}
	};
	
	StartServiceCtrlDispatcher(serviceTable);
}

void InstallService()
{
	SC_HANDLE serviceControlManager = OpenSCManager(0, 0, SC_MANAGER_CREATE_SERVICE);
	
	if (serviceControlManager)
	{
	  TCHAR path[_MAX_PATH + 1];
	  if (GetModuleFileName(0, path, sizeof(path) / sizeof(*path)) > 0)
	  {
	     SC_HANDLE service = CreateService(serviceControlManager,
	                                       SERVICE_NAME, SERVICE_NAME,
	                                       SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS,
	                                       SERVICE_AUTO_START, SERVICE_ERROR_IGNORE, path,
	                                       0, 0, 0, 0, 0);
	     if (service)
	     {
	        CloseServiceHandle(service);
	     }
	  }
	
	  CloseServiceHandle(serviceControlManager);
	}
}

void UninstallService()
{
	SC_HANDLE serviceControlManager = OpenSCManager(0, 0, SC_MANAGER_CONNECT);
	
	if (serviceControlManager)
	{
	  SC_HANDLE service = OpenService(serviceControlManager, SERVICE_NAME, SERVICE_QUERY_STATUS | DELETE);
	  if (service)
	  {
	     SERVICE_STATUS serviceStatus;
	     if (QueryServiceStatus(service, &serviceStatus))
	     {
	        if (serviceStatus.dwCurrentState == SERVICE_STOPPED)
	        {
	           DeleteService(service);
	        }
	     }
	
	     CloseServiceHandle(service);
	  }
	
	  CloseServiceHandle(serviceControlManager);
	}
}

int main(int argc, char** argv)
{
	if (argc > 1 && strcmp(argv[1], COMMAND_INSTALL) == 0)
	{
	  InstallService();
	}
	else if (argc > 1 && strcmp(argv[1], COMMAND_UNINSTALL) == 0)
	{
	  UninstallService();
	}
	else
	{
	  RunService();
	}
	return 0;
}