process.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 PROCESS_CMD_H
#define PROCESS_CMD_H
 
#include <string>
 
#include "libbpf_print.h"
#include "model/tracker.h"
#include "prometheus/counter.h"
#include "prometheus_server.h"
 
extern "C" {
#include <process/process_tracker.h>
}
 
 
struct process_tracker : public tracker_with_config<process_env, process_event>
{
  process_tracker(config_data config);
 
  static std::unique_ptr<process_tracker> create_tracker_with_default_env(tracker_event_handler handler);
  static std::unique_ptr<process_tracker> create_tracker_with_args(
      tracker_event_handler handler,
      const std::vector<std::string> &args)
  {
    return create_tracker_with_default_env(handler);
  }
 
  void start_tracker();
 
  struct prometheus_event_handler : public event_handler<process_event>
  {
    prometheus::Family<prometheus::Counter> &eunomia_process_start_counter;
    prometheus::Family<prometheus::Counter> &eunomia_process_exit_counter;
    void report_prometheus_event(const struct process_event &e, const struct container_info& ct_info);
 
    prometheus_event_handler(prometheus_server &server);
    void handle(tracker_event<process_event> &e);
  };
 
  struct json_event_handler_base : public event_handler<process_event>
  {
    std::string to_json(const struct process_event &e);
  };
 
  struct json_event_printer : public json_event_handler_base
  {
    void handle(tracker_event<process_event> &e);
  };
 
  struct plain_text_event_printer : public event_handler<process_event>
  {
    void handle(tracker_event<process_event> &e);
  };
 
  struct csv_event_printer : public event_handler<process_event>
  {
    void handle(tracker_event<process_event> &e);
  };
 
};
 
#endif