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