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.23 by pajs, Sun Mar 28 18:04:14 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 <
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
72 < #ifdef SOLARIS
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)    
64 >        DIR *proc_dir;
65 >        struct dirent *dir_entry;
66 >        char filename[MAX_FILE_LENGTH];
67 >        FILE *f;
68          psinfo_t process_info;
74 #endif
75 #ifdef ALLBSD
76        int mib[3];
77        size_t size;
78        struct kinfo_proc *kp_stats;    
79        int procs, i;
80 #endif
69  
70 <        process_stat.sleeping=0;
71 <        process_stat.running=0;
72 <        process_stat.zombie=0;
85 <        process_stat.stopped=0;
86 <        process_stat.total=0;
70 >        if((proc_dir=opendir(PROC_LOCATION))==NULL){
71 >                return NULL;
72 >        }
73  
74 < #if defined(SOLARIS) || defined(LINUX)
75 <        if((proc_dir=opendir(PROC_LOCATION))==NULL){
90 <                return NULL;
91 <        }
74 >        while((dir_entry=readdir(proc_dir))!=NULL){
75 >                if(atoi(dir_entry->d_name) == 0) continue;
76  
77 <        while((dir_entry=readdir(proc_dir))!=NULL){
94 <                if(atoi(dir_entry->d_name) == 0) continue;
77 >                snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
78  
79 < #ifdef SOLARIS
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
79 >                if((f=fopen(filename, "r"))==NULL){
80 >                        /* Open failed.. Process since vanished, or the path was too long.
81 >                         * Ah well, move onwards to the next one */
82 >                        continue;
83 >                }
84 >                fread(&process_info, sizeof(psinfo_t), 1, f);
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 <                }
86 >                proc_state = realloc(proc_state, (1+proc_state_size)*sizeof(proc_state_t));
87  
88 < #ifdef SOLARIS
89 <                fread(&process_info, sizeof(psinfo_t), 1, f);
90 <                if(process_info.pr_lwp.pr_state==1) process_stat.sleeping++;
91 <                if(process_info.pr_lwp.pr_state==2) process_stat.running++;
92 <                if(process_info.pr_lwp.pr_state==3) process_stat.zombie++;
93 <                if(process_info.pr_lwp.pr_state==4) process_stat.stopped++;
94 <                if(process_info.pr_lwp.pr_state==6) process_stat.running++;
95 < #endif
96 < #ifdef LINUX
97 <                if((line_ptr=f_read_line(f, "State:"))==NULL){
98 <                        fclose(f);
99 <                        continue;
100 <                }
101 <                if((line_ptr=strchr(line_ptr, '\t'))==NULL){
102 <                        fclose(f);
124 <                        continue;
125 <                }
126 <                line_ptr++;
127 <                if(line_ptr=='\0'){
128 <                        fclose(f);
129 <                        continue;
130 <                }
88 >                proc_state_ptr = proc_state+proc_state_size;
89 >                
90 >                proc_state_ptr->pid = process_info.pr_pid;
91 >                proc_state_ptr->parent = process_info.pr_ppid;
92 >                proc_state_ptr->pgid = process_info.pr_pgid;
93 >                proc_state_ptr->uid = process_info.pr_uid;
94 >                proc_state_ptr->euid = process_info.pr_euid;
95 >                proc_state_ptr->gid = process_info.pr_gid;
96 >                proc_state_ptr->egid = process_info.pr_egid;
97 >                proc_state_ptr->proc_size = (process_info.pr_size) * 1024;
98 >                proc_state_ptr->proc_resident = (process_info.pr_rssize) * 1024;
99 >                proc_state_ptr->time_spent = process_info.pr_time.tv_sec;
100 >                proc_state_ptr->cpu_percent = (process_info.pr_pctcpu * 100.0) / 0x8000;
101 >                proc_state_ptr->process_name = strdup(process_info.pr_fname);
102 >                proc_state_ptr->proctitle = strdup(process_info.pr_psargs);
103  
104 <                if(*line_ptr=='S') process_stat.sleeping++;
133 <                if(*line_ptr=='R') process_stat.running++;
134 <                if(*line_ptr=='Z') process_stat.zombie++;
135 <                if(*line_ptr=='T') process_stat.stopped++;
136 <                if(*line_ptr=='D') process_stat.stopped++;
137 < #endif  
138 <
139 <                fclose(f);
140 <        }
141 <        closedir(proc_dir);
104 >                proc_state_size++;
105   #endif
143 #ifdef ALLBSD
144        mib[0] = CTL_KERN;
145        mib[1] = KERN_PROC;
146        mib[2] = KERN_PROC_ALL;
106  
148        if (sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
149                return NULL;
150        }
151        
152        procs = size / sizeof(struct kinfo_proc);
107  
108 <        kp_stats = malloc(size);
109 <        if (kp_stats == NULL) {
110 <                return NULL;
111 <        }
158 <        
159 <        if (sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
160 <                return NULL;
161 <        }
108 >                fclose(f);
109 >        }
110 >        closedir(proc_dir);
111 >        *ps = proc_state;
112  
113 <        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) {
168 < #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
186 <
187 < free(kp_stats);
188 <
189 < #ifdef CYGWIN
190 <        return NULL;
191 < #endif
192 <
193 <        process_stat.total=process_stat.sleeping+process_stat.running+process_stat.zombie+process_stat.stopped;
194 <
195 <        return &process_stat;
113 >        return proc_state_size;
114   }
115 +

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines