33 |
|
#include <stdlib.h> |
34 |
|
#include <sys/types.h> |
35 |
|
#include <dirent.h> |
36 |
– |
#include <string.h> |
36 |
|
#endif |
37 |
+ |
#include <string.h> |
38 |
|
|
39 |
|
#ifdef SOLARIS |
40 |
|
#include <procfs.h> |
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> |
72 |
|
#include <sys/lwp.h> |
73 |
|
#endif |
74 |
|
#endif |
75 |
+ |
#ifdef HPUX |
76 |
+ |
#include <sys/param.h> |
77 |
+ |
#include <sys/pstat.h> |
78 |
+ |
#include <unistd.h> |
79 |
+ |
#define PROCESS_BATCH 30 |
80 |
+ |
#endif |
81 |
|
|
82 |
|
static void proc_state_init(sg_process_stats *s) { |
83 |
|
s->process_name = NULL; |
94 |
|
proc_state_init, proc_state_destroy); |
95 |
|
int proc_state_size = 0; |
96 |
|
sg_process_stats *proc_state_ptr; |
97 |
+ |
#ifdef HPUX |
98 |
+ |
struct pst_status pstat_procinfo[PROCESS_BATCH]; |
99 |
+ |
long procidx = 0; |
100 |
+ |
long long pagesize; |
101 |
+ |
int num, i; |
102 |
+ |
#endif |
103 |
|
#ifdef ALLBSD |
104 |
|
int mib[4]; |
105 |
|
size_t size; |
112 |
|
int argslen = 0; |
113 |
|
#else |
114 |
|
long buflen; |
115 |
< |
char *p; |
115 |
> |
char *p, *proctitletmp; |
116 |
|
#endif |
117 |
|
#ifdef NETBSD2 |
118 |
|
int lwps; |
143 |
|
|
144 |
|
#ifdef LINUX |
145 |
|
if ((f=fopen("/proc/uptime", "r")) == NULL) { |
146 |
< |
sg_set_error(SG_ERROR_OPEN, "/proc/uptime"); |
146 |
> |
sg_set_error_with_errno(SG_ERROR_OPEN, "/proc/uptime"); |
147 |
|
return NULL; |
148 |
|
} |
149 |
|
if((fscanf(f,"%lu %*d",&uptime)) != 1){ |
154 |
|
#endif |
155 |
|
|
156 |
|
if((proc_dir=opendir(PROC_LOCATION))==NULL){ |
157 |
< |
sg_set_error(SG_ERROR_OPENDIR, PROC_LOCATION); |
157 |
> |
sg_set_error_with_errno(SG_ERROR_OPENDIR, PROC_LOCATION); |
158 |
|
return NULL; |
159 |
|
} |
160 |
|
|
204 |
|
return NULL; |
205 |
|
} |
206 |
|
|
207 |
< |
if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING; |
208 |
< |
if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING; |
209 |
< |
if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE; |
210 |
< |
if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = SG_PROCESS_STATE_STOPPED; |
211 |
< |
if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING; |
207 |
> |
switch (process_info.pr_lwp.pr_state) { |
208 |
> |
case 1: |
209 |
> |
proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING; |
210 |
> |
break; |
211 |
> |
case 2: |
212 |
> |
case 5: |
213 |
> |
proc_state_ptr->state = SG_PROCESS_STATE_RUNNING; |
214 |
> |
break; |
215 |
> |
case 3: |
216 |
> |
proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE; |
217 |
> |
break; |
218 |
> |
case 4: |
219 |
> |
proc_state_ptr->state = SG_PROCESS_STATE_STOPPED; |
220 |
> |
break; |
221 |
> |
} |
222 |
|
#endif |
223 |
|
#ifdef LINUX |
224 |
|
x = fscanf(f, "%d %4096s %c %d %d %*d %*d %*d %*u %*u %*u %*u %*u %lu %lu %*d %*d %*d %d %*d %*d %lu %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), &starttime, &(proc_state_ptr->proc_size), &(proc_state_ptr->proc_resident)); |
225 |
|
/* +3 becuase man page says "Resident Set Size: number of pages the process has in real memory, minus 3 for administrative purposes." */ |
226 |
|
proc_state_ptr->proc_resident = (proc_state_ptr->proc_resident + 3) * getpagesize(); |
227 |
< |
if(s == 'S') proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING; |
228 |
< |
if(s == 'R') proc_state_ptr->state = SG_PROCESS_STATE_RUNNING; |
229 |
< |
if(s == 'Z') proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE; |
230 |
< |
if(s == 'T') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED; |
231 |
< |
if(s == 'D') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED; |
227 |
> |
switch (s) { |
228 |
> |
case 'S': |
229 |
> |
proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING; |
230 |
> |
break; |
231 |
> |
case 'R': |
232 |
> |
proc_state_ptr->state = SG_PROCESS_STATE_RUNNING; |
233 |
> |
break; |
234 |
> |
case 'Z': |
235 |
> |
proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE; |
236 |
> |
break; |
237 |
> |
case 'T': |
238 |
> |
case 'D': |
239 |
> |
proc_state_ptr->state = SG_PROCESS_STATE_STOPPED; |
240 |
> |
break; |
241 |
> |
} |
242 |
|
|
243 |
|
/* pa_name[0] should = '(' */ |
244 |
|
ptr = strchr(&ps_name[1], ')'); |
339 |
|
mib[2] = KERN_PROC_ALL; |
340 |
|
|
341 |
|
if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) { |
342 |
< |
sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_PROC.KERN_PROC_ALL"); |
342 |
> |
sg_set_error_with_errno(SG_ERROR_SYSCTL, |
343 |
> |
"CTL_KERN.KERN_PROC.KERN_PROC_ALL"); |
344 |
|
return NULL; |
345 |
|
} |
346 |
|
|
353 |
|
memset(kp_stats, 0, size); |
354 |
|
|
355 |
|
if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) { |
356 |
< |
sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_PROC.KERN_PROC_ALL"); |
356 |
> |
sg_set_error_with_errno(SG_ERROR_SYSCTL, |
357 |
> |
"CTL_KERN.KERN_PROC.KERN_PROC_ALL"); |
358 |
|
free(kp_stats); |
359 |
|
return NULL; |
360 |
|
} |
396 |
|
} |
397 |
|
|
398 |
|
#if defined(FREEBSD5) || defined(NETBSD) || defined(OPENBSD) |
364 |
– |
size = sizeof(buflen); |
399 |
|
|
400 |
|
#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 |
401 |
|
mib[2] = KERN_PROC_ARGS; |
402 |
|
mib[3] = kp_stats[i].ki_pid; |
403 |
|
#else |
407 |
|
#endif |
408 |
|
|
409 |
|
free(proc_state_ptr->proctitle); |
410 |
< |
if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) { |
411 |
< |
free(proctitle); |
412 |
< |
proc_state_ptr->proctitle = NULL; |
413 |
< |
} |
414 |
< |
else if(size > 0) { |
410 |
> |
proc_state_ptr->proctitle = NULL; |
411 |
> |
|
412 |
> |
/* Starting size - we'll double this straight away */ |
413 |
> |
#define PROCTITLE_START_SIZE 64 |
414 |
> |
buflen = PROCTITLE_START_SIZE; |
415 |
> |
size = buflen; |
416 |
> |
proctitle = NULL; |
417 |
> |
|
418 |
> |
do { |
419 |
> |
if((long) size >= buflen) { |
420 |
> |
buflen *= 2; |
421 |
> |
size = buflen; |
422 |
> |
proctitletmp = sg_realloc(proctitle, buflen); |
423 |
> |
if(proctitletmp == NULL) { |
424 |
> |
free(proctitle); |
425 |
> |
proctitle = NULL; |
426 |
> |
proc_state_ptr->proctitle = NULL; |
427 |
> |
size = 0; |
428 |
> |
break; |
429 |
> |
} |
430 |
> |
proctitle = proctitletmp; |
431 |
> |
bzero(proctitle, buflen); |
432 |
> |
} |
433 |
> |
|
434 |
> |
if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) { |
435 |
> |
free(proctitle); |
436 |
> |
proctitle = NULL; |
437 |
> |
proc_state_ptr->proctitle = NULL; |
438 |
> |
size = 0; |
439 |
> |
break; |
440 |
> |
} |
441 |
> |
} while((long) size >= buflen); |
442 |
> |
|
443 |
> |
if(size > 0) { |
444 |
|
proc_state_ptr->proctitle = sg_malloc(size+1); |
445 |
|
if(proc_state_ptr->proctitle == NULL) { |
446 |
|
return NULL; |
447 |
|
} |
448 |
|
p = proctitle; |
449 |
+ |
#ifdef OPENBSD |
450 |
+ |
/* On OpenBSD, this value has the argv pointers (which |
451 |
+ |
* are terminated by a NULL) at the front, so we have |
452 |
+ |
* to skip over them to get to the strings. */ |
453 |
+ |
while (*(char ***)p != NULL) { |
454 |
+ |
p += sizeof(char **); |
455 |
+ |
} |
456 |
+ |
p += sizeof(char **); |
457 |
+ |
#endif |
458 |
|
proc_state_ptr->proctitle[0] = '\0'; |
459 |
|
do { |
460 |
|
sg_strlcat(proc_state_ptr->proctitle, p, size+1); |
462 |
|
p += strlen(p) + 1; |
463 |
|
} while (p < proctitle + size); |
464 |
|
free(proctitle); |
465 |
+ |
proctitle = NULL; |
466 |
|
/* remove trailing space */ |
467 |
|
proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = '\0'; |
468 |
|
} |
469 |
|
else { |
470 |
< |
free(proctitle); |
470 |
> |
if(proctitle != NULL) { |
471 |
> |
free(proctitle); |
472 |
> |
proctitle = NULL; |
473 |
> |
} |
474 |
|
proc_state_ptr->proctitle = NULL; |
475 |
|
} |
476 |
|
#else |
477 |
|
free(proc_state_ptr->proctitle); |
478 |
+ |
proc_state_ptr->proctitle = NULL; |
479 |
|
if(kvmd != NULL) { |
480 |
|
args = kvm_getargv(kvmd, &(kp_stats[i]), 0); |
481 |
|
if(args != NULL) { |
580 |
|
mib[4] = 0; |
581 |
|
|
582 |
|
if(sysctl(mib, 5, NULL, &size, NULL, 0) < 0) { |
583 |
< |
sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.0"); |
583 |
> |
sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.0"); |
584 |
|
return NULL; |
585 |
|
} |
586 |
|
|
593 |
|
} |
594 |
|
|
595 |
|
if(sysctl(mib, 5, kl_stats, &size, NULL, 0) < 0) { |
596 |
< |
sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.buffersize"); |
596 |
> |
sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.buffersize"); |
597 |
|
return NULL; |
598 |
|
} |
599 |
|
} |
635 |
|
proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN; |
636 |
|
break; |
637 |
|
} |
638 |
+ |
|
639 |
+ |
free(kl_stats); |
640 |
|
#else |
641 |
|
#ifdef FREEBSD5 |
642 |
|
switch (kp_stats[i].ki_stat) { |
679 |
|
free(kp_stats); |
680 |
|
#endif |
681 |
|
|
682 |
+ |
#ifdef HPUX |
683 |
+ |
if ((pagesize = sysconf(_SC_PAGESIZE)) == -1) { |
684 |
+ |
sg_set_error_with_errno(SG_ERROR_SYSCONF, "_SC_PAGESIZE"); |
685 |
+ |
return NULL; |
686 |
+ |
} |
687 |
+ |
|
688 |
+ |
while (1) { |
689 |
+ |
num = pstat_getproc(pstat_procinfo, sizeof pstat_procinfo[0], |
690 |
+ |
PROCESS_BATCH, procidx); |
691 |
+ |
if (num == -1) { |
692 |
+ |
sg_set_error_with_errno(SG_ERROR_PSTAT, |
693 |
+ |
"pstat_getproc"); |
694 |
+ |
return NULL; |
695 |
+ |
} else if (num == 0) { |
696 |
+ |
break; |
697 |
+ |
} |
698 |
+ |
|
699 |
+ |
for (i = 0; i < num; i++) { |
700 |
+ |
struct pst_status *pi = &pstat_procinfo[i]; |
701 |
+ |
|
702 |
+ |
if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) { |
703 |
+ |
return NULL; |
704 |
+ |
} |
705 |
+ |
proc_state_ptr = proc_state+proc_state_size; |
706 |
+ |
|
707 |
+ |
proc_state_ptr->pid = pi->pst_pid; |
708 |
+ |
proc_state_ptr->parent = pi->pst_ppid; |
709 |
+ |
proc_state_ptr->pgid = pi->pst_pgrp; |
710 |
+ |
proc_state_ptr->uid = pi->pst_uid; |
711 |
+ |
proc_state_ptr->euid = pi->pst_euid; |
712 |
+ |
proc_state_ptr->gid = pi->pst_gid; |
713 |
+ |
proc_state_ptr->egid = pi->pst_egid; |
714 |
+ |
proc_state_ptr->proc_size = (pi->pst_dsize + pi->pst_tsize + pi->pst_ssize) * pagesize; |
715 |
+ |
proc_state_ptr->proc_resident = pi->pst_rssize * pagesize; |
716 |
+ |
proc_state_ptr->time_spent = pi->pst_time; |
717 |
+ |
proc_state_ptr->cpu_percent = (pi->pst_pctcpu * 100.0) / 0x8000; |
718 |
+ |
proc_state_ptr->nice = pi->pst_nice; |
719 |
+ |
|
720 |
+ |
if (sg_update_string(&proc_state_ptr->process_name, |
721 |
+ |
pi->pst_ucomm) < 0) { |
722 |
+ |
return NULL; |
723 |
+ |
} |
724 |
+ |
if (sg_update_string(&proc_state_ptr->proctitle, |
725 |
+ |
pi->pst_cmd) < 0) { |
726 |
+ |
return NULL; |
727 |
+ |
} |
728 |
+ |
|
729 |
+ |
switch (pi->pst_stat) { |
730 |
+ |
case PS_SLEEP: |
731 |
+ |
proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING; |
732 |
+ |
break; |
733 |
+ |
case PS_RUN: |
734 |
+ |
proc_state_ptr->state = SG_PROCESS_STATE_RUNNING; |
735 |
+ |
break; |
736 |
+ |
case PS_STOP: |
737 |
+ |
proc_state_ptr->state = SG_PROCESS_STATE_STOPPED; |
738 |
+ |
break; |
739 |
+ |
case PS_ZOMBIE: |
740 |
+ |
proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE; |
741 |
+ |
break; |
742 |
+ |
case PS_IDLE: |
743 |
+ |
case PS_OTHER: |
744 |
+ |
proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN; |
745 |
+ |
break; |
746 |
+ |
} |
747 |
+ |
|
748 |
+ |
proc_state_size++; |
749 |
+ |
} |
750 |
+ |
procidx = pstat_procinfo[num - 1].pst_idx + 1; |
751 |
+ |
} |
752 |
+ |
#endif |
753 |
+ |
|
754 |
|
#ifdef CYGWIN |
755 |
|
sg_set_error(SG_ERROR_UNSUPPORTED, "Cygwin"); |
756 |
|
return NULL; |
803 |
|
return &process_stat; |
804 |
|
} |
805 |
|
|
806 |
+ |
int sg_process_compare_name(const void *va, const void *vb) { |
807 |
+ |
const sg_process_stats *a = (sg_process_stats *)va; |
808 |
+ |
const sg_process_stats *b = (sg_process_stats *)vb; |
809 |
+ |
|
810 |
+ |
return strcmp(a->process_name, b->process_name); |
811 |
+ |
} |
812 |
+ |
|
813 |
|
int sg_process_compare_pid(const void *va, const void *vb) { |
814 |
|
const sg_process_stats *a = (sg_process_stats *)va; |
815 |
|
const sg_process_stats *b = (sg_process_stats *)vb; |
901 |
|
} |
902 |
|
} |
903 |
|
|
770 |
– |
void sg_process_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_process_compare_pid; |
776 |
– |
break; |
777 |
– |
case SG_PS_UID: |
778 |
– |
sortby_ptr = sg_process_compare_uid; |
779 |
– |
break; |
780 |
– |
case SG_PS_GID: |
781 |
– |
sortby_ptr = sg_process_compare_gid; |
782 |
– |
break; |
783 |
– |
case SG_PS_SIZE: |
784 |
– |
sortby_ptr = sg_process_compare_size; |
785 |
– |
break; |
786 |
– |
case SG_PS_RES: |
787 |
– |
sortby_ptr = sg_process_compare_res; |
788 |
– |
break; |
789 |
– |
case SG_PS_CPU: |
790 |
– |
sortby_ptr = sg_process_compare_cpu; |
791 |
– |
break; |
792 |
– |
case SG_PS_TIME: |
793 |
– |
sortby_ptr = sg_process_compare_time; |
794 |
– |
break; |
795 |
– |
} |
796 |
– |
|
797 |
– |
qsort(ps, num_ps, sizeof(*ps), sortby_ptr); |
798 |
– |
|
799 |
– |
return; |
800 |
– |
} |