Fix for #167 - rrdcreate is arguably missing a check for 'step>=1' for RRAs with...
[rrdtool.git] / src / rrd_create.c
1 /*****************************************************************************
2  * RRDtool 1.3.0  Copyright by Tobi Oetiker, 1997-2008
3  *****************************************************************************
4  * rrd_create.c  creates new rrds
5  *****************************************************************************/
6
7 #include <stdlib.h>
8 #include <time.h>
9 #include <locale.h>
10
11 #include "rrd_tool.h"
12 #include "rrd_rpncalc.h"
13 #include "rrd_hw.h"
14
15 #include "rrd_is_thread_safe.h"
16
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 *);
29
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;
47
48     optind = 0;
49     opterr = 0;         /* initialize getopt */
50
51     while (1) {
52         opt = getopt_long(argc, argv, "b:s:", long_options, &option_index);
53
54         if (opt == EOF)
55             break;
56
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             }
69
70             last_up = mktime(&last_up_tv.tm) +last_up_tv.offset;
71
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;
78
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;
87
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));
103
104     return rc;
105 }
106
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;
122
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     }
131
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     }
139
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 */
147
148     /* a default value */
149     rrd.ds_def = NULL;
150     rrd.rra_def = NULL;
151
152     rrd.live_head->last_up = last_up;
153
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;
161
162         if (strncmp(argv[i], "DS:", 3) == 0) {
163             size_t    old_size = sizeof(ds_def_t) * (rrd.stat_head->ds_cnt);
164
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             }
206
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             }
224
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;
236
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));
247
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     }
540
541
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     }
548
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 }
557
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;
565
566     /*
567        int temp;
568
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);
581
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);
586
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 }
600
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;
610
611     /* save index to CF_HWPREDICT */
612     unsigned long hw_index = rrd->stat_head->rra_cnt;
613
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));
627
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;
639
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;
650
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 */
658
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 }
672
673 /* create and empty rrd file according to the specs given */
674
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
689
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     }
696
697     write(rrd_file, rrd->stat_head, sizeof(stat_head_t));
698
699     write(rrd_file, rrd->ds_def, sizeof(ds_def_t) * rrd->stat_head->ds_cnt);
700
701     write(rrd_file, rrd->rra_def,
702           sizeof(rra_def_t) * rrd->stat_head->rra_cnt);
703
704     write(rrd_file, rrd->live_head, sizeof(live_head_t));
705
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     }
713
714     strcpy(rrd->pdp_prep->last_ds, "U");
715
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;
719
720     for (i = 0; i < rrd->stat_head->ds_cnt; i++)
721         write(rrd_file, rrd->pdp_prep, sizeof(pdp_prep_t));
722
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     }
730
731
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         }
762
763         for (ii = 0; ii < rrd->stat_head->ds_cnt; ii++) {
764             write(rrd_file, rrd->cdp_prep, sizeof(cdp_prep_t));
765         }
766     }
767
768     /* now, we must make sure that the rest of the rrd
769        struct is properly initialized */
770
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     }
778
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     }
787
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;
798
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;
802
803     while (unkn_cnt > 0) {
804         write(rrd_file, unknown, sizeof(rrd_value_t) * min(unkn_cnt, 512));
805
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 }
823
824 static int rand_init = 0;
825
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     }
833
834     return random() % rra->row_cnt;
835 }