]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - usr.bin/fstat/fstat.c
Merge llvm-project release/17.x llvmorg-17.0.0-rc4-10-g0176e8729ea4
[FreeBSD/FreeBSD.git] / usr.bin / fstat / fstat.c
1 /*-
2  * SPDX-License-Identifier: BSD-3-Clause
3  *
4  * Copyright (c) 2009 Stanislav Sedov <stas@FreeBSD.org>
5  * Copyright (c) 1988, 1993
6  *      The Regents of the University of California.  All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  * 3. Neither the name of the University nor the names of its contributors
17  *    may be used to endorse or promote products derived from this software
18  *    without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
21  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30  * SUCH DAMAGE.
31  */
32
33 #include <sys/param.h>
34 #include <sys/user.h>
35 #include <sys/stat.h>
36 #include <sys/socket.h>
37 #include <sys/socketvar.h>
38 #include <sys/sysctl.h>
39 #include <sys/queue.h>
40 #include <sys/un.h>
41
42 #include <netinet/in.h>
43
44 #include <arpa/inet.h>
45
46 #include <assert.h>
47 #include <ctype.h>
48 #include <err.h>
49 #include <libprocstat.h>
50 #include <limits.h>
51 #include <pwd.h>
52 #include <stdint.h>
53 #include <stdio.h>
54 #include <stdlib.h>
55 #include <stddef.h>
56 #include <string.h>
57 #include <unistd.h>
58 #include <netdb.h>
59
60 #include "functions.h"
61
62 static int      fsflg,  /* show files on same filesystem as file(s) argument */
63                 pflg,   /* show files open by a particular pid */
64                 sflg,   /* show socket details */
65                 uflg;   /* show files open by a particular (effective) user */
66 static int      checkfile; /* restrict to particular files or filesystems */
67 static int      nflg;   /* (numerical) display f.s. and rdev as dev_t */
68 static int      mflg;   /* include memory-mapped files */
69 static int      vflg;   /* be verbose */
70
71 typedef struct devs {
72         struct devs     *next;
73         uint64_t        fsid;
74         uint64_t        ino;
75         const char      *name;
76 } DEVS;
77
78 static DEVS *devs;
79 static char *memf, *nlistf;
80
81 static int      getfname(const char *filename);
82 static void     dofiles(struct procstat *procstat, struct kinfo_proc *p);
83 static void     print_access_flags(int flags);
84 static void     print_file_info(struct procstat *procstat,
85     struct filestat *fst, const char *uname, const char *cmd, int pid);
86 static void     print_pipe_info(struct procstat *procstat,
87     struct filestat *fst);
88 static void     print_pts_info(struct procstat *procstat,
89     struct filestat *fst);
90 static void     print_sem_info(struct procstat *procstat,
91     struct filestat *fst);
92 static void     print_shm_info(struct procstat *procstat,
93     struct filestat *fst);
94 static void     print_socket_info(struct procstat *procstat,
95     struct filestat *fst);
96 static void     print_vnode_info(struct procstat *procstat,
97     struct filestat *fst);
98 static void     usage(void) __dead2;
99
100 int
101 do_fstat(int argc, char **argv)
102 {
103         struct kinfo_proc *p;
104         struct passwd *passwd;
105         struct procstat *procstat;
106         int arg, ch, what;
107         int cnt, i;
108
109         arg = 0;
110         what = KERN_PROC_PROC;
111         nlistf = memf = NULL;
112         while ((ch = getopt(argc, argv, "fmnp:su:vN:M:")) != -1)
113                 switch((char)ch) {
114                 case 'f':
115                         fsflg = 1;
116                         break;
117                 case 'M':
118                         memf = optarg;
119                         break;
120                 case 'N':
121                         nlistf = optarg;
122                         break;
123                 case 'm':
124                         mflg = 1;
125                         break;
126                 case 'n':
127                         nflg = 1;
128                         break;
129                 case 'p':
130                         if (pflg++)
131                                 usage();
132                         if (!isdigit(*optarg)) {
133                                 warnx("-p requires a process id");
134                                 usage();
135                         }
136                         what = KERN_PROC_PID;
137                         arg = atoi(optarg);
138                         break;
139                 case 's':
140                         sflg = 1;
141                         break;
142                 case 'u':
143                         if (uflg++)
144                                 usage();
145                         if (!(passwd = getpwnam(optarg)))
146                                 errx(1, "%s: unknown uid", optarg);
147                         what = KERN_PROC_UID;
148                         arg = passwd->pw_uid;
149                         break;
150                 case 'v':
151                         vflg = 1;
152                         break;
153                 case '?':
154                 default:
155                         usage();
156                 }
157
158         if (*(argv += optind)) {
159                 for (; *argv; ++argv) {
160                         if (getfname(*argv))
161                                 checkfile = 1;
162                 }
163                 if (!checkfile) /* file(s) specified, but none accessible */
164                         exit(1);
165         }
166
167         if (fsflg && !checkfile) {
168                 /* -f with no files means use wd */
169                 if (getfname(".") == 0)
170                         exit(1);
171                 checkfile = 1;
172         }
173
174         if (memf != NULL)
175                 procstat = procstat_open_kvm(nlistf, memf);
176         else
177                 procstat = procstat_open_sysctl();
178         if (procstat == NULL)
179                 errx(1, "procstat_open()");
180         p = procstat_getprocs(procstat, what, arg, &cnt);
181         if (p == NULL)
182                 errx(1, "procstat_getprocs()");
183
184         /*
185          * Print header.
186          */
187         if (nflg)
188                 printf("%s",
189 "USER     CMD          PID   FD  DEV    INUM       MODE SZ|DV R/W");
190         else
191                 printf("%s",
192 "USER     CMD          PID   FD MOUNT      INUM MODE         SZ|DV R/W");
193         if (checkfile && fsflg == 0)
194                 printf(" NAME\n");
195         else
196                 putchar('\n');
197
198         /*
199          * Go through the process list.
200          */
201         for (i = 0; i < cnt; i++) {
202                 if (p[i].ki_stat == SZOMB)
203                         continue;
204                 dofiles(procstat, &p[i]);
205         }
206         procstat_freeprocs(procstat, p);
207         procstat_close(procstat);
208         return (0);
209 }
210
211 static void
212 dofiles(struct procstat *procstat, struct kinfo_proc *kp)
213 {
214         const char *cmd;
215         const char *uname;
216         struct filestat *fst;
217         struct filestat_list *head;
218         int pid;
219
220         uname = user_from_uid(kp->ki_uid, 0);
221         pid = kp->ki_pid;
222         cmd = kp->ki_comm;
223
224         head = procstat_getfiles(procstat, kp, mflg);
225         if (head == NULL)
226                 return;
227         STAILQ_FOREACH(fst, head, next)
228                 print_file_info(procstat, fst, uname, cmd, pid);
229         procstat_freefiles(procstat, head);
230 }
231
232
233 static void
234 print_file_info(struct procstat *procstat, struct filestat *fst,
235     const char *uname, const char *cmd, int pid)
236 {
237         struct vnstat vn;
238         DEVS *d;
239         const char *filename;
240         int error, fsmatch = 0;
241         char errbuf[_POSIX2_LINE_MAX];
242
243         filename = NULL;
244         if (checkfile != 0) {
245                 if (fst->fs_type != PS_FST_TYPE_VNODE &&
246                     fst->fs_type != PS_FST_TYPE_FIFO)
247                         return;
248                 error = procstat_get_vnode_info(procstat, fst, &vn, errbuf);
249                 if (error != 0)
250                         return;
251
252                 for (d = devs; d != NULL; d = d->next)
253                         if (d->fsid == vn.vn_fsid) {
254                                 fsmatch = 1;
255                                 if (d->ino == vn.vn_fileid) {
256                                         filename = d->name;
257                                         break;
258                                 }
259                         }
260                 if (fsmatch == 0 || (filename == NULL && fsflg == 0))
261                         return;
262         }
263
264         /*
265          * Print entry prefix.
266          */
267         printf("%-8.8s %-10s %5d", uname, cmd, pid);
268         if (fst->fs_uflags & PS_FST_UFLAG_TEXT)
269                 printf(" text");
270         else if (fst->fs_uflags & PS_FST_UFLAG_CDIR)
271                 printf("   wd");
272         else if (fst->fs_uflags & PS_FST_UFLAG_RDIR)
273                 printf(" root");
274         else if (fst->fs_uflags & PS_FST_UFLAG_TRACE)
275                 printf("   tr");
276         else if (fst->fs_uflags & PS_FST_UFLAG_MMAP)
277                 printf(" mmap");
278         else if (fst->fs_uflags & PS_FST_UFLAG_JAIL)
279                 printf(" jail");
280         else if (fst->fs_uflags & PS_FST_UFLAG_CTTY)
281                 printf(" ctty");
282         else
283                 printf(" %4d", fst->fs_fd);
284
285         /*
286          * Print type-specific data.
287          */
288         switch (fst->fs_type) {
289         case PS_FST_TYPE_FIFO:
290         case PS_FST_TYPE_VNODE:
291                 print_vnode_info(procstat, fst);
292                 break;
293         case PS_FST_TYPE_SOCKET:
294                 print_socket_info(procstat, fst);
295                 break;
296         case PS_FST_TYPE_PIPE:
297                 print_pipe_info(procstat, fst);
298                 break;
299         case PS_FST_TYPE_PTS:
300                 print_pts_info(procstat, fst);
301                 break;
302         case PS_FST_TYPE_SHM:
303                 print_shm_info(procstat, fst);
304                 break;
305         case PS_FST_TYPE_SEM:
306                 print_sem_info(procstat, fst);
307                 break;
308         case PS_FST_TYPE_DEV:
309                 break;
310         default:        
311                 if (vflg)
312                         fprintf(stderr,
313                             "unknown file type %d for file %d of pid %d\n",
314                             fst->fs_type, fst->fs_fd, pid);
315         }
316         if (filename && !fsflg)
317                 printf("  %s", filename);
318         putchar('\n');
319 }
320
321 static char *
322 addr_to_string(struct sockaddr_storage *ss, char *buffer, int buflen)
323 {
324         char buffer2[INET6_ADDRSTRLEN];
325         struct sockaddr_in6 *sin6;
326         struct sockaddr_in *sin;
327         struct sockaddr_un *sun;
328
329         switch (ss->ss_family) {
330         case AF_LOCAL:
331                 sun = (struct sockaddr_un *)ss;
332                 if (strlen(sun->sun_path) == 0)
333                         strlcpy(buffer, "-", buflen);
334                 else
335                         strlcpy(buffer, sun->sun_path, buflen);
336                 break;
337
338         case AF_INET:
339                 sin = (struct sockaddr_in *)ss;
340                 if (sin->sin_addr.s_addr == INADDR_ANY)
341                         snprintf(buffer, buflen, "%s:%d", "*",
342                             ntohs(sin->sin_port));
343                 else if (inet_ntop(AF_INET, &sin->sin_addr, buffer2,
344                     sizeof(buffer2)) != NULL)
345                         snprintf(buffer, buflen, "%s:%d", buffer2,
346                             ntohs(sin->sin_port));
347                 break;
348
349         case AF_INET6:
350                 sin6 = (struct sockaddr_in6 *)ss;
351                 if (IN6_IS_ADDR_UNSPECIFIED(&sin6->sin6_addr))
352                         snprintf(buffer, buflen, "%s.%d", "*",
353                             ntohs(sin6->sin6_port));
354                 else if (inet_ntop(AF_INET6, &sin6->sin6_addr, buffer2,
355                     sizeof(buffer2)) != NULL)
356                         snprintf(buffer, buflen, "%s.%d", buffer2,
357                             ntohs(sin6->sin6_port));
358                 else
359                         strlcpy(buffer, "-", buflen);
360                 break;
361
362         default:
363                 strlcpy(buffer, "", buflen);
364                 break;
365         }
366         return buffer;
367 }
368
369
370 static void
371 print_socket_info(struct procstat *procstat, struct filestat *fst)
372 {
373         static const char *stypename[] = {
374                 "unused",       /* 0 */
375                 "stream",       /* 1 */
376                 "dgram",        /* 2 */
377                 "raw",          /* 3 */
378                 "rdm",          /* 4 */
379                 "seqpak"        /* 5 */
380         };
381 #define STYPEMAX 5
382         struct sockstat sock;
383         struct protoent *pe;
384         char errbuf[_POSIX2_LINE_MAX];
385         char src_addr[PATH_MAX], dst_addr[PATH_MAX];
386         struct sockaddr_un *sun;
387         int error;
388         static int isopen;
389
390         error = procstat_get_socket_info(procstat, fst, &sock, errbuf);
391         if (error != 0) {
392                 printf("* error");
393                 return;
394         }
395         if (sock.type > STYPEMAX)
396                 printf("* %s ?%d", sock.dname, sock.type);
397         else
398                 printf("* %s %s", sock.dname, stypename[sock.type]);
399
400         /*
401          * protocol specific formatting
402          *
403          * Try to find interesting things to print.  For tcp, the interesting
404          * thing is the address of the tcpcb, for udp and others, just the
405          * inpcb (socket pcb).  For unix domain, its the address of the socket
406          * pcb and the address of the connected pcb (if connected).  Otherwise
407          * just print the protocol number and address of the socket itself.
408          * The idea is not to duplicate netstat, but to make available enough
409          * information for further analysis.
410          */
411         switch (sock.dom_family) {
412         case AF_INET:
413         case AF_INET6:
414                 if (!isopen)
415                         setprotoent(++isopen);
416                 if ((pe = getprotobynumber(sock.proto)) != NULL)
417                         printf(" %s", pe->p_name);
418                 else
419                         printf(" %d", sock.proto);
420                 if (sock.proto == IPPROTO_TCP ) {
421                         if (sock.inp_ppcb != 0)
422                                 printf(" %lx", (u_long)sock.inp_ppcb);
423                 }
424                 else if (sock.so_pcb != 0)
425                         printf(" %lx", (u_long)sock.so_pcb);
426                 if (!sflg)
427                         break;
428                 printf(" %s <-> %s",
429                     addr_to_string(&sock.sa_local, src_addr, sizeof(src_addr)),
430                     addr_to_string(&sock.sa_peer, dst_addr, sizeof(dst_addr)));
431                 break;
432         case AF_UNIX:
433                 /* print address of pcb and connected pcb */
434                 if (sock.so_pcb != 0) {
435                         printf(" %lx", (u_long)sock.so_pcb);
436                         if (sock.unp_conn) {
437                                 char shoconn[4], *cp;
438
439                                 cp = shoconn;
440                                 if (!(sock.so_rcv_sb_state & SBS_CANTRCVMORE))
441                                         *cp++ = '<';
442                                 *cp++ = '-';
443                                 if (!(sock.so_snd_sb_state & SBS_CANTSENDMORE))
444                                         *cp++ = '>';
445                                 *cp = '\0';
446                                 printf(" %s %lx", shoconn,
447                                     (u_long)sock.unp_conn);
448                         }
449                 }
450                 if (!sflg)
451                         break;
452                 sun = (struct sockaddr_un *)&sock.sa_local;
453                 /*
454                  * While generally we like to print two addresses,
455                  * local and peer, for sockets, it turns out to be
456                  * more useful to print the first non-null address for
457                  * local sockets, as typically they aren't bound and
458                  *  connected, and the path strings can get long.
459                  */
460                 if (sun->sun_path[0] != 0)
461                         addr_to_string(&sock.sa_local,
462                             src_addr, sizeof(src_addr));
463                 else
464                         addr_to_string(&sock.sa_peer,
465                             src_addr, sizeof(src_addr));
466                 printf(" %s", src_addr);
467                 break;
468         default:
469                 /* print protocol number and socket address */
470                 printf(" %d %lx", sock.proto, (u_long)sock.so_addr);
471         }
472 }
473
474 static void
475 print_pipe_info(struct procstat *procstat, struct filestat *fst)
476 {
477         struct pipestat ps;
478         char errbuf[_POSIX2_LINE_MAX];
479         int error;
480
481         error = procstat_get_pipe_info(procstat, fst, &ps, errbuf);
482         if (error != 0) {
483                 printf("* error");
484                 return;
485         }
486         printf("* pipe %8lx <-> %8lx", (u_long)ps.addr, (u_long)ps.peer);
487         printf(" %6zd", ps.buffer_cnt);
488         print_access_flags(fst->fs_fflags);
489 }
490
491 static void
492 print_pts_info(struct procstat *procstat, struct filestat *fst)
493 {
494         struct ptsstat pts;
495         char errbuf[_POSIX2_LINE_MAX];
496         int error;
497
498         error = procstat_get_pts_info(procstat, fst, &pts, errbuf);
499         if (error != 0) {
500                 printf("* error");
501                 return;
502         }
503         printf("* pseudo-terminal master ");
504         if (nflg || !*pts.devname) {
505                 printf("%#10jx", (uintmax_t)pts.dev);
506         } else {
507                 printf("%10s", pts.devname);
508         }
509         print_access_flags(fst->fs_fflags);
510 }
511
512 static void
513 print_sem_info(struct procstat *procstat, struct filestat *fst)
514 {
515         struct semstat sem;
516         char errbuf[_POSIX2_LINE_MAX];
517         char mode[15];
518         int error;
519
520         error = procstat_get_sem_info(procstat, fst, &sem, errbuf);
521         if (error != 0) {
522                 printf("* error");
523                 return;
524         }
525         if (nflg) {
526                 printf("             ");
527                 (void)snprintf(mode, sizeof(mode), "%o", sem.mode);
528         } else {
529                 printf(" %-15s", fst->fs_path != NULL ? fst->fs_path : "-");
530                 strmode(sem.mode, mode);
531         }
532         printf(" %10s %6u", mode, sem.value);
533         print_access_flags(fst->fs_fflags);
534 }
535
536 static void
537 print_shm_info(struct procstat *procstat, struct filestat *fst)
538 {
539         struct shmstat shm;
540         char errbuf[_POSIX2_LINE_MAX];
541         char mode[15];
542         int error;
543
544         error = procstat_get_shm_info(procstat, fst, &shm, errbuf);
545         if (error != 0) {
546                 printf("* error");
547                 return;
548         }
549         if (nflg) {
550                 printf("             ");
551                 (void)snprintf(mode, sizeof(mode), "%o", shm.mode);
552         } else {
553                 printf(" %-15s", fst->fs_path != NULL ? fst->fs_path : "-");
554                 strmode(shm.mode, mode);
555         }
556         printf(" %10s %6ju", mode, shm.size);
557         print_access_flags(fst->fs_fflags);
558 }
559
560 static void
561 print_vnode_info(struct procstat *procstat, struct filestat *fst)
562 {
563         struct vnstat vn;
564         char errbuf[_POSIX2_LINE_MAX];
565         char mode[15];
566         const char *badtype;
567         int error;
568
569         badtype = NULL;
570         error = procstat_get_vnode_info(procstat, fst, &vn, errbuf);
571         if (error != 0)
572                 badtype = errbuf;
573         else if (vn.vn_type == PS_FST_VTYPE_VBAD)
574                 badtype = "bad";
575         else if (vn.vn_type == PS_FST_VTYPE_VNON)
576                 badtype = "none";
577         if (badtype != NULL) {
578                 printf(" -         -  %10s    -", badtype);
579                 return;
580         }
581
582         if (nflg)
583                 printf(" %#5jx", (uintmax_t)vn.vn_fsid);
584         else if (vn.vn_mntdir != NULL)
585                 (void)printf(" %-8s", vn.vn_mntdir);
586
587         /*
588          * Print access mode.
589          */
590         if (nflg)
591                 (void)snprintf(mode, sizeof(mode), "%o", vn.vn_mode);
592         else {
593                 strmode(vn.vn_mode, mode);
594         }
595         (void)printf(" %6jd %10s", (intmax_t)vn.vn_fileid, mode);
596
597         if (vn.vn_type == PS_FST_VTYPE_VBLK || vn.vn_type == PS_FST_VTYPE_VCHR) {
598                 if (nflg || !*vn.vn_devname)
599                         printf(" %#6jx", (uintmax_t)vn.vn_dev);
600                 else {
601                         printf(" %6s", vn.vn_devname);
602                 }
603         } else
604                 printf(" %6ju", (uintmax_t)vn.vn_size);
605         print_access_flags(fst->fs_fflags);
606 }
607
608 static void
609 print_access_flags(int flags)
610 {
611         char rw[3];
612
613         rw[0] = '\0';
614         if (flags & PS_FST_FFLAG_READ)
615                 strcat(rw, "r");
616         if (flags & PS_FST_FFLAG_WRITE)
617                 strcat(rw, "w");
618         printf(" %2s", rw);
619 }
620
621 int
622 getfname(const char *filename)
623 {
624         struct stat statbuf;
625         DEVS *cur;
626
627         if (stat(filename, &statbuf)) {
628                 warn("%s", filename);
629                 return (0);
630         }
631         if ((cur = malloc(sizeof(DEVS))) == NULL)
632                 err(1, NULL);
633         cur->next = devs;
634         devs = cur;
635
636         cur->ino = statbuf.st_ino;
637         cur->fsid = statbuf.st_dev;
638         cur->name = filename;
639         return (1);
640 }
641
642 static void
643 usage(void)
644 {
645         (void)fprintf(stderr,
646  "usage: fstat [-fmnv] [-M core] [-N system] [-p pid] [-u user] [file ...]\n");
647         exit(1);
648 }