src/rrd_tool.c

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