From 8b52a31ed2c299b759f330c4f976b9c70f5765f4 Mon Sep 17 00:00:00 2001
From: Hanoh Haim <hhaim@cisco.com>
Date: Wed, 24 Jun 2015 14:03:29 +0300
Subject: first version

---
 src/nat_check.cpp | 197 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 197 insertions(+)
 create mode 100755 src/nat_check.cpp

(limited to 'src/nat_check.cpp')

diff --git a/src/nat_check.cpp b/src/nat_check.cpp
new file mode 100755
index 00000000..676c1292
--- /dev/null
+++ b/src/nat_check.cpp
@@ -0,0 +1,197 @@
+#include <stdint.h>
+#include "nat_check.h"
+#include "bp_sim.h"
+/*
+ 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.
+*/
+
+
+void CGenNodeNatInfo::dump(FILE *fd){
+
+    fprintf(fd," msg_type :  %d \n",m_msg_type);
+    int i;
+    for (i=0; i<m_cnt; i++) {
+        CNatFlowInfo * lp=&m_data[i];
+        fprintf (fd," id:%d , external ip:%08x:%x , ex_port: %04x , fid: %d \n",i,lp->m_external_ip,lp->m_external_ip_server,lp->m_external_port,lp->m_fid);
+    }
+}
+
+void CGenNodeNatInfo::init(){
+    m_msg_type= CGenNodeMsgBase::NAT_FIRST;
+    m_pad=0;
+    m_cnt=0;
+}
+
+
+void CNatStats::reset(){
+    m_total_rx=0;
+    m_total_msg=0;
+    m_err_no_valid_thread_id=0;
+    m_err_no_valid_proto=0;
+    m_err_queue_full=0;
+}
+
+
+CNatPerThreadInfo * CNatRxManager::get_thread_info(uint8_t thread_id){
+    if (thread_id<m_max_threads) {
+        return (&m_per_thread[thread_id]);
+    }
+    m_stats.m_err_no_valid_thread_id++;
+    return (0);
+}
+
+
+bool CNatRxManager::Create(){
+
+    m_max_threads = CMsgIns::Ins()->get_num_threads() ;
+    m_per_thread = new CNatPerThreadInfo[m_max_threads];
+    CMessagingManager * lpm=CMsgIns::Ins()->getRxDp();
+
+    int i;
+    for (i=0; i<m_max_threads; i++) {
+        m_per_thread[i].m_ring=lpm->getRingCpToDp(i);
+        assert(m_per_thread[i].m_ring);
+    }
+
+
+
+    return (true);
+}
+
+void CNatRxManager::Delete(){
+    if (m_per_thread) {
+        delete m_per_thread;
+        m_per_thread=0;
+    }
+}
+
+void delay(int msec);
+
+
+
+/* check this every 1msec */
+void CNatRxManager::handle_aging(){
+    int i;
+    dsec_t now=now_sec();
+    for (i=0; i<m_max_threads; i++) {
+        CNatPerThreadInfo * thread_info=get_thread_info( i );
+        if ( thread_info->m_cur_nat_msg ){
+            if ( now - thread_info->m_last_time > MAX_TIME_MSG_IN_QUEUE_SEC ){
+                flush_node(thread_info);
+            }
+        }
+    }
+}
+
+void CNatRxManager::flush_node(CNatPerThreadInfo * thread_info){
+        // try send 
+        int cnt=0;
+        while (true) {
+            if ( thread_info->m_ring->Enqueue((CGenNode*)thread_info->m_cur_nat_msg) == 0 ){
+                #ifdef NAT_TRACE_
+                printf("send message \n");
+                #endif
+                break;
+            }
+            m_stats.m_err_queue_full++;
+            delay(1);
+            cnt++;
+            if (cnt>10) {
+                printf(" ERROR  queue from rx->dp is stuck, somthing is wrong here \n");
+                exit(1);
+            }
+        }
+        /* msg will be free by sink */
+        thread_info->m_cur_nat_msg=0;
+}
+
+
+void CNatRxManager::handle_packet_ipv4(CNatOption * option,
+                                  IPHeader * ipv4){
+
+    CNatPerThreadInfo * thread_info=get_thread_info(option->get_thread_id());
+    if (!thread_info) {
+        return;
+    }
+    /* Extract info from the packet ! */
+    uint32_t ext_ip   = ipv4->getSourceIp();
+    uint32_t ext_ip_server = ipv4->getDestIp();
+    uint8_t proto     = ipv4->getProtocol();
+    /* must be TCP/UDP this is the only supported proto */
+    if (!( (proto==6) || (proto==17) )){
+        m_stats.m_err_no_valid_proto++;
+        return;
+    }
+    /* we support only TCP/UDP so take the source port , post IP header */
+    UDPHeader * udp= (UDPHeader *) (((char *)ipv4)+ ipv4->getHeaderLength());
+    uint16_t ext_port = udp->getSourcePort();
+    #ifdef NAT_TRACE_
+    printf("rx msg ext ip : %08x:%08x ext port : %04x  flow_id : %d \n",ext_ip,ext_ip_server,ext_port,option->get_fid());
+    #endif
+
+
+    CGenNodeNatInfo * node=thread_info->m_cur_nat_msg;
+    if ( !node ){
+        node = (CGenNodeNatInfo * )CGlobalInfo::create_node();
+        assert(node);
+        node->init();
+        thread_info->m_cur_nat_msg = node;
+        thread_info->m_last_time   = now_sec();
+    }
+    /* get message */
+    CNatFlowInfo * msg=node->get_next_msg();
+
+    /* fill the message */
+    msg->m_external_ip   = ext_ip;
+    msg->m_external_ip_server = ext_ip_server;
+    msg->m_external_port = ext_port;
+    msg->m_fid           = option->get_fid();
+    msg->m_pad           = 0xee; 
+
+    if ( node->is_full() ){
+        flush_node(thread_info);
+    }
+}
+
+
+#define MYDP(f) if (f)  fprintf(fd," %-40s: %llu \n",#f,f)
+#define MYDP_A(f)     fprintf(fd," %-40s: %llu \n",#f,f)
+
+
+
+void CNatStats::Dump(FILE *fd){
+   MYDP(m_total_rx);
+   MYDP(m_total_msg);
+   MYDP(m_err_no_valid_thread_id);
+   MYDP(m_err_no_valid_proto);
+   MYDP(m_err_queue_full);
+}
+
+
+void CNatRxManager::Dump(FILE *fd){
+    m_stats.Dump(stdout);
+}
+
+void CNatRxManager::DumpShort(FILE *fd){
+    fprintf(fd,"nat check msgs: %lu,  errors: %lu  \n",m_stats.m_total_msg,m_stats.get_errs() );
+}
+
+
+
-- 
cgit 1.2.3-korg