iostat: code shrink ~0.5k
[platform/upstream/busybox.git] / procps / top.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * A tiny 'top' utility.
4  *
5  * This is written specifically for the linux /proc/<PID>/stat(m)
6  * files format.
7  *
8  * This reads the PIDs of all processes and their status and shows
9  * the status of processes (first ones that fit to screen) at given
10  * intervals.
11  *
12  * NOTES:
13  * - At startup this changes to /proc, all the reads are then
14  *   relative to that.
15  *
16  * (C) Eero Tamminen <oak at welho dot com>
17  *
18  * Rewritten by Vladimir Oleynik (C) 2002 <dzo@simtreas.ru>
19  *
20  * Sept 2008: Vineet Gupta <vineet.gupta@arc.com>
21  * Added Support for reporting SMP Information
22  * - CPU where process was last seen running
23  *   (to see effect of sched_setaffinity() etc)
24  * - CPU time split (idle/IO/wait etc) per CPU
25  *
26  * Copyright (c) 1992 Branko Lankester
27  * Copyright (c) 1992 Roger Binns
28  * Copyright (C) 1994-1996 Charles L. Blake.
29  * Copyright (C) 1992-1998 Michael K. Johnson
30  *
31  * Licensed under GPLv2, see file LICENSE in this source tree.
32  */
33 /* How to snapshot /proc for debugging top problems:
34  * for f in /proc/[0-9]*""/stat; do
35  *         n=${f#/proc/}
36  *         n=${n%/stat}_stat
37  *         cp $f $n
38  * done
39  * cp /proc/stat /proc/meminfo /proc/loadavg .
40  * top -bn1 >top.out
41  *
42  * ...and how to run top on it on another machine:
43  * rm -rf proc; mkdir proc
44  * for f in [0-9]*_stat; do
45  *         p=${f%_stat}
46  *         mkdir -p proc/$p
47  *         cp $f proc/$p/stat
48  * done
49  * cp stat meminfo loadavg proc
50  * chroot . ./top -bn1 >top1.out
51  */
52
53 #include "libbb.h"
54
55
56 typedef struct top_status_t {
57         unsigned long vsz;
58 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
59         unsigned long ticks;
60         unsigned pcpu; /* delta of ticks */
61 #endif
62         unsigned pid, ppid;
63         unsigned uid;
64         char state[4];
65         char comm[COMM_LEN];
66 #if ENABLE_FEATURE_TOP_SMP_PROCESS
67         int last_seen_on_cpu;
68 #endif
69 } top_status_t;
70
71 typedef struct jiffy_counts_t {
72         /* Linux 2.4.x has only first four */
73         unsigned long long usr, nic, sys, idle;
74         unsigned long long iowait, irq, softirq, steal;
75         unsigned long long total;
76         unsigned long long busy;
77 } jiffy_counts_t;
78
79 /* This structure stores some critical information from one frame to
80    the next. Used for finding deltas. */
81 typedef struct save_hist {
82         unsigned long ticks;
83         pid_t pid;
84 } save_hist;
85
86 typedef int (*cmp_funcp)(top_status_t *P, top_status_t *Q);
87
88
89 enum { SORT_DEPTH = 3 };
90
91
92 struct globals {
93         top_status_t *top;
94         int ntop;
95         smallint inverted;
96 #if ENABLE_FEATURE_TOPMEM
97         smallint sort_field;
98 #endif
99 #if ENABLE_FEATURE_TOP_SMP_CPU
100         smallint smp_cpu_info; /* one/many cpu info lines? */
101 #endif
102 #if ENABLE_FEATURE_USE_TERMIOS
103         struct termios initial_settings;
104 #endif
105 #if !ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
106         cmp_funcp sort_function[1];
107 #else
108         cmp_funcp sort_function[SORT_DEPTH];
109         struct save_hist *prev_hist;
110         int prev_hist_count;
111         jiffy_counts_t cur_jif, prev_jif;
112         /* int hist_iterations; */
113         unsigned total_pcpu;
114         /* unsigned long total_vsz; */
115 #endif
116 #if ENABLE_FEATURE_TOP_SMP_CPU
117         /* Per CPU samples: current and last */
118         jiffy_counts_t *cpu_jif, *cpu_prev_jif;
119         int num_cpus;
120 #endif
121         char line_buf[80];
122 }; //FIX_ALIASING; - large code growth
123 enum { LINE_BUF_SIZE = COMMON_BUFSIZE - offsetof(struct globals, line_buf) };
124 #define G (*(struct globals*)&bb_common_bufsiz1)
125 struct BUG_bad_size {
126         char BUG_G_too_big[sizeof(G) <= COMMON_BUFSIZE ? 1 : -1];
127         char BUG_line_buf_too_small[LINE_BUF_SIZE > 80 ? 1 : -1];
128 };
129 #define INIT_G() do { } while (0)
130 #define top              (G.top               )
131 #define ntop             (G.ntop              )
132 #define sort_field       (G.sort_field        )
133 #define inverted         (G.inverted          )
134 #define smp_cpu_info     (G.smp_cpu_info      )
135 #define initial_settings (G.initial_settings  )
136 #define sort_function    (G.sort_function     )
137 #define prev_hist        (G.prev_hist         )
138 #define prev_hist_count  (G.prev_hist_count   )
139 #define cur_jif          (G.cur_jif           )
140 #define prev_jif         (G.prev_jif          )
141 #define cpu_jif          (G.cpu_jif           )
142 #define cpu_prev_jif     (G.cpu_prev_jif      )
143 #define num_cpus         (G.num_cpus          )
144 #define total_pcpu       (G.total_pcpu        )
145 #define line_buf         (G.line_buf          )
146
147 enum {
148         OPT_d = (1 << 0),
149         OPT_n = (1 << 1),
150         OPT_b = (1 << 2),
151         OPT_m = (1 << 3),
152         OPT_EOF = (1 << 4), /* pseudo: "we saw EOF in stdin" */
153 };
154 #define OPT_BATCH_MODE (option_mask32 & OPT_b)
155
156
157 #if ENABLE_FEATURE_USE_TERMIOS
158 static int pid_sort(top_status_t *P, top_status_t *Q)
159 {
160         /* Buggy wrt pids with high bit set */
161         /* (linux pids are in [1..2^15-1]) */
162         return (Q->pid - P->pid);
163 }
164 #endif
165
166 static int mem_sort(top_status_t *P, top_status_t *Q)
167 {
168         /* We want to avoid unsigned->signed and truncation errors */
169         if (Q->vsz < P->vsz) return -1;
170         return Q->vsz != P->vsz; /* 0 if ==, 1 if > */
171 }
172
173
174 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
175
176 static int pcpu_sort(top_status_t *P, top_status_t *Q)
177 {
178         /* Buggy wrt ticks with high bit set */
179         /* Affects only processes for which ticks overflow */
180         return (int)Q->pcpu - (int)P->pcpu;
181 }
182
183 static int time_sort(top_status_t *P, top_status_t *Q)
184 {
185         /* We want to avoid unsigned->signed and truncation errors */
186         if (Q->ticks < P->ticks) return -1;
187         return Q->ticks != P->ticks; /* 0 if ==, 1 if > */
188 }
189
190 static int mult_lvl_cmp(void* a, void* b)
191 {
192         int i, cmp_val;
193
194         for (i = 0; i < SORT_DEPTH; i++) {
195                 cmp_val = (*sort_function[i])(a, b);
196                 if (cmp_val != 0)
197                         break;
198         }
199         return inverted ? -cmp_val : cmp_val;
200 }
201
202 static NOINLINE int read_cpu_jiffy(FILE *fp, jiffy_counts_t *p_jif)
203 {
204 #if !ENABLE_FEATURE_TOP_SMP_CPU
205         static const char fmt[] = "cpu %llu %llu %llu %llu %llu %llu %llu %llu";
206 #else
207         static const char fmt[] = "cp%*s %llu %llu %llu %llu %llu %llu %llu %llu";
208 #endif
209         int ret;
210
211         if (!fgets(line_buf, LINE_BUF_SIZE, fp) || line_buf[0] != 'c' /* not "cpu" */)
212                 return 0;
213         ret = sscanf(line_buf, fmt,
214                         &p_jif->usr, &p_jif->nic, &p_jif->sys, &p_jif->idle,
215                         &p_jif->iowait, &p_jif->irq, &p_jif->softirq,
216                         &p_jif->steal);
217         if (ret >= 4) {
218                 p_jif->total = p_jif->usr + p_jif->nic + p_jif->sys + p_jif->idle
219                         + p_jif->iowait + p_jif->irq + p_jif->softirq + p_jif->steal;
220                 /* procps 2.x does not count iowait as busy time */
221                 p_jif->busy = p_jif->total - p_jif->idle - p_jif->iowait;
222         }
223
224         return ret;
225 }
226
227 static void get_jiffy_counts(void)
228 {
229         FILE* fp = xfopen_for_read("stat");
230
231         /* We need to parse cumulative counts even if SMP CPU display is on,
232          * they are used to calculate per process CPU% */
233         prev_jif = cur_jif;
234         if (read_cpu_jiffy(fp, &cur_jif) < 4)
235                 bb_error_msg_and_die("can't read /proc/stat");
236
237 #if !ENABLE_FEATURE_TOP_SMP_CPU
238         fclose(fp);
239         return;
240 #else
241         if (!smp_cpu_info) {
242                 fclose(fp);
243                 return;
244         }
245
246         if (!num_cpus) {
247                 /* First time here. How many CPUs?
248                  * There will be at least 1 /proc/stat line with cpu%d
249                  */
250                 while (1) {
251                         cpu_jif = xrealloc_vector(cpu_jif, 1, num_cpus);
252                         if (read_cpu_jiffy(fp, &cpu_jif[num_cpus]) <= 4)
253                                 break;
254                         num_cpus++;
255                 }
256                 if (num_cpus == 0) /* /proc/stat with only "cpu ..." line?! */
257                         smp_cpu_info = 0;
258
259                 cpu_prev_jif = xzalloc(sizeof(cpu_prev_jif[0]) * num_cpus);
260
261                 /* Otherwise the first per cpu display shows all 100% idles */
262                 usleep(50000);
263         } else { /* Non first time invocation */
264                 jiffy_counts_t *tmp;
265                 int i;
266
267                 /* First switch the sample pointers: no need to copy */
268                 tmp = cpu_prev_jif;
269                 cpu_prev_jif = cpu_jif;
270                 cpu_jif = tmp;
271
272                 /* Get the new samples */
273                 for (i = 0; i < num_cpus; i++)
274                         read_cpu_jiffy(fp, &cpu_jif[i]);
275         }
276 #endif
277         fclose(fp);
278 }
279
280 static void do_stats(void)
281 {
282         top_status_t *cur;
283         pid_t pid;
284         int i, last_i, n;
285         struct save_hist *new_hist;
286
287         get_jiffy_counts();
288         total_pcpu = 0;
289         /* total_vsz = 0; */
290         new_hist = xmalloc(sizeof(new_hist[0]) * ntop);
291         /*
292          * Make a pass through the data to get stats.
293          */
294         /* hist_iterations = 0; */
295         i = 0;
296         for (n = 0; n < ntop; n++) {
297                 cur = top + n;
298
299                 /*
300                  * Calculate time in cur process.  Time is sum of user time
301                  * and system time
302                  */
303                 pid = cur->pid;
304                 new_hist[n].ticks = cur->ticks;
305                 new_hist[n].pid = pid;
306
307                 /* find matching entry from previous pass */
308                 cur->pcpu = 0;
309                 /* do not start at index 0, continue at last used one
310                  * (brought hist_iterations from ~14000 down to 172) */
311                 last_i = i;
312                 if (prev_hist_count) do {
313                         if (prev_hist[i].pid == pid) {
314                                 cur->pcpu = cur->ticks - prev_hist[i].ticks;
315                                 total_pcpu += cur->pcpu;
316                                 break;
317                         }
318                         i = (i+1) % prev_hist_count;
319                         /* hist_iterations++; */
320                 } while (i != last_i);
321                 /* total_vsz += cur->vsz; */
322         }
323
324         /*
325          * Save cur frame's information.
326          */
327         free(prev_hist);
328         prev_hist = new_hist;
329         prev_hist_count = ntop;
330 }
331
332 #endif /* FEATURE_TOP_CPU_USAGE_PERCENTAGE */
333
334 #if ENABLE_FEATURE_TOP_CPU_GLOBAL_PERCENTS && ENABLE_FEATURE_TOP_DECIMALS
335 /* formats 7 char string (8 with terminating NUL) */
336 static char *fmt_100percent_8(char pbuf[8], unsigned value, unsigned total)
337 {
338         unsigned t;
339         if (value >= total) { /* 100% ? */
340                 strcpy(pbuf, "  100% ");
341                 return pbuf;
342         }
343         /* else generate " [N/space]N.N% " string */
344         value = 1000 * value / total;
345         t = value / 100;
346         value = value % 100;
347         pbuf[0] = ' ';
348         pbuf[1] = t ? t + '0' : ' ';
349         pbuf[2] = '0' + (value / 10);
350         pbuf[3] = '.';
351         pbuf[4] = '0' + (value % 10);
352         pbuf[5] = '%';
353         pbuf[6] = ' ';
354         pbuf[7] = '\0';
355         return pbuf;
356 }
357 #endif
358
359 #if ENABLE_FEATURE_TOP_CPU_GLOBAL_PERCENTS
360 static void display_cpus(int scr_width, char *scrbuf, int *lines_rem_p)
361 {
362         /*
363          * xxx% = (cur_jif.xxx - prev_jif.xxx) / (cur_jif.total - prev_jif.total) * 100%
364          */
365         unsigned total_diff;
366         jiffy_counts_t *p_jif, *p_prev_jif;
367         int i;
368 # if ENABLE_FEATURE_TOP_SMP_CPU
369         int n_cpu_lines;
370 # endif
371
372         /* using (unsigned) casts to make operations cheaper */
373 # define  CALC_TOTAL_DIFF do { \
374         total_diff = (unsigned)(p_jif->total - p_prev_jif->total); \
375         if (total_diff == 0) total_diff = 1; \
376 } while (0)
377
378 # if ENABLE_FEATURE_TOP_DECIMALS
379 #  define CALC_STAT(xxx) char xxx[8]
380 #  define SHOW_STAT(xxx) fmt_100percent_8(xxx, (unsigned)(p_jif->xxx - p_prev_jif->xxx), total_diff)
381 #  define FMT "%s"
382 # else
383 #  define CALC_STAT(xxx) unsigned xxx = 100 * (unsigned)(p_jif->xxx - p_prev_jif->xxx) / total_diff
384 #  define SHOW_STAT(xxx) xxx
385 #  define FMT "%4u%% "
386 # endif
387
388 # if !ENABLE_FEATURE_TOP_SMP_CPU
389         {
390                 i = 1;
391                 p_jif = &cur_jif;
392                 p_prev_jif = &prev_jif;
393 # else
394         /* Loop thru CPU(s) */
395         n_cpu_lines = smp_cpu_info ? num_cpus : 1;
396         if (n_cpu_lines > *lines_rem_p)
397                 n_cpu_lines = *lines_rem_p;
398
399         for (i = 0; i < n_cpu_lines; i++) {
400                 p_jif = &cpu_jif[i];
401                 p_prev_jif = &cpu_prev_jif[i];
402 # endif
403                 CALC_TOTAL_DIFF;
404
405                 { /* Need a block: CALC_STAT are declarations */
406                         CALC_STAT(usr);
407                         CALC_STAT(sys);
408                         CALC_STAT(nic);
409                         CALC_STAT(idle);
410                         CALC_STAT(iowait);
411                         CALC_STAT(irq);
412                         CALC_STAT(softirq);
413                         /*CALC_STAT(steal);*/
414
415                         snprintf(scrbuf, scr_width,
416                                 /* Barely fits in 79 chars when in "decimals" mode. */
417 # if ENABLE_FEATURE_TOP_SMP_CPU
418                                 "CPU%s:"FMT"usr"FMT"sys"FMT"nic"FMT"idle"FMT"io"FMT"irq"FMT"sirq",
419                                 (smp_cpu_info ? utoa(i) : ""),
420 # else
421                                 "CPU:"FMT"usr"FMT"sys"FMT"nic"FMT"idle"FMT"io"FMT"irq"FMT"sirq",
422 # endif
423                                 SHOW_STAT(usr), SHOW_STAT(sys), SHOW_STAT(nic), SHOW_STAT(idle),
424                                 SHOW_STAT(iowait), SHOW_STAT(irq), SHOW_STAT(softirq)
425                                 /*, SHOW_STAT(steal) - what is this 'steal' thing? */
426                                 /* I doubt anyone wants to know it */
427                         );
428                         puts(scrbuf);
429                 }
430         }
431 # undef SHOW_STAT
432 # undef CALC_STAT
433 # undef FMT
434         *lines_rem_p -= i;
435 }
436 #else  /* !ENABLE_FEATURE_TOP_CPU_GLOBAL_PERCENTS */
437 # define display_cpus(scr_width, scrbuf, lines_rem) ((void)0)
438 #endif
439
440 static unsigned long display_header(int scr_width, int *lines_rem_p)
441 {
442         FILE *fp;
443         char buf[80];
444         char scrbuf[80];
445         unsigned long total, used, mfree, shared, buffers, cached;
446
447         /* read memory info */
448         fp = xfopen_for_read("meminfo");
449
450         /*
451          * Old kernels (such as 2.4.x) had a nice summary of memory info that
452          * we could parse, however this is gone entirely in 2.6. Try parsing
453          * the old way first, and if that fails, parse each field manually.
454          *
455          * First, we read in the first line. Old kernels will have bogus
456          * strings we don't care about, whereas new kernels will start right
457          * out with MemTotal:
458          *                              -- PFM.
459          */
460         if (fscanf(fp, "MemTotal: %lu %s\n", &total, buf) != 2) {
461                 fgets(buf, sizeof(buf), fp);    /* skip first line */
462
463                 fscanf(fp, "Mem: %lu %lu %lu %lu %lu %lu",
464                         &total, &used, &mfree, &shared, &buffers, &cached);
465                 /* convert to kilobytes */
466                 used /= 1024;
467                 mfree /= 1024;
468                 shared /= 1024;
469                 buffers /= 1024;
470                 cached /= 1024;
471                 total /= 1024;
472         } else {
473                 /*
474                  * Revert to manual parsing, which incidentally already has the
475                  * sizes in kilobytes. This should be safe for both 2.4 and
476                  * 2.6.
477                  */
478                 fscanf(fp, "MemFree: %lu %s\n", &mfree, buf);
479
480                 /*
481                  * MemShared: is no longer present in 2.6. Report this as 0,
482                  * to maintain consistent behavior with normal procps.
483                  */
484                 if (fscanf(fp, "MemShared: %lu %s\n", &shared, buf) != 2)
485                         shared = 0;
486
487                 fscanf(fp, "Buffers: %lu %s\n", &buffers, buf);
488                 fscanf(fp, "Cached: %lu %s\n", &cached, buf);
489
490                 used = total - mfree;
491         }
492         fclose(fp);
493
494         /* output memory info */
495         if (scr_width > (int)sizeof(scrbuf))
496                 scr_width = sizeof(scrbuf);
497         snprintf(scrbuf, scr_width,
498                 "Mem: %luK used, %luK free, %luK shrd, %luK buff, %luK cached",
499                 used, mfree, shared, buffers, cached);
500         /* go to top & clear to the end of screen */
501         printf(OPT_BATCH_MODE ? "%s\n" : "\033[H\033[J%s\n", scrbuf);
502         (*lines_rem_p)--;
503
504         /* Display CPU time split as percentage of total time
505          * This displays either a cumulative line or one line per CPU
506          */
507         display_cpus(scr_width, scrbuf, lines_rem_p);
508
509         /* read load average as a string */
510         buf[0] = '\0';
511         open_read_close("loadavg", buf, sizeof(buf) - 1);
512         buf[sizeof(buf) - 1] = '\n';
513         *strchr(buf, '\n') = '\0';
514         snprintf(scrbuf, scr_width, "Load average: %s", buf);
515         puts(scrbuf);
516         (*lines_rem_p)--;
517
518         return total;
519 }
520
521 static NOINLINE void display_process_list(int lines_rem, int scr_width)
522 {
523         enum {
524                 BITS_PER_INT = sizeof(int) * 8
525         };
526
527         top_status_t *s;
528         char vsz_str_buf[8];
529         unsigned long total_memory = display_header(scr_width, &lines_rem); /* or use total_vsz? */
530         /* xxx_shift and xxx_scale variables allow us to replace
531          * expensive divides with multiply and shift */
532         unsigned pmem_shift, pmem_scale, pmem_half;
533 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
534         unsigned tmp_unsigned;
535         unsigned pcpu_shift, pcpu_scale, pcpu_half;
536         unsigned busy_jifs;
537 #endif
538
539         /* what info of the processes is shown */
540         printf(OPT_BATCH_MODE ? "%.*s" : "\033[7m%.*s\033[0m", scr_width,
541                 "  PID  PPID USER     STAT   VSZ %VSZ"
542                 IF_FEATURE_TOP_SMP_PROCESS(" CPU")
543                 IF_FEATURE_TOP_CPU_USAGE_PERCENTAGE(" %CPU")
544                 " COMMAND");
545         lines_rem--;
546
547 #if ENABLE_FEATURE_TOP_DECIMALS
548 # define UPSCALE 1000
549 # define CALC_STAT(name, val) div_t name = div((val), 10)
550 # define SHOW_STAT(name) name.quot, '0'+name.rem
551 # define FMT "%3u.%c"
552 #else
553 # define UPSCALE 100
554 # define CALC_STAT(name, val) unsigned name = (val)
555 # define SHOW_STAT(name) name
556 # define FMT "%4u%%"
557 #endif
558         /*
559          * %VSZ = s->vsz/MemTotal
560          */
561         pmem_shift = BITS_PER_INT-11;
562         pmem_scale = UPSCALE*(1U<<(BITS_PER_INT-11)) / total_memory;
563         /* s->vsz is in kb. we want (s->vsz * pmem_scale) to never overflow */
564         while (pmem_scale >= 512) {
565                 pmem_scale /= 4;
566                 pmem_shift -= 2;
567         }
568         pmem_half = (1U << pmem_shift) / (ENABLE_FEATURE_TOP_DECIMALS ? 20 : 2);
569 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
570         busy_jifs = cur_jif.busy - prev_jif.busy;
571         /* This happens if there were lots of short-lived processes
572          * between two top updates (e.g. compilation) */
573         if (total_pcpu < busy_jifs) total_pcpu = busy_jifs;
574
575         /*
576          * CPU% = s->pcpu/sum(s->pcpu) * busy_cpu_ticks/total_cpu_ticks
577          * (pcpu is delta of sys+user time between samples)
578          */
579         /* (cur_jif.xxx - prev_jif.xxx) and s->pcpu are
580          * in 0..~64000 range (HZ*update_interval).
581          * we assume that unsigned is at least 32-bit.
582          */
583         pcpu_shift = 6;
584         pcpu_scale = UPSCALE*64 * (uint16_t)busy_jifs;
585         if (pcpu_scale == 0)
586                 pcpu_scale = 1;
587         while (pcpu_scale < (1U << (BITS_PER_INT-2))) {
588                 pcpu_scale *= 4;
589                 pcpu_shift += 2;
590         }
591         tmp_unsigned = (uint16_t)(cur_jif.total - prev_jif.total) * total_pcpu;
592         if (tmp_unsigned != 0)
593                 pcpu_scale /= tmp_unsigned;
594         /* we want (s->pcpu * pcpu_scale) to never overflow */
595         while (pcpu_scale >= 1024) {
596                 pcpu_scale /= 4;
597                 pcpu_shift -= 2;
598         }
599         pcpu_half = (1U << pcpu_shift) / (ENABLE_FEATURE_TOP_DECIMALS ? 20 : 2);
600         /* printf(" pmem_scale=%u pcpu_scale=%u ", pmem_scale, pcpu_scale); */
601 #endif
602
603         /* Ok, all preliminary data is ready, go through the list */
604         scr_width += 2; /* account for leading '\n' and trailing NUL */
605         if (lines_rem > ntop)
606                 lines_rem = ntop;
607         s = top;
608         while (--lines_rem >= 0) {
609                 unsigned col;
610                 CALC_STAT(pmem, (s->vsz*pmem_scale + pmem_half) >> pmem_shift);
611 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
612                 CALC_STAT(pcpu, (s->pcpu*pcpu_scale + pcpu_half) >> pcpu_shift);
613 #endif
614
615                 if (s->vsz >= 100000)
616                         sprintf(vsz_str_buf, "%6ldm", s->vsz/1024);
617                 else
618                         sprintf(vsz_str_buf, "%7ld", s->vsz);
619                 /* PID PPID USER STAT VSZ %VSZ [%CPU] COMMAND */
620                 col = snprintf(line_buf, scr_width,
621                                 "\n" "%5u%6u %-8.8s %s%s" FMT
622                                 IF_FEATURE_TOP_SMP_PROCESS(" %3d")
623                                 IF_FEATURE_TOP_CPU_USAGE_PERCENTAGE(FMT)
624                                 " ",
625                                 s->pid, s->ppid, get_cached_username(s->uid),
626                                 s->state, vsz_str_buf,
627                                 SHOW_STAT(pmem)
628                                 IF_FEATURE_TOP_SMP_PROCESS(, s->last_seen_on_cpu)
629                                 IF_FEATURE_TOP_CPU_USAGE_PERCENTAGE(, SHOW_STAT(pcpu))
630                 );
631                 if ((int)(col + 1) < scr_width)
632                         read_cmdline(line_buf + col, scr_width - col, s->pid, s->comm);
633                 fputs(line_buf, stdout);
634                 /* printf(" %d/%d %lld/%lld", s->pcpu, total_pcpu,
635                         cur_jif.busy - prev_jif.busy, cur_jif.total - prev_jif.total); */
636                 s++;
637         }
638         /* printf(" %d", hist_iterations); */
639         bb_putchar(OPT_BATCH_MODE ? '\n' : '\r');
640         fflush_all();
641 }
642 #undef UPSCALE
643 #undef SHOW_STAT
644 #undef CALC_STAT
645 #undef FMT
646
647 static void clearmems(void)
648 {
649         clear_username_cache();
650         free(top);
651         top = NULL;
652         ntop = 0;
653 }
654
655 #if ENABLE_FEATURE_USE_TERMIOS
656
657 static void reset_term(void)
658 {
659         tcsetattr_stdin_TCSANOW(&initial_settings);
660         if (ENABLE_FEATURE_CLEAN_UP) {
661                 clearmems();
662 # if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
663                 free(prev_hist);
664 # endif
665         }
666 }
667
668 static void sig_catcher(int sig UNUSED_PARAM)
669 {
670         reset_term();
671         _exit(EXIT_FAILURE);
672 }
673
674 #endif /* FEATURE_USE_TERMIOS */
675
676 /*
677  * TOPMEM support
678  */
679
680 typedef unsigned long mem_t;
681
682 typedef struct topmem_status_t {
683         unsigned pid;
684         char comm[COMM_LEN];
685         /* vsz doesn't count /dev/xxx mappings except /dev/zero */
686         mem_t vsz     ;
687         mem_t vszrw   ;
688         mem_t rss     ;
689         mem_t rss_sh  ;
690         mem_t dirty   ;
691         mem_t dirty_sh;
692         mem_t stack   ;
693 } topmem_status_t;
694
695 enum { NUM_SORT_FIELD = 7 };
696
697 #define topmem ((topmem_status_t*)top)
698
699 #if ENABLE_FEATURE_TOPMEM
700
701 static int topmem_sort(char *a, char *b)
702 {
703         int n;
704         mem_t l, r;
705
706         n = offsetof(topmem_status_t, vsz) + (sort_field * sizeof(mem_t));
707         l = *(mem_t*)(a + n);
708         r = *(mem_t*)(b + n);
709         if (l == r) {
710                 l = ((topmem_status_t*)a)->dirty;
711                 r = ((topmem_status_t*)b)->dirty;
712         }
713         /* We want to avoid unsigned->signed and truncation errors */
714         /* l>r: -1, l=r: 0, l<r: 1 */
715         n = (l > r) ? -1 : (l != r);
716         return inverted ? -n : n;
717 }
718
719 /* display header info (meminfo / loadavg) */
720 static void display_topmem_header(int scr_width, int *lines_rem_p)
721 {
722         enum {
723                 TOTAL = 0, MFREE, BUF, CACHE,
724                 SWAPTOTAL, SWAPFREE, DIRTY,
725                 MWRITE, ANON, MAP, SLAB,
726                 NUM_FIELDS
727         };
728         static const char match[NUM_FIELDS][12] = {
729                 "\x09" "MemTotal:",  // TOTAL
730                 "\x08" "MemFree:",   // MFREE
731                 "\x08" "Buffers:",   // BUF
732                 "\x07" "Cached:",    // CACHE
733                 "\x0a" "SwapTotal:", // SWAPTOTAL
734                 "\x09" "SwapFree:",  // SWAPFREE
735                 "\x06" "Dirty:",     // DIRTY
736                 "\x0a" "Writeback:", // MWRITE
737                 "\x0a" "AnonPages:", // ANON
738                 "\x07" "Mapped:",    // MAP
739                 "\x05" "Slab:",      // SLAB
740         };
741         char meminfo_buf[4 * 1024];
742         const char *Z[NUM_FIELDS];
743         unsigned i;
744         int sz;
745
746         for (i = 0; i < NUM_FIELDS; i++)
747                 Z[i] = "?";
748
749         /* read memory info */
750         sz = open_read_close("meminfo", meminfo_buf, sizeof(meminfo_buf) - 1);
751         if (sz >= 0) {
752                 char *p = meminfo_buf;
753                 meminfo_buf[sz] = '\0';
754                 /* Note that fields always appear in the match[] order */
755                 for (i = 0; i < NUM_FIELDS; i++) {
756                         char *found = strstr(p, match[i] + 1);
757                         if (found) {
758                                 /* Cut "NNNN" out of "    NNNN kb" */
759                                 char *s = skip_whitespace(found + match[i][0]);
760                                 p = skip_non_whitespace(s);
761                                 *p++ = '\0';
762                                 Z[i] = s;
763                         }
764                 }
765         }
766
767         snprintf(line_buf, LINE_BUF_SIZE,
768                 "Mem total:%s anon:%s map:%s free:%s",
769                 Z[TOTAL], Z[ANON], Z[MAP], Z[MFREE]);
770         printf(OPT_BATCH_MODE ? "%.*s\n" : "\033[H\033[J%.*s\n", scr_width, line_buf);
771
772         snprintf(line_buf, LINE_BUF_SIZE,
773                 " slab:%s buf:%s cache:%s dirty:%s write:%s",
774                 Z[SLAB], Z[BUF], Z[CACHE], Z[DIRTY], Z[MWRITE]);
775         printf("%.*s\n", scr_width, line_buf);
776
777         snprintf(line_buf, LINE_BUF_SIZE,
778                 "Swap total:%s free:%s", // TODO: % used?
779                 Z[SWAPTOTAL], Z[SWAPFREE]);
780         printf("%.*s\n", scr_width, line_buf);
781
782         (*lines_rem_p) -= 3;
783 }
784
785 static void ulltoa6_and_space(unsigned long long ul, char buf[6])
786 {
787         /* see http://en.wikipedia.org/wiki/Tera */
788         smart_ulltoa5(ul, buf, " mgtpezy");
789         buf[5] = ' ';
790 }
791
792 static NOINLINE void display_topmem_process_list(int lines_rem, int scr_width)
793 {
794 #define HDR_STR "  PID   VSZ VSZRW   RSS (SHR) DIRTY (SHR) STACK"
795 #define MIN_WIDTH sizeof(HDR_STR)
796         const topmem_status_t *s = topmem;
797
798         display_topmem_header(scr_width, &lines_rem);
799         strcpy(line_buf, HDR_STR " COMMAND");
800         line_buf[5 + sort_field * 6] = '*';
801         printf(OPT_BATCH_MODE ? "%.*s" : "\e[7m%.*s\e[0m", scr_width, line_buf);
802         lines_rem--;
803
804         if (lines_rem > ntop)
805                 lines_rem = ntop;
806         while (--lines_rem >= 0) {
807                 /* PID VSZ VSZRW RSS (SHR) DIRTY (SHR) COMMAND */
808                 ulltoa6_and_space(s->pid     , &line_buf[0*6]);
809                 ulltoa6_and_space(s->vsz     , &line_buf[1*6]);
810                 ulltoa6_and_space(s->vszrw   , &line_buf[2*6]);
811                 ulltoa6_and_space(s->rss     , &line_buf[3*6]);
812                 ulltoa6_and_space(s->rss_sh  , &line_buf[4*6]);
813                 ulltoa6_and_space(s->dirty   , &line_buf[5*6]);
814                 ulltoa6_and_space(s->dirty_sh, &line_buf[6*6]);
815                 ulltoa6_and_space(s->stack   , &line_buf[7*6]);
816                 line_buf[8*6] = '\0';
817                 if (scr_width > (int)MIN_WIDTH) {
818                         read_cmdline(&line_buf[8*6], scr_width - MIN_WIDTH, s->pid, s->comm);
819                 }
820                 printf("\n""%.*s", scr_width, line_buf);
821                 s++;
822         }
823         bb_putchar(OPT_BATCH_MODE ? '\n' : '\r');
824         fflush_all();
825 #undef HDR_STR
826 #undef MIN_WIDTH
827 }
828
829 #else
830 void display_topmem_process_list(int lines_rem, int scr_width);
831 int topmem_sort(char *a, char *b);
832 #endif /* TOPMEM */
833
834 /*
835  * end TOPMEM support
836  */
837
838 enum {
839         TOP_MASK = 0
840                 | PSSCAN_PID
841                 | PSSCAN_PPID
842                 | PSSCAN_VSZ
843                 | PSSCAN_STIME
844                 | PSSCAN_UTIME
845                 | PSSCAN_STATE
846                 | PSSCAN_COMM
847                 | PSSCAN_CPU
848                 | PSSCAN_UIDGID,
849         TOPMEM_MASK = 0
850                 | PSSCAN_PID
851                 | PSSCAN_SMAPS
852                 | PSSCAN_COMM,
853         EXIT_MASK = (unsigned)-1,
854 };
855
856 #if ENABLE_FEATURE_USE_TERMIOS
857 static unsigned handle_input(unsigned scan_mask, unsigned interval)
858 {
859         unsigned char c;
860         struct pollfd pfd[1];
861
862         pfd[0].fd = 0;
863         pfd[0].events = POLLIN;
864
865         while (1) {
866                 if (safe_poll(pfd, 1, interval * 1000) <= 0)
867                         return scan_mask;
868                 interval = 0;
869
870                 if (safe_read(STDIN_FILENO, &c, 1) != 1) { /* error/EOF? */
871                         option_mask32 |= OPT_EOF;
872                         return scan_mask;
873                 }
874
875                 if (c == initial_settings.c_cc[VINTR])
876                         return EXIT_MASK;
877                 if (c == initial_settings.c_cc[VEOF])
878                         return EXIT_MASK;
879                 c |= 0x20; /* lowercase */
880                 if (c == 'q')
881                         return EXIT_MASK;
882
883                 if (c == 'n') {
884                         IF_FEATURE_TOPMEM(scan_mask = TOP_MASK;)
885                         sort_function[0] = pid_sort;
886                         continue;
887                 }
888                 if (c == 'm') {
889                         IF_FEATURE_TOPMEM(scan_mask = TOP_MASK;)
890                         sort_function[0] = mem_sort;
891 # if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
892                         sort_function[1] = pcpu_sort;
893                         sort_function[2] = time_sort;
894 # endif
895                         continue;
896                 }
897 # if ENABLE_FEATURE_SHOW_THREADS
898                 if (c == 'h'
899                  IF_FEATURE_TOPMEM(&& scan_mask != TOPMEM_MASK)
900                 ) {
901                         scan_mask ^= PSSCAN_TASKS;
902                         continue;
903                 }
904 # endif
905 # if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
906                 if (c == 'p') {
907                         IF_FEATURE_TOPMEM(scan_mask = TOP_MASK;)
908                         sort_function[0] = pcpu_sort;
909                         sort_function[1] = mem_sort;
910                         sort_function[2] = time_sort;
911                         continue;
912                 }
913                 if (c == 't') {
914                         IF_FEATURE_TOPMEM(scan_mask = TOP_MASK;)
915                         sort_function[0] = time_sort;
916                         sort_function[1] = mem_sort;
917                         sort_function[2] = pcpu_sort;
918                         continue;
919                 }
920 #  if ENABLE_FEATURE_TOPMEM
921                 if (c == 's') {
922                         scan_mask = TOPMEM_MASK;
923                         free(prev_hist);
924                         prev_hist = NULL;
925                         prev_hist_count = 0;
926                         sort_field = (sort_field + 1) % NUM_SORT_FIELD;
927                         continue;
928                 }
929 #  endif
930                 if (c == 'r') {
931                         inverted ^= 1;
932                         continue;
933                 }
934 #  if ENABLE_FEATURE_TOP_SMP_CPU
935                 /* procps-2.0.18 uses 'C', 3.2.7 uses '1' */
936                 if (c == 'c' || c == '1') {
937                         /* User wants to toggle per cpu <> aggregate */
938                         if (smp_cpu_info) {
939                                 free(cpu_prev_jif);
940                                 free(cpu_jif);
941                                 cpu_jif = &cur_jif;
942                                 cpu_prev_jif = &prev_jif;
943                         } else {
944                                 /* Prepare for xrealloc() */
945                                 cpu_jif = cpu_prev_jif = NULL;
946                         }
947                         num_cpus = 0;
948                         smp_cpu_info = !smp_cpu_info;
949                         get_jiffy_counts();
950                         continue;
951                 }
952 #  endif
953 # endif
954                 break; /* unknown key -> force refresh */
955         }
956
957         return scan_mask;
958 }
959 #endif
960
961 //usage:#if ENABLE_FEATURE_SHOW_THREADS || ENABLE_FEATURE_TOP_SMP_CPU
962 //usage:# define IF_SHOW_THREADS_OR_TOP_SMP(...) __VA_ARGS__
963 //usage:#else
964 //usage:# define IF_SHOW_THREADS_OR_TOP_SMP(...)
965 //usage:#endif
966 //usage:#define top_trivial_usage
967 //usage:       "[-b] [-nCOUNT] [-dSECONDS]" IF_FEATURE_TOPMEM(" [-m]")
968 //usage:#define top_full_usage "\n\n"
969 //usage:       "Provide a view of process activity in real time."
970 //usage:   "\n""Read the status of all processes from /proc each SECONDS"
971 //usage:   "\n""and display a screenful of them."
972 //usage:   "\n""Keys:"
973 //usage:   "\n""        N/M"
974 //usage:                IF_FEATURE_TOP_CPU_USAGE_PERCENTAGE("/P")
975 //usage:                IF_FEATURE_TOP_CPU_USAGE_PERCENTAGE("/T")
976 //usage:           ": " IF_FEATURE_TOPMEM("show CPU usage, ") "sort by pid/mem"
977 //usage:                IF_FEATURE_TOP_CPU_USAGE_PERCENTAGE("/cpu")
978 //usage:                IF_FEATURE_TOP_CPU_USAGE_PERCENTAGE("/time")
979 //usage:        IF_FEATURE_TOPMEM(
980 //usage:   "\n""        S: show memory"
981 //usage:        )
982 //usage:   "\n""        R: reverse sort"
983 //usage:        IF_SHOW_THREADS_OR_TOP_SMP(
984 //usage:   "\n""        "
985 //usage:                IF_FEATURE_SHOW_THREADS("H: toggle threads")
986 //usage:                IF_FEATURE_SHOW_THREADS(IF_FEATURE_TOP_SMP_CPU(", "))
987 //usage:                IF_FEATURE_TOP_SMP_CPU("1: toggle SMP")
988 //usage:        )
989 //usage:   "\n""        Q,^C: exit"
990 //usage:   "\n"
991 //usage:   "\n""Options:"
992 //usage:   "\n""        -b      Batch mode"
993 //usage:   "\n""        -n N    Exit after N iterations"
994 //usage:   "\n""        -d N    Delay between updates"
995 //usage:        IF_FEATURE_TOPMEM(
996 //usage:   "\n""        -m      Same as 's' key"
997 //usage:        )
998
999 /* Interactive testing:
1000  * echo sss | ./busybox top
1001  * - shows memory screen
1002  * echo sss | ./busybox top -bn1 >mem
1003  * - saves memory screen - the *whole* list, not first NROWS processes!
1004  * echo .m.s.s.s.s.s.s.q | ./busybox top -b >z
1005  * - saves several different screens, and exits
1006  *
1007  * TODO: -i STRING param as a better alternative?
1008  */
1009
1010 int top_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
1011 int top_main(int argc UNUSED_PARAM, char **argv)
1012 {
1013         int iterations;
1014         unsigned lines, col;
1015         unsigned interval;
1016         char *str_interval, *str_iterations;
1017         unsigned scan_mask = TOP_MASK;
1018 #if ENABLE_FEATURE_USE_TERMIOS
1019         struct termios new_settings;
1020 #endif
1021
1022         INIT_G();
1023
1024         interval = 5; /* default update interval is 5 seconds */
1025         iterations = 0; /* infinite */
1026 #if ENABLE_FEATURE_TOP_SMP_CPU
1027         /*num_cpus = 0;*/
1028         /*smp_cpu_info = 0;*/  /* to start with show aggregate */
1029         cpu_jif = &cur_jif;
1030         cpu_prev_jif = &prev_jif;
1031 #endif
1032
1033         /* all args are options; -n NUM */
1034         opt_complementary = "-"; /* options can be specified w/o dash */
1035         col = getopt32(argv, "d:n:b"IF_FEATURE_TOPMEM("m"), &str_interval, &str_iterations);
1036 #if ENABLE_FEATURE_TOPMEM
1037         if (col & OPT_m) /* -m (busybox specific) */
1038                 scan_mask = TOPMEM_MASK;
1039 #endif
1040         if (col & OPT_d) {
1041                 /* work around for "-d 1" -> "-d -1" done by getopt32
1042                  * (opt_complementary == "-" does this) */
1043                 if (str_interval[0] == '-')
1044                         str_interval++;
1045                 /* Need to limit it to not overflow poll timeout */
1046                 interval = xatou16(str_interval);
1047         }
1048         if (col & OPT_n) {
1049                 if (str_iterations[0] == '-')
1050                         str_iterations++;
1051                 iterations = xatou(str_iterations);
1052         }
1053
1054         /* change to /proc */
1055         xchdir("/proc");
1056
1057 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
1058         sort_function[0] = pcpu_sort;
1059         sort_function[1] = mem_sort;
1060         sort_function[2] = time_sort;
1061 #else
1062         sort_function[0] = mem_sort;
1063 #endif
1064
1065 #if ENABLE_FEATURE_USE_TERMIOS
1066         tcgetattr(0, (void *) &initial_settings);
1067         memcpy(&new_settings, &initial_settings, sizeof(new_settings));
1068         if (!OPT_BATCH_MODE) {
1069                 /* unbuffered input, turn off echo */
1070                 new_settings.c_lflag &= ~(ISIG | ICANON | ECHO | ECHONL);
1071                 tcsetattr_stdin_TCSANOW(&new_settings);
1072         }
1073
1074         bb_signals(BB_FATAL_SIGS, sig_catcher);
1075
1076         /* Eat initial input, if any */
1077         scan_mask = handle_input(scan_mask, 0);
1078 #endif
1079
1080         while (scan_mask != EXIT_MASK) {
1081                 procps_status_t *p = NULL;
1082
1083                 if (OPT_BATCH_MODE) {
1084                         lines = INT_MAX;
1085                         col = LINE_BUF_SIZE - 2; /* +2 bytes for '\n', NUL */
1086                 } else {
1087                         lines = 24; /* default */
1088                         col = 79;
1089 #if ENABLE_FEATURE_USE_TERMIOS
1090                         /* We output to stdout, we need size of stdout (not stdin)! */
1091                         get_terminal_width_height(STDOUT_FILENO, &col, &lines);
1092                         if (lines < 5 || col < 10) {
1093                                 sleep(interval);
1094                                 continue;
1095                         }
1096 #endif
1097                         if (col > LINE_BUF_SIZE - 2)
1098                                 col = LINE_BUF_SIZE - 2;
1099                 }
1100
1101                 /* read process IDs & status for all the processes */
1102                 while ((p = procps_scan(p, scan_mask)) != NULL) {
1103                         int n;
1104 #if ENABLE_FEATURE_TOPMEM
1105                         if (scan_mask != TOPMEM_MASK)
1106 #endif
1107                         {
1108                                 n = ntop;
1109                                 top = xrealloc_vector(top, 6, ntop++);
1110                                 top[n].pid = p->pid;
1111                                 top[n].ppid = p->ppid;
1112                                 top[n].vsz = p->vsz;
1113 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
1114                                 top[n].ticks = p->stime + p->utime;
1115 #endif
1116                                 top[n].uid = p->uid;
1117                                 strcpy(top[n].state, p->state);
1118                                 strcpy(top[n].comm, p->comm);
1119 #if ENABLE_FEATURE_TOP_SMP_PROCESS
1120                                 top[n].last_seen_on_cpu = p->last_seen_on_cpu;
1121 #endif
1122                         }
1123 #if ENABLE_FEATURE_TOPMEM
1124                         else { /* TOPMEM */
1125                                 if (!(p->smaps.mapped_ro | p->smaps.mapped_rw))
1126                                         continue; /* kernel threads are ignored */
1127                                 n = ntop;
1128                                 /* No bug here - top and topmem are the same */
1129                                 top = xrealloc_vector(topmem, 6, ntop++);
1130                                 strcpy(topmem[n].comm, p->comm);
1131                                 topmem[n].pid      = p->pid;
1132                                 topmem[n].vsz      = p->smaps.mapped_rw + p->smaps.mapped_ro;
1133                                 topmem[n].vszrw    = p->smaps.mapped_rw;
1134                                 topmem[n].rss_sh   = p->smaps.shared_clean + p->smaps.shared_dirty;
1135                                 topmem[n].rss      = p->smaps.private_clean + p->smaps.private_dirty + topmem[n].rss_sh;
1136                                 topmem[n].dirty    = p->smaps.private_dirty + p->smaps.shared_dirty;
1137                                 topmem[n].dirty_sh = p->smaps.shared_dirty;
1138                                 topmem[n].stack    = p->smaps.stack;
1139                         }
1140 #endif
1141                 } /* end of "while we read /proc" */
1142                 if (ntop == 0) {
1143                         bb_error_msg("no process info in /proc");
1144                         break;
1145                 }
1146
1147                 if (scan_mask != TOPMEM_MASK) {
1148 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
1149                         if (!prev_hist_count) {
1150                                 do_stats();
1151                                 usleep(100000);
1152                                 clearmems();
1153                                 continue;
1154                         }
1155                         do_stats();
1156                         /* TODO: we don't need to sort all 10000 processes, we need to find top 24! */
1157                         qsort(top, ntop, sizeof(top_status_t), (void*)mult_lvl_cmp);
1158 #else
1159                         qsort(top, ntop, sizeof(top_status_t), (void*)(sort_function[0]));
1160 #endif
1161                 }
1162 #if ENABLE_FEATURE_TOPMEM
1163                 else { /* TOPMEM */
1164                         qsort(topmem, ntop, sizeof(topmem_status_t), (void*)topmem_sort);
1165                 }
1166 #endif
1167                 if (scan_mask != TOPMEM_MASK)
1168                         display_process_list(lines, col);
1169 #if ENABLE_FEATURE_TOPMEM
1170                 else
1171                         display_topmem_process_list(lines, col);
1172 #endif
1173                 clearmems();
1174                 if (iterations >= 0 && !--iterations)
1175                         break;
1176 #if !ENABLE_FEATURE_USE_TERMIOS
1177                 sleep(interval);
1178 #else
1179                 if (option_mask32 & OPT_EOF)
1180                         /* EOF on stdin ("top </dev/null") */
1181                         sleep(interval);
1182                 else
1183                         scan_mask = handle_input(scan_mask, interval);
1184 #endif /* FEATURE_USE_TERMIOS */
1185         } /* end of "while (not Q)" */
1186
1187         bb_putchar('\n');
1188 #if ENABLE_FEATURE_USE_TERMIOS
1189         reset_term();
1190 #endif
1191         return EXIT_SUCCESS;
1192 }