src/rrd_xport.c

   1 /****************************************************************************
   2  * RRDtool 1.2.23  Copyright by Tobi Oetiker, 1997-2007
   3  ****************************************************************************
   4  * rrd_xport.c  export RRD data
   5  ****************************************************************************/
   6
   7 #include <sys/stat.h>
   8
   9 #include "rrd_tool.h"
  10 #include "rrd_graph.h"
  11 #include "rrd_xport.h"
  12 #include "unused.h"
  13
  14 #if defined(_WIN32) && !defined(__CYGWIN__) && !defined(__CYGWIN32__)
  15 #include <io.h>
  16 #include <fcntl.h>
  17 #endif
  18
  19
  20 int       rrd_xport(
  21     int,
  22     char **,
  23     int *,
  24     time_t *,
  25     time_t *,
  26     unsigned long *,
  27     unsigned long *,
  28     char ***,
  29     rrd_value_t **);
  30
  31 int       rrd_xport_fn(
  32     image_desc_t *,
  33     time_t *,
  34     time_t *,
  35     unsigned long *,
  36     unsigned long *,
  37     char ***,
  38     rrd_value_t **);
  39
  40
  41
  42
  43 int rrd_xport(
  44     int argc,
  45     char **argv,
  46     int UNUSED(*xsize),
  47     time_t *start,
  48     time_t *end,        /* which time frame do you want ?
  49                          * will be changed to represent reality */
  50     unsigned long *step,    /* which stepsize do you want?
  51                              * will be changed to represent reality */
  52     unsigned long *col_cnt, /* number of data columns in the result */
  53     char ***legend_v,   /* legend entries */
  54     rrd_value_t **data)
  55 {                       /* two dimensional array containing the data */
  56
  57     image_desc_t im;
  58     time_t    start_tmp = 0, end_tmp = 0;
  59     struct rrd_time_value start_tv, end_tv;
  60     char     *parsetime_error = NULL;
  61
  62     optind = 0;
  63     opterr = 0;         /* initialize getopt */
  64
  65     rrd_graph_init(&im);
  66
  67     parsetime("end-24h", &start_tv);
  68     parsetime("now", &end_tv);
  69
  70     while (1) {
  71         static struct option long_options[] = {
  72             {"start", required_argument, 0, 's'},
  73             {"end", required_argument, 0, 'e'},
  74             {"maxrows", required_argument, 0, 'm'},
  75             {"step", required_argument, 0, 261},
  76             {"enumds", no_argument, 0, 262},    /* these are handled in the frontend ... */
  77             {0, 0, 0, 0}
  78         };
  79         int       option_index = 0;
  80         int       opt;
  81
  82         opt = getopt_long(argc, argv, "s:e:m:", long_options, &option_index);
  83
  84         if (opt == EOF)
  85             break;
  86
  87         switch (opt) {
  88         case 261:
  89             im.step = atoi(optarg);
  90             break;
  91         case 262:
  92             break;
  93         case 's':
  94             if ((parsetime_error = parsetime(optarg, &start_tv))) {
  95                 rrd_set_error("start time: %s", parsetime_error);
  96                 return -1;
  97             }
  98             break;
  99         case 'e':
 100             if ((parsetime_error = parsetime(optarg, &end_tv))) {
 101                 rrd_set_error("end time: %s", parsetime_error);
 102                 return -1;
 103             }
 104             break;
 105         case 'm':
 106             im.xsize = atol(optarg);
 107             if (im.xsize < 10) {
 108                 rrd_set_error("maxrows below 10 rows");
 109                 return -1;
 110             }
 111             break;
 112         case '?':
 113             rrd_set_error("unknown option '%s'", argv[optind - 1]);
 114             return -1;
 115         }
 116     }
 117
 118     if (proc_start_end(&start_tv, &end_tv, &start_tmp, &end_tmp) == -1) {
 119         return -1;
 120     }
 121
 122     if (start_tmp < 3600 * 24 * 365 * 10) {
 123         rrd_set_error("the first entry to fetch should be after 1980 (%ld)",
 124                       start_tmp);
 125         return -1;
 126     }
 127
 128     if (end_tmp < start_tmp) {
 129         rrd_set_error("start (%ld) should be less than end (%ld)",
 130                       start_tmp, end_tmp);
 131         return -1;
 132     }
 133
 134     im.start = start_tmp;
 135     im.end = end_tmp;
 136     im.step = max((long) im.step, (im.end - im.start) / im.xsize);
 137
 138     rrd_graph_script(argc, argv, &im, 0);
 139     if (rrd_test_error()) {
 140         im_free(&im);
 141         return -1;
 142     }
 143
 144     if (im.gdes_c == 0) {
 145         rrd_set_error("can't make a graph without contents");
 146         im_free(&im);
 147         return (-1);
 148     }
 149
 150     if (rrd_xport_fn(&im, start, end, step, col_cnt, legend_v, data) == -1) {
 151         im_free(&im);
 152         return -1;
 153     }
 154
 155     im_free(&im);
 156     return 0;
 157 }
 158
 159
 160
 161 int rrd_xport_fn(
 162     image_desc_t *im,
 163     time_t *start,
 164     time_t *end,        /* which time frame do you want ?
 165                          * will be changed to represent reality */
 166     unsigned long *step,    /* which stepsize do you want?
 167                              * will be changed to represent reality */
 168     unsigned long *col_cnt, /* number of data columns in the result */
 169     char ***legend_v,   /* legend entries */
 170     rrd_value_t **data)
 171 {                       /* two dimensional array containing the data */
 172
 173     int       i = 0, j = 0;
 174     unsigned long *ds_cnt;  /* number of data sources in file */
 175     unsigned long col, dst_row, row_cnt;
 176     rrd_value_t *srcptr, *dstptr;
 177
 178     unsigned long nof_xports = 0;
 179     unsigned long xport_counter = 0;
 180     unsigned long *ref_list;
 181     rrd_value_t **srcptr_list;
 182     char    **legend_list;
 183     int       ii = 0;
 184
 185     time_t    start_tmp = 0;
 186     time_t    end_tmp = 0;
 187     unsigned long step_tmp = 1;
 188
 189     /* pull the data from the rrd files ... */
 190     if (data_fetch(im) == -1)
 191         return -1;
 192
 193     /* evaluate CDEF  operations ... */
 194     if (data_calc(im) == -1)
 195         return -1;
 196
 197     /* how many xports? */
 198     for (i = 0; i < im->gdes_c; i++) {
 199         switch (im->gdes[i].gf) {
 200         case GF_XPORT:
 201             nof_xports++;
 202             break;
 203         default:
 204             break;
 205         }
 206     }
 207
 208     if (nof_xports == 0) {
 209         rrd_set_error("no XPORT found, nothing to do");
 210         return -1;
 211     }
 212
 213     /* a list of referenced gdes */
 214     ref_list = malloc(sizeof(int) * nof_xports);
 215     if (ref_list == NULL)
 216         return -1;
 217
 218     /* a list to save pointers into each gdes data */
 219     srcptr_list = malloc(sizeof(srcptr) * nof_xports);
 220     if (srcptr_list == NULL) {
 221         free(ref_list);
 222         return -1;
 223     }
 224
 225     /* a list to save pointers to the column's legend entry */
 226     /* this is a return value! */
 227     legend_list = malloc(sizeof(char *) * nof_xports);
 228     if (legend_list == NULL) {
 229         free(srcptr_list);
 230         free(ref_list);
 231         return -1;
 232     }
 233
 234     /* find referenced gdes and save their index and */
 235     /* a pointer into their data */
 236     for (i = 0; i < im->gdes_c; i++) {
 237         switch (im->gdes[i].gf) {
 238         case GF_XPORT:
 239             ii = im->gdes[i].vidx;
 240             if (xport_counter > nof_xports) {
 241                 rrd_set_error("too many xports: should not happen. Hmmm");
 242                 free(srcptr_list);
 243                 free(ref_list);
 244                 free(legend_list);
 245                 return -1;
 246             }
 247             srcptr_list[xport_counter] = im->gdes[ii].data;
 248             ref_list[xport_counter++] = i;
 249             break;
 250         default:
 251             break;
 252         }
 253     }
 254
 255     start_tmp = im->gdes[0].start;
 256     end_tmp = im->gdes[0].end;
 257     step_tmp = im->gdes[0].step;
 258
 259     /* fill some return values */
 260     *col_cnt = nof_xports;
 261     *start = start_tmp;
 262     *end = end_tmp;
 263     *step = step_tmp;
 264
 265     row_cnt = ((*end) - (*start)) / (*step);
 266
 267     /* room for rearranged data */
 268     /* this is a return value! */
 269     if (((*data) =
 270          malloc((*col_cnt) * row_cnt * sizeof(rrd_value_t))) == NULL) {
 271         free(srcptr_list);
 272         free(ref_list);
 273         free(legend_list);
 274         rrd_set_error("malloc xport data area");
 275         return (-1);
 276     }
 277     dstptr = (*data);
 278
 279     j = 0;
 280     for (i = 0; i < im->gdes_c; i++) {
 281         switch (im->gdes[i].gf) {
 282         case GF_XPORT:
 283             /* reserve room for one legend entry */
 284             /* is FMT_LEG_LEN + 5 the correct size? */
 285             if ((legend_list[j] =
 286                  malloc(sizeof(char) * (FMT_LEG_LEN + 5))) == NULL) {
 287                 free(srcptr_list);
 288                 free(ref_list);
 289                 free(*data);
 290                 *data = NULL;
 291                 while (--j > -1)
 292                     free(legend_list[j]);
 293                 free(legend_list);
 294                 rrd_set_error("malloc xport legend entry");
 295                 return (-1);
 296             }
 297
 298             if (im->gdes[i].legend)
 299                 /* omit bounds check, should have the same size */
 300                 strcpy(legend_list[j++], im->gdes[i].legend);
 301             else
 302                 legend_list[j++][0] = '\0';
 303
 304             break;
 305         default:
 306             break;
 307         }
 308     }
 309
 310     /* fill data structure */
 311     for (dst_row = 0; (int) dst_row < (int) row_cnt; dst_row++) {
 312         for (i = 0; i < (int) nof_xports; i++) {
 313             j = ref_list[i];
 314             ii = im->gdes[j].vidx;
 315             ds_cnt = &im->gdes[ii].ds_cnt;
 316
 317             srcptr = srcptr_list[i];
 318             for (col = 0; col < (*ds_cnt); col++) {
 319                 rrd_value_t newval = DNAN;
 320
 321                 newval = srcptr[col];
 322
 323                 if (im->gdes[ii].ds_namv && im->gdes[ii].ds_nam) {
 324                     if (strcmp(im->gdes[ii].ds_namv[col], im->gdes[ii].ds_nam)
 325                         == 0)
 326                         (*dstptr++) = newval;
 327                 } else {
 328                     (*dstptr++) = newval;
 329                 }
 330
 331             }
 332             srcptr_list[i] += (*ds_cnt);
 333         }
 334     }
 335
 336     *legend_v = legend_list;
 337     free(srcptr_list);
 338     free(ref_list);
 339     return 0;
 340
 341 }