[ViewVC] Diff of: i-scream/projects/libstatgrab/src/libstatgrab/process

Comparing projects/libstatgrab/src/libstatgrab/process_stats.c (file contents):
Revision 1.25 by pajs, Tue Mar 30 13:41:31 2004 UTC vs.
Revision 1.43 by tdb, Sun Apr 4 22:48:50 2004 UTC

+#else
+#include <sys/proc.h>
+#endif
-+
+#include <string.h>
-+
+#include <paths.h>
-+
+#include <fcntl.h>
-+
+#include <limits.h>
-+
+#if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
-+
+#include <kvm.h>
-+
+#include <tools.h>
+#endif
-+
+#include <unistd.h>
-+
+#endif
+int get_proc_snapshot(proc_state_t **ps){
+        proc_state_t *proc_state = NULL;
+        proc_state_t *proc_state_ptr;
+        int proc_state_size = 0;
-+
+#ifdef ALLBSD
-+
+        int mib[4];
-+
+        size_t size;
-+
+        struct kinfo_proc *kp_stats;
-+
+        int procs, i;
-+
+        char *proctitle;
-+
+#if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
-+
+        kvm_t *kvmd;
-+
+        char **args;
-+
+        int alloc;
-+
+#else
-+
+        long buflen;
-+
+        char *p;
-+
+#endif
-+
+#endif
+#if defined(SOLARIS) || defined(LINUX)
+        DIR *proc_dir;
+        struct dirent *dir_entry;
+                if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = RUNNING;
+#endif
+#ifdef LINUX
-<
+                x = fscanf(f, "%d %4096s %c %d %d %*d %*d %*d %*lu %*lu %*lu %*lu %*lu %lu %lu %*ld %*ld %*ld %d %*ld %*ld %*lu %llu %llu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*d %*d\n", &(proc_state_ptr->pid), ps_name, &s, &(proc_state_ptr->parent), &(proc_state_ptr->pgid), &utime, &stime, &(proc_state_ptr->nice), &(proc_state_ptr->proc_size), &(proc_state_ptr->proc_resident));
->
+                x = fscanf(f, "%d %4096s %c %d %d %*d %*d %*d %*u %*u %*u %*u %*u %lu %lu %*d %*d %*d %d %*d %*d %*u %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), &(proc_state_ptr->proc_size), &(proc_state_ptr->proc_resident));
+                proc_state_ptr->proc_resident = proc_state_ptr->proc_resident * getpagesize();
+                if(s == 'S') proc_state_ptr->state = SLEEPING;
+                if(s == 'R') proc_state_ptr->state = RUNNING;
+#endif
+                proc_state_size++;
-–
+#endif
-–
+                fclose(f);
+        closedir(proc_dir);
-<
+        *ps = proc_state;
->
+#endif
-+
+#ifdef ALLBSD
-+
+        mib[0] = CTL_KERN;
-+
+        mib[1] = KERN_PROC;
-+
+        mib[2] = KERN_PROC_ALL;
-+
-+
+        if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
-+
+                return NULL;
-+
+        }
-+
-+
+        procs = size / sizeof(struct kinfo_proc);
-+
-+
+        kp_stats = malloc(size);
-+
+        if(kp_stats == NULL) {
-+
+                return NULL;
-+
+        }
-+
-+
+        if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
-+
+                free(kp_stats);
-+
+                return NULL;
-+
+        }
-+
-+
+#if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
-+
+        kvmd = get_kvm2();
-+
+#endif
-+
-+
+        for (i = 0; i < procs; i++) {
-+
+                /* replace with something more sensible */
-+
+                proc_state = realloc(proc_state,
-+
+                                (1+proc_state_size)*sizeof(proc_state_t));
-+
+                if(proc_state == NULL ) {
-+
+                        return NULL;
-+
+                }
-+
+                proc_state_ptr = proc_state+proc_state_size;
-+
-+
+#ifdef FREEBSD5
-+
+                proc_state_ptr->process_name =
-+
+                        strdup(kp_stats[i].ki_comm);
-+
+#elif defined(DFBSD)
-+
+                proc_state_ptr->process_name =
-+
+                        strdup(kp_stats[i].kp_thread.td_comm);
-+
+#else
-+
+                proc_state_ptr->process_name =
-+
+                        strdup(kp_stats[i].kp_proc.p_comm);
-+
+#endif
-+
-+
+#if defined(FREEBSD5) || defined(NETBSD) || defined(OPENBSD)
-+
+                size = sizeof(buflen);
-+
-+
+#ifdef FREEBSD5
-+
+                if(sysctlbyname("kern.ps_arg_cache_limit", &buflen, &size, NULL, 0) < 0) {
-+
+                        return NULL;
-+
+                }
-+
+#else
-+
+                mib[1] = KERN_ARGMAX;
-+
-+
+                if(sysctl(mib, 2, &buflen, &size, NULL, 0) < 0) {
-+
+                        return NULL;
-+
+                }
-+
+#endif
-+
-+
+                proctitle = malloc(buflen);
-+
+                if(proctitle == NULL) {
-+
+                        return NULL;
-+
+                }
-+
-+
+                size = buflen;
-+
-+
+#ifdef FREEBSD5
-+
+                mib[2] = KERN_PROC_ARGS;
-+
+                mib[3] = kp_stats[i].ki_pid;
-+
+#else
-+
+                mib[1] = KERN_PROC_ARGS;
-+
+                mib[2] = kp_stats[i].kp_proc.p_pid;
-+
+                mib[3] = KERN_PROC_ARGV;
-+
+#endif
-+
-+
+                if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) {
-+
+                        free(proctitle);
-+
+                        proc_state_ptr->proctitle = NULL;
-+
+                }
-+
+                else if(size > 0) {
-+
+                        proc_state_ptr->proctitle = malloc(size);
-+
+                        if(proc_state_ptr->proctitle == NULL) {
-+
+                                return NULL;
-+
+                        }
-+
+                        p = proctitle;
-+
+                        proc_state_ptr->proctitle[0] = NULL;
-+
+                        do {
-+
+                                strncat(proc_state_ptr->proctitle, p, strlen(p));
-+
+                                strncat(proc_state_ptr->proctitle, " ", 1);
-+
+                                p += strlen(p) + 1;
-+
+                        } while (p < proctitle + size);
-+
+                        free(proctitle);
-+
+                        proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = NULL;
-+
+                }
-+
+                else {
-+
+                        free(proctitle);
-+
+                        proc_state_ptr->proctitle = NULL;
-+
+                }
-+
+#else
-+
+                if(kvmd != NULL) {
-+
+                        args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
-+
+                        if(args != NULL) {
-+
+                                alloc = 1;
-+
+                                proctitle = malloc(alloc);
-+
+                                if(proctitle == NULL) {
-+
+                                        return NULL;
-+
+                                }
-+
+                                while(*args != NULL) {
-+
+                                        if(strlen(proctitle) + strlen(*args) >= alloc) {
-+
+                                                alloc = (alloc + strlen(*args)) << 1;
-+
+                                                proctitle = realloc(proctitle, alloc);
-+
+                                                if(proctitle == NULL) {
-+
+                                                        return NULL;
-+
+                                                }
-+
+                                        }
-+
+                                        strncat(proctitle, *args, strlen(*args));
-+
+                                        strncat(proctitle, " ", 1);
-+
+                                        args++;
-+
+                                }
-+
+                                /* remove trailing space */
-+
+                                proctitle[strlen(proctitle)-1] = NULL;
-+
+                                proc_state_ptr->proctitle = proctitle;
-+
+                        }
-+
+                        else {
-+
+                                proc_state_ptr->proctitle = NULL;
-+
+                        }
-+
+                }
-+
+                else {
-+
+                        proc_state_ptr->proctitle = NULL;
-+
+                }
-+
+#endif
-+
-+
+#ifdef FREEBSD5
-+
+                proc_state_ptr->pid = kp_stats[i].ki_pid;
-+
+                proc_state_ptr->parent = kp_stats[i].ki_ppid;
-+
+                proc_state_ptr->pgid = kp_stats[i].ki_pgid;
-+
+#else
-+
+                proc_state_ptr->pid = kp_stats[i].kp_proc.p_pid;
-+
+                proc_state_ptr->parent = kp_stats[i].kp_eproc.e_ppid;
-+
+                proc_state_ptr->pgid = kp_stats[i].kp_eproc.e_pgid;
-+
+#endif
-+
-+
+#ifdef FREEBSD5
-+
+                proc_state_ptr->uid = kp_stats[i].ki_ruid;
-+
+                proc_state_ptr->euid = kp_stats[i].ki_uid;
-+
+                proc_state_ptr->gid = kp_stats[i].ki_rgid;
-+
+                proc_state_ptr->egid = kp_stats[i].ki_svgid;
-+
+#elif defined(DFBSD)
-+
+                proc_state_ptr->uid = kp_stats[i].kp_eproc.e_ucred.cr_ruid;
-+
+                proc_state_ptr->euid = kp_stats[i].kp_eproc.e_ucred.cr_svuid;
-+
+                proc_state_ptr->gid = kp_stats[i].kp_eproc.e_ucred.cr_rgid;
-+
+                proc_state_ptr->egid = kp_stats[i].kp_eproc.e_ucred.cr_svgid;
-+
+#else
-+
+                proc_state_ptr->uid = kp_stats[i].kp_eproc.e_pcred.p_ruid;
-+
+                proc_state_ptr->euid = kp_stats[i].kp_eproc.e_pcred.p_svuid;
-+
+                proc_state_ptr->gid = kp_stats[i].kp_eproc.e_pcred.p_rgid;
-+
+                proc_state_ptr->egid = kp_stats[i].kp_eproc.e_pcred.p_svgid;
-+
+#endif
-+
-+
+#ifdef FREEBSD5
-+
+                proc_state_ptr->proc_size = kp_stats[i].ki_size;
-+
+                /* This is in pages */
-+
+                proc_state_ptr->proc_resident =
-+
+                        kp_stats[i].ki_rssize * getpagesize();
-+
+                /* This is in microseconds */
-+
+                proc_state_ptr->time_spent = kp_stats[i].ki_runtime / 1000000;
-+
+                proc_state_ptr->cpu_percent =
-+
+                        ((double)kp_stats[i].ki_pctcpu / FSCALE) * 100.0;
-+
+                proc_state_ptr->nice = kp_stats[i].ki_nice;
-+
+#else
-+
+                proc_state_ptr->proc_size =
-+
+                        kp_stats[i].kp_eproc.e_vm.vm_map.size;
-+
+                /* This is in pages */
-+
+                proc_state_ptr->proc_resident =
-+
+                        kp_stats[i].kp_eproc.e_vm.vm_rssize * getpagesize();
-+
+#if defined(NETBSD) || defined(OPENBSD)
-+
+                proc_state_ptr->time_spent =
-+
+                        kp_stats[i].kp_proc.p_rtime.tv_sec;
-+
+#elif defined(DFBSD)
-+
+                proc_state_ptr->time_spent =
-+
+                        ( kp_stats[i].kp_thread.td_uticks +
-+
+                        kp_stats[i].kp_thread.td_sticks +
-+
+                        kp_stats[i].kp_thread.td_iticks ) / 1000000;
-+
+#else
-+
+                /* This is in microseconds */
-+
+                proc_state_ptr->time_spent =
-+
+                        kp_stats[i].kp_proc.p_runtime / 1000000;
-+
+#endif
-+
+                proc_state_ptr->cpu_percent =
-+
+                        ((double)kp_stats[i].kp_proc.p_pctcpu / FSCALE) * 100.0;
-+
+                proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
-+
+#endif
-+
-+
+#ifdef FREEBSD5
-+
+                switch (kp_stats[i].ki_stat) {
-+
+#else
-+
+                switch (kp_stats[i].kp_proc.p_stat) {
-+
+#endif
-+
+                case SIDL:
-+
+                case SRUN:
-+
+#ifdef SONPROC
-+
+                case SONPROC: /* NetBSD */
-+
+#endif
-+
+                        proc_state_ptr->state = RUNNING;
-+
+                        break;
-+
+                case SSLEEP:
-+
+#ifdef SWAIT
-+
+                case SWAIT: /* FreeBSD 5 */
-+
+#endif
-+
+#ifdef SLOCK
-+
+                case SLOCK: /* FreeBSD 5 */
-+
+#endif
-+
+                        proc_state_ptr->state = SLEEPING;
-+
+                        break;
-+
+                case SSTOP:
-+
+                        proc_state_ptr->state = STOPPED;
-+
+                        break;
-+
+                case SZOMB:
-+
+#ifdef SDEAD
-+
+                case SDEAD: /* OpenBSD & NetBSD */
-+
+#endif
-+
+                        proc_state_ptr->state = ZOMBIE;
-+
+                        break;
-+
+                default:
-+
+                        proc_state_ptr->state = UNKNOWN;
-+
+                        break;
-+
+                }
-+
+                proc_state_size++;
-+
+        }
-+
-+
+        free(kp_stats);
-+
+#endif
-+
-+
+        *ps = proc_state;
+        return proc_state_size;
-+
+process_stat_t *get_process_stats() {
-+
+        static process_stat_t process_stat;
-+
+        proc_state_t *ps;
-+
+        int ps_size, x;
-+
-+
+        process_stat.sleeping = 0;
-+
+        process_stat.running = 0;
-+
+        process_stat.zombie = 0;
-+
+        process_stat.stopped = 0;
-+
+        process_stat.total = 0;
-+
-+
+        ps_size = get_proc_snapshot(&ps);
-+
-+
+        if(ps_size == 0) {
-+
+                return NULL;
-+
+        }
-+
-+
+        for(x = 0; x < ps_size; x++) {
-+
+                switch (ps->state) {
-+
+                case RUNNING:
-+
+                        process_stat.running++;
-+
+                        break;
-+
+                case SLEEPING:
-+
+                        process_stat.sleeping++;
-+
+                        break;
-+
+                case STOPPED:
-+
+                        process_stat.stopped++;
-+
+                        break;
-+
+                case ZOMBIE:
-+
+                        process_stat.zombie++;
-+
+                        break;
-+
+                default:
-+
+                        /* currently no mapping for UNKNOWN in process_stat_t */
-+
+                        break;
-+
+                }
-+
+                ps++;
-+
+        }
-+
-+
+        process_stat.total = ps_size;
-+
-+
+        return &process_stat;
-+
+}

Diff Legend

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

Comparing projects/libstatgrab/src/libstatgrab/process_stats.c (file contents): Revision 1.25 by pajs, Tue Mar 30 13:41:31 2004 UTC vs. Revision 1.43 by tdb, Sun Apr 4 22:48:50 2004 UTC

Diff Legend

Comparing projects/libstatgrab/src/libstatgrab/process_stats.c (file contents):
Revision 1.25 by pajs, Tue Mar 30 13:41:31 2004 UTC vs.
Revision 1.43 by tdb, Sun Apr 4 22:48:50 2004 UTC