ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/src/libstatgrab/process_stats.c
(Generate patch)

Comparing projects/libstatgrab/src/libstatgrab/process_stats.c (file contents):
Revision 1.65 by tdb, Wed Apr 7 21:08:40 2004 UTC vs.
Revision 1.74 by ats, Sun Jul 18 21:30:11 2004 UTC

# Line 51 | Line 51
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>
# Line 99 | Line 100 | sg_process_stats *sg_get_process_stats(int *entries){
100          int argslen = 0;
101   #else
102          long buflen;
103 <        char *p;
103 >        char *p, *proctitletmp;
104   #endif
105   #ifdef NETBSD2
106          int lwps;
# Line 130 | Line 131 | sg_process_stats *sg_get_process_stats(int *entries){
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){
# Line 141 | Line 142 | sg_process_stats *sg_get_process_stats(int *entries){
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  
# Line 306 | Line 307 | sg_process_stats *sg_get_process_stats(int *entries){
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  
# Line 319 | Line 321 | sg_process_stats *sg_get_process_stats(int *entries){
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          }
# Line 361 | Line 364 | sg_process_stats *sg_get_process_stats(int *entries){
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
# Line 397 | Line 375 | sg_process_stats *sg_get_process_stats(int *entries){
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);
# Line 414 | Line 430 | sg_process_stats *sg_get_process_stats(int *entries){
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) {
# Line 527 | Line 548 | sg_process_stats *sg_get_process_stats(int *entries){
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  
# Line 540 | Line 561 | sg_process_stats *sg_get_process_stats(int *entries){
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                  }
# Line 582 | Line 603 | sg_process_stats *sg_get_process_stats(int *entries){
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) {
# Line 675 | Line 698 | sg_process_count *sg_get_process_count() {
698  
699          return &process_stat;
700   }
701 +
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) {
716 +                return 0;
717 +        } else {
718 +                return 1;
719 +        }
720 + }
721 +
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 +
726 +        if (a->uid < b->uid) {
727 +                return -1;
728 +        } else if (a->uid == b->uid) {
729 +                return 0;
730 +        } else {
731 +                return 1;
732 +        }
733 + }
734 +
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 +
739 +        if (a->gid < b->gid) {
740 +                return -1;
741 +        } else if (a->gid == b->gid) {
742 +                return 0;
743 +        } else {
744 +                return 1;
745 +        }
746 + }
747 +
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 +
752 +        if (a->proc_size < b->proc_size) {
753 +                return -1;
754 +        } else if (a->proc_size == b->proc_size) {
755 +                return 0;
756 +        } else {
757 +                return 1;
758 +        }
759 + }
760 +
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 +
765 +        if (a->proc_resident < b->proc_resident) {
766 +                return -1;
767 +        } else if (a->proc_resident == b->proc_resident) {
768 +                return 0;
769 +        } else {
770 +                return 1;
771 +        }
772 + }
773 +
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 +
778 +        if (a->cpu_percent < b->cpu_percent) {
779 +                return -1;
780 +        } else if (a->cpu_percent == b->cpu_percent) {
781 +                return 0;
782 +        } else {
783 +                return 1;
784 +        }
785 + }
786 +
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 +
791 +        if (a->time_spent < b->time_spent) {
792 +                return -1;
793 +        } else if (a->time_spent == b->time_spent) {
794 +                return 0;
795 +        } else {
796 +                return 1;
797 +        }
798 + }
799 +

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines