ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/src/libstatgrab/process_stats.c
(Generate patch)

Comparing projects/libstatgrab/src/libstatgrab/process_stats.c (file contents):
Revision 1.11 by pajs, Thu Oct 9 19:40:03 2003 UTC vs.
Revision 1.76 by tdb, Mon Nov 1 18:30:17 2004 UTC

# Line 1 | Line 1
1 < /*
2 < * i-scream central monitoring system
1 > /*
2 > * i-scream libstatgrab
3   * http://www.i-scream.org
4 < * Copyright (C) 2000-2003 i-scream
4 > * Copyright (C) 2000-2004 i-scream
5   *
6 < * This program is free software; you can redistribute it and/or
7 < * modify it under the terms of the GNU General Public License
8 < * as published by the Free Software Foundation; either version 2
9 < * of the License, or (at your option) any later version.
6 > * This library is free software; you can redistribute it and/or
7 > * modify it under the terms of the GNU Lesser General Public
8 > * License as published by the Free Software Foundation; either
9 > * version 2.1 of the License, or (at your option) any later version.
10   *
11 < * This program is distributed in the hope that it will be useful,
11 > * This library is distributed in the hope that it will be useful,
12   * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 < * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 < * GNU General Public License for more details.
13 > * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 > * Lesser General Public License for more details.
15   *
16 < * You should have received a copy of the GNU General Public License
17 < * along with this program; if not, write to the Free Software
18 < * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
16 > * You should have received a copy of the GNU Lesser General Public
17 > * License along with this library; if not, write to the Free Software
18 > * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
19 > * 02111-1307 USA
20 > *
21 > * $Id$
22   */
23  
24   #ifdef HAVE_CONFIG_H
# Line 23 | Line 26
26   #endif
27  
28   #include "statgrab.h"
29 + #include "tools.h"
30 + #include "vector.h"
31   #if defined(SOLARIS) || defined(LINUX)
32   #include <stdio.h>
33   #include <stdlib.h>
# Line 38 | Line 43
43   #define MAX_FILE_LENGTH PATH_MAX
44   #endif
45   #ifdef LINUX
46 < #include "tools.h"
47 < #include <linux/limits.h>
46 > #include <limits.h>
47 > #include <unistd.h>
48 > #include <sys/stat.h>
49 > #include <fcntl.h>
50   #define PROC_LOCATION "/proc"
51   #define MAX_FILE_LENGTH PATH_MAX
52   #endif
53 < #ifdef FREEBSD
54 < #include <kvm.h>
55 < #include <fcntl.h>
53 > #ifdef ALLBSD
54 > #include <errno.h>
55 > #include <stdlib.h>
56   #include <sys/param.h>
57   #include <sys/sysctl.h>
58 + #if defined(FREEBSD) || defined(DFBSD)
59   #include <sys/user.h>
60 + #else
61 + #include <sys/proc.h>
62   #endif
63 + #include <string.h>
64 + #include <paths.h>
65 + #include <fcntl.h>
66 + #include <limits.h>
67 + #if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
68 + #include <kvm.h>
69 + #endif
70 + #include <unistd.h>
71 + #ifdef NETBSD2
72 + #include <sys/lwp.h>
73 + #endif
74 + #endif
75 + #ifdef HPUX
76 + #include <sys/param.h>
77 + #include <sys/pstat.h>
78 + #include <unistd.h>
79 + #endif
80  
81 < process_stat_t *get_process_stats(){
81 > static void proc_state_init(sg_process_stats *s) {
82 >        s->process_name = NULL;
83 >        s->proctitle = NULL;
84 > }
85  
86 <        static process_stat_t process_stat;
86 > static void proc_state_destroy(sg_process_stats *s) {
87 >        free(s->process_name);
88 >        free(s->proctitle);
89 > }
90  
91 + sg_process_stats *sg_get_process_stats(int *entries){
92 +        VECTOR_DECLARE_STATIC(proc_state, sg_process_stats, 64,
93 +                              proc_state_init, proc_state_destroy);
94 +        int proc_state_size = 0;
95 +        sg_process_stats *proc_state_ptr;
96 + #ifdef HPUX
97 +        struct pst_status pstat_procinfo;
98 +        long procidx = 0;
99 +        long long pagesize = 0;
100 + #endif
101 + #ifdef ALLBSD
102 +        int mib[4];
103 +        size_t size;
104 +        struct kinfo_proc *kp_stats;
105 +        int procs, i;
106 +        char *proctitle;
107 + #if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
108 +        kvm_t *kvmd;
109 +        char **args, **argsp;
110 +        int argslen = 0;
111 + #else
112 +        long buflen;
113 +        char *p, *proctitletmp;
114 + #endif
115 + #ifdef NETBSD2
116 +        int lwps;
117 +        struct kinfo_lwp *kl_stats;
118 + #endif
119 + #endif
120   #if defined(SOLARIS) || defined(LINUX)
121 <        DIR *proc_dir;
121 >        DIR *proc_dir;
122          struct dirent *dir_entry;
123          char filename[MAX_FILE_LENGTH];
124          FILE *f;
63 #endif
64 #ifdef LINUX
65        char *line_ptr;
66 #endif
125   #ifdef SOLARIS
126          psinfo_t process_info;
127   #endif
128 < #ifdef FREEBSD
129 <        struct kinfo_proc *kp_stats;    
130 <        kvm_t *kvmd = NULL;
131 <        int procs;
128 > #ifdef LINUX
129 >        char s;
130 >        /* If someone has a executable of 4k filename length, they deserve to get it truncated :) */
131 >        char ps_name[4096];
132 >        char *ptr;
133 >        VECTOR_DECLARE_STATIC(psargs, char, 128, NULL, NULL);
134 >        unsigned long stime, utime, starttime;
135 >        int x;
136 >        int fn;
137 >        int len;
138 >        int rc;
139 >        time_t uptime;
140   #endif
141  
142 <        process_stat.sleeping=0;
143 <        process_stat.running=0;
144 <        process_stat.zombie=0;
145 <        process_stat.stopped=0;
146 <        process_stat.total=0;
142 > #ifdef LINUX
143 >        if ((f=fopen("/proc/uptime", "r")) == NULL) {
144 >                sg_set_error_with_errno(SG_ERROR_OPEN, "/proc/uptime");
145 >                return NULL;
146 >        }
147 >        if((fscanf(f,"%lu %*d",&uptime)) != 1){
148 >                sg_set_error(SG_ERROR_PARSE, NULL);
149 >                return NULL;
150 >        }
151 >        fclose(f);
152 > #endif
153  
82 #if defined(SOLARIS) || defined(LINUX)
154          if((proc_dir=opendir(PROC_LOCATION))==NULL){
155 +                sg_set_error_with_errno(SG_ERROR_OPENDIR, PROC_LOCATION);
156                  return NULL;
157          }
158  
# Line 91 | Line 163 | process_stat_t *get_process_stats(){
163                  snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
164   #endif
165   #ifdef LINUX
166 <                snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/status", dir_entry->d_name);
166 >                snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/stat", dir_entry->d_name);
167   #endif
96
168                  if((f=fopen(filename, "r"))==NULL){
169 <                        /* Open failed.. Process since vanished, or the path was too long.
169 >                        /* Open failed.. Process since vanished, or the path was too long.
170                           * Ah well, move onwards to the next one */
171                          continue;
172                  }
102
173   #ifdef SOLARIS
174                  fread(&process_info, sizeof(psinfo_t), 1, f);
175 <                if(process_info.pr_lwp.pr_state==1) process_stat.sleeping++;
106 <                if(process_info.pr_lwp.pr_state==2) process_stat.running++;
107 <                if(process_info.pr_lwp.pr_state==3) process_stat.zombie++;
108 <                if(process_info.pr_lwp.pr_state==4) process_stat.stopped++;
109 <                if(process_info.pr_lwp.pr_state==6) process_stat.running++;
175 >                fclose(f);
176   #endif
177 +
178 +                if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
179 +                        return NULL;
180 +                }
181 +                proc_state_ptr = proc_state+proc_state_size;
182 +
183 + #ifdef SOLARIS          
184 +                proc_state_ptr->pid = process_info.pr_pid;
185 +                proc_state_ptr->parent = process_info.pr_ppid;
186 +                proc_state_ptr->pgid = process_info.pr_pgid;
187 +                proc_state_ptr->uid = process_info.pr_uid;
188 +                proc_state_ptr->euid = process_info.pr_euid;
189 +                proc_state_ptr->gid = process_info.pr_gid;
190 +                proc_state_ptr->egid = process_info.pr_egid;
191 +                proc_state_ptr->proc_size = (process_info.pr_size) * 1024;
192 +                proc_state_ptr->proc_resident = (process_info.pr_rssize) * 1024;
193 +                proc_state_ptr->time_spent = process_info.pr_time.tv_sec;
194 +                proc_state_ptr->cpu_percent = (process_info.pr_pctcpu * 100.0) / 0x8000;
195 +                proc_state_ptr->nice = (int)process_info.pr_lwp.pr_nice - 20;
196 +                if (sg_update_string(&proc_state_ptr->process_name,
197 +                                     process_info.pr_fname) < 0) {
198 +                        return NULL;
199 +                }
200 +                if (sg_update_string(&proc_state_ptr->proctitle,
201 +                                     process_info.pr_psargs) < 0) {
202 +                        return NULL;
203 +                }
204 +
205 +                switch (process_info.pr_lwp.pr_state) {
206 +                case 1:
207 +                        proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
208 +                        break;
209 +                case 2:
210 +                case 5:
211 +                        proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
212 +                        break;
213 +                case 3:
214 +                        proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
215 +                        break;
216 +                case 4:
217 +                        proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
218 +                        break;
219 +                }
220 + #endif
221   #ifdef LINUX
222 <                if((line_ptr=f_read_line(f, "State:"))==NULL){
223 <                        fclose(f);
222 >                x = fscanf(f, "%d %4096s %c %d %d %*d %*d %*d %*u %*u %*u %*u %*u %lu %lu %*d %*d %*d %d %*d %*d %lu %llu %llu %*u %*u %*u %*u %*u %*u %*u %*u %*u %*u %*u %*u %*u %*d %*d\n", &(proc_state_ptr->pid), ps_name, &s, &(proc_state_ptr->parent), &(proc_state_ptr->pgid), &utime, &stime, &(proc_state_ptr->nice), &starttime, &(proc_state_ptr->proc_size), &(proc_state_ptr->proc_resident));
223 >                /* +3 becuase man page says "Resident  Set Size: number of pages the process has in real memory, minus 3 for administrative purposes." */
224 >                proc_state_ptr->proc_resident = (proc_state_ptr->proc_resident + 3) * getpagesize();
225 >                switch (s) {
226 >                case 'S':
227 >                        proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
228 >                        break;
229 >                case 'R':
230 >                        proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
231 >                        break;
232 >                case 'Z':
233 >                        proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
234 >                        break;
235 >                case 'T':
236 >                case 'D':
237 >                        proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
238 >                        break;
239 >                }
240 >        
241 >                /* pa_name[0] should = '(' */
242 >                ptr = strchr(&ps_name[1], ')');
243 >                if(ptr !=NULL) *ptr='\0';
244 >
245 >                if (sg_update_string(&proc_state_ptr->process_name,
246 >                                     &ps_name[1]) < 0) {
247 >                        return NULL;
248 >                }
249 >
250 >                /* cpu */
251 >                proc_state_ptr->cpu_percent = (100.0 * (utime + stime)) / ((uptime * 100.0) - starttime);
252 >
253 >                fclose(f);
254 >
255 >                /* uid / gid */
256 >                snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/status", dir_entry->d_name);
257 >                if ((f=fopen(filename, "r")) == NULL) {
258 >                        /* Open failed.. Process since vanished, or the path was too long.
259 >                         * Ah well, move onwards to the next one */
260                          continue;
261                  }
262 <                if((line_ptr=strchr(line_ptr, '\t'))==NULL){
262 >
263 >                if((ptr=sg_f_read_line(f, "Uid:"))==NULL){
264                          fclose(f);
265                          continue;
266                  }
267 <                line_ptr++;
268 <                if(line_ptr=='\0'){
267 >                sscanf(ptr, "Uid:\t%d\t%d\t%*d\t%*d\n", &(proc_state_ptr->uid), &(proc_state_ptr->euid));
268 >
269 >                if((ptr=sg_f_read_line(f, "Gid:"))==NULL){
270                          fclose(f);
271                          continue;
272                  }
273 +                sscanf(ptr, "Gid:\t%d\t%d\t%*d\t%*d\n", &(proc_state_ptr->gid), &(proc_state_ptr->egid));
274  
126                if(*line_ptr=='S') process_stat.sleeping++;
127                if(*line_ptr=='R') process_stat.running++;
128                if(*line_ptr=='Z') process_stat.zombie++;
129                if(*line_ptr=='T') process_stat.stopped++;
130                if(*line_ptr=='D') process_stat.stopped++;
131 #endif  
132
275                  fclose(f);
276 +
277 +                /* proctitle */
278 +                snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/cmdline", dir_entry->d_name);
279 +
280 +                if((fn=open(filename, O_RDONLY)) == -1){
281 +                        /* Open failed.. Process since vanished, or the path was too long.
282 +                         * Ah well, move onwards to the next one */
283 +                        continue;
284 +                }
285 +
286 + #define READ_BLOCK_SIZE 128
287 +                len = 0;
288 +                do {
289 +                        if (VECTOR_RESIZE(psargs, len + READ_BLOCK_SIZE) < 0) {
290 +                                return NULL;
291 +                        }
292 +                        rc = read(fn, psargs + len, READ_BLOCK_SIZE);
293 +                        if (rc > 0) {
294 +                                len += rc;
295 +                        }
296 +                } while (rc == READ_BLOCK_SIZE);
297 +                close(fn);
298 +
299 +                if (rc == -1) {
300 +                        /* Read failed; move on. */
301 +                        continue;
302 +                }
303 +
304 +                /* Turn \0s into spaces within the command line. */
305 +                ptr = psargs;
306 +                for(x = 0; x < len; x++) {
307 +                        if (*ptr == '\0') *ptr = ' ';
308 +                        ptr++;
309 +                }
310 +
311 +                if (len == 0) {
312 +                        /* We want psargs to be NULL. */
313 +                        if (VECTOR_RESIZE(psargs, 0) < 0) {
314 +                                return NULL;
315 +                        }
316 +                } else {
317 +                        /* Not empty, so append a \0. */
318 +                        if (VECTOR_RESIZE(psargs, len + 1) < 0) {
319 +                                return NULL;
320 +                        }
321 +                        psargs[len] = '\0';
322 +                }
323 +
324 +                if (sg_update_string(&proc_state_ptr->proctitle, psargs) < 0) {
325 +                        return NULL;
326 +                }
327 + #endif
328 +
329 +                proc_state_size++;
330          }
331          closedir(proc_dir);
332   #endif
333 < #ifdef FREEBSD
334 <        if((kvmd = kvm_openfiles(NULL, NULL, NULL, O_RDONLY, NULL)) == NULL){
333 >
334 > #ifdef ALLBSD
335 >        mib[0] = CTL_KERN;
336 >        mib[1] = KERN_PROC;
337 >        mib[2] = KERN_PROC_ALL;
338 >
339 >        if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
340 >                sg_set_error_with_errno(SG_ERROR_SYSCTL,
341 >                                        "CTL_KERN.KERN_PROC.KERN_PROC_ALL");
342                  return NULL;
343          }
344  
345 <        kp_stats=kvm_getprocs(kvmd, KERN_PROC_ALL, 0, &procs);
345 >        procs = size / sizeof(struct kinfo_proc);
346  
347 <        while(procs--){
347 >        kp_stats = sg_malloc(size);
348 >        if(kp_stats == NULL) {
349 >                return NULL;
350 >        }
351 >        memset(kp_stats, 0, size);
352 >
353 >        if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
354 >                sg_set_error_with_errno(SG_ERROR_SYSCTL,
355 >                                        "CTL_KERN.KERN_PROC.KERN_PROC_ALL");
356 >                free(kp_stats);
357 >                return NULL;
358 >        }
359 >
360 > #if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
361 >        kvmd = sg_get_kvm2();
362 > #endif
363 >
364 >        for (i = 0; i < procs; i++) {
365 >                const char *name;
366 >
367   #ifdef FREEBSD5
368 <                if (kp_stats[procs].ki_stat == SSLEEP) process_stat.sleeping++;
147 <                if (kp_stats[procs].ki_stat == SWAIT) process_stat.sleeping++;
148 <                if (kp_stats[procs].ki_stat == SLOCK) process_stat.sleeping++;
149 <                if (kp_stats[procs].ki_stat == SRUN) process_stat.running++;
150 <                if (kp_stats[procs].ki_stat == SIDL) process_stat.running++;
151 <                if (kp_stats[procs].ki_stat == SZOMB) process_stat.zombie++;
152 <                if (kp_stats[procs].ki_stat == SSTOP) process_stat.stopped++;
153 <                
368 >                if (kp_stats[i].ki_stat == 0) {
369   #else
370 <                if (kp_stats[procs].kp_proc.p_stat == SSLEEP) process_stat.sleeping++;  
156 <                if (kp_stats[procs].kp_proc.p_stat == SRUN) process_stat.running++;
157 <                if (kp_stats[procs].kp_proc.p_stat == SIDL) process_stat.running++;
158 <                if (kp_stats[procs].kp_proc.p_stat == SZOMB) process_stat.zombie++;
159 <                if (kp_stats[procs].kp_proc.p_stat == SSTOP) process_stat.stopped++;
370 >                if (kp_stats[i].kp_proc.p_stat == 0) {
371   #endif
372 +                        /* FreeBSD 5 deliberately overallocates the array that
373 +                         * the sysctl returns, so we'll get a few junk
374 +                         * processes on the end that we have to ignore. (Search
375 +                         * for "overestimate by 5 procs" in
376 +                         * src/sys/kern/kern_proc.c for more details.) */
377 +                        continue;
378 +                }
379 +
380 +                if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
381 +                        return NULL;
382 +                }
383 +                proc_state_ptr = proc_state+proc_state_size;
384 +
385 + #ifdef FREEBSD5
386 +                name = kp_stats[i].ki_comm;
387 + #elif defined(DFBSD)
388 +                name = kp_stats[i].kp_thread.td_comm;
389 + #else
390 +                name = kp_stats[i].kp_proc.p_comm;
391 + #endif
392 +                if (sg_update_string(&proc_state_ptr->process_name, name) < 0) {
393 +                        return NULL;
394 +                }
395 +
396 + #if defined(FREEBSD5) || defined(NETBSD) || defined(OPENBSD)
397 +
398 + #ifdef FREEBSD5
399 +                mib[2] = KERN_PROC_ARGS;
400 +                mib[3] = kp_stats[i].ki_pid;
401 + #else
402 +                mib[1] = KERN_PROC_ARGS;
403 +                mib[2] = kp_stats[i].kp_proc.p_pid;
404 +                mib[3] = KERN_PROC_ARGV;
405 + #endif
406 +
407 +                free(proc_state_ptr->proctitle);
408 +                proc_state_ptr->proctitle = NULL;
409 +
410 + /* Starting size - we'll double this straight away */
411 + #define PROCTITLE_START_SIZE 64
412 +                buflen = PROCTITLE_START_SIZE;
413 +                size = buflen;
414 +                proctitle = NULL;
415 +
416 +                do {
417 +                        if(size >= buflen) {
418 +                                buflen *= 2;
419 +                                size = buflen;
420 +                                proctitletmp = sg_realloc(proctitle, buflen);
421 +                                if(proctitletmp == NULL) {
422 +                                        free(proctitle);
423 +                                        proctitle = NULL;
424 +                                        proc_state_ptr->proctitle = NULL;
425 +                                        size = 0;
426 +                                        break;
427 +                                }
428 +                                proctitle = proctitletmp;
429 +                                bzero(proctitle, buflen);
430 +                        }
431 +
432 +                        if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) {
433 +                                free(proctitle);
434 +                                proctitle = NULL;
435 +                                proc_state_ptr->proctitle = NULL;
436 +                                size = 0;
437 +                                break;
438 +                        }
439 +                } while(size >= buflen);
440 +
441 +                if(size > 0) {
442 +                        proc_state_ptr->proctitle = sg_malloc(size+1);
443 +                        if(proc_state_ptr->proctitle == NULL) {
444 +                                return NULL;
445 +                        }
446 +                        p = proctitle;
447 + #ifdef OPENBSD
448 +                        /* On OpenBSD, this value has the argv pointers (which
449 +                         * are terminated by a NULL) at the front, so we have
450 +                         * to skip over them to get to the strings. */
451 +                        while (*(char ***)p != NULL) {
452 +                                p += sizeof(char **);
453 +                        }
454 +                        p += sizeof(char **);
455 + #endif
456 +                        proc_state_ptr->proctitle[0] = '\0';
457 +                        do {
458 +                                sg_strlcat(proc_state_ptr->proctitle, p, size+1);
459 +                                sg_strlcat(proc_state_ptr->proctitle, " ", size+1);
460 +                                p += strlen(p) + 1;
461 +                        } while (p < proctitle + size);
462 +                        free(proctitle);
463 +                        proctitle = NULL;
464 +                        /* remove trailing space */
465 +                        proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = '\0';
466 +                }
467 +                else {
468 +                        if(proctitle != NULL) {
469 +                                free(proctitle);
470 +                                proctitle = NULL;
471 +                        }
472 +                        proc_state_ptr->proctitle = NULL;
473 +                }
474 + #else
475 +                free(proc_state_ptr->proctitle);
476 +                proc_state_ptr->proctitle = NULL;
477 +                if(kvmd != NULL) {
478 +                        args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
479 +                        if(args != NULL) {
480 +                                argsp = args;
481 +                                while(*argsp != NULL) {
482 +                                        argslen += strlen(*argsp) + 1;
483 +                                        argsp++;
484 +                                }
485 +                                proctitle = sg_malloc(argslen + 1);
486 +                                proctitle[0] = '\0';
487 +                                if(proctitle == NULL) {
488 +                                        return NULL;
489 +                                }
490 +                                while(*args != NULL) {
491 +                                        sg_strlcat(proctitle, *args, argslen + 1);
492 +                                        sg_strlcat(proctitle, " ", argslen + 1);
493 +                                        args++;
494 +                                }
495 +                                /* remove trailing space */
496 +                                proctitle[strlen(proctitle)-1] = '\0';
497 +                                proc_state_ptr->proctitle = proctitle;
498 +                        }
499 +                        else {
500 +                                proc_state_ptr->proctitle = NULL;
501 +                        }
502 +                }
503 +                else {
504 +                        proc_state_ptr->proctitle = NULL;
505 +                }
506 + #endif
507 +
508 + #ifdef FREEBSD5
509 +                proc_state_ptr->pid = kp_stats[i].ki_pid;
510 +                proc_state_ptr->parent = kp_stats[i].ki_ppid;
511 +                proc_state_ptr->pgid = kp_stats[i].ki_pgid;
512 + #else
513 +                proc_state_ptr->pid = kp_stats[i].kp_proc.p_pid;
514 +                proc_state_ptr->parent = kp_stats[i].kp_eproc.e_ppid;
515 +                proc_state_ptr->pgid = kp_stats[i].kp_eproc.e_pgid;
516 + #endif
517 +
518 + #ifdef FREEBSD5
519 +                proc_state_ptr->uid = kp_stats[i].ki_ruid;
520 +                proc_state_ptr->euid = kp_stats[i].ki_uid;
521 +                proc_state_ptr->gid = kp_stats[i].ki_rgid;
522 +                proc_state_ptr->egid = kp_stats[i].ki_svgid;
523 + #elif defined(DFBSD)
524 +                proc_state_ptr->uid = kp_stats[i].kp_eproc.e_ucred.cr_ruid;
525 +                proc_state_ptr->euid = kp_stats[i].kp_eproc.e_ucred.cr_svuid;
526 +                proc_state_ptr->gid = kp_stats[i].kp_eproc.e_ucred.cr_rgid;
527 +                proc_state_ptr->egid = kp_stats[i].kp_eproc.e_ucred.cr_svgid;
528 + #else
529 +                proc_state_ptr->uid = kp_stats[i].kp_eproc.e_pcred.p_ruid;
530 +                proc_state_ptr->euid = kp_stats[i].kp_eproc.e_pcred.p_svuid;
531 +                proc_state_ptr->gid = kp_stats[i].kp_eproc.e_pcred.p_rgid;
532 +                proc_state_ptr->egid = kp_stats[i].kp_eproc.e_pcred.p_svgid;
533 + #endif
534 +
535 + #ifdef FREEBSD5
536 +                proc_state_ptr->proc_size = kp_stats[i].ki_size;
537 +                /* This is in pages */
538 +                proc_state_ptr->proc_resident =
539 +                        kp_stats[i].ki_rssize * getpagesize();
540 +                /* This is in microseconds */
541 +                proc_state_ptr->time_spent = kp_stats[i].ki_runtime / 1000000;
542 +                proc_state_ptr->cpu_percent =
543 +                        ((double)kp_stats[i].ki_pctcpu / FSCALE) * 100.0;
544 +                proc_state_ptr->nice = kp_stats[i].ki_nice;
545 + #else
546 +                proc_state_ptr->proc_size =
547 +                        kp_stats[i].kp_eproc.e_vm.vm_map.size;
548 +                /* This is in pages */
549 +                proc_state_ptr->proc_resident =
550 +                        kp_stats[i].kp_eproc.e_vm.vm_rssize * getpagesize();
551 + #if defined(NETBSD) || defined(OPENBSD)
552 +                proc_state_ptr->time_spent =
553 +                        kp_stats[i].kp_proc.p_rtime.tv_sec;
554 + #elif defined(DFBSD)
555 +                proc_state_ptr->time_spent =
556 +                        ( kp_stats[i].kp_thread.td_uticks +
557 +                        kp_stats[i].kp_thread.td_sticks +
558 +                        kp_stats[i].kp_thread.td_iticks ) / 1000000;
559 + #else
560 +                /* This is in microseconds */
561 +                proc_state_ptr->time_spent =
562 +                        kp_stats[i].kp_proc.p_runtime / 1000000;
563 + #endif
564 +                proc_state_ptr->cpu_percent =
565 +                        ((double)kp_stats[i].kp_proc.p_pctcpu / FSCALE) * 100.0;
566 +                proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
567 + #endif
568 +
569 + #ifdef NETBSD2
570 +                {
571 +                        size_t size;
572 +                        int mib[5];
573 +
574 +                        mib[0] = CTL_KERN;
575 +                        mib[1] = KERN_LWP;
576 +                        mib[2] = kp_stats[i].kp_proc.p_pid;
577 +                        mib[3] = sizeof(struct kinfo_lwp);
578 +                        mib[4] = 0;
579 +
580 +                        if(sysctl(mib, 5, NULL, &size, NULL, 0) < 0) {
581 +                                sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.0");
582 +                                return NULL;
583 +                        }
584 +
585 +                        lwps = size / sizeof(struct kinfo_lwp);
586 +                        mib[4] = lwps;
587 +
588 +                        kl_stats = sg_malloc(size);
589 +                        if(kl_stats == NULL) {
590 +                                return NULL;
591 +                        }
592 +
593 +                        if(sysctl(mib, 5, kl_stats, &size, NULL, 0) < 0) {
594 +                                sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.buffersize");
595 +                                return NULL;
596 +                        }
597 +                }
598 +
599 +                switch(kp_stats[i].kp_proc.p_stat) {
600 +                case SIDL:
601 +                        proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
602 +                        break;
603 +                case SACTIVE:
604 +                        {
605 +                                int i;
606 +
607 +                                for(i = 0; i < lwps; i++) {
608 +                                        switch(kl_stats[i].l_stat) {
609 +                                        case LSONPROC:
610 +                                        case LSRUN:
611 +                                                proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
612 +                                                goto end;
613 +                                        case LSSLEEP:
614 +                                                proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
615 +                                                goto end;
616 +                                        case LSSTOP:
617 +                                        case LSSUSPENDED:
618 +                                                proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
619 +                                                goto end;
620 +                                        }
621 +                                        proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
622 +                                }
623 +                                end: ;
624 +                        }
625 +                        break;
626 +                case SSTOP:
627 +                        proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
628 +                        break;
629 +                case SZOMB:
630 +                        proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
631 +                        break;
632 +                default:
633 +                        proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
634 +                        break;
635 +                }
636 +
637 +                free(kl_stats);
638 + #else
639 + #ifdef FREEBSD5
640 +                switch (kp_stats[i].ki_stat) {
641 + #else
642 +                switch (kp_stats[i].kp_proc.p_stat) {
643 + #endif
644 +                case SIDL:
645 +                case SRUN:
646 + #ifdef SONPROC
647 +                case SONPROC: /* NetBSD */
648 + #endif
649 +                        proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
650 +                        break;
651 +                case SSLEEP:
652 + #ifdef SWAIT
653 +                case SWAIT: /* FreeBSD 5 */
654 + #endif
655 + #ifdef SLOCK
656 +                case SLOCK: /* FreeBSD 5 */
657 + #endif
658 +                        proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
659 +                        break;
660 +                case SSTOP:
661 +                        proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
662 +                        break;
663 +                case SZOMB:
664 + #ifdef SDEAD
665 +                case SDEAD: /* OpenBSD & NetBSD */
666 + #endif
667 +                        proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
668 +                        break;
669 +                default:
670 +                        proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
671 +                        break;
672 +                }
673 + #endif
674 +                proc_state_size++;
675          }
676  
677 <        kvm_close(kvmd);
677 >        free(kp_stats);
678   #endif
679  
680 <        process_stat.total=process_stat.sleeping+process_stat.running+process_stat.zombie+process_stat.stopped;
680 > #ifdef HPUX
681 >        if((pagesize=sysconf(_SC_PAGESIZE)) == -1){
682 >                sg_set_error_with_errno(SG_ERROR_SYSCONF, "_SC_PAGESIZE");
683 >                return NULL;
684 >        }
685  
686 +        for (procidx = 0; procidx < 65536; procidx++) {
687 +                if (pstat_getproc(&pstat_procinfo, sizeof(pstat_procinfo), 1, procidx) == -1) {
688 +                        break;
689 +                }
690 +                if (pstat_procinfo.pst_idx != procidx) {
691 +                        continue;
692 +                }
693 +
694 +                if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
695 +                        return NULL;
696 +                }
697 +                proc_state_ptr = proc_state+proc_state_size;
698 +
699 +                proc_state_ptr->pid = pstat_procinfo.pst_pid;
700 +                proc_state_ptr->parent = pstat_procinfo.pst_ppid;
701 +                proc_state_ptr->pgid = pstat_procinfo.pst_pgrp;
702 +                proc_state_ptr->uid = pstat_procinfo.pst_uid;
703 +                proc_state_ptr->euid = pstat_procinfo.pst_euid;
704 +                proc_state_ptr->gid = pstat_procinfo.pst_gid;
705 +                proc_state_ptr->egid = pstat_procinfo.pst_egid;
706 +                proc_state_ptr->proc_size = (pstat_procinfo.pst_dsize + pstat_procinfo.pst_tsize + pstat_procinfo.pst_ssize) * pagesize;
707 +                proc_state_ptr->proc_resident = pstat_procinfo.pst_rssize * pagesize;
708 +                proc_state_ptr->time_spent = pstat_procinfo.pst_time;
709 +                proc_state_ptr->cpu_percent = (pstat_procinfo.pst_pctcpu * 100.0) / 0x8000;
710 +                proc_state_ptr->nice = pstat_procinfo.pst_nice;
711 +
712 +                if (sg_update_string(&proc_state_ptr->process_name,
713 +                                     pstat_procinfo.pst_ucomm) < 0) {
714 +                        return NULL;
715 +                }
716 +                if (sg_update_string(&proc_state_ptr->proctitle,
717 +                                     pstat_procinfo.pst_cmd) < 0) {
718 +                        return NULL;
719 +                }
720 +
721 +                switch (pstat_procinfo.pst_stat) {
722 +                case PS_SLEEP:
723 +                        proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
724 +                        break;
725 +                case PS_RUN:
726 +                        proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
727 +                        break;
728 +                case PS_STOP:
729 +                        proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
730 +                        break;
731 +                case PS_ZOMBIE:
732 +                        proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
733 +                        break;
734 +                case PS_IDLE:
735 +                case PS_OTHER:
736 +                        proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
737 +                        break;
738 +                }
739 +
740 +                proc_state_size++;
741 +        }
742 +
743 + #endif
744 +
745 + #ifdef CYGWIN
746 +        sg_set_error(SG_ERROR_UNSUPPORTED, "Cygwin");
747 +        return NULL;
748 + #endif
749 +
750 +        *entries = proc_state_size;
751 +        return proc_state;
752 + }
753 +
754 + sg_process_count *sg_get_process_count() {
755 +        static sg_process_count process_stat;
756 +        sg_process_stats *ps;
757 +        int ps_size, x;
758 +
759 +        process_stat.sleeping = 0;
760 +        process_stat.running = 0;
761 +        process_stat.zombie = 0;
762 +        process_stat.stopped = 0;
763 +        process_stat.total = 0;
764 +
765 +        ps = sg_get_process_stats(&ps_size);
766 +        if (ps == NULL) {
767 +                return NULL;
768 +        }
769 +
770 +        for(x = 0; x < ps_size; x++) {
771 +                switch (ps->state) {
772 +                case SG_PROCESS_STATE_RUNNING:
773 +                        process_stat.running++;
774 +                        break;
775 +                case SG_PROCESS_STATE_SLEEPING:
776 +                        process_stat.sleeping++;
777 +                        break;
778 +                case SG_PROCESS_STATE_STOPPED:
779 +                        process_stat.stopped++;
780 +                        break;
781 +                case SG_PROCESS_STATE_ZOMBIE:
782 +                        process_stat.zombie++;
783 +                        break;
784 +                default:
785 +                        /* currently no mapping for SG_PROCESS_STATE_UNKNOWN in
786 +                         * sg_process_count */
787 +                        break;
788 +                }
789 +                ps++;
790 +        }
791 +
792 +        process_stat.total = ps_size;
793 +
794          return &process_stat;
795   }
796 +
797 + int sg_process_compare_name(const void *va, const void *vb) {
798 +        const sg_process_stats *a = (sg_process_stats *)va;
799 +        const sg_process_stats *b = (sg_process_stats *)vb;
800 +
801 +        return strcmp(a->process_name, b->process_name);
802 + }
803 +
804 + int sg_process_compare_pid(const void *va, const void *vb) {
805 +        const sg_process_stats *a = (sg_process_stats *)va;
806 +        const sg_process_stats *b = (sg_process_stats *)vb;
807 +
808 +        if (a->pid < b->pid) {
809 +                return -1;
810 +        } else if (a->pid == b->pid) {
811 +                return 0;
812 +        } else {
813 +                return 1;
814 +        }
815 + }
816 +
817 + int sg_process_compare_uid(const void *va, const void *vb) {
818 +        const sg_process_stats *a = (sg_process_stats *)va;
819 +        const sg_process_stats *b = (sg_process_stats *)vb;
820 +
821 +        if (a->uid < b->uid) {
822 +                return -1;
823 +        } else if (a->uid == b->uid) {
824 +                return 0;
825 +        } else {
826 +                return 1;
827 +        }
828 + }
829 +
830 + int sg_process_compare_gid(const void *va, const void *vb) {
831 +        const sg_process_stats *a = (sg_process_stats *)va;
832 +        const sg_process_stats *b = (sg_process_stats *)vb;
833 +
834 +        if (a->gid < b->gid) {
835 +                return -1;
836 +        } else if (a->gid == b->gid) {
837 +                return 0;
838 +        } else {
839 +                return 1;
840 +        }
841 + }
842 +
843 + int sg_process_compare_size(const void *va, const void *vb) {
844 +        const sg_process_stats *a = (sg_process_stats *)va;
845 +        const sg_process_stats *b = (sg_process_stats *)vb;
846 +
847 +        if (a->proc_size < b->proc_size) {
848 +                return -1;
849 +        } else if (a->proc_size == b->proc_size) {
850 +                return 0;
851 +        } else {
852 +                return 1;
853 +        }
854 + }
855 +
856 + int sg_process_compare_res(const void *va, const void *vb) {
857 +        const sg_process_stats *a = (sg_process_stats *)va;
858 +        const sg_process_stats *b = (sg_process_stats *)vb;
859 +
860 +        if (a->proc_resident < b->proc_resident) {
861 +                return -1;
862 +        } else if (a->proc_resident == b->proc_resident) {
863 +                return 0;
864 +        } else {
865 +                return 1;
866 +        }
867 + }
868 +
869 + int sg_process_compare_cpu(const void *va, const void *vb) {
870 +        const sg_process_stats *a = (sg_process_stats *)va;
871 +        const sg_process_stats *b = (sg_process_stats *)vb;
872 +
873 +        if (a->cpu_percent < b->cpu_percent) {
874 +                return -1;
875 +        } else if (a->cpu_percent == b->cpu_percent) {
876 +                return 0;
877 +        } else {
878 +                return 1;
879 +        }
880 + }
881 +
882 + int sg_process_compare_time(const void *va, const void *vb) {
883 +        const sg_process_stats *a = (sg_process_stats *)va;
884 +        const sg_process_stats *b = (sg_process_stats *)vb;
885 +
886 +        if (a->time_spent < b->time_spent) {
887 +                return -1;
888 +        } else if (a->time_spent == b->time_spent) {
889 +                return 0;
890 +        } else {
891 +                return 1;
892 +        }
893 + }
894 +

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines