xref: /csrg-svn/sys/kern/kern_acct.c (revision 49587)
1*49587Sbostic /*-
2*49587Sbostic  * Copyright (c) 1982, 1986, 1989 The Regents of the University of California.
3*49587Sbostic  * All rights reserved.
423365Smckusick  *
5*49587Sbostic  * %sccs.include.proprietary.c%
6*49587Sbostic  *
7*49587Sbostic  *	@(#)kern_acct.c	7.17 (Berkeley) 05/09/91
823365Smckusick  */
912788Ssam 
1017088Sbloom #include "param.h"
1117088Sbloom #include "systm.h"
1247545Skarels #include "namei.h"
1347545Skarels #include "resourcevar.h"
1438931Skarels #include "proc.h"
1543379Smckusick #include "ioctl.h"
1643379Smckusick #include "termios.h"
1743379Smckusick #include "tty.h"
1837728Smckusick #include "vnode.h"
1937728Smckusick #include "mount.h"
2017088Sbloom #include "kernel.h"
2145055Smckusick #include "file.h"
2217088Sbloom #include "acct.h"
2337728Smckusick #include "syslog.h"
2412788Ssam 
2512788Ssam /*
2637728Smckusick  * Values associated with enabling and disabling accounting
2737728Smckusick  */
2837728Smckusick int	acctsuspend = 2;	/* stop accounting when < 2% free space left */
2937728Smckusick int	acctresume = 4;		/* resume when free space risen to > 4% */
3037728Smckusick struct	timeval chk = { 15, 0 };/* frequency to check space for accounting */
3137728Smckusick 
3237728Smckusick /*
3312788Ssam  * SHOULD REPLACE THIS WITH A DRIVER THAT CAN BE READ TO SIMPLIFY.
3412788Ssam  */
3537728Smckusick struct	vnode *acctp;
3637728Smckusick struct	vnode *savacctp;
3712788Ssam 
3812788Ssam /*
3912788Ssam  * Perform process accounting functions.
4012788Ssam  */
4143379Smckusick /* ARGSUSED */
4243379Smckusick sysacct(p, uap, retval)
4343379Smckusick 	struct proc *p;
4443379Smckusick 	struct args {
4543379Smckusick 		char	*fname;
4643379Smckusick 	} *uap;
4743379Smckusick 	int *retval;
4812788Ssam {
4937728Smckusick 	register struct vnode *vp;
5037728Smckusick 	extern int acctwatch();
5137728Smckusick 	struct vnode *oacctp;
5243379Smckusick 	int error;
5347545Skarels 	struct nameidata nd;
5412788Ssam 
5547545Skarels 	if (error = suser(p->p_ucred, &p->p_acflag))
5644404Skarels 		return (error);
5737553Smckusick 	if (savacctp) {
5837553Smckusick 		acctp = savacctp;
5937553Smckusick 		savacctp = NULL;
6037553Smckusick 	}
6145055Smckusick 	if (uap->fname == NULL) {
6237728Smckusick 		if (vp = acctp) {
6337553Smckusick 			acctp = NULL;
6437728Smckusick 			vrele(vp);
6537728Smckusick 			untimeout(acctwatch, (caddr_t)&chk);
6612788Ssam 		}
6744404Skarels 		return (0);
6812788Ssam 	}
6947545Skarels 	nd.ni_segflg = UIO_USERSPACE;
7047545Skarels 	nd.ni_dirp = uap->fname;
7147545Skarels 	if (error = vn_open(&nd, p, FWRITE, 0644))
7244404Skarels 		return (error);
7347545Skarels 	vp = nd.ni_vp;
7437728Smckusick 	if (vp->v_type != VREG) {
7537728Smckusick 		vrele(vp);
7644404Skarels 		return (EACCES);
7737553Smckusick 	}
7837728Smckusick 	oacctp = acctp;
7937728Smckusick 	acctp = vp;
8037728Smckusick 	if (oacctp)
8137728Smckusick 		vrele(oacctp);
8237728Smckusick 	acctwatch(&chk);
8344404Skarels 	return (0);
8412788Ssam }
8512788Ssam 
8612788Ssam /*
8737728Smckusick  * Periodically check the file system to see if accounting
8837728Smckusick  * should be turned on or off.
8912788Ssam  */
9037728Smckusick acctwatch(resettime)
9137728Smckusick 	struct timeval *resettime;
9212788Ssam {
9337728Smckusick 	struct statfs sb;
9412788Ssam 
9512788Ssam 	if (savacctp) {
9648021Smckusick 		(void)VFS_STATFS(savacctp->v_mount, &sb, (struct proc *)0);
9737728Smckusick 		if (sb.f_bavail > acctresume * sb.f_blocks / 100) {
9812788Ssam 			acctp = savacctp;
9912788Ssam 			savacctp = NULL;
10037728Smckusick 			log(LOG_NOTICE, "Accounting resumed\n");
10137728Smckusick 			return;
10212788Ssam 		}
10312788Ssam 	}
10437728Smckusick 	if (acctp == NULL)
10512788Ssam 		return;
10648021Smckusick 	(void)VFS_STATFS(acctp->v_mount, &sb, (struct proc *)0);
10737728Smckusick 	if (sb.f_bavail <= acctsuspend * sb.f_blocks / 100) {
10812788Ssam 		savacctp = acctp;
10912788Ssam 		acctp = NULL;
11037728Smckusick 		log(LOG_NOTICE, "Accounting suspended\n");
11137728Smckusick 	}
11237728Smckusick 	timeout(acctwatch, (caddr_t)resettime, hzto(resettime));
11337728Smckusick }
11437728Smckusick 
11537728Smckusick /*
11637728Smckusick  * On exit, write a record on the accounting file.
11737728Smckusick  */
11843379Smckusick acct(p)
11943379Smckusick 	register struct proc *p;
12037728Smckusick {
12137728Smckusick 	register struct rusage *ru;
12237728Smckusick 	struct vnode *vp;
12340803Smarc 	struct timeval t, ut, st;
12440803Smarc 	int i, s;
12537728Smckusick 	struct acct acctbuf;
12637728Smckusick 	register struct acct *ap = &acctbuf;
12737728Smckusick 
12837728Smckusick 	if ((vp = acctp) == NULL)
12943379Smckusick 		return (0);
13040803Smarc 	bcopy(p->p_comm, ap->ac_comm, sizeof(ap->ac_comm));
13147545Skarels 	ru = &p->p_stats->p_ru;
13240803Smarc 	s = splclock();
13340803Smarc 	ut = p->p_utime;
13440803Smarc 	st = p->p_stime;
13516715Ssam 	t = time;
13640803Smarc 	splx(s);
13740803Smarc 	ap->ac_utime = compress(ut.tv_sec, ut.tv_usec);
13840803Smarc 	ap->ac_stime = compress(st.tv_sec, st.tv_usec);
13947545Skarels 	timevalsub(&t, &p->p_stats->p_start);
14016715Ssam 	ap->ac_etime = compress(t.tv_sec, t.tv_usec);
14147545Skarels 	ap->ac_btime = p->p_stats->p_start.tv_sec;
14247545Skarels 	ap->ac_uid = p->p_cred->p_ruid;
14347545Skarels 	ap->ac_gid = p->p_cred->p_rgid;
14440803Smarc 	t = st;
14540803Smarc 	timevaladd(&t, &ut);
14616719Skarels 	if (i = t.tv_sec * hz + t.tv_usec / tick)
14747545Skarels 		ap->ac_mem = (ru->ru_ixrss + ru->ru_idrss + ru->ru_isrss) / i;
14816715Ssam 	else
14916715Ssam 		ap->ac_mem = 0;
15026277Skarels 	ap->ac_io = compress(ru->ru_inblock + ru->ru_oublock, (long)0);
15142204Smarc 	if (p->p_flag&SCTTY && p->p_session->s_ttyp)
15240803Smarc 		ap->ac_tty = p->p_session->s_ttyp->t_dev;
15312788Ssam 	else
15412788Ssam 		ap->ac_tty = NODEV;
15547545Skarels 	ap->ac_flag = p->p_acflag;
15648021Smckusick 	return (vn_rdwr(UIO_WRITE, vp, (caddr_t)ap, sizeof (acctbuf), (off_t)0,
15748021Smckusick 		UIO_SYSSPACE, IO_UNIT|IO_APPEND, p->p_ucred, (int *)0,
15848021Smckusick 		(struct proc *)0));
15912788Ssam }
16012788Ssam 
16112788Ssam /*
16212788Ssam  * Produce a pseudo-floating point representation
16312788Ssam  * with 3 bits base-8 exponent, 13 bits fraction.
16412788Ssam  */
16516715Ssam compress(t, ut)
16612788Ssam 	register long t;
16716715Ssam 	long ut;
16812788Ssam {
16912788Ssam 	register exp = 0, round = 0;
17012788Ssam 
17117501Skarels 	t = t * AHZ;  /* compiler will convert only this format to a shift */
17216715Ssam 	if (ut)
17317501Skarels 		t += ut / (1000000 / AHZ);
17412788Ssam 	while (t >= 8192) {
17512788Ssam 		exp++;
17612788Ssam 		round = t&04;
17712788Ssam 		t >>= 3;
17812788Ssam 	}
17912788Ssam 	if (round) {
18012788Ssam 		t++;
18112788Ssam 		if (t >= 8192) {
18212788Ssam 			t >>= 3;
18312788Ssam 			exp++;
18412788Ssam 		}
18512788Ssam 	}
18612788Ssam 	return ((exp<<13) + t);
18712788Ssam }
188