1 /* vi: set sw=4 ts=4: */
3 * Mini ps implementation(s) for busybox
5 * Copyright (C) 1999,2000 by Lineo, inc. Written by Erik Andersen
6 * <andersen@lineo.com>, <andersee@debian.org>
9 * This contains _two_ implementations of ps for Linux. One uses the
10 * traditional /proc virtual filesystem, and the other use the devps kernel
11 * driver (written by Erik Andersen to avoid using /proc thereby saving 100k+).
15 * This program is free software; you can redistribute it and/or modify it
16 * under the terms of the GNU General Public License as published by the Free
17 * Software Foundation; either version 2 of the License, or (at your option)
20 * This program is distributed in the hope that it will be useful, but WITHOUT
21 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
22 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
25 * You should have received a copy of the GNU General Public License along with
26 * this program; if not, write to the Free Software Foundation, Inc., 59 Temple
27 * Place, Suite 330, Boston, MA 02111-1307 USA
38 #include <sys/ioctl.h>
39 #define BB_DECLARE_EXTERN
43 #define TERMINAL_WIDTH 79 /* not 80 in case terminal has linefold bug */
47 #if ! defined BB_FEATURE_USE_DEVPS_PATCH
49 /* The following is the first ps implementation --
50 * the one using the /proc virtual filesystem.
53 #if ! defined BB_FEATURE_USE_PROCFS
54 #error Sorry, I depend on the /proc filesystem right now.
57 typedef struct proc_s {
59 cmd[16]; /* basename of executable file in call to exec(2) */
61 ruid, rgid, /* real only (sorry) */
63 ppid; /* pid of parent process */
65 state; /* single-char code for process state (S=sleeping) */
70 static int file2str(char *filename, char *ret, int cap)
74 if ((fd = open(filename, O_RDONLY, 0)) == -1)
76 if ((num_read = read(fd, ret, cap - 1)) <= 0)
84 static void parse_proc_status(char *S, proc_t * P)
88 memset(P->cmd, 0, sizeof P->cmd);
89 sscanf(S, "Name:\t%15c", P->cmd);
90 tmp = strchr(P->cmd, '\n');
93 tmp = strstr(S, "State");
94 sscanf(tmp, "State:\t%c", &P->state);
96 tmp = strstr(S, "Pid:");
98 sscanf(tmp, "Pid:\t%d\n" "PPid:\t%d\n", &P->pid, &P->ppid);
100 errorMsg("Internal error!\n");
102 /* For busybox, ignoring effective, saved, etc */
103 tmp = strstr(S, "Uid:");
105 sscanf(tmp, "Uid:\t%d", &P->ruid);
107 errorMsg("Internal error!\n");
109 tmp = strstr(S, "Gid:");
111 sscanf(tmp, "Gid:\t%d", &P->rgid);
113 errorMsg("Internal error!\n");
117 extern int ps_main(int argc, char **argv)
122 struct dirent *entry;
123 char path[32], sbuf[512];
124 char uidName[10] = "";
125 char groupName[10] = "";
127 #ifdef BB_FEATURE_AUTOWIDTH
128 struct winsize win = { 0, 0, 0, 0 };
129 int terminal_width = TERMINAL_WIDTH;
131 #define terminal_width TERMINAL_WIDTH
136 if (argc > 1 && strcmp(argv[1], dash_dash_help) == 0)
139 dir = opendir("/proc");
141 fatalError("Can't open /proc\n");
143 #ifdef BB_FEATURE_AUTOWIDTH
144 ioctl(fileno(stdout), TIOCGWINSZ, &win);
146 terminal_width = win.ws_col - 1;
149 fprintf(stdout, "%5s %-8s %-3s %5s %s\n", "PID", "Uid", "Gid",
151 while ((entry = readdir(dir)) != NULL) {
155 if (!isdigit(*entry->d_name))
157 sprintf(path, "/proc/%s/status", entry->d_name);
158 if ((file2str(path, sbuf, sizeof sbuf)) != -1) {
159 parse_proc_status(sbuf, &p);
162 /* Make some adjustments as needed */
163 my_getpwuid(uidName, p.ruid);
164 if (*uidName == '\0')
165 sprintf(uidName, "%d", p.ruid);
166 my_getgrgid(groupName, p.rgid);
167 if (*groupName == '\0')
168 sprintf(groupName, "%d", p.rgid);
170 sprintf(path, "/proc/%s/cmdline", entry->d_name);
171 file = fopen(path, "r");
175 len = fprintf(stdout, "%5d %-8s %-8s %c ", p.pid, uidName, groupName,
177 while (((c = getc(file)) != EOF) && (i < (terminal_width-len))) {
184 fprintf(stdout, "[%s]", p.cmd);
185 fprintf(stdout, "\n");
192 #else /* BB_FEATURE_USE_DEVPS_PATCH */
195 /* The following is the second ps implementation --
196 * this one uses the nifty new devps kernel device.
199 #include <linux/devps.h> /* For Erik's nifty devps device driver */
202 extern int ps_main(int argc, char **argv)
204 char device[] = "/dev/ps";
207 pid_t* pid_array = NULL;
208 struct pid_info info;
209 char uidName[10] = "";
210 char groupName[10] = "";
211 #ifdef BB_FEATURE_AUTOWIDTH
212 struct winsize win = { 0, 0, 0, 0 };
213 int terminal_width = TERMINAL_WIDTH;
215 #define terminal_width TERMINAL_WIDTH
218 if (argc > 1 && **(argv + 1) == '-')
222 fd = open(device, O_RDONLY);
224 fatalError( "open failed for `%s': %s\n", device, strerror (errno));
226 /* Find out how many processes there are */
227 if (ioctl (fd, DEVPS_GET_NUM_PIDS, &num_pids)<0)
228 fatalError( "\nDEVPS_GET_PID_LIST: %s\n", strerror (errno));
230 /* Allocate some memory -- grab a few extras just in case
231 * some new processes start up while we wait. The kernel will
232 * just ignore any extras if we give it too many, and will trunc.
233 * the list if we give it too few. */
234 pid_array = (pid_t*) calloc( num_pids+10, sizeof(pid_t));
235 pid_array[0] = num_pids+10;
237 /* Now grab the pid list */
238 if (ioctl (fd, DEVPS_GET_PID_LIST, pid_array)<0)
239 fatalError("\nDEVPS_GET_PID_LIST: %s\n", strerror (errno));
241 #ifdef BB_FEATURE_AUTOWIDTH
242 ioctl(fileno(stdout), TIOCGWINSZ, &win);
244 terminal_width = win.ws_col - 1;
247 /* Print up a ps listing */
248 fprintf(stdout, "%5s %-8s %-3s %5s %s\n", "PID", "Uid", "Gid",
251 for (i=1; i<pid_array[0] ; i++) {
254 info.pid = pid_array[i];
256 if (ioctl (fd, DEVPS_GET_PID_INFO, &info)<0)
257 fatalError("\nDEVPS_GET_PID_INFO: %s\n", strerror (errno));
259 /* Make some adjustments as needed */
260 my_getpwuid(uidName, info.euid);
261 if (*uidName == '\0')
262 sprintf(uidName, "%ld", info.euid);
263 my_getgrgid(groupName, info.egid);
264 if (*groupName == '\0')
265 sprintf(groupName, "%ld", info.egid);
267 len = fprintf(stdout, "%5d %-8s %-8s %c ", info.pid, uidName, groupName, info.state);
269 if (strlen(info.command_line) > 1) {
270 for( j=0; j<(sizeof(info.command_line)-1) && j < (terminal_width-len); j++) {
271 if (*(info.command_line+j) == '\0' && *(info.command_line+j+1) != '\0') {
272 *(info.command_line+j) = ' ';
275 *(info.command_line+j) = '\0';
276 fprintf(stdout, "%s\n", info.command_line);
278 fprintf(stdout, "[%s]\n", info.name);
287 fatalError("close failed for `%s': %s\n", device, strerror (errno));
292 #endif /* BB_FEATURE_USE_DEVPS_PATCH */