blob: df321612a57924f5f6f95620c4e02c8a2acca820 [file] [log] [blame]
Erik Andersene49d5ec2000-02-08 19:58:47 +00001/* vi: set sw=4 ts=4: */
Eric Andersenef8b6c71999-10-20 08:05:35 +00002/*
Erik Andersen246cc6d2000-03-07 07:41:42 +00003 * Mini ps implementation(s) for busybox
4 *
Erik Andersen61677fe2000-04-13 01:18:56 +00005 * Copyright (C) 1999,2000 by Lineo, inc. Written by Erik Andersen
Erik Andersen246cc6d2000-03-07 07:41:42 +00006 * <andersen@lineo.com>, <andersee@debian.org>
Eric Andersenef8b6c71999-10-20 08:05:35 +00007 *
Eric Andersenc4996011999-10-20 22:08:37 +00008 *
Erik Andersen246cc6d2000-03-07 07:41:42 +00009 * 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+).
Eric Andersenef8b6c71999-10-20 08:05:35 +000012 *
Eric Andersenef8b6c71999-10-20 08:05:35 +000013 *
Eric Andersenef8b6c71999-10-20 08:05:35 +000014 *
Erik Andersen246cc6d2000-03-07 07:41:42 +000015 * 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)
18 * any later version.
19 *
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
23 * more details.
24 *
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
Eric Andersenef8b6c71999-10-20 08:05:35 +000028 *
29 */
30
Eric Andersen3570a342000-09-25 21:45:58 +000031#include "busybox.h"
Erik Andersen246cc6d2000-03-07 07:41:42 +000032#include <stdio.h>
Eric Andersenef8b6c71999-10-20 08:05:35 +000033#include <unistd.h>
34#include <dirent.h>
Erik Andersen246cc6d2000-03-07 07:41:42 +000035#include <errno.h>
Eric Andersend23f9ba1999-10-20 19:18:15 +000036#include <fcntl.h>
37#include <ctype.h>
Erik Andersen2ac2fae2000-03-07 23:32:17 +000038#include <sys/ioctl.h>
Erik Andersen7ab9c7e2000-05-12 19:41:47 +000039#define BB_DECLARE_EXTERN
40#define bb_need_help
41#include "messages.c"
Erik Andersen2ac2fae2000-03-07 23:32:17 +000042
Eric Andersen86ab8a32000-06-02 03:21:42 +000043#define TERMINAL_WIDTH 79 /* not 80 in case terminal has linefold bug */
44
45
Eric Andersend23f9ba1999-10-20 19:18:15 +000046
Erik Andersen269bba22000-03-08 14:50:47 +000047#if ! defined BB_FEATURE_USE_DEVPS_PATCH
Erik Andersen246cc6d2000-03-07 07:41:42 +000048
49/* The following is the first ps implementation --
50 * the one using the /proc virtual filesystem.
51 */
52
Eric Andersen0ecb54a1999-12-05 23:24:55 +000053#if ! defined BB_FEATURE_USE_PROCFS
54#error Sorry, I depend on the /proc filesystem right now.
55#endif
Eric Andersend23f9ba1999-10-20 19:18:15 +000056
57typedef struct proc_s {
Erik Andersene49d5ec2000-02-08 19:58:47 +000058 char
59 cmd[16]; /* basename of executable file in call to exec(2) */
60 int
61 ruid, rgid, /* real only (sorry) */
62 pid, /* process id */
63 ppid; /* pid of parent process */
64 char
65 state; /* single-char code for process state (S=sleeping) */
Eric Andersend23f9ba1999-10-20 19:18:15 +000066} proc_t;
67
68
69
Erik Andersene49d5ec2000-02-08 19:58:47 +000070static int file2str(char *filename, char *ret, int cap)
Eric Andersend23f9ba1999-10-20 19:18:15 +000071{
Erik Andersene49d5ec2000-02-08 19:58:47 +000072 int fd, num_read;
Eric Andersend23f9ba1999-10-20 19:18:15 +000073
Erik Andersene49d5ec2000-02-08 19:58:47 +000074 if ((fd = open(filename, O_RDONLY, 0)) == -1)
75 return -1;
76 if ((num_read = read(fd, ret, cap - 1)) <= 0)
77 return -1;
78 ret[num_read] = 0;
79 close(fd);
80 return num_read;
Eric Andersend23f9ba1999-10-20 19:18:15 +000081}
82
83
Erik Andersene49d5ec2000-02-08 19:58:47 +000084static void parse_proc_status(char *S, proc_t * P)
Eric Andersend23f9ba1999-10-20 19:18:15 +000085{
Erik Andersene49d5ec2000-02-08 19:58:47 +000086 char *tmp;
Eric Andersend23f9ba1999-10-20 19:18:15 +000087
Erik Andersene49d5ec2000-02-08 19:58:47 +000088 memset(P->cmd, 0, sizeof P->cmd);
89 sscanf(S, "Name:\t%15c", P->cmd);
90 tmp = strchr(P->cmd, '\n');
91 if (tmp)
92 *tmp = '\0';
93 tmp = strstr(S, "State");
94 sscanf(tmp, "State:\t%c", &P->state);
Eric Andersend23f9ba1999-10-20 19:18:15 +000095
Erik Andersene49d5ec2000-02-08 19:58:47 +000096 tmp = strstr(S, "Pid:");
97 if (tmp)
98 sscanf(tmp, "Pid:\t%d\n" "PPid:\t%d\n", &P->pid, &P->ppid);
99 else
Matt Kraaid537a952000-07-14 01:51:25 +0000100 errorMsg("Internal error!\n");
Eric Andersend23f9ba1999-10-20 19:18:15 +0000101
Erik Andersene49d5ec2000-02-08 19:58:47 +0000102 /* For busybox, ignoring effective, saved, etc */
103 tmp = strstr(S, "Uid:");
104 if (tmp)
105 sscanf(tmp, "Uid:\t%d", &P->ruid);
106 else
Matt Kraaid537a952000-07-14 01:51:25 +0000107 errorMsg("Internal error!\n");
Erik Andersene49d5ec2000-02-08 19:58:47 +0000108
109 tmp = strstr(S, "Gid:");
110 if (tmp)
111 sscanf(tmp, "Gid:\t%d", &P->rgid);
112 else
Matt Kraaid537a952000-07-14 01:51:25 +0000113 errorMsg("Internal error!\n");
Eric Andersend23f9ba1999-10-20 19:18:15 +0000114
115}
Eric Andersenef8b6c71999-10-20 08:05:35 +0000116
Eric Andersenef8b6c71999-10-20 08:05:35 +0000117extern int ps_main(int argc, char **argv)
118{
Erik Andersene49d5ec2000-02-08 19:58:47 +0000119 proc_t p;
120 DIR *dir;
121 FILE *file;
122 struct dirent *entry;
123 char path[32], sbuf[512];
124 char uidName[10] = "";
125 char groupName[10] = "";
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000126 int len, i, c;
127#ifdef BB_FEATURE_AUTOWIDTH
Eric Andersenfad04fd2000-07-14 06:49:52 +0000128 struct winsize win = { 0, 0, 0, 0 };
Eric Andersen86ab8a32000-06-02 03:21:42 +0000129 int terminal_width = TERMINAL_WIDTH;
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000130#else
Eric Andersen86ab8a32000-06-02 03:21:42 +0000131#define terminal_width TERMINAL_WIDTH
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000132#endif
133
134
Eric Andersenef8b6c71999-10-20 08:05:35 +0000135
Erik Andersene49d5ec2000-02-08 19:58:47 +0000136 dir = opendir("/proc");
Erik Andersen246cc6d2000-03-07 07:41:42 +0000137 if (!dir)
Eric Andersen86ab8a32000-06-02 03:21:42 +0000138 fatalError("Can't open /proc\n");
Eric Andersend23f9ba1999-10-20 19:18:15 +0000139
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000140#ifdef BB_FEATURE_AUTOWIDTH
141 ioctl(fileno(stdout), TIOCGWINSZ, &win);
142 if (win.ws_col > 0)
143 terminal_width = win.ws_col - 1;
144#endif
145
Erik Andersene49d5ec2000-02-08 19:58:47 +0000146 fprintf(stdout, "%5s %-8s %-3s %5s %s\n", "PID", "Uid", "Gid",
147 "State", "Command");
148 while ((entry = readdir(dir)) != NULL) {
149 uidName[0] = '\0';
150 groupName[0] = '\0';
151
152 if (!isdigit(*entry->d_name))
153 continue;
154 sprintf(path, "/proc/%s/status", entry->d_name);
155 if ((file2str(path, sbuf, sizeof sbuf)) != -1) {
156 parse_proc_status(sbuf, &p);
157 }
158
159 /* Make some adjustments as needed */
160 my_getpwuid(uidName, p.ruid);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000161 if (*uidName == '\0')
162 sprintf(uidName, "%d", p.ruid);
Erik Andersen246cc6d2000-03-07 07:41:42 +0000163 my_getgrgid(groupName, p.rgid);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000164 if (*groupName == '\0')
165 sprintf(groupName, "%d", p.rgid);
166
Erik Andersene49d5ec2000-02-08 19:58:47 +0000167 sprintf(path, "/proc/%s/cmdline", entry->d_name);
168 file = fopen(path, "r");
Erik Andersen246cc6d2000-03-07 07:41:42 +0000169 if (file == NULL)
Erik Andersen227a59b2000-04-25 23:24:55 +0000170 continue;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000171 i = 0;
Erik Andersen227a59b2000-04-25 23:24:55 +0000172 len = fprintf(stdout, "%5d %-8s %-8s %c ", p.pid, uidName, groupName,
173 p.state);
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000174 while (((c = getc(file)) != EOF) && (i < (terminal_width-len))) {
Erik Andersene49d5ec2000-02-08 19:58:47 +0000175 i++;
176 if (c == '\0')
177 c = ' ';
178 putc(c, stdout);
179 }
Matt Kraai44e38402000-09-07 04:34:17 +0000180 fclose(file);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000181 if (i == 0)
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000182 fprintf(stdout, "[%s]", p.cmd);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000183 fprintf(stdout, "\n");
Eric Andersend23f9ba1999-10-20 19:18:15 +0000184 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000185 closedir(dir);
Eric Andersenb6106152000-06-19 17:25:40 +0000186 return(TRUE);
Eric Andersenef8b6c71999-10-20 08:05:35 +0000187}
Erik Andersen246cc6d2000-03-07 07:41:42 +0000188
189
Erik Andersen269bba22000-03-08 14:50:47 +0000190#else /* BB_FEATURE_USE_DEVPS_PATCH */
Erik Andersen246cc6d2000-03-07 07:41:42 +0000191
192
193/* The following is the second ps implementation --
194 * this one uses the nifty new devps kernel device.
195 */
196
Eric Andersenc674d702000-07-10 22:57:14 +0000197#include <linux/devps.h> /* For Erik's nifty devps device driver */
Erik Andersen246cc6d2000-03-07 07:41:42 +0000198
199
200extern int ps_main(int argc, char **argv)
201{
202 char device[] = "/dev/ps";
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000203 int i, j, len, fd;
Erik Andersen246cc6d2000-03-07 07:41:42 +0000204 pid_t num_pids;
205 pid_t* pid_array = NULL;
206 struct pid_info info;
207 char uidName[10] = "";
208 char groupName[10] = "";
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000209#ifdef BB_FEATURE_AUTOWIDTH
Pavel Roskinf626dcb2000-07-14 15:55:41 +0000210 struct winsize win = { 0, 0, 0, 0 };
Eric Andersen86ab8a32000-06-02 03:21:42 +0000211 int terminal_width = TERMINAL_WIDTH;
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000212#else
Eric Andersen86ab8a32000-06-02 03:21:42 +0000213#define terminal_width TERMINAL_WIDTH
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000214#endif
Erik Andersen246cc6d2000-03-07 07:41:42 +0000215
216 if (argc > 1 && **(argv + 1) == '-')
Matt Kraai3bd8bd82000-07-14 23:28:47 +0000217 usage(ps_usage);
Erik Andersen246cc6d2000-03-07 07:41:42 +0000218
219 /* open device */
220 fd = open(device, O_RDONLY);
221 if (fd < 0)
222 fatalError( "open failed for `%s': %s\n", device, strerror (errno));
223
224 /* Find out how many processes there are */
225 if (ioctl (fd, DEVPS_GET_NUM_PIDS, &num_pids)<0)
226 fatalError( "\nDEVPS_GET_PID_LIST: %s\n", strerror (errno));
227
228 /* Allocate some memory -- grab a few extras just in case
229 * some new processes start up while we wait. The kernel will
230 * just ignore any extras if we give it too many, and will trunc.
231 * the list if we give it too few. */
Matt Kraai322ae932000-09-13 02:46:14 +0000232 pid_array = (pid_t*) xcalloc( num_pids+10, sizeof(pid_t));
Erik Andersen246cc6d2000-03-07 07:41:42 +0000233 pid_array[0] = num_pids+10;
234
235 /* Now grab the pid list */
236 if (ioctl (fd, DEVPS_GET_PID_LIST, pid_array)<0)
237 fatalError("\nDEVPS_GET_PID_LIST: %s\n", strerror (errno));
238
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000239#ifdef BB_FEATURE_AUTOWIDTH
240 ioctl(fileno(stdout), TIOCGWINSZ, &win);
241 if (win.ws_col > 0)
242 terminal_width = win.ws_col - 1;
243#endif
244
Erik Andersen246cc6d2000-03-07 07:41:42 +0000245 /* Print up a ps listing */
246 fprintf(stdout, "%5s %-8s %-3s %5s %s\n", "PID", "Uid", "Gid",
247 "State", "Command");
248
249 for (i=1; i<pid_array[0] ; i++) {
250 uidName[0] = '\0';
251 groupName[0] = '\0';
252 info.pid = pid_array[i];
253
254 if (ioctl (fd, DEVPS_GET_PID_INFO, &info)<0)
255 fatalError("\nDEVPS_GET_PID_INFO: %s\n", strerror (errno));
256
257 /* Make some adjustments as needed */
258 my_getpwuid(uidName, info.euid);
259 if (*uidName == '\0')
260 sprintf(uidName, "%ld", info.euid);
261 my_getgrgid(groupName, info.egid);
262 if (*groupName == '\0')
263 sprintf(groupName, "%ld", info.egid);
264
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000265 len = fprintf(stdout, "%5d %-8s %-8s %c ", info.pid, uidName, groupName, info.state);
Erik Andersen246cc6d2000-03-07 07:41:42 +0000266
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000267 if (strlen(info.command_line) > 1) {
268 for( j=0; j<(sizeof(info.command_line)-1) && j < (terminal_width-len); j++) {
269 if (*(info.command_line+j) == '\0' && *(info.command_line+j+1) != '\0') {
270 *(info.command_line+j) = ' ';
271 }
272 }
273 *(info.command_line+j) = '\0';
Erik Andersen246cc6d2000-03-07 07:41:42 +0000274 fprintf(stdout, "%s\n", info.command_line);
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000275 } else {
Erik Andersen246cc6d2000-03-07 07:41:42 +0000276 fprintf(stdout, "[%s]\n", info.name);
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000277 }
Erik Andersen246cc6d2000-03-07 07:41:42 +0000278 }
279
280 /* Free memory */
281 free( pid_array);
282
283 /* close device */
284 if (close (fd) != 0)
285 fatalError("close failed for `%s': %s\n", device, strerror (errno));
286
287 exit (0);
288}
289
Erik Andersen269bba22000-03-08 14:50:47 +0000290#endif /* BB_FEATURE_USE_DEVPS_PATCH */
Erik Andersen246cc6d2000-03-07 07:41:42 +0000291