zeek/src/iosource/pcap/Source.h

47 lines
914 B
C++

// See the file "COPYING" in the main distribution directory for copyright.
#pragma once
#include "../PktSrc.h"
extern "C" {
#include <pcap.h>
}
#include <sys/types.h> // for u_char
namespace iosource {
namespace pcap {
class PcapSource : public iosource::PktSrc {
public:
PcapSource(const std::string& path, bool is_live);
~PcapSource() override;
static PktSrc* Instantiate(const std::string& path, bool is_live);
protected:
// PktSrc interface.
void Open() override;
void Close() override;
bool ExtractNextPacket(Packet* pkt) override;
void DoneWithPacket() override;
bool PrecompileFilter(int index, const std::string& filter) override;
bool SetFilter(int index) override;
void Statistics(Stats* stats) override;
private:
void OpenLive();
void OpenOffline();
void PcapError(const char* where = 0);
Properties props;
Stats stats;
pcap_t *pd;
struct pcap_pkthdr current_hdr;
};
}
}