Annotation of embedaddon/dhcp/omapip/mrtrace.c, revision 1.1

1.1     ! misho       1: /* mrtrace.c
        !             2: 
        !             3:    Subroutines that support minires tracing... */
        !             4: 
        !             5: /*
        !             6:  * Copyright (c) 2004,2007,2009 by Internet Systems Consortium, Inc. ("ISC")
        !             7:  * Copyright (c) 2001-2003 by Internet Software Consortium
        !             8:  *
        !             9:  * Permission to use, copy, modify, and distribute this software for any
        !            10:  * purpose with or without fee is hereby granted, provided that the above
        !            11:  * copyright notice and this permission notice appear in all copies.
        !            12:  *
        !            13:  * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
        !            14:  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
        !            15:  * MERCHANTABILITY AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR
        !            16:  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
        !            17:  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
        !            18:  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
        !            19:  * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
        !            20:  *
        !            21:  *   Internet Systems Consortium, Inc.
        !            22:  *   950 Charter Street
        !            23:  *   Redwood City, CA 94063
        !            24:  *   <info@isc.org>
        !            25:  *   https://www.isc.org/
        !            26:  *
        !            27:  * This software has been written for Internet Systems Consortium
        !            28:  * by Ted Lemon, as part of a project for Nominum, Inc.   To learn more
        !            29:  * about Internet Systems Consortium, see https://www.isc.org/.  To
        !            30:  * learn more about Nominum, Inc., see ``http://www.nominum.com''.
        !            31:  */
        !            32: 
        !            33: #include "dhcpd.h"
        !            34: #include <omapip/omapip_p.h>
        !            35: 
        !            36: #include "minires/minires.h"
        !            37: #include "arpa/nameser.h"
        !            38: 
        !            39: #include <errno.h>
        !            40: 
        !            41: #if defined(TRACING)
        !            42: static void trace_mr_output_input (trace_type_t *, unsigned, char *);
        !            43: static void trace_mr_output_stop (trace_type_t *);
        !            44: static void trace_mr_input_input (trace_type_t *, unsigned, char *);
        !            45: static void trace_mr_input_stop (trace_type_t *);
        !            46: static void trace_mr_statp_input (trace_type_t *, unsigned, char *);
        !            47: static void trace_mr_statp_stop (trace_type_t *);
        !            48: static void trace_mr_randomid_input (trace_type_t *, unsigned, char *);
        !            49: static void trace_mr_randomid_stop (trace_type_t *);
        !            50: #endif /* TRACING */
        !            51: trace_type_t *trace_mr_output;
        !            52: trace_type_t *trace_mr_input;
        !            53: trace_type_t *trace_mr_statp;
        !            54: trace_type_t *trace_mr_randomid;
        !            55: ssize_t trace_mr_send (int, void *, size_t, int);
        !            56: ssize_t trace_mr_read_playback (struct sockaddr_in *, void *, size_t);
        !            57: void trace_mr_read_record (struct sockaddr_in *, void *, ssize_t);
        !            58: ssize_t trace_mr_recvfrom (int s, void *, size_t, int,
        !            59:                           struct sockaddr *, SOCKLEN_T *);
        !            60: ssize_t trace_mr_read (int, void *, size_t);
        !            61: int trace_mr_connect (int s, struct sockaddr *, SOCKLEN_T);
        !            62: int trace_mr_socket (int, int, int);
        !            63: int trace_mr_bind (int, struct sockaddr *, SOCKLEN_T);
        !            64: int trace_mr_close (int);
        !            65: time_t trace_mr_time (time_t *);
        !            66: int trace_mr_select (int, fd_set *, fd_set *, fd_set *, struct timeval *);
        !            67: unsigned int trace_mr_res_randomid (unsigned int);
        !            68: 
        !            69: #if defined (TRACING)
        !            70: void trace_mr_init ()
        !            71: {
        !            72:        trace_mr_output = trace_type_register ("mr-output", (void *)0,
        !            73:                                               trace_mr_output_input,
        !            74:                                               trace_mr_output_stop, MDL);
        !            75:        trace_mr_input = trace_type_register ("mr-input", (void *)0,
        !            76:                                              trace_mr_input_input,
        !            77:                                              trace_mr_input_stop, MDL);
        !            78:        trace_mr_statp = trace_type_register ("mr-statp", (void *)0,
        !            79:                                              trace_mr_statp_input,
        !            80:                                              trace_mr_statp_stop, MDL);
        !            81:        trace_mr_randomid = trace_type_register ("mr-randomid", (void *)0,
        !            82:                                                 trace_mr_randomid_input,
        !            83:                                                 trace_mr_randomid_stop, MDL);
        !            84: }
        !            85: 
        !            86: void trace_mr_statp_setup (res_state statp)
        !            87: {
        !            88:        unsigned buflen = 0;
        !            89:        char *buf = (char *)0;
        !            90:        isc_result_t status;
        !            91:        int i;
        !            92: 
        !            93:        if (trace_playback ()) {
        !            94:                int nscount;
        !            95:                status = trace_get_packet (&trace_mr_statp, &buflen, &buf);
        !            96:                if (status != ISC_R_SUCCESS) {
        !            97:                        log_error ("trace_mr_statp: no statp packet found.");
        !            98:                        return;
        !            99:                }
        !           100:                nscount = buflen / sizeof (struct in_addr);
        !           101:                if (nscount * (sizeof (struct in_addr)) != buflen ||
        !           102:                    nscount < 1) {
        !           103:                        log_error ("trace_mr_statp: bogus length: %d",
        !           104:                                   buflen);
        !           105:                        return;
        !           106:                }
        !           107:                if (nscount > MAXNS)
        !           108:                        nscount = MAXNS;
        !           109:                for (i = 0; i < nscount; i++) {
        !           110: #if defined (HAVE_SA_LEN)
        !           111:                        statp -> nsaddr_list [i].sin_len =
        !           112:                                sizeof (struct sockaddr_in);
        !           113: #endif
        !           114:                        memset (&statp -> nsaddr_list [i].sin_zero, 0,
        !           115:                                sizeof statp -> nsaddr_list [i].sin_zero);
        !           116:                        statp -> nsaddr_list [i].sin_port = htons (53); /*XXX*/
        !           117:                        statp -> nsaddr_list [i].sin_family = AF_INET;
        !           118:                        memcpy (&statp -> nsaddr_list [i].sin_addr,
        !           119:                                (buf + i * (sizeof (struct in_addr))),
        !           120:                                sizeof (struct in_addr));
        !           121:                }
        !           122:                statp -> nscount = nscount;
        !           123:                dfree (buf, MDL);
        !           124:                buf = (char *)0;
        !           125:        }
        !           126:        if (trace_record ()) {
        !           127:                trace_iov_t *iov;
        !           128:                iov = dmalloc ((statp -> nscount *
        !           129:                                sizeof (trace_iov_t)), MDL);
        !           130:                if (!iov) {
        !           131:                        trace_stop ();
        !           132:                        log_error ("No memory for statp iov.");
        !           133:                        return;
        !           134:                }
        !           135:                for (i = 0; i < statp -> nscount; i++) {
        !           136:                        iov [i].buf =
        !           137:                                (char *)&statp -> nsaddr_list [i].sin_addr;
        !           138:                        iov [i].len = sizeof (struct in_addr);
        !           139:                }
        !           140:                trace_write_packet_iov (trace_mr_statp, i, iov, MDL);
        !           141:                dfree (iov, MDL);
        !           142:        }
        !           143: }
        !           144: #endif
        !           145: 
        !           146: ssize_t trace_mr_send (int fd, void *msg, size_t len, int flags)
        !           147: {
        !           148:        ssize_t rv;
        !           149: #if defined (TRACING)
        !           150:        isc_result_t status;
        !           151:        unsigned buflen = 0;
        !           152:        char *inbuf = (char *)0;
        !           153:        u_int32_t result;
        !           154:        u_int32_t sflags;
        !           155: 
        !           156:        if (trace_playback()) {
        !           157:                status = trace_get_packet (&trace_mr_output, &buflen, &inbuf);
        !           158:                if (status != ISC_R_SUCCESS) {
        !           159:                        log_error ("trace_mr_recvfrom: no input found.");
        !           160:                        errno = ECONNREFUSED;
        !           161:                        return -1;
        !           162:                }
        !           163:                if (buflen < sizeof result) {
        !           164:                        log_error ("trace_mr_recvfrom: data too short.");
        !           165:                        errno = ECONNREFUSED;
        !           166:                        dfree (inbuf, MDL);
        !           167:                        return -1;
        !           168:                }
        !           169:                memcpy (&result, inbuf, sizeof result);
        !           170:                rv = ntohl (result);
        !           171:                dfree (inbuf, MDL);
        !           172:        } else
        !           173: #endif
        !           174:                rv = send (fd, msg, len, flags);
        !           175: #if defined (TRACING)
        !           176:        if (trace_record ()) {
        !           177:                trace_iov_t iov [3];
        !           178:                result = htonl (rv);
        !           179:                sflags = htonl (flags);
        !           180:                iov [0].len = sizeof result;
        !           181:                iov [0].buf = (char *)&result;
        !           182:                iov [1].len = sizeof sflags;
        !           183:                iov [1].buf = (char *)&flags;
        !           184:                iov [2].len = len;
        !           185:                iov [2].buf = msg;
        !           186:                trace_write_packet_iov (trace_mr_output, 3, iov, MDL);
        !           187:        }
        !           188: #endif
        !           189:        return rv;
        !           190: }
        !           191: 
        !           192: #if defined (TRACING)
        !           193: ssize_t trace_mr_read_playback (struct sockaddr_in *from,
        !           194:                                void *buf, size_t nbytes)
        !           195: {
        !           196:        isc_result_t status;
        !           197:        unsigned buflen = 0, left;
        !           198:        char *inbuf = (char *)0;
        !           199:        char *bufp;
        !           200:        u_int32_t result;
        !           201: 
        !           202:        status = trace_get_packet (&trace_mr_input, &buflen, &inbuf);
        !           203:        if (status != ISC_R_SUCCESS) {
        !           204:                log_error ("trace_mr_recvfrom: no input found.");
        !           205:                errno = ECONNREFUSED;
        !           206:                return -1;
        !           207:        }
        !           208:        if (buflen < sizeof result) {
        !           209:                log_error ("trace_mr_recvfrom: data too short.");
        !           210:                errno = ECONNREFUSED;
        !           211:                dfree (inbuf, MDL);
        !           212:                return -1;
        !           213:        }
        !           214:        bufp = inbuf;
        !           215:        left = buflen;
        !           216:        memcpy (&result, bufp, sizeof result);
        !           217:        result = ntohl (result);
        !           218:        bufp += sizeof result;
        !           219:        left -= sizeof result;
        !           220:        if (result == 0) {
        !           221:                if (left < ((sizeof from -> sin_port) +
        !           222:                            sizeof (from -> sin_addr))) {
        !           223:                        log_error ("trace_mr_recvfrom: data too short.");
        !           224:                        errno = ECONNREFUSED;
        !           225:                        dfree (inbuf, MDL);
        !           226:                        return -1;
        !           227:                }
        !           228:                if (from)
        !           229:                        memcpy (&from -> sin_addr, bufp,
        !           230:                                sizeof from -> sin_addr);
        !           231:                bufp += sizeof from -> sin_addr;
        !           232:                left -= sizeof from -> sin_addr;
        !           233:                if (from)
        !           234:                        memcpy (&from -> sin_port, bufp,
        !           235:                                sizeof from -> sin_port);
        !           236:                bufp += sizeof from -> sin_port;
        !           237:                left -= sizeof from -> sin_port;
        !           238:                if (from) {
        !           239:                        from -> sin_family = AF_INET;
        !           240: #if defined(HAVE_SA_LEN)
        !           241:                        from -> sin_len = sizeof (struct sockaddr_in);
        !           242: #endif
        !           243:                        memset (from -> sin_zero, 0, sizeof from -> sin_zero);
        !           244:                }
        !           245:                if (left > nbytes) {
        !           246:                        log_error ("trace_mr_recvfrom: too much%s",
        !           247:                                   " data.");
        !           248:                        errno = ECONNREFUSED;
        !           249:                        dfree (inbuf, MDL);
        !           250:                        return -1;
        !           251:                }
        !           252:                memcpy (buf, bufp, left);
        !           253:                dfree (inbuf, MDL);
        !           254:                return left;
        !           255:        }
        !           256:        errno = ECONNREFUSED;
        !           257:        return -1;
        !           258: }
        !           259: 
        !           260: void trace_mr_read_record (struct sockaddr_in *from, void *buf, ssize_t rv)
        !           261: {
        !           262:        trace_iov_t iov [4];
        !           263:        u_int32_t result;
        !           264:        int iolen = 0;
        !           265:        static char zero [4] = { 0, 0, 0, 0 };
        !           266:        
        !           267:        if (rv < 0)
        !           268:                result = htonl (errno);         /* XXX */
        !           269:        else
        !           270:                result = 0;
        !           271:        iov [iolen].buf = (char *)&result;
        !           272:        iov [iolen++].len = sizeof result;
        !           273:        if (rv > 0) {
        !           274:                if (from) {
        !           275:                        iov [iolen].buf = (char *)&from -> sin_addr;
        !           276:                        iov [iolen++].len = sizeof from -> sin_addr;
        !           277:                        iov [iolen].buf = (char *)&from -> sin_port;
        !           278:                        iov [iolen++].len = sizeof from -> sin_port;
        !           279:                } else {
        !           280:                        iov [iolen].buf = zero;
        !           281:                        iov [iolen++].len = sizeof from -> sin_addr;
        !           282:                        iov [iolen].buf = zero;
        !           283:                        iov [iolen++].len = sizeof from -> sin_port;
        !           284:                }
        !           285: 
        !           286:                iov [iolen].buf = buf;
        !           287:                iov [iolen++].len = rv;
        !           288:        }
        !           289:        trace_write_packet_iov (trace_mr_input, iolen, iov, MDL);
        !           290: }
        !           291: #endif
        !           292: 
        !           293: ssize_t trace_mr_recvfrom (int s, void *buf, size_t len, int flags,
        !           294:                           struct sockaddr *from, SOCKLEN_T *fromlen)
        !           295: {
        !           296:        ssize_t rv;
        !           297: 
        !           298: #if defined (TRACING)
        !           299:        if (trace_playback ())
        !           300:                rv = trace_mr_read_playback ((struct sockaddr_in *)from,
        !           301:                                             buf, len);
        !           302:        else
        !           303: #endif
        !           304:                rv = recvfrom (s, buf, len, flags, from, fromlen);
        !           305: #if defined (TRACING)
        !           306:        if (trace_record ()) {
        !           307:                trace_mr_read_record ((struct sockaddr_in *)from, buf, rv);
        !           308:        }
        !           309: #endif
        !           310:        return rv;
        !           311: }
        !           312: 
        !           313: ssize_t trace_mr_read (int d, void *buf, size_t nbytes)
        !           314: {
        !           315:        ssize_t rv;
        !           316: 
        !           317: #if defined (TRACING)
        !           318:        if (trace_playback ())
        !           319:                rv = trace_mr_read_playback ((struct sockaddr_in *)0,
        !           320:                                             buf, nbytes);
        !           321:        else
        !           322: #endif
        !           323:                rv = read (d, buf, nbytes);
        !           324: #if defined (TRACING)
        !           325:        if (trace_record ()) {
        !           326:                trace_mr_read_record ((struct sockaddr_in *)0, buf, rv);
        !           327:        }
        !           328: #endif
        !           329:        return rv;
        !           330: }
        !           331: 
        !           332: int trace_mr_connect (int s, struct sockaddr *name, SOCKLEN_T namelen)
        !           333: {
        !           334: #if defined (TRACING)
        !           335:        if (!trace_playback ())
        !           336: #endif
        !           337:                return connect (s, name, namelen);
        !           338: #if defined (TRACING)
        !           339:        return 0;
        !           340: #endif
        !           341: }
        !           342: 
        !           343: int trace_mr_socket (int domain, int type, int protocol)
        !           344: {
        !           345: #if defined (TRACING)
        !           346:        if (!trace_playback ())
        !           347: #endif
        !           348:                return socket (domain, type, protocol);
        !           349: #if defined (TRACING)
        !           350:        return 100;
        !           351: #endif
        !           352: }
        !           353: 
        !           354: int trace_mr_bind (int s, struct sockaddr *name, SOCKLEN_T namelen)
        !           355: {
        !           356: #if defined (TRACING)
        !           357:        if (!trace_playback ())
        !           358: #endif
        !           359:                return bind (s, name, namelen);
        !           360: #if defined (TRACING)
        !           361:        return 0;
        !           362: #endif
        !           363: }
        !           364: 
        !           365: int trace_mr_close (int s)
        !           366: {
        !           367: #if defined (TRACING)
        !           368:        if (!trace_playback ())
        !           369: #endif
        !           370:                return close (s);
        !           371: #if defined (TRACING)
        !           372:        return 0;
        !           373: #endif
        !           374: }
        !           375: 
        !           376: time_t trace_mr_time (time_t *tp)
        !           377: {
        !           378: #if defined (TRACING)
        !           379:        if (trace_playback ()) {
        !           380:                if (tp)
        !           381:                        *tp = cur_time;
        !           382:                return cur_time;
        !           383:        }
        !           384: #endif
        !           385:        return time (tp);
        !           386: }
        !           387: 
        !           388: int trace_mr_select (int s, fd_set *r, fd_set *w, fd_set *x, struct timeval *t)
        !           389: {
        !           390: #if defined (TRACING)
        !           391:        trace_type_t *ttp = (trace_type_t *)0;
        !           392: 
        !           393:        if (trace_playback ()) {
        !           394:                time_t nct = trace_snoop_time (&ttp);
        !           395:                time_t secr = t -> tv_sec;
        !           396:                t -> tv_sec = nct - cur_time;
        !           397:                if (t -> tv_sec > secr)
        !           398:                        return 0;
        !           399:                if (ttp == trace_mr_input)
        !           400:                        return 1;
        !           401:                return 0;
        !           402:        }
        !           403: #endif
        !           404:        return select (s, r, w, x, t);
        !           405: }
        !           406: 
        !           407: unsigned int trace_mr_res_randomid (unsigned int oldid)
        !           408: {
        !           409:        int rid = oldid;
        !           410: #if defined (TRACING)
        !           411:        u_int32_t id;
        !           412:        unsigned buflen = 0;
        !           413:        char *buf = (char *)0;
        !           414:        isc_result_t status;
        !           415: 
        !           416:        if (trace_playback ()) {
        !           417:                status = trace_get_packet (&trace_mr_randomid, &buflen, &buf);
        !           418:                if (status != ISC_R_SUCCESS) {
        !           419:                        log_error ("trace_mr_statp: no statp packet found.");
        !           420:                        return oldid;
        !           421:                }
        !           422:                if (buflen != sizeof id) {
        !           423:                        log_error ("trace_mr_randomid: bogus length: %d",
        !           424:                                   buflen);
        !           425:                        return oldid;
        !           426:                }
        !           427:                memcpy (&id, buf, sizeof id);
        !           428:                dfree (buf, MDL);
        !           429:                buf = (char *)0;
        !           430:                rid = ntohl (id);
        !           431:        }
        !           432:        if (trace_record ()) {
        !           433:                id = htonl (rid);
        !           434:                trace_write_packet (trace_mr_randomid,
        !           435:                                    sizeof id, (char *)&id, MDL);
        !           436:        }
        !           437: #endif
        !           438:        return rid;
        !           439: }
        !           440: 
        !           441: #if defined (TRACING)
        !           442: static void trace_mr_output_input (trace_type_t *ttype,
        !           443:                                   unsigned length, char *buf)
        !           444: {
        !           445: }
        !           446: 
        !           447: static void trace_mr_output_stop (trace_type_t *ttype)
        !           448: {
        !           449: }
        !           450: 
        !           451: static void trace_mr_input_input (trace_type_t *ttype,
        !           452:                                  unsigned length, char *buf)
        !           453: {
        !           454:        log_error ("unaccounted-for minires input.");
        !           455: }
        !           456: 
        !           457: static void trace_mr_input_stop (trace_type_t *ttype)
        !           458: {
        !           459: }
        !           460: 
        !           461: static void trace_mr_statp_input (trace_type_t *ttype,
        !           462:                                  unsigned length, char *buf)
        !           463: {
        !           464:        log_error ("unaccounted-for minires statp input.");
        !           465: }
        !           466: 
        !           467: static void trace_mr_statp_stop (trace_type_t *ttype)
        !           468: {
        !           469: }
        !           470: 
        !           471: static void trace_mr_randomid_input (trace_type_t *ttype,
        !           472:                                     unsigned length, char *buf)
        !           473: {
        !           474:        log_error ("unaccounted-for minires randomid input.");
        !           475: }
        !           476: 
        !           477: static void trace_mr_randomid_stop (trace_type_t *ttype)
        !           478: {
        !           479: }
        !           480: #endif

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>