src/rrd_open.c

   1 /*****************************************************************************
   2  * RRDtool 1.1.x  Copyright Tobias Oetiker, 1997 - 2002
   3  *****************************************************************************
   4  * rrd_open.c  Open an RRD File
   5  *****************************************************************************
   6  * $Id$
   7  * $Log$
   8  * Revision 1.5  2002/06/20 00:21:03  jake
   9  * More Win32 build changes; thanks to Kerry Calvert.
  10  *
  11  * Revision 1.4  2002/02/01 20:34:49  oetiker
  12  * fixed version number and date/time
  13  *
  14  * Revision 1.3  2001/03/04 13:01:55  oetiker
  15  * Aberrant Behavior Detection support. A brief overview added to rrdtool.pod.
  16  * Major updates to rrd_update.c, rrd_create.c. Minor update to other core files.
  17  * This is backwards compatible! But new files using the Aberrant stuff are not readable
  18  * by old rrdtool versions. See http://cricket.sourceforge.net/aberrant/rrd_hw.htm
  19  * -- Jake Brutlag <jakeb@corp.webtv.net>
  20  *
  21  * Revision 1.2  2001/03/04 10:29:20  oetiker
  22  * fixed filedescriptor leak
  23  * -- Mike Franusich <mike@franusich.com>
  24  *
  25  * Revision 1.1.1.1  2001/02/25 22:25:05  oetiker
  26  * checkin
  27  *
  28  *****************************************************************************/
  29
  30 #include "rrd_tool.h"
  31 #define MEMBLK 8192
  32
  33 /* open a database file, return its header and a open filehandle */
  34 /* positioned to the first cdp in the first rra */
  35
  36 int
  37 rrd_open(char *file_name, FILE **in_file, rrd_t *rrd, int rdwr)
  38 {
  39
  40
  41     char *mode = NULL;
  42     rrd_init(rrd);
  43     if (rdwr == RRD_READONLY) {
  44 #ifndef WIN32
  45         mode = "r";
  46 #else
  47         mode = "rb";
  48 #endif
  49     } else {
  50 #ifndef WIN32
  51         mode = "r+";
  52 #else
  53         mode = "rb+";
  54 #endif
  55     }
  56
  57     if (((*in_file) = fopen(file_name,mode)) == NULL ){
  58         rrd_set_error("opening '%s': %s",file_name, strerror(errno));
  59         return (-1);
  60     }
  61 /*
  62         if (rdwr == RRD_READWRITE)
  63         {
  64            if (setvbuf((*in_file),NULL,_IONBF,2)) {
  65                   rrd_set_error("failed to disable the stream buffer\n");
  66                   return (-1);
  67            }
  68         }
  69 */
  70
  71 #define MYFREAD(MYVAR,MYVART,MYCNT) \
  72     if ((MYVAR = malloc(sizeof(MYVART) * MYCNT)) == NULL) {\
  73         rrd_set_error("" #MYVAR " malloc"); \
  74         fclose(*in_file); \
  75     return (-1); } \
  76     fread(MYVAR,sizeof(MYVART),MYCNT, *in_file);
  77
  78
  79     MYFREAD(rrd->stat_head, stat_head_t,  1)
  80
  81         /* lets do some test if we are on track ... */
  82         if (strncmp(rrd->stat_head->cookie,RRD_COOKIE,4) != 0){
  83             rrd_set_error("'%s' is not an RRD file",file_name);
  84             free(rrd->stat_head);
  85             fclose(*in_file);
  86             return(-1);}
  87
  88         if (atoi(rrd->stat_head->version) > atoi(RRD_VERSION)){
  89             rrd_set_error("can't handle RRD file version %s",
  90                         rrd->stat_head->version);
  91             free(rrd->stat_head);
  92             fclose(*in_file);
  93             return(-1);}
  94
  95         if (rrd->stat_head->float_cookie != FLOAT_COOKIE){
  96             rrd_set_error("This RRD was created on other architecture");
  97             free(rrd->stat_head);
  98             fclose(*in_file);
  99             return(-1);}
 100
 101     MYFREAD(rrd->ds_def,    ds_def_t,     rrd->stat_head->ds_cnt)
 102     MYFREAD(rrd->rra_def,   rra_def_t,    rrd->stat_head->rra_cnt)
 103     MYFREAD(rrd->live_head,   live_head_t,  1)
 104     MYFREAD(rrd->pdp_prep,  pdp_prep_t,   rrd->stat_head->ds_cnt)
 105     MYFREAD(rrd->cdp_prep,  cdp_prep_t,   (rrd->stat_head->rra_cnt
 106                                              * rrd->stat_head->ds_cnt))
 107     MYFREAD(rrd->rra_ptr,   rra_ptr_t,    rrd->stat_head->rra_cnt)
 108 #undef MYFREAD
 109
 110     return(0);
 111 }
 112
 113 void rrd_init(rrd_t *rrd)
 114 {
 115     rrd->stat_head = NULL;
 116     rrd->ds_def = NULL;
 117     rrd->rra_def = NULL;
 118     rrd->live_head = NULL;
 119     rrd->rra_ptr = NULL;
 120     rrd->pdp_prep = NULL;
 121     rrd->cdp_prep = NULL;
 122     rrd->rrd_value = NULL;
 123 }
 124
 125 void rrd_free(rrd_t *rrd)
 126 {
 127     free(rrd->stat_head);
 128     free(rrd->ds_def);
 129     free(rrd->rra_def);
 130     free(rrd->live_head);
 131     free(rrd->rra_ptr);
 132     free(rrd->pdp_prep);
 133     free(rrd->cdp_prep);
 134     free(rrd->rrd_value);
 135 }
 136
 137 /* routine used by external libraries to free memory allocated by
 138  * rrd library */
 139 void rrd_freemem(void *mem)
 140 {
 141
 142     free(mem);
 143 }
 144
 145 int readfile(char *file_name, char **buffer, int skipfirst){
 146     long writecnt=0,totalcnt = MEMBLK;
 147     FILE *input=NULL;
 148     char c ;
 149     if ((strcmp("-",file_name) == 0)) { input = stdin; }
 150     else {
 151       if ((input = fopen(file_name,"rb")) == NULL ){
 152         rrd_set_error("opening '%s': %s",file_name,strerror(errno));
 153         return (-1);
 154       }
 155     }
 156     if (skipfirst){
 157       do { c = getc(input); } while (c != '\n' && ! feof(input));
 158     }
 159     if (((*buffer) = (char *) malloc((MEMBLK+4)*sizeof(char))) == NULL) {
 160         perror("Allocate Buffer:");
 161         exit(1);
 162     };
 163     do{
 164       writecnt += fread((*buffer)+writecnt, 1, MEMBLK * sizeof(char) ,input);
 165       if (writecnt >= totalcnt){
 166         totalcnt += MEMBLK;
 167         if (((*buffer)=rrd_realloc((*buffer), (totalcnt+4) * sizeof(char)))==NULL){
 168             perror("Realloc Buffer:");
 169             exit(1);
 170         };
 171       }
 172     } while (! feof(input));
 173     (*buffer)[writecnt] = '\0';
 174     if (strcmp("-",file_name) != 0) {fclose(input);};
 175     return writecnt;
 176 }
 177
 178