src/libstatgrab/process_stats.c

/*
 * i-scream central monitoring system
 * 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.51 2004/04/05 11:21:49 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>
#include <tools.h>
#endif
#include <unistd.h>
#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
#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;
        int x;
        int fn;
        int len;
        int rc;
#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);
#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->process_name = strdup(process_info.pr_fname);
                proc_state_ptr->proctitle = strdup(process_info.pr_psargs);

                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 %*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 = 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]) == NULL) {
                        return NULL;
                }

                /* Need to do cpu */
                

                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 + 1) < 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++;
                }
                /* for safety's sake */
                psargs[len] = '\0';

                if (sg_update_string(&proc_state_ptr->proctitle, psargs) == NULL) {
                        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 = 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 = sg_get_kvm2();
#endif

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

                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) == NULL) {
                        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
                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

                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 = 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 = 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 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;
                }
                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.52
Committed:	Mon Apr 5 15:40:15 2004 UTC (20 years, 1 month ago) by ats
Content type:	text/plain
Branch:	MAIN
Changes since 1.51:	+58 -58 lines
Log Message:	Rename all the functions, types and enums in the API to be consistent. Types are now of the form "sg_mem_stats"; functions of the form "sg_get_mem_stats"; enums of the form "SG_PROCESS_STATE_RUNNING". (The old get_proc_snapshot follows the usual calling convention now.) Make internal functions static where possible. Rename non-static internal functions to have an sg_ prefix so they don't collide with those from other libraries. In particular, strlcpy and strlcat are now called sg_strlcpy and sg_strlcat and are always included. Fix saidar and statgrab to use the new API.
#	User	Rev	Content
1	tdb	1.17	/*
2	pajs	1.1	* i-scream central monitoring system
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	ats	1.52	* $Id: process_stats.c,v 1.51 2004/04/05 11:21:49 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	tdb	1.43	#include <tools.h>
69	pajs	1.6	#endif
70	tdb	1.41	#include <unistd.h>
71			#endif
72	pajs	1.1
73	ats	1.52	static void proc_state_init(sg_process_stats *s) {
74	ats	1.48	s->process_name = NULL;
75			s->proctitle = NULL;
76			}
77
78	ats	1.52	static void proc_state_destroy(sg_process_stats *s) {
79	ats	1.48	free(s->process_name);
80			free(s->proctitle);
81			}
82
83	ats	1.52	sg_process_stats sg_get_process_stats(int entries){
84			VECTOR_DECLARE_STATIC(proc_state, sg_process_stats, 64,
85	ats	1.48	proc_state_init, proc_state_destroy);
86			int proc_state_size = 0;
87	ats	1.52	sg_process_stats *proc_state_ptr;
88	tdb	1.26	#ifdef ALLBSD
89	tdb	1.37	int mib[4];
90	tdb	1.26	size_t size;
91			struct kinfo_proc *kp_stats;
92	tdb	1.41	int procs, i;
93	tdb	1.45	char *proctitle;
94	tdb	1.41	#if (defined(FREEBSD) && !defined(FREEBSD5)) \|\| defined(DFBSD)
95	tdb	1.42	kvm_t *kvmd;
96	tdb	1.50	char args, argsp;
97			int argslen = 0;
98	tdb	1.41	#else
99	tdb	1.37	long buflen;
100			char *p;
101			#endif
102	tdb	1.26	#endif
103	pajs	1.24	#if defined(SOLARIS) \|\| defined(LINUX)
104	pajs	1.23	DIR *proc_dir;
105			struct dirent *dir_entry;
106			char filename[MAX_FILE_LENGTH];
107			FILE *f;
108	pajs	1.24	#ifdef SOLARIS
109	pajs	1.5	psinfo_t process_info;
110	pajs	1.24	#endif
111	pajs	1.25	#ifdef LINUX
112			char s;
113			/* If someone has a executable of 4k filename length, they deserve to get it truncated :) */
114			char ps_name[4096];
115			char *ptr;
116	ats	1.48	VECTOR_DECLARE_STATIC(psargs, char, 128, NULL, NULL);
117	pajs	1.25	unsigned long stime, utime;
118			int x;
119			int fn;
120	ats	1.48	int len;
121			int rc;
122	pajs	1.25	#endif
123	pajs	1.1
124	pajs	1.23	if((proc_dir=opendir(PROC_LOCATION))==NULL){
125	ats	1.52	return NULL;
126	pajs	1.23	}
127
128			while((dir_entry=readdir(proc_dir))!=NULL){
129			if(atoi(dir_entry->d_name) == 0) continue;
130
131	pajs	1.24	#ifdef SOLARIS
132	pajs	1.23	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
133	pajs	1.24	#endif
134	pajs	1.25	#ifdef LINUX
135			snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/stat", dir_entry->d_name);
136			#endif
137	pajs	1.23	if((f=fopen(filename, "r"))==NULL){
138			/* Open failed.. Process since vanished, or the path was too long.
139			* Ah well, move onwards to the next one */
140			continue;
141			}
142	pajs	1.24	#ifdef SOLARIS
143	pajs	1.23	fread(&process_info, sizeof(psinfo_t), 1, f);
144	pajs	1.25	#endif
145	pajs	1.1
146	ats	1.48	if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
147	ats	1.52	return NULL;
148	ats	1.48	}
149	pajs	1.23	proc_state_ptr = proc_state+proc_state_size;
150	ats	1.48
151	pajs	1.25	#ifdef SOLARIS
152	pajs	1.23	proc_state_ptr->pid = process_info.pr_pid;
153			proc_state_ptr->parent = process_info.pr_ppid;
154			proc_state_ptr->pgid = process_info.pr_pgid;
155			proc_state_ptr->uid = process_info.pr_uid;
156			proc_state_ptr->euid = process_info.pr_euid;
157			proc_state_ptr->gid = process_info.pr_gid;
158			proc_state_ptr->egid = process_info.pr_egid;
159			proc_state_ptr->proc_size = (process_info.pr_size) * 1024;
160			proc_state_ptr->proc_resident = (process_info.pr_rssize) * 1024;
161			proc_state_ptr->time_spent = process_info.pr_time.tv_sec;
162			proc_state_ptr->cpu_percent = (process_info.pr_pctcpu * 100.0) / 0x8000;
163			proc_state_ptr->process_name = strdup(process_info.pr_fname);
164			proc_state_ptr->proctitle = strdup(process_info.pr_psargs);
165	pajs	1.24
166	ats	1.52	if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
167			if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
168			if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
169			if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
170			if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
171	pajs	1.25	#endif
172			#ifdef LINUX
173	ats	1.38	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));
174	pajs	1.25	proc_state_ptr->proc_resident = proc_state_ptr->proc_resident * getpagesize();
175	ats	1.52	if(s == 'S') proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
176			if(s == 'R') proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
177			if(s == 'Z') proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
178			if(s == 'T') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
179			if(s == 'D') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
180	pajs	1.25
181			/* pa_name[0] should = '(' */
182			ptr = strchr(&ps_name[1], ')');
183			if(ptr !=NULL) *ptr='\0';
184	ats	1.48
185	ats	1.52	if (sg_update_string(&proc_state_ptr->process_name,
186	ats	1.48	&ps_name[1]) == NULL) {
187	ats	1.52	return NULL;
188	ats	1.48	}
189	pajs	1.25
190			/* Need to do cpu */
191
192
193	ats	1.48	fclose(f);
194
195	pajs	1.25	/* proctitle */
196			snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/cmdline", dir_entry->d_name);
197
198			if((fn=open(filename, O_RDONLY)) == -1){
199			/* Open failed.. Process since vanished, or the path was too long.
200			* Ah well, move onwards to the next one */
201			continue;
202			}
203	ats	1.48
204			#define READ_BLOCK_SIZE 128
205			len = 0;
206			do {
207			if (VECTOR_RESIZE(psargs, len + READ_BLOCK_SIZE + 1) < 0) {
208	ats	1.52	return NULL;
209	ats	1.48	}
210			rc = read(fn, psargs + len, READ_BLOCK_SIZE);
211			if (rc > 0) {
212			len += rc;
213			}
214			} while (rc == READ_BLOCK_SIZE);
215			close(fn);
216
217			if (rc == -1) {
218			/* Read failed; move on. */
219			continue;
220	pajs	1.25	}
221
222	ats	1.48	/* Turn \0s into spaces within the command line. */
223	pajs	1.25	ptr = psargs;
224	ats	1.48	for(x = 0; x < len; x++) {
225	pajs	1.25	if (ptr == '\0') ptr = ' ';
226			ptr++;
227			}
228	ats	1.48	/* for safety's sake */
229			psargs[len] = '\0';
230	pajs	1.25
231	ats	1.52	if (sg_update_string(&proc_state_ptr->proctitle, psargs) == NULL) {
232			return NULL;
233	ats	1.48	}
234	pajs	1.24	#endif
235	pajs	1.23
236			proc_state_size++;
237			}
238			closedir(proc_dir);
239	tdb	1.26	#endif
240
241			#ifdef ALLBSD
242	tdb	1.36	mib[0] = CTL_KERN;
243			mib[1] = KERN_PROC;
244			mib[2] = KERN_PROC_ALL;
245
246			if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
247	ats	1.52	return NULL;
248	tdb	1.36	}
249
250			procs = size / sizeof(struct kinfo_proc);
251
252			kp_stats = malloc(size);
253			if(kp_stats == NULL) {
254	ats	1.52	return NULL;
255	tdb	1.36	}
256
257			if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
258			free(kp_stats);
259	ats	1.52	return NULL;
260	tdb	1.36	}
261	tdb	1.33
262	tdb	1.37	#if (defined(FREEBSD) && !defined(FREEBSD5)) \|\| defined(DFBSD)
263	ats	1.52	kvmd = sg_get_kvm2();
264	tdb	1.33	#endif
265	tdb	1.26
266			for (i = 0; i < procs; i++) {
267	ats	1.48	const char *name;
268
269			if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
270	ats	1.52	return NULL;
271	tdb	1.29	}
272	tdb	1.26	proc_state_ptr = proc_state+proc_state_size;
273	tdb	1.30
274			#ifdef FREEBSD5
275	ats	1.48	name = kp_stats[i].ki_comm;
276	tdb	1.35	#elif defined(DFBSD)
277	ats	1.48	name = kp_stats[i].kp_thread.td_comm;
278	tdb	1.30	#else
279	ats	1.48	name = kp_stats[i].kp_proc.p_comm;
280	tdb	1.30	#endif
281	ats	1.52	if (sg_update_string(&proc_state_ptr->process_name, name) == NULL) {
282			return NULL;
283	ats	1.48	}
284	tdb	1.29
285	tdb	1.37	#if defined(FREEBSD5) \|\| defined(NETBSD) \|\| defined(OPENBSD)
286			size = sizeof(buflen);
287
288			#ifdef FREEBSD5
289			if(sysctlbyname("kern.ps_arg_cache_limit", &buflen, &size, NULL, 0) < 0) {
290	ats	1.52	return NULL;
291	tdb	1.37	}
292			#else
293			mib[1] = KERN_ARGMAX;
294
295			if(sysctl(mib, 2, &buflen, &size, NULL, 0) < 0) {
296	ats	1.52	return NULL;
297	tdb	1.37	}
298			#endif
299
300			proctitle = malloc(buflen);
301			if(proctitle == NULL) {
302	ats	1.52	return NULL;
303	tdb	1.37	}
304
305			size = buflen;
306
307			#ifdef FREEBSD5
308			mib[2] = KERN_PROC_ARGS;
309			mib[3] = kp_stats[i].ki_pid;
310			#else
311			mib[1] = KERN_PROC_ARGS;
312			mib[2] = kp_stats[i].kp_proc.p_pid;
313			mib[3] = KERN_PROC_ARGV;
314			#endif
315
316	ats	1.48	free(proc_state_ptr->proctitle);
317	tdb	1.37	if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) {
318			free(proctitle);
319			proc_state_ptr->proctitle = NULL;
320			}
321			else if(size > 0) {
322	tdb	1.44	proc_state_ptr->proctitle = malloc(size+1);
323	tdb	1.37	if(proc_state_ptr->proctitle == NULL) {
324	ats	1.52	return NULL;
325	tdb	1.29	}
326	tdb	1.37	p = proctitle;
327	ats	1.49	proc_state_ptr->proctitle[0] = '\0';
328	tdb	1.37	do {
329	ats	1.52	sg_strlcat(proc_state_ptr->proctitle, p, size+1);
330			sg_strlcat(proc_state_ptr->proctitle, " ", size+1);
331	tdb	1.37	p += strlen(p) + 1;
332			} while (p < proctitle + size);
333			free(proctitle);
334	tdb	1.45	/* remove trailing space */
335			proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = '\0';
336	tdb	1.37	}
337			else {
338			free(proctitle);
339			proc_state_ptr->proctitle = NULL;
340			}
341			#else
342	ats	1.48	free(proc_state_ptr->proctitle);
343	tdb	1.37	if(kvmd != NULL) {
344			args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
345			if(args != NULL) {
346	tdb	1.50	argsp = args;
347			while(*argsp != NULL) {
348	tdb	1.51	argslen += strlen(*argsp) + 1;
349			argsp++;
350	tdb	1.50	}
351			proctitle = malloc(argslen + 1);
352			proctitle[0] = '\0';
353	tdb	1.37	if(proctitle == NULL) {
354	ats	1.52	return NULL;
355	tdb	1.37	}
356			while(*args != NULL) {
357	ats	1.52	sg_strlcat(proctitle, *args, argslen + 1);
358			sg_strlcat(proctitle, " ", argslen + 1);
359	tdb	1.37	args++;
360	tdb	1.29	}
361	tdb	1.37	/* remove trailing space */
362	tdb	1.45	proctitle[strlen(proctitle)-1] = '\0';
363	tdb	1.37	proc_state_ptr->proctitle = proctitle;
364			}
365			else {
366			proc_state_ptr->proctitle = NULL;
367	tdb	1.29	}
368			}
369			else {
370	tdb	1.37	proc_state_ptr->proctitle = NULL;
371	tdb	1.29	}
372	tdb	1.37	#endif
373	tdb	1.26
374	tdb	1.30	#ifdef FREEBSD5
375	tdb	1.26	proc_state_ptr->pid = kp_stats[i].ki_pid;
376			proc_state_ptr->parent = kp_stats[i].ki_ppid;
377			proc_state_ptr->pgid = kp_stats[i].ki_pgid;
378	tdb	1.30	#else
379			proc_state_ptr->pid = kp_stats[i].kp_proc.p_pid;
380			proc_state_ptr->parent = kp_stats[i].kp_eproc.e_ppid;
381			proc_state_ptr->pgid = kp_stats[i].kp_eproc.e_pgid;
382			#endif
383	tdb	1.26
384	tdb	1.30	#ifdef FREEBSD5
385	tdb	1.26	proc_state_ptr->uid = kp_stats[i].ki_ruid;
386			proc_state_ptr->euid = kp_stats[i].ki_uid;
387			proc_state_ptr->gid = kp_stats[i].ki_rgid;
388			proc_state_ptr->egid = kp_stats[i].ki_svgid;
389	tdb	1.35	#elif defined(DFBSD)
390			proc_state_ptr->uid = kp_stats[i].kp_eproc.e_ucred.cr_ruid;
391			proc_state_ptr->euid = kp_stats[i].kp_eproc.e_ucred.cr_svuid;
392			proc_state_ptr->gid = kp_stats[i].kp_eproc.e_ucred.cr_rgid;
393			proc_state_ptr->egid = kp_stats[i].kp_eproc.e_ucred.cr_svgid;
394	tdb	1.30	#else
395			proc_state_ptr->uid = kp_stats[i].kp_eproc.e_pcred.p_ruid;
396			proc_state_ptr->euid = kp_stats[i].kp_eproc.e_pcred.p_svuid;
397			proc_state_ptr->gid = kp_stats[i].kp_eproc.e_pcred.p_rgid;
398			proc_state_ptr->egid = kp_stats[i].kp_eproc.e_pcred.p_svgid;
399			#endif
400	tdb	1.26
401	tdb	1.30	#ifdef FREEBSD5
402	tdb	1.26	proc_state_ptr->proc_size = kp_stats[i].ki_size;
403			/* This is in pages */
404	tdb	1.30	proc_state_ptr->proc_resident =
405			kp_stats[i].ki_rssize * getpagesize();
406	tdb	1.28	/* This is in microseconds */
407	tdb	1.26	proc_state_ptr->time_spent = kp_stats[i].ki_runtime / 1000000;
408	tdb	1.30	proc_state_ptr->cpu_percent =
409			((double)kp_stats[i].ki_pctcpu / FSCALE) * 100.0;
410	tdb	1.26	proc_state_ptr->nice = kp_stats[i].ki_nice;
411	tdb	1.30	#else
412			proc_state_ptr->proc_size =
413			kp_stats[i].kp_eproc.e_vm.vm_map.size;
414			/* This is in pages */
415			proc_state_ptr->proc_resident =
416			kp_stats[i].kp_eproc.e_vm.vm_rssize * getpagesize();
417	tdb	1.34	#if defined(NETBSD) \|\| defined(OPENBSD)
418	tdb	1.33	proc_state_ptr->time_spent =
419			kp_stats[i].kp_proc.p_rtime.tv_sec;
420	tdb	1.35	#elif defined(DFBSD)
421			proc_state_ptr->time_spent =
422			( kp_stats[i].kp_thread.td_uticks +
423			kp_stats[i].kp_thread.td_sticks +
424			kp_stats[i].kp_thread.td_iticks ) / 1000000;
425	tdb	1.33	#else
426	tdb	1.36	/* This is in microseconds */
427	tdb	1.30	proc_state_ptr->time_spent =
428			kp_stats[i].kp_proc.p_runtime / 1000000;
429	tdb	1.33	#endif
430	tdb	1.30	proc_state_ptr->cpu_percent =
431			((double)kp_stats[i].kp_proc.p_pctcpu / FSCALE) * 100.0;
432			proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
433			#endif
434	tdb	1.26
435	tdb	1.30	#ifdef FREEBSD5
436	tdb	1.26	switch (kp_stats[i].ki_stat) {
437	tdb	1.30	#else
438			switch (kp_stats[i].kp_proc.p_stat) {
439			#endif
440	tdb	1.26	case SIDL:
441			case SRUN:
442			#ifdef SONPROC
443			case SONPROC: /* NetBSD */
444			#endif
445	ats	1.52	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
446	tdb	1.26	break;
447			case SSLEEP:
448			#ifdef SWAIT
449			case SWAIT: /* FreeBSD 5 */
450			#endif
451			#ifdef SLOCK
452			case SLOCK: /* FreeBSD 5 */
453			#endif
454	ats	1.52	proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
455	tdb	1.26	break;
456			case SSTOP:
457	ats	1.52	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
458	tdb	1.26	break;
459			case SZOMB:
460			#ifdef SDEAD
461			case SDEAD: /* OpenBSD & NetBSD */
462			#endif
463	ats	1.52	proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
464	tdb	1.31	break;
465			default:
466	ats	1.52	proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
467	tdb	1.26	break;
468			}
469			proc_state_size++;
470			}
471
472			free(kp_stats);
473	tdb	1.47	#endif
474
475			#ifdef CYGWIN
476	ats	1.52	return NULL;
477	tdb	1.26	#endif
478
479	ats	1.52	*entries = proc_state_size;
480			return proc_state;
481	tdb	1.27	}
482
483	ats	1.52	sg_process_count *sg_get_process_count() {
484			static sg_process_count process_stat;
485			sg_process_stats *ps;
486	tdb	1.27	int ps_size, x;
487
488	tdb	1.30	process_stat.sleeping = 0;
489			process_stat.running = 0;
490			process_stat.zombie = 0;
491			process_stat.stopped = 0;
492			process_stat.total = 0;
493	tdb	1.27
494	ats	1.52	ps = sg_get_process_stats(&ps_size);
495			if (ps == NULL) {
496	tdb	1.32	return NULL;
497			}
498	tdb	1.27
499			for(x = 0; x < ps_size; x++) {
500			switch (ps->state) {
501	ats	1.52	case SG_PROCESS_STATE_RUNNING:
502	tdb	1.27	process_stat.running++;
503			break;
504	ats	1.52	case SG_PROCESS_STATE_SLEEPING:
505	tdb	1.27	process_stat.sleeping++;
506			break;
507	ats	1.52	case SG_PROCESS_STATE_STOPPED:
508	tdb	1.27	process_stat.stopped++;
509			break;
510	ats	1.52	case SG_PROCESS_STATE_ZOMBIE:
511	tdb	1.27	process_stat.zombie++;
512	ats	1.40	break;
513			default:
514	ats	1.52	/* currently no mapping for SG_PROCESS_STATE_UNKNOWN in
515			* sg_process_count */
516	tdb	1.27	break;
517			}
518			ps++;
519			}
520
521			process_stat.total = ps_size;
522
523			return &process_stat;
524	pajs	1.23	}