m_bsd386.c

unix系统下top命令的源代码

/*
 * top - a top users display for Unix
 *
 * SYNOPSIS:  For a BSD/386 system
 *	      Note memory statistic and process sizes could be wrong,
 *	      but ps gets them wrong too...
 *
 * DESCRIPTION:
 * This is the machine-dependent module for BSD/386
 * Works for:
 *	hp300
 *	i386
 *
 * CFLAGS: -DHAVE_GETOPT
 *
 * LIBS: -lkvm
 *
 * AUTHOR:  Christos Zoulas <[e-mail]>
 */

#include <sys/types.h>
#include <sys/signal.h>
#include <sys/param.h>

#include <stdio.h>
#include <nlist.h>
#include <math.h>
#ifdef __bsdi__
#include <sys/time.h>
#include <sys/proc.h>
#include <sys/vmmeter.h>
#endif
#include <kvm.h>
#include <sys/errno.h>
#include <sys/kinfo.h>
#include <sys/kinfo_proc.h>
#ifdef notyet
#define time __time
#define hz __hz
#include <sys/kernel.h>
#undef time
#undef hz
#endif
#include <sys/dir.h>
#ifdef __bsdi__
#include <sys/cpustats.h>
#include <sys/sysinfo.h>
#else
#include <sys/dkstat.h>
#endif
#include <sys/file.h>
#include <sys/time.h>


#define DOSWAP

#include "top.h"
#include "machine.h"
#include "utils.h"

#ifdef __bsdi__
#define	VMUNIX	"/bsd"
#else
#define VMUNIX	"/vmunix"
#endif
#define KMEM	"/dev/kmem"
#define MEM	"/dev/mem"
#ifdef DOSWAP
#define SWAP	"/dev/drum"
#endif

/* get_process_info passes back a handle.  This is what it looks like: */

struct handle
{
    struct kinfo_proc **next_proc;	/* points to next valid proc pointer */
    int remaining;		/* number of pointers remaining */
};

/* declarations for load_avg */
#include "loadavg.h"

#define PP(pp, field) ((pp)->kp_proc . field)
#define EP(pp, field) ((pp)->kp_eproc . field)
#define VP(pp, field) ((pp)->kp_eproc.e_vm . field)

/* define what weighted cpu is.  */
#define weighted_cpu(pct, pp) (PP((pp), p_time) == 0 ? 0.0 : \
			 ((pct) / (1.0 - exp(PP((pp), p_time) * logcpu))))

/* what we consider to be process size: */
#define PROCSIZE(pp) (VP((pp), vm_tsize) \
		      + VP((pp), vm_dsize) \
		      + VP((pp), vm_ssize))

/* definitions for indices in the nlist array */
#define X_CCPU		0
#ifdef __bsdi__
#define X_TOTAL	1
#else
#define X_CP_TIME	1
#endif
#define X_HZ		2
#define X_AVENRUN	3

static struct nlist nlst[] = {
    { "_ccpu" },		/* 0 */
#ifdef __bsdi__
    { "_total" },		/* 1 */
#else
    { "_cp_time" },		/* 1 */
#endif
    { "_hz" },			/* 2 */
    { "_averunnable" },		/* 3 */
    { 0 }
};

/*
 *  These definitions control the format of the per-process area
 */

static char header[] =
  "  PID X        PRI NICE  SIZE   RES STATE   TIME   WCPU    CPU COMMAND";
/* 0123456   -- field to fill in starts at header+6 */
#define UNAME_START 6

#define Proc_format \
"%5d %-8.8s %3d %4d %5s %5s %-5s %6s %5.2f%% %5.2f%% %.16s"


/* process state names for the "STATE" column of the display */
/* the extra nulls in the string "run" are for adding a slash and
   the processor number when needed */

static char *state_abbrev[] =
{
    "", "sleep", "WAIT", "run\0\0\0", "start", "zomb", "stop"
};


static kvm_t *kd;

/* values that we stash away in _init and use in later routines */

static double logcpu;

/* these are retrieved from the kernel in _init */

static long	hz;
static load_avg	ccpu;
static int	ncpu = 0;

/* these are offsets obtained via nlist and used in the get_ functions */

#ifdef __bsdi__
static unsigned long total_offset;
#else
static unsigned long cp_time_offset;
#endif
static unsigned long avenrun_offset;

#ifndef __bsdi__
/* these are for calculating cpu state percentages */

static u_long cp_time[CPUSTATES];
static u_long cp_old[CPUSTATES];
static u_long cp_diff[CPUSTATES];
#endif

/* these are for detailing the process states */

static int process_states[7];
static char *procstatenames[] = {
    "", " sleeping, ", " ABANDONED, ", " running, ", " starting, ",
    " zombie, ", " stopped, ",
    NULL
};

/* these are for detailing the cpu states */

static int cpu_states[CPUSTATES];
static char *cpustatenames[CPUSTATES+1] = {
    "user", "nice", "system", "idle", NULL
};

/* these are for detailing the memory statistics */

static int memory_stats[8];
static char *memorynames[] = {
    "Real: ", "K/", "K ", "Virt: ", "K/",
    "K ", "Free: ", "K", NULL
};

/* these are for keeping track of the proc array */

static int bytes;
static int nproc;
static int onproc = -1;
static int pref_len;
static struct kinfo_proc *pbase;
static struct kinfo_proc **pref;

/* these are for getting the memory statistics */

static int pageshift;		/* log base 2 of the pagesize */

/* define pagetok in terms of pageshift */

#define pagetok(size) ((size) << pageshift)

machine_init(statics)

struct statics *statics;

{
    register int i = 0;
    register int pagesize;

    if ((kd = kvm_open(VMUNIX, MEM, SWAP, O_RDONLY, "kvm_open")) == NULL)
	return -1;


    /* get the list of symbols we want to access in the kernel */
    (void) kvm_nlist(kd, nlst);
    if (nlst[0].n_type == 0)
    {
	fprintf(stderr, "top: nlist failed\n");
	return(-1);
    }

    /* make sure they were all found */
    if (i > 0 && check_nlist(nlst) > 0)
    {
	return(-1);
    }

    /* get the symbol values out of kmem */
    (void) getkval(nlst[X_HZ].n_value,     (int *)(&hz),	sizeof(hz),
	    nlst[X_HZ].n_name);
    (void) getkval(nlst[X_CCPU].n_value,   (int *)(&ccpu),	sizeof(ccpu),
	    nlst[X_CCPU].n_name);

    /* stash away certain offsets for later use */
#ifdef __bsdi__
    total_offset = nlst[X_TOTAL].n_value;
#else
    cp_time_offset = nlst[X_CP_TIME].n_value;
#endif
    avenrun_offset = nlst[X_AVENRUN].n_value;

    /* this is used in calculating WCPU -- calculate it ahead of time */
    logcpu = log(loaddouble(ccpu));

    pbase = NULL;
    pref = NULL;
    nproc = 0;
    onproc = -1;
    /* get the page size with "getpagesize" and calculate pageshift from it */
    pagesize = getpagesize();
    pageshift = 0;
    while (pagesize > 1)
    {
	pageshift++;
	pagesize >>= 1;
    }

    /* we only need the amount of log(2)1024 for our conversion */
    pageshift -= LOG1024;

    /* fill in the statics information */
    statics->procstate_names = procstatenames;
    statics->cpustate_names = cpustatenames;
    statics->memory_names = memorynames;

    /* all done! */
    return(0);
}

char *format_header(uname_field)

register char *uname_field;

{
    register char *ptr;

    ptr = header + UNAME_START;
    while (*uname_field != '\0')
    {
	*ptr++ = *uname_field++;
    }

    return(header);
}

get_system_info(si)

struct system_info *si;

{
    register u_long total;
    load_avg avenrun[3];
#ifdef __bsdi__
    struct cpustats cpu;
    struct sysinfo sys;
    int size;
#else
    load_avg *avenrunp = avenrun;
#endif

    /* get the various high-level data structures */
#ifdef __bsdi__
    size = sizeof(struct cpustats);
    if (getkerninfo(KINFO_CPU, &cpu, &size, 0) < 0) {
	perror("getkerninfo#1");
	abort();
    }
#ifdef notyet
    size = sizeof(struct sysinfo);
    if (getkerninfo(KINFO_SYSINFO, &sys, &size, 0) < 0) {
	perror("getkerninfo#2");
	abort();
    }
#endif /*notyet*/
#else
    (void) getkval(cp_time_offset, (int *)cp_time, sizeof(cp_time),
		   "_cp_time");
    (void) getkval(avenrun_offset, (int *)avenrun, sizeof(avenrun),
		   "_avenrun");
#endif

    /* convert load averages to doubles */
    {
	register int i;
	register double *infoloadp = si->load_avg;

	for (i = 0; i < CPUSTATES; i++)
	{
#ifdef __bsdi__
	    *infoloadp++ = ((double) cpu.cp_averunnable[i]) / FSCALE;
#else
	    *infoloadp++ = loaddouble(*avenrunp++);
#endif
	}
    }

    /* convert cp_time counts to percentages */
#ifdef __bsdi__
    {
	register int i;
	register double total, pct;

	total = 0.0;
	for (i = 0; i < CPUSTATES; i++)
		total += (double) cpu.cp_time[i];
	if (total == 0)
		pct = 0;
	else
		pct = 100 / total;