Додано патч від Alexey Osipov <lion-simba@pridelands.ru>

[stg.git] / include / raw_ip_packet.h

diff --git a/include/raw_ip_packet.h b/include/raw_ip_packet.h
index af62d05184dc5ab4dc2d7e8dad44a3eadd9e80ce..cd89126630ecbeab42f3077cd0adbada1d1f1ade 100644 (file)
--- a/include/raw_ip_packet.h
+++ b/include/raw_ip_packet.h
@@ -1,6 +1,10 @@
 #ifndef RAW_IP_PACKET_H
 #define RAW_IP_PACKET_H
 
 #ifndef RAW_IP_PACKET_H
 #define RAW_IP_PACKET_H
 

+#if defined(FREE_BSD) || defined(FREE_BSD5)
+#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.h> // for htons
 #include <netinet/ip.h> // for struct ip
 


@@ -49,77 +53,77 @@ union
         // Only for packets without options field
         uint16_t    sPort;
         uint16_t    dPort;
         // Only for packets without options field
         uint16_t    sPort;
         uint16_t    dPort;

-        } __attribute__ ((packed));
+        } header __attribute__ ((packed));

     };
 int32_t     dataLen;                // äÌÉÎÁ IP ÐÁËÅÔÁ. åÓÌÉ -1, ÔÏ ÉÓÐÏÌØÚÏ×ÁÔØ ÄÌÉÎÕ ÉÚ ÚÁÇÏÌÏ×ËÁ ÓÁÍÏÇÏ ÐÁËÅÔÁ.
 };
 //-----------------------------------------------------------------------------
 inline uint16_t RAW_PACKET::GetIPVersion() const
 {
     };
 int32_t     dataLen;                // äÌÉÎÁ IP ÐÁËÅÔÁ. åÓÌÉ -1, ÔÏ ÉÓÐÏÌØÚÏ×ÁÔØ ÄÌÉÎÕ ÉÚ ÚÁÇÏÌÏ×ËÁ ÓÁÍÏÇÏ ÐÁËÅÔÁ.
 };
 //-----------------------------------------------------------------------------
 inline uint16_t RAW_PACKET::GetIPVersion() const
 {

-return ipHeader.ip_v;
+return header.ipHeader.ip_v;

 }
 //-----------------------------------------------------------------------------
 inline uint8_t RAW_PACKET::GetHeaderLen() const
 {
 }
 //-----------------------------------------------------------------------------
 inline uint8_t RAW_PACKET::GetHeaderLen() const
 {

-return ipHeader.ip_hl * 4;
+return header.ipHeader.ip_hl * 4;

 }
 //-----------------------------------------------------------------------------
 inline uint8_t RAW_PACKET::GetProto() const
 {
 }
 //-----------------------------------------------------------------------------
 inline uint8_t RAW_PACKET::GetProto() const
 {

-return ipHeader.ip_p;
+return header.ipHeader.ip_p;

 }
 //-----------------------------------------------------------------------------
 inline uint32_t RAW_PACKET::GetLen() const
 {
 if (dataLen != -1)
     return dataLen;
 }
 //-----------------------------------------------------------------------------
 inline uint32_t RAW_PACKET::GetLen() const
 {
 if (dataLen != -1)
     return dataLen;

-return ntohs(ipHeader.ip_len);
+return ntohs(header.ipHeader.ip_len);

 }
 //-----------------------------------------------------------------------------
 inline uint32_t RAW_PACKET::GetSrcIP() const
 {
 }
 //-----------------------------------------------------------------------------
 inline uint32_t RAW_PACKET::GetSrcIP() const
 {

-return ipHeader.ip_src.s_addr;
+return header.ipHeader.ip_src.s_addr;

 }
 //-----------------------------------------------------------------------------
 inline uint32_t RAW_PACKET::GetDstIP() const
 {
 }
 //-----------------------------------------------------------------------------
 inline uint32_t RAW_PACKET::GetDstIP() const
 {

-return ipHeader.ip_dst.s_addr;
+return header.ipHeader.ip_dst.s_addr;

 }
 //-----------------------------------------------------------------------------
 inline uint16_t RAW_PACKET::GetSrcPort() const
 {
 }
 //-----------------------------------------------------------------------------
 inline uint16_t RAW_PACKET::GetSrcPort() const
 {

-if (ipHeader.ip_p == 1) // for icmp proto return port 0
+if (header.ipHeader.ip_p == 1) // for icmp proto return port 0

     return 0;
     return 0;

-return ntohs(*((uint16_t*)(pckt + ipHeader.ip_hl * 4)));
+return ntohs(*((uint16_t*)(pckt + header.ipHeader.ip_hl * 4)));

 }
 //-----------------------------------------------------------------------------
 inline uint16_t RAW_PACKET::GetDstPort() const
 {
 }
 //-----------------------------------------------------------------------------
 inline uint16_t RAW_PACKET::GetDstPort() const
 {

-if (ipHeader.ip_p == 1) // for icmp proto return port 0
+if (header.ipHeader.ip_p == 1) // for icmp proto return port 0

     return 0;
     return 0;

-return ntohs(*((uint16_t*)(pckt + ipHeader.ip_hl * 4 + 2)));
+return ntohs(*((uint16_t*)(pckt + header.ipHeader.ip_hl * 4 + 2)));

 }
 //-----------------------------------------------------------------------------
 inline bool RAW_PACKET::operator==(const RAW_PACKET & rvalue) const
 {
 }
 //-----------------------------------------------------------------------------
 inline bool RAW_PACKET::operator==(const RAW_PACKET & rvalue) const
 {

-if (ipHeader.ip_src.s_addr != rvalue.ipHeader.ip_src.s_addr)
+if (header.ipHeader.ip_src.s_addr != rvalue.header.ipHeader.ip_src.s_addr)

     return false;
 
     return false;
 

-if (ipHeader.ip_dst.s_addr != rvalue.ipHeader.ip_dst.s_addr)
+if (header.ipHeader.ip_dst.s_addr != rvalue.header.ipHeader.ip_dst.s_addr)

     return false;
 
     return false;
 

-if (ipHeader.ip_p != 1 && rvalue.ipHeader.ip_p != 1)
+if (header.ipHeader.ip_p != 1 && rvalue.header.ipHeader.ip_p != 1)

     {
     {

-    if (*((uint16_t *)(pckt + ipHeader.ip_hl * 4)) !=
-        *((uint16_t *)(rvalue.pckt + rvalue.ipHeader.ip_hl * 4)))
+    if (*((uint16_t *)(pckt + header.ipHeader.ip_hl * 4)) !=
+        *((uint16_t *)(rvalue.pckt + rvalue.header.ipHeader.ip_hl * 4)))

         return false;
 
         return false;
 

-    if (*((uint16_t *)(pckt + ipHeader.ip_hl * 4 + 2)) !=
-        *((uint16_t *)(rvalue.pckt + rvalue.ipHeader.ip_hl * 4 + 2)))
+    if (*((uint16_t *)(pckt + header.ipHeader.ip_hl * 4 + 2)) !=
+        *((uint16_t *)(rvalue.pckt + rvalue.header.ipHeader.ip_hl * 4 + 2)))

         return false;
     }
 
         return false;
     }
 

-if (ipHeader.ip_p != rvalue.ipHeader.ip_p)
+if (header.ipHeader.ip_p != rvalue.header.ipHeader.ip_p)

     return false;
 
 return true;
     return false;
 
 return true;


@@ -147,34 +151,34 @@ return true;
 //-----------------------------------------------------------------------------
 inline bool RAW_PACKET::operator<(const RAW_PACKET & rvalue) const
 {
 //-----------------------------------------------------------------------------
 inline bool RAW_PACKET::operator<(const RAW_PACKET & rvalue) const
 {

-if (ipHeader.ip_src.s_addr < rvalue.ipHeader.ip_src.s_addr) 
+if (header.ipHeader.ip_src.s_addr < rvalue.header.ipHeader.ip_src.s_addr) 

     return true;
     return true;

-if (ipHeader.ip_src.s_addr > rvalue.ipHeader.ip_src.s_addr) 
+if (header.ipHeader.ip_src.s_addr > rvalue.header.ipHeader.ip_src.s_addr) 

     return false;
 
     return false;
 

-if (ipHeader.ip_dst.s_addr < rvalue.ipHeader.ip_dst.s_addr) 
+if (header.ipHeader.ip_dst.s_addr < rvalue.header.ipHeader.ip_dst.s_addr) 

     return true;
     return true;

-if (ipHeader.ip_dst.s_addr > rvalue.ipHeader.ip_dst.s_addr) 
+if (header.ipHeader.ip_dst.s_addr > rvalue.header.ipHeader.ip_dst.s_addr) 

     return false;
 
     return false;
 

-if (ipHeader.ip_p != 1 && rvalue.ipHeader.ip_p != 1)
+if (header.ipHeader.ip_p != 1 && rvalue.header.ipHeader.ip_p != 1)

     {
     {

-    if (*((uint16_t *)(pckt + ipHeader.ip_hl * 4)) <
-        *((uint16_t *)(rvalue.pckt + rvalue.ipHeader.ip_hl * 4))) 
+    if (*((uint16_t *)(pckt + header.ipHeader.ip_hl * 4)) <
+        *((uint16_t *)(rvalue.pckt + rvalue.header.ipHeader.ip_hl * 4))) 

         return true;
         return true;

-    if (*((uint16_t *)(pckt + ipHeader.ip_hl * 4)) >
-        *((uint16_t *)(rvalue.pckt + rvalue.ipHeader.ip_hl * 4))) 
+    if (*((uint16_t *)(pckt + header.ipHeader.ip_hl * 4)) >
+        *((uint16_t *)(rvalue.pckt + rvalue.header.ipHeader.ip_hl * 4))) 

         return false;
 
         return false;
 

-    if (*((uint16_t *)(pckt + ipHeader.ip_hl * 4 + 2)) <
-        *((uint16_t *)(rvalue.pckt + rvalue.ipHeader.ip_hl * 4 + 2))) 
+    if (*((uint16_t *)(pckt + header.ipHeader.ip_hl * 4 + 2)) <
+        *((uint16_t *)(rvalue.pckt + rvalue.header.ipHeader.ip_hl * 4 + 2))) 

         return true;
         return true;

-    if (*((uint16_t *)(pckt + ipHeader.ip_hl * 4 + 2)) >
-        *((uint16_t *)(rvalue.pckt + rvalue.ipHeader.ip_hl * 4 + 2))) 
+    if (*((uint16_t *)(pckt + header.ipHeader.ip_hl * 4 + 2)) >
+        *((uint16_t *)(rvalue.pckt + rvalue.header.ipHeader.ip_hl * 4 + 2))) 

         return false;
     }
 
         return false;
     }
 

-if (ipHeader.ip_p < rvalue.ipHeader.ip_p) 
+if (header.ipHeader.ip_p < rvalue.header.ipHeader.ip_p) 

     return true;
 
 return false;
     return true;
 
 return false;


@@ -210,5 +214,3 @@ return false;
 //-----------------------------------------------------------------------------
 
 #endif
 //-----------------------------------------------------------------------------
 
 #endif

-
-