grpc.hpp File Reference

#include <Windows.h>
#include <string>

Go to the source code of this file.

Classes
class	PanoptesContainerClient

Functions
void	SelfQueuePeScan (std::string pePath, std::string fileHash)
VOID	RunServiceServer (LPVOID lpParam)

Function Documentation

RunServiceServer()

VOID RunServiceServer

(

LPVOID

lpParam

)

Definition at line 371 of file grpc.cpp.

{
    serviceContext = reinterpret_cast<PanoptesContext*>(lpParam);
    PanoptesImpl service;
    grpc::ServerBuilder builder;
    int selected_port = 0;
    std::string server_url = "localhost:0";
 
    //Setting the server address to localhost:0 will allow the OS to assign an available port
    builder.AddListeningPort(server_url, grpc::InsecureServerCredentials(), &selected_port);
    builder.RegisterService(&service);
 
    std::unique_ptr<grpc::Server> server(builder.BuildAndStart());
    if (server == nullptr) {
        auto threadError = &serviceContext->threadError;
        bool threadState = threadError->load();
        threadError->store(true);
    }
 
    bool updatedReg = CreateRegistryEntryWithPort(selected_port);
    if (!updatedReg) {
        auto threadError = &serviceContext->threadError;
        bool threadState = threadError->load();
        threadError->store(true);
    }
 
    server->Wait();
}

References CreateRegistryEntryWithPort(), serviceContext, and PanoptesContext::threadError.

Referenced by WinMain().

SelfQueuePeScan()

void SelfQueuePeScan	(	std::string	pePath,
		std::string	fileHash
	)

Definition at line 224 of file grpc.cpp.

                                                           {
    auto configuration = serviceContext->config;
    if (isPathInExclusions(configuration->m_exclusions, pePath)) {
        return;
    }
 
    DWORD containerPort = 0;
    GetRegistryPortValue(containerPort);
    std::string server_url = "localhost:" + std::to_string(containerPort);
    std::shared_ptr<grpc::Channel> channel = grpc::CreateChannel(server_url, grpc::InsecureChannelCredentials());
    selfStub_ = PanoptesService::NewStub(channel);
 
    AckMessage reply;
    ClientContext context;
    PeScanInfo request;
    request.set_file_hash(fileHash);
    request.set_portable_executable_path(pePath);
 
    Status status = selfStub_->QueuePeScan(&context, request, &reply);
    if (!status.ok()) {
        std::cout << status.error_code() << ": " << status.error_message() << std::endl;
    }
}

References PanoptesContext::config, GetRegistryPortValue(), isPathInExclusions(), selfStub_, and serviceContext.

Referenced by DisplayEventInfo().