date and time created 94/04/13 10:27:13 by eric

[unix-history] / usr / src / usr.bin / systat / pigs.c

/*-
* Copyright (c) 1980, 1992, 1993
*      The Regents of the University of California.  All rights reserved.
*
* %sccs.include.redist.c%
*/

#ifndef lint
static char sccsid[] = "@(#)pigs.c      8.2 (Berkeley) %G%";
#endif /* not lint */

/*
* Pigs display from Bill Reeves at Lucasfilm
*/

#include <sys/param.h>
#include <sys/dkstat.h>
#include <sys/dir.h>
#include <sys/time.h>
#include <sys/proc.h>
#include <sys/sysctl.h>

#include <curses.h>
#include <math.h>
#include <nlist.h>
#include <pwd.h>
#include <stdlib.h>

#include "extern.h"
#include "systat.h"

int compar __P((const void *, const void *));

static int nproc;
static struct p_times {
       float pt_pctcpu;
       struct kinfo_proc *pt_kp;
} *pt;

static long stime[CPUSTATES];
static int     fscale;
static double  lccpu;

WINDOW *
openpigs()
{
       return (subwin(stdscr, LINES-5-1, 0, 5, 0));
}

void
closepigs(w)
       WINDOW *w;
{
       if (w == NULL)
               return;
       wclear(w);
       wrefresh(w);
       delwin(w);
}


void
showpigs()
{
       register int i, j, y, k;
       struct  eproc *ep;
       float total;
       int factor;
       char *uname, *pname, pidname[30];

       if (pt == NULL)
               return;
       /* Accumulate the percent of cpu per user. */
       total = 0.0;
       for (i = 0; i <= nproc; i++) {
               /* Accumulate the percentage. */
               total += pt[i].pt_pctcpu;
       }

       if (total < 1.0)
               total = 1.0;
       factor = 50.0/total;

       qsort(pt, nproc + 1, sizeof (struct p_times), compar);
       y = 1;
       i = nproc + 1;
       if (i > wnd->maxy-1)
               i = wnd->maxy-1;
       for (k = 0; i > 0 && pt[k].pt_pctcpu > 0.01; i--, y++, k++) {
               if (pt[k].pt_kp == NULL) {
                       uname = "";
                       pname = "<idle>";
               }
               else {
                       ep = &pt[k].pt_kp->kp_eproc;
                       uname = (char *)user_from_uid(ep->e_ucred.cr_uid, 0);
                       pname = pt[k].pt_kp->kp_proc.p_comm;
               }
               wmove(wnd, y, 0);
               wclrtoeol(wnd);
               mvwaddstr(wnd, y, 0, uname);
               sprintf(pidname, "%10.10s", pname, 0);
               mvwaddstr(wnd, y, 9, pidname);
               wmove(wnd, y, 20);
               for (j = pt[k].pt_pctcpu*factor + 0.5; j > 0; j--)
                       waddch(wnd, 'X');
       }
       wmove(wnd, y, 0); wclrtobot(wnd);
}

static struct nlist namelist[] = {
#define X_FIRST         0
#define X_CPTIME        0
       { "_cp_time" },
#define X_CCPU          1
       { "_ccpu" },
#define X_FSCALE        2
       { "_fscale" },

       { "" }
};

int
initpigs()
{
       fixpt_t ccpu;

       if (namelist[X_FIRST].n_type == 0) {
               if (kvm_nlist(kd, namelist)) {
                       nlisterr(namelist);
                       return(0);
               }
               if (namelist[X_FIRST].n_type == 0) {
                       error("namelist failed");
                       return(0);
               }
       }
       KREAD(NPTR(X_CPTIME), stime, sizeof (stime));
       NREAD(X_CCPU, &ccpu, LONG);
       NREAD(X_FSCALE,  &fscale, LONG);
       lccpu = log((double) ccpu / fscale);

       return(1);
}

void
fetchpigs()
{
       register int i;
       register float time;
       register struct proc *pp;
       register float *pctp;
       struct kinfo_proc *kpp;
       long ctime[CPUSTATES];
       double t;
       static int lastnproc = 0;

       if (namelist[X_FIRST].n_type == 0)
               return;
       if ((kpp = kvm_getprocs(kd, KERN_PROC_ALL, 0, &nproc)) == NULL) {
               error("%s", kvm_geterr(kd));
               if (pt)
                       free(pt);
               return;
       }
       if (nproc > lastnproc) {
               free(pt);
               if ((pt =
                   malloc((nproc + 1) * sizeof(struct p_times))) == NULL) {
                       error("Out of memory");
                       die(0);
               }
       }
       lastnproc = nproc;
       /*
        * calculate %cpu for each proc
        */
       for (i = 0; i < nproc; i++) {
               pt[i].pt_kp = &kpp[i];
               pp = &kpp[i].kp_proc;
               pctp = &pt[i].pt_pctcpu;
               time = pp->p_swtime;
               if (time == 0 || (pp->p_flag & P_INMEM) == 0)
                       *pctp = 0;
               else
                       *pctp = ((double) pp->p_pctcpu / 
                                       fscale) / (1.0 - exp(time * lccpu));
       }
       /*
        * and for the imaginary "idle" process
        */
       KREAD(NPTR(X_CPTIME), ctime, sizeof (ctime));
       t = 0;
       for (i = 0; i < CPUSTATES; i++)
               t += ctime[i] - stime[i];
       if (t == 0.0)
               t = 1.0;
       pt[nproc].pt_kp = NULL;
       pt[nproc].pt_pctcpu = (ctime[CP_IDLE] - stime[CP_IDLE]) / t;
       for (i = 0; i < CPUSTATES; i++)
               stime[i] = ctime[i];
}

void
labelpigs()
{
       wmove(wnd, 0, 0);
       wclrtoeol(wnd);
       mvwaddstr(wnd, 0, 20,
           "/0   /10  /20  /30  /40  /50  /60  /70  /80  /90  /100");
}

int
compar(a, b)
       const void *a, *b;
{
       return (((struct p_times *) a)->pt_pctcpu >
               ((struct p_times *) b)->pt_pctcpu)? -1: 1;
}