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; |
131 |
|
|
132 |
|
#ifdef LINUX |
133 |
|
if ((f=fopen("/proc/uptime", "r")) == NULL) { |
134 |
< |
sg_set_error(SG_ERROR_OPEN, "/proc/uptime"); |
134 |
> |
sg_set_error_with_errno(SG_ERROR_OPEN, "/proc/uptime"); |
135 |
|
return NULL; |
136 |
|
} |
137 |
|
if((fscanf(f,"%lu %*d",&uptime)) != 1){ |
142 |
|
#endif |
143 |
|
|
144 |
|
if((proc_dir=opendir(PROC_LOCATION))==NULL){ |
145 |
< |
sg_set_error(SG_ERROR_OPENDIR, PROC_LOCATION); |
145 |
> |
sg_set_error_with_errno(SG_ERROR_OPENDIR, PROC_LOCATION); |
146 |
|
return NULL; |
147 |
|
} |
148 |
|
|
307 |
|
mib[2] = KERN_PROC_ALL; |
308 |
|
|
309 |
|
if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) { |
310 |
< |
sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_PROC.KERN_PROC_ALL"); |
310 |
> |
sg_set_error_with_errno(SG_ERROR_SYSCTL, |
311 |
> |
"CTL_KERN.KERN_PROC.KERN_PROC_ALL"); |
312 |
|
return NULL; |
313 |
|
} |
314 |
|
|
321 |
|
memset(kp_stats, 0, size); |
322 |
|
|
323 |
|
if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) { |
324 |
< |
sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_PROC.KERN_PROC_ALL"); |
324 |
> |
sg_set_error_with_errno(SG_ERROR_SYSCTL, |
325 |
> |
"CTL_KERN.KERN_PROC.KERN_PROC_ALL"); |
326 |
|
free(kp_stats); |
327 |
|
return NULL; |
328 |
|
} |
364 |
|
} |
365 |
|
|
366 |
|
#if defined(FREEBSD5) || defined(NETBSD) || defined(OPENBSD) |
364 |
– |
size = sizeof(buflen); |
367 |
|
|
368 |
|
#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 |
369 |
|
mib[2] = KERN_PROC_ARGS; |
370 |
|
mib[3] = kp_stats[i].ki_pid; |
371 |
|
#else |
375 |
|
#endif |
376 |
|
|
377 |
|
free(proc_state_ptr->proctitle); |
378 |
< |
if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) { |
379 |
< |
free(proctitle); |
380 |
< |
proc_state_ptr->proctitle = NULL; |
381 |
< |
} |
382 |
< |
else if(size > 0) { |
378 |
> |
proc_state_ptr->proctitle = NULL; |
379 |
> |
|
380 |
> |
/* Starting size - we'll double this straight away */ |
381 |
> |
#define PROCTITLE_START_SIZE 64 |
382 |
> |
buflen = PROCTITLE_START_SIZE; |
383 |
> |
size = buflen; |
384 |
> |
proctitle = NULL; |
385 |
> |
|
386 |
> |
do { |
387 |
> |
if(size >= buflen) { |
388 |
> |
buflen *= 2; |
389 |
> |
size = buflen; |
390 |
> |
proctitletmp = sg_realloc(proctitle, buflen); |
391 |
> |
if(proctitletmp == NULL) { |
392 |
> |
free(proctitle); |
393 |
> |
proctitle = NULL; |
394 |
> |
proc_state_ptr->proctitle = NULL; |
395 |
> |
size = 0; |
396 |
> |
break; |
397 |
> |
} |
398 |
> |
proctitle = proctitletmp; |
399 |
> |
bzero(proctitle, buflen); |
400 |
> |
} |
401 |
> |
|
402 |
> |
if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) { |
403 |
> |
free(proctitle); |
404 |
> |
proctitle = NULL; |
405 |
> |
proc_state_ptr->proctitle = NULL; |
406 |
> |
size = 0; |
407 |
> |
break; |
408 |
> |
} |
409 |
> |
} while(size >= buflen); |
410 |
> |
|
411 |
> |
if(size > 0) { |
412 |
|
proc_state_ptr->proctitle = sg_malloc(size+1); |
413 |
|
if(proc_state_ptr->proctitle == NULL) { |
414 |
|
return NULL; |
415 |
|
} |
416 |
|
p = proctitle; |
417 |
+ |
#ifdef OPENBSD |
418 |
+ |
/* On OpenBSD, this value has the argv pointers (which |
419 |
+ |
* are terminated by a NULL) at the front, so we have |
420 |
+ |
* to skip over them to get to the strings. */ |
421 |
+ |
while (*(char ***)p != NULL) { |
422 |
+ |
p += sizeof(char **); |
423 |
+ |
} |
424 |
+ |
p += sizeof(char **); |
425 |
+ |
#endif |
426 |
|
proc_state_ptr->proctitle[0] = '\0'; |
427 |
|
do { |
428 |
|
sg_strlcat(proc_state_ptr->proctitle, p, size+1); |
430 |
|
p += strlen(p) + 1; |
431 |
|
} while (p < proctitle + size); |
432 |
|
free(proctitle); |
433 |
+ |
proctitle = NULL; |
434 |
|
/* remove trailing space */ |
435 |
|
proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = '\0'; |
436 |
|
} |
437 |
|
else { |
438 |
< |
free(proctitle); |
438 |
> |
if(proctitle != NULL) { |
439 |
> |
free(proctitle); |
440 |
> |
proctitle = NULL; |
441 |
> |
} |
442 |
|
proc_state_ptr->proctitle = NULL; |
443 |
|
} |
444 |
|
#else |
445 |
|
free(proc_state_ptr->proctitle); |
446 |
+ |
proc_state_ptr->proctitle = NULL; |
447 |
|
if(kvmd != NULL) { |
448 |
|
args = kvm_getargv(kvmd, &(kp_stats[i]), 0); |
449 |
|
if(args != NULL) { |
548 |
|
mib[4] = 0; |
549 |
|
|
550 |
|
if(sysctl(mib, 5, NULL, &size, NULL, 0) < 0) { |
551 |
< |
sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.0"); |
551 |
> |
sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.0"); |
552 |
|
return NULL; |
553 |
|
} |
554 |
|
|
561 |
|
} |
562 |
|
|
563 |
|
if(sysctl(mib, 5, kl_stats, &size, NULL, 0) < 0) { |
564 |
< |
sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.buffersize"); |
564 |
> |
sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.buffersize"); |
565 |
|
return NULL; |
566 |
|
} |
567 |
|
} |
603 |
|
proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN; |
604 |
|
break; |
605 |
|
} |
606 |
+ |
|
607 |
+ |
free(kl_stats); |
608 |
|
#else |
609 |
|
#ifdef FREEBSD5 |
610 |
|
switch (kp_stats[i].ki_stat) { |
699 |
|
return &process_stat; |
700 |
|
} |
701 |
|
|
702 |
< |
int sg_compare_pid(const void *va, const void *vb) { |
702 |
> |
int sg_process_compare_name(const void *va, const void *vb) { |
703 |
|
const sg_process_stats *a = (sg_process_stats *)va; |
704 |
|
const sg_process_stats *b = (sg_process_stats *)vb; |
705 |
|
|
706 |
+ |
return strcmp(a->process_name, b->process_name); |
707 |
+ |
} |
708 |
+ |
|
709 |
+ |
int sg_process_compare_pid(const void *va, const void *vb) { |
710 |
+ |
const sg_process_stats *a = (sg_process_stats *)va; |
711 |
+ |
const sg_process_stats *b = (sg_process_stats *)vb; |
712 |
+ |
|
713 |
|
if (a->pid < b->pid) { |
714 |
|
return -1; |
715 |
|
} else if (a->pid == b->pid) { |
719 |
|
} |
720 |
|
} |
721 |
|
|
722 |
< |
int sg_compare_uid(const void *va, const void *vb) { |
722 |
> |
int sg_process_compare_uid(const void *va, const void *vb) { |
723 |
|
const sg_process_stats *a = (sg_process_stats *)va; |
724 |
|
const sg_process_stats *b = (sg_process_stats *)vb; |
725 |
|
|
732 |
|
} |
733 |
|
} |
734 |
|
|
735 |
< |
int sg_compare_gid(const void *va, const void *vb) { |
735 |
> |
int sg_process_compare_gid(const void *va, const void *vb) { |
736 |
|
const sg_process_stats *a = (sg_process_stats *)va; |
737 |
|
const sg_process_stats *b = (sg_process_stats *)vb; |
738 |
|
|
745 |
|
} |
746 |
|
} |
747 |
|
|
748 |
< |
int sg_compare_size(const void *va, const void *vb) { |
748 |
> |
int sg_process_compare_size(const void *va, const void *vb) { |
749 |
|
const sg_process_stats *a = (sg_process_stats *)va; |
750 |
|
const sg_process_stats *b = (sg_process_stats *)vb; |
751 |
|
|
758 |
|
} |
759 |
|
} |
760 |
|
|
761 |
< |
int sg_compare_res(const void *va, const void *vb) { |
761 |
> |
int sg_process_compare_res(const void *va, const void *vb) { |
762 |
|
const sg_process_stats *a = (sg_process_stats *)va; |
763 |
|
const sg_process_stats *b = (sg_process_stats *)vb; |
764 |
|
|
771 |
|
} |
772 |
|
} |
773 |
|
|
774 |
< |
int sg_compare_cpu(const void *va, const void *vb) { |
774 |
> |
int sg_process_compare_cpu(const void *va, const void *vb) { |
775 |
|
const sg_process_stats *a = (sg_process_stats *)va; |
776 |
|
const sg_process_stats *b = (sg_process_stats *)vb; |
777 |
|
|
784 |
|
} |
785 |
|
} |
786 |
|
|
787 |
< |
int sg_compare_time(const void *va, const void *vb) { |
787 |
> |
int sg_process_compare_time(const void *va, const void *vb) { |
788 |
|
const sg_process_stats *a = (sg_process_stats *)va; |
789 |
|
const sg_process_stats *b = (sg_process_stats *)vb; |
790 |
|
|
797 |
|
} |
798 |
|
} |
799 |
|
|
770 |
– |
void sg_sort(sg_process_stats *ps, int num_ps, sg_sort_method sm){ |
771 |
– |
int (*sortby_ptr)(const void *a, const void *b); |
772 |
– |
|
773 |
– |
switch(sm){ |
774 |
– |
case SG_PS_PID: |
775 |
– |
sortby_ptr = sg_compare_pid; |
776 |
– |
break; |
777 |
– |
case SG_PS_UID: |
778 |
– |
sortby_ptr = sg_compare_uid; |
779 |
– |
break; |
780 |
– |
case SG_PS_GID: |
781 |
– |
sortby_ptr = sg_compare_gid; |
782 |
– |
break; |
783 |
– |
case SG_PS_SIZE: |
784 |
– |
sortby_ptr = sg_compare_size; |
785 |
– |
break; |
786 |
– |
case SG_PS_RES: |
787 |
– |
sortby_ptr = sg_compare_res; |
788 |
– |
break; |
789 |
– |
case SG_PS_CPU: |
790 |
– |
sortby_ptr = sg_compare_cpu; |
791 |
– |
break; |
792 |
– |
case SG_PS_TIME: |
793 |
– |
sortby_ptr = sg_compare_time; |
794 |
– |
break; |
795 |
– |
} |
796 |
– |
|
797 |
– |
qsort(ps, num_ps, sizeof(*ps), sortby_ptr); |
798 |
– |
|
799 |
– |
return; |
800 |
– |
} |