]> git.stg.codes - stg.git/blobdiff - tests/test_raw_ip.cpp
Use std::lock_guard instead of STG_LOCKER.
[stg.git] / tests / test_raw_ip.cpp
index 90a2cdd5a44758ec193cdfa6a196bc2c8ebb770e..4c7ce121c5e05a83f02ed594d996fa8fd5d85807 100644 (file)
-#include <sys/socket.h>
-#include <netinet/in.h>
-#include <arpa/inet.h>
+#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 <boost/test/unit_test.hpp>
+#pragma GCC diagnostic pop
 
 #include <cstdlib>
 #include <cstdint>
 #include <ctime>
 #include <iostream>
 
-#include "tut/tut.hpp"
-
-#include "raw_ip_packet_old.h"
-#include "stg/raw_ip_packet.h"
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
 
-#ifndef ITERATIONS
-#define ITERATIONS 1000
-#endif
+namespace
+{
 
-void genVector(uint8_t * buf);
+constexpr size_t ITERATIONS = 1000;
 
-namespace STG
+std::array<uint8_t, 68> genVector()
 {
-std::ostream & operator<<(std::ostream & stream, const RawPacket & p);
+    std::array<uint8_t, 68> res;
+    for (size_t i = 0; i < res.size(); ++i)
+        res[i] = rand() % 256;
+    res[0] = (res[0] & 0xF0) | 0x05; // Fix header length
+    return res;
 }
 
-namespace tut
-{
-    struct rp_data {
-    };
-
-    typedef test_group<rp_data> 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<size_t>(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<uint32_t>(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)
+namespace std
 {
-    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)
+std::ostream & operator<<(std::ostream& stream, const STG::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<unsigned>(p.rawPacket.data[i]);
     }
-    stream.unsetf(std::ios::hex);
     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<size_t>(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<uint32_t>(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)
+    {
+        RawPacketOld 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()