diff options
Diffstat (limited to 'omapip')
| -rw-r--r-- | omapip/mrtrace.c | 291 |
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 |