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.64 by pajs, Wed Apr 7 15:46:34 2004 UTC vs.
Revision 1.75 by ats, Fri Sep 10 15:15:00 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_with_errno(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_with_errno(SG_ERROR_OPENDIR, PROC_LOCATION);
146                  return NULL;
147          }
148  
# Line 188 | Line 192 | sg_process_stats *sg_get_process_stats(int *entries){
192                          return NULL;
193                  }
194  
195 <                if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
196 <                if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
197 <                if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
198 <                if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
199 <                if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
195 >                switch (process_info.pr_lwp.pr_state) {
196 >                case 1:
197 >                        proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
198 >                        break;
199 >                case 2:
200 >                case 5:
201 >                        proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
202 >                        break;
203 >                case 3:
204 >                        proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
205 >                        break;
206 >                case 4:
207 >                        proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
208 >                        break;
209 >                }
210   #endif
211   #ifdef LINUX
212                  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));
213                  /* +3 becuase man page says "Resident  Set Size: number of pages the process has in real memory, minus 3 for administrative purposes." */
214                  proc_state_ptr->proc_resident = (proc_state_ptr->proc_resident + 3) * getpagesize();
215 <                if(s == 'S') proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
216 <                if(s == 'R') proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
217 <                if(s == 'Z') proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
218 <                if(s == 'T') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
219 <                if(s == 'D') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
215 >                switch (s) {
216 >                case 'S':
217 >                        proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
218 >                        break;
219 >                case 'R':
220 >                        proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
221 >                        break;
222 >                case 'Z':
223 >                        proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
224 >                        break;
225 >                case 'T':
226 >                case 'D':
227 >                        proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
228 >                        break;
229 >                }
230          
231                  /* pa_name[0] should = '(' */
232                  ptr = strchr(&ps_name[1], ')');
# Line 303 | Line 327 | sg_process_stats *sg_get_process_stats(int *entries){
327          mib[2] = KERN_PROC_ALL;
328  
329          if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
330 +                sg_set_error_with_errno(SG_ERROR_SYSCTL,
331 +                                        "CTL_KERN.KERN_PROC.KERN_PROC_ALL");
332                  return NULL;
333          }
334  
# Line 315 | Line 341 | sg_process_stats *sg_get_process_stats(int *entries){
341          memset(kp_stats, 0, size);
342  
343          if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
344 +                sg_set_error_with_errno(SG_ERROR_SYSCTL,
345 +                                        "CTL_KERN.KERN_PROC.KERN_PROC_ALL");
346                  free(kp_stats);
347                  return NULL;
348          }
# Line 356 | Line 384 | sg_process_stats *sg_get_process_stats(int *entries){
384                  }
385  
386   #if defined(FREEBSD5) || defined(NETBSD) || defined(OPENBSD)
359                size = sizeof(buflen);
387  
388   #ifdef FREEBSD5
362                if(sysctlbyname("kern.ps_arg_cache_limit", &buflen, &size, NULL, 0) < 0) {
363                        return NULL;
364                }
365 #else
366                /* FIXME - this value can be too large on some of
367                   the BSD's, which causes sysctl not to return
368                   anything. Maybe we need something smaller? */
369                mib[1] = KERN_ARGMAX;
370
371                if(sysctl(mib, 2, &buflen, &size, NULL, 0) < 0) {
372                        return NULL;
373                }
374 #endif
375
376                proctitle = sg_malloc(buflen);
377                if(proctitle == NULL) {
378                        return NULL;
379                }
380
381                size = buflen;
382
383 #ifdef FREEBSD5
389                  mib[2] = KERN_PROC_ARGS;
390                  mib[3] = kp_stats[i].ki_pid;
391   #else
# Line 390 | Line 395 | sg_process_stats *sg_get_process_stats(int *entries){
395   #endif
396  
397                  free(proc_state_ptr->proctitle);
398 <                if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) {
399 <                        free(proctitle);
400 <                        proc_state_ptr->proctitle = NULL;
401 <                }
402 <                else if(size > 0) {
398 >                proc_state_ptr->proctitle = NULL;
399 >
400 > /* Starting size - we'll double this straight away */
401 > #define PROCTITLE_START_SIZE 64
402 >                buflen = PROCTITLE_START_SIZE;
403 >                size = buflen;
404 >                proctitle = NULL;
405 >
406 >                do {
407 >                        if(size >= buflen) {
408 >                                buflen *= 2;
409 >                                size = buflen;
410 >                                proctitletmp = sg_realloc(proctitle, buflen);
411 >                                if(proctitletmp == NULL) {
412 >                                        free(proctitle);
413 >                                        proctitle = NULL;
414 >                                        proc_state_ptr->proctitle = NULL;
415 >                                        size = 0;
416 >                                        break;
417 >                                }
418 >                                proctitle = proctitletmp;
419 >                                bzero(proctitle, buflen);
420 >                        }
421 >
422 >                        if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) {
423 >                                free(proctitle);
424 >                                proctitle = NULL;
425 >                                proc_state_ptr->proctitle = NULL;
426 >                                size = 0;
427 >                                break;
428 >                        }
429 >                } while(size >= buflen);
430 >
431 >                if(size > 0) {
432                          proc_state_ptr->proctitle = sg_malloc(size+1);
433                          if(proc_state_ptr->proctitle == NULL) {
434                                  return NULL;
435                          }
436                          p = proctitle;
437 + #ifdef OPENBSD
438 +                        /* On OpenBSD, this value has the argv pointers (which
439 +                         * are terminated by a NULL) at the front, so we have
440 +                         * to skip over them to get to the strings. */
441 +                        while (*(char ***)p != NULL) {
442 +                                p += sizeof(char **);
443 +                        }
444 +                        p += sizeof(char **);
445 + #endif
446                          proc_state_ptr->proctitle[0] = '\0';
447                          do {
448                                  sg_strlcat(proc_state_ptr->proctitle, p, size+1);
# Line 407 | Line 450 | sg_process_stats *sg_get_process_stats(int *entries){
450                                  p += strlen(p) + 1;
451                          } while (p < proctitle + size);
452                          free(proctitle);
453 +                        proctitle = NULL;
454                          /* remove trailing space */
455                          proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = '\0';
456                  }
457                  else {
458 <                        free(proctitle);
458 >                        if(proctitle != NULL) {
459 >                                free(proctitle);
460 >                                proctitle = NULL;
461 >                        }
462                          proc_state_ptr->proctitle = NULL;
463                  }
464   #else
465                  free(proc_state_ptr->proctitle);
466 +                proc_state_ptr->proctitle = NULL;
467                  if(kvmd != NULL) {
468                          args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
469                          if(args != NULL) {
# Line 520 | Line 568 | sg_process_stats *sg_get_process_stats(int *entries){
568                          mib[4] = 0;
569  
570                          if(sysctl(mib, 5, NULL, &size, NULL, 0) < 0) {
571 +                                sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.0");
572                                  return NULL;
573                          }
574  
# Line 532 | Line 581 | sg_process_stats *sg_get_process_stats(int *entries){
581                          }
582  
583                          if(sysctl(mib, 5, kl_stats, &size, NULL, 0) < 0) {
584 +                                sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.buffersize");
585                                  return NULL;
586                          }
587                  }
# Line 573 | Line 623 | sg_process_stats *sg_get_process_stats(int *entries){
623                          proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
624                          break;
625                  }
626 +
627 +                free(kl_stats);
628   #else
629   #ifdef FREEBSD5
630                  switch (kp_stats[i].ki_stat) {
# Line 616 | Line 668 | sg_process_stats *sg_get_process_stats(int *entries){
668   #endif
669  
670   #ifdef CYGWIN
671 +        sg_set_error(SG_ERROR_UNSUPPORTED, "Cygwin");
672          return NULL;
673   #endif
674  
# Line 665 | Line 718 | sg_process_count *sg_get_process_count() {
718  
719          return &process_stat;
720   }
721 +
722 + int sg_process_compare_name(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 +        return strcmp(a->process_name, b->process_name);
727 + }
728 +
729 + int sg_process_compare_pid(const void *va, const void *vb) {
730 +        const sg_process_stats *a = (sg_process_stats *)va;
731 +        const sg_process_stats *b = (sg_process_stats *)vb;
732 +
733 +        if (a->pid < b->pid) {
734 +                return -1;
735 +        } else if (a->pid == b->pid) {
736 +                return 0;
737 +        } else {
738 +                return 1;
739 +        }
740 + }
741 +
742 + int sg_process_compare_uid(const void *va, const void *vb) {
743 +        const sg_process_stats *a = (sg_process_stats *)va;
744 +        const sg_process_stats *b = (sg_process_stats *)vb;
745 +
746 +        if (a->uid < b->uid) {
747 +                return -1;
748 +        } else if (a->uid == b->uid) {
749 +                return 0;
750 +        } else {
751 +                return 1;
752 +        }
753 + }
754 +
755 + int sg_process_compare_gid(const void *va, const void *vb) {
756 +        const sg_process_stats *a = (sg_process_stats *)va;
757 +        const sg_process_stats *b = (sg_process_stats *)vb;
758 +
759 +        if (a->gid < b->gid) {
760 +                return -1;
761 +        } else if (a->gid == b->gid) {
762 +                return 0;
763 +        } else {
764 +                return 1;
765 +        }
766 + }
767 +
768 + int sg_process_compare_size(const void *va, const void *vb) {
769 +        const sg_process_stats *a = (sg_process_stats *)va;
770 +        const sg_process_stats *b = (sg_process_stats *)vb;
771 +
772 +        if (a->proc_size < b->proc_size) {
773 +                return -1;
774 +        } else if (a->proc_size == b->proc_size) {
775 +                return 0;
776 +        } else {
777 +                return 1;
778 +        }
779 + }
780 +
781 + int sg_process_compare_res(const void *va, const void *vb) {
782 +        const sg_process_stats *a = (sg_process_stats *)va;
783 +        const sg_process_stats *b = (sg_process_stats *)vb;
784 +
785 +        if (a->proc_resident < b->proc_resident) {
786 +                return -1;
787 +        } else if (a->proc_resident == b->proc_resident) {
788 +                return 0;
789 +        } else {
790 +                return 1;
791 +        }
792 + }
793 +
794 + int sg_process_compare_cpu(const void *va, const void *vb) {
795 +        const sg_process_stats *a = (sg_process_stats *)va;
796 +        const sg_process_stats *b = (sg_process_stats *)vb;
797 +
798 +        if (a->cpu_percent < b->cpu_percent) {
799 +                return -1;
800 +        } else if (a->cpu_percent == b->cpu_percent) {
801 +                return 0;
802 +        } else {
803 +                return 1;
804 +        }
805 + }
806 +
807 + int sg_process_compare_time(const void *va, const void *vb) {
808 +        const sg_process_stats *a = (sg_process_stats *)va;
809 +        const sg_process_stats *b = (sg_process_stats *)vb;
810 +
811 +        if (a->time_spent < b->time_spent) {
812 +                return -1;
813 +        } else if (a->time_spent == b->time_spent) {
814 +                return 0;
815 +        } else {
816 +                return 1;
817 +        }
818 + }
819 +

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines