xref: /freebsd/usr.bin/netstat/unix.c (revision f05cddf9)
1 /*-
2  * Copyright (c) 1983, 1988, 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  * 4. Neither the name of the University nor the names of its contributors
14  *    may be used to endorse or promote products derived from this software
15  *    without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27  * SUCH DAMAGE.
28  */
29 
30 #if 0
31 #ifndef lint
32 static char sccsid[] = "@(#)unix.c	8.1 (Berkeley) 6/6/93";
33 #endif /* not lint */
34 #endif
35 
36 #include <sys/cdefs.h>
37 __FBSDID("$FreeBSD$");
38 
39 /*
40  * Display protocol blocks in the unix domain.
41  */
42 #include <sys/param.h>
43 #include <sys/queue.h>
44 #include <sys/protosw.h>
45 #include <sys/socket.h>
46 #include <sys/socketvar.h>
47 #include <sys/mbuf.h>
48 #include <sys/sysctl.h>
49 #include <sys/un.h>
50 #include <sys/unpcb.h>
51 
52 #include <netinet/in.h>
53 
54 #include <errno.h>
55 #include <err.h>
56 #include <stddef.h>
57 #include <stdint.h>
58 #include <stdio.h>
59 #include <stdlib.h>
60 #include <strings.h>
61 #include <kvm.h>
62 #include "netstat.h"
63 
64 static	void unixdomainpr(struct xunpcb *, struct xsocket *);
65 
66 static	const char *const socktype[] =
67     { "#0", "stream", "dgram", "raw", "rdm", "seqpacket" };
68 
69 static int
70 pcblist_sysctl(int type, char **bufp)
71 {
72 	char 	*buf;
73 	size_t	len;
74 	char mibvar[sizeof "net.local.seqpacket.pcblist"];
75 
76 	sprintf(mibvar, "net.local.%s.pcblist", socktype[type]);
77 
78 	len = 0;
79 	if (sysctlbyname(mibvar, 0, &len, 0, 0) < 0) {
80 		if (errno != ENOENT)
81 			warn("sysctl: %s", mibvar);
82 		return (-1);
83 	}
84 	if ((buf = malloc(len)) == 0) {
85 		warnx("malloc %lu bytes", (u_long)len);
86 		return (-2);
87 	}
88 	if (sysctlbyname(mibvar, buf, &len, 0, 0) < 0) {
89 		warn("sysctl: %s", mibvar);
90 		free(buf);
91 		return (-2);
92 	}
93 	*bufp = buf;
94 	return (0);
95 }
96 
97 static int
98 pcblist_kvm(u_long count_off, u_long gencnt_off, u_long head_off, char **bufp)
99 {
100 	struct unp_head head;
101 	struct unpcb *unp, unp_conn;
102 	u_char sun_len;
103 	struct socket so;
104 	struct xunpgen xug;
105 	struct xunpcb xu;
106 	unp_gen_t unp_gencnt;
107 	u_int	unp_count;
108 	char 	*buf, *p;
109 	size_t	len;
110 
111 	if (count_off == 0 || gencnt_off == 0)
112 		return (-2);
113 	if (head_off == 0)
114 		return (-1);
115 	kread(count_off, &unp_count, sizeof(unp_count));
116 	len = 2 * sizeof(xug) + (unp_count + unp_count / 8) * sizeof(xu);
117 	if ((buf = malloc(len)) == 0) {
118 		warnx("malloc %lu bytes", (u_long)len);
119 		return (-2);
120 	}
121 	p = buf;
122 
123 #define	COPYOUT(obj, size) do {						\
124 	if (len < (size)) {						\
125 		warnx("buffer size exceeded");				\
126 		goto fail;						\
127 	}								\
128 	bcopy((obj), p, (size));					\
129 	len -= (size);							\
130 	p += (size);							\
131 } while (0)
132 
133 #define	KREAD(off, buf, len) do {					\
134 	if (kread((uintptr_t)(off), (buf), (len)) != 0)			\
135 		goto fail;						\
136 } while (0)
137 
138 	/* Write out header. */
139 	kread(gencnt_off, &unp_gencnt, sizeof(unp_gencnt));
140 	xug.xug_len = sizeof xug;
141 	xug.xug_count = unp_count;
142 	xug.xug_gen = unp_gencnt;
143 	xug.xug_sogen = 0;
144 	COPYOUT(&xug, sizeof xug);
145 
146 	/* Walk the PCB list. */
147 	xu.xu_len = sizeof xu;
148 	KREAD(head_off, &head, sizeof(head));
149 	LIST_FOREACH(unp, &head, unp_link) {
150 		xu.xu_unpp = unp;
151 		KREAD(unp, &xu.xu_unp, sizeof (*unp));
152 		unp = &xu.xu_unp;
153 
154 		if (unp->unp_gencnt > unp_gencnt)
155 			continue;
156 		if (unp->unp_addr != NULL) {
157 			KREAD(unp->unp_addr, &sun_len, sizeof(sun_len));
158 			KREAD(unp->unp_addr, &xu.xu_addr, sun_len);
159 		}
160 		if (unp->unp_conn != NULL) {
161 			KREAD(unp->unp_conn, &unp_conn, sizeof(unp_conn));
162 			if (unp_conn.unp_addr != NULL) {
163 				KREAD(unp_conn.unp_addr, &sun_len,
164 				    sizeof(sun_len));
165 				KREAD(unp_conn.unp_addr, &xu.xu_caddr, sun_len);
166 			}
167 		}
168 		KREAD(unp->unp_socket, &so, sizeof(so));
169 		if (sotoxsocket(&so, &xu.xu_socket) != 0)
170 			goto fail;
171 		COPYOUT(&xu, sizeof(xu));
172 	}
173 
174 	/* Reread the counts and write the footer. */
175 	kread(count_off, &unp_count, sizeof(unp_count));
176 	kread(gencnt_off, &unp_gencnt, sizeof(unp_gencnt));
177 	xug.xug_count = unp_count;
178 	xug.xug_gen = unp_gencnt;
179 	COPYOUT(&xug, sizeof xug);
180 
181 	*bufp = buf;
182 	return (0);
183 
184 fail:
185 	free(buf);
186 	return (-1);
187 #undef COPYOUT
188 #undef KREAD
189 }
190 
191 void
192 unixpr(u_long count_off, u_long gencnt_off, u_long dhead_off, u_long shead_off,
193     u_long sphead_off)
194 {
195 	char 	*buf;
196 	int	ret, type;
197 	struct	xsocket *so;
198 	struct	xunpgen *xug, *oxug;
199 	struct	xunpcb *xunp;
200 	u_long	head_off;
201 
202 	buf = NULL;
203 	for (type = SOCK_STREAM; type <= SOCK_SEQPACKET; type++) {
204 		if (live)
205 			ret = pcblist_sysctl(type, &buf);
206 		else {
207 			head_off = 0;
208 			switch (type) {
209 			case SOCK_STREAM:
210 				head_off = shead_off;
211 				break;
212 
213 			case SOCK_DGRAM:
214 				head_off = dhead_off;
215 				break;
216 
217 			case SOCK_SEQPACKET:
218 				head_off = sphead_off;
219 				break;
220 			}
221 			ret = pcblist_kvm(count_off, gencnt_off, head_off,
222 			    &buf);
223 		}
224 		if (ret == -1)
225 			continue;
226 		if (ret < 0)
227 			return;
228 
229 		oxug = xug = (struct xunpgen *)buf;
230 		for (xug = (struct xunpgen *)((char *)xug + xug->xug_len);
231 		     xug->xug_len > sizeof(struct xunpgen);
232 		     xug = (struct xunpgen *)((char *)xug + xug->xug_len)) {
233 			xunp = (struct xunpcb *)xug;
234 			so = &xunp->xu_socket;
235 
236 			/* Ignore PCBs which were freed during copyout. */
237 			if (xunp->xu_unp.unp_gencnt > oxug->xug_gen)
238 				continue;
239 			unixdomainpr(xunp, so);
240 		}
241 		if (xug != oxug && xug->xug_gen != oxug->xug_gen) {
242 			if (oxug->xug_count > xug->xug_count) {
243 				printf("Some %s sockets may have been deleted.\n",
244 				       socktype[type]);
245 			} else if (oxug->xug_count < xug->xug_count) {
246 				printf("Some %s sockets may have been created.\n",
247 			       socktype[type]);
248 			} else {
249 				printf("Some %s sockets may have been created or deleted",
250 			       socktype[type]);
251 			}
252 		}
253 		free(buf);
254 	}
255 }
256 
257 static void
258 unixdomainpr(struct xunpcb *xunp, struct xsocket *so)
259 {
260 	struct unpcb *unp;
261 	struct sockaddr_un *sa;
262 	static int first = 1;
263 	char buf1[15];
264 
265 	unp = &xunp->xu_unp;
266 	if (unp->unp_addr)
267 		sa = &xunp->xu_addr;
268 	else
269 		sa = (struct sockaddr_un *)0;
270 
271 	if (first && !Lflag) {
272 		printf("Active UNIX domain sockets\n");
273 		printf(
274 "%-8.8s %-6.6s %-6.6s %-6.6s %8.8s %8.8s %8.8s %8.8s Addr\n",
275 		    "Address", "Type", "Recv-Q", "Send-Q",
276 		    "Inode", "Conn", "Refs", "Nextref");
277 		first = 0;
278 	}
279 
280 	if (Lflag && so->so_qlimit == 0)
281 		return;
282 
283 	if (Lflag) {
284 		snprintf(buf1, 15, "%d/%d/%d", so->so_qlen,
285 		    so->so_incqlen, so->so_qlimit);
286 		printf("unix  %-14.14s", buf1);
287 	} else {
288 		printf("%8lx %-6.6s %6u %6u %8lx %8lx %8lx %8lx",
289 		    (long)so->so_pcb, socktype[so->so_type], so->so_rcv.sb_cc,
290 		    so->so_snd.sb_cc, (long)unp->unp_vnode, (long)unp->unp_conn,
291 		    (long)LIST_FIRST(&unp->unp_refs),
292 		    (long)LIST_NEXT(unp, unp_reflink));
293 	}
294 	if (sa)
295 		printf(" %.*s",
296 		    (int)(sa->sun_len - offsetof(struct sockaddr_un, sun_path)),
297 		    sa->sun_path);
298 	putchar('\n');
299 }
300