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.1 by pajs, Tue Feb 18 19:28:30 2003 UTC vs.
Revision 1.58 by pajs, Tue Apr 6 16:29:44 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
25   #include "config.h"
26   #endif
27  
25 #include <stdio.h>
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>
34 < #include <unistd.h>
34 > #include <sys/types.h>
35 > #include <dirent.h>
36 > #include <string.h>
37 > #endif
38  
39   #ifdef SOLARIS
40   #include <procfs.h>
41   #include <limits.h>
33 #include <sys/types.h>
34 #include <dirent.h>
35 #include <string.h>
42   #define PROC_LOCATION "/proc"
43   #define MAX_FILE_LENGTH PATH_MAX
44   #endif
45 + #ifdef LINUX
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 ALLBSD
54 + #include <stdlib.h>
55 + #include <sys/param.h>
56 + #include <sys/sysctl.h>
57 + #if defined(FREEBSD) || defined(DFBSD)
58 + #include <sys/user.h>
59 + #else
60 + #include <sys/proc.h>
61 + #endif
62 + #include <string.h>
63 + #include <paths.h>
64 + #include <fcntl.h>
65 + #include <limits.h>
66 + #if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
67 + #include <kvm.h>
68 + #include <tools.h>
69 + #endif
70 + #include <unistd.h>
71 + #ifdef NETBSD2
72 + #include <sys/lwp.h>
73 + #endif
74 + #endif
75  
76 < process_stat_t *get_process_stats(){
76 > static void proc_state_init(sg_process_stats *s) {
77 >        s->process_name = NULL;
78 >        s->proctitle = NULL;
79 > }
80  
81 <        static process_stat_t process_stat;
81 > static void proc_state_destroy(sg_process_stats *s) {
82 >        free(s->process_name);
83 >        free(s->proctitle);
84 > }
85  
86 + sg_process_stats *sg_get_process_stats(int *entries){
87 +        VECTOR_DECLARE_STATIC(proc_state, sg_process_stats, 64,
88 +                              proc_state_init, proc_state_destroy);
89 +        int proc_state_size = 0;
90 +        sg_process_stats *proc_state_ptr;
91 + #ifdef ALLBSD
92 +        int mib[4];
93 +        size_t size;
94 +        struct kinfo_proc *kp_stats;
95 +        int procs, i;
96 +        char *proctitle;
97 + #if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
98 +        kvm_t *kvmd;
99 +        char **args, **argsp;
100 +        int argslen = 0;
101 + #else
102 +        long buflen;
103 +        char *p;
104 + #endif
105 + #ifdef NETBSD2
106 +        int lwps;
107 +        struct kinfo_lwp *kl_stats;
108 + #endif
109 + #endif
110 + #if defined(SOLARIS) || defined(LINUX)
111 +        DIR *proc_dir;
112 +        struct dirent *dir_entry;
113 +        char filename[MAX_FILE_LENGTH];
114 +        FILE *f;
115 + #ifdef SOLARIS
116          psinfo_t process_info;
117 <        DIR *proc_dir;
118 <        struct dirent *dir_entry;
119 <        char filename[MAX_FILE_LENGTH];
120 <        FILE *f;
117 > #endif
118 > #ifdef LINUX
119 >        char s;
120 >        /* If someone has a executable of 4k filename length, they deserve to get it truncated :) */
121 >        char ps_name[4096];
122 >        char *ptr;
123 >        VECTOR_DECLARE_STATIC(psargs, char, 128, NULL, NULL);
124 >        unsigned long stime, utime, starttime;
125 >        int x;
126 >        int fn;
127 >        int len;
128 >        int rc;
129 >        time_t uptime;
130 > #endif
131  
132 <        process_stat.sleeping=0;
133 <        process_stat.running=0;
134 <        process_stat.zombie=0;
135 <        process_stat.stopped=0;
136 <        process_stat.running=0;
132 > #ifdef LINUX
133 >        if ((f=fopen("/proc/uptime", "r")) == NULL) {
134 >                return NULL;
135 >        }
136 >        if((fscanf(f,"%lu %*d",&uptime)) != 1){
137 >                return NULL;
138 >        }
139 >        fclose(f);
140 > #endif
141  
142 <        if((proc_dir=opendir(PROC_LOCATION))==NULL){
142 >        if((proc_dir=opendir(PROC_LOCATION))==NULL){
143 >                return NULL;
144 >        }
145 >
146 >        while((dir_entry=readdir(proc_dir))!=NULL){
147 >                if(atoi(dir_entry->d_name) == 0) continue;
148 >
149 > #ifdef SOLARIS
150 >                snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
151 > #endif
152 > #ifdef LINUX
153 >                snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/stat", dir_entry->d_name);
154 > #endif
155 >                if((f=fopen(filename, "r"))==NULL){
156 >                        /* Open failed.. Process since vanished, or the path was too long.
157 >                         * Ah well, move onwards to the next one */
158 >                        continue;
159 >                }
160 > #ifdef SOLARIS
161 >                fread(&process_info, sizeof(psinfo_t), 1, f);
162 > #endif
163 >
164 >                if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
165 >                        return NULL;
166 >                }
167 >                proc_state_ptr = proc_state+proc_state_size;
168 >
169 > #ifdef SOLARIS          
170 >                proc_state_ptr->pid = process_info.pr_pid;
171 >                proc_state_ptr->parent = process_info.pr_ppid;
172 >                proc_state_ptr->pgid = process_info.pr_pgid;
173 >                proc_state_ptr->uid = process_info.pr_uid;
174 >                proc_state_ptr->euid = process_info.pr_euid;
175 >                proc_state_ptr->gid = process_info.pr_gid;
176 >                proc_state_ptr->egid = process_info.pr_egid;
177 >                proc_state_ptr->proc_size = (process_info.pr_size) * 1024;
178 >                proc_state_ptr->proc_resident = (process_info.pr_rssize) * 1024;
179 >                proc_state_ptr->time_spent = process_info.pr_time.tv_sec;
180 >                proc_state_ptr->cpu_percent = (process_info.pr_pctcpu * 100.0) / 0x8000;
181 >                proc_state_ptr->process_name = strdup(process_info.pr_fname);
182 >                proc_state_ptr->proctitle = strdup(process_info.pr_psargs);
183 >
184 >                if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
185 >                if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
186 >                if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
187 >                if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
188 >                if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
189 > #endif
190 > #ifdef LINUX
191 >                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));
192 >                /* +3 becuase man page says "Resident  Set Size: number of pages the process has in real memory, minus 3 for administrative purposes." */
193 >                proc_state_ptr->proc_resident = (proc_state_ptr->proc_resident + 3) * getpagesize();
194 >                if(s == 'S') proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
195 >                if(s == 'R') proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
196 >                if(s == 'Z') proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
197 >                if(s == 'T') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
198 >                if(s == 'D') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
199 >        
200 >                /* pa_name[0] should = '(' */
201 >                ptr = strchr(&ps_name[1], ')');
202 >                if(ptr !=NULL) *ptr='\0';
203 >
204 >                if (sg_update_string(&proc_state_ptr->process_name,
205 >                                     &ps_name[1]) < 0) {
206 >                        return NULL;
207 >                }
208 >
209 >                /* cpu */
210 >                proc_state_ptr->cpu_percent = (100.0 * (utime + stime)) / ((uptime * 100.0) - starttime);
211 >                printf("%s  ut: %lu st: %lu up: %lu start: %lu\n", proc_state_ptr->process_name, utime, stime, uptime , starttime);
212 >
213 >                fclose(f);
214 >
215 >                /* proctitle */
216 >                snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/cmdline", dir_entry->d_name);
217 >
218 >                if((fn=open(filename, O_RDONLY)) == -1){
219 >                        /* Open failed.. Process since vanished, or the path was too long.
220 >                         * Ah well, move onwards to the next one */
221 >                        continue;
222 >                }
223 >
224 > #define READ_BLOCK_SIZE 128
225 >                len = 0;
226 >                do {
227 >                        if (VECTOR_RESIZE(psargs, len + READ_BLOCK_SIZE) < 0) {
228 >                                return NULL;
229 >                        }
230 >                        rc = read(fn, psargs + len, READ_BLOCK_SIZE);
231 >                        if (rc > 0) {
232 >                                len += rc;
233 >                        }
234 >                } while (rc == READ_BLOCK_SIZE);
235 >                close(fn);
236 >
237 >                if (rc == -1) {
238 >                        /* Read failed; move on. */
239 >                        continue;
240 >                }
241 >
242 >                /* Turn \0s into spaces within the command line. */
243 >                ptr = psargs;
244 >                for(x = 0; x < len; x++) {
245 >                        if (*ptr == '\0') *ptr = ' ';
246 >                        ptr++;
247 >                }
248 >
249 >                if (len == 0) {
250 >                        /* We want psargs to be NULL. */
251 >                        if (VECTOR_RESIZE(psargs, 0) < 0) {
252 >                                return NULL;
253 >                        }
254 >                } else {
255 >                        /* Not empty, so append a \0. */
256 >                        if (VECTOR_RESIZE(psargs, len + 1) < 0) {
257 >                                return NULL;
258 >                        }
259 >                        psargs[len] = '\0';
260 >                }
261 >
262 >                if (sg_update_string(&proc_state_ptr->proctitle, psargs) < 0) {
263 >                        return NULL;
264 >                }
265 > #endif
266 >
267 >                proc_state_size++;
268 >        }
269 >        closedir(proc_dir);
270 > #endif
271 >
272 > #ifdef ALLBSD
273 >        mib[0] = CTL_KERN;
274 >        mib[1] = KERN_PROC;
275 >        mib[2] = KERN_PROC_ALL;
276 >
277 >        if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
278                  return NULL;
279          }
280  
281 <        while((dir_entry=readdir(proc_dir))!=NULL){
61 <                if(atoi(dir_entry->d_name) == 0) continue;
62 <                snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
281 >        procs = size / sizeof(struct kinfo_proc);
282  
283 <                printf("Filename : %s\n", filename);
284 <                if((f=fopen(filename, "r"))==NULL){
285 <                        /* Open failed.. Process since vanished, or the path was too long.
286 <                         * Ah well, move onwards to the next one */
283 >        kp_stats = malloc(size);
284 >        if(kp_stats == NULL) {
285 >                return NULL;
286 >        }
287 >        memset(kp_stats, 0, size);
288 >
289 >        if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
290 >                free(kp_stats);
291 >                return NULL;
292 >        }
293 >
294 > #if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
295 >        kvmd = sg_get_kvm2();
296 > #endif
297 >
298 >        for (i = 0; i < procs; i++) {
299 >                const char *name;
300 >
301 > #ifdef FREEBSD5
302 >                if (kp_stats[i].ki_stat == 0) {
303 > #else
304 >                if (kp_stats[i].kp_proc.p_stat == 0) {
305 > #endif
306 >                        /* FreeBSD 5 deliberately overallocates the array that
307 >                         * the sysctl returns, so we'll get a few junk
308 >                         * processes on the end that we have to ignore. (Search
309 >                         * for "overestimate by 5 procs" in
310 >                         * src/sys/kern/kern_proc.c for more details.) */
311                          continue;
312                  }
313  
314 <                fread(&process_info, sizeof(psinfo_t), 1, f);
315 <                fclose(f);
314 >                if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
315 >                        return NULL;
316 >                }
317 >                proc_state_ptr = proc_state+proc_state_size;
318  
319 <                if(process_info.pr_lwp.pr_state==1) process_stat.sleeping++;
320 <                if(process_info.pr_lwp.pr_state==2) process_stat.running++;
321 <                if(process_info.pr_lwp.pr_state==3) process_stat.zombie++;
322 <                if(process_info.pr_lwp.pr_state==4) process_stat.stopped++;
323 <                if(process_info.pr_lwp.pr_state==6) process_stat.running++;
319 > #ifdef FREEBSD5
320 >                name = kp_stats[i].ki_comm;
321 > #elif defined(DFBSD)
322 >                name = kp_stats[i].kp_thread.td_comm;
323 > #else
324 >                name = kp_stats[i].kp_proc.p_comm;
325 > #endif
326 >                if (sg_update_string(&proc_state_ptr->process_name, name) < 0) {
327 >                        return NULL;
328 >                }
329 >
330 > #if defined(FREEBSD5) || defined(NETBSD) || defined(OPENBSD)
331 >                size = sizeof(buflen);
332 >
333 > #ifdef FREEBSD5
334 >                if(sysctlbyname("kern.ps_arg_cache_limit", &buflen, &size, NULL, 0) < 0) {
335 >                        return NULL;
336 >                }
337 > #else
338 >                /* FIXME - this value can be too large on some of
339 >                   the BSD's, which causes sysctl not to return
340 >                   anything. Maybe we need something smaller? */
341 >                mib[1] = KERN_ARGMAX;
342 >
343 >                if(sysctl(mib, 2, &buflen, &size, NULL, 0) < 0) {
344 >                        return NULL;
345 >                }
346 > #endif
347 >
348 >                proctitle = malloc(buflen);
349 >                if(proctitle == NULL) {
350 >                        return NULL;
351 >                }
352 >
353 >                size = buflen;
354 >
355 > #ifdef FREEBSD5
356 >                mib[2] = KERN_PROC_ARGS;
357 >                mib[3] = kp_stats[i].ki_pid;
358 > #else
359 >                mib[1] = KERN_PROC_ARGS;
360 >                mib[2] = kp_stats[i].kp_proc.p_pid;
361 >                mib[3] = KERN_PROC_ARGV;
362 > #endif
363 >
364 >                free(proc_state_ptr->proctitle);
365 >                if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) {
366 >                        free(proctitle);
367 >                        proc_state_ptr->proctitle = NULL;
368 >                }
369 >                else if(size > 0) {
370 >                        proc_state_ptr->proctitle = malloc(size+1);
371 >                        if(proc_state_ptr->proctitle == NULL) {
372 >                                return NULL;
373 >                        }
374 >                        p = proctitle;
375 >                        proc_state_ptr->proctitle[0] = '\0';
376 >                        do {
377 >                                sg_strlcat(proc_state_ptr->proctitle, p, size+1);
378 >                                sg_strlcat(proc_state_ptr->proctitle, " ", size+1);
379 >                                p += strlen(p) + 1;
380 >                        } while (p < proctitle + size);
381 >                        free(proctitle);
382 >                        /* remove trailing space */
383 >                        proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = '\0';
384 >                }
385 >                else {
386 >                        free(proctitle);
387 >                        proc_state_ptr->proctitle = NULL;
388 >                }
389 > #else
390 >                free(proc_state_ptr->proctitle);
391 >                if(kvmd != NULL) {
392 >                        args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
393 >                        if(args != NULL) {
394 >                                argsp = args;
395 >                                while(*argsp != NULL) {
396 >                                        argslen += strlen(*argsp) + 1;
397 >                                        argsp++;
398 >                                }
399 >                                proctitle = malloc(argslen + 1);
400 >                                proctitle[0] = '\0';
401 >                                if(proctitle == NULL) {
402 >                                        return NULL;
403 >                                }
404 >                                while(*args != NULL) {
405 >                                        sg_strlcat(proctitle, *args, argslen + 1);
406 >                                        sg_strlcat(proctitle, " ", argslen + 1);
407 >                                        args++;
408 >                                }
409 >                                /* remove trailing space */
410 >                                proctitle[strlen(proctitle)-1] = '\0';
411 >                                proc_state_ptr->proctitle = proctitle;
412 >                        }
413 >                        else {
414 >                                proc_state_ptr->proctitle = NULL;
415 >                        }
416 >                }
417 >                else {
418 >                        proc_state_ptr->proctitle = NULL;
419 >                }
420 > #endif
421 >
422 > #ifdef FREEBSD5
423 >                proc_state_ptr->pid = kp_stats[i].ki_pid;
424 >                proc_state_ptr->parent = kp_stats[i].ki_ppid;
425 >                proc_state_ptr->pgid = kp_stats[i].ki_pgid;
426 > #else
427 >                proc_state_ptr->pid = kp_stats[i].kp_proc.p_pid;
428 >                proc_state_ptr->parent = kp_stats[i].kp_eproc.e_ppid;
429 >                proc_state_ptr->pgid = kp_stats[i].kp_eproc.e_pgid;
430 > #endif
431 >
432 > #ifdef FREEBSD5
433 >                proc_state_ptr->uid = kp_stats[i].ki_ruid;
434 >                proc_state_ptr->euid = kp_stats[i].ki_uid;
435 >                proc_state_ptr->gid = kp_stats[i].ki_rgid;
436 >                proc_state_ptr->egid = kp_stats[i].ki_svgid;
437 > #elif defined(DFBSD)
438 >                proc_state_ptr->uid = kp_stats[i].kp_eproc.e_ucred.cr_ruid;
439 >                proc_state_ptr->euid = kp_stats[i].kp_eproc.e_ucred.cr_svuid;
440 >                proc_state_ptr->gid = kp_stats[i].kp_eproc.e_ucred.cr_rgid;
441 >                proc_state_ptr->egid = kp_stats[i].kp_eproc.e_ucred.cr_svgid;
442 > #else
443 >                proc_state_ptr->uid = kp_stats[i].kp_eproc.e_pcred.p_ruid;
444 >                proc_state_ptr->euid = kp_stats[i].kp_eproc.e_pcred.p_svuid;
445 >                proc_state_ptr->gid = kp_stats[i].kp_eproc.e_pcred.p_rgid;
446 >                proc_state_ptr->egid = kp_stats[i].kp_eproc.e_pcred.p_svgid;
447 > #endif
448 >
449 > #ifdef FREEBSD5
450 >                proc_state_ptr->proc_size = kp_stats[i].ki_size;
451 >                /* This is in pages */
452 >                proc_state_ptr->proc_resident =
453 >                        kp_stats[i].ki_rssize * getpagesize();
454 >                /* This is in microseconds */
455 >                proc_state_ptr->time_spent = kp_stats[i].ki_runtime / 1000000;
456 >                proc_state_ptr->cpu_percent =
457 >                        ((double)kp_stats[i].ki_pctcpu / FSCALE) * 100.0;
458 >                proc_state_ptr->nice = kp_stats[i].ki_nice;
459 > #else
460 >                proc_state_ptr->proc_size =
461 >                        kp_stats[i].kp_eproc.e_vm.vm_map.size;
462 >                /* This is in pages */
463 >                proc_state_ptr->proc_resident =
464 >                        kp_stats[i].kp_eproc.e_vm.vm_rssize * getpagesize();
465 > #if defined(NETBSD) || defined(OPENBSD)
466 >                proc_state_ptr->time_spent =
467 >                        kp_stats[i].kp_proc.p_rtime.tv_sec;
468 > #elif defined(DFBSD)
469 >                proc_state_ptr->time_spent =
470 >                        ( kp_stats[i].kp_thread.td_uticks +
471 >                        kp_stats[i].kp_thread.td_sticks +
472 >                        kp_stats[i].kp_thread.td_iticks ) / 1000000;
473 > #else
474 >                /* This is in microseconds */
475 >                proc_state_ptr->time_spent =
476 >                        kp_stats[i].kp_proc.p_runtime / 1000000;
477 > #endif
478 >                proc_state_ptr->cpu_percent =
479 >                        ((double)kp_stats[i].kp_proc.p_pctcpu / FSCALE) * 100.0;
480 >                proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
481 > #endif
482 >
483 > #ifdef NETBSD2
484 >                {
485 >                        size_t size;
486 >                        int mib[5];
487 >
488 >                        mib[0] = CTL_KERN;
489 >                        mib[1] = KERN_LWP;
490 >                        mib[2] = kp_stats[i].kp_proc.p_pid;
491 >                        mib[3] = sizeof(struct kinfo_lwp);
492 >                        mib[4] = 0;
493 >
494 >                        if(sysctl(mib, 5, NULL, &size, NULL, 0) < 0) {
495 >                                return NULL;
496 >                        }
497 >
498 >                        lwps = size / sizeof(struct kinfo_lwp);
499 >                        mib[4] = lwps;
500 >
501 >                        kl_stats = malloc(size);
502 >                        if(kl_stats == NULL) {
503 >                                return NULL;
504 >                        }
505 >
506 >                        if(sysctl(mib, 5, kl_stats, &size, NULL, 0) < 0) {
507 >                                return NULL;
508 >                        }
509 >                }
510 >
511 >                switch(kp_stats[i].kp_proc.p_stat) {
512 >                case SIDL:
513 >                        proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
514 >                        break;
515 >                case SACTIVE:
516 >                        {
517 >                                int i;
518 >
519 >                                for(i = 0; i < lwps; i++) {
520 >                                        switch(kl_stats[i].l_stat) {
521 >                                        case LSONPROC:
522 >                                        case LSRUN:
523 >                                                proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
524 >                                                goto end;
525 >                                        case LSSLEEP:
526 >                                                proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
527 >                                                goto end;
528 >                                        case LSSTOP:
529 >                                        case LSSUSPENDED:
530 >                                                proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
531 >                                                goto end;
532 >                                        }
533 >                                        proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
534 >                                }
535 >                                end: ;
536 >                        }
537 >                        break;
538 >                case SSTOP:
539 >                        proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
540 >                        break;
541 >                case SZOMB:
542 >                        proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
543 >                        break;
544 >                default:
545 >                        proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
546 >                        break;
547 >                }
548 > #else
549 > #ifdef FREEBSD5
550 >                switch (kp_stats[i].ki_stat) {
551 > #else
552 >                switch (kp_stats[i].kp_proc.p_stat) {
553 > #endif
554 >                case SIDL:
555 >                case SRUN:
556 > #ifdef SONPROC
557 >                case SONPROC: /* NetBSD */
558 > #endif
559 >                        proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
560 >                        break;
561 >                case SSLEEP:
562 > #ifdef SWAIT
563 >                case SWAIT: /* FreeBSD 5 */
564 > #endif
565 > #ifdef SLOCK
566 >                case SLOCK: /* FreeBSD 5 */
567 > #endif
568 >                        proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
569 >                        break;
570 >                case SSTOP:
571 >                        proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
572 >                        break;
573 >                case SZOMB:
574 > #ifdef SDEAD
575 >                case SDEAD: /* OpenBSD & NetBSD */
576 > #endif
577 >                        proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
578 >                        break;
579 >                default:
580 >                        proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
581 >                        break;
582 >                }
583 > #endif
584 >                proc_state_size++;
585          }
586 <        
587 <        closedir(proc_dir);
586 >
587 >        free(kp_stats);
588 > #endif
589 >
590 > #ifdef CYGWIN
591 >        return NULL;
592 > #endif
593 >
594 >        *entries = proc_state_size;
595 >        return proc_state;
596 > }
597 >
598 > sg_process_count *sg_get_process_count() {
599 >        static sg_process_count process_stat;
600 >        sg_process_stats *ps;
601 >        int ps_size, x;
602 >
603 >        process_stat.sleeping = 0;
604 >        process_stat.running = 0;
605 >        process_stat.zombie = 0;
606 >        process_stat.stopped = 0;
607 >        process_stat.total = 0;
608 >
609 >        ps = sg_get_process_stats(&ps_size);
610 >        if (ps == NULL) {
611 >                return NULL;
612 >        }
613 >
614 >        for(x = 0; x < ps_size; x++) {
615 >                switch (ps->state) {
616 >                case SG_PROCESS_STATE_RUNNING:
617 >                        process_stat.running++;
618 >                        break;
619 >                case SG_PROCESS_STATE_SLEEPING:
620 >                        process_stat.sleeping++;
621 >                        break;
622 >                case SG_PROCESS_STATE_STOPPED:
623 >                        process_stat.stopped++;
624 >                        break;
625 >                case SG_PROCESS_STATE_ZOMBIE:
626 >                        process_stat.zombie++;
627 >                        break;
628 >                default:
629 >                        /* currently no mapping for SG_PROCESS_STATE_UNKNOWN in
630 >                         * sg_process_count */
631 >                        break;
632 >                }
633 >                ps++;
634 >        }
635 >
636 >        process_stat.total = ps_size;
637  
638          return &process_stat;
639   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines