f221c99d535491fa511bbfa0eae2b55b13834aff
1 /*****************************************************************************
2 * RRDtool 1.4.3 Copyright by Tobi Oetiker, 1997-2010
3 *****************************************************************************
4 * rrd_create.c creates new rrds
5 *****************************************************************************/
7 #include <stdlib.h>
8 #include <time.h>
9 #include <locale.h>
11 #include "rrd_tool.h"
12 #include "rrd_rpncalc.h"
13 #include "rrd_hw.h"
14 #include "rrd_client.h"
16 #include "rrd_is_thread_safe.h"
18 #ifdef WIN32
19 # include <process.h>
20 #endif
22 unsigned long FnvHash(
23 const char *str);
24 int create_hw_contingent_rras(
25 rrd_t *rrd,
26 unsigned short period,
27 unsigned long hashed_name);
28 void parseGENERIC_DS(
29 const char *def,
30 rrd_t *rrd,
31 int ds_idx);
33 static void rrd_free2(
34 rrd_t *rrd); /* our onwn copy, immmune to mmap */
36 int rrd_create(
37 int argc,
38 char **argv)
39 {
40 struct option long_options[] = {
41 {"start", required_argument, 0, 'b'},
42 {"step", required_argument, 0, 's'},
43 {"daemon", required_argument, 0, 'd'},
44 {"no-overwrite", no_argument, 0, 'O'},
45 {0, 0, 0, 0}
46 };
47 int option_index = 0;
48 int opt;
49 time_t last_up = time(NULL) - 10;
50 unsigned long pdp_step = 300;
51 rrd_time_value_t last_up_tv;
52 char *parsetime_error = NULL;
53 long long_tmp;
54 int rc;
55 char * opt_daemon = NULL;
56 int opt_no_overwrite = 0;
58 optind = 0;
59 opterr = 0; /* initialize getopt */
61 while (1) {
62 opt = getopt_long(argc, argv, "Ob:s:d:", long_options, &option_index);
64 if (opt == EOF)
65 break;
67 switch (opt) {
68 case 'd':
69 if (opt_daemon != NULL)
70 free (opt_daemon);
71 opt_daemon = strdup (optarg);
72 if (opt_daemon == NULL)
73 {
74 rrd_set_error ("strdup failed.");
75 return (-1);
76 }
77 break;
79 case 'b':
80 if ((parsetime_error = rrd_parsetime(optarg, &last_up_tv))) {
81 rrd_set_error("start time: %s", parsetime_error);
82 return (-1);
83 }
84 if (last_up_tv.type == RELATIVE_TO_END_TIME ||
85 last_up_tv.type == RELATIVE_TO_START_TIME) {
86 rrd_set_error("specifying time relative to the 'start' "
87 "or 'end' makes no sense here");
88 return (-1);
89 }
91 last_up = mktime(&last_up_tv.tm) +last_up_tv.offset;
93 if (last_up < 3600 * 24 * 365 * 10) {
94 rrd_set_error
95 ("the first entry to the RRD should be after 1980");
96 return (-1);
97 }
98 break;
100 case 's':
101 long_tmp = atol(optarg);
102 if (long_tmp < 1) {
103 rrd_set_error("step size should be no less than one second");
104 return (-1);
105 }
106 pdp_step = long_tmp;
107 break;
109 case 'O':
110 opt_no_overwrite = 1;
111 break;
113 case '?':
114 if (optopt != 0)
115 rrd_set_error("unknown option '%c'", optopt);
116 else
117 rrd_set_error("unknown option '%s'", argv[optind - 1]);
118 return (-1);
119 }
120 }
121 if (optind == argc) {
122 rrd_set_error("need name of an rrd file to create");
123 return -1;
124 }
126 rrdc_connect (opt_daemon);
127 if (rrdc_is_connected (opt_daemon)) {
128 rc = rrdc_create (argv[optind],
129 pdp_step, last_up, opt_no_overwrite,
130 argc - optind - 1, (const char **) (argv + optind + 1));
131 } else {
132 rc = rrd_create_r2(argv[optind],
133 pdp_step, last_up, opt_no_overwrite,
134 argc - optind - 1, (const char **) (argv + optind + 1));
135 }
137 return rc;
138 }
140 /* #define DEBUG */
141 /* For backwards compatibility with previous API. Use rrd_create_r2 if you
142 need to have the no_overwrite parameter. */
143 int rrd_create_r(
144 const char *filename,
145 unsigned long pdp_step,
146 time_t last_up,
147 int argc,
148 const char **argv)
149 {
150 return rrd_create_r2(filename,pdp_step,last_up,0,argc,argv);
151 }
152 int rrd_create_r2(
153 const char *filename,
154 unsigned long pdp_step,
155 time_t last_up,
156 int no_overwrite,
157 int argc,
158 const char **argv)
159 {
160 rrd_t rrd;
161 long i;
162 int offset;
163 char *token;
164 char dummychar1[2], dummychar2[2];
165 unsigned short token_idx, error_flag, period = 0;
166 unsigned long hashed_name;
168 /* clear any previous errors */
169 rrd_clear_error();
171 /* init rrd clean */
172 rrd_init(&rrd);
173 /* static header */
174 if ((rrd.stat_head = (stat_head_t*)calloc(1, sizeof(stat_head_t))) == NULL) {
175 rrd_set_error("allocating rrd.stat_head");
176 rrd_free2(&rrd);
177 return (-1);
178 }
180 /* live header */
181 if ((rrd.live_head = (live_head_t*)calloc(1, sizeof(live_head_t))) == NULL) {
182 rrd_set_error("allocating rrd.live_head");
183 rrd_free2(&rrd);
184 return (-1);
185 }
187 /* set some defaults */
188 strcpy(rrd.stat_head->cookie, RRD_COOKIE);
189 strcpy(rrd.stat_head->version, RRD_VERSION3); /* by default we are still version 3 */
190 rrd.stat_head->float_cookie = FLOAT_COOKIE;
191 rrd.stat_head->ds_cnt = 0; /* this will be adjusted later */
192 rrd.stat_head->rra_cnt = 0; /* ditto */
193 rrd.stat_head->pdp_step = pdp_step; /* 5 minute default */
195 /* a default value */
196 rrd.ds_def = NULL;
197 rrd.rra_def = NULL;
199 rrd.live_head->last_up = last_up;
201 /* optind points to the first non-option command line arg,
202 * in this case, the file name. */
203 /* Compute the FNV hash value (used by SEASONAL and DEVSEASONAL
204 * arrays. */
205 hashed_name = FnvHash(filename);
206 for (i = 0; i < argc; i++) {
207 unsigned int ii;
209 if (strncmp(argv[i], "DS:", 3) == 0) {
210 size_t old_size = sizeof(ds_def_t) * (rrd.stat_head->ds_cnt);
212 if ((rrd.ds_def = (ds_def_t*)rrd_realloc(rrd.ds_def,
213 old_size + sizeof(ds_def_t))) ==
214 NULL) {
215 rrd_set_error("allocating rrd.ds_def");
216 rrd_free2(&rrd);
217 return (-1);
218 }
219 memset(&rrd.ds_def[rrd.stat_head->ds_cnt], 0, sizeof(ds_def_t));
220 /* extract the name and type */
221 switch (sscanf(&argv[i][3],
222 DS_NAM_FMT "%1[:]" DST_FMT "%1[:]%n",
223 rrd.ds_def[rrd.stat_head->ds_cnt].ds_nam,
224 dummychar1,
225 rrd.ds_def[rrd.stat_head->ds_cnt].dst,
226 dummychar2, &offset)) {
227 case 0:
228 case 1:
229 rrd_set_error("Invalid DS name in [%s]",&argv[i][3]);
230 break;
231 case 2:
232 case 3:
233 rrd_set_error("Invalid DS type in [%s]",&argv[i][3]);
234 break;
235 case 4: /* (%n may or may not be counted) */
236 case 5: /* check for duplicate datasource names */
237 for (ii = 0; ii < rrd.stat_head->ds_cnt; ii++)
238 if (strcmp(rrd.ds_def[rrd.stat_head->ds_cnt].ds_nam,
239 rrd.ds_def[ii].ds_nam) == 0)
240 rrd_set_error("Duplicate DS name: %s",
241 rrd.ds_def[ii].ds_nam);
242 /* DS_type may be valid or not. Checked later */
243 break;
244 default:
245 rrd_set_error("invalid DS format");
246 }
247 if (rrd_test_error()) {
248 rrd_free2(&rrd);
249 return -1;
250 }
252 /* parse the remainder of the arguments */
253 switch (dst_conv(rrd.ds_def[rrd.stat_head->ds_cnt].dst)) {
254 case DST_COUNTER:
255 case DST_ABSOLUTE:
256 case DST_GAUGE:
257 case DST_DERIVE:
258 parseGENERIC_DS(&argv[i][offset + 3], &rrd,
259 rrd.stat_head->ds_cnt);
260 break;
261 case DST_CDEF:
262 parseCDEF_DS(&argv[i][offset + 3], &rrd,
263 rrd.stat_head->ds_cnt);
264 break;
265 default:
266 rrd_set_error("invalid DS type specified");
267 break;
268 }
270 if (rrd_test_error()) {
271 rrd_free2(&rrd);
272 return -1;
273 }
274 rrd.stat_head->ds_cnt++;
275 } else if (strncmp(argv[i], "RRA:", 4) == 0) {
276 char *argvcopy;
277 char *tokptr = "";
278 size_t old_size = sizeof(rra_def_t) * (rrd.stat_head->rra_cnt);
279 int row_cnt;
281 if ((rrd.rra_def = (rra_def_t*)rrd_realloc(rrd.rra_def,
282 old_size + sizeof(rra_def_t))) ==
283 NULL) {
284 rrd_set_error("allocating rrd.rra_def");
285 rrd_free2(&rrd);
286 return (-1);
287 }
288 memset(&rrd.rra_def[rrd.stat_head->rra_cnt], 0,
289 sizeof(rra_def_t));
291 argvcopy = strdup(argv[i]);
292 token = strtok_r(&argvcopy[4], ":", &tokptr);
293 token_idx = error_flag = 0;
294 while (token != NULL) {
295 switch (token_idx) {
296 case 0:
297 if (sscanf(token, CF_NAM_FMT,
298 rrd.rra_def[rrd.stat_head->rra_cnt].cf_nam) !=
299 1)
300 rrd_set_error("Failed to parse CF name");
301 switch (cf_conv
302 (rrd.rra_def[rrd.stat_head->rra_cnt].cf_nam)) {
303 case CF_MHWPREDICT:
304 strcpy(rrd.stat_head->version, RRD_VERSION); /* MHWPREDICT causes Version 4 */
305 case CF_HWPREDICT:
306 /* initialize some parameters */
307 rrd.rra_def[rrd.stat_head->rra_cnt].par[RRA_hw_alpha].
308 u_val = 0.1;
309 rrd.rra_def[rrd.stat_head->rra_cnt].par[RRA_hw_beta].
310 u_val = 1.0 / 288;
311 rrd.rra_def[rrd.stat_head->rra_cnt].
312 par[RRA_dependent_rra_idx].u_cnt =
313 rrd.stat_head->rra_cnt;
314 break;
315 case CF_DEVSEASONAL:
316 case CF_SEASONAL:
317 /* initialize some parameters */
318 rrd.rra_def[rrd.stat_head->rra_cnt].
319 par[RRA_seasonal_gamma].u_val = 0.1;
320 rrd.rra_def[rrd.stat_head->rra_cnt].
321 par[RRA_seasonal_smoothing_window].u_val = 0.05;
322 /* fall through */
323 case CF_DEVPREDICT:
324 rrd.rra_def[rrd.stat_head->rra_cnt].
325 par[RRA_dependent_rra_idx].u_cnt = -1;
326 break;
327 case CF_FAILURES:
328 rrd.rra_def[rrd.stat_head->rra_cnt].
329 par[RRA_delta_pos].u_val = 2.0;
330 rrd.rra_def[rrd.stat_head->rra_cnt].
331 par[RRA_delta_neg].u_val = 2.0;
332 rrd.rra_def[rrd.stat_head->rra_cnt].
333 par[RRA_window_len].u_cnt = 3;
334 rrd.rra_def[rrd.stat_head->rra_cnt].
335 par[RRA_failure_threshold].u_cnt = 2;
336 rrd.rra_def[rrd.stat_head->rra_cnt].
337 par[RRA_dependent_rra_idx].u_cnt = -1;
338 break;
339 /* invalid consolidation function */
340 case -1:
341 rrd_set_error
342 ("Unrecognized consolidation function %s",
343 rrd.rra_def[rrd.stat_head->rra_cnt].cf_nam);
344 default:
345 break;
346 }
347 /* default: 1 pdp per cdp */
348 rrd.rra_def[rrd.stat_head->rra_cnt].pdp_cnt = 1;
349 break;
350 case 1:
351 switch (cf_conv
352 (rrd.rra_def[rrd.stat_head->rra_cnt].cf_nam)) {
353 case CF_HWPREDICT:
354 case CF_MHWPREDICT:
355 case CF_DEVSEASONAL:
356 case CF_SEASONAL:
357 case CF_DEVPREDICT:
358 case CF_FAILURES:
359 row_cnt = atoi(token);
360 if (row_cnt <= 0)
361 rrd_set_error("Invalid row count: %i", row_cnt);
362 rrd.rra_def[rrd.stat_head->rra_cnt].row_cnt = row_cnt;
363 break;
364 default:
365 rrd.rra_def[rrd.stat_head->rra_cnt].
366 par[RRA_cdp_xff_val].u_val = atof(token);
367 if (rrd.rra_def[rrd.stat_head->rra_cnt].
368 par[RRA_cdp_xff_val].u_val < 0.0
369 || rrd.rra_def[rrd.stat_head->rra_cnt].
370 par[RRA_cdp_xff_val].u_val >= 1.0)
371 rrd_set_error
372 ("Invalid xff: must be between 0 and 1");
373 break;
374 }
375 break;
376 case 2:
377 switch (cf_conv
378 (rrd.rra_def[rrd.stat_head->rra_cnt].cf_nam)) {
379 case CF_HWPREDICT:
380 case CF_MHWPREDICT:
381 rrd.rra_def[rrd.stat_head->rra_cnt].par[RRA_hw_alpha].
382 u_val = atof(token);
383 if (atof(token) <= 0.0 || atof(token) >= 1.0)
384 rrd_set_error
385 ("Invalid alpha: must be between 0 and 1");
386 break;
387 case CF_DEVSEASONAL:
388 case CF_SEASONAL:
389 rrd.rra_def[rrd.stat_head->rra_cnt].
390 par[RRA_seasonal_gamma].u_val = atof(token);
391 if (atof(token) <= 0.0 || atof(token) >= 1.0)
392 rrd_set_error
393 ("Invalid gamma: must be between 0 and 1");
394 rrd.rra_def[rrd.stat_head->rra_cnt].
395 par[RRA_seasonal_smooth_idx].u_cnt =
396 hashed_name %
397 rrd.rra_def[rrd.stat_head->rra_cnt].row_cnt;
398 break;
399 case CF_FAILURES:
400 /* specifies the # of violations that constitutes the failure threshold */
401 rrd.rra_def[rrd.stat_head->rra_cnt].
402 par[RRA_failure_threshold].u_cnt = atoi(token);
403 if (atoi(token) < 1
404 || atoi(token) > MAX_FAILURES_WINDOW_LEN)
405 rrd_set_error
406 ("Failure threshold is out of range %d, %d",
407 1, MAX_FAILURES_WINDOW_LEN);
408 break;
409 case CF_DEVPREDICT:
410 /* specifies the index (1-based) of CF_DEVSEASONAL array
411 * associated with this CF_DEVPREDICT array. */
412 rrd.rra_def[rrd.stat_head->rra_cnt].
413 par[RRA_dependent_rra_idx].u_cnt =
414 atoi(token) - 1;
415 break;
416 default:
417 rrd.rra_def[rrd.stat_head->rra_cnt].pdp_cnt =
418 atoi(token);
419 if (atoi(token) < 1)
420 rrd_set_error("Invalid step: must be >= 1");
421 break;
422 }
423 break;
424 case 3:
425 switch (cf_conv
426 (rrd.rra_def[rrd.stat_head->rra_cnt].cf_nam)) {
427 case CF_HWPREDICT:
428 case CF_MHWPREDICT:
429 rrd.rra_def[rrd.stat_head->rra_cnt].par[RRA_hw_beta].
430 u_val = atof(token);
431 if (atof(token) < 0.0 || atof(token) > 1.0)
432 rrd_set_error
433 ("Invalid beta: must be between 0 and 1");
434 break;
435 case CF_DEVSEASONAL:
436 case CF_SEASONAL:
437 /* specifies the index (1-based) of CF_HWPREDICT array
438 * associated with this CF_DEVSEASONAL or CF_SEASONAL array.
439 * */
440 rrd.rra_def[rrd.stat_head->rra_cnt].
441 par[RRA_dependent_rra_idx].u_cnt =
442 atoi(token) - 1;
443 break;
444 case CF_FAILURES:
445 /* specifies the window length */
446 rrd.rra_def[rrd.stat_head->rra_cnt].
447 par[RRA_window_len].u_cnt = atoi(token);
448 if (atoi(token) < 1
449 || atoi(token) > MAX_FAILURES_WINDOW_LEN)
450 rrd_set_error
451 ("Window length is out of range %d, %d", 1,
452 MAX_FAILURES_WINDOW_LEN);
453 /* verify that window length exceeds the failure threshold */
454 if (rrd.rra_def[rrd.stat_head->rra_cnt].
455 par[RRA_window_len].u_cnt <
456 rrd.rra_def[rrd.stat_head->rra_cnt].
457 par[RRA_failure_threshold].u_cnt)
458 rrd_set_error
459 ("Window length is shorter than the failure threshold");
460 break;
461 case CF_DEVPREDICT:
462 /* shouldn't be any more arguments */
463 rrd_set_error
464 ("Unexpected extra argument for consolidation function DEVPREDICT");
465 break;
466 default:
467 row_cnt = atoi(token);
468 if (row_cnt <= 0)
469 rrd_set_error("Invalid row count: %i", row_cnt);
470 rrd.rra_def[rrd.stat_head->rra_cnt].row_cnt = row_cnt;
471 break;
472 }
473 break;
474 case 4:
475 switch (cf_conv
476 (rrd.rra_def[rrd.stat_head->rra_cnt].cf_nam)) {
477 case CF_FAILURES:
478 /* specifies the index (1-based) of CF_DEVSEASONAL array
479 * associated with this CF_DEVFAILURES array. */
480 rrd.rra_def[rrd.stat_head->rra_cnt].
481 par[RRA_dependent_rra_idx].u_cnt =
482 atoi(token) - 1;
483 break;
484 case CF_DEVSEASONAL:
485 case CF_SEASONAL:
486 /* optional smoothing window */
487 if (sscanf(token, "smoothing-window=%lf",
488 &(rrd.rra_def[rrd.stat_head->rra_cnt].
489 par[RRA_seasonal_smoothing_window].
490 u_val))) {
491 strcpy(rrd.stat_head->version, RRD_VERSION); /* smoothing-window causes Version 4 */
492 if (rrd.rra_def[rrd.stat_head->rra_cnt].
493 par[RRA_seasonal_smoothing_window].u_val < 0.0
494 || rrd.rra_def[rrd.stat_head->rra_cnt].
495 par[RRA_seasonal_smoothing_window].u_val >
496 1.0) {
497 rrd_set_error
498 ("Invalid smoothing-window %f: must be between 0 and 1",
499 rrd.rra_def[rrd.stat_head->rra_cnt].
500 par[RRA_seasonal_smoothing_window].
501 u_val);
502 }
503 } else {
504 rrd_set_error("Invalid option %s", token);
505 }
506 break;
507 case CF_HWPREDICT:
508 case CF_MHWPREDICT:
509 /* length of the associated CF_SEASONAL and CF_DEVSEASONAL arrays. */
510 period = atoi(token);
511 if (period >
512 rrd.rra_def[rrd.stat_head->rra_cnt].row_cnt)
513 rrd_set_error
514 ("Length of seasonal cycle exceeds length of HW prediction array");
515 break;
516 default:
517 /* shouldn't be any more arguments */
518 rrd_set_error
519 ("Unexpected extra argument for consolidation function %s",
520 rrd.rra_def[rrd.stat_head->rra_cnt].cf_nam);
521 break;
522 }
523 break;
524 case 5:
525 /* If we are here, this must be a CF_HWPREDICT RRA.
526 * Specifies the index (1-based) of CF_SEASONAL array
527 * associated with this CF_HWPREDICT array. If this argument
528 * is missing, then the CF_SEASONAL, CF_DEVSEASONAL, CF_DEVPREDICT,
529 * CF_FAILURES.
530 * arrays are created automatically. */
531 rrd.rra_def[rrd.stat_head->rra_cnt].
532 par[RRA_dependent_rra_idx].u_cnt = atoi(token) - 1;
533 break;
534 default:
535 /* should never get here */
536 rrd_set_error("Unknown error");
537 break;
538 } /* end switch */
539 if (rrd_test_error()) {
540 /* all errors are unrecoverable */
541 free(argvcopy);
542 rrd_free2(&rrd);
543 return (-1);
544 }
545 token = strtok_r(NULL, ":", &tokptr);
546 token_idx++;
547 } /* end while */
548 free(argvcopy);
549 #ifdef DEBUG
550 fprintf(stderr,
551 "Creating RRA CF: %s, dep idx %lu, current idx %lu\n",
552 rrd.rra_def[rrd.stat_head->rra_cnt].cf_nam,
553 rrd.rra_def[rrd.stat_head->rra_cnt].
554 par[RRA_dependent_rra_idx].u_cnt, rrd.stat_head->rra_cnt);
555 #endif
556 /* should we create CF_SEASONAL, CF_DEVSEASONAL, and CF_DEVPREDICT? */
557 if ((cf_conv(rrd.rra_def[rrd.stat_head->rra_cnt].cf_nam) ==
558 CF_HWPREDICT
559 || cf_conv(rrd.rra_def[rrd.stat_head->rra_cnt].cf_nam) ==
560 CF_MHWPREDICT)
561 && rrd.rra_def[rrd.stat_head->rra_cnt].
562 par[RRA_dependent_rra_idx].u_cnt == rrd.stat_head->rra_cnt) {
563 #ifdef DEBUG
564 fprintf(stderr, "Creating HW contingent RRAs\n");
565 #endif
566 if (create_hw_contingent_rras(&rrd, period, hashed_name) ==
567 -1) {
568 rrd_set_error("creating contingent RRA");
569 rrd_free2(&rrd);
570 return -1;
571 }
572 }
573 rrd.stat_head->rra_cnt++;
574 } else {
575 rrd_set_error("can't parse argument '%s'", argv[i]);
576 rrd_free2(&rrd);
577 return -1;
578 }
579 }
582 if (rrd.stat_head->rra_cnt < 1) {
583 rrd_set_error("you must define at least one Round Robin Archive");
584 rrd_free2(&rrd);
585 return (-1);
586 }
588 if (rrd.stat_head->ds_cnt < 1) {
589 rrd_set_error("you must define at least one Data Source");
590 rrd_free2(&rrd);
591 return (-1);
592 }
593 return rrd_create_fn(filename, &rrd, no_overwrite);
594 }
596 void parseGENERIC_DS(
597 const char *def,
598 rrd_t *rrd,
599 int ds_idx)
600 {
601 char minstr[DS_NAM_SIZE], maxstr[DS_NAM_SIZE];
602 char *old_locale;
604 /*
605 int temp;
607 temp = sscanf(def,"%lu:%18[^:]:%18[^:]",
608 &(rrd -> ds_def[ds_idx].par[DS_mrhb_cnt].u_cnt),
609 minstr,maxstr);
610 */
611 old_locale = setlocale(LC_NUMERIC, NULL);
612 setlocale(LC_NUMERIC, "C");
613 if (sscanf(def, "%lu:%18[^:]:%18[^:]",
614 &(rrd->ds_def[ds_idx].par[DS_mrhb_cnt].u_cnt),
615 minstr, maxstr) == 3) {
616 if (minstr[0] == 'U' && minstr[1] == 0)
617 rrd->ds_def[ds_idx].par[DS_min_val].u_val = DNAN;
618 else
619 rrd->ds_def[ds_idx].par[DS_min_val].u_val = atof(minstr);
621 if (maxstr[0] == 'U' && maxstr[1] == 0)
622 rrd->ds_def[ds_idx].par[DS_max_val].u_val = DNAN;
623 else
624 rrd->ds_def[ds_idx].par[DS_max_val].u_val = atof(maxstr);
626 if (!isnan(rrd->ds_def[ds_idx].par[DS_min_val].u_val) &&
627 !isnan(rrd->ds_def[ds_idx].par[DS_max_val].u_val) &&
628 rrd->ds_def[ds_idx].par[DS_min_val].u_val
629 >= rrd->ds_def[ds_idx].par[DS_max_val].u_val) {
630 rrd_set_error("min must be less than max in DS definition");
631 setlocale(LC_NUMERIC, old_locale);
632 return;
633 }
634 } else {
635 rrd_set_error("failed to parse data source %s", def);
636 }
637 setlocale(LC_NUMERIC, old_locale);
638 }
640 /* Create the CF_DEVPREDICT, CF_DEVSEASONAL, CF_SEASONAL, and CF_FAILURES RRAs
641 * associated with a CF_HWPREDICT RRA. */
642 int create_hw_contingent_rras(
643 rrd_t *rrd,
644 unsigned short period,
645 unsigned long hashed_name)
646 {
647 size_t old_size;
648 rra_def_t *current_rra;
650 /* save index to CF_HWPREDICT */
651 unsigned long hw_index = rrd->stat_head->rra_cnt;
653 /* advance the pointer */
654 (rrd->stat_head->rra_cnt)++;
655 /* allocate the memory for the 4 contingent RRAs */
656 old_size = sizeof(rra_def_t) * (rrd->stat_head->rra_cnt);
657 if ((rrd->rra_def = (rra_def_t*)rrd_realloc(rrd->rra_def,
658 old_size + 4 * sizeof(rra_def_t))) ==
659 NULL) {
660 rrd_free2(rrd);
661 rrd_set_error("allocating rrd.rra_def");
662 return (-1);
663 }
664 /* clear memory */
665 memset(&(rrd->rra_def[rrd->stat_head->rra_cnt]), 0,
666 4 * sizeof(rra_def_t));
668 /* create the CF_SEASONAL RRA */
669 current_rra = &(rrd->rra_def[rrd->stat_head->rra_cnt]);
670 strcpy(current_rra->cf_nam, "SEASONAL");
671 current_rra->row_cnt = period;
672 current_rra->par[RRA_seasonal_smooth_idx].u_cnt = hashed_name % period;
673 current_rra->pdp_cnt = 1;
674 current_rra->par[RRA_seasonal_gamma].u_val =
675 rrd->rra_def[hw_index].par[RRA_hw_alpha].u_val;
676 current_rra->par[RRA_dependent_rra_idx].u_cnt = hw_index;
677 rrd->rra_def[hw_index].par[RRA_dependent_rra_idx].u_cnt =
678 rrd->stat_head->rra_cnt;
680 /* create the CF_DEVSEASONAL RRA */
681 (rrd->stat_head->rra_cnt)++;
682 current_rra = &(rrd->rra_def[rrd->stat_head->rra_cnt]);
683 strcpy(current_rra->cf_nam, "DEVSEASONAL");
684 current_rra->row_cnt = period;
685 current_rra->par[RRA_seasonal_smooth_idx].u_cnt = hashed_name % period;
686 current_rra->pdp_cnt = 1;
687 current_rra->par[RRA_seasonal_gamma].u_val =
688 rrd->rra_def[hw_index].par[RRA_hw_alpha].u_val;
689 current_rra->par[RRA_dependent_rra_idx].u_cnt = hw_index;
691 /* create the CF_DEVPREDICT RRA */
692 (rrd->stat_head->rra_cnt)++;
693 current_rra = &(rrd->rra_def[rrd->stat_head->rra_cnt]);
694 strcpy(current_rra->cf_nam, "DEVPREDICT");
695 current_rra->row_cnt = (rrd->rra_def[hw_index]).row_cnt;
696 current_rra->pdp_cnt = 1;
697 current_rra->par[RRA_dependent_rra_idx].u_cnt = hw_index + 2; /* DEVSEASONAL */
699 /* create the CF_FAILURES RRA */
700 (rrd->stat_head->rra_cnt)++;
701 current_rra = &(rrd->rra_def[rrd->stat_head->rra_cnt]);
702 strcpy(current_rra->cf_nam, "FAILURES");
703 current_rra->row_cnt = period;
704 current_rra->pdp_cnt = 1;
705 current_rra->par[RRA_delta_pos].u_val = 2.0;
706 current_rra->par[RRA_delta_neg].u_val = 2.0;
707 current_rra->par[RRA_failure_threshold].u_cnt = 7;
708 current_rra->par[RRA_window_len].u_cnt = 9;
709 current_rra->par[RRA_dependent_rra_idx].u_cnt = hw_index + 2; /* DEVSEASONAL */
710 return 0;
711 }
713 /* create and empty rrd file according to the specs given */
715 int rrd_create_fn(
716 const char *file_name,
717 rrd_t *rrd,
718 int no_overwrite )
719 {
720 unsigned long i, ii;
721 rrd_value_t *unknown;
722 int unkn_cnt;
723 rrd_file_t *rrd_file_dn;
724 rrd_t rrd_dn;
725 unsigned rrd_flags = RRD_READWRITE | RRD_CREAT;
727 if (no_overwrite) {
728 rrd_flags |= RRD_EXCL ;
729 }
731 unkn_cnt = 0;
732 for (i = 0; i < rrd->stat_head->rra_cnt; i++)
733 unkn_cnt += rrd->stat_head->ds_cnt * rrd->rra_def[i].row_cnt;
735 if ((rrd_file_dn = rrd_open(file_name, rrd, rrd_flags)) == NULL) {
736 rrd_set_error("creating '%s': %s", file_name, rrd_strerror(errno));
737 rrd_free2(rrd);
738 return (-1);
739 }
741 rrd_write(rrd_file_dn, rrd->stat_head, sizeof(stat_head_t));
743 rrd_write(rrd_file_dn, rrd->ds_def, sizeof(ds_def_t) * rrd->stat_head->ds_cnt);
745 rrd_write(rrd_file_dn, rrd->rra_def,
746 sizeof(rra_def_t) * rrd->stat_head->rra_cnt);
748 rrd_write(rrd_file_dn, rrd->live_head, sizeof(live_head_t));
750 if ((rrd->pdp_prep = (pdp_prep_t*)calloc(1, sizeof(pdp_prep_t))) == NULL) {
751 rrd_set_error("allocating pdp_prep");
752 rrd_free2(rrd);
753 rrd_close(rrd_file_dn);
754 return (-1);
755 }
757 strcpy(rrd->pdp_prep->last_ds, "U");
759 rrd->pdp_prep->scratch[PDP_val].u_val = 0.0;
760 rrd->pdp_prep->scratch[PDP_unkn_sec_cnt].u_cnt =
761 rrd->live_head->last_up % rrd->stat_head->pdp_step;
763 for (i = 0; i < rrd->stat_head->ds_cnt; i++)
764 rrd_write(rrd_file_dn, rrd->pdp_prep, sizeof(pdp_prep_t));
766 if ((rrd->cdp_prep = (cdp_prep_t*)calloc(1, sizeof(cdp_prep_t))) == NULL) {
767 rrd_set_error("allocating cdp_prep");
768 rrd_free2(rrd);
769 rrd_close(rrd_file_dn);
770 return (-1);
771 }
774 for (i = 0; i < rrd->stat_head->rra_cnt; i++) {
775 switch (cf_conv(rrd->rra_def[i].cf_nam)) {
776 case CF_HWPREDICT:
777 case CF_MHWPREDICT:
778 init_hwpredict_cdp(rrd->cdp_prep);
779 break;
780 case CF_SEASONAL:
781 case CF_DEVSEASONAL:
782 init_seasonal_cdp(rrd->cdp_prep);
783 break;
784 case CF_FAILURES:
785 /* initialize violation history to 0 */
786 for (ii = 0; ii < MAX_CDP_PAR_EN; ii++) {
787 /* We can zero everything out, by setting u_val to the
788 * NULL address. Each array entry in scratch is 8 bytes
789 * (a double), but u_cnt only accessed 4 bytes (long) */
790 rrd->cdp_prep->scratch[ii].u_val = 0.0;
791 }
792 break;
793 default:
794 /* can not be zero because we don't know anything ... */
795 rrd->cdp_prep->scratch[CDP_val].u_val = DNAN;
796 /* startup missing pdp count */
797 rrd->cdp_prep->scratch[CDP_unkn_pdp_cnt].u_cnt =
798 ((rrd->live_head->last_up -
799 rrd->pdp_prep->scratch[PDP_unkn_sec_cnt].u_cnt)
800 % (rrd->stat_head->pdp_step
801 * rrd->rra_def[i].pdp_cnt)) / rrd->stat_head->pdp_step;
802 break;
803 }
805 for (ii = 0; ii < rrd->stat_head->ds_cnt; ii++) {
806 rrd_write(rrd_file_dn, rrd->cdp_prep, sizeof(cdp_prep_t));
807 }
808 }
810 /* now, we must make sure that the rest of the rrd
811 struct is properly initialized */
813 if ((rrd->rra_ptr = (rra_ptr_t*)calloc(1, sizeof(rra_ptr_t))) == NULL) {
814 rrd_set_error("allocating rra_ptr");
815 rrd_free2(rrd);
816 rrd_close(rrd_file_dn);
817 return (-1);
818 }
820 /* changed this initialization to be consistent with
821 * rrd_restore. With the old value (0), the first update
822 * would occur for cur_row = 1 because rrd_update increments
823 * the pointer a priori. */
824 for (i = 0; i < rrd->stat_head->rra_cnt; i++) {
825 rrd->rra_ptr->cur_row = rrd_select_initial_row(rrd_file_dn, i, &rrd->rra_def[i]);
826 rrd_write(rrd_file_dn, rrd->rra_ptr, sizeof(rra_ptr_t));
827 }
829 /* write the empty data area */
830 if ((unknown = (rrd_value_t *) malloc(512 * sizeof(rrd_value_t))) == NULL) {
831 rrd_set_error("allocating unknown");
832 rrd_free2(rrd);
833 rrd_close(rrd_file_dn);
834 return (-1);
835 }
836 for (i = 0; i < 512; ++i)
837 unknown[i] = DNAN;
839 while (unkn_cnt > 0) {
840 if(rrd_write(rrd_file_dn, unknown, sizeof(rrd_value_t) * min(unkn_cnt, 512)) < 0)
841 {
842 rrd_set_error("creating rrd: %s", rrd_strerror(errno));
843 return -1;
844 }
846 unkn_cnt -= 512;
847 }
848 free(unknown);
849 rrd_free2(rrd);
850 if (rrd_close(rrd_file_dn) == -1) {
851 rrd_set_error("creating rrd: %s", rrd_strerror(errno));
852 return -1;
853 }
854 /* flush all we don't need out of the cache */
855 rrd_init(&rrd_dn);
856 if((rrd_file_dn = rrd_open(file_name, &rrd_dn, RRD_READONLY)) != NULL)
857 {
858 rrd_dontneed(rrd_file_dn, &rrd_dn);
859 /* rrd_free(&rrd_dn); */
860 rrd_close(rrd_file_dn);
861 }
862 return (0);
863 }
866 static void rrd_free2(
867 rrd_t *rrd)
868 {
869 free(rrd->live_head);
870 free(rrd->stat_head);
871 free(rrd->ds_def);
872 free(rrd->rra_def);
873 free(rrd->rra_ptr);
874 free(rrd->pdp_prep);
875 free(rrd->cdp_prep);
876 free(rrd->rrd_value);
877 }