1df8bae1dSRodney W. Grimes /*- 2df8bae1dSRodney W. Grimes * Copyright (c) 1982, 1986, 1989, 1993 3df8bae1dSRodney W. Grimes * The Regents of the University of California. All rights reserved. 4df8bae1dSRodney W. Grimes * (c) UNIX System Laboratories, Inc. 5fcdc50ebSRobert Watson * Copyright (c) 2005 Robert N. M. Watson 6fcdc50ebSRobert Watson * All rights reserved. 7fcdc50ebSRobert Watson * 8df8bae1dSRodney W. Grimes * All or some portions of this file are derived from material licensed 9df8bae1dSRodney W. Grimes * to the University of California by American Telephone and Telegraph 10df8bae1dSRodney W. Grimes * Co. or Unix System Laboratories, Inc. and are reproduced herein with 11df8bae1dSRodney W. Grimes * the permission of UNIX System Laboratories, Inc. 12df8bae1dSRodney W. Grimes * 13fcdc50ebSRobert Watson * Redistribution and use in source and binary forms, with or without 14fcdc50ebSRobert Watson * modification, are permitted provided that the following conditions 15fcdc50ebSRobert Watson * are met: 16fcdc50ebSRobert Watson * 1. Redistributions of source code must retain the above copyright 17fcdc50ebSRobert Watson * notice, this list of conditions and the following disclaimer. 18fcdc50ebSRobert Watson * 2. Redistributions in binary form must reproduce the above copyright 19fcdc50ebSRobert Watson * notice, this list of conditions and the following disclaimer in the 20fcdc50ebSRobert Watson * documentation and/or other materials provided with the distribution. 21fcdc50ebSRobert Watson * 4. Neither the name of the University nor the names of its contributors 22fcdc50ebSRobert Watson * may be used to endorse or promote products derived from this software 23fcdc50ebSRobert Watson * without specific prior written permission. 24fcdc50ebSRobert Watson * 25fcdc50ebSRobert Watson * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 26fcdc50ebSRobert Watson * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 27fcdc50ebSRobert Watson * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 28fcdc50ebSRobert Watson * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 29fcdc50ebSRobert Watson * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 30fcdc50ebSRobert Watson * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 31fcdc50ebSRobert Watson * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 32fcdc50ebSRobert Watson * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 33fcdc50ebSRobert Watson * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 34fcdc50ebSRobert Watson * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 35fcdc50ebSRobert Watson * SUCH DAMAGE. 36fcdc50ebSRobert Watson * 3771909edeSRobert Watson * Copyright (c) 1994 Christopher G. Demetriou 3871909edeSRobert Watson * 39df8bae1dSRodney W. Grimes * Redistribution and use in source and binary forms, with or without 40df8bae1dSRodney W. Grimes * modification, are permitted provided that the following conditions 41df8bae1dSRodney W. Grimes * are met: 42df8bae1dSRodney W. Grimes * 1. Redistributions of source code must retain the above copyright 43df8bae1dSRodney W. Grimes * notice, this list of conditions and the following disclaimer. 44df8bae1dSRodney W. Grimes * 2. Redistributions in binary form must reproduce the above copyright 45df8bae1dSRodney W. Grimes * notice, this list of conditions and the following disclaimer in the 46df8bae1dSRodney W. Grimes * documentation and/or other materials provided with the distribution. 47df8bae1dSRodney W. Grimes * 3. All advertising materials mentioning features or use of this software 48df8bae1dSRodney W. Grimes * must display the following acknowledgement: 49df8bae1dSRodney W. Grimes * This product includes software developed by the University of 50df8bae1dSRodney W. Grimes * California, Berkeley and its contributors. 51df8bae1dSRodney W. Grimes * 4. Neither the name of the University nor the names of its contributors 52df8bae1dSRodney W. Grimes * may be used to endorse or promote products derived from this software 53df8bae1dSRodney W. Grimes * without specific prior written permission. 54df8bae1dSRodney W. Grimes * 55df8bae1dSRodney W. Grimes * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 56df8bae1dSRodney W. Grimes * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 57df8bae1dSRodney W. Grimes * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 58df8bae1dSRodney W. Grimes * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 59df8bae1dSRodney W. Grimes * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 60df8bae1dSRodney W. Grimes * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 61df8bae1dSRodney W. Grimes * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 62df8bae1dSRodney W. Grimes * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 63df8bae1dSRodney W. Grimes * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 64df8bae1dSRodney W. Grimes * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 65df8bae1dSRodney W. Grimes * SUCH DAMAGE. 66df8bae1dSRodney W. Grimes * 67c7d893deSDavid Greenman * @(#)kern_acct.c 8.1 (Berkeley) 6/14/93 68df8bae1dSRodney W. Grimes */ 69df8bae1dSRodney W. Grimes 70677b542eSDavid E. O'Brien #include <sys/cdefs.h> 71677b542eSDavid E. O'Brien __FBSDID("$FreeBSD$"); 72677b542eSDavid E. O'Brien 73df8bae1dSRodney W. Grimes #include <sys/param.h> 740ad076d5SBruce Evans #include <sys/systm.h> 75b0864d13SJohn Baldwin #include <sys/acct.h> 76b0864d13SJohn Baldwin #include <sys/fcntl.h> 77b0864d13SJohn Baldwin #include <sys/kernel.h> 78505a1493SJohn Baldwin #include <sys/kthread.h> 7972de1b37SDiomidis Spinellis #include <sys/limits.h> 80fb919e4dSMark Murray #include <sys/lock.h> 81df8bae1dSRodney W. Grimes #include <sys/mount.h> 82b0864d13SJohn Baldwin #include <sys/mutex.h> 83c7d893deSDavid Greenman #include <sys/namei.h> 84acd3428bSRobert Watson #include <sys/priv.h> 85b0864d13SJohn Baldwin #include <sys/proc.h> 86c7d893deSDavid Greenman #include <sys/resourcevar.h> 87505a1493SJohn Baldwin #include <sys/sched.h> 88b0864d13SJohn Baldwin #include <sys/sx.h> 89b0864d13SJohn Baldwin #include <sys/sysctl.h> 90b0864d13SJohn Baldwin #include <sys/sysent.h> 91b0864d13SJohn Baldwin #include <sys/syslog.h> 92b0864d13SJohn Baldwin #include <sys/sysproto.h> 93c7d893deSDavid Greenman #include <sys/tty.h> 94b0864d13SJohn Baldwin #include <sys/vnode.h> 95df8bae1dSRodney W. Grimes 96aed55708SRobert Watson #include <security/mac/mac_framework.h> 97aed55708SRobert Watson 98df8bae1dSRodney W. Grimes /* 99c7d893deSDavid Greenman * The routines implemented in this file are described in: 100c7d893deSDavid Greenman * Leffler, et al.: The Design and Implementation of the 4.3BSD 101c7d893deSDavid Greenman * UNIX Operating System (Addison Welley, 1989) 102c7d893deSDavid Greenman * on pages 62-63. 103fdbe5babSDiomidis Spinellis * On May 2007 the historic 3 bits base 8 exponent, 13 bit fraction 104fdbe5babSDiomidis Spinellis * compt_t representation described in the above reference was replaced 105fdbe5babSDiomidis Spinellis * with that of IEEE-754 floats. 106c7d893deSDavid Greenman * 107c7d893deSDavid Greenman * Arguably, to simplify accounting operations, this mechanism should 108c7d893deSDavid Greenman * be replaced by one in which an accounting log file (similar to /dev/klog) 109c7d893deSDavid Greenman * is read by a user process, etc. However, that has its own problems. 110df8bae1dSRodney W. Grimes */ 111df8bae1dSRodney W. Grimes 112fdbe5babSDiomidis Spinellis /* Floating point definitions from <float.h>. */ 113fdbe5babSDiomidis Spinellis #define FLT_MANT_DIG 24 /* p */ 114fdbe5babSDiomidis Spinellis #define FLT_MAX_EXP 128 /* emax */ 115fdbe5babSDiomidis Spinellis 116df8bae1dSRodney W. Grimes /* 117c7d893deSDavid Greenman * Internal accounting functions. 118c7d893deSDavid Greenman * The former's operation is described in Leffler, et al., and the latter 119c7d893deSDavid Greenman * was provided by UCB with the 4.4BSD-Lite release 120df8bae1dSRodney W. Grimes */ 121fdbe5babSDiomidis Spinellis static uint32_t encode_timeval(struct timeval); 122fdbe5babSDiomidis Spinellis static uint32_t encode_long(long); 123505a1493SJohn Baldwin static void acctwatch(void); 124505a1493SJohn Baldwin static void acct_thread(void *); 125505a1493SJohn Baldwin static int acct_disable(struct thread *); 126ab36c067SJustin T. Gibbs 127ab36c067SJustin T. Gibbs /* 1285b606744SJohan Karlsson * Accounting vnode pointer, saved vnode pointer, and flags for each. 12971909edeSRobert Watson * acct_sx protects against changes to the active vnode and credentials 13071909edeSRobert Watson * while accounting records are being committed to disk. 131c7d893deSDavid Greenman */ 132101581b0SRobert Watson static int acct_configured; 13371909edeSRobert Watson static int acct_suspended; 13471909edeSRobert Watson static struct vnode *acct_vp; 13571909edeSRobert Watson static struct ucred *acct_cred; 13671909edeSRobert Watson static int acct_flags; 13771909edeSRobert Watson static struct sx acct_sx; 138df8bae1dSRodney W. Grimes 13971909edeSRobert Watson SX_SYSINIT(acct, &acct_sx, "acct_sx"); 1404f39d5d5SAndrew R. Reiter 141df8bae1dSRodney W. Grimes /* 142505a1493SJohn Baldwin * State of the accounting kthread. 143505a1493SJohn Baldwin */ 144505a1493SJohn Baldwin static int acct_state; 145505a1493SJohn Baldwin 146505a1493SJohn Baldwin #define ACCT_RUNNING 1 /* Accounting kthread is running. */ 147505a1493SJohn Baldwin #define ACCT_EXITREQ 2 /* Accounting kthread should exit. */ 148505a1493SJohn Baldwin 149505a1493SJohn Baldwin /* 150df8bae1dSRodney W. Grimes * Values associated with enabling and disabling accounting 151df8bae1dSRodney W. Grimes */ 15287b6de2bSPoul-Henning Kamp static int acctsuspend = 2; /* stop accounting when < 2% free space left */ 15387b6de2bSPoul-Henning Kamp SYSCTL_INT(_kern, OID_AUTO, acct_suspend, CTLFLAG_RW, 15447fdd692SNeil Blakey-Milner &acctsuspend, 0, "percentage of free disk space below which accounting stops"); 15587b6de2bSPoul-Henning Kamp 15687b6de2bSPoul-Henning Kamp static int acctresume = 4; /* resume when free space risen to > 4% */ 15787b6de2bSPoul-Henning Kamp SYSCTL_INT(_kern, OID_AUTO, acct_resume, CTLFLAG_RW, 15847fdd692SNeil Blakey-Milner &acctresume, 0, "percentage of free disk space above which accounting resumes"); 15987b6de2bSPoul-Henning Kamp 16087b6de2bSPoul-Henning Kamp static int acctchkfreq = 15; /* frequency (in seconds) to check space */ 161222fdf4bSJohn Baldwin 162222fdf4bSJohn Baldwin static int 163222fdf4bSJohn Baldwin sysctl_acct_chkfreq(SYSCTL_HANDLER_ARGS) 164222fdf4bSJohn Baldwin { 165222fdf4bSJohn Baldwin int error, value; 166222fdf4bSJohn Baldwin 167222fdf4bSJohn Baldwin /* Write out the old value. */ 168222fdf4bSJohn Baldwin error = SYSCTL_OUT(req, &acctchkfreq, sizeof(int)); 169222fdf4bSJohn Baldwin if (error || req->newptr == NULL) 170222fdf4bSJohn Baldwin return (error); 171222fdf4bSJohn Baldwin 172222fdf4bSJohn Baldwin /* Read in and verify the new value. */ 173222fdf4bSJohn Baldwin error = SYSCTL_IN(req, &value, sizeof(int)); 174222fdf4bSJohn Baldwin if (error) 175222fdf4bSJohn Baldwin return (error); 176222fdf4bSJohn Baldwin if (value <= 0) 177222fdf4bSJohn Baldwin return (EINVAL); 178222fdf4bSJohn Baldwin acctchkfreq = value; 179222fdf4bSJohn Baldwin return (0); 180222fdf4bSJohn Baldwin } 181222fdf4bSJohn Baldwin SYSCTL_PROC(_kern, OID_AUTO, acct_chkfreq, CTLTYPE_INT|CTLFLAG_RW, 182222fdf4bSJohn Baldwin &acctchkfreq, 0, sysctl_acct_chkfreq, "I", 183222fdf4bSJohn Baldwin "frequency for checking the free space"); 184df8bae1dSRodney W. Grimes 185101581b0SRobert Watson SYSCTL_INT(_kern, OID_AUTO, acct_configured, CTLFLAG_RD, &acct_configured, 0, 186101581b0SRobert Watson "Accounting configured or not"); 187101581b0SRobert Watson 18871909edeSRobert Watson SYSCTL_INT(_kern, OID_AUTO, acct_suspended, CTLFLAG_RD, &acct_suspended, 0, 18971909edeSRobert Watson "Accounting suspended or not"); 19071909edeSRobert Watson 191df8bae1dSRodney W. Grimes /* 1920c14ff0eSRobert Watson * Accounting system call. Written based on the specification and previous 1930c14ff0eSRobert Watson * implementation done by Mark Tinguely. 194df8bae1dSRodney W. Grimes */ 195c7d893deSDavid Greenman int 19671909edeSRobert Watson acct(struct thread *td, struct acct_args *uap) 197c7d893deSDavid Greenman { 198c7d893deSDavid Greenman struct nameidata nd; 19911178ee4SJohn Baldwin int error, flags, vfslocked; 200c7d893deSDavid Greenman 201acd3428bSRobert Watson error = priv_check(td, PRIV_ACCT); 202797f2d22SPoul-Henning Kamp if (error) 20316e7bc7bSJohn Baldwin return (error); 204c7d893deSDavid Greenman 205c7d893deSDavid Greenman /* 206c7d893deSDavid Greenman * If accounting is to be started to a file, open that file for 20711178ee4SJohn Baldwin * appending and make sure it's a 'normal'. 208c7d893deSDavid Greenman */ 209d1e405c5SAlfred Perlstein if (uap->path != NULL) { 210d3778141SRobert Watson NDINIT(&nd, LOOKUP, NOFOLLOW | MPSAFE | AUDITVNODE1, 211d3778141SRobert Watson UIO_USERSPACE, uap->path, td); 21292da2e76SJohan Karlsson flags = FWRITE | O_APPEND; 2139e223287SKonstantin Belousov error = vn_open(&nd, &flags, 0, NULL); 214797f2d22SPoul-Henning Kamp if (error) 21511178ee4SJohn Baldwin return (error); 21611178ee4SJohn Baldwin vfslocked = NDHASGIANT(&nd); 217762e6b85SEivind Eklund NDFREE(&nd, NDF_ONLY_PNBUF); 218e5e820fdSRobert Watson #ifdef MAC 21930d239bcSRobert Watson error = mac_system_check_acct(td->td_ucred, nd.ni_vp); 220e5e820fdSRobert Watson if (error) { 22122db15c0SAttilio Rao VOP_UNLOCK(nd.ni_vp, 0); 222e5e820fdSRobert Watson vn_close(nd.ni_vp, flags, td->td_ucred, td); 22311178ee4SJohn Baldwin VFS_UNLOCK_GIANT(vfslocked); 22411178ee4SJohn Baldwin return (error); 225e5e820fdSRobert Watson } 226e5e820fdSRobert Watson #endif 22722db15c0SAttilio Rao VOP_UNLOCK(nd.ni_vp, 0); 228c7d893deSDavid Greenman if (nd.ni_vp->v_type != VREG) { 2295b606744SJohan Karlsson vn_close(nd.ni_vp, flags, td->td_ucred, td); 23011178ee4SJohn Baldwin VFS_UNLOCK_GIANT(vfslocked); 23111178ee4SJohn Baldwin return (EACCES); 232c7d893deSDavid Greenman } 23311178ee4SJohn Baldwin VFS_UNLOCK_GIANT(vfslocked); 234e5e820fdSRobert Watson #ifdef MAC 235e5e820fdSRobert Watson } else { 23630d239bcSRobert Watson error = mac_system_check_acct(td->td_ucred, NULL); 237e5e820fdSRobert Watson if (error) 23811178ee4SJohn Baldwin return (error); 239e5e820fdSRobert Watson #endif 240c7d893deSDavid Greenman } 241c7d893deSDavid Greenman 24271909edeSRobert Watson /* 24371909edeSRobert Watson * Disallow concurrent access to the accounting vnode while we swap 24471909edeSRobert Watson * it out, in order to prevent access after close. 24571909edeSRobert Watson */ 24671909edeSRobert Watson sx_xlock(&acct_sx); 24701e3f3aeSBruce Evans 248c7d893deSDavid Greenman /* 249c7d893deSDavid Greenman * If accounting was previously enabled, kill the old space-watcher, 25071909edeSRobert Watson * close the file, and (if no new file was specified, leave). Reset 25171909edeSRobert Watson * the suspended state regardless of whether accounting remains 25271909edeSRobert Watson * enabled. 253c7d893deSDavid Greenman */ 25471909edeSRobert Watson acct_suspended = 0; 25511178ee4SJohn Baldwin if (acct_vp != NULL) { 25611178ee4SJohn Baldwin vfslocked = VFS_LOCK_GIANT(acct_vp->v_mount); 257505a1493SJohn Baldwin error = acct_disable(td); 25811178ee4SJohn Baldwin VFS_UNLOCK_GIANT(vfslocked); 25911178ee4SJohn Baldwin } 260d1e405c5SAlfred Perlstein if (uap->path == NULL) { 261505a1493SJohn Baldwin if (acct_state & ACCT_RUNNING) { 262505a1493SJohn Baldwin acct_state |= ACCT_EXITREQ; 263505a1493SJohn Baldwin wakeup(&acct_state); 264505a1493SJohn Baldwin } 26571909edeSRobert Watson sx_xunlock(&acct_sx); 26611178ee4SJohn Baldwin return (error); 267b4dcc46aSAndrew R. Reiter } 268c7d893deSDavid Greenman 269c7d893deSDavid Greenman /* 270c7d893deSDavid Greenman * Save the new accounting file vnode, and schedule the new 271c7d893deSDavid Greenman * free space watcher. 272c7d893deSDavid Greenman */ 27371909edeSRobert Watson acct_vp = nd.ni_vp; 27471909edeSRobert Watson acct_cred = crhold(td->td_ucred); 27571909edeSRobert Watson acct_flags = flags; 276505a1493SJohn Baldwin if (acct_state & ACCT_RUNNING) 277505a1493SJohn Baldwin acct_state &= ~ACCT_EXITREQ; 278505a1493SJohn Baldwin else { 279505a1493SJohn Baldwin /* 280505a1493SJohn Baldwin * Try to start up an accounting kthread. We may start more 281505a1493SJohn Baldwin * than one, but if so the extras will commit suicide as 282505a1493SJohn Baldwin * soon as they start up. 283505a1493SJohn Baldwin */ 2843745c395SJulian Elischer error = kproc_create(acct_thread, NULL, NULL, 0, 0, 285505a1493SJohn Baldwin "accounting"); 286505a1493SJohn Baldwin if (error) { 28711178ee4SJohn Baldwin vfslocked = VFS_LOCK_GIANT(acct_vp->v_mount); 288505a1493SJohn Baldwin (void) vn_close(acct_vp, acct_flags, acct_cred, td); 28911178ee4SJohn Baldwin VFS_UNLOCK_GIANT(vfslocked); 290505a1493SJohn Baldwin crfree(acct_cred); 291101581b0SRobert Watson acct_configured = 0; 292505a1493SJohn Baldwin acct_vp = NULL; 293505a1493SJohn Baldwin acct_cred = NULL; 294505a1493SJohn Baldwin acct_flags = 0; 295505a1493SJohn Baldwin sx_xunlock(&acct_sx); 296505a1493SJohn Baldwin log(LOG_NOTICE, "Unable to start accounting thread\n"); 29711178ee4SJohn Baldwin return (error); 298505a1493SJohn Baldwin } 299505a1493SJohn Baldwin } 300101581b0SRobert Watson acct_configured = 1; 30171909edeSRobert Watson sx_xunlock(&acct_sx); 30248719ca7SBosko Milekic log(LOG_NOTICE, "Accounting enabled\n"); 303c7d893deSDavid Greenman return (error); 304c7d893deSDavid Greenman } 305c7d893deSDavid Greenman 306c7d893deSDavid Greenman /* 307505a1493SJohn Baldwin * Disable currently in-progress accounting by closing the vnode, dropping 308505a1493SJohn Baldwin * our reference to the credential, and clearing the vnode's flags. 309505a1493SJohn Baldwin */ 310505a1493SJohn Baldwin static int 311505a1493SJohn Baldwin acct_disable(struct thread *td) 312505a1493SJohn Baldwin { 313505a1493SJohn Baldwin int error; 314505a1493SJohn Baldwin 315505a1493SJohn Baldwin sx_assert(&acct_sx, SX_XLOCKED); 316505a1493SJohn Baldwin error = vn_close(acct_vp, acct_flags, acct_cred, td); 317505a1493SJohn Baldwin crfree(acct_cred); 318101581b0SRobert Watson acct_configured = 0; 319505a1493SJohn Baldwin acct_vp = NULL; 320505a1493SJohn Baldwin acct_cred = NULL; 321505a1493SJohn Baldwin acct_flags = 0; 322505a1493SJohn Baldwin log(LOG_NOTICE, "Accounting disabled\n"); 323505a1493SJohn Baldwin return (error); 324505a1493SJohn Baldwin } 325505a1493SJohn Baldwin 326505a1493SJohn Baldwin /* 327c7d893deSDavid Greenman * Write out process accounting information, on process exit. 328c7d893deSDavid Greenman * Data to be written out is specified in Leffler, et al. 329c7d893deSDavid Greenman * and are enumerated below. (They're also noted in the system 330c7d893deSDavid Greenman * "acct.h" header file.) 331c7d893deSDavid Greenman */ 332c7d893deSDavid Greenman int 33371909edeSRobert Watson acct_process(struct thread *td) 334c7d893deSDavid Greenman { 335fdbe5babSDiomidis Spinellis struct acctv2 acct; 336c7d893deSDavid Greenman struct timeval ut, st, tmp; 33791d5354aSJohn Baldwin struct plimit *newlim, *oldlim; 33801e3f3aeSBruce Evans struct proc *p; 3391c4bcd05SJeff Roberson struct rusage ru; 34071909edeSRobert Watson int t, ret, vfslocked; 3414f39d5d5SAndrew R. Reiter 3422b05b557SRobert Watson /* 3432b05b557SRobert Watson * Lockless check of accounting condition before doing the hard 3442b05b557SRobert Watson * work. 3452b05b557SRobert Watson */ 34671909edeSRobert Watson if (acct_vp == NULL || acct_suspended) 3472b05b557SRobert Watson return (0); 3482b05b557SRobert Watson 34971909edeSRobert Watson sx_slock(&acct_sx); 350c7d893deSDavid Greenman 3512b05b557SRobert Watson /* 3522b05b557SRobert Watson * If accounting isn't enabled, don't bother. Have to check again 3532b05b557SRobert Watson * once we own the lock in case we raced with disabling of accounting 3542b05b557SRobert Watson * by another thread. 3552b05b557SRobert Watson */ 35671909edeSRobert Watson if (acct_vp == NULL || acct_suspended) { 35771909edeSRobert Watson sx_sunlock(&acct_sx); 358c7d893deSDavid Greenman return (0); 3594f39d5d5SAndrew R. Reiter } 360c7d893deSDavid Greenman 36101e3f3aeSBruce Evans p = td->td_proc; 36201e3f3aeSBruce Evans 363c7d893deSDavid Greenman /* 364c7d893deSDavid Greenman * Get process accounting information. 365c7d893deSDavid Greenman */ 366c7d893deSDavid Greenman 36740572ab3SEd Schouten sx_slock(&proctree_lock); 3687e653dbdSJohn Baldwin PROC_LOCK(p); 36940572ab3SEd Schouten 37040572ab3SEd Schouten /* (1) The terminal from which the process was started */ 37140572ab3SEd Schouten if ((p->p_flag & P_CONTROLT) && p->p_pgrp->pg_session->s_ttyp) 37240572ab3SEd Schouten acct.ac_tty = tty_udev(p->p_pgrp->pg_session->s_ttyp); 37340572ab3SEd Schouten else 37440572ab3SEd Schouten acct.ac_tty = NODEV; 37540572ab3SEd Schouten sx_sunlock(&proctree_lock); 37640572ab3SEd Schouten 37740572ab3SEd Schouten /* (2) The name of the command that ran */ 378c7d893deSDavid Greenman bcopy(p->p_comm, acct.ac_comm, sizeof acct.ac_comm); 379c7d893deSDavid Greenman 38040572ab3SEd Schouten /* (3) The amount of user and system time that was used */ 381a1fe14bcSAttilio Rao rufetchcalc(p, &ru, &ut, &st); 382fdbe5babSDiomidis Spinellis acct.ac_utime = encode_timeval(ut); 383fdbe5babSDiomidis Spinellis acct.ac_stime = encode_timeval(st); 384c7d893deSDavid Greenman 38540572ab3SEd Schouten /* (4) The elapsed time the command ran (and its starting time) */ 38687ccef7bSDag-Erling Smørgrav tmp = boottime; 38787ccef7bSDag-Erling Smørgrav timevaladd(&tmp, &p->p_stats->p_start); 38887ccef7bSDag-Erling Smørgrav acct.ac_btime = tmp.tv_sec; 38987ccef7bSDag-Erling Smørgrav microuptime(&tmp); 390c7d893deSDavid Greenman timevalsub(&tmp, &p->p_stats->p_start); 391fdbe5babSDiomidis Spinellis acct.ac_etime = encode_timeval(tmp); 392c7d893deSDavid Greenman 39340572ab3SEd Schouten /* (5) The average amount of memory used */ 394c7d893deSDavid Greenman tmp = ut; 395c7d893deSDavid Greenman timevaladd(&tmp, &st); 396fdbe5babSDiomidis Spinellis /* Convert tmp (i.e. u + s) into hz units to match ru_i*. */ 397c7d893deSDavid Greenman t = tmp.tv_sec * hz + tmp.tv_usec / tick; 398c7d893deSDavid Greenman if (t) 3991c4bcd05SJeff Roberson acct.ac_mem = encode_long((ru.ru_ixrss + ru.ru_idrss + 4001c4bcd05SJeff Roberson + ru.ru_isrss) / t); 401c7d893deSDavid Greenman else 402c7d893deSDavid Greenman acct.ac_mem = 0; 403c7d893deSDavid Greenman 40440572ab3SEd Schouten /* (6) The number of disk I/O operations done */ 4051c4bcd05SJeff Roberson acct.ac_io = encode_long(ru.ru_inblock + ru.ru_oublock); 406c7d893deSDavid Greenman 40740572ab3SEd Schouten /* (7) The UID and GID of the process */ 408b1fc0ec1SRobert Watson acct.ac_uid = p->p_ucred->cr_ruid; 409b1fc0ec1SRobert Watson acct.ac_gid = p->p_ucred->cr_rgid; 410c7d893deSDavid Greenman 411c7d893deSDavid Greenman /* (8) The boolean flags that tell how the process terminated, etc. */ 412fdbe5babSDiomidis Spinellis acct.ac_flagx = p->p_acflag; 4137e653dbdSJohn Baldwin PROC_UNLOCK(p); 414c7d893deSDavid Greenman 415fdbe5babSDiomidis Spinellis /* Setup ancillary structure fields. */ 416fdbe5babSDiomidis Spinellis acct.ac_flagx |= ANVER; 417fdbe5babSDiomidis Spinellis acct.ac_zero = 0; 418fdbe5babSDiomidis Spinellis acct.ac_version = 2; 419fdbe5babSDiomidis Spinellis acct.ac_len = acct.ac_len2 = sizeof(acct); 420fdbe5babSDiomidis Spinellis 421c7d893deSDavid Greenman /* 422b5afad71SDavid Greenman * Eliminate any file size rlimit. 423b5afad71SDavid Greenman */ 42491d5354aSJohn Baldwin newlim = lim_alloc(); 42591d5354aSJohn Baldwin PROC_LOCK(p); 42691d5354aSJohn Baldwin oldlim = p->p_limit; 42791d5354aSJohn Baldwin lim_copy(newlim, oldlim); 42891d5354aSJohn Baldwin newlim->pl_rlimit[RLIMIT_FSIZE].rlim_cur = RLIM_INFINITY; 42991d5354aSJohn Baldwin p->p_limit = newlim; 43091d5354aSJohn Baldwin PROC_UNLOCK(p); 43191d5354aSJohn Baldwin lim_free(oldlim); 432b5afad71SDavid Greenman 43301e3f3aeSBruce Evans /* 43401e3f3aeSBruce Evans * Write the accounting information to the file. 43501e3f3aeSBruce Evans */ 43671909edeSRobert Watson vfslocked = VFS_LOCK_GIANT(acct_vp->v_mount); 43771909edeSRobert Watson ret = vn_rdwr(UIO_WRITE, acct_vp, (caddr_t)&acct, sizeof (acct), 43871909edeSRobert Watson (off_t)0, UIO_SYSSPACE, IO_APPEND|IO_UNIT, acct_cred, NOCRED, 43901ed1748SKonstantin Belousov NULL, td); 44071909edeSRobert Watson VFS_UNLOCK_GIANT(vfslocked); 44171909edeSRobert Watson sx_sunlock(&acct_sx); 4424f39d5d5SAndrew R. Reiter return (ret); 443c7d893deSDavid Greenman } 444c7d893deSDavid Greenman 445fdbe5babSDiomidis Spinellis /* FLOAT_CONVERSION_START (Regression testing; don't remove this line.) */ 446fdbe5babSDiomidis Spinellis 447fdbe5babSDiomidis Spinellis /* Convert timevals and longs into IEEE-754 bit patterns. */ 448fdbe5babSDiomidis Spinellis 449fdbe5babSDiomidis Spinellis /* Mantissa mask (MSB is implied, so subtract 1). */ 450fdbe5babSDiomidis Spinellis #define MANT_MASK ((1 << (FLT_MANT_DIG - 1)) - 1) 451fdbe5babSDiomidis Spinellis 452c7d893deSDavid Greenman /* 453fdbe5babSDiomidis Spinellis * We calculate integer values to a precision of approximately 454fdbe5babSDiomidis Spinellis * 28 bits. 455fdbe5babSDiomidis Spinellis * This is high-enough precision to fill the 24 float bits 456fdbe5babSDiomidis Spinellis * and low-enough to avoid overflowing the 32 int bits. 457c7d893deSDavid Greenman */ 458fdbe5babSDiomidis Spinellis #define CALC_BITS 28 459c7d893deSDavid Greenman 460fdbe5babSDiomidis Spinellis /* log_2(1000000). */ 461fdbe5babSDiomidis Spinellis #define LOG2_1M 20 462c7d893deSDavid Greenman 463fdbe5babSDiomidis Spinellis /* 464fdbe5babSDiomidis Spinellis * Convert the elements of a timeval into a 32-bit word holding 465fdbe5babSDiomidis Spinellis * the bits of a IEEE-754 float. 466fdbe5babSDiomidis Spinellis * The float value represents the timeval's value in microsecond units. 467fdbe5babSDiomidis Spinellis */ 468fdbe5babSDiomidis Spinellis static uint32_t 469fdbe5babSDiomidis Spinellis encode_timeval(struct timeval tv) 470c7d893deSDavid Greenman { 471fdbe5babSDiomidis Spinellis int log2_s; 472fdbe5babSDiomidis Spinellis int val, exp; /* Unnormalized value and exponent */ 473fdbe5babSDiomidis Spinellis int norm_exp; /* Normalized exponent */ 474fdbe5babSDiomidis Spinellis int shift; 475c7d893deSDavid Greenman 476fdbe5babSDiomidis Spinellis /* 477fdbe5babSDiomidis Spinellis * First calculate value and exponent to about CALC_BITS precision. 478fdbe5babSDiomidis Spinellis * Note that the following conditionals have been ordered so that 479fdbe5babSDiomidis Spinellis * the most common cases appear first. 480fdbe5babSDiomidis Spinellis */ 481fdbe5babSDiomidis Spinellis if (tv.tv_sec == 0) { 482fdbe5babSDiomidis Spinellis if (tv.tv_usec == 0) 483fdbe5babSDiomidis Spinellis return (0); 484c7d893deSDavid Greenman exp = 0; 485fdbe5babSDiomidis Spinellis val = tv.tv_usec; 486fdbe5babSDiomidis Spinellis } else { 487fdbe5babSDiomidis Spinellis /* 488fdbe5babSDiomidis Spinellis * Calculate the value to a precision of approximately 489fdbe5babSDiomidis Spinellis * CALC_BITS. 490fdbe5babSDiomidis Spinellis */ 491fdbe5babSDiomidis Spinellis log2_s = fls(tv.tv_sec) - 1; 492fdbe5babSDiomidis Spinellis if (log2_s + LOG2_1M < CALC_BITS) { 493fdbe5babSDiomidis Spinellis exp = 0; 494fdbe5babSDiomidis Spinellis val = 1000000 * tv.tv_sec + tv.tv_usec; 495fdbe5babSDiomidis Spinellis } else { 496fdbe5babSDiomidis Spinellis exp = log2_s + LOG2_1M - CALC_BITS; 497*60ae52f7SEd Schouten val = (unsigned int)(((uint64_t)1000000 * tv.tv_sec + 498fdbe5babSDiomidis Spinellis tv.tv_usec) >> exp); 499fdbe5babSDiomidis Spinellis } 500fdbe5babSDiomidis Spinellis } 501fdbe5babSDiomidis Spinellis /* Now normalize and pack the value into an IEEE-754 float. */ 502fdbe5babSDiomidis Spinellis norm_exp = fls(val) - 1; 503fdbe5babSDiomidis Spinellis shift = FLT_MANT_DIG - norm_exp - 1; 504fdbe5babSDiomidis Spinellis #ifdef ACCT_DEBUG 505fdbe5babSDiomidis Spinellis printf("val=%d exp=%d shift=%d log2(val)=%d\n", 506fdbe5babSDiomidis Spinellis val, exp, shift, norm_exp); 507fdbe5babSDiomidis Spinellis printf("exp=%x mant=%x\n", FLT_MAX_EXP - 1 + exp + norm_exp, 508fdbe5babSDiomidis Spinellis ((shift > 0 ? (val << shift) : (val >> -shift)) & MANT_MASK)); 509fdbe5babSDiomidis Spinellis #endif 510fdbe5babSDiomidis Spinellis return (((FLT_MAX_EXP - 1 + exp + norm_exp) << (FLT_MANT_DIG - 1)) | 511fdbe5babSDiomidis Spinellis ((shift > 0 ? val << shift : val >> -shift) & MANT_MASK)); 512c7d893deSDavid Greenman } 513c7d893deSDavid Greenman 514fdbe5babSDiomidis Spinellis /* 515fdbe5babSDiomidis Spinellis * Convert a non-negative long value into the bit pattern of 516fdbe5babSDiomidis Spinellis * an IEEE-754 float value. 517fdbe5babSDiomidis Spinellis */ 518fdbe5babSDiomidis Spinellis static uint32_t 519fdbe5babSDiomidis Spinellis encode_long(long val) 520fdbe5babSDiomidis Spinellis { 521fdbe5babSDiomidis Spinellis int norm_exp; /* Normalized exponent */ 522fdbe5babSDiomidis Spinellis int shift; 523fdbe5babSDiomidis Spinellis 524fdbe5babSDiomidis Spinellis if (val == 0) 525fdbe5babSDiomidis Spinellis return (0); 52672de1b37SDiomidis Spinellis if (val < 0) { 52772de1b37SDiomidis Spinellis log(LOG_NOTICE, 528d5b6981eSDiomidis Spinellis "encode_long: negative value %ld in accounting record\n", 52972de1b37SDiomidis Spinellis val); 53072de1b37SDiomidis Spinellis val = LONG_MAX; 53172de1b37SDiomidis Spinellis } 532fdbe5babSDiomidis Spinellis norm_exp = fls(val) - 1; 533fdbe5babSDiomidis Spinellis shift = FLT_MANT_DIG - norm_exp - 1; 534fdbe5babSDiomidis Spinellis #ifdef ACCT_DEBUG 535fdbe5babSDiomidis Spinellis printf("val=%d shift=%d log2(val)=%d\n", 536fdbe5babSDiomidis Spinellis val, shift, norm_exp); 537fdbe5babSDiomidis Spinellis printf("exp=%x mant=%x\n", FLT_MAX_EXP - 1 + exp + norm_exp, 538fdbe5babSDiomidis Spinellis ((shift > 0 ? (val << shift) : (val >> -shift)) & MANT_MASK)); 539fdbe5babSDiomidis Spinellis #endif 540fdbe5babSDiomidis Spinellis return (((FLT_MAX_EXP - 1 + norm_exp) << (FLT_MANT_DIG - 1)) | 541fdbe5babSDiomidis Spinellis ((shift > 0 ? val << shift : val >> -shift) & MANT_MASK)); 542c7d893deSDavid Greenman } 543c7d893deSDavid Greenman 544fdbe5babSDiomidis Spinellis /* FLOAT_CONVERSION_END (Regression testing; don't remove this line.) */ 545c7d893deSDavid Greenman 546c7d893deSDavid Greenman /* 547c7d893deSDavid Greenman * Periodically check the filesystem to see if accounting 548c7d893deSDavid Greenman * should be turned on or off. Beware the case where the vnode 549c7d893deSDavid Greenman * has been vgone()'d out from underneath us, e.g. when the file 550c7d893deSDavid Greenman * system containing the accounting file has been forcibly unmounted. 551c7d893deSDavid Greenman */ 552df8bae1dSRodney W. Grimes /* ARGSUSED */ 55387b6de2bSPoul-Henning Kamp static void 554505a1493SJohn Baldwin acctwatch(void) 555df8bae1dSRodney W. Grimes { 556df8bae1dSRodney W. Grimes struct statfs sb; 55771909edeSRobert Watson int vfslocked; 558df8bae1dSRodney W. Grimes 559505a1493SJohn Baldwin sx_assert(&acct_sx, SX_XLOCKED); 560505a1493SJohn Baldwin 561505a1493SJohn Baldwin /* 562505a1493SJohn Baldwin * If accounting was disabled before our kthread was scheduled, 563505a1493SJohn Baldwin * then acct_vp might be NULL. If so, just ask our kthread to 564505a1493SJohn Baldwin * exit and return. 565505a1493SJohn Baldwin */ 566505a1493SJohn Baldwin if (acct_vp == NULL) { 567505a1493SJohn Baldwin acct_state |= ACCT_EXITREQ; 568c7d893deSDavid Greenman return; 569c7d893deSDavid Greenman } 570505a1493SJohn Baldwin 571505a1493SJohn Baldwin /* 572505a1493SJohn Baldwin * If our vnode is no longer valid, tear it down and signal the 573505a1493SJohn Baldwin * accounting thread to die. 574505a1493SJohn Baldwin */ 575505a1493SJohn Baldwin vfslocked = VFS_LOCK_GIANT(acct_vp->v_mount); 576505a1493SJohn Baldwin if (acct_vp->v_type == VBAD) { 577505a1493SJohn Baldwin (void) acct_disable(NULL); 578505a1493SJohn Baldwin VFS_UNLOCK_GIANT(vfslocked); 579505a1493SJohn Baldwin acct_state |= ACCT_EXITREQ; 580505a1493SJohn Baldwin return; 581505a1493SJohn Baldwin } 582505a1493SJohn Baldwin 58371909edeSRobert Watson /* 58471909edeSRobert Watson * Stopping here is better than continuing, maybe it will be VBAD 58571909edeSRobert Watson * next time around. 58671909edeSRobert Watson */ 587dfd233edSAttilio Rao if (VFS_STATFS(acct_vp->v_mount, &sb) < 0) { 58871909edeSRobert Watson VFS_UNLOCK_GIANT(vfslocked); 58971909edeSRobert Watson return; 59071909edeSRobert Watson } 59171909edeSRobert Watson VFS_UNLOCK_GIANT(vfslocked); 59271909edeSRobert Watson if (acct_suspended) { 59371909edeSRobert Watson if (sb.f_bavail > (int64_t)(acctresume * sb.f_blocks / 59471909edeSRobert Watson 100)) { 59571909edeSRobert Watson acct_suspended = 0; 596df8bae1dSRodney W. Grimes log(LOG_NOTICE, "Accounting resumed\n"); 597df8bae1dSRodney W. Grimes } 598996c772fSJohn Dyson } else { 59971909edeSRobert Watson if (sb.f_bavail <= (int64_t)(acctsuspend * sb.f_blocks / 60071909edeSRobert Watson 100)) { 60171909edeSRobert Watson acct_suspended = 1; 602df8bae1dSRodney W. Grimes log(LOG_NOTICE, "Accounting suspended\n"); 603df8bae1dSRodney W. Grimes } 604996c772fSJohn Dyson } 605505a1493SJohn Baldwin } 606505a1493SJohn Baldwin 607505a1493SJohn Baldwin /* 608505a1493SJohn Baldwin * The main loop for the dedicated kernel thread that periodically calls 609505a1493SJohn Baldwin * acctwatch(). 610505a1493SJohn Baldwin */ 611505a1493SJohn Baldwin static void 612505a1493SJohn Baldwin acct_thread(void *dummy) 613505a1493SJohn Baldwin { 614505a1493SJohn Baldwin u_char pri; 615505a1493SJohn Baldwin 616505a1493SJohn Baldwin /* This is a low-priority kernel thread. */ 617505a1493SJohn Baldwin pri = PRI_MAX_KERN; 618982d11f8SJeff Roberson thread_lock(curthread); 619505a1493SJohn Baldwin sched_prio(curthread, pri); 620982d11f8SJeff Roberson thread_unlock(curthread); 621505a1493SJohn Baldwin 622505a1493SJohn Baldwin /* If another accounting kthread is already running, just die. */ 623505a1493SJohn Baldwin sx_xlock(&acct_sx); 624505a1493SJohn Baldwin if (acct_state & ACCT_RUNNING) { 62571909edeSRobert Watson sx_xunlock(&acct_sx); 6263745c395SJulian Elischer kproc_exit(0); 627505a1493SJohn Baldwin } 628505a1493SJohn Baldwin acct_state |= ACCT_RUNNING; 629505a1493SJohn Baldwin 630505a1493SJohn Baldwin /* Loop until we are asked to exit. */ 631505a1493SJohn Baldwin while (!(acct_state & ACCT_EXITREQ)) { 632505a1493SJohn Baldwin 633505a1493SJohn Baldwin /* Perform our periodic checks. */ 634505a1493SJohn Baldwin acctwatch(); 635505a1493SJohn Baldwin 636505a1493SJohn Baldwin /* 637505a1493SJohn Baldwin * We check this flag again before sleeping since the 638505a1493SJohn Baldwin * acctwatch() might have shut down accounting and asked us 639505a1493SJohn Baldwin * to exit. 640505a1493SJohn Baldwin */ 641505a1493SJohn Baldwin if (!(acct_state & ACCT_EXITREQ)) { 6426caa5f40SJohn Baldwin sx_sleep(&acct_state, &acct_sx, 0, "-", 6436caa5f40SJohn Baldwin acctchkfreq * hz); 644505a1493SJohn Baldwin } 645505a1493SJohn Baldwin } 646505a1493SJohn Baldwin 647505a1493SJohn Baldwin /* 648505a1493SJohn Baldwin * Acknowledge the exit request and shutdown. We clear both the 649505a1493SJohn Baldwin * exit request and running flags. 650505a1493SJohn Baldwin */ 651505a1493SJohn Baldwin acct_state = 0; 652505a1493SJohn Baldwin sx_xunlock(&acct_sx); 6533745c395SJulian Elischer kproc_exit(0); 654df8bae1dSRodney W. Grimes } 655