ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/src/libstatgrab/disk_stats.c
Revision: 1.10
Committed: Tue Mar 4 18:06:16 2003 UTC (21 years, 2 months ago) by pajs
Content type: text/plain
Branch: MAIN
Changes since 1.9: +66 -4 lines
Log Message:
Diskio stats for linux. Not been checked. Relies on tools.c (seperate comit)
If someone knows a nicer way of getting the device name, please tell me, as
the way im doing it is horrible! :)

File Contents

# User Rev Content
1 pajs 1.1 /*
2     * i-scream central monitoring system
3     * http://www.i-scream.org.uk
4     * 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     #ifdef HAVE_CONFIG_H
22     #include "config.h"
23     #endif
24    
25     #include <stdlib.h>
26     #include <string.h>
27 tdb 1.7 #include "statgrab.h"
28 pajs 1.1
29     #ifdef SOLARIS
30 pajs 1.9 #include <stdio.h>
31 pajs 1.1 #include <sys/mnttab.h>
32     #include <sys/statvfs.h>
33 pajs 1.4 #include <kstat.h>
34 pajs 1.1 #define VALID_FS_TYPES {"ufs", "tmpfs"}
35     #endif
36    
37 pajs 1.9 #ifdef LINUX
38     #include <stdio.h>
39     #include <sys/vfs.h>
40     #include <mntent.h>
41 pajs 1.10 #include "tools.h"
42 pajs 1.9 #define VALID_FS_TYPES {"ext2", "ext3", "xfs", "reiserfs", "vfat", "tmpfs"}
43     #endif
44    
45 pajs 1.1 #define START_VAL 1
46    
47     char *copy_string(char *orig_ptr, const char *newtext){
48    
49     /* Maybe free if not NULL, and strdup rather than realloc and strcpy? */
50     orig_ptr=realloc(orig_ptr, (1+strlen(newtext)));
51     if(orig_ptr==NULL){
52     return NULL;
53     }
54     strcpy(orig_ptr, newtext);
55    
56     return orig_ptr;
57     }
58    
59    
60     void init_disk_stat(int start, int end, disk_stat_t *disk_stats){
61    
62     for(disk_stats+=start; start<=end; start++){
63     disk_stats->device_name=NULL;
64     disk_stats->fs_type=NULL;
65     disk_stats->mnt_point=NULL;
66    
67     disk_stats++;
68     }
69     }
70    
71     disk_stat_t *get_disk_stats(int *entries){
72    
73     static disk_stat_t *disk_stats;
74     static int watermark=-1;
75    
76     char *fs_types[] = VALID_FS_TYPES;
77     int x, valid_type;
78     int num_disks=0;
79     FILE *f;
80    
81     disk_stat_t *disk_ptr;
82    
83 pajs 1.9 #ifdef SOLARIS
84     struct mnttab *mp;
85     struct statvfs fs;
86     #endif
87     #ifdef LINUX
88     struct mntent *mp;
89     struct statfs fs;
90     #endif
91    
92 pajs 1.1 if(watermark==-1){
93     disk_stats=malloc(START_VAL * sizeof(disk_stat_t));
94     if(disk_stats==NULL){
95     return NULL;
96     }
97     watermark=START_VAL;
98     init_disk_stat(0, watermark-1, disk_stats);
99     }
100 pajs 1.9 #ifdef LINUX
101     if ((f=setmntent("/etc/mtab", "r" ))==NULL){
102     return NULL;
103     }
104 pajs 1.1
105 pajs 1.9 while((mp=getmntent(f))){
106     if((statfs(mp->mnt_dir, &fs)) !=0){
107     continue;
108     }
109 pajs 1.1
110 pajs 1.9 valid_type=0;
111     for(x=0;x<((sizeof(fs_types))/(sizeof(char*)));x++){
112     if(strcmp(mp->mnt_type, fs_types[x]) ==0){
113     valid_type=1;
114     break;
115     }
116     }
117     #endif
118    
119     #ifdef SOLARIS
120 pajs 1.1 if ((f=fopen("/etc/mnttab", "r" ))==NULL){
121     return NULL;
122     }
123     while((getmntent(f, &mp)) == 0){
124     if ((statvfs(mp.mnt_mountp, &fs)) !=0){
125     continue;
126     }
127     valid_type=0;
128     for(x=0;x<((sizeof(fs_types))/(sizeof(char*)));x++){
129     if(strcmp(mp.mnt_fstype, fs_types[x]) ==0){
130     valid_type=1;
131     break;
132     }
133     }
134 pajs 1.9 #endif
135 pajs 1.1
136     if(valid_type){
137     if(num_disks>watermark-1){
138     disk_ptr=disk_stats;
139     if((disk_stats=realloc(disk_stats, (watermark*2 * sizeof(disk_stat_t))))==NULL){
140     disk_stats=disk_ptr;
141     return NULL;
142     }
143    
144     watermark=watermark*2;
145     init_disk_stat(num_disks, watermark-1, disk_stats);
146     }
147    
148     disk_ptr=disk_stats+num_disks;
149 pajs 1.9 #ifdef LINUX
150     if((disk_ptr->device_name=copy_string(disk_ptr->device_name, mp->mnt_fsname))==NULL){
151     return NULL;
152     }
153    
154     if((disk_ptr->fs_type=copy_string(disk_ptr->fs_type, mp->mnt_type))==NULL){
155     return NULL;
156     }
157    
158     if((disk_ptr->mnt_point=copy_string(disk_ptr->mnt_point, mp->mnt_dir))==NULL){
159     return NULL;
160     }
161     disk_ptr->size = (long long)fs.f_bsize * (long long)fs.f_blocks;
162     disk_ptr->avail = (long long)fs.f_bsize * (long long)fs.f_bavail;
163     disk_ptr->used = (disk_ptr->size) - ((long long)fs.f_bsize * (long long)fs.f_bfree);
164    
165     disk_ptr->total_inodes=(long long)fs.f_files;
166     disk_ptr->free_inodes=(long long)fs.f_ffree;
167     /* Linux doesn't have a "available" inodes */
168     disk_ptr->used_inodes=disk_ptr->total_inodes-disk_ptr->free_inodes;
169     #endif
170    
171     #ifdef SOLARIS
172 pajs 1.1 /* Memory leak in event of realloc failing */
173     /* Maybe make this char[bigenough] and do strncpy's and put a null in the end?
174 pajs 1.9 * Downside is its a bit hungry for a lot of mounts, as MNT_MAX_SIZE would prob
175     * be upwards of a k each
176     */
177 pajs 1.1 if((disk_ptr->device_name=copy_string(disk_ptr->device_name, mp.mnt_special))==NULL){
178     return NULL;
179     }
180    
181     if((disk_ptr->fs_type=copy_string(disk_ptr->fs_type, mp.mnt_fstype))==NULL){
182     return NULL;
183     }
184    
185     if((disk_ptr->mnt_point=copy_string(disk_ptr->mnt_point, mp.mnt_mountp))==NULL){
186     return NULL;
187     }
188    
189     disk_ptr->size = (long long)fs.f_frsize * (long long)fs.f_blocks;
190     disk_ptr->avail = (long long)fs.f_frsize * (long long)fs.f_bavail;
191     disk_ptr->used = (disk_ptr->size) - ((long long)fs.f_frsize * (long long)fs.f_bfree);
192    
193     disk_ptr->total_inodes=(long long)fs.f_files;
194     disk_ptr->used_inodes=disk_ptr->total_inodes - (long long)fs.f_ffree;
195     disk_ptr->free_inodes=(long long)fs.f_favail;
196 pajs 1.9 #endif
197 pajs 1.1 num_disks++;
198     }
199     }
200    
201     *entries=num_disks;
202    
203     /* If this fails, there is very little i can do about it, so i'll ignore it :) */
204     fclose(f);
205    
206     return disk_stats;
207    
208     }
209 pajs 1.4 void diskio_stat_init(int start, int end, diskio_stat_t *diskio_stats){
210    
211     for(diskio_stats+=start; start<end; start++){
212     diskio_stats->disk_name=NULL;
213    
214     diskio_stats++;
215     }
216     }
217    
218     diskio_stat_t *diskio_stat_malloc(int needed_entries, int *cur_entries, diskio_stat_t *diskio_stats){
219    
220     if(diskio_stats==NULL){
221    
222     if((diskio_stats=malloc(needed_entries * sizeof(diskio_stat_t)))==NULL){
223     return NULL;
224     }
225     diskio_stat_init(0, needed_entries, diskio_stats);
226     *cur_entries=needed_entries;
227    
228     return diskio_stats;
229     }
230    
231    
232     if(*cur_entries<needed_entries){
233     diskio_stats=realloc(diskio_stats, (sizeof(diskio_stat_t)*needed_entries));
234     if(diskio_stats==NULL){
235     return NULL;
236     }
237     diskio_stat_init(*cur_entries, needed_entries, diskio_stats);
238     *cur_entries=needed_entries;
239     }
240    
241     return diskio_stats;
242     }
243    
244     static diskio_stat_t *diskio_stats=NULL;
245 pajs 1.5 static int num_diskio=0;
246 pajs 1.4
247 tdb 1.3 diskio_stat_t *get_diskio_stats(int *entries){
248 pajs 1.2
249 pajs 1.4 static int sizeof_diskio_stats=0;
250     diskio_stat_t *diskio_stats_ptr;
251    
252 pajs 1.10 #ifdef SOLARIS
253 pajs 1.4 kstat_ctl_t *kc;
254     kstat_t *ksp;
255     kstat_io_t kios;
256 pajs 1.10 #endif
257     #ifdef LINUX
258     FILE *f;
259     char *line_ptr;
260     int major, minor;
261     char dev_letter;
262     #endif
263    
264     num_diskio=0;
265 pajs 1.4
266 pajs 1.10 #ifdef SOLARIS
267 pajs 1.4 if ((kc = kstat_open()) == NULL) {
268     return NULL;
269     }
270    
271     for (ksp = kc->kc_chain; ksp; ksp = ksp->ks_next) {
272     if (!strcmp(ksp->ks_class, "disk")) {
273    
274     if(ksp->ks_type != KSTAT_TYPE_IO) continue;
275 pajs 1.5 /* We dont want metadevices appearins as num_diskio */
276 pajs 1.4 if(strcmp(ksp->ks_module, "md")==0) continue;
277     if((kstat_read(kc, ksp, &kios))==-1){
278     }
279    
280 pajs 1.5 if((diskio_stats=diskio_stat_malloc(num_diskio+1, &sizeof_diskio_stats, diskio_stats))==NULL){
281 pajs 1.4 kstat_close(kc);
282     return NULL;
283     }
284 pajs 1.5 diskio_stats_ptr=diskio_stats+num_diskio;
285 pajs 1.4
286     diskio_stats_ptr->read_bytes=kios.nread;
287    
288     diskio_stats_ptr->write_bytes=kios.nwritten;
289    
290     if(diskio_stats_ptr->disk_name!=NULL) free(diskio_stats_ptr->disk_name);
291    
292     diskio_stats_ptr->disk_name=strdup(ksp->ks_name);
293 pajs 1.5 num_diskio++;
294 pajs 1.4 }
295     }
296    
297     kstat_close(kc);
298 pajs 1.10 #endif
299 pajs 1.4
300 pajs 1.10 #ifdef LINUX
301     f=fopen("/proc/stat", "r");
302     if(f==NULL){
303     return NULL;
304     }
305     if((line_ptr=f_read_line(f, "disk_io:"))==NULL){
306     return NULL;
307     }
308     while((line_ptr=strchr(line_ptr, ' '))!=NULL){
309     if((diskio_stats=diskio_stat_malloc(num_diskio+1, &sizeof_diskio_stats, diskio_stats))==NULL){
310     fclose(f);
311     return NULL;
312     }
313     diskio_stats_ptr=diskio_stats+num_diskio;
314    
315    
316     sscanf(line_ptr, "(%d,%d):(%*d, %lld, %*d, %lld, %*d)", \
317     &major, \
318     &minor, \
319     &diskio_stats_ptr->read_bytes, \
320     &diskio_stats_ptr->write_bytes);
321    
322     if(diskio_stats_ptr->disk_name!=NULL) free(diskio_stats_ptr->disk_name);
323    
324     switch(major){
325     case 3:
326     if(minor==0){
327     diskio_stats_ptr->disk_name=strdup("hda");
328     }else{
329     diskio_stats_ptr->disk_name=strdup("hdb");
330     }
331     break;
332    
333     case 22:
334     if(minor==0){
335     diskio_stats_ptr->disk_name=strdup("hdc");
336     }else{
337     diskio_stats_ptr->disk_name=strdup("hdd");
338     }
339     case 8:
340     dev_letter='a'+(minor/16);
341     diskio_stats_ptr->disk_name=malloc(4);
342     snprintf(diskio_stats_ptr->disk_name, 4, "sd%c", dev_letter);
343     default:
344     /* I have no idea what it is then :) */
345     diskio_stats_ptr->disk_name=malloc(16);
346     snprintf(diskio_stats_ptr->disk_name, 16, "%d %d", major, minor);
347     }
348    
349     num_diskio++;
350     }
351    
352     #endif
353 pajs 1.5 *entries=num_diskio;
354 pajs 1.4
355     return diskio_stats;
356 pajs 1.5 }
357    
358 pajs 1.6 diskio_stat_t *get_diskio_stats_diff(int *entries){
359 pajs 1.5 static diskio_stat_t *diskio_stats_diff=NULL;
360     static int sizeof_diskio_stats_diff=0;
361     diskio_stat_t *diskio_stats_diff_ptr, *diskio_stats_ptr;
362     int disks, x, y;
363    
364     if(diskio_stats==NULL){
365     diskio_stats_ptr=get_diskio_stats(&disks);
366     *entries=disks;
367     return diskio_stats_ptr;
368     }
369    
370     diskio_stats_diff=diskio_stat_malloc(num_diskio, &sizeof_diskio_stats_diff, diskio_stats_diff);
371     if(diskio_stats_diff==NULL){
372     return NULL;
373     }
374    
375     diskio_stats_diff_ptr=diskio_stats_diff;
376     diskio_stats_ptr=diskio_stats;
377    
378     for(disks=0;disks<num_diskio;disks++){
379     if(diskio_stats_diff_ptr->disk_name!=NULL){
380     free(diskio_stats_diff_ptr->disk_name);
381     }
382     diskio_stats_diff_ptr->disk_name=strdup(diskio_stats_ptr->disk_name);
383     diskio_stats_diff_ptr->read_bytes=diskio_stats_ptr->read_bytes;
384     diskio_stats_diff_ptr->write_bytes=diskio_stats_ptr->write_bytes;
385     diskio_stats_diff_ptr->systime=diskio_stats_ptr->systime;
386    
387     diskio_stats_diff_ptr++;
388     diskio_stats_ptr++;
389     }
390    
391     diskio_stats_ptr=get_diskio_stats(&disks);
392     diskio_stats_diff_ptr=diskio_stats_diff;
393    
394     for(x=0;x<sizeof_diskio_stats_diff;x++){
395    
396     if((strcmp(diskio_stats_diff_ptr->disk_name, diskio_stats_ptr->disk_name))==0){
397     diskio_stats_diff_ptr->read_bytes=diskio_stats_ptr->read_bytes-diskio_stats_diff_ptr->read_bytes;
398     diskio_stats_diff_ptr->write_bytes=diskio_stats_ptr->write_bytes-diskio_stats_diff_ptr->write_bytes;
399     diskio_stats_diff_ptr->systime=diskio_stats_ptr->systime-diskio_stats_diff_ptr->systime;
400     }else{
401     diskio_stats_ptr=diskio_stats;
402     for(y=0;y<disks;y++){
403     if((strcmp(diskio_stats_diff_ptr->disk_name, diskio_stats_ptr->disk_name))==0){
404     diskio_stats_diff_ptr->read_bytes=diskio_stats_ptr->read_bytes-diskio_stats_diff_ptr->read_bytes;
405     diskio_stats_diff_ptr->write_bytes=diskio_stats_ptr->write_bytes-diskio_stats_diff_ptr->write_bytes;
406     diskio_stats_diff_ptr->systime=diskio_stats_ptr->systime-diskio_stats_diff_ptr->systime;
407    
408     break;
409     }
410    
411     diskio_stats_ptr++;
412     }
413     }
414    
415     diskio_stats_ptr++;
416     diskio_stats_diff_ptr++;
417    
418     }
419 pajs 1.8
420     *entries=sizeof_diskio_stats_diff;
421 pajs 1.5 return diskio_stats_diff;
422 pajs 1.2 }