Configuration.hpp

Go to the documentation of this file.

#pragma once
#include <string>
#include <fstream>
#include <vector>
#include <nlohmann/json.hpp>
 
class Configuration {
protected:
    nlohmann::json m_data;
    std::ifstream m_fileStream;
 
    void GetExtensibilitySelected();
 
    void GetScannerExclusions();
 
    void GetIgnoreDriver();
 
    void GetQuarantineMaliciousFiles();
 
    void GetEventProviders();
 
public:
    enum ContainerType : int {
      CONTAINER_TYPE_NONE = 0,
      CONTAINER_TYPE_AMSI = 10,
      CONTAINER_TYPE_PE = 20,
      CONTAINER_TYPE_YARA = 30
    };
 
    std::vector<std::string> m_exclusions;
 
    std::vector<Configuration::ContainerType> m_extensibility;
 
    std::vector<std::string> m_extensibilityListName;
 
    std::vector <std::tuple<std::string, unsigned long, unsigned long>> m_eventProviders;
 
    bool m_ignoreDriver = false;
 
    bool m_quartine = false;
 
    Configuration(std::string configurationPath);
 
    void Parse();
 
    void IsValidJson();
 
    std::vector<std::string> GetJsonKeys();
};