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 uint32_t get_random_color (void) /* {{{ */
 163 {
 164   double hsv[3] = { 0.0, 1.0, 1.0 };
 165   double rgb[3] = { 0.0, 0.0, 0.0 };
 166
 167   hsv[0] = 360.0 * ((double) rand ()) / (((double) RAND_MAX) + 1.0);
 168
 169   hsv_to_rgb (hsv, rgb);
 170
 171   return (rgb_to_uint32 (rgb));
 172 } /* }}} uint32_t get_random_color */
 173
 174 int print_debug (const char *format, ...) /* {{{ */
 175 {
 176   static _Bool have_header = 0;
 177
 178   va_list ap;
 179   int status;
 180
 181   if (!have_header)
 182   {
 183     printf ("Content-Type: text/plain\n\n");
 184     have_header = 1;
 185   }
 186
 187   va_start (ap, format);
 188   status = vprintf (format, ap);
 189   va_end (ap);
 190
 191   return (status);
 192 } /* }}} int print_debug */
 193
 194 char *strtolower (char *str) /* {{{ */
 195 {
 196   unsigned int i;
 197
 198   if (str == NULL)
 199     return (NULL);
 200
 201   for (i = 0; str[i] != 0; i++)
 202     str[i] = (char) tolower ((int) str[i]);
 203
 204   return (str);
 205 } /* }}} char *strtolower */
 206
 207 char *strtolower_copy (const char *str)
 208 {
 209   if (str == NULL)
 210     return (NULL);
 211
 212   return (strtolower (strdup (str)));
 213 }
 214
 215 /* vim: set sw=2 sts=2 et fdm=marker : */