Minires trace support.

1 files changed, 291 insertions, 0 deletions

diff --git a/omapip/mrtrace.c b/omapip/mrtrace.c
new file mode 100644
index 00000000..f90b1e3e
--- /dev/null
+++ b/omapip/mrtrace.c
@@ -0,0 +1,291 @@
+/* mrtrace.c
+
+   Subroutines that support minires tracing... */
+
+/*
+ * Copyright (c) 2001 Internet Software Consortium.
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ * 3. Neither the name of The Internet Software Consortium nor the names
+ *    of its contributors may be used to endorse or promote products derived
+ *    from this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE INTERNET SOFTWARE CONSORTIUM AND
+ * CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
+ * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED.  IN NO EVENT SHALL THE INTERNET SOFTWARE CONSORTIUM OR
+ * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
+ * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+ * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
+ * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
+ *
+ * This software has been written for the Internet Software Consortium
+ * by Ted Lemon, as part of a project for Nominum, Inc.   To learn more
+ * about the Internet Software Consortium, see http://www.isc.org/.  To
+ * learn more about Nominum, Inc., see ``http://www.nominum.com''.
+ */
+
+#include <omapip/omapip_p.h>
+
+#include "minires/minires.h"
+#include "arpa/nameser.h"
+
+#if defined (TRACING)
+static void trace_mr_output_input (trace_type_t *, unsigned, char *);
+static void trace_mr_output_stop (trace_type_t *);
+static void trace_mr_input_input (trace_type_t *, unsigned, char *);
+static void trace_mr_input_stop (trace_type_t *);
+static void trace_mr_statp_input (trace_type_t *, unsigned, char *);
+static void trace_mr_statp_stop (trace_type_t *);
+trace_type_t *trace_mr_output;
+trace_type_t *trace_mr_input;
+trace_type_t *trace_mr_statp;
+ssize_t trace_mr_send (int, void *, size_t, int);
+ssize_t trace_mr_read_playback (void *, size_t);
+void trace_mr_read_record (void *, ssize_t);
+ssize_t trace_mr_recvfrom (int s, void *, size_t, int,
+			   struct sockaddr *, socklen_t *);
+ssize_t trace_mr_read (int, void *, size_t);
+int trace_mr_connect (int s, const struct sockaddr *, socklen_t);
+int trace_mr_socket (int, int, int);
+int trace_mr_bind (int, const struct sockaddr *, socklen_t);
+int trace_mr_close (int);
+time_t trace_mr_time (time_t *);
+int trace_mr_select (int, fd_set *, fd_set *, fd_set *, struct timeval *);
+
+extern time_t cur_time;
+
+void trace_mr_init ()
+{
+	trace_mr_output = trace_type_register ("mr-output", (void *)0,
+					       trace_mr_output_input,
+					       trace_mr_output_stop, MDL);
+	trace_mr_input = trace_type_register ("mr-input", (void *)0,
+					      trace_mr_input_input,
+					      trace_mr_input_stop, MDL);
+	trace_mr_statp = trace_type_register ("mr-statp", (void *)0,
+					      trace_mr_statp_input,
+					      trace_mr_statp_stop, MDL);
+}
+
+void trace_mr_statp_setup (res_state statp)
+{
+	unsigned buflen = 0;
+	char *buf = (char *)0;
+	isc_result_t status;
+
+	if (trace_playback ()) {
+		int nscount;
+		status = trace_get_packet (&trace_mr_statp, &buflen, &buf);
+		if (status != ISC_R_SUCCESS) {
+			log_error ("trace_mr_statp: no statp packet found.");
+			return;
+		}
+		nscount = buflen / sizeof (struct sockaddr_in);
+		if (nscount * (sizeof (struct sockaddr_in)) != buflen) {
+			log_error ("trace_mr_statp: bogus length: %d",
+				   buflen);
+			return;
+		}
+		if (nscount > MAXNS)
+			nscount = MAXNS;
+		memcpy (statp, buf, nscount * sizeof (struct sockaddr_in));
+		dfree (buf, MDL);
+		buf = (char *)0;
+	}
+	if (trace_record ()) {
+	    trace_write_packet (trace_mr_statp,
+				statp -> nscount * sizeof (struct sockaddr_in),
+				(char *)statp -> nsaddr_list, MDL);
+	}
+}
+
+
+ssize_t trace_mr_send (int fd, void *msg, size_t len, int flags)
+{
+	if (trace_record ())
+		trace_write_packet (trace_mr_output, len, msg, MDL);
+	return send (fd, msg, len, flags);
+}
+
+ssize_t trace_mr_read_playback (void *buf, size_t nbytes)
+{
+	isc_result_t status;
+	unsigned buflen = 0;
+	char *inbuf = (char *)0;
+	u_int32_t result;
+	ssize_t rv;
+
+	status = trace_get_packet (&trace_mr_input, &buflen, &inbuf);
+	if (status != ISC_R_SUCCESS) {
+		log_error ("trace_mr_recvfrom: no input found.");
+		errno = ECONNREFUSED;
+		return -1;
+	}
+	if (buflen < sizeof result) {
+		log_error ("trace_mr_recvfrom: data too short.");
+		errno = ECONNREFUSED;
+		dfree (buf, MDL);
+		return -1;
+	}
+	memcpy (&result, inbuf, sizeof result);
+	result = ntohl (result);
+	if (result == 0) {
+		rv = buflen - sizeof result;
+		if (rv > nbytes) {
+			log_error ("trace_mr_recvfrom: too much%s",
+				   " data.");
+			errno = ECONNREFUSED;
+			dfree (buf, MDL);
+			return -1;
+		}
+		memcpy (buf, inbuf, (unsigned)rv);
+		return rv;
+	}
+	errno = ECONNREFUSED;
+	return -1;
+}
+
+void trace_mr_read_record (void *buf, ssize_t rv)
+{
+	trace_iov_t iov [2];
+	u_int32_t result;
+	
+	result = htonl (rv);
+	iov [0].buf = (char *)&result;
+	iov [0].len = sizeof result;
+	if (rv > 0) {
+		iov [1].buf = buf;
+		iov [1].len = rv;
+	}
+	trace_write_packet_iov (trace_mr_input,
+				rv > 0 ? 2 : 1, iov, MDL);
+}
+
+ssize_t trace_mr_recvfrom (int s, void *buf, size_t len, int flags,
+			   struct sockaddr *from, socklen_t *fromlen)
+{
+	ssize_t rv;
+
+	if (trace_playback ()) {
+		trace_mr_read_playback (buf, len);
+	} else {
+		rv = recvfrom (s, buf, len, flags, from, fromlen);
+		if (trace_record ()) {
+			trace_mr_read_record (buf, rv);
+		}
+	}
+	return rv;
+}
+
+ssize_t trace_mr_read (int d, void *buf, size_t nbytes)
+{
+	ssize_t rv;
+
+	if (trace_playback ()) {
+		trace_mr_read_playback (buf, nbytes);
+	} else {
+		rv = read (d, buf, nbytes);
+		if (trace_record ()) {
+			trace_mr_read_record (buf, rv);
+		}
+	}
+	return rv;
+}
+
+int trace_mr_connect (int s, const struct sockaddr *name, socklen_t namelen)
+{
+	if (!trace_playback ())
+		return connect (s, name, namelen);
+	return 1000;
+}
+
+int trace_mr_socket (int domain, int type, int protocol)
+{
+	if (!trace_playback ())
+		return socket (domain, type, protocol);
+	return 1000;
+}
+
+int trace_mr_bind (int s, const struct sockaddr *name, socklen_t namelen)
+{
+	if (!trace_playback ())
+		return bind (s, name, namelen);
+	return 0;
+}
+
+int trace_mr_close (int s)
+{
+	if (!trace_playback ())
+		return close (s);
+	return 0;
+}
+
+time_t trace_mr_time (time_t *tp)
+{
+	if (trace_playback ()) {
+		if (tp)
+			*tp = cur_time;
+		return cur_time;
+	}
+	return time (tp);
+}
+
+int trace_mr_select (int s, fd_set *r, fd_set *w, fd_set *x, struct timeval *t)
+{
+	if (trace_playback ()) {
+		time_t nct = trace_snoop_time ();
+		time_t secr = t -> tv_sec;
+		t -> tv_sec = nct - cur_time;
+		if (t -> tv_sec > secr)
+			return 0;
+		return 1;
+	}
+	return select (s, r, w, x, t);
+}
+
+static void trace_mr_output_input (trace_type_t *ttype,
+				   unsigned length, char *buf)
+{
+}
+
+static void trace_mr_output_stop (trace_type_t *ttype)
+{
+}
+
+static void trace_mr_input_input (trace_type_t *ttype,
+				  unsigned length, char *buf)
+{
+	log_error ("unaccounted-for minires input.");
+}
+
+static void trace_mr_input_stop (trace_type_t *ttype)
+{
+}
+
+static void trace_mr_statp_input (trace_type_t *ttype,
+				  unsigned length, char *buf)
+{
+	log_error ("unaccounted-for minires statp input.");
+}
+
+static void trace_mr_statp_stop (trace_type_t *ttype)
+{
+}
+
+
+#endif