#include #include #include #include #include #include #include #include "tut/tut.hpp" #include "raw_ip_packet_old.h" #include "stg/raw_ip_packet.h" #ifndef ITERATIONS #define ITERATIONS 1000 #endif void genVector(uint8_t * buf); namespace STG { std::ostream & operator<<(std::ostream & stream, const RawPacket & p); } namespace tut { struct rp_data { }; typedef test_group tg; tg rp_test_group("STG::RawPacket tests group"); typedef tg::object testobject; template<> template<> void testobject::test<1>() { set_test_name("Check structure consistency"); STG::RawPacket rp; rp.rawPacket.header.ipHeader.ip_v = 4; rp.rawPacket.header.ipHeader.ip_hl = 5; rp.rawPacket.header.ipHeader.ip_tos = 0; rp.rawPacket.header.ipHeader.ip_len = htons(40); // 20 of header + 20 of data rp.rawPacket.header.ipHeader.ip_p = 6; rp.rawPacket.header.ipHeader.ip_src.s_addr = inet_addr("192.168.0.1"); rp.rawPacket.header.ipHeader.ip_dst.s_addr = inet_addr("192.168.0.101"); rp.rawPacket.header.sPort = htons(80); rp.rawPacket.header.dPort = htons(38546); ensure_equals("IP header size (explicitly)", sizeof(rp.rawPacket.header.ipHeader), static_cast(20)); ensure_equals("IP version", rp.GetIPVersion(), 4); ensure_equals("IP header size (with options)", rp.GetHeaderLen(), 20); ensure_equals("Underlying protocol version", rp.GetProto(), 6); ensure_equals("Packet length", rp.GetLen(), static_cast(40)); ensure_equals("Source IP address", rp.GetSrcIP(), inet_addr("192.168.0.1")); ensure_equals("Destination IP address", rp.GetDstIP(), inet_addr("192.168.0.101")); ensure_equals("Source port number", rp.GetSrcPort(), 80); ensure_equals("Destination port number", rp.GetDstPort(), 38546); } template<> template<> void testobject::test<2>() { srand(time(NULL)); for (size_t i = 0; i < ITERATIONS; ++i) { RAW_PACKET_OLD p1; STG::RawPacket p2; STG::RawPacket p3; uint8_t buf[68]; genVector(buf); memcpy(p1.pckt, buf, 68); memcpy(p2.rawPacket.data, buf, 68); memcpy(p3.rawPacket.data, buf, 68); ensure_equals("IP versions", p1.GetIPVersion(), p2.GetIPVersion()); ensure_equals("IP headers length", p1.GetHeaderLen(), p2.GetHeaderLen()); ensure_equals("Protocols", p1.GetProto(), p2.GetProto()); ensure_equals("Source IPs", p1.GetSrcIP(), p2.GetSrcIP()); ensure_equals("Destination IPs", p1.GetDstIP(), p2.GetDstIP()); ensure_equals("Source ports", p1.GetSrcPort(), p2.GetSrcPort()); ensure_equals("Destination ports", p1.GetDstPort(), p2.GetDstPort()); ensure_equals("Self equallity", p2, p3); ensure_equals("Reverse self equallity", p3, p2); } } } inline void genVector(uint8_t * buf) { for (size_t i = 0; i < 68; ++i) { buf[i] = rand() % 256; } buf[0] = (buf[0] & 0xF0) | 0x05; // Fix header length } std::ostream & STG::operator<<(std::ostream & stream, const RawPacket & p) { stream.unsetf(std::ios::dec); stream.setf(std::ios::hex); for (size_t i = 0; i < sizeof(p.rawPacket.data); ++i) { stream << static_cast(p.rawPacket.data[i]); } stream.unsetf(std::ios::hex); stream.setf(std::ios::dec); return stream; }