tests/test_raw_ip.cpp

   1 #include <sys/socket.h>
   2 #include <netinet/in.h>
   3 #include <arpa/inet.h>
   4
   5 #include <cstdlib>
   6 #include <ctime>
   7 #include <iostream>
   8
   9 #include "tut/tut.hpp"
  10
  11 #include "raw_ip_packet_old.h"
  12 #include "raw_ip_packet.h"
  13
  14 #ifndef ITERATIONS
  15 #define ITERATIONS 1000
  16 #endif
  17
  18 void genVector(uint8_t * buf);
  19
  20 std::ostream & operator<<(std::ostream & stream, const RAW_PACKET & p);
  21
  22 namespace tut
  23 {
  24     struct rp_data {
  25     };
  26
  27     typedef test_group<rp_data> tg;
  28     tg rp_test_group("RAW_PACKET tests group");
  29
  30     typedef tg::object testobject;
  31
  32     template<>
  33     template<>
  34     void testobject::test<1>()
  35     {
  36         set_test_name("Check structure consistency");
  37
  38         RAW_PACKET rp;
  39         rp.ipHeader.ip_v = 4;
  40         rp.ipHeader.ip_hl = 5;
  41         rp.ipHeader.ip_tos = 0;
  42         rp.ipHeader.ip_len = htons(40); // 20 of header + 20 of data
  43         rp.ipHeader.ip_p = 6;
  44         rp.ipHeader.ip_src.s_addr = inet_addr("192.168.0.1");
  45         rp.ipHeader.ip_dst.s_addr = inet_addr("192.168.0.101");
  46         rp.sPort = htons(80);
  47         rp.dPort = htons(38546);
  48
  49         ensure_equals("IP header size (explicitly)", sizeof(rp.ipHeader), 20);
  50         ensure_equals("IP version", rp.GetIPVersion(), 4);
  51         ensure_equals("IP header size (with options)", rp.GetHeaderLen(), 20);
  52         ensure_equals("Underlying protocol version", rp.GetProto(), 6);
  53         ensure_equals("Packet length", rp.GetLen(), 40);
  54         ensure_equals("Source IP address", rp.GetSrcIP(), inet_addr("192.168.0.1"));
  55         ensure_equals("Destination IP address", rp.GetDstIP(), inet_addr("192.168.0.101"));
  56         ensure_equals("Source port number", rp.GetSrcPort(), 80);
  57         ensure_equals("Destination port number", rp.GetDstPort(), 38546);
  58     }
  59
  60     template<>
  61     template<>
  62     void testobject::test<2>()
  63     {
  64         srand(time(NULL));
  65         for (size_t i = 0; i < ITERATIONS; ++i) {
  66             RAW_PACKET_OLD p1;
  67             RAW_PACKET p2;
  68             RAW_PACKET p3;
  69
  70             uint8_t buf[68];
  71             genVector(buf);
  72
  73             memcpy(p1.pckt, buf, 68);
  74             memcpy(p2.pckt, buf, 68);
  75             memcpy(p3.pckt, buf, 68);
  76
  77             ensure_equals("IP versions", p1.GetIPVersion(), p2.GetIPVersion());
  78             ensure_equals("IP headers length", p1.GetHeaderLen(), p2.GetHeaderLen());
  79             ensure_equals("Protocols", p1.GetProto(), p2.GetProto());
  80             ensure_equals("Source IPs", p1.GetSrcIP(), p2.GetSrcIP());
  81             ensure_equals("Destination IPs", p1.GetDstIP(), p2.GetDstIP());
  82             ensure_equals("Source ports", p1.GetSrcPort(), p2.GetSrcPort());
  83             ensure_equals("Destination ports", p1.GetDstPort(), p2.GetDstPort());
  84
  85             ensure_equals("Self equallity", p2, p3);
  86             ensure_equals("Reverse self equallity", p3, p2);
  87         }
  88     }
  89 }
  90
  91 inline
  92 void genVector(uint8_t * buf)
  93 {
  94     for (size_t i = 0; i < 68; ++i) {
  95         buf[i] = rand() % 256;
  96     }
  97     buf[0] = (buf[0] & 0xF0) | 0x05; // Fix header length
  98 }
  99
 100 std::ostream & operator<<(std::ostream & stream, const RAW_PACKET & p)
 101 {
 102     stream.unsetf(std::ios::dec);
 103     stream.setf(std::ios::hex);
 104     for (size_t i = 0; i < sizeof(p.pckt); ++i) {
 105         stream << static_cast<unsigned>(p.pckt[i]);
 106     }
 107     stream.unsetf(std::ios::hex);
 108     stream.setf(std::ios::dec);
 109     return stream;
 110 }