/*-
 * SPDX-License-Identifier: BSD-4-Clause
 *
 * Copyright (c) 2009 Stanislav Sedov <stas@FreeBSD.org>
 * Copyright (c) 1988, 1993
 *      The Regents of the University of California.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *      This product includes software developed by the University of
 *      California, Berkeley and its contributors.
 * 4. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <sys/param.h>
#include <sys/user.h>
#include <sys/stat.h>
#include <sys/vnode.h>
#include <sys/conf.h>
#include <sys/pipe.h>
#define _WANT_MOUNT
#include <sys/mount.h>
#include <ufs/ufs/quota.h>
#include <ufs/ufs/inode.h>
#include <ufs/ufs/extattr.h>
#include <ufs/ufs/ufsmount.h>
#include <fs/devfs/devfs.h>
#include <fs/devfs/devfs_int.h>
#include <nfs/nfsproto.h>
#include <nfsclient/nfs.h>
#include <nfsclient/nfsnode.h>

#include <assert.h>
#include <err.h>
#include <kvm.h>
#include <stddef.h>
#include <string.h>

#include <libprocstat.h>
#include "common_kvm.h"

int
kvm_read_all(kvm_t *kd, unsigned long addr, void *buf, size_t nbytes)
{
        ssize_t error;

        if (nbytes >= SSIZE_MAX)
                return (0);
        error = kvm_read(kd, addr, buf, nbytes);
        return (error == (ssize_t)(nbytes));
}

int
kdevtoname(kvm_t *kd, struct cdev *dev, char *buf)
{
        struct cdev si;

        assert(buf);
        if (!kvm_read_all(kd, (unsigned long)dev, &si, sizeof(si)))
                return (1);
        strlcpy(buf, si.si_name, SPECNAMELEN + 1);
        return (0);
}

int
ufs_filestat(kvm_t *kd, struct vnode *vp, struct vnstat *vn)
{
        struct inode inode;
        struct ufsmount um;

        if (!kvm_read_all(kd, (unsigned long)VTOI(vp), &inode, sizeof(inode))) {
                warnx("can't read inode at %p", (void *)VTOI(vp));
                return (1);
        }
        if (!kvm_read_all(kd, (unsigned long)inode.i_ump, &um, sizeof(um))) {
                warnx("can't read ufsmount at %p", (void *)inode.i_ump);
                return (1);
        }
        /*
         * The st_dev from stat(2) is a dev_t. These kernel structures
         * contain cdev pointers. We need to convert to dev_t to make
         * comparisons
         */
        vn->vn_fsid = dev2udev(kd, um.um_dev);
        vn->vn_fileid = inode.i_number;
        vn->vn_mode = (mode_t)inode.i_mode;
        vn->vn_size = inode.i_size;
        return (0);
}

int
devfs_filestat(kvm_t *kd, struct vnode *vp, struct vnstat *vn)
{
        struct devfs_dirent devfs_dirent;
        struct mount mount;

        if (!kvm_read_all(kd, (unsigned long)getvnodedata(vp), &devfs_dirent,
            sizeof(devfs_dirent))) {
                warnx("can't read devfs_dirent at %p",
                    (void *)vp->v_data);
                return (1);
        }
        if (!kvm_read_all(kd, (unsigned long)getvnodemount(vp), &mount,
            sizeof(mount))) {
                warnx("can't read mount at %p",
                    (void *)getvnodemount(vp));
                return (1);
        }
        vn->vn_fsid = mount.mnt_stat.f_fsid.val[0];
        vn->vn_fileid = devfs_dirent.de_inode;
        vn->vn_mode = (devfs_dirent.de_mode & ~S_IFMT) | S_IFCHR;
        vn->vn_size = 0;
        return (0);
}

int
nfs_filestat(kvm_t *kd, struct vnode *vp, struct vnstat *vn)
{
        struct nfsnode nfsnode;
        mode_t mode;

        if (!kvm_read_all(kd, (unsigned long)VTONFS(vp), &nfsnode,
            sizeof(nfsnode))) {
                warnx("can't read nfsnode at %p",
                    (void *)VTONFS(vp));
                return (1);
        }
        vn->vn_fsid = nfsnode.n_vattr.va_fsid;
        vn->vn_fileid = nfsnode.n_vattr.va_fileid;
        vn->vn_size = nfsnode.n_size;
        mode = (mode_t)nfsnode.n_vattr.va_mode;
        switch (vp->v_type) {
        case VREG:
                mode |= S_IFREG;
                break;
        case VDIR:
                mode |= S_IFDIR;
                break;
        case VBLK:
                mode |= S_IFBLK;
                break;
        case VCHR:
                mode |= S_IFCHR;
                break;
        case VLNK:
                mode |= S_IFLNK;
                break;
        case VSOCK:
                mode |= S_IFSOCK;
                break;
        case VFIFO:
                mode |= S_IFIFO;
                break;
        default:
                break;
        };
        vn->vn_mode = mode;
        return (0);
}

/*
 * Read the cdev structure in the kernel in order to work out the
 * associated dev_t
 */
dev_t
dev2udev(kvm_t *kd, struct cdev *dev)
{
        struct cdev_priv priv;

        assert(kd);
        if (kvm_read_all(kd, (unsigned long)cdev2priv(dev), &priv,
            sizeof(priv))) {
                return ((dev_t)priv.cdp_inode);
        } else {
                warnx("can't convert cdev *%p to a dev_t\n", dev);
                return (-1);
        }
}

void *
getvnodedata(struct vnode *vp)
{
        return (vp->v_data);
}

struct mount *
getvnodemount(struct vnode *vp)
{
        return (vp->v_mount);
}