Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 1 | /* vi: set sw=4 ts=4: */ |
Eric Andersen | ef8b6c7 | 1999-10-20 08:05:35 +0000 | [diff] [blame] | 2 | /* |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 3 | * Mini ps implementation(s) for busybox |
| 4 | * |
Eric Andersen | bdfd0d7 | 2001-10-24 05:00:29 +0000 | [diff] [blame] | 5 | * Copyright (C) 1999,2000 by Lineo, inc. and Erik Andersen |
| 6 | * Copyright (C) 1999,2000,2001 by Erik Andersen <andersee@debian.org> |
Eric Andersen | ef8b6c7 | 1999-10-20 08:05:35 +0000 | [diff] [blame] | 7 | * |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 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 |
Eric Andersen | bdfd0d7 | 2001-10-24 05:00:29 +0000 | [diff] [blame] | 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+). |
Eric Andersen | ef8b6c7 | 1999-10-20 08:05:35 +0000 | [diff] [blame] | 27 | */ |
| 28 | |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 29 | #include <stdio.h> |
Eric Andersen | ed3ef50 | 2001-01-27 08:24:39 +0000 | [diff] [blame] | 30 | #include <stdlib.h> |
Eric Andersen | ef8b6c7 | 1999-10-20 08:05:35 +0000 | [diff] [blame] | 31 | #include <unistd.h> |
| 32 | #include <dirent.h> |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 33 | #include <errno.h> |
Eric Andersen | d23f9ba | 1999-10-20 19:18:15 +0000 | [diff] [blame] | 34 | #include <fcntl.h> |
| 35 | #include <ctype.h> |
Eric Andersen | ed3ef50 | 2001-01-27 08:24:39 +0000 | [diff] [blame] | 36 | #include <string.h> |
Eric Andersen | 8d79ce8 | 2001-07-22 23:00:15 +0000 | [diff] [blame] | 37 | #include <termios.h> |
Erik Andersen | 2ac2fae | 2000-03-07 23:32:17 +0000 | [diff] [blame] | 38 | #include <sys/ioctl.h> |
Eric Andersen | cbe31da | 2001-02-20 06:14:08 +0000 | [diff] [blame] | 39 | #include "busybox.h" |
Erik Andersen | 2ac2fae | 2000-03-07 23:32:17 +0000 | [diff] [blame] | 40 | |
Mark Whitley | 59ab025 | 2001-01-23 22:30:04 +0000 | [diff] [blame] | 41 | static const int TERMINAL_WIDTH = 79; /* not 80 in case terminal has linefold bug */ |
Eric Andersen | 86ab8a3 | 2000-06-02 03:21:42 +0000 | [diff] [blame] | 42 | |
| 43 | |
Eric Andersen | d23f9ba | 1999-10-20 19:18:15 +0000 | [diff] [blame] | 44 | |
Eric Andersen | bdfd0d7 | 2001-10-24 05:00:29 +0000 | [diff] [blame] | 45 | #if ! defined CONFIG_FEATURE_USE_DEVPS_PATCH |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 46 | |
Eric Andersen | ef8b6c7 | 1999-10-20 08:05:35 +0000 | [diff] [blame] | 47 | extern int ps_main(int argc, char **argv) |
| 48 | { |
Eric Andersen | 44608e9 | 2002-10-22 12:21:15 +0000 | [diff] [blame] | 49 | procps_status_t * p; |
| 50 | int i, len; |
Eric Andersen | bdfd0d7 | 2001-10-24 05:00:29 +0000 | [diff] [blame] | 51 | #ifdef CONFIG_FEATURE_AUTOWIDTH |
Eric Andersen | fad04fd | 2000-07-14 06:49:52 +0000 | [diff] [blame] | 52 | struct winsize win = { 0, 0, 0, 0 }; |
Eric Andersen | 86ab8a3 | 2000-06-02 03:21:42 +0000 | [diff] [blame] | 53 | int terminal_width = TERMINAL_WIDTH; |
Erik Andersen | 2ac2fae | 2000-03-07 23:32:17 +0000 | [diff] [blame] | 54 | #else |
Eric Andersen | 86ab8a3 | 2000-06-02 03:21:42 +0000 | [diff] [blame] | 55 | #define terminal_width TERMINAL_WIDTH |
Erik Andersen | 2ac2fae | 2000-03-07 23:32:17 +0000 | [diff] [blame] | 56 | #endif |
| 57 | |
| 58 | |
Eric Andersen | bdfd0d7 | 2001-10-24 05:00:29 +0000 | [diff] [blame] | 59 | #ifdef CONFIG_FEATURE_AUTOWIDTH |
Erik Andersen | 2ac2fae | 2000-03-07 23:32:17 +0000 | [diff] [blame] | 60 | ioctl(fileno(stdout), TIOCGWINSZ, &win); |
| 61 | if (win.ws_col > 0) |
| 62 | terminal_width = win.ws_col - 1; |
| 63 | #endif |
| 64 | |
Eric Andersen | 1372780 | 2002-04-27 06:06:11 +0000 | [diff] [blame] | 65 | printf(" PID Uid VmSize Stat Command\n"); |
Eric Andersen | 44608e9 | 2002-10-22 12:21:15 +0000 | [diff] [blame] | 66 | while ((p = procps_scan(1)) != 0) { |
| 67 | char *namecmd = p->cmd; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 68 | |
Eric Andersen | 44608e9 | 2002-10-22 12:21:15 +0000 | [diff] [blame] | 69 | if(p->rss == 0) |
| 70 | len = printf("%5d %-8s %s ", p->pid, p->user, p->state); |
Eric Andersen | 1372780 | 2002-04-27 06:06:11 +0000 | [diff] [blame] | 71 | else |
Eric Andersen | 44608e9 | 2002-10-22 12:21:15 +0000 | [diff] [blame] | 72 | len = printf("%5d %-8s %6ld %s ", p->pid, p->user, p->rss, p->state); |
| 73 | i = terminal_width-len; |
| 74 | |
| 75 | if(namecmd != 0 && namecmd[0] != 0) { |
| 76 | if(i < 0) |
| 77 | i = 0; |
| 78 | if(strlen(namecmd) > i) |
| 79 | namecmd[i] = 0; |
| 80 | printf("%s\n", namecmd); |
| 81 | } else { |
| 82 | namecmd = p->short_cmd; |
| 83 | if(i < 2) |
| 84 | i = 2; |
| 85 | if(strlen(namecmd) > (i-2)) |
| 86 | namecmd[i-2] = 0; |
| 87 | printf("[%s]\n", namecmd); |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 88 | } |
Eric Andersen | 44608e9 | 2002-10-22 12:21:15 +0000 | [diff] [blame] | 89 | free(p->cmd); |
Eric Andersen | d23f9ba | 1999-10-20 19:18:15 +0000 | [diff] [blame] | 90 | } |
Matt Kraai | 3e856ce | 2000-12-01 02:55:13 +0000 | [diff] [blame] | 91 | return EXIT_SUCCESS; |
Eric Andersen | ef8b6c7 | 1999-10-20 08:05:35 +0000 | [diff] [blame] | 92 | } |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 93 | |
| 94 | |
Eric Andersen | bdfd0d7 | 2001-10-24 05:00:29 +0000 | [diff] [blame] | 95 | #else /* CONFIG_FEATURE_USE_DEVPS_PATCH */ |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 96 | |
| 97 | |
| 98 | /* The following is the second ps implementation -- |
| 99 | * this one uses the nifty new devps kernel device. |
| 100 | */ |
| 101 | |
Eric Andersen | c674d70 | 2000-07-10 22:57:14 +0000 | [diff] [blame] | 102 | #include <linux/devps.h> /* For Erik's nifty devps device driver */ |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 103 | |
| 104 | |
| 105 | extern int ps_main(int argc, char **argv) |
| 106 | { |
| 107 | char device[] = "/dev/ps"; |
Erik Andersen | 2ac2fae | 2000-03-07 23:32:17 +0000 | [diff] [blame] | 108 | int i, j, len, fd; |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 109 | pid_t num_pids; |
| 110 | pid_t* pid_array = NULL; |
| 111 | struct pid_info info; |
Eric Andersen | bd193a4 | 2000-12-13 01:52:39 +0000 | [diff] [blame] | 112 | char uidName[9]; |
Eric Andersen | bdfd0d7 | 2001-10-24 05:00:29 +0000 | [diff] [blame] | 113 | #ifdef CONFIG_FEATURE_AUTOWIDTH |
Pavel Roskin | f626dcb | 2000-07-14 15:55:41 +0000 | [diff] [blame] | 114 | struct winsize win = { 0, 0, 0, 0 }; |
Eric Andersen | 86ab8a3 | 2000-06-02 03:21:42 +0000 | [diff] [blame] | 115 | int terminal_width = TERMINAL_WIDTH; |
Erik Andersen | 2ac2fae | 2000-03-07 23:32:17 +0000 | [diff] [blame] | 116 | #else |
Eric Andersen | 86ab8a3 | 2000-06-02 03:21:42 +0000 | [diff] [blame] | 117 | #define terminal_width TERMINAL_WIDTH |
Erik Andersen | 2ac2fae | 2000-03-07 23:32:17 +0000 | [diff] [blame] | 118 | #endif |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 119 | |
| 120 | if (argc > 1 && **(argv + 1) == '-') |
Eric Andersen | 67991cf | 2001-02-14 21:23:06 +0000 | [diff] [blame] | 121 | show_usage(); |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 122 | |
| 123 | /* open device */ |
| 124 | fd = open(device, O_RDONLY); |
| 125 | if (fd < 0) |
Matt Kraai | 1fa1ade | 2000-12-18 03:57:16 +0000 | [diff] [blame] | 126 | perror_msg_and_die( "open failed for `%s'", device); |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 127 | |
| 128 | /* Find out how many processes there are */ |
| 129 | if (ioctl (fd, DEVPS_GET_NUM_PIDS, &num_pids)<0) |
Matt Kraai | 1fa1ade | 2000-12-18 03:57:16 +0000 | [diff] [blame] | 130 | perror_msg_and_die( "\nDEVPS_GET_PID_LIST"); |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 131 | |
| 132 | /* Allocate some memory -- grab a few extras just in case |
| 133 | * some new processes start up while we wait. The kernel will |
| 134 | * just ignore any extras if we give it too many, and will trunc. |
| 135 | * the list if we give it too few. */ |
Matt Kraai | 322ae93 | 2000-09-13 02:46:14 +0000 | [diff] [blame] | 136 | pid_array = (pid_t*) xcalloc( num_pids+10, sizeof(pid_t)); |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 137 | pid_array[0] = num_pids+10; |
| 138 | |
| 139 | /* Now grab the pid list */ |
| 140 | if (ioctl (fd, DEVPS_GET_PID_LIST, pid_array)<0) |
Matt Kraai | 1fa1ade | 2000-12-18 03:57:16 +0000 | [diff] [blame] | 141 | perror_msg_and_die("\nDEVPS_GET_PID_LIST"); |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 142 | |
Eric Andersen | bdfd0d7 | 2001-10-24 05:00:29 +0000 | [diff] [blame] | 143 | #ifdef CONFIG_FEATURE_AUTOWIDTH |
Erik Andersen | 2ac2fae | 2000-03-07 23:32:17 +0000 | [diff] [blame] | 144 | ioctl(fileno(stdout), TIOCGWINSZ, &win); |
| 145 | if (win.ws_col > 0) |
| 146 | terminal_width = win.ws_col - 1; |
| 147 | #endif |
| 148 | |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 149 | /* Print up a ps listing */ |
Eric Andersen | 0392b86 | 2001-06-26 23:11:44 +0000 | [diff] [blame] | 150 | printf(" PID Uid Stat Command\n"); |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 151 | |
| 152 | for (i=1; i<pid_array[0] ; i++) { |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 153 | info.pid = pid_array[i]; |
| 154 | |
| 155 | if (ioctl (fd, DEVPS_GET_PID_INFO, &info)<0) |
Matt Kraai | 1fa1ade | 2000-12-18 03:57:16 +0000 | [diff] [blame] | 156 | perror_msg_and_die("\nDEVPS_GET_PID_INFO"); |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 157 | |
| 158 | /* Make some adjustments as needed */ |
| 159 | my_getpwuid(uidName, info.euid); |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 160 | |
Eric Andersen | 1372780 | 2002-04-27 06:06:11 +0000 | [diff] [blame] | 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); |
Erik Andersen | 2ac2fae | 2000-03-07 23:32:17 +0000 | [diff] [blame] | 165 | if (strlen(info.command_line) > 1) { |
| 166 | for( j=0; j<(sizeof(info.command_line)-1) && j < (terminal_width-len); j++) { |
| 167 | if (*(info.command_line+j) == '\0' && *(info.command_line+j+1) != '\0') { |
| 168 | *(info.command_line+j) = ' '; |
| 169 | } |
| 170 | } |
| 171 | *(info.command_line+j) = '\0'; |
Matt Kraai | 12f417e | 2001-01-18 02:57:08 +0000 | [diff] [blame] | 172 | puts(info.command_line); |
Erik Andersen | 2ac2fae | 2000-03-07 23:32:17 +0000 | [diff] [blame] | 173 | } else { |
Matt Kraai | 12f417e | 2001-01-18 02:57:08 +0000 | [diff] [blame] | 174 | printf("[%s]\n", info.name); |
Erik Andersen | 2ac2fae | 2000-03-07 23:32:17 +0000 | [diff] [blame] | 175 | } |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 176 | } |
| 177 | |
| 178 | /* Free memory */ |
| 179 | free( pid_array); |
| 180 | |
| 181 | /* close device */ |
| 182 | if (close (fd) != 0) |
Matt Kraai | 1fa1ade | 2000-12-18 03:57:16 +0000 | [diff] [blame] | 183 | perror_msg_and_die("close failed for `%s'", device); |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 184 | |
| 185 | exit (0); |
| 186 | } |
| 187 | |
Eric Andersen | bdfd0d7 | 2001-10-24 05:00:29 +0000 | [diff] [blame] | 188 | #endif /* CONFIG_FEATURE_USE_DEVPS_PATCH */ |
Erik Andersen | 246cc6d | 2000-03-07 07:41:42 +0000 | [diff] [blame] | 189 | |