X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/a258b9f408911f1a0e6f221180ee9993d100f58b..e6c0bd53164bd30bbe038c80a370707847fc266c:/include/stg/raw_ip_packet.h

diff --git a/include/stg/raw_ip_packet.h b/include/stg/raw_ip_packet.h
index 16e43402..663ff1bc 100644
--- a/include/stg/raw_ip_packet.h
+++ b/include/stg/raw_ip_packet.h
@@ -1,170 +1,185 @@
-#ifndef RAW_IP_PACKET_H
-#define RAW_IP_PACKET_H
+#pragma once
 
-#if defined(FREE_BSD) || defined(FREE_BSD5)
+#include <cstring>
+
+#if defined(FREE_BSD)
+#include <sys/types.h> // u_char, u_int32_t, etc.
 #include <netinet/in_systm.h> // n_long in netinet/ip.h
 #endif
 
 #include <netinet/in.h> // for htons
 #include <netinet/ip.h> // for struct ip
 
-#include <cstring>
-
-#include "const.h"
-#include "common.h"
-
 #define IPv4 (2)
 
-enum { pcktSize = 68 }; //60(max) ip + 8 udp or tcp (part of tcp or udp header to ports)
+namespace STG
+{
+
+enum { packetSize = 68 }; //60(max) ip + 8 udp or tcp (part of tcp or udp header to ports)
 //-----------------------------------------------------------------------------
-struct RAW_PACKET
+struct RawPacket
 {
-    RAW_PACKET()
-        : rawPacket(),
-          dataLen(-1)
+    RawPacket()
+        : dataLen(-1)
     {
-    memset(rawPacket.pckt, 0, pcktSize);
+        memset(rawPacket.data, 0, packetSize);
     }
 
-    RAW_PACKET(const RAW_PACKET & rp)
-        : rawPacket(),
-          dataLen(rp.dataLen)
+    RawPacket(const RawPacket& rhs) noexcept
     {
-    memcpy(rawPacket.pckt, rp.rawPacket.pckt, pcktSize);
+        memcpy(rawPacket.data, rhs.rawPacket.data, packetSize);
+    }
+    RawPacket& operator=(const RawPacket& rhs) noexcept
+    {
+        memcpy(rawPacket.data, rhs.rawPacket.data, packetSize);
+        return *this;
+    }
+    RawPacket(RawPacket&& rhs) noexcept
+    {
+        memcpy(rawPacket.data, rhs.rawPacket.data, packetSize);
+    }
+    RawPacket& operator=(RawPacket&& rhs) noexcept
+    {
+        memcpy(rawPacket.data, rhs.rawPacket.data, packetSize);
+        return *this;
     }
 
-uint16_t    GetIPVersion() const;
-uint8_t     GetHeaderLen() const;
-uint8_t     GetProto() const;
-uint32_t    GetLen() const;
-uint32_t    GetSrcIP() const;
-uint32_t    GetDstIP() const;
-uint16_t    GetSrcPort() const;
-uint16_t    GetDstPort() const;
+    uint16_t GetIPVersion() const noexcept;
+    uint8_t  GetHeaderLen() const noexcept;
+    uint8_t  GetProto() const noexcept;
+    uint32_t GetLen() const noexcept;
+    uint32_t GetSrcIP() const noexcept;
+    uint32_t GetDstIP() const noexcept;
+    uint16_t GetSrcPort() const noexcept;
+    uint16_t GetDstPort() const noexcept;
 
-bool        operator==(const RAW_PACKET & rvalue) const;
-bool        operator!=(const RAW_PACKET & rvalue) const { return !(*this == rvalue); };
-bool        operator<(const RAW_PACKET & rvalue) const;
+    bool     operator==(const RawPacket& rhs) const noexcept;
+    bool     operator!=(const RawPacket& rhs) const noexcept { return !(*this == rhs); }
+    bool     operator<(const RawPacket& rhs) const noexcept;
 
-union
+    union
     {
-    uint8_t pckt[pcktSize]; // Packet header as a raw data
-    struct
+        uint8_t data[packetSize]; // Packet header as a raw data
+        struct
         {
-        struct ip   ipHeader;
-        // Only for packets without options field
-        uint16_t    sPort;
-        uint16_t    dPort;
-        } header __attribute__ ((packed));
+            struct ip   ipHeader;
+            // Only for packets without options field
+            uint16_t    sPort;
+            uint16_t    dPort;
+        } header;
     } rawPacket;
-int32_t dataLen; // IP packet length. Set to -1 to use length field from the header
+    int32_t dataLen; // IP packet length. Set to -1 to use length field from the header
 };
 //-----------------------------------------------------------------------------
-inline uint16_t RAW_PACKET::GetIPVersion() const
+inline uint16_t RawPacket::GetIPVersion() const noexcept
 {
-return rawPacket.header.ipHeader.ip_v;
+    return rawPacket.header.ipHeader.ip_v;
 }
 //-----------------------------------------------------------------------------
-inline uint8_t RAW_PACKET::GetHeaderLen() const
+inline uint8_t RawPacket::GetHeaderLen() const noexcept
 {
-return rawPacket.header.ipHeader.ip_hl * 4;
+    return rawPacket.header.ipHeader.ip_hl * 4;
 }
 //-----------------------------------------------------------------------------
-inline uint8_t RAW_PACKET::GetProto() const
+inline uint8_t RawPacket::GetProto() const noexcept
 {
-return rawPacket.header.ipHeader.ip_p;
+    return rawPacket.header.ipHeader.ip_p;
 }
 //-----------------------------------------------------------------------------
-inline uint32_t RAW_PACKET::GetLen() const
+inline uint32_t RawPacket::GetLen() const noexcept
 {
-if (dataLen != -1)
-    return dataLen;
-return ntohs(rawPacket.header.ipHeader.ip_len);
+    if (dataLen != -1)
+        return dataLen;
+    return ntohs(rawPacket.header.ipHeader.ip_len);
 }
 //-----------------------------------------------------------------------------
-inline uint32_t RAW_PACKET::GetSrcIP() const
+inline uint32_t RawPacket::GetSrcIP() const noexcept
 {
-return rawPacket.header.ipHeader.ip_src.s_addr;
+    return rawPacket.header.ipHeader.ip_src.s_addr;
 }
 //-----------------------------------------------------------------------------
-inline uint32_t RAW_PACKET::GetDstIP() const
+inline uint32_t RawPacket::GetDstIP() const noexcept
 {
-return rawPacket.header.ipHeader.ip_dst.s_addr;
+    return rawPacket.header.ipHeader.ip_dst.s_addr;
 }
 //-----------------------------------------------------------------------------
-inline uint16_t RAW_PACKET::GetSrcPort() const
+inline uint16_t RawPacket::GetSrcPort() const noexcept
 {
-if (rawPacket.header.ipHeader.ip_p == 1) // for icmp proto return port 0
-    return 0;
-return ntohs(*((uint16_t*)(rawPacket.pckt + rawPacket.header.ipHeader.ip_hl * 4)));
+    if (rawPacket.header.ipHeader.ip_p == 1) // for icmp proto return port 0
+        return 0;
+    const uint8_t* pos = rawPacket.data + rawPacket.header.ipHeader.ip_hl * 4;
+    return ntohs(*reinterpret_cast<const uint16_t *>(pos));
 }
 //-----------------------------------------------------------------------------
-inline uint16_t RAW_PACKET::GetDstPort() const
+inline uint16_t RawPacket::GetDstPort() const noexcept
 {
-if (rawPacket.header.ipHeader.ip_p == 1) // for icmp proto return port 0
-    return 0;
-return ntohs(*((uint16_t*)(rawPacket.pckt + rawPacket.header.ipHeader.ip_hl * 4 + 2)));
+    if (rawPacket.header.ipHeader.ip_p == 1) // for icmp proto return port 0
+        return 0;
+    const uint8_t * pos = rawPacket.data + rawPacket.header.ipHeader.ip_hl * 4 + 2;
+    return ntohs(*reinterpret_cast<const uint16_t *>(pos));
 }
 //-----------------------------------------------------------------------------
-inline bool RAW_PACKET::operator==(const RAW_PACKET & rvalue) const
+inline bool RawPacket::operator==(const RawPacket& rhs) const noexcept
 {
-if (rawPacket.header.ipHeader.ip_src.s_addr != rvalue.rawPacket.header.ipHeader.ip_src.s_addr)
-    return false;
-
-if (rawPacket.header.ipHeader.ip_dst.s_addr != rvalue.rawPacket.header.ipHeader.ip_dst.s_addr)
-    return false;
-
-if (rawPacket.header.ipHeader.ip_p != 1 && rvalue.rawPacket.header.ipHeader.ip_p != 1)
-    {
-    if (*((uint16_t *)(rawPacket.pckt + rawPacket.header.ipHeader.ip_hl * 4)) !=
-        *((uint16_t *)(rvalue.rawPacket.pckt + rvalue.rawPacket.header.ipHeader.ip_hl * 4)))
+    if (rawPacket.header.ipHeader.ip_src.s_addr != rhs.rawPacket.header.ipHeader.ip_src.s_addr)
         return false;
 
-    if (*((uint16_t *)(rawPacket.pckt + rawPacket.header.ipHeader.ip_hl * 4 + 2)) !=
-        *((uint16_t *)(rvalue.rawPacket.pckt + rvalue.rawPacket.header.ipHeader.ip_hl * 4 + 2)))
+    if (rawPacket.header.ipHeader.ip_dst.s_addr != rhs.rawPacket.header.ipHeader.ip_dst.s_addr)
         return false;
+
+    if (rawPacket.header.ipHeader.ip_p != 1 && rhs.rawPacket.header.ipHeader.ip_p != 1)
+    {
+        const uint8_t * pos = rawPacket.data + rawPacket.header.ipHeader.ip_hl * 4;
+        const uint8_t * rpos = rhs.rawPacket.data + rhs.rawPacket.header.ipHeader.ip_hl * 4;
+        if (*reinterpret_cast<const uint16_t *>(pos) != *reinterpret_cast<const uint16_t *>(rpos))
+            return false;
+
+        pos += 2;
+        rpos += 2;
+        if (*reinterpret_cast<const uint16_t *>(pos) != *reinterpret_cast<const uint16_t *>(rpos))
+            return false;
     }
 
-if (rawPacket.header.ipHeader.ip_p != rvalue.rawPacket.header.ipHeader.ip_p)
-    return false;
+    if (rawPacket.header.ipHeader.ip_p != rhs.rawPacket.header.ipHeader.ip_p)
+        return false;
 
-return true;
+    return true;
 }
 //-----------------------------------------------------------------------------
-inline bool RAW_PACKET::operator<(const RAW_PACKET & rvalue) const
+inline bool RawPacket::operator<(const RawPacket& rhs) const noexcept
 {
-if (rawPacket.header.ipHeader.ip_src.s_addr < rvalue.rawPacket.header.ipHeader.ip_src.s_addr) 
-    return true;
-if (rawPacket.header.ipHeader.ip_src.s_addr > rvalue.rawPacket.header.ipHeader.ip_src.s_addr) 
-    return false;
-
-if (rawPacket.header.ipHeader.ip_dst.s_addr < rvalue.rawPacket.header.ipHeader.ip_dst.s_addr) 
-    return true;
-if (rawPacket.header.ipHeader.ip_dst.s_addr > rvalue.rawPacket.header.ipHeader.ip_dst.s_addr) 
-    return false;
-
-if (rawPacket.header.ipHeader.ip_p != 1 && rvalue.rawPacket.header.ipHeader.ip_p != 1)
-    {
-    if (*((uint16_t *)(rawPacket.pckt + rawPacket.header.ipHeader.ip_hl * 4)) <
-        *((uint16_t *)(rvalue.rawPacket.pckt + rvalue.rawPacket.header.ipHeader.ip_hl * 4))) 
+    if (rawPacket.header.ipHeader.ip_src.s_addr < rhs.rawPacket.header.ipHeader.ip_src.s_addr)
         return true;
-    if (*((uint16_t *)(rawPacket.pckt + rawPacket.header.ipHeader.ip_hl * 4)) >
-        *((uint16_t *)(rvalue.rawPacket.pckt + rvalue.rawPacket.header.ipHeader.ip_hl * 4))) 
+    if (rawPacket.header.ipHeader.ip_src.s_addr > rhs.rawPacket.header.ipHeader.ip_src.s_addr)
         return false;
 
-    if (*((uint16_t *)(rawPacket.pckt + rawPacket.header.ipHeader.ip_hl * 4 + 2)) <
-        *((uint16_t *)(rvalue.rawPacket.pckt + rvalue.rawPacket.header.ipHeader.ip_hl * 4 + 2))) 
+    if (rawPacket.header.ipHeader.ip_dst.s_addr < rhs.rawPacket.header.ipHeader.ip_dst.s_addr)
         return true;
-    if (*((uint16_t *)(rawPacket.pckt + rawPacket.header.ipHeader.ip_hl * 4 + 2)) >
-        *((uint16_t *)(rvalue.rawPacket.pckt + rvalue.rawPacket.header.ipHeader.ip_hl * 4 + 2))) 
+    if (rawPacket.header.ipHeader.ip_dst.s_addr > rhs.rawPacket.header.ipHeader.ip_dst.s_addr)
         return false;
+
+    if (rawPacket.header.ipHeader.ip_p != 1 && rhs.rawPacket.header.ipHeader.ip_p != 1)
+    {
+        const uint8_t * pos = rawPacket.data + rawPacket.header.ipHeader.ip_hl * 4;
+        const uint8_t * rpos = rhs.rawPacket.data + rhs.rawPacket.header.ipHeader.ip_hl * 4;
+        if (*reinterpret_cast<const uint16_t *>(pos) < *reinterpret_cast<const uint16_t *>(rpos))
+            return true;
+        if (*reinterpret_cast<const uint16_t *>(pos) > *reinterpret_cast<const uint16_t *>(rpos))
+            return false;
+
+        pos += 2;
+        rpos += 2;
+        if (*reinterpret_cast<const uint16_t *>(pos) < *reinterpret_cast<const uint16_t *>(rpos))
+            return true;
+        if (*reinterpret_cast<const uint16_t *>(pos) > *reinterpret_cast<const uint16_t *>(rpos))
+            return false;
     }
 
-if (rawPacket.header.ipHeader.ip_p < rvalue.rawPacket.header.ipHeader.ip_p) 
-    return true;
+    if (rawPacket.header.ipHeader.ip_p < rhs.rawPacket.header.ipHeader.ip_p)
+        return true;
 
-return false;
+    return false;
 }
 //-----------------------------------------------------------------------------
 
-#endif
+}