51 |
|
#define MAX_FILE_LENGTH PATH_MAX |
52 |
|
#endif |
53 |
|
#ifdef ALLBSD |
54 |
+ |
#include <errno.h> |
55 |
|
#include <stdlib.h> |
56 |
|
#include <sys/param.h> |
57 |
|
#include <sys/sysctl.h> |
100 |
|
int argslen = 0; |
101 |
|
#else |
102 |
|
long buflen; |
103 |
< |
char *p; |
103 |
> |
char *p, *proctitletmp; |
104 |
|
#endif |
105 |
|
#ifdef NETBSD2 |
106 |
|
int lwps; |
362 |
|
} |
363 |
|
|
364 |
|
#if defined(FREEBSD5) || defined(NETBSD) || defined(OPENBSD) |
364 |
– |
size = sizeof(buflen); |
365 |
|
|
366 |
|
#ifdef FREEBSD5 |
367 |
– |
if(sysctlbyname("kern.ps_arg_cache_limit", &buflen, &size, NULL, 0) < 0) { |
368 |
– |
sg_set_error(SG_ERROR_SYSCTLBYNAME, "kern.ps_arg_cache_limit"); |
369 |
– |
return NULL; |
370 |
– |
} |
371 |
– |
#else |
372 |
– |
/* FIXME - this value can be too large on some of |
373 |
– |
the BSD's, which causes sysctl not to return |
374 |
– |
anything. Maybe we need something smaller? */ |
375 |
– |
mib[1] = KERN_ARGMAX; |
376 |
– |
|
377 |
– |
if(sysctl(mib, 2, &buflen, &size, NULL, 0) < 0) { |
378 |
– |
sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_ARGMAX"); |
379 |
– |
return NULL; |
380 |
– |
} |
381 |
– |
#endif |
382 |
– |
|
383 |
– |
proctitle = sg_malloc(buflen); |
384 |
– |
if(proctitle == NULL) { |
385 |
– |
return NULL; |
386 |
– |
} |
387 |
– |
|
388 |
– |
size = buflen; |
389 |
– |
|
390 |
– |
#ifdef FREEBSD5 |
367 |
|
mib[2] = KERN_PROC_ARGS; |
368 |
|
mib[3] = kp_stats[i].ki_pid; |
369 |
|
#else |
373 |
|
#endif |
374 |
|
|
375 |
|
free(proc_state_ptr->proctitle); |
376 |
< |
if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) { |
377 |
< |
free(proctitle); |
378 |
< |
proc_state_ptr->proctitle = NULL; |
379 |
< |
} |
380 |
< |
else if(size > 0) { |
376 |
> |
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 |
|
proc_state_ptr->proctitle = sg_malloc(size+1); |
411 |
|
if(proc_state_ptr->proctitle == NULL) { |
412 |
|
return NULL; |
419 |
|
p += strlen(p) + 1; |
420 |
|
} while (p < proctitle + size); |
421 |
|
free(proctitle); |
422 |
+ |
proctitle = NULL; |
423 |
|
/* remove trailing space */ |
424 |
|
proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = '\0'; |
425 |
|
} |
426 |
|
else { |
427 |
< |
free(proctitle); |
427 |
> |
if(proctitle != NULL) { |
428 |
> |
free(proctitle); |
429 |
> |
proctitle = NULL; |
430 |
> |
} |
431 |
|
proc_state_ptr->proctitle = NULL; |
432 |
|
} |
433 |
|
#else |
434 |
|
free(proc_state_ptr->proctitle); |
435 |
+ |
proc_state_ptr->proctitle = NULL; |
436 |
|
if(kvmd != NULL) { |
437 |
|
args = kvm_getargv(kvmd, &(kp_stats[i]), 0); |
438 |
|
if(args != NULL) { |