1 files changed, 171 insertions, 90 deletions

diff --git a/libbb/procps.c b/libbb/procps.c
index 15f7715..dee5638 100644
--- a/libbb/procps.c
+++ b/libbb/procps.c
@@ -11,6 +11,35 @@
 #include "libbb.h"
 
 
+typedef struct {
+	uid_t uid;
+	char username[12];
+} user_map_t;
+
+static user_map_t *username_cache;
+static int username_cache_size;
+
+void clear_username_cache(void)
+{
+	free(username_cache);
+	username_cache = NULL;
+	username_cache_size = 0;
+}
+
+const char* get_cached_username(uid_t uid)
+{
+	int i;
+	for (i = 0; i < username_cache_size; i++)
+		if (username_cache[i].uid == uid)
+			return username_cache[i].username;
+	i = username_cache_size++;
+	username_cache = xrealloc(username_cache, username_cache_size * sizeof(*username_cache));
+	username_cache[i].uid = uid;
+	bb_getpwuid(username_cache[i].username, uid, sizeof(username_cache[i].username));
+	return username_cache[i].username;
+}
+
+
 #define PROCPS_BUFSIZE 1024
 
 static int read_to_buf(const char *filename, void *buf)
@@ -21,119 +50,171 @@ static int read_to_buf(const char *filename, void *buf)
 	return ret;
 }
 
+procps_status_t* alloc_procps_scan(int flags)
+{
+	procps_status_t* sp = xzalloc(sizeof(procps_status_t));
+	sp->dir = xopendir("/proc");
+	return sp;
+}
 
-procps_status_t * procps_scan(int save_user_arg0)
+void free_procps_scan(procps_status_t* sp)
 {
-	static DIR *dir;
-	static procps_status_t ret_status;
+	closedir(sp->dir);
+	free(sp->cmd);
+	free(sp);
+}
 
+void BUG_comm_size(void);
+procps_status_t* procps_scan(procps_status_t* sp, int flags)
+{
 	struct dirent *entry;
-	char *name;
 	char buf[PROCPS_BUFSIZE];
-	char status[sizeof("/proc//cmdline") + sizeof(int)*3];
-	char *status_tail;
-	procps_status_t curstatus;
+	char filename[sizeof("/proc//cmdline") + sizeof(int)*3];
+	char *filename_tail;
 	long tasknice;
-	int pid;
+	unsigned pid;
 	int n;
 	struct stat sb;
 
-	if (!dir) {
-		dir = xopendir("/proc");
-	}
+	if (!sp)
+		sp = alloc_procps_scan(flags);
+
 	for (;;) {
-		entry = readdir(dir);
+		entry = readdir(sp->dir);
 		if (entry == NULL) {
-			closedir(dir);
-			dir = 0;
-			return 0;
+			free_procps_scan(sp);
+			return NULL;
 		}
-		name = entry->d_name;
-		if (!(*name >= '0' && *name <= '9'))
+		if (safe_strtou(entry->d_name, &pid))
 			continue;
 
-		memset(&curstatus, 0, sizeof(procps_status_t));
-		pid = atoi(name);
-		curstatus.pid = pid;
+		/* After this point we have to break, not continue
+		 * ("continue" would mean that current /proc/NNN
+		 * is not a valid process info) */
 
-		status_tail = status + sprintf(status, "/proc/%d", pid);
-		if (stat(status, &sb))
-			continue;
-		bb_getpwuid(curstatus.user, sb.st_uid, sizeof(curstatus.user));
+		memset(&sp->rss, 0, sizeof(*sp) - offsetof(procps_status_t, rss));
 
-		/* see proc(5) for some details on this */
-		strcpy(status_tail, "/stat");
-		n = read_to_buf(status, buf);
-		if (n < 0)
-			continue;
-		name = strrchr(buf, ')'); /* split into "PID (cmd" and "<rest>" */
-		if (name == 0 || name[1] != ' ')
-			continue;
-		*name = 0;
-		sscanf(buf, "%*s (%15c", curstatus.short_cmd);
-		n = sscanf(name+2,
-		"%c %d "
-		"%*s %*s %*s %*s "     /* pgrp, session, tty, tpgid */
-		"%*s %*s %*s %*s %*s " /* flags, min_flt, cmin_flt, maj_flt, cmaj_flt */
-#ifdef CONFIG_FEATURE_TOP_CPU_USAGE_PERCENTAGE
-		"%lu %lu "             /* utime, stime */
-#else
-		"%*s %*s "             /* utime, stime */
-#endif
-		"%*s %*s %*s "         /* cutime, cstime, priority */
-		"%ld "                 /* nice */
-		"%*s %*s %*s "         /* timeout, it_real_value, start_time */
-		"%*s "                 /* vsize */
-		"%ld",                 /* rss */
-		curstatus.state, &curstatus.ppid,
-#ifdef CONFIG_FEATURE_TOP_CPU_USAGE_PERCENTAGE
-		&curstatus.utime, &curstatus.stime,
-#endif
-		&tasknice,
-		&curstatus.rss);
-#ifdef CONFIG_FEATURE_TOP_CPU_USAGE_PERCENTAGE
-		if (n != 6)
-#else
-		if (n != 4)
-#endif
-			continue;
+		sp->pid = pid;
+		if (!(flags & ~PSSCAN_PID)) break;
 
-		if (curstatus.rss == 0 && curstatus.state[0] != 'Z')
-			curstatus.state[1] = 'W';
-		else
-			curstatus.state[1] = ' ';
-		if (tasknice < 0)
-			curstatus.state[2] = '<';
-		else if (tasknice > 0)
-			curstatus.state[2] = 'N';
-		else
-			curstatus.state[2] = ' ';
+		filename_tail = filename + sprintf(filename, "/proc/%d", pid);
+
+		if (flags & PSSCAN_UIDGID) {
+			if (stat(filename, &sb))
+				break;
+			/* Need comment - is this effective or read UID/GID? */
+			sp->uid = sb.st_uid;
+			sp->gid = sb.st_gid;
+		}
+	
+		if (flags & PSSCAN_STAT) {
+			char *cp;
+			/* see proc(5) for some details on this */
+			strcpy(filename_tail, "/stat");
+			n = read_to_buf(filename, buf);
+			if (n < 0)
+				break;
+			cp = strrchr(buf, ')'); /* split into "PID (cmd" and "<rest>" */
+			if (!cp || cp[1] != ' ')
+				break;
+			cp[0] = '\0';
+			if (sizeof(sp->comm) < 16)
+				BUG_comm_size();
+			sscanf(buf, "%*s (%15c", sp->comm);
+			n = sscanf(cp+2,
+				"%c %u "               /* state, ppid */
+				"%u %u %*s %*s "       /* pgid, sid, tty, tpgid */
+				"%*s %*s %*s %*s %*s " /* flags, min_flt, cmin_flt, maj_flt, cmaj_flt */
+				"%lu %lu "             /* utime, stime */
+				"%*s %*s %*s "         /* cutime, cstime, priority */
+				"%ld "                 /* nice */
+				"%*s %*s %*s "         /* timeout, it_real_value, start_time */
+				"%*s "                 /* vsize */
+				"%lu",                 /* rss */
+				sp->state, &sp->ppid,
+				&sp->pgid, &sp->sid,
+				&sp->utime, &sp->stime,
+				&tasknice,
+				&sp->rss);
+			if (n != 8)
+				break;
+
+			if (sp->rss == 0 && sp->state[0] != 'Z')
+				sp->state[1] = 'W';
+			else
+				sp->state[1] = ' ';
+			if (tasknice < 0)
+				sp->state[2] = '<';
+			else if (tasknice > 0)
+				sp->state[2] = 'N';
+			else
+				sp->state[2] = ' ';
 
 #ifdef PAGE_SHIFT
-		curstatus.rss <<= (PAGE_SHIFT - 10);     /* 2**10 = 1kb */
+			sp->rss <<= (PAGE_SHIFT - 10);     /* 2**10 = 1kb */
 #else
-		curstatus.rss *= (getpagesize() >> 10);     /* 2**10 = 1kb */
+			sp->rss *= (getpagesize() >> 10);     /* 2**10 = 1kb */
 #endif
+		}
 
-		if (save_user_arg0) {
-			strcpy(status_tail, "/cmdline");
-			n = read_to_buf(status, buf);
-			if (n > 0) {
-				if (buf[n-1]=='\n')
-					buf[--n] = 0;
-				name = buf;
-				while (n) {
-					if (((unsigned char)*name) < ' ')
-						*name = ' ';
-					name++;
-					n--;
-				}
-				*name = 0;
-				if (buf[0])
-					curstatus.cmd = strdup(buf);
-				/* if NULL it work true also */
+		if (flags & PSSCAN_CMD) {
+			free(sp->cmd);
+			sp->cmd = NULL;
+			strcpy(filename_tail, "/cmdline");
+			n = read_to_buf(filename, buf);
+			if (n <= 0)
+				break;
+			if (buf[n-1] == '\n') {
+				if (!--n)
+					break;
+				buf[n] = '\0';
 			}
+			do {
+				n--;
+				if ((unsigned char)(buf[n]) < ' ')
+					buf[n] = ' ';
+			} while (n);
+			sp->cmd = strdup(buf);
 		}
-		return memcpy(&ret_status, &curstatus, sizeof(procps_status_t));
+		break;
 	}
+	return sp;
 }
+/* from kernel:
+	//             pid comm S ppid pgid sid tty_nr tty_pgrp flg
+        sprintf(buffer,"%d (%s) %c %d  %d   %d  %d     %d       %lu %lu \
+%lu %lu %lu %lu %lu %ld %ld %ld %ld %d 0 %llu %lu %ld %lu %lu %lu %lu %lu \
+%lu %lu %lu %lu %lu %lu %lu %lu %d %d %lu %lu %llu\n",
+                task->pid,
+                tcomm,
+                state,
+                ppid,
+                pgid,
+                sid,
+                tty_nr,
+                tty_pgrp,
+                task->flags,
+                min_flt,
+
+                cmin_flt,
+                maj_flt,
+                cmaj_flt,
+                cputime_to_clock_t(utime),
+                cputime_to_clock_t(stime),
+                cputime_to_clock_t(cutime),
+                cputime_to_clock_t(cstime),
+                priority,
+                nice,
+                num_threads,
+	// 0,
+                start_time,
+                vsize,
+                mm ? get_mm_rss(mm) : 0,
+                rsslim,
+                mm ? mm->start_code : 0,
+                mm ? mm->end_code : 0,
+                mm ? mm->start_stack : 0,
+                esp,
+                eip,
+the rest is some obsolete cruft
+*/