src/libstatgrab/process_stats.c

/* 
 * i-scream central monitoring system
 * http://www.i-scream.org
 * Copyright (C) 2000-2003 i-scream
 * 
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 * 
 * This program 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 General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */

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

#include "statgrab.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 "tools.h"
#include <linux/limits.h>
#define PROC_LOCATION "/proc"
#define MAX_FILE_LENGTH PATH_MAX
#endif
#ifdef FREEBSD
#include <kvm.h>
#include <fcntl.h>
#include <sys/param.h>
#include <sys/sysctl.h>
#include <sys/user.h>
#endif

process_stat_t *get_process_stats(){

        static process_stat_t process_stat;

#if defined(SOLARIS) || defined(LINUX)
        DIR *proc_dir; 
        struct dirent *dir_entry;
        char filename[MAX_FILE_LENGTH];
        FILE *f;
#endif
#ifdef LINUX
        char *line_ptr;
#endif
#ifdef SOLARIS
        psinfo_t process_info;
#endif
#ifdef FREEBSD
        struct kinfo_proc *kp_stats;    
        kvm_t *kvmd = NULL;
        int procs;
#endif

        process_stat.sleeping=0;
        process_stat.running=0;
        process_stat.zombie=0;
        process_stat.stopped=0;
        process_stat.total=0;

#if defined(SOLARIS) || defined(LINUX)
        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/status", 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);
                if(process_info.pr_lwp.pr_state==1) process_stat.sleeping++;
                if(process_info.pr_lwp.pr_state==2) process_stat.running++;
                if(process_info.pr_lwp.pr_state==3) process_stat.zombie++;
                if(process_info.pr_lwp.pr_state==4) process_stat.stopped++;
                if(process_info.pr_lwp.pr_state==6) process_stat.running++;
#endif
#ifdef LINUX
                if((line_ptr=f_read_line(f, "State:"))==NULL){
                        fclose(f);
                        continue;
                }
                if((line_ptr=strchr(line_ptr, '\t'))==NULL){
                        fclose(f);
                        continue;
                }
                line_ptr++;
                if(line_ptr=='\0'){
                        fclose(f);
                        continue;
                }

                if(*line_ptr=='S') process_stat.sleeping++;
                if(*line_ptr=='R') process_stat.running++;
                if(*line_ptr=='Z') process_stat.zombie++;
                if(*line_ptr=='T') process_stat.stopped++;
                if(*line_ptr=='D') process_stat.stopped++;
#endif  

                fclose(f);
        }
        closedir(proc_dir);
#endif
#ifdef FREEBSD
        if((kvmd = kvm_openfiles(NULL, NULL, NULL, O_RDONLY, NULL)) == NULL){
                return NULL;
        }

        kp_stats=kvm_getprocs(kvmd, KERN_PROC_ALL, 0, &procs);

        while(procs--){
#ifdef FREEBSD5
                if (kp_stats[procs].ki_stat == SSLEEP) process_stat.sleeping++;
                if (kp_stats[procs].ki_stat == SRUN) process_stat.running++;
                if (kp_stats[procs].ki_stat == SIDL) process_stat.running++;
                if (kp_stats[procs].ki_stat == SZOMB) process_stat.zombie++;
                if (kp_stats[procs].ki_stat == SSTOP) process_stat.stopped++;
#else
                if (kp_stats[procs].kp_proc.p_stat == SSLEEP) process_stat.sleeping++;  
                if (kp_stats[procs].kp_proc.p_stat == SRUN) process_stat.running++;
                if (kp_stats[procs].kp_proc.p_stat == SIDL) process_stat.running++;
                if (kp_stats[procs].kp_proc.p_stat == SZOMB) process_stat.zombie++;
                if (kp_stats[procs].kp_proc.p_stat == SSTOP) process_stat.stopped++;
                process_stat.total++;
#endif
        }

        kvm_close(kvmd);
#endif

#ifndef FREEBSD5
        process_stat.total=process_stat.sleeping+process_stat.running+process_stat.zombie+process_stat.stopped;
#endif

        return &process_stat;
}
Revision:	1.10
Committed:	Thu Oct 9 16:48:33 2003 UTC (20 years, 7 months ago) by pajs
Content type:	text/plain
Branch:	MAIN
Changes since 1.9:	+4 -0 lines
Log Message:	Possibly making sure that the total is always correct on freebsd 5, even if some of the states are left out.
#	User	Rev	Content
1	pajs	1.1	/*
2			* i-scream central monitoring system
3	tdb	1.8	* http://www.i-scream.org
4			* Copyright (C) 2000-2003 i-scream
5	pajs	1.1	*
6			* This program is free software; you can redistribute it and/or
7			* modify it under the terms of the GNU General Public License
8			* as published by the Free Software Foundation; either version 2
9			* of the License, or (at your option) any later version.
10			*
11			* This program is distributed in the hope that it will be useful,
12			* but WITHOUT ANY WARRANTY; without even the implied warranty of
13			* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14			* GNU General Public License for more details.
15			*
16			* You should have received a copy of the GNU General Public License
17			* along with this program; if not, write to the Free Software
18			* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
19			*/
20
21			#ifdef HAVE_CONFIG_H
22			#include "config.h"
23			#endif
24
25	pajs	1.6	#include "statgrab.h"
26			#if defined(SOLARIS) \|\| defined(LINUX)
27	pajs	1.1	#include <stdio.h>
28			#include <stdlib.h>
29	pajs	1.5	#include <sys/types.h>
30			#include <dirent.h>
31			#include <string.h>
32	pajs	1.6	#endif
33	pajs	1.1
34			#ifdef SOLARIS
35			#include <procfs.h>
36			#include <limits.h>
37	pajs	1.5	#define PROC_LOCATION "/proc"
38			#define MAX_FILE_LENGTH PATH_MAX
39			#endif
40			#ifdef LINUX
41			#include "tools.h"
42			#include <linux/limits.h>
43	pajs	1.1	#define PROC_LOCATION "/proc"
44			#define MAX_FILE_LENGTH PATH_MAX
45			#endif
46	pajs	1.6	#ifdef FREEBSD
47			#include <kvm.h>
48			#include <fcntl.h>
49			#include <sys/param.h>
50			#include <sys/sysctl.h>
51			#include <sys/user.h>
52			#endif
53	pajs	1.1
54			process_stat_t *get_process_stats(){
55
56			static process_stat_t process_stat;
57
58	pajs	1.6	#if defined(SOLARIS) \|\| defined(LINUX)
59	pajs	1.1	DIR *proc_dir;
60			struct dirent *dir_entry;
61			char filename[MAX_FILE_LENGTH];
62			FILE *f;
63	pajs	1.6	#endif
64	pajs	1.5	#ifdef LINUX
65			char *line_ptr;
66			#endif
67			#ifdef SOLARIS
68			psinfo_t process_info;
69			#endif
70	pajs	1.6	#ifdef FREEBSD
71			struct kinfo_proc *kp_stats;
72			kvm_t *kvmd = NULL;
73			int procs;
74			#endif
75	pajs	1.1
76			process_stat.sleeping=0;
77			process_stat.running=0;
78			process_stat.zombie=0;
79			process_stat.stopped=0;
80	pajs	1.5	process_stat.total=0;
81	pajs	1.1
82	pajs	1.7	#if defined(SOLARIS) \|\| defined(LINUX)
83	pajs	1.1	if((proc_dir=opendir(PROC_LOCATION))==NULL){
84			return NULL;
85			}
86
87			while((dir_entry=readdir(proc_dir))!=NULL){
88			if(atoi(dir_entry->d_name) == 0) continue;
89	pajs	1.5
90			#ifdef SOLARIS
91	pajs	1.1	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
92	pajs	1.5	#endif
93			#ifdef LINUX
94			snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/status", dir_entry->d_name);
95			#endif
96	pajs	1.1
97			if((f=fopen(filename, "r"))==NULL){
98			/* Open failed.. Process since vanished, or the path was too long.
99			* Ah well, move onwards to the next one */
100			continue;
101			}
102
103	pajs	1.5	#ifdef SOLARIS
104	pajs	1.1	fread(&process_info, sizeof(psinfo_t), 1, f);
105			if(process_info.pr_lwp.pr_state==1) process_stat.sleeping++;
106			if(process_info.pr_lwp.pr_state==2) process_stat.running++;
107			if(process_info.pr_lwp.pr_state==3) process_stat.zombie++;
108			if(process_info.pr_lwp.pr_state==4) process_stat.stopped++;
109			if(process_info.pr_lwp.pr_state==6) process_stat.running++;
110	pajs	1.5	#endif
111			#ifdef LINUX
112			if((line_ptr=f_read_line(f, "State:"))==NULL){
113			fclose(f);
114			continue;
115			}
116			if((line_ptr=strchr(line_ptr, '\t'))==NULL){
117			fclose(f);
118			continue;
119			}
120			line_ptr++;
121			if(line_ptr=='\0'){
122			fclose(f);
123			continue;
124			}
125
126			if(*line_ptr=='S') process_stat.sleeping++;
127			if(*line_ptr=='R') process_stat.running++;
128			if(*line_ptr=='Z') process_stat.zombie++;
129			if(*line_ptr=='T') process_stat.stopped++;
130			if(*line_ptr=='D') process_stat.stopped++;
131			#endif
132
133			fclose(f);
134	pajs	1.1	}
135			closedir(proc_dir);
136	pajs	1.6	#endif
137			#ifdef FREEBSD
138			if((kvmd = kvm_openfiles(NULL, NULL, NULL, O_RDONLY, NULL)) == NULL){
139			return NULL;
140			}
141
142			kp_stats=kvm_getprocs(kvmd, KERN_PROC_ALL, 0, &procs);
143	pajs	1.1
144	pajs	1.6	while(procs--){
145	pajs	1.9	#ifdef FREEBSD5
146			if (kp_stats[procs].ki_stat == SSLEEP) process_stat.sleeping++;
147			if (kp_stats[procs].ki_stat == SRUN) process_stat.running++;
148			if (kp_stats[procs].ki_stat == SIDL) process_stat.running++;
149			if (kp_stats[procs].ki_stat == SZOMB) process_stat.zombie++;
150			if (kp_stats[procs].ki_stat == SSTOP) process_stat.stopped++;
151			#else
152	pajs	1.6	if (kp_stats[procs].kp_proc.p_stat == SSLEEP) process_stat.sleeping++;
153			if (kp_stats[procs].kp_proc.p_stat == SRUN) process_stat.running++;
154			if (kp_stats[procs].kp_proc.p_stat == SIDL) process_stat.running++;
155			if (kp_stats[procs].kp_proc.p_stat == SZOMB) process_stat.zombie++;
156			if (kp_stats[procs].kp_proc.p_stat == SSTOP) process_stat.stopped++;
157	pajs	1.10	process_stat.total++;
158	pajs	1.9	#endif
159	pajs	1.6	}
160
161			kvm_close(kvmd);
162			#endif
163	pajs	1.4
164	pajs	1.10	#ifndef FREEBSD5
165	pajs	1.4	process_stat.total=process_stat.sleeping+process_stat.running+process_stat.zombie+process_stat.stopped;
166	pajs	1.10	#endif
167
168	pajs	1.1	return &process_stat;
169			}