75a0c92cc0
Submitted by: Antranig Vartanian <antranigv@freebsd.am> Reviewed by: imp, philip Differential Revision: https://reviews.freebsd.org/D22871
1222 lines
27 KiB
C
1222 lines
27 KiB
C
/*-
|
|
* Top users/processes display for Unix
|
|
*
|
|
* This program may be freely redistributed,
|
|
* but this entire comment MUST remain intact.
|
|
*
|
|
* Copyright (c) 1984, 1989, William LeFebvre, Rice University
|
|
* Copyright (c) 1989 - 1994, William LeFebvre, Northwestern University
|
|
* Copyright (c) 1994, 1995, William LeFebvre, Argonne National Laboratory
|
|
* Copyright (c) 1996, William LeFebvre, Group sys Consulting
|
|
*
|
|
* $FreeBSD$
|
|
*/
|
|
|
|
#include <sys/types.h>
|
|
#include <sys/time.h>
|
|
#include <sys/cdefs.h>
|
|
#include <sys/limits.h>
|
|
#include <sys/resource.h>
|
|
#include <sys/select.h>
|
|
#include <sys/signal.h>
|
|
|
|
#include <assert.h>
|
|
#include <err.h>
|
|
#include <errno.h>
|
|
#include <getopt.h>
|
|
#include <jail.h>
|
|
#include <locale.h>
|
|
#include <stdbool.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <signal.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
#include "commands.h"
|
|
#include "display.h" /* interface to display package */
|
|
#include "screen.h" /* interface to screen package */
|
|
#include "top.h"
|
|
#include "machine.h"
|
|
#include "utils.h"
|
|
#include "username.h"
|
|
|
|
/* Size of the stdio buffer given to stdout */
|
|
#define Buffersize 2048
|
|
|
|
char copyright[] =
|
|
"Copyright (c) 1984 through 1996, William LeFebvre";
|
|
|
|
typedef void sigret_t;
|
|
|
|
/* The buffer that stdio will use */
|
|
static char stdoutbuf[Buffersize];
|
|
|
|
static int fmt_flags = 0;
|
|
int pcpu_stats = false;
|
|
|
|
/* signal handling routines */
|
|
static sigret_t leave(int);
|
|
static sigret_t tstop(int);
|
|
static sigret_t top_winch(int);
|
|
|
|
static volatile sig_atomic_t leaveflag;
|
|
static volatile sig_atomic_t tstopflag;
|
|
static volatile sig_atomic_t winchflag;
|
|
|
|
/* values which need to be accessed by signal handlers */
|
|
static int max_topn; /* maximum displayable processes */
|
|
|
|
/* miscellaneous things */
|
|
struct process_select ps;
|
|
pid_t mypid;
|
|
|
|
/* pointers to display routines */
|
|
static void (*d_loadave)(int mpid, double *avenrun) = i_loadave;
|
|
static void (*d_procstates)(int total, int *brkdn) = i_procstates;
|
|
static void (*d_cpustates)(int *states) = i_cpustates;
|
|
static void (*d_memory)(int *stats) = i_memory;
|
|
static void (*d_arc)(int *stats) = i_arc;
|
|
static void (*d_carc)(int *stats) = i_carc;
|
|
static void (*d_swap)(int *stats) = i_swap;
|
|
static void (*d_message)(void) = i_message;
|
|
static void (*d_header)(const char *text) = i_header;
|
|
static void (*d_process)(int line, char *thisline) = i_process;
|
|
|
|
static void reset_display(void);
|
|
|
|
static const struct option longopts[] = {
|
|
{ "cpu-display-mode", no_argument, NULL, 'C' }, /* differs from orignal */
|
|
/* D reserved */
|
|
{ "thread", no_argument, NULL, 'H' },
|
|
{ "idle-procs", no_argument, NULL, 'I' },
|
|
{ "jail", required_argument, NULL, 'J' },
|
|
{ "per-cpu", no_argument, NULL, 'P' },
|
|
{ "system-procs", no_argument, NULL, 'S' },
|
|
{ "thread-id", no_argument, NULL, 'T' }, /* differs from orignal */
|
|
{ "user", required_argument, NULL, 'U' },
|
|
{ "all", no_argument, NULL, 'a' },
|
|
{ "batch", no_argument, NULL, 'b' },
|
|
/* c reserved */
|
|
{ "displays", required_argument, NULL, 'd' },
|
|
{ "interactive", no_argument, NULL, 'i' },
|
|
{ "jail-id", no_argument, NULL, 'j' },
|
|
{ "display-mode", required_argument, NULL, 'm' },
|
|
/* n is identical to batch */
|
|
{ "sort-order", required_argument, NULL, 'o' },
|
|
{ "pid", required_argument, NULL, 'p' },
|
|
{ "quick", no_argument, NULL, 'q' },
|
|
{ "delay", required_argument, NULL, 's' },
|
|
{ "threads", no_argument, NULL, 't' },
|
|
{ "uids", no_argument, NULL, 'u' },
|
|
{ "version", no_argument, NULL, 'v' },
|
|
{ "swap", no_argument, NULL, 'w' },
|
|
{ "system-idle-procs", no_argument, NULL, 'z' },
|
|
{ NULL, 0, NULL, 0 }
|
|
};
|
|
|
|
static void
|
|
reset_uids(void)
|
|
{
|
|
for (size_t i = 0; i < TOP_MAX_UIDS; ++i)
|
|
ps.uid[i] = -1;
|
|
}
|
|
|
|
static int
|
|
add_uid(int uid)
|
|
{
|
|
size_t i = 0;
|
|
|
|
/* Add the uid if there's room */
|
|
for (; i < TOP_MAX_UIDS; ++i)
|
|
{
|
|
if (ps.uid[i] == -1 || ps.uid[i] == uid)
|
|
{
|
|
ps.uid[i] = uid;
|
|
break;
|
|
}
|
|
}
|
|
|
|
return (i == TOP_MAX_UIDS);
|
|
}
|
|
|
|
static void
|
|
rem_uid(int uid)
|
|
{
|
|
size_t i = 0;
|
|
size_t where = TOP_MAX_UIDS;
|
|
|
|
/* Look for the user to remove - no problem if it's not there */
|
|
for (; i < TOP_MAX_UIDS; ++i)
|
|
{
|
|
if (ps.uid[i] == -1)
|
|
break;
|
|
if (ps.uid[i] == uid)
|
|
where = i;
|
|
}
|
|
|
|
/* Make sure we don't leave a hole in the middle */
|
|
if (where != TOP_MAX_UIDS)
|
|
{
|
|
ps.uid[where] = ps.uid[i-1];
|
|
ps.uid[i-1] = -1;
|
|
}
|
|
}
|
|
|
|
static int
|
|
handle_user(char *buf, size_t buflen)
|
|
{
|
|
int rc = 0;
|
|
int uid = -1;
|
|
char *buf2 = buf;
|
|
|
|
new_message(MT_standout, "Username to show (+ for all): ");
|
|
if (readline(buf, buflen, false) <= 0)
|
|
{
|
|
clear_message();
|
|
return (rc);
|
|
}
|
|
|
|
if (buf[0] == '+' || buf[0] == '-')
|
|
{
|
|
if (buf[1] == '\0')
|
|
{
|
|
reset_uids();
|
|
goto end;
|
|
}
|
|
else
|
|
++buf2;
|
|
}
|
|
|
|
if ((uid = userid(buf2)) == -1)
|
|
{
|
|
new_message(MT_standout, " %s: unknown user", buf2);
|
|
rc = 1;
|
|
goto end;
|
|
}
|
|
|
|
if (buf2 == buf)
|
|
{
|
|
reset_uids();
|
|
ps.uid[0] = uid;
|
|
goto end;
|
|
}
|
|
|
|
if (buf[0] == '+')
|
|
{
|
|
if (add_uid(uid))
|
|
{
|
|
new_message(MT_standout, " too many users, reset with '+'");
|
|
rc = 1;
|
|
goto end;
|
|
}
|
|
}
|
|
else
|
|
rem_uid(uid);
|
|
|
|
end:
|
|
putchar('\r');
|
|
return (rc);
|
|
}
|
|
|
|
int
|
|
main(int argc, const char *argv[])
|
|
{
|
|
int i;
|
|
int active_procs;
|
|
|
|
struct system_info system_info;
|
|
struct statics statics;
|
|
void * processes;
|
|
|
|
static char tempbuf1[50];
|
|
static char tempbuf2[50];
|
|
sigset_t old_sigmask, new_sigmask;
|
|
int topn = Infinity;
|
|
struct timeval delay = { 2, 0 };
|
|
int displays = 0; /* indicates unspecified */
|
|
int sel_ret = 0;
|
|
time_t curr_time;
|
|
char *(*get_userid)(int) = username;
|
|
const char *uname_field = "USERNAME";
|
|
const char *header_text;
|
|
char *env_top;
|
|
const char **preset_argv;
|
|
int preset_argc = 0;
|
|
const char **av = NULL;
|
|
int ac = -1;
|
|
bool do_unames = true;
|
|
char interactive = 2;
|
|
char warnings = 0;
|
|
char topn_specified = false;
|
|
char ch;
|
|
char no_command = 1;
|
|
struct timeval timeout;
|
|
char *order_name = NULL;
|
|
int order_index = 0;
|
|
fd_set readfds;
|
|
char *nptr;
|
|
|
|
/* set the buffer for stdout */
|
|
#ifdef DEBUG
|
|
extern FILE *debug;
|
|
debug = fopen("debug.run", "w");
|
|
setbuffer(stdout, NULL, 0);
|
|
#else
|
|
setbuffer(stdout, stdoutbuf, Buffersize);
|
|
#endif
|
|
|
|
if (setlocale(LC_ALL, "") == NULL) {
|
|
fprintf(stderr, "invalid locale.\n");
|
|
exit(1);
|
|
}
|
|
|
|
mypid = getpid();
|
|
|
|
/* get our name */
|
|
/* initialize some selection options */
|
|
ps.idle = true;
|
|
ps.self = true;
|
|
ps.system = false;
|
|
reset_uids();
|
|
ps.thread = false;
|
|
ps.wcpu = 1;
|
|
ps.jid = -1;
|
|
ps.jail = false;
|
|
ps.swap = false;
|
|
ps.kidle = true;
|
|
ps.pid = -1;
|
|
ps.command = NULL;
|
|
ps.thread_id = false;
|
|
|
|
/* get preset options from the environment */
|
|
if ((env_top = getenv("TOP")) != NULL)
|
|
{
|
|
av = preset_argv = argparse(env_top, &preset_argc);
|
|
ac = preset_argc;
|
|
|
|
/* set the dummy argument to an explanatory message, in case
|
|
getopt encounters a bad argument */
|
|
preset_argv[0] = "while processing environment";
|
|
}
|
|
|
|
/* process options */
|
|
do {
|
|
/* if we're done doing the presets, then process the real arguments */
|
|
if (preset_argc == 0)
|
|
{
|
|
ac = argc;
|
|
av = argv;
|
|
|
|
/* this should keep getopt happy... */
|
|
optind = 1;
|
|
}
|
|
|
|
while ((i = getopt_long(ac, __DECONST(char * const *, av), "CSIHPabijJ:nquvzs:d:U:m:o:p:Ttw", longopts, NULL)) != EOF)
|
|
{
|
|
switch(i)
|
|
{
|
|
case 'v': /* show version number */
|
|
errx(0, "version FreeBSD");
|
|
break;
|
|
|
|
case 'u': /* toggle uid/username display */
|
|
do_unames = !do_unames;
|
|
break;
|
|
|
|
case 'U': /* display only username's processes */
|
|
if ((ps.uid[0] = userid(optarg)) == -1)
|
|
{
|
|
errx(1, "%s: unknown user\n", optarg);
|
|
}
|
|
break;
|
|
|
|
case 'S': /* show system processes */
|
|
ps.system = true;
|
|
break;
|
|
|
|
case 'I': /* show idle processes */
|
|
ps.idle = !ps.idle;
|
|
break;
|
|
|
|
case 'i': /* go interactive regardless */
|
|
interactive = 1;
|
|
break;
|
|
|
|
case 'n': /* batch, or non-interactive */
|
|
case 'b':
|
|
interactive = 0;
|
|
break;
|
|
|
|
case 'a':
|
|
fmt_flags ^= FMT_SHOWARGS;
|
|
break;
|
|
|
|
case 'd': /* number of displays to show */
|
|
if ((i = atoiwi(optarg)) == Invalid || i == 0)
|
|
{
|
|
warnx("warning: display count should be positive -- option ignored");
|
|
warnings++;
|
|
}
|
|
else
|
|
{
|
|
displays = i;
|
|
}
|
|
break;
|
|
case 'p': {
|
|
unsigned long long num;
|
|
const char *errstr;
|
|
|
|
num = strtonum(optarg, 0, INT_MAX, &errstr);
|
|
if (errstr != NULL || !find_pid(num)) {
|
|
fprintf(stderr, "%s: unknown pid\n", optarg);
|
|
exit(1);
|
|
}
|
|
ps.pid = (pid_t)num;
|
|
ps.system = true;
|
|
break;
|
|
}
|
|
|
|
case 's':
|
|
{
|
|
double delay_d = strtod(optarg, &nptr);
|
|
if (nptr == optarg)
|
|
{
|
|
warnx("warning: invalid delay");
|
|
warnings++;
|
|
}
|
|
else if (delay_d <= 0)
|
|
{
|
|
warnx("warning: seconds delay should be positive -- using default");
|
|
warnings++;
|
|
}
|
|
else
|
|
{
|
|
delay.tv_sec = delay_d;
|
|
delay.tv_usec = (delay_d - delay.tv_sec) * 1e6;
|
|
}
|
|
break;
|
|
}
|
|
|
|
case 'q': /* be quick about it */
|
|
errno = 0;
|
|
i = setpriority(PRIO_PROCESS, 0, PRIO_MIN);
|
|
if (i == -1 && errno != 0) {
|
|
warnx("warning: `-q' option failed (%m)");
|
|
warnings++;
|
|
}
|
|
break;
|
|
|
|
case 'm': /* select display mode */
|
|
if (strcmp(optarg, "io") == 0) {
|
|
displaymode = DISP_IO;
|
|
} else if (strcmp(optarg, "cpu") == 0) {
|
|
displaymode = DISP_CPU;
|
|
} else {
|
|
errx(1, "warning: `-m' option can only take args 'io' or 'cpu'");
|
|
}
|
|
break;
|
|
|
|
case 'o': /* select sort order */
|
|
order_name = optarg;
|
|
break;
|
|
|
|
case 't':
|
|
ps.self = !ps.self;
|
|
break;
|
|
|
|
case 'C':
|
|
ps.wcpu = !ps.wcpu;
|
|
break;
|
|
|
|
case 'H':
|
|
ps.thread = !ps.thread;
|
|
break;
|
|
|
|
case 'T':
|
|
ps.thread_id = !ps.thread_id;
|
|
break;
|
|
|
|
case 'j':
|
|
ps.jail = !ps.jail;
|
|
break;
|
|
|
|
case 'J': /* display only jail's processes */
|
|
if ((ps.jid = jail_getid(optarg)) == -1)
|
|
{
|
|
fprintf(stderr, "%s: unknown jail\n", optarg);
|
|
exit(1);
|
|
}
|
|
ps.jail = 1;
|
|
break;
|
|
|
|
case 'P':
|
|
pcpu_stats = !pcpu_stats;
|
|
break;
|
|
|
|
case 'w':
|
|
ps.swap = 1;
|
|
break;
|
|
|
|
case 'z':
|
|
ps.kidle = !ps.kidle;
|
|
break;
|
|
|
|
default:
|
|
errx(1,
|
|
"[-abCHIijnPqStuvwz] [-d count] [-m io | cpu] [-o field] [-p pid]\n"
|
|
" [-s time] [-J jail] [-U username] [number]");
|
|
}
|
|
}
|
|
|
|
/* get count of top processes to display (if any) */
|
|
if (optind < ac)
|
|
{
|
|
if ((topn = atoiwi(av[optind])) == Invalid)
|
|
{
|
|
warnx("warning: process display count should be non-negative -- using default");
|
|
warnings++;
|
|
}
|
|
else
|
|
{
|
|
topn_specified = true;
|
|
}
|
|
}
|
|
|
|
/* tricky: remember old value of preset_argc & set preset_argc = 0 */
|
|
i = preset_argc;
|
|
preset_argc = 0;
|
|
|
|
/* repeat only if we really did the preset arguments */
|
|
} while (i != 0);
|
|
|
|
/* set constants for username/uid display correctly */
|
|
if (!do_unames)
|
|
{
|
|
uname_field = " UID ";
|
|
get_userid = itoa7;
|
|
}
|
|
|
|
/* initialize the kernel memory interface */
|
|
if (machine_init(&statics) == -1)
|
|
{
|
|
exit(1);
|
|
}
|
|
|
|
/* determine sorting order index, if necessary */
|
|
if (order_name != NULL)
|
|
{
|
|
if ((order_index = string_index(order_name, statics.order_names)) == -1)
|
|
{
|
|
const char * const *pp;
|
|
|
|
warnx("'%s' is not a recognized sorting order.", order_name);
|
|
fprintf(stderr, "\tTry one of these:");
|
|
pp = statics.order_names;
|
|
while (*pp != NULL)
|
|
{
|
|
fprintf(stderr, " %s", *pp++);
|
|
}
|
|
fputc('\n', stderr);
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
/* initialize termcap */
|
|
init_termcap(interactive);
|
|
|
|
/* get the string to use for the process area header */
|
|
header_text = format_header(uname_field);
|
|
|
|
/* initialize display interface */
|
|
if ((max_topn = display_init(&statics)) == -1)
|
|
{
|
|
errx(4, "can't allocate sufficient memory");
|
|
}
|
|
|
|
/* print warning if user requested more processes than we can display */
|
|
if (topn > max_topn)
|
|
{
|
|
warnx("warning: this terminal can only display %d processes.", max_topn);
|
|
warnings++;
|
|
}
|
|
|
|
/* adjust for topn == Infinity */
|
|
if (topn == Infinity)
|
|
{
|
|
/*
|
|
* For smart terminals, infinity really means everything that can
|
|
* be displayed, or Largest.
|
|
* On dumb terminals, infinity means every process in the system!
|
|
* We only really want to do that if it was explicitly specified.
|
|
* This is always the case when "Default_TOPN != Infinity". But if
|
|
* topn wasn't explicitly specified and we are on a dumb terminal
|
|
* and the default is Infinity, then (and only then) we use
|
|
* "Nominal_TOPN" instead.
|
|
*/
|
|
topn = smart_terminal ? Largest :
|
|
(topn_specified ? Largest : Nominal_TOPN);
|
|
}
|
|
|
|
/* set header display accordingly */
|
|
display_header(topn > 0);
|
|
|
|
/* determine interactive state */
|
|
if (interactive == 2)
|
|
{
|
|
interactive = smart_terminal;
|
|
}
|
|
|
|
/* if # of displays not specified, fill it in */
|
|
if (displays == 0)
|
|
{
|
|
displays = smart_terminal ? Infinity : 1;
|
|
}
|
|
|
|
/* hold interrupt signals while setting up the screen and the handlers */
|
|
|
|
sigemptyset(&new_sigmask);
|
|
sigaddset(&new_sigmask, SIGINT);
|
|
sigaddset(&new_sigmask, SIGQUIT);
|
|
sigaddset(&new_sigmask, SIGTSTP);
|
|
sigprocmask(SIG_BLOCK, &new_sigmask, &old_sigmask);
|
|
init_screen();
|
|
signal(SIGINT, leave);
|
|
signal(SIGQUIT, leave);
|
|
signal(SIGTSTP, tstop);
|
|
signal(SIGWINCH, top_winch);
|
|
sigprocmask(SIG_SETMASK, &old_sigmask, NULL);
|
|
if (warnings)
|
|
{
|
|
fputs("....", stderr);
|
|
fflush(stderr);
|
|
sleep(3 * warnings);
|
|
fputc('\n', stderr);
|
|
}
|
|
|
|
restart:
|
|
|
|
/*
|
|
* main loop -- repeat while display count is positive or while it
|
|
* indicates infinity (by being -1)
|
|
*/
|
|
|
|
while ((displays == -1) || (displays-- > 0))
|
|
{
|
|
int (*compare)(const void * const, const void * const);
|
|
|
|
|
|
/* get the current stats */
|
|
get_system_info(&system_info);
|
|
|
|
compare = compares[order_index];
|
|
|
|
/* get the current set of processes */
|
|
processes =
|
|
get_process_info(&system_info, &ps, compare);
|
|
|
|
/* display the load averages */
|
|
(*d_loadave)(system_info.last_pid,
|
|
system_info.load_avg);
|
|
|
|
/* display the battery info (if any) */
|
|
i_battery(statics.nbatteries, system_info.battery);
|
|
|
|
/* display the current time */
|
|
/* this method of getting the time SHOULD be fairly portable */
|
|
time(&curr_time);
|
|
i_uptime(&system_info.boottime, &curr_time);
|
|
i_timeofday(&curr_time);
|
|
|
|
/* display process state breakdown */
|
|
(*d_procstates)(system_info.p_total,
|
|
system_info.procstates);
|
|
(*d_cpustates)(system_info.cpustates);
|
|
|
|
/* display memory stats */
|
|
(*d_memory)(system_info.memory);
|
|
(*d_arc)(system_info.arc);
|
|
(*d_carc)(system_info.carc);
|
|
|
|
/* display swap stats */
|
|
(*d_swap)(system_info.swap);
|
|
|
|
/* handle message area */
|
|
(*d_message)();
|
|
|
|
/* update the header area */
|
|
(*d_header)(header_text);
|
|
|
|
if (topn > 0)
|
|
{
|
|
/* determine number of processes to actually display */
|
|
/* this number will be the smallest of: active processes,
|
|
number user requested, number current screen accomodates */
|
|
active_procs = system_info.p_pactive;
|
|
if (active_procs > topn)
|
|
{
|
|
active_procs = topn;
|
|
}
|
|
if (active_procs > max_topn)
|
|
{
|
|
active_procs = max_topn;
|
|
}
|
|
|
|
/* now show the top "n" processes. */
|
|
for (i = 0; i < active_procs; i++)
|
|
{
|
|
(*d_process)(i, format_next_process(processes, get_userid,
|
|
fmt_flags));
|
|
}
|
|
}
|
|
else
|
|
{
|
|
i = 0;
|
|
}
|
|
|
|
/* do end-screen processing */
|
|
u_endscreen(i);
|
|
|
|
/* now, flush the output buffer */
|
|
if (fflush(stdout) != 0)
|
|
{
|
|
new_message(MT_standout, " Write error on stdout");
|
|
putchar('\r');
|
|
quit(1);
|
|
}
|
|
|
|
/* only do the rest if we have more displays to show */
|
|
if (displays)
|
|
{
|
|
/* switch out for new display on smart terminals */
|
|
if (smart_terminal)
|
|
{
|
|
if (overstrike)
|
|
{
|
|
reset_display();
|
|
}
|
|
else
|
|
{
|
|
d_loadave = u_loadave;
|
|
d_procstates = u_procstates;
|
|
d_cpustates = u_cpustates;
|
|
d_memory = u_memory;
|
|
d_arc = u_arc;
|
|
d_carc = u_carc;
|
|
d_swap = u_swap;
|
|
d_message = u_message;
|
|
d_header = u_header;
|
|
d_process = u_process;
|
|
}
|
|
}
|
|
|
|
no_command = true;
|
|
if (!interactive)
|
|
{
|
|
timeout = delay;
|
|
select(0, NULL, NULL, NULL, &timeout);
|
|
if (leaveflag) {
|
|
end_screen();
|
|
exit(0);
|
|
}
|
|
}
|
|
else while (no_command)
|
|
{
|
|
/* assume valid command unless told otherwise */
|
|
no_command = false;
|
|
|
|
/* set up arguments for select with timeout */
|
|
FD_ZERO(&readfds);
|
|
FD_SET(0, &readfds); /* for standard input */
|
|
timeout = delay;
|
|
|
|
if (leaveflag) {
|
|
end_screen();
|
|
exit(0);
|
|
}
|
|
|
|
if (tstopflag) {
|
|
/* move to the lower left */
|
|
end_screen();
|
|
fflush(stdout);
|
|
|
|
/* default the signal handler action */
|
|
signal(SIGTSTP, SIG_DFL);
|
|
|
|
/* unblock the signal and send ourselves one */
|
|
sigsetmask(sigblock(0) & ~(1 << (SIGTSTP - 1)));
|
|
kill(0, SIGTSTP);
|
|
|
|
/* reset the signal handler */
|
|
signal(SIGTSTP, tstop);
|
|
|
|
/* reinit screen */
|
|
reinit_screen();
|
|
reset_display();
|
|
tstopflag = 0;
|
|
goto restart;
|
|
}
|
|
|
|
if (winchflag) {
|
|
/* reascertain the screen dimensions */
|
|
get_screensize();
|
|
|
|
/* tell display to resize */
|
|
max_topn = display_resize();
|
|
|
|
/* reset the signal handler */
|
|
signal(SIGWINCH, top_winch);
|
|
|
|
reset_display();
|
|
winchflag = 0;
|
|
goto restart;
|
|
}
|
|
|
|
/* wait for either input or the end of the delay period */
|
|
sel_ret = select(2, &readfds, NULL, NULL, &timeout);
|
|
if (sel_ret < 0 && errno != EINTR)
|
|
quit(0);
|
|
if (sel_ret > 0)
|
|
{
|
|
int newval;
|
|
const char *errmsg;
|
|
const struct command *cptr;
|
|
|
|
/* something to read -- clear the message area first */
|
|
clear_message();
|
|
|
|
/* now read it and convert to command strchr */
|
|
/* (use "change" as a temporary to hold strchr) */
|
|
if (read(0, &ch, 1) != 1)
|
|
{
|
|
/* read error: either 0 or -1 */
|
|
new_message(MT_standout, " Read error on stdin");
|
|
putchar('\r');
|
|
quit(1);
|
|
}
|
|
if (ch == '\r' || ch == '\n') {
|
|
continue;
|
|
}
|
|
cptr = all_commands;
|
|
while (cptr->c != '\0') {
|
|
if (cptr->c == ch) {
|
|
break;
|
|
}
|
|
cptr++;
|
|
}
|
|
if (cptr->c == '\0') {
|
|
new_message(MT_standout, " Command not understood");
|
|
putchar('\r');
|
|
no_command = true;
|
|
}
|
|
if (overstrike && !cptr->available_to_dumb)
|
|
{
|
|
new_message(MT_standout,
|
|
" Command cannot be handled by this terminal");
|
|
putchar('\r');
|
|
no_command = true;
|
|
}
|
|
if (!no_command) {
|
|
switch(cptr->id)
|
|
{
|
|
case CMD_redraw: /* redraw screen */
|
|
reset_display();
|
|
break;
|
|
|
|
case CMD_update: /* merely update display */
|
|
break;
|
|
|
|
case CMD_quit:
|
|
quit(0);
|
|
break;
|
|
|
|
case CMD_help:
|
|
reset_display();
|
|
top_clear();
|
|
show_help();
|
|
top_standout("Hit any key to continue: ");
|
|
fflush(stdout);
|
|
read(0, &ch, 1);
|
|
break;
|
|
|
|
case CMD_errors: /* show errors */
|
|
if (error_count() == 0)
|
|
{
|
|
new_message(MT_standout,
|
|
" Currently no errors to report.");
|
|
putchar('\r');
|
|
no_command = true;
|
|
}
|
|
else
|
|
{
|
|
reset_display();
|
|
top_clear();
|
|
show_errors();
|
|
top_standout("Hit any key to continue: ");
|
|
fflush(stdout);
|
|
read(0, &ch, 1);
|
|
}
|
|
break;
|
|
|
|
case CMD_number:
|
|
new_message(MT_standout,
|
|
"Number of processes to show: ");
|
|
newval = readline(tempbuf1, 8, true);
|
|
if (newval > -1)
|
|
{
|
|
if (newval > max_topn)
|
|
{
|
|
new_message(MT_standout | MT_delayed,
|
|
" This terminal can only display %d processes.",
|
|
max_topn);
|
|
putchar('\r');
|
|
}
|
|
|
|
if (newval == 0)
|
|
{
|
|
/* inhibit the header */
|
|
display_header(false);
|
|
}
|
|
else if (newval > topn && topn == 0)
|
|
{
|
|
/* redraw the header */
|
|
display_header(true);
|
|
d_header = i_header;
|
|
}
|
|
topn = newval;
|
|
}
|
|
break;
|
|
|
|
case CMD_delay: /* new seconds delay */
|
|
new_message(MT_standout, "Seconds to delay: ");
|
|
if ((i = readline(tempbuf1, 8, false)) > 0)
|
|
{
|
|
double delay_d = strtod(tempbuf1, &nptr);
|
|
if (nptr == tempbuf1 || delay_d <= 0)
|
|
{
|
|
new_message(MT_standout, " Invalid delay");
|
|
putchar('\r');
|
|
no_command = true;
|
|
}
|
|
else
|
|
{
|
|
delay.tv_sec = delay_d;
|
|
delay.tv_usec = (delay_d - delay.tv_sec) * 1e6;
|
|
clear_message();
|
|
}
|
|
}
|
|
break;
|
|
|
|
case CMD_displays: /* change display count */
|
|
new_message(MT_standout,
|
|
"Displays to show (currently %s): ",
|
|
displays == -1 ? "infinite" :
|
|
itoa(displays));
|
|
if ((i = readline(tempbuf1, 10, true)) > 0)
|
|
{
|
|
displays = i;
|
|
}
|
|
else if (i == 0)
|
|
{
|
|
quit(0);
|
|
}
|
|
clear_message();
|
|
break;
|
|
|
|
case CMD_kill: /* kill program */
|
|
new_message(0, "kill ");
|
|
if (readline(tempbuf2, sizeof(tempbuf2), false) > 0)
|
|
{
|
|
if ((errmsg = kill_procs(tempbuf2)) != NULL)
|
|
{
|
|
new_message(MT_standout, "%s", errmsg);
|
|
putchar('\r');
|
|
no_command = true;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
clear_message();
|
|
}
|
|
break;
|
|
|
|
case CMD_renice: /* renice program */
|
|
new_message(0, "renice ");
|
|
if (readline(tempbuf2, sizeof(tempbuf2), false) > 0)
|
|
{
|
|
if ((errmsg = renice_procs(tempbuf2)) != NULL)
|
|
{
|
|
new_message(MT_standout, "%s", errmsg);
|
|
putchar('\r');
|
|
no_command = true;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
clear_message();
|
|
}
|
|
break;
|
|
|
|
case CMD_idletog:
|
|
ps.idle = !ps.idle;
|
|
new_message(MT_standout | MT_delayed,
|
|
" %sisplaying idle processes.",
|
|
ps.idle ? "D" : "Not d");
|
|
putchar('\r');
|
|
break;
|
|
|
|
case CMD_selftog:
|
|
ps.self = !ps.self;
|
|
new_message(MT_standout | MT_delayed,
|
|
" %sisplaying self.",
|
|
(ps.self) ? "D" : "Not d");
|
|
putchar('\r');
|
|
break;
|
|
|
|
case CMD_user:
|
|
if (handle_user(tempbuf2, sizeof(tempbuf2)))
|
|
no_command = true;
|
|
break;
|
|
|
|
case CMD_thrtog:
|
|
ps.thread = !ps.thread;
|
|
new_message(MT_standout | MT_delayed,
|
|
" Displaying threads %s",
|
|
ps.thread ? "separately" : "as a count");
|
|
header_text = format_header(uname_field);
|
|
reset_display();
|
|
putchar('\r');
|
|
break;
|
|
|
|
case CMD_toggletid:
|
|
ps.thread_id = !ps.thread_id;
|
|
new_message(MT_standout | MT_delayed,
|
|
" Displaying %s",
|
|
ps.thread_id ? "tid" : "pid");
|
|
header_text = format_header(uname_field);
|
|
reset_display();
|
|
putchar('\r');
|
|
break;
|
|
|
|
case CMD_wcputog:
|
|
ps.wcpu = !ps.wcpu;
|
|
new_message(MT_standout | MT_delayed,
|
|
" Displaying %s CPU",
|
|
ps.wcpu ? "weighted" : "raw");
|
|
header_text = format_header(uname_field);
|
|
reset_display();
|
|
putchar('\r');
|
|
break;
|
|
case CMD_viewtog:
|
|
displaymode = displaymode == DISP_IO ? DISP_CPU : DISP_IO;
|
|
new_message(MT_standout | MT_delayed,
|
|
" Displaying %s statistics.",
|
|
displaymode == DISP_IO ? "IO" : "CPU");
|
|
header_text = format_header(uname_field);
|
|
display_header(true);
|
|
d_header = i_header;
|
|
reset_display();
|
|
break;
|
|
case CMD_viewsys:
|
|
ps.system = !ps.system;
|
|
new_message(MT_standout | MT_delayed,
|
|
" %sisplaying system processes.",
|
|
ps.system ? "D" : "Not d");
|
|
break;
|
|
case CMD_showargs:
|
|
fmt_flags ^= FMT_SHOWARGS;
|
|
new_message(MT_standout | MT_delayed,
|
|
" %sisplaying process arguments.",
|
|
fmt_flags & FMT_SHOWARGS ? "D" : "Not d");
|
|
break;
|
|
case CMD_order:
|
|
new_message(MT_standout,
|
|
"Order to sort: ");
|
|
if (readline(tempbuf2, sizeof(tempbuf2), false) > 0)
|
|
{
|
|
if ((i = string_index(tempbuf2, statics.order_names)) == -1)
|
|
{
|
|
new_message(MT_standout,
|
|
" %s: unrecognized sorting order", tempbuf2);
|
|
no_command = true;
|
|
}
|
|
else
|
|
{
|
|
order_index = i;
|
|
}
|
|
putchar('\r');
|
|
}
|
|
else
|
|
{
|
|
clear_message();
|
|
}
|
|
break;
|
|
case CMD_jidtog:
|
|
ps.jail = !ps.jail;
|
|
new_message(MT_standout | MT_delayed,
|
|
" %sisplaying jail ID.",
|
|
ps.jail ? "D" : "Not d");
|
|
header_text = format_header(uname_field);
|
|
reset_display();
|
|
putchar('\r');
|
|
break;
|
|
|
|
case CMD_jail:
|
|
new_message(MT_standout,
|
|
"Jail to show (+ for all): ");
|
|
if (readline(tempbuf2, sizeof(tempbuf2), false) > 0)
|
|
{
|
|
if (tempbuf2[0] == '+' &&
|
|
tempbuf2[1] == '\0')
|
|
{
|
|
ps.jid = -1;
|
|
}
|
|
else if ((i = jail_getid(tempbuf2)) == -1)
|
|
{
|
|
new_message(MT_standout,
|
|
" %s: unknown jail", tempbuf2);
|
|
no_command = true;
|
|
}
|
|
else
|
|
{
|
|
ps.jid = i;
|
|
}
|
|
if (ps.jail == 0) {
|
|
ps.jail = 1;
|
|
new_message(MT_standout |
|
|
MT_delayed, " Displaying jail "
|
|
"ID.");
|
|
header_text =
|
|
format_header(uname_field);
|
|
reset_display();
|
|
}
|
|
putchar('\r');
|
|
}
|
|
else
|
|
{
|
|
clear_message();
|
|
}
|
|
break;
|
|
|
|
case CMD_kidletog:
|
|
ps.kidle = !ps.kidle;
|
|
new_message(MT_standout | MT_delayed,
|
|
" %sisplaying system idle process.",
|
|
ps.kidle ? "D" : "Not d");
|
|
putchar('\r');
|
|
break;
|
|
case CMD_pcputog:
|
|
pcpu_stats = !pcpu_stats;
|
|
new_message(MT_standout | MT_delayed,
|
|
" Displaying %sCPU statistics.",
|
|
pcpu_stats ? "per-" : "global ");
|
|
toggle_pcpustats();
|
|
max_topn = display_updatecpus(&statics);
|
|
reset_display();
|
|
putchar('\r');
|
|
break;
|
|
case CMD_swaptog:
|
|
ps.swap = !ps.swap;
|
|
new_message(MT_standout | MT_delayed,
|
|
" %sisplaying per-process swap usage.",
|
|
ps.swap ? "D" : "Not d");
|
|
header_text = format_header(uname_field);
|
|
reset_display();
|
|
putchar('\r');
|
|
break;
|
|
case CMD_pid:
|
|
new_message(MT_standout,
|
|
"Process id to show (+ for all): ");
|
|
if (readline(tempbuf2, sizeof(tempbuf2), false) > 0) {
|
|
if (tempbuf2[0] == '+' &&
|
|
tempbuf2[1] == '\0') {
|
|
ps.pid = (pid_t)-1;
|
|
} else {
|
|
unsigned long long num;
|
|
const char *errstr;
|
|
|
|
num = strtonum(tempbuf2, 0, INT_MAX,
|
|
&errstr);
|
|
if (errstr != NULL || !find_pid(num)) {
|
|
new_message(MT_standout,
|
|
" %s: unknown pid",
|
|
tempbuf2);
|
|
no_command = true;
|
|
} else {
|
|
ps.pid = (pid_t)num;
|
|
}
|
|
}
|
|
putchar('\r');
|
|
} else
|
|
clear_message();
|
|
break;
|
|
case CMD_NONE:
|
|
assert(false && "reached switch without command");
|
|
}
|
|
}
|
|
}
|
|
|
|
/* flush out stuff that may have been written */
|
|
fflush(stdout);
|
|
}
|
|
}
|
|
}
|
|
|
|
#ifdef DEBUG
|
|
fclose(debug);
|
|
#endif
|
|
quit(0);
|
|
}
|
|
|
|
/*
|
|
* reset_display() - reset all the display routine pointers so that entire
|
|
* screen will get redrawn.
|
|
*/
|
|
|
|
static void
|
|
reset_display(void)
|
|
{
|
|
d_loadave = i_loadave;
|
|
d_procstates = i_procstates;
|
|
d_cpustates = i_cpustates;
|
|
d_memory = i_memory;
|
|
d_arc = i_arc;
|
|
d_carc = i_carc;
|
|
d_swap = i_swap;
|
|
d_message = i_message;
|
|
d_header = i_header;
|
|
d_process = i_process;
|
|
}
|
|
|
|
/*
|
|
* signal handlers
|
|
*/
|
|
|
|
static sigret_t
|
|
leave(int i __unused) /* exit under normal conditions -- INT handler */
|
|
{
|
|
|
|
leaveflag = 1;
|
|
}
|
|
|
|
static sigret_t
|
|
tstop(int i __unused) /* SIGTSTP handler */
|
|
{
|
|
|
|
tstopflag = 1;
|
|
}
|
|
|
|
static sigret_t
|
|
top_winch(int i __unused) /* SIGWINCH handler */
|
|
{
|
|
|
|
winchflag = 1;
|
|
}
|
|
|
|
void __dead2
|
|
quit(int status) /* exit under duress */
|
|
{
|
|
end_screen();
|
|
exit(status);
|
|
}
|