lib/libprocstat/common_kvm.c

   1 /*-
   2  * SPDX-License-Identifier: BSD-4-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. All advertising materials mentioning features or use of this software
  17  *    must display the following acknowledgement:
  18  *      This product includes software developed by the University of
  19  *      California, Berkeley and its contributors.
  20  * 4. Neither the name of the University nor the names of its contributors
  21  *    may be used to endorse or promote products derived from this software
  22  *    without specific prior written permission.
  23  *
  24  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  25  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  26  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  27  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  28  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  29  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  30  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  31  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  32  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  33  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  34  * SUCH DAMAGE.
  35  */
  36
  37 #include <sys/cdefs.h>
  38 __FBSDID("$FreeBSD$");
  39
  40 #include <sys/param.h>
  41 #include <sys/user.h>
  42 #include <sys/stat.h>
  43 #include <sys/vnode.h>
  44 #include <sys/conf.h>
  45 #include <sys/pipe.h>
  46 #define _WANT_MOUNT
  47 #include <sys/mount.h>
  48 #include <ufs/ufs/quota.h>
  49 #include <ufs/ufs/inode.h>
  50 #include <ufs/ufs/extattr.h>
  51 #include <ufs/ufs/ufsmount.h>
  52 #include <fs/devfs/devfs.h>
  53 #include <fs/devfs/devfs_int.h>
  54 #include <nfs/nfsproto.h>
  55 #include <nfsclient/nfs.h>
  56 #include <nfsclient/nfsnode.h>
  57
  58 #include <assert.h>
  59 #include <err.h>
  60 #include <kvm.h>
  61 #include <stddef.h>
  62 #include <string.h>
  63
  64 #include <libprocstat.h>
  65 #include "common_kvm.h"
  66
  67 int
  68 kvm_read_all(kvm_t *kd, unsigned long addr, void *buf, size_t nbytes)
  69 {
  70         ssize_t error;
  71
  72         if (nbytes >= SSIZE_MAX)
  73                 return (0);
  74         error = kvm_read(kd, addr, buf, nbytes);
  75         return (error == (ssize_t)(nbytes));
  76 }
  77
  78 int
  79 kdevtoname(kvm_t *kd, struct cdev *dev, char *buf)
  80 {
  81         struct cdev si;
  82
  83         assert(buf);
  84         if (!kvm_read_all(kd, (unsigned long)dev, &si, sizeof(si)))
  85                 return (1);
  86         strlcpy(buf, si.si_name, SPECNAMELEN + 1);
  87         return (0);
  88 }
  89
  90 int
  91 ufs_filestat(kvm_t *kd, struct vnode *vp, struct vnstat *vn)
  92 {
  93         struct inode inode;
  94         struct ufsmount um;
  95
  96         if (!kvm_read_all(kd, (unsigned long)VTOI(vp), &inode, sizeof(inode))) {
  97                 warnx("can't read inode at %p", (void *)VTOI(vp));
  98                 return (1);
  99         }
 100         if (!kvm_read_all(kd, (unsigned long)inode.i_ump, &um, sizeof(um))) {
 101                 warnx("can't read ufsmount at %p", (void *)inode.i_ump);
 102                 return (1);
 103         }
 104         /*
 105          * The st_dev from stat(2) is a dev_t. These kernel structures
 106          * contain cdev pointers. We need to convert to dev_t to make
 107          * comparisons
 108          */
 109         vn->vn_fsid = dev2udev(kd, um.um_dev);
 110         vn->vn_fileid = inode.i_number;
 111         vn->vn_mode = (mode_t)inode.i_mode;
 112         vn->vn_size = inode.i_size;
 113         return (0);
 114 }
 115
 116 int
 117 devfs_filestat(kvm_t *kd, struct vnode *vp, struct vnstat *vn)
 118 {
 119         struct devfs_dirent devfs_dirent;
 120         struct mount mount;
 121
 122         if (!kvm_read_all(kd, (unsigned long)getvnodedata(vp), &devfs_dirent,
 123             sizeof(devfs_dirent))) {
 124                 warnx("can't read devfs_dirent at %p",
 125                     (void *)vp->v_data);
 126                 return (1);
 127         }
 128         if (!kvm_read_all(kd, (unsigned long)getvnodemount(vp), &mount,
 129             sizeof(mount))) {
 130                 warnx("can't read mount at %p",
 131                     (void *)getvnodemount(vp));
 132                 return (1);
 133         }
 134         vn->vn_fsid = mount.mnt_stat.f_fsid.val[0];
 135         vn->vn_fileid = devfs_dirent.de_inode;
 136         vn->vn_mode = (devfs_dirent.de_mode & ~S_IFMT) | S_IFCHR;
 137         vn->vn_size = 0;
 138         return (0);
 139 }
 140
 141 int
 142 nfs_filestat(kvm_t *kd, struct vnode *vp, struct vnstat *vn)
 143 {
 144         struct nfsnode nfsnode;
 145         mode_t mode;
 146
 147         if (!kvm_read_all(kd, (unsigned long)VTONFS(vp), &nfsnode,
 148             sizeof(nfsnode))) {
 149                 warnx("can't read nfsnode at %p",
 150                     (void *)VTONFS(vp));
 151                 return (1);
 152         }
 153         vn->vn_fsid = nfsnode.n_vattr.va_fsid;
 154         vn->vn_fileid = nfsnode.n_vattr.va_fileid;
 155         vn->vn_size = nfsnode.n_size;
 156         mode = (mode_t)nfsnode.n_vattr.va_mode;
 157         switch (vp->v_type) {
 158         case VREG:
 159                 mode |= S_IFREG;
 160                 break;
 161         case VDIR:
 162                 mode |= S_IFDIR;
 163                 break;
 164         case VBLK:
 165                 mode |= S_IFBLK;
 166                 break;
 167         case VCHR:
 168                 mode |= S_IFCHR;
 169                 break;
 170         case VLNK:
 171                 mode |= S_IFLNK;
 172                 break;
 173         case VSOCK:
 174                 mode |= S_IFSOCK;
 175                 break;
 176         case VFIFO:
 177                 mode |= S_IFIFO;
 178                 break;
 179         default:
 180                 break;
 181         };
 182         vn->vn_mode = mode;
 183         return (0);
 184 }
 185
 186 /*
 187  * Read the cdev structure in the kernel in order to work out the
 188  * associated dev_t
 189  */
 190 dev_t
 191 dev2udev(kvm_t *kd, struct cdev *dev)
 192 {
 193         struct cdev_priv priv;
 194
 195         assert(kd);
 196         if (kvm_read_all(kd, (unsigned long)cdev2priv(dev), &priv,
 197             sizeof(priv))) {
 198                 return ((dev_t)priv.cdp_inode);
 199         } else {
 200                 warnx("can't convert cdev *%p to a dev_t\n", dev);
 201                 return (-1);
 202         }
 203 }
 204
 205 void *
 206 getvnodedata(struct vnode *vp)
 207 {
 208         return (vp->v_data);
 209 }
 210
 211 struct mount *
 212 getvnodemount(struct vnode *vp)
 213 {
 214         return (vp->v_mount);
 215 }