xref: /dragonfly/usr.bin/systat/netstat.c (revision 783d47c4) 
1 /*-
2  * Copyright (c) 1980, 1992, 1993
3  *	The Regents of the University of California.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *	This product includes software developed by the University of
16  *	California, Berkeley and its contributors.
17  * 4. Neither the name of the University nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  *
33  * @(#)netstat.c	8.1 (Berkeley) 6/6/93
34  * $FreeBSD: src/usr.bin/systat/netstat.c,v 1.13 1999/08/30 08:18:08 peter Exp $
35  */
36 
37 /*
38  * netstat
39  */
40 #include <sys/param.h>
41 #include <sys/queue.h>
42 #include <sys/socket.h>
43 #include <sys/socketvar.h>
44 #include <sys/protosw.h>
45 #include <sys/sysctl.h>
46 
47 #include <netinet/in.h>
48 #include <arpa/inet.h>
49 #include <net/route.h>
50 #include <netinet/in_systm.h>
51 #include <netinet/ip.h>
52 #ifdef INET6
53 #include <netinet/ip6.h>
54 #endif
55 #include <netinet/in_pcb.h>
56 #include <netinet/ip_icmp.h>
57 #include <netinet/icmp_var.h>
58 #include <netinet/ip_var.h>
59 #include <netinet/tcp.h>
60 #include <netinet/tcpip.h>
61 #include <netinet/tcp_seq.h>
62 #define TCPSTATES
63 #include <netinet/tcp_fsm.h>
64 #include <netinet/tcp_timer.h>
65 #include <netinet/tcp_var.h>
66 #include <netinet/tcp_debug.h>
67 #include <netinet/udp.h>
68 #include <netinet/udp_var.h>
69 
70 #include <err.h>
71 #include <errno.h>
72 #include <netdb.h>
73 #include <stdlib.h>
74 #include <string.h>
75 #include <nlist.h>
76 #include <paths.h>
77 #include "systat.h"
78 #include "extern.h"
79 
80 static void enter(struct inpcb *, struct xsocket *, int, const char *);
81 static char *inetname(struct in_addr);
82 static void inetprint(struct in_addr *, int, const char *);
83 
84 #define	streq(a,b)	(strcmp(a,b)==0)
85 #define	YMAX(w)		((w)->_maxy-1)
86 
87 WINDOW *
88 opennetstat(void)
89 {
90 	sethostent(1);
91 	setnetent(1);
92 	return (subwin(stdscr, LINES-5-1, 0, 5, 0));
93 }
94 
95 struct netinfo {
96 	struct	netinfo *ni_forw, *ni_prev;
97 	short	ni_line;		/* line on screen */
98 	short	ni_seen;		/* 0 when not present in list */
99 	short	ni_flags;
100 #define	NIF_LACHG	0x1		/* local address changed */
101 #define	NIF_FACHG	0x2		/* foreign address changed */
102 	short	ni_state;		/* tcp state */
103 	const char *ni_proto;		/* protocol */
104 	struct	in_addr ni_laddr;	/* local address */
105 	long	ni_lport;		/* local port */
106 	struct	in_addr	ni_faddr;	/* foreign address */
107 	long	ni_fport;		/* foreign port */
108 	long	ni_rcvcc;		/* rcv buffer character count */
109 	long	ni_sndcc;		/* snd buffer character count */
110 };
111 
112 static struct {
113 	struct	netinfo *ni_forw, *ni_prev;
114 } netcb;
115 
116 static	int aflag = 0;
117 static	int nflag = 0;
118 static	int lastrow = 1;
119 
120 void
121 closenetstat(WINDOW *w)
122 {
123 	struct netinfo *p;
124 
125 	endhostent();
126 	endnetent();
127 	p = (struct netinfo *)netcb.ni_forw;
128 	while (p != (struct netinfo *)&netcb) {
129 		if (p->ni_line != -1)
130 			lastrow--;
131 		p->ni_line = -1;
132 		p = p->ni_forw;
133 	}
134         if (w != NULL) {
135 		wclear(w);
136 		wrefresh(w);
137 		delwin(w);
138 	}
139 }
140 
141 int
142 initnetstat(void)
143 {
144 	netcb.ni_forw = netcb.ni_prev = (struct netinfo *)&netcb;
145 	protos = TCP|UDP;
146 	return(1);
147 }
148 
149 static void
150 enter_tcp(void *xig)
151 {
152 	struct xtcpcb *xtcp = (struct xtcpcb *)xig;
153 	struct xsocket *xso;
154 	int state;
155 
156 	if (xtcp->xt_len < sizeof(*xtcp))
157 		return;
158 	xso = &xtcp->xt_socket;
159 	state = xtcp->xt_tp.t_state;
160 	enter(&xtcp->xt_inp, xso, state, "tcp");
161 }
162 
163 static void
164 enter_udp(void *xig)
165 {
166 	struct xinpcb *xinp = (struct xinpcb *)xig;
167 	struct xsocket *xso;
168 
169 	if (xinp->xi_len < sizeof(*xinp))
170 		return;
171 	xso = &xinp->xi_socket;
172 	enter(&xinp->xi_inp, xso, 0, "udp");
173 }
174 
175 static void
176 fetchnetstat_proto(void (*enter_proto)(void *),
177     const char *mibvar)
178 {
179 	char *buf, *buf2;
180 	size_t i, len, elem_len;
181 
182 	if (sysctlbyname(mibvar, 0, &len, 0, 0) < 0) {
183 		if (errno != ENOENT)
184 			warn("sysctl: %s", mibvar);
185 		return;
186 	}
187 	if ((buf = malloc(len)) == NULL) {
188 		warn("malloc %lu bytes", (u_long)len);
189 		return;
190 	}
191 	if (sysctlbyname(mibvar, buf, &len, 0, 0) < 0) {
192 		warn("sysctl: %s", mibvar);
193 		free(buf);
194 		return;
195 	}
196 
197 	/*
198 	 * XXX this is better with a single PCB type
199 	 */
200 	if (len == 0) {
201 		free(buf);
202 		return;
203 	}
204 	if (len < sizeof(size_t)) {
205 		warnx("sysctl: short read");
206 		free(buf);
207 		return;
208 	}
209 	elem_len = *(size_t *)buf;
210 	len /= elem_len;
211 	buf2 = buf;
212 	for (i = 0; i < len; i++, buf2 += elem_len) {
213 		if (*(size_t *)(buf2) != elem_len) {
214 			warn("sysctl: inconsistent PCB len");
215 			free(buf);
216 			return;
217 		}
218 		enter_proto(buf2);
219 	}
220 	free(buf);
221 }
222 
223 void
224 fetchnetstat(void)
225 {
226 	struct netinfo *p;
227 
228 	for (p = netcb.ni_forw; p != (struct netinfo *)&netcb; p = p->ni_forw)
229 		p->ni_seen = 0;
230 	if (protos & TCP)
231 		fetchnetstat_proto(enter_tcp, "net.inet.tcp.pcblist");
232 	if (protos & UDP)
233 		fetchnetstat_proto(enter_udp, "net.inet.udp.pcblist");
234 }
235 
236 static void
237 enter(struct inpcb *inp, struct xsocket *so, int state, const char *proto)
238 {
239 	struct netinfo *p;
240 
241 	if (!aflag && inet_lnaof(inp->inp_laddr) == INADDR_ANY)
242 		return;
243 	if (nhosts && !checkhost(inp))
244 		return;
245 	if (nports && !checkport(inp))
246 		return;
247 	/*
248 	 * pcblist may return non-ipv4 sockets, but at the moment
249 	 * -netstat code doesn't support other than ipv4.
250 	 */
251 	if ((inp->inp_vflag & INP_IPV4) == 0)
252 		return;
253 	/*
254 	 * Only take exact matches, any sockets with
255 	 * previously unbound addresses will be deleted
256 	 * below in the display routine because they
257 	 * will appear as ``not seen'' in the kernel
258 	 * data structures.
259 	 */
260 	for (p = netcb.ni_forw; p != (struct netinfo *)&netcb; p = p->ni_forw) {
261 		if (!streq(proto, p->ni_proto))
262 			continue;
263 		if (p->ni_lport != inp->inp_lport ||
264 		    p->ni_laddr.s_addr != inp->inp_laddr.s_addr)
265 			continue;
266 		if (p->ni_faddr.s_addr == inp->inp_faddr.s_addr &&
267 		    p->ni_fport == inp->inp_fport)
268 			break;
269 	}
270 	if (p == (struct netinfo *)&netcb) {
271 		if ((p = malloc(sizeof(*p))) == NULL) {
272 			error("Out of memory");
273 			return;
274 		}
275 		p->ni_prev = (struct netinfo *)&netcb;
276 		p->ni_forw = netcb.ni_forw;
277 		netcb.ni_forw->ni_prev = p;
278 		netcb.ni_forw = p;
279 		p->ni_line = -1;
280 		p->ni_laddr = inp->inp_laddr;
281 		p->ni_lport = inp->inp_lport;
282 		p->ni_faddr = inp->inp_faddr;
283 		p->ni_fport = inp->inp_fport;
284 		p->ni_proto = proto;
285 		p->ni_flags = NIF_LACHG|NIF_FACHG;
286 	}
287 	p->ni_rcvcc = so->so_rcv.sb_cc;
288 	p->ni_sndcc = so->so_snd.sb_cc;
289 	p->ni_state = state;
290 	p->ni_seen = 1;
291 }
292 
293 /* column locations */
294 #define	LADDR	0
295 #define	FADDR	LADDR+22
296 #define	CPUID	FADDR+22
297 #define	PROTO	CPUID+4
298 #define	RCVCC	PROTO+6
299 #define	SNDCC	RCVCC+7
300 #define	STATE	SNDCC+7
301 
302 
303 void
304 labelnetstat(void)
305 {
306 	wmove(wnd, 0, 0); wclrtobot(wnd);
307 	mvwaddstr(wnd, 0, LADDR, "Local Address");
308 	mvwaddstr(wnd, 0, FADDR, "Foreign Address");
309 	mvwaddstr(wnd, 0, PROTO, "Proto");
310 	mvwaddstr(wnd, 0, RCVCC, "Recv-Q");
311 	mvwaddstr(wnd, 0, SNDCC, "Send-Q");
312 	mvwaddstr(wnd, 0, STATE, "(state)");
313 }
314 
315 void
316 shownetstat(void)
317 {
318 	struct netinfo *p, *q;
319 
320 	/*
321 	 * First, delete any connections that have gone
322 	 * away and adjust the position of connections
323 	 * below to reflect the deleted line.
324 	 */
325 	p = netcb.ni_forw;
326 	while (p != (struct netinfo *)&netcb) {
327 		if (p->ni_line == -1 || p->ni_seen) {
328 			p = p->ni_forw;
329 			continue;
330 		}
331 		wmove(wnd, p->ni_line, 0); wdeleteln(wnd);
332 		q = netcb.ni_forw;
333 		for (; q != (struct netinfo *)&netcb; q = q->ni_forw)
334 			if (q != p && q->ni_line > p->ni_line) {
335 				q->ni_line--;
336 				/* this shouldn't be necessary */
337 				q->ni_flags |= NIF_LACHG|NIF_FACHG;
338 			}
339 		lastrow--;
340 		q = p->ni_forw;
341 		p->ni_prev->ni_forw = p->ni_forw;
342 		p->ni_forw->ni_prev = p->ni_prev;
343 		free(p);
344 		p = q;
345 	}
346 	/*
347 	 * Update existing connections and add new ones.
348 	 */
349 	for (p = netcb.ni_forw; p != (struct netinfo *)&netcb; p = p->ni_forw) {
350 		if (p->ni_line == -1) {
351 			/*
352 			 * Add a new entry if possible.
353 			 */
354 			if (lastrow > YMAX(wnd))
355 				continue;
356 			p->ni_line = lastrow++;
357 			p->ni_flags |= NIF_LACHG|NIF_FACHG;
358 		}
359 		if (p->ni_flags & NIF_LACHG) {
360 			wmove(wnd, p->ni_line, LADDR);
361 			inetprint(&p->ni_laddr, p->ni_lport, p->ni_proto);
362 			p->ni_flags &= ~NIF_LACHG;
363 		}
364 		if (p->ni_flags & NIF_FACHG) {
365 			wmove(wnd, p->ni_line, FADDR);
366 			inetprint(&p->ni_faddr, p->ni_fport, p->ni_proto);
367 			p->ni_flags &= ~NIF_FACHG;
368 		}
369 		mvwaddstr(wnd, p->ni_line, PROTO, p->ni_proto);
370 		mvwprintw(wnd, p->ni_line, RCVCC, "%6d", p->ni_rcvcc);
371 		mvwprintw(wnd, p->ni_line, SNDCC, "%6d", p->ni_sndcc);
372 		if (streq(p->ni_proto, "tcp")) {
373 			if (p->ni_state < 0 || p->ni_state >= TCP_NSTATES)
374 				mvwprintw(wnd, p->ni_line, STATE, "%d",
375 				    p->ni_state);
376 			else
377 				mvwaddstr(wnd, p->ni_line, STATE,
378 				    tcpstates[p->ni_state]);
379 		}
380 		wclrtoeol(wnd);
381 	}
382 	if (lastrow < YMAX(wnd)) {
383 		wmove(wnd, lastrow, 0); wclrtobot(wnd);
384 		wmove(wnd, YMAX(wnd), 0); wdeleteln(wnd);	/* XXX */
385 	}
386 }
387 
388 /*
389  * Pretty print an Internet address (net address + port).
390  * If the nflag was specified, use numbers instead of names.
391  */
392 static void
393 inetprint(struct in_addr *in, int port, const char *proto)
394 {
395 	struct servent *sp = NULL;
396 	char line[80], *cp;
397 
398 	snprintf(line, sizeof(line), "%.*s.", 16, inetname(*in));
399 	cp = strchr(line, '\0');
400 	if (!nflag && port)
401 		sp = getservbyport(port, proto);
402 	if (sp || port == 0)
403 		snprintf(cp, sizeof(line) - (cp - line), "%.8s",
404 		    sp ? sp->s_name : "*");
405 	else
406 		snprintf(cp, sizeof(line) - (cp - line), "%d",
407 		    ntohs((u_short)port));
408 	/* pad to full column to clear any garbage */
409 	cp = strchr(line, '\0');
410 	while (cp - line < 22)
411 		*cp++ = ' ';
412 	line[22] = '\0';
413 	waddstr(wnd, line);
414 }
415 
416 /*
417  * Construct an Internet address representation.
418  * If the nflag has been supplied, give
419  * numeric value, otherwise try for symbolic name.
420  */
421 static char *
422 inetname(struct in_addr in)
423 {
424 	char *cp = NULL;
425 	static char line[50];
426 	struct hostent *hp;
427 	struct netent *np;
428 
429 	if (!nflag && in.s_addr != INADDR_ANY) {
430 		int net = inet_netof(in);
431 		int lna = inet_lnaof(in);
432 
433 		if (lna == INADDR_ANY) {
434 			np = getnetbyaddr(net, AF_INET);
435 			if (np)
436 				cp = np->n_name;
437 		}
438 		if (cp == NULL) {
439 			hp = gethostbyaddr(&in, sizeof (in), AF_INET);
440 			if (hp)
441 				cp = hp->h_name;
442 		}
443 	}
444 	if (in.s_addr == INADDR_ANY)
445 		strcpy(line, "*");
446 	else if (cp)
447 		snprintf(line, sizeof(line), "%s", cp);
448 	else {
449 		in.s_addr = ntohl(in.s_addr);
450 #define C(x)	((x) & 0xff)
451 		snprintf(line, sizeof(line), "%u.%u.%u.%u", C(in.s_addr >> 24),
452 			C(in.s_addr >> 16), C(in.s_addr >> 8), C(in.s_addr));
453 	}
454 	return (line);
455 }
456 
457 int
458 cmdnetstat(const char *cmd, char *args)
459 {
460 	struct netinfo *p;
461 
462 	if (prefix(cmd, "all")) {
463 		aflag = !aflag;
464 		goto fixup;
465 	}
466 	if  (prefix(cmd, "numbers") || prefix(cmd, "names")) {
467 		int new;
468 
469 		new = prefix(cmd, "numbers");
470 		if (new == nflag)
471 			return (1);
472 		p = netcb.ni_forw;
473 		for (; p != (struct netinfo *)&netcb; p = p->ni_forw) {
474 			if (p->ni_line == -1)
475 				continue;
476 			p->ni_flags |= NIF_LACHG|NIF_FACHG;
477 		}
478 		nflag = new;
479 		goto redisplay;
480 	}
481 	if (!netcmd(cmd, args))
482 		return (0);
483 fixup:
484 	fetchnetstat();
485 redisplay:
486 	shownetstat();
487 	refresh();
488 	return (1);
489 }
490