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.63 by tdb, Wed Apr 7 14:53:40 2004 UTC vs.
Revision 1.73 by tdb, Mon Jun 21 14:48:48 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");
135                  return NULL;
136          }
137          if((fscanf(f,"%lu %*d",&uptime)) != 1){
138 +                sg_set_error(SG_ERROR_PARSE, NULL);
139                  return NULL;
140          }
141          fclose(f);
142   #endif
143  
144          if((proc_dir=opendir(PROC_LOCATION))==NULL){
145 +                sg_set_error(SG_ERROR_OPENDIR, PROC_LOCATION);
146                  return NULL;
147          }
148  
# Line 158 | Line 162 | sg_process_stats *sg_get_process_stats(int *entries){
162                  }
163   #ifdef SOLARIS
164                  fread(&process_info, sizeof(psinfo_t), 1, f);
165 +                fclose(f);
166   #endif
167  
168                  if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
# Line 177 | Line 182 | sg_process_stats *sg_get_process_stats(int *entries){
182                  proc_state_ptr->proc_resident = (process_info.pr_rssize) * 1024;
183                  proc_state_ptr->time_spent = process_info.pr_time.tv_sec;
184                  proc_state_ptr->cpu_percent = (process_info.pr_pctcpu * 100.0) / 0x8000;
185 +                proc_state_ptr->nice = (int)process_info.pr_lwp.pr_nice - 20;
186                  if (sg_update_string(&proc_state_ptr->process_name,
187                                       process_info.pr_fname) < 0) {
188                          return NULL;
# Line 301 | 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");
311                  return NULL;
312          }
313  
# Line 313 | Line 320 | sg_process_stats *sg_get_process_stats(int *entries){
320          memset(kp_stats, 0, size);
321  
322          if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
323 +                sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_PROC.KERN_PROC_ALL");
324                  free(kp_stats);
325                  return NULL;
326          }
# Line 354 | Line 362 | sg_process_stats *sg_get_process_stats(int *entries){
362                  }
363  
364   #if defined(FREEBSD5) || defined(NETBSD) || defined(OPENBSD)
357                size = sizeof(buflen);
365  
366   #ifdef FREEBSD5
360                if(sysctlbyname("kern.ps_arg_cache_limit", &buflen, &size, NULL, 0) < 0) {
361                        return NULL;
362                }
363 #else
364                /* FIXME - this value can be too large on some of
365                   the BSD's, which causes sysctl not to return
366                   anything. Maybe we need something smaller? */
367                mib[1] = KERN_ARGMAX;
368
369                if(sysctl(mib, 2, &buflen, &size, NULL, 0) < 0) {
370                        return NULL;
371                }
372 #endif
373
374                proctitle = sg_malloc(buflen);
375                if(proctitle == NULL) {
376                        return NULL;
377                }
378
379                size = buflen;
380
381 #ifdef FREEBSD5
367                  mib[2] = KERN_PROC_ARGS;
368                  mib[3] = kp_stats[i].ki_pid;
369   #else
# Line 388 | Line 373 | sg_process_stats *sg_get_process_stats(int *entries){
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;
413                          }
414                          p = proctitle;
415 + #ifdef OPENBSD
416 +                        /* On OpenBSD, this value has the argv pointers (which
417 +                         * are terminated by a NULL) at the front, so we have
418 +                         * to skip over them to get to the strings. */
419 +                        while (*(char ***)p != NULL) {
420 +                                p += sizeof(char **);
421 +                        }
422 +                        p += sizeof(char **);
423 + #endif
424                          proc_state_ptr->proctitle[0] = '\0';
425                          do {
426                                  sg_strlcat(proc_state_ptr->proctitle, p, size+1);
# Line 405 | Line 428 | sg_process_stats *sg_get_process_stats(int *entries){
428                                  p += strlen(p) + 1;
429                          } while (p < proctitle + size);
430                          free(proctitle);
431 +                        proctitle = NULL;
432                          /* remove trailing space */
433                          proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = '\0';
434                  }
435                  else {
436 <                        free(proctitle);
436 >                        if(proctitle != NULL) {
437 >                                free(proctitle);
438 >                                proctitle = NULL;
439 >                        }
440                          proc_state_ptr->proctitle = NULL;
441                  }
442   #else
443                  free(proc_state_ptr->proctitle);
444 +                proc_state_ptr->proctitle = NULL;
445                  if(kvmd != NULL) {
446                          args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
447                          if(args != NULL) {
# Line 518 | Line 546 | sg_process_stats *sg_get_process_stats(int *entries){
546                          mib[4] = 0;
547  
548                          if(sysctl(mib, 5, NULL, &size, NULL, 0) < 0) {
549 +                                sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.0");
550                                  return NULL;
551                          }
552  
# Line 530 | Line 559 | sg_process_stats *sg_get_process_stats(int *entries){
559                          }
560  
561                          if(sysctl(mib, 5, kl_stats, &size, NULL, 0) < 0) {
562 +                                sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.buffersize");
563                                  return NULL;
564                          }
565                  }
# Line 571 | Line 601 | sg_process_stats *sg_get_process_stats(int *entries){
601                          proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
602                          break;
603                  }
604 +
605 +                free(kl_stats);
606   #else
607   #ifdef FREEBSD5
608                  switch (kp_stats[i].ki_stat) {
# Line 614 | Line 646 | sg_process_stats *sg_get_process_stats(int *entries){
646   #endif
647  
648   #ifdef CYGWIN
649 +        sg_set_error(SG_ERROR_UNSUPPORTED, "Cygwin");
650          return NULL;
651   #endif
652  
# Line 663 | Line 696 | sg_process_count *sg_get_process_count() {
696  
697          return &process_stat;
698   }
699 +
700 + int sg_process_compare_name(const void *va, const void *vb) {
701 +        const sg_process_stats *a = (sg_process_stats *)va;
702 +        const sg_process_stats *b = (sg_process_stats *)vb;
703 +
704 +        return strcmp(a->process_name, b->process_name);
705 + }
706 +
707 + int sg_process_compare_pid(const void *va, const void *vb) {
708 +        const sg_process_stats *a = (sg_process_stats *)va;
709 +        const sg_process_stats *b = (sg_process_stats *)vb;
710 +
711 +        if (a->pid < b->pid) {
712 +                return -1;
713 +        } else if (a->pid == b->pid) {
714 +                return 0;
715 +        } else {
716 +                return 1;
717 +        }
718 + }
719 +
720 + int sg_process_compare_uid(const void *va, const void *vb) {
721 +        const sg_process_stats *a = (sg_process_stats *)va;
722 +        const sg_process_stats *b = (sg_process_stats *)vb;
723 +
724 +        if (a->uid < b->uid) {
725 +                return -1;
726 +        } else if (a->uid == b->uid) {
727 +                return 0;
728 +        } else {
729 +                return 1;
730 +        }
731 + }
732 +
733 + int sg_process_compare_gid(const void *va, const void *vb) {
734 +        const sg_process_stats *a = (sg_process_stats *)va;
735 +        const sg_process_stats *b = (sg_process_stats *)vb;
736 +
737 +        if (a->gid < b->gid) {
738 +                return -1;
739 +        } else if (a->gid == b->gid) {
740 +                return 0;
741 +        } else {
742 +                return 1;
743 +        }
744 + }
745 +
746 + int sg_process_compare_size(const void *va, const void *vb) {
747 +        const sg_process_stats *a = (sg_process_stats *)va;
748 +        const sg_process_stats *b = (sg_process_stats *)vb;
749 +
750 +        if (a->proc_size < b->proc_size) {
751 +                return -1;
752 +        } else if (a->proc_size == b->proc_size) {
753 +                return 0;
754 +        } else {
755 +                return 1;
756 +        }
757 + }
758 +
759 + int sg_process_compare_res(const void *va, const void *vb) {
760 +        const sg_process_stats *a = (sg_process_stats *)va;
761 +        const sg_process_stats *b = (sg_process_stats *)vb;
762 +
763 +        if (a->proc_resident < b->proc_resident) {
764 +                return -1;
765 +        } else if (a->proc_resident == b->proc_resident) {
766 +                return 0;
767 +        } else {
768 +                return 1;
769 +        }
770 + }
771 +
772 + int sg_process_compare_cpu(const void *va, const void *vb) {
773 +        const sg_process_stats *a = (sg_process_stats *)va;
774 +        const sg_process_stats *b = (sg_process_stats *)vb;
775 +
776 +        if (a->cpu_percent < b->cpu_percent) {
777 +                return -1;
778 +        } else if (a->cpu_percent == b->cpu_percent) {
779 +                return 0;
780 +        } else {
781 +                return 1;
782 +        }
783 + }
784 +
785 + int sg_process_compare_time(const void *va, const void *vb) {
786 +        const sg_process_stats *a = (sg_process_stats *)va;
787 +        const sg_process_stats *b = (sg_process_stats *)vb;
788 +
789 +        if (a->time_spent < b->time_spent) {
790 +                return -1;
791 +        } else if (a->time_spent == b->time_spent) {
792 +                return 0;
793 +        } else {
794 +                return 1;
795 +        }
796 + }
797 +

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines