int main (int argc, char** argv)
{
output_file = stdout;
program_name = basename (argv[0]);
init_command (argc-1, argv+1);
source_path = getenv ("COVERAGE_PATH");
if (source_path == NULL) {
source_path = "."; }
augment_source_path (source_path);
while (TRUE) {
get_command ();
switch ((int)command) {
case QUIT: {
exit (0);
break; }
case DUMMY: {
break; }
case EXEC_FILE: {
FILE *f = fopen (*arg_v, "r");
if (f == NULL) {
sprintf (error_message, "cannot open command file %s",
*arg_v);
error (); }
command_file_stack [++nb_command_files] = f;
command_source = FROM_FILE;
break; }
case READ_DATABASE: {
while (arg_c-- > 0) {
read_coverage_data (*arg_v++); }
break; }
case WRITE_DATABASE: {
ensure_that_database_is_loaded ();
while (arg_c-- > 0) {
write_coverage_data (*arg_v++); }
break; }
case SHOW_LINES: {
ensure_that_database_is_loaded ();
while (arg_c-- > 0) {
show_lines (basename (*arg_v++)); }
break; }
case SHOW_LINES_ALL: {
collection c;
ensure_that_database_is_loaded ();
for (c = collections; c != NULL; c = c->next) {
show_lines (c->file_name); }
break; }
case SHOW_PROCS: {
ensure_that_database_is_loaded ();
while (arg_c-- > 0) {
show_procs (basename (*arg_v++)); }
break; }
case SHOW_PROCS_ALL: {
collection c;
ensure_that_database_is_loaded ();
for (c = collections; c != NULL; c = c->next) {
show_procs (c->file_name); }
break; }
case AUGMENT_SOURCE_PATH: {
while (arg_c-- > 0) {
augment_source_path (*arg_v++); }
break; }
case SELECT_OUTPUT: {
FILE* f = fopen (*arg_v, "w");
if (f == NULL) {
sprintf (error_message,
"cannot open %s for output - %s", *arg_v,
"redirection ignored\n");
warning (); }
else {
if (output_file != stdout) {
fclose (output_file); }
if (verbose_mode) {
fprintf (stderr, "output file is now %s\n", *arg_v); }
output_file = f; }
break; }
case VERBOSE_ON: {
verbose_mode = TRUE;
break; }}}
/* exit (0); */
}
int main(int argc, char **argv)
{
/* For select(2). */
struct timeval tv;
fd_set in;
/* For parsing arguments. */
char *cp;
/* The key read in. */
char c;
struct sigaction sact;
setlocale(LC_ALL, "");
get_options();
/* set to PCPU sorting */
register_sort_function( -1, (cmp_t)pcpu_sort);
#ifdef HZ
Hertz = HZ;
#endif
/*
* Parse arguments.
*/
argv++;
while (*argv) {
cp = *argv++;
while (*cp) {
switch (*cp) {
case 'd':
if (cp[1]) {
if (sscanf(++cp, "%f", &Sleeptime) != 1) {
fprintf(stderr, PROGNAME ": Bad delay time `%s'\n", cp);
exit(1);
}
goto breakargv;
} else if (*argv) { /* last char in an argv, use next as arg */
if (sscanf(cp = *argv++, "%f", &Sleeptime) != 1) {
fprintf(stderr, PROGNAME ": Bad delay time `%s'\n", cp);
exit(1);
}
goto breakargv;
} else {
fprintf(stderr, "-d requires an argument\n");
exit(1);
}
break;
case 'n':
if (cp[1]) {
if (sscanf(++cp, "%d", &Loops) != 1) {
fprintf(stderr, PROGNAME ": Bad value `%s'\n", cp);
exit(1);
}
goto breakargv;
} else if (*argv) { /* last char in an argv, use next as arg */
if (sscanf(cp = *argv++, "%d", &Loops) != 1) {
fprintf(stderr, PROGNAME ": Bad value `%s'\n", cp);
exit(1);
}
goto breakargv;
}
break;
case 'q':
if (!getuid())
/* set priority to -10 in order to stay above kswapd */
if (setpriority(PRIO_PROCESS, getpid(), -10)) {
/* We check this just for paranoia. It's not
fatal, and shouldn't happen. */
perror(PROGNAME ": setpriority() failed");
}
Sleeptime = 0;
break;
case 'p':
if (monpids_index >= monpids_max) {
fprintf(stderr, PROGNAME ": More than %u process ids specified\n",
monpids_max);
exit(1);
}
if (cp[1]) {
if (sscanf(++cp, "%d", &monpids[monpids_index]) != 1 ||
monpids[monpids_index] < 0 || monpids[monpids_index] > 65535) {
fprintf(stderr, PROGNAME ": Bad process id `%s'\n", cp);
exit(1);
}
} else if (*argv) { /* last char in an argv, use next as arg */
if (sscanf(cp = *argv++, "%d", &monpids[monpids_index]) != 1 ||
monpids[monpids_index] < 0 || monpids[monpids_index] > 65535) {
fprintf(stderr, PROGNAME ": Bad process id `%s'\n", cp);
exit(1);
}
} else {
fprintf(stderr, "-p requires an argument\n");
exit(1);
}
if (!monpids[monpids_index])
monpids[monpids_index] = getpid();
/* default to no sorting when monitoring process ids */
if (!monpids_index++) {
sort_type = S_NONE;
reset_sort_options();
}
cp = "_";
break;
case 'b':
Batch = 1;
break;
case 'c':
show_cmd = !show_cmd;
break;
case 'S':
Cumulative = 1;
break;
case 'i':
Noidle = 1;
break;
case 's':
Secure = 1;
break;
case 'C':
CPU_states = 1;
break;
case '-':
break; /* Just ignore it */
case 'v':
case 'V':
fprintf(stdout, "top (%s)\n", procps_version);
exit(0);
case 'h':
fprintf(stdout, "usage: " PROGNAME " -hvbcisqS -d delay -p pid -n iterations\n");
exit(0);
default:
fprintf(stderr, PROGNAME ": Unknown argument `%c'\n", *cp);
fprintf(stdout, "usage: " PROGNAME " -hvbcisqS -d delay -p pid -n iterations\n");
exit(1);
}
cp++;
}
breakargv:;
}
if (nr_cpu > 1 && CPU_states)
header_lines++;
setup_terminal();
window_size(0);
/*
* Set up signal handlers.
*/
sact.sa_handler = sig_end;
sact.sa_flags = 0;
sigemptyset(&sact.sa_mask);
sigaction(SIGHUP, &sact, NULL);
sigaction(SIGINT, &sact, NULL);
sigaction(SIGQUIT, &sact, NULL);
sact.sa_handler = sig_stop;
sact.sa_flags = SA_RESTART;
sigaction(SIGTSTP, &sact, NULL);
sact.sa_handler = window_size;
sigaction(SIGWINCH, &sact, NULL);
sigaction(SIGCONT, &sact, NULL);
/* loop, collecting process info and sleeping */
while (1) {
if (Loops > 0)
Loops--;
/* display the tasks */
show_procs();
/* sleep & wait for keyboard input */
if (Loops == 0)
sig_end(0);
if (!Batch)
{
tv.tv_sec = Sleeptime;
tv.tv_usec = (Sleeptime - (int) Sleeptime) * 1000000;
FD_ZERO(&in);
FD_SET(0, &in);
if (select(1, &in, 0, 0, &tv) > 0 && read(0, &c, 1) == 1)
do_key(c);
} else {
sleep(Sleeptime);
}
}
}
int main(int argc, char **argv)
{
/* For select(2). */
struct timeval tv;
fd_set in;
/* For parsing arguments. */
char *cp;
/* The key read in. */
char c;
get_options();
/*
* Parse arguments.
*/
argv++;
while (*argv) {
cp = *argv++;
while (*cp) {
switch (*cp) {
case 'd':
if (cp[1]) {
if (sscanf(++cp, "%f", &Sleeptime) != 1) {
fprintf(stderr, PROGNAME ": Bad delay time `%s'\n", cp);
exit(1);
}
goto breakargv;
} else if (*argv) { /* last char in an argv, use next as arg */
if (sscanf(cp = *argv++, "%f", &Sleeptime) != 1) {
fprintf(stderr, PROGNAME ": Bad delay time `%s'\n", cp);
exit(1);
}
goto breakargv;
} else {
fprintf(stderr, "-d requires an argument\n");
exit(1);
}
break;
case 'q':
if (!getuid())
/* set priority to -10 in order to stay above kswapd */
if (setpriority(PRIO_PROCESS, getpid(), -10)) {
/* We check this just for paranoia. It's not
fatal, and shouldn't happen. */
perror(PROGNAME ": setpriority() failed");
}
Sleeptime = 0;
break;
case 'c':
show_cmd = !show_cmd;
break;
case 'S':
Cumulative = 1;
break;
case 'i':
Noidle = 1;
break;
case 's':
Secure = 1;
break;
case '-':
break; /* Just ignore it */
#if defined (SHOWFLAG)
case '/': showall++;
#endif
default:
fprintf(stderr, PROGNAME ": Unknown argument `%c'\n", *cp);
exit(1);
}
cp++;
}
breakargv:
}
/* set to PCPU sorting */
register_sort_function( -1, (cmp_t)pcpu_sort);
/* for correct handling of some fields, we have to do distinguish
* between kernel versions */
set_linux_version();
/* get kernel symbol table, if needed */
if (!CL_wchan_nout) {
if (open_psdb()) {
CL_wchan_nout = 1;
} else {
psdbsucc = 1;
}
}
setup_terminal();
window_size();
/*
* calculate header size, length of cmdline field ...
*/
Numfields = make_header();
/*
* Set up signal handlers.
*/
signal(SIGHUP, (void *) (int) end);
signal(SIGINT, (void *) (int) end);
signal(SIGQUIT, (void *) (int) end);
signal(SIGTSTP, (void *) (int) stop);
signal(SIGWINCH, (void *) (int) window_size);
/* loop, collecting process info and sleeping */
while (1) {
if (setjmp(redraw_jmp))
clear_screen();
/* display the tasks */
show_procs();
/* sleep & wait for keyboard input */
tv.tv_sec = Sleeptime;
tv.tv_usec = (Sleeptime - (int) Sleeptime) * 1000000;
FD_ZERO(&in);
FD_SET(0, &in);
if (select(16, &in, 0, 0, &tv) > 0 && read(0, &c, 1) == 1)
do_key(c);
}
}
/*#######################################################################
*#### Signal handled routines: error_end, end, stop, window_size ###
*#### Small utilities: make_header, getstr, getint, getfloat, getsig ###
*#######################################################################
*/
/*
* end when exiting with an error.
*/
void error_end(int rno)
{
if (psdbsucc)
close_psdb();
ioctl(0, TCSETAF, &Savetty);
PUTP(tgoto(cm, 0, Lines - 1));
fputs("\r\n", stdout);
exit(rno);
}
/*
* Normal end of execution.
*/
void end(void)
{
if (psdbsucc)
close_psdb();
ioctl(0, TCSETAF, &Savetty);
PUTP(tgoto(cm, 0, Lines - 1));
fputs("\r\n", stdout);
exit(0);
}
/*
* SIGTSTP catcher.
*/
void stop(void)
{
/* Reset terminal. */
if (psdbsucc)
close_psdb();
ioctl(0, TCSETAF, &Savetty);
PUTP(tgoto(cm, 0, Lines - 3));
fflush(stdout);
raise(SIGTSTP);
/* Later... */
ioctl(0, TCSETAF, &Rawtty);
signal(SIGTSTP, (void *) (int) stop);
longjmp(redraw_jmp, 1);
}
/*
* Reads the window size and clear the window. This is called on setup,
* and also catches SIGWINCHs, and adjusts Maxlines. Basically, this is
* the central place for window size stuff.
*/
void window_size(void)
{
struct winsize ws;
if (ioctl(1, TIOCGWINSZ, &ws) != -1) {
Cols = ws.ws_col;
Lines = ws.ws_row;
} else {
Cols = tgetnum("co");
Lines = tgetnum("li");
}
clear_screen();
}
/*
* this adjusts the lines needed for the header to the current value
*/
int make_header(void)
{
int i, j;
j = 0;
for (i = 0; i < strlen(Fields); i++) {
if (isupper(Fields[i])) {
pflags[j++] = Fields[i] - 'A';
}
}
strcpy(Header, "");
for (i = 0; i < j; i++)
strcat(Header, headers[pflags[i]]);
/* readjust window size ... */
Maxcmd = Cols - strlen(Header) + 7;
Maxlines = Display_procs ? Display_procs : Lines - header_lines;
if (Maxlines > Lines - header_lines)
Maxlines = Lines - header_lines;
return (j);
}
