usr.sbin/pstat/pstat.c

   1 /*-
   2  * Copyright (c) 1980, 1991, 1993, 1994
   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  * @(#) Copyright (c) 1980, 1991, 1993, 1994 The Regents of the University of California.  All rights reserved.
  34  * @(#)pstat.c  8.16 (Berkeley) 5/9/95
  35  * $FreeBSD: src/usr.sbin/pstat/pstat.c,v 1.49.2.5 2002/07/12 09:12:49 des Exp $
  36  */
  37
  38 #include <sys/user.h>
  39 #include <sys/kinfo.h>
  40 #include <sys/param.h>
  41 #include <sys/time.h>
  42 #include <sys/vnode.h>
  43 #include <sys/ucred.h>
  44 #include <sys/file.h>
  45 #include <vfs/ufs/quota.h>
  46 #include <vfs/ufs/inode.h>
  47 #include <sys/mount.h>
  48 #include <sys/uio.h>
  49 #include <sys/namei.h>
  50 #include <vfs/union/union.h>
  51 #include <sys/stat.h>
  52 #include <nfs/rpcv2.h>
  53 #include <nfs/nfsproto.h>
  54 #include <nfs/nfs.h>
  55 #include <nfs/nfsnode.h>
  56 #include <sys/ioctl.h>
  57 #include <sys/ioctl_compat.h>   /* XXX NTTYDISC is too well hidden */
  58 #include <sys/tty.h>
  59 #include <sys/conf.h>
  60 #include <sys/blist.h>
  61
  62 #include <sys/sysctl.h>
  63
  64 #include <err.h>
  65 #include <fcntl.h>
  66 #include <kvm.h>
  67 #include <limits.h>
  68 #include <nlist.h>
  69 #include <stdio.h>
  70 #include <stdlib.h>
  71 #include <string.h>
  72 #include <unistd.h>
  73
  74 #ifdef USE_KCORE
  75 #  define KCORE_KINFO_WRAPPER
  76 #  include <kcore.h>
  77 #else
  78 #  include <kinfo.h>
  79 #endif
  80
  81 enum {
  82         NL_MOUNTLIST,
  83         NL_NUMVNODES,
  84         NL_RC_TTY,
  85         NL_NRC_TTY,
  86         NL_CY_TTY,
  87         NL_NCY_TTY,
  88         NL_SI_TTY,
  89         NL_SI_NPORTS,
  90 };
  91
  92 const size_t NL_LAST_MANDATORY = NL_NUMVNODES;
  93
  94 struct nlist nl[] = {
  95         { "_mountlist", 0, 0, 0, 0 },   /* address of head of mount list. */
  96         { "_numvnodes", 0, 0, 0, 0 },
  97         { "_rc_tty", 0, 0, 0, 0 },
  98         { "_nrc_tty", 0, 0, 0, 0 },
  99         { "_cy_tty", 0, 0, 0, 0 },
 100         { "_ncy_tty", 0, 0, 0, 0 },
 101         { "_si__tty", 0, 0, 0, 0 },
 102         { "_si_Nports", 0, 0, 0, 0 },
 103         { "", 0, 0, 0, 0 }
 104 };
 105
 106 int     usenumflag;
 107 int     totalflag;
 108 int     swapflag;
 109 char    *nlistf = NULL;
 110 char    *memf   = NULL;
 111 kvm_t   *kd;
 112
 113 const char      *usagestr;
 114
 115 struct {
 116         int m_flag;
 117         const char *m_name;
 118 } mnt_flags[] = {
 119         { MNT_RDONLY, "rdonly" },
 120         { MNT_SYNCHRONOUS, "sync" },
 121         { MNT_NOEXEC, "noexec" },
 122         { MNT_NOSUID, "nosuid" },
 123         { MNT_NODEV, "nodev" },
 124         { MNT_UNION, "union" },
 125         { MNT_ASYNC, "async" },
 126         { MNT_SUIDDIR, "suiddir" },
 127         { MNT_SOFTDEP, "softdep" },
 128         { MNT_NOSYMFOLLOW, "nosymfollow" },
 129         { MNT_NOATIME, "noatime" },
 130         { MNT_NOCLUSTERR, "noclusterread" },
 131         { MNT_NOCLUSTERW, "noclusterwrite" },
 132         { MNT_EXRDONLY, "exrdonly" },
 133         { MNT_EXPORTED, "exported" },
 134         { MNT_DEFEXPORTED, "defexported" },
 135         { MNT_EXPORTANON, "exportanon" },
 136         { MNT_EXKERB, "exkerb" },
 137         { MNT_EXPUBLIC, "public" },
 138         { MNT_LOCAL, "local" },
 139         { MNT_QUOTA, "quota" },
 140         { MNT_ROOTFS, "rootfs" },
 141         { MNT_USER, "user" },
 142         { MNT_IGNORE, "ignore" },
 143         { MNT_UPDATE, "update" },
 144         { MNT_DELEXPORT, "delexport" },
 145         { MNT_RELOAD, "reload" },
 146         { MNT_FORCE, "force" },
 147         { 0, NULL }
 148 };
 149
 150
 151 #define SVAR(var) __STRING(var) /* to force expansion */
 152 #define KGET(idx, var)                                                  \
 153         KGET1(idx, &var, sizeof(var), SVAR(var))
 154 #define KGET1(idx, p, s, msg)                                           \
 155         KGET2(nl[idx].n_value, p, s, msg)
 156 #define KGET2(addr, p, s, msg)                                          \
 157         if (kvm_read(kd, (u_long)(addr), p, s) != s)                    \
 158                 warnx("cannot read %s: %s", msg, kvm_geterr(kd))
 159 #define KGETN(idx, var)                                                 \
 160         KGET1N(idx, &var, sizeof(var), SVAR(var))
 161 #define KGET1N(idx, p, s, msg)                                          \
 162         KGET2N(nl[idx].n_value, p, s, msg)
 163 #define KGET2N(addr, p, s, msg)                                         \
 164         ((kvm_read(kd, (u_long)(addr), p, s) == s) ? 1 : 0)
 165 #define KGETRET(addr, p, s, msg)                                        \
 166         if (kvm_read(kd, (u_long)(addr), p, s) != s) {                  \
 167                 warnx("cannot read %s: %s", msg, kvm_geterr(kd));       \
 168                 return (0);                                             \
 169         }
 170
 171 void    filemode(void);
 172 int     getfiles(char **, int *);
 173 struct mount *
 174         getmnt(struct mount *);
 175 struct e_vnode *
 176         kinfo_vnodes(int *);
 177 struct e_vnode *
 178         loadvnodes(int *);
 179 void    mount_print(struct mount *);
 180 void    nfs_header(void);
 181 int     nfs_print(struct vnode *);
 182 void    swapmode(void);
 183 void    ttymode(void);
 184 void    ttyprt(struct tty *, int);
 185 void    ttytype(struct tty *, const char *, int, int, int);
 186 void    ufs_header(void);
 187 int     ufs_print(struct vnode *);
 188 void    union_header(void);
 189 int     union_print(struct vnode *);
 190 static void usage(void);
 191 void    vnode_header(void);
 192 void    vnode_print(struct vnode *, struct vnode *);
 193 void    vnodemode(void);
 194
 195 int
 196 main(int argc, char **argv)
 197 {
 198         int ch, ret;
 199         int fileflag, ttyflag, vnodeflag;
 200         char buf[_POSIX2_LINE_MAX];
 201         const char *opts;
 202
 203         fileflag = swapflag = ttyflag = vnodeflag = 0;
 204
 205         /* We will behave like good old swapinfo if thus invoked */
 206         opts = strrchr(argv[0],'/');
 207         if (opts)
 208                 opts++;
 209         else
 210                 opts = argv[0];
 211         if (!strcmp(opts,"swapinfo")) {
 212                 swapflag = 1;
 213                 opts = "kM:N:";
 214                 usagestr = "swapinfo [-k] [-M core] [-N system]";
 215         } else {
 216                 opts = "TM:N:fiknstv";
 217                 usagestr = "pstat [-Tfknst] [-M core] [-N system]";
 218         }
 219
 220         while ((ch = getopt(argc, argv, opts)) != -1)
 221                 switch (ch) {
 222                 case 'f':
 223                         fileflag = 1;
 224                         break;
 225                 case 'k':
 226                         if (setenv("BLOCKSIZE", "1K", 1) == -1)
 227                                 warn("setenv: cannot set BLOCKSIZE=1K");
 228                         break;
 229                 case 'M':
 230                         memf = optarg;
 231                         break;
 232                 case 'N':
 233                         nlistf = optarg;
 234                         break;
 235                 case 'n':
 236                         usenumflag = 1;
 237                         break;
 238                 case 's':
 239                         ++swapflag;
 240                         break;
 241                 case 'T':
 242                         totalflag = 1;
 243                         break;
 244                 case 't':
 245                         ttyflag = 1;
 246                         break;
 247                 case 'v':
 248                 case 'i':               /* Backward compatibility. */
 249                         errx(1, "vnode mode not supported");
 250 #if 0
 251                         vnodeflag = 1;
 252                         break;
 253 #endif
 254                 default:
 255                         usage();
 256                 }
 257         argc -= optind;
 258         argv += optind;
 259
 260         /*
 261          * Discard setgid privileges if not the running kernel so that bad
 262          * guys can't print interesting stuff from kernel memory.
 263          */
 264         if (nlistf != NULL || memf != NULL)
 265                 setgid(getgid());
 266
 267         if ((kd = kvm_openfiles(nlistf, memf, NULL, O_RDONLY, buf)) == NULL)
 268                 errx(1, "kvm_openfiles: %s", buf);
 269 #ifdef USE_KCORE
 270         if (kcore_wrapper_open(nlistf, memf, buf))
 271                 errx(1, "kcore_open: %s", buf);
 272 #endif
 273         if ((ret = kvm_nlist(kd, nl)) != 0) {
 274                 size_t i;
 275                 int quit = 0;
 276
 277                 if (ret == -1)
 278                         errx(1, "kvm_nlist: %s", kvm_geterr(kd));
 279                 for (i = 0; i < NL_LAST_MANDATORY; i++) {
 280                         if (!nl[i].n_value) {
 281                                 quit = 1;
 282                                 warnx("undefined symbol: %s", nl[i].n_name);
 283                         }
 284                 }
 285                 if (quit)
 286                         exit(1);
 287         }
 288         if (!(fileflag | vnodeflag | ttyflag | swapflag | totalflag))
 289                 usage();
 290         if (fileflag || totalflag)
 291                 filemode();
 292         if (vnodeflag)
 293                 vnodemode();
 294         if (ttyflag)
 295                 ttymode();
 296         if (swapflag || totalflag)
 297                 swapmode();
 298         exit (0);
 299 }
 300
 301 static void
 302 usage(void)
 303 {
 304         fprintf(stderr, "usage: %s\n", usagestr);
 305         exit (1);
 306 }
 307
 308 struct e_vnode {
 309         struct vnode *avnode;
 310         struct vnode vnode;
 311 };
 312
 313 void
 314 vnodemode(void)
 315 {
 316         struct e_vnode *e_vnodebase, *endvnode, *evp;
 317         struct vnode *vp;
 318         struct mount *maddr, *mp = NULL;
 319         int numvnodes;
 320
 321         e_vnodebase = loadvnodes(&numvnodes);
 322         if (totalflag) {
 323                 printf("%7d vnodes\n", numvnodes);
 324                 return;
 325         }
 326         endvnode = e_vnodebase + numvnodes;
 327         printf("%d active vnodes\n", numvnodes);
 328
 329
 330 #define ST      mp->mnt_stat
 331         maddr = NULL;
 332         for (evp = e_vnodebase; evp < endvnode; evp++) {
 333                 vp = &evp->vnode;
 334                 if (vp->v_mount != maddr) {
 335                         /*
 336                          * New filesystem
 337                          */
 338                         if ((mp = getmnt(vp->v_mount)) == NULL)
 339                                 continue;
 340                         maddr = vp->v_mount;
 341                         mount_print(mp);
 342                         vnode_header();
 343                         if (!strcmp(ST.f_fstypename, "ufs") ||
 344                             !strcmp(ST.f_fstypename, "mfs"))
 345                                 ufs_header();
 346                         else if (!strcmp(ST.f_fstypename, "nfs"))
 347                                 nfs_header();
 348                         else if (!strcmp(ST.f_fstypename, "union"))
 349                                 union_header();
 350                         printf("\n");
 351                 }
 352                 vnode_print(evp->avnode, vp);
 353                 if (!strcmp(ST.f_fstypename, "ufs") ||
 354                     !strcmp(ST.f_fstypename, "mfs"))
 355                         ufs_print(vp);
 356                 else if (!strcmp(ST.f_fstypename, "nfs"))
 357                         nfs_print(vp);
 358                 else if (!strcmp(ST.f_fstypename, "union"))
 359                         union_print(vp);
 360                 printf("\n");
 361         }
 362         free(e_vnodebase);
 363 }
 364
 365 void
 366 vnode_header(void)
 367 {
 368         printf("ADDR     TYP VFLAG  USE HOLD");
 369 }
 370
 371 void
 372 vnode_print(struct vnode *avnode, struct vnode *vp)
 373 {
 374         const char *type;
 375         char flags[32];
 376         char *fp = flags;
 377         int flag;
 378         int refs;
 379
 380         /*
 381          * set type
 382          */
 383         switch (vp->v_type) {
 384         case VNON:
 385                 type = "non"; break;
 386         case VREG:
 387                 type = "reg"; break;
 388         case VDIR:
 389                 type = "dir"; break;
 390         case VBLK:
 391                 type = "blk"; break;
 392         case VCHR:
 393                 type = "chr"; break;
 394         case VLNK:
 395                 type = "lnk"; break;
 396         case VSOCK:
 397                 type = "soc"; break;
 398         case VFIFO:
 399                 type = "fif"; break;
 400         case VBAD:
 401                 type = "bad"; break;
 402         default:
 403                 type = "unk"; break;
 404         }
 405         /*
 406          * gather flags
 407          */
 408         flag = vp->v_flag;
 409         if (flag & VROOT)
 410                 *fp++ = 'R';
 411         if (flag & VTEXT)
 412                 *fp++ = 'T';
 413         if (flag & VSYSTEM)
 414                 *fp++ = 'S';
 415         if (flag & VISTTY)
 416                 *fp++ = 't';
 417 #ifdef VXLOCK
 418         if (flag & VXLOCK)
 419                 *fp++ = 'L';
 420         if (flag & VXWANT)
 421                 *fp++ = 'W';
 422 #endif
 423         if (flag & VOBJBUF)
 424                 *fp++ = 'V';
 425         if (flag & (VAGE0 | VAGE1))
 426                 *fp++ = 'a';
 427         if (flag & VOLOCK)
 428                 *fp++ = 'l';
 429         if (flag & VOWANT)
 430                 *fp++ = 'w';
 431 #ifdef VDOOMED
 432         if (flag & VDOOMED)
 433                 *fp++ = 'D';
 434 #endif
 435         if (flag & VFREE)
 436                 *fp++ = 'F';
 437         if (flag & VONWORKLST)
 438                 *fp++ = 'O';
 439 #ifdef VRECLAIMED
 440         if (flag & VINACTIVE)
 441                 *fp++ = 'I';
 442         if (flag & VRECLAIMED)
 443                 *fp++ = 'X';
 444 #endif
 445
 446         if (flag == 0)
 447                 *fp++ = '-';
 448         *fp = '\0';
 449
 450         /*
 451          * Convert SYSREF ref counts into something more
 452          * human readable for display.
 453          */
 454         if ((refs = vp->v_sysref.refcnt) < 0)
 455                 refs = -(refs + 0x40000001);
 456         printf("%8lx %s %5s %4d %4d",
 457             (u_long)(void *)avnode, type, flags, refs, vp->v_auxrefs);
 458 }
 459
 460 void
 461 ufs_header(void)
 462 {
 463         printf(" FILEID IFLAG RDEV|SZ");
 464 }
 465
 466 int
 467 ufs_print(struct vnode *vp)
 468 {
 469         int flag;
 470         struct inode inode, *ip = &inode;
 471         char flagbuf[16], *flags = flagbuf;
 472         char *name;
 473         mode_t type;
 474
 475         KGETRET(VTOI(vp), &inode, sizeof(struct inode), "vnode's inode");
 476         flag = ip->i_flag;
 477         if (flag & IN_ACCESS)
 478                 *flags++ = 'A';
 479         if (flag & IN_CHANGE)
 480                 *flags++ = 'C';
 481         if (flag & IN_UPDATE)
 482                 *flags++ = 'U';
 483         if (flag & IN_MODIFIED)
 484                 *flags++ = 'M';
 485         if (flag & IN_RENAME)
 486                 *flags++ = 'R';
 487         if (flag & IN_SHLOCK)
 488                 *flags++ = 'S';
 489         if (flag & IN_EXLOCK)
 490                 *flags++ = 'E';
 491         if (flag & IN_HASHED)
 492                 *flags++ = 'H';
 493         if (flag & IN_LAZYMOD)
 494                 *flags++ = 'L';
 495         if (flag == 0)
 496                 *flags++ = '-';
 497         *flags = '\0';
 498
 499         printf(" %6ju %5s", (uintmax_t)ip->i_number, flagbuf);
 500         type = ip->i_mode & S_IFMT;
 501         if (S_ISCHR(ip->i_mode) || S_ISBLK(ip->i_mode))
 502                 if (usenumflag || ((name = devname(ip->i_rdev, type)) == NULL))
 503                         printf("   %2d,%-2d",
 504                             major(ip->i_rdev), minor(ip->i_rdev));
 505                 else
 506                         printf(" %7s", name);
 507         else
 508                 printf(" %7qd", ip->i_size);
 509         return (0);
 510 }
 511
 512 void
 513 nfs_header(void)
 514 {
 515         printf(" FILEID NFLAG RDEV|SZ");
 516 }
 517
 518 int
 519 nfs_print(struct vnode *vp)
 520 {
 521         struct nfsnode nfsnode, *np = &nfsnode;
 522         char flagbuf[16], *flags = flagbuf;
 523         int flag;
 524         char *name;
 525         mode_t type;
 526
 527         KGETRET(VTONFS(vp), &nfsnode, sizeof(nfsnode), "vnode's nfsnode");
 528         flag = np->n_flag;
 529         if (flag & NFLUSHWANT)
 530                 *flags++ = 'W';
 531         if (flag & NFLUSHINPROG)
 532                 *flags++ = 'P';
 533         if (flag & NLMODIFIED)
 534                 *flags++ = 'M';
 535         if (flag & NRMODIFIED)
 536                 *flags++ = 'R';
 537         if (flag & NWRITEERR)
 538                 *flags++ = 'E';
 539         if (flag & NQNFSEVICTED)
 540                 *flags++ = 'G';
 541         if (flag & NACC)
 542                 *flags++ = 'A';
 543         if (flag & NUPD)
 544                 *flags++ = 'U';
 545         if (flag & NCHG)
 546                 *flags++ = 'C';
 547         if (flag & NLOCKED)
 548                 *flags++ = 'L';
 549         if (flag & NWANTED)
 550                 *flags++ = 'w';
 551         if (flag == 0)
 552                 *flags++ = '-';
 553         *flags = '\0';
 554
 555 #define VT      np->n_vattr
 556         printf(" %6ju %5s", (uintmax_t)VT.va_fileid, flagbuf);
 557         type = VT.va_mode & S_IFMT;
 558         if (S_ISCHR(VT.va_mode) || S_ISBLK(VT.va_mode))
 559                 if (usenumflag || ((name = devname((VT.va_rmajor << 8) | VT.va_rminor, type)) == NULL))
 560                         printf("   %2d,%-2d",
 561                             VT.va_rmajor, VT.va_rminor);
 562                 else
 563                         printf(" %7s", name);
 564         else
 565                 printf(" %7qd", np->n_size);
 566         return (0);
 567 }
 568
 569 void
 570 union_header(void)
 571 {
 572         printf("    UPPER    LOWER");
 573 }
 574
 575 int
 576 union_print(struct vnode *vp)
 577 {
 578         struct union_node unode, *up = &unode;
 579
 580         KGETRET(VTOUNION(vp), &unode, sizeof(unode), "vnode's unode");
 581
 582         printf(" %8lx %8lx", (u_long)(void *)up->un_uppervp,
 583             (u_long)(void *)up->un_lowervp);
 584         return (0);
 585 }
 586
 587 /*
 588  * Given a pointer to a mount structure in kernel space,
 589  * read it in and return a usable pointer to it.
 590  */
 591 struct mount *
 592 getmnt(struct mount *maddr)
 593 {
 594         static struct mtab {
 595                 struct mtab *next;
 596                 struct mount *maddr;
 597                 struct mount mount;
 598         } *mhead = NULL;
 599         struct mtab *mt;
 600
 601         for (mt = mhead; mt != NULL; mt = mt->next)
 602                 if (maddr == mt->maddr)
 603                         return (&mt->mount);
 604         if ((mt = malloc(sizeof(struct mtab))) == NULL)
 605                 errx(1, "malloc");
 606         KGETRET(maddr, &mt->mount, sizeof(struct mount), "mount table");
 607         mt->maddr = maddr;
 608         mt->next = mhead;
 609         mhead = mt;
 610         return (&mt->mount);
 611 }
 612
 613 void
 614 mount_print(struct mount *mp)
 615 {
 616         int flags;
 617
 618 #define ST      mp->mnt_stat
 619         printf("*** MOUNT %s %s on %s", ST.f_fstypename,
 620             ST.f_mntfromname, ST.f_mntonname);
 621         if ((flags = mp->mnt_flag)) {
 622                 int i;
 623                 const char *sep = " (";
 624
 625                 for (i = 0; mnt_flags[i].m_flag; i++) {
 626                         if (flags & mnt_flags[i].m_flag) {
 627                                 printf("%s%s", sep, mnt_flags[i].m_name);
 628                                 flags &= ~mnt_flags[i].m_flag;
 629                                 sep = ",";
 630                         }
 631                 }
 632                 if (flags)
 633                         printf("%sunknown_flags:%x", sep, flags);
 634                 printf(")");
 635         }
 636         printf("\n");
 637 #undef ST
 638 }
 639
 640 struct e_vnode *
 641 loadvnodes(int *avnodes)
 642 {
 643         int mib[2];
 644         size_t copysize;
 645         struct e_vnode *vnodebase;
 646
 647         if (memf != NULL) {
 648                 /*
 649                  * do it by hand
 650                  */
 651                 return (kinfo_vnodes(avnodes));
 652         }
 653         mib[0] = CTL_KERN;
 654         mib[1] = KERN_VNODE;
 655         if (sysctl(mib, 2, NULL, &copysize, NULL, 0) == -1)
 656                 err(1, "sysctl: KERN_VNODE");
 657         if ((vnodebase = malloc(copysize)) == NULL)
 658                 errx(1, "malloc");
 659         if (sysctl(mib, 2, vnodebase, &copysize, NULL, 0) == -1)
 660                 err(1, "sysctl: KERN_VNODE");
 661         if (copysize % sizeof(struct e_vnode))
 662                 errx(1, "vnode size mismatch");
 663         *avnodes = copysize / sizeof(struct e_vnode);
 664
 665         return (vnodebase);
 666 }
 667
 668 /*
 669  * simulate what a running kernel does in in kinfo_vnode
 670  */
 671 struct e_vnode *
 672 kinfo_vnodes(int *avnodes)
 673 {
 674         struct mntlist mountlist;
 675         struct mount *mp, mounth, *mp_next;
 676         struct vnode *vp, vnode, *vp_next;
 677         char *vbuf, *evbuf, *bp;
 678         int num, numvnodes;
 679
 680 #define VPTRSZ  sizeof(struct vnode *)
 681 #define VNODESZ sizeof(struct vnode)
 682
 683         KGET(NL_NUMVNODES, numvnodes);
 684         if ((vbuf = malloc((numvnodes + 20) * (VPTRSZ + VNODESZ))) == NULL)
 685                 errx(1, "malloc");
 686         bp = vbuf;
 687         evbuf = vbuf + (numvnodes + 20) * (VPTRSZ + VNODESZ);
 688         KGET(NL_MOUNTLIST, mountlist);
 689         for (num = 0, mp = TAILQ_FIRST(&mountlist); ; mp = mp_next) {
 690                 KGET2(mp, &mounth, sizeof(mounth), "mount entry");
 691                 mp_next = TAILQ_NEXT(&mounth, mnt_list);
 692                 for (vp = TAILQ_FIRST(&mounth.mnt_nvnodelist);
 693                     vp != NULL; vp = vp_next) {
 694                         KGET2(vp, &vnode, sizeof(vnode), "vnode");
 695                         vp_next = TAILQ_NEXT(&vnode, v_nmntvnodes);
 696                         if ((bp + VPTRSZ + VNODESZ) > evbuf)
 697                                 /* XXX - should realloc */
 698                                 errx(1, "no more room for vnodes");
 699                         memmove(bp, &vp, VPTRSZ);
 700                         bp += VPTRSZ;
 701                         memmove(bp, &vnode, VNODESZ);
 702                         bp += VNODESZ;
 703                         num++;
 704                 }
 705                 if (mp == TAILQ_LAST(&mountlist, mntlist))
 706                         break;
 707         }
 708         *avnodes = num;
 709         return ((struct e_vnode *)vbuf);
 710 }
 711
 712 char hdr[] =
 713 "  LINE RAW CAN OUT IHIWT ILOWT OHWT LWT     COL STATE  SESS      PGID DISC\n";
 714 int ttyspace = 128;
 715
 716 void
 717 ttymode(void)
 718 {
 719         struct tty *tty;
 720         struct tty ttyb[1000];
 721         int error;
 722         size_t len, i;
 723
 724         printf(hdr);
 725         len = sizeof(ttyb);
 726         error = sysctlbyname("kern.ttys", &ttyb, &len, 0, 0);
 727         if (!error) {
 728                 len /= sizeof(ttyb[0]);
 729                 for (i = 0; i < len; i++) {
 730                         ttyprt(&ttyb[i], 0);
 731                 }
 732         }
 733         if ((tty = malloc(ttyspace * sizeof(*tty))) == NULL)
 734                 errx(1, "malloc");
 735         if (nl[NL_NRC_TTY].n_type != 0)
 736                 ttytype(tty, "rc", NL_RC_TTY, NL_NRC_TTY, 0);
 737         if (nl[NL_NCY_TTY].n_type != 0)
 738                 ttytype(tty, "cy", NL_CY_TTY, NL_NCY_TTY, 0);
 739         if (nl[NL_SI_NPORTS].n_type != 0)
 740                 ttytype(tty, "si", NL_SI_TTY, NL_SI_NPORTS, 1);
 741         free(tty);
 742 }
 743
 744 void
 745 ttytype(struct tty *tty, const char *name, int type, int number, int indir)
 746 {
 747         struct tty *tp;
 748         int ntty;
 749         struct tty **ttyaddr;
 750
 751         if (tty == NULL)
 752                 return;
 753         KGET(number, ntty);
 754         printf("%d %s %s\n", ntty, name, (ntty == 1) ? "line" : "lines");
 755         if (ntty > ttyspace) {
 756                 ttyspace = ntty;
 757                 if ((tty = realloc(tty, ttyspace * sizeof(*tty))) == NULL)
 758                         errx(1, "realloc");
 759         }
 760         if (indir) {
 761                 KGET(type, ttyaddr);
 762                 KGET2(ttyaddr, tty, (ssize_t)(ntty * sizeof(struct tty)),
 763                       "tty structs");
 764         } else {
 765                 KGET1(type, tty, (ssize_t)(ntty * sizeof(struct tty)),
 766                       "tty structs");
 767         }
 768         printf(hdr);
 769         for (tp = tty; tp < &tty[ntty]; tp++)
 770                 ttyprt(tp, tp - tty);
 771 }
 772
 773 struct {
 774         int flag;
 775         char val;
 776 } ttystates[] = {
 777 #ifdef TS_WOPEN
 778         { TS_WOPEN,     'W'},
 779 #endif
 780         { TS_ISOPEN,    'O'},
 781         { TS_CARR_ON,   'C'},
 782 #ifdef TS_CONNECTED
 783         { TS_CONNECTED, 'c'},
 784 #endif
 785         { TS_TIMEOUT,   'T'},
 786         { TS_FLUSH,     'F'},
 787         { TS_BUSY,      'B'},
 788 #ifdef TS_ASLEEP
 789         { TS_ASLEEP,    'A'},
 790 #endif
 791 #ifdef TS_SO_OLOWAT
 792         { TS_SO_OLOWAT, 'A'},
 793 #endif
 794 #ifdef TS_SO_OCOMPLETE
 795         { TS_SO_OCOMPLETE, 'a'},
 796 #endif
 797         { TS_XCLUDE,    'X'},
 798         { TS_TTSTOP,    'S'},
 799 #ifdef TS_CAR_OFLOW
 800         { TS_CAR_OFLOW, 'm'},
 801 #endif
 802 #ifdef TS_CTS_OFLOW
 803         { TS_CTS_OFLOW, 'o'},
 804 #endif
 805 #ifdef TS_DSR_OFLOW
 806         { TS_DSR_OFLOW, 'd'},
 807 #endif
 808         { TS_TBLOCK,    'K'},
 809         { TS_ASYNC,     'Y'},
 810         { TS_BKSL,      'D'},
 811         { TS_ERASE,     'E'},
 812         { TS_LNCH,      'L'},
 813         { TS_TYPEN,     'P'},
 814         { TS_CNTTB,     'N'},
 815 #ifdef TS_CAN_BYPASS_L_RINT
 816         { TS_CAN_BYPASS_L_RINT, 'l'},
 817 #endif
 818 #ifdef TS_SNOOP
 819         { TS_SNOOP,     's'},
 820 #endif
 821 #ifdef TS_ZOMBIE
 822         { TS_ZOMBIE,    'Z'},
 823 #endif
 824         { 0,           '\0'},
 825 };
 826
 827 void
 828 ttyprt(struct tty *tp, int line)
 829 {
 830         int i, j;
 831         pid_t pgid;
 832         char *name, state[20];
 833
 834         if (usenumflag || tp->t_dev == 0 ||
 835            (name = devname(tp->t_dev, S_IFCHR)) == NULL)
 836                 printf("%7d ", line);
 837         else
 838                 printf("%7s ", name);
 839         printf("%2d %3d ", tp->t_rawq.c_cc, tp->t_canq.c_cc);
 840         printf("%3d %5d %5d %4d %3d %7d ", tp->t_outq.c_cc,
 841                 tp->t_ihiwat, tp->t_ilowat, tp->t_ohiwat, tp->t_olowat,
 842                 tp->t_column);
 843         for (i = j = 0; ttystates[i].flag; i++)
 844                 if (tp->t_state&ttystates[i].flag)
 845                         state[j++] = ttystates[i].val;
 846         if (j == 0)
 847                 state[j++] = '-';
 848         state[j] = '\0';
 849         printf("%-6s %8lx", state, (u_long)(void *)tp->t_session);
 850         pgid = 0;
 851         if (tp->t_pgrp != NULL)
 852                 KGET2(&tp->t_pgrp->pg_id, &pgid, sizeof(pid_t), "pgid");
 853         printf("%6d ", pgid);
 854         switch (tp->t_line) {
 855         case TTYDISC:
 856                 printf("term\n");
 857                 break;
 858         case NTTYDISC:
 859                 printf("ntty\n");
 860                 break;
 861         case SLIPDISC:
 862                 printf("slip\n");
 863                 break;
 864         case PPPDISC:
 865                 printf("ppp\n");
 866                 break;
 867         default:
 868                 printf("%d\n", tp->t_line);
 869                 break;
 870         }
 871 }
 872
 873 void
 874 filemode(void)
 875 {
 876         struct kinfo_file *fp, *ofp;
 877         size_t len;
 878         char flagbuf[16], *fbp;
 879         int maxfile, nfile;
 880         static const char *dtypes[] = { "???", "inode", "socket" };
 881
 882         if (kinfo_get_maxfiles(&maxfile))
 883                 err(1, "kinfo_get_maxfiles");
 884         if (totalflag) {
 885                 if (kinfo_get_openfiles(&nfile))
 886                         err(1, "kinfo_get_openfiles");
 887                 printf("%3d/%3d files\n", nfile, maxfile);
 888                 return;
 889         }
 890         if (kinfo_get_files(&fp, &len))
 891                 err(1, "kinfo_get_files");
 892         ofp = fp;
 893
 894         printf("%d/%d open files\n", len, maxfile);
 895         printf("   LOC   TYPE    FLG     CNT  MSG    DATA    OFFSET\n");
 896         for (; len-- > 0; fp++) {
 897                 if ((unsigned)fp->f_type > DTYPE_SOCKET)
 898                         continue;
 899                 printf("%p ", fp->f_file);
 900                 printf("%-8.8s", dtypes[fp->f_type]);
 901                 fbp = flagbuf;
 902                 if (fp->f_flag & FREAD)
 903                         *fbp++ = 'R';
 904                 if (fp->f_flag & FWRITE)
 905                         *fbp++ = 'W';
 906                 if (fp->f_flag & FAPPEND)
 907                         *fbp++ = 'A';
 908 #ifdef FSHLOCK  /* currently gone */
 909                 if (fp->f_flag & FSHLOCK)
 910                         *fbp++ = 'S';
 911                 if (fp->f_flag & FEXLOCK)
 912                         *fbp++ = 'X';
 913 #endif
 914                 if (fp->f_flag & FASYNC)
 915                         *fbp++ = 'I';
 916                 *fbp = '\0';
 917                 printf("%6s  %3d", flagbuf, fp->f_count);
 918                 printf("  %3d", fp->f_msgcount);
 919                 printf("  %8lx", (u_long)(void *)fp->f_data);
 920                 if (fp->f_offset < 0)
 921                         printf("  %qx\n", fp->f_offset);
 922                 else
 923                         printf("  %qd\n", fp->f_offset);
 924         }
 925         free(ofp);
 926 }
 927
 928 /*
 929  * swapmode is based on a program called swapinfo written
 930  * by Kevin Lahey <kml@rokkaku.atl.ga.us>.
 931  */
 932 void
 933 swapmode(void)
 934 {
 935         struct kvm_swap kswap[16];
 936         int i;
 937         int n;
 938         int pagesize = getpagesize();
 939         const char *header;
 940         int hlen;
 941         long blocksize;
 942
 943         n = kvm_getswapinfo(
 944             kd,
 945             kswap,
 946             sizeof(kswap)/sizeof(kswap[0]),
 947             ((swapflag > 1) ? SWIF_DUMP_TREE : 0) | SWIF_DEV_PREFIX
 948         );
 949
 950 #define CONVERT(v)      ((int)((quad_t)(v) * pagesize / blocksize))
 951
 952         header = getbsize(&hlen, &blocksize);
 953         if (totalflag == 0) {
 954                 printf("%-15s %*s %8s %8s %8s  %s\n",
 955                     "Device", hlen, header,
 956                     "Used", "Avail", "Capacity", "Type");
 957
 958                 for (i = 0; i < n; ++i) {
 959                         printf(
 960                             "%-15s %*d ",
 961                             kswap[i].ksw_devname,
 962                             hlen,
 963                             CONVERT(kswap[i].ksw_total)
 964                         );
 965                         printf(
 966                             "%8d %8d %5.0f%%    %s\n",
 967                             CONVERT(kswap[i].ksw_used),
 968                             CONVERT(kswap[i].ksw_total - kswap[i].ksw_used),
 969                             (double)kswap[i].ksw_used * 100.0 /
 970                                 (double)kswap[i].ksw_total,
 971                             (kswap[i].ksw_flags & SW_SEQUENTIAL) ?
 972                                 "Sequential" : "Interleaved"
 973                         );
 974                 }
 975         }
 976
 977         if (totalflag) {
 978                 blocksize = 1024 * 1024;
 979
 980                 printf(
 981                     "%dM/%dM swap space\n",
 982                     CONVERT(kswap[n].ksw_used),
 983                     CONVERT(kswap[n].ksw_total)
 984                 );
 985         } else if (n > 1) {
 986                 printf(
 987                     "%-15s %*d %8d %8d %5.0f%%\n",
 988                     "Total",
 989                     hlen,
 990                     CONVERT(kswap[n].ksw_total),
 991                     CONVERT(kswap[n].ksw_used),
 992                     CONVERT(kswap[n].ksw_total - kswap[n].ksw_used),
 993                     (double)kswap[n].ksw_used * 100.0 /
 994                         (double)kswap[n].ksw_total
 995                 );
 996         }
 997 }