blob: bf162bbe53e21ef7775d01842bd47b94c1b88011 [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 *
Eric Andersen8ec10a92001-01-27 09:33:39 +00005 * Copyright (C) 1999,2000,2001 by Lineo, inc.
6 * Written by Erik Andersen <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
Erik Andersen246cc6d2000-03-07 07:41:42 +000031#include <stdio.h>
Eric Andersened3ef502001-01-27 08:24:39 +000032#include <stdlib.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>
Eric Andersened3ef502001-01-27 08:24:39 +000038#include <string.h>
Erik Andersen2ac2fae2000-03-07 23:32:17 +000039#include <sys/ioctl.h>
Eric Andersencbe31da2001-02-20 06:14:08 +000040#include "busybox.h"
Erik Andersen2ac2fae2000-03-07 23:32:17 +000041
Mark Whitley59ab0252001-01-23 22:30:04 +000042static const int TERMINAL_WIDTH = 79; /* not 80 in case terminal has linefold bug */
Eric Andersen86ab8a32000-06-02 03:21:42 +000043
44
Eric Andersend23f9ba1999-10-20 19:18:15 +000045
Erik Andersen269bba22000-03-08 14:50:47 +000046#if ! defined BB_FEATURE_USE_DEVPS_PATCH
Erik Andersen246cc6d2000-03-07 07:41:42 +000047
48/* The following is the first ps implementation --
49 * the one using the /proc virtual filesystem.
50 */
51
Eric Andersend23f9ba1999-10-20 19:18:15 +000052typedef struct proc_s {
Erik Andersene49d5ec2000-02-08 19:58:47 +000053 char
54 cmd[16]; /* basename of executable file in call to exec(2) */
55 int
56 ruid, rgid, /* real only (sorry) */
57 pid, /* process id */
58 ppid; /* pid of parent process */
59 char
60 state; /* single-char code for process state (S=sleeping) */
Eric Andersend23f9ba1999-10-20 19:18:15 +000061} proc_t;
62
63
64
Erik Andersene49d5ec2000-02-08 19:58:47 +000065static int file2str(char *filename, char *ret, int cap)
Eric Andersend23f9ba1999-10-20 19:18:15 +000066{
Erik Andersene49d5ec2000-02-08 19:58:47 +000067 int fd, num_read;
Eric Andersend23f9ba1999-10-20 19:18:15 +000068
Erik Andersene49d5ec2000-02-08 19:58:47 +000069 if ((fd = open(filename, O_RDONLY, 0)) == -1)
70 return -1;
71 if ((num_read = read(fd, ret, cap - 1)) <= 0)
72 return -1;
73 ret[num_read] = 0;
74 close(fd);
75 return num_read;
Eric Andersend23f9ba1999-10-20 19:18:15 +000076}
77
78
Erik Andersene49d5ec2000-02-08 19:58:47 +000079static void parse_proc_status(char *S, proc_t * P)
Eric Andersend23f9ba1999-10-20 19:18:15 +000080{
Erik Andersene49d5ec2000-02-08 19:58:47 +000081 char *tmp;
Eric Andersend23f9ba1999-10-20 19:18:15 +000082
Erik Andersene49d5ec2000-02-08 19:58:47 +000083 memset(P->cmd, 0, sizeof P->cmd);
84 sscanf(S, "Name:\t%15c", P->cmd);
85 tmp = strchr(P->cmd, '\n');
86 if (tmp)
87 *tmp = '\0';
88 tmp = strstr(S, "State");
89 sscanf(tmp, "State:\t%c", &P->state);
Eric Andersend23f9ba1999-10-20 19:18:15 +000090
Erik Andersene49d5ec2000-02-08 19:58:47 +000091 tmp = strstr(S, "Pid:");
92 if (tmp)
93 sscanf(tmp, "Pid:\t%d\n" "PPid:\t%d\n", &P->pid, &P->ppid);
94 else
Matt Kraaidd19c692001-01-31 19:00:21 +000095 error_msg("Internal error!");
Eric Andersend23f9ba1999-10-20 19:18:15 +000096
Eric Andersen77d92682001-05-23 20:32:09 +000097 /* For busybox, ignoring effective, saved, etc. */
Erik Andersene49d5ec2000-02-08 19:58:47 +000098 tmp = strstr(S, "Uid:");
99 if (tmp)
100 sscanf(tmp, "Uid:\t%d", &P->ruid);
101 else
Matt Kraaidd19c692001-01-31 19:00:21 +0000102 error_msg("Internal error!");
Erik Andersene49d5ec2000-02-08 19:58:47 +0000103
104 tmp = strstr(S, "Gid:");
105 if (tmp)
106 sscanf(tmp, "Gid:\t%d", &P->rgid);
107 else
Matt Kraaidd19c692001-01-31 19:00:21 +0000108 error_msg("Internal error!");
Eric Andersend23f9ba1999-10-20 19:18:15 +0000109
110}
Eric Andersenef8b6c71999-10-20 08:05:35 +0000111
Eric Andersenef8b6c71999-10-20 08:05:35 +0000112extern int ps_main(int argc, char **argv)
113{
Erik Andersene49d5ec2000-02-08 19:58:47 +0000114 proc_t p;
115 DIR *dir;
116 FILE *file;
117 struct dirent *entry;
118 char path[32], sbuf[512];
Eric Andersenbd193a42000-12-13 01:52:39 +0000119 char uidName[9];
120 char groupName[9];
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000121 int len, i, c;
122#ifdef BB_FEATURE_AUTOWIDTH
Eric Andersenfad04fd2000-07-14 06:49:52 +0000123 struct winsize win = { 0, 0, 0, 0 };
Eric Andersen86ab8a32000-06-02 03:21:42 +0000124 int terminal_width = TERMINAL_WIDTH;
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000125#else
Eric Andersen86ab8a32000-06-02 03:21:42 +0000126#define terminal_width TERMINAL_WIDTH
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000127#endif
128
129
Eric Andersenef8b6c71999-10-20 08:05:35 +0000130
Erik Andersene49d5ec2000-02-08 19:58:47 +0000131 dir = opendir("/proc");
Erik Andersen246cc6d2000-03-07 07:41:42 +0000132 if (!dir)
Matt Kraaidd19c692001-01-31 19:00:21 +0000133 error_msg_and_die("Can't open /proc");
Eric Andersend23f9ba1999-10-20 19:18:15 +0000134
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000135#ifdef BB_FEATURE_AUTOWIDTH
136 ioctl(fileno(stdout), TIOCGWINSZ, &win);
137 if (win.ws_col > 0)
138 terminal_width = win.ws_col - 1;
139#endif
140
Matt Kraai12f417e2001-01-18 02:57:08 +0000141 printf("%5s %-8s %-3s %5s %s\n", "PID", "Uid", "Gid", "State", "Command");
Erik Andersene49d5ec2000-02-08 19:58:47 +0000142 while ((entry = readdir(dir)) != NULL) {
Erik Andersene49d5ec2000-02-08 19:58:47 +0000143 if (!isdigit(*entry->d_name))
144 continue;
145 sprintf(path, "/proc/%s/status", entry->d_name);
146 if ((file2str(path, sbuf, sizeof sbuf)) != -1) {
147 parse_proc_status(sbuf, &p);
148 }
149
150 /* Make some adjustments as needed */
151 my_getpwuid(uidName, p.ruid);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000152 if (*uidName == '\0')
153 sprintf(uidName, "%d", p.ruid);
Erik Andersen246cc6d2000-03-07 07:41:42 +0000154 my_getgrgid(groupName, p.rgid);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000155 if (*groupName == '\0')
156 sprintf(groupName, "%d", p.rgid);
157
Erik Andersene49d5ec2000-02-08 19:58:47 +0000158 sprintf(path, "/proc/%s/cmdline", entry->d_name);
159 file = fopen(path, "r");
Erik Andersen246cc6d2000-03-07 07:41:42 +0000160 if (file == NULL)
Erik Andersen227a59b2000-04-25 23:24:55 +0000161 continue;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000162 i = 0;
Matt Kraai12f417e2001-01-18 02:57:08 +0000163 len = printf("%5d %-8s %-8s %c ", p.pid, uidName, groupName, p.state);
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000164 while (((c = getc(file)) != EOF) && (i < (terminal_width-len))) {
Erik Andersene49d5ec2000-02-08 19:58:47 +0000165 i++;
166 if (c == '\0')
167 c = ' ';
168 putc(c, stdout);
169 }
Matt Kraai44e38402000-09-07 04:34:17 +0000170 fclose(file);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000171 if (i == 0)
Matt Kraai12f417e2001-01-18 02:57:08 +0000172 printf("[%s]", p.cmd);
173 putchar('\n');
Eric Andersend23f9ba1999-10-20 19:18:15 +0000174 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000175 closedir(dir);
Matt Kraai3e856ce2000-12-01 02:55:13 +0000176 return EXIT_SUCCESS;
Eric Andersenef8b6c71999-10-20 08:05:35 +0000177}
Erik Andersen246cc6d2000-03-07 07:41:42 +0000178
179
Erik Andersen269bba22000-03-08 14:50:47 +0000180#else /* BB_FEATURE_USE_DEVPS_PATCH */
Erik Andersen246cc6d2000-03-07 07:41:42 +0000181
182
183/* The following is the second ps implementation --
184 * this one uses the nifty new devps kernel device.
185 */
186
Eric Andersenc674d702000-07-10 22:57:14 +0000187#include <linux/devps.h> /* For Erik's nifty devps device driver */
Erik Andersen246cc6d2000-03-07 07:41:42 +0000188
189
190extern int ps_main(int argc, char **argv)
191{
192 char device[] = "/dev/ps";
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000193 int i, j, len, fd;
Erik Andersen246cc6d2000-03-07 07:41:42 +0000194 pid_t num_pids;
195 pid_t* pid_array = NULL;
196 struct pid_info info;
Eric Andersenbd193a42000-12-13 01:52:39 +0000197 char uidName[9];
198 char groupName[9];
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000199#ifdef BB_FEATURE_AUTOWIDTH
Pavel Roskinf626dcb2000-07-14 15:55:41 +0000200 struct winsize win = { 0, 0, 0, 0 };
Eric Andersen86ab8a32000-06-02 03:21:42 +0000201 int terminal_width = TERMINAL_WIDTH;
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000202#else
Eric Andersen86ab8a32000-06-02 03:21:42 +0000203#define terminal_width TERMINAL_WIDTH
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000204#endif
Erik Andersen246cc6d2000-03-07 07:41:42 +0000205
206 if (argc > 1 && **(argv + 1) == '-')
Eric Andersen67991cf2001-02-14 21:23:06 +0000207 show_usage();
Erik Andersen246cc6d2000-03-07 07:41:42 +0000208
209 /* open device */
210 fd = open(device, O_RDONLY);
211 if (fd < 0)
Matt Kraai1fa1ade2000-12-18 03:57:16 +0000212 perror_msg_and_die( "open failed for `%s'", device);
Erik Andersen246cc6d2000-03-07 07:41:42 +0000213
214 /* Find out how many processes there are */
215 if (ioctl (fd, DEVPS_GET_NUM_PIDS, &num_pids)<0)
Matt Kraai1fa1ade2000-12-18 03:57:16 +0000216 perror_msg_and_die( "\nDEVPS_GET_PID_LIST");
Erik Andersen246cc6d2000-03-07 07:41:42 +0000217
218 /* Allocate some memory -- grab a few extras just in case
219 * some new processes start up while we wait. The kernel will
220 * just ignore any extras if we give it too many, and will trunc.
221 * the list if we give it too few. */
Matt Kraai322ae932000-09-13 02:46:14 +0000222 pid_array = (pid_t*) xcalloc( num_pids+10, sizeof(pid_t));
Erik Andersen246cc6d2000-03-07 07:41:42 +0000223 pid_array[0] = num_pids+10;
224
225 /* Now grab the pid list */
226 if (ioctl (fd, DEVPS_GET_PID_LIST, pid_array)<0)
Matt Kraai1fa1ade2000-12-18 03:57:16 +0000227 perror_msg_and_die("\nDEVPS_GET_PID_LIST");
Erik Andersen246cc6d2000-03-07 07:41:42 +0000228
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000229#ifdef BB_FEATURE_AUTOWIDTH
230 ioctl(fileno(stdout), TIOCGWINSZ, &win);
231 if (win.ws_col > 0)
232 terminal_width = win.ws_col - 1;
233#endif
234
Erik Andersen246cc6d2000-03-07 07:41:42 +0000235 /* Print up a ps listing */
Matt Kraai12f417e2001-01-18 02:57:08 +0000236 printf("%5s %-8s %-3s %5s %s\n", "PID", "Uid", "Gid", "State", "Command");
Erik Andersen246cc6d2000-03-07 07:41:42 +0000237
238 for (i=1; i<pid_array[0] ; i++) {
Erik Andersen246cc6d2000-03-07 07:41:42 +0000239 info.pid = pid_array[i];
240
241 if (ioctl (fd, DEVPS_GET_PID_INFO, &info)<0)
Matt Kraai1fa1ade2000-12-18 03:57:16 +0000242 perror_msg_and_die("\nDEVPS_GET_PID_INFO");
Erik Andersen246cc6d2000-03-07 07:41:42 +0000243
244 /* Make some adjustments as needed */
245 my_getpwuid(uidName, info.euid);
246 if (*uidName == '\0')
247 sprintf(uidName, "%ld", info.euid);
248 my_getgrgid(groupName, info.egid);
249 if (*groupName == '\0')
250 sprintf(groupName, "%ld", info.egid);
251
Matt Kraai12f417e2001-01-18 02:57:08 +0000252 len = printf("%5d %-8s %-8s %c ", info.pid, uidName, groupName, info.state);
Erik Andersen246cc6d2000-03-07 07:41:42 +0000253
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000254 if (strlen(info.command_line) > 1) {
255 for( j=0; j<(sizeof(info.command_line)-1) && j < (terminal_width-len); j++) {
256 if (*(info.command_line+j) == '\0' && *(info.command_line+j+1) != '\0') {
257 *(info.command_line+j) = ' ';
258 }
259 }
260 *(info.command_line+j) = '\0';
Matt Kraai12f417e2001-01-18 02:57:08 +0000261 puts(info.command_line);
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000262 } else {
Matt Kraai12f417e2001-01-18 02:57:08 +0000263 printf("[%s]\n", info.name);
Erik Andersen2ac2fae2000-03-07 23:32:17 +0000264 }
Erik Andersen246cc6d2000-03-07 07:41:42 +0000265 }
266
267 /* Free memory */
268 free( pid_array);
269
270 /* close device */
271 if (close (fd) != 0)
Matt Kraai1fa1ade2000-12-18 03:57:16 +0000272 perror_msg_and_die("close failed for `%s'", device);
Erik Andersen246cc6d2000-03-07 07:41:42 +0000273
274 exit (0);
275}
276
Erik Andersen269bba22000-03-08 14:50:47 +0000277#endif /* BB_FEATURE_USE_DEVPS_PATCH */
Erik Andersen246cc6d2000-03-07 07:41:42 +0000278