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