services.cpp

#define _CRT_SECURE_NO_WARNINGS

#include <windows.h>
#include <stdio.h>
#include "ErrorParser.h"

BOOL InstallServiceEx(HWND hWnd, SC_HANDLE schSCManager, char *scName, DWORD scStartType, char *szPath) {
    SC_HANDLE schService;

	// Ñúçäàâà ãî ñåðâèñà.
    schService = CreateService(
        schSCManager,              // SCM áàçà ñ äàííè
        scName,                    // èìå íà ñåðâèñà
        scName,                    // èçëîæåíîòî èìå íà ñåðâèñà
        SERVICE_ALL_ACCESS,        // æåëàåù äîñòúï
        SERVICE_KERNEL_DRIVER,     // òèï íà ñåðâèñà
        scStartType,			   // ñòàðòóâàù òèï
        SERVICE_ERROR_NORMAL,      // êîíòðîëåí òèï çà ãðåøêè
        szPath,                    // ïúòå÷êà äî ñåðâèñà(*.sys)
        NULL,                      // áåç çàðåæäàùà ãðóïà
        NULL,                      // áåç ïðèáàâêà çà óñòàíîâÿâàíå
        NULL,                      // áåç çàâèñèìîñòè
        NULL,                      // ëîêàëåí îïèñ
        NULL);                     // áåç ïàðîëà
    if (schService == NULL) {
		MessageBox(hWnd, GetErrorText(GetLastError()), "FileHide", MB_ICONERROR);
		return FALSE;
    }

	MessageBox(hWnd, "Ñåðâèñúò å óñïåøíî èíñòàëèðàí", "FileHide", MB_ICONEXCLAMATION);
    CloseServiceHandle(schService);
	return TRUE;
}

BOOL DeleteServiceEx(HWND hWnd, SC_HANDLE schSCManager, char *szSvcName)
{
    SC_HANDLE schService;

    schService = OpenService(
        schSCManager,       // SCM áàçà ñ äàííè
        szSvcName,          // èìå íà ñåðâèñà
        DELETE);            // äîñòúï çà èçòðèâàíå
    if (schService == NULL) {
		MessageBox(hWnd, GetErrorText(GetLastError()), "FileHide", MB_ICONERROR);
        return FALSE;
    }

    // Èçòðèâà ãî ñåðâèñà
    if (!DeleteService(schService)) {
		MessageBox(hWnd, GetErrorText(GetLastError()), "FileHide", MB_ICONERROR);
		CloseServiceHandle(schService);
        return FALSE;
    }

	MessageBox(hWnd, "Ñåðâèñúò å óñïåøíî èçòðèåí", "FileHide", MB_ICONEXCLAMATION);
    CloseServiceHandle(schService);
	return TRUE;
}

BOOL StartServiceEx(HWND hWnd, SC_HANDLE schSCManager, char *szSvcName)
{
	SC_HANDLE schService;
    SERVICE_STATUS_PROCESS ssStatus;
    DWORD dwOldCheckPoint;
    DWORD dwStartTickCount;
    DWORD dwWaitTime;
    DWORD dwBytesNeeded;
	char str[50];

    schService = OpenService(
        schSCManager,         // SCM áàçà ñ äàííè
        szSvcName,            // èìå íà ñåðâèñà
        SERVICE_ALL_ACCESS);  // ïúëåí äîñòúï

    if (schService == NULL) {
		MessageBox(hWnd, GetErrorText(GetLastError()), "FileHide", MB_ICONERROR);
        return FALSE;
    }

	// Ïðîâåðêà íà ñúñòîÿíèåòî â ñëó÷àé, ÷å ñåðâèñà íå å ñïðÿë.
    if (!QueryServiceStatusEx(
            schService,                     // äðúæêà äî ñåðâèñà
            SC_STATUS_PROCESS_INFO,         // èíôîðìàöèîííî íèâî
            (LPBYTE) &ssStatus,             // àäðåñà íà ñòðóêòóðà
            sizeof(SERVICE_STATUS_PROCESS), // ãîëåìèíà íà ñòðóêòóðà
            &dwBytesNeeded))              	// íåîáõîäèìà ãîëåìèíà, àêî áóôåðà å ìíîãî ìàëúê
    {
        MessageBox(hWnd, GetErrorText(GetLastError()), "FileHide", MB_ICONERROR);
        CloseServiceHandle(schService);
        return FALSE;
    }

	// Ïðîâåðÿâà äàëè ñåðâèñà ðàáîòè.
    if(ssStatus.dwCurrentState != SERVICE_STOPPED && ssStatus.dwCurrentState != SERVICE_STOP_PENDING) {
		MessageBox(hWnd, "Ñåðâèñúò íå ìîæå äà ñå ñòàðòóâà, çàùîòî å âå÷å ñòàðòóâàí è ðàáîòè", "FileHide", MB_ICONERROR);
        CloseServiceHandle(schService);
        return FALSE;
    }

	// Ñïàñÿâàíå íà öúêàíèÿòà è ïúðâîíà÷àëíèÿ êîíòðîëíî-ïðîïóñêâàòåëåí ïóíêò.
    dwStartTickCount = GetTickCount();
    dwOldCheckPoint = ssStatus.dwCheckPoint;

	// Èç÷àêâàíå íà ñåðâèñà äà ñïðå ïðåäè äà ñå îïèòà äà ñòàðòóâà.
    while (ssStatus.dwCurrentState == SERVICE_STOP_PENDING) {
		// Äà íå ñå ÷àêà ïî-äúëãî, îòêîëêîòî èç÷àêâàùèÿ íàìåê. Èíòåðâàëà å
		// åäíà äåñåòà îò èç÷àêâàùèÿ íàìåê, íî íå ïîìàëêî îò åäíà ñåêóíäà
		// è íå ïî-âå÷å îò 10 ñåêóíäè.
        dwWaitTime = ssStatus.dwWaitHint / 10;

        if(dwWaitTime < 1000)
            dwWaitTime = 1000;
        else if (dwWaitTime > 10000)
            dwWaitTime = 10000;

        Sleep(dwWaitTime);

		// Ïðîâåðêà íà ñúñòîÿíèåòî, äîêàòî ñåðâèñà íå ñïðå â î÷àêâàíå.
        if (!QueryServiceStatusEx(
                schService,                     // äðúæêà äî ñåðâèñà
                SC_STATUS_PROCESS_INFO,         // èíôîðìàöèîííî íèâî
                (LPBYTE) &ssStatus,             // àäðåñà íà ñòðóêòóðà
                sizeof(SERVICE_STATUS_PROCESS), // ãîëåìèíà íà ñòðóêòóðà
                &dwBytesNeeded))              	// íåîáõîäèìà ãîëåìèíà, àêî áóôåðà å ìíîãî ìàëúê
        {
			MessageBox(hWnd, GetErrorText(GetLastError()), "FileHide", MB_ICONERROR);
            CloseServiceHandle(schService);
            return FALSE;
        }

        if (ssStatus.dwCheckPoint > dwOldCheckPoint) {
			// Ïðîäúëæàâàíå íà èç÷àêâàíåòî è ïðîâåðêàòà.
            dwStartTickCount = GetTickCount();
            dwOldCheckPoint = ssStatus.dwCheckPoint;
        } else {
            if(GetTickCount()-dwStartTickCount > ssStatus.dwWaitHint) {
				MessageBox(hWnd, "Âðåìåòî çà ÷àêàíå íà ñåðâèñà çà ñïèðàíå èçìèíà", "FileHide", MB_ICONERROR);
                CloseServiceHandle(schService);
                return FALSE;
            }
        }
    }

	// Îïèò äà ñå ñòàðòóâà ñåðâèñà.
    if (!StartService(
            schService,  // handle to service
            0,           // number of arguments
            NULL) )      // no arguments
    {
        MessageBox(hWnd, GetErrorText(GetLastError()), "FileHide", MB_ICONERROR);
        CloseServiceHandle(schService);
        return FALSE;
    }

	// Çàïî÷âà î÷àêâàíåòî...
	// Ïðîâåðêà íà ñúñòîÿíèåòî, äîêàòî ñåðâèñà íå çàïî÷íå â î÷àêâàíå.
    if (!QueryServiceStatusEx(
            schService,                     // äðúæêà äî ñåðâèñà
            SC_STATUS_PROCESS_INFO,         // èíôîðìàöèîííî íèâî
            (LPBYTE) &ssStatus,             // àäðåñà íà ñòðóêòóðà
            sizeof(SERVICE_STATUS_PROCESS), // ãîëåìèíà íà ñòðóêòóðà
            &dwBytesNeeded))             	// àêî áóôåðà å ìíîãî ìàëúê
    {
        MessageBox(hWnd, GetErrorText(GetLastError()), "FileHide", MB_ICONERROR);
        CloseServiceHandle(schService);
        return FALSE;
    }

	// Ñïàñÿâàíå íà öúêàíèÿòà è ïúðâîíà÷àëíèÿ êîíòðîëíî-ïðîïóñêâàòåëåí ïóíêò.
    dwStartTickCount = GetTickCount();
    dwOldCheckPoint = ssStatus.dwCheckPoint;

    while (ssStatus.dwCurrentState == SERVICE_START_PENDING) {

        dwWaitTime = ssStatus.dwWaitHint / 10;

        if(dwWaitTime < 1000)
            dwWaitTime = 1000;
        else if (dwWaitTime > 10000)
            dwWaitTime = 10000;

        Sleep(dwWaitTime);

        // Îòíîâî ïðîâåðêà íà ñúñòîÿíèåòî.
        if (!QueryServiceStatusEx(
            schService,
            SC_STATUS_PROCESS_INFO,
            (LPBYTE) &ssStatus,
            sizeof(SERVICE_STATUS_PROCESS),
            &dwBytesNeeded))
        {
            MessageBox(hWnd, GetErrorText(GetLastError()), "FileHide", MB_ICONERROR);
            break;
        }

        if (ssStatus.dwCheckPoint > dwOldCheckPoint) {
			// Ïðîäúëæàâàíå íà èç÷àêâàíåòî è ïðîâåðêàòà.
            dwStartTickCount = GetTickCount();
            dwOldCheckPoint = ssStatus.dwCheckPoint;
        } else {
            if(GetTickCount()-dwStartTickCount > ssStatus.dwWaitHint) {
				// Áåç íàïðåäúê â èç÷åêàùèÿ íàìåê.
                break;
            }
        }
    }

	// Îïðåäåëÿ äàëè ñåðâèñà å ñòàðòóâàí.
    if (ssStatus.dwCurrentState == SERVICE_RUNNING) {
        MessageBox(hWnd, "Ñåðâèñúò å ñòàðòóâàí", "FileHide", MB_ICONEXCLAMATION);
		CloseServiceHandle(schService);
		return TRUE;
    } else {
		sprintf(str, "Ñåðâèñúò íå å ñòàðòóâàí.\n");
		sprintf(str, "Òåêóùî ñúñòîÿíèå: %d\n", ssStatus.dwCurrentState);
        sprintf(str, "  Exit Code: %d\n", ssStatus.dwWin32ExitCode);
        sprintf(str, "  Check Point: %d\n", ssStatus.dwCheckPoint);
        sprintf(str, "  Wait Hint: %d\n", ssStatus.dwWaitHint);
		MessageBox(hWnd, str, "FileHide", MB_ICONERROR);
		CloseServiceHandle(schService);
		return FALSE;
    }

}

BOOL StopDependentServices(SC_HANDLE hSCManager, SC_HANDLE hService);

BOOL StopServiceEx(HWND hWnd, SC_HANDLE schSCManager, char *szSvcName)
{
	SC_HANDLE schService;
    SERVICE_STATUS_PROCESS ssp;
    DWORD dwStartTime = GetTickCount();
    DWORD dwBytesNeeded;
    DWORD dwTimeout = 30000; // 30 ñåêóíäíà ïàóçà
    DWORD dwWaitTime;

    schService = OpenService(schSCManager, szSvcName,
							SERVICE_STOP | SERVICE_QUERY_STATUS | SERVICE_ENUMERATE_DEPENDENTS);
    if (schService == NULL) {
        MessageBox(hWnd, GetErrorText(GetLastError()), "FileHide", MB_ICONERROR);
        return FALSE;
    }

	// Ïðîâåðÿâà äàëè ñåðâèñà íå å âå÷å ñïðÿí.
    if (!QueryServiceStatusEx(schService, SC_STATUS_PROCESS_INFO, (LPBYTE)&ssp,
								sizeof(SERVICE_STATUS_PROCESS), &dwBytesNeeded))
    {
        MessageBox(hWnd, GetErrorText(GetLastError()), "FileHide", MB_ICONERROR);
        CloseServiceHandle(schService);
		return FALSE;
    }

    if (ssp.dwCurrentState == SERVICE_STOPPED) {
        MessageBox(hWnd, "Ñåðâèñúò å áèë ñïðåí", "FileHide", MB_ICONERROR);
        CloseServiceHandle(schService);
		return FALSE;
    }

	// Àêî ñòîïèðàíåòî å â î÷àêâàíå, èç÷àêàé ãî.
    while (ssp.dwCurrentState == SERVICE_STOP_PENDING) {
        dwWaitTime = ssp.dwWaitHint / 10;
        if(dwWaitTime < 1000)
            dwWaitTime = 1000;
        else if (dwWaitTime > 10000)
            dwWaitTime = 10000;

        Sleep(dwWaitTime);

        if ( !QueryServiceStatusEx(schService, SC_STATUS_PROCESS_INFO, (LPBYTE)&ssp,
									sizeof(SERVICE_STATUS_PROCESS), &dwBytesNeeded))
        {
			MessageBox(hWnd, GetErrorText(GetLastError()), "FileHide", MB_ICONERROR);
			CloseServiceHandle(schService);
			return FALSE;
        }

        if (ssp.dwCurrentState == SERVICE_STOPPED) {
			MessageBox(hWnd, "Ñåðâèñúò å ñïðÿí", "FileHide", MB_ICONEXCLAMATION);
			CloseServiceHandle(schService);
			return TRUE;
        }

        if (GetTickCount() - dwStartTime > dwTimeout) {
			MessageBox(hWnd, "Âðåìåòî çà ñïèðàíå íà ñåðâèñà èçòå÷å", "FileHide", MB_ICONERROR);
			CloseServiceHandle(schService);
			return FALSE;
        }
    }

	// Àêî ñåðâèñà ðàáîòè, çàâèñèìèòå ñåðâèñè ïúðâî ñå ñïèðàò.
    StopDependentServices(schSCManager, schService);

	// Èçïðàùà ñòîï êîä êúì ñåðâèñà.
    if (!ControlService(schService, SERVICE_CONTROL_STOP, (LPSERVICE_STATUS)&ssp)) {
		MessageBox(hWnd, GetErrorText(GetLastError()), "FileHide", MB_ICONERROR);
		CloseServiceHandle(schService);
        return FALSE;
    }

	// Èç÷àêà ñåðâèñà äà ñïðå.
    while (ssp.dwCurrentState != SERVICE_STOPPED) {
        Sleep(ssp.dwWaitHint);

        if (!QueryServiceStatusEx(schService, SC_STATUS_PROCESS_INFO, (LPBYTE)&ssp,
									sizeof(SERVICE_STATUS_PROCESS), &dwBytesNeeded))
        {
			MessageBox(hWnd, GetErrorText(GetLastError()), "FileHide", MB_ICONERROR);
			CloseServiceHandle(schService);
			return FALSE;
        }

        if (ssp.dwCurrentState == SERVICE_STOPPED)
            break;

        if (GetTickCount() - dwStartTime > dwTimeout) {
			MessageBox(hWnd, "Âðåìåòî çà ñïèðàíå íà ñåðâèñà èçòå÷å", "FileHide", MB_ICONERROR);
			CloseServiceHandle(schService);
			return FALSE;
        }
    }

	MessageBox(hWnd, "Ñåðâèñúò å ñïðÿí", "FileHide", MB_ICONEXCLAMATION);
	CloseServiceHandle(schService);
	return TRUE;
}

BOOL StopDependentServices(SC_HANDLE hSCManager, SC_HANDLE hService)
{
    LPENUM_SERVICE_STATUS   lpDependencies = NULL;
    ENUM_SERVICE_STATUS     ess;
    SC_HANDLE               hDepService;
    SERVICE_STATUS_PROCESS  ssp;
	DWORD i, dwBytesNeeded, dwCount;

    DWORD dwStartTime = GetTickCount();
    DWORD dwTimeout = 30000; // 30 ñåêóíäíà ïàóçà

	// Ïóñêàíå íà áóôåð ñ íóëåâà äúëæèíà, çà äà ñå ïîëó÷è æåëàíèÿ ðàçìåð íà áóôåðà.
    if (EnumDependentServices(hService, SERVICE_ACTIVE, lpDependencies, 0, &dwBytesNeeded, &dwCount))
		// Àêî èçâèêâàíåòî íà Enum-à å óñïåøíî, òîãàâà íÿìà çàâèñèìè
		// ñåðâèñè, òàêà ÷å íå ñå ïðàâè íèùî.
        return TRUE;
	
	if (GetLastError() != ERROR_MORE_DATA)
		return FALSE; // Íåî÷àêâàíà ãðåøêà

	// Àëîêèðàíå íà áóôåð çà çàâèñèìîñòèòå.
	lpDependencies = (LPENUM_SERVICE_STATUS)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, dwBytesNeeded);
	if (!lpDependencies)
		return FALSE;

	__try {
		// Èçáðîéâàò ñå çàâèñèìîñòèòå.
		if (!EnumDependentServices(hService, SERVICE_ACTIVE, lpDependencies,
									dwBytesNeeded, &dwBytesNeeded, &dwCount))
			return FALSE;

		for (i=0; i<dwCount; i++) {
			ess = *(lpDependencies + i);
			// Îòâàðÿ ñå ñåðâèñà
			hDepService = OpenService(hSCManager, ess.lpServiceName, SERVICE_STOP|SERVICE_QUERY_STATUS);
			if (!hDepService)
			   return FALSE;

			__try {
				// Èçïðàùà ñå ñòîï êîä.
				if (!ControlService(hDepService, SERVICE_CONTROL_STOP, (LPSERVICE_STATUS)&ssp))
					return FALSE;

				// Èç÷àêâà ñå ñåðâèñà äà ñïðå.
				while (ssp.dwCurrentState != SERVICE_STOPPED) {
					Sleep(ssp.dwWaitHint);
					if (!QueryServiceStatusEx(hDepService, SC_STATUS_PROCESS_INFO, (LPBYTE)&ssp,
												sizeof(SERVICE_STATUS_PROCESS), &dwBytesNeeded))
						return FALSE;

					if (ssp.dwCurrentState == SERVICE_STOPPED)
						break;

					if (GetTickCount() - dwStartTime > dwTimeout)
						return FALSE;
				}
			} __finally {
				// Âèíàãè ñå èçïóñêà äðúæêàòà íà ñåðâèñà.
				CloseServiceHandle(hDepService);
			}
		}
	} __finally {
		// Âèíàãè ñå îñâîáîæäàâà èçáðîåíèÿ áóôåð.
		HeapFree(GetProcessHeap(), 0, lpDependencies);
	}
	return TRUE;
}

BOOL IsServiceInstalled(SC_HANDLE schSCManager, char *szSvcName)
{
	SC_HANDLE schService;

	schService = OpenService(
        schSCManager,         // SCM áàçà ñ äàííè
        szSvcName,            // èìå íà ñåðâèñà
        SERVICE_ALL_ACCESS);  // ïúëåí äîñòúï
    if (schService != NULL)
        return TRUE;
	return FALSE;
}

BOOL IsServiceRunning(HWND hWnd, SC_HANDLE schSCManager, char *szSvcName)
{
	SC_HANDLE schService;
    SERVICE_STATUS_PROCESS ssStatus;
    DWORD dwBytesNeeded;

	schService = OpenService(
        schSCManager,         // SCM áàçà ñ äàííè
        szSvcName,            // èìå íà ñåðâèñà
        SERVICE_ALL_ACCESS);  // ïúëåí äîñòúï

    if (schService == NULL) {
        return FALSE;
    }

    if (!QueryServiceStatusEx(
            schService,                     // äðúæêà äî ñåðâèñà
            SC_STATUS_PROCESS_INFO,         // èíôîðìàöèîííî íèâî
            (LPBYTE) &ssStatus,             // àäðåñà íà ñòðóêòóðà
            sizeof(SERVICE_STATUS_PROCESS), // ãîëåìèíà íà ñòðóêòóðà
            &dwBytesNeeded))              	// íåîáõîäèìà ãîëåìèíà, àêî áóôåðà å ìíîãî ìàëúê
    {
        MessageBox(hWnd, GetErrorText(GetLastError()), "FileHide", MB_ICONERROR);
        CloseServiceHandle(schService);
        return FALSE;
    }
	// Ïðîâåðÿâà äàëè ñåðâèñà ðàáîòè.
    if(ssStatus.dwCurrentState == SERVICE_RUNNING) {
        CloseServiceHandle(schService);
        return TRUE;
    }

	return FALSE;
}