procps/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. and Erik Andersen
   6  * Copyright (C) 1999,2000,2001 by Erik Andersen <andersee@debian.org>
   7  *
   8  * This program is free software; you can redistribute it and/or modify it
   9  * under the terms of the GNU General Public License as published by the Free
  10  * Software Foundation; either version 2 of the License, or (at your option)
  11  * any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful, but WITHOUT
  14  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  15  * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
  16  * more details.
  17  *
  18  * You should have received a copy of the GNU General Public License along with
  19  * this program; if not, write to the Free Software Foundation, Inc., 59 Temple
  20  * Place, Suite 330, Boston, MA 02111-1307 USA
  21  */
  22
  23 /*
  24  * This contains _two_ implementations of ps for Linux.  One uses the
  25  * traditional /proc virtual filesystem, and the other use the devps kernel
  26  * driver (written by Erik Andersen to avoid using /proc thereby saving 100k+).
  27  */
  28
  29 #include <stdio.h>
  30 #include <stdlib.h>
  31 #include <unistd.h>
  32 #include <dirent.h>
  33 #include <errno.h>
  34 #include <fcntl.h>
  35 #include <ctype.h>
  36 #include <string.h>
  37 #include <termios.h>
  38 #include <sys/ioctl.h>
  39 #include "busybox.h"
  40
  41 static const int TERMINAL_WIDTH = 79;      /* not 80 in case terminal has linefold bug */
  42
  43
  44
  45 #if ! defined CONFIG_FEATURE_USE_DEVPS_PATCH
  46
  47 /* The following is the first ps implementation --
  48  * the one using the /proc virtual filesystem.
  49  */
  50
  51 typedef struct proc_s {
  52         char cmd[16];                                   /* basename of executable file in call to exec(2) */
  53         int ruid;                                               /* real only (sorry) */
  54         int pid;                                                /* process id */
  55         int ppid;                                               /* pid of parent process */
  56         char state;                                             /* single-char code for process state (S=sleeping) */
  57         unsigned int vmsize;                    /* size of process as far as the vm is concerned */
  58 } proc_t;
  59
  60
  61
  62 static int file2str(char *filename, char *ret, int cap)
  63 {
  64         int fd, num_read;
  65
  66         if ((fd = open(filename, O_RDONLY, 0)) == -1)
  67                 return -1;
  68         if ((num_read = read(fd, ret, cap - 1)) <= 0)
  69                 return -1;
  70         ret[num_read] = 0;
  71         close(fd);
  72         return num_read;
  73 }
  74
  75
  76 static void parse_proc_status(char *S, proc_t * P)
  77 {
  78         char *tmp;
  79
  80         memset(P->cmd, 0, sizeof P->cmd);
  81         sscanf(S, "Name:\t%15c", P->cmd);
  82         tmp = strchr(P->cmd, '\n');
  83         if (tmp)
  84                 *tmp = '\0';
  85         tmp = strstr(S, "State");
  86         sscanf(tmp, "State:\t%c", &P->state);
  87
  88         P->pid = 0;
  89         P->ppid = 0;
  90         tmp = strstr(S, "Pid:");
  91         if (tmp)
  92                 sscanf(tmp, "Pid:\t%d\n" "PPid:\t%d\n", &P->pid, &P->ppid);
  93         else
  94                 error_msg("Internal error!");
  95
  96         /* For busybox, ignoring effective, saved, etc. */
  97         P->ruid = 0;
  98         tmp = strstr(S, "Uid:");
  99         if (tmp)
 100                 sscanf(tmp, "Uid:\t%d", &P->ruid);
 101         else
 102                 error_msg("Internal error!");
 103
 104         P->vmsize = 0;
 105         tmp = strstr(S, "VmSize:");
 106         if (tmp)
 107                 sscanf(tmp, "VmSize:\t%d", &P->vmsize);
 108 #if 0
 109         else
 110                 error_msg("Internal error!");
 111 #endif
 112 }
 113
 114 extern int ps_main(int argc, char **argv)
 115 {
 116         proc_t p;
 117         DIR *dir;
 118         FILE *file;
 119         struct dirent *entry;
 120         char path[32], sbuf[512];
 121         char uidName[9];
 122         int len, i, c;
 123 #ifdef CONFIG_FEATURE_AUTOWIDTH
 124         struct winsize win = { 0, 0, 0, 0 };
 125         int terminal_width = TERMINAL_WIDTH;
 126 #else
 127 #define terminal_width  TERMINAL_WIDTH
 128 #endif
 129
 130
 131
 132         dir = opendir("/proc");
 133         if (!dir)
 134                 error_msg_and_die("Can't open /proc");
 135
 136 #ifdef CONFIG_FEATURE_AUTOWIDTH
 137                 ioctl(fileno(stdout), TIOCGWINSZ, &win);
 138                 if (win.ws_col > 0)
 139                         terminal_width = win.ws_col - 1;
 140 #endif
 141
 142         printf("  PID  Uid     VmSize Stat Command\n");
 143         while ((entry = readdir(dir)) != NULL) {
 144                 if (!isdigit(*entry->d_name))
 145                         continue;
 146                 sprintf(path, "/proc/%s/status", entry->d_name);
 147                 if ((file2str(path, sbuf, sizeof sbuf)) != -1) {
 148                         parse_proc_status(sbuf, &p);
 149                 }
 150
 151                 /* Make some adjustments as needed */
 152                 my_getpwuid(uidName, p.ruid);
 153                 if (*uidName == '\0')
 154                         sprintf(uidName, "%d", p.ruid);
 155
 156                 sprintf(path, "/proc/%s/cmdline", entry->d_name);
 157                 file = fopen(path, "r");
 158                 if (file == NULL)
 159                         continue;
 160                 i = 0;
 161                 if(p.vmsize == 0)
 162                         len = printf("%5d %-8s        %c    ", p.pid, uidName, p.state);
 163                 else
 164                         len = printf("%5d %-8s %6d %c    ", p.pid, uidName, p.vmsize, p.state);
 165                 while (((c = getc(file)) != EOF) && (i < (terminal_width-len))) {
 166                         i++;
 167                         if (c == '\0')
 168                                 c = ' ';
 169                         putc(c, stdout);
 170                 }
 171                 fclose(file);
 172                 if (i == 0)
 173                         printf("[%s]", p.cmd);
 174                 putchar('\n');
 175         }
 176         closedir(dir);
 177         return EXIT_SUCCESS;
 178 }
 179
 180
 181 #else /* CONFIG_FEATURE_USE_DEVPS_PATCH */
 182
 183
 184 /* The following is the second ps implementation --
 185  * this one uses the nifty new devps kernel device.
 186  */
 187
 188 #include <linux/devps.h> /* For Erik's nifty devps device driver */
 189
 190
 191 extern int ps_main(int argc, char **argv)
 192 {
 193         char device[] = "/dev/ps";
 194         int i, j, len, fd;
 195         pid_t num_pids;
 196         pid_t* pid_array = NULL;
 197         struct pid_info info;
 198         char uidName[9];
 199 #ifdef CONFIG_FEATURE_AUTOWIDTH
 200         struct winsize win = { 0, 0, 0, 0 };
 201         int terminal_width = TERMINAL_WIDTH;
 202 #else
 203 #define terminal_width  TERMINAL_WIDTH
 204 #endif
 205
 206         if (argc > 1 && **(argv + 1) == '-')
 207                 show_usage();
 208
 209         /* open device */
 210         fd = open(device, O_RDONLY);
 211         if (fd < 0)
 212                 perror_msg_and_die( "open failed for `%s'", device);
 213
 214         /* Find out how many processes there are */
 215         if (ioctl (fd, DEVPS_GET_NUM_PIDS, &num_pids)<0)
 216                 perror_msg_and_die( "\nDEVPS_GET_PID_LIST");
 217
 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.  */
 222         pid_array = (pid_t*) xcalloc( num_pids+10, sizeof(pid_t));
 223         pid_array[0] = num_pids+10;
 224
 225         /* Now grab the pid list */
 226         if (ioctl (fd, DEVPS_GET_PID_LIST, pid_array)<0)
 227                 perror_msg_and_die("\nDEVPS_GET_PID_LIST");
 228
 229 #ifdef CONFIG_FEATURE_AUTOWIDTH
 230                 ioctl(fileno(stdout), TIOCGWINSZ, &win);
 231                 if (win.ws_col > 0)
 232                         terminal_width = win.ws_col - 1;
 233 #endif
 234
 235         /* Print up a ps listing */
 236         printf("  PID  Uid     Stat Command\n");
 237
 238         for (i=1; i<pid_array[0] ; i++) {
 239             info.pid = pid_array[i];
 240
 241             if (ioctl (fd, DEVPS_GET_PID_INFO, &info)<0)
 242                         perror_msg_and_die("\nDEVPS_GET_PID_INFO");
 243
 244                 /* Make some adjustments as needed */
 245                 my_getpwuid(uidName, info.euid);
 246                 if (*uidName == '\0')
 247                         sprintf(uidName, "%ld", info.euid);
 248
 249                 if(p.vmsize == 0)
 250                         len = printf("%5d %-8s        %c    ", p.pid, uidName, p.state);
 251                 else
 252                         len = printf("%5d %-8s %6d %c    ", p.pid, uidName, p.vmsize, p.state);
 253                 if (strlen(info.command_line) > 1) {
 254                         for( j=0; j<(sizeof(info.command_line)-1) && j < (terminal_width-len); j++) {
 255                                 if (*(info.command_line+j) == '\0' && *(info.command_line+j+1) != '\0') {
 256                                         *(info.command_line+j) = ' ';
 257                                 }
 258                         }
 259                         *(info.command_line+j) = '\0';
 260                         puts(info.command_line);
 261                 } else {
 262                         printf("[%s]\n", info.name);
 263                 }
 264         }
 265
 266         /* Free memory */
 267         free( pid_array);
 268
 269         /* close device */
 270         if (close (fd) != 0)
 271                 perror_msg_and_die("close failed for `%s'", device);
 272
 273         exit (0);
 274 }
 275
 276 #endif /* CONFIG_FEATURE_USE_DEVPS_PATCH */
 277