pcap.hpp

Go to the documentation of this file.

#pragma once
 
 
#include <cstddef>
#include <cstdint>
#include <optional>
#include <span>
#include <string>
#include <vector>
 
#include "itch/parser.hpp"
#include "itch/transport/moldudp64.hpp"
 
namespace itch::transport {
 
class PcapReader {
   public:
    explicit PcapReader(MessageCallback callback);
 
    auto read(std::span<const std::byte> capture) -> bool;
 
    auto read_file(const std::string& path) -> bool;
 
    auto set_udp_port_filter(std::uint16_t port) -> void { m_port_filter = port; }
 
    auto clear_udp_port_filter() noexcept -> void { m_port_filter = std::nullopt; }
 
    [[nodiscard]] auto mold_decoder() noexcept -> MoldUdp64Decoder& { return m_mold; }
    [[nodiscard]] auto mold_decoder() const noexcept -> const MoldUdp64Decoder& { return m_mold; }
 
    [[nodiscard]] auto udp_datagrams() const noexcept -> std::uint64_t { return m_udp_datagrams; }
 
    [[nodiscard]] auto messages_decoded() const noexcept -> std::uint64_t {
        return m_mold.messages_decoded();
    }
 
   private:
    auto read_classic_pcap(std::span<const std::byte> capture, bool swapped) -> bool;
 
    auto read_pcapng(std::span<const std::byte> capture) -> bool;
 
    auto handle_frame(std::span<const std::byte> frame, std::uint32_t link_type) -> void;
 
    [[nodiscard]] auto extract_udp_payload(std::span<const std::byte> network
    ) const -> std::optional<std::span<const std::byte>>;
 
    MoldUdp64Decoder             m_mold;
    std::optional<std::uint16_t> m_port_filter {};
    std::uint64_t                m_udp_datagrams {0};
};
 
}  // namespace itch::transport