Magellan Linux

Contents of /tags/mkinitrd-6_1_5/busybox/procps/top.c

Parent Directory Parent Directory | Revision Log Revision Log


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