src/rrd_tool.c

   1 /*****************************************************************************
   2  * RRDtool 1.0.33  Copyright Tobias Oetiker, 1997 - 2001
   3  *****************************************************************************
   4  * rrd_tool.c  Startup wrapper
   5  *****************************************************************************/
   6
   7 #include "rrd_tool.h"
   8
   9 void PrintUsage(char *cmd);
  10 int CountArgs(char *aLine);
  11 int CreateArgs(char *, char *, int, char **);
  12 int HandleInputLine(int, char **, FILE*);
  13 #define TRUE            1
  14 #define FALSE           0
  15 #define MAX_LENGTH      10000
  16
  17
  18 void PrintUsage(char *cmd)
  19 {
  20
  21     char help_main[] =
  22            "RRDtool 1.0.33  Copyright 1997-2001 by Tobias Oetiker <tobi@oetiker.ch>\n\n"
  23            "Usage: rrdtool [options] command command_options\n\n";
  24
  25     char help_list[] =
  26            "Valid commands: create, update, graph, dump, restore,\n"
  27            "\t\tlast, info, fetch, tune, resize\n\n";
  28
  29     char help_create[] =
  30            "* create - create a new RRD\n\n"
  31            "\trrdtool create filename [--start|-b start time]\n"
  32            "\t\t[--step|-s step]\n"
  33            "\t\t[DS:ds-name:DST:heartbeat:min:max] [RRA:CF:xff:steps:rows]\n\n";
  34
  35     char help_dump[] =
  36            "* dump - dump an RRD to XML\n\n"
  37            "\trrdtool dump filename.rrd >filename.xml\n\n";
  38
  39     char help_info[] =
  40            "* info - returns the configuration and status of the\n\n"
  41            "\trrdtool info filename.rrd\n\n";
  42
  43     char help_restore[] =
  44            "* restore - restore an RRD file from its XML form\n\n"
  45            "\trrdtool restore [--range-check|-r] filename.xml filename.rrd\n\n";
  46
  47     char help_last[] =
  48            "* last - show last update time for RRD\n\n"
  49            "\trrdtool last filename.rrd\n\n";
  50
  51     char help_update[] =
  52            "* update - update an RRD\n\n"
  53            "\trrdtool update filename\n"
  54            "\t\t--template|-t ds-name:ds-name:...\n"
  55            "\t\ttime|N:value[:value...]\n\n"
  56            "\t\t[ time:value[:value...] ..]\n\n";
  57
  58     char help_fetch[] =
  59            "* fetch - fetch data out of an RRD\n\n"
  60            "\trrdtool fetch filename.rrd CF\n"
  61            "\t\t[--resolution|-r resolution]\n"
  62            "\t\t[--start|-s start] [--end|-e end]\n\n";
  63
  64     char help_graph[] =
  65            "* graph - generate a graph from one or several RRD\n\n"
  66            "\trrdtool graph filename [-s|--start seconds] [-e|--end seconds]\n"
  67            "\t\t[-x|--x-grid x-axis grid and label]\n"
  68            "\t\t[--alt-y-grid]\n"
  69            "\t\t[-y|--y-grid y-axis grid and label]\n"
  70            "\t\t[-v|--vertical-label string] [-w|--width pixels]\n"
  71            "\t\t[-h|--height pixels] [-o|--logarithmic]\n"
  72            "\t\t[-u|--upper-limit value] [-z|--lazy]\n"
  73            "\t\t[-l|--lower-limit value] [-r|--rigid]\n"
  74            "\t\t[-g|--no-legend]\n"
  75            "\t\t[--alt-autoscale]\n"
  76            "\t\t[--alt-autoscale-max]\n"
  77            "\t\t[--units-exponent value]\n"
  78            "\t\t[--step seconds]\n"
  79            "\t\t[-f|--imginfo printfstr]\n"
  80            "\t\t[-a|--imgformat GIF|PNG]\n"
  81            "\t\t[-c|--color COLORTAG#rrggbb] [-t|--title string]\n"
  82            "\t\t[DEF:vname=rrd:ds-name:CF]\n"
  83            "\t\t[CDEF:vname=rpn-expression]\n"
  84            "\t\t[PRINT:vname:CF:format]\n"
  85            "\t\t[GPRINT:vname:CF:format]\n"
  86            "\t\t[HRULE:value#rrggbb[:legend]]\n"
  87            "\t\t[VRULE:value#rrggbb[:legend]]\n"
  88            "\t\t[LINE{1|2|3}:vname[#rrggbb[:legend]]]\n"
  89            "\t\t[AREA:vname[#rrggbb[:legend]]]\n"
  90            "\t\t[STACK:vname[#rrggbb[:legend]]]\n\n";
  91
  92     char help_tune[] =
  93            " * tune -  Modify some basic properties of an RRD\n\n"
  94            "\trrdtool tune filename\n"
  95            "\t\t[--heartbeat|-h ds-name:heartbeat]\n"
  96            "\t\t[--data-source-type|-d ds-name:DST\n"
  97            "\t\t[--data-source-rename|-r old-name:new-name\n"
  98            "\t\t[--minimum|-i ds-name:min] [--maximum|-a ds-name:max]\n\n";
  99
 100     char help_resize[] =
 101            " * resize - alter the lenght of one of the RRAs in an RRD\n\n"
 102            "\trrdtool resize filename rranum GROW|SHRINK rows\n\n";
 103
 104     char help_lic[] =
 105            "RRDtool is distributed under the Terms of the GNU General\n"
 106            "Public License Version 2. (www.gnu.org/copyleft/gpl.html)\n\n"
 107
 108            "For more information read the RRD manpages\n\n";
 109
 110     enum { C_NONE, C_CREATE, C_DUMP, C_INFO, C_RESTORE, C_LAST,
 111            C_UPDATE, C_FETCH, C_GRAPH, C_TUNE, C_RESIZE };
 112
 113     int help_cmd = C_NONE;
 114
 115     if (cmd)
 116         {
 117             if (!strcmp(cmd,"create"))
 118                 help_cmd = C_CREATE;
 119             else if (!strcmp(cmd,"dump"))
 120                 help_cmd = C_DUMP;
 121             else if (!strcmp(cmd,"info"))
 122                 help_cmd = C_INFO;
 123             else if (!strcmp(cmd,"restore"))
 124                 help_cmd = C_RESTORE;
 125             else if (!strcmp(cmd,"last"))
 126                 help_cmd = C_LAST;
 127             else if (!strcmp(cmd,"update"))
 128                 help_cmd = C_UPDATE;
 129             else if (!strcmp(cmd,"fetch"))
 130                 help_cmd = C_FETCH;
 131             else if (!strcmp(cmd,"graph"))
 132                 help_cmd = C_GRAPH;
 133             else if (!strcmp(cmd,"tune"))
 134                 help_cmd = C_TUNE;
 135             else if (!strcmp(cmd,"resize"))
 136                 help_cmd = C_RESIZE;
 137         }
 138     fputs(help_main, stdout);
 139     switch (help_cmd)
 140         {
 141             case C_NONE:
 142                 fputs(help_list, stdout);
 143                 break;
 144             case C_CREATE:
 145                 fputs(help_create, stdout);
 146                 break;
 147             case C_DUMP:
 148                 fputs(help_dump, stdout);
 149                 break;
 150             case C_INFO:
 151                 fputs(help_info, stdout);
 152                 break;
 153             case C_RESTORE:
 154                 fputs(help_restore, stdout);
 155                 break;
 156             case C_LAST:
 157                 fputs(help_last, stdout);
 158                 break;
 159             case C_UPDATE:
 160                 fputs(help_update, stdout);
 161                 break;
 162             case C_FETCH:
 163                 fputs(help_fetch, stdout);
 164                 break;
 165             case C_GRAPH:
 166                 fputs(help_graph, stdout);
 167                 break;
 168             case C_TUNE:
 169                 fputs(help_tune, stdout);
 170                 break;
 171             case C_RESIZE:
 172                 fputs(help_resize, stdout);
 173                 break;
 174         }
 175     fputs(help_lic, stdout);
 176 }
 177
 178
 179 int main(int argc, char *argv[])
 180 {
 181     char **myargv;
 182     char aLine[MAX_LENGTH];
 183 #ifdef MUST_DISABLE_SIGFPE
 184     signal(SIGFPE,SIG_IGN);
 185 #endif
 186 #ifdef MUST_DISABLE_FPMASK
 187     fpsetmask(0);
 188 #endif
 189     if (argc == 1)
 190         {
 191             PrintUsage("");
 192             return 0;
 193         }
 194
 195     if ((argc == 2) && !strcmp("-",argv[1]))
 196         {
 197 #if HAVE_GETRUSAGE
 198           struct rusage  myusage;
 199           struct timeval starttime;
 200           struct timeval currenttime;
 201           struct timezone tz;
 202
 203             tz.tz_minuteswest =0;
 204             tz.tz_dsttime=0;
 205             gettimeofday(&starttime,&tz);
 206 #endif
 207
 208             while (fgets(aLine, sizeof(aLine)-1, stdin)){
 209                 if ((argc = CountArgs(aLine)) == 0)  {
 210                     fprintf(stderr,"ERROR: not enough arguments\n");
 211                 }
 212                 if ((myargv = (char **) malloc((argc+1) *
 213                                                sizeof(char *))) == NULL)   {
 214                     perror("malloc");
 215                     return -1;
 216                 }
 217                 if ((argc=CreateArgs(argv[0], aLine, argc, myargv)) < 0) {
 218                     fprintf(stderr, "ERROR: creating arguments\n");
 219                     return -1;
 220                 }
 221
 222                 if (HandleInputLine(argc, myargv, stdout))
 223                     return -1;
 224                 free(myargv);
 225
 226 #if HAVE_GETRUSAGE
 227                 getrusage(RUSAGE_SELF,&myusage);
 228                 gettimeofday(&currenttime,&tz);
 229                 printf("OK u:%1.2f s:%1.2f r:%1.2f\n",
 230                        (double)myusage.ru_utime.tv_sec+
 231                        (double)myusage.ru_utime.tv_usec/1000000.0,
 232                        (double)myusage.ru_stime.tv_sec+
 233                        (double)myusage.ru_stime.tv_usec/1000000.0,
 234                        (double)(currenttime.tv_sec-starttime.tv_sec)
 235                        +(double)(currenttime.tv_usec-starttime.tv_usec)
 236                        /1000000.0);
 237 #else
 238                 printf("OK\n");
 239 #endif
 240                 fflush(stdout); /* this is important for pipes to work */
 241             }
 242         }
 243     else if (argc == 2)
 244         {
 245                 PrintUsage(argv[1]);
 246                 exit(0);
 247         }
 248     else
 249         HandleInputLine(argc, argv, stderr);
 250     return 0;
 251 }
 252
 253 int HandleInputLine(int argc, char **argv, FILE* out)
 254 {
 255     optind=0; /* reset gnu getopt */
 256     opterr=0; /* no error messages */
 257
 258     if (argc < 3
 259         || strcmp("help", argv[1]) == 0
 260         || strcmp("--help", argv[1]) == 0
 261         || strcmp("-help", argv[1]) == 0
 262         || strcmp("-?", argv[1]) == 0
 263         || strcmp("-h", argv[1]) == 0 ) {
 264         PrintUsage("");
 265         return 0;
 266     }
 267
 268     if (strcmp("create", argv[1]) == 0)
 269         rrd_create(argc-1, &argv[1]);
 270     else if (strcmp("dump", argv[1]) == 0)
 271         rrd_dump(argc-1, &argv[1]);
 272     else if (strcmp("info", argv[1]) == 0){
 273         info_t *data,*save;
 274         data=rrd_info(argc-1, &argv[1]);
 275         while (data) {
 276             save=data;
 277             printf ("%s = ", data->key);
 278             free(data->key);
 279
 280             switch (data->type) {
 281             case RD_I_VAL:
 282                 if (isnan (data->value.u_val))
 283                     printf("NaN");
 284                 else
 285                     printf ("%0.10e", data->value.u_val);
 286                 break;
 287             case RD_I_CNT:
 288                 printf ("%lu", data->value.u_cnt);
 289                 break;
 290             case RD_I_STR:
 291                 printf ("\"%s\"", data->value.u_str);
 292                 free(data->value.u_str);
 293                 break;
 294             }
 295             data = data->next;
 296             free(save);
 297             printf ("\n");
 298         }
 299         free(data);
 300     }
 301
 302     else if (strcmp("--version", argv[1]) == 0 ||
 303              strcmp("version", argv[1]) == 0 ||
 304              strcmp("v", argv[1]) == 0 ||
 305              strcmp("-v", argv[1]) == 0  ||
 306              strcmp("-version", argv[1]) == 0  )
 307         printf("RRDtool 1.0.33  Copyright (C) 1997-2001 by Tobias Oetiker <tobi@oetiker.ch>\n");
 308     else if (strcmp("restore", argv[1]) == 0)
 309         rrd_restore(argc-1, &argv[1]);
 310     else if (strcmp("resize", argv[1]) == 0)
 311         rrd_resize(argc-1, &argv[1]);
 312     else if (strcmp("last", argv[1]) == 0)
 313         printf("%ld\n",rrd_last(argc-1, &argv[1]));
 314     else if (strcmp("update", argv[1]) == 0)
 315         rrd_update(argc-1, &argv[1]);
 316     else if (strcmp("fetch", argv[1]) == 0) {
 317         time_t        start,end;
 318         unsigned long step, ds_cnt,i,ii;
 319         rrd_value_t   *data,*datai;
 320         char          **ds_namv;
 321         if (rrd_fetch(argc-1, &argv[1],&start,&end,&step,&ds_cnt,&ds_namv,&data) != -1) {
 322             datai=data;
 323             printf("           ");
 324             for (i = 0; i<ds_cnt;i++)
 325                 printf("%14s",ds_namv[i]);
 326             printf ("\n\n");
 327             for (i = start; i <= end; i += step){
 328                 printf("%10lu:", i);
 329                 for (ii = 0; ii < ds_cnt; ii++)
 330                     printf(" %0.10e", *(datai++));
 331                 printf("\n");
 332             }
 333             for (i=0;i<ds_cnt;i++)
 334                   free(ds_namv[i]);
 335             free(ds_namv);
 336             free (data);
 337         }
 338     }
 339     else if (strcmp("graph", argv[1]) == 0) {
 340         char **calcpr;
 341         int xsize, ysize;
 342         int i;
 343         if( rrd_graph(argc-1, &argv[1], &calcpr, &xsize, &ysize) != -1 ) {
 344             if (strcmp(argv[2],"-") != 0)
 345                 printf ("%dx%d\n",xsize,ysize);
 346             if (calcpr) {
 347                 for(i=0;calcpr[i];i++){
 348                     if (strcmp(argv[2],"-") != 0)
 349                         printf("%s\n",calcpr[i]);
 350                     free(calcpr[i]);
 351                 }
 352                 free(calcpr);
 353             }
 354         }
 355
 356     } else if (strcmp("tune", argv[1]) == 0)
 357                 rrd_tune(argc-1, &argv[1]);
 358     else {
 359                 rrd_set_error("unknown function '%s'",argv[1]);
 360     }
 361     if (rrd_test_error()) {
 362         fprintf(out, "ERROR: %s\n",rrd_get_error());
 363         rrd_clear_error();
 364     }
 365     return(0);
 366 }
 367
 368 int CountArgs(char *aLine)
 369 {
 370     int i=0;
 371     int aCount = 0;
 372     int inarg = 0;
 373     while (aLine[i] == ' ') i++;
 374     while (aLine[i] != 0){
 375         if((aLine[i]== ' ') && inarg){
 376             inarg = 0;
 377         }
 378         if((aLine[i]!= ' ') && ! inarg){
 379             inarg = 1;
 380             aCount++;
 381         }
 382         i++;
 383     }
 384     return aCount;
 385 }
 386
 387 /*
 388  * CreateArgs - take a string (aLine) and tokenize
 389  */
 390 int CreateArgs(char *pName, char *aLine, int argc, char **argv)
 391 {
 392     char        *getP, *putP;
 393     char        **pargv = argv;
 394     char        Quote = 0;
 395     int inArg = 0;
 396     int len;
 397
 398     len = strlen(aLine);
 399     /* remove trailing space and newlines */
 400     while (len && aLine[len] <= ' ') {
 401         aLine[len] = 0 ; len--;
 402     }
 403     /* sikp leading blanks */
 404     while (*aLine && *aLine <= ' ') aLine++;
 405
 406     pargv[0] = pName;
 407     argc = 1;
 408     getP = aLine;
 409     putP = aLine;
 410     while (*getP){
 411         switch (*getP) {
 412         case ' ':
 413             if (Quote){
 414                 *(putP++)=*getP;
 415             } else
 416                 if(inArg) {
 417                     *(putP++) = 0;
 418                     inArg = 0;
 419                 }
 420             break;
 421         case '"':
 422         case '\'':
 423             if (Quote != 0) {
 424                 if (Quote == *getP)
 425                     Quote = 0;
 426                 else {
 427                     *(putP++)=*getP;
 428                 }
 429             } else {
 430                 if(!inArg){
 431                     pargv[argc++] = putP;
 432                     inArg=1;
 433                 }
 434                 Quote = *getP;
 435             }
 436             break;
 437         default:
 438             if(!inArg){
 439                 pargv[argc++] = putP;
 440                 inArg=1;
 441             }
 442             *(putP++)=*getP;
 443             break;
 444         }
 445         getP++;
 446     }
 447
 448     *putP = '\0';
 449
 450     if (Quote)
 451         return -1;
 452     else
 453         return argc;
 454 }
 455
 456