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.19 by tdb, Sat Feb 14 02:56:11 2004 UTC vs.
Revision 1.24 by pajs, Sun Mar 28 18:22:58 2004 UTC

# Line 34 | Line 34
34   #include <string.h>
35   #endif
36  
37 #include "tools.h"
37   #ifdef SOLARIS
38   #include <procfs.h>
39   #include <limits.h>
# Line 50 | Line 49
49   #include <stdlib.h>
50   #include <sys/param.h>
51   #include <sys/sysctl.h>
52 + #if defined(FREEBSD) || defined(DFBSD)
53 + #include <sys/user.h>
54 + #else
55   #include <sys/proc.h>
56   #endif
55 #ifdef FREEBSD
56 #include <sys/user.h>
57   #endif
58  
59 < process_stat_t *get_process_stats(){
60 <
61 <        static process_stat_t process_stat;
62 <
59 > int get_proc_snapshot(proc_state_t **ps){
60 >        proc_state_t *proc_state = NULL;
61 >        proc_state_t *proc_state_ptr;
62 >        int proc_state_size = 0;
63   #if defined(SOLARIS) || defined(LINUX)
64 <        DIR *proc_dir;
65 <        struct dirent *dir_entry;
66 <        char filename[MAX_FILE_LENGTH];
67 <        FILE *f;
68 < #endif
69 < #ifdef LINUX
70 <        char *line_ptr;
71 < #endif
64 >        DIR *proc_dir;
65 >        struct dirent *dir_entry;
66 >        char filename[MAX_FILE_LENGTH];
67 >        FILE *f;
68   #ifdef SOLARIS
69          psinfo_t process_info;
70   #endif
75 #ifdef ALLBSD
76        int mib[3];
77        size_t size;
78        struct kinfo_proc *kp_stats;    
79        int procs, i;
80 #endif
71  
72 <        process_stat.sleeping=0;
73 <        process_stat.running=0;
74 <        process_stat.zombie=0;
85 <        process_stat.stopped=0;
86 <        process_stat.total=0;
72 >        if((proc_dir=opendir(PROC_LOCATION))==NULL){
73 >                return NULL;
74 >        }
75  
76 < #if defined(SOLARIS) || defined(LINUX)
77 <        if((proc_dir=opendir(PROC_LOCATION))==NULL){
90 <                return NULL;
91 <        }
76 >        while((dir_entry=readdir(proc_dir))!=NULL){
77 >                if(atoi(dir_entry->d_name) == 0) continue;
78  
93        while((dir_entry=readdir(proc_dir))!=NULL){
94                if(atoi(dir_entry->d_name) == 0) continue;
95
79   #ifdef SOLARIS
80 <                snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
80 >                snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
81   #endif
82 < #ifdef LINUX
83 <                snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/status", dir_entry->d_name);
84 < #endif
85 <
86 <                if((f=fopen(filename, "r"))==NULL){
104 <                        /* Open failed.. Process since vanished, or the path was too long.
105 <                         * Ah well, move onwards to the next one */
106 <                        continue;
107 <                }
108 <
82 >                if((f=fopen(filename, "r"))==NULL){
83 >                        /* Open failed.. Process since vanished, or the path was too long.
84 >                         * Ah well, move onwards to the next one */
85 >                        continue;
86 >                }
87   #ifdef SOLARIS
88 <                fread(&process_info, sizeof(psinfo_t), 1, f);
111 <                if(process_info.pr_lwp.pr_state==1) process_stat.sleeping++;
112 <                if(process_info.pr_lwp.pr_state==2) process_stat.running++;
113 <                if(process_info.pr_lwp.pr_state==3) process_stat.zombie++;
114 <                if(process_info.pr_lwp.pr_state==4) process_stat.stopped++;
115 <                if(process_info.pr_lwp.pr_state==6) process_stat.running++;
116 < #endif
117 < #ifdef LINUX
118 <                if((line_ptr=f_read_line(f, "State:"))==NULL){
119 <                        fclose(f);
120 <                        continue;
121 <                }
122 <                if((line_ptr=strchr(line_ptr, '\t'))==NULL){
123 <                        fclose(f);
124 <                        continue;
125 <                }
126 <                line_ptr++;
127 <                if(line_ptr=='\0'){
128 <                        fclose(f);
129 <                        continue;
130 <                }
88 >                fread(&process_info, sizeof(psinfo_t), 1, f);
89  
90 <                if(*line_ptr=='S') process_stat.sleeping++;
91 <                if(*line_ptr=='R') process_stat.running++;
92 <                if(*line_ptr=='Z') process_stat.zombie++;
93 <                if(*line_ptr=='T') process_stat.stopped++;
94 <                if(*line_ptr=='D') process_stat.stopped++;
95 < #endif  
90 >                proc_state = realloc(proc_state, (1+proc_state_size)*sizeof(proc_state_t));
91 >                proc_state_ptr = proc_state+proc_state_size;
92 >                
93 >                proc_state_ptr->pid = process_info.pr_pid;
94 >                proc_state_ptr->parent = process_info.pr_ppid;
95 >                proc_state_ptr->pgid = process_info.pr_pgid;
96 >                proc_state_ptr->uid = process_info.pr_uid;
97 >                proc_state_ptr->euid = process_info.pr_euid;
98 >                proc_state_ptr->gid = process_info.pr_gid;
99 >                proc_state_ptr->egid = process_info.pr_egid;
100 >                proc_state_ptr->proc_size = (process_info.pr_size) * 1024;
101 >                proc_state_ptr->proc_resident = (process_info.pr_rssize) * 1024;
102 >                proc_state_ptr->time_spent = process_info.pr_time.tv_sec;
103 >                proc_state_ptr->cpu_percent = (process_info.pr_pctcpu * 100.0) / 0x8000;
104 >                proc_state_ptr->process_name = strdup(process_info.pr_fname);
105 >                proc_state_ptr->proctitle = strdup(process_info.pr_psargs);
106  
107 <                fclose(f);
108 <        }
109 <        closedir(proc_dir);
107 >                if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SLEEPING;
108 >                if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = RUNNING;
109 >                if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = ZOMBIE;
110 >                if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = STOPPED;
111 >                if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = RUNNING;
112   #endif
143 #ifdef ALLBSD
144        mib[0] = CTL_KERN;
145        mib[1] = KERN_PROC;
146        mib[2] = KERN_PROC_ALL;
113  
114 <        if (sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
149 <                return NULL;
150 <        }
151 <        
152 <        procs = size / sizeof(struct kinfo_proc);
153 <
154 <        kp_stats = malloc(size);
155 <        if (kp_stats == NULL) {
156 <                return NULL;
157 <        }
158 <        
159 <        if (sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
160 <                return NULL;
161 <        }
162 <
163 <        for (i = 0; i < procs; i++) {
164 < #ifdef FREEBSD5
165 <                switch (kp_stats[i].ki_stat) {
166 < #else
167 <                switch (kp_stats[i].kp_proc.p_stat) {
114 >                proc_state_size++;
115   #endif
169                case SSLEEP:
170                        process_stat.sleeping++;
171                        break;
172                case SRUN:
173                case SIDL:
174                        process_stat.running++;
175                        break;
176                case SZOMB:
177                /*case SDEAD:*/
178                        process_stat.zombie++;
179                        break;
180                case SSTOP:
181                        process_stat.stopped++;
182                        break;
183                }
184        }
185 #endif
116  
187 free(kp_stats);
117  
118 < #ifdef CYGWIN
119 <        return NULL;
120 < #endif
118 >                fclose(f);
119 >        }
120 >        closedir(proc_dir);
121 >        *ps = proc_state;
122  
123 <        process_stat.total=process_stat.sleeping+process_stat.running+process_stat.zombie+process_stat.stopped;
194 <
195 <        return &process_stat;
123 >        return proc_state_size;
124   }
125 +

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines