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
  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 "libbb.h"
  32
  33
  34 typedef struct top_status_t {
  35         unsigned long vsz;
  36 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
  37         unsigned long ticks;
  38         unsigned pcpu; /* delta of ticks */
  39 #endif
  40         unsigned pid, ppid;
  41         unsigned uid;
  42         char state[4];
  43         char comm[COMM_LEN];
  44 } top_status_t;
  45
  46 typedef struct jiffy_counts_t {
  47         unsigned long long usr,nic,sys,idle,iowait,irq,softirq,steal;
  48         unsigned long long total;
  49         unsigned long long busy;
  50 } jiffy_counts_t;
  51
  52 /* This structure stores some critical information from one frame to
  53    the next. Used for finding deltas. */
  54 typedef struct save_hist {
  55         unsigned long ticks;
  56         unsigned pid;
  57 } save_hist;
  58
  59 typedef int (*cmp_funcp)(top_status_t *P, top_status_t *Q);
  60
  61
  62 enum { SORT_DEPTH = 3 };
  63
  64
  65 struct globals {
  66         top_status_t *top;
  67         int ntop;
  68 #if ENABLE_FEATURE_TOPMEM
  69         smallint sort_field;
  70         smallint inverted;
  71 #endif
  72 #if ENABLE_FEATURE_USE_TERMIOS
  73         struct termios initial_settings;
  74 #endif
  75 #if !ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
  76         cmp_funcp sort_function[1];
  77 #else
  78         cmp_funcp sort_function[SORT_DEPTH];
  79         struct save_hist *prev_hist;
  80         int prev_hist_count;
  81         jiffy_counts_t jif, prev_jif;
  82         /* int hist_iterations; */
  83         unsigned total_pcpu;
  84         /* unsigned long total_vsz; */
  85 #endif
  86         char line_buf[80];
  87 };
  88
  89 enum { LINE_BUF_SIZE = COMMON_BUFSIZE - offsetof(struct globals, line_buf) };
  90
  91 #define G (*(struct globals*)&bb_common_bufsiz1)
  92 #define INIT_G() \
  93         do { \
  94                 struct G_sizecheck { \
  95                         char G_sizecheck[sizeof(G) > COMMON_BUFSIZE ? -1 : 1]; \
  96                 }; \
  97         } while (0)
  98 #define top              (G.top               )
  99 #define ntop             (G.ntop              )
 100 #define sort_field       (G.sort_field        )
 101 #define inverted         (G.inverted          )
 102 #define initial_settings (G.initial_settings  )
 103 #define sort_function    (G.sort_function     )
 104 #define prev_hist        (G.prev_hist         )
 105 #define prev_hist_count  (G.prev_hist_count   )
 106 #define jif              (G.jif               )
 107 #define prev_jif         (G.prev_jif          )
 108 #define total_pcpu       (G.total_pcpu        )
 109 #define line_buf         (G.line_buf          )
 110
 111
 112 #define OPT_BATCH_MODE (option_mask32 & 0x4)
 113
 114
 115 #if ENABLE_FEATURE_USE_TERMIOS
 116 static int pid_sort(top_status_t *P, top_status_t *Q)
 117 {
 118         /* Buggy wrt pids with high bit set */
 119         /* (linux pids are in [1..2^15-1]) */
 120         return (Q->pid - P->pid);
 121 }
 122 #endif
 123
 124 static int mem_sort(top_status_t *P, top_status_t *Q)
 125 {
 126         /* We want to avoid unsigned->signed and truncation errors */
 127         if (Q->vsz < P->vsz) return -1;
 128         return Q->vsz != P->vsz; /* 0 if ==, 1 if > */
 129 }
 130
 131
 132 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
 133
 134 static int pcpu_sort(top_status_t *P, top_status_t *Q)
 135 {
 136         /* Buggy wrt ticks with high bit set */
 137         /* Affects only processes for which ticks overflow */
 138         return (int)Q->pcpu - (int)P->pcpu;
 139 }
 140
 141 static int time_sort(top_status_t *P, top_status_t *Q)
 142 {
 143         /* We want to avoid unsigned->signed and truncation errors */
 144         if (Q->ticks < P->ticks) return -1;
 145         return Q->ticks != P->ticks; /* 0 if ==, 1 if > */
 146 }
 147
 148 static int mult_lvl_cmp(void* a, void* b)
 149 {
 150         int i, cmp_val;
 151
 152         for (i = 0; i < SORT_DEPTH; i++) {
 153                 cmp_val = (*sort_function[i])(a, b);
 154                 if (cmp_val != 0)
 155                         return cmp_val;
 156         }
 157         return 0;
 158 }
 159
 160
 161 static void get_jiffy_counts(void)
 162 {
 163         FILE* fp = xfopen("stat", "r");
 164         prev_jif = jif;
 165         if (fscanf(fp, "cpu  %lld %lld %lld %lld %lld %lld %lld %lld",
 166                         &jif.usr,&jif.nic,&jif.sys,&jif.idle,
 167                         &jif.iowait,&jif.irq,&jif.softirq,&jif.steal) < 4) {
 168                 bb_error_msg_and_die("failed to read /proc/stat");
 169         }
 170         fclose(fp);
 171         jif.total = jif.usr + jif.nic + jif.sys + jif.idle
 172                         + jif.iowait + jif.irq + jif.softirq + jif.steal;
 173         /* procps 2.x does not count iowait as busy time */
 174         jif.busy = jif.total - jif.idle - jif.iowait;
 175 }
 176
 177
 178 static void do_stats(void)
 179 {
 180         top_status_t *cur;
 181         pid_t pid;
 182         int i, last_i, n;
 183         struct save_hist *new_hist;
 184
 185         get_jiffy_counts();
 186         total_pcpu = 0;
 187         /* total_vsz = 0; */
 188         new_hist = xmalloc(sizeof(struct save_hist)*ntop);
 189         /*
 190          * Make a pass through the data to get stats.
 191          */
 192         /* hist_iterations = 0; */
 193         i = 0;
 194         for (n = 0; n < ntop; n++) {
 195                 cur = top + n;
 196
 197                 /*
 198                  * Calculate time in cur process.  Time is sum of user time
 199                  * and system time
 200                  */
 201                 pid = cur->pid;
 202                 new_hist[n].ticks = cur->ticks;
 203                 new_hist[n].pid = pid;
 204
 205                 /* find matching entry from previous pass */
 206                 cur->pcpu = 0;
 207                 /* do not start at index 0, continue at last used one
 208                  * (brought hist_iterations from ~14000 down to 172) */
 209                 last_i = i;
 210                 if (prev_hist_count) do {
 211                         if (prev_hist[i].pid == pid) {
 212                                 cur->pcpu = cur->ticks - prev_hist[i].ticks;
 213                                 total_pcpu += cur->pcpu;
 214                                 break;
 215                         }
 216                         i = (i+1) % prev_hist_count;
 217                         /* hist_iterations++; */
 218                 } while (i != last_i);
 219                 /* total_vsz += cur->vsz; */
 220         }
 221
 222         /*
 223          * Save cur frame's information.
 224          */
 225         free(prev_hist);
 226         prev_hist = new_hist;
 227         prev_hist_count = ntop;
 228 }
 229 #endif /* FEATURE_TOP_CPU_USAGE_PERCENTAGE */
 230
 231 #if ENABLE_FEATURE_TOP_CPU_GLOBAL_PERCENTS && ENABLE_FEATURE_TOP_DECIMALS
 232 /* formats 7 char string (8 with terminating NUL) */
 233 static char *fmt_100percent_8(char pbuf[8], unsigned value, unsigned total)
 234 {
 235         unsigned t;
 236         if (value >= total) { /* 100% ? */
 237                 strcpy(pbuf, "  100% ");
 238                 return pbuf;
 239         }
 240         /* else generate " [N/space]N.N% " string */
 241         value = 1000 * value / total;
 242         t = value / 100;
 243         value = value % 100;
 244         pbuf[0] = ' ';
 245         pbuf[1] = t ? t + '0' : ' ';
 246         pbuf[2] = '0' + (value / 10);
 247         pbuf[3] = '.';
 248         pbuf[4] = '0' + (value % 10);
 249         pbuf[5] = '%';
 250         pbuf[6] = ' ';
 251         pbuf[7] = '\0';
 252         return pbuf;
 253 }
 254 #endif
 255
 256 static unsigned long display_header(int scr_width)
 257 {
 258         FILE *fp;
 259         char buf[80];
 260         char scrbuf[80];
 261         unsigned long total, used, mfree, shared, buffers, cached;
 262 #if ENABLE_FEATURE_TOP_CPU_GLOBAL_PERCENTS
 263         unsigned total_diff;
 264 #endif
 265
 266         /* read memory info */
 267         fp = xfopen("meminfo", "r");
 268
 269         /*
 270          * Old kernels (such as 2.4.x) had a nice summary of memory info that
 271          * we could parse, however this is gone entirely in 2.6. Try parsing
 272          * the old way first, and if that fails, parse each field manually.
 273          *
 274          * First, we read in the first line. Old kernels will have bogus
 275          * strings we don't care about, whereas new kernels will start right
 276          * out with MemTotal:
 277          *                              -- PFM.
 278          */
 279         if (fscanf(fp, "MemTotal: %lu %s\n", &total, buf) != 2) {
 280                 fgets(buf, sizeof(buf), fp);    /* skip first line */
 281
 282                 fscanf(fp, "Mem: %lu %lu %lu %lu %lu %lu",
 283                         &total, &used, &mfree, &shared, &buffers, &cached);
 284                 /* convert to kilobytes */
 285                 used /= 1024;
 286                 mfree /= 1024;
 287                 shared /= 1024;
 288                 buffers /= 1024;
 289                 cached /= 1024;
 290                 total /= 1024;
 291         } else {
 292                 /*
 293                  * Revert to manual parsing, which incidentally already has the
 294                  * sizes in kilobytes. This should be safe for both 2.4 and
 295                  * 2.6.
 296                  */
 297
 298                 fscanf(fp, "MemFree: %lu %s\n", &mfree, buf);
 299
 300                 /*
 301                  * MemShared: is no longer present in 2.6. Report this as 0,
 302                  * to maintain consistent behavior with normal procps.
 303                  */
 304                 if (fscanf(fp, "MemShared: %lu %s\n", &shared, buf) != 2)
 305                         shared = 0;
 306
 307                 fscanf(fp, "Buffers: %lu %s\n", &buffers, buf);
 308                 fscanf(fp, "Cached: %lu %s\n", &cached, buf);
 309
 310                 used = total - mfree;
 311         }
 312         fclose(fp);
 313
 314         /* output memory info */
 315         if (scr_width > sizeof(scrbuf))
 316                 scr_width = sizeof(scrbuf);
 317         snprintf(scrbuf, scr_width,
 318                 "Mem: %luK used, %luK free, %luK shrd, %luK buff, %luK cached",
 319                 used, mfree, shared, buffers, cached);
 320         /* clear screen & go to top */
 321         printf(OPT_BATCH_MODE ? "%s\n" : "\e[H\e[J%s\n", scrbuf);
 322
 323 #if ENABLE_FEATURE_TOP_CPU_GLOBAL_PERCENTS
 324         /*
 325          * xxx% = (jif.xxx - prev_jif.xxx) / (jif.total - prev_jif.total) * 100%
 326          */
 327         /* using (unsigned) casts to make operations cheaper */
 328         total_diff = ((unsigned)(jif.total - prev_jif.total) ? : 1);
 329 #if ENABLE_FEATURE_TOP_DECIMALS
 330 /* Generated code is approx +0.3k */
 331 #define CALC_STAT(xxx) char xxx[8]
 332 #define SHOW_STAT(xxx) fmt_100percent_8(xxx, (unsigned)(jif.xxx - prev_jif.xxx), total_diff)
 333 #define FMT "%s"
 334 #else
 335 #define CALC_STAT(xxx) unsigned xxx = 100 * (unsigned)(jif.xxx - prev_jif.xxx) / total_diff
 336 #define SHOW_STAT(xxx) xxx
 337 #define FMT "%4u%% "
 338 #endif
 339         { /* need block: CALC_STAT are declarations */
 340                 CALC_STAT(usr);
 341                 CALC_STAT(sys);
 342                 CALC_STAT(nic);
 343                 CALC_STAT(idle);
 344                 CALC_STAT(iowait);
 345                 CALC_STAT(irq);
 346                 CALC_STAT(softirq);
 347                 //CALC_STAT(steal);
 348
 349                 snprintf(scrbuf, scr_width,
 350                         /* Barely fits in 79 chars when in "decimals" mode. */
 351                         "CPU:"FMT"usr"FMT"sys"FMT"nice"FMT"idle"FMT"io"FMT"irq"FMT"softirq",
 352                         SHOW_STAT(usr), SHOW_STAT(sys), SHOW_STAT(nic), SHOW_STAT(idle),
 353                         SHOW_STAT(iowait), SHOW_STAT(irq), SHOW_STAT(softirq)
 354                         //, SHOW_STAT(steal) - what is this 'steal' thing?
 355                         // I doubt anyone wants to know it
 356                 );
 357         }
 358         puts(scrbuf);
 359 #undef SHOW_STAT
 360 #undef CALC_STAT
 361 #undef FMT
 362 #endif
 363
 364         /* read load average as a string */
 365         buf[0] = '\0';
 366         open_read_close("loadavg", buf, sizeof("N.NN N.NN N.NN")-1);
 367         buf[sizeof("N.NN N.NN N.NN")-1] = '\0';
 368         snprintf(scrbuf, scr_width, "Load average: %s", buf);
 369         puts(scrbuf);
 370
 371         return total;
 372 }
 373
 374 static NOINLINE void display_process_list(int count, int scr_width)
 375 {
 376         enum {
 377                 BITS_PER_INT = sizeof(int)*8
 378         };
 379
 380         top_status_t *s = top;
 381         char vsz_str_buf[8];
 382         unsigned long total_memory = display_header(scr_width); /* or use total_vsz? */
 383         /* xxx_shift and xxx_scale variables allow us to replace
 384          * expensive divides with multiply and shift */
 385         unsigned pmem_shift, pmem_scale, pmem_half;
 386 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
 387         unsigned pcpu_shift, pcpu_scale, pcpu_half;
 388         unsigned busy_jifs;
 389
 390         /* what info of the processes is shown */
 391         printf(OPT_BATCH_MODE ? "%.*s" : "\e[7m%.*s\e[0m", scr_width,
 392                 "  PID  PPID USER     STAT   VSZ %MEM %CPU COMMAND");
 393 #else
 394
 395         /* !CPU_USAGE_PERCENTAGE */
 396         printf(OPT_BATCH_MODE ? "%.*s" : "\e[7m%.*s\e[0m", scr_width,
 397                 "  PID  PPID USER     STAT   VSZ %MEM COMMAND");
 398 #endif
 399
 400 #if ENABLE_FEATURE_TOP_DECIMALS
 401 #define UPSCALE 1000
 402 #define CALC_STAT(name, val) div_t name = div((val), 10)
 403 #define SHOW_STAT(name) name.quot, '0'+name.rem
 404 #define FMT "%3u.%c"
 405 #else
 406 #define UPSCALE 100
 407 #define CALC_STAT(name, val) unsigned name = (val)
 408 #define SHOW_STAT(name) name
 409 #define FMT "%4u%%"
 410 #endif
 411         /*
 412          * MEM% = s->vsz/MemTotal
 413          */
 414         pmem_shift = BITS_PER_INT-11;
 415         pmem_scale = UPSCALE*(1U<<(BITS_PER_INT-11)) / total_memory;
 416         /* s->vsz is in kb. we want (s->vsz * pmem_scale) to never overflow */
 417         while (pmem_scale >= 512) {
 418                 pmem_scale /= 4;
 419                 pmem_shift -= 2;
 420         }
 421         pmem_half = (1U << pmem_shift) / (ENABLE_FEATURE_TOP_DECIMALS? 20 : 2);
 422 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
 423         busy_jifs = jif.busy - prev_jif.busy;
 424         /* This happens if there were lots of short-lived processes
 425          * between two top updates (e.g. compilation) */
 426         if (total_pcpu < busy_jifs) total_pcpu = busy_jifs;
 427
 428         /*
 429          * CPU% = s->pcpu/sum(s->pcpu) * busy_cpu_ticks/total_cpu_ticks
 430          * (pcpu is delta of sys+user time between samples)
 431          */
 432         /* (jif.xxx - prev_jif.xxx) and s->pcpu are
 433          * in 0..~64000 range (HZ*update_interval).
 434          * we assume that unsigned is at least 32-bit.
 435          */
 436         pcpu_shift = 6;
 437         pcpu_scale = (UPSCALE*64*(uint16_t)busy_jifs ? : 1);
 438         while (pcpu_scale < (1U<<(BITS_PER_INT-2))) {
 439                 pcpu_scale *= 4;
 440                 pcpu_shift += 2;
 441         }
 442         pcpu_scale /= ( (uint16_t)(jif.total-prev_jif.total)*total_pcpu ? : 1);
 443         /* we want (s->pcpu * pcpu_scale) to never overflow */
 444         while (pcpu_scale >= 1024) {
 445                 pcpu_scale /= 4;
 446                 pcpu_shift -= 2;
 447         }
 448         pcpu_half = (1U << pcpu_shift) / (ENABLE_FEATURE_TOP_DECIMALS? 20 : 2);
 449         /* printf(" pmem_scale=%u pcpu_scale=%u ", pmem_scale, pcpu_scale); */
 450 #endif
 451
 452         scr_width += 2; /* account for leading '\n' and trailing NUL */
 453         /* Ok, all preliminary data is ready, go through the list */
 454         while (count-- > 0) {
 455                 unsigned col;
 456                 CALC_STAT(pmem, (s->vsz*pmem_scale + pmem_half) >> pmem_shift);
 457 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
 458                 CALC_STAT(pcpu, (s->pcpu*pcpu_scale + pcpu_half) >> pcpu_shift);
 459 #endif
 460
 461                 if (s->vsz >= 100000)
 462                         sprintf(vsz_str_buf, "%6ldm", s->vsz/1024);
 463                 else
 464                         sprintf(vsz_str_buf, "%7ld", s->vsz);
 465                 // PID PPID USER STAT VSZ %MEM [%CPU] COMMAND
 466                 col = snprintf(line_buf, scr_width,
 467                                 "\n" "%5u%6u %-8.8s %s%s" FMT
 468 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
 469                                 FMT
 470 #endif
 471                                 " ",
 472                                 s->pid, s->ppid, get_cached_username(s->uid),
 473                                 s->state, vsz_str_buf,
 474                                 SHOW_STAT(pmem)
 475 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
 476                                 , SHOW_STAT(pcpu)
 477 #endif
 478                 );
 479                 if (col + 1 < scr_width)
 480                         read_cmdline(line_buf + col, scr_width - col - 1, s->pid, s->comm);
 481                 fputs(line_buf, stdout);
 482                 /* printf(" %d/%d %lld/%lld", s->pcpu, total_pcpu,
 483                         jif.busy - prev_jif.busy, jif.total - prev_jif.total); */
 484                 s++;
 485         }
 486         /* printf(" %d", hist_iterations); */
 487         bb_putchar(OPT_BATCH_MODE ? '\n' : '\r');
 488         fflush(stdout);
 489 }
 490 #undef UPSCALE
 491 #undef SHOW_STAT
 492 #undef CALC_STAT
 493 #undef FMT
 494
 495 static void clearmems(void)
 496 {
 497         clear_username_cache();
 498         free(top);
 499         top = NULL;
 500         ntop = 0;
 501 }
 502
 503 #if ENABLE_FEATURE_USE_TERMIOS
 504 #include <termios.h>
 505 #include <signal.h>
 506
 507 static void reset_term(void)
 508 {
 509         tcsetattr(0, TCSANOW, (void *) &initial_settings);
 510         if (ENABLE_FEATURE_CLEAN_UP) {
 511                 clearmems();
 512 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
 513                 free(prev_hist);
 514 #endif
 515         }
 516 }
 517
 518 static void sig_catcher(int sig ATTRIBUTE_UNUSED)
 519 {
 520         reset_term();
 521         exit(1);
 522 }
 523 #endif /* FEATURE_USE_TERMIOS */
 524
 525 /*
 526  * TOPMEM support
 527  */
 528
 529 typedef unsigned long mem_t;
 530
 531 typedef struct topmem_status_t {
 532         unsigned pid;
 533         char comm[COMM_LEN];
 534         /* vsz doesn't count /dev/xxx mappings except /dev/zero */
 535         mem_t vsz     ;
 536         mem_t vszrw   ;
 537         mem_t rss     ;
 538         mem_t rss_sh  ;
 539         mem_t dirty   ;
 540         mem_t dirty_sh;
 541         mem_t stack   ;
 542 } topmem_status_t;
 543
 544 enum { NUM_SORT_FIELD = 7 };
 545
 546 #define topmem ((topmem_status_t*)top)
 547
 548 #if ENABLE_FEATURE_TOPMEM
 549 static int topmem_sort(char *a, char *b)
 550 {
 551         int n;
 552         mem_t l, r;
 553
 554         n = offsetof(topmem_status_t, vsz) + (sort_field * sizeof(mem_t));
 555         l = *(mem_t*)(a + n);
 556         r = *(mem_t*)(b + n);
 557 //      if (l == r) {
 558 //              l = a->mapped_rw;
 559 //              r = b->mapped_rw;
 560 //      }
 561         /* We want to avoid unsigned->signed and truncation errors */
 562         /* l>r: -1, l=r: 0, l<r: 1 */
 563         n = (l > r) ? -1 : (l != r);
 564         return inverted ? -n : n;
 565 }
 566
 567 /* Cut "NNNN " out of "    NNNN kb" */
 568 static char *grab_number(char *str, const char *match, unsigned sz)
 569 {
 570         if (strncmp(str, match, sz) == 0) {
 571                 str = skip_whitespace(str + sz);
 572                 (skip_non_whitespace(str))[1] = '\0';
 573                 return xstrdup(str);
 574         }
 575         return NULL;
 576 }
 577
 578 /* display header info (meminfo / loadavg) */
 579 static void display_topmem_header(int scr_width)
 580 {
 581         char linebuf[128];
 582         int i;
 583         FILE *fp;
 584         union {
 585                 struct {
 586                         /*  1 */ char *total;
 587                         /*  2 */ char *mfree;
 588                         /*  3 */ char *buf;
 589                         /*  4 */ char *cache;
 590                         /*  5 */ char *swaptotal;
 591                         /*  6 */ char *swapfree;
 592                         /*  7 */ char *dirty;
 593                         /*  8 */ char *mwrite;
 594                         /*  9 */ char *anon;
 595                         /* 10 */ char *map;
 596                         /* 11 */ char *slab;
 597                 } u;
 598                 char *str[11];
 599         } Z;
 600 #define total     Z.u.total
 601 #define mfree     Z.u.mfree
 602 #define buf       Z.u.buf
 603 #define cache     Z.u.cache
 604 #define swaptotal Z.u.swaptotal
 605 #define swapfree  Z.u.swapfree
 606 #define dirty     Z.u.dirty
 607 #define mwrite    Z.u.mwrite
 608 #define anon      Z.u.anon
 609 #define map       Z.u.map
 610 #define slab      Z.u.slab
 611 #define str       Z.str
 612
 613         memset(&Z, 0, sizeof(Z));
 614
 615         /* read memory info */
 616         fp = xfopen("meminfo", "r");
 617         while (fgets(linebuf, sizeof(linebuf), fp)) {
 618                 char *p;
 619
 620 #define SCAN(match, name) \
 621                 p = grab_number(linebuf, match, sizeof(match)-1); \
 622                 if (p) { name = p; continue; }
 623
 624                 SCAN("MemTotal:", total);
 625                 SCAN("MemFree:", mfree);
 626                 SCAN("Buffers:", buf);
 627                 SCAN("Cached:", cache);
 628                 SCAN("SwapTotal:", swaptotal);
 629                 SCAN("SwapFree:", swapfree);
 630                 SCAN("Dirty:", dirty);
 631                 SCAN("Writeback:", mwrite);
 632                 SCAN("AnonPages:", anon);
 633                 SCAN("Mapped:", map);
 634                 SCAN("Slab:", slab);
 635 #undef SCAN
 636         }
 637         fclose(fp);
 638
 639 #define S(s) (s ? s : "0 ")
 640         snprintf(linebuf, sizeof(linebuf),
 641                 "Mem %stotal %sanon %smap %sfree",
 642                 S(total), S(anon), S(map), S(mfree));
 643         printf(OPT_BATCH_MODE ? "%.*s\n" : "\e[H\e[J%.*s\n", scr_width, linebuf);
 644
 645         snprintf(linebuf, sizeof(linebuf),
 646                 " %sslab %sbuf %scache %sdirty %swrite",
 647                 S(slab), S(buf), S(cache), S(dirty), S(mwrite));
 648         printf("%.*s\n", scr_width, linebuf);
 649
 650         snprintf(linebuf, sizeof(linebuf),
 651                 "Swap %stotal %sfree", // TODO: % used?
 652                 S(swaptotal), S(swapfree));
 653         printf("%.*s\n", scr_width, linebuf);
 654 #undef S
 655
 656         for (i = 0; i < ARRAY_SIZE(str); i++)
 657                 free(str[i]);
 658 #undef total
 659 #undef free
 660 #undef buf
 661 #undef cache
 662 #undef swaptotal
 663 #undef swapfree
 664 #undef dirty
 665 #undef write
 666 #undef anon
 667 #undef map
 668 #undef slab
 669 #undef str
 670 }
 671
 672 static void ulltoa6_and_space(unsigned long long ul, char buf[6])
 673 {
 674         /* see http://en.wikipedia.org/wiki/Tera */
 675         smart_ulltoa5(ul, buf, " mgtpezy");
 676         buf[5] = ' ';
 677 }
 678
 679 static NOINLINE void display_topmem_process_list(int count, int scr_width)
 680 {
 681 #define HDR_STR "  PID   VSZ VSZRW   RSS (SHR) DIRTY (SHR) STACK"
 682 #define MIN_WIDTH sizeof(HDR_STR)
 683         const topmem_status_t *s = topmem;
 684
 685         display_topmem_header(scr_width);
 686         strcpy(line_buf, HDR_STR " COMMAND");
 687         line_buf[5 + sort_field * 6] = '*';
 688         printf(OPT_BATCH_MODE ? "%.*s" : "\e[7m%.*s\e[0m", scr_width, line_buf);
 689
 690         while (--count >= 0) {
 691                 // PID VSZ VSZRW RSS (SHR) DIRTY (SHR) COMMAND
 692                 ulltoa6_and_space(s->pid     , &line_buf[0*6]);
 693                 ulltoa6_and_space(s->vsz     , &line_buf[1*6]);
 694                 ulltoa6_and_space(s->vszrw   , &line_buf[2*6]);
 695                 ulltoa6_and_space(s->rss     , &line_buf[3*6]);
 696                 ulltoa6_and_space(s->rss_sh  , &line_buf[4*6]);
 697                 ulltoa6_and_space(s->dirty   , &line_buf[5*6]);
 698                 ulltoa6_and_space(s->dirty_sh, &line_buf[6*6]);
 699                 ulltoa6_and_space(s->stack   , &line_buf[7*6]);
 700                 line_buf[8*6] = '\0';
 701                 if (scr_width > MIN_WIDTH) {
 702                         read_cmdline(&line_buf[8*6], scr_width - MIN_WIDTH, s->pid, s->comm);
 703                 }
 704                 printf("\n""%.*s", scr_width, line_buf);
 705                 s++;
 706         }
 707         bb_putchar(OPT_BATCH_MODE ? '\n' : '\r');
 708         fflush(stdout);
 709 #undef HDR_STR
 710 #undef MIN_WIDTH
 711 }
 712 #else
 713 void display_topmem_process_list(int count, int scr_width);
 714 int topmem_sort(char *a, char *b);
 715 #endif /* TOPMEM */
 716
 717 /*
 718  * end TOPMEM support
 719  */
 720
 721 enum {
 722         TOP_MASK = 0
 723                 | PSSCAN_PID
 724                 | PSSCAN_PPID
 725                 | PSSCAN_VSZ
 726                 | PSSCAN_STIME
 727                 | PSSCAN_UTIME
 728                 | PSSCAN_STATE
 729                 | PSSCAN_COMM
 730                 | PSSCAN_UIDGID,
 731         TOPMEM_MASK = 0
 732                 | PSSCAN_PID
 733                 | PSSCAN_SMAPS
 734                 | PSSCAN_COMM,
 735 };
 736
 737 int top_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 738 int top_main(int argc, char **argv)
 739 {
 740         int count, lines, col;
 741         unsigned interval;
 742         int iterations;
 743         char *sinterval;
 744         SKIP_FEATURE_TOPMEM(const) unsigned scan_mask = TOP_MASK;
 745 #if ENABLE_FEATURE_USE_TERMIOS
 746         struct termios new_settings;
 747         struct pollfd pfd[1];
 748         unsigned char c;
 749
 750         pfd[0].fd = 0;
 751         pfd[0].events = POLLIN;
 752 #endif /* FEATURE_USE_TERMIOS */
 753
 754         INIT_G();
 755
 756         interval = 5; /* default update rate is 5 seconds */
 757         iterations = 0; /* infinite */
 758
 759         /* do normal option parsing */
 760         opt_complementary = "-:n+";
 761         getopt32(argv, "d:n:b", &sinterval, &iterations);
 762         if (option_mask32 & 0x1) {
 763                 /* Need to limit it to not overflow poll timeout */
 764                 interval = xatou16(sinterval); // -d
 765         }
 766
 767         /* change to /proc */
 768         xchdir("/proc");
 769 #if ENABLE_FEATURE_USE_TERMIOS
 770         tcgetattr(0, (void *) &initial_settings);
 771         memcpy(&new_settings, &initial_settings, sizeof(new_settings));
 772         /* unbuffered input, turn off echo */
 773         new_settings.c_lflag &= ~(ISIG | ICANON | ECHO | ECHONL);
 774
 775         bb_signals(0
 776                 + (1 << SIGTERM)
 777                 + (1 << SIGINT)
 778                 , sig_catcher);
 779         tcsetattr(0, TCSANOW, (void *) &new_settings);
 780         atexit(reset_term);
 781 #endif /* FEATURE_USE_TERMIOS */
 782
 783 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
 784         sort_function[0] = pcpu_sort;
 785         sort_function[1] = mem_sort;
 786         sort_function[2] = time_sort;
 787 #else
 788         sort_function[0] = mem_sort;
 789 #endif /* FEATURE_TOP_CPU_USAGE_PERCENTAGE */
 790
 791         while (1) {
 792                 procps_status_t *p = NULL;
 793
 794                 lines = 24; /* default */
 795                 col = 79;
 796 #if ENABLE_FEATURE_USE_TERMIOS
 797                 get_terminal_width_height(0, &col, &lines);
 798                 if (lines < 5 || col < 10) {
 799                         sleep(interval);
 800                         continue;
 801                 }
 802 #endif /* FEATURE_USE_TERMIOS */
 803                 if (col > LINE_BUF_SIZE-2) /* +2 bytes for '\n', NUL, */
 804                         col = LINE_BUF_SIZE-2;
 805                 if (!ENABLE_FEATURE_TOP_CPU_GLOBAL_PERCENTS && scan_mask == TOP_MASK)
 806                         lines -= 3;
 807                 else
 808                         lines -= 4;
 809
 810                 /* read process IDs & status for all the processes */
 811                 while ((p = procps_scan(p, scan_mask)) != NULL) {
 812                         int n;
 813                         if (scan_mask == TOP_MASK) {
 814                                 n = ntop;
 815                                 top = xrealloc(top, (++ntop) * sizeof(*top));
 816                                 top[n].pid = p->pid;
 817                                 top[n].ppid = p->ppid;
 818                                 top[n].vsz = p->vsz;
 819 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
 820                                 top[n].ticks = p->stime + p->utime;
 821 #endif
 822                                 top[n].uid = p->uid;
 823                                 strcpy(top[n].state, p->state);
 824                                 strcpy(top[n].comm, p->comm);
 825                         } else { /* TOPMEM */
 826 #if ENABLE_FEATURE_TOPMEM
 827                                 if (!(p->mapped_ro | p->mapped_rw))
 828                                         continue; /* kernel threads are ignored */
 829                                 n = ntop;
 830                                 top = xrealloc(topmem, (++ntop) * sizeof(*topmem));
 831                                 strcpy(topmem[n].comm, p->comm);
 832                                 topmem[n].pid      = p->pid;
 833                                 topmem[n].vsz      = p->mapped_rw + p->mapped_ro;
 834                                 topmem[n].vszrw    = p->mapped_rw;
 835                                 topmem[n].rss_sh   = p->shared_clean + p->shared_dirty;
 836                                 topmem[n].rss      = p->private_clean + p->private_dirty + topmem[n].rss_sh;
 837                                 topmem[n].dirty    = p->private_dirty + p->shared_dirty;
 838                                 topmem[n].dirty_sh = p->shared_dirty;
 839                                 topmem[n].stack    = p->stack;
 840 #endif
 841                         }
 842                 }
 843                 if (ntop == 0) {
 844                         bb_error_msg_and_die("no process info in /proc");
 845                 }
 846
 847                 if (scan_mask == TOP_MASK) {
 848 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
 849                         if (!prev_hist_count) {
 850                                 do_stats();
 851                                 usleep(100000);
 852                                 clearmems();
 853                                 continue;
 854                         }
 855                         do_stats();
 856 /* TODO: we don't need to sort all 10000 processes, we need to find top 24! */
 857                         qsort(top, ntop, sizeof(top_status_t), (void*)mult_lvl_cmp);
 858 #else
 859                         qsort(top, ntop, sizeof(top_status_t), (void*)(sort_function[0]));
 860 #endif /* FEATURE_TOP_CPU_USAGE_PERCENTAGE */
 861                 } else { /* TOPMEM */
 862                         qsort(topmem, ntop, sizeof(topmem_status_t), (void*)topmem_sort);
 863                 }
 864                 count = lines;
 865                 if (OPT_BATCH_MODE || count > ntop) {
 866                         count = ntop;
 867                 }
 868                 if (scan_mask == TOP_MASK)
 869                         display_process_list(count, col);
 870                 else
 871                         display_topmem_process_list(count, col);
 872                 clearmems();
 873                 if (iterations >= 0 && !--iterations)
 874                         break;
 875 #if !ENABLE_FEATURE_USE_TERMIOS
 876                 sleep(interval);
 877 #else
 878                 if (safe_poll(pfd, 1, interval * 1000) > 0) {
 879                         if (read(0, &c, 1) != 1)    /* signal */
 880                                 break;
 881                         if (c == initial_settings.c_cc[VINTR])
 882                                 break;
 883                         c |= 0x20; /* lowercase */
 884                         if (c == 'q')
 885                                 break;
 886                         if (c == 'n') {
 887                                 USE_FEATURE_TOPMEM(scan_mask = TOP_MASK;)
 888                                 sort_function[0] = pid_sort;
 889                         }
 890                         if (c == 'm') {
 891                                 USE_FEATURE_TOPMEM(scan_mask = TOP_MASK;)
 892                                 sort_function[0] = mem_sort;
 893 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
 894                                 sort_function[1] = pcpu_sort;
 895                                 sort_function[2] = time_sort;
 896 #endif
 897                         }
 898 #if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
 899                         if (c == 'p') {
 900                                 USE_FEATURE_TOPMEM(scan_mask = TOP_MASK;)
 901                                 sort_function[0] = pcpu_sort;
 902                                 sort_function[1] = mem_sort;
 903                                 sort_function[2] = time_sort;
 904                         }
 905                         if (c == 't') {
 906                                 USE_FEATURE_TOPMEM(scan_mask = TOP_MASK;)
 907                                 sort_function[0] = time_sort;
 908                                 sort_function[1] = mem_sort;
 909                                 sort_function[2] = pcpu_sort;
 910                         }
 911 #if ENABLE_FEATURE_TOPMEM
 912                         if (c == 's') {
 913                                 scan_mask = TOPMEM_MASK;
 914                                 free(prev_hist);
 915                                 prev_hist = NULL;
 916                                 prev_hist_count = 0;
 917                                 sort_field = (sort_field + 1) % NUM_SORT_FIELD;
 918                         }
 919                         if (c == 'r')
 920                                 inverted ^= 1;
 921 #endif
 922 #endif
 923                 }
 924 #endif /* FEATURE_USE_TERMIOS */
 925         }
 926         bb_putchar('\n');
 927         return EXIT_SUCCESS;
 928 }