ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/src/libstatgrab/process_stats.c
Revision: 1.45
Committed: Sun Apr 4 23:35:45 2004 UTC (20 years, 1 month ago) by tdb
Content type: text/plain
Branch: MAIN
Changes since 1.44: +6 -5 lines
Log Message:
Minor bugfixes.

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