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