git.itanic.dy.fi Git - rrdd/blob - rrdtool.c

   1 #include <time.h>
   2
   3 #include <sys/types.h>
   4 #include <sys/stat.h>
   5 #include <unistd.h>
   6 #include <string.h>
   7
   8 #include "rrdtool.h"
   9 #include "process.h"
  10 #include "parser.h"
  11 #include "debug.h"
  12 #include "string.h"
  13
  14 #define MAX_ARGS        512
  15 #define ARGSTR_LEN      32768
  16
  17 #define RRDTOOL_CMD "/usr/bin/rrdtool"
  18
  19 /*
  20  * Add new argument to a argument list
  21  *
  22  * args         pointer list to arguments
  23  * argcnt       argument counter
  24  * argstr       array where the actual arguments are stored
  25  * idx          index in the argstr where the new argument will be appended
  26  */
  27 #define add_arg(args, argcnt, argstr, idx, fmt, arg...) \
  28         args[argcnt] = argstr + idx;                    \
  29         idx += sprintf(argstr + idx, fmt, ##arg);       \
  30         argcnt++;                                       \
  31         argstr[++idx] = 0
  32
  33 int rrdtool_draw_image(struct rrd_image *image)
  34 {
  35         char cmd[] = RRDTOOL_CMD;
  36 //      char cmd[] = "echo";
  37         char *args[512], argstr[ARGSTR_LEN];
  38         int idx = 0, argcnt = 0, i,j;
  39         char timestamp[256];
  40         char tmp[256];
  41         time_t t = time(0);
  42
  43         pr_info("Drawing image %s\n", image->image_filename);
  44
  45         strftime(tmp, 256, "%d.%m.%Y %T (%Z) ", localtime(&t));
  46         for (i = 0, j = 0; j < 256;) {
  47                 if (tmp[i] == ':') {
  48                         timestamp[j++] = '\\';
  49                 }
  50                 timestamp[j++] = tmp[i++];
  51                 if (!tmp[i])
  52                         break;
  53         }
  54         timestamp[j] = 0;
  55
  56
  57         add_arg(args, argcnt, argstr, idx, " ");
  58         add_arg(args, argcnt, argstr, idx, "graph");
  59         add_arg(args, argcnt, argstr, idx, "%s", image->image_filename);
  60
  61         add_arg(args, argcnt, argstr, idx, "--start");
  62         add_arg(args, argcnt, argstr, idx, "%s", image->timestart);
  63         add_arg(args, argcnt, argstr, idx, "--end");
  64         add_arg(args, argcnt, argstr, idx, "%s", image->timeend);
  65         add_arg(args, argcnt, argstr, idx, "--width");
  66         add_arg(args, argcnt, argstr, idx, "%d", image->width);
  67         add_arg(args, argcnt, argstr, idx, "--height");
  68         add_arg(args, argcnt, argstr, idx, "%d", image->height);
  69         add_arg(args, argcnt, argstr, idx, "--imgformat");
  70         add_arg(args, argcnt, argstr, idx, "%s", image->imageformat);
  71
  72         for (i = 0; image->options[i]; i++) {
  73                 add_arg(args, argcnt, argstr, idx, "%s", image->options[i]);
  74         }
  75
  76         for (i = 0; image->text[i]; i++) {
  77                 args[argcnt++] = image->text[i];
  78         }
  79
  80         add_arg(args, argcnt, argstr, idx, "COMMENT:Last update %s\\c", timestamp);
  81
  82         args[argcnt] = 0;
  83
  84         run(cmd, args);
  85
  86         return 0;
  87 }
  88
  89 int rrdtool_draw_images(struct rrd_image **image)
  90 {
  91         int i;
  92         for (i = 0; image[i]; i++)
  93                 rrdtool_draw_image(image[i]);
  94
  95         return 0;
  96 }
  97
  98 static int sanitize_rrd_update_data(char *data)
  99 {
 100         char clean_data[RRD_DATA_MAX_LEN];
 101         int entries = 0;
 102         char *src, *end, *cln;
 103
 104         data[RRD_DATA_MAX_LEN - 1] = 0;
 105         src = data;
 106         cln = clean_data;
 107
 108         while (src < data + RRD_DATA_MAX_LEN && *src) {
 109                 if (*src == '-')
 110                         src++;
 111
 112                 end = skip_numbers(src);
 113
 114                 if (*end == '.') {
 115                         end++;
 116                         end = skip_numbers(end);
 117                 }
 118
 119                 if (*end == ':' || !*end) {
 120                         for (; src <= end; src++, cln++)
 121                                 *cln = *src;
 122
 123                         goto next;
 124                 }
 125
 126                 while (*end != ':' && *end)
 127                         end++;
 128                 *cln = 'U';
 129                 cln++;
 130                 *cln = ':';
 131                 cln++;
 132         next:
 133                 end++;
 134                 src = end;
 135                 entries++;
 136         }
 137
 138         strncpy(data, clean_data, RRD_DATA_MAX_LEN);
 139         return entries;
 140 }
 141
 142 int rrdtool_update_data(struct rrd_database *rrd)
 143 {
 144         char data[RRD_DATA_MAX_LEN + 2];
 145         char cmd[] = RRDTOOL_CMD;
 146 //      char cmd[] = "echo";
 147         char *cmdline[] = {
 148                 "",
 149                 "update",
 150                 rrd->filename,
 151                 data,
 152                 0
 153         };
 154         int l;
 155
 156         rrd->last_update = time(0);
 157         if (do_fork())
 158                 return 0;
 159
 160         l = sprintf(data, "N:");
 161
 162         if (rrd->parse) {
 163                 rrd->parse(data + l, rrd->parser_data);
 164                 sanitize_rrd_update_data(data + l);
 165                 run(cmd, cmdline);
 166         }
 167
 168         if (rrd->images)
 169                 rrdtool_draw_images(rrd->images);
 170
 171         while (harvest_zombies(0));
 172         exit(0);
 173 }
 174
 175 static int database_exists(struct rrd_database *db)
 176 {
 177         struct stat s;
 178
 179         /* If the filename exists, stat will return zero */
 180         if (db->filename)
 181                 return !stat(db->filename, &s);
 182
 183         return 1;
 184 }
 185
 186 static int create_database(struct rrd_database *db)
 187 {
 188         char cmd[] = RRDTOOL_CMD;
 189 //      char cmd[] = "echo";
 190         char *args[512], argstr[ARGSTR_LEN];
 191         int idx = 0, argcnt = 0;
 192         int child, i;
 193
 194         if (!db->sources || !db->archives) {
 195                 printf("Cannot create db \"%s\", insufficient source data\n",
 196                         db->filename);
 197                 return -1;
 198         }
 199
 200         add_arg(args, argcnt, argstr, idx, " ");
 201         add_arg(args, argcnt, argstr, idx, "create");
 202         add_arg(args, argcnt, argstr, idx, "%s", db->filename);
 203         add_arg(args, argcnt, argstr, idx, "--step");
 204         add_arg(args, argcnt, argstr, idx, "%d", db->interval);
 205
 206         for (i = 0; db->sources[i].type; i++) {
 207                 add_arg(args, argcnt, argstr, idx, "DS:%s:%s:%d:%f:%f",
 208                         db->sources[i].name,
 209                         db->sources[i].type,
 210                         db->sources[i].heartbeat,
 211                         db->sources[i].min,
 212                         db->sources[i].max);
 213         }
 214
 215         for (i = 0; db->archives[i].type; i++) {
 216                 add_arg(args, argcnt, argstr, idx, "RRA:%s:%f:%d:%d",
 217                         db->archives[i].type,
 218                         db->archives[i].xff,
 219                         db->archives[i].steps,
 220                         db->archives[i].rows);
 221         }
 222
 223         child = run(cmd, args);
 224
 225         harvest_zombies(child);
 226
 227         return 0;
 228 }
 229
 230 int rrdtool_check_databases(struct rrd_database *dbs[])
 231 {
 232         struct rrd_database *db;
 233         int i;
 234
 235         for (i = 0, db = dbs[i]; db; i++, db = dbs[i]) {
 236                 if (database_exists(db)) {
 237                         printf("database %s found\n", db->filename);
 238                         continue;
 239                 }
 240                 printf("Database %s missing, creating\n", db->filename);
 241                 create_database(db);
 242         }
 243         printf("All done\n");
 244
 245         return 0;
 246 }