mirror of
https://github.com/sheumann/hush.git
synced 2024-08-16 09:29:16 +00:00
Use it where appropriate. Stop scanning /etc/passwd *for every process*!!! (uid->username) top: reduce memory usage - we won't save unneeded fields from /proc info anymore. Downside: ~+250 bytes of code
567 lines
14 KiB
C
567 lines
14 KiB
C
/* vi: set sw=4 ts=4: */
|
|
/*
|
|
* A tiny 'top' utility.
|
|
*
|
|
* This is written specifically for the linux /proc/<PID>/stat(m)
|
|
* files format.
|
|
|
|
* This reads the PIDs of all processes and their status and shows
|
|
* the status of processes (first ones that fit to screen) at given
|
|
* intervals.
|
|
*
|
|
* NOTES:
|
|
* - At startup this changes to /proc, all the reads are then
|
|
* relative to that.
|
|
*
|
|
* (C) Eero Tamminen <oak at welho dot com>
|
|
*
|
|
* Rewritten by Vladimir Oleynik (C) 2002 <dzo@simtreas.ru>
|
|
*/
|
|
|
|
/* Original code Copyrights */
|
|
/*
|
|
* Copyright (c) 1992 Branko Lankester
|
|
* Copyright (c) 1992 Roger Binns
|
|
* Copyright (C) 1994-1996 Charles L. Blake.
|
|
* Copyright (C) 1992-1998 Michael K. Johnson
|
|
* May be distributed under the conditions of the
|
|
* GNU Library General Public License
|
|
*/
|
|
|
|
#include "busybox.h"
|
|
|
|
|
|
typedef struct {
|
|
unsigned long rss;
|
|
#if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
|
|
unsigned long ticks;
|
|
unsigned pcpu; /* delta of ticks */
|
|
#endif
|
|
unsigned pid, ppid;
|
|
unsigned uid;
|
|
char state[4];
|
|
char comm[COMM_LEN];
|
|
} top_status_t;
|
|
static top_status_t *top;
|
|
static int ntop;
|
|
/* This structure stores some critical information from one frame to
|
|
the next. Used for finding deltas. */
|
|
struct save_hist {
|
|
unsigned long ticks;
|
|
unsigned pid;
|
|
};
|
|
static struct save_hist *prev_hist;
|
|
static int prev_hist_count;
|
|
/* static int hist_iterations; */
|
|
static unsigned total_pcpu;
|
|
/* static unsigned long total_rss; */
|
|
|
|
|
|
#define OPT_BATCH_MODE (option_mask32 & 0x4)
|
|
|
|
#if ENABLE_FEATURE_USE_TERMIOS
|
|
static int pid_sort(top_status_t *P, top_status_t *Q)
|
|
{
|
|
/* Buggy wrt pids with high bit set */
|
|
/* (linux pids are in [1..2^15-1]) */
|
|
return (Q->pid - P->pid);
|
|
}
|
|
#endif
|
|
|
|
static int mem_sort(top_status_t *P, top_status_t *Q)
|
|
{
|
|
/* We want to avoid unsigned->signed and truncation errors */
|
|
if (Q->rss < P->rss) return -1;
|
|
return Q->rss != P->rss; /* 0 if ==, 1 if > */
|
|
}
|
|
|
|
|
|
typedef int (*cmp_funcp)(top_status_t *P, top_status_t *Q);
|
|
|
|
#if !ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
|
|
|
|
static cmp_funcp sort_function;
|
|
|
|
#else
|
|
|
|
enum { SORT_DEPTH = 3 };
|
|
|
|
static cmp_funcp sort_function[SORT_DEPTH];
|
|
|
|
static int pcpu_sort(top_status_t *P, top_status_t *Q)
|
|
{
|
|
/* Buggy wrt ticks with high bit set */
|
|
/* Affects only processes for which ticks overflow */
|
|
return (int)Q->pcpu - (int)P->pcpu;
|
|
}
|
|
|
|
static int time_sort(top_status_t *P, top_status_t *Q)
|
|
{
|
|
/* We want to avoid unsigned->signed and truncation errors */
|
|
if (Q->ticks < P->ticks) return -1;
|
|
return Q->ticks != P->ticks; /* 0 if ==, 1 if > */
|
|
}
|
|
|
|
static int mult_lvl_cmp(void* a, void* b) {
|
|
int i, cmp_val;
|
|
|
|
for (i = 0; i < SORT_DEPTH; i++) {
|
|
cmp_val = (*sort_function[i])(a, b);
|
|
if (cmp_val != 0)
|
|
return cmp_val;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
|
|
typedef struct {
|
|
unsigned long long usr,nic,sys,idle,iowait,irq,softirq,steal;
|
|
unsigned long long total;
|
|
unsigned long long busy;
|
|
} jiffy_counts_t;
|
|
static jiffy_counts_t jif, prev_jif;
|
|
static void get_jiffy_counts(void)
|
|
{
|
|
FILE* fp = xfopen("stat", "r");
|
|
prev_jif = jif;
|
|
if (fscanf(fp, "cpu %lld %lld %lld %lld %lld %lld %lld %lld",
|
|
&jif.usr,&jif.nic,&jif.sys,&jif.idle,
|
|
&jif.iowait,&jif.irq,&jif.softirq,&jif.steal) < 4) {
|
|
bb_error_msg_and_die("failed to read /proc/stat");
|
|
}
|
|
fclose(fp);
|
|
jif.total = jif.usr + jif.nic + jif.sys + jif.idle
|
|
+ jif.iowait + jif.irq + jif.softirq + jif.steal;
|
|
/* procps 2.x does not count iowait as busy time */
|
|
jif.busy = jif.total - jif.idle - jif.iowait;
|
|
}
|
|
|
|
|
|
static void do_stats(void)
|
|
{
|
|
top_status_t *cur;
|
|
pid_t pid;
|
|
int i, last_i, n;
|
|
struct save_hist *new_hist;
|
|
|
|
get_jiffy_counts();
|
|
total_pcpu = 0;
|
|
/* total_rss = 0; */
|
|
new_hist = xmalloc(sizeof(struct save_hist)*ntop);
|
|
/*
|
|
* Make a pass through the data to get stats.
|
|
*/
|
|
/* hist_iterations = 0; */
|
|
i = 0;
|
|
for (n = 0; n < ntop; n++) {
|
|
cur = top + n;
|
|
|
|
/*
|
|
* Calculate time in cur process. Time is sum of user time
|
|
* and system time
|
|
*/
|
|
pid = cur->pid;
|
|
new_hist[n].ticks = cur->ticks;
|
|
new_hist[n].pid = pid;
|
|
|
|
/* find matching entry from previous pass */
|
|
cur->pcpu = 0;
|
|
/* do not start at index 0, continue at last used one
|
|
* (brought hist_iterations from ~14000 down to 172) */
|
|
last_i = i;
|
|
if (prev_hist_count) do {
|
|
if (prev_hist[i].pid == pid) {
|
|
cur->pcpu = cur->ticks - prev_hist[i].ticks;
|
|
total_pcpu += cur->pcpu;
|
|
break;
|
|
}
|
|
i = (i+1) % prev_hist_count;
|
|
/* hist_iterations++; */
|
|
} while (i != last_i);
|
|
/* total_rss += cur->rss; */
|
|
}
|
|
|
|
/*
|
|
* Save cur frame's information.
|
|
*/
|
|
free(prev_hist);
|
|
prev_hist = new_hist;
|
|
prev_hist_count = ntop;
|
|
}
|
|
#endif /* FEATURE_TOP_CPU_USAGE_PERCENTAGE */
|
|
|
|
|
|
/* display generic info (meminfo / loadavg) */
|
|
static unsigned long display_generic(int scr_width)
|
|
{
|
|
FILE *fp;
|
|
char buf[80];
|
|
char scrbuf[80];
|
|
char *end;
|
|
unsigned long total, used, mfree, shared, buffers, cached;
|
|
unsigned int needs_conversion = 1;
|
|
|
|
/* read memory info */
|
|
fp = xfopen("meminfo", "r");
|
|
|
|
/*
|
|
* Old kernels (such as 2.4.x) had a nice summary of memory info that
|
|
* we could parse, however this is gone entirely in 2.6. Try parsing
|
|
* the old way first, and if that fails, parse each field manually.
|
|
*
|
|
* First, we read in the first line. Old kernels will have bogus
|
|
* strings we don't care about, whereas new kernels will start right
|
|
* out with MemTotal:
|
|
* -- PFM.
|
|
*/
|
|
if (fscanf(fp, "MemTotal: %lu %s\n", &total, buf) != 2) {
|
|
fgets(buf, sizeof(buf), fp); /* skip first line */
|
|
|
|
fscanf(fp, "Mem: %lu %lu %lu %lu %lu %lu",
|
|
&total, &used, &mfree, &shared, &buffers, &cached);
|
|
} else {
|
|
/*
|
|
* Revert to manual parsing, which incidentally already has the
|
|
* sizes in kilobytes. This should be safe for both 2.4 and
|
|
* 2.6.
|
|
*/
|
|
needs_conversion = 0;
|
|
|
|
fscanf(fp, "MemFree: %lu %s\n", &mfree, buf);
|
|
|
|
/*
|
|
* MemShared: is no longer present in 2.6. Report this as 0,
|
|
* to maintain consistent behavior with normal procps.
|
|
*/
|
|
if (fscanf(fp, "MemShared: %lu %s\n", &shared, buf) != 2)
|
|
shared = 0;
|
|
|
|
fscanf(fp, "Buffers: %lu %s\n", &buffers, buf);
|
|
fscanf(fp, "Cached: %lu %s\n", &cached, buf);
|
|
|
|
used = total - mfree;
|
|
}
|
|
fclose(fp);
|
|
|
|
/* read load average as a string */
|
|
buf[0] = '\0';
|
|
open_read_close("loadavg", buf, sizeof(buf));
|
|
end = strchr(buf, ' ');
|
|
if (end) end = strchr(end+1, ' ');
|
|
if (end) end = strchr(end+1, ' ');
|
|
if (end) *end = '\0';
|
|
|
|
if (needs_conversion) {
|
|
/* convert to kilobytes */
|
|
used /= 1024;
|
|
mfree /= 1024;
|
|
shared /= 1024;
|
|
buffers /= 1024;
|
|
cached /= 1024;
|
|
total /= 1024;
|
|
}
|
|
|
|
/* output memory info and load average */
|
|
/* clear screen & go to top */
|
|
if (scr_width > sizeof(scrbuf))
|
|
scr_width = sizeof(scrbuf);
|
|
snprintf(scrbuf, scr_width,
|
|
"Mem: %ldK used, %ldK free, %ldK shrd, %ldK buff, %ldK cached",
|
|
used, mfree, shared, buffers, cached);
|
|
|
|
printf(OPT_BATCH_MODE ? "%s\n" : "\e[H\e[J%s\n", scrbuf);
|
|
|
|
snprintf(scrbuf, scr_width, "Load average: %s", buf);
|
|
printf("%s\n", scrbuf);
|
|
|
|
return total;
|
|
}
|
|
|
|
|
|
/* display process statuses */
|
|
static void display_status(int count, int scr_width)
|
|
{
|
|
enum {
|
|
bits_per_int = sizeof(int)*8
|
|
};
|
|
|
|
top_status_t *s = top;
|
|
char rss_str_buf[8];
|
|
unsigned long total_memory = display_generic(scr_width); /* or use total_rss? */
|
|
unsigned pmem_shift, pmem_scale;
|
|
|
|
#if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
|
|
unsigned pcpu_shift, pcpu_scale;
|
|
unsigned busy_jifs;
|
|
|
|
/* what info of the processes is shown */
|
|
printf(OPT_BATCH_MODE ? "%.*s" : "\e[7m%.*s\e[0m", scr_width,
|
|
" PID USER STATUS RSS PPID %CPU %MEM COMMAND");
|
|
#define MIN_WIDTH \
|
|
sizeof( " PID USER STATUS RSS PPID %CPU %MEM C")
|
|
#else
|
|
printf(OPT_BATCH_MODE ? "%.*s" : "\e[7m%.*s\e[0m", scr_width,
|
|
" PID USER STATUS RSS PPID %MEM COMMAND");
|
|
#define MIN_WIDTH \
|
|
sizeof( " PID USER STATUS RSS PPID %MEM C")
|
|
#endif
|
|
|
|
/*
|
|
* MEM% = s->rss/MemTotal
|
|
*/
|
|
pmem_shift = bits_per_int-11;
|
|
pmem_scale = 1000*(1U<<(bits_per_int-11)) / total_memory;
|
|
/* s->rss is in kb. we want (s->rss * pmem_scale) to never overflow */
|
|
while (pmem_scale >= 512) {
|
|
pmem_scale /= 4;
|
|
pmem_shift -= 2;
|
|
}
|
|
#if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
|
|
busy_jifs = jif.busy - prev_jif.busy;
|
|
/* This happens if there were lots of short-lived processes
|
|
* between two top updates (e.g. compilation) */
|
|
if (total_pcpu < busy_jifs) total_pcpu = busy_jifs;
|
|
|
|
/*
|
|
* CPU% = s->pcpu/sum(s->pcpu) * busy_cpu_ticks/total_cpu_ticks
|
|
* (pcpu is delta of sys+user time between samples)
|
|
*/
|
|
/* (jif.xxx - prev_jif.xxx) and s->pcpu are
|
|
* in 0..~64000 range (HZ*update_interval).
|
|
* we assume that unsigned is at least 32-bit.
|
|
*/
|
|
pcpu_shift = 6;
|
|
pcpu_scale = (1000*64*(uint16_t)busy_jifs ? : 1);
|
|
while (pcpu_scale < (1U<<(bits_per_int-2))) {
|
|
pcpu_scale *= 4;
|
|
pcpu_shift += 2;
|
|
}
|
|
pcpu_scale /= ( (uint16_t)(jif.total-prev_jif.total)*total_pcpu ? : 1);
|
|
/* we want (s->pcpu * pcpu_scale) to never overflow */
|
|
while (pcpu_scale >= 1024) {
|
|
pcpu_scale /= 4;
|
|
pcpu_shift -= 2;
|
|
}
|
|
/* printf(" pmem_scale=%u pcpu_scale=%u ", pmem_scale, pcpu_scale); */
|
|
#endif
|
|
while (count-- > 0) {
|
|
div_t pmem = div((s->rss*pmem_scale) >> pmem_shift, 10);
|
|
int col = scr_width+1;
|
|
USE_FEATURE_TOP_CPU_USAGE_PERCENTAGE(div_t pcpu;)
|
|
|
|
if (s->rss >= 100*1024)
|
|
sprintf(rss_str_buf, "%6ldM", s->rss/1024);
|
|
else
|
|
sprintf(rss_str_buf, "%7ld", s->rss);
|
|
USE_FEATURE_TOP_CPU_USAGE_PERCENTAGE(
|
|
pcpu = div((s->pcpu*pcpu_scale) >> pcpu_shift, 10);
|
|
)
|
|
col -= printf("\n%5u %-8s %s "
|
|
"%s%6u"
|
|
USE_FEATURE_TOP_CPU_USAGE_PERCENTAGE("%3u.%c")
|
|
"%3u.%c ",
|
|
s->pid, get_cached_username(s->uid), s->state,
|
|
rss_str_buf, s->ppid,
|
|
USE_FEATURE_TOP_CPU_USAGE_PERCENTAGE(pcpu.quot, '0'+pcpu.rem,)
|
|
pmem.quot, '0'+pmem.rem);
|
|
if (col > 0)
|
|
printf("%.*s", col, s->comm);
|
|
/* printf(" %d/%d %lld/%lld", s->pcpu, total_pcpu,
|
|
jif.busy - prev_jif.busy, jif.total - prev_jif.total); */
|
|
s++;
|
|
}
|
|
/* printf(" %d", hist_iterations); */
|
|
putchar(OPT_BATCH_MODE ? '\n' : '\r');
|
|
fflush(stdout);
|
|
}
|
|
|
|
|
|
static void clearmems(void)
|
|
{
|
|
clear_username_cache();
|
|
free(top);
|
|
top = 0;
|
|
ntop = 0;
|
|
}
|
|
|
|
|
|
#if ENABLE_FEATURE_USE_TERMIOS
|
|
#include <termios.h>
|
|
#include <signal.h>
|
|
|
|
static struct termios initial_settings;
|
|
|
|
static void reset_term(void)
|
|
{
|
|
tcsetattr(0, TCSANOW, (void *) &initial_settings);
|
|
#if ENABLE_FEATURE_CLEAN_UP
|
|
clearmems();
|
|
#if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
|
|
free(prev_hist);
|
|
#endif
|
|
#endif /* FEATURE_CLEAN_UP */
|
|
}
|
|
|
|
static void sig_catcher(int sig ATTRIBUTE_UNUSED)
|
|
{
|
|
reset_term();
|
|
exit(1);
|
|
}
|
|
#endif /* FEATURE_USE_TERMIOS */
|
|
|
|
|
|
int top_main(int argc, char **argv)
|
|
{
|
|
int count, lines, col;
|
|
unsigned interval = 5; /* default update rate is 5 seconds */
|
|
unsigned iterations = UINT_MAX; /* 2^32 iterations by default :) */
|
|
char *sinterval, *siterations;
|
|
#if ENABLE_FEATURE_USE_TERMIOS
|
|
struct termios new_settings;
|
|
struct timeval tv;
|
|
fd_set readfds;
|
|
unsigned char c;
|
|
#endif /* FEATURE_USE_TERMIOS */
|
|
|
|
/* do normal option parsing */
|
|
interval = 5;
|
|
opt_complementary = "-";
|
|
getopt32(argc, argv, "d:n:b", &sinterval, &siterations);
|
|
if (option_mask32 & 0x1) interval = xatou(sinterval); // -d
|
|
if (option_mask32 & 0x2) iterations = xatou(siterations); // -n
|
|
//if (option_mask32 & 0x4) // -b
|
|
|
|
/* change to /proc */
|
|
xchdir("/proc");
|
|
#if ENABLE_FEATURE_USE_TERMIOS
|
|
tcgetattr(0, (void *) &initial_settings);
|
|
memcpy(&new_settings, &initial_settings, sizeof(struct termios));
|
|
/* unbuffered input, turn off echo */
|
|
new_settings.c_lflag &= ~(ISIG | ICANON | ECHO | ECHONL);
|
|
|
|
signal(SIGTERM, sig_catcher);
|
|
signal(SIGINT, sig_catcher);
|
|
tcsetattr(0, TCSANOW, (void *) &new_settings);
|
|
atexit(reset_term);
|
|
#endif /* FEATURE_USE_TERMIOS */
|
|
|
|
#if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
|
|
sort_function[0] = pcpu_sort;
|
|
sort_function[1] = mem_sort;
|
|
sort_function[2] = time_sort;
|
|
#else
|
|
sort_function = mem_sort;
|
|
#endif /* FEATURE_TOP_CPU_USAGE_PERCENTAGE */
|
|
|
|
while (1) {
|
|
procps_status_t *p = NULL;
|
|
|
|
/* Default to 25 lines - 5 lines for status */
|
|
lines = 24 - 3;
|
|
col = 79;
|
|
#if ENABLE_FEATURE_USE_TERMIOS
|
|
get_terminal_width_height(0, &col, &lines);
|
|
if (lines < 5 || col < MIN_WIDTH) {
|
|
sleep(interval);
|
|
continue;
|
|
}
|
|
lines -= 3;
|
|
#endif /* FEATURE_USE_TERMIOS */
|
|
|
|
/* read process IDs & status for all the processes */
|
|
while ((p = procps_scan(p, 0
|
|
| PSSCAN_PID
|
|
| PSSCAN_PPID
|
|
| PSSCAN_RSS
|
|
| PSSCAN_STIME
|
|
| PSSCAN_UTIME
|
|
| PSSCAN_STATE
|
|
| PSSCAN_COMM
|
|
| PSSCAN_SID
|
|
| PSSCAN_UIDGID
|
|
))) {
|
|
int n = ntop;
|
|
top = xrealloc(top, (++ntop)*sizeof(top_status_t));
|
|
top[n].pid = p->pid;
|
|
top[n].ppid = p->ppid;
|
|
top[n].rss = p->rss;
|
|
top[n].ticks = p->stime + p->utime;
|
|
top[n].uid = p->uid;
|
|
strcpy(top[n].state, p->state);
|
|
strcpy(top[n].comm, p->comm);
|
|
}
|
|
if (ntop == 0) {
|
|
bb_error_msg_and_die("can't find process info in /proc");
|
|
}
|
|
#if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
|
|
if (!prev_hist_count) {
|
|
do_stats();
|
|
sleep(1);
|
|
clearmems();
|
|
continue;
|
|
}
|
|
do_stats();
|
|
qsort(top, ntop, sizeof(top_status_t), (void*)mult_lvl_cmp);
|
|
#else
|
|
qsort(top, ntop, sizeof(top_status_t), (void*)sort_function);
|
|
#endif /* FEATURE_TOP_CPU_USAGE_PERCENTAGE */
|
|
count = lines;
|
|
if (OPT_BATCH_MODE || count > ntop) {
|
|
count = ntop;
|
|
}
|
|
/* show status for each of the processes */
|
|
display_status(count, col);
|
|
#if ENABLE_FEATURE_USE_TERMIOS
|
|
tv.tv_sec = interval;
|
|
tv.tv_usec = 0;
|
|
FD_ZERO(&readfds);
|
|
FD_SET(0, &readfds);
|
|
select(1, &readfds, NULL, NULL, &tv);
|
|
if (FD_ISSET(0, &readfds)) {
|
|
if (read(0, &c, 1) <= 0) { /* signal */
|
|
return EXIT_FAILURE;
|
|
}
|
|
if (c == 'q' || c == initial_settings.c_cc[VINTR])
|
|
break;
|
|
if (c == 'M') {
|
|
#if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
|
|
sort_function[0] = mem_sort;
|
|
sort_function[1] = pcpu_sort;
|
|
sort_function[2] = time_sort;
|
|
#else
|
|
sort_function = mem_sort;
|
|
#endif
|
|
}
|
|
#if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
|
|
if (c == 'P') {
|
|
sort_function[0] = pcpu_sort;
|
|
sort_function[1] = mem_sort;
|
|
sort_function[2] = time_sort;
|
|
}
|
|
if (c == 'T') {
|
|
sort_function[0] = time_sort;
|
|
sort_function[1] = mem_sort;
|
|
sort_function[2] = pcpu_sort;
|
|
}
|
|
#endif
|
|
if (c == 'N') {
|
|
#if ENABLE_FEATURE_TOP_CPU_USAGE_PERCENTAGE
|
|
sort_function[0] = pid_sort;
|
|
#else
|
|
sort_function = pid_sort;
|
|
#endif
|
|
}
|
|
}
|
|
if (!--iterations)
|
|
break;
|
|
#else
|
|
sleep(interval);
|
|
#endif /* FEATURE_USE_TERMIOS */
|
|
clearmems();
|
|
}
|
|
if (ENABLE_FEATURE_CLEAN_UP)
|
|
clearmems();
|
|
putchar('\n');
|
|
return EXIT_SUCCESS;
|
|
}
|