ps.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini ps implementation(s) for busybox
   4  *
   5  * Copyright (C) 1999,2000,2001 by Lineo, inc.
   6  * Written by Erik Andersen <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 <stdio.h>
  32 #include <stdlib.h>
  33 #include <unistd.h>
  34 #include <dirent.h>
  35 #include <errno.h>
  36 #include <fcntl.h>
  37 #include <ctype.h>
  38 #include <string.h>
  39 #include <sys/ioctl.h>
  40 #include "busybox.h"
  41
  42 static const int TERMINAL_WIDTH = 79;      /* not 80 in case terminal has linefold bug */
  43
  44
  45
  46 #if ! defined BB_FEATURE_USE_DEVPS_PATCH
  47
  48 /* The following is the first ps implementation --
  49  * the one using the /proc virtual filesystem.
  50  */
  51
  52 typedef struct proc_s {
  53         char
  54          cmd[16];                                       /* basename of executable file in call to exec(2) */
  55         int
  56          ruid,                                          /* 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) */
  61 } proc_t;
  62
  63
  64
  65 static int file2str(char *filename, char *ret, int cap)
  66 {
  67         int fd, num_read;
  68
  69         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;
  76 }
  77
  78
  79 static void parse_proc_status(char *S, proc_t * P)
  80 {
  81         char *tmp;
  82
  83         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);
  90
  91         tmp = strstr(S, "Pid:");
  92         if (tmp)
  93                 sscanf(tmp, "Pid:\t%d\n" "PPid:\t%d\n", &P->pid, &P->ppid);
  94         else
  95                 error_msg("Internal error!");
  96
  97         /* For busybox, ignoring effective, saved, etc. */
  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
 105 }
 106
 107 extern int ps_main(int argc, char **argv)
 108 {
 109         proc_t p;
 110         DIR *dir;
 111         FILE *file;
 112         struct dirent *entry;
 113         char path[32], sbuf[512];
 114         char uidName[9];
 115         int len, i, c;
 116 #ifdef BB_FEATURE_AUTOWIDTH
 117         struct winsize win = { 0, 0, 0, 0 };
 118         int terminal_width = TERMINAL_WIDTH;
 119 #else
 120 #define terminal_width  TERMINAL_WIDTH
 121 #endif
 122
 123
 124
 125         dir = opendir("/proc");
 126         if (!dir)
 127                 error_msg_and_die("Can't open /proc");
 128
 129 #ifdef BB_FEATURE_AUTOWIDTH
 130                 ioctl(fileno(stdout), TIOCGWINSZ, &win);
 131                 if (win.ws_col > 0)
 132                         terminal_width = win.ws_col - 1;
 133 #endif
 134
 135         printf("  PID  Uid     Stat Command\n");
 136         while ((entry = readdir(dir)) != NULL) {
 137                 if (!isdigit(*entry->d_name))
 138                         continue;
 139                 sprintf(path, "/proc/%s/status", entry->d_name);
 140                 if ((file2str(path, sbuf, sizeof sbuf)) != -1) {
 141                         parse_proc_status(sbuf, &p);
 142                 }
 143
 144                 /* Make some adjustments as needed */
 145                 my_getpwuid(uidName, p.ruid);
 146                 if (*uidName == '\0')
 147                         sprintf(uidName, "%d", p.ruid);
 148
 149                 sprintf(path, "/proc/%s/cmdline", entry->d_name);
 150                 file = fopen(path, "r");
 151                 if (file == NULL)
 152                         continue;
 153                 i = 0;
 154                 len = printf("%5d %-8s %c    ", p.pid, uidName, p.state);
 155                 while (((c = getc(file)) != EOF) && (i < (terminal_width-len))) {
 156                         i++;
 157                         if (c == '\0')
 158                                 c = ' ';
 159                         putc(c, stdout);
 160                 }
 161                 fclose(file);
 162                 if (i == 0)
 163                         printf("[%s]", p.cmd);
 164                 putchar('\n');
 165         }
 166         closedir(dir);
 167         return EXIT_SUCCESS;
 168 }
 169
 170
 171 #else /* BB_FEATURE_USE_DEVPS_PATCH */
 172
 173
 174 /* The following is the second ps implementation --
 175  * this one uses the nifty new devps kernel device.
 176  */
 177
 178 #include <linux/devps.h> /* For Erik's nifty devps device driver */
 179
 180
 181 extern int ps_main(int argc, char **argv)
 182 {
 183         char device[] = "/dev/ps";
 184         int i, j, len, fd;
 185         pid_t num_pids;
 186         pid_t* pid_array = NULL;
 187         struct pid_info info;
 188         char uidName[9];
 189 #ifdef BB_FEATURE_AUTOWIDTH
 190         struct winsize win = { 0, 0, 0, 0 };
 191         int terminal_width = TERMINAL_WIDTH;
 192 #else
 193 #define terminal_width  TERMINAL_WIDTH
 194 #endif
 195
 196         if (argc > 1 && **(argv + 1) == '-')
 197                 show_usage();
 198
 199         /* open device */
 200         fd = open(device, O_RDONLY);
 201         if (fd < 0)
 202                 perror_msg_and_die( "open failed for `%s'", device);
 203
 204         /* Find out how many processes there are */
 205         if (ioctl (fd, DEVPS_GET_NUM_PIDS, &num_pids)<0)
 206                 perror_msg_and_die( "\nDEVPS_GET_PID_LIST");
 207
 208         /* Allocate some memory -- grab a few extras just in case
 209          * some new processes start up while we wait. The kernel will
 210          * just ignore any extras if we give it too many, and will trunc.
 211          * the list if we give it too few.  */
 212         pid_array = (pid_t*) xcalloc( num_pids+10, sizeof(pid_t));
 213         pid_array[0] = num_pids+10;
 214
 215         /* Now grab the pid list */
 216         if (ioctl (fd, DEVPS_GET_PID_LIST, pid_array)<0)
 217                 perror_msg_and_die("\nDEVPS_GET_PID_LIST");
 218
 219 #ifdef BB_FEATURE_AUTOWIDTH
 220                 ioctl(fileno(stdout), TIOCGWINSZ, &win);
 221                 if (win.ws_col > 0)
 222                         terminal_width = win.ws_col - 1;
 223 #endif
 224
 225         /* Print up a ps listing */
 226         printf("  PID  Uid     Stat Command\n");
 227
 228         for (i=1; i<pid_array[0] ; i++) {
 229             info.pid = pid_array[i];
 230
 231             if (ioctl (fd, DEVPS_GET_PID_INFO, &info)<0)
 232                         perror_msg_and_die("\nDEVPS_GET_PID_INFO");
 233
 234                 /* Make some adjustments as needed */
 235                 my_getpwuid(uidName, info.euid);
 236                 if (*uidName == '\0')
 237                         sprintf(uidName, "%ld", info.euid);
 238
 239                 len = printf("%5d %-8s %c    ", info.pid, uidName, info.state);
 240
 241                 if (strlen(info.command_line) > 1) {
 242                         for( j=0; j<(sizeof(info.command_line)-1) && j < (terminal_width-len); j++) {
 243                                 if (*(info.command_line+j) == '\0' && *(info.command_line+j+1) != '\0') {
 244                                         *(info.command_line+j) = ' ';
 245                                 }
 246                         }
 247                         *(info.command_line+j) = '\0';
 248                         puts(info.command_line);
 249                 } else {
 250                         printf("[%s]\n", info.name);
 251                 }
 252         }
 253
 254         /* Free memory */
 255         free( pid_array);
 256
 257         /* close device */
 258         if (close (fd) != 0)
 259                 perror_msg_and_die("close failed for `%s'", device);
 260
 261         exit (0);
 262 }
 263
 264 #endif /* BB_FEATURE_USE_DEVPS_PATCH */
 265