ANSIfication; bug report 4.3BSD/bin/223

[unix-history] / usr / src / sys / kern / kern_resource.c

/*
* Copyright (c) 1982, 1986 Regents of the University of California.
* All rights reserved.  The Berkeley software License Agreement
* specifies the terms and conditions for redistribution.
*
*      @(#)kern_resource.c     7.3 (Berkeley) %G%
*/

#include "param.h"
#include "dir.h"
#include "user.h"
#include "proc.h"

/*
* Resource controls and accounting.
*/

getpriority()
{
       register struct a {
               int     which;
               int     who;
       } *uap = (struct a *)u.u_ap;
       register struct proc *p;
       register int low = PRIO_MAX + 1;

       switch (uap->which) {

       case PRIO_PROCESS:
               if (uap->who == 0)
                       p = u.u_procp;
               else
                       p = pfind(uap->who);
               if (p == 0)
                       break;
               low = p->p_nice;
               break;

       case PRIO_PGRP: {
               register struct pgrp *pg;

               if (uap->who == 0)
                       pg = u.u_procp->p_pgrp;
               else if ((pg = pgfind(uap->who)) == NULL)
                       break;
               for (p = pg->pg_mem; p != NULL; p = p->p_pgrpnxt) {
                       if (p->p_nice < low)
                               low = p->p_nice;
               }
               break;
       }

       case PRIO_USER:
               if (uap->who == 0)
                       uap->who = u.u_uid;
               for (p = allproc; p != NULL; p = p->p_nxt) {
                       if (p->p_uid == uap->who &&
                           p->p_nice < low)
                               low = p->p_nice;
               }
               break;

       default:
               u.u_error = EINVAL;
               return;
       }
       if (low == PRIO_MAX + 1) {
               u.u_error = ESRCH;
               return;
       }
       u.u_r.r_val1 = low;
}

setpriority()
{
       register struct a {
               int     which;
               int     who;
               int     prio;
       } *uap = (struct a *)u.u_ap;
       register struct proc *p;
       int found = 0;

       switch (uap->which) {

       case PRIO_PROCESS:
               if (uap->who == 0)
                       p = u.u_procp;
               else
                       p = pfind(uap->who);
               if (p == 0)
                       break;
               donice(p, uap->prio);
               found++;
               break;

       case PRIO_PGRP: {
               register struct pgrp *pg;
                
               if (uap->who == 0)
                       pg = u.u_procp->p_pgrp;
               else if ((pg = pgfind(uap->who)) == NULL)
                       break;
               for (p = pg->pg_mem; p != NULL; p = p->p_pgrpnxt) {
                       donice(p, uap->prio);
                       found++;
               }
               break;
       }

       case PRIO_USER:
               if (uap->who == 0)
                       uap->who = u.u_uid;
               for (p = allproc; p != NULL; p = p->p_nxt)
                       if (p->p_uid == uap->who) {
                               donice(p, uap->prio);
                               found++;
                       }
               break;

       default:
               u.u_error = EINVAL;
               return;
       }
       if (found == 0)
               u.u_error = ESRCH;
}

donice(p, n)
       register struct proc *p;
       register int n;
{

       if (u.u_uid && u.u_ruid &&
           u.u_uid != p->p_uid && u.u_ruid != p->p_uid) {
               u.u_error = EPERM;
               return;
       }
       if (n > PRIO_MAX)
               n = PRIO_MAX;
       if (n < PRIO_MIN)
               n = PRIO_MIN;
       if (n < p->p_nice && !suser()) {
               u.u_error = EACCES;
               return;
       }
       p->p_nice = n;
       (void) setpri(p);
}

setrlimit()
{
       register struct a {
               u_int   which;
               struct  rlimit *lim;
       } *uap = (struct a *)u.u_ap;
       struct rlimit alim;
       register struct rlimit *alimp;
       extern unsigned maxdmap;

       if (uap->which >= RLIM_NLIMITS) {
               u.u_error = EINVAL;
               return;
       }
       alimp = &u.u_rlimit[uap->which];
       u.u_error = copyin((caddr_t)uap->lim, (caddr_t)&alim,
               sizeof (struct rlimit));
       if (u.u_error)
               return;
       if (alim.rlim_cur > alimp->rlim_max || alim.rlim_max > alimp->rlim_max)
               if (!suser())
                       return;
       switch (uap->which) {

       case RLIMIT_DATA:
               if (alim.rlim_cur > maxdmap)
                       alim.rlim_cur = maxdmap;
               if (alim.rlim_max > maxdmap)
                       alim.rlim_max = maxdmap;
               break;

       case RLIMIT_STACK:
               if (alim.rlim_cur > maxdmap)
                       alim.rlim_cur = maxdmap;
               if (alim.rlim_max > maxdmap)
                       alim.rlim_max = maxdmap;
               break;
       }
       *alimp = alim;
       if (uap->which == RLIMIT_RSS)
               u.u_procp->p_maxrss = alim.rlim_cur/NBPG;
}

getrlimit()
{
       register struct a {
               u_int   which;
               struct  rlimit *rlp;
       } *uap = (struct a *)u.u_ap;

       if (uap->which >= RLIM_NLIMITS) {
               u.u_error = EINVAL;
               return;
       }
       u.u_error = copyout((caddr_t)&u.u_rlimit[uap->which], (caddr_t)uap->rlp,
           sizeof (struct rlimit));
}

getrusage()
{
       register struct a {
               int     who;
               struct  rusage *rusage;
       } *uap = (struct a *)u.u_ap;
       register struct rusage *rup;

       switch (uap->who) {

       case RUSAGE_SELF:
               rup = &u.u_ru;
               break;

       case RUSAGE_CHILDREN:
               rup = &u.u_cru;
               break;

       default:
               u.u_error = EINVAL;
               return;
       }
       u.u_error = copyout((caddr_t)rup, (caddr_t)uap->rusage,
           sizeof (struct rusage));
}

ruadd(ru, ru2)
       register struct rusage *ru, *ru2;
{
       register long *ip, *ip2;
       register int i;

       timevaladd(&ru->ru_utime, &ru2->ru_utime);
       timevaladd(&ru->ru_stime, &ru2->ru_stime);
       if (ru->ru_maxrss < ru2->ru_maxrss)
               ru->ru_maxrss = ru2->ru_maxrss;
       ip = &ru->ru_first; ip2 = &ru2->ru_first;
       for (i = &ru->ru_last - &ru->ru_first; i > 0; i--)
               *ip++ += *ip2++;
}