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>
 
namespace yarilo {
 
class Recording {
public:
  enum class DataLinkType {
    RADIOTAP,
    RAW80211,
    ETH2,
  };
 
  struct info {
    uuid::UUIDv4 uuid;
    std::string filename;
    std::string display_name;
    DataLinkType datalink;
    uint32_t count;
  };
 
  Recording(const std::filesystem::path &save_dir, bool dump_raw);
 
  void set_name(const std::string &basename) { this->basename = basename; }
 
  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 filename for the recording file.
   * @return The generated filename as a path object.
   */
  std::filesystem::path generate_filename() const;
 
  std::shared_ptr<spdlog::logger> logger;
  const std::filesystem::path save_dir;
  const bool dump_raw = false;
  std::string basename = "recording";
  uuid::UUIDv4 uuid;
};
 
} // namespace yarilo
 
#endif // SNIFF_RECORDING