Disable NFQ on non-linux.

[stg.git] / include / stg / raw_ip_packet.h

diff --git a/include/stg/raw_ip_packet.h b/include/stg/raw_ip_packet.h
index f07bf2280b4ab2db68c5b1532ca819bd1236579c..3139cafe763c40cab8c6a951b46b0e1394b10c07 100644 (file)
--- a/include/stg/raw_ip_packet.h
+++ b/include/stg/raw_ip_packet.h
@@ -1,167 +1,184 @@
-#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 <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 <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>
-

 #define IPv4 (2)
 
 #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.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.pckt, rp.rawPacket.pckt, pcktSize);
+        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;
     } 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;
 
         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;
         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;
         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;
 
         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;
         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;
         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
+}