src/rrd_restore.c

   1 /*****************************************************************************
   2  * RRDtool 1.2.28  Copyright by Tobi Oetiker, 1997-2008
   3  *****************************************************************************
   4  * rrd_restore.c  creates new rrd from data dumped by rrd_dump.c
   5  *****************************************************************************/
   6
   7 #include "rrd_tool.h"
   8 #include "rrd_rpncalc.h"
   9 #include <fcntl.h>
  10
  11 #if defined(_WIN32) && !defined(__CYGWIN__) && !defined(__CYGWIN32__)
  12 #include <io.h>
  13 #define open _open
  14 #define close _close
  15 #endif
  16
  17 /* Prototypes */
  18
  19 void xml_lc(char*);
  20 int skip(char **);
  21 int skipxml(char **);
  22 int eat_tag(char **, char *);
  23 int read_tag(char **, char *, char *, void *);
  24 int xml2rrd(char*, rrd_t*, char);
  25 int rrd_write(char *, rrd_t *, char);
  26 void parse_patch1028_RRA_params(char **buf, rrd_t *rrd, int rra_index);
  27 void parse_patch1028_CDP_params(char **buf, rrd_t *rrd, int rra_index, int ds_index);
  28 void parse_FAILURES_history(char **buf, rrd_t *rrd, int rra_index, int ds_index);
  29 long int rra_random_row(rra_def_t *);
  30
  31 /* convert all occurrences of <BlaBlaBla> to <blablabla> */
  32
  33 void xml_lc(char* buf){
  34   int intag=0;
  35   while((*buf)){
  36     if (intag ==0 && (*buf) == '<') {
  37       intag = 1;
  38     }
  39     else if (intag ==1 && (*buf) == '>') {
  40       intag = 0;
  41       continue;
  42     } else  if (intag ==1) {
  43       *buf = tolower(*buf);
  44     }
  45     buf++;
  46   }
  47 }
  48
  49 int skipxml(char **buf){
  50   char *ptr;
  51   ptr=(*buf);
  52   do {
  53     (*buf)=ptr;
  54     while((*(ptr+1)) && ((*ptr)==' ' ||  (*ptr)=='\r' || (*ptr)=='\n' || (*ptr)=='\t')) ptr++;
  55     if (strncmp(ptr,"<?xml",4) == 0) {
  56       ptr= strstr(ptr,"?>");
  57       if (ptr) ptr+=2; else {
  58         rrd_set_error("Dangling XML header");
  59         (*buf) = NULL;
  60         return -1;
  61       }
  62     }
  63   } while ((*buf)!=ptr);
  64   return 1;
  65 }
  66
  67 int skip(char **buf){
  68   char *ptr;
  69   if ((buf == NULL) || (*buf == NULL))
  70     return -1;
  71   ptr=(*buf);
  72   do {
  73     (*buf)=ptr;
  74     while((*(ptr+1)) && ((*ptr)==' ' ||  (*ptr)=='\r' || (*ptr)=='\n' || (*ptr)=='\t')) ptr++;
  75     if (strncmp(ptr,"<!--",4) == 0) {
  76       ptr= strstr(ptr,"-->");
  77       if (ptr) ptr+=3; else {
  78         rrd_set_error("Dangling Comment");
  79         (*buf) = NULL;
  80         return -1;
  81       }
  82     }
  83   } while ((*buf)!=ptr);
  84   return 1;
  85 }
  86
  87 int eat_tag(char **buf, char *tag){
  88   if ((*buf)==NULL) return -1;   /* fall though clause */
  89
  90   rrd_clear_error();
  91   skip(buf);
  92   if ((**buf)=='<'
  93       && strncmp((*buf)+1,tag,strlen(tag)) == 0
  94       && *((*buf)+strlen(tag)+1)=='>') {
  95     (*buf) += strlen(tag)+2;
  96   }
  97   else {
  98     rrd_set_error("No <%s> tag found",tag);
  99     (*buf) = NULL;
 100     return -1;
 101   }
 102   skip(buf);
 103   return 1;
 104 }
 105
 106 int read_tag(char **buf, char *tag, char *format, void *value){
 107     char *end_tag;
 108     int matches;
 109     if ((*buf)==NULL) return -1;   /* fall though clause */
 110     rrd_clear_error();
 111     if (eat_tag(buf,tag)==1){
 112         char *temp;
 113         temp = (*buf);
 114         while(*((*buf)+1) && (*(*buf) != '<')) (*buf)++; /*find start of endtag*/
 115         *(*buf) = '\0';
 116         matches =sscanf(temp,format,value);
 117         *(*buf) = '<';
 118         end_tag = malloc((strlen(tag)+2)*sizeof(char));
 119         sprintf(end_tag,"/%s",tag);
 120         eat_tag(buf,end_tag);
 121         free(end_tag);
 122         if (matches == 0 && strcmp(format,"%lf") == 0)
 123             (*((double* )(value))) = DNAN;
 124         if (matches != 1)       return 0;
 125         return 1;
 126     }
 127     return -1;
 128 }
 129
 130
 131 /* parse the data stored in buf and return a filled rrd structure */
 132 int xml2rrd(char* buf, rrd_t* rrd, char rc){
 133   /* pass 1 identify number of RRAs  */
 134   char *ptr,*ptr2,*ptr3; /* walks thought the buffer */
 135   long rows=0,mempool=0,i=0;
 136   int rra_index;
 137   int input_version;
 138   xml_lc(buf); /* lets lowercase all active parts of the xml */
 139   ptr=buf;
 140   ptr2=buf;
 141   ptr3=buf;
 142   /* start with an RRD tag */
 143
 144   skipxml(&ptr);
 145
 146   eat_tag(&ptr,"rrd");
 147   /* allocate static header */
 148   if((rrd->stat_head = calloc(1,sizeof(stat_head_t)))==NULL){
 149     rrd_set_error("allocating rrd.stat_head");
 150     return -1;
 151   };
 152
 153   strcpy(rrd->stat_head->cookie,RRD_COOKIE);
 154   read_tag(&ptr,"version","%4[0-9]",rrd->stat_head->version);
 155   input_version = atoi(rrd->stat_head->version);
 156   /* added primitive version checking */
 157   if (input_version > atoi(RRD_VERSION) || input_version < 1)
 158   {
 159     rrd_set_error("Incompatible file version, detected version %s. This is not supported by the version %s restore tool.\n",
 160                   rrd -> stat_head -> version, RRD_VERSION );
 161     free(rrd -> stat_head);
 162     rrd->stat_head = NULL;
 163     return -1;
 164   }
 165   /* make sure we output the right version */
 166   strcpy(rrd->stat_head->version,RRD_VERSION);
 167
 168   /*  if (atoi(rrd -> stat_head -> version) < 2)
 169   {
 170     rrd_set_error("Can only restore version >= 2 (Not %s). Dump your old rrd using a current rrdtool dump.",  rrd -> stat_head -> version );
 171     return -1;
 172   } */
 173
 174   rrd->stat_head->float_cookie = FLOAT_COOKIE;
 175   rrd->stat_head->ds_cnt = 0;
 176   rrd->stat_head->rra_cnt = 0;
 177   read_tag(&ptr,"step","%lu",&(rrd->stat_head->pdp_step));
 178
 179   /* allocate live head */
 180   if((rrd->live_head = calloc(1,sizeof(live_head_t)))==NULL){
 181     rrd_set_error("allocating rrd.live_head");
 182     return -1;
 183   }
 184   read_tag(&ptr,"lastupdate","%lu",&(rrd->live_head->last_up));
 185
 186   /* Data Source Definition Part */
 187   ptr2 = ptr;
 188   while (eat_tag(&ptr2,"ds") == 1){
 189       rrd->stat_head->ds_cnt++;
 190       if((rrd->ds_def = rrd_realloc(rrd->ds_def,rrd->stat_head->ds_cnt*sizeof(ds_def_t)))==NULL){
 191           rrd_set_error("allocating rrd.ds_def");
 192           return -1;
 193       };
 194       /* clean out memory to make sure no data gets stored from previous tasks */
 195       memset(&(rrd->ds_def[rrd->stat_head->ds_cnt-1]), 0, sizeof(ds_def_t));
 196       if((rrd->pdp_prep = rrd_realloc(rrd->pdp_prep,rrd->stat_head->ds_cnt
 197                                   *sizeof(pdp_prep_t)))==NULL){
 198         rrd_set_error("allocating pdp_prep");
 199         return(-1);
 200       }
 201       /* clean out memory to make sure no data gets stored from previous tasks */
 202       memset(&(rrd->pdp_prep[rrd->stat_head->ds_cnt-1]), 0, sizeof(pdp_prep_t));
 203
 204       read_tag(&ptr2,"name",DS_NAM_FMT,rrd->ds_def[rrd->stat_head->ds_cnt-1].ds_nam);
 205
 206       read_tag(&ptr2,"type",DST_FMT,rrd->ds_def[rrd->stat_head->ds_cnt-1].dst);
 207       /* test for valid type */
 208       if( (int)dst_conv(rrd->ds_def[rrd->stat_head->ds_cnt-1].dst) == -1) return -1;
 209
 210           if (dst_conv(rrd->ds_def[rrd->stat_head->ds_cnt-1].dst) != DST_CDEF)
 211           {
 212       read_tag(&ptr2,"minimal_heartbeat","%lu",
 213                &(rrd->ds_def[rrd->stat_head->ds_cnt-1].par[DS_mrhb_cnt].u_cnt));
 214       read_tag(&ptr2,"min","%lf",&(rrd->ds_def[rrd->stat_head->ds_cnt-1].par[DS_min_val].u_val));
 215       read_tag(&ptr2,"max","%lf",&(rrd->ds_def[rrd->stat_head->ds_cnt-1].par[DS_max_val].u_val));
 216           } else { /* DST_CDEF */
 217                  char buffer[1024];
 218                  read_tag(&ptr2,"cdef","%1000s",buffer);
 219                  parseCDEF_DS(buffer,rrd,rrd -> stat_head -> ds_cnt - 1);
 220                  if (rrd_test_error()) return -1;
 221           }
 222
 223       read_tag(&ptr2,"last_ds","%30s",rrd->pdp_prep[rrd->stat_head->ds_cnt-1].last_ds);
 224       read_tag(&ptr2,"value","%lf",&(rrd->pdp_prep[rrd->stat_head->ds_cnt-1].scratch[PDP_val].u_val));
 225       read_tag(&ptr2,"unknown_sec","%lu",&(rrd->pdp_prep[rrd->stat_head->ds_cnt-1].scratch[PDP_unkn_sec_cnt].u_cnt));
 226       eat_tag(&ptr2,"/ds");
 227       ptr=ptr2;
 228   }
 229
 230   ptr2 = ptr;
 231   while (eat_tag(&ptr2,"rra") == 1){
 232       rrd->stat_head->rra_cnt++;
 233
 234       /* allocate and reset rra definition areas */
 235       if((rrd->rra_def = rrd_realloc(rrd->rra_def,rrd->stat_head->rra_cnt*sizeof(rra_def_t)))==NULL){
 236           rrd_set_error("allocating rra_def"); return -1; }
 237       memset(&(rrd->rra_def[rrd->stat_head->rra_cnt-1]), 0, sizeof(rra_def_t));
 238
 239       /* allocate and reset consolidation point areas */
 240       if((rrd->cdp_prep = rrd_realloc(rrd->cdp_prep,
 241                                   rrd->stat_head->rra_cnt
 242                                   *rrd->stat_head->ds_cnt*sizeof(cdp_prep_t)))==NULL){
 243          rrd_set_error("allocating cdp_prep"); return -1; }
 244
 245       memset(&(rrd->cdp_prep[rrd->stat_head->ds_cnt*(rrd->stat_head->rra_cnt-1)]),
 246              0, rrd->stat_head->ds_cnt*sizeof(cdp_prep_t));
 247
 248
 249       read_tag(&ptr2,"cf",CF_NAM_FMT,rrd->rra_def[rrd->stat_head->rra_cnt-1].cf_nam);
 250       /* test for valid type */
 251       if( (int)cf_conv(rrd->rra_def[rrd->stat_head->rra_cnt-1].cf_nam) == -1) return -1;
 252
 253       read_tag(&ptr2,"pdp_per_row","%lu",&(rrd->rra_def[rrd->stat_head->rra_cnt-1].pdp_cnt));
 254       /* support to read RRA parameters */
 255       rra_index = rrd->stat_head->rra_cnt - 1;
 256       if ( input_version < 2 ){
 257          read_tag(&ptr2, "xff","%lf",
 258             &(rrd->rra_def[rra_index].par[RRA_cdp_xff_val].u_val));
 259       } else {
 260         if (eat_tag(&ptr2, "params") != 1) {
 261           rrd_set_error("could not find params tag to eat and skip");
 262           return -1;
 263         }
 264         skip(&ptr2);
 265         /* backwards compatibility w/ old patch */
 266       if (strncmp(ptr2, "<value>",7) == 0) {
 267           parse_patch1028_RRA_params(&ptr2,rrd,rra_index);
 268       } else {
 269       switch(cf_conv(rrd -> rra_def[rra_index].cf_nam)) {
 270       case CF_HWPREDICT:
 271          read_tag(&ptr2, "hw_alpha", "%lf",
 272             &(rrd->rra_def[rra_index].par[RRA_hw_alpha].u_val));
 273          read_tag(&ptr2, "hw_beta", "%lf",
 274             &(rrd->rra_def[rra_index].par[RRA_hw_beta].u_val));
 275          read_tag(&ptr2, "dependent_rra_idx", "%lu",
 276             &(rrd->rra_def[rra_index].par[RRA_dependent_rra_idx].u_cnt));
 277          break;
 278       case CF_SEASONAL:
 279       case CF_DEVSEASONAL:
 280          read_tag(&ptr2, "seasonal_gamma", "%lf",
 281             &(rrd->rra_def[rra_index].par[RRA_seasonal_gamma].u_val));
 282          read_tag(&ptr2, "seasonal_smooth_idx", "%lu",
 283             &(rrd->rra_def[rra_index].par[RRA_seasonal_smooth_idx].u_cnt));
 284          read_tag(&ptr2, "dependent_rra_idx", "%lu",
 285             &(rrd->rra_def[rra_index].par[RRA_dependent_rra_idx].u_cnt));
 286          break;
 287       case CF_FAILURES:
 288          read_tag(&ptr2, "delta_pos", "%lf",
 289             &(rrd->rra_def[rra_index].par[RRA_delta_pos].u_val));
 290          read_tag(&ptr2, "delta_neg", "%lf",
 291             &(rrd->rra_def[rra_index].par[RRA_delta_neg].u_val));
 292          read_tag(&ptr2, "window_len", "%lu",
 293             &(rrd->rra_def[rra_index].par[RRA_window_len].u_cnt));
 294          read_tag(&ptr2, "failure_threshold", "%lu",
 295             &(rrd->rra_def[rra_index].par[RRA_failure_threshold].u_cnt));
 296          /* fall thru */
 297       case CF_DEVPREDICT:
 298          read_tag(&ptr2, "dependent_rra_idx", "%lu",
 299             &(rrd->rra_def[rra_index].par[RRA_dependent_rra_idx].u_cnt));
 300          break;
 301       case CF_AVERAGE:
 302       case CF_MAXIMUM:
 303       case CF_MINIMUM:
 304       case CF_LAST:
 305       default:
 306          read_tag(&ptr2, "xff","%lf",
 307             &(rrd->rra_def[rra_index].par[RRA_cdp_xff_val].u_val));
 308       }
 309       }
 310       eat_tag(&ptr2, "/params");
 311    }
 312
 313
 314       eat_tag(&ptr2,"cdp_prep");
 315       for(i=0;i< (int)rrd->stat_head->ds_cnt;i++)
 316       {
 317          if (eat_tag(&ptr2, "ds") != 1){
 318             rrd_set_error("expected to find %lu <ds> entries in <cdp_prep>",rrd->stat_head->ds_cnt);
 319             return -1;
 320          }
 321       /* support to read CDP parameters */
 322       rra_index = rrd->stat_head->rra_cnt-1;
 323       skip(&ptr2);
 324       if ( input_version < 2 ){
 325           rrd->cdp_prep[rrd->stat_head->ds_cnt*(rra_index)+i].scratch[CDP_primary_val].u_val = 0.0;
 326           rrd->cdp_prep[rrd->stat_head->ds_cnt*(rra_index)+i].scratch[CDP_secondary_val].u_val = 0.0;
 327           read_tag(&ptr2,"value","%lf",&(rrd->cdp_prep[rrd->stat_head->ds_cnt
 328                *(rra_index) +i].scratch[CDP_val].u_val));
 329           read_tag(&ptr2,"unknown_datapoints","%lu",&(rrd->cdp_prep[rrd->stat_head->ds_cnt
 330               *(rra_index) +i].scratch[CDP_unkn_pdp_cnt].u_cnt));
 331       } else {
 332
 333       if (strncmp(ptr2, "<value>",7) == 0) {
 334          parse_patch1028_CDP_params(&ptr2,rrd,rra_index,i);
 335       } else {
 336          read_tag(&ptr2, "primary_value","%lf",
 337                &(rrd->cdp_prep[rrd->stat_head->ds_cnt*(rra_index)
 338                +i].scratch[CDP_primary_val].u_val));
 339          read_tag(&ptr2, "secondary_value","%lf",
 340                &(rrd->cdp_prep[rrd->stat_head->ds_cnt*(rra_index)
 341                +i].scratch[CDP_secondary_val].u_val));
 342          switch(cf_conv(rrd->rra_def[rra_index].cf_nam)) {
 343          case CF_HWPREDICT:
 344             read_tag(&ptr2,"intercept","%lf",
 345                &(rrd->cdp_prep[rrd->stat_head->ds_cnt*(rra_index)
 346                +i].scratch[CDP_hw_intercept].u_val));
 347             read_tag(&ptr2,"last_intercept","%lf",
 348                &(rrd->cdp_prep[rrd->stat_head->ds_cnt*(rra_index)
 349                +i].scratch[CDP_hw_last_intercept].u_val));
 350             read_tag(&ptr2,"slope","%lf",
 351                &(rrd->cdp_prep[rrd->stat_head->ds_cnt*(rra_index)
 352                +i].scratch[CDP_hw_slope].u_val));
 353             read_tag(&ptr2,"last_slope","%lf",
 354                &(rrd->cdp_prep[rrd->stat_head->ds_cnt*(rra_index)
 355                +i].scratch[CDP_hw_last_slope].u_val));
 356             read_tag(&ptr2,"nan_count","%lu",
 357                &(rrd->cdp_prep[rrd->stat_head->ds_cnt*(rra_index)
 358                +i].scratch[CDP_null_count].u_cnt));
 359             read_tag(&ptr2,"last_nan_count","%lu",
 360                &(rrd->cdp_prep[rrd->stat_head->ds_cnt*(rra_index)
 361                +i].scratch[CDP_last_null_count].u_cnt));
 362             break;
 363          case CF_SEASONAL:
 364          case CF_DEVSEASONAL:
 365             read_tag(&ptr2,"seasonal","%lf",
 366                &(rrd->cdp_prep[rrd->stat_head->ds_cnt*(rra_index)
 367                +i].scratch[CDP_hw_seasonal].u_val));
 368             read_tag(&ptr2,"last_seasonal","%lf",
 369                &(rrd->cdp_prep[rrd->stat_head->ds_cnt*(rra_index)
 370                +i].scratch[CDP_hw_last_seasonal].u_val));
 371             read_tag(&ptr2,"init_flag","%lu",
 372                &(rrd->cdp_prep[rrd->stat_head->ds_cnt*(rra_index)
 373                +i].scratch[CDP_init_seasonal].u_cnt));
 374             break;
 375          case CF_DEVPREDICT:
 376             break;
 377          case CF_FAILURES:
 378             parse_FAILURES_history(&ptr2,rrd,rra_index,i);
 379             break;
 380          case CF_AVERAGE:
 381          case CF_MAXIMUM:
 382          case CF_MINIMUM:
 383          case CF_LAST:
 384          default:
 385             read_tag(&ptr2,"value","%lf",&(rrd->cdp_prep[rrd->stat_head->ds_cnt
 386                *(rra_index) +i].scratch[CDP_val].u_val));
 387             read_tag(&ptr2,"unknown_datapoints","%lu",&(rrd->cdp_prep[rrd->stat_head->ds_cnt
 388                *(rra_index) +i].scratch[CDP_unkn_pdp_cnt].u_cnt));
 389             break;
 390          }
 391       }
 392       }
 393       eat_tag(&ptr2,"/ds");
 394       }
 395       eat_tag(&ptr2,"/cdp_prep");
 396       rrd->rra_def[rrd->stat_head->rra_cnt-1].row_cnt=0;
 397       eat_tag(&ptr2,"database");
 398       ptr3 = ptr2;
 399       while (eat_tag(&ptr3,"row") == 1){
 400
 401           if(mempool==0){
 402             mempool = 1000;
 403             if((rrd->rrd_value = rrd_realloc(rrd->rrd_value,
 404                                          (rows+mempool)*(rrd->stat_head->ds_cnt)
 405                                          *sizeof(rrd_value_t)))==NULL) {
 406               rrd_set_error("allocating rrd_values"); return -1; }
 407           }
 408           rows++;
 409           mempool--;
 410           rrd->rra_def[rrd->stat_head->rra_cnt-1].row_cnt++;
 411           for(i=0;i< (int)rrd->stat_head->ds_cnt;i++){
 412
 413                   rrd_value_t  * value = &(rrd->rrd_value[(rows-1)*rrd->stat_head->ds_cnt+i]);
 414
 415                   read_tag(&ptr3,"v","%lf", value);
 416
 417                   if (
 418                           (rc == 1)                     /* do we have to check for the ranges */
 419                           &&
 420                       (!isnan(*value))  /* not a NAN value */
 421                       &&
 422                           (dst_conv(rrd->ds_def[i].dst) != DST_CDEF)
 423                           &&
 424                       (                                 /* min defined and in the range ? */
 425                           (!isnan(rrd->ds_def[i].par[DS_min_val].u_val)
 426                                 && (*value < rrd->ds_def[i].par[DS_min_val].u_val))
 427                           ||                            /* max defined and in the range ? */
 428                           (!isnan(rrd->ds_def[i].par[DS_max_val].u_val)
 429                                 && (*value > rrd->ds_def[i].par[DS_max_val].u_val))
 430                       )
 431                   ) {
 432                       fprintf (stderr, "out of range found [ds: %lu], [value : %0.10e]\n", i, *value);
 433                       *value = DNAN;
 434                   }
 435           }
 436           eat_tag(&ptr3,"/row");
 437           ptr2=ptr3;
 438       }
 439       eat_tag(&ptr2,"/database");
 440       eat_tag(&ptr2,"/rra");
 441       ptr=ptr2;
 442   }
 443   eat_tag(&ptr,"/rrd");
 444
 445   if((rrd->rra_ptr = calloc(1,sizeof(rra_ptr_t)*rrd->stat_head->rra_cnt)) == NULL) {
 446       rrd_set_error("allocating rra_ptr");
 447       return(-1);
 448   }
 449
 450   if (ptr==NULL)
 451       return -1;
 452   return 1;
 453 }
 454
 455
 456
 457
 458
 459 /* create and empty rrd file according to the specs given */
 460
 461 int
 462 rrd_write(char *file_name, rrd_t *rrd, char force_overwrite)
 463 {
 464     unsigned long    i,ii,rra_offset;
 465     FILE             *rrd_file=NULL;
 466     int                 fdflags;
 467     int                 fd;
 468
 469     if (strcmp("-",file_name)==0){
 470       rrd_file= stdout;
 471     } else {
 472 #if defined(_WIN32) && !defined(__CYGWIN__) && !defined(__CYGWIN32__)
 473       fdflags = O_RDWR|O_BINARY|O_CREAT;
 474 #else
 475       fdflags = O_WRONLY|O_CREAT;
 476 #endif
 477       if (force_overwrite == 0) {
 478         fdflags |= O_EXCL;
 479       }
 480       fd = open(file_name,fdflags,0666);
 481       if (fd == -1 || (rrd_file = fdopen(fd,"wb")) == NULL) {
 482         rrd_set_error("creating '%s': %s",file_name,rrd_strerror(errno));
 483         if (fd != -1)
 484           close(fd);
 485         return(-1);
 486       }
 487     }
 488     fwrite(rrd->stat_head,
 489            sizeof(stat_head_t), 1, rrd_file);
 490
 491     fwrite(rrd->ds_def,
 492            sizeof(ds_def_t), rrd->stat_head->ds_cnt, rrd_file);
 493
 494     fwrite(rrd->rra_def,
 495            sizeof(rra_def_t), rrd->stat_head->rra_cnt, rrd_file);
 496
 497     fwrite(rrd->live_head, sizeof(live_head_t),1, rrd_file);
 498
 499     fwrite( rrd->pdp_prep, sizeof(pdp_prep_t),rrd->stat_head->ds_cnt,rrd_file);
 500
 501     fwrite( rrd->cdp_prep, sizeof(cdp_prep_t),rrd->stat_head->rra_cnt*
 502             rrd->stat_head->ds_cnt,rrd_file);
 503
 504     for(i=0; i < rrd->stat_head->rra_cnt; i++)
 505       rrd->rra_ptr[i].cur_row = rra_random_row(&rrd->rra_def[i]);
 506
 507     fwrite( rrd->rra_ptr, sizeof(rra_ptr_t), rrd->stat_head->rra_cnt,rrd_file);
 508
 509
 510
 511     /* Dump RRD values */
 512     rra_offset=0;
 513     for(i=0; i <  rrd->stat_head->rra_cnt; i++)
 514     {
 515         unsigned long num_rows = rrd->rra_def[i].row_cnt;
 516         unsigned long cur_row = rrd->rra_ptr[i].cur_row;
 517         unsigned long ds_cnt = rrd->stat_head->ds_cnt;
 518
 519         fwrite(rrd->rrd_value + (rra_offset + num_rows-1 - cur_row) * ds_cnt,
 520                sizeof(rrd_value_t), (cur_row+1)*ds_cnt, rrd_file);
 521
 522         fwrite(rrd->rrd_value + rra_offset * ds_cnt,
 523                sizeof(rrd_value_t), (num_rows-1 - cur_row)*ds_cnt, rrd_file);
 524
 525         rra_offset += num_rows;
 526     }
 527
 528     /* lets see if we had an error */
 529     if(ferror(rrd_file)){
 530         rrd_set_error("a file error occurred while creating '%s'",file_name);
 531         fclose(rrd_file);
 532         return(-1);
 533     }
 534
 535     fclose(rrd_file);
 536     return 0;
 537 }
 538
 539
 540 int
 541 rrd_restore(int argc, char **argv)
 542 {
 543     rrd_t          rrd;
 544     char          *buf;
 545         char                    rc = 0;
 546         char                    force_overwrite = 0;
 547
 548     /* init rrd clean */
 549     optind = 0; opterr = 0;  /* initialize getopt */
 550         while (1) {
 551                 static struct option long_options[] =
 552                 {
 553                         {"range-check",      no_argument, 0,  'r'},
 554                         {"force-overwrite",  no_argument, 0,  'f'},
 555                         {0,0,0,0}
 556                 };
 557                 int option_index = 0;
 558                 int opt;
 559
 560
 561                 opt = getopt_long(argc, argv, "rf", long_options, &option_index);
 562
 563                 if (opt == EOF)
 564                         break;
 565
 566                 switch(opt) {
 567                 case 'r':
 568                         rc=1;
 569                         break;
 570                 case 'f':
 571                         force_overwrite=1;
 572                         break;
 573                 default:
 574                         rrd_set_error("usage rrdtool %s [--range-check|-r] [--force-overwrite/-f]  file.xml file.rrd",argv[0]);
 575                         return -1;
 576                         break;
 577                 }
 578     }
 579
 580     if (argc-optind != 2) {
 581                 rrd_set_error("usage rrdtool %s [--range-check/-r] [--force-overwrite/-f] file.xml file.rrd",argv[0]);
 582                 return -1;
 583     }
 584
 585     if (readfile(argv[optind],&buf,0)==-1){
 586       return -1;
 587     }
 588
 589     rrd_init(&rrd);
 590
 591     if (xml2rrd(buf,&rrd,rc)==-1) {
 592         rrd_free(&rrd);
 593         free(buf);
 594         return -1;
 595     }
 596
 597     free(buf);
 598
 599     if(rrd_write(argv[optind+1],&rrd,force_overwrite)==-1){
 600         rrd_free(&rrd);
 601         return -1;
 602     };
 603     rrd_free(&rrd);
 604     return 0;
 605 }
 606
 607 /* a backwards compatibility routine that will parse the RRA params section
 608  * generated by the aberrant patch to 1.0.28. */
 609
 610 void
 611 parse_patch1028_RRA_params(char **buf, rrd_t *rrd, int rra_index)
 612 {
 613    int i;
 614    for (i = 0; i < MAX_RRA_PAR_EN; i++)
 615    {
 616    if (i == RRA_dependent_rra_idx ||
 617        i == RRA_seasonal_smooth_idx ||
 618        i == RRA_failure_threshold)
 619       read_tag(buf, "value","%lu",
 620          &(rrd->rra_def[rra_index].par[i].u_cnt));
 621    else
 622       read_tag(buf, "value","%lf",
 623          &(rrd->rra_def[rra_index].par[i].u_val));
 624    }
 625 }
 626
 627 /* a backwards compatibility routine that will parse the CDP params section
 628  * generated by the aberrant patch to 1.0.28. */
 629 void
 630 parse_patch1028_CDP_params(char **buf, rrd_t *rrd, int rra_index, int ds_index)
 631 {
 632    int ii;
 633    for (ii = 0; ii < MAX_CDP_PAR_EN; ii++)
 634    {
 635    if (cf_conv(rrd->rra_def[rra_index].cf_nam) == CF_FAILURES ||
 636        ii == CDP_unkn_pdp_cnt ||
 637        ii == CDP_null_count ||
 638        ii == CDP_last_null_count)
 639    {
 640       read_tag(buf,"value","%lu",
 641        &(rrd->cdp_prep[rrd->stat_head->ds_cnt*(rra_index) + ds_index].scratch[ii].u_cnt));
 642    } else {
 643       read_tag(buf,"value","%lf",&(rrd->cdp_prep[rrd->stat_head->ds_cnt*
 644        (rra_index) + ds_index].scratch[ii].u_val));
 645    }
 646    }
 647 }
 648
 649 void
 650 parse_FAILURES_history(char **buf, rrd_t *rrd, int rra_index, int ds_index)
 651 {
 652    char history[MAX_FAILURES_WINDOW_LEN + 1];
 653    char *violations_array;
 654    unsigned short i;
 655
 656    /* 28 = MAX_FAILURES_WINDOW_LEN */
 657    read_tag(buf, "history", "%28[0-1]", history);
 658    violations_array = (char*) rrd -> cdp_prep[rrd->stat_head->ds_cnt*(rra_index)
 659       + ds_index].scratch;
 660
 661    for (i = 0; i < rrd -> rra_def[rra_index].par[RRA_window_len].u_cnt; ++i)
 662       violations_array[i] = (history[i] == '1') ? 1 : 0;
 663
 664 }