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.6 by pajs, Thu Apr 3 23:14:57 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
3 < * http://www.i-scream.org.uk
4 < * Copyright (C) 2000-2002 i-scream
1 > /*
2 > * i-scream libstatgrab
3 > * http://www.i-scream.org
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 < #if defined(SOLARIS) || defined(LINUX)
143 <        process_stat.sleeping=0;
144 <        process_stat.running=0;
145 <        process_stat.zombie=0;
146 <        process_stat.stopped=0;
147 <        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  
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--){
348 <                if (kp_stats[procs].kp_proc.p_stat == SSLEEP) process_stat.sleeping++;  
349 <                if (kp_stats[procs].kp_proc.p_stat == SRUN) process_stat.running++;
147 <                if (kp_stats[procs].kp_proc.p_stat == SIDL) process_stat.running++;
148 <                if (kp_stats[procs].kp_proc.p_stat == SZOMB) process_stat.zombie++;
149 <                if (kp_stats[procs].kp_proc.p_stat == SSTOP) process_stat.stopped++;
347 >        kp_stats = sg_malloc(size);
348 >        if(kp_stats == NULL) {
349 >                return NULL;
350          }
351 +        memset(kp_stats, 0, size);
352  
353 <        kvm_close(kvmd);
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 <        process_stat.total=process_stat.sleeping+process_stat.running+process_stat.zombie+process_stat.stopped;
364 >        for (i = 0; i < procs; i++) {
365 >                const char *name;
366 >
367 > #ifdef FREEBSD5
368 >                if (kp_stats[i].ki_stat == 0) {
369 > #else
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 >        free(kp_stats);
678 > #endif
679 >
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