src/rrd_create.c

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