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