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.21 by ats, Sat Feb 14 16:43:55 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 <                free(kp_stats);
161 <                return NULL;
162 <        }
108 >                fclose(f);
109 >        }
110 >        closedir(proc_dir);
111 >        *ps = proc_state;
112  
113 <        for (i = 0; i < procs; i++) {
165 < #ifdef FREEBSD5
166 <                switch (kp_stats[i].ki_stat) {
167 < #else
168 <                switch (kp_stats[i].kp_proc.p_stat) {
169 < #endif
170 <                case SIDL:
171 <                case SRUN:
172 < #ifdef SONPROC
173 <                case SONPROC: /* NetBSD */
174 < #endif
175 <                        process_stat.running++;
176 <                        break;
177 <                case SSLEEP:
178 < #ifdef SWAIT
179 <                case SWAIT: /* FreeBSD 5 */
180 < #endif
181 < #ifdef SLOCK
182 <                case SLOCK: /* FreeBSD 5 */
183 < #endif
184 <                        process_stat.sleeping++;
185 <                        break;
186 <                case SSTOP:
187 <                        process_stat.stopped++;
188 <                        break;
189 <                case SZOMB:
190 < #ifdef SDEAD
191 <                case SDEAD: /* OpenBSD & NetBSD */
192 < #endif
193 <                        process_stat.zombie++;
194 <                        break;
195 <                }
196 <        }
197 <
198 <        free(kp_stats);
199 < #endif
200 <
201 < #ifdef CYGWIN
202 <        return NULL;
203 < #endif
204 <
205 <        process_stat.total=process_stat.sleeping+process_stat.running+process_stat.zombie+process_stat.stopped;
206 <
207 <        return &process_stat;
113 >        return proc_state_size;
114   }
115 +

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines