/*	$NetBSD: fstat.c,v 1.117 2022/10/28 05:27:16 ozaki-r Exp $	*/

/*-
 * 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. 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/cdefs.h>
#ifndef lint
__COPYRIGHT("@(#) Copyright (c) 1988, 1993\
 The Regents of the University of California.  All rights reserved.");
#endif /* not lint */

#ifndef lint
#if 0
static char sccsid[] = "@(#)fstat.c	8.3 (Berkeley) 5/2/95";
#else
__RCSID("$NetBSD: fstat.c,v 1.117 2022/10/28 05:27:16 ozaki-r Exp $");
#endif
#endif /* not lint */

#include <sys/types.h>
#include <sys/param.h>
#include <sys/time.h>
#include <sys/proc.h>
#include <sys/stat.h>
#include <sys/vnode.h>
#include <sys/socket.h>
#include <sys/socketvar.h>
#include <sys/domain.h>
#include <sys/protosw.h>
#include <sys/unpcb.h>
#include <sys/sysctl.h>
#include <sys/filedesc.h>
#include <sys/pipe.h>
#define _KERNEL
#include <sys/mount.h>
#undef _KERNEL
#define	_KERNEL
#include <sys/file.h>
#include <ufs/ufs/inode.h>
#include <ufs/ufs/ufsmount.h>
#undef _KERNEL
#define NFS
#include <nfs/nfsproto.h>
#include <nfs/rpcv2.h>
#include <nfs/nfs.h>
#include <nfs/nfsnode.h>
#undef NFS
#include <msdosfs/denode.h>
#include <msdosfs/bpb.h>
#define	_KERNEL
#include <msdosfs/msdosfsmount.h>
#undef _KERNEL
#define	_KERNEL
#include <miscfs/genfs/layer.h>
#undef _KERNEL

#include <net/route.h>
#include <netinet/in.h>
#include <netinet/in_systm.h>
#include <netinet/ip.h>
#include <netinet/in_pcb.h>

#ifdef INET6
#include <netinet/ip6.h>
#include <netinet6/in6.h>
#include <netinet6/ip6_var.h>
#include <netinet6/in6_pcb.h>
#endif

#include <netatalk/at.h>
#include <netatalk/ddp_var.h>

#include <netdb.h>
#include <arpa/inet.h>

#include <ctype.h>
#include <errno.h>
#include <kvm.h>
#include <limits.h>
#include <nlist.h>
#include <paths.h>
#include <pwd.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <err.h>
#include <util.h>

#include "fstat.h"

#define	TEXT	-1
#define	CDIR	-2
#define	RDIR	-3
#define	TRACE	-4

typedef struct devs {
	struct	devs *next;
	long	fsid;
	ino_t	ino;
	const char *name;
} DEVS;
static DEVS *devs;

static int 	fsflg,	/* show files on same filesystem as file(s) argument */
	pflg,	/* show files open by a particular pid */
	uflg;	/* show files open by a particular (effective) user */
static int 	checkfile; /* true if restricting to particular files or filesystems */
static int	nflg;	/* (numerical) display f.s. and rdev as dev_t */
static int	Aflg;	/* prefix with address of file structure */
static int	Oflg;	/* print offset instead of size */
int	vflg;	/* display errors in locating kernel data objects etc... */

static fdfile_t **ofiles; /* buffer of pointers to file structures */
static int fstat_maxfiles;
#define ALLOC_OFILES(d)	\
	if ((d) > fstat_maxfiles) { \
		size_t len = (d) * sizeof(fdfile_t *); \
		free(ofiles); \
		ofiles = malloc(len); \
		if (ofiles == NULL) { \
			err(1, "malloc(%zu)", len);	\
		} \
		fstat_maxfiles = (d); \
	}

kvm_t *kd;

static const char *const dtypes[] = {
	DTYPE_NAMES
};

static void	dofiles(struct kinfo_proc2 *);
static int	ext2fs_filestat(struct vnode *, struct filestat *);
static int	getfname(const char *);
static void	getinetproto(char *, size_t, int);
static void	getatproto(char *, size_t, int);
static char   *getmnton(struct mount *);
static const char   *layer_filestat(struct vnode *, struct filestat *);
static int	msdosfs_filestat(struct vnode *, struct filestat *);
static int	nfs_filestat(struct vnode *, struct filestat *);
static const char *inet_addrstr(char *, size_t, const struct in_addr *,
    uint16_t, bool);
#ifdef INET6
static const char *inet6_addrstr(char *, size_t, const struct in6_addr *,
    uint16_t, bool);
#endif
static const char *at_addrstr(char *, size_t, const struct sockaddr_at *);
static void	socktrans(struct file *, struct socket *, int);
static void	misctrans(struct file *, int);
static int	ufs_filestat(struct vnode *, struct filestat *);
static void	usage(void) __dead;
static const char   *vfilestat(struct vnode *, struct filestat *);
static void	vtrans(struct file *, struct vnode *, int, int, long);
static void	ftrans(fdfile_t *, int);
static void	ptrans(struct file *, struct pipe *, int);
static void	kdriver_init(void);
static void	check_privs(void);

int
main(int argc, char **argv)
{
	struct passwd *passwd;
	struct kinfo_proc2 *p, *plast;
	int arg, ch, what;
	char *memf, *nlistf;
	char buf[_POSIX2_LINE_MAX];
	int cnt;
	gid_t egid = getegid();

	(void)setegid(getgid());
	arg = 0;
	what = KERN_PROC_ALL;
	nlistf = memf = NULL;
	while ((ch = getopt(argc, argv, "fnAOp:u:vN:M:")) != -1)
		switch((char)ch) {
		case 'f':
			fsflg = 1;
			break;
		case 'M':
			memf = optarg;
			break;
		case 'N':
			nlistf = optarg;
			break;
		case 'n':
			nflg = 1;
			break;
		case 'A':
			Aflg = 1;
			break;
		case 'O':
			Oflg = 1;
			break;
		case 'p':
			if (pflg++)
				usage();
			if (!isdigit((unsigned char)*optarg)) {
				warnx("-p requires a process id");
				usage();
			}
			what = KERN_PROC_PID;
			arg = atoi(optarg);
			break;
		case 'u':
			if (uflg++)
				usage();
			if (!(passwd = getpwnam(optarg))) {
				errx(1, "%s: unknown uid", optarg);
			}
			what = KERN_PROC_UID;
			arg = passwd->pw_uid;
			break;
		case 'v':
			vflg = 1;
			break;
		case '?':
		default:
			usage();
		}

	check_privs();

	kdriver_init();

	if (*(argv += optind)) {
		for (; *argv; ++argv) {
			if (getfname(*argv))
				checkfile = 1;
		}
		if (!checkfile)	/* file(s) specified, but none accessible */
			exit(1);
	}

	ALLOC_OFILES(256);	/* reserve space for file pointers */

	if (fsflg && !checkfile) {	
		/* -f with no files means use wd */
		if (getfname(".") == 0)
			exit(1);
		checkfile = 1;
	}

	/*
	 * Discard setgid privileges.  If not the running kernel, we toss
	 * them away totally so that bad guys can't print interesting stuff
	 * from kernel memory, otherwise switch back to kmem for the
	 * duration of the kvm_openfiles() call.
	 */
	if (nlistf != NULL || memf != NULL)
		(void)setgid(getgid());
	else
		(void)setegid(egid);

	if ((kd = kvm_openfiles(nlistf, memf, NULL, O_RDONLY, buf)) == NULL)
		errx(1, "%s", buf);

	/* get rid of it now anyway */
	if (nlistf == NULL && memf == NULL)
		(void)setgid(getgid());

	if ((p = kvm_getproc2(kd, what, arg, sizeof *p, &cnt)) == NULL) {
		errx(1, "%s", kvm_geterr(kd));
	}
	if (Aflg)
		(void)printf("%-*s ", 2*(int)(sizeof(void*)), "ADDR");
	if (nflg)
		(void)printf(
"USER     CMD          PID   FD    DEV       INUM   MODE  %s R/W",
    Oflg ? " OFFS" : "SZ|DV" );
	else
		(void)printf(
"USER     CMD          PID   FD  MOUNT         INUM MODE         %s R/W",
    Oflg ? " OFFS" : "SZ|DV" );

	if (checkfile && fsflg == 0)
		(void)printf(" NAME\n");
	else
		(void)putchar('\n');

	for (plast = &p[cnt]; p < plast; ++p) {
		if (p->p_stat == SZOMB)
			continue;
		dofiles(p);
	}
	return 0;
}

static void
check_privs(void)
{
	int expaddr;
	size_t expsize = sizeof(expaddr);
	const char *expname = "kern.expose_address";

	if (geteuid() == 0)
		return;

	if (sysctlbyname(expname, &expaddr, &expsize, NULL, 0) == -1)
		err(EXIT_FAILURE, "Can't get sysctl `%s'", expname);
	if (expaddr == 0)
		errx(EXIT_FAILURE, "This program does not work without "
		    "sysctl `%s' being set", expname);
}

static const	char *Uname, *Comm;
pid_t	Pid;

#define PREFIX(i) (void)printf("%-8.8s %-10s %5d", Uname, Comm, Pid); \
	switch(i) { \
	case TEXT: \
		(void)printf(" text"); \
		break; \
	case CDIR: \
		(void)printf("   wd"); \
		break; \
	case RDIR: \
		(void)printf(" root"); \
		break; \
	case TRACE: \
		(void)printf("   tr"); \
		break; \
	default: \
		(void)printf(" %4d", i); \
		break; \
	}

static struct kinfo_drivers *kdriver;
static size_t kdriverlen;

static int
kdriver_comp(const void *a, const void *b)
{
	const struct kinfo_drivers *ka = a;
	const struct kinfo_drivers *kb = b;
	int kac = ka->d_cmajor == -1 ? 0 : ka->d_cmajor;
	int kbc = kb->d_cmajor == -1 ? 0 : kb->d_cmajor;
	int kab = ka->d_bmajor == -1 ? 0 : ka->d_bmajor;
	int kbb = kb->d_bmajor == -1 ? 0 : kb->d_bmajor;
	int c = kac - kbc;
	if (c == 0)
		return kab - kbb;
	else
		return c;
}

static const char *
kdriver_search(int type, dev_t num)
{
	struct kinfo_drivers k, *kp;
	static char buf[64];

	if (nflg)
		goto out;

	if (type == VBLK) {
		k.d_bmajor = num;
		k.d_cmajor = -1;
	} else {
		k.d_bmajor = -1;
		k.d_cmajor = num;
	}
	kp = bsearch(&k, kdriver, kdriverlen, sizeof(*kdriver), kdriver_comp);
	if (kp)
		return kp->d_name;
out:	
	snprintf(buf, sizeof(buf), "%llu", (unsigned long long)num);
	return buf;
}


static void
kdriver_init(void)
{
	size_t sz;
	int error;
	static const int name[2] = { CTL_KERN, KERN_DRIVERS };

	error = sysctl(name, __arraycount(name), NULL, &sz, NULL, 0);
	if (error == -1) {
		warn("sysctl kern.drivers");
		return;
	}

	if (sz % sizeof(*kdriver)) {
		warnx("bad size %zu for kern.drivers", sz);
		return;
	}

	kdriver = malloc(sz);
	if (kdriver == NULL) {
		warn("malloc");
		return;
	}

	error = sysctl(name, __arraycount(name), kdriver, &sz, NULL, 0);
	if (error == -1) {
		warn("sysctl kern.drivers");
		return;
	}

	kdriverlen = sz / sizeof(*kdriver);
	qsort(kdriver, kdriverlen, sizeof(*kdriver), kdriver_comp);
#ifdef DEBUG
	for (size_t i = 0; i < kdriverlen; i++)
		printf("%d %d %s\n", kdriver[i].d_cmajor, kdriver[i].d_bmajor,
		    kdriver[i].d_name);
#endif
}

/*
 * print open files attributed to this process
 */
static void
dofiles(struct kinfo_proc2 *p)
{
	int i;
	struct filedesc filed;
	struct cwdinfo cwdi;
	struct fdtab dt;

	Uname = user_from_uid(p->p_uid, 0);
	Pid = p->p_pid;
	Comm = p->p_comm;

	if (p->p_fd == 0 || p->p_cwdi == 0)
		return;
	if (!KVM_READ(p->p_fd, &filed, sizeof (filed))) {
		warnx("can't read filedesc at %p for pid %d",
		    (void *)(uintptr_t)p->p_fd, Pid);
		return;
	}
	if (filed.fd_lastfile == -1)
		return;
	if (!KVM_READ(p->p_cwdi, &cwdi, sizeof(cwdi))) {
		warnx("can't read cwdinfo at %p for pid %d",
		    (void *)(uintptr_t)p->p_cwdi, Pid);
		return;
	}
	if (!KVM_READ(filed.fd_dt, &dt, sizeof(dt))) {
		warnx("can't read dtab at %p for pid %d", filed.fd_dt, Pid);
		return;
	}
	if ((unsigned)filed.fd_lastfile >= dt.dt_nfiles ||
	    filed.fd_freefile > filed.fd_lastfile + 1) {
		dprintf("filedesc corrupted at %p for pid %d",
		    (void *)(uintptr_t)p->p_fd, Pid);
		return;
	}
	/*
	 * root directory vnode, if one
	 */
	if (cwdi.cwdi_rdir)
		vtrans(NULL, cwdi.cwdi_rdir, RDIR, FREAD, (long)cwdi.cwdi_rdir);
	/*
	 * current working directory vnode
	 */
	vtrans(NULL, cwdi.cwdi_cdir, CDIR, FREAD, (long)cwdi.cwdi_cdir);
#if 0
	/*
	 * Disable for now, since p->p_tracep appears to point to a ktr_desc *
	 * ktrace vnode, if one
	 */
	if (p->p_tracep)
		ftrans(p->p_tracep, TRACE);
#endif
	/*
	 * open files
	 */
#define FPSIZE	(sizeof (fdfile_t *))
	ALLOC_OFILES(filed.fd_lastfile+1);
	if (!KVM_READ(&filed.fd_dt->dt_ff, ofiles,
	    (filed.fd_lastfile+1) * FPSIZE)) {
		dprintf("can't read file structures at %p for pid %d",
		    &filed.fd_dt->dt_ff, Pid);
		return;
	}
	for (i = 0; i <= filed.fd_lastfile; i++) {
		if (ofiles[i] == NULL)
			continue;
		ftrans(ofiles[i], i);
	}
}

static void
ftrans(fdfile_t *fp, int i)
{
	struct file file;
	fdfile_t fdfile;

	if (!KVM_READ(fp, &fdfile, sizeof(fdfile))) {
		dprintf("can't read file %d at %p for pid %d",
		    i, fp, Pid);
		return;
	}
	if (fdfile.ff_file == NULL) {
		dprintf("null ff_file for %d at %p for pid %d",
		    i, fp, Pid);
		return;
	}
	if (!KVM_READ(fdfile.ff_file, &file, sizeof(file))) {
		dprintf("can't read file %d at %p for pid %d",
		    i, fdfile.ff_file, Pid);
		return;
	}
	if (Aflg && file.f_type != DTYPE_VNODE && checkfile == 0)
		(void)printf("%*lx ",
			2*(int)(sizeof(void*)), (long)fdfile.ff_file);
	switch (file.f_type) {
	case DTYPE_VNODE:
		vtrans(&file, file.f_data, i, file.f_flag, (long)fdfile.ff_file);
		break;
	case DTYPE_SOCKET:
		socktrans(&file, file.f_data, i);
		break;
	case DTYPE_PIPE:
		if (checkfile == 0)
			ptrans(&file, file.f_data, i);
		break;
	case DTYPE_MISC:
	case DTYPE_KQUEUE:
	case DTYPE_CRYPTO:
	case DTYPE_MQUEUE:
	case DTYPE_SEM:
		if (checkfile == 0)
			misctrans(&file, i);
		break;
	default:
		dprintf("unknown file type %d for file %d of pid %d",
		    file.f_type, i, Pid);
		break;
	}
}

static const char dead[] = "dead";
static const char *vnode_tag[] = {
	VNODE_TAGS
};

static const char *
vfilestat(struct vnode *vp, struct filestat *fsp)
{
	const char *badtype = NULL;

	if (vp->v_type == VNON)
		badtype = "none";
	else if (vp->v_type == VBAD)
		badtype = "bad";
	else
		switch (vp->v_tag) {
		case VT_NON:
			badtype = dead;
			break;
		case VT_UFS:
		case VT_LFS:
		case VT_MFS:
			if (!ufs_filestat(vp, fsp))
				badtype = "error";
			break;
		case VT_MSDOSFS:
			if (!msdosfs_filestat(vp, fsp))
				badtype = "error";
			break;
		case VT_NFS:
			if (!nfs_filestat(vp, fsp))
				badtype = "error";
			break;
		case VT_EXT2FS:
			if (!ext2fs_filestat(vp, fsp))
				badtype = "error";
			break;
		case VT_ISOFS:
			if (!isofs_filestat(vp, fsp))
				badtype = "error";
			break;
		case VT_NTFS:
			if (!ntfs_filestat(vp, fsp))
				badtype = "error";
			break;
		case VT_PTYFS:
			if (!ptyfs_filestat(vp, fsp))
				badtype = "error";
			break;
		case VT_TMPFS:
			if (!tmpfs_filestat(vp, fsp))
				badtype = "error";
			break;
#ifdef HAVE_ZFS
		case VT_ZFS:
			if (!zfs_filestat(vp, fsp))
				badtype = "error";
			break;
#endif
		case VT_NULL:
		case VT_OVERLAY:
		case VT_UMAP:
			badtype = layer_filestat(vp, fsp);
			break;
		default: {
			static char unknown[10];
			(void)snprintf(unknown, sizeof unknown, "%s(%#x)",
			    (size_t)vp->v_tag < __arraycount(vnode_tag) ?
			    vnode_tag[vp->v_tag] : "?", vp->v_tag);
			badtype = unknown;
			break;
		}
	}
	return badtype;
}

static int
checkfs(struct vnode *vp, struct vnode *vn, struct filestat *fst,
    const char **type, const char **fname)
{
	*fname = NULL;
	if (!KVM_READ(vp, vn, sizeof(*vn))) {
		dprintf("can't read vnode at %p for pid %d", vp, Pid);
		return 0;
	}
	*type = vfilestat(vn, fst);
	if (checkfile) {
		int fsmatch = 0;
		DEVS *d;
#if 0
		if (*type && *type != dead)
			return 0;
#endif
		for (d = devs; d != NULL; d = d->next) {
			if (d->fsid == fst->fsid) {
				fsmatch = 1;
				if (d->ino == fst->fileid) {
					*fname = d->name;
					break;
				}
			}
		}
		if (fsmatch == 0 || (*fname == NULL && fsflg == 0))
			return 0;
	}
	return 1;
}

static void
vprint(struct vnode *vn, struct filestat *fst)
{
	switch (vn->v_type) {
	case VBLK:
	case VCHR: {
		const char *name;

		if (nflg || ((name = devname(fst->rdev, vn->v_type == VCHR ? 
		    S_IFCHR : S_IFBLK)) == NULL))
			(void)printf("  %s,%-2llu",
			    kdriver_search(vn->v_type, major(fst->rdev)),
			    (unsigned long long)minor(fst->rdev));
		else
			(void)printf(" %6s", name);
		break;
	}
	default:
		(void)printf(" %6lld", (long long)fst->size);
	}
}

void
oprint(struct file *fp, const char *str)
{
	if (Oflg)
		(void)printf(" %6lld", (long long)(fp ? fp->f_offset : 0));
	fputs(str, stdout);
}

static void
vtrans(struct file *fp, struct vnode *vp, int i, int flag, long addr)
{
	struct vnode vn;
	char mode[15], rw[3];
	const char *badtype, *filename;
	struct filestat fst;

	if (!checkfs(vp, &vn, &fst, &badtype, &filename))
		return;

	if (Aflg)
		(void)printf("%*lx ", 2*(int)(sizeof(void*)), addr);
	PREFIX(i);
	if (badtype == dead) {
		char buf[1024];
		(void)snprintb(buf, sizeof(buf), VNODE_FLAGBITS,
		    vn.v_iflag | vn.v_vflag | vn.v_uflag);
		(void)printf("  flags %s\n", buf);
		return;
	} else if (badtype) {
		(void)printf("  -         -  %10s    -\n", badtype);
		return;
	}
	if (nflg)
		(void)printf("  %3llu,%-2llu",
		    (unsigned long long)major(fst.fsid),
		    (unsigned long long)minor(fst.fsid));
	else
		(void)printf("  %-8s", getmnton(vn.v_mount));
	if (nflg)
		(void)snprintf(mode, sizeof mode, "%6o", fst.mode);
	else
		strmode(fst.mode, mode);
	(void)printf("  %8"PRIu64" %*s", fst.fileid, nflg ? 5 : 10, mode);
	if (Oflg) {
		oprint(fp, "");
	} else {
		vprint(&vn, &fst);
	}
	rw[0] = '\0';
	if (flag & FREAD)
		(void)strlcat(rw, "r", sizeof(rw));
	if (flag & FWRITE)
		(void)strlcat(rw, "w", sizeof(rw));
	(void)printf(" %-2s", rw);
	if (filename && !fsflg)
		(void)printf("  %s", filename);
	(void)putchar('\n');
}

static int
ufs_filestat(struct vnode *vp, struct filestat *fsp)
{
	struct inode inode;
	struct ufsmount ufsmount;
	union dinode {
		struct ufs1_dinode dp1;
		struct ufs2_dinode dp2;
	} dip;

	if (!KVM_READ(VTOI(vp), &inode, sizeof (inode))) {
		dprintf("can't read inode at %p for pid %d", VTOI(vp), Pid);
		return 0;
	}

	if (!KVM_READ(inode.i_ump, &ufsmount, sizeof (struct ufsmount))) {
		dprintf("can't read ufsmount at %p for pid %d", inode.i_ump, Pid);
		return 0;
	}

	switch (ufsmount.um_fstype) {
	case UFS1:
		if (!KVM_READ(inode.i_din.ffs1_din, &dip,
		    sizeof(struct ufs1_dinode))) {
			dprintf("can't read dinode at %p for pid %d",
				inode.i_din.ffs1_din, Pid);
			return 0;
		}
		fsp->rdev = dip.dp1.di_rdev;
		break;
	case UFS2:
		if (!KVM_READ(inode.i_din.ffs2_din, &dip,
		    sizeof(struct ufs2_dinode))) {
			dprintf("can't read dinode at %p for pid %d",
			    inode.i_din.ffs2_din, Pid);
			return 0;
		}
		fsp->rdev = dip.dp2.di_rdev;
		break;
	default:
		dprintf("unknown ufs type %ld for pid %d",
			ufsmount.um_fstype, Pid);
		break;
	}
	fsp->fsid = inode.i_dev & 0xffff;
	fsp->fileid = inode.i_number;
	fsp->mode = (mode_t)inode.i_mode;
	fsp->size = inode.i_size;

	return 1;
}

static int
ext2fs_filestat(struct vnode *vp, struct filestat *fsp)
{
	struct inode inode;
	struct ext2fs_dinode dinode;

	if (!KVM_READ(VTOI(vp), &inode, sizeof (inode))) {
		dprintf("can't read inode at %p for pid %d", VTOI(vp), Pid);
		return 0;
	}
	fsp->fsid = inode.i_dev & 0xffff;
	fsp->fileid = inode.i_number;

	if (!KVM_READ(inode.i_din.e2fs_din, &dinode, sizeof dinode)) {
		dprintf("can't read ext2fs_dinode at %p for pid %d",
			inode.i_din.e2fs_din, Pid);
		return 0;
	}
	fsp->mode = dinode.e2di_mode;
	fsp->size = dinode.e2di_size;
	fsp->rdev = dinode.e2di_rdev;

	return 1;
}

static int
nfs_filestat(struct vnode *vp, struct filestat *fsp)
{
	struct nfsnode nfsnode;
	struct vattr va;

	if (!KVM_READ(VTONFS(vp), &nfsnode, sizeof (nfsnode))) {
		dprintf("can't read nfsnode at %p for pid %d", VTONFS(vp),
		    Pid);
		return 0;
	}
	if (!KVM_READ(nfsnode.n_vattr, &va, sizeof(va))) {
		dprintf("can't read vnode attributes at %p for pid %d",
		    nfsnode.n_vattr, Pid);
		return 0;
	}
	fsp->fsid = va.va_fsid;
	fsp->fileid = va.va_fileid;
	fsp->size = nfsnode.n_size;
	fsp->rdev = va.va_rdev;
	fsp->mode = (mode_t)va.va_mode | getftype(vp->v_type);

	return 1;
}

static int
msdosfs_filestat(struct vnode *vp, struct filestat *fsp)
{
	struct denode de;
	struct msdosfsmount mp;

	if (!KVM_READ(VTONFS(vp), &de, sizeof(de))) {
		dprintf("can't read denode at %p for pid %d", VTONFS(vp),
		    Pid);
		return 0;
	}
	if (!KVM_READ(de.de_pmp, &mp, sizeof(mp))) {
		dprintf("can't read mount struct at %p for pid %d", de.de_pmp,
		    Pid);
		return 0;
	}

	fsp->fsid = de.de_dev & 0xffff;
	fsp->fileid = 0; /* XXX see msdosfs_vptofh() for more info */
	fsp->size = de.de_FileSize;
	fsp->rdev = 0;	/* msdosfs doesn't support device files */
	fsp->mode = (0777 & mp.pm_mask) | getftype(vp->v_type);
	return 1;
}

static const char *
layer_filestat(struct vnode *vp, struct filestat *fsp)
{
	struct layer_node layer_node;
	struct mount mount;
	struct vnode vn;
	const char *badtype;

	if (!KVM_READ(VTOLAYER(vp), &layer_node, sizeof(layer_node))) {
		dprintf("can't read layer_node at %p for pid %d",
		    VTOLAYER(vp), Pid);
		return "error";
	}
	if (!KVM_READ(vp->v_mount, &mount, sizeof(struct mount))) {
		dprintf("can't read mount struct at %p for pid %d",
		    vp->v_mount, Pid);
		return "error";
	}
	vp = layer_node.layer_lowervp;
	if (!KVM_READ(vp, &vn, sizeof(struct vnode))) {
		dprintf("can't read vnode at %p for pid %d", vp, Pid);
		return "error";
	}
	if ((badtype = vfilestat(&vn, fsp)) == NULL)
		fsp->fsid = mount.mnt_stat.f_fsidx.__fsid_val[0];
	return badtype;
}

static char *
getmnton(struct mount *m)
{
	static struct mount mount;
	static struct mtab {
		struct mtab *next;
		struct mount *m;
		char mntonname[MNAMELEN];
	} *mhead = NULL;
	struct mtab *mt;

	for (mt = mhead; mt != NULL; mt = mt->next)
		if (m == mt->m)
			return mt->mntonname;
	if (!KVM_READ(m, &mount, sizeof(struct mount))) {
		warnx("can't read mount table at %p", m);
		return NULL;
	}
	if ((mt = malloc(sizeof (struct mtab))) == NULL) {
		err(1, "malloc(%u)", (unsigned int)sizeof(struct mtab));
	}
	mt->m = m;
	(void)memmove(&mt->mntonname[0], &mount.mnt_stat.f_mntonname[0],
	    MNAMELEN);
	mt->next = mhead;
	mhead = mt;
	return mt->mntonname;
}

static const char *
inet_addrstr(char *buf, size_t len, const struct in_addr *a, uint16_t p, bool isdg)
{
	char addr[256], serv[256];
	struct sockaddr_in sin;
	const int niflags =
		(nflg ? (NI_NUMERICHOST|NI_NUMERICSERV) : 0) |
		(isdg ? NI_DGRAM : 0);


	(void)memset(&sin, 0, sizeof(sin));
	sin.sin_family = AF_INET;
	sin.sin_len = sizeof(sin);
	sin.sin_addr = *a;
	sin.sin_port = htons(p);

	serv[0] = '\0';

	if (getnameinfo((struct sockaddr *)&sin, sin.sin_len,
	    addr, sizeof(addr), serv, sizeof(serv), niflags)) {
		if (inet_ntop(AF_INET, a, addr, sizeof(addr)) == NULL)
			strlcpy(addr, "invalid", sizeof(addr));
	}

	if (serv[0] == '\0')
		snprintf(serv, sizeof(serv), "%u", p);

	if (a->s_addr == INADDR_ANY) {
		if (p == 0)
			buf[0] = '\0';
		else
			snprintf(buf, len, "*:%s", serv);
		return buf;
	}

	snprintf(buf, len, "%s:%s", addr, serv);
	return buf;
}

#ifdef INET6
static const char *
inet6_addrstr(char *buf, size_t len, const struct in6_addr *a, uint16_t p, bool isdg)
{
	char addr[256], serv[256];
	struct sockaddr_in6 sin6;
	const int niflags =
		(nflg ? (NI_NUMERICHOST|NI_NUMERICSERV) : 0) |
		(isdg ? NI_DGRAM : 0);

	(void)memset(&sin6, 0, sizeof(sin6));
	sin6.sin6_family = AF_INET6;
	sin6.sin6_len = sizeof(sin6);
	sin6.sin6_addr = *a;
	sin6.sin6_port = htons(p);

	inet6_getscopeid(&sin6, INET6_IS_ADDR_LINKLOCAL);
	serv[0] = '\0';

	if (getnameinfo((struct sockaddr *)&sin6, sin6.sin6_len,
	    addr, sizeof(addr), serv, sizeof(serv), niflags)) {
		if (inet_ntop(AF_INET6, a, addr, sizeof(addr)) == NULL)
			strlcpy(addr, "invalid", sizeof(addr));
	}

	if (serv[0] == '\0')
		snprintf(serv, sizeof(serv), "%u", p);

	if (IN6_IS_ADDR_UNSPECIFIED(a)) {
		if (p == 0)
			buf[0] = '\0';
		else
			snprintf(buf, len, "*:%s", serv);
		return buf;
	}

	if (strchr(addr, ':') == NULL)
		snprintf(buf, len, "%s:%s", addr, serv);
	else
		snprintf(buf, len, "[%s]:%s", addr, serv);

	return buf;
}
#endif

static const char *
at_addrstr(char *buf, size_t len, const struct sockaddr_at *sat)
{
	const struct netrange *nr = &sat->sat_range.r_netrange;
	const struct at_addr *at = &sat->sat_addr;
	char addr[64], phase[64], range[64];

	if (sat->sat_port || at->s_net || at->s_node) {
		if (at->s_net || at->s_node)
			snprintf(addr, sizeof(addr), "%u.%u:%u",
			    ntohs(at->s_net), at->s_node, sat->sat_port);
		else
			snprintf(addr, sizeof(addr), "*:%u", sat->sat_port);
	} else
		addr[0] = '\0';

	if (nr->nr_phase)
		snprintf(phase, sizeof(phase), " phase %u", nr->nr_phase);
	else
		phase[0] = '\0';

	if (nr->nr_firstnet || nr->nr_lastnet)
		snprintf(range, sizeof(range), " range [%u-%u]",
		    ntohs(nr->nr_firstnet), ntohs(nr->nr_lastnet));
	else
		range[0] = '\0';

	snprintf(buf, len, "%s%s%s", addr, phase, range);
	return buf;
}

static void
socktrans(struct file *f, struct socket *sock, int i)
{
	static const char *stypename[] = {
		"unused",	/* 0 */
		"stream", 	/* 1 */
		"dgram",	/* 2 */
		"raw",		/* 3 */
		"rdm",		/* 4 */
		"seqpak"	/* 5 */
	};
#define	STYPEMAX 5
	struct socket	so;
	struct protosw	proto;
	struct domain	dom;
	struct in4pcb	in4pcb;
	struct in6pcb	in6pcb;
	struct unpcb	unpcb;
	struct ddpcb	ddpcb;
	int len;
	char dname[32];
	char lbuf[512], fbuf[512], pbuf[24];
	bool isdgram;

	pbuf[0] = '\0';
	/* fill in socket */
	if (!KVM_READ(sock, &so, sizeof(struct socket))) {
		dprintf("can't read sock at %p", sock);
		goto bad;
	}

	/* fill in protosw entry */
	if (!KVM_READ(so.so_proto, &proto, sizeof(struct protosw))) {
		dprintf("can't read protosw at %p", so.so_proto);
		goto bad;
	}

	/* fill in domain */
	if (!KVM_READ(proto.pr_domain, &dom, sizeof(struct domain))) {
		dprintf("can't read domain at %p", proto.pr_domain);
		goto bad;
	}

	if (checkfile && dom.dom_family != AF_LOCAL)
		return;

	if ((len = kvm_read(kd, (u_long)dom.dom_name, dname,
	    sizeof(dname) - 1)) != sizeof(dname) -1) {
		dprintf("can't read domain name at %p", dom.dom_name);
		dname[0] = '\0';
	}
	else
		dname[len] = '\0';

	/* 
	 * protocol specific formatting
	 *
	 * Try to find interesting things to print.  For TCP, the interesting
	 * thing is the address of the tcpcb, for UDP and others, just the
	 * inpcb (socket pcb).  For UNIX domain, its the address of the socket
	 * pcb and the address of the connected pcb (if connected).  Otherwise
	 * just print the protocol number and address of the socket itself.
	 * The idea is not to duplicate netstat, but to make available enough
	 * information for further analysis.
	 */
	fbuf[0] = '\0';
	lbuf[0] = '\0';
	isdgram = false;
	switch(dom.dom_family) {
	case AF_INET:
		getinetproto(pbuf, sizeof(pbuf), proto.pr_protocol);
		switch (proto.pr_protocol) {
		case IPPROTO_UDP:
			isdgram = true;
			/* FALLTHROUGH */
		case IPPROTO_TCP:
			if (so.so_pcb == NULL)
				break;
			if (kvm_read(kd, (u_long)so.so_pcb, (char *)&in4pcb,
			    sizeof(in4pcb)) != sizeof(in4pcb)) {
				dprintf("can't read in4pcb at %p", so.so_pcb);
				goto bad;
			}
			struct inpcb *inp = (struct inpcb *)&in4pcb;
			inet_addrstr(lbuf, sizeof(lbuf), &in4p_laddr(inp),
			    ntohs(inp->inp_lport), isdgram);
			inet_addrstr(fbuf, sizeof(fbuf), &in4p_faddr(inp),
			    ntohs(inp->inp_fport), isdgram);
			break;
		default:
			break;
		}
		break;
#ifdef INET6
	case AF_INET6:
		getinetproto(pbuf, sizeof(pbuf), proto.pr_protocol);
		switch (proto.pr_protocol) {
		case IPPROTO_UDP:
			isdgram = true;
			/* FALLTHROUGH */
		case IPPROTO_TCP:
			if (so.so_pcb == NULL)
				break;
			if (kvm_read(kd, (u_long)so.so_pcb, (char *)&in6pcb,
			    sizeof(in6pcb)) != sizeof(in6pcb)) {
				dprintf("can't read in6pcb at %p", so.so_pcb);
				goto bad;
			}
			struct inpcb *inp = (struct inpcb *)&in6pcb;
			inet6_addrstr(lbuf, sizeof(lbuf), &in6p_laddr(inp),
			    ntohs(inp->inp_lport), isdgram);
			inet6_addrstr(fbuf, sizeof(fbuf), &in6p_faddr(inp),
			    ntohs(inp->inp_fport), isdgram);
			break;
		default:
			break;
		}
		break;
#endif
	case AF_LOCAL:
		/* print address of pcb and connected pcb */
		if (so.so_pcb) {
			char shoconn[4], *cp;
			void *pcb[2];
			size_t p = 0;

			pcb[0] = so.so_pcb;

			cp = shoconn;
			if (!(so.so_state & SS_CANTRCVMORE))
				*cp++ = '<';
			*cp++ = '-';
			if (!(so.so_state & SS_CANTSENDMORE))
				*cp++ = '>';
			*cp = '\0';
again:
			if (kvm_read(kd, (u_long)pcb[p], (char *)&unpcb,
			    sizeof(struct unpcb)) != sizeof(struct unpcb)){
				dprintf("can't read unpcb at %p", so.so_pcb);
				goto bad;
			}
			if (checkfile) {
				struct vnode vn;
				struct filestat fst;
				const char *badtype, *filename;
				if (unpcb.unp_vnode == NULL)
					return;
				if (!checkfs(unpcb.unp_vnode, &vn, &fst,
				    &badtype, &filename))
					return;
			}

			if (unpcb.unp_addr) {
				struct sockaddr_un *sun = 
					malloc(unpcb.unp_addrlen);
				if (sun == NULL)
				    err(1, "malloc(%zu)",
					unpcb.unp_addrlen);
				if (kvm_read(kd, (u_long)unpcb.unp_addr,
				    sun, unpcb.unp_addrlen) !=
				    (ssize_t)unpcb.unp_addrlen) {
					dprintf("can't read sun at %p",
					    unpcb.unp_addr);
					free(sun);
				} else {
					snprintf(fbuf, sizeof(fbuf), " %s %s %s",
					    shoconn, sun->sun_path,
					    p == 0 ? "[creat]" : "[using]");
					free(sun);
					break;
				}
			}
			if (unpcb.unp_conn) {
				if (p == 0) {
					pcb[++p] = unpcb.unp_conn;
					goto again;
				} else
					snprintf(fbuf, sizeof(fbuf),
					    " %p %s %p", pcb[0], shoconn,
					    pcb[1]);
			}
		}
		break;
	case AF_APPLETALK:
		getatproto(pbuf, sizeof(pbuf), proto.pr_protocol);
		if (so.so_pcb) {
			if (kvm_read(kd, (u_long)so.so_pcb, (char *)&ddpcb,
			    sizeof(ddpcb)) != sizeof(ddpcb)){
				dprintf("can't read ddpcb at %p", so.so_pcb);
				goto bad;
			}
			at_addrstr(fbuf, sizeof(fbuf), &ddpcb.ddp_fsat);
			at_addrstr(lbuf, sizeof(lbuf), &ddpcb.ddp_lsat);
		}
		break;
	default:
		/* print protocol number and socket address */
		snprintf(fbuf, sizeof(fbuf), " %d %jx", proto.pr_protocol,
		    (uintmax_t)(uintptr_t)sock);
		break;
	}
	PREFIX(i);
	if ((u_short)so.so_type > STYPEMAX)
		(void)printf("* %s ?%d", dname, so.so_type);
	else
		(void)printf("* %s %s", dname, stypename[so.so_type]);

	if (pbuf[0])
		printf("%s", pbuf);
	if (fbuf[0] || lbuf[0])
		printf(" %s%s%s", fbuf, (fbuf[0] && lbuf[0]) ? " <-> " : "",
		    lbuf);
	else if (so.so_pcb)
		printf(" %jx", (uintmax_t)(uintptr_t)so.so_pcb);
	oprint(f, "\n");
	return;
bad:
	(void)printf("* error\n");
}

static void
ptrans(struct file *fp, struct pipe *cpipe, int i)
{
	struct pipe cp;

	PREFIX(i);
	
	/* fill in pipe */
	if (!KVM_READ(cpipe, &cp, sizeof(struct pipe))) {
		dprintf("can't read pipe at %p", cpipe);
		goto bad;
	}

	/* pipe descriptor is either read or write, never both */
	(void)printf("* pipe %p %s %p %s%s%s", cpipe,
		(fp->f_flag & FWRITE) ? "->" : "<-",
		cp.pipe_peer,
		(fp->f_flag & FWRITE) ? "w" : "r",
		(fp->f_flag & FNONBLOCK) ? "n" : "",
		(cp.pipe_state & PIPE_ASYNC) ? "a" : "");
	oprint(fp, "\n");
	return;
bad:
	(void)printf("* error\n");
}

static void
misctrans(struct file *file, int i)
{

	PREFIX(i);
	pmisc(file, dtypes[file->f_type]);
}

/*
 * getinetproto --
 *	print name of protocol number
 */
static void
getinetproto(char *buf, size_t len, int number)
{
	const char *cp;

	switch (number) {
	case IPPROTO_IP:
		cp = "ip"; break;
	case IPPROTO_ICMP:
		cp ="icmp"; break;
	case IPPROTO_GGP:
		cp ="ggp"; break;
	case IPPROTO_TCP:
		cp ="tcp"; break;
	case IPPROTO_EGP:
		cp ="egp"; break;
	case IPPROTO_PUP:
		cp ="pup"; break;
	case IPPROTO_UDP:
		cp ="udp"; break;
	case IPPROTO_IDP:
		cp ="idp"; break;
	case IPPROTO_RAW:
		cp ="raw"; break;
	case IPPROTO_ICMPV6:
		cp ="icmp6"; break;
	default:
		(void)snprintf(buf, len, " %d", number);
		return;
	}
	(void)snprintf(buf, len, " %s", cp);
}

/*
 * getatproto --
 *	print name of protocol number
 */
static void
getatproto(char *buf, size_t len, int number)
{
	const char *cp;

	switch (number) {
	case ATPROTO_DDP:
		cp = "ddp"; break;
	case ATPROTO_AARP:
		cp ="aarp"; break;
	default:
		(void)snprintf(buf, len, " %d", number);
		return;
	}
	(void)snprintf(buf, len, " %s", cp);
}

static int
getfname(const char *filename)
{
	struct stat statbuf;
	DEVS *cur;

	if (stat(filename, &statbuf)) {
		warn("stat(%s)", filename);
		return 0;
	}
	if ((cur = malloc(sizeof(*cur))) == NULL) {
		err(1, "malloc(%zu)", sizeof(*cur));
	}
	cur->next = devs;
	devs = cur;

	cur->ino = statbuf.st_ino;
	cur->fsid = statbuf.st_dev & 0xffff;
	cur->name = filename;
	return 1;
}

mode_t
getftype(enum vtype v_type)
{
	mode_t ftype;

	switch (v_type) {
	case VREG:
		ftype = S_IFREG;
		break;
	case VDIR:
		ftype = S_IFDIR;
		break;
	case VBLK:
		ftype = S_IFBLK;
		break;
	case VCHR:
		ftype = S_IFCHR;
		break;
	case VLNK:
		ftype = S_IFLNK;
		break;
	case VSOCK:
		ftype = S_IFSOCK;
		break;
	case VFIFO:
		ftype = S_IFIFO;
		break;
	default:
		ftype = 0;
		break;
	};

	return ftype;
}

static void
usage(void)
{
	(void)fprintf(stderr, "Usage: %s [-Afnv] [-M core] [-N system] "
	    "[-p pid] [-u user] [file ...]\n", getprogname());
	exit(1);
}
