#define BOOST_TEST_MODULE STGRawIP #include "raw_ip_packet_old.h" #include "stg/raw_ip_packet.h" #pragma GCC diagnostic push #pragma GCC diagnostic ignored "-Wold-style-cast" #pragma GCC diagnostic ignored "-Wunused-parameter" #pragma GCC diagnostic ignored "-Wsign-compare" #pragma GCC diagnostic ignored "-Wparentheses" #include #pragma GCC diagnostic pop #include #include #include #include #include #include #include namespace { constexpr size_t ITERATIONS = 1000; std::array genVector() { std::array res; for (size_t i = 0; i < 68; ++i) res[i] = rand() % 256; res[0] = (res[0] & 0xF0) | 0x05; // Fix header length return res; } } namespace std { std::ostream & operator<<(std::ostream& stream, const STG::RawPacket& p) { stream.setf(std::ios::hex); for (size_t i = 0; i < sizeof(p.rawPacket.data); ++i) { stream << static_cast(p.rawPacket.data[i]); } stream.setf(std::ios::dec); return stream; } } BOOST_AUTO_TEST_SUITE(RawIP) BOOST_AUTO_TEST_CASE(StructureConsistency) { 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); BOOST_CHECK_EQUAL(sizeof(rp.rawPacket.header.ipHeader), static_cast(20)); BOOST_CHECK_EQUAL(rp.GetIPVersion(), 4); BOOST_CHECK_EQUAL(rp.GetHeaderLen(), 20); BOOST_CHECK_EQUAL(rp.GetProto(), 6); BOOST_CHECK_EQUAL(rp.GetLen(), static_cast(40)); BOOST_CHECK_EQUAL(rp.GetSrcIP(), inet_addr("192.168.0.1")); BOOST_CHECK_EQUAL(rp.GetDstIP(), inet_addr("192.168.0.101")); BOOST_CHECK_EQUAL(rp.GetSrcPort(), 80); BOOST_CHECK_EQUAL(rp.GetDstPort(), 38546); } BOOST_AUTO_TEST_CASE(RandomTests) { srand(time(NULL)); for (size_t i = 0; i < ITERATIONS; ++i) { RAW_PACKET_OLD p1; STG::RawPacket p2; STG::RawPacket p3; const auto buf = genVector(); memcpy(p1.pckt, buf.data(), 68); memcpy(p2.rawPacket.data, buf.data(), 68); memcpy(p3.rawPacket.data, buf.data(), 68); BOOST_CHECK_EQUAL(p1.GetIPVersion(), p2.GetIPVersion()); BOOST_CHECK_EQUAL(p1.GetHeaderLen(), p2.GetHeaderLen()); BOOST_CHECK_EQUAL(p1.GetProto(), p2.GetProto()); BOOST_CHECK_EQUAL(p1.GetSrcIP(), p2.GetSrcIP()); BOOST_CHECK_EQUAL(p1.GetDstIP(), p2.GetDstIP()); BOOST_CHECK_EQUAL(p1.GetSrcPort(), p2.GetSrcPort()); BOOST_CHECK_EQUAL(p1.GetDstPort(), p2.GetDstPort()); BOOST_CHECK_EQUAL(p2, p3); BOOST_CHECK_EQUAL(p3, p2); } } BOOST_AUTO_TEST_SUITE_END()