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.28 by tdb, Sat Apr 3 16:57:32 2004 UTC vs.
Revision 1.35 by tdb, Sun Apr 4 15:56:53 2004 UTC

# Line 57 | Line 57
57   #else
58   #include <sys/proc.h>
59   #endif
60 + #include <string.h>
61 + #include <paths.h>
62 + #include <fcntl.h>
63 + #include <limits.h>
64 + #include <kvm.h>
65   #endif
66  
67   int get_proc_snapshot(proc_state_t **ps){
# Line 67 | Line 72 | int get_proc_snapshot(proc_state_t **ps){
72          int mib[3];
73          size_t size;
74          struct kinfo_proc *kp_stats;
75 <        int procs, i;
75 >        int procs, i, alloc;
76 >        static kvm_t *kvmd;
77 >        char **args;
78 >        char *proctitle;
79   #endif
80   #if defined(SOLARIS) || defined(LINUX)
81          DIR *proc_dir;
# Line 199 | Line 207 | int get_proc_snapshot(proc_state_t **ps){
207   #endif
208  
209   #ifdef ALLBSD
202 –        mib[0] = CTL_KERN;
203 –        mib[1] = KERN_PROC;
204 –        mib[2] = KERN_PROC_ALL;
210  
211 <        if (sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
212 <                return NULL;
213 <        }
211 > #if defined(NETBSD) || defined(OPENBSD)
212 >        kvmd = kvm_openfiles(NULL, NULL, NULL, O_RDONLY, NULL);
213 > #else
214 >        kvmd = kvm_openfiles(_PATH_DEVNULL, _PATH_DEVNULL, NULL, O_RDONLY, NULL);
215 > #endif
216  
217 <        procs = size / sizeof(struct kinfo_proc);
217 >        if(kvmd == NULL) return NULL;
218  
219 <        kp_stats = malloc(size);
213 <        if (kp_stats == NULL) {
214 <                return NULL;
215 <        }
219 >        kp_stats = kvm_getprocs(kvmd, KERN_PROC_ALL, 0, &procs);
220  
221 <        if (sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
218 <                free(kp_stats);
221 >        if (kp_stats == NULL || procs < 0) {
222                  return NULL;
223          }
224  
225          for (i = 0; i < procs; i++) {
226 <                proc_state = realloc(proc_state, (1+proc_state_size)*sizeof(proc_state_t));
226 >                /* replace with something more sensible */
227 >                proc_state = realloc(proc_state,
228 >                                (1+proc_state_size)*sizeof(proc_state_t));
229 >                if(proc_state == NULL ) {
230 >                        return NULL;
231 >                }
232                  proc_state_ptr = proc_state+proc_state_size;
225 –                
226 –                proc_state_ptr->process_name = strdup(kp_stats[i].ki_comm);
227 –                /* Seems we don't have access to that bit of memory */
228 –                /*proc_state_ptr->proctitle = strdup(kp_stats[i].ki_args->ar_args);*/
229 –                proc_state_ptr->proctitle = NULL;
233  
234 + #ifdef FREEBSD5
235 +                proc_state_ptr->process_name =
236 +                        strdup(kp_stats[i].ki_comm);
237 + #elif defined(DFBSD)
238 +                proc_state_ptr->process_name =
239 +                        strdup(kp_stats[i].kp_thread.td_comm);
240 + #else
241 +                proc_state_ptr->process_name =
242 +                        strdup(kp_stats[i].kp_proc.p_comm);
243 + #endif
244 +
245 +                args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
246 +                if(args != NULL) {
247 +                        alloc = 1;
248 +                        proctitle = malloc(alloc);
249 +                        if(proctitle == NULL) {
250 +                                return NULL;
251 +                        }
252 +                        while(*args != NULL) {
253 +                                if(strlen(proctitle) + strlen(*args) >= alloc) {
254 +                                        alloc = (alloc + strlen(*args)) << 1;
255 +                                        proctitle = realloc(proctitle, alloc);
256 +                                        if(proctitle == NULL) {
257 +                                                return NULL;
258 +                                        }
259 +                                }
260 +                                strncat(proctitle, *args, strlen(*args));
261 +                                strncat(proctitle, " ", 1);
262 +                                args++;
263 +                        }
264 +                        /* remove trailing space */
265 +                        proctitle[strlen(proctitle)-1] = NULL;
266 +                        proc_state_ptr->proctitle = proctitle;
267 +                }
268 +                else {
269 +                        proc_state_ptr->proctitle =
270 +                                malloc(strlen(proc_state_ptr->process_name)+4);
271 +                        if(proc_state_ptr->proctitle == NULL) {
272 +                                return NULL;
273 +                        }
274 +                        sprintf(proc_state_ptr->proctitle, " (%s)",
275 +                                proc_state_ptr->process_name);
276 +                }
277 +
278 + #ifdef FREEBSD5
279                  proc_state_ptr->pid = kp_stats[i].ki_pid;
280                  proc_state_ptr->parent = kp_stats[i].ki_ppid;
281                  proc_state_ptr->pgid = kp_stats[i].ki_pgid;
282 + #else
283 +                proc_state_ptr->pid = kp_stats[i].kp_proc.p_pid;
284 +                proc_state_ptr->parent = kp_stats[i].kp_eproc.e_ppid;
285 +                proc_state_ptr->pgid = kp_stats[i].kp_eproc.e_pgid;
286 + #endif
287  
288 + #ifdef FREEBSD5
289                  proc_state_ptr->uid = kp_stats[i].ki_ruid;
290                  proc_state_ptr->euid = kp_stats[i].ki_uid;
291                  proc_state_ptr->gid = kp_stats[i].ki_rgid;
292                  proc_state_ptr->egid = kp_stats[i].ki_svgid;
293 + #elif defined(DFBSD)
294 +                proc_state_ptr->uid = kp_stats[i].kp_eproc.e_ucred.cr_ruid;
295 +                proc_state_ptr->euid = kp_stats[i].kp_eproc.e_ucred.cr_svuid;
296 +                proc_state_ptr->gid = kp_stats[i].kp_eproc.e_ucred.cr_rgid;
297 +                proc_state_ptr->egid = kp_stats[i].kp_eproc.e_ucred.cr_svgid;
298 + #else
299 +                proc_state_ptr->uid = kp_stats[i].kp_eproc.e_pcred.p_ruid;
300 +                proc_state_ptr->euid = kp_stats[i].kp_eproc.e_pcred.p_svuid;
301 +                proc_state_ptr->gid = kp_stats[i].kp_eproc.e_pcred.p_rgid;
302 +                proc_state_ptr->egid = kp_stats[i].kp_eproc.e_pcred.p_svgid;
303 + #endif
304  
305 + #ifdef FREEBSD5
306                  proc_state_ptr->proc_size = kp_stats[i].ki_size;
307                  /* This is in pages */
308 <                proc_state_ptr->proc_resident = kp_stats[i].ki_rssize * getpagesize();
308 >                proc_state_ptr->proc_resident =
309 >                        kp_stats[i].ki_rssize * getpagesize();
310                  /* This is in microseconds */
311                  proc_state_ptr->time_spent = kp_stats[i].ki_runtime / 1000000;
312 <                proc_state_ptr->cpu_percent = ((double) kp_stats[i].ki_pctcpu / FSCALE) * 100.0;
312 >                proc_state_ptr->cpu_percent =
313 >                        ((double)kp_stats[i].ki_pctcpu / FSCALE) * 100.0;
314                  proc_state_ptr->nice = kp_stats[i].ki_nice;
315 + #else
316 +                proc_state_ptr->proc_size =
317 +                        kp_stats[i].kp_eproc.e_vm.vm_map.size;
318 +                /* This is in pages */
319 +                proc_state_ptr->proc_resident =
320 +                        kp_stats[i].kp_eproc.e_vm.vm_rssize * getpagesize();
321 +                /* This is in microseconds */
322 + #if defined(NETBSD) || defined(OPENBSD)
323 +                proc_state_ptr->time_spent =
324 +                        kp_stats[i].kp_proc.p_rtime.tv_sec;
325 + #elif defined(DFBSD)
326 +                proc_state_ptr->time_spent =
327 +                        ( kp_stats[i].kp_thread.td_uticks +
328 +                        kp_stats[i].kp_thread.td_sticks +
329 +                        kp_stats[i].kp_thread.td_iticks ) / 1000000;
330 + #else
331 +                proc_state_ptr->time_spent =
332 +                        kp_stats[i].kp_proc.p_runtime / 1000000;
333 + #endif
334 +                proc_state_ptr->cpu_percent =
335 +                        ((double)kp_stats[i].kp_proc.p_pctcpu / FSCALE) * 100.0;
336 +                proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
337 + #endif
338  
339 + #ifdef FREEBSD5
340                  switch (kp_stats[i].ki_stat) {
341 + #else
342 +                switch (kp_stats[i].kp_proc.p_stat) {
343 + #endif
344                  case SIDL:
345                  case SRUN:
346   #ifdef SONPROC
# Line 271 | Line 366 | int get_proc_snapshot(proc_state_t **ps){
366   #endif
367                          proc_state_ptr->state = ZOMBIE;
368                          break;
369 +                default:
370 +                        proc_state_ptr->state = UNKNOWN;
371 +                        break;
372                  }
373                  proc_state_size++;
374          }
# Line 287 | Line 385 | process_stat_t *get_process_stats() {
385          proc_state_t *ps;
386          int ps_size, x;
387  
388 <        process_stat.sleeping=0;
389 <        process_stat.running=0;
390 <        process_stat.zombie=0;
391 <        process_stat.stopped=0;
392 <        process_stat.total=0;
388 >        process_stat.sleeping = 0;
389 >        process_stat.running = 0;
390 >        process_stat.zombie = 0;
391 >        process_stat.stopped = 0;
392 >        process_stat.total = 0;
393  
394          ps_size = get_proc_snapshot(&ps);
395  
396 +        if(ps_size == NULL) {
397 +                return NULL;
398 +        }
399 +
400          for(x = 0; x < ps_size; x++) {
401                  switch (ps->state) {
402 +                /* currently no mapping for UNKNOWN in process_stat_t */
403                  case RUNNING:
404                          process_stat.running++;
405                          break;

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines