ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/src/libstatgrab/process_stats.c
Revision: 1.48
Committed: Mon Apr 5 01:41:25 2004 UTC (20 years, 1 month ago) by ats
Content type: text/plain
Branch: MAIN
Changes since 1.47: +65 -42 lines
Log Message:
Use vectors and update_string. (I haven't really touched the BSD code
here yet.)

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