1 /* vi: set sw=4 ts=4: */
5 * Copyright 1998 by Albert Cahalan; all rights reserved.
6 * Copyright (C) 2002 by Vladimir Oleynik <dzo@simtreas.ru>
7 * SELinux support: (c) 2007 by Yuichi Nakamura <ynakam@hitachisoft.jp>
9 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
15 typedef struct unsigned_to_name_map_t {
17 char name[USERNAME_MAX_SIZE];
18 } unsigned_to_name_map_t;
20 typedef struct cache_t {
21 unsigned_to_name_map_t *cache;
25 static cache_t username, groupname;
27 static void clear_cache(cache_t *cp)
33 void FAST_FUNC clear_username_cache(void)
35 clear_cache(&username);
36 clear_cache(&groupname);
39 #if 0 /* more generic, but we don't need that yet */
40 /* Returns -N-1 if not found. */
41 /* cp->cache[N] is allocated and must be filled in this case */
42 static int get_cached(cache_t *cp, unsigned id)
45 for (i = 0; i < cp->size; i++)
46 if (cp->cache[i].id == id)
49 cp->cache = xrealloc_vector(cp->cache, 2, i);
50 cp->cache[i++].id = id;
55 static char* get_cached(cache_t *cp, long id,
56 char* FAST_FUNC x2x_utoa(long id))
59 for (i = 0; i < cp->size; i++)
60 if (cp->cache[i].id == id)
61 return cp->cache[i].name;
63 cp->cache = xrealloc_vector(cp->cache, 2, i);
65 /* Never fails. Generates numeric string if name isn't found */
66 safe_strncpy(cp->cache[i].name, x2x_utoa(id), sizeof(cp->cache[i].name));
67 return cp->cache[i].name;
69 const char* FAST_FUNC get_cached_username(uid_t uid)
71 return get_cached(&username, uid, uid2uname_utoa);
73 const char* FAST_FUNC get_cached_groupname(gid_t gid)
75 return get_cached(&groupname, gid, gid2group_utoa);
79 #define PROCPS_BUFSIZE 1024
81 static int read_to_buf(const char *filename, void *buf)
84 /* open_read_close() would do two reads, checking for EOF.
85 * When you have 10000 /proc/$NUM/stat to read, it isn't desirable */
87 fd = open(filename, O_RDONLY);
89 ret = read(fd, buf, PROCPS_BUFSIZE-1);
92 ((char *)buf)[ret > 0 ? ret : 0] = '\0';
96 static procps_status_t* FAST_FUNC alloc_procps_scan(void)
98 unsigned n = getpagesize();
99 procps_status_t* sp = xzalloc(sizeof(procps_status_t));
100 sp->dir = xopendir("/proc");
104 sp->shift_pages_to_bytes++;
106 sp->shift_pages_to_kb = sp->shift_pages_to_bytes - 10;
110 void FAST_FUNC free_procps_scan(procps_status_t* sp)
115 IF_SELINUX(free(sp->context);)
119 #if ENABLE_FEATURE_TOPMEM
120 static unsigned long fast_strtoul_16(char **endptr)
126 while ((c = *str++) != ' ') {
127 c = ((c|0x20) - '0');
129 // c = c + '0' - 'a' + 10:
130 c = c - ('a' - '0' - 10);
133 *endptr = str; /* We skip trailing space! */
136 /* TOPMEM uses fast_strtoul_10, so... */
137 #undef ENABLE_FEATURE_FAST_TOP
138 #define ENABLE_FEATURE_FAST_TOP 1
141 #if ENABLE_FEATURE_FAST_TOP
142 /* We cut a lot of corners here for speed */
143 static unsigned long fast_strtoul_10(char **endptr)
147 unsigned long n = *str - '0';
149 while ((c = *++str) != ' ')
150 n = n*10 + (c - '0');
152 *endptr = str + 1; /* We skip trailing space! */
156 static long fast_strtol_10(char **endptr)
159 return fast_strtoul_10(endptr);
162 return - (long)fast_strtoul_10(endptr);
165 static char *skip_fields(char *str, int count)
168 while (*str++ != ' ')
170 /* we found a space char, str points after it */
176 void BUG_comm_size(void);
177 procps_status_t* FAST_FUNC procps_scan(procps_status_t* sp, int flags)
179 struct dirent *entry;
180 char buf[PROCPS_BUFSIZE];
181 char filename[sizeof("/proc//cmdline") + sizeof(int)*3];
189 sp = alloc_procps_scan();
192 entry = readdir(sp->dir);
194 free_procps_scan(sp);
197 pid = bb_strtou(entry->d_name, NULL, 10);
201 /* After this point we have to break, not continue
202 * ("continue" would mean that current /proc/NNN
203 * is not a valid process info) */
205 memset(&sp->vsz, 0, sizeof(*sp) - offsetof(procps_status_t, vsz));
208 if (!(flags & ~PSSCAN_PID)) break;
211 if (flags & PSSCAN_CONTEXT) {
212 if (getpidcon(sp->pid, &sp->context) < 0)
217 filename_tail = filename + sprintf(filename, "/proc/%u/", pid);
219 if (flags & PSSCAN_UIDGID) {
220 if (stat(filename, &sb))
222 /* Effective UID/GID, not real */
227 if (flags & PSSCAN_STAT) {
230 #if !ENABLE_FEATURE_FAST_TOP
231 unsigned long vsz, rss;
233 /* see proc(5) for some details on this */
234 strcpy(filename_tail, "stat");
235 n = read_to_buf(filename, buf);
238 cp = strrchr(buf, ')'); /* split into "PID (cmd" and "<rest>" */
239 /*if (!cp || cp[1] != ' ')
242 if (sizeof(sp->comm) < 16)
244 comm1 = strchr(buf, '(');
246 safe_strncpy(sp->comm, comm1 + 1, sizeof(sp->comm));
248 #if !ENABLE_FEATURE_FAST_TOP
250 "%c %u " /* state, ppid */
251 "%u %u %d %*s " /* pgid, sid, tty, tpgid */
252 "%*s %*s %*s %*s %*s " /* flags, min_flt, cmin_flt, maj_flt, cmaj_flt */
253 "%lu %lu " /* utime, stime */
254 "%*s %*s %*s " /* cutime, cstime, priority */
256 "%*s %*s " /* timeout, it_real_value */
257 "%lu " /* start_time */
260 #if ENABLE_FEATURE_TOP_SMP_PROCESS
261 "%*s %*s %*s %*s %*s %*s " /*rss_rlim, start_code, end_code, start_stack, kstk_esp, kstk_eip */
262 "%*s %*s %*s %*s " /*signal, blocked, sigignore, sigcatch */
263 "%*s %*s %*s %*s " /*wchan, nswap, cnswap, exit_signal */
264 "%d" /*cpu last seen on*/
267 sp->state, &sp->ppid,
268 &sp->pgid, &sp->sid, &tty,
269 &sp->utime, &sp->stime,
274 #if ENABLE_FEATURE_TOP_SMP_PROCESS
275 , &sp->last_seen_on_cpu
281 #if ENABLE_FEATURE_TOP_SMP_PROCESS
283 sp->last_seen_on_cpu = 0;
286 /* vsz is in bytes and we want kb */
288 /* vsz is in bytes but rss is in *PAGES*! Can you believe that? */
289 sp->rss = rss << sp->shift_pages_to_kb;
290 sp->tty_major = (tty >> 8) & 0xfff;
291 sp->tty_minor = (tty & 0xff) | ((tty >> 12) & 0xfff00);
293 /* This costs ~100 bytes more but makes top faster by 20%
294 * If you run 10000 processes, this may be important for you */
295 sp->state[0] = cp[2];
297 sp->ppid = fast_strtoul_10(&cp);
298 sp->pgid = fast_strtoul_10(&cp);
299 sp->sid = fast_strtoul_10(&cp);
300 tty = fast_strtoul_10(&cp);
301 sp->tty_major = (tty >> 8) & 0xfff;
302 sp->tty_minor = (tty & 0xff) | ((tty >> 12) & 0xfff00);
303 cp = skip_fields(cp, 6); /* tpgid, flags, min_flt, cmin_flt, maj_flt, cmaj_flt */
304 sp->utime = fast_strtoul_10(&cp);
305 sp->stime = fast_strtoul_10(&cp);
306 cp = skip_fields(cp, 3); /* cutime, cstime, priority */
307 tasknice = fast_strtol_10(&cp);
308 cp = skip_fields(cp, 2); /* timeout, it_real_value */
309 sp->start_time = fast_strtoul_10(&cp);
310 /* vsz is in bytes and we want kb */
311 sp->vsz = fast_strtoul_10(&cp) >> 10;
312 /* vsz is in bytes but rss is in *PAGES*! Can you believe that? */
313 sp->rss = fast_strtoul_10(&cp) << sp->shift_pages_to_kb;
314 #if ENABLE_FEATURE_TOP_SMP_PROCESS
315 /* (6): rss_rlim, start_code, end_code, start_stack, kstk_esp, kstk_eip */
316 /* (4): signal, blocked, sigignore, sigcatch */
317 /* (4): wchan, nswap, cnswap, exit_signal */
318 cp = skip_fields(cp, 14);
319 //FIXME: is it safe to assume this field exists?
320 sp->last_seen_on_cpu = fast_strtoul_10(&cp);
322 #endif /* end of !ENABLE_FEATURE_TOP_SMP_PROCESS */
324 #if ENABLE_FEATURE_PS_ADDITIONAL_COLUMNS
325 sp->niceness = tasknice;
328 if (sp->vsz == 0 && sp->state[0] != 'Z')
334 else if (tasknice) /* > 0 */
340 #if ENABLE_FEATURE_TOPMEM
341 if (flags & (PSSCAN_SMAPS)) {
344 strcpy(filename_tail, "smaps");
345 file = fopen_for_read(filename);
348 while (fgets(buf, sizeof(buf), file)) {
352 #define SCAN(str, name) \
353 if (strncmp(buf, str, sizeof(str)-1) == 0) { \
354 tp = skip_whitespace(buf + sizeof(str)-1); \
355 sp->name += fast_strtoul_10(&tp); \
358 SCAN("Shared_Clean:" , shared_clean );
359 SCAN("Shared_Dirty:" , shared_dirty );
360 SCAN("Private_Clean:", private_clean);
361 SCAN("Private_Dirty:", private_dirty);
363 // f7d29000-f7d39000 rw-s ADR M:m OFS FILE
364 tp = strchr(buf, '-');
368 sz = fast_strtoul_16(&tp); /* start */
369 sz = (fast_strtoul_16(&tp) - sz) >> 10; /* end - start */
370 // tp -> "rw-s" string
372 // skipping "rw-s ADR M:m OFS "
373 tp = skip_whitespace(skip_fields(tp, 4));
374 // filter out /dev/something (something != zero)
375 if (strncmp(tp, "/dev/", 5) != 0 || strcmp(tp, "/dev/zero\n") == 0) {
378 } else if (w == '-') {
382 //else printf("DROPPING %s (%s)\n", buf, tp);
383 if (strcmp(tp, "[stack]\n") == 0)
390 #if ENABLE_FEATURE_PS_ADDITIONAL_COLUMNS
391 if (flags & PSSCAN_RUIDGID) {
394 strcpy(filename_tail, "status");
395 file = fopen_for_read(filename);
398 while (fgets(buf, sizeof(buf), file)) {
400 #define SCAN_TWO(str, name, statement) \
401 if (strncmp(buf, str, sizeof(str)-1) == 0) { \
402 tp = skip_whitespace(buf + sizeof(str)-1); \
403 sscanf(tp, "%u", &sp->name); \
406 SCAN_TWO("Uid:", ruid, continue);
407 SCAN_TWO("Gid:", rgid, break);
412 #endif /* PS_ADDITIONAL_COLUMNS */
413 #if 0 /* PSSCAN_CMD is not used */
414 if (flags & (PSSCAN_CMD|PSSCAN_ARGV0)) {
419 strcpy(filename_tail, "cmdline");
420 /* TODO: to get rid of size limits, read into malloc buf,
421 * then realloc it down to real size. */
422 n = read_to_buf(filename, buf);
425 if (flags & PSSCAN_ARGV0)
426 sp->argv0 = xstrdup(buf);
427 if (flags & PSSCAN_CMD) {
430 if ((unsigned char)(buf[n]) < ' ')
433 sp->cmd = xstrdup(buf);
437 if (flags & (PSSCAN_ARGV0|PSSCAN_ARGVN)) {
440 strcpy(filename_tail, "cmdline");
441 n = read_to_buf(filename, buf);
444 if (flags & PSSCAN_ARGVN) {
446 sp->argv0 = xmalloc(n + 1);
447 memcpy(sp->argv0, buf, n + 1);
448 /* sp->argv0[n] = '\0'; - buf has it */
451 sp->argv0 = xstrdup(buf);
455 if (flags & PSSCAN_EXE) {
456 strcpy(filename_tail, "exe");
458 sp->exe = xmalloc_readlink(filename);
465 void FAST_FUNC read_cmdline(char *buf, int col, unsigned pid, const char *comm)
468 char filename[sizeof("/proc//cmdline") + sizeof(int)*3];
470 sprintf(filename, "/proc/%u/cmdline", pid);
471 sz = open_read_close(filename, buf, col);
475 if ((unsigned char)(buf[sz]) < ' ')
478 snprintf(buf, col, "[%s]", comm);
483 // pid comm S ppid pgid sid tty_nr tty_pgrp flg
484 sprintf(buffer,"%d (%s) %c %d %d %d %d %d %lu %lu \
485 %lu %lu %lu %lu %lu %ld %ld %ld %ld %d 0 %llu %lu %ld %lu %lu %lu %lu %lu \
486 %lu %lu %lu %lu %lu %lu %lu %lu %d %d %lu %lu %llu\n",
500 cputime_to_clock_t(utime),
501 cputime_to_clock_t(stime),
502 cputime_to_clock_t(cutime),
503 cputime_to_clock_t(cstime),
510 mm ? get_mm_rss(mm) : 0,
512 mm ? mm->start_code : 0,
513 mm ? mm->end_code : 0,
514 mm ? mm->start_stack : 0,
517 the rest is some obsolete cruft