xref: /freebsd/usr.bin/netstat/unix.c (revision 2be1a816)
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  * 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 
34 #if 0
35 #ifndef lint
36 static char sccsid[] = "@(#)unix.c	8.1 (Berkeley) 6/6/93";
37 #endif /* not lint */
38 #endif
39 
40 #include <sys/cdefs.h>
41 __FBSDID("$FreeBSD$");
42 
43 /*
44  * Display protocol blocks in the unix domain.
45  */
46 #include <sys/param.h>
47 #include <sys/queue.h>
48 #include <sys/protosw.h>
49 #include <sys/socket.h>
50 #include <sys/socketvar.h>
51 #include <sys/mbuf.h>
52 #include <sys/sysctl.h>
53 #include <sys/un.h>
54 #include <sys/unpcb.h>
55 
56 #include <netinet/in.h>
57 
58 #include <errno.h>
59 #include <err.h>
60 #include <stddef.h>
61 #include <stdint.h>
62 #include <stdio.h>
63 #include <stdlib.h>
64 #include <strings.h>
65 #include <kvm.h>
66 #include "netstat.h"
67 
68 static	void unixdomainpr(struct xunpcb *, struct xsocket *);
69 
70 static	const char *const socktype[] =
71     { "#0", "stream", "dgram", "raw", "rdm", "seqpacket" };
72 
73 static int
74 pcblist_sysctl(int type, char **bufp)
75 {
76 	char 	*buf;
77 	size_t	len;
78 	char mibvar[sizeof "net.local.seqpacket.pcblist"];
79 
80 	sprintf(mibvar, "net.local.%s.pcblist", socktype[type]);
81 
82 	len = 0;
83 	if (sysctlbyname(mibvar, 0, &len, 0, 0) < 0) {
84 		if (errno != ENOENT)
85 			warn("sysctl: %s", mibvar);
86 		return (-1);
87 	}
88 	if ((buf = malloc(len)) == 0) {
89 		warnx("malloc %lu bytes", (u_long)len);
90 		return (-2);
91 	}
92 	if (sysctlbyname(mibvar, buf, &len, 0, 0) < 0) {
93 		warn("sysctl: %s", mibvar);
94 		free(buf);
95 		return (-2);
96 	}
97 	*bufp = buf;
98 	return (0);
99 }
100 
101 static int
102 pcblist_kvm(u_long count_off, u_long gencnt_off, u_long head_off, char **bufp)
103 {
104 	struct unp_head head;
105 	struct unpcb *unp, unp_conn;
106 	u_char sun_len;
107 	struct socket so;
108 	struct xunpgen xug;
109 	struct xunpcb xu;
110 	unp_gen_t unp_gencnt;
111 	u_int	unp_count;
112 	char 	*buf, *p;
113 	size_t	len;
114 
115 	if (count_off == 0 || gencnt_off == 0)
116 		return (-2);
117 	if (head_off == 0)
118 		return (-1);
119 	kread(count_off, &unp_count, sizeof(unp_count));
120 	len = 2 * sizeof(xug) + (unp_count + unp_count / 8) * sizeof(xu);
121 	if ((buf = malloc(len)) == 0) {
122 		warnx("malloc %lu bytes", (u_long)len);
123 		return (-2);
124 	}
125 	p = buf;
126 
127 #define	COPYOUT(obj, size) do {						\
128 	if (len < (size)) {						\
129 		warnx("buffer size exceeded");				\
130 		goto fail;						\
131 	}								\
132 	bcopy((obj), p, (size));					\
133 	len -= (size);							\
134 	p += (size);							\
135 } while (0)
136 
137 #define	KREAD(off, buf, len) do {					\
138 	if (kread((uintptr_t)(off), (buf), (len)) != 0)			\
139 		goto fail;						\
140 } while (0)
141 
142 	/* Write out header. */
143 	kread(gencnt_off, &unp_gencnt, sizeof(unp_gencnt));
144 	xug.xug_len = sizeof xug;
145 	xug.xug_count = unp_count;
146 	xug.xug_gen = unp_gencnt;
147 	xug.xug_sogen = 0;
148 	COPYOUT(&xug, sizeof xug);
149 
150 	/* Walk the PCB list. */
151 	xu.xu_len = sizeof xu;
152 	KREAD(head_off, &head, sizeof(head));
153 	LIST_FOREACH(unp, &head, unp_link) {
154 		xu.xu_unpp = unp;
155 		KREAD(unp, &xu.xu_unp, sizeof (*unp));
156 		unp = &xu.xu_unp;
157 
158 		if (unp->unp_gencnt > unp_gencnt)
159 			continue;
160 		if (unp->unp_addr != NULL) {
161 			KREAD(unp->unp_addr, &sun_len, sizeof(sun_len));
162 			KREAD(unp->unp_addr, &xu.xu_addr, sun_len);
163 		}
164 		if (unp->unp_conn != NULL) {
165 			KREAD(unp->unp_conn, &unp_conn, sizeof(unp_conn));
166 			if (unp_conn.unp_addr != NULL) {
167 				KREAD(unp_conn.unp_addr, &sun_len,
168 				    sizeof(sun_len));
169 				KREAD(unp_conn.unp_addr, &xu.xu_caddr, sun_len);
170 			}
171 		}
172 		KREAD(unp->unp_socket, &so, sizeof(so));
173 		if (sotoxsocket(&so, &xu.xu_socket) != 0)
174 			goto fail;
175 		COPYOUT(&xu, sizeof(xu));
176 	}
177 
178 	/* Reread the counts and write the footer. */
179 	kread(count_off, &unp_count, sizeof(unp_count));
180 	kread(gencnt_off, &unp_gencnt, sizeof(unp_gencnt));
181 	xug.xug_count = unp_count;
182 	xug.xug_gen = unp_gencnt;
183 	COPYOUT(&xug, sizeof xug);
184 
185 	*bufp = buf;
186 	return (0);
187 
188 fail:
189 	free(buf);
190 	return (-1);
191 #undef COPYOUT
192 #undef KREAD
193 }
194 
195 void
196 unixpr(u_long count_off, u_long gencnt_off, u_long dhead_off, u_long shead_off)
197 {
198 	char 	*buf;
199 	int	ret, type;
200 	struct	xsocket *so;
201 	struct	xunpgen *xug, *oxug;
202 	struct	xunpcb *xunp;
203 
204 	for (type = SOCK_STREAM; type <= SOCK_SEQPACKET; type++) {
205 		if (live)
206 			ret = pcblist_sysctl(type, &buf);
207 		else
208 			ret = pcblist_kvm(count_off, gencnt_off,
209 			    type == SOCK_STREAM ? shead_off :
210 			    (type == SOCK_DGRAM ? dhead_off : 0), &buf);
211 		if (ret == -1)
212 			continue;
213 		if (ret < 0)
214 			return;
215 
216 		oxug = xug = (struct xunpgen *)buf;
217 		for (xug = (struct xunpgen *)((char *)xug + xug->xug_len);
218 		     xug->xug_len > sizeof(struct xunpgen);
219 		     xug = (struct xunpgen *)((char *)xug + xug->xug_len)) {
220 			xunp = (struct xunpcb *)xug;
221 			so = &xunp->xu_socket;
222 
223 			/* Ignore PCBs which were freed during copyout. */
224 			if (xunp->xu_unp.unp_gencnt > oxug->xug_gen)
225 				continue;
226 			unixdomainpr(xunp, so);
227 		}
228 		if (xug != oxug && xug->xug_gen != oxug->xug_gen) {
229 			if (oxug->xug_count > xug->xug_count) {
230 				printf("Some %s sockets may have been deleted.\n",
231 				       socktype[type]);
232 			} else if (oxug->xug_count < xug->xug_count) {
233 				printf("Some %s sockets may have been created.\n",
234 			       socktype[type]);
235 			} else {
236 				printf("Some %s sockets may have been created or deleted",
237 			       socktype[type]);
238 			}
239 		}
240 		free(buf);
241 	}
242 }
243 
244 static void
245 unixdomainpr(struct xunpcb *xunp, struct xsocket *so)
246 {
247 	struct unpcb *unp;
248 	struct sockaddr_un *sa;
249 	static int first = 1;
250 
251 	unp = &xunp->xu_unp;
252 	if (unp->unp_addr)
253 		sa = &xunp->xu_addr;
254 	else
255 		sa = (struct sockaddr_un *)0;
256 
257 	if (first) {
258 		printf("Active UNIX domain sockets\n");
259 		printf(
260 "%-8.8s %-6.6s %-6.6s %-6.6s %8.8s %8.8s %8.8s %8.8s Addr\n",
261 		    "Address", "Type", "Recv-Q", "Send-Q",
262 		    "Inode", "Conn", "Refs", "Nextref");
263 		first = 0;
264 	}
265 	printf("%8lx %-6.6s %6u %6u %8lx %8lx %8lx %8lx",
266 	       (long)so->so_pcb, socktype[so->so_type], so->so_rcv.sb_cc,
267 	       so->so_snd.sb_cc,
268 	       (long)unp->unp_vnode, (long)unp->unp_conn,
269 	       (long)LIST_FIRST(&unp->unp_refs), (long)LIST_NEXT(unp, unp_reflink));
270 	if (sa)
271 		printf(" %.*s",
272 		    (int)(sa->sun_len - offsetof(struct sockaddr_un, sun_path)),
273 		    sa->sun_path);
274 	putchar('\n');
275 }
276