src/rrd_open.c

   1 /*****************************************************************************
   2  * RRDtool 1.0.33  Copyright Tobias Oetiker, 1997 - 2000
   3  *****************************************************************************
   4  * rrd_open.c  Open an RRD File
   5  *****************************************************************************
   6  * $Id$
   7  * $Log$
   8  * Revision 1.2  2001/03/04 10:29:20  oetiker
   9  * fixed filedescriptor leak
  10  * -- Mike Franusich <mike@franusich.com>
  11  *
  12  * Revision 1.1.1.1  2001/02/25 22:25:05  oetiker
  13  * checkin
  14  *
  15  *****************************************************************************/
  16
  17 #include "rrd_tool.h"
  18 #define MEMBLK 8192
  19
  20 /* open a database file, return its header and a open filehandle */
  21 /* positioned to the first cdp in the first rra */
  22
  23 int
  24 rrd_open(char *file_name, FILE **in_file, rrd_t *rrd, int rdwr)
  25 {
  26
  27
  28     char *mode = NULL;
  29     rrd_init(rrd);
  30     if (rdwr == RRD_READONLY) {
  31 #ifndef WIN32
  32         mode = "r";
  33 #else
  34         mode = "rb";
  35 #endif
  36     } else {
  37 #ifndef WIN32
  38         mode = "r+";
  39 #else
  40         mode = "rb+";
  41 #endif
  42     }
  43
  44     if (((*in_file) = fopen(file_name,mode)) == NULL ){
  45         rrd_set_error("opening '%s': %s",file_name, strerror(errno));
  46         return (-1);
  47     }
  48
  49 #define MYFREAD(MYVAR,MYVART,MYCNT) \
  50     if ((MYVAR = malloc(sizeof(MYVART) * MYCNT)) == NULL) {\
  51         rrd_set_error("" #MYVAR " malloc"); \
  52         fclose(*in_file); \
  53     return (-1); } \
  54     fread(MYVAR,sizeof(MYVART),MYCNT, *in_file);
  55
  56
  57     MYFREAD(rrd->stat_head, stat_head_t,  1)
  58
  59         /* lets do some test if we are on track ... */
  60         if (strncmp(rrd->stat_head->cookie,RRD_COOKIE,4) != 0){
  61             rrd_set_error("'%s' is not an RRD file",file_name);
  62             free(rrd->stat_head);
  63             fclose(*in_file);
  64             return(-1);}
  65
  66         if (strncmp(rrd->stat_head->version,RRD_VERSION,5) != 0){
  67             rrd_set_error("can't handle RRD file version %s",
  68                         rrd->stat_head->version);
  69             free(rrd->stat_head);
  70             fclose(*in_file);
  71             return(-1);}
  72
  73         if (rrd->stat_head->float_cookie != FLOAT_COOKIE){
  74             rrd_set_error("This RRD was created on other architecture");
  75             free(rrd->stat_head);
  76             fclose(*in_file);
  77             return(-1);}
  78
  79     MYFREAD(rrd->ds_def,    ds_def_t,     rrd->stat_head->ds_cnt)
  80     MYFREAD(rrd->rra_def,   rra_def_t,    rrd->stat_head->rra_cnt)
  81     MYFREAD(rrd->live_head,   live_head_t,  1)
  82     MYFREAD(rrd->pdp_prep,  pdp_prep_t,   rrd->stat_head->ds_cnt)
  83     MYFREAD(rrd->cdp_prep,  cdp_prep_t,   (rrd->stat_head->rra_cnt
  84                                              * rrd->stat_head->ds_cnt))
  85     MYFREAD(rrd->rra_ptr,   rra_ptr_t,    rrd->stat_head->rra_cnt)
  86 #undef MYFREAD
  87
  88     return(0);
  89 }
  90
  91 void rrd_init(rrd_t *rrd)
  92 {
  93     rrd->stat_head = NULL;
  94     rrd->ds_def = NULL;
  95     rrd->rra_def = NULL;
  96     rrd->live_head = NULL;
  97     rrd->rra_ptr = NULL;
  98     rrd->pdp_prep = NULL;
  99     rrd->cdp_prep = NULL;
 100     rrd->rrd_value = NULL;
 101 }
 102
 103 void rrd_free(rrd_t *rrd)
 104 {
 105     free(rrd->stat_head);
 106     free(rrd->ds_def);
 107     free(rrd->rra_def);
 108     free(rrd->live_head);
 109     free(rrd->rra_ptr);
 110     free(rrd->pdp_prep);
 111     free(rrd->cdp_prep);
 112     free(rrd->rrd_value);
 113 }
 114
 115 int readfile(char *file_name, char **buffer, int skipfirst){
 116     long writecnt=0,totalcnt = MEMBLK;
 117     FILE *input=NULL;
 118     char c ;
 119     if ((strcmp("-",file_name) == 0)) { input = stdin; }
 120     else {
 121       if ((input = fopen(file_name,"rb")) == NULL ){
 122         rrd_set_error("opening '%s': %s",file_name,strerror(errno));
 123         return (-1);
 124       }
 125     }
 126     if (skipfirst){
 127       do { c = getc(input); } while (c != '\n' && ! feof(input));
 128     }
 129     if (((*buffer) = (char *) malloc((MEMBLK+4)*sizeof(char))) == NULL) {
 130         perror("Allocate Buffer:");
 131         exit(1);
 132     };
 133     do{
 134       writecnt += fread((*buffer)+writecnt, 1, MEMBLK * sizeof(char) ,input);
 135       if (writecnt >= totalcnt){
 136         totalcnt += MEMBLK;
 137         if (((*buffer)=rrd_realloc((*buffer), (totalcnt+4) * sizeof(char)))==NULL){
 138             perror("Realloc Buffer:");
 139             exit(1);
 140         };
 141       }
 142     } while (! feof(input));
 143     (*buffer)[writecnt] = '\0';
 144     if (strcmp("-",file_name) != 0) {fclose(input);};
 145     return writecnt;
 146 }
 147
 148