freebsd-dev/usr.bin/top/top.c
Philip Paeps fc8ae86a8f top: display battery capacity remaining
Submitted by: Antranig Vartanian <antranigv@freebsd.am>
Reviewed by:  imp, philip
Differential Revision:        https://reviews.freebsd.org/D22871
2019-12-21 05:03:21 +00:00

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);
}