TCP_Rewriter.h File Reference

#include <queue>
#include <set>
#include <pcap.h>
#include "Val.h"
#include "TCP.h"
#include "Anon.h"

Include dependency graph for TCP_Rewriter.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	PacketDumper
struct	IP_ID
struct	ltipid
class	TCP_TracePacket
class	TCP_RewriterEndpoint
class	TCP_RewriteSlot
class	TCP_Rewriter
class	TCP_SourcePacket
class	TCP_SourcePacketWriter
Defines
#define	END_BY_FIN   1
#define	END_BY_RST   2
#define	END_BY_PEER_RST   4
Typedefs
typedef set< IP_ID, ltipid >	IP_IDSet
Functions
uint16	NextIP_ID (const uint32 src_addr, const uint16 id)
void	schedule_flush (TCP_RewriterEndpoint *endp)
void	schedule_funeral (TCP_Rewriter *rewriter)
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
PacketDumper *	transformed_pkt_dump
PacketDumper *	source_pkt_dump
int	transformed_pkt_dump_MTU

---

Define Documentation

#define END_BY_FIN   1

Definition at line 142 of file TCP_Rewriter.h. Referenced by TCP_RewriterEndpoint::GenerateFIN(), and TCP_RewriterEndpoint::WriteData().

#define END_BY_PEER_RST   4

Definition at line 144 of file TCP_Rewriter.h. Referenced by TCP_RewriterEndpoint::Reset().

#define END_BY_RST   2

Definition at line 143 of file TCP_Rewriter.h. Referenced by TCP_RewriterEndpoint::Reset(), and TCP_RewriterEndpoint::WriteData().

---

Typedef Documentation

typedef set<IP_ID, ltipid> IP_IDSet

Definition at line 43 of file TCP_Rewriter.h. Referenced by NextIP_ID().

---

Function Documentation

void flush_rewriter_packet (   )

Definition at line 1511 of file TCP_Rewriter.cc. References TCP_RewriterEndpoint::Conn(), TCP_Rewriter::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 1612 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 1543 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 1537 of file TCP_Rewriter.cc. References get_trace_rewriter(). { 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 433 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 1505 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 1499 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); }

---

Variable Documentation

PacketDumper* source_pkt_dump

Definition at line 395 of file TCP_Rewriter.h. Referenced by TCP_Connection::Init(), and net_init().

PacketDumper* transformed_pkt_dump

Definition at line 394 of file TCP_Rewriter.h. Referenced by get_trace_rewriter(), TCP_Connection::Init(), net_delete(), and net_init().

int transformed_pkt_dump_MTU

Definition at line 396 of file TCP_Rewriter.h. Referenced by TCP_Connection::Init().

---