src/utils_array.c: Fix a segfault in "array_prepend".
[collection4.git] / src / common.c
1 #include <stdlib.h>
2 #include <stdio.h>
3 #include <stdint.h>
4 #include <inttypes.h>
5 #include <string.h>
6 #include <ctype.h>
7 #include <errno.h>
8 #include <sys/types.h>
9 #include <sys/stat.h>
10 #include <unistd.h>
11 #include <dirent.h>
12 #include <assert.h>
13 #include <math.h>
14
15 #include <rrd.h>
16
17 #include "common.h"
18 #include "graph_list.h"
19
20 #include <fcgiapp.h>
21 #include <fcgi_stdio.h>
22
23 size_t c_strlcat (char *dst, const char *src, size_t size) /* {{{ */
24 {
25   size_t retval;
26   size_t dst_len;
27   size_t src_len;
28
29   dst_len = strlen (dst);
30   src_len = strlen (src);
31   retval = dst_len + src_len;
32
33   if ((dst_len + 1) >= size)
34     return (retval);
35
36   dst  += dst_len;
37   size -= dst_len;
38   assert (size >= 2);
39
40   /* Result will be truncated. */
41   if (src_len >= size)
42     src_len = size - 1;
43
44   memcpy (dst, src, src_len);
45   dst[src_len] = 0;
46
47   return (retval);
48 } /* }}} size_t c_strlcat */
49
50 int ds_list_from_rrd_file (char *file, /* {{{ */
51     size_t *ret_dses_num, char ***ret_dses)
52 {
53   char *rrd_argv[] = { "info", file, NULL };
54   int rrd_argc = (sizeof (rrd_argv) / sizeof (rrd_argv[0])) - 1;
55
56   rrd_info_t *info;
57   rrd_info_t *ptr;
58
59   char **dses = NULL;
60   size_t dses_num = 0;
61
62   info = rrd_info (rrd_argc, rrd_argv);
63   if (info == NULL)
64   {
65     printf ("%s: rrd_info (%s) failed.\n", __func__, file);
66     return (-1);
67   }
68
69   for (ptr = info; ptr != NULL; ptr = ptr->next)
70   {
71     size_t keylen;
72     size_t dslen;
73     char *ds;
74     char **tmp;
75
76     if (strncmp ("ds[", ptr->key, strlen ("ds[")) != 0)
77       continue;
78
79     keylen = strlen (ptr->key);
80     if (keylen < strlen ("ds[?].index"))
81       continue;
82
83     dslen = keylen - strlen ("ds[].index");
84     assert (dslen >= 1);
85
86     if (strcmp ("].index", ptr->key + (strlen ("ds[") + dslen)) != 0)
87       continue;
88
89     ds = malloc (dslen + 1);
90     if (ds == NULL)
91       continue;
92
93     memcpy (ds, ptr->key + strlen ("ds["), dslen);
94     ds[dslen] = 0;
95
96     tmp = realloc (dses, sizeof (*dses) * (dses_num + 1));
97     if (tmp == NULL)
98     {
99       free (ds);
100       continue;
101     }
102     dses = tmp;
103
104     dses[dses_num] = ds;
105     dses_num++;
106   }
107
108   rrd_info_free (info);
109
110   if (dses_num < 1)
111   {
112     assert (dses == NULL);
113     return (ENOENT);
114   }
115
116   *ret_dses_num = dses_num;
117   *ret_dses = dses;
118
119   return (0);
120 } /* }}} int ds_list_from_rrd_file */
121
122 static int hsv_to_rgb (double *hsv, double *rgb) /* {{{ */
123 {
124   double c = hsv[2] * hsv[1];
125   double h = hsv[0] / 60.0;
126   double x = c * (1.0 - fabs (fmod (h, 2.0) - 1));
127   double m = hsv[2] - c;
128
129   rgb[0] = 0.0;
130   rgb[1] = 0.0;
131   rgb[2] = 0.0;
132
133        if ((0.0 <= h) && (h < 1.0)) { rgb[0] = 1.0; rgb[1] = x; rgb[2] = 0.0; }
134   else if ((1.0 <= h) && (h < 2.0)) { rgb[0] = x; rgb[1] = 1.0; rgb[2] = 0.0; }
135   else if ((2.0 <= h) && (h < 3.0)) { rgb[0] = 0.0; rgb[1] = 1.0; rgb[2] = x; }
136   else if ((3.0 <= h) && (h < 4.0)) { rgb[0] = 0.0; rgb[1] = x; rgb[2] = 1.0; }
137   else if ((4.0 <= h) && (h < 5.0)) { rgb[0] = x; rgb[1] = 0.0; rgb[2] = 1.0; }
138   else if ((5.0 <= h) && (h < 6.0)) { rgb[0] = 1.0; rgb[1] = 0.0; rgb[2] = x; }
139
140   rgb[0] += m;
141   rgb[1] += m;
142   rgb[2] += m;
143
144   return (0);
145 } /* }}} int hsv_to_rgb */
146
147 static uint32_t rgb_to_uint32 (double *rgb) /* {{{ */
148 {
149   uint8_t r;
150   uint8_t g;
151   uint8_t b;
152
153   r = (uint8_t) (255.0 * rgb[0]);
154   g = (uint8_t) (255.0 * rgb[1]);
155   b = (uint8_t) (255.0 * rgb[2]);
156
157   return ((((uint32_t) r) << 16)
158       | (((uint32_t) g) << 8)
159       | ((uint32_t) b));
160 } /* }}} uint32_t rgb_to_uint32 */
161
162 static int uint32_to_rgb (uint32_t color, double *rgb)
163 {
164   uint8_t r;
165   uint8_t g;
166   uint8_t b;
167
168   r = (uint8_t) ((color >> 16) & 0x00ff);
169   g = (uint8_t) ((color >>  8) & 0x00ff);
170   b = (uint8_t) ((color >>  0) & 0x00ff);
171
172   rgb[0] = ((double) r) / 255.0;
173   rgb[1] = ((double) g) / 255.0;
174   rgb[2] = ((double) b) / 255.0;
175
176   return (0);
177 } /* }}} int uint32_to_rgb */
178
179 uint32_t get_random_color (void) /* {{{ */
180 {
181   double hsv[3] = { 0.0, 1.0, 1.0 };
182   double rgb[3] = { 0.0, 0.0, 0.0 };
183
184   hsv[0] = 360.0 * ((double) rand ()) / (((double) RAND_MAX) + 1.0);
185
186   hsv_to_rgb (hsv, rgb);
187
188   return (rgb_to_uint32 (rgb));
189 } /* }}} uint32_t get_random_color */
190
191 uint32_t fade_color (uint32_t color) /* {{{ */
192 {
193   double rgb[3];
194
195   uint32_to_rgb (color, rgb);
196   rgb[0] = 1.0 - ((1.0 - rgb[0]) * 0.1);
197   rgb[1] = 1.0 - ((1.0 - rgb[1]) * 0.1);
198   rgb[2] = 1.0 - ((1.0 - rgb[2]) * 0.1);
199
200   return (rgb_to_uint32 (rgb));
201 } /* }}} uint32_t fade_color */
202
203 int print_debug (const char *format, ...) /* {{{ */
204 {
205   static _Bool have_header = 0;
206
207   va_list ap;
208   int status;
209
210   if (!have_header)
211   {
212     printf ("Content-Type: text/plain\n\n");
213     have_header = 1;
214   }
215
216   va_start (ap, format);
217   status = vprintf (format, ap);
218   va_end (ap);
219
220   return (status);
221 } /* }}} int print_debug */
222
223 char *strtolower (char *str) /* {{{ */
224 {
225   unsigned int i;
226
227   if (str == NULL)
228     return (NULL);
229
230   for (i = 0; str[i] != 0; i++)
231     str[i] = (char) tolower ((int) str[i]);
232
233   return (str);
234 } /* }}} char *strtolower */
235
236 char *strtolower_copy (const char *str)
237 {
238   if (str == NULL)
239     return (NULL);
240
241   return (strtolower (strdup (str)));
242 }
243
244 /* vim: set sw=2 sts=2 et fdm=marker : */