Use unsigned printf/scanf conversion where more appropriate
[oweals/busybox.git] / procps / top.c
index ee65551883733cba4a1f07a0aa58f9e9c275d0ba..51f1c1aedf7c21a3073061d69514685400111814 100644 (file)
  * chroot . ./top -bn1 >top1.out
  */
 
+//config:config TOP
+//config:      bool "top"
+//config:      default y
+//config:      help
+//config:        The top program provides a dynamic real-time view of a running
+//config:        system.
+//config:
+//config:config FEATURE_TOP_CPU_USAGE_PERCENTAGE
+//config:      bool "Show CPU per-process usage percentage"
+//config:      default y
+//config:      depends on TOP
+//config:      help
+//config:        Make top display CPU usage for each process.
+//config:        This adds about 2k.
+//config:
+//config:config FEATURE_TOP_CPU_GLOBAL_PERCENTS
+//config:      bool "Show CPU global usage percentage"
+//config:      default y
+//config:      depends on FEATURE_TOP_CPU_USAGE_PERCENTAGE
+//config:      help
+//config:        Makes top display "CPU: NN% usr NN% sys..." line.
+//config:        This adds about 0.5k.
+//config:
+//config:config FEATURE_TOP_SMP_CPU
+//config:      bool "SMP CPU usage display ('c' key)"
+//config:      default y
+//config:      depends on FEATURE_TOP_CPU_GLOBAL_PERCENTS
+//config:      help
+//config:        Allow 'c' key to switch between individual/cumulative CPU stats
+//config:        This adds about 0.5k.
+//config:
+//config:config FEATURE_TOP_DECIMALS
+//config:      bool "Show 1/10th of a percent in CPU/mem statistics"
+//config:      default y
+//config:      depends on FEATURE_TOP_CPU_USAGE_PERCENTAGE
+//config:      help
+//config:        Show 1/10th of a percent in CPU/mem statistics.
+//config:        This adds about 0.3k.
+//config:
+//config:config FEATURE_TOP_SMP_PROCESS
+//config:      bool "Show CPU process runs on ('j' field)"
+//config:      default y
+//config:      depends on TOP
+//config:      help
+//config:        Show CPU where process was last found running on.
+//config:        This is the 'j' field.
+//config:
+//config:config FEATURE_TOPMEM
+//config:      bool "Topmem command ('s' key)"
+//config:      default y
+//config:      depends on TOP
+//config:      help
+//config:        Enable 's' in top (gives lots of memory info).
+
 #include "libbb.h"
 
 
@@ -92,15 +146,20 @@ enum { SORT_DEPTH = 3 };
 struct globals {
        top_status_t *top;
        int ntop;
+       smallint inverted;
 #if ENABLE_FEATURE_TOPMEM
        smallint sort_field;
-       smallint inverted;
 #endif
 #if ENABLE_FEATURE_TOP_SMP_CPU
        smallint smp_cpu_info; /* one/many cpu info lines? */
 #endif
+       unsigned lines;  /* screen height */
 #if ENABLE_FEATURE_USE_TERMIOS
        struct termios initial_settings;
+       int scroll_ofs;
+#define G_scroll_ofs G.scroll_ofs
+#else
+#define G_scroll_ofs 0
 #endif
 #if !ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
        cmp_funcp sort_function[1];
@@ -117,6 +176,9 @@ struct globals {
        /* Per CPU samples: current and last */
        jiffy_counts_t *cpu_jif, *cpu_prev_jif;
        int num_cpus;
+#endif
+#if ENABLE_FEATURE_USE_TERMIOS
+       char kbd_input[KEYCODE_BUFFER_SIZE];
 #endif
        char line_buf[80];
 }; //FIX_ALIASING; - large code growth
@@ -126,7 +188,6 @@ struct BUG_bad_size {
        char BUG_G_too_big[sizeof(G) <= COMMON_BUFSIZE ? 1 : -1];
        char BUG_line_buf_too_small[LINE_BUF_SIZE > 80 ? 1 : -1];
 };
-#define INIT_G() do { } while (0)
 #define top              (G.top               )
 #define ntop             (G.ntop              )
 #define sort_field       (G.sort_field        )
@@ -143,6 +204,7 @@ struct BUG_bad_size {
 #define num_cpus         (G.num_cpus          )
 #define total_pcpu       (G.total_pcpu        )
 #define line_buf         (G.line_buf          )
+#define INIT_G() do { } while (0)
 
 enum {
        OPT_d = (1 << 0),
@@ -194,9 +256,9 @@ static int mult_lvl_cmp(void* a, void* b)
        for (i = 0; i < SORT_DEPTH; i++) {
                cmp_val = (*sort_function[i])(a, b);
                if (cmp_val != 0)
-                       return cmp_val;
+                       break;
        }
-       return 0;
+       return inverted ? -cmp_val : cmp_val;
 }
 
 static NOINLINE int read_cpu_jiffy(FILE *fp, jiffy_counts_t *p_jif)
@@ -232,7 +294,7 @@ static void get_jiffy_counts(void)
         * they are used to calculate per process CPU% */
        prev_jif = cur_jif;
        if (read_cpu_jiffy(fp, &cur_jif) < 4)
-               bb_error_msg_and_die("can't read /proc/stat");
+               bb_error_msg_and_die("can't read '%s'", "/proc/stat");
 
 #if !ENABLE_FEATURE_TOP_SMP_CPU
        fclose(fp);
@@ -602,9 +664,9 @@ static NOINLINE void display_process_list(int lines_rem, int scr_width)
 
        /* Ok, all preliminary data is ready, go through the list */
        scr_width += 2; /* account for leading '\n' and trailing NUL */
-       if (lines_rem > ntop)
-               lines_rem = ntop;
-       s = top;
+       if (lines_rem > ntop - G_scroll_ofs)
+               lines_rem = ntop - G_scroll_ofs;
+       s = top + G_scroll_ofs;
        while (--lines_rem >= 0) {
                unsigned col;
                CALC_STAT(pmem, (s->vsz*pmem_scale + pmem_half) >> pmem_shift);
@@ -615,7 +677,7 @@ static NOINLINE void display_process_list(int lines_rem, int scr_width)
                if (s->vsz >= 100000)
                        sprintf(vsz_str_buf, "%6ldm", s->vsz/1024);
                else
-                       sprintf(vsz_str_buf, "%7ld", s->vsz);
+                       sprintf(vsz_str_buf, "%7lu", s->vsz);
                /* PID PPID USER STAT VSZ %VSZ [%CPU] COMMAND */
                col = snprintf(line_buf, scr_width,
                                "\n" "%5u%6u %-8.8s %s%s" FMT
@@ -649,14 +711,14 @@ static void clearmems(void)
        clear_username_cache();
        free(top);
        top = NULL;
-       ntop = 0;
 }
 
 #if ENABLE_FEATURE_USE_TERMIOS
 
 static void reset_term(void)
 {
-       tcsetattr_stdin_TCSANOW(&initial_settings);
+       if (!OPT_BATCH_MODE)
+               tcsetattr_stdin_TCSANOW(&initial_settings);
        if (ENABLE_FEATURE_CLEAN_UP) {
                clearmems();
 # if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
@@ -665,10 +727,10 @@ static void reset_term(void)
        }
 }
 
-static void sig_catcher(int sig UNUSED_PARAM)
+static void sig_catcher(int sig)
 {
        reset_term();
-       _exit(EXIT_FAILURE);
+       kill_myself_with_sig(sig);
 }
 
 #endif /* FEATURE_USE_TERMIOS */
@@ -785,24 +847,23 @@ static void display_topmem_header(int scr_width, int *lines_rem_p)
 static void ulltoa6_and_space(unsigned long long ul, char buf[6])
 {
        /* see http://en.wikipedia.org/wiki/Tera */
-       smart_ulltoa5(ul, buf, " mgtpezy");
-       buf[5] = ' ';
+       smart_ulltoa5(ul, buf, " mgtpezy")[0] = ' ';
 }
 
 static NOINLINE void display_topmem_process_list(int lines_rem, int scr_width)
 {
 #define HDR_STR "  PID   VSZ VSZRW   RSS (SHR) DIRTY (SHR) STACK"
 #define MIN_WIDTH sizeof(HDR_STR)
-       const topmem_status_t *s = topmem;
+       const topmem_status_t *s = topmem + G_scroll_ofs;
 
        display_topmem_header(scr_width, &lines_rem);
        strcpy(line_buf, HDR_STR " COMMAND");
-       line_buf[5 + sort_field * 6] = '*';
+       line_buf[11 + sort_field * 6] = "^_"[inverted];
        printf(OPT_BATCH_MODE ? "%.*s" : "\e[7m%.*s\e[0m", scr_width, line_buf);
        lines_rem--;
 
-       if (lines_rem > ntop)
-               lines_rem = ntop;
+       if (lines_rem > ntop - G_scroll_ofs)
+               lines_rem = ntop - G_scroll_ofs;
        while (--lines_rem >= 0) {
                /* PID VSZ VSZRW RSS (SHR) DIRTY (SHR) COMMAND */
                ulltoa6_and_space(s->pid     , &line_buf[0*6]);
@@ -850,8 +911,151 @@ enum {
                | PSSCAN_PID
                | PSSCAN_SMAPS
                | PSSCAN_COMM,
+       EXIT_MASK = (unsigned)-1,
 };
 
+#if ENABLE_FEATURE_USE_TERMIOS
+static unsigned handle_input(unsigned scan_mask, unsigned interval)
+{
+       struct pollfd pfd[1];
+
+       if (option_mask32 & OPT_EOF) {
+               /* EOF on stdin ("top </dev/null") */
+               sleep(interval);
+               return scan_mask;
+       }
+
+       pfd[0].fd = 0;
+       pfd[0].events = POLLIN;
+
+       while (1) {
+               int32_t c;
+
+               c = read_key(STDIN_FILENO, G.kbd_input, interval * 1000);
+               if (c == -1 && errno != EAGAIN) {
+                       /* error/EOF */
+                       option_mask32 |= OPT_EOF;
+                       break;
+               }
+               interval = 0;
+
+               if (c == initial_settings.c_cc[VINTR])
+                       return EXIT_MASK;
+               if (c == initial_settings.c_cc[VEOF])
+                       return EXIT_MASK;
+
+               if (c == KEYCODE_UP) {
+                       G_scroll_ofs--;
+                       goto normalize_ofs;
+               }
+               if (c == KEYCODE_DOWN) {
+                       G_scroll_ofs++;
+                       goto normalize_ofs;
+               }
+               if (c == KEYCODE_HOME) {
+                       G_scroll_ofs = 0;
+                       break;
+               }
+               if (c == KEYCODE_END) {
+                       G_scroll_ofs = ntop - G.lines / 2;
+                       goto normalize_ofs;
+               }
+               if (c == KEYCODE_PAGEUP) {
+                       G_scroll_ofs -= G.lines / 2;
+                       goto normalize_ofs;
+               }
+               if (c == KEYCODE_PAGEDOWN) {
+                       G_scroll_ofs += G.lines / 2;
+ normalize_ofs:
+                       if (G_scroll_ofs >= ntop)
+                               G_scroll_ofs = ntop - 1;
+                       if (G_scroll_ofs < 0)
+                               G_scroll_ofs = 0;
+                       break;
+               }
+
+               c |= 0x20; /* lowercase */
+               if (c == 'q')
+                       return EXIT_MASK;
+
+               if (c == 'n') {
+                       IF_FEATURE_TOPMEM(scan_mask = TOP_MASK;)
+                       sort_function[0] = pid_sort;
+                       continue;
+               }
+               if (c == 'm') {
+                       IF_FEATURE_TOPMEM(scan_mask = TOP_MASK;)
+                       sort_function[0] = mem_sort;
+# if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
+                       sort_function[1] = pcpu_sort;
+                       sort_function[2] = time_sort;
+# endif
+                       continue;
+               }
+# if ENABLE_FEATURE_SHOW_THREADS
+               if (c == 'h'
+               IF_FEATURE_TOPMEM(&& scan_mask != TOPMEM_MASK)
+               ) {
+                       scan_mask ^= PSSCAN_TASKS;
+                       continue;
+               }
+# endif
+# if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
+               if (c == 'p') {
+                       IF_FEATURE_TOPMEM(scan_mask = TOP_MASK;)
+                       sort_function[0] = pcpu_sort;
+                       sort_function[1] = mem_sort;
+                       sort_function[2] = time_sort;
+                       continue;
+               }
+               if (c == 't') {
+                       IF_FEATURE_TOPMEM(scan_mask = TOP_MASK;)
+                       sort_function[0] = time_sort;
+                       sort_function[1] = mem_sort;
+                       sort_function[2] = pcpu_sort;
+                       continue;
+               }
+#  if ENABLE_FEATURE_TOPMEM
+               if (c == 's') {
+                       scan_mask = TOPMEM_MASK;
+                       free(prev_hist);
+                       prev_hist = NULL;
+                       prev_hist_count = 0;
+                       sort_field = (sort_field + 1) % NUM_SORT_FIELD;
+                       continue;
+               }
+#  endif
+               if (c == 'r') {
+                       inverted ^= 1;
+                       continue;
+               }
+#  if ENABLE_FEATURE_TOP_SMP_CPU
+               /* procps-2.0.18 uses 'C', 3.2.7 uses '1' */
+               if (c == 'c' || c == '1') {
+                       /* User wants to toggle per cpu <> aggregate */
+                       if (smp_cpu_info) {
+                               free(cpu_prev_jif);
+                               free(cpu_jif);
+                               cpu_jif = &cur_jif;
+                               cpu_prev_jif = &prev_jif;
+                       } else {
+                               /* Prepare for xrealloc() */
+                               cpu_jif = cpu_prev_jif = NULL;
+                       }
+                       num_cpus = 0;
+                       smp_cpu_info = !smp_cpu_info;
+                       get_jiffy_counts();
+                       continue;
+               }
+#  endif
+# endif
+               break; /* unknown key -> force refresh */
+       }
+
+       return scan_mask;
+}
+#endif
+
 //usage:#if ENABLE_FEATURE_SHOW_THREADS || ENABLE_FEATURE_TOP_SMP_CPU
 //usage:# define IF_SHOW_THREADS_OR_TOP_SMP(...) __VA_ARGS__
 //usage:#else
@@ -871,8 +1075,9 @@ enum {
 //usage:                IF_FEATURE_TOP_CPU_USAGE_PERCENTAGE("/cpu")
 //usage:                IF_FEATURE_TOP_CPU_USAGE_PERCENTAGE("/time")
 //usage:       IF_FEATURE_TOPMEM(
-//usage:   "\n""       S: show memory, R: reverse memory sort"
+//usage:   "\n""       S: show memory"
 //usage:       )
+//usage:   "\n""       R: reverse sort"
 //usage:       IF_SHOW_THREADS_OR_TOP_SMP(
 //usage:   "\n""       "
 //usage:                IF_FEATURE_SHOW_THREADS("H: toggle threads")
@@ -880,23 +1085,36 @@ enum {
 //usage:                IF_FEATURE_TOP_SMP_CPU("1: toggle SMP")
 //usage:       )
 //usage:   "\n""       Q,^C: exit"
+//usage:   "\n"
+//usage:   "\n""Options:"
+//usage:   "\n""       -b      Batch mode"
+//usage:   "\n""       -n N    Exit after N iterations"
+//usage:   "\n""       -d N    Delay between updates"
+//usage:       IF_FEATURE_TOPMEM(
+//usage:   "\n""       -m      Same as 's' key"
+//usage:       )
+
+/* Interactive testing:
+ * echo sss | ./busybox top
+ * - shows memory screen
+ * echo sss | ./busybox top -bn1 >mem
+ * - saves memory screen - the *whole* list, not first NROWS processes!
+ * echo .m.s.s.s.s.s.s.q | ./busybox top -b >z
+ * - saves several different screens, and exits
+ *
+ * TODO: -i STRING param as a better alternative?
+ */
 
 int top_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 int top_main(int argc UNUSED_PARAM, char **argv)
 {
        int iterations;
-       unsigned lines, col;
-       int lines_rem;
+       unsigned col;
        unsigned interval;
        char *str_interval, *str_iterations;
        unsigned scan_mask = TOP_MASK;
 #if ENABLE_FEATURE_USE_TERMIOS
        struct termios new_settings;
-       struct pollfd pfd[1];
-       unsigned char c;
-
-       pfd[0].fd = 0;
-       pfd[0].events = POLLIN;
 #endif
 
        INIT_G();
@@ -933,15 +1151,6 @@ int top_main(int argc UNUSED_PARAM, char **argv)
 
        /* change to /proc */
        xchdir("/proc");
-#if ENABLE_FEATURE_USE_TERMIOS
-       tcgetattr(0, (void *) &initial_settings);
-       memcpy(&new_settings, &initial_settings, sizeof(new_settings));
-       /* unbuffered input, turn off echo */
-       new_settings.c_lflag &= ~(ISIG | ICANON | ECHO | ECHONL);
-
-       bb_signals(BB_FATAL_SIGS, sig_catcher);
-       tcsetattr_stdin_TCSANOW(&new_settings);
-#endif
 
 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
        sort_function[0] = pcpu_sort;
@@ -951,23 +1160,47 @@ int top_main(int argc UNUSED_PARAM, char **argv)
        sort_function[0] = mem_sort;
 #endif
 
-       while (1) {
+       if (OPT_BATCH_MODE) {
+               option_mask32 |= OPT_EOF;
+       }
+#if ENABLE_FEATURE_USE_TERMIOS
+       else {
+               tcgetattr(0, (void *) &initial_settings);
+               memcpy(&new_settings, &initial_settings, sizeof(new_settings));
+               /* unbuffered input, turn off echo */
+               new_settings.c_lflag &= ~(ISIG | ICANON | ECHO | ECHONL);
+               tcsetattr_stdin_TCSANOW(&new_settings);
+       }
+
+       bb_signals(BB_FATAL_SIGS, sig_catcher);
+
+       /* Eat initial input, if any */
+       scan_mask = handle_input(scan_mask, 0);
+#endif
+
+       while (scan_mask != EXIT_MASK) {
                procps_status_t *p = NULL;
 
-               lines = 24; /* default */
-               col = 79;
+               if (OPT_BATCH_MODE) {
+                       G.lines = INT_MAX;
+                       col = LINE_BUF_SIZE - 2; /* +2 bytes for '\n', NUL */
+               } else {
+                       G.lines = 24; /* default */
+                       col = 79;
 #if ENABLE_FEATURE_USE_TERMIOS
-               /* We output to stdout, we need size of stdout (not stdin)! */
-               get_terminal_width_height(STDOUT_FILENO, &col, &lines);
-               if (lines < 5 || col < 10) {
-                       sleep(interval);
-                       continue;
-               }
+                       /* We output to stdout, we need size of stdout (not stdin)! */
+                       get_terminal_width_height(STDOUT_FILENO, &col, &G.lines);
+                       if (G.lines < 5 || col < 10) {
+                               sleep(interval);
+                               continue;
+                       }
 #endif
-               if (col > LINE_BUF_SIZE-2) /* +2 bytes for '\n', NUL, */
-                       col = LINE_BUF_SIZE-2;
+                       if (col > LINE_BUF_SIZE - 2)
+                               col = LINE_BUF_SIZE - 2;
+               }
 
                /* read process IDs & status for all the processes */
+               ntop = 0;
                while ((p = procps_scan(p, scan_mask)) != NULL) {
                        int n;
 #if ENABLE_FEATURE_TOPMEM
@@ -1033,15 +1266,11 @@ int top_main(int argc UNUSED_PARAM, char **argv)
                        qsort(topmem, ntop, sizeof(topmem_status_t), (void*)topmem_sort);
                }
 #endif
-               lines_rem = lines;
-               if (OPT_BATCH_MODE) {
-                       lines_rem = INT_MAX;
-               }
                if (scan_mask != TOPMEM_MASK)
-                       display_process_list(lines_rem, col);
+                       display_process_list(G.lines, col);
 #if ENABLE_FEATURE_TOPMEM
                else
-                       display_topmem_process_list(lines_rem, col);
+                       display_topmem_process_list(G.lines, col);
 #endif
                clearmems();
                if (iterations >= 0 && !--iterations)
@@ -1049,84 +1278,9 @@ int top_main(int argc UNUSED_PARAM, char **argv)
 #if !ENABLE_FEATURE_USE_TERMIOS
                sleep(interval);
 #else
-               if (option_mask32 & (OPT_b|OPT_EOF))
-                        /* batch mode, or EOF on stdin ("top </dev/null") */
-                       sleep(interval);
-               else if (safe_poll(pfd, 1, interval * 1000) > 0) {
-                       if (safe_read(STDIN_FILENO, &c, 1) != 1) { /* error/EOF? */
-                               option_mask32 |= OPT_EOF;
-                               continue;
-                       }
-                       if (c == initial_settings.c_cc[VINTR])
-                               break;
-                       c |= 0x20; /* lowercase */
-                       if (c == 'q')
-                               break;
-                       if (c == 'n') {
-                               IF_FEATURE_TOPMEM(scan_mask = TOP_MASK;)
-                               sort_function[0] = pid_sort;
-                       }
-                       if (c == 'm') {
-                               IF_FEATURE_TOPMEM(scan_mask = TOP_MASK;)
-                               sort_function[0] = mem_sort;
-# if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
-                               sort_function[1] = pcpu_sort;
-                               sort_function[2] = time_sort;
-# endif
-                       }
-# if ENABLE_FEATURE_SHOW_THREADS
-                       if (c == 'h'
-                        IF_FEATURE_TOPMEM(&& scan_mask != TOPMEM_MASK)
-                       ) {
-                               scan_mask ^= PSSCAN_TASKS;
-                       }
-# endif
-# if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
-                       if (c == 'p') {
-                               IF_FEATURE_TOPMEM(scan_mask = TOP_MASK;)
-                               sort_function[0] = pcpu_sort;
-                               sort_function[1] = mem_sort;
-                               sort_function[2] = time_sort;
-                       }
-                       if (c == 't') {
-                               IF_FEATURE_TOPMEM(scan_mask = TOP_MASK;)
-                               sort_function[0] = time_sort;
-                               sort_function[1] = mem_sort;
-                               sort_function[2] = pcpu_sort;
-                       }
-#  if ENABLE_FEATURE_TOPMEM
-                       if (c == 's') {
-                               scan_mask = TOPMEM_MASK;
-                               free(prev_hist);
-                               prev_hist = NULL;
-                               prev_hist_count = 0;
-                               sort_field = (sort_field + 1) % NUM_SORT_FIELD;
-                       }
-                       if (c == 'r')
-                               inverted ^= 1;
-#  endif
-#  if ENABLE_FEATURE_TOP_SMP_CPU
-                       /* procps-2.0.18 uses 'C', 3.2.7 uses '1' */
-                       if (c == 'c' || c == '1') {
-                               /* User wants to toggle per cpu <> aggregate */
-                               if (smp_cpu_info) {
-                                       free(cpu_prev_jif);
-                                       free(cpu_jif);
-                                       cpu_jif = &cur_jif;
-                                       cpu_prev_jif = &prev_jif;
-                               } else {
-                                       /* Prepare for xrealloc() */
-                                       cpu_jif = cpu_prev_jif = NULL;
-                               }
-                               num_cpus = 0;
-                               smp_cpu_info = !smp_cpu_info;
-                               get_jiffy_counts();
-                       }
-#  endif
-# endif
-               }
+               scan_mask = handle_input(scan_mask, interval);
 #endif /* FEATURE_USE_TERMIOS */
-       } /* end of "while (1)" */
+       } /* end of "while (not Q)" */
 
        bb_putchar('\n');
 #if ENABLE_FEATURE_USE_TERMIOS