]> CyberLeo.Net >> Repos - FreeBSD/releng/9.2.git/blob - usr.bin/netstat/unix.c
- Copy stable/9 to releng/9.2 as part of the 9.2-RELEASE cycle.
[FreeBSD/releng/9.2.git] / usr.bin / netstat / unix.c
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         for (type = SOCK_STREAM; type <= SOCK_SEQPACKET; type++) {
203                 if (live)
204                         ret = pcblist_sysctl(type, &buf);
205                 else {
206                         head_off = 0;
207                         switch (type) {
208                         case SOCK_STREAM:
209                                 head_off = shead_off;
210                                 break;
211
212                         case SOCK_DGRAM:
213                                 head_off = dhead_off;
214                                 break;
215
216                         case SOCK_SEQPACKET:
217                                 head_off = sphead_off;
218                                 break;
219                         }
220                         ret = pcblist_kvm(count_off, gencnt_off, head_off,
221                             &buf);
222                 }
223                 if (ret == -1)
224                         continue;
225                 if (ret < 0)
226                         return;
227
228                 oxug = xug = (struct xunpgen *)buf;
229                 for (xug = (struct xunpgen *)((char *)xug + xug->xug_len);
230                      xug->xug_len > sizeof(struct xunpgen);
231                      xug = (struct xunpgen *)((char *)xug + xug->xug_len)) {
232                         xunp = (struct xunpcb *)xug;
233                         so = &xunp->xu_socket;
234
235                         /* Ignore PCBs which were freed during copyout. */
236                         if (xunp->xu_unp.unp_gencnt > oxug->xug_gen)
237                                 continue;
238                         unixdomainpr(xunp, so);
239                 }
240                 if (xug != oxug && xug->xug_gen != oxug->xug_gen) {
241                         if (oxug->xug_count > xug->xug_count) {
242                                 printf("Some %s sockets may have been deleted.\n",
243                                        socktype[type]);
244                         } else if (oxug->xug_count < xug->xug_count) {
245                                 printf("Some %s sockets may have been created.\n",
246                                socktype[type]);
247                         } else {
248                                 printf("Some %s sockets may have been created or deleted",
249                                socktype[type]);
250                         }
251                 }
252                 free(buf);
253         }
254 }
255
256 static void
257 unixdomainpr(struct xunpcb *xunp, struct xsocket *so)
258 {
259         struct unpcb *unp;
260         struct sockaddr_un *sa;
261         static int first = 1;
262         char buf1[15];
263
264         unp = &xunp->xu_unp;
265         if (unp->unp_addr)
266                 sa = &xunp->xu_addr;
267         else
268                 sa = (struct sockaddr_un *)0;
269
270         if (first && !Lflag) {
271                 printf("Active UNIX domain sockets\n");
272                 printf(
273 "%-8.8s %-6.6s %-6.6s %-6.6s %8.8s %8.8s %8.8s %8.8s Addr\n",
274                     "Address", "Type", "Recv-Q", "Send-Q",
275                     "Inode", "Conn", "Refs", "Nextref");
276                 first = 0;
277         }
278
279         if (Lflag && so->so_qlimit == 0)
280                 return;
281
282         if (Lflag) {
283                 snprintf(buf1, 15, "%d/%d/%d", so->so_qlen,
284                     so->so_incqlen, so->so_qlimit);
285                 printf("unix  %-14.14s", buf1);
286         } else {
287                 printf("%8lx %-6.6s %6u %6u %8lx %8lx %8lx %8lx",
288                     (long)so->so_pcb, socktype[so->so_type], so->so_rcv.sb_cc,
289                     so->so_snd.sb_cc, (long)unp->unp_vnode, (long)unp->unp_conn,
290                     (long)LIST_FIRST(&unp->unp_refs),
291                     (long)LIST_NEXT(unp, unp_reflink));
292         }
293         if (sa)
294                 printf(" %.*s",
295                     (int)(sa->sun_len - offsetof(struct sockaddr_un, sun_path)),
296                     sa->sun_path);
297         putchar('\n');
298 }