ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/src/libstatgrab/disk_stats.c
(Generate patch)

Comparing projects/libstatgrab/src/libstatgrab/disk_stats.c (file contents):
Revision 1.39 by tdb, Mon Nov 10 21:07:04 2003 UTC vs.
Revision 1.59 by ats, Sun Apr 4 23:55:48 2004 UTC

# Line 1 | Line 1
1 < /*
1 > /*
2   * i-scream central monitoring system
3   * http://www.i-scream.org
4 < * Copyright (C) 2000-2003 i-scream
4 > * Copyright (C) 2000-2004 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.
6 > * 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   *
11 < * This program is distributed in the hope that it will be useful,
11 > * This library 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.
13 > * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 > * Lesser 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.
16 > * 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 > *
21 > * $Id$
22   */
23  
24   #ifdef HAVE_CONFIG_H
# Line 25 | Line 28
28   #include <stdio.h>
29   #include <stdlib.h>
30   #include <string.h>
31 + #include <time.h>
32   #include "statgrab.h"
33 + #include "vector.h"
34  
35   #ifdef SOLARIS
36   #include <sys/mnttab.h>
# Line 34 | Line 39
39   #define VALID_FS_TYPES {"ufs", "tmpfs"}
40   #endif
41  
42 < #ifdef LINUX
38 < #include <time.h>
39 < #include <sys/vfs.h>
42 > #if defined(LINUX) || defined(CYGWIN)
43   #include <mntent.h>
44 + #include <sys/vfs.h>
45   #include "tools.h"
46 < #ifdef CYGWIN
47 < #define VALID_FS_TYPES {"user"}
48 < #else
46 > #endif
47 >
48 > #ifdef LINUX
49   #define VALID_FS_TYPES {"adfs", "affs", "befs", "bfs", "efs", "ext2", \
50                          "ext3", "vxfs", "hfs", "hfsplus", "hpfs", "jffs", \
51                          "jffs2", "minix", "msdos", "ntfs", "qnx4", "ramfs", \
52                          "rootfs", "reiserfs", "sysv", "v7", "udf", "ufs", \
53                          "umsdos", "vfat", "xfs", "jfs"}
54   #endif
55 +
56 + #ifdef CYGWIN
57 + #define VALID_FS_TYPES {"user"}
58   #endif
59  
60   #ifdef ALLBSD
# Line 55 | Line 62
62   #include <sys/ucred.h>
63   #include <sys/mount.h>
64   #endif
65 < #ifdef FREEBSD
65 > #if defined(FREEBSD) || defined(DFBSD)
66   #include <sys/dkstat.h>
67   #include <devstat.h>
68   #define VALID_FS_TYPES {"hpfs", "msdosfs", "ntfs", "udf", "ext2fs", \
69                          "ufs", "mfs"}
70   #endif
71 < #ifdef NETBSD
71 > #if defined(NETBSD) || defined(OPENBSD)
72   #include <sys/param.h>
73   #include <sys/sysctl.h>
74   #include <sys/disk.h>
# Line 69 | Line 76
76                          "ntfs"}
77   #endif
78  
79 < #define START_VAL 1
79 > static void disk_stat_init(disk_stat_t *d) {
80 >        d->device_name = NULL;
81 >        d->fs_type = NULL;
82 >        d->mnt_point = NULL;
83 > }
84  
85 < char *copy_string(char *orig_ptr, const char *newtext){
86 <
87 <        /* Maybe free if not NULL, and strdup rather than realloc and strcpy? */
88 <        orig_ptr=realloc(orig_ptr, (1+strlen(newtext)));
78 <        if(orig_ptr==NULL){
79 <                return NULL;
80 <        }
81 <        strcpy(orig_ptr, newtext);
82 <
83 <        return orig_ptr;
85 > static void disk_stat_destroy(disk_stat_t *d) {
86 >        free(d->device_name);
87 >        free(d->fs_type);
88 >        free(d->mnt_point);
89   }
90  
91 + int is_valid_fs_type(const char *type) {
92 +        const char *types[] = VALID_FS_TYPES;
93 +        int i;
94  
95 < void init_disk_stat(int start, int end, disk_stat_t *disk_stats){
96 <
97 <        for(disk_stats+=start; start<=end; start++){
98 <                disk_stats->device_name=NULL;
91 <                disk_stats->fs_type=NULL;
92 <                disk_stats->mnt_point=NULL;
93 <                
94 <                disk_stats++;
95 >        for (i = 0; i < (sizeof types / sizeof *types); i++) {
96 >                if (strcmp(types[i], type) == 0) {
97 >                        return 1;
98 >                }
99          }
100 +        return 0;
101   }
102  
103   disk_stat_t *get_disk_stats(int *entries){
104 +        VECTOR_DECLARE_STATIC(disk_stats, disk_stat_t, 10,
105 +                              disk_stat_init, disk_stat_destroy);
106  
107 <        static disk_stat_t *disk_stats;
101 <        static int watermark=-1;
102 <
103 <        char *fs_types[] = VALID_FS_TYPES;
104 <        int x, valid_type;
107 >        int valid_type;
108          int num_disks=0;
109 < #if defined(LINUX) || defined (SOLARIS)
109 > #if defined(LINUX) || defined (SOLARIS) || defined(CYGWIN)
110          FILE *f;
111   #endif
112  
# Line 113 | Line 116 | disk_stat_t *get_disk_stats(int *entries){
116          struct mnttab mp;
117          struct statvfs fs;
118   #endif
119 < #ifdef LINUX
119 > #if defined(LINUX) || defined(CYGWIN)
120          struct mntent *mp;
121          struct statfs fs;
122   #endif
# Line 122 | Line 125 | disk_stat_t *get_disk_stats(int *entries){
125          struct statfs *mp;
126   #endif
127  
125        if(watermark==-1){
126                disk_stats=malloc(START_VAL * sizeof(disk_stat_t));
127                if(disk_stats==NULL){
128                        return NULL;
129                }
130                watermark=START_VAL;
131                init_disk_stat(0, watermark-1, disk_stats);
132        }
128   #ifdef ALLBSD
129          nummnt=getmntinfo(&mp , MNT_LOCAL);
130          if (nummnt<=0){
131                  return NULL;
132          }
133          for(;nummnt--; mp++){
134 <                valid_type=0;
140 <                for(x=0;x<((sizeof(fs_types))/(sizeof(char*)));x++){
141 <                        if(strcmp(mp->f_fstypename, fs_types[x]) ==0){
142 <                                valid_type=1;
143 <                                break;
144 <                        }
145 <                }
134 >                valid_type = is_valid_fs_type(mp->f_fstypename);
135   #endif
136  
137 < #ifdef LINUX
137 > #if defined(LINUX) || defined(CYGWIN)
138          if ((f=setmntent("/etc/mtab", "r" ))==NULL){
139                  return NULL;
140          }
# Line 155 | Line 144 | disk_stat_t *get_disk_stats(int *entries){
144                          continue;
145                  }      
146  
147 <                valid_type=0;
159 <                for(x=0;x<((sizeof(fs_types))/(sizeof(char*)));x++){
160 <                        if(strcmp(mp->mnt_type, fs_types[x]) ==0){
161 <                                valid_type=1;
162 <                                break;
163 <                        }
164 <                }
147 >                valid_type = is_valid_fs_type(mp->mnt_type);
148   #endif
149  
150   #ifdef SOLARIS
# Line 172 | Line 155 | disk_stat_t *get_disk_stats(int *entries){
155                  if ((statvfs(mp.mnt_mountp, &fs)) !=0){
156                          continue;
157                  }
158 <                valid_type=0;
176 <                for(x=0;x<((sizeof(fs_types))/(sizeof(char*)));x++){
177 <                        if(strcmp(mp.mnt_fstype, fs_types[x]) ==0){
178 <                                valid_type=1;
179 <                                break;
180 <                        }
181 <                }
158 >                valid_type = is_valid_fs_type(mp.mnt_fstype);
159   #endif
160  
161                  if(valid_type){
162 <                        if(num_disks>watermark-1){
163 <                                disk_ptr=disk_stats;
187 <                                if((disk_stats=realloc(disk_stats, (watermark*2 * sizeof(disk_stat_t))))==NULL){
188 <                                        disk_stats=disk_ptr;
189 <                                        return NULL;
190 <                                }
191 <
192 <                                watermark=watermark*2;
193 <                                init_disk_stat(num_disks, watermark-1, disk_stats);
162 >                        if (VECTOR_RESIZE(disk_stats, num_disks + 1) < 0) {
163 >                                return NULL;
164                          }
195
165                          disk_ptr=disk_stats+num_disks;
166 +
167   #ifdef ALLBSD
168 <                        if((disk_ptr->device_name=copy_string(disk_ptr->device_name, mp->f_mntfromname))==NULL){
168 >                        if (update_string(&disk_ptr->device_name, mp->f_mntfromname) == NULL) {
169                                  return NULL;
170                          }
171 <
202 <                        if((disk_ptr->fs_type=copy_string(disk_ptr->fs_type, mp->f_fstypename))==NULL){
171 >                        if (update_string(&disk_ptr->fs_type, mp->f_fstypename) == NULL) {
172                                  return NULL;
173                          }
174 <
206 <                        if((disk_ptr->mnt_point=copy_string(disk_ptr->mnt_point, mp->f_mntonname))==NULL){
174 >                        if (update_string(&disk_ptr->mnt_point, mp->f_mntonname) == NULL) {
175                                  return NULL;
176                          }
177  
# Line 216 | Line 184 | disk_stat_t *get_disk_stats(int *entries){
184                          /* Freebsd doesn't have a "available" inodes */
185                          disk_ptr->used_inodes=disk_ptr->total_inodes-disk_ptr->free_inodes;
186   #endif
187 < #ifdef LINUX
188 <                        if((disk_ptr->device_name=copy_string(disk_ptr->device_name, mp->mnt_fsname))==NULL){
187 > #if defined(LINUX) || defined(CYGWIN)
188 >                        if (update_string(&disk_ptr->device_name, mp->mnt_fsname) == NULL) {
189                                  return NULL;
190                          }
191                                  
192 <                        if((disk_ptr->fs_type=copy_string(disk_ptr->fs_type, mp->mnt_type))==NULL){    
192 >                        if (update_string(&disk_ptr->fs_type, mp->mnt_type) == NULL) {  
193                                  return NULL;
194                          }
195  
196 <                        if((disk_ptr->mnt_point=copy_string(disk_ptr->mnt_point, mp->mnt_dir))==NULL){
196 >                        if (update_string(&disk_ptr->mnt_point, mp->mnt_dir) == NULL) {
197                                  return NULL;
198                          }
199                          disk_ptr->size = (long long)fs.f_bsize * (long long)fs.f_blocks;
# Line 239 | Line 207 | disk_stat_t *get_disk_stats(int *entries){
207   #endif
208  
209   #ifdef SOLARIS
242                        /* Memory leak in event of realloc failing */
210                          /* Maybe make this char[bigenough] and do strncpy's and put a null in the end?
211                           * Downside is its a bit hungry for a lot of mounts, as MNT_MAX_SIZE would prob
212                           * be upwards of a k each
213                           */
214 <                        if((disk_ptr->device_name=copy_string(disk_ptr->device_name, mp.mnt_special))==NULL){
214 >                        if (update_string(&disk_ptr->device_name, mp.mnt_special) == NULL) {
215                                  return NULL;
216                          }
217  
218 <                        if((disk_ptr->fs_type=copy_string(disk_ptr->fs_type, mp.mnt_fstype))==NULL){
218 >                        if (update_string(&disk_ptr->fs_type, mp.mnt_fstype) == NULL) {
219                                  return NULL;
220                          }
221          
222 <                        if((disk_ptr->mnt_point=copy_string(disk_ptr->mnt_point, mp.mnt_mountp))==NULL){
222 >                        if (update_string(&disk_ptr->mnt_point, mp.mnt_mountp) == NULL) {
223                                  return NULL;
224                          }
225                          
# Line 272 | Line 239 | disk_stat_t *get_disk_stats(int *entries){
239  
240          /* If this fails, there is very little i can do about it, so
241             I'll ignore it :) */
242 < #if defined(LINUX)
242 > #if defined(LINUX) || defined(CYGWIN)
243          endmntent(f);
244   #endif
245   #if defined(SOLARIS)
# Line 282 | Line 249 | disk_stat_t *get_disk_stats(int *entries){
249          return disk_stats;
250  
251   }
285 void diskio_stat_init(int start, int end, diskio_stat_t *diskio_stats){
252  
253 <        for(diskio_stats+=start; start<end; start++){
254 <                diskio_stats->disk_name=NULL;
289 <                
290 <                diskio_stats++;
291 <        }
253 > static void diskio_stat_init(diskio_stat_t *d) {
254 >        d->disk_name = NULL;
255   }
256  
257 < diskio_stat_t *diskio_stat_malloc(int needed_entries, int *cur_entries, diskio_stat_t *diskio_stats){
258 <
296 <        if(diskio_stats==NULL){
297 <
298 <                if((diskio_stats=malloc(needed_entries * sizeof(diskio_stat_t)))==NULL){
299 <                        return NULL;
300 <                }
301 <                diskio_stat_init(0, needed_entries, diskio_stats);
302 <                *cur_entries=needed_entries;
303 <
304 <                return diskio_stats;
305 <        }
306 <
307 <
308 <        if(*cur_entries<needed_entries){
309 <                diskio_stats=realloc(diskio_stats, (sizeof(diskio_stat_t)*needed_entries));
310 <                if(diskio_stats==NULL){
311 <                        return NULL;
312 <                }
313 <                diskio_stat_init(*cur_entries, needed_entries, diskio_stats);
314 <                *cur_entries=needed_entries;
315 <        }
316 <
317 <        return diskio_stats;
257 > static void diskio_stat_destroy(diskio_stat_t *d) {
258 >        free(d->disk_name);
259   }
260  
261 < static diskio_stat_t *diskio_stats=NULL;        
262 < static int num_diskio=0;        
261 > VECTOR_DECLARE_STATIC(diskio_stats, diskio_stat_t, 10,
262 >                      diskio_stat_init, diskio_stat_destroy);
263  
264   #ifdef LINUX
265   typedef struct {
# Line 328 | Line 269 | typedef struct {
269   #endif
270  
271   diskio_stat_t *get_diskio_stats(int *entries){
272 <
332 <        static int sizeof_diskio_stats=0;
272 >        int num_diskio;
273   #ifndef LINUX
274          diskio_stat_t *diskio_stats_ptr;
275   #endif
# Line 344 | Line 284 | diskio_stat_t *get_diskio_stats(int *entries){
284          char *line_ptr;
285          int major, minor;
286          int has_pp_stats = 1;
287 <        static partition *parts = NULL;
348 <        static int alloc_parts = 0;
287 >        VECTOR_DECLARE_STATIC(parts, partition, 16, NULL, NULL);
288          int i, n;
289          time_t now;
290          const char *format;
291   #endif
292 < #ifdef FREEBSD
292 > #if defined(FREEBSD) || defined(DFBSD)
293          static struct statinfo stats;
294          static int stats_init = 0;
295          int counter;
# Line 361 | Line 300 | diskio_stat_t *get_diskio_stats(int *entries){
300   #endif
301   #ifdef NETBSD
302          struct disk_sysctl *stats;
303 + #endif
304 + #ifdef OPENBSD
305 +        int diskcount;
306 +        char *disknames, *name, *bufpp;
307 +        char **dk_name;
308 +        struct diskstats *stats;
309 + #endif
310 + #ifdef NETBSD
311 + #define MIBSIZE 3
312 + #endif
313 + #ifdef OPENBSD
314 + #define MIBSIZE 2
315 + #endif
316 + #if defined(NETBSD) || defined(OPENBSD)
317          int num_disks, i;
318 <        int mib[3];
318 >        int mib[MIBSIZE];
319          size_t size;
320   #endif
321  
322          num_diskio=0;
323  
324 < #ifdef NETBSD
324 > #ifdef OPENBSD
325          mib[0] = CTL_HW;
326 +        mib[1] = HW_DISKCOUNT;
327 +
328 +        size = sizeof(diskcount);
329 +        if (sysctl(mib, MIBSIZE, &diskcount, &size, NULL, 0) < 0) {
330 +                return NULL;
331 +        }
332 +
333 +        mib[0] = CTL_HW;
334 +        mib[1] = HW_DISKNAMES;
335 +
336 +        if (sysctl(mib, MIBSIZE, NULL, &size, NULL, 0) < 0) {
337 +                return NULL;
338 +        }
339 +
340 +        disknames = malloc(size);
341 +        if (disknames == NULL) {
342 +                return NULL;
343 +        }
344 +
345 +        if (sysctl(mib, MIBSIZE, disknames, &size, NULL, 0) < 0) {
346 +                return NULL;
347 +        }
348 +
349 +        dk_name = calloc(diskcount, sizeof(char *));
350 +        bufpp = disknames;
351 +        for (i = 0; i < diskcount && (name = strsep(&bufpp, ",")) != NULL; i++) {
352 +                dk_name[i] = name;
353 +        }
354 + #endif
355 +
356 + #if defined(NETBSD) || defined(OPENBSD)
357 +        mib[0] = CTL_HW;
358          mib[1] = HW_DISKSTATS;
359 + #ifdef NETBSD
360          mib[2] = sizeof(struct disk_sysctl);
361 + #endif
362  
363 <        if (sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
363 >        if (sysctl(mib, MIBSIZE, NULL, &size, NULL, 0) < 0) {
364                  return NULL;
365          }
366 +
367 + #ifdef NETBSD
368          num_disks = size / sizeof(struct disk_sysctl);
369 + #else
370 +        num_disks = size / sizeof(struct diskstats);
371 + #endif
372  
373          stats = malloc(size);
374          if (stats == NULL) {
375                  return NULL;
376          }
377  
378 <        if (sysctl(mib, 3, stats, &size, NULL, 0) < 0) {
378 >        if (sysctl(mib, MIBSIZE, stats, &size, NULL, 0) < 0) {
379                  return NULL;
380          }
381  
382          for (i = 0; i < num_disks; i++) {
383                  u_int64_t rbytes, wbytes;
384  
385 + #ifdef NETBSD
386   #ifdef HAVE_DK_RBYTES
387                  rbytes = stats[i].dk_rbytes;
388                  wbytes = stats[i].dk_wbytes;
# Line 397 | Line 390 | diskio_stat_t *get_diskio_stats(int *entries){
390                  /* Before 1.7, NetBSD merged reads and writes. */
391                  rbytes = wbytes = stats[i].dk_bytes;
392   #endif
393 + #else
394 +                rbytes = wbytes = stats[i].ds_bytes;
395 + #endif
396  
397                  /* Don't keep stats for disks that have never been used. */
398                  if (rbytes == 0 && wbytes == 0) {
399                          continue;
400                  }
401  
402 <                diskio_stats = diskio_stat_malloc(num_diskio + 1,
407 <                                                  &sizeof_diskio_stats,
408 <                                                  diskio_stats);
409 <                if (diskio_stats == NULL) {
402 >                if (VECTOR_RESIZE(diskio_stats, num_diskio + 1) < 0) {
403                          return NULL;
404                  }
405                  diskio_stats_ptr = diskio_stats + num_diskio;
# Line 416 | Line 409 | diskio_stat_t *get_diskio_stats(int *entries){
409                  if (diskio_stats_ptr->disk_name != NULL) {
410                          free(diskio_stats_ptr->disk_name);
411                  }
412 + #ifdef NETBSD
413                  diskio_stats_ptr->disk_name = strdup(stats[i].dk_name);
414 + #else
415 +                diskio_stats_ptr->disk_name = strdup(dk_name[i]);
416 + #endif
417                  diskio_stats_ptr->systime = time(NULL);
418          
419                  num_diskio++;  
420          }
421  
422          free(stats);
423 + #ifdef OPENBSD
424 +        free(dk_name);
425 +        free(disknames);
426   #endif
427 + #endif
428  
429 < #ifdef FREEBSD
429 > #if defined(FREEBSD) || defined(DFBSD)
430          if (!stats_init) {
431                  stats.dinfo=malloc(sizeof(struct devinfo));
432                  if(stats.dinfo==NULL) return NULL;
# Line 460 | Line 461 | diskio_stat_t *get_diskio_stats(int *entries){
461   #else
462                  if((dev_ptr->bytes_read==0) && (dev_ptr->bytes_written==0)) continue;
463   #endif
464 <                if((diskio_stats=diskio_stat_malloc(num_diskio+1, &sizeof_diskio_stats, diskio_stats))==NULL){
464 >
465 >                if (VECTOR_RESIZE(diskio_stats, num_diskio + 1) < 0) {
466                          return NULL;
467                  }
468                  diskio_stats_ptr=diskio_stats+num_diskio;
# Line 495 | Line 497 | diskio_stat_t *get_diskio_stats(int *entries){
497                          if((kstat_read(kc, ksp, &kios))==-1){  
498                          }
499                          
500 <                        if((diskio_stats=diskio_stat_malloc(num_diskio+1, &sizeof_diskio_stats, diskio_stats))==NULL){
500 >                        if (VECTOR_RESIZE(diskio_stats, num_diskio + 1) < 0) {
501                                  kstat_close(kc);
502                                  return NULL;
503                          }
# Line 507 | Line 509 | diskio_stat_t *get_diskio_stats(int *entries){
509  
510                          if(diskio_stats_ptr->disk_name!=NULL) free(diskio_stats_ptr->disk_name);
511  
512 <                        diskio_stats_ptr->disk_name=strdup(ksp->ks_name);
512 >                        diskio_stats_ptr->disk_name=strdup((char *) get_svr_from_bsd(ksp->ks_name));
513                          diskio_stats_ptr->systime=time(NULL);
514                          num_diskio++;
515                  }
# Line 526 | Line 528 | diskio_stat_t *get_diskio_stats(int *entries){
528             the same format. */
529  
530          f = fopen("/proc/diskstats", "r");
531 <        format = " %d %d %19s %*d %*d %lld %*d %*d %*d %lld";
531 >        format = " %d %d %99s %*d %*d %lld %*d %*d %*d %lld";
532          if (f == NULL) {
533                  f = fopen("/proc/partitions", "r");
534 <                format = " %d %d %*d %19s %*d %*d %lld %*d %*d %*d %lld";
534 >                format = " %d %d %*d %99s %*d %*d %lld %*d %*d %*d %lld";
535          }
536          if (f == NULL) goto out;
537          now = time(NULL);
538  
539          while ((line_ptr = f_read_line(f, "")) != NULL) {
540 <                char name[20];
540 >                char name[100];
541                  char *s;
542                  long long rsect, wsect;
543  
# Line 556 | Line 558 | diskio_stat_t *get_diskio_stats(int *entries){
558                          wsect = 0;
559                  }
560  
561 <                diskio_stats = diskio_stat_malloc(n + 1, &sizeof_diskio_stats,
562 <                        diskio_stats);
561 <                if (diskio_stats == NULL) goto out;
562 <                if (n >= alloc_parts) {
563 <                        alloc_parts += 16;
564 <                        parts = realloc(parts, alloc_parts * sizeof *parts);
565 <                        if (parts == NULL) {
566 <                                alloc_parts = 0;
567 <                                goto out;
568 <                        }
561 >                if (VECTOR_RESIZE(diskio_stats, n + 1) < 0) {
562 >                        goto out;
563                  }
564 +                if (VECTOR_RESIZE(parts, n + 1) < 0) {
565 +                        goto out;
566 +                }
567  
568                  if (diskio_stats[n].disk_name != NULL)
569                          free(diskio_stats[n].disk_name);
# Line 580 | Line 577 | diskio_stat_t *get_diskio_stats(int *entries){
577                  n++;
578          }
579  
580 +        fclose(f);
581 +        f = NULL;
582 +
583          if (!has_pp_stats) {
584 <                /* This is an older kernel without stats in /proc/partitions.
585 <                   Read what we can from /proc/stat instead. */
584 >                /* This is an older kernel where /proc/partitions doesn't
585 >                   contain stats. Read what we can from /proc/stat instead, and
586 >                   fill in the appropriate bits of the list allocated above. */
587  
588                  f = fopen("/proc/stat", "r");
589                  if (f == NULL) goto out;
# Line 649 | Line 650 | diskio_stat_t *get_diskio_stats(int *entries){
650          num_diskio = n;
651   out:
652          if (f != NULL) fclose(f);
653 + #endif
654  
655 + #ifdef CYGWIN
656 +        return NULL;
657   #endif
658 +
659          *entries=num_diskio;
660  
661          return diskio_stats;
662   }
663  
664   diskio_stat_t *get_diskio_stats_diff(int *entries){
665 <        static diskio_stat_t *diskio_stats_diff=NULL;
666 <        static int sizeof_diskio_stats_diff=0;
667 <        diskio_stat_t *diskio_stats_diff_ptr, *diskio_stats_ptr;
668 <        int disks, x, y;
665 >        VECTOR_DECLARE_STATIC(diff, diskio_stat_t, 1,
666 >                              diskio_stat_init, diskio_stat_destroy);
667 >        diskio_stat_t *src = NULL, *dest;
668 >        int i, j, diff_count, new_count;
669  
670 <        if(diskio_stats==NULL){
671 <                diskio_stats_ptr=get_diskio_stats(&disks);
672 <                *entries=disks;
668 <                return diskio_stats_ptr;
670 >        if (diskio_stats == NULL) {
671 >                /* No previous stats, so we can't calculate a difference. */
672 >                return get_diskio_stats(entries);
673          }
674  
675 <        diskio_stats_diff=diskio_stat_malloc(num_diskio, &sizeof_diskio_stats_diff, diskio_stats_diff);
676 <        if(diskio_stats_diff==NULL){
675 >        /* Resize the results array to match the previous stats. */
676 >        diff_count = VECTOR_SIZE(diskio_stats);
677 >        if (VECTOR_RESIZE(diff, diff_count) < 0) {
678                  return NULL;
679          }
680  
681 <        diskio_stats_diff_ptr=diskio_stats_diff;
682 <        diskio_stats_ptr=diskio_stats;
681 >        /* Copy the previous stats into the result. */
682 >        for (i = 0; i < diff_count; i++) {
683 >                src = &diskio_stats[i];
684 >                dest = &diff[i];
685  
686 <        for(disks=0;disks<num_diskio;disks++){
687 <                if(diskio_stats_diff_ptr->disk_name!=NULL){
681 <                        free(diskio_stats_diff_ptr->disk_name);
686 >                if (dest->disk_name != NULL) {
687 >                        free(dest->disk_name);
688                  }
689 <                diskio_stats_diff_ptr->disk_name=strdup(diskio_stats_ptr->disk_name);
690 <                diskio_stats_diff_ptr->read_bytes=diskio_stats_ptr->read_bytes;
691 <                diskio_stats_diff_ptr->write_bytes=diskio_stats_ptr->write_bytes;
692 <                diskio_stats_diff_ptr->systime=diskio_stats_ptr->systime;
689 >                dest->disk_name = strdup(src->disk_name);
690 >                dest->read_bytes = src->read_bytes;
691 >                dest->write_bytes = src->write_bytes;
692 >                dest->systime = src->systime;
693 >        }
694  
695 <                diskio_stats_diff_ptr++;
696 <                diskio_stats_ptr++;
695 >        /* Get a new set of stats. */
696 >        if (get_diskio_stats(&new_count) == NULL) {
697 >                return NULL;
698          }
699  
700 <        diskio_stats_ptr=get_diskio_stats(&disks);
701 <        diskio_stats_diff_ptr=diskio_stats_diff;
700 >        /* For each previous stat... */
701 >        for (i = 0; i < diff_count; i++) {
702 >                dest = &diff[i];
703  
704 <        for(x=0;x<sizeof_diskio_stats_diff;x++){
705 <
706 <                if((strcmp(diskio_stats_diff_ptr->disk_name, diskio_stats_ptr->disk_name))==0){
707 <                        diskio_stats_diff_ptr->read_bytes=diskio_stats_ptr->read_bytes-diskio_stats_diff_ptr->read_bytes;
708 <                        diskio_stats_diff_ptr->write_bytes=diskio_stats_ptr->write_bytes-diskio_stats_diff_ptr->write_bytes;
709 <                        diskio_stats_diff_ptr->systime=diskio_stats_ptr->systime-diskio_stats_diff_ptr->systime;
710 <                }else{
702 <                        diskio_stats_ptr=diskio_stats;
703 <                        for(y=0;y<disks;y++){
704 <                                if((strcmp(diskio_stats_diff_ptr->disk_name, diskio_stats_ptr->disk_name))==0){
705 <                                        diskio_stats_diff_ptr->read_bytes=diskio_stats_ptr->read_bytes-diskio_stats_diff_ptr->read_bytes;
706 <                                        diskio_stats_diff_ptr->write_bytes=diskio_stats_ptr->write_bytes-diskio_stats_diff_ptr->write_bytes;
707 <                                        diskio_stats_diff_ptr->systime=diskio_stats_ptr->systime-diskio_stats_diff_ptr->systime;
708 <
709 <                                        break;
710 <                                }
711 <                                
712 <                                diskio_stats_ptr++;
704 >                /* ... find the corresponding new stat ... */
705 >                for (j = 0; j < new_count; j++) {
706 >                        /* Try the new stat in the same position first,
707 >                           since that's most likely to be it. */
708 >                        src = &diskio_stats[(i + j) % new_count];
709 >                        if (strcmp(src->disk_name, dest->disk_name) == 0) {
710 >                                break;
711                          }
712                  }
713 +                if (j == new_count) {
714 +                        /* No match found. */
715 +                        continue;
716 +                }
717  
718 <                diskio_stats_ptr++;
719 <                diskio_stats_diff_ptr++;        
720 <
718 >                /* ... and subtract the previous stat from it to get the
719 >                   difference. */
720 >                dest->read_bytes = src->read_bytes - dest->read_bytes;
721 >                dest->write_bytes = src->write_bytes - dest->write_bytes;
722 >                dest->systime = src->systime - dest->systime;
723          }
724 <        
725 <        *entries=sizeof_diskio_stats_diff;
726 <        return diskio_stats_diff;
724 >
725 >        *entries = diff_count;
726 >        return diff;
727   }
728 +

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines