eunomia_core.h

Go to the documentation of this file.

/* SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
 *
 * Copyright (c) 2022, 郑昱笙，濮雯旭，张典典（牛校牛子队）
 * All rights reserved.
 */
 
#ifndef EUNOMIA_TRACKER_FACTORY_H
#define EUNOMIA_TRACKER_FACTORY_H
 
#include <optional>
 
#include "config.h"
#include "eunomia/config.h"
#include "eunomia/container_manager.h"
#include "eunomia/files.h"
#include "eunomia/ipc.h"
#include "eunomia/myseccomp.h"
#include "eunomia/process.h"
#include "eunomia/prometheus_server.h"
#include "eunomia/sec_analyzer.h"
#include "eunomia/tcp.h"
#include "eunomia/tracker_manager.h"
 
 
struct eunomia_core
{
 private:
  eunomia_config_data core_config;
 
  tracker_manager core_tracker_manager;
  container_manager core_container_manager;
  prometheus_server core_prometheus_server;
 
  std::shared_ptr<sec_analyzer> core_sec_analyzer;
 
 
  template<tracker_concept TRACKER>
  TRACKER::tracker_event_handler create_tracker_event_handler(const handler_config_data& config);
  template<tracker_concept TRACKER>
  TRACKER::tracker_event_handler create_tracker_event_handlers(const std::vector<handler_config_data>& handler_configs);
 
  template<tracker_concept TRACKER>
  TRACKER::tracker_event_handler create_print_event_handler(const TRACKER* tracker_ptr);
 
  template<tracker_concept TRACKER>
  std::unique_ptr<TRACKER> create_default_tracker(const tracker_config_data& base);
 
  template<tracker_concept TRACKER>
  std::unique_ptr<TRACKER> create_default_tracker_with_handler(
      const tracker_config_data& base,
      TRACKER::tracker_event_handler);
 
  template<tracker_concept TRACKER, typename SEC_ANALYZER_HANDLER>
  std::unique_ptr<TRACKER> create_default_tracker_with_sec_analyzer(const tracker_config_data& base);
 
  std::unique_ptr<process_tracker> create_process_tracker_with_container_tracking(const tracker_config_data& base);
 
  void start_trackers(void);
  void check_auto_exit(void);
  void start_prometheus_server(void);
  void start_container_manager(void);
  void start_sec_analyzer(void);
 
 public:
  eunomia_core(eunomia_config_data& config);
  int start_eunomia(void);
  std::optional<std::size_t> start_tracker(const tracker_config_data& config);
  std::vector<std::tuple<int, std::string>> list_all_trackers(void);
  void stop_tracker(std::size_t tracker_id);
};
 
#endif