src/rrd_tool.c

   1 /*****************************************************************************
   2  * RRDtool 1.3.2  Copyright by Tobi Oetiker, 1997-2008
   3  *****************************************************************************
   4  * rrd_tool.c  Startup wrapper
   5  *****************************************************************************/
   6
   7 #if defined(_WIN32) && !defined(__CYGWIN__) && !defined(__CYGWIN32__) && !defined(HAVE_CONFIG_H)
   8 #include "../win32/config.h"
   9 #include <stdlib.h>
  10 #include <sys/stat.h>
  11 #else
  12 #ifdef HAVE_CONFIG_H
  13 #include "../rrd_config.h"
  14 #endif
  15 #endif
  16
  17 #include "rrd_tool.h"
  18 #include "rrd_xport.h"
  19 #include "rrd_i18n.h"
  20
  21 #ifdef HAVE_LOCALE_H
  22 #include <locale.h>
  23 #endif
  24
  25 void      PrintUsage(
  26     char *cmd);
  27 int       CountArgs(
  28     char *aLine);
  29 int       CreateArgs(
  30     char *,
  31     char *,
  32     int,
  33     char **);
  34 int       HandleInputLine(
  35     int,
  36     char **,
  37     FILE *);
  38 int       RemoteMode = 0;
  39 int       ChangeRoot = 0;
  40
  41 #define TRUE            1
  42 #define FALSE           0
  43 #define MAX_LENGTH      10000
  44
  45
  46 void PrintUsage(
  47     char *cmd)
  48 {
  49
  50     const char *help_main =
  51         N_("RRDtool %s"
  52            "  Copyright 1997-2008 by Tobias Oetiker <tobi@oetiker.ch>\n"
  53            "               Compiled %s %s\n\n"
  54            "Usage: rrdtool [options] command command_options\n\n");
  55
  56     const char *help_list =
  57         N_
  58         ("Valid commands: create, update, updatev, graph, graphv,  dump, restore,\n"
  59          "\t\tlast, lastupdate, first, info, fetch, tune,\n"
  60          "\t\tresize, xport, flush\n\n");
  61
  62     const char *help_listremote =
  63         N_("Valid remote commands: quit, ls, cd, mkdir, pwd\n\n");
  64
  65
  66     const char *help_create =
  67         N_("* create - create a new RRD\n\n"
  68            "\trrdtool create filename [--start|-b start time]\n"
  69            "\t\t[--step|-s step]\n"
  70            "\t\t[DS:ds-name:DST:dst arguments]\n"
  71            "\t\t[RRA:CF:cf arguments]\n\n");
  72
  73     const char *help_dump =
  74         N_("* dump - dump an RRD to XML\n\n"
  75            "\trrdtool dump filename.rrd >filename.xml\n\n");
  76
  77     const char *help_info =
  78         N_("* info - returns the configuration and status of the RRD\n\n"
  79            "\trrdtool info filename.rrd\n\n");
  80
  81     const char *help_restore =
  82         N_("* restore - restore an RRD file from its XML form\n\n"
  83            "\trrdtool restore [--range-check|-r] [--force-overwrite|-f] filename.xml filename.rrd\n\n");
  84
  85     const char *help_last =
  86         N_("* last - show last update time for RRD\n\n"
  87            "\trrdtool last filename.rrd\n\n");
  88
  89     const char *help_lastupdate =
  90         N_("* lastupdate - returns the most recent datum stored for\n"
  91            "  each DS in an RRD\n\n" "\trrdtool lastupdate filename.rrd\n\n");
  92
  93     const char *help_first =
  94         N_("* first - show first update time for RRA within an RRD\n\n"
  95            "\trrdtool first filename.rrd [--rraindex number]\n\n");
  96
  97     const char *help_update =
  98         N_("* update - update an RRD\n\n"
  99            "\trrdtool update filename\n"
 100            "\t\t[--template|-t ds-name:ds-name:...]\n"
 101            "\t\t[--daemon <address>]\n"
 102            "\t\ttime|N:value[:value...]\n\n"
 103            "\t\tat-time@value[:value...]\n\n"
 104            "\t\t[ time:value[:value...] ..]\n\n");
 105
 106     const char *help_updatev =
 107         N_("* updatev - a verbose version of update\n"
 108            "\treturns information about values, RRAs, and datasources updated\n\n"
 109            "\trrdtool updatev filename\n"
 110            "\t\t[--template|-t ds-name:ds-name:...]\n"
 111            "\t\ttime|N:value[:value...]\n\n"
 112            "\t\tat-time@value[:value...]\n\n"
 113            "\t\t[ time:value[:value...] ..]\n\n");
 114
 115     const char *help_fetch =
 116         N_("* fetch - fetch data out of an RRD\n\n"
 117            "\trrdtool fetch filename.rrd CF\n"
 118            "\t\t[-r|--resolution resolution]\n"
 119            "\t\t[-s|--start start] [-e|--end end]\n"
 120            "\t\t[--daemon <address>]\n\n");
 121
 122     const char *help_flush =
 123         N_("* flush - flush cached data out to an RRD file\n\n"
 124            "\trrdtool flush filename.rrd\n"
 125            "\t\t[--daemon <address>]\n\n");
 126
 127 /* break up very large strings (help_graph, help_tune) for ISO C89 compliance*/
 128
 129     const char *help_graph0 =
 130         N_("* graph - generate a graph from one or several RRD\n\n"
 131            "\trrdtool graph filename [-s|--start seconds] [-e|--end seconds]\n");
 132     const char *help_graphv0 =
 133         N_("* graphv - generate a graph from one or several RRD\n"
 134            "           with meta data printed before the graph\n\n"
 135            "\trrdtool graphv filename [-s|--start seconds] [-e|--end seconds]\n");
 136     const char *help_graph1 =
 137         N_("\t\t[-x|--x-grid x-axis grid and label]\n"
 138            "\t\t[-Y|--alt-y-grid] [--full-size-mode]\n"
 139            "\t\t[-y|--y-grid y-axis grid and label]\n"
 140            "\t\t[-v|--vertical-label string] [-w|--width pixels]\n"
 141            "\t\t[--right-axis scale:shift] [--right-axis-label label]\n"
 142            "\t\t[--right-axis-format format]\n"
 143            "\t\t[-h|--height pixels] [-o|--logarithmic]\n"
 144            "\t\t[-u|--upper-limit value] [-z|--lazy]\n"
 145            "\t\t[-l|--lower-limit value] [-r|--rigid]\n"
 146            "\t\t[-g|--no-legend] [--daemon <address>]\n"
 147            "\t\t[-F|--force-rules-legend]\n" "\t\t[-j|--only-graph]\n");
 148     const char *help_graph2 =
 149         N_("\t\t[-n|--font FONTTAG:size:font]\n"
 150            "\t\t[-m|--zoom factor]\n"
 151            "\t\t[-A|--alt-autoscale]\n"
 152            "\t\t[-M|--alt-autoscale-max]\n"
 153            "\t\t[-R|--font-render-mode {normal,light,mono}]\n"
 154            "\t\t[-B|--font-smoothing-threshold size]\n"
 155            "\t\t[-T|--tabwidth width]\n"
 156            "\t\t[-E|--slope-mode]\n"
 157            "\t\t[-N|--no-gridfit]\n"
 158            "\t\t[-X|--units-exponent value]\n"
 159            "\t\t[-L|--units-length value]\n"
 160            "\t\t[-S|--step seconds]\n"
 161            "\t\t[-f|--imginfo printfstr]\n"
 162            "\t\t[-a|--imgformat PNG]\n"
 163            "\t\t[-c|--color COLORTAG#rrggbb[aa]] [-t|--title string]\n"
 164            "\t\t[-W|--watermark string]\n"
 165            "\t\t[DEF:vname=rrd:ds-name:CF]\n");
 166     const char *help_graph3 =
 167         N_("\t\t[CDEF:vname=rpn-expression]\n"
 168            "\t\t[VDEF:vdefname=rpn-expression]\n"
 169            "\t\t[PRINT:vdefname:format]\n"
 170            "\t\t[GPRINT:vdefname:format]\n" "\t\t[COMMENT:text]\n"
 171            "\t\t[SHIFT:vname:offset]\n"
 172            "\t\t[TICK:vname#rrggbb[aa][:[fraction][:legend]]]\n"
 173            "\t\t[HRULE:value#rrggbb[aa][:legend]]\n"
 174            "\t\t[VRULE:value#rrggbb[aa][:legend]]\n"
 175            "\t\t[LINE[width]:vname[#rrggbb[aa][:[legend][:STACK]]]]\n"
 176            "\t\t[AREA:vname[#rrggbb[aa][:[legend][:STACK]]]]\n"
 177            "\t\t[PRINT:vname:CF:format] (deprecated)\n"
 178            "\t\t[GPRINT:vname:CF:format] (deprecated)\n"
 179            "\t\t[STACK:vname[#rrggbb[aa][:legend]]] (deprecated)\n\n");
 180     const char *help_tune1 =
 181         N_(" * tune -  Modify some basic properties of an RRD\n\n"
 182            "\trrdtool tune filename\n"
 183            "\t\t[--heartbeat|-h ds-name:heartbeat]\n"
 184            "\t\t[--data-source-type|-d ds-name:DST]\n"
 185            "\t\t[--data-source-rename|-r old-name:new-name]\n"
 186            "\t\t[--minimum|-i ds-name:min] [--maximum|-a ds-name:max]\n"
 187            "\t\t[--deltapos scale-value] [--deltaneg scale-value]\n"
 188            "\t\t[--failure-threshold integer]\n"
 189            "\t\t[--window-length integer]\n"
 190            "\t\t[--alpha adaptation-parameter]\n");
 191     const char *help_tune2 =
 192         N_(" * tune -  Modify some basic properties of an RRD\n\n"
 193            "\t\t[--beta adaptation-parameter]\n"
 194            "\t\t[--gamma adaptation-parameter]\n"
 195            "\t\t[--gamma-deviation adaptation-parameter]\n"
 196            "\t\t[--aberrant-reset ds-name]\n\n");
 197     const char *help_resize =
 198         N_
 199         (" * resize - alter the length of one of the RRAs in an RRD\n\n"
 200          "\trrdtool resize filename rranum GROW|SHRINK rows\n\n");
 201     const char *help_xport =
 202         N_("* xport - generate XML dump from one or several RRD\n\n"
 203            "\trrdtool xport [-s|--start seconds] [-e|--end seconds]\n"
 204            "\t\t[-m|--maxrows rows]\n" "\t\t[--step seconds]\n"
 205            "\t\t[--enumds]\n" "\t\t[DEF:vname=rrd:ds-name:CF]\n"
 206            "\t\t[CDEF:vname=rpn-expression]\n"
 207            "\t\t[XPORT:vname:legend]\n\n");
 208     const char *help_quit =
 209         N_(" * quit - closing a session in remote mode\n\n"
 210            "\trrdtool quit\n\n");
 211     const char *help_ls =
 212         N_(" * ls - lists all *.rrd files in current directory\n\n"
 213            "\trrdtool ls\n\n");
 214     const char *help_cd =
 215         N_(" * cd - changes the current directory\n\n"
 216            "\trrdtool cd new directory\n\n");
 217     const char *help_mkdir =
 218         N_(" * mkdir - creates a new directory\n\n"
 219            "\trrdtool mkdir newdirectoryname\n\n");
 220     const char *help_pwd =
 221         N_(" * pwd - returns the current working directory\n\n"
 222            "\trrdtool pwd\n\n");
 223     const char *help_lic =
 224         N_("RRDtool is distributed under the Terms of the GNU General\n"
 225            "Public License Version 2. (www.gnu.org/copyleft/gpl.html)\n\n"
 226            "For more information read the RRD manpages\n\n");
 227     enum { C_NONE, C_CREATE, C_DUMP, C_INFO, C_RESTORE, C_LAST,
 228         C_LASTUPDATE, C_FIRST, C_UPDATE, C_FETCH, C_GRAPH, C_GRAPHV,
 229         C_TUNE,
 230         C_RESIZE, C_XPORT, C_QUIT, C_LS, C_CD, C_MKDIR, C_PWD,
 231         C_UPDATEV, C_FLUSH
 232     };
 233     int       help_cmd = C_NONE;
 234
 235     if (cmd) {
 236         if (!strcmp(cmd, "create"))
 237             help_cmd = C_CREATE;
 238         else if (!strcmp(cmd, "dump"))
 239             help_cmd = C_DUMP;
 240         else if (!strcmp(cmd, "info"))
 241             help_cmd = C_INFO;
 242         else if (!strcmp(cmd, "restore"))
 243             help_cmd = C_RESTORE;
 244         else if (!strcmp(cmd, "last"))
 245             help_cmd = C_LAST;
 246         else if (!strcmp(cmd, "lastupdate"))
 247             help_cmd = C_LASTUPDATE;
 248         else if (!strcmp(cmd, "first"))
 249             help_cmd = C_FIRST;
 250         else if (!strcmp(cmd, "update"))
 251             help_cmd = C_UPDATE;
 252         else if (!strcmp(cmd, "updatev"))
 253             help_cmd = C_UPDATEV;
 254         else if (!strcmp(cmd, "fetch"))
 255             help_cmd = C_FETCH;
 256         else if (!strcmp(cmd, "flush"))
 257             help_cmd = C_FLUSH;
 258         else if (!strcmp(cmd, "graph"))
 259             help_cmd = C_GRAPH;
 260         else if (!strcmp(cmd, "graphv"))
 261             help_cmd = C_GRAPHV;
 262         else if (!strcmp(cmd, "tune"))
 263             help_cmd = C_TUNE;
 264         else if (!strcmp(cmd, "resize"))
 265             help_cmd = C_RESIZE;
 266         else if (!strcmp(cmd, "xport"))
 267             help_cmd = C_XPORT;
 268         else if (!strcmp(cmd, "quit"))
 269             help_cmd = C_QUIT;
 270         else if (!strcmp(cmd, "ls"))
 271             help_cmd = C_LS;
 272         else if (!strcmp(cmd, "cd"))
 273             help_cmd = C_CD;
 274         else if (!strcmp(cmd, "mkdir"))
 275             help_cmd = C_MKDIR;
 276         else if (!strcmp(cmd, "pwd"))
 277             help_cmd = C_PWD;
 278     }
 279     fprintf(stdout, _(help_main), PACKAGE_VERSION, __DATE__, __TIME__);
 280     fflush(stdout);
 281     switch (help_cmd) {
 282     case C_NONE:
 283         fputs(_(help_list), stdout);
 284         if (RemoteMode) {
 285             fputs(_(help_listremote), stdout);
 286         }
 287         break;
 288     case C_CREATE:
 289         fputs(_(help_create), stdout);
 290         break;
 291     case C_DUMP:
 292         fputs(_(help_dump), stdout);
 293         break;
 294     case C_INFO:
 295         fputs(_(help_info), stdout);
 296         break;
 297     case C_RESTORE:
 298         fputs(_(help_restore), stdout);
 299         break;
 300     case C_LAST:
 301         fputs(_(help_last), stdout);
 302         break;
 303     case C_LASTUPDATE:
 304         fputs(_(help_lastupdate), stdout);
 305         break;
 306     case C_FIRST:
 307         fputs(_(help_first), stdout);
 308         break;
 309     case C_UPDATE:
 310         fputs(_(help_update), stdout);
 311         break;
 312     case C_UPDATEV:
 313         fputs(_(help_updatev), stdout);
 314         break;
 315     case C_FETCH:
 316         fputs(_(help_fetch), stdout);
 317         break;
 318     case C_FLUSH:
 319         fputs(_(help_flush), stdout);
 320         break;
 321     case C_GRAPH:
 322         fputs(_(help_graph0), stdout);
 323         fputs(_(help_graph1), stdout);
 324         fputs(_(help_graph2), stdout);
 325         fputs(_(help_graph3), stdout);
 326         break;
 327     case C_GRAPHV:
 328         fputs(_(help_graphv0), stdout);
 329         fputs(_(help_graph1), stdout);
 330         fputs(_(help_graph2), stdout);
 331         fputs(_(help_graph3), stdout);
 332         break;
 333     case C_TUNE:
 334         fputs(_(help_tune1), stdout);
 335         fputs(_(help_tune2), stdout);
 336         break;
 337     case C_RESIZE:
 338         fputs(_(help_resize), stdout);
 339         break;
 340     case C_XPORT:
 341         fputs(_(help_xport), stdout);
 342         break;
 343     case C_QUIT:
 344         fputs(_(help_quit), stdout);
 345         break;
 346     case C_LS:
 347         fputs(_(help_ls), stdout);
 348         break;
 349     case C_CD:
 350         fputs(_(help_cd), stdout);
 351         break;
 352     case C_MKDIR:
 353         fputs(_(help_mkdir), stdout);
 354         break;
 355     case C_PWD:
 356         fputs(_(help_pwd), stdout);
 357         break;
 358     }
 359     fputs(_(help_lic), stdout);
 360 }
 361
 362 static char *fgetslong(
 363     char **aLinePtr,
 364     FILE * stream)
 365 {
 366     char     *linebuf;
 367     size_t    bufsize = MAX_LENGTH;
 368     int       eolpos = 0;
 369
 370     if (feof(stream))
 371         return *aLinePtr = 0;
 372     if (!(linebuf = malloc(bufsize))) {
 373         perror("fgetslong: malloc");
 374         exit(1);
 375     }
 376     linebuf[0] = '\0';
 377     while (fgets(linebuf + eolpos, MAX_LENGTH, stream)) {
 378         eolpos += strlen(linebuf + eolpos);
 379         if (linebuf[eolpos - 1] == '\n')
 380             return *aLinePtr = linebuf;
 381         bufsize += MAX_LENGTH;
 382         if (!(linebuf = realloc(linebuf, bufsize))) {
 383             free(linebuf);
 384             perror("fgetslong: realloc");
 385             exit(1);
 386         }
 387     }
 388     if (linebuf[0]){
 389         return  *aLinePtr = linebuf;
 390     }
 391     free(linebuf);
 392     return *aLinePtr = 0;
 393 }
 394
 395 int main(
 396     int argc,
 397     char *argv[])
 398 {
 399     char    **myargv;
 400     char     *aLine;
 401     char     *firstdir = "";
 402
 403 #ifdef MUST_DISABLE_SIGFPE
 404     signal(SIGFPE, SIG_IGN);
 405 #endif
 406 #ifdef MUST_DISABLE_FPMASK
 407     fpsetmask(0);
 408 #endif
 409 #ifdef HAVE_LOCALE_H
 410     setlocale(LC_ALL, "");
 411 #endif
 412
 413 #if defined(HAVE_LIBINTL_H) && defined(BUILD_LIBINTL)
 414     bindtextdomain(GETTEXT_PACKAGE, LOCALEDIR);
 415     bind_textdomain_codeset(GETTEXT_PACKAGE, "UTF-8");
 416     textdomain(GETTEXT_PACKAGE);
 417 #endif
 418     if (argc == 1) {
 419         PrintUsage("");
 420         return 0;
 421     }
 422
 423     if (((argc == 2) || (argc == 3)) && !strcmp("-", argv[1])) {
 424 #if HAVE_GETRUSAGE
 425         struct rusage myusage;
 426         struct timeval starttime;
 427         struct timeval currenttime;
 428
 429         gettimeofday(&starttime, NULL);
 430 #endif
 431         RemoteMode = 1;
 432         if ((argc == 3) && strcmp("", argv[2])) {
 433
 434             if (
 435 #ifdef HAVE_GETUID
 436                    getuid()
 437 #else
 438                    1
 439 #endif
 440                    == 0) {
 441
 442 #ifdef HAVE_CHROOT
 443                 if (chroot(argv[2]) != 0){
 444                     fprintf(stderr, "ERROR: chroot %s: %s\n", argv[2],rrd_strerror(errno));
 445                     exit(errno);
 446                 }
 447                 ChangeRoot = 1;
 448                 firstdir = "/";
 449 #else
 450                 fprintf(stderr,
 451                         "ERROR: change root is not supported by your OS "
 452                         "or at least by this copy of rrdtool\n");
 453                 exit(1);
 454 #endif
 455             } else {
 456                 firstdir = argv[2];
 457             }
 458         }
 459         if (strcmp(firstdir, "")) {
 460             if (chdir(firstdir) != 0){
 461                 fprintf(stderr, "ERROR: chdir %s %s\n", firstdir,rrd_strerror(errno));
 462                 exit(errno);
 463             }
 464         }
 465
 466         while (fgetslong(&aLine, stdin)) {
 467             if ((argc = CountArgs(aLine)) == 0) {
 468                 free(aLine);
 469                 printf("ERROR: not enough arguments\n");
 470             }
 471             if ((myargv = (char **) malloc((argc + 1) *
 472                                            sizeof(char *))) == NULL) {
 473                 perror("malloc");
 474                 exit(1);
 475             }
 476             if ((argc = CreateArgs(argv[0], aLine, argc, myargv)) < 0) {
 477                 free(aLine);
 478                 free(myargv);
 479                 printf("ERROR: creating arguments\n");
 480             } else {
 481                 int       ret = HandleInputLine(argc, myargv, stdout);
 482
 483                 free(myargv);
 484                 if (ret == 0) {
 485 #if HAVE_GETRUSAGE
 486                     getrusage(RUSAGE_SELF, &myusage);
 487                     gettimeofday(&currenttime, NULL);
 488                     printf("OK u:%1.2f s:%1.2f r:%1.2f\n",
 489                            (double) myusage.ru_utime.tv_sec +
 490                            (double) myusage.ru_utime.tv_usec / 1000000.0,
 491                            (double) myusage.ru_stime.tv_sec +
 492                            (double) myusage.ru_stime.tv_usec / 1000000.0,
 493                            (double) (currenttime.tv_sec - starttime.tv_sec)
 494                            + (double) (currenttime.tv_usec -
 495                                        starttime.tv_usec)
 496                            / 1000000.0);
 497 #else
 498                     printf("OK\n");
 499 #endif
 500                 }
 501             }
 502             fflush(stdout); /* this is important for pipes to work */
 503             free(aLine);
 504         }
 505     } else if (argc == 2) {
 506         PrintUsage(argv[1]);
 507         exit(0);
 508     } else if (argc == 3 && !strcmp(argv[1], "help")) {
 509         PrintUsage(argv[2]);
 510         exit(0);
 511     } else {
 512         exit(HandleInputLine(argc, argv, stderr));
 513     }
 514     return 0;
 515 }
 516
 517 /* HandleInputLine is NOT thread safe - due to readdir issues,
 518    resolving them portably is not really simple. */
 519 int HandleInputLine(
 520     int argc,
 521     char **argv,
 522     FILE * out)
 523 {
 524 #if defined(HAVE_OPENDIR) && defined (HAVE_READDIR)
 525     DIR      *curdir;   /* to read current dir with ls */
 526     struct dirent *dent;
 527 #endif
 528 #if defined(HAVE_SYS_STAT_H)
 529     struct stat st;
 530 #endif
 531
 532     /* Reset errno to 0 before we start.
 533      */
 534     if (RemoteMode) {
 535         if (argc > 1 && strcmp("quit", argv[1]) == 0) {
 536             if (argc > 2) {
 537                 printf("ERROR: invalid parameter count for quit\n");
 538                 return (1);
 539             }
 540             exit(0);
 541         }
 542 #if defined(HAVE_OPENDIR) && defined(HAVE_READDIR) && defined(HAVE_CHDIR)
 543         if (argc > 1 && strcmp("cd", argv[1]) == 0) {
 544             if (argc > 3) {
 545                 printf("ERROR: invalid parameter count for cd\n");
 546                 return (1);
 547             }
 548 #if ! defined(HAVE_CHROOT) || ! defined(HAVE_GETUID)
 549             if (getuid() == 0 && !ChangeRoot) {
 550                 printf
 551                     ("ERROR: chdir security problem - rrdtool is running as "
 552                      "root but not chroot!\n");
 553                 return (1);
 554             }
 555 #endif
 556             if (chdir(argv[2]) != 0){
 557                 printf("ERROR: chdir %s %s\n", argv[2], rrd_strerror(errno));
 558                 return (1);
 559             }
 560             return (0);
 561         }
 562         if (argc > 1 && strcmp("pwd", argv[1]) == 0) {
 563             char     *cwd;      /* To hold current working dir on call to pwd */
 564             if (argc > 2) {
 565                 printf("ERROR: invalid parameter count for pwd\n");
 566                 return (1);
 567             }
 568             cwd = getcwd(NULL, MAXPATH);
 569             if (cwd == NULL) {
 570                 printf("ERROR: getcwd %s\n", rrd_strerror(errno));
 571                 return (1);
 572             }
 573             printf("%s\n", cwd);
 574             free(cwd);
 575             return (0);
 576         }
 577         if (argc > 1 && strcmp("mkdir", argv[1]) == 0) {
 578             if (argc > 3) {
 579                 printf("ERROR: invalid parameter count for mkdir\n");
 580                 return (1);
 581             }
 582 #if ! defined(HAVE_CHROOT) || ! defined(HAVE_GETUID)
 583             if (getuid() == 0 && !ChangeRoot) {
 584                 printf
 585                     ("ERROR: mkdir security problem - rrdtool is running as "
 586                      "root but not chroot!\n");
 587                 return (1);
 588             }
 589 #endif
 590             if(mkdir(argv[2], 0777)!=0){
 591                 printf("ERROR: mkdir %s: %s\n", argv[2],rrd_strerror(errno));
 592                 return (1);
 593             }
 594             return (0);
 595         }
 596         if (argc > 1 && strcmp("ls", argv[1]) == 0) {
 597             if (argc > 2) {
 598                 printf("ERROR: invalid parameter count for ls\n");
 599                 return (1);
 600             }
 601             if ((curdir = opendir(".")) != NULL) {
 602                 while ((dent = readdir(curdir)) != NULL) {
 603                     if (!stat(dent->d_name, &st)) {
 604                         if (S_ISDIR(st.st_mode)) {
 605                             printf("d %s\n", dent->d_name);
 606                         }
 607                         if (strlen(dent->d_name) > 4 && S_ISREG(st.st_mode)) {
 608                             if (!strcmp
 609                                 (dent->d_name + NAMLEN(dent) - 4, ".rrd")
 610                                 || !strcmp(dent->d_name + NAMLEN(dent) - 4,
 611                                            ".RRD")) {
 612                                 printf("- %s\n", dent->d_name);
 613                             }
 614                         }
 615                     }
 616                 }
 617                 closedir(curdir);
 618             } else {
 619                 printf("ERROR: opendir .: %s\n", rrd_strerror(errno));
 620                 return (errno);
 621             }
 622             return (0);
 623         }
 624 #endif                          /* opendir and readdir */
 625
 626     }
 627     if (argc < 3
 628         || strcmp("help", argv[1]) == 0
 629         || strcmp("--help", argv[1]) == 0
 630         || strcmp("-help", argv[1]) == 0
 631         || strcmp("-?", argv[1]) == 0 || strcmp("-h", argv[1]) == 0) {
 632         PrintUsage("");
 633         return 0;
 634     }
 635
 636     if (strcmp("create", argv[1]) == 0)
 637         rrd_create(argc - 1, &argv[1]);
 638     else if (strcmp("dump", argv[1]) == 0)
 639         rrd_dump(argc - 1, &argv[1]);
 640     else if (strcmp("info", argv[1]) == 0 || strcmp("updatev", argv[1]) == 0) {
 641         rrd_info_t *data;
 642
 643         if (strcmp("info", argv[1]) == 0)
 644
 645             data = rrd_info(argc - 1, &argv[1]);
 646         else
 647             data = rrd_update_v(argc - 1, &argv[1]);
 648         rrd_info_print(data);
 649         rrd_info_free(data);
 650     }
 651
 652     else if (strcmp("--version", argv[1]) == 0 ||
 653              strcmp("version", argv[1]) == 0 ||
 654              strcmp("v", argv[1]) == 0 ||
 655              strcmp("-v", argv[1]) == 0 || strcmp("-version", argv[1]) == 0)
 656         printf("RRDtool " PACKAGE_VERSION
 657                "  Copyright by Tobi Oetiker, 1997-2008 (%f)\n",
 658                rrd_version());
 659     else if (strcmp("restore", argv[1]) == 0)
 660         rrd_restore(argc - 1, &argv[1]);
 661     else if (strcmp("resize", argv[1]) == 0)
 662         rrd_resize(argc - 1, &argv[1]);
 663     else if (strcmp("last", argv[1]) == 0)
 664         printf("%ld\n", rrd_last(argc - 1, &argv[1]));
 665     else if (strcmp("lastupdate", argv[1]) == 0) {
 666         rrd_lastupdate(argc - 1, &argv[1]);
 667     } else if (strcmp("first", argv[1]) == 0)
 668         printf("%ld\n", rrd_first(argc - 1, &argv[1]));
 669     else if (strcmp("update", argv[1]) == 0)
 670         rrd_update(argc - 1, &argv[1]);
 671     else if (strcmp("fetch", argv[1]) == 0) {
 672         time_t    start, end, ti;
 673         unsigned long step, ds_cnt, i, ii;
 674         rrd_value_t *data, *datai;
 675         char    **ds_namv;
 676
 677         if (rrd_fetch
 678             (argc - 1, &argv[1], &start, &end, &step, &ds_cnt, &ds_namv,
 679              &data) != -1) {
 680             datai = data;
 681             printf("           ");
 682             for (i = 0; i < ds_cnt; i++)
 683                 printf("%20s", ds_namv[i]);
 684             printf("\n\n");
 685             for (ti = start + step; ti <= end; ti += step) {
 686                 printf("%10lu:", ti);
 687                 for (ii = 0; ii < ds_cnt; ii++)
 688                     printf(" %0.10e", *(datai++));
 689                 printf("\n");
 690             }
 691             for (i = 0; i < ds_cnt; i++)
 692                 free(ds_namv[i]);
 693             free(ds_namv);
 694             free(data);
 695         }
 696     } else if (strcmp("xport", argv[1]) == 0) {
 697         int       xxsize;
 698         unsigned long int j = 0;
 699         time_t    start, end, ti;
 700         unsigned long step, col_cnt, row_cnt;
 701         rrd_value_t *data, *ptr;
 702         char    **legend_v;
 703         int       enumds = 0;
 704         int       i;
 705         size_t    vtag_s = strlen(COL_DATA_TAG) + 10;
 706         char     *vtag = malloc(vtag_s);
 707
 708         for (i = 2; i < argc; i++) {
 709             if (strcmp("--enumds", argv[i]) == 0)
 710                 enumds = 1;
 711         }
 712
 713         if (rrd_xport
 714             (argc - 1, &argv[1], &xxsize, &start, &end, &step, &col_cnt,
 715              &legend_v, &data) != -1) {
 716             row_cnt = (end - start) / step;
 717             ptr = data;
 718             printf("<?xml version=\"1.0\" encoding=\"%s\"?>\n\n",
 719                    XML_ENCODING);
 720             printf("<%s>\n", ROOT_TAG);
 721             printf("  <%s>\n", META_TAG);
 722             printf("    <%s>%lu</%s>\n", META_START_TAG,
 723                    (unsigned long) start + step, META_START_TAG);
 724             printf("    <%s>%lu</%s>\n", META_STEP_TAG, step, META_STEP_TAG);
 725             printf("    <%s>%lu</%s>\n", META_END_TAG, (unsigned long) end,
 726                    META_END_TAG);
 727             printf("    <%s>%lu</%s>\n", META_ROWS_TAG, row_cnt,
 728                    META_ROWS_TAG);
 729             printf("    <%s>%lu</%s>\n", META_COLS_TAG, col_cnt,
 730                    META_COLS_TAG);
 731             printf("    <%s>\n", LEGEND_TAG);
 732             for (j = 0; j < col_cnt; j++) {
 733                 char     *entry = NULL;
 734
 735                 entry = legend_v[j];
 736                 printf("      <%s>%s</%s>\n", LEGEND_ENTRY_TAG, entry,
 737                        LEGEND_ENTRY_TAG);
 738                 free(entry);
 739             }
 740             free(legend_v);
 741             printf("    </%s>\n", LEGEND_TAG);
 742             printf("  </%s>\n", META_TAG);
 743             printf("  <%s>\n", DATA_TAG);
 744             for (ti = start + step; ti <= end; ti += step) {
 745                 printf("    <%s>", DATA_ROW_TAG);
 746                 printf("<%s>%lu</%s>", COL_TIME_TAG, ti, COL_TIME_TAG);
 747                 for (j = 0; j < col_cnt; j++) {
 748                     rrd_value_t newval = DNAN;
 749
 750                     if (enumds == 1)
 751
 752                         snprintf(vtag, vtag_s, "%s%lu", COL_DATA_TAG, j);
 753                     else
 754                         snprintf(vtag, vtag_s, "%s", COL_DATA_TAG);
 755                     newval = *ptr;
 756                     if (isnan(newval)) {
 757                         printf("<%s>NaN</%s>", vtag, vtag);
 758                     } else {
 759                         printf("<%s>%0.10e</%s>", vtag, newval, vtag);
 760                     };
 761                     ptr++;
 762                 }
 763                 printf("</%s>\n", DATA_ROW_TAG);
 764             }
 765             free(data);
 766             printf("  </%s>\n", DATA_TAG);
 767             printf("</%s>\n", ROOT_TAG);
 768         }
 769         free(vtag);
 770     } else if (strcmp("graph", argv[1]) == 0) {
 771         char    **calcpr;
 772
 773 #ifdef notused /*XXX*/
 774         const char *imgfile = argv[2];  /* rrd_graph changes argv pointer */
 775 #endif
 776         int       xsize, ysize;
 777         double    ymin, ymax;
 778         int       i;
 779         int       tostdout = (strcmp(argv[2], "-") == 0);
 780         int       imginfo = 0;
 781
 782         for (i = 2; i < argc; i++) {
 783             if (strcmp(argv[i], "--imginfo") == 0
 784                 || strcmp(argv[i], "-f") == 0) {
 785                 imginfo = 1;
 786                 break;
 787             }
 788         }
 789         if (rrd_graph
 790             (argc - 1, &argv[1], &calcpr, &xsize, &ysize, NULL, &ymin,
 791              &ymax) != -1) {
 792             if (!tostdout && !imginfo)
 793                 printf("%dx%d\n", xsize, ysize);
 794             if (calcpr) {
 795                 for (i = 0; calcpr[i]; i++) {
 796                     if (!tostdout)
 797                         printf("%s\n", calcpr[i]);
 798                     free(calcpr[i]);
 799                 }
 800                 free(calcpr);
 801             }
 802         }
 803
 804     } else if (strcmp("graphv", argv[1]) == 0) {
 805         rrd_info_t *grinfo = NULL;  /* 1 to distinguish it from the NULL that rrd_graph sends in */
 806
 807         grinfo = rrd_graph_v(argc - 1, &argv[1]);
 808         if (grinfo) {
 809             rrd_info_print(grinfo);
 810             rrd_info_free(grinfo);
 811         }
 812
 813     } else if (strcmp("tune", argv[1]) == 0)
 814         rrd_tune(argc - 1, &argv[1]);
 815     else if (strcmp("flush", argv[1]) == 0)
 816         rrd_cmd_flush(argc - 1, &argv[1]);
 817     else {
 818         rrd_set_error("unknown function '%s'", argv[1]);
 819     }
 820     if (rrd_test_error()) {
 821         fprintf(out, "ERROR: %s\n", rrd_get_error());
 822         rrd_clear_error();
 823         return 1;
 824     }
 825     return (0);
 826 }
 827
 828 int CountArgs(
 829     char *aLine)
 830 {
 831     int       i = 0;
 832     int       aCount = 0;
 833     int       inarg = 0;
 834
 835     while (aLine[i] == ' ')
 836         i++;
 837     while (aLine[i] != 0) {
 838         if ((aLine[i] == ' ') && inarg) {
 839             inarg = 0;
 840         }
 841         if ((aLine[i] != ' ') && !inarg) {
 842             inarg = 1;
 843             aCount++;
 844         }
 845         i++;
 846     }
 847     return aCount;
 848 }
 849
 850 /*
 851  * CreateArgs - take a string (aLine) and tokenize
 852  */
 853 int CreateArgs(
 854     char *pName,
 855     char *aLine,
 856     int argc,
 857     char **argv)
 858 {
 859     char     *getP, *putP;
 860     char    **pargv = argv;
 861     char      Quote = 0;
 862     int       inArg = 0;
 863     int       len;
 864
 865     len = strlen(aLine);
 866     /* remove trailing space and newlines */
 867     while (len && aLine[len] <= ' ') {
 868         aLine[len] = 0;
 869         len--;
 870     }
 871     /* sikp leading blanks */
 872     while (*aLine && *aLine <= ' ')
 873         aLine++;
 874     pargv[0] = pName;
 875     argc = 1;
 876     getP = aLine;
 877     putP = aLine;
 878     while (*getP) {
 879         switch (*getP) {
 880         case ' ':
 881             if (Quote) {
 882                 *(putP++) = *getP;
 883             } else if (inArg) {
 884                 *(putP++) = 0;
 885                 inArg = 0;
 886             }
 887             break;
 888         case '"':
 889         case '\'':
 890             if (Quote != 0) {
 891                 if (Quote == *getP)
 892                     Quote = 0;
 893                 else {
 894                     *(putP++) = *getP;
 895                 }
 896             } else {
 897                 if (!inArg) {
 898                     pargv[argc++] = putP;
 899                     inArg = 1;
 900                 }
 901                 Quote = *getP;
 902             }
 903             break;
 904         default:
 905             if (!inArg) {
 906                 pargv[argc++] = putP;
 907                 inArg = 1;
 908             }
 909             *(putP++) = *getP;
 910             break;
 911         }
 912         getP++;
 913     }
 914
 915     *putP = '\0';
 916     if (Quote)
 917         return -1;
 918     else
 919         return argc;
 920 }