Initial import of svn+ssh:://svn.icir.org/bro/trunk/bro as of r7088

src/Rewriter.cc

@@ -0,0 +1,35 @@
+// $Id:$
+//
+// See the file "COPYING" in the main distribution directory for copyright.
+
+#include "TCP_Rewriter.h"
+#include "UDP_Rewriter.h"
+
+// The following two are called from .bif's to obtain handle of Rewriter.
+Rewriter* get_trace_rewriter(Val* conn_val)
+	{
+	Connection* conn = (Connection*) conn_val->AsRecordVal()->GetOrigin();
+	return get_trace_rewriter(conn);
+	}
+
+Rewriter* get_trace_rewriter(Connection* conn)
+	{
+	if ( ! conn ||
+	     (conn->ConnTransport() != TRANSPORT_TCP &&
+	      conn->ConnTransport() != TRANSPORT_UDP) )
+		internal_error("connection for the trace rewriter does not exist");
+
+	Rewriter* rewriter = conn->TraceRewriter();
+	if ( rewriter )
+		return 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 0;
+	}