Home | History | Annotate | Line # | Download | only in trpt
trpt.c revision 1.21
      1 /*	$NetBSD: trpt.c,v 1.21 2005/09/06 03:04:15 rpaulo Exp $	*/
      2 
      3 /*-
      4  * Copyright (c) 1997 The NetBSD Foundation, Inc.
      5  * All rights reserved.
      6  *
      7  * This code is derived from software contributed to The NetBSD Foundation
      8  * by Jason R. Thorpe of the Numerical Aerospace Simulation Facility,
      9  * NASA Ames Research Center.
     10  *
     11  * Redistribution and use in source and binary forms, with or without
     12  * modification, are permitted provided that the following conditions
     13  * are met:
     14  * 1. Redistributions of source code must retain the above copyright
     15  *    notice, this list of conditions and the following disclaimer.
     16  * 2. Redistributions in binary form must reproduce the above copyright
     17  *    notice, this list of conditions and the following disclaimer in the
     18  *    documentation and/or other materials provided with the distribution.
     19  * 3. All advertising materials mentioning features or use of this software
     20  *    must display the following acknowledgement:
     21  *	This product includes software developed by the NetBSD
     22  *	Foundation, Inc. and its contributors.
     23  * 4. Neither the name of The NetBSD Foundation nor the names of its
     24  *    contributors may be used to endorse or promote products derived
     25  *    from this software without specific prior written permission.
     26  *
     27  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     28  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     29  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     30  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     31  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     32  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     33  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     34  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     35  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     36  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     37  * POSSIBILITY OF SUCH DAMAGE.
     38  */
     39 
     40 /*
     41  * Copyright (c) 1983, 1988, 1993
     42  *	The Regents of the University of California.  All rights reserved.
     43  *
     44  * Redistribution and use in source and binary forms, with or without
     45  * modification, are permitted provided that the following conditions
     46  * are met:
     47  * 1. Redistributions of source code must retain the above copyright
     48  *    notice, this list of conditions and the following disclaimer.
     49  * 2. Redistributions in binary form must reproduce the above copyright
     50  *    notice, this list of conditions and the following disclaimer in the
     51  *    documentation and/or other materials provided with the distribution.
     52  * 3. Neither the name of the University nor the names of its contributors
     53  *    may be used to endorse or promote products derived from this software
     54  *    without specific prior written permission.
     55  *
     56  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     57  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     58  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     59  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     60  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     61  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     62  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     63  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     64  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     65  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     66  * SUCH DAMAGE.
     67  */
     68 
     69 #include <sys/cdefs.h>
     70 #ifndef lint
     71 __COPYRIGHT(
     72 "@(#) Copyright (c) 1983, 1988, 1993\n\
     73 	The Regents of the University of California.  All rights reserved.\n");
     74 #endif /* not lint */
     75 
     76 #ifndef lint
     77 #if 0
     78 static char sccsid[] = "@(#)trpt.c	8.1 (Berkeley) 6/6/93";
     79 #else
     80 __RCSID("$NetBSD: trpt.c,v 1.21 2005/09/06 03:04:15 rpaulo Exp $");
     81 #endif
     82 #endif /* not lint */
     83 
     84 #include <sys/param.h>
     85 #include <sys/queue.h>
     86 #include <sys/socket.h>
     87 #include <sys/socketvar.h>
     88 #include <sys/sysctl.h>
     89 #define PRUREQUESTS
     90 #include <sys/protosw.h>
     91 #include <sys/file.h>
     92 
     93 #include <net/route.h>
     94 #include <net/if.h>
     95 
     96 #include <netinet/in.h>
     97 #include <netinet/in_systm.h>
     98 #include <netinet/ip.h>
     99 #include <netinet/in_pcb.h>
    100 #include <netinet/ip_var.h>
    101 
    102 #ifdef INET6
    103 #ifndef INET
    104 #include <netinet/in.h>
    105 #endif
    106 #include <netinet/ip6.h>
    107 #endif
    108 
    109 #include <netinet/tcp.h>
    110 #define TCPSTATES
    111 #include <netinet/tcp_fsm.h>
    112 #include <netinet/tcp_seq.h>
    113 #define	TCPTIMERS
    114 #include <netinet/tcp_timer.h>
    115 #include <netinet/tcp_var.h>
    116 #include <netinet/tcpip.h>
    117 #define	TANAMES
    118 #include <netinet/tcp_debug.h>
    119 
    120 #include <arpa/inet.h>
    121 
    122 #include <err.h>
    123 #include <stdio.h>
    124 #include <errno.h>
    125 #include <kvm.h>
    126 #include <nlist.h>
    127 #include <paths.h>
    128 #include <limits.h>
    129 #include <stdlib.h>
    130 #include <unistd.h>
    131 
    132 struct nlist nl[] = {
    133 #define	N_HARDCLOCK_TICKS	0
    134 	{ "_hardclock_ticks" },
    135 #define	N_TCP_DEBUG		1
    136 	{ "_tcp_debug" },
    137 #define	N_TCP_DEBX		2
    138 	{ "_tcp_debx" },
    139 	{ NULL },
    140 };
    141 
    142 static caddr_t tcp_pcbs[TCP_NDEBUG];
    143 static n_time ntime;
    144 static int aflag, follow, sflag, tflag;
    145 
    146 /* see sys/netinet/tcp_debug.c */
    147 struct  tcp_debug tcp_debug[TCP_NDEBUG];
    148 int tcp_debx;
    149 
    150 int	main(int, char *[]);
    151 void	dotrace(caddr_t);
    152 void	tcp_trace(short, short, struct tcpcb *, struct tcpcb *,
    153 	    int, void *, int);
    154 int	numeric(const void *, const void *);
    155 void	usage(void);
    156 
    157 kvm_t	*kd;
    158 int     use_sysctl;
    159 
    160 int
    161 main(int argc, char *argv[])
    162 {
    163 	int ch, i, jflag, npcbs;
    164 	char *system, *core, *cp, errbuf[_POSIX2_LINE_MAX];
    165 	unsigned long l;
    166 
    167 	jflag = npcbs = 0;
    168 	system = core = NULL;
    169 
    170 	while ((ch = getopt(argc, argv, "afjp:stN:M:")) != -1) {
    171 		switch (ch) {
    172 		case 'a':
    173 			++aflag;
    174 			break;
    175 		case 'f':
    176 			++follow;
    177 			setlinebuf(stdout);
    178 			break;
    179 		case 'j':
    180 			++jflag;
    181 			break;
    182 		case 'p':
    183 			if (npcbs >= TCP_NDEBUG)
    184 				errx(1, "too many pcbs specified");
    185 			errno = 0;
    186 			cp = NULL;
    187 			l = strtoul(optarg, &cp, 16);
    188 			tcp_pcbs[npcbs] = (caddr_t)l;
    189 			if (*optarg == '\0' || *cp != '\0' || errno ||
    190 			    (unsigned long)tcp_pcbs[npcbs] != l)
    191 				errx(1, "invalid address: %s", optarg);
    192 			npcbs++;
    193 			break;
    194 		case 's':
    195 			++sflag;
    196 			break;
    197 		case 't':
    198 			++tflag;
    199 			break;
    200 		case 'N':
    201 			system = optarg;
    202 			break;
    203 		case 'M':
    204 			core = optarg;
    205 			break;
    206 		default:
    207 			usage();
    208 			/* NOTREACHED */
    209 		}
    210 	}
    211 	argc -= optind;
    212 	argv += optind;
    213 
    214 	if (argc)
    215 		usage();
    216 
    217 	use_sysctl = (system == NULL && core == NULL);
    218 
    219 	if (use_sysctl) {
    220 		size_t lenx = sizeof(tcp_debx);
    221 		size_t lend = sizeof(tcp_debug);
    222 
    223 		if (sysctlbyname("net.inet.tcp.debx", &tcp_debx, &lenx,
    224 		    NULL, 0) == -1)
    225 			err(1, "net.inet.tcp.debx");
    226 		if (sysctlbyname("net.inet.tcp.debug", &tcp_debug, &lend,
    227 		    NULL, 0) == -1)
    228 			err(1, "net.inet.tcp.debug");
    229 	} else {
    230 		kd = kvm_openfiles(system, core, NULL, O_RDONLY, errbuf);
    231 		if (kd == NULL)
    232 			errx(1, "can't open kmem: %s", errbuf);
    233 
    234 		if (kvm_nlist(kd, nl))
    235 			errx(2, "%s: no namelist", system);
    236 
    237 		if (kvm_read(kd, nl[N_TCP_DEBX].n_value, (char *)&tcp_debx,
    238 		    sizeof(tcp_debx)) != sizeof(tcp_debx))
    239 			errx(3, "tcp_debx: %s", kvm_geterr(kd));
    240 
    241 		if (kvm_read(kd, nl[N_TCP_DEBUG].n_value, (char *)tcp_debug,
    242 		    sizeof(tcp_debug)) != sizeof(tcp_debug))
    243 			errx(3, "tcp_debug: %s", kvm_geterr(kd));
    244 	}
    245 
    246 	/*
    247 	 * If no control blocks have been specified, figure
    248 	 * out how many distinct one we have and summarize
    249 	 * them in tcp_pcbs for sorting the trace records
    250 	 * below.
    251 	 */
    252 	if (npcbs == 0) {
    253 		for (i = 0; i < TCP_NDEBUG; i++) {
    254 			struct tcp_debug *td = &tcp_debug[i];
    255 			int j;
    256 
    257 			if (td->td_tcb == 0)
    258 				continue;
    259 			for (j = 0; j < npcbs; j++)
    260 				if (tcp_pcbs[j] == td->td_tcb)
    261 					break;
    262 			if (j >= npcbs)
    263 				tcp_pcbs[npcbs++] = td->td_tcb;
    264 		}
    265 		if (npcbs == 0)
    266 			exit(0);
    267 	}
    268 	qsort(tcp_pcbs, npcbs, sizeof(caddr_t), numeric);
    269 	if (jflag) {
    270 		for (i = 0;;) {
    271 			printf("%lx", (long)tcp_pcbs[i]);
    272 			if (++i == npcbs)
    273 				break;
    274 			fputs(", ", stdout);
    275 		}
    276 		putchar('\n');
    277 	} else {
    278 		for (i = 0; i < npcbs; i++) {
    279 			printf("\n%lx:\n", (long)tcp_pcbs[i]);
    280 			dotrace(tcp_pcbs[i]);
    281 		}
    282 	}
    283 	exit(0);
    284 }
    285 
    286 void
    287 dotrace(caddr_t tcpcb)
    288 {
    289 	struct tcp_debug *td;
    290 	int prev_debx = tcp_debx;
    291 	int i;
    292 
    293  again:
    294 	if (--tcp_debx < 0)
    295 		tcp_debx = TCP_NDEBUG - 1;
    296 	for (i = prev_debx % TCP_NDEBUG; i < TCP_NDEBUG; i++) {
    297 		td = &tcp_debug[i];
    298 		if (tcpcb && td->td_tcb != tcpcb)
    299 			continue;
    300 		ntime = ntohl(td->td_time);
    301 		switch (td->td_family) {
    302 		case AF_INET:
    303 			tcp_trace(td->td_act, td->td_ostate,
    304 			    (struct tcpcb *)td->td_tcb, &td->td_cb,
    305 			    td->td_family, &td->td_ti, td->td_req);
    306 			break;
    307 #ifdef INET6
    308 		case AF_INET6:
    309 			tcp_trace(td->td_act, td->td_ostate,
    310 			    (struct tcpcb *)td->td_tcb, &td->td_cb,
    311 			    td->td_family, &td->td_ti6, td->td_req);
    312 			break;
    313 #endif
    314 		default:
    315 			tcp_trace(td->td_act, td->td_ostate,
    316 			    (struct tcpcb *)td->td_tcb, &td->td_cb,
    317 			    td->td_family, NULL, td->td_req);
    318 			break;
    319 		}
    320 		if (i == tcp_debx)
    321 			goto done;
    322 	}
    323 	for (i = 0; i <= tcp_debx % TCP_NDEBUG; i++) {
    324 		td = &tcp_debug[i];
    325 		if (tcpcb && td->td_tcb != tcpcb)
    326 			continue;
    327 		ntime = ntohl(td->td_time);
    328 		switch (td->td_family) {
    329 		case AF_INET:
    330 			tcp_trace(td->td_act, td->td_ostate,
    331 			    (struct tcpcb *)td->td_tcb, &td->td_cb,
    332 			    td->td_family, &td->td_ti, td->td_req);
    333 			break;
    334 #ifdef INET6
    335 		case AF_INET6:
    336 			tcp_trace(td->td_act, td->td_ostate,
    337 			    (struct tcpcb *)td->td_tcb, &td->td_cb,
    338 			    td->td_family, &td->td_ti6, td->td_req);
    339 			break;
    340 #endif
    341 		default:
    342 			tcp_trace(td->td_act, td->td_ostate,
    343 			    (struct tcpcb *)td->td_tcb, &td->td_cb,
    344 			    td->td_family, NULL, td->td_req);
    345 			break;
    346 		}
    347 	}
    348  done:
    349 	if (follow) {
    350 		prev_debx = tcp_debx + 1;
    351 		if (prev_debx >= TCP_NDEBUG)
    352 			prev_debx = 0;
    353 		do {
    354 			sleep(1);
    355 			if (use_sysctl) {
    356 				size_t len = sizeof(tcp_debx);
    357 
    358 				if (sysctlbyname("net.inet.tcp.debx",
    359 				    &tcp_debx, &len, NULL, 0) == -1)
    360 					err(1, "net.inet.tcp.debx");
    361 			} else
    362 				if (kvm_read(kd, nl[N_TCP_DEBX].n_value,
    363 				    (char *)&tcp_debx, sizeof(tcp_debx)) !=
    364 				    sizeof(tcp_debx))
    365 					errx(3, "tcp_debx: %s",
    366 					    kvm_geterr(kd));
    367 		} while (tcp_debx == prev_debx);
    368 
    369 		if (use_sysctl) {
    370 			size_t len = sizeof(tcp_debug);
    371 
    372 			if (sysctlbyname("net.inet.tcp.debug", &tcp_debug,
    373 			    &len, NULL, 0) == -1)
    374 				err(1, "net.inet.tcp.debug");
    375 		} else
    376 			if (kvm_read(kd, nl[N_TCP_DEBUG].n_value,
    377 			    (char *)tcp_debug,
    378 			    sizeof(tcp_debug)) != sizeof(tcp_debug))
    379 				errx(3, "tcp_debug: %s", kvm_geterr(kd));
    380 
    381 		goto again;
    382 	}
    383 }
    384 
    385 /*
    386  * Tcp debug routines
    387  */
    388 /*ARGSUSED*/
    389 void
    390 tcp_trace(short act, short ostate, struct tcpcb *atp, struct tcpcb *tp,
    391     int family, void *packet, int req)
    392 {
    393 	tcp_seq seq, ack;
    394 	int flags, len, win, timer;
    395 	struct tcphdr *th = NULL;
    396 	struct ip *ip = NULL;
    397 #ifdef INET6
    398 	struct ip6_hdr *ip6 = NULL;
    399 #endif
    400 	char hbuf[MAXHOSTNAMELEN];
    401 
    402 	len = 0;	/* XXXGCC -Wuninitialized */
    403 
    404 	switch (family) {
    405 	case AF_INET:
    406 		if (packet) {
    407 			ip = (struct ip *)packet;
    408 			th = (struct tcphdr *)(ip + 1);
    409 		}
    410 		break;
    411 #ifdef INET6
    412 	case AF_INET6:
    413 		if (packet) {
    414 			ip6 = (struct ip6_hdr *)packet;
    415 			th = (struct tcphdr *)(ip6 + 1);
    416 		}
    417 		break;
    418 #endif
    419 	default:
    420 		return;
    421 	}
    422 
    423 	printf("%03d %s:%s ", (ntime/10) % 1000, tcpstates[ostate],
    424 	    tanames[act]);
    425 
    426 #ifndef INET6
    427 	if (!ip)
    428 #else
    429 	if (!(ip || ip6))
    430 #endif
    431 		goto skipact;
    432 
    433 	switch (act) {
    434 	case TA_INPUT:
    435 	case TA_OUTPUT:
    436 	case TA_DROP:
    437 		if (aflag) {
    438 			inet_ntop(family,
    439 #ifndef INET6
    440 				(void *)&ip->ip_src,
    441 #else
    442 				family == AF_INET ? (void *)&ip->ip_src
    443 						  : (void *)&ip6->ip6_src,
    444 #endif
    445 				hbuf, sizeof(hbuf));
    446 			printf("(src=%s,%u, ",
    447 			    hbuf, ntohs(th->th_sport));
    448 			inet_ntop(family,
    449 #ifndef INET6
    450 				(void *)&ip->ip_dst,
    451 #else
    452 				family == AF_INET ? (void *)&ip->ip_dst
    453 						  : (void *)&ip6->ip6_dst,
    454 #endif
    455 				hbuf, sizeof(hbuf));
    456 			printf("dst=%s,%u)",
    457 			    hbuf, ntohs(th->th_dport));
    458 		}
    459 		seq = th->th_seq;
    460 		ack = th->th_ack;
    461 		if (ip)
    462 			len = ip->ip_len;
    463 #ifdef INET6
    464 		else if (ip6)
    465 			len = ip6->ip6_plen;
    466 #endif
    467 		win = th->th_win;
    468 		if (act == TA_OUTPUT) {
    469 			NTOHL(seq);
    470 			NTOHL(ack);
    471 			NTOHS(len);
    472 			NTOHS(win);
    473 		}
    474 		if (act == TA_OUTPUT)
    475 			len -= sizeof(struct tcphdr);
    476 		if (len)
    477 			printf("[%x..%x)", seq, seq + len);
    478 		else
    479 			printf("%x", seq);
    480 		printf("@%x", ack);
    481 		if (win)
    482 			printf("(win=%x)", win);
    483 		flags = th->th_flags;
    484 		if (flags) {
    485 			char *cp = "<";
    486 #define	pf(flag, string) { \
    487 	if (th->th_flags&flag) { \
    488 		(void)printf("%s%s", cp, string); \
    489 		cp = ","; \
    490 	} \
    491 }
    492 			pf(TH_SYN, "SYN");
    493 			pf(TH_ACK, "ACK");
    494 			pf(TH_FIN, "FIN");
    495 			pf(TH_RST, "RST");
    496 			pf(TH_PUSH, "PUSH");
    497 			pf(TH_URG, "URG");
    498 			printf(">");
    499 		}
    500 		break;
    501 	case TA_USER:
    502 		timer = req >> 8;
    503 		req &= 0xff;
    504 		printf("%s", prurequests[req]);
    505 		if (req == PRU_SLOWTIMO || req == PRU_FASTTIMO)
    506 			printf("<%s>", tcptimers[timer]);
    507 		break;
    508 	}
    509 
    510 skipact:
    511 	printf(" -> %s", tcpstates[tp->t_state]);
    512 	/* print out internal state of tp !?! */
    513 	printf("\n");
    514 	if (sflag) {
    515 		printf("\trcv_nxt %x rcv_wnd %lx snd_una %x snd_nxt %x snd_max %x\n",
    516 		    tp->rcv_nxt, tp->rcv_wnd, tp->snd_una, tp->snd_nxt,
    517 		    tp->snd_max);
    518 		printf("\tsnd_wl1 %x snd_wl2 %x snd_wnd %lx\n", tp->snd_wl1,
    519 		    tp->snd_wl2, tp->snd_wnd);
    520 	}
    521 	/* print out timers? */
    522 	if (tflag) {
    523 		char *cp = "\t";
    524 		int i;
    525 		int hardticks;
    526 
    527 		if (use_sysctl) {
    528 			size_t len = sizeof(hardticks);
    529 
    530 			if (sysctlbyname("kern.hardclock_ticks", &hardticks,
    531 			    &len, NULL, 0) == -1)
    532 				err(1, "kern.hardclock_ticks");
    533 		} else {
    534 			if (kvm_read(kd, nl[N_HARDCLOCK_TICKS].n_value,
    535 			    (char *)&hardticks,
    536 			    sizeof(hardticks)) != sizeof(hardticks))
    537 				errx(3, "hardclock_ticks: %s", kvm_geterr(kd));
    538 
    539 			for (i = 0; i < TCPT_NTIMERS; i++) {
    540 				if ((tp->t_timer[i].c_flags
    541 				    & CALLOUT_PENDING) == 0)
    542 					continue;
    543 				printf("%s%s=%d", cp, tcptimers[i],
    544 				    tp->t_timer[i].c_time - hardticks);
    545 				if (i == TCPT_REXMT)
    546 					printf(" (t_rxtshft=%d)",
    547 					    tp->t_rxtshift);
    548 				cp = ", ";
    549 			}
    550 			if (*cp != '\t')
    551 				putchar('\n');
    552 		}
    553 	}
    554 }
    555 
    556 int
    557 numeric(const void *v1, const void *v2)
    558 {
    559 	const caddr_t *c1 = v1;
    560 	const caddr_t *c2 = v2;
    561 	int rv;
    562 
    563 	if (*c1 < *c2)
    564 		rv = -1;
    565 	else if (*c1 > *c2)
    566 		rv = 1;
    567 	else
    568 		rv = 0;
    569 
    570 	return (rv);
    571 }
    572 
    573 void
    574 usage(void)
    575 {
    576 
    577 	(void) fprintf(stderr, "usage: %s [-afjst] [-p hex-address]"
    578 	    " [-N system] [-M core]\n", getprogname());
    579 	exit(1);
    580 }
    581