pdns/dnspcap.hh

   1 #ifndef PDNS_DNSPCAP_HH
   2 #define PDNS_DNSPCAP_HH
   3
   4 #include <cstdio>
   5 #include <stdexcept>
   6 #include "iputils.hh"
   7 #include <string>
   8 #include "misc.hh"
   9 #include <iostream>
  10 #define __FAVOR_BSD
  11 #include <netinet/in_systm.h>
  12 #include <netinet/ip.h>
  13 #include <netinet/ip6.h>
  14 #include <netinet/udp.h>
  15 #if defined(__NetBSD__)
  16 #include <net/if.h>
  17 #include <net/if_ether.h>
  18 #elif defined (__OpenBSD__)
  19 #include <net/if.h>
  20 #include <netinet/if_ether.h>
  21 #elif defined (__SVR4) && defined (__sun)
  22 #include <sys/ethernet.h>
  23 #else
  24 #include <net/ethernet.h>
  25 #endif
  26 #include <vector>
  27 #include <boost/format.hpp>
  28 #include "namespaces.hh"
  29
  30 struct pdns_pcap_file_header {
  31   uint32_t magic;
  32   uint16_t version_major;
  33   uint16_t version_minor;
  34   uint32_t thiszone;     /* gmt to local correction */
  35   uint32_t sigfigs;    /* accuracy of timestamps */
  36   uint32_t snaplen;    /* max length saved portion of each pkt */
  37   uint32_t linktype;   /* data link type (LINKTYPE_*) */
  38 };
  39
  40
  41 struct pdns_timeval
  42 {
  43   uint32_t tv_sec;
  44   uint32_t tv_usec;
  45 };
  46
  47 struct pdns_pcap_pkthdr {
  48   struct pdns_timeval ts;      /* time stamp */
  49   uint32_t caplen;     /* length of portion present */
  50   uint32_t len;        /* length this packet (off wire) */
  51 };
  52
  53 struct pdns_lcc_header {
  54   uint16_t lcc_pkttype;/* packet type */
  55   uint16_t lcc_hatype;/* link-layer address type */
  56   uint16_t lcc_halen;/* link-layer address length */
  57   uint8_t lcc_addr[8];/* link-layer address */
  58   uint16_t lcc_protocol;/* protocol */
  59 };
  60
  61 class PcapPacketReader
  62 {
  63 public:
  64   class EofException : public runtime_error
  65   {
  66   public:
  67     EofException(const string& str="") : runtime_error(str)
  68     {
  69     }
  70   };
  71
  72   PcapPacketReader(const string& fname);
  73
  74   ~PcapPacketReader();
  75
  76   template<typename T>
  77   void checkedFread(T* ptr)
  78   {
  79     checkedFreadSize(ptr, sizeof(*ptr));
  80   }
  81
  82   void checkedFreadSize(void* ptr, size_t size) ;
  83
  84   bool getUDPPacket();
  85
  86
  87   ComboAddress getSource() const;
  88   ComboAddress getDest() const;
  89
  90   struct pdns_lcc_header* d_lcc;
  91   struct ether_header* d_ether;
  92   struct ip *d_ip;
  93   struct ip6_hdr *d_ip6;
  94   const struct tcphdr *d_tcp;
  95   const struct udphdr *d_udp;
  96   const uint8_t* d_payload;
  97   unsigned int d_len;
  98   struct pdns_pcap_pkthdr d_pheader;
  99
 100   pdns_pcap_file_header d_pfh;
 101   unsigned int d_runts, d_oversized, d_correctpackets, d_nonetheripudp;
 102   char d_buffer[32768];
 103 private:
 104   FILE* d_fp;
 105   string d_fname;
 106   int d_skipMediaHeader;
 107 };
 108
 109 class PcapPacketWriter
 110 {
 111 public:
 112   PcapPacketWriter(const string& fname, PcapPacketReader& ppr);
 113
 114   void write();
 115
 116   ~PcapPacketWriter();
 117
 118 private:
 119   string d_fname;
 120   const PcapPacketReader& d_ppr;
 121
 122   FILE *d_fp;
 123 };
 124
 125 #endif // DNSPCAP_HH