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