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.72 by ats, Thu Apr 8 23:30:29 2004 UTC vs.
Revision 1.81 by tdb, Sat Sep 24 13:29:22 2005 UTC

# Line 33 | Line 33
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>
# Line 72 | Line 72
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 + #ifdef WIN32
82 + #include <windows.h>
83 + #include <psapi.h>
84 + #endif
85  
86   static void proc_state_init(sg_process_stats *s) {
87          s->process_name = NULL;
# Line 88 | Line 98 | sg_process_stats *sg_get_process_stats(int *entries){
98                                proc_state_init, proc_state_destroy);
99          int proc_state_size = 0;
100          sg_process_stats *proc_state_ptr;
101 + #ifdef HPUX
102 +        struct pst_status pstat_procinfo[PROCESS_BATCH];
103 +        long procidx = 0;
104 +        long long pagesize;
105 +        int num, i;
106 + #endif
107   #ifdef ALLBSD
108          int mib[4];
109          size_t size;
# Line 131 | Line 147 | sg_process_stats *sg_get_process_stats(int *entries){
147  
148   #ifdef LINUX
149          if ((f=fopen("/proc/uptime", "r")) == NULL) {
150 <                sg_set_error(SG_ERROR_OPEN, "/proc/uptime");
150 >                sg_set_error_with_errno(SG_ERROR_OPEN, "/proc/uptime");
151                  return NULL;
152          }
153          if((fscanf(f,"%lu %*d",&uptime)) != 1){
# Line 142 | Line 158 | sg_process_stats *sg_get_process_stats(int *entries){
158   #endif
159  
160          if((proc_dir=opendir(PROC_LOCATION))==NULL){
161 <                sg_set_error(SG_ERROR_OPENDIR, PROC_LOCATION);
161 >                sg_set_error_with_errno(SG_ERROR_OPENDIR, PROC_LOCATION);
162                  return NULL;
163          }
164  
# Line 192 | Line 208 | sg_process_stats *sg_get_process_stats(int *entries){
208                          return NULL;
209                  }
210  
211 <                if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
212 <                if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
213 <                if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
214 <                if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
215 <                if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
211 >                switch (process_info.pr_lwp.pr_state) {
212 >                case 1:
213 >                        proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
214 >                        break;
215 >                case 2:
216 >                case 5:
217 >                        proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
218 >                        break;
219 >                case 3:
220 >                        proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
221 >                        break;
222 >                case 4:
223 >                        proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
224 >                        break;
225 >                }
226   #endif
227   #ifdef LINUX
228                  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));
229                  /* +3 becuase man page says "Resident  Set Size: number of pages the process has in real memory, minus 3 for administrative purposes." */
230                  proc_state_ptr->proc_resident = (proc_state_ptr->proc_resident + 3) * getpagesize();
231 <                if(s == 'S') proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
232 <                if(s == 'R') proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
233 <                if(s == 'Z') proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
234 <                if(s == 'T') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
235 <                if(s == 'D') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
231 >                switch (s) {
232 >                case 'S':
233 >                        proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
234 >                        break;
235 >                case 'R':
236 >                        proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
237 >                        break;
238 >                case 'Z':
239 >                        proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
240 >                        break;
241 >                case 'T':
242 >                case 'D':
243 >                        proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
244 >                        break;
245 >                }
246          
247                  /* pa_name[0] should = '(' */
248                  ptr = strchr(&ps_name[1], ')');
# Line 307 | Line 343 | sg_process_stats *sg_get_process_stats(int *entries){
343          mib[2] = KERN_PROC_ALL;
344  
345          if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
346 <                sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_PROC.KERN_PROC_ALL");
346 >                sg_set_error_with_errno(SG_ERROR_SYSCTL,
347 >                                        "CTL_KERN.KERN_PROC.KERN_PROC_ALL");
348                  return NULL;
349          }
350  
# Line 320 | Line 357 | sg_process_stats *sg_get_process_stats(int *entries){
357          memset(kp_stats, 0, size);
358  
359          if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
360 <                sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_PROC.KERN_PROC_ALL");
360 >                sg_set_error_with_errno(SG_ERROR_SYSCTL,
361 >                                        "CTL_KERN.KERN_PROC.KERN_PROC_ALL");
362                  free(kp_stats);
363                  return NULL;
364          }
# Line 382 | Line 420 | sg_process_stats *sg_get_process_stats(int *entries){
420                  proctitle = NULL;
421  
422                  do {
423 <                        if(size >= buflen) {
423 >                        if((long) size >= buflen) {
424                                  buflen *= 2;
425                                  size = buflen;
426                                  proctitletmp = sg_realloc(proctitle, buflen);
# Line 404 | Line 442 | sg_process_stats *sg_get_process_stats(int *entries){
442                                  size = 0;
443                                  break;
444                          }
445 <                } while(size >= buflen);
445 >                } while((long) size >= buflen);
446  
447                  if(size > 0) {
448                          proc_state_ptr->proctitle = sg_malloc(size+1);
# Line 546 | Line 584 | sg_process_stats *sg_get_process_stats(int *entries){
584                          mib[4] = 0;
585  
586                          if(sysctl(mib, 5, NULL, &size, NULL, 0) < 0) {
587 <                                sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.0");
587 >                                sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.0");
588                                  return NULL;
589                          }
590  
# Line 559 | Line 597 | sg_process_stats *sg_get_process_stats(int *entries){
597                          }
598  
599                          if(sysctl(mib, 5, kl_stats, &size, NULL, 0) < 0) {
600 <                                sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.buffersize");
600 >                                sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.buffersize");
601                                  return NULL;
602                          }
603                  }
# Line 601 | Line 639 | sg_process_stats *sg_get_process_stats(int *entries){
639                          proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
640                          break;
641                  }
642 +
643 +                free(kl_stats);
644   #else
645   #ifdef FREEBSD5
646                  switch (kp_stats[i].ki_stat) {
# Line 643 | Line 683 | sg_process_stats *sg_get_process_stats(int *entries){
683          free(kp_stats);
684   #endif
685  
686 + #ifdef HPUX
687 +        if ((pagesize = sysconf(_SC_PAGESIZE)) == -1) {
688 +                sg_set_error_with_errno(SG_ERROR_SYSCONF, "_SC_PAGESIZE");
689 +                return NULL;
690 +        }
691 +
692 +        while (1) {
693 +                num = pstat_getproc(pstat_procinfo, sizeof pstat_procinfo[0],
694 +                                    PROCESS_BATCH, procidx);
695 +                if (num == -1) {
696 +                        sg_set_error_with_errno(SG_ERROR_PSTAT,
697 +                                                "pstat_getproc");
698 +                        return NULL;
699 +                } else if (num == 0) {
700 +                        break;
701 +                }
702 +
703 +                for (i = 0; i < num; i++) {
704 +                        struct pst_status *pi = &pstat_procinfo[i];
705 +
706 +                        if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
707 +                                return NULL;
708 +                        }
709 +                        proc_state_ptr = proc_state+proc_state_size;
710 +        
711 +                        proc_state_ptr->pid = pi->pst_pid;
712 +                        proc_state_ptr->parent = pi->pst_ppid;
713 +                        proc_state_ptr->pgid = pi->pst_pgrp;
714 +                        proc_state_ptr->uid = pi->pst_uid;
715 +                        proc_state_ptr->euid = pi->pst_euid;
716 +                        proc_state_ptr->gid = pi->pst_gid;
717 +                        proc_state_ptr->egid = pi->pst_egid;
718 +                        proc_state_ptr->proc_size = (pi->pst_dsize + pi->pst_tsize + pi->pst_ssize) * pagesize;
719 +                        proc_state_ptr->proc_resident = pi->pst_rssize * pagesize;
720 +                        proc_state_ptr->time_spent = pi->pst_time;
721 +                        proc_state_ptr->cpu_percent = (pi->pst_pctcpu * 100.0) / 0x8000;
722 +                        proc_state_ptr->nice = pi->pst_nice;
723 +        
724 +                        if (sg_update_string(&proc_state_ptr->process_name,
725 +                                             pi->pst_ucomm) < 0) {
726 +                                return NULL;
727 +                        }
728 +                        if (sg_update_string(&proc_state_ptr->proctitle,
729 +                                             pi->pst_cmd) < 0) {
730 +                                return NULL;
731 +                        }
732 +        
733 +                        switch (pi->pst_stat) {
734 +                        case PS_SLEEP:
735 +                                proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
736 +                                break;
737 +                        case PS_RUN:
738 +                                proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
739 +                                break;
740 +                        case PS_STOP:
741 +                                proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
742 +                                break;
743 +                        case PS_ZOMBIE:
744 +                                proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
745 +                                break;
746 +                        case PS_IDLE:
747 +                        case PS_OTHER:
748 +                                proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
749 +                                break;
750 +                        }
751 +        
752 +                        proc_state_size++;
753 +                }
754 +                procidx = pstat_procinfo[num - 1].pst_idx + 1;
755 +        }
756 + #endif
757 +
758   #ifdef CYGWIN
759          sg_set_error(SG_ERROR_UNSUPPORTED, "Cygwin");
760          return NULL;
761   #endif
762 + #ifdef WIN32
763 +        /* FIXME The data needed for this is probably do able with the
764 +         * "performance registry". Although using this appears to be a black
765 +         * art and closely guarded secret.
766 +         * This is not directly used in ihost, so not considered a priority */
767 +        sg_set_error(SG_ERROR_UNSUPPORTED, "Win32");
768 +        return NULL;
769 + #endif
770  
771          *entries = proc_state_size;
772          return proc_state;
# Line 654 | Line 774 | sg_process_stats *sg_get_process_stats(int *entries){
774  
775   sg_process_count *sg_get_process_count() {
776          static sg_process_count process_stat;
777 + #ifndef WIN32
778          sg_process_stats *ps;
779          int ps_size, x;
780 + #else
781 +        DWORD aProcesses[1024];
782 +        DWORD cbNeeded;
783 + #endif
784  
785          process_stat.sleeping = 0;
786          process_stat.running = 0;
# Line 663 | Line 788 | sg_process_count *sg_get_process_count() {
788          process_stat.stopped = 0;
789          process_stat.total = 0;
790  
791 + #ifndef WIN32
792          ps = sg_get_process_stats(&ps_size);
793          if (ps == NULL) {
794                  return NULL;
# Line 691 | Line 817 | sg_process_count *sg_get_process_count() {
817          }
818  
819          process_stat.total = ps_size;
820 + #else
821 +        if (!EnumProcesses(aProcesses, sizeof(aProcesses), &cbNeeded))
822 +                return NULL;
823 +        process_stat.total = cbNeeded / sizeof(DWORD);
824 + #endif
825  
826          return &process_stat;
827   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines