src/utils_search.c

   1 /**
   2  * collection4 - utils_search.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  **/
  23
  24 #include <stdlib.h>
  25 #include <stdio.h>
  26 #include <string.h>
  27 #include <ctype.h>
  28 #include <errno.h>
  29
  30 #include "utils_search.h"
  31 #include "graph_ident.h"
  32 #include "graph_instance.h"
  33 #include "utils_array.h"
  34
  35 #include <fcgiapp.h>
  36 #include <fcgi_stdio.h>
  37
  38 struct search_info_s
  39 {
  40   char *host;
  41   char *plugin;
  42   char *plugin_instance;
  43   char *type;
  44   char *type_instance;
  45
  46   str_array_t *terms;
  47 };
  48
  49 /*
  50  * Private functions
  51  */
  52 static char *read_quoted_string (const char **buffer) /* {{{ */
  53 {
  54   const char *ptr = *buffer;
  55   char *ret;
  56   size_t ret_len;
  57
  58   if (ptr[0] != '"')
  59     return (NULL);
  60   ptr++;
  61
  62   ret_len = 0;
  63   while ((*ptr != '"') && (*ptr != 0))
  64   {
  65     ret_len++;
  66
  67     if (*ptr == '\\')
  68       ptr += 2;
  69     else
  70       ptr++;
  71   }
  72
  73   if ((ret_len < 1) || (*ptr != '"'))
  74     return (NULL);
  75
  76   ret = malloc (ret_len + 1);
  77   if (ret == NULL)
  78     return (NULL);
  79
  80   ptr = *buffer + 1;
  81   ret_len = 0;
  82   while ((*ptr != '"') && (*ptr != 0))
  83   {
  84     if (*ptr == '"')
  85       break;
  86
  87     if (*ptr == '\\')
  88       ptr++;
  89
  90     ret[ret_len] = *ptr;
  91
  92     ptr++;
  93     ret_len++;
  94   }
  95
  96   /* terminate string */
  97   ret[ret_len] = 0;
  98
  99   /* "ptr" points to the '"' sign, so advance one more */
 100   ptr++;
 101   *buffer = ptr;
 102
 103   return (ret);
 104 } /* }}} char *read_quoted_string */
 105
 106 static char *read_unquoted_word (const char **buffer) /* {{{ */
 107 {
 108   const char *ptr = *buffer;
 109   char *ret;
 110   size_t ret_len;
 111
 112   ret_len = 0;
 113   while (!isspace ((int) ptr[ret_len]) && (ptr[ret_len] != 0))
 114     ret_len++;
 115
 116   if (ret_len < 1)
 117     return (NULL);
 118
 119   ret = malloc (ret_len + 1);
 120   if (ret == NULL)
 121     return (NULL);
 122
 123   memcpy (ret, ptr, ret_len);
 124   ret[ret_len] = 0;
 125
 126   ptr += ret_len;
 127   *buffer = ptr;
 128
 129   return (ret);
 130 } /* }}} char *read_unquoted_word */
 131
 132 static char *next_token (const char **buffer) /* {{{ */
 133 {
 134   const char *ptr = *buffer;
 135   char *ret;
 136
 137   while (isspace ((int) (*ptr)))
 138     ptr++;
 139
 140   if (ptr[0] == 0)
 141     return (NULL);
 142   else if (ptr[0] == '"')
 143   {
 144     ret = read_quoted_string (&ptr);
 145     if (ret != NULL)
 146     {
 147       *buffer = ptr;
 148       return (ret);
 149     }
 150   }
 151
 152   ret = read_unquoted_word (&ptr);
 153   if (ret != NULL)
 154     *buffer = ptr;
 155
 156   return (ret);
 157 } /* }}} char *next_token */
 158
 159 static int store_token_field (char **field, const char *token) /* {{{ */
 160 {
 161   char *copy;
 162
 163   if ((field == NULL) || (token == NULL))
 164     return (EINVAL);
 165
 166   copy = strdup (token);
 167   if (copy == NULL)
 168     return (ENOMEM);
 169
 170   free (*field);
 171   *field = copy;
 172
 173   return (0);
 174 } /* }}} int store_token_field */
 175
 176 static int store_token (search_info_t *si, const char *token) /* {{{ */
 177 {
 178   if (strncmp ("host:", token, strlen ("host:")) == 0)
 179     return (store_token_field (&si->host, token + strlen ("host:")));
 180   else if (strncmp ("plugin:", token, strlen ("plugin:")) == 0)
 181     return (store_token_field (&si->plugin, token + strlen ("plugin:")));
 182   else if (strncmp ("plugin_instance:", token, strlen ("plugin_instance:")) == 0)
 183     return (store_token_field (&si->plugin_instance, token + strlen ("plugin_instance:")));
 184   else if (strncmp ("type:", token, strlen ("type:")) == 0)
 185     return (store_token_field (&si->type, token + strlen ("type:")));
 186   else if (strncmp ("type_instance:", token, strlen ("type_instance:")) == 0)
 187     return (store_token_field (&si->type_instance, token + strlen ("type_instance:")));
 188
 189   return (array_append (si->terms, token));
 190 } /* }}} int store_token */
 191
 192 /*
 193  * Public functions
 194  */
 195 search_info_t *search_parse (const char *search) /* {{{ */
 196 {
 197   const char *ptr;
 198   char *token;
 199   search_info_t *si;
 200
 201   si = malloc (sizeof (*si));
 202   if (si == NULL)
 203     return (NULL);
 204   memset (si, 0, sizeof (*si));
 205
 206   si->terms = array_create ();
 207   if (si->terms == NULL)
 208   {
 209     free (si);
 210     return (NULL);
 211   }
 212
 213   ptr = search;
 214
 215   while ((token = next_token (&ptr)) != NULL)
 216   {
 217     store_token (si, token);
 218     free (token);
 219   }
 220
 221   return (si);
 222 } /* }}} search_info_t *search_parse */
 223
 224 void search_destroy (search_info_t *si) /* {{{ */
 225 {
 226   if (si == NULL)
 227     return;
 228
 229   free (si->host);
 230   free (si->plugin);
 231   free (si->plugin_instance);
 232   free (si->type);
 233   free (si->type_instance);
 234
 235   array_destroy (si->terms);
 236 } /* }}} void search_destroy */
 237
 238 _Bool search_has_selector (search_info_t *si) /* {{{ */
 239 {
 240   if (si == NULL)
 241     return (0);
 242
 243   if ((si->host != NULL)
 244       || (si->plugin != NULL) || (si->plugin_instance != NULL)
 245       || (si->type != NULL) || (si->type_instance != NULL))
 246     return (1);
 247
 248   return (0);
 249 } /* }}} _Bool search_has_selector */
 250
 251 graph_ident_t *search_to_ident (search_info_t *si) /* {{{ */
 252 {
 253   if (si == NULL)
 254     return (NULL);
 255
 256   return (ident_create ((si->host == NULL) ? ANY_TOKEN : si->host,
 257         (si->plugin == NULL) ? ANY_TOKEN : si->plugin,
 258         (si->plugin_instance == NULL) ? ANY_TOKEN : si->plugin_instance,
 259         (si->type == NULL) ? ANY_TOKEN : si->type,
 260         (si->type_instance == NULL) ? ANY_TOKEN : si->type_instance));
 261 } /* }}} graph_ident_t *search_to_ident */
 262
 263 search_info_t *search_from_ident (const graph_ident_t *ident) /* {{{ */
 264 {
 265   search_info_t *si;
 266
 267   if (ident == NULL)
 268     return (NULL);
 269
 270   si = malloc (sizeof (*si));
 271   if (si == NULL)
 272     return (NULL);
 273   memset (si, 0, sizeof (*si));
 274   si->terms = NULL;
 275
 276 #define COPY_FIELD(f) do {                                                   \
 277   const char *tmp = ident_get_##f (ident);                                   \
 278   if (tmp == NULL)                                                           \
 279     si->f = NULL;                                                            \
 280   else                                                                       \
 281     si->f = strdup (tmp);                                                    \
 282 } while (0)
 283
 284   COPY_FIELD(host);
 285   COPY_FIELD(plugin);
 286   COPY_FIELD(plugin_instance);
 287   COPY_FIELD(type);
 288   COPY_FIELD(type_instance);
 289
 290 #undef COPY_FIELD
 291
 292   return (si);
 293 } /* }}} search_info_t *search_from_ident */
 294
 295 _Bool search_graph_title_matches (search_info_t *si, /* {{{ */
 296     const char *title)
 297 {
 298   char **argv;
 299   int argc;
 300   int i;
 301
 302   if ((si == NULL) || (title == NULL))
 303     return (0);
 304
 305   if (si->terms == NULL)
 306     return (1);
 307
 308   argc = array_argc (si->terms);
 309   argv = array_argv (si->terms);
 310   for (i = 0; i < argc; i++)
 311     if (strstr (title, argv[i]) == NULL)
 312       return (0);
 313
 314   return (1);
 315 } /* }}} _Bool search_graph_title_matches */
 316
 317 _Bool search_graph_inst_matches (search_info_t *si, /* {{{ */
 318     graph_config_t *cfg, graph_instance_t *inst,
 319     const char *title)
 320 {
 321   char **argv;
 322   int argc;
 323   int i;
 324
 325   if ((si == NULL) || (cfg == NULL) || (inst == NULL))
 326     return (0);
 327
 328   if ((si->host != NULL)
 329       && !inst_matches_field (inst, GIF_HOST, si->host))
 330     return (0);
 331   else if ((si->plugin != NULL)
 332       && !inst_matches_field (inst, GIF_PLUGIN, si->plugin))
 333     return (0);
 334   else if ((si->plugin_instance != NULL)
 335       && !inst_matches_field (inst, GIF_PLUGIN_INSTANCE, si->plugin_instance))
 336     return (0);
 337   else if ((si->type != NULL)
 338       && !inst_matches_field (inst, GIF_TYPE, si->type))
 339     return (0);
 340   else if ((si->type_instance != NULL)
 341       && !inst_matches_field (inst, GIF_TYPE_INSTANCE, si->type_instance))
 342     return (0);
 343
 344   if (si->terms == NULL)
 345     return (1);
 346
 347   argc = array_argc (si->terms);
 348   argv = array_argv (si->terms);
 349   for (i = 0; i < argc; i++)
 350   {
 351     if (inst_matches_string (cfg, inst, argv[i]))
 352       continue;
 353
 354     if ((title != NULL) && (strstr (title, argv[i]) != NULL))
 355       continue;
 356
 357     return (0);
 358   }
 359
 360   return (1);
 361 } /* }}} _Bool search_graph_inst_matches */
 362
 363 /* vim: set sw=2 sts=2 et fdm=marker : */