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
32static 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
64static	void unixdomainpr(struct xunpcb *, struct xsocket *);
65
66static	const char *const socktype[] =
67    { "#0", "stream", "dgram", "raw", "rdm", "seqpacket" };
68
69static int
70pcblist_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
97static int
98pcblist_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
184fail:
185	free(buf);
186	return (-1);
187#undef COPYOUT
188#undef KREAD
189}
190
191void
192unixpr(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
257static void
258unixdomainpr(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