#include <grpcpp/grpcpp.h>
#include "panoptes.grpc.pb.h"
#include "panoptes.pb.h"
#include "container_ipc.hpp"
#include <ExtensibilityCore.h>
#include <iostream>
#include <atomic>

Functions
bool	GetRegistryPortValue (DWORD &portValue)
	Get the gRPC port value from the registry.

void	MessageQueueThread ()
	The MessageQueueThread function is a thread that dequeues messages from the message queue and processes them in the PanoEntry function of the loaded extensibility DLL.

void	HealthCheck ()
	The HealthCheck function is a thread that checks the health of the Panoptes main service.

bool	LoadExtensibility (std::string dllFile)
	The LoadExtensibility function is a function that loads the extensibility DLL.

bool	BindToServiceWithRetry (int containerPort, int maxAttempts, int baseSleep, double multiplier)
	The BindToServiceWithRetry function is a function that binds to the Panoptes main service with a retry mechanism.

int WINAPI	WinMain (HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow)
	The WinMain function is the entry point for the container.

Variables
ExtensibilityCore::PanoBindPtr	PanoBind

ExtensibilityCore::PanoEntryPtr	PanoEntry

ExtensibilityCore::PanoUnbindPtr	PanoUnbind

PeMessageQueue *	message_queue_pe_

MemoryMessageQueue *	message_queue_mem_

std::atomic_bool	isConnected = true

std::unique_ptr< PanoptesService::Stub >	stub

int	BindPort = 0

Function Documentation

◆ BindToServiceWithRetry()

bool BindToServiceWithRetry	(	int	containerPort,
		int	maxAttempts,
		int	baseSleep,
		double	multiplier
	)

The BindToServiceWithRetry function is a function that binds to the Panoptes main service with a retry mechanism.

Parameters

containerPort	The port value that is read from the registry.
maxAttempts	The maximum number of attempts to bind to the service.
baseSleep	The base sleep time in milliseconds.
multiplier	The multiplier for the sleep time.

Returns: True if the service is bound successfully, false otherwise.

Definition at line 149 of file container.cpp.

                                                                                                  {
 
    for (int attempt = 0; attempt < maxAttempts; ++attempt) {
        PanoBind(containerPort);
        if (BindPort > 0)
            break;
 
        if (attempt < maxAttempts - 1) {  // Don't sleep after the last attempt
            int sleep_duration_ms = baseSleep * std::pow(multiplier, attempt);
            std::cout << "Sleeping for " << sleep_duration_ms << " ms" << std::endl;
            std::this_thread::sleep_for(std::chrono::milliseconds(sleep_duration_ms));
        }
    }
 
    return true;
}

References BindPort, and PanoBind.

Referenced by WinMain().

◆ GetRegistryPortValue()

bool GetRegistryPortValue ( DWORD & portValue )

Get the gRPC port value from the registry.

Parameters

portValue The port value that is read from the registry.

Returns: True if the port value is read successfully, false otherwise.

Definition at line 24 of file container.cpp.

{
    HKEY hKey;
    DWORD dwType = REG_DWORD;
    DWORD dwSize = sizeof(DWORD);
 
    // Open the key
    LONG lResult = RegOpenKeyExA(
        HKEY_LOCAL_MACHINE,
        "SOFTWARE\\Panoptes",
        0,
        KEY_READ,
        &hKey
    );
 
    if (lResult != ERROR_SUCCESS) {
        std::cerr << "Error opening registry key. Error code: " << lResult << std::endl;
        return false;
    }
 
    // Read the SRV_PORT value
    lResult = RegQueryValueExA(
        hKey,
        "SRV_PORT",
        NULL,
        &dwType,
        reinterpret_cast<LPBYTE>(&portValue),
        &dwSize
    );
 
    RegCloseKey(hKey);
 
    if (lResult != ERROR_SUCCESS) {
        std::cerr << "Error reading registry value. Error code: " << lResult << std::endl;
        return false;
    }
 
    if (dwType != REG_DWORD) {
        std::cerr << "Unexpected value type in registry." << std::endl;
        return false;
    }
 
    return true;
}

Referenced by HealthCheck().

◆ HealthCheck()

void HealthCheck ( )

The HealthCheck function is a thread that checks the health of the Panoptes main service.

Definition at line 83 of file container.cpp.

                   {
    DWORD portValue = 0;
    if (!GetRegistryPortValue(portValue)) {
        std::cerr << "Failed to get registry port value." << std::endl;
        return;
    }
    std::string server_url = "localhost:" + std::to_string(portValue);
 
    while (true)
    {
        std::shared_ptr<grpc::Channel> channel = grpc::CreateChannel(server_url, 
        grpc::InsecureChannelCredentials());
 
        stub = PanoptesService::NewStub(channel);
 
        grpc::ClientContext context;
        HealthCheckRequest request;
        HealthCheckResponse response;
        request.set_ping(true);
 
 
        grpc::Status status = stub->HealthCheck(&context, request, &response);
 
        if (!status.ok()) {
            isConnected.store(false);
            return;
        }
 
        Sleep(2500);
    }
}

References GetRegistryPortValue(), isConnected, and stub.

Referenced by WinMain().

◆ LoadExtensibility()

bool LoadExtensibility ( std::string dllFile )

The LoadExtensibility function is a function that loads the extensibility DLL.

Parameters

dllFile The path to the extensibility DLL.

Returns: True if the extensibility DLL is loaded successfully, false otherwise.

Definition at line 118 of file container.cpp.

                                          {
    HMODULE extensibility = LoadLibraryA(dllFile.c_str());
    if (extensibility == NULL) {
        DWORD error = GetLastError();
        return false;
    }
    else {
        PanoBind = (ExtensibilityCore::PanoBindPtr)GetProcAddress(extensibility, "PanoBind");
        if (PanoBind == NULL) {
            return false;
        }
        PanoEntry = (ExtensibilityCore::PanoEntryPtr)GetProcAddress(extensibility, "PanoEntry");
        if (PanoEntry == NULL) {
            return false;
        }
        PanoUnbind = (ExtensibilityCore::PanoUnbindPtr)GetProcAddress(extensibility, "PanoUnbind");
        if (PanoUnbind == NULL) {
            return false;
        }
    }
 
    return true;
}

References PanoBind, PanoEntry, and PanoUnbind.

Referenced by WinMain().

◆ MessageQueueThread()

void MessageQueueThread ( )

The MessageQueueThread function is a thread that dequeues messages from the message queue and processes them in the PanoEntry function of the loaded extensibility DLL.

Definition at line 71 of file container.cpp.

                          {
    message_queue_pe_ = new PeMessageQueue();
    while (isConnected) {
        PeScan pe_data = message_queue_pe_->dequeue();
        if (pe_data.PePath != "")
        {
            PanoEntry(&pe_data, NULL);
        }
    }
}

References PeMessageQueue::dequeue(), isConnected, message_queue_pe_, PanoEntry, and PeScan::PePath.

Referenced by WinMain().

◆ WinMain()

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

The WinMain function is the entry point for the container.

Parameters

hInstance	The instance of the application.
hPrevInstance	The previous instance of the application.
lpCmdLine	The command line arguments.
nCmdShow	The command line arguments.

Returns: The exit code of the application.

Definition at line 172 of file container.cpp.

{
    if (__argc < 1) {
        printf("No Args");
        exit(1);
    }
 
    std::string dllPath(__argv[1]);
    if (!LoadExtensibility(dllPath)) {
        printf("Error Loading Extensibility");
        exit(1);
    }
 
    HANDLE grpcServerThread = CreateThread(
        NULL,                   // default security attributes
        0,                      // default stack size
        (LPTHREAD_START_ROUTINE)RunContainerServer,   // thread function
        &BindPort,                   // no thread function arguments
        0,                      // default creation flags
        NULL                    // receive thread identifier
    );
    Sleep(2000);
    if (grpcServerThread == NULL || BindPort == 0)
    {
        std::cout << "Error creating thread: " << GetLastError() << std::endl;
        PanoUnbind();
        return 1;
    }
 
    if (!BindToServiceWithRetry(BindPort, 7, 1000, 2))
    {
        std::cout << "Error Binding to service: " << GetLastError() << std::endl;
        PanoUnbind();
        return 1;
    }
 
    HANDLE healthCheckThread = CreateThread(
        NULL,                   // default security attributes
        0,                      // default stack size
        (LPTHREAD_START_ROUTINE)HealthCheck,   // thread function
        NULL,                   // no thread function arguments
        0,                      // default creation flags
        NULL                    // receive thread identifier
    );
    Sleep(2000);
    if (healthCheckThread == NULL || BindPort == 0)
    {
        std::cout << "Error creating thread: " << GetLastError() << std::endl;
 
        PanoUnbind();
        return 1;
    }
 
    HANDLE messageQueueThread = CreateThread(
        NULL,                   // default security attributes
        0,                      // default stack size
        (LPTHREAD_START_ROUTINE)MessageQueueThread,   // thread function
        NULL,                   // no thread function arguments
        0,                      // default creation flags
        NULL                    // receive thread identifier
    );
    Sleep(2000);
    if (healthCheckThread == NULL || BindPort == 0)
    {
        std::cout << "Error creating thread: " << GetLastError() << std::endl;
        PanoUnbind();
        return 1;
    }
 
    while (true)
    {
        if (isConnected == false) {
            exit(1);
        }
        Sleep(2500);
    }
 
    return 0;
}