src/libstatgrab/process_stats.c

/*
 * i-scream libstatgrab
 * http://www.i-scream.org
 * Copyright (C) 2000-2004 i-scream
 * 
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 * 
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
 * 02111-1307 USA
 *
 * $Id: process_stats.c,v 1.63 2004/04/07 14:53:40 tdb Exp $
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include "statgrab.h"
#include "tools.h"
#include "vector.h"
#if defined(SOLARIS) || defined(LINUX)
#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <dirent.h>
#include <string.h>
#endif

#ifdef SOLARIS
#include <procfs.h>
#include <limits.h>
#define PROC_LOCATION "/proc"
#define MAX_FILE_LENGTH PATH_MAX
#endif
#ifdef LINUX
#include <limits.h>
#include <unistd.h>
#include <sys/stat.h>
#include <fcntl.h>
#define PROC_LOCATION "/proc"
#define MAX_FILE_LENGTH PATH_MAX
#endif
#ifdef ALLBSD
#include <stdlib.h>
#include <sys/param.h>
#include <sys/sysctl.h>
#if defined(FREEBSD) || defined(DFBSD)
#include <sys/user.h>
#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>
#endif
#include <unistd.h>
#ifdef NETBSD2
#include <sys/lwp.h>
#endif
#endif

static void proc_state_init(sg_process_stats *s) {
        s->process_name = NULL;
        s->proctitle = NULL;
}

static void proc_state_destroy(sg_process_stats *s) {
        free(s->process_name);
        free(s->proctitle);
}

sg_process_stats *sg_get_process_stats(int *entries){
        VECTOR_DECLARE_STATIC(proc_state, sg_process_stats, 64,
                              proc_state_init, proc_state_destroy);
        int proc_state_size = 0;
        sg_process_stats *proc_state_ptr;
#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, **argsp;
        int argslen = 0;
#else
        long buflen;
        char *p;
#endif
#ifdef NETBSD2
        int lwps;
        struct kinfo_lwp *kl_stats;
#endif
#endif
#if defined(SOLARIS) || defined(LINUX)
        DIR *proc_dir;
        struct dirent *dir_entry;
        char filename[MAX_FILE_LENGTH];
        FILE *f;
#ifdef SOLARIS
        psinfo_t process_info;
#endif
#ifdef LINUX
        char s;
        /* If someone has a executable of 4k filename length, they deserve to get it truncated :) */
        char ps_name[4096];
        char *ptr;
        VECTOR_DECLARE_STATIC(psargs, char, 128, NULL, NULL);
        unsigned long stime, utime, starttime;
        int x;
        int fn;
        int len;
        int rc;
        time_t uptime;
#endif

#ifdef LINUX
        if ((f=fopen("/proc/uptime", "r")) == NULL) {
                return NULL;
        }
        if((fscanf(f,"%lu %*d",&uptime)) != 1){
                return NULL;
        }
        fclose(f);
#endif

        if((proc_dir=opendir(PROC_LOCATION))==NULL){
                return NULL;
        }

        while((dir_entry=readdir(proc_dir))!=NULL){
                if(atoi(dir_entry->d_name) == 0) continue;

#ifdef SOLARIS
                snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
#endif
#ifdef LINUX
                snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/stat", dir_entry->d_name);
#endif
                if((f=fopen(filename, "r"))==NULL){
                        /* Open failed.. Process since vanished, or the path was too long.
                         * Ah well, move onwards to the next one */
                        continue;
                }
#ifdef SOLARIS
                fread(&process_info, sizeof(psinfo_t), 1, f);
                fclose(f);
#endif

                if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
                        return NULL;
                }
                proc_state_ptr = proc_state+proc_state_size;

#ifdef SOLARIS          
                proc_state_ptr->pid = process_info.pr_pid;
                proc_state_ptr->parent = process_info.pr_ppid;
                proc_state_ptr->pgid = process_info.pr_pgid;
                proc_state_ptr->uid = process_info.pr_uid;
                proc_state_ptr->euid = process_info.pr_euid;
                proc_state_ptr->gid = process_info.pr_gid;
                proc_state_ptr->egid = process_info.pr_egid;
                proc_state_ptr->proc_size = (process_info.pr_size) * 1024;
                proc_state_ptr->proc_resident = (process_info.pr_rssize) * 1024;
                proc_state_ptr->time_spent = process_info.pr_time.tv_sec;
                proc_state_ptr->cpu_percent = (process_info.pr_pctcpu * 100.0) / 0x8000;
                proc_state_ptr->nice = (int)process_info.pr_lwp.pr_nice - 20;
                if (sg_update_string(&proc_state_ptr->process_name,
                                     process_info.pr_fname) < 0) {
                        return NULL;
                }
                if (sg_update_string(&proc_state_ptr->proctitle,
                                     process_info.pr_psargs) < 0) {
                        return NULL;
                }

                if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
                if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING; 
                if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE; 
                if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = SG_PROCESS_STATE_STOPPED; 
                if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING; 
#endif
#ifdef LINUX
                x = fscanf(f, "%d %4096s %c %d %d %*d %*d %*d %*u %*u %*u %*u %*u %lu %lu %*d %*d %*d %d %*d %*d %lu %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), &starttime, &(proc_state_ptr->proc_size), &(proc_state_ptr->proc_resident));
                /* +3 becuase man page says "Resident  Set Size: number of pages the process has in real memory, minus 3 for administrative purposes." */
                proc_state_ptr->proc_resident = (proc_state_ptr->proc_resident + 3) * getpagesize();
                if(s == 'S') proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
                if(s == 'R') proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
                if(s == 'Z') proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
                if(s == 'T') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
                if(s == 'D') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
        
                /* pa_name[0] should = '(' */
                ptr = strchr(&ps_name[1], ')'); 
                if(ptr !=NULL) *ptr='\0';

                if (sg_update_string(&proc_state_ptr->process_name,
                                     &ps_name[1]) < 0) {
                        return NULL;
                }

                /* cpu */
                proc_state_ptr->cpu_percent = (100.0 * (utime + stime)) / ((uptime * 100.0) - starttime);

                fclose(f);

                /* uid / gid */
                snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/status", dir_entry->d_name);
                if ((f=fopen(filename, "r")) == NULL) {
                        /* Open failed.. Process since vanished, or the path was too long.
                         * Ah well, move onwards to the next one */
                        continue;
                }

                if((ptr=sg_f_read_line(f, "Uid:"))==NULL){
                        fclose(f);
                        continue;
                }
                sscanf(ptr, "Uid:\t%d\t%d\t%*d\t%*d\n", &(proc_state_ptr->uid), &(proc_state_ptr->euid));

                if((ptr=sg_f_read_line(f, "Gid:"))==NULL){
                        fclose(f);
                        continue;
                }
                sscanf(ptr, "Gid:\t%d\t%d\t%*d\t%*d\n", &(proc_state_ptr->gid), &(proc_state_ptr->egid));

                fclose(f);

                /* proctitle */ 
                snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/cmdline", dir_entry->d_name);

                if((fn=open(filename, O_RDONLY)) == -1){
                        /* Open failed.. Process since vanished, or the path was too long.
                         * Ah well, move onwards to the next one */
                        continue;
                }

#define READ_BLOCK_SIZE 128
                len = 0;
                do {
                        if (VECTOR_RESIZE(psargs, len + READ_BLOCK_SIZE) < 0) {
                                return NULL;
                        }
                        rc = read(fn, psargs + len, READ_BLOCK_SIZE);
                        if (rc > 0) {
                                len += rc;
                        }
                } while (rc == READ_BLOCK_SIZE);
                close(fn);

                if (rc == -1) {
                        /* Read failed; move on. */
                        continue;
                }

                /* Turn \0s into spaces within the command line. */
                ptr = psargs;
                for(x = 0; x < len; x++) {
                        if (*ptr == '\0') *ptr = ' ';
                        ptr++;
                }

                if (len == 0) {
                        /* We want psargs to be NULL. */
                        if (VECTOR_RESIZE(psargs, 0) < 0) {
                                return NULL;
                        }
                } else {
                        /* Not empty, so append a \0. */
                        if (VECTOR_RESIZE(psargs, len + 1) < 0) {
                                return NULL;
                        }
                        psargs[len] = '\0';
                }

                if (sg_update_string(&proc_state_ptr->proctitle, psargs) < 0) {
                        return NULL;
                }
#endif

                proc_state_size++;
        }
        closedir(proc_dir);
#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 = sg_malloc(size);
        if(kp_stats == NULL) {
                return NULL;
        }
        memset(kp_stats, 0, size);

        if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
                free(kp_stats);
                return NULL;
        }

#if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
        kvmd = sg_get_kvm2();
#endif

        for (i = 0; i < procs; i++) {
                const char *name;

#ifdef FREEBSD5
                if (kp_stats[i].ki_stat == 0) {
#else
                if (kp_stats[i].kp_proc.p_stat == 0) {
#endif
                        /* FreeBSD 5 deliberately overallocates the array that
                         * the sysctl returns, so we'll get a few junk
                         * processes on the end that we have to ignore. (Search
                         * for "overestimate by 5 procs" in
                         * src/sys/kern/kern_proc.c for more details.) */
                        continue;
                }

                if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
                        return NULL;
                }
                proc_state_ptr = proc_state+proc_state_size;

#ifdef FREEBSD5
                name = kp_stats[i].ki_comm;
#elif defined(DFBSD)
                name = kp_stats[i].kp_thread.td_comm;
#else
                name = kp_stats[i].kp_proc.p_comm;
#endif
                if (sg_update_string(&proc_state_ptr->process_name, name) < 0) {
                        return NULL;
                }

#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
                /* FIXME - this value can be too large on some of
                   the BSD's, which causes sysctl not to return
                   anything. Maybe we need something smaller? */
                mib[1] = KERN_ARGMAX;

                if(sysctl(mib, 2, &buflen, &size, NULL, 0) < 0) {
                        return NULL;
                }
#endif

                proctitle = sg_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

                free(proc_state_ptr->proctitle);
                if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) {
                        free(proctitle);
                        proc_state_ptr->proctitle = NULL;
                }
                else if(size > 0) {
                        proc_state_ptr->proctitle = sg_malloc(size+1);
                        if(proc_state_ptr->proctitle == NULL) {
                                return NULL;
                        }
                        p = proctitle;
                        proc_state_ptr->proctitle[0] = '\0';
                        do {
                                sg_strlcat(proc_state_ptr->proctitle, p, size+1);
                                sg_strlcat(proc_state_ptr->proctitle, " ", size+1);
                                p += strlen(p) + 1;
                        } while (p < proctitle + size);
                        free(proctitle);
                        /* remove trailing space */
                        proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = '\0';
                }
                else {
                        free(proctitle);
                        proc_state_ptr->proctitle = NULL;
                }
#else
                free(proc_state_ptr->proctitle);
                if(kvmd != NULL) {
                        args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
                        if(args != NULL) {
                                argsp = args;
                                while(*argsp != NULL) {
                                        argslen += strlen(*argsp) + 1;
                                        argsp++;
                                }
                                proctitle = sg_malloc(argslen + 1);
                                proctitle[0] = '\0';
                                if(proctitle == NULL) {
                                        return NULL;
                                }
                                while(*args != NULL) {
                                        sg_strlcat(proctitle, *args, argslen + 1);
                                        sg_strlcat(proctitle, " ", argslen + 1);
                                        args++;
                                }
                                /* remove trailing space */
                                proctitle[strlen(proctitle)-1] = '\0';
                                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 NETBSD2
                {
                        size_t size;
                        int mib[5];

                        mib[0] = CTL_KERN;
                        mib[1] = KERN_LWP;
                        mib[2] = kp_stats[i].kp_proc.p_pid;
                        mib[3] = sizeof(struct kinfo_lwp);
                        mib[4] = 0;

                        if(sysctl(mib, 5, NULL, &size, NULL, 0) < 0) {
                                return NULL;
                        }

                        lwps = size / sizeof(struct kinfo_lwp);
                        mib[4] = lwps;

                        kl_stats = sg_malloc(size);
                        if(kl_stats == NULL) {
                                return NULL;
                        }

                        if(sysctl(mib, 5, kl_stats, &size, NULL, 0) < 0) {
                                return NULL;
                        }
                }

                switch(kp_stats[i].kp_proc.p_stat) {
                case SIDL:
                        proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
                        break;
                case SACTIVE:
                        {
                                int i;

                                for(i = 0; i < lwps; i++) {
                                        switch(kl_stats[i].l_stat) {
                                        case LSONPROC:
                                        case LSRUN:
                                                proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
                                                goto end;
                                        case LSSLEEP:
                                                proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
                                                goto end;
                                        case LSSTOP:
                                        case LSSUSPENDED:
                                                proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
                                                goto end;
                                        }
                                        proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
                                }
                                end: ;
                        }
                        break;
                case SSTOP:
                        proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
                        break;
                case SZOMB:
                        proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
                        break;
                default:
                        proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
                        break;
                }
#else
#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 = SG_PROCESS_STATE_RUNNING;
                        break;
                case SSLEEP:
#ifdef SWAIT
                case SWAIT: /* FreeBSD 5 */
#endif
#ifdef SLOCK
                case SLOCK: /* FreeBSD 5 */
#endif
                        proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
                        break;
                case SSTOP:
                        proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
                        break;
                case SZOMB:
#ifdef SDEAD
                case SDEAD: /* OpenBSD & NetBSD */
#endif
                        proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
                        break;
                default:
                        proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
                        break;
                }
#endif
                proc_state_size++;
        }

        free(kp_stats);
#endif

#ifdef CYGWIN
        return NULL;
#endif

        *entries = proc_state_size;
        return proc_state;
}

sg_process_count *sg_get_process_count() {
        static sg_process_count process_stat;
        sg_process_stats *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 = sg_get_process_stats(&ps_size);
        if (ps == NULL) {
                return NULL;
        }

        for(x = 0; x < ps_size; x++) {
                switch (ps->state) {
                case SG_PROCESS_STATE_RUNNING:
                        process_stat.running++;
                        break;
                case SG_PROCESS_STATE_SLEEPING:
                        process_stat.sleeping++;
                        break;
                case SG_PROCESS_STATE_STOPPED:
                        process_stat.stopped++;
                        break;
                case SG_PROCESS_STATE_ZOMBIE:
                        process_stat.zombie++;
                        break;
                default:
                        /* currently no mapping for SG_PROCESS_STATE_UNKNOWN in
                         * sg_process_count */
                        break;
                }
                ps++;
        }

        process_stat.total = ps_size;

        return &process_stat;
}
Revision:	1.64
Committed:	Wed Apr 7 15:46:34 2004 UTC (20 years, 1 month ago) by pajs
Content type:	text/plain
Branch:	MAIN
Changes since 1.63:	+3 -1 lines
Log Message:	Fixed tools.c change of enum Fixed a FD leak in solaris and added nice value.
#	User	Rev	Content
1	tdb	1.17	/*
2	tdb	1.57	* i-scream libstatgrab
3	tdb	1.8	* http://www.i-scream.org
4	tdb	1.17	* Copyright (C) 2000-2004 i-scream
5	pajs	1.1	*
6	tdb	1.17	* This library is free software; you can redistribute it and/or
7			* modify it under the terms of the GNU Lesser General Public
8			* License as published by the Free Software Foundation; either
9			* version 2.1 of the License, or (at your option) any later version.
10	pajs	1.1	*
11	tdb	1.17	* This library is distributed in the hope that it will be useful,
12	pajs	1.1	* but WITHOUT ANY WARRANTY; without even the implied warranty of
13	tdb	1.17	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14			* Lesser General Public License for more details.
15	pajs	1.1	*
16	tdb	1.17	* You should have received a copy of the GNU Lesser General Public
17			* License along with this library; if not, write to the Free Software
18			* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
19			* 02111-1307 USA
20	tdb	1.18	*
21	pajs	1.64	* $Id: process_stats.c,v 1.63 2004/04/07 14:53:40 tdb Exp $
22	pajs	1.1	*/
23
24			#ifdef HAVE_CONFIG_H
25			#include "config.h"
26			#endif
27
28	pajs	1.6	#include "statgrab.h"
29	ats	1.48	#include "tools.h"
30			#include "vector.h"
31	pajs	1.6	#if defined(SOLARIS) \|\| defined(LINUX)
32	pajs	1.1	#include <stdio.h>
33			#include <stdlib.h>
34	pajs	1.5	#include <sys/types.h>
35			#include <dirent.h>
36			#include <string.h>
37	pajs	1.6	#endif
38	pajs	1.1
39			#ifdef SOLARIS
40			#include <procfs.h>
41			#include <limits.h>
42	pajs	1.5	#define PROC_LOCATION "/proc"
43			#define MAX_FILE_LENGTH PATH_MAX
44			#endif
45			#ifdef LINUX
46	ats	1.16	#include <limits.h>
47	pajs	1.25	#include <unistd.h>
48			#include <sys/stat.h>
49			#include <fcntl.h>
50	pajs	1.1	#define PROC_LOCATION "/proc"
51			#define MAX_FILE_LENGTH PATH_MAX
52			#endif
53	ats	1.13	#ifdef ALLBSD
54	tdb	1.19	#include <stdlib.h>
55	pajs	1.6	#include <sys/param.h>
56			#include <sys/sysctl.h>
57	tdb	1.22	#if defined(FREEBSD) \|\| defined(DFBSD)
58			#include <sys/user.h>
59			#else
60	tdb	1.19	#include <sys/proc.h>
61	ats	1.13	#endif
62	tdb	1.29	#include <string.h>
63			#include <paths.h>
64			#include <fcntl.h>
65			#include <limits.h>
66	tdb	1.41	#if (defined(FREEBSD) && !defined(FREEBSD5)) \|\| defined(DFBSD)
67	tdb	1.29	#include <kvm.h>
68	pajs	1.6	#endif
69	tdb	1.41	#include <unistd.h>
70	tdb	1.55	#ifdef NETBSD2
71			#include <sys/lwp.h>
72			#endif
73	tdb	1.41	#endif
74	pajs	1.1
75	ats	1.52	static void proc_state_init(sg_process_stats *s) {
76	ats	1.48	s->process_name = NULL;
77			s->proctitle = NULL;
78			}
79
80	ats	1.52	static void proc_state_destroy(sg_process_stats *s) {
81	ats	1.48	free(s->process_name);
82			free(s->proctitle);
83			}
84
85	ats	1.52	sg_process_stats sg_get_process_stats(int entries){
86			VECTOR_DECLARE_STATIC(proc_state, sg_process_stats, 64,
87	tdb	1.63	proc_state_init, proc_state_destroy);
88	ats	1.48	int proc_state_size = 0;
89	ats	1.52	sg_process_stats *proc_state_ptr;
90	tdb	1.26	#ifdef ALLBSD
91	tdb	1.37	int mib[4];
92	tdb	1.26	size_t size;
93			struct kinfo_proc *kp_stats;
94	tdb	1.41	int procs, i;
95	tdb	1.45	char *proctitle;
96	tdb	1.41	#if (defined(FREEBSD) && !defined(FREEBSD5)) \|\| defined(DFBSD)
97	tdb	1.42	kvm_t *kvmd;
98	tdb	1.50	char args, argsp;
99			int argslen = 0;
100	tdb	1.41	#else
101	tdb	1.37	long buflen;
102			char *p;
103			#endif
104	tdb	1.55	#ifdef NETBSD2
105			int lwps;
106			struct kinfo_lwp *kl_stats;
107			#endif
108	tdb	1.26	#endif
109	pajs	1.24	#if defined(SOLARIS) \|\| defined(LINUX)
110	tdb	1.63	DIR *proc_dir;
111			struct dirent *dir_entry;
112			char filename[MAX_FILE_LENGTH];
113			FILE *f;
114	pajs	1.24	#ifdef SOLARIS
115	pajs	1.5	psinfo_t process_info;
116	pajs	1.24	#endif
117	pajs	1.25	#ifdef LINUX
118			char s;
119			/* If someone has a executable of 4k filename length, they deserve to get it truncated :) */
120			char ps_name[4096];
121			char *ptr;
122	ats	1.48	VECTOR_DECLARE_STATIC(psargs, char, 128, NULL, NULL);
123	pajs	1.58	unsigned long stime, utime, starttime;
124	pajs	1.25	int x;
125			int fn;
126	ats	1.48	int len;
127			int rc;
128	pajs	1.58	time_t uptime;
129			#endif
130
131			#ifdef LINUX
132			if ((f=fopen("/proc/uptime", "r")) == NULL) {
133			return NULL;
134			}
135			if((fscanf(f,"%lu %*d",&uptime)) != 1){
136			return NULL;
137			}
138			fclose(f);
139	pajs	1.25	#endif
140	pajs	1.1
141	tdb	1.63	if((proc_dir=opendir(PROC_LOCATION))==NULL){
142			return NULL;
143			}
144	pajs	1.23
145	tdb	1.63	while((dir_entry=readdir(proc_dir))!=NULL){
146			if(atoi(dir_entry->d_name) == 0) continue;
147	pajs	1.23
148	pajs	1.24	#ifdef SOLARIS
149	tdb	1.63	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
150	pajs	1.24	#endif
151	pajs	1.25	#ifdef LINUX
152			snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/stat", dir_entry->d_name);
153			#endif
154	tdb	1.63	if((f=fopen(filename, "r"))==NULL){
155			/* Open failed.. Process since vanished, or the path was too long.
156			* Ah well, move onwards to the next one */
157			continue;
158			}
159	pajs	1.24	#ifdef SOLARIS
160	tdb	1.63	fread(&process_info, sizeof(psinfo_t), 1, f);
161	pajs	1.64	fclose(f);
162	pajs	1.25	#endif
163	pajs	1.1
164	ats	1.48	if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
165	ats	1.52	return NULL;
166	ats	1.48	}
167	pajs	1.23	proc_state_ptr = proc_state+proc_state_size;
168	ats	1.48
169	pajs	1.25	#ifdef SOLARIS
170	pajs	1.23	proc_state_ptr->pid = process_info.pr_pid;
171			proc_state_ptr->parent = process_info.pr_ppid;
172			proc_state_ptr->pgid = process_info.pr_pgid;
173			proc_state_ptr->uid = process_info.pr_uid;
174			proc_state_ptr->euid = process_info.pr_euid;
175			proc_state_ptr->gid = process_info.pr_gid;
176			proc_state_ptr->egid = process_info.pr_egid;
177			proc_state_ptr->proc_size = (process_info.pr_size) * 1024;
178			proc_state_ptr->proc_resident = (process_info.pr_rssize) * 1024;
179			proc_state_ptr->time_spent = process_info.pr_time.tv_sec;
180			proc_state_ptr->cpu_percent = (process_info.pr_pctcpu * 100.0) / 0x8000;
181	pajs	1.64	proc_state_ptr->nice = (int)process_info.pr_lwp.pr_nice - 20;
182	ats	1.60	if (sg_update_string(&proc_state_ptr->process_name,
183	tdb	1.63	process_info.pr_fname) < 0) {
184	ats	1.60	return NULL;
185			}
186			if (sg_update_string(&proc_state_ptr->proctitle,
187	tdb	1.63	process_info.pr_psargs) < 0) {
188	ats	1.60	return NULL;
189			}
190	pajs	1.24
191	tdb	1.63	if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
192			if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
193			if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
194			if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
195			if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
196	pajs	1.25	#endif
197			#ifdef LINUX
198	pajs	1.58	x = fscanf(f, "%d %4096s %c %d %d %d %d %d %u %u %u %u %u %lu %lu %d %d %d %d %d %d %lu %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), &starttime, &(proc_state_ptr->proc_size), &(proc_state_ptr->proc_resident));
199			/* +3 becuase man page says "Resident Set Size: number of pages the process has in real memory, minus 3 for administrative purposes." */
200			proc_state_ptr->proc_resident = (proc_state_ptr->proc_resident + 3) * getpagesize();
201	ats	1.52	if(s == 'S') proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
202			if(s == 'R') proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
203			if(s == 'Z') proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
204			if(s == 'T') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
205			if(s == 'D') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
206	pajs	1.25
207			/* pa_name[0] should = '(' */
208			ptr = strchr(&ps_name[1], ')');
209			if(ptr !=NULL) *ptr='\0';
210	ats	1.48
211	ats	1.52	if (sg_update_string(&proc_state_ptr->process_name,
212	tdb	1.63	&ps_name[1]) < 0) {
213	ats	1.52	return NULL;
214	ats	1.48	}
215	pajs	1.25
216	pajs	1.58	/* cpu */
217			proc_state_ptr->cpu_percent = (100.0 * (utime + stime)) / ((uptime * 100.0) - starttime);
218	pajs	1.25
219	tdb	1.63	fclose(f);
220	pajs	1.62
221			/* uid / gid */
222			snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/status", dir_entry->d_name);
223	tdb	1.63	if ((f=fopen(filename, "r")) == NULL) {
224			/* Open failed.. Process since vanished, or the path was too long.
225			* Ah well, move onwards to the next one */
226			continue;
227			}
228	pajs	1.62
229			if((ptr=sg_f_read_line(f, "Uid:"))==NULL){
230			fclose(f);
231			continue;
232			}
233			sscanf(ptr, "Uid:\t%d\t%d\t%d\t%d\n", &(proc_state_ptr->uid), &(proc_state_ptr->euid));
234
235			if((ptr=sg_f_read_line(f, "Gid:"))==NULL){
236	tdb	1.63	fclose(f);
237			continue;
238			}
239			sscanf(ptr, "Gid:\t%d\t%d\t%d\t%d\n", &(proc_state_ptr->gid), &(proc_state_ptr->egid));
240	pajs	1.62
241			fclose(f);
242	ats	1.48
243	pajs	1.25	/* proctitle */
244			snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/cmdline", dir_entry->d_name);
245
246	tdb	1.63	if((fn=open(filename, O_RDONLY)) == -1){
247			/* Open failed.. Process since vanished, or the path was too long.
248			* Ah well, move onwards to the next one */
249			continue;
250			}
251	ats	1.48
252			#define READ_BLOCK_SIZE 128
253			len = 0;
254			do {
255	ats	1.53	if (VECTOR_RESIZE(psargs, len + READ_BLOCK_SIZE) < 0) {
256	ats	1.52	return NULL;
257	ats	1.48	}
258			rc = read(fn, psargs + len, READ_BLOCK_SIZE);
259			if (rc > 0) {
260			len += rc;
261			}
262			} while (rc == READ_BLOCK_SIZE);
263			close(fn);
264
265			if (rc == -1) {
266			/* Read failed; move on. */
267			continue;
268	pajs	1.25	}
269
270	ats	1.48	/* Turn \0s into spaces within the command line. */
271	pajs	1.25	ptr = psargs;
272	ats	1.48	for(x = 0; x < len; x++) {
273	pajs	1.25	if (ptr == '\0') ptr = ' ';
274			ptr++;
275			}
276
277	ats	1.53	if (len == 0) {
278			/* We want psargs to be NULL. */
279			if (VECTOR_RESIZE(psargs, 0) < 0) {
280			return NULL;
281			}
282			} else {
283			/* Not empty, so append a \0. */
284			if (VECTOR_RESIZE(psargs, len + 1) < 0) {
285			return NULL;
286			}
287			psargs[len] = '\0';
288			}
289
290			if (sg_update_string(&proc_state_ptr->proctitle, psargs) < 0) {
291	ats	1.52	return NULL;
292	ats	1.48	}
293	pajs	1.24	#endif
294	pajs	1.23
295			proc_state_size++;
296	tdb	1.63	}
297			closedir(proc_dir);
298	tdb	1.26	#endif
299
300			#ifdef ALLBSD
301	tdb	1.36	mib[0] = CTL_KERN;
302			mib[1] = KERN_PROC;
303			mib[2] = KERN_PROC_ALL;
304
305			if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
306	ats	1.52	return NULL;
307	tdb	1.36	}
308
309			procs = size / sizeof(struct kinfo_proc);
310
311	ats	1.61	kp_stats = sg_malloc(size);
312	tdb	1.36	if(kp_stats == NULL) {
313	ats	1.52	return NULL;
314	tdb	1.36	}
315	ats	1.54	memset(kp_stats, 0, size);
316	tdb	1.36
317			if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
318			free(kp_stats);
319	ats	1.52	return NULL;
320	tdb	1.36	}
321	tdb	1.33
322	tdb	1.37	#if (defined(FREEBSD) && !defined(FREEBSD5)) \|\| defined(DFBSD)
323	ats	1.52	kvmd = sg_get_kvm2();
324	tdb	1.33	#endif
325	tdb	1.26
326			for (i = 0; i < procs; i++) {
327	ats	1.48	const char *name;
328	ats	1.54
329	tdb	1.55	#ifdef FREEBSD5
330	ats	1.54	if (kp_stats[i].ki_stat == 0) {
331	tdb	1.55	#else
332			if (kp_stats[i].kp_proc.p_stat == 0) {
333			#endif
334	ats	1.54	/* FreeBSD 5 deliberately overallocates the array that
335			* the sysctl returns, so we'll get a few junk
336			* processes on the end that we have to ignore. (Search
337			* for "overestimate by 5 procs" in
338			* src/sys/kern/kern_proc.c for more details.) */
339			continue;
340			}
341	ats	1.48
342			if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
343	ats	1.52	return NULL;
344	tdb	1.29	}
345	tdb	1.26	proc_state_ptr = proc_state+proc_state_size;
346	tdb	1.30
347			#ifdef FREEBSD5
348	ats	1.48	name = kp_stats[i].ki_comm;
349	tdb	1.35	#elif defined(DFBSD)
350	ats	1.48	name = kp_stats[i].kp_thread.td_comm;
351	tdb	1.30	#else
352	ats	1.48	name = kp_stats[i].kp_proc.p_comm;
353	tdb	1.30	#endif
354	ats	1.53	if (sg_update_string(&proc_state_ptr->process_name, name) < 0) {
355	ats	1.52	return NULL;
356	ats	1.48	}
357	tdb	1.29
358	tdb	1.37	#if defined(FREEBSD5) \|\| defined(NETBSD) \|\| defined(OPENBSD)
359			size = sizeof(buflen);
360
361			#ifdef FREEBSD5
362			if(sysctlbyname("kern.ps_arg_cache_limit", &buflen, &size, NULL, 0) < 0) {
363	ats	1.52	return NULL;
364	tdb	1.37	}
365			#else
366	tdb	1.56	/* FIXME - this value can be too large on some of
367			the BSD's, which causes sysctl not to return
368			anything. Maybe we need something smaller? */
369	tdb	1.37	mib[1] = KERN_ARGMAX;
370
371			if(sysctl(mib, 2, &buflen, &size, NULL, 0) < 0) {
372	ats	1.52	return NULL;
373	tdb	1.37	}
374			#endif
375
376	ats	1.61	proctitle = sg_malloc(buflen);
377	tdb	1.37	if(proctitle == NULL) {
378	ats	1.52	return NULL;
379	tdb	1.37	}
380
381			size = buflen;
382
383			#ifdef FREEBSD5
384			mib[2] = KERN_PROC_ARGS;
385			mib[3] = kp_stats[i].ki_pid;
386			#else
387			mib[1] = KERN_PROC_ARGS;
388			mib[2] = kp_stats[i].kp_proc.p_pid;
389			mib[3] = KERN_PROC_ARGV;
390			#endif
391
392	ats	1.48	free(proc_state_ptr->proctitle);
393	tdb	1.37	if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) {
394			free(proctitle);
395			proc_state_ptr->proctitle = NULL;
396			}
397			else if(size > 0) {
398	ats	1.61	proc_state_ptr->proctitle = sg_malloc(size+1);
399	tdb	1.37	if(proc_state_ptr->proctitle == NULL) {
400	ats	1.52	return NULL;
401	tdb	1.29	}
402	tdb	1.37	p = proctitle;
403	ats	1.49	proc_state_ptr->proctitle[0] = '\0';
404	tdb	1.37	do {
405	ats	1.52	sg_strlcat(proc_state_ptr->proctitle, p, size+1);
406			sg_strlcat(proc_state_ptr->proctitle, " ", size+1);
407	tdb	1.37	p += strlen(p) + 1;
408			} while (p < proctitle + size);
409			free(proctitle);
410	tdb	1.45	/* remove trailing space */
411			proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = '\0';
412	tdb	1.37	}
413			else {
414			free(proctitle);
415			proc_state_ptr->proctitle = NULL;
416			}
417			#else
418	ats	1.48	free(proc_state_ptr->proctitle);
419	tdb	1.37	if(kvmd != NULL) {
420			args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
421			if(args != NULL) {
422	tdb	1.50	argsp = args;
423			while(*argsp != NULL) {
424	tdb	1.51	argslen += strlen(*argsp) + 1;
425			argsp++;
426	tdb	1.50	}
427	ats	1.61	proctitle = sg_malloc(argslen + 1);
428	tdb	1.50	proctitle[0] = '\0';
429	tdb	1.37	if(proctitle == NULL) {
430	ats	1.52	return NULL;
431	tdb	1.37	}
432			while(*args != NULL) {
433	ats	1.52	sg_strlcat(proctitle, *args, argslen + 1);
434			sg_strlcat(proctitle, " ", argslen + 1);
435	tdb	1.37	args++;
436	tdb	1.29	}
437	tdb	1.37	/* remove trailing space */
438	tdb	1.45	proctitle[strlen(proctitle)-1] = '\0';
439	tdb	1.37	proc_state_ptr->proctitle = proctitle;
440			}
441			else {
442			proc_state_ptr->proctitle = NULL;
443	tdb	1.29	}
444			}
445			else {
446	tdb	1.37	proc_state_ptr->proctitle = NULL;
447	tdb	1.29	}
448	tdb	1.37	#endif
449	tdb	1.26
450	tdb	1.30	#ifdef FREEBSD5
451	tdb	1.26	proc_state_ptr->pid = kp_stats[i].ki_pid;
452			proc_state_ptr->parent = kp_stats[i].ki_ppid;
453			proc_state_ptr->pgid = kp_stats[i].ki_pgid;
454	tdb	1.30	#else
455			proc_state_ptr->pid = kp_stats[i].kp_proc.p_pid;
456			proc_state_ptr->parent = kp_stats[i].kp_eproc.e_ppid;
457			proc_state_ptr->pgid = kp_stats[i].kp_eproc.e_pgid;
458			#endif
459	tdb	1.26
460	tdb	1.30	#ifdef FREEBSD5
461	tdb	1.26	proc_state_ptr->uid = kp_stats[i].ki_ruid;
462			proc_state_ptr->euid = kp_stats[i].ki_uid;
463			proc_state_ptr->gid = kp_stats[i].ki_rgid;
464			proc_state_ptr->egid = kp_stats[i].ki_svgid;
465	tdb	1.35	#elif defined(DFBSD)
466			proc_state_ptr->uid = kp_stats[i].kp_eproc.e_ucred.cr_ruid;
467			proc_state_ptr->euid = kp_stats[i].kp_eproc.e_ucred.cr_svuid;
468			proc_state_ptr->gid = kp_stats[i].kp_eproc.e_ucred.cr_rgid;
469			proc_state_ptr->egid = kp_stats[i].kp_eproc.e_ucred.cr_svgid;
470	tdb	1.30	#else
471			proc_state_ptr->uid = kp_stats[i].kp_eproc.e_pcred.p_ruid;
472			proc_state_ptr->euid = kp_stats[i].kp_eproc.e_pcred.p_svuid;
473			proc_state_ptr->gid = kp_stats[i].kp_eproc.e_pcred.p_rgid;
474			proc_state_ptr->egid = kp_stats[i].kp_eproc.e_pcred.p_svgid;
475			#endif
476	tdb	1.26
477	tdb	1.30	#ifdef FREEBSD5
478	tdb	1.26	proc_state_ptr->proc_size = kp_stats[i].ki_size;
479			/* This is in pages */
480	tdb	1.30	proc_state_ptr->proc_resident =
481			kp_stats[i].ki_rssize * getpagesize();
482	tdb	1.28	/* This is in microseconds */
483	tdb	1.26	proc_state_ptr->time_spent = kp_stats[i].ki_runtime / 1000000;
484	tdb	1.30	proc_state_ptr->cpu_percent =
485			((double)kp_stats[i].ki_pctcpu / FSCALE) * 100.0;
486	tdb	1.26	proc_state_ptr->nice = kp_stats[i].ki_nice;
487	tdb	1.30	#else
488			proc_state_ptr->proc_size =
489			kp_stats[i].kp_eproc.e_vm.vm_map.size;
490			/* This is in pages */
491			proc_state_ptr->proc_resident =
492			kp_stats[i].kp_eproc.e_vm.vm_rssize * getpagesize();
493	tdb	1.34	#if defined(NETBSD) \|\| defined(OPENBSD)
494	tdb	1.33	proc_state_ptr->time_spent =
495			kp_stats[i].kp_proc.p_rtime.tv_sec;
496	tdb	1.35	#elif defined(DFBSD)
497			proc_state_ptr->time_spent =
498			( kp_stats[i].kp_thread.td_uticks +
499			kp_stats[i].kp_thread.td_sticks +
500			kp_stats[i].kp_thread.td_iticks ) / 1000000;
501	tdb	1.33	#else
502	tdb	1.36	/* This is in microseconds */
503	tdb	1.30	proc_state_ptr->time_spent =
504			kp_stats[i].kp_proc.p_runtime / 1000000;
505	tdb	1.33	#endif
506	tdb	1.30	proc_state_ptr->cpu_percent =
507			((double)kp_stats[i].kp_proc.p_pctcpu / FSCALE) * 100.0;
508			proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
509			#endif
510	tdb	1.26
511	tdb	1.55	#ifdef NETBSD2
512			{
513			size_t size;
514			int mib[5];
515
516			mib[0] = CTL_KERN;
517			mib[1] = KERN_LWP;
518			mib[2] = kp_stats[i].kp_proc.p_pid;
519			mib[3] = sizeof(struct kinfo_lwp);
520			mib[4] = 0;
521
522			if(sysctl(mib, 5, NULL, &size, NULL, 0) < 0) {
523			return NULL;
524			}
525
526			lwps = size / sizeof(struct kinfo_lwp);
527			mib[4] = lwps;
528
529	ats	1.61	kl_stats = sg_malloc(size);
530	tdb	1.55	if(kl_stats == NULL) {
531			return NULL;
532			}
533
534			if(sysctl(mib, 5, kl_stats, &size, NULL, 0) < 0) {
535			return NULL;
536			}
537			}
538
539			switch(kp_stats[i].kp_proc.p_stat) {
540			case SIDL:
541			proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
542			break;
543			case SACTIVE:
544			{
545			int i;
546
547			for(i = 0; i < lwps; i++) {
548			switch(kl_stats[i].l_stat) {
549			case LSONPROC:
550			case LSRUN:
551			proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
552			goto end;
553			case LSSLEEP:
554			proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
555			goto end;
556			case LSSTOP:
557			case LSSUSPENDED:
558			proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
559			goto end;
560			}
561			proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
562			}
563			end: ;
564			}
565			break;
566			case SSTOP:
567			proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
568			break;
569			case SZOMB:
570			proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
571			break;
572			default:
573			proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
574			break;
575			}
576			#else
577	tdb	1.30	#ifdef FREEBSD5
578	tdb	1.26	switch (kp_stats[i].ki_stat) {
579	tdb	1.30	#else
580			switch (kp_stats[i].kp_proc.p_stat) {
581			#endif
582	tdb	1.26	case SIDL:
583			case SRUN:
584			#ifdef SONPROC
585			case SONPROC: /* NetBSD */
586			#endif
587	ats	1.52	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
588	tdb	1.26	break;
589			case SSLEEP:
590			#ifdef SWAIT
591			case SWAIT: /* FreeBSD 5 */
592			#endif
593			#ifdef SLOCK
594			case SLOCK: /* FreeBSD 5 */
595			#endif
596	ats	1.52	proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
597	tdb	1.26	break;
598			case SSTOP:
599	ats	1.52	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
600	tdb	1.26	break;
601			case SZOMB:
602			#ifdef SDEAD
603			case SDEAD: /* OpenBSD & NetBSD */
604			#endif
605	ats	1.52	proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
606	tdb	1.31	break;
607			default:
608	ats	1.52	proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
609	tdb	1.26	break;
610			}
611	tdb	1.55	#endif
612	tdb	1.26	proc_state_size++;
613			}
614
615			free(kp_stats);
616	tdb	1.47	#endif
617
618			#ifdef CYGWIN
619	ats	1.52	return NULL;
620	tdb	1.26	#endif
621
622	ats	1.52	*entries = proc_state_size;
623			return proc_state;
624	tdb	1.27	}
625
626	ats	1.52	sg_process_count *sg_get_process_count() {
627			static sg_process_count process_stat;
628			sg_process_stats *ps;
629	tdb	1.27	int ps_size, x;
630
631	tdb	1.30	process_stat.sleeping = 0;
632			process_stat.running = 0;
633			process_stat.zombie = 0;
634			process_stat.stopped = 0;
635			process_stat.total = 0;
636	tdb	1.27
637	ats	1.52	ps = sg_get_process_stats(&ps_size);
638			if (ps == NULL) {
639	tdb	1.32	return NULL;
640			}
641	tdb	1.27
642			for(x = 0; x < ps_size; x++) {
643			switch (ps->state) {
644	ats	1.52	case SG_PROCESS_STATE_RUNNING:
645	tdb	1.27	process_stat.running++;
646			break;
647	ats	1.52	case SG_PROCESS_STATE_SLEEPING:
648	tdb	1.27	process_stat.sleeping++;
649			break;
650	ats	1.52	case SG_PROCESS_STATE_STOPPED:
651	tdb	1.27	process_stat.stopped++;
652			break;
653	ats	1.52	case SG_PROCESS_STATE_ZOMBIE:
654	tdb	1.27	process_stat.zombie++;
655	ats	1.40	break;
656			default:
657	ats	1.52	/* currently no mapping for SG_PROCESS_STATE_UNKNOWN in
658			* sg_process_count */
659	tdb	1.27	break;
660			}
661			ps++;
662			}
663
664			process_stat.total = ps_size;
665
666			return &process_stat;
667	pajs	1.23	}