ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/src/libstatgrab/process_stats.c
Revision: 1.74
Committed: Sun Jul 18 21:30:11 2004 UTC (19 years, 10 months ago) by ats
Content type: text/plain
Branch: MAIN
CVS Tags: LIBSTATGRAB_0_10_3, LIBSTATGRAB_0_10_2, LIBSTATGRAB_0_10_1
Changes since 1.73: +9 -7 lines
Log Message:
Use sg_set_error_with_errno whenever errno's valid.
Change the one user of SG_ERROR_ENOENT to SG_ERROR_SYSCTL instead (since
that's what it should have been).

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.74 * $Id: process_stats.c,v 1.73 2004/06/21 14:48:48 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.71 #include <errno.h>
55 tdb 1.19 #include <stdlib.h>
56 pajs 1.6 #include <sys/param.h>
57     #include <sys/sysctl.h>
58 tdb 1.22 #if defined(FREEBSD) || defined(DFBSD)
59     #include <sys/user.h>
60     #else
61 tdb 1.19 #include <sys/proc.h>
62 ats 1.13 #endif
63 tdb 1.29 #include <string.h>
64     #include <paths.h>
65     #include <fcntl.h>
66     #include <limits.h>
67 tdb 1.41 #if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
68 tdb 1.29 #include <kvm.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 tdb 1.63 proc_state_init, proc_state_destroy);
89 ats 1.48 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 tdb 1.71 char *p, *proctitletmp;
104 tdb 1.37 #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 tdb 1.63 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 ats 1.74 sg_set_error_with_errno(SG_ERROR_OPEN, "/proc/uptime");
135 pajs 1.58 return NULL;
136     }
137     if((fscanf(f,"%lu %*d",&uptime)) != 1){
138 tdb 1.65 sg_set_error(SG_ERROR_PARSE, NULL);
139 pajs 1.58 return NULL;
140     }
141     fclose(f);
142 pajs 1.25 #endif
143 pajs 1.1
144 tdb 1.63 if((proc_dir=opendir(PROC_LOCATION))==NULL){
145 ats 1.74 sg_set_error_with_errno(SG_ERROR_OPENDIR, PROC_LOCATION);
146 tdb 1.63 return NULL;
147     }
148 pajs 1.23
149 tdb 1.63 while((dir_entry=readdir(proc_dir))!=NULL){
150     if(atoi(dir_entry->d_name) == 0) continue;
151 pajs 1.23
152 pajs 1.24 #ifdef SOLARIS
153 tdb 1.63 snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
154 pajs 1.24 #endif
155 pajs 1.25 #ifdef LINUX
156     snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/stat", dir_entry->d_name);
157     #endif
158 tdb 1.63 if((f=fopen(filename, "r"))==NULL){
159     /* Open failed.. Process since vanished, or the path was too long.
160     * Ah well, move onwards to the next one */
161     continue;
162     }
163 pajs 1.24 #ifdef SOLARIS
164 tdb 1.63 fread(&process_info, sizeof(psinfo_t), 1, f);
165 pajs 1.64 fclose(f);
166 pajs 1.25 #endif
167 pajs 1.1
168 ats 1.48 if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
169 ats 1.52 return NULL;
170 ats 1.48 }
171 pajs 1.23 proc_state_ptr = proc_state+proc_state_size;
172 ats 1.48
173 pajs 1.25 #ifdef SOLARIS
174 pajs 1.23 proc_state_ptr->pid = process_info.pr_pid;
175     proc_state_ptr->parent = process_info.pr_ppid;
176     proc_state_ptr->pgid = process_info.pr_pgid;
177     proc_state_ptr->uid = process_info.pr_uid;
178     proc_state_ptr->euid = process_info.pr_euid;
179     proc_state_ptr->gid = process_info.pr_gid;
180     proc_state_ptr->egid = process_info.pr_egid;
181     proc_state_ptr->proc_size = (process_info.pr_size) * 1024;
182     proc_state_ptr->proc_resident = (process_info.pr_rssize) * 1024;
183     proc_state_ptr->time_spent = process_info.pr_time.tv_sec;
184     proc_state_ptr->cpu_percent = (process_info.pr_pctcpu * 100.0) / 0x8000;
185 pajs 1.64 proc_state_ptr->nice = (int)process_info.pr_lwp.pr_nice - 20;
186 ats 1.60 if (sg_update_string(&proc_state_ptr->process_name,
187 tdb 1.63 process_info.pr_fname) < 0) {
188 ats 1.60 return NULL;
189     }
190     if (sg_update_string(&proc_state_ptr->proctitle,
191 tdb 1.63 process_info.pr_psargs) < 0) {
192 ats 1.60 return NULL;
193     }
194 pajs 1.24
195 tdb 1.63 if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
196     if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
197     if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
198     if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
199     if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
200 pajs 1.25 #endif
201     #ifdef LINUX
202 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));
203     /* +3 becuase man page says "Resident Set Size: number of pages the process has in real memory, minus 3 for administrative purposes." */
204     proc_state_ptr->proc_resident = (proc_state_ptr->proc_resident + 3) * getpagesize();
205 ats 1.52 if(s == 'S') proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
206     if(s == 'R') proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
207     if(s == 'Z') proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
208     if(s == 'T') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
209     if(s == 'D') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
210 pajs 1.25
211     /* pa_name[0] should = '(' */
212     ptr = strchr(&ps_name[1], ')');
213     if(ptr !=NULL) *ptr='\0';
214 ats 1.48
215 ats 1.52 if (sg_update_string(&proc_state_ptr->process_name,
216 tdb 1.63 &ps_name[1]) < 0) {
217 ats 1.52 return NULL;
218 ats 1.48 }
219 pajs 1.25
220 pajs 1.58 /* cpu */
221     proc_state_ptr->cpu_percent = (100.0 * (utime + stime)) / ((uptime * 100.0) - starttime);
222 pajs 1.25
223 tdb 1.63 fclose(f);
224 pajs 1.62
225     /* uid / gid */
226     snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/status", dir_entry->d_name);
227 tdb 1.63 if ((f=fopen(filename, "r")) == NULL) {
228     /* Open failed.. Process since vanished, or the path was too long.
229     * Ah well, move onwards to the next one */
230     continue;
231     }
232 pajs 1.62
233     if((ptr=sg_f_read_line(f, "Uid:"))==NULL){
234     fclose(f);
235     continue;
236     }
237     sscanf(ptr, "Uid:\t%d\t%d\t%*d\t%*d\n", &(proc_state_ptr->uid), &(proc_state_ptr->euid));
238    
239     if((ptr=sg_f_read_line(f, "Gid:"))==NULL){
240 tdb 1.63 fclose(f);
241     continue;
242     }
243     sscanf(ptr, "Gid:\t%d\t%d\t%*d\t%*d\n", &(proc_state_ptr->gid), &(proc_state_ptr->egid));
244 pajs 1.62
245     fclose(f);
246 ats 1.48
247 pajs 1.25 /* proctitle */
248     snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/cmdline", dir_entry->d_name);
249    
250 tdb 1.63 if((fn=open(filename, O_RDONLY)) == -1){
251     /* Open failed.. Process since vanished, or the path was too long.
252     * Ah well, move onwards to the next one */
253     continue;
254     }
255 ats 1.48
256     #define READ_BLOCK_SIZE 128
257     len = 0;
258     do {
259 ats 1.53 if (VECTOR_RESIZE(psargs, len + READ_BLOCK_SIZE) < 0) {
260 ats 1.52 return NULL;
261 ats 1.48 }
262     rc = read(fn, psargs + len, READ_BLOCK_SIZE);
263     if (rc > 0) {
264     len += rc;
265     }
266     } while (rc == READ_BLOCK_SIZE);
267     close(fn);
268    
269     if (rc == -1) {
270     /* Read failed; move on. */
271     continue;
272 pajs 1.25 }
273    
274 ats 1.48 /* Turn \0s into spaces within the command line. */
275 pajs 1.25 ptr = psargs;
276 ats 1.48 for(x = 0; x < len; x++) {
277 pajs 1.25 if (*ptr == '\0') *ptr = ' ';
278     ptr++;
279     }
280    
281 ats 1.53 if (len == 0) {
282     /* We want psargs to be NULL. */
283     if (VECTOR_RESIZE(psargs, 0) < 0) {
284     return NULL;
285     }
286     } else {
287     /* Not empty, so append a \0. */
288     if (VECTOR_RESIZE(psargs, len + 1) < 0) {
289     return NULL;
290     }
291     psargs[len] = '\0';
292     }
293    
294     if (sg_update_string(&proc_state_ptr->proctitle, psargs) < 0) {
295 ats 1.52 return NULL;
296 ats 1.48 }
297 pajs 1.24 #endif
298 pajs 1.23
299     proc_state_size++;
300 tdb 1.63 }
301     closedir(proc_dir);
302 tdb 1.26 #endif
303    
304     #ifdef ALLBSD
305 tdb 1.36 mib[0] = CTL_KERN;
306     mib[1] = KERN_PROC;
307     mib[2] = KERN_PROC_ALL;
308    
309     if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
310 ats 1.74 sg_set_error_with_errno(SG_ERROR_SYSCTL,
311     "CTL_KERN.KERN_PROC.KERN_PROC_ALL");
312 ats 1.52 return NULL;
313 tdb 1.36 }
314    
315     procs = size / sizeof(struct kinfo_proc);
316    
317 ats 1.61 kp_stats = sg_malloc(size);
318 tdb 1.36 if(kp_stats == NULL) {
319 ats 1.52 return NULL;
320 tdb 1.36 }
321 ats 1.54 memset(kp_stats, 0, size);
322 tdb 1.36
323     if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
324 ats 1.74 sg_set_error_with_errno(SG_ERROR_SYSCTL,
325     "CTL_KERN.KERN_PROC.KERN_PROC_ALL");
326 tdb 1.36 free(kp_stats);
327 ats 1.52 return NULL;
328 tdb 1.36 }
329 tdb 1.33
330 tdb 1.37 #if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
331 ats 1.52 kvmd = sg_get_kvm2();
332 tdb 1.33 #endif
333 tdb 1.26
334     for (i = 0; i < procs; i++) {
335 ats 1.48 const char *name;
336 ats 1.54
337 tdb 1.55 #ifdef FREEBSD5
338 ats 1.54 if (kp_stats[i].ki_stat == 0) {
339 tdb 1.55 #else
340     if (kp_stats[i].kp_proc.p_stat == 0) {
341     #endif
342 ats 1.54 /* FreeBSD 5 deliberately overallocates the array that
343     * the sysctl returns, so we'll get a few junk
344     * processes on the end that we have to ignore. (Search
345     * for "overestimate by 5 procs" in
346     * src/sys/kern/kern_proc.c for more details.) */
347     continue;
348     }
349 ats 1.48
350     if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
351 ats 1.52 return NULL;
352 tdb 1.29 }
353 tdb 1.26 proc_state_ptr = proc_state+proc_state_size;
354 tdb 1.30
355     #ifdef FREEBSD5
356 ats 1.48 name = kp_stats[i].ki_comm;
357 tdb 1.35 #elif defined(DFBSD)
358 ats 1.48 name = kp_stats[i].kp_thread.td_comm;
359 tdb 1.30 #else
360 ats 1.48 name = kp_stats[i].kp_proc.p_comm;
361 tdb 1.30 #endif
362 ats 1.53 if (sg_update_string(&proc_state_ptr->process_name, name) < 0) {
363 ats 1.52 return NULL;
364 ats 1.48 }
365 tdb 1.29
366 tdb 1.37 #if defined(FREEBSD5) || defined(NETBSD) || defined(OPENBSD)
367    
368     #ifdef FREEBSD5
369     mib[2] = KERN_PROC_ARGS;
370     mib[3] = kp_stats[i].ki_pid;
371     #else
372     mib[1] = KERN_PROC_ARGS;
373     mib[2] = kp_stats[i].kp_proc.p_pid;
374     mib[3] = KERN_PROC_ARGV;
375     #endif
376    
377 ats 1.48 free(proc_state_ptr->proctitle);
378 tdb 1.71 proc_state_ptr->proctitle = NULL;
379    
380     /* Starting size - we'll double this straight away */
381     #define PROCTITLE_START_SIZE 64
382     buflen = PROCTITLE_START_SIZE;
383     size = buflen;
384     proctitle = NULL;
385    
386     do {
387     if(size >= buflen) {
388     buflen *= 2;
389     size = buflen;
390     proctitletmp = sg_realloc(proctitle, buflen);
391     if(proctitletmp == NULL) {
392     free(proctitle);
393     proctitle = NULL;
394     proc_state_ptr->proctitle = NULL;
395     size = 0;
396     break;
397     }
398     proctitle = proctitletmp;
399     bzero(proctitle, buflen);
400     }
401    
402     if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) {
403     free(proctitle);
404     proctitle = NULL;
405     proc_state_ptr->proctitle = NULL;
406     size = 0;
407     break;
408     }
409     } while(size >= buflen);
410    
411     if(size > 0) {
412 ats 1.61 proc_state_ptr->proctitle = sg_malloc(size+1);
413 tdb 1.37 if(proc_state_ptr->proctitle == NULL) {
414 ats 1.52 return NULL;
415 tdb 1.29 }
416 tdb 1.37 p = proctitle;
417 ats 1.72 #ifdef OPENBSD
418     /* On OpenBSD, this value has the argv pointers (which
419     * are terminated by a NULL) at the front, so we have
420     * to skip over them to get to the strings. */
421     while (*(char ***)p != NULL) {
422     p += sizeof(char **);
423     }
424     p += sizeof(char **);
425     #endif
426 ats 1.49 proc_state_ptr->proctitle[0] = '\0';
427 tdb 1.37 do {
428 ats 1.52 sg_strlcat(proc_state_ptr->proctitle, p, size+1);
429     sg_strlcat(proc_state_ptr->proctitle, " ", size+1);
430 tdb 1.37 p += strlen(p) + 1;
431     } while (p < proctitle + size);
432     free(proctitle);
433 tdb 1.71 proctitle = NULL;
434 tdb 1.45 /* remove trailing space */
435     proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = '\0';
436 tdb 1.37 }
437     else {
438 tdb 1.71 if(proctitle != NULL) {
439     free(proctitle);
440     proctitle = NULL;
441     }
442 tdb 1.37 proc_state_ptr->proctitle = NULL;
443     }
444     #else
445 ats 1.48 free(proc_state_ptr->proctitle);
446 tdb 1.71 proc_state_ptr->proctitle = NULL;
447 tdb 1.37 if(kvmd != NULL) {
448     args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
449     if(args != NULL) {
450 tdb 1.50 argsp = args;
451     while(*argsp != NULL) {
452 tdb 1.51 argslen += strlen(*argsp) + 1;
453     argsp++;
454 tdb 1.50 }
455 ats 1.61 proctitle = sg_malloc(argslen + 1);
456 tdb 1.50 proctitle[0] = '\0';
457 tdb 1.37 if(proctitle == NULL) {
458 ats 1.52 return NULL;
459 tdb 1.37 }
460     while(*args != NULL) {
461 ats 1.52 sg_strlcat(proctitle, *args, argslen + 1);
462     sg_strlcat(proctitle, " ", argslen + 1);
463 tdb 1.37 args++;
464 tdb 1.29 }
465 tdb 1.37 /* remove trailing space */
466 tdb 1.45 proctitle[strlen(proctitle)-1] = '\0';
467 tdb 1.37 proc_state_ptr->proctitle = proctitle;
468     }
469     else {
470     proc_state_ptr->proctitle = NULL;
471 tdb 1.29 }
472     }
473     else {
474 tdb 1.37 proc_state_ptr->proctitle = NULL;
475 tdb 1.29 }
476 tdb 1.37 #endif
477 tdb 1.26
478 tdb 1.30 #ifdef FREEBSD5
479 tdb 1.26 proc_state_ptr->pid = kp_stats[i].ki_pid;
480     proc_state_ptr->parent = kp_stats[i].ki_ppid;
481     proc_state_ptr->pgid = kp_stats[i].ki_pgid;
482 tdb 1.30 #else
483     proc_state_ptr->pid = kp_stats[i].kp_proc.p_pid;
484     proc_state_ptr->parent = kp_stats[i].kp_eproc.e_ppid;
485     proc_state_ptr->pgid = kp_stats[i].kp_eproc.e_pgid;
486     #endif
487 tdb 1.26
488 tdb 1.30 #ifdef FREEBSD5
489 tdb 1.26 proc_state_ptr->uid = kp_stats[i].ki_ruid;
490     proc_state_ptr->euid = kp_stats[i].ki_uid;
491     proc_state_ptr->gid = kp_stats[i].ki_rgid;
492     proc_state_ptr->egid = kp_stats[i].ki_svgid;
493 tdb 1.35 #elif defined(DFBSD)
494     proc_state_ptr->uid = kp_stats[i].kp_eproc.e_ucred.cr_ruid;
495     proc_state_ptr->euid = kp_stats[i].kp_eproc.e_ucred.cr_svuid;
496     proc_state_ptr->gid = kp_stats[i].kp_eproc.e_ucred.cr_rgid;
497     proc_state_ptr->egid = kp_stats[i].kp_eproc.e_ucred.cr_svgid;
498 tdb 1.30 #else
499     proc_state_ptr->uid = kp_stats[i].kp_eproc.e_pcred.p_ruid;
500     proc_state_ptr->euid = kp_stats[i].kp_eproc.e_pcred.p_svuid;
501     proc_state_ptr->gid = kp_stats[i].kp_eproc.e_pcred.p_rgid;
502     proc_state_ptr->egid = kp_stats[i].kp_eproc.e_pcred.p_svgid;
503     #endif
504 tdb 1.26
505 tdb 1.30 #ifdef FREEBSD5
506 tdb 1.26 proc_state_ptr->proc_size = kp_stats[i].ki_size;
507     /* This is in pages */
508 tdb 1.30 proc_state_ptr->proc_resident =
509     kp_stats[i].ki_rssize * getpagesize();
510 tdb 1.28 /* This is in microseconds */
511 tdb 1.26 proc_state_ptr->time_spent = kp_stats[i].ki_runtime / 1000000;
512 tdb 1.30 proc_state_ptr->cpu_percent =
513     ((double)kp_stats[i].ki_pctcpu / FSCALE) * 100.0;
514 tdb 1.26 proc_state_ptr->nice = kp_stats[i].ki_nice;
515 tdb 1.30 #else
516     proc_state_ptr->proc_size =
517     kp_stats[i].kp_eproc.e_vm.vm_map.size;
518     /* This is in pages */
519     proc_state_ptr->proc_resident =
520     kp_stats[i].kp_eproc.e_vm.vm_rssize * getpagesize();
521 tdb 1.34 #if defined(NETBSD) || defined(OPENBSD)
522 tdb 1.33 proc_state_ptr->time_spent =
523     kp_stats[i].kp_proc.p_rtime.tv_sec;
524 tdb 1.35 #elif defined(DFBSD)
525     proc_state_ptr->time_spent =
526     ( kp_stats[i].kp_thread.td_uticks +
527     kp_stats[i].kp_thread.td_sticks +
528     kp_stats[i].kp_thread.td_iticks ) / 1000000;
529 tdb 1.33 #else
530 tdb 1.36 /* This is in microseconds */
531 tdb 1.30 proc_state_ptr->time_spent =
532     kp_stats[i].kp_proc.p_runtime / 1000000;
533 tdb 1.33 #endif
534 tdb 1.30 proc_state_ptr->cpu_percent =
535     ((double)kp_stats[i].kp_proc.p_pctcpu / FSCALE) * 100.0;
536     proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
537     #endif
538 tdb 1.26
539 tdb 1.55 #ifdef NETBSD2
540     {
541     size_t size;
542     int mib[5];
543    
544     mib[0] = CTL_KERN;
545     mib[1] = KERN_LWP;
546     mib[2] = kp_stats[i].kp_proc.p_pid;
547     mib[3] = sizeof(struct kinfo_lwp);
548     mib[4] = 0;
549    
550     if(sysctl(mib, 5, NULL, &size, NULL, 0) < 0) {
551 ats 1.74 sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.0");
552 tdb 1.55 return NULL;
553     }
554    
555     lwps = size / sizeof(struct kinfo_lwp);
556     mib[4] = lwps;
557    
558 ats 1.61 kl_stats = sg_malloc(size);
559 tdb 1.55 if(kl_stats == NULL) {
560     return NULL;
561     }
562    
563     if(sysctl(mib, 5, kl_stats, &size, NULL, 0) < 0) {
564 ats 1.74 sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.buffersize");
565 tdb 1.55 return NULL;
566     }
567     }
568    
569     switch(kp_stats[i].kp_proc.p_stat) {
570     case SIDL:
571     proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
572     break;
573     case SACTIVE:
574     {
575     int i;
576    
577     for(i = 0; i < lwps; i++) {
578     switch(kl_stats[i].l_stat) {
579     case LSONPROC:
580     case LSRUN:
581     proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
582     goto end;
583     case LSSLEEP:
584     proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
585     goto end;
586     case LSSTOP:
587     case LSSUSPENDED:
588     proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
589     goto end;
590     }
591     proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
592     }
593     end: ;
594     }
595     break;
596     case SSTOP:
597     proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
598     break;
599     case SZOMB:
600     proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
601     break;
602     default:
603     proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
604     break;
605     }
606 tdb 1.73
607     free(kl_stats);
608 tdb 1.55 #else
609 tdb 1.30 #ifdef FREEBSD5
610 tdb 1.26 switch (kp_stats[i].ki_stat) {
611 tdb 1.30 #else
612     switch (kp_stats[i].kp_proc.p_stat) {
613     #endif
614 tdb 1.26 case SIDL:
615     case SRUN:
616     #ifdef SONPROC
617     case SONPROC: /* NetBSD */
618     #endif
619 ats 1.52 proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
620 tdb 1.26 break;
621     case SSLEEP:
622     #ifdef SWAIT
623     case SWAIT: /* FreeBSD 5 */
624     #endif
625     #ifdef SLOCK
626     case SLOCK: /* FreeBSD 5 */
627     #endif
628 ats 1.52 proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
629 tdb 1.26 break;
630     case SSTOP:
631 ats 1.52 proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
632 tdb 1.26 break;
633     case SZOMB:
634     #ifdef SDEAD
635     case SDEAD: /* OpenBSD & NetBSD */
636     #endif
637 ats 1.52 proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
638 tdb 1.31 break;
639     default:
640 ats 1.52 proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
641 tdb 1.26 break;
642     }
643 tdb 1.55 #endif
644 tdb 1.26 proc_state_size++;
645     }
646    
647     free(kp_stats);
648 tdb 1.47 #endif
649    
650     #ifdef CYGWIN
651 tdb 1.65 sg_set_error(SG_ERROR_UNSUPPORTED, "Cygwin");
652 ats 1.52 return NULL;
653 tdb 1.26 #endif
654    
655 ats 1.52 *entries = proc_state_size;
656     return proc_state;
657 tdb 1.27 }
658    
659 ats 1.52 sg_process_count *sg_get_process_count() {
660     static sg_process_count process_stat;
661     sg_process_stats *ps;
662 tdb 1.27 int ps_size, x;
663    
664 tdb 1.30 process_stat.sleeping = 0;
665     process_stat.running = 0;
666     process_stat.zombie = 0;
667     process_stat.stopped = 0;
668     process_stat.total = 0;
669 tdb 1.27
670 ats 1.52 ps = sg_get_process_stats(&ps_size);
671     if (ps == NULL) {
672 tdb 1.32 return NULL;
673     }
674 tdb 1.27
675     for(x = 0; x < ps_size; x++) {
676     switch (ps->state) {
677 ats 1.52 case SG_PROCESS_STATE_RUNNING:
678 tdb 1.27 process_stat.running++;
679     break;
680 ats 1.52 case SG_PROCESS_STATE_SLEEPING:
681 tdb 1.27 process_stat.sleeping++;
682     break;
683 ats 1.52 case SG_PROCESS_STATE_STOPPED:
684 tdb 1.27 process_stat.stopped++;
685     break;
686 ats 1.52 case SG_PROCESS_STATE_ZOMBIE:
687 tdb 1.27 process_stat.zombie++;
688 ats 1.40 break;
689     default:
690 ats 1.52 /* currently no mapping for SG_PROCESS_STATE_UNKNOWN in
691     * sg_process_count */
692 tdb 1.27 break;
693     }
694     ps++;
695     }
696    
697     process_stat.total = ps_size;
698    
699     return &process_stat;
700 ats 1.70 }
701    
702     int sg_process_compare_name(const void *va, const void *vb) {
703     const sg_process_stats *a = (sg_process_stats *)va;
704     const sg_process_stats *b = (sg_process_stats *)vb;
705    
706     return strcmp(a->process_name, b->process_name);
707 pajs 1.66 }
708    
709 pajs 1.67 int sg_process_compare_pid(const void *va, const void *vb) {
710 pajs 1.66 const sg_process_stats *a = (sg_process_stats *)va;
711     const sg_process_stats *b = (sg_process_stats *)vb;
712    
713     if (a->pid < b->pid) {
714     return -1;
715     } else if (a->pid == b->pid) {
716     return 0;
717     } else {
718     return 1;
719     }
720     }
721    
722 pajs 1.67 int sg_process_compare_uid(const void *va, const void *vb) {
723 pajs 1.66 const sg_process_stats *a = (sg_process_stats *)va;
724     const sg_process_stats *b = (sg_process_stats *)vb;
725    
726     if (a->uid < b->uid) {
727     return -1;
728     } else if (a->uid == b->uid) {
729     return 0;
730     } else {
731     return 1;
732     }
733     }
734    
735 pajs 1.67 int sg_process_compare_gid(const void *va, const void *vb) {
736 pajs 1.66 const sg_process_stats *a = (sg_process_stats *)va;
737     const sg_process_stats *b = (sg_process_stats *)vb;
738    
739     if (a->gid < b->gid) {
740     return -1;
741     } else if (a->gid == b->gid) {
742     return 0;
743     } else {
744     return 1;
745     }
746     }
747    
748 pajs 1.67 int sg_process_compare_size(const void *va, const void *vb) {
749 pajs 1.66 const sg_process_stats *a = (sg_process_stats *)va;
750     const sg_process_stats *b = (sg_process_stats *)vb;
751    
752     if (a->proc_size < b->proc_size) {
753     return -1;
754     } else if (a->proc_size == b->proc_size) {
755     return 0;
756     } else {
757     return 1;
758     }
759     }
760    
761 pajs 1.67 int sg_process_compare_res(const void *va, const void *vb) {
762 pajs 1.66 const sg_process_stats *a = (sg_process_stats *)va;
763     const sg_process_stats *b = (sg_process_stats *)vb;
764    
765     if (a->proc_resident < b->proc_resident) {
766     return -1;
767     } else if (a->proc_resident == b->proc_resident) {
768     return 0;
769     } else {
770     return 1;
771     }
772     }
773    
774 pajs 1.67 int sg_process_compare_cpu(const void *va, const void *vb) {
775 pajs 1.66 const sg_process_stats *a = (sg_process_stats *)va;
776     const sg_process_stats *b = (sg_process_stats *)vb;
777    
778     if (a->cpu_percent < b->cpu_percent) {
779     return -1;
780     } else if (a->cpu_percent == b->cpu_percent) {
781     return 0;
782     } else {
783     return 1;
784     }
785     }
786    
787 pajs 1.67 int sg_process_compare_time(const void *va, const void *vb) {
788 pajs 1.66 const sg_process_stats *a = (sg_process_stats *)va;
789     const sg_process_stats *b = (sg_process_stats *)vb;
790    
791     if (a->time_spent < b->time_spent) {
792     return -1;
793     } else if (a->time_spent == b->time_spent) {
794     return 0;
795     } else {
796     return 1;
797     }
798     }
799