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