ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/src/libstatgrab/process_stats.c
Revision: 1.58
Committed: Tue Apr 6 16:29:44 2004 UTC (20 years, 1 month ago) by pajs
Content type: text/plain
Branch: MAIN
Changes since 1.57: +19 -6 lines
Log Message:
A bit more of the linux code.. Need to do uid/gid bit still.
Also, haven't done nice on solaris yet.

File Contents

# User Rev Content
1 tdb 1.17 /*
2 tdb 1.57 * i-scream libstatgrab
3 tdb 1.8 * http://www.i-scream.org
4 tdb 1.17 * Copyright (C) 2000-2004 i-scream
5 pajs 1.1 *
6 tdb 1.17 * 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 pajs 1.1 *
11 tdb 1.17 * This library is distributed in the hope that it will be useful,
12 pajs 1.1 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 tdb 1.17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14     * Lesser General Public License for more details.
15 pajs 1.1 *
16 tdb 1.17 * 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 tdb 1.18 *
21 pajs 1.58 * $Id: process_stats.c,v 1.57 2004/04/06 14:52:58 tdb Exp $
22 pajs 1.1 */
23    
24     #ifdef HAVE_CONFIG_H
25     #include "config.h"
26     #endif
27    
28 pajs 1.6 #include "statgrab.h"
29 ats 1.48 #include "tools.h"
30     #include "vector.h"
31 pajs 1.6 #if defined(SOLARIS) || defined(LINUX)
32 pajs 1.1 #include <stdio.h>
33     #include <stdlib.h>
34 pajs 1.5 #include <sys/types.h>
35     #include <dirent.h>
36     #include <string.h>
37 pajs 1.6 #endif
38 pajs 1.1
39     #ifdef SOLARIS
40     #include <procfs.h>
41     #include <limits.h>
42 pajs 1.5 #define PROC_LOCATION "/proc"
43     #define MAX_FILE_LENGTH PATH_MAX
44     #endif
45     #ifdef LINUX
46 ats 1.16 #include <limits.h>
47 pajs 1.25 #include <unistd.h>
48     #include <sys/stat.h>
49     #include <fcntl.h>
50 pajs 1.1 #define PROC_LOCATION "/proc"
51     #define MAX_FILE_LENGTH PATH_MAX
52     #endif
53 ats 1.13 #ifdef ALLBSD
54 tdb 1.19 #include <stdlib.h>
55 pajs 1.6 #include <sys/param.h>
56     #include <sys/sysctl.h>
57 tdb 1.22 #if defined(FREEBSD) || defined(DFBSD)
58     #include <sys/user.h>
59     #else
60 tdb 1.19 #include <sys/proc.h>
61 ats 1.13 #endif
62 tdb 1.29 #include <string.h>
63     #include <paths.h>
64     #include <fcntl.h>
65     #include <limits.h>
66 tdb 1.41 #if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
67 tdb 1.29 #include <kvm.h>
68 tdb 1.43 #include <tools.h>
69 pajs 1.6 #endif
70 tdb 1.41 #include <unistd.h>
71 tdb 1.55 #ifdef NETBSD2
72     #include <sys/lwp.h>
73     #endif
74 tdb 1.41 #endif
75 pajs 1.1
76 ats 1.52 static void proc_state_init(sg_process_stats *s) {
77 ats 1.48 s->process_name = NULL;
78     s->proctitle = NULL;
79     }
80    
81 ats 1.52 static void proc_state_destroy(sg_process_stats *s) {
82 ats 1.48 free(s->process_name);
83     free(s->proctitle);
84     }
85    
86 ats 1.52 sg_process_stats *sg_get_process_stats(int *entries){
87     VECTOR_DECLARE_STATIC(proc_state, sg_process_stats, 64,
88 ats 1.48 proc_state_init, proc_state_destroy);
89     int proc_state_size = 0;
90 ats 1.52 sg_process_stats *proc_state_ptr;
91 tdb 1.26 #ifdef ALLBSD
92 tdb 1.37 int mib[4];
93 tdb 1.26 size_t size;
94     struct kinfo_proc *kp_stats;
95 tdb 1.41 int procs, i;
96 tdb 1.45 char *proctitle;
97 tdb 1.41 #if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
98 tdb 1.42 kvm_t *kvmd;
99 tdb 1.50 char **args, **argsp;
100     int argslen = 0;
101 tdb 1.41 #else
102 tdb 1.37 long buflen;
103     char *p;
104     #endif
105 tdb 1.55 #ifdef NETBSD2
106     int lwps;
107     struct kinfo_lwp *kl_stats;
108     #endif
109 tdb 1.26 #endif
110 pajs 1.24 #if defined(SOLARIS) || defined(LINUX)
111 pajs 1.23 DIR *proc_dir;
112     struct dirent *dir_entry;
113     char filename[MAX_FILE_LENGTH];
114     FILE *f;
115 pajs 1.24 #ifdef SOLARIS
116 pajs 1.5 psinfo_t process_info;
117 pajs 1.24 #endif
118 pajs 1.25 #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 ats 1.48 VECTOR_DECLARE_STATIC(psargs, char, 128, NULL, NULL);
124 pajs 1.58 unsigned long stime, utime, starttime;
125 pajs 1.25 int x;
126     int fn;
127 ats 1.48 int len;
128     int rc;
129 pajs 1.58 time_t uptime;
130     #endif
131    
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 pajs 1.25 #endif
141 pajs 1.1
142 pajs 1.23 if((proc_dir=opendir(PROC_LOCATION))==NULL){
143 ats 1.52 return NULL;
144 pajs 1.23 }
145    
146     while((dir_entry=readdir(proc_dir))!=NULL){
147     if(atoi(dir_entry->d_name) == 0) continue;
148    
149 pajs 1.24 #ifdef SOLARIS
150 pajs 1.23 snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
151 pajs 1.24 #endif
152 pajs 1.25 #ifdef LINUX
153     snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/stat", dir_entry->d_name);
154     #endif
155 pajs 1.23 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 pajs 1.24 #ifdef SOLARIS
161 pajs 1.23 fread(&process_info, sizeof(psinfo_t), 1, f);
162 pajs 1.25 #endif
163 pajs 1.1
164 ats 1.48 if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
165 ats 1.52 return NULL;
166 ats 1.48 }
167 pajs 1.23 proc_state_ptr = proc_state+proc_state_size;
168 ats 1.48
169 pajs 1.25 #ifdef SOLARIS
170 pajs 1.23 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 pajs 1.24
184 ats 1.52 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 pajs 1.25 #endif
190     #ifdef LINUX
191 pajs 1.58 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 ats 1.52 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 pajs 1.25
200     /* pa_name[0] should = '(' */
201     ptr = strchr(&ps_name[1], ')');
202     if(ptr !=NULL) *ptr='\0';
203 ats 1.48
204 ats 1.52 if (sg_update_string(&proc_state_ptr->process_name,
205 ats 1.53 &ps_name[1]) < 0) {
206 ats 1.52 return NULL;
207 ats 1.48 }
208 pajs 1.25
209 pajs 1.58 /* 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 pajs 1.25
213 ats 1.48 fclose(f);
214    
215 pajs 1.25 /* 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 ats 1.48
224     #define READ_BLOCK_SIZE 128
225     len = 0;
226     do {
227 ats 1.53 if (VECTOR_RESIZE(psargs, len + READ_BLOCK_SIZE) < 0) {
228 ats 1.52 return NULL;
229 ats 1.48 }
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 pajs 1.25 }
241    
242 ats 1.48 /* Turn \0s into spaces within the command line. */
243 pajs 1.25 ptr = psargs;
244 ats 1.48 for(x = 0; x < len; x++) {
245 pajs 1.25 if (*ptr == '\0') *ptr = ' ';
246     ptr++;
247     }
248    
249 ats 1.53 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 ats 1.52 return NULL;
264 ats 1.48 }
265 pajs 1.24 #endif
266 pajs 1.23
267     proc_state_size++;
268     }
269     closedir(proc_dir);
270 tdb 1.26 #endif
271    
272     #ifdef ALLBSD
273 tdb 1.36 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 ats 1.52 return NULL;
279 tdb 1.36 }
280    
281     procs = size / sizeof(struct kinfo_proc);
282    
283     kp_stats = malloc(size);
284     if(kp_stats == NULL) {
285 ats 1.52 return NULL;
286 tdb 1.36 }
287 ats 1.54 memset(kp_stats, 0, size);
288 tdb 1.36
289     if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
290     free(kp_stats);
291 ats 1.52 return NULL;
292 tdb 1.36 }
293 tdb 1.33
294 tdb 1.37 #if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
295 ats 1.52 kvmd = sg_get_kvm2();
296 tdb 1.33 #endif
297 tdb 1.26
298     for (i = 0; i < procs; i++) {
299 ats 1.48 const char *name;
300 ats 1.54
301 tdb 1.55 #ifdef FREEBSD5
302 ats 1.54 if (kp_stats[i].ki_stat == 0) {
303 tdb 1.55 #else
304     if (kp_stats[i].kp_proc.p_stat == 0) {
305     #endif
306 ats 1.54 /* 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 ats 1.48
314     if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
315 ats 1.52 return NULL;
316 tdb 1.29 }
317 tdb 1.26 proc_state_ptr = proc_state+proc_state_size;
318 tdb 1.30
319     #ifdef FREEBSD5
320 ats 1.48 name = kp_stats[i].ki_comm;
321 tdb 1.35 #elif defined(DFBSD)
322 ats 1.48 name = kp_stats[i].kp_thread.td_comm;
323 tdb 1.30 #else
324 ats 1.48 name = kp_stats[i].kp_proc.p_comm;
325 tdb 1.30 #endif
326 ats 1.53 if (sg_update_string(&proc_state_ptr->process_name, name) < 0) {
327 ats 1.52 return NULL;
328 ats 1.48 }
329 tdb 1.29
330 tdb 1.37 #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 ats 1.52 return NULL;
336 tdb 1.37 }
337     #else
338 tdb 1.56 /* 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 tdb 1.37 mib[1] = KERN_ARGMAX;
342    
343     if(sysctl(mib, 2, &buflen, &size, NULL, 0) < 0) {
344 ats 1.52 return NULL;
345 tdb 1.37 }
346     #endif
347    
348     proctitle = malloc(buflen);
349     if(proctitle == NULL) {
350 ats 1.52 return NULL;
351 tdb 1.37 }
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 ats 1.48 free(proc_state_ptr->proctitle);
365 tdb 1.37 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 tdb 1.44 proc_state_ptr->proctitle = malloc(size+1);
371 tdb 1.37 if(proc_state_ptr->proctitle == NULL) {
372 ats 1.52 return NULL;
373 tdb 1.29 }
374 tdb 1.37 p = proctitle;
375 ats 1.49 proc_state_ptr->proctitle[0] = '\0';
376 tdb 1.37 do {
377 ats 1.52 sg_strlcat(proc_state_ptr->proctitle, p, size+1);
378     sg_strlcat(proc_state_ptr->proctitle, " ", size+1);
379 tdb 1.37 p += strlen(p) + 1;
380     } while (p < proctitle + size);
381     free(proctitle);
382 tdb 1.45 /* remove trailing space */
383     proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = '\0';
384 tdb 1.37 }
385     else {
386     free(proctitle);
387     proc_state_ptr->proctitle = NULL;
388     }
389     #else
390 ats 1.48 free(proc_state_ptr->proctitle);
391 tdb 1.37 if(kvmd != NULL) {
392     args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
393     if(args != NULL) {
394 tdb 1.50 argsp = args;
395     while(*argsp != NULL) {
396 tdb 1.51 argslen += strlen(*argsp) + 1;
397     argsp++;
398 tdb 1.50 }
399     proctitle = malloc(argslen + 1);
400     proctitle[0] = '\0';
401 tdb 1.37 if(proctitle == NULL) {
402 ats 1.52 return NULL;
403 tdb 1.37 }
404     while(*args != NULL) {
405 ats 1.52 sg_strlcat(proctitle, *args, argslen + 1);
406     sg_strlcat(proctitle, " ", argslen + 1);
407 tdb 1.37 args++;
408 tdb 1.29 }
409 tdb 1.37 /* remove trailing space */
410 tdb 1.45 proctitle[strlen(proctitle)-1] = '\0';
411 tdb 1.37 proc_state_ptr->proctitle = proctitle;
412     }
413     else {
414     proc_state_ptr->proctitle = NULL;
415 tdb 1.29 }
416     }
417     else {
418 tdb 1.37 proc_state_ptr->proctitle = NULL;
419 tdb 1.29 }
420 tdb 1.37 #endif
421 tdb 1.26
422 tdb 1.30 #ifdef FREEBSD5
423 tdb 1.26 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 tdb 1.30 #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 tdb 1.26
432 tdb 1.30 #ifdef FREEBSD5
433 tdb 1.26 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 tdb 1.35 #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 tdb 1.30 #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 tdb 1.26
449 tdb 1.30 #ifdef FREEBSD5
450 tdb 1.26 proc_state_ptr->proc_size = kp_stats[i].ki_size;
451     /* This is in pages */
452 tdb 1.30 proc_state_ptr->proc_resident =
453     kp_stats[i].ki_rssize * getpagesize();
454 tdb 1.28 /* This is in microseconds */
455 tdb 1.26 proc_state_ptr->time_spent = kp_stats[i].ki_runtime / 1000000;
456 tdb 1.30 proc_state_ptr->cpu_percent =
457     ((double)kp_stats[i].ki_pctcpu / FSCALE) * 100.0;
458 tdb 1.26 proc_state_ptr->nice = kp_stats[i].ki_nice;
459 tdb 1.30 #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 tdb 1.34 #if defined(NETBSD) || defined(OPENBSD)
466 tdb 1.33 proc_state_ptr->time_spent =
467     kp_stats[i].kp_proc.p_rtime.tv_sec;
468 tdb 1.35 #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 tdb 1.33 #else
474 tdb 1.36 /* This is in microseconds */
475 tdb 1.30 proc_state_ptr->time_spent =
476     kp_stats[i].kp_proc.p_runtime / 1000000;
477 tdb 1.33 #endif
478 tdb 1.30 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 tdb 1.26
483 tdb 1.55 #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 tdb 1.30 #ifdef FREEBSD5
550 tdb 1.26 switch (kp_stats[i].ki_stat) {
551 tdb 1.30 #else
552     switch (kp_stats[i].kp_proc.p_stat) {
553     #endif
554 tdb 1.26 case SIDL:
555     case SRUN:
556     #ifdef SONPROC
557     case SONPROC: /* NetBSD */
558     #endif
559 ats 1.52 proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
560 tdb 1.26 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 ats 1.52 proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
569 tdb 1.26 break;
570     case SSTOP:
571 ats 1.52 proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
572 tdb 1.26 break;
573     case SZOMB:
574     #ifdef SDEAD
575     case SDEAD: /* OpenBSD & NetBSD */
576     #endif
577 ats 1.52 proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
578 tdb 1.31 break;
579     default:
580 ats 1.52 proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
581 tdb 1.26 break;
582     }
583 tdb 1.55 #endif
584 tdb 1.26 proc_state_size++;
585     }
586    
587     free(kp_stats);
588 tdb 1.47 #endif
589    
590     #ifdef CYGWIN
591 ats 1.52 return NULL;
592 tdb 1.26 #endif
593    
594 ats 1.52 *entries = proc_state_size;
595     return proc_state;
596 tdb 1.27 }
597    
598 ats 1.52 sg_process_count *sg_get_process_count() {
599     static sg_process_count process_stat;
600     sg_process_stats *ps;
601 tdb 1.27 int ps_size, x;
602    
603 tdb 1.30 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 tdb 1.27
609 ats 1.52 ps = sg_get_process_stats(&ps_size);
610     if (ps == NULL) {
611 tdb 1.32 return NULL;
612     }
613 tdb 1.27
614     for(x = 0; x < ps_size; x++) {
615     switch (ps->state) {
616 ats 1.52 case SG_PROCESS_STATE_RUNNING:
617 tdb 1.27 process_stat.running++;
618     break;
619 ats 1.52 case SG_PROCESS_STATE_SLEEPING:
620 tdb 1.27 process_stat.sleeping++;
621     break;
622 ats 1.52 case SG_PROCESS_STATE_STOPPED:
623 tdb 1.27 process_stat.stopped++;
624     break;
625 ats 1.52 case SG_PROCESS_STATE_ZOMBIE:
626 tdb 1.27 process_stat.zombie++;
627 ats 1.40 break;
628     default:
629 ats 1.52 /* currently no mapping for SG_PROCESS_STATE_UNKNOWN in
630     * sg_process_count */
631 tdb 1.27 break;
632     }
633     ps++;
634     }
635    
636     process_stat.total = ps_size;
637    
638     return &process_stat;
639 pajs 1.23 }