1 /* $NetBSD: trpt.c,v 1.23 2007/07/10 22:23:13 jmcneill Exp $ */ 2 3 /*- 4 * Copyright (c) 1997, 2005, 2006 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.23 2007/07/10 22:23:13 jmcneill Exp $"); 81 #endif 82 #endif /* not lint */ 83 84 #define _CALLOUT_PRIVATE /* for defs in sys/callout.h */ 85 86 #include <sys/param.h> 87 #include <sys/queue.h> 88 #include <sys/socket.h> 89 #include <sys/socketvar.h> 90 #include <sys/sysctl.h> 91 #define PRUREQUESTS 92 #include <sys/protosw.h> 93 #include <sys/file.h> 94 95 #include <net/route.h> 96 #include <net/if.h> 97 98 #include <netinet/in.h> 99 #include <netinet/in_systm.h> 100 #include <netinet/ip.h> 101 #include <netinet/in_pcb.h> 102 #include <netinet/ip_var.h> 103 104 #ifdef INET6 105 #ifndef INET 106 #include <netinet/in.h> 107 #endif 108 #include <netinet/ip6.h> 109 #endif 110 111 #include <netinet/tcp.h> 112 #define TCPSTATES 113 #include <netinet/tcp_fsm.h> 114 #include <netinet/tcp_seq.h> 115 #define TCPTIMERS 116 #include <netinet/tcp_timer.h> 117 #include <netinet/tcp_var.h> 118 #include <netinet/tcpip.h> 119 #define TANAMES 120 #include <netinet/tcp_debug.h> 121 122 #include <arpa/inet.h> 123 124 #include <err.h> 125 #include <stdio.h> 126 #include <errno.h> 127 #include <kvm.h> 128 #include <nlist.h> 129 #include <paths.h> 130 #include <limits.h> 131 #include <stdlib.h> 132 #include <unistd.h> 133 134 struct nlist nl[] = { 135 #define N_HARDCLOCK_TICKS 0 136 { "_hardclock_ticks" }, 137 #define N_TCP_DEBUG 1 138 { "_tcp_debug" }, 139 #define N_TCP_DEBX 2 140 { "_tcp_debx" }, 141 { NULL }, 142 }; 143 144 static caddr_t tcp_pcbs[TCP_NDEBUG]; 145 static n_time ntime; 146 static int aflag, follow, sflag, tflag; 147 148 /* see sys/netinet/tcp_debug.c */ 149 struct tcp_debug tcp_debug[TCP_NDEBUG]; 150 int tcp_debx; 151 152 int main(int, char *[]); 153 void dotrace(caddr_t); 154 void tcp_trace(short, short, struct tcpcb *, struct tcpcb *, 155 int, void *, int); 156 int numeric(const void *, const void *); 157 void usage(void); 158 159 kvm_t *kd; 160 int use_sysctl; 161 162 int 163 main(int argc, char *argv[]) 164 { 165 int ch, i, jflag, npcbs; 166 char *system, *core, *cp, errbuf[_POSIX2_LINE_MAX]; 167 unsigned long l; 168 169 jflag = npcbs = 0; 170 system = core = NULL; 171 172 while ((ch = getopt(argc, argv, "afjp:stN:M:")) != -1) { 173 switch (ch) { 174 case 'a': 175 ++aflag; 176 break; 177 case 'f': 178 ++follow; 179 setlinebuf(stdout); 180 break; 181 case 'j': 182 ++jflag; 183 break; 184 case 'p': 185 if (npcbs >= TCP_NDEBUG) 186 errx(1, "too many pcbs specified"); 187 errno = 0; 188 cp = NULL; 189 l = strtoul(optarg, &cp, 16); 190 tcp_pcbs[npcbs] = (caddr_t)l; 191 if (*optarg == '\0' || *cp != '\0' || errno || 192 (unsigned long)tcp_pcbs[npcbs] != l) 193 errx(1, "invalid address: %s", optarg); 194 npcbs++; 195 break; 196 case 's': 197 ++sflag; 198 break; 199 case 't': 200 ++tflag; 201 break; 202 case 'N': 203 system = optarg; 204 break; 205 case 'M': 206 core = optarg; 207 break; 208 default: 209 usage(); 210 /* NOTREACHED */ 211 } 212 } 213 argc -= optind; 214 argv += optind; 215 216 if (argc) 217 usage(); 218 219 use_sysctl = (system == NULL && core == NULL); 220 221 if (use_sysctl) { 222 size_t lenx = sizeof(tcp_debx); 223 size_t lend = sizeof(tcp_debug); 224 225 if (sysctlbyname("net.inet.tcp.debx", &tcp_debx, &lenx, 226 NULL, 0) == -1) 227 err(1, "net.inet.tcp.debx"); 228 if (sysctlbyname("net.inet.tcp.debug", &tcp_debug, &lend, 229 NULL, 0) == -1) 230 err(1, "net.inet.tcp.debug"); 231 } else { 232 kd = kvm_openfiles(system, core, NULL, O_RDONLY, errbuf); 233 if (kd == NULL) 234 errx(1, "can't open kmem: %s", errbuf); 235 236 if (kvm_nlist(kd, nl)) 237 errx(2, "%s: no namelist", system); 238 239 if (kvm_read(kd, nl[N_TCP_DEBX].n_value, (char *)&tcp_debx, 240 sizeof(tcp_debx)) != sizeof(tcp_debx)) 241 errx(3, "tcp_debx: %s", kvm_geterr(kd)); 242 243 if (kvm_read(kd, nl[N_TCP_DEBUG].n_value, (char *)tcp_debug, 244 sizeof(tcp_debug)) != sizeof(tcp_debug)) 245 errx(3, "tcp_debug: %s", kvm_geterr(kd)); 246 } 247 248 /* 249 * If no control blocks have been specified, figure 250 * out how many distinct one we have and summarize 251 * them in tcp_pcbs for sorting the trace records 252 * below. 253 */ 254 if (npcbs == 0) { 255 for (i = 0; i < TCP_NDEBUG; i++) { 256 struct tcp_debug *td = &tcp_debug[i]; 257 int j; 258 259 if (td->td_tcb == 0) 260 continue; 261 for (j = 0; j < npcbs; j++) 262 if (tcp_pcbs[j] == td->td_tcb) 263 break; 264 if (j >= npcbs) 265 tcp_pcbs[npcbs++] = td->td_tcb; 266 } 267 if (npcbs == 0) 268 exit(0); 269 } 270 qsort(tcp_pcbs, npcbs, sizeof(caddr_t), numeric); 271 if (jflag) { 272 for (i = 0;;) { 273 printf("%lx", (long)tcp_pcbs[i]); 274 if (++i == npcbs) 275 break; 276 fputs(", ", stdout); 277 } 278 putchar('\n'); 279 } else { 280 for (i = 0; i < npcbs; i++) { 281 printf("\n%lx:\n", (long)tcp_pcbs[i]); 282 dotrace(tcp_pcbs[i]); 283 } 284 } 285 exit(0); 286 } 287 288 void 289 dotrace(caddr_t tcpcb) 290 { 291 struct tcp_debug *td; 292 int prev_debx = tcp_debx; 293 int i; 294 295 again: 296 if (--tcp_debx < 0) 297 tcp_debx = TCP_NDEBUG - 1; 298 for (i = prev_debx % TCP_NDEBUG; i < TCP_NDEBUG; i++) { 299 td = &tcp_debug[i]; 300 if (tcpcb && td->td_tcb != tcpcb) 301 continue; 302 ntime = ntohl(td->td_time); 303 switch (td->td_family) { 304 case AF_INET: 305 tcp_trace(td->td_act, td->td_ostate, 306 (struct tcpcb *)td->td_tcb, &td->td_cb, 307 td->td_family, &td->td_ti, td->td_req); 308 break; 309 #ifdef INET6 310 case AF_INET6: 311 tcp_trace(td->td_act, td->td_ostate, 312 (struct tcpcb *)td->td_tcb, &td->td_cb, 313 td->td_family, &td->td_ti6, td->td_req); 314 break; 315 #endif 316 default: 317 tcp_trace(td->td_act, td->td_ostate, 318 (struct tcpcb *)td->td_tcb, &td->td_cb, 319 td->td_family, NULL, td->td_req); 320 break; 321 } 322 if (i == tcp_debx) 323 goto done; 324 } 325 for (i = 0; i <= tcp_debx % TCP_NDEBUG; i++) { 326 td = &tcp_debug[i]; 327 if (tcpcb && td->td_tcb != tcpcb) 328 continue; 329 ntime = ntohl(td->td_time); 330 switch (td->td_family) { 331 case AF_INET: 332 tcp_trace(td->td_act, td->td_ostate, 333 (struct tcpcb *)td->td_tcb, &td->td_cb, 334 td->td_family, &td->td_ti, td->td_req); 335 break; 336 #ifdef INET6 337 case AF_INET6: 338 tcp_trace(td->td_act, td->td_ostate, 339 (struct tcpcb *)td->td_tcb, &td->td_cb, 340 td->td_family, &td->td_ti6, td->td_req); 341 break; 342 #endif 343 default: 344 tcp_trace(td->td_act, td->td_ostate, 345 (struct tcpcb *)td->td_tcb, &td->td_cb, 346 td->td_family, NULL, td->td_req); 347 break; 348 } 349 } 350 done: 351 if (follow) { 352 prev_debx = tcp_debx + 1; 353 if (prev_debx >= TCP_NDEBUG) 354 prev_debx = 0; 355 do { 356 sleep(1); 357 if (use_sysctl) { 358 size_t len = sizeof(tcp_debx); 359 360 if (sysctlbyname("net.inet.tcp.debx", 361 &tcp_debx, &len, NULL, 0) == -1) 362 err(1, "net.inet.tcp.debx"); 363 } else 364 if (kvm_read(kd, nl[N_TCP_DEBX].n_value, 365 (char *)&tcp_debx, sizeof(tcp_debx)) != 366 sizeof(tcp_debx)) 367 errx(3, "tcp_debx: %s", 368 kvm_geterr(kd)); 369 } while (tcp_debx == prev_debx); 370 371 if (use_sysctl) { 372 size_t len = sizeof(tcp_debug); 373 374 if (sysctlbyname("net.inet.tcp.debug", &tcp_debug, 375 &len, NULL, 0) == -1) 376 err(1, "net.inet.tcp.debug"); 377 } else 378 if (kvm_read(kd, nl[N_TCP_DEBUG].n_value, 379 (char *)tcp_debug, 380 sizeof(tcp_debug)) != sizeof(tcp_debug)) 381 errx(3, "tcp_debug: %s", kvm_geterr(kd)); 382 383 goto again; 384 } 385 } 386 387 /* 388 * Tcp debug routines 389 */ 390 /*ARGSUSED*/ 391 void 392 tcp_trace(short act, short ostate, struct tcpcb *atp, struct tcpcb *tp, 393 int family, void *packet, int req) 394 { 395 tcp_seq seq, ack; 396 int flags, len, win, timer; 397 struct tcphdr *th = NULL; 398 struct ip *ip = NULL; 399 #ifdef INET6 400 struct ip6_hdr *ip6 = NULL; 401 #endif 402 callout_impl_t *ci; 403 char hbuf[MAXHOSTNAMELEN]; 404 405 len = 0; /* XXXGCC -Wuninitialized */ 406 407 switch (family) { 408 case AF_INET: 409 if (packet) { 410 ip = (struct ip *)packet; 411 th = (struct tcphdr *)(ip + 1); 412 } 413 break; 414 #ifdef INET6 415 case AF_INET6: 416 if (packet) { 417 ip6 = (struct ip6_hdr *)packet; 418 th = (struct tcphdr *)(ip6 + 1); 419 } 420 break; 421 #endif 422 default: 423 return; 424 } 425 426 printf("%03d %s:%s ", (ntime/10) % 1000, tcpstates[ostate], 427 tanames[act]); 428 429 #ifndef INET6 430 if (!ip) 431 #else 432 if (!(ip || ip6)) 433 #endif 434 goto skipact; 435 436 switch (act) { 437 case TA_INPUT: 438 case TA_OUTPUT: 439 case TA_DROP: 440 if (aflag) { 441 inet_ntop(family, 442 #ifndef INET6 443 (void *)&ip->ip_src, 444 #else 445 family == AF_INET ? (void *)&ip->ip_src 446 : (void *)&ip6->ip6_src, 447 #endif 448 hbuf, sizeof(hbuf)); 449 printf("(src=%s,%u, ", 450 hbuf, ntohs(th->th_sport)); 451 inet_ntop(family, 452 #ifndef INET6 453 (void *)&ip->ip_dst, 454 #else 455 family == AF_INET ? (void *)&ip->ip_dst 456 : (void *)&ip6->ip6_dst, 457 #endif 458 hbuf, sizeof(hbuf)); 459 printf("dst=%s,%u)", 460 hbuf, ntohs(th->th_dport)); 461 } 462 seq = th->th_seq; 463 ack = th->th_ack; 464 if (ip) 465 len = ip->ip_len; 466 #ifdef INET6 467 else if (ip6) 468 len = ip6->ip6_plen; 469 #endif 470 win = th->th_win; 471 if (act == TA_OUTPUT) { 472 NTOHL(seq); 473 NTOHL(ack); 474 NTOHS(len); 475 NTOHS(win); 476 } 477 if (act == TA_OUTPUT) 478 len -= sizeof(struct tcphdr); 479 if (len) 480 printf("[%x..%x)", seq, seq + len); 481 else 482 printf("%x", seq); 483 printf("@%x", ack); 484 if (win) 485 printf("(win=%x)", win); 486 flags = th->th_flags; 487 if (flags) { 488 char *cp = "<"; 489 #define pf(flag, string) { \ 490 if (th->th_flags&flag) { \ 491 (void)printf("%s%s", cp, string); \ 492 cp = ","; \ 493 } \ 494 } 495 pf(TH_SYN, "SYN"); 496 pf(TH_ACK, "ACK"); 497 pf(TH_FIN, "FIN"); 498 pf(TH_RST, "RST"); 499 pf(TH_PUSH, "PUSH"); 500 pf(TH_URG, "URG"); 501 pf(TH_CWR, "CWR"); 502 pf(TH_ECE, "ECE"); 503 printf(">"); 504 } 505 break; 506 case TA_USER: 507 timer = req >> 8; 508 req &= 0xff; 509 printf("%s", prurequests[req]); 510 if (req == PRU_SLOWTIMO || req == PRU_FASTTIMO) 511 printf("<%s>", tcptimers[timer]); 512 break; 513 } 514 515 skipact: 516 printf(" -> %s", tcpstates[tp->t_state]); 517 /* print out internal state of tp !?! */ 518 printf("\n"); 519 if (sflag) { 520 printf("\trcv_nxt %x rcv_wnd %lx snd_una %x snd_nxt %x snd_max %x\n", 521 tp->rcv_nxt, tp->rcv_wnd, tp->snd_una, tp->snd_nxt, 522 tp->snd_max); 523 printf("\tsnd_wl1 %x snd_wl2 %x snd_wnd %lx\n", tp->snd_wl1, 524 tp->snd_wl2, tp->snd_wnd); 525 } 526 /* print out timers? */ 527 if (tflag) { 528 char *cp = "\t"; 529 int i; 530 int hardticks; 531 532 if (use_sysctl) { 533 size_t len = sizeof(hardticks); 534 535 if (sysctlbyname("kern.hardclock_ticks", &hardticks, 536 &len, NULL, 0) == -1) 537 err(1, "kern.hardclock_ticks"); 538 } else { 539 if (kvm_read(kd, nl[N_HARDCLOCK_TICKS].n_value, 540 (char *)&hardticks, 541 sizeof(hardticks)) != sizeof(hardticks)) 542 errx(3, "hardclock_ticks: %s", kvm_geterr(kd)); 543 544 for (i = 0; i < TCPT_NTIMERS; i++) { 545 ci = (callout_impl_t *)&tp->t_timer[i]; 546 if ((ci->c_flags & CALLOUT_PENDING) == 0) 547 continue; 548 printf("%s%s=%d", cp, tcptimers[i], 549 ci->c_time - hardticks); 550 if (i == TCPT_REXMT) 551 printf(" (t_rxtshft=%d)", 552 tp->t_rxtshift); 553 cp = ", "; 554 } 555 if (*cp != '\t') 556 putchar('\n'); 557 } 558 } 559 } 560 561 int 562 numeric(const void *v1, const void *v2) 563 { 564 const caddr_t *c1 = v1; 565 const caddr_t *c2 = v2; 566 int rv; 567 568 if (*c1 < *c2) 569 rv = -1; 570 else if (*c1 > *c2) 571 rv = 1; 572 else 573 rv = 0; 574 575 return (rv); 576 } 577 578 void 579 usage(void) 580 { 581 582 (void) fprintf(stderr, "usage: %s [-afjst] [-p hex-address]" 583 " [-N system] [-M core]\n", getprogname()); 584 exit(1); 585 } 586