usr.bin/fstat/fstat.c

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