ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/src/libstatgrab/process_stats.c
Revision: 1.71
Committed: Thu Apr 8 17:20:51 2004 UTC (20 years, 1 month ago) by tdb
Content type: text/plain
Branch: MAIN
Changes since 1.70: +43 -33 lines
Log Message:
Attempt to rework the command line argument fetching - mallocing 256Kb was
a bit on the large side. Now we start with something sensible and make it
bigger as required.

This produces odd results on OpenBSD.

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 tdb 1.71 * $Id: process_stats.c,v 1.70 2004/04/08 13:44:44 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 tdb 1.65 sg_set_error(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 tdb 1.65 sg_set_error(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 tdb 1.65 sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_PROC.KERN_PROC_ALL");
311 ats 1.52 return NULL;
312 tdb 1.36 }
313    
314     procs = size / sizeof(struct kinfo_proc);
315    
316 ats 1.61 kp_stats = sg_malloc(size);
317 tdb 1.36 if(kp_stats == NULL) {
318 ats 1.52 return NULL;
319 tdb 1.36 }
320 ats 1.54 memset(kp_stats, 0, size);
321 tdb 1.36
322     if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
323 tdb 1.65 sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_PROC.KERN_PROC_ALL");
324 tdb 1.36 free(kp_stats);
325 ats 1.52 return NULL;
326 tdb 1.36 }
327 tdb 1.33
328 tdb 1.37 #if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
329 ats 1.52 kvmd = sg_get_kvm2();
330 tdb 1.33 #endif
331 tdb 1.26
332     for (i = 0; i < procs; i++) {
333 ats 1.48 const char *name;
334 ats 1.54
335 tdb 1.55 #ifdef FREEBSD5
336 ats 1.54 if (kp_stats[i].ki_stat == 0) {
337 tdb 1.55 #else
338     if (kp_stats[i].kp_proc.p_stat == 0) {
339     #endif
340 ats 1.54 /* FreeBSD 5 deliberately overallocates the array that
341     * the sysctl returns, so we'll get a few junk
342     * processes on the end that we have to ignore. (Search
343     * for "overestimate by 5 procs" in
344     * src/sys/kern/kern_proc.c for more details.) */
345     continue;
346     }
347 ats 1.48
348     if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
349 ats 1.52 return NULL;
350 tdb 1.29 }
351 tdb 1.26 proc_state_ptr = proc_state+proc_state_size;
352 tdb 1.30
353     #ifdef FREEBSD5
354 ats 1.48 name = kp_stats[i].ki_comm;
355 tdb 1.35 #elif defined(DFBSD)
356 ats 1.48 name = kp_stats[i].kp_thread.td_comm;
357 tdb 1.30 #else
358 ats 1.48 name = kp_stats[i].kp_proc.p_comm;
359 tdb 1.30 #endif
360 ats 1.53 if (sg_update_string(&proc_state_ptr->process_name, name) < 0) {
361 ats 1.52 return NULL;
362 ats 1.48 }
363 tdb 1.29
364 tdb 1.37 #if defined(FREEBSD5) || defined(NETBSD) || defined(OPENBSD)
365    
366     #ifdef FREEBSD5
367     mib[2] = KERN_PROC_ARGS;
368     mib[3] = kp_stats[i].ki_pid;
369     #else
370     mib[1] = KERN_PROC_ARGS;
371     mib[2] = kp_stats[i].kp_proc.p_pid;
372     mib[3] = KERN_PROC_ARGV;
373     #endif
374    
375 ats 1.48 free(proc_state_ptr->proctitle);
376 tdb 1.71 proc_state_ptr->proctitle = NULL;
377    
378     /* Starting size - we'll double this straight away */
379     #define PROCTITLE_START_SIZE 64
380     buflen = PROCTITLE_START_SIZE;
381     size = buflen;
382     proctitle = NULL;
383    
384     do {
385     if(size >= buflen) {
386     buflen *= 2;
387     size = buflen;
388     proctitletmp = sg_realloc(proctitle, buflen);
389     if(proctitletmp == NULL) {
390     free(proctitle);
391     proctitle = NULL;
392     proc_state_ptr->proctitle = NULL;
393     size = 0;
394     break;
395     }
396     proctitle = proctitletmp;
397     bzero(proctitle, buflen);
398     }
399    
400     if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) {
401     free(proctitle);
402     proctitle = NULL;
403     proc_state_ptr->proctitle = NULL;
404     size = 0;
405     break;
406     }
407     } while(size >= buflen);
408    
409     if(size > 0) {
410 ats 1.61 proc_state_ptr->proctitle = sg_malloc(size+1);
411 tdb 1.37 if(proc_state_ptr->proctitle == NULL) {
412 ats 1.52 return NULL;
413 tdb 1.29 }
414 tdb 1.37 p = proctitle;
415 ats 1.49 proc_state_ptr->proctitle[0] = '\0';
416 tdb 1.37 do {
417 ats 1.52 sg_strlcat(proc_state_ptr->proctitle, p, size+1);
418     sg_strlcat(proc_state_ptr->proctitle, " ", size+1);
419 tdb 1.37 p += strlen(p) + 1;
420     } while (p < proctitle + size);
421     free(proctitle);
422 tdb 1.71 proctitle = NULL;
423 tdb 1.45 /* remove trailing space */
424     proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = '\0';
425 tdb 1.37 }
426     else {
427 tdb 1.71 if(proctitle != NULL) {
428     free(proctitle);
429     proctitle = NULL;
430     }
431 tdb 1.37 proc_state_ptr->proctitle = NULL;
432     }
433     #else
434 ats 1.48 free(proc_state_ptr->proctitle);
435 tdb 1.71 proc_state_ptr->proctitle = NULL;
436 tdb 1.37 if(kvmd != NULL) {
437     args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
438     if(args != NULL) {
439 tdb 1.50 argsp = args;
440     while(*argsp != NULL) {
441 tdb 1.51 argslen += strlen(*argsp) + 1;
442     argsp++;
443 tdb 1.50 }
444 ats 1.61 proctitle = sg_malloc(argslen + 1);
445 tdb 1.50 proctitle[0] = '\0';
446 tdb 1.37 if(proctitle == NULL) {
447 ats 1.52 return NULL;
448 tdb 1.37 }
449     while(*args != NULL) {
450 ats 1.52 sg_strlcat(proctitle, *args, argslen + 1);
451     sg_strlcat(proctitle, " ", argslen + 1);
452 tdb 1.37 args++;
453 tdb 1.29 }
454 tdb 1.37 /* remove trailing space */
455 tdb 1.45 proctitle[strlen(proctitle)-1] = '\0';
456 tdb 1.37 proc_state_ptr->proctitle = proctitle;
457     }
458     else {
459     proc_state_ptr->proctitle = NULL;
460 tdb 1.29 }
461     }
462     else {
463 tdb 1.37 proc_state_ptr->proctitle = NULL;
464 tdb 1.29 }
465 tdb 1.37 #endif
466 tdb 1.26
467 tdb 1.30 #ifdef FREEBSD5
468 tdb 1.26 proc_state_ptr->pid = kp_stats[i].ki_pid;
469     proc_state_ptr->parent = kp_stats[i].ki_ppid;
470     proc_state_ptr->pgid = kp_stats[i].ki_pgid;
471 tdb 1.30 #else
472     proc_state_ptr->pid = kp_stats[i].kp_proc.p_pid;
473     proc_state_ptr->parent = kp_stats[i].kp_eproc.e_ppid;
474     proc_state_ptr->pgid = kp_stats[i].kp_eproc.e_pgid;
475     #endif
476 tdb 1.26
477 tdb 1.30 #ifdef FREEBSD5
478 tdb 1.26 proc_state_ptr->uid = kp_stats[i].ki_ruid;
479     proc_state_ptr->euid = kp_stats[i].ki_uid;
480     proc_state_ptr->gid = kp_stats[i].ki_rgid;
481     proc_state_ptr->egid = kp_stats[i].ki_svgid;
482 tdb 1.35 #elif defined(DFBSD)
483     proc_state_ptr->uid = kp_stats[i].kp_eproc.e_ucred.cr_ruid;
484     proc_state_ptr->euid = kp_stats[i].kp_eproc.e_ucred.cr_svuid;
485     proc_state_ptr->gid = kp_stats[i].kp_eproc.e_ucred.cr_rgid;
486     proc_state_ptr->egid = kp_stats[i].kp_eproc.e_ucred.cr_svgid;
487 tdb 1.30 #else
488     proc_state_ptr->uid = kp_stats[i].kp_eproc.e_pcred.p_ruid;
489     proc_state_ptr->euid = kp_stats[i].kp_eproc.e_pcred.p_svuid;
490     proc_state_ptr->gid = kp_stats[i].kp_eproc.e_pcred.p_rgid;
491     proc_state_ptr->egid = kp_stats[i].kp_eproc.e_pcred.p_svgid;
492     #endif
493 tdb 1.26
494 tdb 1.30 #ifdef FREEBSD5
495 tdb 1.26 proc_state_ptr->proc_size = kp_stats[i].ki_size;
496     /* This is in pages */
497 tdb 1.30 proc_state_ptr->proc_resident =
498     kp_stats[i].ki_rssize * getpagesize();
499 tdb 1.28 /* This is in microseconds */
500 tdb 1.26 proc_state_ptr->time_spent = kp_stats[i].ki_runtime / 1000000;
501 tdb 1.30 proc_state_ptr->cpu_percent =
502     ((double)kp_stats[i].ki_pctcpu / FSCALE) * 100.0;
503 tdb 1.26 proc_state_ptr->nice = kp_stats[i].ki_nice;
504 tdb 1.30 #else
505     proc_state_ptr->proc_size =
506     kp_stats[i].kp_eproc.e_vm.vm_map.size;
507     /* This is in pages */
508     proc_state_ptr->proc_resident =
509     kp_stats[i].kp_eproc.e_vm.vm_rssize * getpagesize();
510 tdb 1.34 #if defined(NETBSD) || defined(OPENBSD)
511 tdb 1.33 proc_state_ptr->time_spent =
512     kp_stats[i].kp_proc.p_rtime.tv_sec;
513 tdb 1.35 #elif defined(DFBSD)
514     proc_state_ptr->time_spent =
515     ( kp_stats[i].kp_thread.td_uticks +
516     kp_stats[i].kp_thread.td_sticks +
517     kp_stats[i].kp_thread.td_iticks ) / 1000000;
518 tdb 1.33 #else
519 tdb 1.36 /* This is in microseconds */
520 tdb 1.30 proc_state_ptr->time_spent =
521     kp_stats[i].kp_proc.p_runtime / 1000000;
522 tdb 1.33 #endif
523 tdb 1.30 proc_state_ptr->cpu_percent =
524     ((double)kp_stats[i].kp_proc.p_pctcpu / FSCALE) * 100.0;
525     proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
526     #endif
527 tdb 1.26
528 tdb 1.55 #ifdef NETBSD2
529     {
530     size_t size;
531     int mib[5];
532    
533     mib[0] = CTL_KERN;
534     mib[1] = KERN_LWP;
535     mib[2] = kp_stats[i].kp_proc.p_pid;
536     mib[3] = sizeof(struct kinfo_lwp);
537     mib[4] = 0;
538    
539     if(sysctl(mib, 5, NULL, &size, NULL, 0) < 0) {
540 tdb 1.65 sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.0");
541 tdb 1.55 return NULL;
542     }
543    
544     lwps = size / sizeof(struct kinfo_lwp);
545     mib[4] = lwps;
546    
547 ats 1.61 kl_stats = sg_malloc(size);
548 tdb 1.55 if(kl_stats == NULL) {
549     return NULL;
550     }
551    
552     if(sysctl(mib, 5, kl_stats, &size, NULL, 0) < 0) {
553 tdb 1.65 sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.buffersize");
554 tdb 1.55 return NULL;
555     }
556     }
557    
558     switch(kp_stats[i].kp_proc.p_stat) {
559     case SIDL:
560     proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
561     break;
562     case SACTIVE:
563     {
564     int i;
565    
566     for(i = 0; i < lwps; i++) {
567     switch(kl_stats[i].l_stat) {
568     case LSONPROC:
569     case LSRUN:
570     proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
571     goto end;
572     case LSSLEEP:
573     proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
574     goto end;
575     case LSSTOP:
576     case LSSUSPENDED:
577     proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
578     goto end;
579     }
580     proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
581     }
582     end: ;
583     }
584     break;
585     case SSTOP:
586     proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
587     break;
588     case SZOMB:
589     proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
590     break;
591     default:
592     proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
593     break;
594     }
595     #else
596 tdb 1.30 #ifdef FREEBSD5
597 tdb 1.26 switch (kp_stats[i].ki_stat) {
598 tdb 1.30 #else
599     switch (kp_stats[i].kp_proc.p_stat) {
600     #endif
601 tdb 1.26 case SIDL:
602     case SRUN:
603     #ifdef SONPROC
604     case SONPROC: /* NetBSD */
605     #endif
606 ats 1.52 proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
607 tdb 1.26 break;
608     case SSLEEP:
609     #ifdef SWAIT
610     case SWAIT: /* FreeBSD 5 */
611     #endif
612     #ifdef SLOCK
613     case SLOCK: /* FreeBSD 5 */
614     #endif
615 ats 1.52 proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
616 tdb 1.26 break;
617     case SSTOP:
618 ats 1.52 proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
619 tdb 1.26 break;
620     case SZOMB:
621     #ifdef SDEAD
622     case SDEAD: /* OpenBSD & NetBSD */
623     #endif
624 ats 1.52 proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
625 tdb 1.31 break;
626     default:
627 ats 1.52 proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
628 tdb 1.26 break;
629     }
630 tdb 1.55 #endif
631 tdb 1.26 proc_state_size++;
632     }
633    
634     free(kp_stats);
635 tdb 1.47 #endif
636    
637     #ifdef CYGWIN
638 tdb 1.65 sg_set_error(SG_ERROR_UNSUPPORTED, "Cygwin");
639 ats 1.52 return NULL;
640 tdb 1.26 #endif
641    
642 ats 1.52 *entries = proc_state_size;
643     return proc_state;
644 tdb 1.27 }
645    
646 ats 1.52 sg_process_count *sg_get_process_count() {
647     static sg_process_count process_stat;
648     sg_process_stats *ps;
649 tdb 1.27 int ps_size, x;
650    
651 tdb 1.30 process_stat.sleeping = 0;
652     process_stat.running = 0;
653     process_stat.zombie = 0;
654     process_stat.stopped = 0;
655     process_stat.total = 0;
656 tdb 1.27
657 ats 1.52 ps = sg_get_process_stats(&ps_size);
658     if (ps == NULL) {
659 tdb 1.32 return NULL;
660     }
661 tdb 1.27
662     for(x = 0; x < ps_size; x++) {
663     switch (ps->state) {
664 ats 1.52 case SG_PROCESS_STATE_RUNNING:
665 tdb 1.27 process_stat.running++;
666     break;
667 ats 1.52 case SG_PROCESS_STATE_SLEEPING:
668 tdb 1.27 process_stat.sleeping++;
669     break;
670 ats 1.52 case SG_PROCESS_STATE_STOPPED:
671 tdb 1.27 process_stat.stopped++;
672     break;
673 ats 1.52 case SG_PROCESS_STATE_ZOMBIE:
674 tdb 1.27 process_stat.zombie++;
675 ats 1.40 break;
676     default:
677 ats 1.52 /* currently no mapping for SG_PROCESS_STATE_UNKNOWN in
678     * sg_process_count */
679 tdb 1.27 break;
680     }
681     ps++;
682     }
683    
684     process_stat.total = ps_size;
685    
686     return &process_stat;
687 ats 1.70 }
688    
689     int sg_process_compare_name(const void *va, const void *vb) {
690     const sg_process_stats *a = (sg_process_stats *)va;
691     const sg_process_stats *b = (sg_process_stats *)vb;
692    
693     return strcmp(a->process_name, b->process_name);
694 pajs 1.66 }
695    
696 pajs 1.67 int sg_process_compare_pid(const void *va, const void *vb) {
697 pajs 1.66 const sg_process_stats *a = (sg_process_stats *)va;
698     const sg_process_stats *b = (sg_process_stats *)vb;
699    
700     if (a->pid < b->pid) {
701     return -1;
702     } else if (a->pid == b->pid) {
703     return 0;
704     } else {
705     return 1;
706     }
707     }
708    
709 pajs 1.67 int sg_process_compare_uid(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->uid < b->uid) {
714     return -1;
715     } else if (a->uid == b->uid) {
716     return 0;
717     } else {
718     return 1;
719     }
720     }
721    
722 pajs 1.67 int sg_process_compare_gid(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->gid < b->gid) {
727     return -1;
728     } else if (a->gid == b->gid) {
729     return 0;
730     } else {
731     return 1;
732     }
733     }
734    
735 pajs 1.67 int sg_process_compare_size(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->proc_size < b->proc_size) {
740     return -1;
741     } else if (a->proc_size == b->proc_size) {
742     return 0;
743     } else {
744     return 1;
745     }
746     }
747    
748 pajs 1.67 int sg_process_compare_res(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_resident < b->proc_resident) {
753     return -1;
754     } else if (a->proc_resident == b->proc_resident) {
755     return 0;
756     } else {
757     return 1;
758     }
759     }
760    
761 pajs 1.67 int sg_process_compare_cpu(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->cpu_percent < b->cpu_percent) {
766     return -1;
767     } else if (a->cpu_percent == b->cpu_percent) {
768     return 0;
769     } else {
770     return 1;
771     }
772     }
773    
774 pajs 1.67 int sg_process_compare_time(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->time_spent < b->time_spent) {
779     return -1;
780     } else if (a->time_spent == b->time_spent) {
781     return 0;
782     } else {
783     return 1;
784     }
785     }
786