ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/src/libstatgrab/process_stats.c
Revision: 1.43
Committed: Sun Apr 4 22:48:50 2004 UTC (20 years, 1 month ago) by tdb
Content type: text/plain
Branch: MAIN
Changes since 1.42: +2 -1 lines
Log Message:
Fix a bunch of warnings on the BSD's.

File Contents

# Content
1 /*
2 * i-scream central monitoring system
3 * http://www.i-scream.org
4 * Copyright (C) 2000-2004 i-scream
5 *
6 * 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 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * 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 *
21 * $Id: process_stats.c,v 1.42 2004/04/04 22:09:32 tdb Exp $
22 */
23
24 #ifdef HAVE_CONFIG_H
25 #include "config.h"
26 #endif
27
28 #include "statgrab.h"
29 #if defined(SOLARIS) || defined(LINUX)
30 #include <stdio.h>
31 #include <stdlib.h>
32 #include <sys/types.h>
33 #include <dirent.h>
34 #include <string.h>
35 #endif
36
37 #ifdef SOLARIS
38 #include <procfs.h>
39 #include <limits.h>
40 #define PROC_LOCATION "/proc"
41 #define MAX_FILE_LENGTH PATH_MAX
42 #endif
43 #ifdef LINUX
44 #include <limits.h>
45 #include <unistd.h>
46 #include <sys/stat.h>
47 #include <fcntl.h>
48 #define PROC_LOCATION "/proc"
49 #define MAX_FILE_LENGTH PATH_MAX
50 #endif
51 #ifdef ALLBSD
52 #include <stdlib.h>
53 #include <sys/param.h>
54 #include <sys/sysctl.h>
55 #if defined(FREEBSD) || defined(DFBSD)
56 #include <sys/user.h>
57 #else
58 #include <sys/proc.h>
59 #endif
60 #include <string.h>
61 #include <paths.h>
62 #include <fcntl.h>
63 #include <limits.h>
64 #if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
65 #include <kvm.h>
66 #include <tools.h>
67 #endif
68 #include <unistd.h>
69 #endif
70
71 int get_proc_snapshot(proc_state_t **ps){
72 proc_state_t *proc_state = NULL;
73 proc_state_t *proc_state_ptr;
74 int proc_state_size = 0;
75 #ifdef ALLBSD
76 int mib[4];
77 size_t size;
78 struct kinfo_proc *kp_stats;
79 int procs, i;
80 char *proctitle;
81 #if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
82 kvm_t *kvmd;
83 char **args;
84 int alloc;
85 #else
86 long buflen;
87 char *p;
88 #endif
89 #endif
90 #if defined(SOLARIS) || defined(LINUX)
91 DIR *proc_dir;
92 struct dirent *dir_entry;
93 char filename[MAX_FILE_LENGTH];
94 FILE *f;
95 #ifdef SOLARIS
96 psinfo_t process_info;
97 #endif
98 #ifdef LINUX
99 char s;
100 /* If someone has a executable of 4k filename length, they deserve to get it truncated :) */
101 char ps_name[4096];
102 char *ptr;
103 static char *psargs = NULL;
104 static int psarg_size = 0;
105 unsigned long stime, utime;
106 int x;
107 int fn;
108 int toread;
109 ssize_t size;
110 int t_read;
111 #endif
112
113 if((proc_dir=opendir(PROC_LOCATION))==NULL){
114 return -1;
115 }
116
117 while((dir_entry=readdir(proc_dir))!=NULL){
118 if(atoi(dir_entry->d_name) == 0) continue;
119
120 #ifdef SOLARIS
121 snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
122 #endif
123 #ifdef LINUX
124 snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/stat", dir_entry->d_name);
125 #endif
126 if((f=fopen(filename, "r"))==NULL){
127 /* Open failed.. Process since vanished, or the path was too long.
128 * Ah well, move onwards to the next one */
129 continue;
130 }
131 #ifdef SOLARIS
132 fread(&process_info, sizeof(psinfo_t), 1, f);
133 #endif
134
135 proc_state = realloc(proc_state, (1+proc_state_size)*sizeof(proc_state_t));
136 proc_state_ptr = proc_state+proc_state_size;
137 #ifdef SOLARIS
138 proc_state_ptr->pid = process_info.pr_pid;
139 proc_state_ptr->parent = process_info.pr_ppid;
140 proc_state_ptr->pgid = process_info.pr_pgid;
141 proc_state_ptr->uid = process_info.pr_uid;
142 proc_state_ptr->euid = process_info.pr_euid;
143 proc_state_ptr->gid = process_info.pr_gid;
144 proc_state_ptr->egid = process_info.pr_egid;
145 proc_state_ptr->proc_size = (process_info.pr_size) * 1024;
146 proc_state_ptr->proc_resident = (process_info.pr_rssize) * 1024;
147 proc_state_ptr->time_spent = process_info.pr_time.tv_sec;
148 proc_state_ptr->cpu_percent = (process_info.pr_pctcpu * 100.0) / 0x8000;
149 proc_state_ptr->process_name = strdup(process_info.pr_fname);
150 proc_state_ptr->proctitle = strdup(process_info.pr_psargs);
151
152 if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SLEEPING;
153 if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = RUNNING;
154 if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = ZOMBIE;
155 if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = STOPPED;
156 if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = RUNNING;
157 #endif
158 #ifdef LINUX
159 x = fscanf(f, "%d %4096s %c %d %d %*d %*d %*d %*u %*u %*u %*u %*u %lu %lu %*d %*d %*d %d %*d %*d %*u %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), &(proc_state_ptr->proc_size), &(proc_state_ptr->proc_resident));
160 proc_state_ptr->proc_resident = proc_state_ptr->proc_resident * getpagesize();
161 if(s == 'S') proc_state_ptr->state = SLEEPING;
162 if(s == 'R') proc_state_ptr->state = RUNNING;
163 if(s == 'Z') proc_state_ptr->state = ZOMBIE;
164 if(s == 'T') proc_state_ptr->state = STOPPED;
165 if(s == 'D') proc_state_ptr->state = STOPPED;
166
167 /* pa_name[0] should = '(' */
168 ptr = strchr(&ps_name[1], ')');
169 if(ptr !=NULL) *ptr='\0';
170 proc_state_ptr->process_name = strdup(&ps_name[1]);
171
172 /* Need to do cpu */
173
174
175 /* proctitle */
176 snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/cmdline", dir_entry->d_name);
177
178 if((fn=open(filename, O_RDONLY)) == -1){
179 /* Open failed.. Process since vanished, or the path was too long.
180 * Ah well, move onwards to the next one */
181 continue;
182 }
183 #define PSARG_START_SIZE 128
184 if(psargs == NULL){
185 psargs = malloc(PSARG_START_SIZE);
186 psarg_size = PSARG_START_SIZE;
187 }
188 ptr = psargs;
189 t_read = 0;
190 toread = psarg_size;
191 while((size = read(fn, ptr, toread)) == toread){
192 psargs = realloc(psargs, (psarg_size + PSARG_START_SIZE));
193 ptr = psargs+psarg_size;
194 t_read = psarg_size;
195 psarg_size+=PSARG_START_SIZE;
196 toread = PSARG_START_SIZE;
197 }
198 if(size != -1) t_read+=size;
199
200 ptr = psargs;
201 for(x=0; x<t_read; x++){
202 if (*ptr == '\0') *ptr = ' ';
203 ptr++;
204 }
205 /* for safety sake */
206 psargs[t_read] = '\0';
207
208 proc_state_ptr->proctitle = strdup(psargs);
209
210 #endif
211
212 proc_state_size++;
213
214 fclose(f);
215 }
216 closedir(proc_dir);
217 #endif
218
219 #ifdef ALLBSD
220 mib[0] = CTL_KERN;
221 mib[1] = KERN_PROC;
222 mib[2] = KERN_PROC_ALL;
223
224 if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
225 return NULL;
226 }
227
228 procs = size / sizeof(struct kinfo_proc);
229
230 kp_stats = malloc(size);
231 if(kp_stats == NULL) {
232 return NULL;
233 }
234
235 if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
236 free(kp_stats);
237 return NULL;
238 }
239
240 #if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
241 kvmd = get_kvm2();
242 #endif
243
244 for (i = 0; i < procs; i++) {
245 /* replace with something more sensible */
246 proc_state = realloc(proc_state,
247 (1+proc_state_size)*sizeof(proc_state_t));
248 if(proc_state == NULL ) {
249 return NULL;
250 }
251 proc_state_ptr = proc_state+proc_state_size;
252
253 #ifdef FREEBSD5
254 proc_state_ptr->process_name =
255 strdup(kp_stats[i].ki_comm);
256 #elif defined(DFBSD)
257 proc_state_ptr->process_name =
258 strdup(kp_stats[i].kp_thread.td_comm);
259 #else
260 proc_state_ptr->process_name =
261 strdup(kp_stats[i].kp_proc.p_comm);
262 #endif
263
264 #if defined(FREEBSD5) || defined(NETBSD) || defined(OPENBSD)
265 size = sizeof(buflen);
266
267 #ifdef FREEBSD5
268 if(sysctlbyname("kern.ps_arg_cache_limit", &buflen, &size, NULL, 0) < 0) {
269 return NULL;
270 }
271 #else
272 mib[1] = KERN_ARGMAX;
273
274 if(sysctl(mib, 2, &buflen, &size, NULL, 0) < 0) {
275 return NULL;
276 }
277 #endif
278
279 proctitle = malloc(buflen);
280 if(proctitle == NULL) {
281 return NULL;
282 }
283
284 size = buflen;
285
286 #ifdef FREEBSD5
287 mib[2] = KERN_PROC_ARGS;
288 mib[3] = kp_stats[i].ki_pid;
289 #else
290 mib[1] = KERN_PROC_ARGS;
291 mib[2] = kp_stats[i].kp_proc.p_pid;
292 mib[3] = KERN_PROC_ARGV;
293 #endif
294
295 if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) {
296 free(proctitle);
297 proc_state_ptr->proctitle = NULL;
298 }
299 else if(size > 0) {
300 proc_state_ptr->proctitle = malloc(size);
301 if(proc_state_ptr->proctitle == NULL) {
302 return NULL;
303 }
304 p = proctitle;
305 proc_state_ptr->proctitle[0] = NULL;
306 do {
307 strncat(proc_state_ptr->proctitle, p, strlen(p));
308 strncat(proc_state_ptr->proctitle, " ", 1);
309 p += strlen(p) + 1;
310 } while (p < proctitle + size);
311 free(proctitle);
312 proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = NULL;
313 }
314 else {
315 free(proctitle);
316 proc_state_ptr->proctitle = NULL;
317 }
318 #else
319 if(kvmd != NULL) {
320 args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
321 if(args != NULL) {
322 alloc = 1;
323 proctitle = malloc(alloc);
324 if(proctitle == NULL) {
325 return NULL;
326 }
327 while(*args != NULL) {
328 if(strlen(proctitle) + strlen(*args) >= alloc) {
329 alloc = (alloc + strlen(*args)) << 1;
330 proctitle = realloc(proctitle, alloc);
331 if(proctitle == NULL) {
332 return NULL;
333 }
334 }
335 strncat(proctitle, *args, strlen(*args));
336 strncat(proctitle, " ", 1);
337 args++;
338 }
339 /* remove trailing space */
340 proctitle[strlen(proctitle)-1] = NULL;
341 proc_state_ptr->proctitle = proctitle;
342 }
343 else {
344 proc_state_ptr->proctitle = NULL;
345 }
346 }
347 else {
348 proc_state_ptr->proctitle = NULL;
349 }
350 #endif
351
352 #ifdef FREEBSD5
353 proc_state_ptr->pid = kp_stats[i].ki_pid;
354 proc_state_ptr->parent = kp_stats[i].ki_ppid;
355 proc_state_ptr->pgid = kp_stats[i].ki_pgid;
356 #else
357 proc_state_ptr->pid = kp_stats[i].kp_proc.p_pid;
358 proc_state_ptr->parent = kp_stats[i].kp_eproc.e_ppid;
359 proc_state_ptr->pgid = kp_stats[i].kp_eproc.e_pgid;
360 #endif
361
362 #ifdef FREEBSD5
363 proc_state_ptr->uid = kp_stats[i].ki_ruid;
364 proc_state_ptr->euid = kp_stats[i].ki_uid;
365 proc_state_ptr->gid = kp_stats[i].ki_rgid;
366 proc_state_ptr->egid = kp_stats[i].ki_svgid;
367 #elif defined(DFBSD)
368 proc_state_ptr->uid = kp_stats[i].kp_eproc.e_ucred.cr_ruid;
369 proc_state_ptr->euid = kp_stats[i].kp_eproc.e_ucred.cr_svuid;
370 proc_state_ptr->gid = kp_stats[i].kp_eproc.e_ucred.cr_rgid;
371 proc_state_ptr->egid = kp_stats[i].kp_eproc.e_ucred.cr_svgid;
372 #else
373 proc_state_ptr->uid = kp_stats[i].kp_eproc.e_pcred.p_ruid;
374 proc_state_ptr->euid = kp_stats[i].kp_eproc.e_pcred.p_svuid;
375 proc_state_ptr->gid = kp_stats[i].kp_eproc.e_pcred.p_rgid;
376 proc_state_ptr->egid = kp_stats[i].kp_eproc.e_pcred.p_svgid;
377 #endif
378
379 #ifdef FREEBSD5
380 proc_state_ptr->proc_size = kp_stats[i].ki_size;
381 /* This is in pages */
382 proc_state_ptr->proc_resident =
383 kp_stats[i].ki_rssize * getpagesize();
384 /* This is in microseconds */
385 proc_state_ptr->time_spent = kp_stats[i].ki_runtime / 1000000;
386 proc_state_ptr->cpu_percent =
387 ((double)kp_stats[i].ki_pctcpu / FSCALE) * 100.0;
388 proc_state_ptr->nice = kp_stats[i].ki_nice;
389 #else
390 proc_state_ptr->proc_size =
391 kp_stats[i].kp_eproc.e_vm.vm_map.size;
392 /* This is in pages */
393 proc_state_ptr->proc_resident =
394 kp_stats[i].kp_eproc.e_vm.vm_rssize * getpagesize();
395 #if defined(NETBSD) || defined(OPENBSD)
396 proc_state_ptr->time_spent =
397 kp_stats[i].kp_proc.p_rtime.tv_sec;
398 #elif defined(DFBSD)
399 proc_state_ptr->time_spent =
400 ( kp_stats[i].kp_thread.td_uticks +
401 kp_stats[i].kp_thread.td_sticks +
402 kp_stats[i].kp_thread.td_iticks ) / 1000000;
403 #else
404 /* This is in microseconds */
405 proc_state_ptr->time_spent =
406 kp_stats[i].kp_proc.p_runtime / 1000000;
407 #endif
408 proc_state_ptr->cpu_percent =
409 ((double)kp_stats[i].kp_proc.p_pctcpu / FSCALE) * 100.0;
410 proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
411 #endif
412
413 #ifdef FREEBSD5
414 switch (kp_stats[i].ki_stat) {
415 #else
416 switch (kp_stats[i].kp_proc.p_stat) {
417 #endif
418 case SIDL:
419 case SRUN:
420 #ifdef SONPROC
421 case SONPROC: /* NetBSD */
422 #endif
423 proc_state_ptr->state = RUNNING;
424 break;
425 case SSLEEP:
426 #ifdef SWAIT
427 case SWAIT: /* FreeBSD 5 */
428 #endif
429 #ifdef SLOCK
430 case SLOCK: /* FreeBSD 5 */
431 #endif
432 proc_state_ptr->state = SLEEPING;
433 break;
434 case SSTOP:
435 proc_state_ptr->state = STOPPED;
436 break;
437 case SZOMB:
438 #ifdef SDEAD
439 case SDEAD: /* OpenBSD & NetBSD */
440 #endif
441 proc_state_ptr->state = ZOMBIE;
442 break;
443 default:
444 proc_state_ptr->state = UNKNOWN;
445 break;
446 }
447 proc_state_size++;
448 }
449
450 free(kp_stats);
451 #endif
452
453 *ps = proc_state;
454 return proc_state_size;
455 }
456
457 process_stat_t *get_process_stats() {
458 static process_stat_t process_stat;
459 proc_state_t *ps;
460 int ps_size, x;
461
462 process_stat.sleeping = 0;
463 process_stat.running = 0;
464 process_stat.zombie = 0;
465 process_stat.stopped = 0;
466 process_stat.total = 0;
467
468 ps_size = get_proc_snapshot(&ps);
469
470 if(ps_size == 0) {
471 return NULL;
472 }
473
474 for(x = 0; x < ps_size; x++) {
475 switch (ps->state) {
476 case RUNNING:
477 process_stat.running++;
478 break;
479 case SLEEPING:
480 process_stat.sleeping++;
481 break;
482 case STOPPED:
483 process_stat.stopped++;
484 break;
485 case ZOMBIE:
486 process_stat.zombie++;
487 break;
488 default:
489 /* currently no mapping for UNKNOWN in process_stat_t */
490 break;
491 }
492 ps++;
493 }
494
495 process_stat.total = ps_size;
496
497 return &process_stat;
498 }