diff options
Diffstat (limited to 'src/nat_check.cpp')
| -rwxr-xr-x | src/nat_check.cpp | 197 |
1 files changed, 197 insertions, 0 deletions
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() ); +} + + + |