procps/ps.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini ps implementation(s) for busybox
   4  *
   5  * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
   6  *
   7  * Licensed under the GPL v2, see the file LICENSE in this tarball.
   8  */
   9
  10 #include <stdio.h>
  11 #include <stdlib.h>
  12 #include <unistd.h>
  13 #include <dirent.h>
  14 #include <errno.h>
  15 #include <fcntl.h>
  16 #include <ctype.h>
  17 #include <string.h>
  18 #include <termios.h>
  19 #include <sys/ioctl.h>
  20 #include "busybox.h"
  21 #if ENABLE_SELINUX
  22 #include <selinux/selinux.h>  /* for is_selinux_enabled()  */
  23 #endif
  24
  25 extern int ps_main(int argc, char **argv)
  26 {
  27         procps_status_t * p;
  28         int i, len;
  29
  30 #if ENABLE_SELINUX
  31         int use_selinux = 0;
  32         security_context_t sid=NULL;
  33 #endif
  34
  35 #if ENABLE_FEATURE_PS_WIDE
  36         int terminal_width;
  37         int w_count = 0;
  38
  39         bb_opt_complementally="-:ww";
  40 #else
  41 # define terminal_width 79
  42 #endif
  43
  44 #if ENABLE_FEATURE_PS_WIDE || ENABLE_SELINUX
  45         /* handle arguments */
  46 #if ENABLE_FEATURE_PS_WIDE && ENABLE_SELINUX
  47         i = bb_getopt_ulflags(argc, argv, "wc", &w_count);
  48 #elif ENABLE_FEATURE_PS_WIDE && !ENABLE_SELINUX
  49         bb_getopt_ulflags(argc, argv, "w", &w_count);
  50 #else /* !ENABLE_FEATURE_PS_WIDE && ENABLE_SELINUX */
  51         i = bb_getopt_ulflags(argc, argv, "c");
  52 #endif
  53 #if ENABLE_FEATURE_PS_WIDE
  54         /* if w is given once, GNU ps sets the width to 132,
  55          * if w is given more than once, it is "unlimited"
  56          */
  57         if(w_count) {
  58                 terminal_width = (w_count==1) ? 132 : INT_MAX;
  59         } else {
  60                 get_terminal_width_height(1, &terminal_width, NULL);
  61                 /* Go one less... */
  62                 terminal_width--;
  63         }
  64 #endif
  65 #if ENABLE_SELINUX
  66         if ((i & (1+ENABLE_FEATURE_PS_WIDE)) && is_selinux_enabled())
  67                 use_selinux = 1;
  68 #endif
  69 #endif  /* ENABLE_FEATURE_PS_WIDE || ENABLE_SELINUX */
  70
  71 #if ENABLE_SELINUX
  72         if (use_selinux)
  73           printf("  PID Context                          Stat Command\n");
  74         else
  75 #endif
  76           printf("  PID  Uid     VmSize Stat Command\n");
  77
  78         while ((p = procps_scan(1)) != 0)  {
  79                 char *namecmd = p->cmd;
  80 #if ENABLE_SELINUX
  81                 if (use_selinux)
  82                   {
  83                         char sbuf[128];
  84                         len = sizeof(sbuf);
  85
  86                         if (is_selinux_enabled()) {
  87                           if (getpidcon(p->pid,&sid)<0)
  88                             sid=NULL;
  89                         }
  90
  91                         if (sid) {
  92                           /*  I assume sid initilized with NULL  */
  93                           len = strlen(sid)+1;
  94                           safe_strncpy(sbuf, sid, len);
  95                           freecon(sid);
  96                           sid=NULL;
  97                         }else {
  98                           safe_strncpy(sbuf, "unknown",7);
  99                         }
 100                         len = printf("%5d %-32s %s ", p->pid, sbuf, p->state);
 101                 }
 102                 else
 103 #endif
 104                   if(p->rss == 0)
 105                     len = printf("%5d %-8s        %s ", p->pid, p->user, p->state);
 106                   else
 107                     len = printf("%5d %-8s %6ld %s ", p->pid, p->user, p->rss, p->state);
 108
 109                 i = terminal_width-len;
 110
 111                 if(namecmd && namecmd[0]) {
 112                         if(i < 0)
 113                                 i = 0;
 114                         if(strlen(namecmd) > (size_t)i)
 115                                 namecmd[i] = 0;
 116                         printf("%s\n", namecmd);
 117                 } else {
 118                         namecmd = p->short_cmd;
 119                         if(i < 2)
 120                                 i = 2;
 121                         if(strlen(namecmd) > ((size_t)i-2))
 122                                 namecmd[i-2] = 0;
 123                         printf("[%s]\n", namecmd);
 124                 }
 125                 /* no check needed, but to make valgrind happy..  */
 126                 if (ENABLE_FEATURE_CLEAN_UP && p->cmd)
 127                         free(p->cmd);
 128         }
 129         return EXIT_SUCCESS;
 130 }