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.25 by tdb, Tue Oct 7 20:32:50 2003 UTC vs.
Revision 1.31 by ats, Sun Oct 19 02:03:02 2003 UTC

# Line 31 | Line 31
31   #include <sys/mnttab.h>
32   #include <sys/statvfs.h>
33   #include <kstat.h>
34 #include <sys/types.h>
35 #include <dirent.h>
36 #include <limits.h>
37 #include <unistd.h>
34   #define VALID_FS_TYPES {"ufs", "tmpfs"}
35   #endif
36  
37   #ifdef LINUX
38 + #include <time.h>
39   #include <sys/vfs.h>
40   #include <mntent.h>
41   #include "tools.h"
42   #define VALID_FS_TYPES {"ext2", "ext3", "xfs", "reiserfs", "vfat", "tmpfs"}
43   #endif
44  
45 < #ifdef FREEBSD
45 > #ifdef ALLBSD
46   #include <sys/param.h>
47   #include <sys/ucred.h>
48   #include <sys/mount.h>
49 + #define VALID_FS_TYPES {"ufs", "mfs", "ffs"}
50 + #endif
51 + #ifdef FREEBSD
52   #include <sys/dkstat.h>
53   #include <devstat.h>
54 #define VALID_FS_TYPES {"ufs", "mfs"}
54   #endif
55 +
56   #define START_VAL 1
57  
58 /* Solaris kernel stores the disk names in the old BSD format
59 * and we would prefer it in th modern format.
60 */
61 #ifdef SOLARIS
62 /* Lines from the path_to_inst file */
63 typedef struct {
64        /* unlikely to get a line that long */
65        char line[256];
66 }file_line_t;
67
68 /* Linked list of the mappings from the older bsd-based sunos names
69 * to the modern solaris names
70 */
71 struct disk_mapping_ent_t{
72        char *sunos_name;
73        char *solaris_name;
74        struct disk_mapping_ent_t *next;
75 };
76 typedef struct disk_mapping_ent_t disk_mapping_t;
77 #endif
78
58   char *copy_string(char *orig_ptr, const char *newtext){
59  
60          /* Maybe free if not NULL, and strdup rather than realloc and strcpy? */
# Line 122 | Line 101 | disk_stat_t *get_disk_stats(int *entries){
101          struct mntent *mp;
102          struct statfs fs;
103   #endif
104 < #ifdef FREEBSD
104 > #ifdef ALLBSD
105          int nummnt;
106          struct statfs *mp;
107   #endif
# Line 135 | Line 114 | disk_stat_t *get_disk_stats(int *entries){
114                  watermark=START_VAL;
115                  init_disk_stat(0, watermark-1, disk_stats);
116          }
117 < #ifdef FREEBSD
117 > #ifdef ALLBSD
118          nummnt=getmntinfo(&mp , MNT_LOCAL);
119          if (nummnt<=0){
120                  return NULL;
# Line 199 | Line 178 | disk_stat_t *get_disk_stats(int *entries){
178                          }
179  
180                          disk_ptr=disk_stats+num_disks;
181 < #ifdef FREEBSD
181 > #ifdef ALLBSD
182                          if((disk_ptr->device_name=copy_string(disk_ptr->device_name, mp->f_mntfromname))==NULL){
183                                  return NULL;
184                          }
# Line 292 | Line 271 | void diskio_stat_init(int start, int end, diskio_stat_
271          }
272   }
273  
295 #ifdef SOLARIS
296 char *drive_map(char *sunos_name){
297        file_line_t *file_lines = NULL;
298        file_line_t *f_ptr;
299        static disk_mapping_t *disk_mapping = NULL;
300        disk_mapping_t *d_ptr;
301        int x = 0;
302        int y = 256;
303        FILE *f;
304        DIR *dsk;
305        struct dirent *dsk_ent;
306        char linkpointer[PATH_MAX];
307        char link_path[PATH_MAX];
308        char *p, *r, *q;
309
310        int dev_num;
311        char dev_name[51];
312
313        if(disk_mapping == NULL){
314                /* Read in the path_to_inst file into memory */
315                file_lines = malloc(sizeof(file_line_t) * y);
316                f_ptr = file_lines;
317                if(file_lines == NULL) return NULL;
318                f = fopen("/etc/path_to_inst", "r");
319                if (f == NULL){
320                        free(file_lines);
321                        return NULL;
322                }      
323                for(x=0;fgets(f_ptr->line, sizeof(f_ptr->line), f);x++){
324                        if ((x+1) >= y){
325                                y = y*2;
326                                file_lines = realloc(file_lines, sizeof(file_line_t) * y);
327                                if (file_lines == NULL) return NULL;
328                                f_ptr=file_lines+x;
329                        }
330                        f_ptr++;
331                }
332                f_ptr = NULL;
333                fclose(f);
334
335                dsk = opendir("/dev/dsk");
336                if (dsk == NULL){
337                        free(file_lines);
338                        return NULL;
339                }
340                
341                while((dsk_ent = readdir(dsk)) != NULL){
342                        snprintf(link_path, sizeof(link_path), "/dev/dsk/%s", dsk_ent->d_name);
343                        /* Ignore the non symlinks, e.g. ".." and "." */
344                        if((readlink(link_path, linkpointer, PATH_MAX)) == -1) continue;
345                        /* We are only intrested in the string past the ../../devices/ part */
346                        p = strstr(linkpointer, "devices");
347                        p+=8;
348                
349                        /* We are not intrested in the :a etc slices */
350                        q = strrchr(p, ':');
351                        *q = '\0';
352
353                        /* char *p should not contain the info we need to look up in the
354                         * path_to_inst file.
355                         */
356                        
357                        for(f_ptr = file_lines; f_ptr != NULL; f_ptr++){
358                                r = strstr(f_ptr->line, p);
359                                if (r != NULL) break;
360                        }
361
362                        if(r == NULL){
363                                free(file_lines);
364                                closedir(dsk);
365                                return NULL;
366                        }
367
368                        /* f_ptr should be pointing to the line of path_to_inst we are intrested in now */
369
370                        sscanf(r, "%*s %d \"%50s", &dev_num, dev_name);
371                        q = strrchr(dev_name, '"');
372                        if (q == NULL){
373                                free(file_lines);
374                                closedir(dsk);
375                                return NULL;
376                        }
377                        *q = '\0';
378                        
379                        /* Nasty... Add the number to the end of the string.. This means
380                         * dev_name now contains the sunos name.. E.g. ssd1
381                         */
382                        snprintf(dev_name, 50, "%s%d", dev_name, dev_num);
383                        if (disk_mapping == NULL){
384                                disk_mapping = malloc(sizeof(disk_mapping_t));
385                                d_ptr = disk_mapping;
386                        }else{
387                                d_ptr->next = malloc(sizeof(disk_mapping_t));
388                                d_ptr = d_ptr->next;
389                        }
390
391                        if (d_ptr == NULL){
392                                free(file_lines);
393                                closedir(dsk);
394                                return NULL;
395                        }
396
397                        if( ((d_ptr->sunos_name = strdup(dev_name)) == NULL) ||
398                           ((d_ptr->solaris_name = strdup(dsk_ent->d_name))==NULL) ){
399                                free(file_lines);
400                                closedir(dsk);
401                                return NULL;
402                        }
403                }
404                free(file_lines);
405                closedir(dsk);
406        }
407                
408        for(d_ptr = disk_mapping; d_ptr !=NULL; d_ptr=d_ptr->next){
409                if(!strcmp(sunos_name, d_ptr->sunos_name)){
410                        return (strdup(d_ptr->solaris_name));
411                }
412        }
413
414        /* If we really fail to find it.. Return the original name back */
415        return strdup(sunos_name);
416 }
417
418 #endif
419
274   diskio_stat_t *diskio_stat_malloc(int needed_entries, int *cur_entries, diskio_stat_t *diskio_stats){
275  
276          if(diskio_stats==NULL){
# Line 483 | Line 337 | diskio_stat_t *get_diskio_stats(int *entries){
337          long sel_gen;
338          struct devstat *dev_ptr;
339   #endif
340 + #ifdef NETBSD
341 +        /* FIXME get_diskio_stats NYI on NetBSD.
342 +         * See vmstat/dkstats.c in NetBSD source for examples.
343 +         */
344 + #endif
345          num_diskio=0;
346  
347   #ifdef FREEBSD
# Line 492 | Line 351 | diskio_stat_t *get_diskio_stats(int *entries){
351                  if(stats.dinfo==NULL) return NULL;
352                  stats_init = 1;
353          }
354 + #ifdef FREEBSD5
355 +        if ((devstat_getdevs(NULL, &stats)) < 0) return NULL;
356 +        /* Not aware of a get all devices, so i said 999. If we ever
357 +         * find a machine with more than 999 disks, then i'll change
358 +         * this number :)
359 +         */
360 +        if (devstat_selectdevs(&dev_sel, &n_selected, &n_selections, &sel_gen, stats.dinfo->generation, stats.dinfo->devices, stats.dinfo->numdevs, NULL, 0, NULL, 0, DS_SELECT_ONLY, 999, 1) < 0) return NULL;
361 + #else
362          if ((getdevs(&stats)) < 0) return NULL;
363          /* Not aware of a get all devices, so i said 999. If we ever
364           * find a machine with more than 999 disks, then i'll change
365           * this number :)
366           */
367          if (selectdevs(&dev_sel, &n_selected, &n_selections, &sel_gen, stats.dinfo->generation, stats.dinfo->devices, stats.dinfo->numdevs, NULL, 0, NULL, 0, DS_SELECT_ONLY, 999, 1) < 0) return NULL;
368 + #endif
369  
370          for(counter=0;counter<stats.dinfo->numdevs;counter++){
371                  dev_ptr=&stats.dinfo->devices[dev_sel[counter].position];
# Line 506 | Line 374 | diskio_stat_t *get_diskio_stats(int *entries){
374                   * devices.. like mem, proc.. and also doesn't report floppy
375                   * drives etc unless they are doing stuff :)
376                   */
377 + #ifdef FREEBSD5
378 +                if((dev_ptr->bytes[DEVSTAT_READ]==0) && (dev_ptr->bytes[DEVSTAT_WRITE]==0)) continue;
379 + #else
380                  if((dev_ptr->bytes_read==0) && (dev_ptr->bytes_written==0)) continue;
381 + #endif
382                  if((diskio_stats=diskio_stat_malloc(num_diskio+1, &sizeof_diskio_stats, diskio_stats))==NULL){
383                          return NULL;
384                  }
385                  diskio_stats_ptr=diskio_stats+num_diskio;
386 <                
386 >
387 > #ifdef FREEBSD5        
388 >                diskio_stats_ptr->read_bytes=dev_ptr->bytes[DEVSTAT_READ];
389 >                diskio_stats_ptr->write_bytes=dev_ptr->bytes[DEVSTAT_WRITE];
390 > #else
391                  diskio_stats_ptr->read_bytes=dev_ptr->bytes_read;
392                  diskio_stats_ptr->write_bytes=dev_ptr->bytes_written;
393 + #endif
394                  if(diskio_stats_ptr->disk_name!=NULL) free(diskio_stats_ptr->disk_name);
395                  asprintf((&diskio_stats_ptr->disk_name), "%s%d", dev_ptr->device_name, dev_ptr->unit_number);
396                  diskio_stats_ptr->systime=time(NULL);
# Line 549 | Line 426 | diskio_stat_t *get_diskio_stats(int *entries){
426  
427                          if(diskio_stats_ptr->disk_name!=NULL) free(diskio_stats_ptr->disk_name);
428  
429 <                        diskio_stats_ptr->disk_name=drive_map(ksp->ks_name);
429 >                        diskio_stats_ptr->disk_name=strdup(ksp->ks_name);
430                          diskio_stats_ptr->systime=time(NULL);
431                          num_diskio++;
432                  }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines