TCP_Rewriter.cc File Reference

#include "config.h"
#include <assert.h>
#include <stdlib.h>
#include "Event.h"
#include "Net.h"
#include "TCP_Rewriter.h"
#include "common-rw.bif.func_def"

Include dependency graph for TCP_Rewriter.cc:

Go to the source code of this file.

Classes
struct	TCPOptionRewriting
Defines
#define	MSG_PREFIX   "TCP trace rewriter: "
#define	DEBUG_MSG_A(x...)
#define	KEEP_ORIG   1
#define	REUSE_OPT   1
#define	TO_NOP   0
#define	MAX_TCP_OPTION_REWRITING   9
Functions
timeval	double_to_timeval (double t)
int	time_compare (struct timeval *tv_a, struct timeval *tv_b)
uint16	NextIP_ID (const uint32 src_addr, const uint16 id)
void	schedule_funeral (TCP_Rewriter *rewriter)
void	schedule_flush (TCP_RewriterEndpoint *endp)
void	flush_rewriter_packet ()
TCP_Rewriter *	get_trace_rewriter (Val *conn_val)
TCP_Rewriter *	get_trace_rewriter (Connection *conn)
TCP_SourcePacketWriter *	get_src_pkt_writer (Connection *conn)
Variables
IP_IDSet *	ip_id_set = 0
int	num_packets_held
int	num_packets_cleaned
TCPOptionRewriting	tcp_option_rewriting [MAX_TCP_OPTION_REWRITING]
std::queue< TCP_Rewriter * >	rewriter_funerals
std::queue< TCP_RewriterEndpoint * >	rewriters_to_flush

---

Define Documentation

#define DEBUG_MSG_A (  x...   )

Definition at line 47 of file TCP_Rewriter.cc. Referenced by TCP_Rewriter::add_slot(), TCP_RewriteSlot::Dump(), TCP_Rewriter::DumpPacket(), TCP_RewriterEndpoint::Flush(), TCP_RewriterEndpoint::GenerateFIN(), TCP_Rewriter::NextPacket(), TCP_Rewriter::ReleaseSlot(), TCP_Rewriter::ReserveSlot(), TCP_RewriterEndpoint::Reset(), and TCP_RewriterEndpoint::ScheduleFlush().

#define KEEP_ORIG   1

Definition at line 1261 of file TCP_Rewriter.cc.

#define MAX_TCP_OPTION_REWRITING   9

Definition at line 1264 of file TCP_Rewriter.cc. Referenced by TCP_Rewriter::RewriteTCPOption().

#define MSG_PREFIX   "TCP trace rewriter: "

Definition at line 46 of file TCP_Rewriter.cc. Referenced by TCP_TracePacket::AppendData(), TCP_TracePacket::AppendIPHeader(), TCP_TracePacket::AppendLinkHeader(), TCP_TracePacket::AppendTCPHeader(), TCP_Rewriter::DumpPacket(), TCP_RewriterEndpoint::Flush(), TCP_Rewriter::Funeral(), TCP_RewriterEndpoint::GenerateFIN(), TCP_Rewriter::NextPacket(), TCP_RewriterEndpoint::NextPacket(), PacketDumper::PacketDumper(), TCP_RewriterEndpoint::PushPacket(), TCP_RewriterEndpoint::SetNextPacket(), TCP_RewriterEndpoint::WriteData(), and TCP_RewriterEndpoint::~TCP_RewriterEndpoint().

#define REUSE_OPT   1

Definition at line 1262 of file TCP_Rewriter.cc.

#define TO_NOP   0

Definition at line 1263 of file TCP_Rewriter.cc.

---

Function Documentation

struct timeval double_to_timeval (  double  t  )  [static]

Definition at line 53 of file TCP_Rewriter.cc. References double, double_to_timeval(), and int. Referenced by double_to_timeval(), and TCP_TracePacket::Finish(). { struct timeval tv; double t1 = floor(t); tv.tv_sec = int(t1); // Round to integer to tolerate floating arithmetic // imprecision (which can be up to 0.05 here!). tv.tv_usec = int((t - t1) * 1e6 + 0.5); return tv; }

void flush_rewriter_packet (   )

Definition at line 1515 of file TCP_Rewriter.cc. References TCP_Rewriter::Conn(), TCP_RewriterEndpoint::Conn(), TCP_RewriterEndpoint::Flush(), TCP_Rewriter::Funeral(), EventMgr::HasEvents(), internal_error(), mgr, rewriter_funerals, rewriters_to_flush, and Unref(). Referenced by EventMgr::Drain(). { while ( ! rewriter_funerals.empty() ) { TCP_Rewriter* rewriter = rewriter_funerals.front(); rewriter_funerals.pop(); rewriter->Funeral(); Unref(rewriter->Conn()); } while ( ! rewriters_to_flush.empty() ) { TCP_RewriterEndpoint* endp = rewriters_to_flush.front(); rewriters_to_flush.pop(); if ( endp ) { endp->Flush(); Unref(endp->Conn()); } } if ( mgr.HasEvents() ) internal_error("flushing packets generates additional events!"); }

TCP_SourcePacketWriter* get_src_pkt_writer (  Connection *  conn  )

Definition at line 1616 of file TCP_Rewriter.cc. References builtin_run_time(), CONN_TCP, Connection::ConnType(), internal_error(), and pkt_dump. { if ( ! conn || conn->ConnType() != CONN_TCP ) internal_error("connection for the trace rewriter does not exist"); TCP_SourcePacketWriter* writer = ((TCP_Connection*) conn)->SourcePacketWriter(); if ( ! writer ) { if ( ! pkt_dump ) return 0; // don't complain if no output file else if ( ! dump_source_packets ) builtin_run_time("flag dump_source_packets is not set"); else internal_error("source packet writer not initialized"); } return writer; }

TCP_Rewriter* get_trace_rewriter (  Connection *  conn  )

Definition at line 1547 of file TCP_Rewriter.cc. References builtin_run_time(), CONN_TCP, Connection::ConnType(), internal_error(), Connection::RewritingTrace(), and transformed_pkt_dump. Referenced by get_trace_rewriter(). { if ( ! conn || conn->ConnType() != CONN_TCP ) internal_error("connection for the trace rewriter does not exist"); TCP_Rewriter* rewriter = ((TCP_Connection*) conn)->TraceRewriter(); if ( ! rewriter ) { if ( ! transformed_pkt_dump ) return 0; // okay if we don't have an output file else if ( ! conn->RewritingTrace() ) builtin_run_time("flag rewriting_..._trace is not set properly"); else internal_error("trace rewriter not initialized"); } return rewriter; }

TCP_Rewriter* get_trace_rewriter (  Val *  conn_val  )

Definition at line 1541 of file TCP_Rewriter.cc. { Connection* conn = (Connection*) conn_val->AsRecordVal()->GetOrigin(); return get_trace_rewriter(conn); }

uint16 NextIP_ID (  const uint32  src_addr, const uint16  id )

Definition at line 438 of file TCP_Rewriter.cc. References IP_ID::id, IP_ID::ip, ip_id_set, IP_IDSet, uint16, and uint32. { if ( ip_id_set == 0 ) ip_id_set = new IP_IDSet(); IP_ID ipid; ipid.ip = src_addr; ipid.id = id; while ( ip_id_set->find(ipid) != ip_id_set->end() ) { ipid.id = (ipid.id + 1) & 0xffff; if ( ipid.id == id ) { // clear all entries of the IP IP_ID first_id, last_id; first_id.ip = last_id.ip = src_addr; first_id.id = 0; last_id.id = 0xffff; ip_id_set->erase(ip_id_set->find(first_id), ip_id_set->find(last_id)); } } ip_id_set->insert(ipid); return uint16(ipid.id & 0xffff); }

void schedule_flush (  TCP_RewriterEndpoint *  endp  )

Definition at line 1509 of file TCP_Rewriter.cc. References TCP_RewriterEndpoint::Conn(), Ref(), and rewriters_to_flush. Referenced by TCP_RewriterEndpoint::ScheduleFlush(). { Ref(endp->Conn()); rewriters_to_flush.push(endp); }

void schedule_funeral (  TCP_Rewriter *  rewriter  )

Definition at line 1503 of file TCP_Rewriter.cc. References TCP_Rewriter::Conn(), Ref(), and rewriter_funerals. Referenced by TCP_Rewriter::Done(). { Ref(rewriter->Conn()); rewriter_funerals.push(rewriter); }

int time_compare (  struct timeval *  tv_a, struct timeval *  tv_b )  [inline]

Definition at line 67 of file TCP_Rewriter.cc. Referenced by PacketDumper::SortTimeStamp(). { if ( tv_a->tv_sec == tv_b->tv_sec ) return tv_a->tv_usec - tv_b->tv_usec; else return tv_a->tv_sec - tv_b->tv_sec; }

---

Variable Documentation

IP_IDSet* ip_id_set = 0 [static]

Definition at line 50 of file TCP_Rewriter.cc. Referenced by NextIP_ID().

int num_packets_cleaned

Definition at line 51 of file TCP_Rewriter.cc. Referenced by TCP_Rewriter::CleanUpEmptyPlaceHolders(), and net_finish().

int num_packets_held

Definition at line 51 of file TCP_Rewriter.cc. Referenced by TCP_Rewriter::HoldPacket(), net_finish(), and TCP_Rewriter::NextPacket().

std::queue<TCP_Rewriter*> rewriter_funerals [static]

Definition at line 1500 of file TCP_Rewriter.cc. Referenced by flush_rewriter_packet(), and schedule_funeral().

std::queue<TCP_RewriterEndpoint*> rewriters_to_flush [static]

Definition at line 1501 of file TCP_Rewriter.cc. Referenced by flush_rewriter_packet(), and schedule_flush().

struct TCPOptionRewriting tcp_option_rewriting[MAX_TCP_OPTION_REWRITING]

Referenced by TCP_Rewriter::RewriteTCPOption().

---