Code

Attempt at detecting 64bit compilation flags between gcc, old, and new versions of...
[nagiosplug.git] / plugins-root / pst3.c
1 /*****************************************************************************
2
3 * pst3
4
5 * License: GPL
6 * Copyright (c) 2008 Nagios Plugin 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 *****************************************************************************/
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>
54 /*
55  *  Constants
56  */
58 #define PROC_DIR  "/proc"
59 #define ARGS            30
61 /*
62  *  Globals
63  */
65 static char *        szProg;
67 /*
68  *  Prototypes
69  */
70 void usage();
72 /*----------------------------------------------------------------------------*/
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;
82   /* Set our program name global */
83   if ((szProg = strrchr(argv[0], '/')) != NULL)
84     szProg++;
85   else
86     szProg = argv[0];
88   /* if given any parameters, print out help */
89   if(argc > 1) {
90     (void)usage();
91     exit(1);
92   }
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   }
101   if ((procdir = opendir(PROC_DIR)) == NULL) {
102     fprintf(stderr, "%s: cannot open PROC directory %s\n", szProg, PROC_DIR);
103     exit(1);
104   }
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   );
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;
133     if(proc->d_name[0] == '.')
134       continue;
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;
142     if((as_fd = open(as_name, O_RDONLY)) == -1)
143       continue;
145     if(read(ps_fd, &psinfo, sizeof(psinfo)) != sizeof(psinfo)) {
146       int err = errno;
147       close(ps_fd);
148       close(as_fd);
149       if(err == EAGAIN) goto try_again;
150       if(err != ENOENT)
151         fprintf(stderr, "%s: read() on %s: %s\n", szProg,
152           ps_name, strerror(err));
153       continue;
154     }
155     close(ps_fd);
157     /* system process, ignore since the previous version did */
158     if(
159       psinfo.pr_nlwp == 0 ||
160       strcmp(psinfo.pr_lwp.pr_clname, "SYS") == 0
161     ) {
162       continue;
163     }
165     /* get the procname to match previous versions */
166     procname = strdup(psinfo.pr_psargs);
167     if((ptr = strchr(procname, ' ')) != NULL)
168         *ptr = '\0';
169     if((ptr = strrchr(procname, '/')) != NULL)
170         ptr++;
171     else
172         ptr = procname;
174     /*
175      * print out what we currently know
176      */
177     printf("%c %5d %5d %5d %6lu %6lu %4.1f %s ",
178       psinfo.pr_lwp.pr_sname,
179       psinfo.pr_euid,
180       psinfo.pr_pid,
181       psinfo.pr_ppid,
182       psinfo.pr_size,
183       psinfo.pr_rssize,
184       ((float)(psinfo.pr_pctcpu) / 0x8000 * 100.0),
185       ptr
186     );
187     free(procname);
189     /*
190      * and now for the command line stuff
191      */
193     args_addr = psinfo.pr_argv;
194     args_count = psinfo.pr_argc;
195     args_vecs = malloc(args_count * sizeof(uintptr_t));
197     if(psinfo.pr_dmodel == PR_MODEL_NATIVE) {
198       /* this process matches target process */
199       pread(as_fd,args_vecs, args_count * sizeof(uintptr_t),
200         args_addr);
201     } else {
202       /* this process is 64bit, target process is 32 bit*/
203       caddr32_t *args_vecs32 = (caddr32_t *)args_vecs;
204       pread(as_fd,args_vecs32,args_count * sizeof(caddr32_t),
205         args_addr);
206       for (i=args_count-1;i>=0;--i)
207         args_vecs[i]=args_vecs32[i];
208     }
210     /*
211      * now read in the args - if what we read in fills buffer
212      * resize buffer and reread that bit again
213      */
214     argslen=ARGS;
215     args=malloc(argslen+1);
216     for(i=0;i<args_count;i++) {
217       memset(args,'\0',argslen+1);
218       if(pread(as_fd, args, argslen, args_vecs[i]) <= 0) {
219         break;
220       }
221       args[argslen]='\0';
222       if(strlen(args) == argslen){
223         argslen += ARGS;
224         args = realloc(args, argslen + 1);
225         i--;
226         continue;
227       }
228       printf(" %s", args);
229     }
230     free(args_vecs);
231     free(args);
232     close(as_fd);
233     printf("\n");
234   }
236   (void) closedir(procdir);
238   return (0);
241 /*----------------------------------------------------------------------------*/
243 void usage() {
244   printf("%s: Help output\n\n", szProg);
245   printf("If this program is given any arguments, this help is displayed.\n");
246   printf("This command is used to print out the full command line for all\n");
247   printf("running processes because /usr/bin/ps is limited to 80 chars and\n");
248   printf("/usr/ucb/ps can merge columns together.\n\n");
249   printf("Columns are:\n");
250   printf("\tS        - State of process - see 'ps' man page\n");
251   printf("\tUID      - UID of the process owner\n");
252   printf("\tPID      - PID of the process\n");
253   printf("\tPPID     - PID of the parent process\n");
254   printf("\tVSZ      - Virtual memory usage (kilobytes)\n");
255   printf("\tRSS      - Real memory usage (kilobytes)\n");
256   printf("\t%%CPU     - CPU usage\n");
257   printf("\tCOMMAND  - Command being run\n");
258   printf("\tARGS     - Full command line with arguements\n");
259   return;