492e5e7245581233a7c158f26e3a90c05257bb5d
1 /**
2 * collection4 - action_graph.c
3 * Copyright (C) 2010 Florian octo Forster
4 *
5 * This program is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser General Public
7 * License as published by the Free Software Foundation; either
8 * version 2.1 of the License, or (at your option) any later version.
9 *
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Lesser General Public License for more details.
14 *
15 * You should have received a copy of the GNU Lesser General Public
16 * License along with this program; if not, write to the Free Software
17 * Foundation, Inc., 51 Franklin Street, Fifth Floor,
18 * Boston, MA 02110-1301 USA
19 *
20 * Authors:
21 * Florian octo Forster <ff at octo.it>
22 **/
24 #include <stdlib.h>
25 #include <stdio.h>
26 #include <string.h>
27 #include <errno.h>
28 #include <stdint.h>
29 #include <inttypes.h>
30 #include <dirent.h> /* for PATH_MAX */
31 #include <assert.h>
32 #include <math.h>
34 #include <rrd.h>
36 #include "common.h"
37 #include "action_graph.h"
38 #include "graph.h"
39 #include "graph_instance.h"
40 #include "graph_list.h"
41 #include "utils_cgi.h"
42 #include "utils_array.h"
44 #include <fcgiapp.h>
45 #include <fcgi_stdio.h>
47 struct graph_data_s
48 {
49 rrd_args_t *args;
50 rrd_info_t *info;
51 time_t mtime;
52 time_t expires;
53 long now;
54 long begin;
55 long end;
56 };
57 typedef struct graph_data_s graph_data_t;
59 static int get_time_args (graph_data_t *data) /* {{{ */
60 {
61 const char *begin_str;
62 const char *end_str;
63 long now;
64 long begin;
65 long end;
66 char *endptr;
67 long tmp;
69 begin_str = param ("begin");
70 end_str = param ("end");
72 now = (long) time (NULL);
73 data->now = now;
74 data->begin = now - 86400;
75 data->end = now;
77 if (begin_str != NULL)
78 {
79 endptr = NULL;
80 errno = 0;
81 tmp = strtol (begin_str, &endptr, /* base = */ 0);
82 if ((endptr == begin_str) || (errno != 0))
83 return (-1);
84 if (tmp <= 0)
85 begin = now + tmp;
86 else
87 begin = tmp;
88 }
89 else /* if (begin_str == NULL) */
90 {
91 begin = now - 86400;
92 }
94 if (end_str != NULL)
95 {
96 endptr = NULL;
97 errno = 0;
98 tmp = strtol (end_str, &endptr, /* base = */ 0);
99 if ((endptr == end_str) || (errno != 0))
100 return (-1);
101 end = tmp;
102 if (tmp <= 0)
103 end = now + tmp;
104 else
105 end = tmp;
106 }
107 else /* if (end_str == NULL) */
108 {
109 end = now;
110 }
112 if (begin == end)
113 return (-1);
115 if (begin > end)
116 {
117 tmp = begin;
118 begin = end;
119 end = tmp;
120 }
122 data->begin = begin;
123 data->end = end;
125 array_append (data->args->options, "-s");
126 array_append_format (data->args->options, "%li", begin);
127 array_append (data->args->options, "-e");
128 array_append_format (data->args->options, "%li", end);
130 return (0);
131 } /* }}} int get_time_args */
133 static void emulate_graph (int argc, char **argv) /* {{{ */
134 {
135 int i;
137 printf ("rrdtool \\\n");
138 for (i = 0; i < argc; i++)
139 {
140 if (i < (argc - 1))
141 printf (" \"%s\" \\\n", argv[i]);
142 else
143 printf (" \"%s\"\n", argv[i]);
144 }
145 } /* }}} void emulate_graph */
147 static int ag_info_print (rrd_info_t *info) /* {{{ */
148 {
149 if (info->type == RD_I_VAL)
150 printf ("[info] %s = %g;\n", info->key, info->value.u_val);
151 else if (info->type == RD_I_CNT)
152 printf ("[info] %s = %lu;\n", info->key, info->value.u_cnt);
153 else if (info->type == RD_I_STR)
154 printf ("[info] %s = %s;\n", info->key, info->value.u_str);
155 else if (info->type == RD_I_INT)
156 printf ("[info] %s = %i;\n", info->key, info->value.u_int);
157 else if (info->type == RD_I_BLO)
158 printf ("[info] %s = [blob, %lu bytes];\n", info->key, info->value.u_blo.size);
159 else
160 printf ("[info] %s = [unknown type %#x];\n", info->key, info->type);
162 return (0);
163 } /* }}} int ag_info_print */
165 static int output_graph (graph_data_t *data) /* {{{ */
166 {
167 rrd_info_t *img;
168 char time_buffer[256];
169 time_t expires;
170 int status;
172 for (img = data->info; img != NULL; img = img->next)
173 if ((strcmp ("image", img->key) == 0)
174 && (img->type == RD_I_BLO))
175 break;
177 if (img == NULL)
178 return (ENOENT);
180 printf ("Content-Type: image/png\n"
181 "Content-Length: %lu\n",
182 img->value.u_blo.size);
183 if (data->mtime > 0)
184 {
185 int status;
187 status = time_to_rfc1123 (data->mtime, time_buffer, sizeof (time_buffer));
188 if (status == 0)
189 printf ("Last-Modified: %s\n", time_buffer);
190 }
192 /* Print Expires header. */
193 if (data->end >= data->now)
194 {
195 /* The end of the timespan can be seen. */
196 long secs_per_pixel;
198 /* FIXME: Handle graphs with width != 400. */
199 secs_per_pixel = (data->end - data->begin) / 400;
201 expires = (time_t) (data->now + secs_per_pixel);
202 }
203 else /* if (data->end < data->now) */
204 {
205 expires = (time_t) (data->now + 86400);
206 }
207 status = time_to_rfc1123 (expires, time_buffer, sizeof (time_buffer));
208 if (status == 0)
209 printf ("Expires: %s\n", time_buffer);
211 printf ("\n");
213 fwrite (img->value.u_blo.ptr, img->value.u_blo.size,
214 /* nmemb = */ 1, stdout);
216 return (0);
217 } /* }}} int output_graph */
219 #define OUTPUT_ERROR(...) do { \
220 printf ("Content-Type: text/plain\n\n"); \
221 printf (__VA_ARGS__); \
222 return (0); \
223 } while (0)
225 int action_graph (void) /* {{{ */
226 {
227 graph_data_t data;
228 graph_config_t *cfg;
229 graph_instance_t *inst;
230 int status;
232 int argc;
233 char **argv;
235 cfg = gl_graph_get_selected ();
236 if (cfg == NULL)
237 OUTPUT_ERROR ("gl_graph_get_selected () failed.\n");
239 inst = inst_get_selected (cfg);
240 if (inst == NULL)
241 OUTPUT_ERROR ("inst_get_selected (%p) failed.\n", (void *) cfg);
243 data.args = ra_create ();
244 if (data.args == NULL)
245 return (ENOMEM);
247 array_append (data.args->options, "graph");
248 array_append (data.args->options, "-");
249 array_append (data.args->options, "--imgformat");
250 array_append (data.args->options, "PNG");
252 get_time_args (&data);
254 status = inst_get_rrdargs (cfg, inst, data.args);
255 if (status != 0)
256 {
257 ra_destroy (data.args);
258 OUTPUT_ERROR ("inst_get_rrdargs failed with status %i.\n", status);
259 }
261 argc = ra_argc (data.args);
262 argv = ra_argv (data.args);
263 if ((argc < 0) || (argv == NULL))
264 {
265 ra_destroy (data.args);
266 return (-1);
267 }
269 rrd_clear_error ();
270 data.info = rrd_graph_v (argc, argv);
271 if ((data.info == NULL) || rrd_test_error ())
272 {
273 printf ("Content-Type: text/plain\n\n");
274 printf ("rrd_graph_v failed: %s\n", rrd_get_error ());
275 emulate_graph (argc, argv);
276 }
277 else
278 {
279 int status;
281 data.mtime = inst_get_mtime (inst);
283 status = output_graph (&data);
284 if (status != 0)
285 {
286 rrd_info_t *ptr;
288 printf ("Content-Type: text/plain\n\n");
289 printf ("output_graph failed. Maybe the \"image\" info was not found?\n\n");
291 for (ptr = data.info; ptr != NULL; ptr = ptr->next)
292 {
293 ag_info_print (ptr);
294 }
295 }
296 }
298 if (data.info != NULL)
299 rrd_info_free (data.info);
301 ra_argv_free (argv);
302 ra_destroy (data.args);
303 data.args = NULL;
305 return (0);
306 } /* }}} int action_graph */
308 /* vim: set sw=2 sts=2 et fdm=marker : */