common.c

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