ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/src/libstatgrab/process_stats.c
Revision: 1.60
Committed: Wed Apr 7 10:27:53 2004 UTC (20 years, 1 month ago) by ats
Content type: text/plain
Branch: MAIN
Changes since 1.59: +9 -3 lines
Log Message:
Change the last few instances of strdup to use sg_update_string.

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