service.cpp File Reference

#include "panoptes_service.h"
#include "ResourceCore.h"
#include <iostream>
#include "grpc.hpp"
#include "TrayNotifications.h"
#include "events.h"
#include "error_message.h"
#include "mutex.hpp"
#include "containers.h"
#include "utils.h"
#include "database.hpp"
#include "driver.h"
#include "Configuration.hpp"

Go to the source code of this file.

Functions
ERRORCODE	CleanupWithError (ERRORCODE err)
int WINAPI	WinMain (HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow)

Function Documentation

CleanupWithError()

ERRORCODE CleanupWithError

(

ERRORCODE

err

)

Definition at line 15 of file service.cpp.

                                          {
    DestroyMutex();
    std::string msg = GetErrorMessage(err);
    DisplayErrorMessage(msg);
    return err;
}

References DestroyMutex(), DisplayErrorMessage(), and GetErrorMessage().

Referenced by WinMain().

WinMain()

int WINAPI WinMain	(	HINSTANCE	hInstance,
		HINSTANCE	hPrevInstance,
		LPSTR	lpCmdLine,
		int	nCmdShow
	)

Definition at line 23 of file service.cpp.

{
 
#ifndef _DEBUG
    if (IsDebuggerPresent()) {
        auto msg = GetErrorMessage(DEBUGGER);
        DisplayErrorMessage(msg);
        return DEBUGGER;
    }
#endif 
 
    PanoptesContext serviceContext;
    serviceContext.threadError = false;
    
#pragma region Admin Check
    if (!IsRunningAsAdmin()) {
        auto msg = GetErrorMessage(NOT_ADMIN);
        DisplayErrorMessage(msg);
        return NOT_ADMIN;
    }
#pragma endregion
 
#pragma region Mutex Check
    if (MutexExist() != PANO_SUCCESS) {
        if (SetEnvironmentMutex() != PANO_SUCCESS) {
            auto msg = GetErrorMessage(MUTEX_SET);
            DisplayErrorMessage(msg);
            return MUTEX_SET;
        }
    }
    else {
        auto msg = GetErrorMessage(MUTEX_SET);
        DisplayErrorMessage(msg);
        return MUTEX_SET;
    }
#pragma endregion
 
#pragma region Panoptes Configuration
#ifdef _DEBUG
    std::string configPath = GetCurrentPath() + "\\panoptes.config";
#else
    std::string configPath = "C:\\ProgramData\\Panoptes\\Panoptes.config";
#endif // _DEBUG
 
    Configuration* configuration = new Configuration(configPath);
    try {
        configuration->Parse();
        serviceContext.config = configuration;
    }
    catch (const int& err) {
        return CleanupWithError(err);
    }
#pragma endregion
 
#pragma region Database
    auto dbTmp = serviceContext.database.load();
    ERRORCODE errCode = dbTmp.InitializeDatabase();
    if (errCode != PANO_SUCCESS) {
        return CleanupWithError(errCode);
    }
#pragma endregion
 
#pragma region GRPC Server
    HANDLE grpcThread = CreateThread(
        NULL,                   // default security attributes
        0,                      // default stack size
        (LPTHREAD_START_ROUTINE)RunServiceServer,   // thread function
        &serviceContext,                   // no thread function arguments
        0,                      // default creation flags
        NULL                    // receive thread identifier
    );
    Sleep(2000);
 
    if (grpcThread == NULL || serviceContext.threadError) {
        return CleanupWithError(GRPC_SERVER_ERROR);
    }
#pragma endregion
 
#pragma region Container Start
    errCode = StartContainers(configuration->m_extensibility);
    if (errCode != PANO_SUCCESS) {
        return CleanupWithError(errCode);
    }
#pragma endregion
 
#pragma region ETW Trace
    HANDLE etwThread = CreateThread(
        NULL,                                       // default security attributes
        0,                                          // default stack size
        (LPTHREAD_START_ROUTINE)StartPanoptesTrace, // thread function
        &serviceContext,                            // no thread function arguments
        0,                                          // default creation flags
        NULL                                        // receive thread identifier
    );
    Sleep(2000);
    
    if (etwThread == NULL || serviceContext.threadError){
        return CleanupWithError(START_TRACE);
    }
#pragma endregion
 
#pragma region Driver Check
    //Check to see if the driver is installed and running/stopped
    if (!configuration->m_ignoreDriver) {
        errCode = GetKernelServiceStatus();
        if (errCode == NOT_INSTALLED)
        {
            return CleanupWithError(errCode);
        }
        else if (errCode == PANO_SERVICE_RUNNING) {
            errCode = StopWindowsDriver();
            if (errCode != PANO_SUCCESS)
            {
                return CleanupWithError(errCode);
            }
        }
        else if (errCode == PANO_SERVICE_STOPPED) {
            errCode = StartWindowsDriver();
            if (errCode != PANO_SUCCESS)
            {
                return CleanupWithError(errCode);
            }
        }
    }
#pragma endregion
 
    HANDLE proc = GetCurrentProcess();
    while (1) {
        WaitForSingleObject(proc, INFINITE);
    }
 
    return 0;
}

References CleanupWithError(), PanoptesContext::config, PanoptesContext::database, DEBUGGER, DisplayErrorMessage(), ERRORCODE, GetCurrentPath(), GetErrorMessage(), GetKernelServiceStatus(), GRPC_SERVER_ERROR, IsRunningAsAdmin(), Configuration::m_extensibility, Configuration::m_ignoreDriver, MUTEX_SET, MutexExist(), NOT_ADMIN, NOT_INSTALLED, PANO_SERVICE_RUNNING, PANO_SERVICE_STOPPED, PANO_SUCCESS, Configuration::Parse(), RunServiceServer(), serviceContext, SetEnvironmentMutex(), START_TRACE, StartContainers(), StartPanoptesTrace(), StartWindowsDriver(), StopWindowsDriver(), and PanoptesContext::threadError.