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