1 /*-
2 * SPDX-License-Identifier: BSD-3-Clause
3 *
4 * Copyright (c) 1983, 1988, 1993
5 * Regents of the University of California. All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 * 3. Neither the name of the University nor the names of its contributors
16 * may be used to endorse or promote products derived from this software
17 * without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 * SUCH DAMAGE.
30 */
31
32 #ifndef lint
33 static char const copyright[] =
34 "@(#) Copyright (c) 1983, 1988, 1993\n\
35 Regents of the University of California. All rights reserved.\n";
36 #endif /* not lint */
37
38 #if 0
39 #ifndef lint
40 static char sccsid[] = "@(#)main.c 8.4 (Berkeley) 3/1/94";
41 #endif /* not lint */
42 #endif
43
44 #include <sys/cdefs.h>
45 __FBSDID("$FreeBSD$");
46
47 #include <sys/param.h>
48 #include <sys/file.h>
49 #include <sys/protosw.h>
50 #include <sys/socket.h>
51 #include <sys/socketvar.h>
52 #include <sys/sysctl.h>
53
54 #include <netinet/in.h>
55
56 #ifdef NETGRAPH
57 #include <netgraph/ng_socket.h>
58 #endif
59
60 #include <ctype.h>
61 #include <err.h>
62 #include <errno.h>
63 #include <kvm.h>
64 #include <limits.h>
65 #include <netdb.h>
66 #include <nlist.h>
67 #include <paths.h>
68 #include <stdint.h>
69 #include <stdio.h>
70 #include <stdlib.h>
71 #include <stdbool.h>
72 #include <string.h>
73 #include <unistd.h>
74 #include "netstat.h"
75 #include "nl_defs.h"
76 #include <libxo/xo.h>
77
78 static struct protox {
79 int pr_index; /* index into nlist of cb head */
80 int pr_sindex; /* index into nlist of stat block */
81 u_char pr_wanted; /* 1 if wanted, 0 otherwise */
82 void (*pr_cblocks)(u_long, const char *, int, int);
83 /* control blocks printing routine */
84 void (*pr_stats)(u_long, const char *, int, int);
85 /* statistics printing routine */
86 void (*pr_istats)(char *); /* per/if statistics printing routine */
87 const char *pr_name; /* well-known name */
88 int pr_usesysctl; /* non-zero if we use sysctl, not kvm */
89 int pr_protocol;
90 } protox[] = {
91 { N_TCBINFO, N_TCPSTAT, 1, protopr,
92 tcp_stats, NULL, "tcp", 1, IPPROTO_TCP },
93 { N_UDBINFO, N_UDPSTAT, 1, protopr,
94 udp_stats, NULL, "udp", 1, IPPROTO_UDP },
95 #ifdef SCTP
96 { -1, N_SCTPSTAT, 1, sctp_protopr,
97 sctp_stats, NULL, "sctp", 1, IPPROTO_SCTP },
98 #endif
99 #ifdef SDP
100 { -1, -1, 1, protopr,
101 NULL, NULL, "sdp", 1, IPPROTO_TCP },
102 #endif
103 { N_DIVCBINFO, -1, 1, protopr,
104 NULL, NULL, "divert", 1, IPPROTO_DIVERT },
105 { N_RIPCBINFO, N_IPSTAT, 1, protopr,
106 ip_stats, NULL, "ip", 1, IPPROTO_RAW },
107 { N_RIPCBINFO, N_ICMPSTAT, 1, protopr,
108 icmp_stats, NULL, "icmp", 1, IPPROTO_ICMP },
109 { N_RIPCBINFO, N_IGMPSTAT, 1, protopr,
110 igmp_stats, NULL, "igmp", 1, IPPROTO_IGMP },
111 #ifdef IPSEC
112 { -1, N_IPSEC4STAT, 1, NULL, /* keep as compat */
113 ipsec_stats, NULL, "ipsec", 1, 0},
114 { -1, N_AHSTAT, 1, NULL,
115 ah_stats, NULL, "ah", 1, 0},
116 { -1, N_ESPSTAT, 1, NULL,
117 esp_stats, NULL, "esp", 1, 0},
118 { -1, N_IPCOMPSTAT, 1, NULL,
119 ipcomp_stats, NULL, "ipcomp", 1, 0},
120 #endif
121 { N_RIPCBINFO, N_PIMSTAT, 1, protopr,
122 pim_stats, NULL, "pim", 1, IPPROTO_PIM },
123 { -1, N_CARPSTATS, 1, NULL,
124 carp_stats, NULL, "carp", 1, 0 },
125 #ifdef PF
126 { -1, N_PFSYNCSTATS, 1, NULL,
127 pfsync_stats, NULL, "pfsync", 1, 0 },
128 #endif
129 { -1, N_ARPSTAT, 1, NULL,
130 arp_stats, NULL, "arp", 1, 0 },
131 { -1, -1, 0, NULL,
132 NULL, NULL, NULL, 0, 0 }
133 };
134
135 #ifdef INET6
136 static struct protox ip6protox[] = {
137 { N_TCBINFO, N_TCPSTAT, 1, protopr,
138 tcp_stats, NULL, "tcp", 1, IPPROTO_TCP },
139 { N_UDBINFO, N_UDPSTAT, 1, protopr,
140 udp_stats, NULL, "udp", 1, IPPROTO_UDP },
141 { N_RIPCBINFO, N_IP6STAT, 1, protopr,
142 ip6_stats, ip6_ifstats, "ip6", 1, IPPROTO_RAW },
143 { N_RIPCBINFO, N_ICMP6STAT, 1, protopr,
144 icmp6_stats, icmp6_ifstats, "icmp6", 1, IPPROTO_ICMPV6 },
145 #ifdef SDP
146 { -1, -1, 1, protopr,
147 NULL, NULL, "sdp", 1, IPPROTO_TCP },
148 #endif
149 #ifdef IPSEC
150 { -1, N_IPSEC6STAT, 1, NULL,
151 ipsec_stats, NULL, "ipsec6", 1, 0 },
152 #endif
153 #ifdef notyet
154 { -1, N_PIM6STAT, 1, NULL,
155 pim6_stats, NULL, "pim6", 1, 0 },
156 #endif
157 { -1, N_RIP6STAT, 1, NULL,
158 rip6_stats, NULL, "rip6", 1, 0 },
159 { -1, -1, 0, NULL,
160 NULL, NULL, NULL, 0, 0 }
161 };
162 #endif /*INET6*/
163
164 #ifdef IPSEC
165 static struct protox pfkeyprotox[] = {
166 { -1, N_PFKEYSTAT, 1, NULL,
167 pfkey_stats, NULL, "pfkey", 0, 0 },
168 { -1, -1, 0, NULL,
169 NULL, NULL, NULL, 0, 0 }
170 };
171 #endif
172
173 #ifdef NETGRAPH
174 static struct protox netgraphprotox[] = {
175 { N_NGSOCKLIST, -1, 1, netgraphprotopr,
176 NULL, NULL, "ctrl", 0, 0 },
177 { N_NGSOCKLIST, -1, 1, netgraphprotopr,
178 NULL, NULL, "data", 0, 0 },
179 { -1, -1, 0, NULL,
180 NULL, NULL, NULL, 0, 0 }
181 };
182 #endif
183
184 static struct protox *protoprotox[] = {
185 protox,
186 #ifdef INET6
187 ip6protox,
188 #endif
189 #ifdef IPSEC
190 pfkeyprotox,
191 #endif
192 NULL };
193
194 static void printproto(struct protox *, const char *, bool *);
195 static void usage(void);
196 static struct protox *name2protox(const char *);
197 static struct protox *knownname(const char *);
198
199 static int kresolve_list(struct nlist *_nl);
200
201 static kvm_t *kvmd;
202 static char *nlistf = NULL, *memf = NULL;
203
204 int Aflag; /* show addresses of protocol control block */
205 int aflag; /* show all sockets (including servers) */
206 static int Bflag; /* show information about bpf consumers */
207 int bflag; /* show i/f total bytes in/out */
208 int cflag; /* show TCP congestion control stack */
209 int Cflag; /* show congestion control algo and vars */
210 int dflag; /* show i/f dropped packets */
211 int gflag; /* show group (multicast) routing or stats */
212 int hflag; /* show counters in human readable format */
213 int iflag; /* show interfaces */
214 int Lflag; /* show size of listen queues */
215 int mflag; /* show memory stats */
216 int noutputs = 0; /* how much outputs before we exit */
217 int numeric_addr; /* show addresses numerically */
218 int numeric_port; /* show ports numerically */
219 int Oflag; /* show nhgrp objects*/
220 int oflag; /* show nexthop objects*/
221 int Pflag; /* show TCP log ID */
222 static int pflag; /* show given protocol */
223 static int Qflag; /* show netisr information */
224 int rflag; /* show routing tables (or routing stats) */
225 int Rflag; /* show flow / RSS statistics */
226 int sflag; /* show protocol statistics */
227 int Wflag; /* wide display */
228 int Tflag; /* TCP Information */
229 int xflag; /* extra information, includes all socket buffer info */
230 int zflag; /* zero stats */
231
232 int interval; /* repeat interval for i/f stats */
233
234 char *interface; /* desired i/f for stats, or NULL for all i/fs */
235 int unit; /* unit number for above */
236
237 static int af; /* address family */
238 int live; /* true if we are examining a live system */
239
240 int
main(int argc,char * argv[])241 main(int argc, char *argv[])
242 {
243 struct protox *tp = NULL; /* for printing cblocks & stats */
244 int ch;
245 int fib = -1;
246 char *endptr;
247 bool first = true;
248
249 af = AF_UNSPEC;
250
251 argc = xo_parse_args(argc, argv);
252 if (argc < 0)
253 exit(EXIT_FAILURE);
254
255 while ((ch = getopt(argc, argv, "46AaBbCcdF:f:ghI:iLlM:mN:nOoPp:Qq:RrSTsuWw:xz"))
256 != -1)
257 switch(ch) {
258 case '4':
259 #ifdef INET
260 af = AF_INET;
261 #else
262 errx(1, "IPv4 support is not compiled in");
263 #endif
264 break;
265 case '6':
266 #ifdef INET6
267 af = AF_INET6;
268 #else
269 errx(1, "IPv6 support is not compiled in");
270 #endif
271 break;
272 case 'A':
273 Aflag = 1;
274 break;
275 case 'a':
276 aflag = 1;
277 break;
278 case 'B':
279 Bflag = 1;
280 break;
281 case 'b':
282 bflag = 1;
283 break;
284 case 'c':
285 cflag = 1;
286 break;
287 case 'C':
288 Cflag = 1;
289 break;
290 case 'd':
291 dflag = 1;
292 break;
293 case 'F':
294 fib = strtol(optarg, &endptr, 0);
295 if (*endptr != '\0' ||
296 (fib == 0 && (errno == EINVAL || errno == ERANGE)))
297 xo_errx(1, "%s: invalid fib", optarg);
298 break;
299 case 'f':
300 if (strcmp(optarg, "inet") == 0)
301 af = AF_INET;
302 #ifdef INET6
303 else if (strcmp(optarg, "inet6") == 0)
304 af = AF_INET6;
305 #endif
306 #ifdef IPSEC
307 else if (strcmp(optarg, "pfkey") == 0)
308 af = PF_KEY;
309 #endif
310 else if (strcmp(optarg, "unix") == 0 ||
311 strcmp(optarg, "local") == 0)
312 af = AF_UNIX;
313 #ifdef NETGRAPH
314 else if (strcmp(optarg, "ng") == 0
315 || strcmp(optarg, "netgraph") == 0)
316 af = AF_NETGRAPH;
317 #endif
318 else if (strcmp(optarg, "link") == 0)
319 af = AF_LINK;
320 else {
321 xo_errx(1, "%s: unknown address family",
322 optarg);
323 }
324 break;
325 case 'g':
326 gflag = 1;
327 break;
328 case 'h':
329 hflag = 1;
330 break;
331 case 'I': {
332 char *cp;
333
334 iflag = 1;
335 for (cp = interface = optarg; isalpha(*cp); cp++)
336 continue;
337 unit = atoi(cp);
338 break;
339 }
340 case 'i':
341 iflag = 1;
342 break;
343 case 'L':
344 Lflag = 1;
345 break;
346 case 'M':
347 memf = optarg;
348 break;
349 case 'm':
350 mflag = 1;
351 break;
352 case 'N':
353 nlistf = optarg;
354 break;
355 case 'n':
356 numeric_addr = numeric_port = 1;
357 break;
358 case 'o':
359 oflag = 1;
360 break;
361 case 'O':
362 Oflag = 1;
363 break;
364 case 'P':
365 Pflag = 1;
366 break;
367 case 'p':
368 if ((tp = name2protox(optarg)) == NULL) {
369 xo_errx(1, "%s: unknown or uninstrumented "
370 "protocol", optarg);
371 }
372 pflag = 1;
373 break;
374 case 'Q':
375 Qflag = 1;
376 break;
377 case 'q':
378 noutputs = atoi(optarg);
379 if (noutputs != 0)
380 noutputs++;
381 break;
382 case 'r':
383 rflag = 1;
384 break;
385 case 'R':
386 Rflag = 1;
387 break;
388 case 's':
389 ++sflag;
390 break;
391 case 'S':
392 numeric_addr = 1;
393 break;
394 case 'u':
395 af = AF_UNIX;
396 break;
397 case 'W':
398 case 'l':
399 Wflag = 1;
400 break;
401 case 'w':
402 interval = atoi(optarg);
403 iflag = 1;
404 break;
405 case 'T':
406 Tflag = 1;
407 break;
408 case 'x':
409 xflag = 1;
410 break;
411 case 'z':
412 zflag = 1;
413 break;
414 case '?':
415 default:
416 usage();
417 }
418 argv += optind;
419 argc -= optind;
420
421 #define BACKWARD_COMPATIBILITY
422 #ifdef BACKWARD_COMPATIBILITY
423 if (*argv) {
424 if (isdigit(**argv)) {
425 interval = atoi(*argv);
426 if (interval <= 0)
427 usage();
428 ++argv;
429 iflag = 1;
430 }
431 if (*argv) {
432 nlistf = *argv;
433 if (*++argv)
434 memf = *argv;
435 }
436 }
437 #endif
438
439 /*
440 * Discard setgid privileges if not the running kernel so that bad
441 * guys can't print interesting stuff from kernel memory.
442 */
443 live = (nlistf == NULL && memf == NULL);
444 if (!live) {
445 if (setgid(getgid()) != 0)
446 xo_err(-1, "setgid");
447 /* Load all necessary kvm symbols */
448 kresolve_list(nl);
449 }
450
451 if (xflag && Tflag)
452 xo_errx(1, "-x and -T are incompatible, pick one.");
453
454 if (Bflag) {
455 if (!live)
456 usage();
457 bpf_stats(interface);
458 xo_finish();
459 exit(0);
460 }
461 if (mflag) {
462 if (!live) {
463 if (kread(0, NULL, 0) == 0)
464 mbpr(kvmd, nl[N_SFSTAT].n_value);
465 } else
466 mbpr(NULL, 0);
467 xo_finish();
468 exit(0);
469 }
470 if (Qflag) {
471 if (!live) {
472 if (kread(0, NULL, 0) == 0)
473 netisr_stats();
474 } else
475 netisr_stats();
476 xo_finish();
477 exit(0);
478 }
479 #if 0
480 /*
481 * Keep file descriptors open to avoid overhead
482 * of open/close on each call to get* routines.
483 */
484 sethostent(1);
485 setnetent(1);
486 #else
487 /*
488 * This does not make sense any more with DNS being default over
489 * the files. Doing a setXXXXent(1) causes a tcp connection to be
490 * used for the queries, which is slower.
491 */
492 #endif
493 if (iflag && !sflag) {
494 xo_open_container("statistics");
495 intpr(NULL, af);
496 xo_close_container("statistics");
497 xo_finish();
498 exit(0);
499 }
500 if (rflag) {
501 xo_open_container("statistics");
502 if (sflag) {
503 if (live) {
504 kresolve_list(nl);
505 }
506 rt_stats();
507 } else
508 routepr(fib, af);
509 xo_close_container("statistics");
510 xo_finish();
511 exit(0);
512 }
513 if (oflag) {
514 xo_open_container("statistics");
515 nhops_print(fib, af);
516 xo_close_container("statistics");
517 xo_finish();
518 exit(0);
519 }
520 if (Oflag) {
521 xo_open_container("statistics");
522 nhgrp_print(fib, af);
523 xo_close_container("statistics");
524 xo_finish();
525 exit(0);
526 }
527
528
529
530 if (gflag) {
531 xo_open_container("statistics");
532 if (sflag) {
533 if (af == AF_INET || af == AF_UNSPEC)
534 mrt_stats();
535 #ifdef INET6
536 if (af == AF_INET6 || af == AF_UNSPEC)
537 mrt6_stats();
538 #endif
539 } else {
540 if (af == AF_INET || af == AF_UNSPEC)
541 mroutepr();
542 #ifdef INET6
543 if (af == AF_INET6 || af == AF_UNSPEC)
544 mroute6pr();
545 #endif
546 }
547 xo_close_container("statistics");
548 xo_finish();
549 exit(0);
550 }
551
552 if (tp) {
553 xo_open_container("statistics");
554 printproto(tp, tp->pr_name, &first);
555 if (!first)
556 xo_close_list("socket");
557 xo_close_container("statistics");
558 xo_finish();
559 exit(0);
560 }
561
562 xo_open_container("statistics");
563 if (af == AF_INET || af == AF_UNSPEC)
564 for (tp = protox; tp->pr_name; tp++)
565 printproto(tp, tp->pr_name, &first);
566 #ifdef INET6
567 if (af == AF_INET6 || af == AF_UNSPEC)
568 for (tp = ip6protox; tp->pr_name; tp++)
569 printproto(tp, tp->pr_name, &first);
570 #endif /*INET6*/
571 #ifdef IPSEC
572 if (af == PF_KEY || af == AF_UNSPEC)
573 for (tp = pfkeyprotox; tp->pr_name; tp++)
574 printproto(tp, tp->pr_name, &first);
575 #endif /*IPSEC*/
576 #ifdef NETGRAPH
577 if (af == AF_NETGRAPH || af == AF_UNSPEC)
578 for (tp = netgraphprotox; tp->pr_name; tp++)
579 printproto(tp, tp->pr_name, &first);
580 #endif /* NETGRAPH */
581 if ((af == AF_UNIX || af == AF_UNSPEC) && !sflag)
582 unixpr(nl[N_UNP_COUNT].n_value, nl[N_UNP_GENCNT].n_value,
583 nl[N_UNP_DHEAD].n_value, nl[N_UNP_SHEAD].n_value,
584 nl[N_UNP_SPHEAD].n_value, &first);
585
586 if (!first)
587 xo_close_list("socket");
588 xo_close_container("statistics");
589 xo_finish();
590 exit(0);
591 }
592
593 static int
fetch_stats_internal(const char * sysctlname,u_long off,void * stats,size_t len,kreadfn_t kreadfn,int zero)594 fetch_stats_internal(const char *sysctlname, u_long off, void *stats,
595 size_t len, kreadfn_t kreadfn, int zero)
596 {
597 int error;
598
599 if (live) {
600 memset(stats, 0, len);
601 if (zero)
602 error = sysctlbyname(sysctlname, NULL, NULL, stats,
603 len);
604 else
605 error = sysctlbyname(sysctlname, stats, &len, NULL, 0);
606 if (error == -1 && errno != ENOENT)
607 xo_warn("sysctl %s", sysctlname);
608 } else {
609 if (off == 0)
610 return (1);
611 error = kreadfn(off, stats, len);
612 }
613 return (error);
614 }
615
616 int
fetch_stats(const char * sysctlname,u_long off,void * stats,size_t len,kreadfn_t kreadfn)617 fetch_stats(const char *sysctlname, u_long off, void *stats,
618 size_t len, kreadfn_t kreadfn)
619 {
620
621 return (fetch_stats_internal(sysctlname, off, stats, len, kreadfn,
622 zflag));
623 }
624
625 int
fetch_stats_ro(const char * sysctlname,u_long off,void * stats,size_t len,kreadfn_t kreadfn)626 fetch_stats_ro(const char *sysctlname, u_long off, void *stats,
627 size_t len, kreadfn_t kreadfn)
628 {
629
630 return (fetch_stats_internal(sysctlname, off, stats, len, kreadfn, 0));
631 }
632
633 /*
634 * Print out protocol statistics or control blocks (per sflag).
635 * If the interface was not specifically requested, and the symbol
636 * is not in the namelist, ignore this one.
637 */
638 static void
printproto(struct protox * tp,const char * name,bool * first)639 printproto(struct protox *tp, const char *name, bool *first)
640 {
641 void (*pr)(u_long, const char *, int, int);
642 u_long off;
643 bool doingdblocks = false;
644
645 if (sflag) {
646 if (iflag) {
647 if (tp->pr_istats)
648 intpr(tp->pr_istats, af);
649 else if (pflag)
650 xo_message("%s: no per-interface stats routine",
651 tp->pr_name);
652 return;
653 } else {
654 pr = tp->pr_stats;
655 if (!pr) {
656 if (pflag)
657 xo_message("%s: no stats routine",
658 tp->pr_name);
659 return;
660 }
661 if (tp->pr_usesysctl && live)
662 off = 0;
663 else if (tp->pr_sindex < 0) {
664 if (pflag)
665 xo_message("%s: stats routine doesn't "
666 "work on cores", tp->pr_name);
667 return;
668 } else
669 off = nl[tp->pr_sindex].n_value;
670 }
671 } else {
672 doingdblocks = true;
673 pr = tp->pr_cblocks;
674 if (!pr) {
675 if (pflag)
676 xo_message("%s: no PCB routine", tp->pr_name);
677 return;
678 }
679 if (tp->pr_usesysctl && live)
680 off = 0;
681 else if (tp->pr_index < 0) {
682 if (pflag)
683 xo_message("%s: PCB routine doesn't work on "
684 "cores", tp->pr_name);
685 return;
686 } else
687 off = nl[tp->pr_index].n_value;
688 }
689 if (pr != NULL && (off || (live && tp->pr_usesysctl) ||
690 af != AF_UNSPEC)) {
691 if (doingdblocks && *first) {
692 xo_open_list("socket");
693 *first = false;
694 }
695
696 (*pr)(off, name, af, tp->pr_protocol);
697 }
698 }
699
700 static int
kvmd_init(void)701 kvmd_init(void)
702 {
703 char errbuf[_POSIX2_LINE_MAX];
704
705 if (kvmd != NULL)
706 return (0);
707
708 kvmd = kvm_openfiles(nlistf, memf, NULL, O_RDONLY, errbuf);
709 if (setgid(getgid()) != 0)
710 xo_err(-1, "setgid");
711
712 if (kvmd == NULL) {
713 xo_warnx("kvm not available: %s", errbuf);
714 return (-1);
715 }
716
717 return (0);
718 }
719
720 /*
721 * Resolve symbol list, return 0 on success.
722 */
723 static int
kresolve_list(struct nlist * _nl)724 kresolve_list(struct nlist *_nl)
725 {
726
727 if ((kvmd == NULL) && (kvmd_init() != 0))
728 return (-1);
729
730 if (_nl[0].n_type != 0)
731 return (0);
732
733 if (kvm_nlist(kvmd, _nl) < 0) {
734 if (nlistf)
735 xo_errx(1, "%s: kvm_nlist: %s", nlistf,
736 kvm_geterr(kvmd));
737 else
738 xo_errx(1, "kvm_nlist: %s", kvm_geterr(kvmd));
739 }
740
741 return (0);
742 }
743
744 /*
745 * Wrapper of kvm_dpcpu_setcpu().
746 */
747 void
kset_dpcpu(u_int cpuid)748 kset_dpcpu(u_int cpuid)
749 {
750
751 if ((kvmd == NULL) && (kvmd_init() != 0))
752 xo_errx(-1, "%s: kvm is not available", __func__);
753
754 if (kvm_dpcpu_setcpu(kvmd, cpuid) < 0)
755 xo_errx(-1, "%s: kvm_dpcpu_setcpu(%u): %s", __func__,
756 cpuid, kvm_geterr(kvmd));
757 return;
758 }
759
760 /*
761 * Read kernel memory, return 0 on success.
762 */
763 int
kread(u_long addr,void * buf,size_t size)764 kread(u_long addr, void *buf, size_t size)
765 {
766
767 if (kvmd_init() < 0)
768 return (-1);
769
770 if (!buf)
771 return (0);
772 if (kvm_read(kvmd, addr, buf, size) != (ssize_t)size) {
773 xo_warnx("%s", kvm_geterr(kvmd));
774 return (-1);
775 }
776 return (0);
777 }
778
779 /*
780 * Read single counter(9).
781 */
782 uint64_t
kread_counter(u_long addr)783 kread_counter(u_long addr)
784 {
785
786 if (kvmd_init() < 0)
787 return (-1);
788
789 return (kvm_counter_u64_fetch(kvmd, addr));
790 }
791
792 /*
793 * Read an array of N counters in kernel memory into array of N uint64_t's.
794 */
795 int
kread_counters(u_long addr,void * buf,size_t size)796 kread_counters(u_long addr, void *buf, size_t size)
797 {
798 uint64_t *c;
799 u_long *counters;
800 size_t i, n;
801
802 if (kvmd_init() < 0)
803 return (-1);
804
805 if (size % sizeof(uint64_t) != 0) {
806 xo_warnx("kread_counters: invalid counter set size");
807 return (-1);
808 }
809
810 n = size / sizeof(uint64_t);
811 if ((counters = malloc(n * sizeof(u_long))) == NULL)
812 xo_err(-1, "malloc");
813 if (kread(addr, counters, n * sizeof(u_long)) < 0) {
814 free(counters);
815 return (-1);
816 }
817
818 c = buf;
819 for (i = 0; i < n; i++)
820 c[i] = kvm_counter_u64_fetch(kvmd, counters[i]);
821
822 free(counters);
823 return (0);
824 }
825
826 const char *
plural(uintmax_t n)827 plural(uintmax_t n)
828 {
829 return (n != 1 ? "s" : "");
830 }
831
832 const char *
plurales(uintmax_t n)833 plurales(uintmax_t n)
834 {
835 return (n != 1 ? "es" : "");
836 }
837
838 const char *
pluralies(uintmax_t n)839 pluralies(uintmax_t n)
840 {
841 return (n != 1 ? "ies" : "y");
842 }
843
844 /*
845 * Find the protox for the given "well-known" name.
846 */
847 static struct protox *
knownname(const char * name)848 knownname(const char *name)
849 {
850 struct protox **tpp, *tp;
851
852 for (tpp = protoprotox; *tpp; tpp++)
853 for (tp = *tpp; tp->pr_name; tp++)
854 if (strcmp(tp->pr_name, name) == 0)
855 return (tp);
856 return (NULL);
857 }
858
859 /*
860 * Find the protox corresponding to name.
861 */
862 static struct protox *
name2protox(const char * name)863 name2protox(const char *name)
864 {
865 struct protox *tp;
866 char **alias; /* alias from p->aliases */
867 struct protoent *p;
868
869 /*
870 * Try to find the name in the list of "well-known" names. If that
871 * fails, check if name is an alias for an Internet protocol.
872 */
873 if ((tp = knownname(name)) != NULL)
874 return (tp);
875
876 setprotoent(1); /* make protocol lookup cheaper */
877 while ((p = getprotoent()) != NULL) {
878 /* assert: name not same as p->name */
879 for (alias = p->p_aliases; *alias; alias++)
880 if (strcmp(name, *alias) == 0) {
881 endprotoent();
882 return (knownname(p->p_name));
883 }
884 }
885 endprotoent();
886 return (NULL);
887 }
888
889 static void
usage(void)890 usage(void)
891 {
892 (void)xo_error("%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n",
893 "usage: netstat [-46AaCcLnRSTWx] [-f protocol_family | -p protocol]\n"
894 " [-M core] [-N system]",
895 " netstat -i | -I interface [-46abdhnW] [-f address_family]\n"
896 " [-M core] [-N system]",
897 " netstat -w wait [-I interface] [-46d] [-M core] [-N system]\n"
898 " [-q howmany]",
899 " netstat -s [-46sz] [-f protocol_family | -p protocol]\n"
900 " [-M core] [-N system]",
901 " netstat -i | -I interface -s [-46s]\n"
902 " [-f protocol_family | -p protocol] [-M core] [-N system]",
903 " netstat -m [-M core] [-N system]",
904 " netstat -B [-z] [-I interface]",
905 " netstat -r [-46AnW] [-F fibnum] [-f address_family]\n"
906 " [-M core] [-N system]",
907 " netstat -rs [-s] [-M core] [-N system]",
908 " netstat -g [-46W] [-f address_family] [-M core] [-N system]",
909 " netstat -gs [-46s] [-f address_family] [-M core] [-N system]",
910 " netstat -Q");
911 xo_finish();
912 exit(1);
913 }
914