plugins-root/pst3.c

   1 /*****************************************************************************
   2 *
   3 * pst3
   4 *
   5 * License: GPL
   6 * Copyright (c) 2008 Monitoring Plugins Development Team
   7 *
   8 * Description:
   9 *
  10 * This file contains the pst3 executable. This is a replacement ps command
  11 * for Solaris to get output which provides a long argument listing, which
  12 * is not possible with the standard ps command (due to truncation). /usr/ucb/ps
  13 * also has issues where some fields run into each other.
  14 *
  15 * This executable works by reading process address structures, so needs
  16 * to be executed as root
  17 *
  18 * Originally written by R.W.Ingraham
  19 * Rewritten by Duncan Ferguson (Altinity Ltd, June 2008)
  20 *   The rewrite was necessary as /dev/kmem is not available within
  21 *   non-global zones on Solaris 10
  22 *
  23 *   Details for rewrite came from
  24 *    source of /usr/ucb/ps on Solaris:
  25 *     http://cvs.opensolaris.org/source/xref/onnv/onnv-gate/usr/src/ucbcmd/ps/ps.c#argvoff
  26 *    usenet group posting
  27 *     http://groups.google.com/group/comp.unix.solaris/tree/browse_frm/month/2001-09/bfa40c08bac819a2?rnum=141&_done=%2Fgroup%2Fcomp.unix.solaris%2Fbrowse_frm%2Fmonth%2F2001-09%3F
  28 *
  29 * This program is free software: you can redistribute it and/or modify
  30 * it under the terms of the GNU General Public License as published by
  31 * the Free Software Foundation, either version 3 of the License, or
  32 * (at your option) any later version.
  33 *
  34 * This program is distributed in the hope that it will be useful,
  35 * but WITHOUT ANY WARRANTY; without even the implied warranty of
  36 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  37 * GNU General Public License for more details.
  38 *
  39 * You should have received a copy of the GNU General Public License
  40 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  41 *
  42 *****************************************************************************/
  43
  44 #include <stdio.h>
  45 #include <stdlib.h>
  46 #include <string.h>
  47 #include <unistd.h>
  48 #include <dirent.h>
  49 #include <errno.h>
  50 #include <fcntl.h>
  51 #include <procfs.h>
  52 #include <sys/types32.h>
  53
  54 /*
  55  *  Constants
  56  */
  57
  58 #define PROC_DIR  "/proc"
  59 #define ARGS            30
  60
  61 /*
  62  *  Globals
  63  */
  64
  65 static char *        szProg;
  66
  67 /*
  68  *  Prototypes
  69  */
  70 void usage();
  71
  72 /*----------------------------------------------------------------------------*/
  73
  74 int main (int argc, char **argv)
  75 {
  76   DIR *procdir;
  77   struct dirent *proc;
  78   char ps_name[ARGS];
  79   char as_name[ARGS];
  80   psinfo_t psinfo;
  81
  82   /* Set our program name global */
  83   if ((szProg = strrchr(argv[0], '/')) != NULL)
  84     szProg++;
  85   else
  86     szProg = argv[0];
  87
  88   /* if given any parameters, print out help */
  89   if(argc > 1) {
  90     (void)usage();
  91     exit(1);
  92   }
  93
  94   /* Make sure that our euid is root */
  95   if (geteuid() != 0)
  96   {
  97     fprintf(stderr, "%s: This program can only be run by the root user!\n", szProg);
  98     exit(1);
  99   }
 100
 101   if ((procdir = opendir(PROC_DIR)) == NULL) {
 102     fprintf(stderr, "%s: cannot open PROC directory %s\n", szProg, PROC_DIR);
 103     exit(1);
 104   }
 105
 106   /* Display column headings */
 107   printf("%c %5s %5s %5s %6s %6s %4s %s %s\n",
 108     'S',
 109     "UID",
 110     "PID",
 111     "PPID",
 112     "VSZ",
 113     "RSS",
 114     "%CPU",
 115     "COMMAND",
 116     "ARGS"
 117   );
 118
 119   /* Zip through all of the process entries */
 120   while((proc = readdir(procdir))) {
 121     int ps_fd;
 122     int as_fd;
 123     off_t argoff;
 124     int i;
 125     char *args;
 126     char *procname;
 127     char *ptr;
 128     int argslen;
 129     uintptr_t args_addr;;
 130     uintptr_t *args_vecs;;
 131     int args_count;
 132
 133     if(proc->d_name[0] == '.')
 134       continue;
 135
 136     sprintf(ps_name,"%s/%s/%s",PROC_DIR,proc->d_name,"psinfo");
 137     sprintf(as_name,"%s/%s/%s",PROC_DIR,proc->d_name,"as");
 138 try_again:
 139     if((ps_fd = open(ps_name, O_RDONLY)) == -1)
 140       continue;
 141
 142     if((as_fd = open(as_name, O_RDONLY)) == -1) {
 143       close(ps_fd);
 144       continue;
 145     }
 146
 147     if(read(ps_fd, &psinfo, sizeof(psinfo)) != sizeof(psinfo)) {
 148       int err = errno;
 149       close(ps_fd);
 150       close(as_fd);
 151       if(err == EAGAIN) goto try_again;
 152       if(err != ENOENT)
 153         fprintf(stderr, "%s: read() on %s: %s\n", szProg,
 154           ps_name, strerror(err));
 155       continue;
 156     }
 157     close(ps_fd);
 158
 159     /* system process, ignore since the previous version did */
 160     if(
 161       psinfo.pr_nlwp == 0 ||
 162       strcmp(psinfo.pr_lwp.pr_clname, "SYS") == 0
 163     ) {
 164       continue;
 165     }
 166
 167     /* get the procname to match previous versions */
 168     procname = strdup(psinfo.pr_psargs);
 169     if((ptr = strchr(procname, ' ')) != NULL)
 170         *ptr = '\0';
 171     if((ptr = strrchr(procname, '/')) != NULL)
 172         ptr++;
 173     else
 174         ptr = procname;
 175
 176     /*
 177      * print out what we currently know
 178      */
 179     printf("%c %5d %5d %5d %6lu %6lu %4.1f %s ",
 180       psinfo.pr_lwp.pr_sname,
 181       psinfo.pr_euid,
 182       psinfo.pr_pid,
 183       psinfo.pr_ppid,
 184       psinfo.pr_size,
 185       psinfo.pr_rssize,
 186       ((float)(psinfo.pr_pctcpu) / 0x8000 * 100.0),
 187       ptr
 188     );
 189     free(procname);
 190
 191     /*
 192      * and now for the command line stuff
 193      */
 194
 195     args_addr = psinfo.pr_argv;
 196     args_count = psinfo.pr_argc;
 197     args_vecs = malloc(args_count * sizeof(uintptr_t));
 198
 199     if(psinfo.pr_dmodel == PR_MODEL_NATIVE) {
 200       /* this process matches target process */
 201       pread(as_fd,args_vecs, args_count * sizeof(uintptr_t),
 202         args_addr);
 203     } else {
 204       /* this process is 64bit, target process is 32 bit*/
 205       caddr32_t *args_vecs32 = (caddr32_t *)args_vecs;
 206       pread(as_fd,args_vecs32,args_count * sizeof(caddr32_t),
 207         args_addr);
 208       for (i=args_count-1;i>=0;--i)
 209         args_vecs[i]=args_vecs32[i];
 210     }
 211
 212     /*
 213      * now read in the args - if what we read in fills buffer
 214      * resize buffer and reread that bit again
 215      */
 216     argslen=ARGS;
 217     args=malloc(argslen+1);
 218     for(i=0;i<args_count;i++) {
 219       memset(args,'\0',argslen+1);
 220       if(pread(as_fd, args, argslen, args_vecs[i]) <= 0) {
 221         break;
 222       }
 223       args[argslen]='\0';
 224       if(strlen(args) == argslen){
 225         argslen += ARGS;
 226         args = realloc(args, argslen + 1);
 227         i--;
 228         continue;
 229       }
 230       printf(" %s", args);
 231     }
 232     free(args_vecs);
 233     free(args);
 234     close(as_fd);
 235     printf("\n");
 236   }
 237
 238   (void) closedir(procdir);
 239
 240   return (0);
 241 }
 242
 243 /*----------------------------------------------------------------------------*/
 244
 245 void usage() {
 246   printf("%s: Help output\n\n", szProg);
 247   printf("If this program is given any arguments, this help is displayed.\n");
 248   printf("This command is used to print out the full command line for all\n");
 249   printf("running processes because /usr/bin/ps is limited to 80 chars and\n");
 250   printf("/usr/ucb/ps can merge columns together.\n\n");
 251   printf("Columns are:\n");
 252   printf("\tS        - State of process - see 'ps' man page\n");
 253   printf("\tUID      - UID of the process owner\n");
 254   printf("\tPID      - PID of the process\n");
 255   printf("\tPPID     - PID of the parent process\n");
 256   printf("\tVSZ      - Virtual memory usage (kilobytes)\n");
 257   printf("\tRSS      - Real memory usage (kilobytes)\n");
 258   printf("\t%%CPU     - CPU usage\n");
 259   printf("\tCOMMAND  - Command being run\n");
 260   printf("\tARGS     - Full command line with arguments\n");
 261   return;
 262 }