/*
* Mini ps implementation(s) for busybox
*
- * Copyright (C) 1999,2000,2001 by Lineo, inc.
- * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
- *
- *
- * This contains _two_ implementations of ps for Linux. One uses the
- * traditional /proc virtual filesystem, and the other use the devps kernel
- * driver (written by Erik Andersen to avoid using /proc thereby saving 100k+).
- *
- *
- *
- * This program is free software; you can redistribute it and/or modify it
- * under the terms of the GNU General Public License as published by the Free
- * Software Foundation; either version 2 of the License, or (at your option)
- * any later version.
- *
- * This program is distributed in the hope that it will be useful, but WITHOUT
- * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
- * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
- * more details.
- *
- * You should have received a copy of the GNU General Public License along with
- * this program; if not, write to the Free Software Foundation, Inc., 59 Temple
- * Place, Suite 330, Boston, MA 02111-1307 USA
+ * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
*
+ * Licensed under the GPL v2, see the file LICENSE in this tarball.
*/
#include "busybox.h"
#include <fcntl.h>
#include <ctype.h>
#include <string.h>
+#include <termios.h>
#include <sys/ioctl.h>
-#define BB_DECLARE_EXTERN
-#define bb_need_help
-#include "messages.c"
-
-static const int TERMINAL_WIDTH = 79; /* not 80 in case terminal has linefold bug */
-
-
-
-#if ! defined BB_FEATURE_USE_DEVPS_PATCH
-
-/* The following is the first ps implementation --
- * the one using the /proc virtual filesystem.
- */
-
-typedef struct proc_s {
- char
- cmd[16]; /* basename of executable file in call to exec(2) */
- int
- ruid, rgid, /* real only (sorry) */
- pid, /* process id */
- ppid; /* pid of parent process */
- char
- state; /* single-char code for process state (S=sleeping) */
-} proc_t;
-
-
-
-static int file2str(char *filename, char *ret, int cap)
-{
- int fd, num_read;
-
- if ((fd = open(filename, O_RDONLY, 0)) == -1)
- return -1;
- if ((num_read = read(fd, ret, cap - 1)) <= 0)
- return -1;
- ret[num_read] = 0;
- close(fd);
- return num_read;
-}
-
+#if ENABLE_SELINUX
+#include <selinux/selinux.h> /* for is_selinux_enabled() */
+#endif
-static void parse_proc_status(char *S, proc_t * P)
+int ps_main(int argc, char **argv)
{
- char *tmp;
-
- memset(P->cmd, 0, sizeof P->cmd);
- sscanf(S, "Name:\t%15c", P->cmd);
- tmp = strchr(P->cmd, '\n');
- if (tmp)
- *tmp = '\0';
- tmp = strstr(S, "State");
- sscanf(tmp, "State:\t%c", &P->state);
-
- tmp = strstr(S, "Pid:");
- if (tmp)
- sscanf(tmp, "Pid:\t%d\n" "PPid:\t%d\n", &P->pid, &P->ppid);
- else
- error_msg("Internal error!");
-
- /* For busybox, ignoring effective, saved, etc */
- tmp = strstr(S, "Uid:");
- if (tmp)
- sscanf(tmp, "Uid:\t%d", &P->ruid);
- else
- error_msg("Internal error!");
+ procps_status_t * p;
+ int i, len;
- tmp = strstr(S, "Gid:");
- if (tmp)
- sscanf(tmp, "Gid:\t%d", &P->rgid);
- else
- error_msg("Internal error!");
+#if ENABLE_SELINUX
+ int use_selinux = 0;
+ security_context_t sid=NULL;
+#endif
-}
+#if ENABLE_FEATURE_PS_WIDE
+ int terminal_width;
+ int w_count = 0;
-extern int ps_main(int argc, char **argv)
-{
- proc_t p;
- DIR *dir;
- FILE *file;
- struct dirent *entry;
- char path[32], sbuf[512];
- char uidName[9];
- char groupName[9];
- int len, i, c;
-#ifdef BB_FEATURE_AUTOWIDTH
- struct winsize win = { 0, 0, 0, 0 };
- int terminal_width = TERMINAL_WIDTH;
+ bb_opt_complementally="-:ww";
#else
-#define terminal_width TERMINAL_WIDTH
+# define terminal_width 79
#endif
-
-
- dir = opendir("/proc");
- if (!dir)
- error_msg_and_die("Can't open /proc");
-
-#ifdef BB_FEATURE_AUTOWIDTH
- ioctl(fileno(stdout), TIOCGWINSZ, &win);
- if (win.ws_col > 0)
- terminal_width = win.ws_col - 1;
+#if ENABLE_FEATURE_PS_WIDE || ENABLE_SELINUX
+ /* handle arguments */
+#if ENABLE_FEATURE_PS_WIDE && ENABLE_SELINUX
+ i = bb_getopt_ulflags(argc, argv, "wc", &w_count);
+#elif ENABLE_FEATURE_PS_WIDE && !ENABLE_SELINUX
+ bb_getopt_ulflags(argc, argv, "w", &w_count);
+#else /* !ENABLE_FEATURE_PS_WIDE && ENABLE_SELINUX */
+ i = bb_getopt_ulflags(argc, argv, "c");
#endif
-
- printf("%5s %-8s %-3s %5s %s\n", "PID", "Uid", "Gid", "State", "Command");
- while ((entry = readdir(dir)) != NULL) {
- if (!isdigit(*entry->d_name))
- continue;
- sprintf(path, "/proc/%s/status", entry->d_name);
- if ((file2str(path, sbuf, sizeof sbuf)) != -1) {
- parse_proc_status(sbuf, &p);
- }
-
- /* Make some adjustments as needed */
- my_getpwuid(uidName, p.ruid);
- if (*uidName == '\0')
- sprintf(uidName, "%d", p.ruid);
- my_getgrgid(groupName, p.rgid);
- if (*groupName == '\0')
- sprintf(groupName, "%d", p.rgid);
-
- sprintf(path, "/proc/%s/cmdline", entry->d_name);
- file = fopen(path, "r");
- if (file == NULL)
- continue;
- i = 0;
- len = printf("%5d %-8s %-8s %c ", p.pid, uidName, groupName, p.state);
- while (((c = getc(file)) != EOF) && (i < (terminal_width-len))) {
- i++;
- if (c == '\0')
- c = ' ';
- putc(c, stdout);
- }
- fclose(file);
- if (i == 0)
- printf("[%s]", p.cmd);
- putchar('\n');
+#if ENABLE_FEATURE_PS_WIDE
+ /* if w is given once, GNU ps sets the width to 132,
+ * if w is given more than once, it is "unlimited"
+ */
+ if(w_count) {
+ terminal_width = (w_count==1) ? 132 : INT_MAX;
+ } else {
+ get_terminal_width_height(1, &terminal_width, NULL);
+ /* Go one less... */
+ terminal_width--;
}
- closedir(dir);
- return EXIT_SUCCESS;
-}
-
-
-#else /* BB_FEATURE_USE_DEVPS_PATCH */
-
-
-/* The following is the second ps implementation --
- * this one uses the nifty new devps kernel device.
- */
-
-#include <linux/devps.h> /* For Erik's nifty devps device driver */
-
-
-extern int ps_main(int argc, char **argv)
-{
- char device[] = "/dev/ps";
- int i, j, len, fd;
- pid_t num_pids;
- pid_t* pid_array = NULL;
- struct pid_info info;
- char uidName[9];
- char groupName[9];
-#ifdef BB_FEATURE_AUTOWIDTH
- struct winsize win = { 0, 0, 0, 0 };
- int terminal_width = TERMINAL_WIDTH;
-#else
-#define terminal_width TERMINAL_WIDTH
#endif
-
- if (argc > 1 && **(argv + 1) == '-')
- show_usage();
-
- /* open device */
- fd = open(device, O_RDONLY);
- if (fd < 0)
- perror_msg_and_die( "open failed for `%s'", device);
-
- /* Find out how many processes there are */
- if (ioctl (fd, DEVPS_GET_NUM_PIDS, &num_pids)<0)
- perror_msg_and_die( "\nDEVPS_GET_PID_LIST");
-
- /* Allocate some memory -- grab a few extras just in case
- * some new processes start up while we wait. The kernel will
- * just ignore any extras if we give it too many, and will trunc.
- * the list if we give it too few. */
- pid_array = (pid_t*) xcalloc( num_pids+10, sizeof(pid_t));
- pid_array[0] = num_pids+10;
-
- /* Now grab the pid list */
- if (ioctl (fd, DEVPS_GET_PID_LIST, pid_array)<0)
- perror_msg_and_die("\nDEVPS_GET_PID_LIST");
-
-#ifdef BB_FEATURE_AUTOWIDTH
- ioctl(fileno(stdout), TIOCGWINSZ, &win);
- if (win.ws_col > 0)
- terminal_width = win.ws_col - 1;
+#if ENABLE_SELINUX
+ if ((i & (1+ENABLE_FEATURE_PS_WIDE)) && is_selinux_enabled())
+ use_selinux = 1;
#endif
+#endif /* ENABLE_FEATURE_PS_WIDE || ENABLE_SELINUX */
- /* Print up a ps listing */
- printf("%5s %-8s %-3s %5s %s\n", "PID", "Uid", "Gid", "State", "Command");
-
- for (i=1; i<pid_array[0] ; i++) {
- info.pid = pid_array[i];
-
- if (ioctl (fd, DEVPS_GET_PID_INFO, &info)<0)
- perror_msg_and_die("\nDEVPS_GET_PID_INFO");
-
- /* Make some adjustments as needed */
- my_getpwuid(uidName, info.euid);
- if (*uidName == '\0')
- sprintf(uidName, "%ld", info.euid);
- my_getgrgid(groupName, info.egid);
- if (*groupName == '\0')
- sprintf(groupName, "%ld", info.egid);
-
- len = printf("%5d %-8s %-8s %c ", info.pid, uidName, groupName, info.state);
+#if ENABLE_SELINUX
+ if (use_selinux)
+ printf(" PID Context Stat Command\n");
+ else
+#endif
+ printf(" PID Uid VmSize Stat Command\n");
+
+ while ((p = procps_scan(1)) != 0) {
+ char *namecmd = p->cmd;
+#if ENABLE_SELINUX
+ if (use_selinux)
+ {
+ char sbuf[128];
+ len = sizeof(sbuf);
+
+ if (is_selinux_enabled()) {
+ if (getpidcon(p->pid,&sid)<0)
+ sid=NULL;
+ }
- if (strlen(info.command_line) > 1) {
- for( j=0; j<(sizeof(info.command_line)-1) && j < (terminal_width-len); j++) {
- if (*(info.command_line+j) == '\0' && *(info.command_line+j+1) != '\0') {
- *(info.command_line+j) = ' ';
- }
+ if (sid) {
+ /* I assume sid initilized with NULL */
+ len = strlen(sid)+1;
+ safe_strncpy(sbuf, sid, len);
+ freecon(sid);
+ sid=NULL;
+ }else {
+ safe_strncpy(sbuf, "unknown",7);
}
- *(info.command_line+j) = '\0';
- puts(info.command_line);
+ len = printf("%5d %-32s %s ", p->pid, sbuf, p->state);
+ }
+ else
+#endif
+ if(p->rss == 0)
+ len = printf("%5d %-8s %s ", p->pid, p->user, p->state);
+ else
+ len = printf("%5d %-8s %6ld %s ", p->pid, p->user, p->rss, p->state);
+
+ i = terminal_width-len;
+
+ if(namecmd && namecmd[0]) {
+ if(i < 0)
+ i = 0;
+ if(strlen(namecmd) > (size_t)i)
+ namecmd[i] = 0;
+ printf("%s\n", namecmd);
} else {
- printf("[%s]\n", info.name);
+ namecmd = p->short_cmd;
+ if(i < 2)
+ i = 2;
+ if(strlen(namecmd) > ((size_t)i-2))
+ namecmd[i-2] = 0;
+ printf("[%s]\n", namecmd);
}
+ /* no check needed, but to make valgrind happy.. */
+ if (ENABLE_FEATURE_CLEAN_UP && p->cmd)
+ free(p->cmd);
}
-
- /* Free memory */
- free( pid_array);
-
- /* close device */
- if (close (fd) != 0)
- perror_msg_and_die("close failed for `%s'", device);
-
- exit (0);
+ return EXIT_SUCCESS;
}
-
-#endif /* BB_FEATURE_USE_DEVPS_PATCH */
-