NetBSD/sys/kern/kern_kinfo.c

312 lines
7.4 KiB
C

/*
* Copyright (c) 1982, 1986, 1989 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.
*
* from: @(#)kern_kinfo.c 7.17 (Berkeley) 6/26/91
* $Id: kern_kinfo.c,v 1.5 1993/06/27 06:01:36 andrew Exp $
*/
#include "param.h"
#include "systm.h"
#include "proc.h"
#include "kinfo.h"
#include "ioctl.h"
#include "tty.h"
#include "buf.h"
#include "file.h"
#include "vm/vm.h"
#include "kinfo_proc.h"
#define snderr(e) { error = (e); goto release;}
extern int kinfo_doproc(), kinfo_rtable(), kinfo_vnode(), kinfo_file();
struct kinfo_lock kinfo_lock;
/* ARGSUSED */
int
getkerninfo(p, uap, retval)
struct proc *p;
register struct args {
int op;
char *where;
int *size;
int arg;
} *uap;
int *retval;
{
int bufsize; /* max size of users buffer */
int needed, locked, (*server)(), error = 0;
switch (ki_type(uap->op)) {
case KINFO_PROC:
server = kinfo_doproc;
break;
case KINFO_RT:
server = kinfo_rtable;
break;
case KINFO_VNODE:
server = kinfo_vnode;
break;
case KINFO_FILE:
server = kinfo_file;
break;
default:
error = EINVAL;
goto done;
}
if (uap->where == NULL || uap->size == NULL) {
error = (*server)(uap->op, NULL, NULL, uap->arg, &needed);
goto done;
}
if (error = copyin((caddr_t)uap->size, (caddr_t)&bufsize,
sizeof (bufsize)))
goto done;
while (kinfo_lock.kl_lock) {
kinfo_lock.kl_want++;
sleep((caddr_t)&kinfo_lock, PRIBIO+1);
kinfo_lock.kl_want--;
kinfo_lock.kl_locked++;
}
kinfo_lock.kl_lock++;
if (!useracc(uap->where, bufsize, B_WRITE))
snderr(EFAULT);
if (server != kinfo_vnode) /* XXX */
vslock(uap->where, bufsize);
locked = bufsize;
error = (*server)(uap->op, uap->where, &bufsize, uap->arg, &needed);
if (server != kinfo_vnode) /* XXX */
vsunlock(uap->where, locked, B_WRITE);
if (error == 0)
error = copyout((caddr_t)&bufsize,
(caddr_t)uap->size, sizeof (bufsize));
release:
kinfo_lock.kl_lock--;
if (kinfo_lock.kl_want)
wakeup((caddr_t)&kinfo_lock);
done:
if (!error)
*retval = needed;
return (error);
}
/*
* try over estimating by 5 procs
*/
#define KINFO_PROCSLOP (5 * sizeof (struct kinfo_proc))
int
kinfo_doproc(op, where, acopysize, arg, aneeded)
int op;
char *where;
int *acopysize, *aneeded;
int arg;
{
register struct proc *p;
register struct kinfo_proc *dp = (struct kinfo_proc *)where;
register needed = 0;
int buflen;
int doingzomb;
struct eproc eproc;
int error = 0;
if (where != NULL)
buflen = *acopysize;
p = allproc;
doingzomb = 0;
again:
for (; p != NULL; p = p->p_nxt) {
/*
* TODO - make more efficient (see notes below).
* do by session.
*/
switch (ki_op(op)) {
case KINFO_PROC_PID:
/* could do this with just a lookup */
if (p->p_pid != (pid_t)arg)
continue;
break;
case KINFO_PROC_PGRP:
/* could do this by traversing pgrp */
if (p->p_pgrp->pg_id != (pid_t)arg)
continue;
break;
case KINFO_PROC_TTY:
if ((p->p_flag&SCTTY) == 0 ||
p->p_session->s_ttyp == NULL ||
p->p_session->s_ttyp->t_dev != (dev_t)arg)
continue;
break;
case KINFO_PROC_UID:
if (p->p_ucred->cr_uid != (uid_t)arg)
continue;
break;
case KINFO_PROC_RUID:
if (p->p_cred->p_ruid != (uid_t)arg)
continue;
break;
}
if (where != NULL && buflen >= sizeof (struct kinfo_proc)) {
fill_eproc(p, &eproc);
if (error = copyout((caddr_t)p, &dp->kp_proc,
sizeof (struct proc)))
return (error);
if (error = copyout((caddr_t)&eproc, &dp->kp_eproc,
sizeof (eproc)))
return (error);
dp++;
buflen -= sizeof (struct kinfo_proc);
}
needed += sizeof (struct kinfo_proc);
}
if (doingzomb == 0) {
p = zombproc;
doingzomb++;
goto again;
}
if (where != NULL)
*acopysize = (caddr_t)dp - where;
else
needed += KINFO_PROCSLOP;
*aneeded = needed;
return (0);
}
/*
* Fill in an eproc structure for the specified process.
*/
void
fill_eproc(p, ep)
register struct proc *p;
register struct eproc *ep;
{
register struct tty *tp;
ep->e_paddr = p;
ep->e_sess = p->p_pgrp->pg_session;
ep->e_pcred = *p->p_cred;
ep->e_ucred = *p->p_ucred;
ep->e_vm = *p->p_vmspace;
if (p->p_pptr)
ep->e_ppid = p->p_pptr->p_pid;
else
ep->e_ppid = 0;
ep->e_pgid = p->p_pgrp->pg_id;
ep->e_jobc = p->p_pgrp->pg_jobc;
if ((p->p_flag&SCTTY) &&
(tp = ep->e_sess->s_ttyp)) {
ep->e_tdev = tp->t_dev;
ep->e_tpgid = tp->t_pgrp ? tp->t_pgrp->pg_id : NO_PID;
ep->e_tsess = tp->t_session;
} else
ep->e_tdev = NODEV;
ep->e_flag = ep->e_sess->s_ttyvp ? EPROC_CTTY : 0;
if (SESS_LEADER(p))
ep->e_flag |= EPROC_SLEADER;
if (p->p_wmesg)
strncpy(ep->e_wmesg, p->p_wmesg, WMESGLEN);
ep->e_xsize = ep->e_xrssize = 0;
ep->e_xccount = ep->e_xswrss = 0;
}
/*
* Get file structures.
*/
int
kinfo_file(op, where, acopysize, arg, aneeded)
int op;
register char *where;
int *acopysize, *aneeded;
int arg;
{
int buflen, needed, error;
struct file *fp;
char *start = where;
if (where == NULL) {
/*
* overestimate by 10 files
*/
*aneeded = sizeof (filehead) +
(nfiles + 10) * sizeof (struct file);
return (0);
}
buflen = *acopysize;
needed = 0;
/*
* first copyout filehead
*/
if (buflen > sizeof (filehead)) {
if (error = copyout((caddr_t)&filehead, where,
sizeof (filehead)))
return (error);
buflen -= sizeof (filehead);
where += sizeof (filehead);
}
needed += sizeof (filehead);
/*
* followed by an array of file structures
*/
for (fp = filehead; fp != NULL; fp = fp->f_filef) {
if (buflen > sizeof (struct file)) {
if (error = copyout((caddr_t)fp, where,
sizeof (struct file)))
return (error);
buflen -= sizeof (struct file);
where += sizeof (struct file);
}
needed += sizeof (struct file);
}
*acopysize = where - start;
*aneeded = needed;
return (0);
}