source: MondoRescue/trunk/mindi-busybox/procps/top.c@ 863

Last change on this file since 863 was 821, checked in by Bruno Cornec, 18 years ago

Addition of busybox 1.2.1 as a mindi-busybox new package
This should avoid delivering binary files in mindi not built there (Fedora and Debian are quite serious about that)
File size: 13.1 KB
Line 
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
21/* Original code Copyrights */
22/*
23 * Copyright (c) 1992 Branko Lankester
24 * Copyright (c) 1992 Roger Binns
25 * Copyright (C) 1994-1996 Charles L. Blake.
26 * Copyright (C) 1992-1998 Michael K. Johnson
27 * May be distributed under the conditions of the
28 * GNU Library General Public License
29 */
30
31#include "busybox.h"
32#include <sys/types.h>
33#include <stdio.h>
34#include <stdlib.h>
35#include <unistd.h>
36#include <string.h>
37#include <sys/ioctl.h>
38
39//#define CONFIG_FEATURE_TOP_CPU_USAGE_PERCENTAGE /* + 2k */
40
41#ifdef CONFIG_FEATURE_TOP_CPU_USAGE_PERCENTAGE
42#include <time.h>
43#include <fcntl.h>
44#include <netinet/in.h> /* htons */
45#endif
46
47
48typedef int (*cmp_t)(procps_status_t *P, procps_status_t *Q);
49
50static procps_status_t *top; /* Hehe */
51static int ntop;
52
53#ifdef CONFIG_FEATURE_USE_TERMIOS
54static int pid_sort(procps_status_t *P, procps_status_t *Q)
55{
56 return (Q->pid - P->pid);
57}
58#endif
59
60static int mem_sort(procps_status_t *P, procps_status_t *Q)
61{
62 return (int)(Q->rss - P->rss);
63}
64
65#ifdef CONFIG_FEATURE_TOP_CPU_USAGE_PERCENTAGE
66
67#define sort_depth 3
68static cmp_t sort_function[sort_depth];
69
70static int pcpu_sort(procps_status_t *P, procps_status_t *Q)
71{
72 return (Q->pcpu - P->pcpu);
73}
74
75static int time_sort(procps_status_t *P, procps_status_t *Q)
76{
77 return (int)((Q->stime + Q->utime) - (P->stime + P->utime));
78}
79
80static int mult_lvl_cmp(void* a, void* b) {
81 int i, cmp_val;
82
83 for (i = 0; i < sort_depth; i++) {
84 cmp_val = (*sort_function[i])(a, b);
85 if (cmp_val != 0)
86 return cmp_val;
87 }
88 return 0;
89}
90
91/* This structure stores some critical information from one frame to
92 the next. mostly used for sorting. Added cumulative and resident fields. */
93struct save_hist {
94 int ticks;
95 int pid;
96};
97
98/*
99 * Calculates percent cpu usage for each task.
100 */
101
102static struct save_hist *prev_hist;
103static int prev_hist_count;
104/* static int hist_iterations; */
105
106
107static unsigned total_pcpu;
108/* static unsigned long total_rss; */
109
110struct jiffy_counts {
111 unsigned long long usr,nic,sys,idle,iowait,irq,softirq,steal;
112 unsigned long long total;
113 unsigned long long busy;
114};
115static struct jiffy_counts jif, prev_jif;
116
117static void get_jiffy_counts(void)
118{
119 FILE* fp = bb_xfopen("stat", "r");
120 prev_jif = jif;
121 if (fscanf(fp, "cpu %lld %lld %lld %lld %lld %lld %lld %lld",
122 &jif.usr,&jif.nic,&jif.sys,&jif.idle,
123 &jif.iowait,&jif.irq,&jif.softirq,&jif.steal) < 4) {
124 bb_error_msg_and_die("failed to read 'stat'");
125 }
126 fclose(fp);
127 jif.total = jif.usr + jif.nic + jif.sys + jif.idle
128 + jif.iowait + jif.irq + jif.softirq + jif.steal;
129 /* procps 2.x does not count iowait as busy time */
130 jif.busy = jif.total - jif.idle - jif.iowait;
131}
132
133static void do_stats(void)
134{
135 procps_status_t *cur;
136 int pid, total_time, i, last_i, n;
137 struct save_hist *new_hist;
138
139 get_jiffy_counts();
140 total_pcpu = 0;
141 /* total_rss = 0; */
142 new_hist = xmalloc(sizeof(struct save_hist)*ntop);
143 /*
144 * Make a pass through the data to get stats.
145 */
146 /* hist_iterations = 0; */
147 i = 0;
148 for (n = 0; n < ntop; n++) {
149 cur = top + n;
150
151 /*
152 * Calculate time in cur process. Time is sum of user time
153 * and system time
154 */
155 pid = cur->pid;
156 total_time = cur->stime + cur->utime;
157 new_hist[n].ticks = total_time;
158 new_hist[n].pid = pid;
159
160 /* find matching entry from previous pass */
161 cur->pcpu = 0;
162 /* do not start at index 0, continue at last used one
163 * (brought hist_iterations from ~14000 down to 172) */
164 last_i = i;
165 if (prev_hist_count) do {
166 if (prev_hist[i].pid == pid) {
167 cur->pcpu = total_time - prev_hist[i].ticks;
168 break;
169 }
170 i = (i+1) % prev_hist_count;
171 /* hist_iterations++; */
172 } while (i != last_i);
173 total_pcpu += cur->pcpu;
174 /* total_rss += cur->rss; */
175 }
176
177 /*
178 * Save cur frame's information.
179 */
180 free(prev_hist);
181 prev_hist = new_hist;
182 prev_hist_count = ntop;
183}
184#else
185static cmp_t sort_function;
186#endif /* CONFIG_FEATURE_TOP_CPU_USAGE_PERCENTAGE */
187
188/* display generic info (meminfo / loadavg) */
189static unsigned long display_generic(int scr_width)
190{
191 FILE *fp;
192 char buf[80];
193 char scrbuf[80];
194 char *end;
195 unsigned long total, used, mfree, shared, buffers, cached;
196 unsigned int needs_conversion = 1;
197
198 /* read memory info */
199 fp = bb_xfopen("meminfo", "r");
200
201 /*
202 * Old kernels (such as 2.4.x) had a nice summary of memory info that
203 * we could parse, however this is gone entirely in 2.6. Try parsing
204 * the old way first, and if that fails, parse each field manually.
205 *
206 * First, we read in the first line. Old kernels will have bogus
207 * strings we don't care about, whereas new kernels will start right
208 * out with MemTotal:
209 * -- PFM.
210 */
211 if (fscanf(fp, "MemTotal: %lu %s\n", &total, buf) != 2) {
212 fgets(buf, sizeof(buf), fp); /* skip first line */
213
214 fscanf(fp, "Mem: %lu %lu %lu %lu %lu %lu",
215 &total, &used, &mfree, &shared, &buffers, &cached);
216 } else {
217 /*
218 * Revert to manual parsing, which incidentally already has the
219 * sizes in kilobytes. This should be safe for both 2.4 and
220 * 2.6.
221 */
222 needs_conversion = 0;
223
224 fscanf(fp, "MemFree: %lu %s\n", &mfree, buf);
225
226 /*
227 * MemShared: is no longer present in 2.6. Report this as 0,
228 * to maintain consistent behavior with normal procps.
229 */
230 if (fscanf(fp, "MemShared: %lu %s\n", &shared, buf) != 2)
231 shared = 0;
232
233 fscanf(fp, "Buffers: %lu %s\n", &buffers, buf);
234 fscanf(fp, "Cached: %lu %s\n", &cached, buf);
235
236 used = total - mfree;
237 }
238 fclose(fp);
239
240 /* read load average as a string */
241 fp = bb_xfopen("loadavg", "r");
242 buf[0] = '\0';
243 fgets(buf, sizeof(buf), fp);
244 end = strchr(buf, ' ');
245 if (end) end = strchr(end+1, ' ');
246 if (end) end = strchr(end+1, ' ');
247 if (end) *end = '\0';
248 fclose(fp);
249
250 if (needs_conversion) {
251 /* convert to kilobytes */
252 used /= 1024;
253 mfree /= 1024;
254 shared /= 1024;
255 buffers /= 1024;
256 cached /= 1024;
257 total /= 1024;
258 }
259
260 /* output memory info and load average */
261 /* clear screen & go to top */
262 if (scr_width > sizeof(scrbuf))
263 scr_width = sizeof(scrbuf);
264 snprintf(scrbuf, scr_width,
265 "Mem: %ldK used, %ldK free, %ldK shrd, %ldK buff, %ldK cached",
266 used, mfree, shared, buffers, cached);
267 printf("\e[H\e[J%s\n", scrbuf);
268 snprintf(scrbuf, scr_width,
269 "Load average: %s (Status: S=sleeping R=running, W=waiting)", buf);
270 printf("%s\n", scrbuf);
271
272 return total;
273}
274
275
276/* display process statuses */
277static void display_status(int count, int scr_width)
278{
279 enum {
280 bits_per_int = sizeof(int)*8
281 };
282
283 procps_status_t *s = top;
284 char rss_str_buf[8];
285 unsigned long total_memory = display_generic(scr_width); /* or use total_rss? */
286 unsigned pmem_shift, pmem_scale;
287
288#ifdef CONFIG_FEATURE_TOP_CPU_USAGE_PERCENTAGE
289 unsigned pcpu_shift, pcpu_scale;
290
291 /* what info of the processes is shown */
292 printf("\e[7m%.*s\e[0m", scr_width,
293 " PID USER STATUS RSS PPID %CPU %MEM COMMAND");
294#define MIN_WIDTH \
295 sizeof( " PID USER STATUS RSS PPID %CPU %MEM C")
296#else
297 printf("\e[7m%.*s\e[0m", scr_width,
298 " PID USER STATUS RSS PPID %MEM COMMAND");
299#define MIN_WIDTH \
300 sizeof( " PID USER STATUS RSS PPID %MEM C")
301#endif
302
303 /*
304 * MEM% = s->rss/MemTotal
305 */
306 pmem_shift = bits_per_int-11;
307 pmem_scale = 1000*(1U<<(bits_per_int-11)) / total_memory;
308 /* s->rss is in kb. we want (s->rss * pmem_scale) to never overflow */
309 while (pmem_scale >= 512) {
310 pmem_scale /= 4;
311 pmem_shift -= 2;
312 }
313#ifdef CONFIG_FEATURE_TOP_CPU_USAGE_PERCENTAGE
314 /*
315 * CPU% = s->pcpu/sum(s->pcpu) * busy_cpu_ticks/total_cpu_ticks
316 * (pcpu is delta of sys+user time between samples)
317 */
318 /* (jif.xxx - prev_jif.xxx) and s->pcpu are
319 * in 0..~64000 range (HZ*update_interval).
320 * we assume that unsigned is at least 32-bit.
321 */
322 pcpu_shift = 6;
323 pcpu_scale = (1000*64*(uint16_t)(jif.busy-prev_jif.busy) ? : 1);
324 while (pcpu_scale < (1U<<(bits_per_int-2))) {
325 pcpu_scale *= 4;
326 pcpu_shift += 2;
327 }
328 pcpu_scale /= ( (uint16_t)(jif.total-prev_jif.total)*total_pcpu ? : 1);
329 /* we want (s->pcpu * pcpu_scale) to never overflow */
330 while (pcpu_scale >= 1024) {
331 pcpu_scale /= 4;
332 pcpu_shift -= 2;
333 }
334 /* printf(" pmem_scale=%u pcpu_scale=%u ", pmem_scale, pcpu_scale); */
335#endif
336
337 while (count--) {
338 div_t pmem = div( (s->rss*pmem_scale) >> pmem_shift, 10);
339 int col = scr_width+1;
340 USE_FEATURE_TOP_CPU_USAGE_PERCENTAGE(div_t pcpu;)
341
342 if (s->rss >= 100*1024)
343 sprintf(rss_str_buf, "%6ldM", s->rss/1024);
344 else
345 sprintf(rss_str_buf, "%7ld", s->rss);
346 USE_FEATURE_TOP_CPU_USAGE_PERCENTAGE(pcpu = div((s->pcpu*pcpu_scale) >> pcpu_shift, 10);)
347 col -= printf("\n%5d %-8s %s %s%6d%3u.%c" \
348 USE_FEATURE_TOP_CPU_USAGE_PERCENTAGE("%3u.%c") " ",
349 s->pid, s->user, s->state, rss_str_buf, s->ppid,
350 USE_FEATURE_TOP_CPU_USAGE_PERCENTAGE(pcpu.quot, '0'+pcpu.rem,)
351 pmem.quot, '0'+pmem.rem);
352 if (col>0)
353 printf("%.*s", col, s->short_cmd);
354 /* printf(" %d/%d %lld/%lld", s->pcpu, total_pcpu,
355 jif.busy - prev_jif.busy, jif.total - prev_jif.total); */
356 s++;
357 }
358 /* printf(" %d", hist_iterations); */
359 putchar('\r');
360 fflush(stdout);
361}
362
363static void clearmems(void)
364{
365 free(top);
366 top = 0;
367 ntop = 0;
368}
369
370#ifdef CONFIG_FEATURE_USE_TERMIOS
371#include <termios.h>
372#include <signal.h>
373
374
375static struct termios initial_settings;
376
377static void reset_term(void)
378{
379 tcsetattr(0, TCSANOW, (void *) &initial_settings);
380#ifdef CONFIG_FEATURE_CLEAN_UP
381 clearmems();
382#ifdef CONFIG_FEATURE_TOP_CPU_USAGE_PERCENTAGE
383 free(prev_hist);
384#endif
385#endif /* CONFIG_FEATURE_CLEAN_UP */
386}
387
388static void sig_catcher(int sig ATTRIBUTE_UNUSED)
389{
390 reset_term();
391 exit(1);
392}
393#endif /* CONFIG_FEATURE_USE_TERMIOS */
394
395
396int top_main(int argc, char **argv)
397{
398 int opt, interval, lines, col;
399 char *sinterval;
400#ifdef CONFIG_FEATURE_USE_TERMIOS
401 struct termios new_settings;
402 struct timeval tv;
403 fd_set readfds;
404 unsigned char c;
405#endif /* CONFIG_FEATURE_USE_TERMIOS */
406
407 /* do normal option parsing */
408 opt = bb_getopt_ulflags(argc, argv, "d:", &sinterval);
409 if ((opt & 1)) {
410 interval = atoi(sinterval);
411 } else {
412 /* Default update rate is 5 seconds */
413 interval = 5;
414 }
415
416 /* change to /proc */
417 bb_xchdir("/proc");
418#ifdef CONFIG_FEATURE_USE_TERMIOS
419 tcgetattr(0, (void *) &initial_settings);
420 memcpy(&new_settings, &initial_settings, sizeof(struct termios));
421 new_settings.c_lflag &= ~(ISIG | ICANON); /* unbuffered input */
422 /* Turn off echoing */
423 new_settings.c_lflag &= ~(ECHO | ECHONL);
424
425 signal(SIGTERM, sig_catcher);
426 signal(SIGINT, sig_catcher);
427 tcsetattr(0, TCSANOW, (void *) &new_settings);
428 atexit(reset_term);
429#endif /* CONFIG_FEATURE_USE_TERMIOS */
430
431#ifdef CONFIG_FEATURE_TOP_CPU_USAGE_PERCENTAGE
432 sort_function[0] = pcpu_sort;
433 sort_function[1] = mem_sort;
434 sort_function[2] = time_sort;
435#else
436 sort_function = mem_sort;
437#endif /* CONFIG_FEATURE_TOP_CPU_USAGE_PERCENTAGE */
438
439 while (1) {
440 procps_status_t *p;
441
442 /* Default to 25 lines - 5 lines for status */
443 lines = 24 - 3;
444 col = 79;
445#ifdef CONFIG_FEATURE_USE_TERMIOS
446 get_terminal_width_height(0, &col, &lines);
447 if (lines < 5 || col < MIN_WIDTH) {
448 sleep(interval);
449 continue;
450 }
451 lines -= 3;
452#endif /* CONFIG_FEATURE_USE_TERMIOS */
453
454 /* read process IDs & status for all the processes */
455 while ((p = procps_scan(0)) != 0) {
456 int n = ntop;
457
458 top = xrealloc(top, (++ntop)*sizeof(procps_status_t));
459 memcpy(top + n, p, sizeof(procps_status_t));
460 }
461 if (ntop == 0) {
462 bb_error_msg_and_die("Can't find process info in /proc");
463 }
464#ifdef CONFIG_FEATURE_TOP_CPU_USAGE_PERCENTAGE
465 if (!prev_hist_count) {
466 do_stats();
467 sleep(1);
468 clearmems();
469 continue;
470 }
471 do_stats();
472 qsort(top, ntop, sizeof(procps_status_t), (void*)mult_lvl_cmp);
473#else
474 qsort(top, ntop, sizeof(procps_status_t), (void*)sort_function);
475#endif /* CONFIG_FEATURE_TOP_CPU_USAGE_PERCENTAGE */
476 opt = lines;
477 if (opt > ntop) {
478 opt = ntop;
479 }
480 /* show status for each of the processes */
481 display_status(opt, col);
482#ifdef CONFIG_FEATURE_USE_TERMIOS
483 tv.tv_sec = interval;
484 tv.tv_usec = 0;
485 FD_ZERO(&readfds);
486 FD_SET(0, &readfds);
487 select(1, &readfds, NULL, NULL, &tv);
488 if (FD_ISSET(0, &readfds)) {
489 if (read(0, &c, 1) <= 0) { /* signal */
490 return EXIT_FAILURE;
491 }
492 if (c == 'q' || c == initial_settings.c_cc[VINTR])
493 break;
494 if (c == 'M') {
495#ifdef CONFIG_FEATURE_TOP_CPU_USAGE_PERCENTAGE
496 sort_function[0] = mem_sort;
497 sort_function[1] = pcpu_sort;
498 sort_function[2] = time_sort;
499#else
500 sort_function = mem_sort;
501#endif
502 }
503#ifdef CONFIG_FEATURE_TOP_CPU_USAGE_PERCENTAGE
504 if (c == 'P') {
505 sort_function[0] = pcpu_sort;
506 sort_function[1] = mem_sort;
507 sort_function[2] = time_sort;
508 }
509 if (c == 'T') {
510 sort_function[0] = time_sort;
511 sort_function[1] = mem_sort;
512 sort_function[2] = pcpu_sort;
513 }
514#endif
515 if (c == 'N') {
516#ifdef CONFIG_FEATURE_TOP_CPU_USAGE_PERCENTAGE
517 sort_function[0] = pid_sort;
518#else
519 sort_function = pid_sort;
520#endif
521 }
522 }
523#else
524 sleep(interval);
525#endif /* CONFIG_FEATURE_USE_TERMIOS */
526 clearmems();
527 }
528 if (ENABLE_FEATURE_CLEAN_UP)
529 clearmems();
530 putchar('\n');
531 return EXIT_SUCCESS;
532}
Note: See TracBrowser for help on using the repository browser.