ps.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini ps implementation(s) for busybox
   4  *
   5  * Copyright (C) 1999,2000 by Lineo, inc.  Written by Erik Andersen
   6  * <andersen@lineo.com>, <andersee@debian.org>
   7  *
   8  *
   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+).
  12  *
  13  *
  14  *
  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)
  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
  28  *
  29  */
  30
  31 #include "internal.h"
  32 #include <stdio.h>
  33 #include <unistd.h>
  34 #include <dirent.h>
  35 #include <errno.h>
  36 #include <fcntl.h>
  37 #include <ctype.h>
  38 #include <sys/ioctl.h>
  39 #define BB_DECLARE_EXTERN
  40 #define bb_need_help
  41 #include "messages.c"
  42
  43 #define TERMINAL_WIDTH  79      /* not 80 in case terminal has linefold bug */
  44
  45
  46
  47 #if ! defined BB_FEATURE_USE_DEVPS_PATCH
  48
  49 /* The following is the first ps implementation --
  50  * the one using the /proc virtual filesystem.
  51  */
  52
  53 #if ! defined BB_FEATURE_USE_PROCFS
  54 #error Sorry, I depend on the /proc filesystem right now.
  55 #endif
  56
  57 typedef struct proc_s {
  58         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) */
  66 } proc_t;
  67
  68
  69
  70 static int file2str(char *filename, char *ret, int cap)
  71 {
  72         int fd, num_read;
  73
  74         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;
  81 }
  82
  83
  84 static void parse_proc_status(char *S, proc_t * P)
  85 {
  86         char *tmp;
  87
  88         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);
  95
  96         tmp = strstr(S, "Pid:");
  97         if (tmp)
  98                 sscanf(tmp, "Pid:\t%d\n" "PPid:\t%d\n", &P->pid, &P->ppid);
  99         else
 100                 fprintf(stderr, "Internal error!\n");
 101
 102         /* For busybox, ignoring effective, saved, etc */
 103         tmp = strstr(S, "Uid:");
 104         if (tmp)
 105                 sscanf(tmp, "Uid:\t%d", &P->ruid);
 106         else
 107                 fprintf(stderr, "Internal error!\n");
 108
 109         tmp = strstr(S, "Gid:");
 110         if (tmp)
 111                 sscanf(tmp, "Gid:\t%d", &P->rgid);
 112         else
 113                 fprintf(stderr, "Internal error!\n");
 114
 115 }
 116
 117
 118 extern int ps_main(int argc, char **argv)
 119 {
 120         proc_t p;
 121         DIR *dir;
 122         FILE *file;
 123         struct dirent *entry;
 124         char path[32], sbuf[512];
 125         char uidName[10] = "";
 126         char groupName[10] = "";
 127         int len, i, c;
 128 #ifdef BB_FEATURE_AUTOWIDTH
 129         struct winsize win = { 0, 0 };
 130         int terminal_width = TERMINAL_WIDTH;
 131 #else
 132 #define terminal_width  TERMINAL_WIDTH
 133 #endif
 134
 135
 136
 137         if (argc > 1 && strcmp(argv[1], dash_dash_help) == 0) {
 138                 usage ("ps\n"
 139 #ifndef BB_FEATURE_TRIVIAL_HELP
 140                                 "\nReport process status\n"
 141                                 "\nThis version of ps accepts no options.\n"
 142 #endif
 143                                 );
 144         }
 145
 146         dir = opendir("/proc");
 147         if (!dir)
 148                 fatalError("Can't open /proc\n");
 149
 150 #ifdef BB_FEATURE_AUTOWIDTH
 151                 ioctl(fileno(stdout), TIOCGWINSZ, &win);
 152                 if (win.ws_col > 0)
 153                         terminal_width = win.ws_col - 1;
 154 #endif
 155
 156         fprintf(stdout, "%5s  %-8s %-3s %5s %s\n", "PID", "Uid", "Gid",
 157                         "State", "Command");
 158         while ((entry = readdir(dir)) != NULL) {
 159                 uidName[0] = '\0';
 160                 groupName[0] = '\0';
 161
 162                 if (!isdigit(*entry->d_name))
 163                         continue;
 164                 sprintf(path, "/proc/%s/status", entry->d_name);
 165                 if ((file2str(path, sbuf, sizeof sbuf)) != -1) {
 166                         parse_proc_status(sbuf, &p);
 167                 }
 168
 169                 /* Make some adjustments as needed */
 170                 my_getpwuid(uidName, p.ruid);
 171                 if (*uidName == '\0')
 172                         sprintf(uidName, "%d", p.ruid);
 173                 my_getgrgid(groupName, p.rgid);
 174                 if (*groupName == '\0')
 175                         sprintf(groupName, "%d", p.rgid);
 176
 177                 sprintf(path, "/proc/%s/cmdline", entry->d_name);
 178                 file = fopen(path, "r");
 179                 if (file == NULL)
 180                         continue;
 181                 i = 0;
 182                 len = fprintf(stdout, "%5d %-8s %-8s %c ", p.pid, uidName, groupName,
 183                                 p.state);
 184                 while (((c = getc(file)) != EOF) && (i < (terminal_width-len))) {
 185                         i++;
 186                         if (c == '\0')
 187                                 c = ' ';
 188                         putc(c, stdout);
 189                 }
 190                 if (i == 0)
 191                         fprintf(stdout, "[%s]", p.cmd);
 192                 fprintf(stdout, "\n");
 193         }
 194         closedir(dir);
 195         exit(TRUE);
 196 }
 197
 198
 199 #else /* BB_FEATURE_USE_DEVPS_PATCH */
 200
 201
 202 /* The following is the second ps implementation --
 203  * this one uses the nifty new devps kernel device.
 204  */
 205
 206 #include <linux/devps.h>
 207
 208
 209 extern int ps_main(int argc, char **argv)
 210 {
 211         char device[] = "/dev/ps";
 212         int i, j, len, fd;
 213         pid_t num_pids;
 214         pid_t* pid_array = NULL;
 215         struct pid_info info;
 216         char uidName[10] = "";
 217         char groupName[10] = "";
 218 #ifdef BB_FEATURE_AUTOWIDTH
 219         struct winsize win = { 0, 0 };
 220         int terminal_width = TERMINAL_WIDTH;
 221 #else
 222 #define terminal_width  TERMINAL_WIDTH
 223 #endif
 224
 225         if (argc > 1 && **(argv + 1) == '-')
 226                 usage("ps-devps\n\nReport process status\n\nThis version of ps accepts no options.\n\n");
 227
 228         /* open device */
 229         fd = open(device, O_RDONLY);
 230         if (fd < 0)
 231                 fatalError( "open failed for `%s': %s\n", device, strerror (errno));
 232
 233         /* Find out how many processes there are */
 234         if (ioctl (fd, DEVPS_GET_NUM_PIDS, &num_pids)<0)
 235                 fatalError( "\nDEVPS_GET_PID_LIST: %s\n", strerror (errno));
 236
 237         /* Allocate some memory -- grab a few extras just in case
 238          * some new processes start up while we wait. The kernel will
 239          * just ignore any extras if we give it too many, and will trunc.
 240          * the list if we give it too few.  */
 241         pid_array = (pid_t*) calloc( num_pids+10, sizeof(pid_t));
 242         pid_array[0] = num_pids+10;
 243
 244         /* Now grab the pid list */
 245         if (ioctl (fd, DEVPS_GET_PID_LIST, pid_array)<0)
 246                 fatalError("\nDEVPS_GET_PID_LIST: %s\n", strerror (errno));
 247
 248 #ifdef BB_FEATURE_AUTOWIDTH
 249                 ioctl(fileno(stdout), TIOCGWINSZ, &win);
 250                 if (win.ws_col > 0)
 251                         terminal_width = win.ws_col - 1;
 252 #endif
 253
 254         /* Print up a ps listing */
 255         fprintf(stdout, "%5s  %-8s %-3s %5s %s\n", "PID", "Uid", "Gid",
 256                         "State", "Command");
 257
 258         for (i=1; i<pid_array[0] ; i++) {
 259                 uidName[0] = '\0';
 260                 groupName[0] = '\0';
 261             info.pid = pid_array[i];
 262
 263             if (ioctl (fd, DEVPS_GET_PID_INFO, &info)<0)
 264                         fatalError("\nDEVPS_GET_PID_INFO: %s\n", strerror (errno));
 265
 266                 /* Make some adjustments as needed */
 267                 my_getpwuid(uidName, info.euid);
 268                 if (*uidName == '\0')
 269                         sprintf(uidName, "%ld", info.euid);
 270                 my_getgrgid(groupName, info.egid);
 271                 if (*groupName == '\0')
 272                         sprintf(groupName, "%ld", info.egid);
 273
 274                 len = fprintf(stdout, "%5d %-8s %-8s %c ", info.pid, uidName, groupName, info.state);
 275
 276                 if (strlen(info.command_line) > 1) {
 277                         for( j=0; j<(sizeof(info.command_line)-1) && j < (terminal_width-len); j++) {
 278                                 if (*(info.command_line+j) == '\0' && *(info.command_line+j+1) != '\0') {
 279                                         *(info.command_line+j) = ' ';
 280                                 }
 281                         }
 282                         *(info.command_line+j) = '\0';
 283                         fprintf(stdout, "%s\n", info.command_line);
 284                 } else {
 285                         fprintf(stdout, "[%s]\n", info.name);
 286                 }
 287         }
 288
 289         /* Free memory */
 290         free( pid_array);
 291
 292         /* close device */
 293         if (close (fd) != 0)
 294                 fatalError("close failed for `%s': %s\n", device, strerror (errno));
 295
 296         exit (0);
 297 }
 298
 299 #endif /* BB_FEATURE_USE_DEVPS_PATCH */
 300