Do some fairly major include file cleanups to further separate kernelland
[dragonfly.git] / usr.bin / top / machine.c
1 /*
2  * top - a top users display for Unix
3  *
4  * SYNOPSIS:  For FreeBSD-2.x and later
5  *
6  * DESCRIPTION:
7  * Originally written for BSD4.4 system by Christos Zoulas.
8  * Ported to FreeBSD 2.x by Steven Wallace && Wolfram Schneider
9  * Order support hacked in from top-3.5beta6/machine/m_aix41.c
10  *   by Monte Mitzelfelt (for latest top see http://www.groupsys.com/topinfo/)
11  *
12  * This is the machine-dependent module for FreeBSD 2.2
13  * Works for:
14  *      FreeBSD 2.2.x, 3.x, 4.x, and probably FreeBSD 2.1.x
15  *
16  * LIBS: -lkvm
17  *
18  * AUTHOR:  Christos Zoulas <christos@ee.cornell.edu>
19  *          Steven Wallace  <swallace@freebsd.org>
20  *          Wolfram Schneider <wosch@FreeBSD.org>
21  *
22  * $FreeBSD: src/usr.bin/top/machine.c,v 1.29.2.2 2001/07/31 20:27:05 tmm Exp $
23  * $DragonFly: src/usr.bin/top/machine.c,v 1.10 2003/11/21 22:46:14 dillon Exp $
24  */
25
26
27 #include <sys/time.h>
28 #include <sys/types.h>
29 #include <sys/signal.h>
30 #include <sys/param.h>
31
32 #include "os.h"
33 #include <stdio.h>
34 #include <nlist.h>
35 #include <math.h>
36 #include <kvm.h>
37 #include <pwd.h>
38 #include <sys/errno.h>
39 #include <sys/sysctl.h>
40 #include <sys/dkstat.h>
41 #include <sys/file.h>
42 #include <sys/time.h>
43 #include <sys/user.h>
44 #include <sys/vmmeter.h>
45 #include <sys/resource.h>
46 #include <sys/rtprio.h>
47
48 /* Swap */
49 #include <stdlib.h>
50 #include <sys/conf.h>
51
52 #include <osreldate.h> /* for changes in kernel structures */
53
54 #include "top.h"
55 #include "machine.h"
56
57 static int check_nlist(struct nlist *);
58 static int getkval(unsigned long, int *, int, char *);
59 extern char* printable(char *);
60 int swapmode(int *retavail, int *retfree);
61 static int smpmode;
62 static int namelength;
63 static int cmdlength;
64
65
66 /* get_process_info passes back a handle.  This is what it looks like: */
67
68 struct handle
69 {
70     struct kinfo_proc **next_proc;      /* points to next valid proc pointer */
71     int remaining;              /* number of pointers remaining */
72 };
73
74 /* declarations for load_avg */
75 #include "loadavg.h"
76
77 #define PP(pp, field) ((pp)->kp_proc . field)
78 #define EP(pp, field) ((pp)->kp_eproc . field)
79 #define TP(pp, field) ((pp)->kp_thread . field)
80 #define VP(pp, field) ((pp)->kp_eproc.e_vm . field)
81
82 /* define what weighted cpu is.  */
83 #define weighted_cpu(pct, pp) (PP((pp), p_swtime) == 0 ? 0.0 : \
84                          ((pct) / (1.0 - exp(PP((pp), p_swtime) * logcpu))))
85
86 /* what we consider to be process size: */
87 #define PROCSIZE(pp) (VP((pp), vm_map.size) / 1024)
88
89 /* definitions for indices in the nlist array */
90
91 static struct nlist nlst[] = {
92 #define X_CCPU          0
93     { "_ccpu" },
94 #define X_CP_TIME       1
95     { "_cp_time" },
96 #define X_AVENRUN       2
97     { "_averunnable" },
98
99 #define X_BUFSPACE      3
100         { "_bufspace" },        /* K in buffer cache */
101
102 /* Last pid */
103 #define X_LASTPID       4
104     { "_nextpid" },             
105     { 0 }
106 };
107
108 /*
109  *  These definitions control the format of the per-process area
110  */
111
112 static char smp_header[] =
113   "  PID %-*.*s PRI NICE  SIZE    RES STATE  C   TIME   WCPU    CPU COMMAND";
114
115 #define smp_Proc_format \
116         "%5d %-*.*s %3d %3d%7s %6s %-6.6s %1x%7s %5.2f%% %5.2f%% %.*s"
117
118 static char up_header[] =
119   "  PID %-*.*s PRI NICE  SIZE    RES STATE    TIME   WCPU    CPU COMMAND";
120
121 #define up_Proc_format \
122         "%5d %-*.*s %3d %3d%7s %6s %-6.6s%.0d%7s %5.2f%% %5.2f%% %.*s"
123
124
125
126 /* process state names for the "STATE" column of the display */
127 /* the extra nulls in the string "run" are for adding a slash and
128    the processor number when needed */
129
130 char *state_abbrev[] =
131 {
132     "", "START", "RUN\0\0\0", "SLEEP", "STOP", "ZOMB",
133 };
134
135
136 static kvm_t *kd;
137
138 /* values that we stash away in _init and use in later routines */
139
140 static double logcpu;
141
142 /* these are retrieved from the kernel in _init */
143
144 static load_avg  ccpu;
145
146 /* these are offsets obtained via nlist and used in the get_ functions */
147
148 static unsigned long cp_time_offset;
149 static unsigned long avenrun_offset;
150 static unsigned long lastpid_offset;
151 static long lastpid;
152 static unsigned long bufspace_offset;
153 static long cnt;
154
155 /* these are for calculating cpu state percentages */
156
157 static long cp_time[CPUSTATES];
158 static long cp_old[CPUSTATES];
159 static long cp_diff[CPUSTATES];
160
161 /* these are for detailing the process states */
162
163 int process_states[6];
164 char *procstatenames[] = {
165     "", " starting, ", " running, ", " sleeping, ", " stopped, ",
166     " zombie, ",
167     NULL
168 };
169
170 /* these are for detailing the cpu states */
171
172 int cpu_states[CPUSTATES];
173 char *cpustatenames[] = {
174     "user", "nice", "system", "interrupt", "idle", NULL
175 };
176
177 /* these are for detailing the memory statistics */
178
179 int memory_stats[7];
180 char *memorynames[] = {
181     "K Active, ", "K Inact, ", "K Wired, ", "K Cache, ", "K Buf, ", "K Free",
182     NULL
183 };
184
185 int swap_stats[7];
186 char *swapnames[] = {
187 /*   0           1            2           3            4       5 */
188     "K Total, ", "K Used, ", "K Free, ", "% Inuse, ", "K In, ", "K Out",
189     NULL
190 };
191
192
193 /* these are for keeping track of the proc array */
194
195 static int nproc;
196 static int onproc = -1;
197 static int pref_len;
198 static struct kinfo_proc *pbase;
199 static struct kinfo_proc **pref;
200
201 /* these are for getting the memory statistics */
202
203 static int pageshift;           /* log base 2 of the pagesize */
204
205 /* define pagetok in terms of pageshift */
206
207 #define pagetok(size) ((size) << pageshift)
208
209 /* useful externals */
210 long percentages();
211
212 #ifdef ORDER
213 /* sorting orders. first is default */
214 char *ordernames[] = {
215     "cpu", "size", "res", "time", "pri", NULL
216 };
217 #endif
218
219 int
220 machine_init(struct statics *statics)
221 {
222     register int i = 0;
223     register int pagesize;
224     size_t modelen;
225     struct passwd *pw;
226
227     modelen = sizeof(smpmode);
228     if ((sysctlbyname("machdep.smp_active", &smpmode, &modelen, NULL, 0) < 0 &&
229          sysctlbyname("smp.smp_active", &smpmode, &modelen, NULL, 0) < 0) ||
230         modelen != sizeof(smpmode))
231             smpmode = 0;
232
233     while ((pw = getpwent()) != NULL) {
234         if (strlen(pw->pw_name) > namelength)
235             namelength = strlen(pw->pw_name);
236     }
237     if (namelength < 8)
238         namelength = 8;
239     if (smpmode && namelength > 13)
240         namelength = 13;
241     else if (namelength > 15)
242         namelength = 15;
243
244     if ((kd = kvm_open(NULL, NULL, NULL, O_RDONLY, "kvm_open")) == NULL)
245         return -1;
246
247
248     /* get the list of symbols we want to access in the kernel */
249     (void) kvm_nlist(kd, nlst);
250     if (nlst[0].n_type == 0)
251     {
252         fprintf(stderr, "top: nlist failed\n");
253         return(-1);
254     }
255
256     /* make sure they were all found */
257     if (i > 0 && check_nlist(nlst) > 0)
258     {
259         return(-1);
260     }
261
262     (void) getkval(nlst[X_CCPU].n_value,   (int *)(&ccpu),      sizeof(ccpu),
263             nlst[X_CCPU].n_name);
264
265     /* stash away certain offsets for later use */
266     cp_time_offset = nlst[X_CP_TIME].n_value;
267     avenrun_offset = nlst[X_AVENRUN].n_value;
268     lastpid_offset =  nlst[X_LASTPID].n_value;
269     bufspace_offset = nlst[X_BUFSPACE].n_value;
270
271     /* this is used in calculating WCPU -- calculate it ahead of time */
272     logcpu = log(loaddouble(ccpu));
273
274     pbase = NULL;
275     pref = NULL;
276     nproc = 0;
277     onproc = -1;
278     /* get the page size with "getpagesize" and calculate pageshift from it */
279     pagesize = getpagesize();
280     pageshift = 0;
281     while (pagesize > 1)
282     {
283         pageshift++;
284         pagesize >>= 1;
285     }
286
287     /* we only need the amount of log(2)1024 for our conversion */
288     pageshift -= LOG1024;
289
290     /* fill in the statics information */
291     statics->procstate_names = procstatenames;
292     statics->cpustate_names = cpustatenames;
293     statics->memory_names = memorynames;
294     statics->swap_names = swapnames;
295 #ifdef ORDER
296     statics->order_names = ordernames;
297 #endif
298
299     /* all done! */
300     return(0);
301 }
302
303 char *format_header(register char *uname_field)
304 {
305     register char *ptr;
306     static char Header[128];
307
308     snprintf(Header, sizeof(Header), smpmode ? smp_header : up_header,
309              namelength, namelength, uname_field);
310
311     cmdlength = 80 - strlen(Header) + 6;
312
313     return Header;
314 }
315
316 static int swappgsin = -1;
317 static int swappgsout = -1;
318 extern struct timeval timeout;
319
320 void
321 get_system_info(struct system_info *si)
322 {
323     long total;
324     load_avg avenrun[3];
325     int mib[2];
326     struct timeval boottime;
327     size_t bt_size;
328
329     /* get the cp_time array */
330     (void) getkval(cp_time_offset, (int *)cp_time, sizeof(cp_time),
331                    nlst[X_CP_TIME].n_name);
332     (void) getkval(avenrun_offset, (int *)avenrun, sizeof(avenrun),
333                    nlst[X_AVENRUN].n_name);
334
335     (void) getkval(lastpid_offset, (int *)(&lastpid), sizeof(lastpid),
336                    "!");
337
338     /* convert load averages to doubles */
339     {
340         register int i;
341         register double *infoloadp;
342         load_avg *avenrunp;
343
344 #ifdef notyet
345         struct loadavg sysload;
346         int size;
347         getkerninfo(KINFO_LOADAVG, &sysload, &size, 0);
348 #endif
349
350         infoloadp = si->load_avg;
351         avenrunp = avenrun;
352         for (i = 0; i < 3; i++)
353         {
354 #ifdef notyet
355             *infoloadp++ = ((double) sysload.ldavg[i]) / sysload.fscale;
356 #endif
357             *infoloadp++ = loaddouble(*avenrunp++);
358         }
359     }
360
361     /* convert cp_time counts to percentages */
362     total = percentages(CPUSTATES, cpu_states, cp_time, cp_old, cp_diff);
363
364     /* sum memory & swap statistics */
365     {
366         struct vmmeter vmm;
367         struct vmstats vms;
368         int vms_size = sizeof(vms);
369         int vmm_size = sizeof(vmm);
370         static unsigned int swap_delay = 0;
371         static int swapavail = 0;
372         static int swapfree = 0;
373         static int bufspace = 0;
374
375         if (sysctlbyname("vm.vmstats", &vms, &vms_size, NULL, 0)) {
376                 perror("sysctlbyname: vm.vmstats");
377                 exit(1);
378         }
379         if (sysctlbyname("vm.vmmeter", &vmm, &vmm_size, NULL, 0)) {
380                 perror("sysctlbyname: vm.vmstats");
381                 exit(1);
382         }
383         (void) getkval(bufspace_offset, (int *)(&bufspace), sizeof(bufspace),
384                    "_bufspace");
385
386         /* convert memory stats to Kbytes */
387         memory_stats[0] = pagetok(vms.v_active_count);
388         memory_stats[1] = pagetok(vms.v_inactive_count);
389         memory_stats[2] = pagetok(vms.v_wire_count);
390         memory_stats[3] = pagetok(vms.v_cache_count);
391         memory_stats[4] = bufspace / 1024;
392         memory_stats[5] = pagetok(vms.v_free_count);
393         memory_stats[6] = -1;
394
395         /* first interval */
396         if (swappgsin < 0) {
397             swap_stats[4] = 0;
398             swap_stats[5] = 0;
399         } 
400
401         /* compute differences between old and new swap statistic */
402         else {
403             swap_stats[4] = pagetok(((vmm.v_swappgsin - swappgsin)));
404             swap_stats[5] = pagetok(((vmm.v_swappgsout - swappgsout)));
405         }
406
407         swappgsin = vmm.v_swappgsin;
408         swappgsout = vmm.v_swappgsout;
409
410         /* call CPU heavy swapmode() only for changes */
411         if (swap_stats[4] > 0 || swap_stats[5] > 0 || swap_delay == 0) {
412             swap_stats[3] = swapmode(&swapavail, &swapfree);
413             swap_stats[0] = swapavail;
414             swap_stats[1] = swapavail - swapfree;
415             swap_stats[2] = swapfree;
416         }
417         swap_delay = 1;
418         swap_stats[6] = -1;
419     }
420
421     /* set arrays and strings */
422     si->cpustates = cpu_states;
423     si->memory = memory_stats;
424     si->swap = swap_stats;
425
426
427     if(lastpid > 0) {
428         si->last_pid = lastpid;
429     } else {
430         si->last_pid = -1;
431     }
432
433     /*
434      * Print how long system has been up.
435      * (Found by looking getting "boottime" from the kernel)
436      */
437     mib[0] = CTL_KERN;
438     mib[1] = KERN_BOOTTIME;
439     bt_size = sizeof(boottime);
440     if (sysctl(mib, 2, &boottime, &bt_size, NULL, 0) != -1 &&
441         boottime.tv_sec != 0) {
442         si->boottime = boottime;
443     } else {
444         si->boottime.tv_sec = -1;
445     }
446 }
447
448 static struct handle handle;
449
450 caddr_t get_process_info(struct system_info *si, struct process_select *sel,
451                          int (*compare)())
452 {
453     register int i;
454     register int total_procs;
455     register int active_procs;
456     register struct kinfo_proc **prefp;
457     register struct kinfo_proc *pp;
458
459     /* these are copied out of sel for speed */
460     int show_idle;
461     int show_self;
462     int show_system;
463     int show_uid;
464     int show_command;
465
466     
467     pbase = kvm_getprocs(kd, KERN_PROC_ALL, 0, &nproc);
468     if (nproc > onproc)
469         pref = (struct kinfo_proc **) realloc(pref, sizeof(struct kinfo_proc *)
470                 * (onproc = nproc));
471     if (pref == NULL || pbase == NULL) {
472         (void) fprintf(stderr, "top: Out of memory.\n");
473         quit(23);
474     }
475     /* get a pointer to the states summary array */
476     si->procstates = process_states;
477
478     /* set up flags which define what we are going to select */
479     show_idle = sel->idle;
480     show_self = sel->self;
481     show_system = sel->system;
482     show_uid = sel->uid != -1;
483     show_command = sel->command != NULL;
484
485     /* count up process states and get pointers to interesting procs */
486     total_procs = 0;
487     active_procs = 0;
488     memset((char *)process_states, 0, sizeof(process_states));
489     prefp = pref;
490     for (pp = pbase, i = 0; i < nproc; pp++, i++)
491     {
492         /*
493          *  Place pointers to each valid proc structure in pref[].
494          *  Process slots that are actually in use have a non-zero
495          *  status field.  Processes with P_SYSTEM set are system
496          *  processes---these get ignored unless show_sysprocs is set.
497          */
498         if (PP(pp, p_stat) != 0 &&
499             (show_self != PP(pp, p_pid)) &&
500             (show_system || ((PP(pp, p_flag) & P_SYSTEM) == 0)))
501         {
502             total_procs++;
503             process_states[(unsigned char) PP(pp, p_stat)]++;
504             if ((PP(pp, p_stat) != SZOMB) &&
505                 (show_idle || (PP(pp, p_pctcpu) != 0) || 
506                  (PP(pp, p_stat) == SRUN)) &&
507                 (!show_uid || EP(pp, e_ucred.cr_ruid) == (uid_t)sel->uid))
508             {
509                 *prefp++ = pp;
510                 active_procs++;
511             }
512         }
513     }
514
515     /* if requested, sort the "interesting" processes */
516     if (compare != NULL)
517     {
518         qsort((char *)pref, active_procs, sizeof(struct kinfo_proc *), compare);
519     }
520
521     /* remember active and total counts */
522     si->p_total = total_procs;
523     si->p_active = pref_len = active_procs;
524
525     /* pass back a handle */
526     handle.next_proc = pref;
527     handle.remaining = active_procs;
528     return((caddr_t)&handle);
529 }
530
531 char fmt[128];          /* static area where result is built */
532
533 char *format_next_process(caddr_t handle, char *(*get_userid)())
534 {
535     struct kinfo_proc *pp;
536     long cputime;
537     double pct;
538     struct handle *hp;
539     char status[16];
540     int state;
541     int nice;
542
543     /* find and remember the next proc structure */
544     hp = (struct handle *)handle;
545     pp = *(hp->next_proc++);
546     hp->remaining--;
547     
548     /* get the process's command name */
549     if ((PP(pp, p_flag) & P_INMEM) == 0) {
550         /*
551          * Print swapped processes as <pname>
552          */
553         char *comm = TP(pp, td_comm);
554 #define COMSIZ sizeof(TP(pp, td_comm))
555         char buf[COMSIZ];
556         (void) strncpy(buf, comm, COMSIZ);
557         comm[0] = '<';
558         (void) strncpy(&comm[1], buf, COMSIZ - 2);
559         comm[COMSIZ - 2] = '\0';
560         (void) strncat(comm, ">", COMSIZ - 1);
561         comm[COMSIZ - 1] = '\0';
562     }
563
564     /*
565      * Convert the process's runtime from microseconds to seconds.  This
566      * time includes the interrupt time although that is not wanted here.
567      * ps(1) is similarly sloppy.
568      */
569     cputime = (EP(pp, e_uticks) + EP(pp, e_sticks)) / 1000000;
570
571     /* calculate the base for cpu percentages */
572     pct = pctdouble(PP(pp, p_pctcpu));
573
574     /* generate "STATE" field */
575     switch (state = PP(pp, p_stat)) {
576         case SRUN:
577             if (smpmode && TP(pp, td_flags) & TDF_RUNNING)
578                 sprintf(status, "CPU%d", EP(pp, e_cpuid));
579             else
580                 strcpy(status, "RUN");
581             break;
582         case SSLEEP:
583             if (TP(pp, td_wmesg) != NULL) {
584                 sprintf(status, "%.6s", EP(pp, e_wmesg));
585                 break;
586             }
587             /* fall through */
588         default:
589
590             if (state >= 0 &&
591                 state < sizeof(state_abbrev) / sizeof(*state_abbrev))
592                     sprintf(status, "%.6s", state_abbrev[(unsigned char) state]);
593             else
594                     sprintf(status, "?%5d", state);
595             break;
596     }
597
598     /*
599      * idle time 0 - 31 -> nice value +21 - +52
600      * normal time      -> nice value -20 - +20 
601      * real time 0 - 31 -> nice value -52 - -21
602      * thread    0 - 31 -> nice value -53 -
603      */
604     switch(PP(pp, p_rtprio.type)) {
605     case RTP_PRIO_REALTIME:
606         nice = PRIO_MIN - 1 - RTP_PRIO_MAX + PP(pp, p_rtprio.prio);
607         break;
608     case RTP_PRIO_IDLE:
609         nice = PRIO_MAX + 1 + PP(pp, p_rtprio.prio);
610         break;
611     case RTP_PRIO_THREAD:
612         nice = PRIO_MIN - 1 - RTP_PRIO_MAX - PP(pp, p_rtprio.prio);
613         break;
614     default:
615         nice = PP(pp, p_nice);
616         break;
617     }
618
619
620     /* format this entry */
621     sprintf(fmt,
622             smpmode ? smp_Proc_format : up_Proc_format,
623             PP(pp, p_pid),
624             namelength, namelength,
625             (*get_userid)(EP(pp, e_ucred.cr_ruid)),
626             PP(pp, p_priority),
627             nice,
628             format_k2(PROCSIZE(pp)),
629             format_k2(pagetok(VP(pp, vm_rssize))),
630             status,
631             smpmode ? EP(pp, e_cpuid) : 0,
632             format_time(cputime),
633             100.0 * weighted_cpu(pct, pp),
634             100.0 * pct,
635             cmdlength,
636             printable(TP(pp, td_comm)));
637
638     /* return the result */
639     return(fmt);
640 }
641
642
643 /*
644  * check_nlist(nlst) - checks the nlist to see if any symbols were not
645  *              found.  For every symbol that was not found, a one-line
646  *              message is printed to stderr.  The routine returns the
647  *              number of symbols NOT found.
648  */
649
650 static int check_nlist(register struct nlist *nlst)
651 {
652     register int i;
653
654     /* check to see if we got ALL the symbols we requested */
655     /* this will write one line to stderr for every symbol not found */
656
657     i = 0;
658     while (nlst->n_name != NULL)
659     {
660         if (nlst->n_type == 0)
661         {
662             /* this one wasn't found */
663             (void) fprintf(stderr, "kernel: no symbol named `%s'\n",
664                            nlst->n_name);
665             i = 1;
666         }
667         nlst++;
668     }
669
670     return(i);
671 }
672
673
674 /*
675  *  getkval(offset, ptr, size, refstr) - get a value out of the kernel.
676  *      "offset" is the byte offset into the kernel for the desired value,
677  *      "ptr" points to a buffer into which the value is retrieved,
678  *      "size" is the size of the buffer (and the object to retrieve),
679  *      "refstr" is a reference string used when printing error meessages,
680  *          if "refstr" starts with a '!', then a failure on read will not
681  *          be fatal (this may seem like a silly way to do things, but I
682  *          really didn't want the overhead of another argument).
683  *      
684  */
685
686 static int getkval(unsigned long offset, int *ptr, int size, char *refstr)
687 {
688     if (kvm_read(kd, offset, (char *) ptr, size) != size)
689     {
690         if (*refstr == '!')
691         {
692             return(0);
693         }
694         else
695         {
696             fprintf(stderr, "top: kvm_read for %s: %s\n",
697                 refstr, strerror(errno));
698             quit(23);
699         }
700     }
701     return(1);
702 }
703     
704 /* comparison routines for qsort */
705
706 /*
707  *  proc_compare - comparison function for "qsort"
708  *      Compares the resource consumption of two processes using five
709  *      distinct keys.  The keys (in descending order of importance) are:
710  *      percent cpu, cpu ticks, state, resident set size, total virtual
711  *      memory usage.  The process states are ordered as follows (from least
712  *      to most important):  WAIT, zombie, sleep, stop, start, run.  The
713  *      array declaration below maps a process state index into a number
714  *      that reflects this ordering.
715  */
716
717 static unsigned char sorted_state[] =
718 {
719     0,  /* not used             */
720     3,  /* sleep                */
721     1,  /* ABANDONED (WAIT)     */
722     6,  /* run                  */
723     5,  /* start                */
724     2,  /* zombie               */
725     4   /* stop                 */
726 };
727  
728
729 #define ORDERKEY_PCTCPU \
730   if (lresult = (long) PP(p2, p_pctcpu) - (long) PP(p1, p_pctcpu), \
731      (result = lresult > 0 ? 1 : lresult < 0 ? -1 : 0) == 0)
732
733 #define CPTICKS(p)      (EP(p, e_uticks) + EP(p, e_sticks))
734
735 #define ORDERKEY_CPTICKS \
736   if ((result = CPTICKS(p2) > CPTICKS(p1) ? 1 : \
737                 CPTICKS(p2) < CPTICKS(p1) ? -1 : 0) == 0)
738
739 #define ORDERKEY_STATE \
740   if ((result = sorted_state[(unsigned char) PP(p2, p_stat)] - \
741                 sorted_state[(unsigned char) PP(p1, p_stat)]) == 0)
742
743 #define ORDERKEY_PRIO \
744   if ((result = PP(p2, p_priority) - PP(p1, p_priority)) == 0)
745
746 #define ORDERKEY_RSSIZE \
747   if ((result = VP(p2, vm_rssize) - VP(p1, vm_rssize)) == 0) 
748
749 #define ORDERKEY_MEM \
750   if ( (result = PROCSIZE(p2) - PROCSIZE(p1)) == 0 )
751
752 /* compare_cpu - the comparison function for sorting by cpu percentage */
753
754 int
755 #ifdef ORDER
756 compare_cpu(struct proc **pp1, struct proc **pp2)
757 #else
758 proc_compare(struct proc **pp1, struct proc **pp2)
759 #endif
760 {
761     register struct kinfo_proc *p1;
762     register struct kinfo_proc *p2;
763     register int result;
764     register pctcpu lresult;
765
766     /* remove one level of indirection */
767     p1 = *(struct kinfo_proc **) pp1;
768     p2 = *(struct kinfo_proc **) pp2;
769
770     ORDERKEY_PCTCPU
771     ORDERKEY_CPTICKS
772     ORDERKEY_STATE
773     ORDERKEY_PRIO
774     ORDERKEY_RSSIZE
775     ORDERKEY_MEM
776     ;
777
778     return(result);
779 }
780
781 #ifdef ORDER
782 /* compare routines */
783 int compare_size(), compare_res(), compare_time(), compare_prio();
784
785 int (*proc_compares[])() = {
786     compare_cpu,
787     compare_size,
788     compare_res,
789     compare_time,
790     compare_prio,
791     NULL
792 };
793
794 /* compare_size - the comparison function for sorting by total memory usage */
795
796 int
797 compare_size(struct proc **pp1, struct proc **pp2)
798 {
799     register struct kinfo_proc *p1;
800     register struct kinfo_proc *p2;
801     register int result;
802     register pctcpu lresult;
803
804     /* remove one level of indirection */
805     p1 = *(struct kinfo_proc **) pp1;
806     p2 = *(struct kinfo_proc **) pp2;
807
808     ORDERKEY_MEM
809     ORDERKEY_RSSIZE
810     ORDERKEY_PCTCPU
811     ORDERKEY_CPTICKS
812     ORDERKEY_STATE
813     ORDERKEY_PRIO
814     ;
815
816     return(result);
817 }
818
819 /* compare_res - the comparison function for sorting by resident set size */
820
821 int
822 compare_res(struct proc **pp1, struct proc **pp2)
823 {
824     register struct kinfo_proc *p1;
825     register struct kinfo_proc *p2;
826     register int result;
827     register pctcpu lresult;
828
829     /* remove one level of indirection */
830     p1 = *(struct kinfo_proc **) pp1;
831     p2 = *(struct kinfo_proc **) pp2;
832
833     ORDERKEY_RSSIZE
834     ORDERKEY_MEM
835     ORDERKEY_PCTCPU
836     ORDERKEY_CPTICKS
837     ORDERKEY_STATE
838     ORDERKEY_PRIO
839     ;
840
841     return(result);
842 }
843
844 /* compare_time - the comparison function for sorting by total cpu time */
845
846 int
847 compare_time(struct proc **pp1, struct proc **pp2)
848 {
849     register struct kinfo_proc *p1;
850     register struct kinfo_proc *p2;
851     register int result;
852     register pctcpu lresult;
853   
854     /* remove one level of indirection */
855     p1 = *(struct kinfo_proc **) pp1;
856     p2 = *(struct kinfo_proc **) pp2;
857
858     ORDERKEY_CPTICKS
859     ORDERKEY_PCTCPU
860     ORDERKEY_STATE
861     ORDERKEY_PRIO
862     ORDERKEY_RSSIZE
863     ORDERKEY_MEM
864     ;
865
866       return(result);
867   }
868   
869 /* compare_prio - the comparison function for sorting by cpu percentage */
870
871 int
872 compare_prio(struct proc **pp1, struct proc **pp2)
873 {
874     register struct kinfo_proc *p1;
875     register struct kinfo_proc *p2;
876     register int result;
877     register pctcpu lresult;
878
879     /* remove one level of indirection */
880     p1 = *(struct kinfo_proc **) pp1;
881     p2 = *(struct kinfo_proc **) pp2;
882
883     ORDERKEY_PRIO
884     ORDERKEY_CPTICKS
885     ORDERKEY_PCTCPU
886     ORDERKEY_STATE
887     ORDERKEY_RSSIZE
888     ORDERKEY_MEM
889     ;
890
891     return(result);
892 }
893 #endif
894
895 /*
896  * proc_owner(pid) - returns the uid that owns process "pid", or -1 if
897  *              the process does not exist.
898  *              It is EXTREMLY IMPORTANT that this function work correctly.
899  *              If top runs setuid root (as in SVR4), then this function
900  *              is the only thing that stands in the way of a serious
901  *              security problem.  It validates requests for the "kill"
902  *              and "renice" commands.
903  */
904
905 int proc_owner(int pid)
906 {
907     register int cnt;
908     register struct kinfo_proc **prefp;
909     register struct kinfo_proc *pp;
910
911     prefp = pref;
912     cnt = pref_len;
913     while (--cnt >= 0)
914     {
915         pp = *prefp++;  
916         if (PP(pp, p_pid) == (pid_t)pid)
917         {
918             return((int)EP(pp, e_ucred.cr_ruid));
919         }
920     }
921     return(-1);
922 }
923
924
925 /*
926  * swapmode is based on a program called swapinfo written
927  * by Kevin Lahey <kml@rokkaku.atl.ga.us>.
928  */
929
930 #define SVAR(var) __STRING(var) /* to force expansion */
931 #define KGET(idx, var)                                                  \
932         KGET1(idx, &var, sizeof(var), SVAR(var))
933 #define KGET1(idx, p, s, msg)                                           \
934         KGET2(nlst[idx].n_value, p, s, msg)
935 #define KGET2(addr, p, s, msg)                                          \
936         if (kvm_read(kd, (u_long)(addr), p, s) != s) {                  \
937                 warnx("cannot read %s: %s", msg, kvm_geterr(kd));       \
938                 return (0);                                             \
939        }
940 #define KGETRET(addr, p, s, msg)                                        \
941         if (kvm_read(kd, (u_long)(addr), p, s) != s) {                  \
942                 warnx("cannot read %s: %s", msg, kvm_geterr(kd));       \
943                 return (0);                                             \
944         }
945
946
947 int
948 swapmode(int *retavail, int *retfree)
949 {
950         int n;
951         int pagesize = getpagesize();
952         struct kvm_swap swapary[1];
953
954         *retavail = 0;
955         *retfree = 0;
956
957 #define CONVERT(v)      ((quad_t)(v) * pagesize / 1024)
958
959         n = kvm_getswapinfo(kd, swapary, 1, 0);
960         if (n < 0 || swapary[0].ksw_total == 0)
961                 return(0);
962
963         *retavail = CONVERT(swapary[0].ksw_total);
964         *retfree = CONVERT(swapary[0].ksw_total - swapary[0].ksw_used);
965
966         n = (int)((double)swapary[0].ksw_used * 100.0 /
967             (double)swapary[0].ksw_total);
968         return(n);
969 }
970