ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/src/libstatgrab/process_stats.c
Revision: 1.49
Committed: Mon Apr 5 01:44:20 2004 UTC (20 years, 1 month ago) by ats
Content type: text/plain
Branch: MAIN
Changes since 1.48: +4 -2 lines
Log Message:
NULL -> \0, initialise malloced memory to \0, and plug a memory leak.

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.48 2004/04/05 01:41:25 ats 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] = '\0';
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 proctitle[0] = '\0';
353 while(*args != NULL) {
354 if(strlen(proctitle) + strlen(*args) >= alloc) {
355 alloc = (alloc + strlen(*args)) * 2;
356 proctitletmp = realloc(proctitle, alloc);
357 if(proctitletmp == NULL) {
358 free(proctitle);
359 return -1;
360 }
361 proctitle = proctitletmp;
362 }
363 strlcat(proctitle, *args, alloc);
364 strlcat(proctitle, " ", alloc);
365 args++;
366 }
367 /* remove trailing space */
368 proctitle[strlen(proctitle)-1] = '\0';
369 proc_state_ptr->proctitle = proctitle;
370 }
371 else {
372 proc_state_ptr->proctitle = NULL;
373 }
374 }
375 else {
376 proc_state_ptr->proctitle = NULL;
377 }
378 #endif
379
380 #ifdef FREEBSD5
381 proc_state_ptr->pid = kp_stats[i].ki_pid;
382 proc_state_ptr->parent = kp_stats[i].ki_ppid;
383 proc_state_ptr->pgid = kp_stats[i].ki_pgid;
384 #else
385 proc_state_ptr->pid = kp_stats[i].kp_proc.p_pid;
386 proc_state_ptr->parent = kp_stats[i].kp_eproc.e_ppid;
387 proc_state_ptr->pgid = kp_stats[i].kp_eproc.e_pgid;
388 #endif
389
390 #ifdef FREEBSD5
391 proc_state_ptr->uid = kp_stats[i].ki_ruid;
392 proc_state_ptr->euid = kp_stats[i].ki_uid;
393 proc_state_ptr->gid = kp_stats[i].ki_rgid;
394 proc_state_ptr->egid = kp_stats[i].ki_svgid;
395 #elif defined(DFBSD)
396 proc_state_ptr->uid = kp_stats[i].kp_eproc.e_ucred.cr_ruid;
397 proc_state_ptr->euid = kp_stats[i].kp_eproc.e_ucred.cr_svuid;
398 proc_state_ptr->gid = kp_stats[i].kp_eproc.e_ucred.cr_rgid;
399 proc_state_ptr->egid = kp_stats[i].kp_eproc.e_ucred.cr_svgid;
400 #else
401 proc_state_ptr->uid = kp_stats[i].kp_eproc.e_pcred.p_ruid;
402 proc_state_ptr->euid = kp_stats[i].kp_eproc.e_pcred.p_svuid;
403 proc_state_ptr->gid = kp_stats[i].kp_eproc.e_pcred.p_rgid;
404 proc_state_ptr->egid = kp_stats[i].kp_eproc.e_pcred.p_svgid;
405 #endif
406
407 #ifdef FREEBSD5
408 proc_state_ptr->proc_size = kp_stats[i].ki_size;
409 /* This is in pages */
410 proc_state_ptr->proc_resident =
411 kp_stats[i].ki_rssize * getpagesize();
412 /* This is in microseconds */
413 proc_state_ptr->time_spent = kp_stats[i].ki_runtime / 1000000;
414 proc_state_ptr->cpu_percent =
415 ((double)kp_stats[i].ki_pctcpu / FSCALE) * 100.0;
416 proc_state_ptr->nice = kp_stats[i].ki_nice;
417 #else
418 proc_state_ptr->proc_size =
419 kp_stats[i].kp_eproc.e_vm.vm_map.size;
420 /* This is in pages */
421 proc_state_ptr->proc_resident =
422 kp_stats[i].kp_eproc.e_vm.vm_rssize * getpagesize();
423 #if defined(NETBSD) || defined(OPENBSD)
424 proc_state_ptr->time_spent =
425 kp_stats[i].kp_proc.p_rtime.tv_sec;
426 #elif defined(DFBSD)
427 proc_state_ptr->time_spent =
428 ( kp_stats[i].kp_thread.td_uticks +
429 kp_stats[i].kp_thread.td_sticks +
430 kp_stats[i].kp_thread.td_iticks ) / 1000000;
431 #else
432 /* This is in microseconds */
433 proc_state_ptr->time_spent =
434 kp_stats[i].kp_proc.p_runtime / 1000000;
435 #endif
436 proc_state_ptr->cpu_percent =
437 ((double)kp_stats[i].kp_proc.p_pctcpu / FSCALE) * 100.0;
438 proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
439 #endif
440
441 #ifdef FREEBSD5
442 switch (kp_stats[i].ki_stat) {
443 #else
444 switch (kp_stats[i].kp_proc.p_stat) {
445 #endif
446 case SIDL:
447 case SRUN:
448 #ifdef SONPROC
449 case SONPROC: /* NetBSD */
450 #endif
451 proc_state_ptr->state = RUNNING;
452 break;
453 case SSLEEP:
454 #ifdef SWAIT
455 case SWAIT: /* FreeBSD 5 */
456 #endif
457 #ifdef SLOCK
458 case SLOCK: /* FreeBSD 5 */
459 #endif
460 proc_state_ptr->state = SLEEPING;
461 break;
462 case SSTOP:
463 proc_state_ptr->state = STOPPED;
464 break;
465 case SZOMB:
466 #ifdef SDEAD
467 case SDEAD: /* OpenBSD & NetBSD */
468 #endif
469 proc_state_ptr->state = ZOMBIE;
470 break;
471 default:
472 proc_state_ptr->state = UNKNOWN;
473 break;
474 }
475 proc_state_size++;
476 }
477
478 free(kp_stats);
479 #endif
480
481 #ifdef CYGWIN
482 return -1;
483 #endif
484
485 *ps = proc_state;
486 return proc_state_size;
487 }
488
489 process_stat_t *get_process_stats() {
490 static process_stat_t process_stat;
491 proc_state_t *ps;
492 int ps_size, x;
493
494 process_stat.sleeping = 0;
495 process_stat.running = 0;
496 process_stat.zombie = 0;
497 process_stat.stopped = 0;
498 process_stat.total = 0;
499
500 ps_size = get_proc_snapshot(&ps);
501
502 if(ps_size < 0) {
503 return NULL;
504 }
505
506 for(x = 0; x < ps_size; x++) {
507 switch (ps->state) {
508 case RUNNING:
509 process_stat.running++;
510 break;
511 case SLEEPING:
512 process_stat.sleeping++;
513 break;
514 case STOPPED:
515 process_stat.stopped++;
516 break;
517 case ZOMBIE:
518 process_stat.zombie++;
519 break;
520 default:
521 /* currently no mapping for UNKNOWN in process_stat_t */
522 break;
523 }
524 ps++;
525 }
526
527 process_stat.total = ps_size;
528
529 return &process_stat;
530 }