ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/src/libstatgrab/process_stats.c
Revision: 1.75
Committed: Fri Sep 10 15:15:00 2004 UTC (19 years, 8 months ago) by ats
Content type: text/plain
Branch: MAIN
Changes since 1.74: +31 -11 lines
Log Message:
Fix some nasty repeated if()s (reported by Fred Barnes).

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