first version

1 files changed, 212 insertions, 0 deletions

diff --git a/src/rx_check_header.h b/src/rx_check_header.h
new file mode 100755
index 00000000..3ac5dd1f
--- /dev/null
+++ b/src/rx_check_header.h
@@ -0,0 +1,212 @@
+#ifndef RX_CHECK_HEADER
+#define RX_CHECK_HEADER
+/*
+ Hanoh Haim
+ Cisco Systems, Inc.
+*/
+
+/*
+Copyright (c) 2015-2015 Cisco Systems, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+
+#include <stdint.h>
+#include <stdio.h>
+#include <common/bitMan.h> 
+#include <common/Network/Packet/CPktCmn.h>
+
+
+#include "os_time.h"
+
+#define RX_CHECK_LEN    (sizeof(struct CRx_check_header))
+
+// IPv4 option type:
+//   bit[0] = copy flag (0=do not copy to fragment)
+//   bit[1:2] = option class (2=debugging)
+//   bit[3:7] = option number (23=TRex)
+#define RX_CHECK_V4_OPT_TYPE    0x57
+#define RX_CHECK_V4_OPT_LEN   RX_CHECK_LEN
+
+// IPv6 extension header:
+//   type = 60 (destination options)
+#define RX_CHECK_V6_OPT_TYPE    0x3c
+#define RX_CHECK_V6_OPT_LEN ((RX_CHECK_LEN - 8) / 8)
+
+// IPv6 subfield option type:
+//   bit[0:1] = unrecognied option action (00=skip option)
+//   bit[2] = change allowed flag (0=no changes enroute)
+//   bit[3:7] = option number (23=TRex)
+#define RX_CHECK_V6_SF_OPT_TYPE    0x17
+#define RX_CHECK_V6_SF_OPT_LEN   (RX_CHECK_LEN - 2)
+
+// Magic field overlays IPv6 subfield option type
+#define RX_CHECK_MAGIC  (((RX_CHECK_V6_SF_OPT_LEN & 0xff) << 8) | \
+                                           RX_CHECK_V6_SF_OPT_TYPE)
+
+struct  CRx_check_header {
+    // 24 bytes, watch alignment and keep this in 8 byte increments
+    uint8_t m_option_type;
+    uint8_t m_option_len;
+    uint16_t m_magic;
+
+    uint32_t m_time_stamp;
+    uint64_t m_flow_id; // include thread_id , template_id
+    uint16_t m_pkt_id;    /* pkt_id inside the flow - zero base 0,1,2*/
+    uint16_t m_flow_size; /* how many packets in flow */
+
+    uint8_t m_flags;
+    uint8_t m_template_id;
+    uint16_t m_aging_sec;
+
+public:
+    bool is_fif_dir(void){
+        return (m_pkt_id==0?true:false);
+    }
+    bool is_eof_dir(void){
+        return ( (m_pkt_id==m_flow_size-1)?true:false);
+    }
+
+    void set_dir(int dir){
+        btSetMaskBit8(m_flags,0,0,dir?1:0);
+    }
+
+    int get_dir(void){
+        return (btGetMaskBit8(m_flags,0,0) ? 1:0);
+    }
+
+    /* need to mark if we expect to see both sides of the flow, this is know offline  */
+    void set_both_dir(int both){
+        btSetMaskBit8(m_flags,1,1,both?1:0);
+    }
+
+    int get_both_dir(void){
+        return (btGetMaskBit8(m_flags,1,1) ? 1:0);
+    }
+
+
+
+    void dump(FILE *fd);
+};
+
+
+class  CNatOption {
+public:
+    enum {
+        noIPV4_OPTION   = 0x10, /* dummy IPV4 option */
+        noOPTION_LEN    = 0x8,
+        noIPV4_MAGIC    = 0xEE,
+        noIPV4_MAGIC_RX    = 0xED,
+
+        noIPV6_OPTION_LEN    = (noOPTION_LEN/8)-1,
+        noIPV6_OPTION = 0x3C, /*IPv6-Opts	Destination Options for IPv6	RFC 2460*/
+    };
+    void set_option_type(uint8_t id){
+        u.m_data[0]=id;
+    }
+    uint8_t get_option_type(){
+        return (u.m_data[0]);
+    }
+
+    void set_option_len(uint8_t len){
+        u.m_data[1]=len;
+    }
+    uint8_t get_option_len(){
+        return ( u.m_data[1]);
+    }
+
+    void set_thread_id(uint8_t thread_id){
+        u.m_data[3]=thread_id;
+    }
+    uint8_t get_thread_id(){
+        return (u.m_data[3]);
+    }
+
+    void set_magic(uint8_t magic){
+        u.m_data[2]=magic;
+    }
+
+    uint8_t get_magic(){
+        return (u.m_data[2]);
+    }
+
+    void set_rx_check(bool enable){
+        if (enable) {
+            set_magic(CNatOption::noIPV4_MAGIC_RX);
+        }else{
+            set_magic(CNatOption::noIPV4_MAGIC);
+        }
+    }
+    bool is_rx_check(){
+        if (get_magic() ==CNatOption::noIPV4_MAGIC_RX) {
+            return(true);
+        }else{
+            return (false);
+        }
+    }
+
+
+    void set_fid(uint32_t fid){
+        u.m_data_uint32[1]=fid;
+    }
+    uint32_t get_fid(){
+        return (u.m_data_uint32[1]);
+    }
+
+    bool is_valid_ipv4_magic_op0(void){
+        return ( ( PKT_NTOHL( u.m_data_uint32[0] )& 0xFFFFFF00 ) ==
+                 (CNatOption::noIPV4_OPTION <<24) +  (CNatOption::noOPTION_LEN<<16) + (CNatOption::noIPV4_MAGIC<<8) ?true:false);
+    }
+
+    bool is_valid_ipv4_magic_op1(void){
+        return ( ( PKT_NTOHL( u.m_data_uint32[0] )& 0xFFFFFF00 ) ==
+                 (CNatOption::noIPV4_OPTION <<24) +  (CNatOption::noOPTION_LEN<<16) + (CNatOption::noIPV4_MAGIC_RX<<8) ?true:false);
+    }
+
+    bool is_valid_ipv4_magic(void){
+        return (is_valid_ipv4_magic_op0() ||is_valid_ipv4_magic_op1() );
+    }
+
+
+    bool is_valid_ipv6_magic(void){
+        return ( ( PKT_NTOHL( u.m_data_uint32[0] )& 0x00FFFF00 ) ==
+                 (CNatOption::noIPV6_OPTION_LEN<<16) + (CNatOption::noIPV4_MAGIC<<8) ?true:false);
+
+    }
+
+    void set_init_ipv4_header(){
+        set_option_type(CNatOption::noIPV4_OPTION);
+        set_option_len(CNatOption::noOPTION_LEN);
+        set_magic(CNatOption::noIPV4_MAGIC);
+    }
+
+    void set_init_ipv6_header(void){
+        set_option_len(noIPV6_OPTION_LEN);
+        set_magic(CNatOption::noIPV4_MAGIC);
+    }
+
+    void dump(FILE *fd);
+
+
+private:
+    union u_ {
+        uint8_t  m_data[8];
+        uint32_t m_data_uint32[2];
+    } u;
+};
+
+
+#endif
+
+