recording.h

Go to the documentation of this file.

#ifndef SNIFF_RECORDING
#define SNIFF_RECORDING
 
#include "channel.h"
#include "uuid.h"
#include <filesystem>
#include <memory>
#include <optional>
#include <spdlog/logger.h>
#include <tins/snap.h>
 
#include "database.h"
 
namespace yarilo {
 
class Recording {
public:
  enum class DataLinkType {
    UNKNOWN,
    RADIOTAP,
    RAW80211,
    ETH2,
  };
 
  struct info {
    uuid::UUIDv4 uuid;
    std::string filename;
    std::string display_name;
    DataLinkType datalink;
    uint32_t count;
 
    uuid::UUIDv4 get_uuid() const { return uuid; }
  };
 
  Recording(const std::filesystem::path &save_dir, bool dump_raw, Database &db,
            const std::string &display_name);
 
  std::optional<info> dump(std::shared_ptr<PacketChannel> channel) const;
 
  std::optional<info> dump(std::vector<Tins::Packet *> *packets) const;
 
  static std::unique_ptr<Tins::Packet> make_eth_packet(Tins::Packet *pkt);
 
private:
  /*
   * Generates a path for the recording file.
   * @return The generated recording path as a path object.
   */
  std::filesystem::path generate_filepath() const;
 
  std::shared_ptr<spdlog::logger> logger;
  const std::filesystem::path save_dir;
  const bool dump_raw;
  const std::string basename;
  const uuid::UUIDv4 uuid;
  Database &db;
};
 
} // namespace yarilo
 
#endif // SNIFF_RECORDING