ihost/libstatgrab/disk_stat.c

/* 
 * i-scream central monitoring system
 * http://www.i-scream.org.uk
 * Copyright (C) 2000-2002 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.
 */

#include <stdlib.h>
#include <stdio.h>
#include "ukcprog.h"
#include <string.h>

#ifdef SOLARIS
#include <sys/mnttab.h>
#include <sys/types.h>
#include <sys/statvfs.h>
#endif
#ifdef LINUX
#include <mntent.h>
#include <sys/vfs.h>
#endif
#ifdef FREEBSD
#include <sys/param.h>
#include <sys/ucred.h>
#include <sys/mount.h>
#endif
typedef struct {
        char *device_name;
        char *mnt_point;
        long size;
        long used;
        long avail;
        long t_inodes;
        long f_inodes;
}disk_stat_t;

typedef struct system_disks_t{
        disk_stat_t *disk;
        struct system_disks_t *next_disk;
}system_disks_t;

char *get_disk_stats(){
        system_disks_t *sd=NULL;
        system_disks_t *sd_ptr=sd;
        int counter=0;
        char *xml_disk_stats;
        char *xml_disk_stats_ptr;
#ifdef SOLARIS
        struct mnttab mp;
        struct statvfs df;
        FILE *f;
#endif
#ifdef LINUX
        struct mntent *mp;
        struct statfs df;
        FILE *f;
#endif
#ifdef FREEBSD
        int nummnt;
        struct statfs *mp;
#endif

#ifdef SOLARIS
        if ((f=fopen("/etc/mnttab", "r" ))==NULL){
                errf("Failed to open mounts (%m)");
                return NULL;
        }
        while((getmntent(f, &mp)) == 0){
                if ((statvfs(mp.mnt_mountp, &df)) !=0){
                        errf("Failed to gets fs stats (%m)");
                        continue;
                }
                if((((strcmp(mp.mnt_fstype,"ufs"))==0) || (strcmp(mp.mnt_fstype,"tmpfs")) ==0)){
                        if((sd_ptr=malloc(sizeof(system_disks_t))) == NULL){
                                errf("malloc failed (%m)");
                                return NULL;
                        }
                        if((sd_ptr->disk=malloc(sizeof(disk_stat_t))) == NULL){
                                errf("malloc failed (%m)");
                                return NULL;
                        }
                        if((sd_ptr->disk->device_name=strdup(mp.mnt_special)) == NULL) {
                                errf("strdup failed (%m)");     
                                return NULL;
                        }
                        if((sd_ptr->disk->mnt_point=strdup(mp.mnt_mountp)) == NULL){
                                errf("strdup failed (%m)");     
                                return NULL;
                        }
                        sd_ptr->disk->size=((df.f_frsize/1024) * df.f_blocks);
                        sd_ptr->disk->used=(((df.f_frsize/1024) * df.f_blocks) -((df.f_frsize/1024) * df.f_bfree));
                        sd_ptr->disk->avail=(df.f_frsize/1024) * df.f_bavail;
                        sd_ptr->disk->t_inodes=df.f_files;
                        sd_ptr->disk->f_inodes=df.f_ffree;
                        sd_ptr->next_disk=sd;
                        sd=sd_ptr;
                }

        }

        if((fclose(f))!=0){
                errf("Failed to close mnttab");
                return NULL;
        }

#endif
#ifdef LINUX
        if ((f=setmntent("/etc/mtab", "r" ))==NULL){
                errf("Failed to open mounts (%m)");
                return NULL;
        }

        while((mp=getmntent(f))){
                if ((statfs(mp->mnt_dir, &df)) !=0){
                        errf("Failed to gets fs stats (%m)");
                        continue;
                }

                if((((strcmp(mp->mnt_type, MNTTYPE_NFS))==0) ||  (strcmp(mp->mnt_type,MNTTYPE_IGNORE)) ==0)) continue;
                if((sd_ptr=malloc(sizeof(system_disks_t))) == NULL){
                        errf("malloc failed (%m)");
                        return NULL;
                }
                if((sd_ptr->disk=malloc(sizeof(disk_stat_t))) == NULL){
                        errf("malloc failed (%m)");
                        return NULL;
                }
                if((sd_ptr->disk->device_name=strdup(mp->mnt_fsname)) == NULL){ 
                        errf("strdup failed (%m)");
                        return NULL;
                }
                if((sd_ptr->disk->mnt_point=strdup(mp->mnt_dir)) == NULL){
                        errf("strdup failed (%m)");
                        return NULL;
                }
                sd_ptr->disk->size=((df.f_bsize/1024) * df.f_blocks);
                sd_ptr->disk->used=((df.f_bsize/1024) * df.f_blocks) -((df.f_bsize/1024) * df.f_bfree);
                sd_ptr->disk->avail=((df.f_bsize/1024) * df.f_bavail);
                sd_ptr->disk->t_inodes=df.f_files;
                sd_ptr->disk->f_inodes=df.f_ffree;
                sd_ptr->next_disk=sd;
                sd=sd_ptr;
        }

        endmntent(f);

#endif
#ifdef FREEBSD
        nummnt=getmntinfo(&mp , MNT_LOCAL);
        if (nummnt<=0){
                errf("Failed to get disk stats (%m)");
                return NULL;
        }
        
        for(counter=0;counter<nummnt;counter++){
                if((sd_ptr=malloc(sizeof(system_disks_t))) == NULL){
                        errf("malloc failed (%m)");
                        return NULL;
                }
                if((sd_ptr->disk=malloc(sizeof(disk_stat_t))) == NULL){
                        errf("malloc failed (%m)");
                        return NULL;
                }
                if((sd_ptr->disk->device_name=strdup(mp->f_mntfromname))==NULL){
                        errf("strdup failed (%m)");
                        return NULL;
                }
                if((sd_ptr->disk->mnt_point=strdup(mp->f_mntonname))==NULL){
                        errf("strdup failed (%m)");
                        return NULL;
                }
                sd_ptr->disk->size=((mp->f_bsize/1024) * mp->f_blocks);
                sd_ptr->disk->used=((mp->f_bsize/1024) * mp->f_blocks) -((mp->f_bsize/1024) * mp->f_bfree);
                sd_ptr->disk->avail=((mp->f_bsize/1024) * mp->f_bavail);
                sd_ptr->disk->t_inodes=mp->f_files;
                sd_ptr->disk->f_inodes=mp->f_ffree;
                sd_ptr->next_disk=sd;
                sd=sd_ptr;
                mp++;
        }
#endif


        sd_ptr=sd;
        xml_disk_stats=strdup("<disk>");
        
        for(counter=0;sd_ptr!=NULL;counter++){
                xml_disk_stats_ptr=xml_disk_stats;
                if((xml_disk_stats=strf("%s<p%d name=\"%s\" mount=\"%s\" kbytes=\"%ld\" used=\"%ld\" avail=\"%ld\" totalinodes=\"%ld\" freeinodes=\"%ld\"></p%d>", xml_disk_stats, counter, sd_ptr->disk->device_name, sd_ptr->disk->mnt_point, sd_ptr->disk->size, sd_ptr->disk->used, sd_ptr->disk->avail, sd_ptr->disk->t_inodes, sd_ptr->disk->f_inodes, counter)) == NULL){
                        errf("strf failed (%m)");
                        return NULL;
                }
                free(xml_disk_stats_ptr);
                sd_ptr=sd_ptr->next_disk;
        }
        xml_disk_stats_ptr=xml_disk_stats;
        xml_disk_stats=strf("%s</disk>",xml_disk_stats);
        free(xml_disk_stats_ptr);       

        /* Cleaning up */
        sd_ptr=sd;
        while(sd_ptr!=NULL){
                sd=sd_ptr->next_disk;
                free(sd_ptr->disk->device_name);
                free(sd_ptr->disk->mnt_point);
                free(sd_ptr->disk);
                free(sd_ptr);
                sd_ptr=sd;
        }

        return xml_disk_stats;
}

Revision:	1.12
Committed:	Wed May 22 08:54:53 2002 UTC (22 years, 4 months ago) by pajs
Content type:	text/plain
Branch:	MAIN
Changes since 1.11:	+1 -4 lines
Log Message:	endmntend always succeeds
#	User	Rev	Content
1	tdb	1.7	/*
2			* i-scream central monitoring system
3	tdb	1.11	* http://www.i-scream.org.uk
4	tdb	1.7	* Copyright (C) 2000-2002 i-scream
5			*
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	pajs	1.1	#include <stdlib.h>
22			#include <stdio.h>
23	pajs	1.4	#include "ukcprog.h"
24	pajs	1.2	#include <string.h>
25	pajs	1.1
26			#ifdef SOLARIS
27			#include <sys/mnttab.h>
28			#include <sys/types.h>
29			#include <sys/statvfs.h>
30			#endif
31	pajs	1.2	#ifdef LINUX
32			#include <mntent.h>
33			#include <sys/vfs.h>
34			#endif
35			#ifdef FREEBSD
36			#include <sys/param.h>
37			#include <sys/ucred.h>
38			#include <sys/mount.h>
39			#endif
40	pajs	1.1	typedef struct {
41			char *device_name;
42			char *mnt_point;
43			long size;
44			long used;
45			long avail;
46			long t_inodes;
47			long f_inodes;
48			}disk_stat_t;
49
50			typedef struct system_disks_t{
51			disk_stat_t *disk;
52			struct system_disks_t *next_disk;
53			}system_disks_t;
54
55	pajs	1.2	char *get_disk_stats(){
56	pajs	1.1	system_disks_t *sd=NULL;
57			system_disks_t *sd_ptr=sd;
58			int counter=0;
59	pajs	1.4	char *xml_disk_stats;
60	pajs	1.5	char *xml_disk_stats_ptr;
61	pajs	1.1	#ifdef SOLARIS
62			struct mnttab mp;
63			struct statvfs df;
64	pajs	1.3	FILE *f;
65	pajs	1.1	#endif
66			#ifdef LINUX
67			struct mntent *mp;
68			struct statfs df;
69	pajs	1.3	FILE *f;
70	pajs	1.1	#endif
71			#ifdef FREEBSD
72			int nummnt;
73			struct statfs *mp;
74			#endif
75
76			#ifdef SOLARIS
77			if ((f=fopen("/etc/mnttab", "r" ))==NULL){
78			errf("Failed to open mounts (%m)");
79			return NULL;
80			}
81			while((getmntent(f, &mp)) == 0){
82			if ((statvfs(mp.mnt_mountp, &df)) !=0){
83			errf("Failed to gets fs stats (%m)");
84			continue;
85			}
86			if((((strcmp(mp.mnt_fstype,"ufs"))==0) \|\| (strcmp(mp.mnt_fstype,"tmpfs")) ==0)){
87	pajs	1.4	if((sd_ptr=malloc(sizeof(system_disks_t))) == NULL){
88			errf("malloc failed (%m)");
89			return NULL;
90			}
91			if((sd_ptr->disk=malloc(sizeof(disk_stat_t))) == NULL){
92			errf("malloc failed (%m)");
93			return NULL;
94			}
95			if((sd_ptr->disk->device_name=strdup(mp.mnt_special)) == NULL) {
96			errf("strdup failed (%m)");
97			return NULL;
98			}
99			if((sd_ptr->disk->mnt_point=strdup(mp.mnt_mountp)) == NULL){
100			errf("strdup failed (%m)");
101			return NULL;
102			}
103	pajs	1.1	sd_ptr->disk->size=((df.f_frsize/1024) * df.f_blocks);
104			sd_ptr->disk->used=(((df.f_frsize/1024) * df.f_blocks) -((df.f_frsize/1024) * df.f_bfree));
105			sd_ptr->disk->avail=(df.f_frsize/1024) * df.f_bavail;
106			sd_ptr->disk->t_inodes=df.f_files;
107			sd_ptr->disk->f_inodes=df.f_ffree;
108			sd_ptr->next_disk=sd;
109			sd=sd_ptr;
110			}
111
112			}
113
114	pajs	1.6	if((fclose(f))!=0){
115			errf("Failed to close mnttab");
116			return NULL;
117			}
118
119	pajs	1.1	#endif
120	pajs	1.10	#ifdef LINUX
121			if ((f=setmntent("/etc/mtab", "r" ))==NULL){
122	pajs	1.2	errf("Failed to open mounts (%m)");
123			return NULL;
124			}
125
126			while((mp=getmntent(f))){
127			if ((statfs(mp->mnt_dir, &df)) !=0){
128			errf("Failed to gets fs stats (%m)");
129			continue;
130			}
131
132			if((((strcmp(mp->mnt_type, MNTTYPE_NFS))==0) \|\| (strcmp(mp->mnt_type,MNTTYPE_IGNORE)) ==0)) continue;
133	pajs	1.4	if((sd_ptr=malloc(sizeof(system_disks_t))) == NULL){
134			errf("malloc failed (%m)");
135			return NULL;
136			}
137			if((sd_ptr->disk=malloc(sizeof(disk_stat_t))) == NULL){
138			errf("malloc failed (%m)");
139			return NULL;
140			}
141			if((sd_ptr->disk->device_name=strdup(mp->mnt_fsname)) == NULL){
142			errf("strdup failed (%m)");
143			return NULL;
144			}
145			if((sd_ptr->disk->mnt_point=strdup(mp->mnt_dir)) == NULL){
146			errf("strdup failed (%m)");
147			return NULL;
148			}
149	pajs	1.2	sd_ptr->disk->size=((df.f_bsize/1024) * df.f_blocks);
150			sd_ptr->disk->used=((df.f_bsize/1024) * df.f_blocks) -((df.f_bsize/1024) * df.f_bfree);
151			sd_ptr->disk->avail=((df.f_bsize/1024) * df.f_bavail);
152			sd_ptr->disk->t_inodes=df.f_files;
153			sd_ptr->disk->f_inodes=df.f_ffree;
154			sd_ptr->next_disk=sd;
155			sd=sd_ptr;
156			}
157	pajs	1.8
158	pajs	1.12	endmntent(f);
159	pajs	1.8
160	pajs	1.2	#endif
161			#ifdef FREEBSD
162			nummnt=getmntinfo(&mp , MNT_LOCAL);
163			if (nummnt<=0){
164			errf("Failed to get disk stats (%m)");
165			return NULL;
166			}
167
168	pajs	1.4	for(counter=0;counter<nummnt;counter++){
169			if((sd_ptr=malloc(sizeof(system_disks_t))) == NULL){
170			errf("malloc failed (%m)");
171			return NULL;
172			}
173			if((sd_ptr->disk=malloc(sizeof(disk_stat_t))) == NULL){
174			errf("malloc failed (%m)");
175			return NULL;
176			}
177			if((sd_ptr->disk->device_name=strdup(mp->f_mntfromname))==NULL){
178			errf("strdup failed (%m)");
179			return NULL;
180			}
181			if((sd_ptr->disk->mnt_point=strdup(mp->f_mntonname))==NULL){
182			errf("strdup failed (%m)");
183			return NULL;
184			}
185	pajs	1.2	sd_ptr->disk->size=((mp->f_bsize/1024) * mp->f_blocks);
186			sd_ptr->disk->used=((mp->f_bsize/1024) * mp->f_blocks) -((mp->f_bsize/1024) * mp->f_bfree);
187			sd_ptr->disk->avail=((mp->f_bsize/1024) * mp->f_bavail);
188			sd_ptr->disk->t_inodes=mp->f_files;
189			sd_ptr->disk->f_inodes=mp->f_ffree;
190			sd_ptr->next_disk=sd;
191			sd=sd_ptr;
192	pajs	1.3	mp++;
193	pajs	1.2	}
194			#endif
195
196
197	pajs	1.1	sd_ptr=sd;
198	pajs	1.4	xml_disk_stats=strdup("<disk>");
199
200	pajs	1.9	for(counter=0;sd_ptr!=NULL;counter++){
201	pajs	1.5	xml_disk_stats_ptr=xml_disk_stats;
202	pajs	1.4	if((xml_disk_stats=strf("%s<p%d name=\"%s\" mount=\"%s\" kbytes=\"%ld\" used=\"%ld\" avail=\"%ld\" totalinodes=\"%ld\" freeinodes=\"%ld\"></p%d>", xml_disk_stats, counter, sd_ptr->disk->device_name, sd_ptr->disk->mnt_point, sd_ptr->disk->size, sd_ptr->disk->used, sd_ptr->disk->avail, sd_ptr->disk->t_inodes, sd_ptr->disk->f_inodes, counter)) == NULL){
203			errf("strf failed (%m)");
204			return NULL;
205			}
206	pajs	1.5	free(xml_disk_stats_ptr);
207	pajs	1.1	sd_ptr=sd_ptr->next_disk;
208			}
209	pajs	1.6	xml_disk_stats_ptr=xml_disk_stats;
210	pajs	1.4	xml_disk_stats=strf("%s</disk>",xml_disk_stats);
211	pajs	1.6	free(xml_disk_stats_ptr);
212	pajs	1.5
213			/* Cleaning up */
214			sd_ptr=sd;
215			while(sd_ptr!=NULL){
216			sd=sd_ptr->next_disk;
217			free(sd_ptr->disk->device_name);
218			free(sd_ptr->disk->mnt_point);
219			free(sd_ptr->disk);
220			free(sd_ptr);
221			sd_ptr=sd;
222			}
223	pajs	1.4
224			return xml_disk_stats;
225	pajs	1.1	}
226