X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=src%2Fgraph_list.c;h=2112c76faa88dc7b84623df2ee975ee93aaa6259;hb=99f31aafefdd00b788d9ab2ba74b1cfb70c679e3;hp=f5cb2e9895e1096f77a2eb9e64ea622a2694bed7;hpb=721000defaa6fa0b5e27bbf80d0896798368ef98;p=collection4.git

diff --git a/src/graph_list.c b/src/graph_list.c
index f5cb2e9..2112c76 100644
--- a/src/graph_list.c
+++ b/src/graph_list.c
@@ -28,6 +28,11 @@
 #include <string.h>
 #include <time.h>
 #include <errno.h>
+#include <assert.h>
+#include <unistd.h>
+#include <fcntl.h>
+
+#include <yajl/yajl_parse.h>
 
 #include "graph_list.h"
 #include "common.h"
@@ -36,7 +41,9 @@
 #include "graph_config.h"
 #include "graph_def.h"
 #include "graph_ident.h"
+#include "graph_instance.h"
 #include "utils_cgi.h"
+#include "utils_search.h"
 
 #include <fcgiapp.h>
 #include <fcgi_stdio.h>
@@ -45,6 +52,7 @@
  * Defines
  */
 #define UPDATE_INTERVAL 900
+#define CACHE_FILE "/tmp/collection4.json"
 
 /*
  * Global variables
@@ -173,7 +181,7 @@ static int gl_register_file (const graph_ident_t *file, /* {{{ */
     graph_config_t *cfg = gl_active[i];
     int status;
 
-    if (!graph_matches_ident (cfg, file))
+    if (!graph_ident_matches (cfg, file))
       continue;
 
     status = graph_add_file (cfg, file);
@@ -221,6 +229,352 @@ static int gl_clear_instances (void) /* {{{ */
   return (0);
 } /* }}} int gl_clear_instances */
 
+static void gl_dump_cb (void *ctx, /* {{{ */
+    const char *str, unsigned int len)
+{
+  FILE *fh = ctx;
+
+  /* FIXME: Has everything been written? */
+  fwrite ((void *) str, /* size = */ 1, /* nmemb = */ len, fh);
+} /* }}} void gl_dump_cb */
+
+static int gl_dump (void) /* {{{ */
+{
+  FILE *fh;
+  yajl_gen handler;
+  yajl_gen_config handler_config = { /* pretty = */ 1, /* indent = */ "  " };
+  size_t i;
+
+  /* FIXME: Lock the file */
+  fh = fopen (CACHE_FILE, "w");
+  if (fh == NULL)
+    return (errno);
+
+  handler = yajl_gen_alloc2 (gl_dump_cb, &handler_config,
+      /* alloc funcs = */ NULL, /* ctx = */ fh);
+  if (handler == NULL)
+  {
+    fclose (fh);
+    return (-1);
+  }
+
+  yajl_gen_array_open (handler);
+
+  for (i = 0; i < gl_active_num; i++)
+    graph_to_json (gl_active[i], handler);
+
+  for (i = 0; i < gl_dynamic_num; i++)
+    graph_to_json (gl_dynamic[i], handler);
+
+  yajl_gen_array_close (handler);
+
+  yajl_gen_free (handler);
+  fclose (fh);
+
+  return (0);
+} /* }}} int gl_dump */
+
+/*
+ * JSON parsing functions
+ */
+#define CTX_MASK                  0xff000000
+#define CTX_GRAPH                 0x01000000
+#define CTX_GRAPH_SELECT          0x02000000
+#define CTX_INST                  0x03000000
+#define CTX_INST_SELECT           0x04000000
+#define CTX_INST_FILE             0x05000000
+
+#define CTX_IDENT_MASK            0x00ff0000
+#define CTX_IDENT_HOST            0x00010000
+#define CTX_IDENT_PLUGIN          0x00020000
+#define CTX_IDENT_PLUGIN_INSTANCE 0x00030000
+#define CTX_IDENT_TYPE            0x00040000
+#define CTX_IDENT_TYPE_INSTANCE   0x00050000
+
+struct gl_json_context_s
+{
+  uint32_t          state;
+
+  graph_config_t   *cfg;
+  graph_instance_t *inst;
+  graph_ident_t    *ident;
+
+  _Bool             dynamic_graph;
+};
+typedef struct gl_json_context_s gl_json_context_t;
+
+static void set_state (gl_json_context_t *ctx, /* {{{ */
+    uint32_t new_state, uint32_t mask)
+{
+  uint32_t old_state = ctx->state;
+  ctx->state = (old_state & ~mask) | (new_state & mask);
+} /* }}} void set_state */
+
+static int gl_json_string (void *user_data, /* {{{ */
+    const unsigned char *str,
+    unsigned int str_length)
+{
+  gl_json_context_t *ctx = user_data;
+  char buffer[str_length + 1];
+
+  memcpy (buffer, str, str_length);
+  buffer[str_length] = 0;
+
+  if (((ctx->state & CTX_MASK) == CTX_GRAPH_SELECT)
+      || ((ctx->state & CTX_MASK) == CTX_INST_SELECT)
+      || ((ctx->state & CTX_MASK) == CTX_INST_FILE))
+  {
+    switch (ctx->state & CTX_IDENT_MASK)
+    {
+      case CTX_IDENT_HOST:
+        ident_set_host (ctx->ident, buffer);
+        break;
+      case CTX_IDENT_PLUGIN:
+        ident_set_plugin (ctx->ident, buffer);
+        break;
+      case CTX_IDENT_PLUGIN_INSTANCE:
+        ident_set_plugin_instance (ctx->ident, buffer);
+        break;
+      case CTX_IDENT_TYPE:
+        ident_set_type (ctx->ident, buffer);
+        break;
+      case CTX_IDENT_TYPE_INSTANCE:
+        ident_set_type_instance (ctx->ident, buffer);
+        break;
+    }
+  }
+
+  return (1);
+} /* }}} int gl_json_string */
+
+static int gl_json_end_map (void *user_data) /* {{{ */
+{
+  gl_json_context_t *ctx = user_data;
+
+  if ((ctx->state & CTX_MASK) == CTX_GRAPH_SELECT)
+  {
+    size_t i;
+
+    /* ctx->ident should now hold the valid selector */
+    assert (ctx->cfg == NULL);
+    assert (ctx->inst  == NULL);
+    assert (ctx->ident != NULL);
+
+    for (i = 0; i < gl_active_num; i++)
+    {
+      if (graph_compare (gl_active[i], ctx->ident) != 0)
+        continue;
+
+      ctx->cfg = gl_active[i];
+      ctx->dynamic_graph = 0;
+      break;
+    }
+
+    if (ctx->cfg == NULL)
+    {
+      ctx->cfg = graph_create (ctx->ident);
+      ctx->dynamic_graph = 1;
+    }
+
+    ident_destroy (ctx->ident);
+    ctx->ident = NULL;
+
+    set_state (ctx, CTX_GRAPH, CTX_MASK);
+  }
+  else if ((ctx->state & CTX_MASK) == CTX_INST_SELECT)
+  {
+    /* ctx->ident should now hold the valid selector */
+    assert (ctx->cfg   != NULL);
+    assert (ctx->inst  == NULL);
+    assert (ctx->ident != NULL);
+
+    ctx->inst = inst_create (ctx->cfg, ctx->ident);
+    ident_destroy (ctx->ident);
+    ctx->ident = NULL;
+
+    set_state (ctx, CTX_INST, CTX_MASK);
+  }
+  else if ((ctx->state & CTX_MASK) == CTX_INST_FILE)
+  {
+    /* ctx->ident should now hold the valid file */
+    assert (ctx->cfg   != NULL);
+    assert (ctx->inst  != NULL);
+    assert (ctx->ident != NULL);
+
+    inst_add_file (ctx->inst, ctx->ident);
+    ident_destroy (ctx->ident);
+    ctx->ident = NULL;
+
+    /* Don't reset the state here, files are in an array. */
+  }
+  else if ((ctx->state & CTX_MASK) == CTX_INST)
+  {
+    /* ctx->inst should now hold a complete instance */
+    assert (ctx->cfg   != NULL);
+    assert (ctx->inst  != NULL);
+    assert (ctx->ident == NULL);
+
+    graph_add_inst (ctx->cfg, ctx->inst);
+    /* don't destroy / free ctx->inst */
+    ctx->inst = NULL;
+
+    /* Don't reset the state here, instances are in an array. */
+  }
+  else if ((ctx->state & CTX_MASK) == CTX_GRAPH)
+  {
+    /* ctx->cfg should now hold a complete graph */
+    assert (ctx->cfg   != NULL);
+    assert (ctx->inst  == NULL);
+    assert (ctx->ident == NULL);
+
+    if (ctx->dynamic_graph)
+      gl_add_graph_internal (ctx->cfg, &gl_dynamic, &gl_dynamic_num);
+    /* else: already contained in gl_active */
+    ctx->cfg = NULL;
+
+    /* Don't reset the state here, graphs are in an array. */
+  }
+
+  return (1);
+} /* }}} int gl_json_end_map */
+
+static int gl_json_end_array (void *user_data) /* {{{ */
+{
+  gl_json_context_t *ctx = user_data;
+
+  if ((ctx->state & CTX_MASK) == CTX_INST_FILE)
+    set_state (ctx, CTX_INST, CTX_MASK);
+  else if ((ctx->state & CTX_MASK) == CTX_INST)
+    set_state (ctx, CTX_GRAPH, CTX_MASK);
+  else if ((ctx->state & CTX_MASK) == CTX_GRAPH)
+  {
+    /* We're done */
+  }
+
+  return (1);
+} /* }}} int gl_json_end_array */
+
+static int gl_json_key (void *user_data, /* {{{ */
+    const unsigned char *str,
+    unsigned int str_length)
+{
+  gl_json_context_t *ctx = user_data;
+  char buffer[str_length + 1];
+
+  memcpy (buffer, str, str_length);
+  buffer[str_length] = 0;
+
+  if ((ctx->state & CTX_MASK) == CTX_GRAPH)
+  {
+    if (strcasecmp ("select", buffer) == 0)
+      set_state (ctx, CTX_GRAPH_SELECT, CTX_MASK);
+    else if (strcasecmp ("instances", buffer) == 0)
+      set_state (ctx, CTX_INST, CTX_MASK);
+  }
+  else if ((ctx->state & CTX_MASK) == CTX_INST)
+  {
+    if (strcasecmp ("select", buffer) == 0)
+      set_state (ctx, CTX_INST_SELECT, CTX_MASK);
+    else if (strcasecmp ("files", buffer) == 0)
+      set_state (ctx, CTX_INST_FILE, CTX_MASK);
+  }
+  else if (((ctx->state & CTX_MASK) == CTX_GRAPH_SELECT)
+      || ((ctx->state & CTX_MASK) == CTX_INST_SELECT)
+      || ((ctx->state & CTX_MASK) == CTX_INST_FILE))
+  {
+    if (strcasecmp ("host", buffer) == 0)
+      set_state (ctx, CTX_IDENT_HOST, CTX_IDENT_MASK);
+    else if (strcasecmp ("plugin", buffer) == 0)
+      set_state (ctx, CTX_IDENT_PLUGIN, CTX_IDENT_MASK);
+    else if (strcasecmp ("plugin_instance", buffer) == 0)
+      set_state (ctx, CTX_IDENT_PLUGIN_INSTANCE, CTX_IDENT_MASK);
+    else if (strcasecmp ("type", buffer) == 0)
+      set_state (ctx, CTX_IDENT_TYPE, CTX_IDENT_MASK);
+    else if (strcasecmp ("type_instance", buffer) == 0)
+      set_state (ctx, CTX_IDENT_TYPE_INSTANCE, CTX_IDENT_MASK);
+  }
+
+  return (0);
+} /* }}} int gl_json_key */
+
+yajl_callbacks gl_json_callbacks =
+{
+  /*        null = */ NULL,
+  /*     boolean = */ NULL,
+  /*     integer = */ NULL,
+  /*      double = */ NULL,
+  /*      number = */ NULL,
+  /*      string = */ gl_json_string,
+  /*   start_map = */ NULL,
+  /*     map_key = */ gl_json_key,
+  /*     end_map = */ gl_json_end_map,
+  /* start_array = */ NULL,
+  /*   end_array = */ gl_json_end_array
+};
+
+static int gl_read_cache (_Bool block) /* {{{ */
+{
+  yajl_handle handle;
+  gl_json_context_t context;
+  yajl_parser_config handle_config = { /* comments = */ 0, /* check UTF-8 */ 0 };
+
+  int fd;
+  int cmd;
+  struct flock lock;
+  int status;
+
+  fd = open (CACHE_FILE, O_RDONLY);
+  if (fd < 0)
+  {
+    fprintf (stderr, "gl_read_cache: open(2) failed with status %i\n", errno);
+    return (errno);
+  }
+
+  if (block)
+    cmd = F_SETLKW;
+  else
+    cmd = F_SETLK;
+
+  memset (&lock, 0, sizeof (lock));
+  lock.l_type = F_RDLCK;
+  lock.l_whence = SEEK_SET;
+  lock.l_start = 0;
+  lock.l_len = 0; /* lock everything */
+
+  while (42)
+  {
+    status = fcntl (fd, cmd, &lock);
+    if (status == 0)
+      break;
+
+    if (!block && ((errno == EACCES) || (errno == EAGAIN)))
+    {
+      close (fd);
+      return (0);
+    }
+
+    if (errno == EINTR)
+      continue;
+
+    fprintf (stderr, "gl_read_cache: fcntl(2) failed with status %i\n",
+        errno);
+    return (errno);
+  }
+
+  memset (&context, 0, sizeof (context));
+  context.cfg = NULL;
+  context.inst = NULL;
+  context.ident = NULL;
+
+  handle = yajl_alloc (&gl_json_callbacks,
+      &handle_config,
+      /* alloc funcs = */ NULL,
+      &context);
+
+  close (fd);
+
+} /* }}} int gl_read_cache */
+
 /*
  * Global functions
  */
@@ -381,16 +735,37 @@ int gl_instance_get_all (graph_inst_callback_t callback, /* {{{ */
 } /* }}} int gl_instance_get_all */
 /* }}} gl_instance_get_all, gl_graph_instance_get_all */
 
-int gl_search (const char *term, graph_inst_callback_t callback, /* {{{ */
-    void *user_data)
+int gl_search (search_info_t *si, /* {{{ */
+    graph_inst_callback_t callback, void *user_data)
 {
   size_t i;
+  graph_ident_t *ident;
+
+  if ((si == NULL) || (callback == NULL))
+    return (EINVAL);
+
+  if (search_has_selector (si))
+  {
+    ident = search_to_ident (si);
+    if (ident == NULL)
+    {
+      fprintf (stderr, "gl_search: search_to_ident failed\n");
+      return (-1);
+    }
+  }
+  else
+  {
+    ident = NULL;
+  }
 
   for (i = 0; i < gl_active_num; i++)
   {
     int status;
 
-    status = graph_inst_search (gl_active[i], term,
+    if ((ident != NULL) && !graph_ident_intersect (gl_active[i], ident))
+      continue;
+
+    status = graph_search_inst (gl_active[i], si,
         /* callback  = */ callback,
         /* user data = */ user_data);
     if (status != 0)
@@ -401,7 +776,10 @@ int gl_search (const char *term, graph_inst_callback_t callback, /* {{{ */
   {
     int status;
 
-    status = graph_inst_search (gl_dynamic[i], term,
+    if ((ident != NULL) && !graph_ident_intersect (gl_dynamic[i], ident))
+      continue;
+
+    status = graph_search_inst (gl_dynamic[i], si,
         /* callback  = */ callback,
         /* user data = */ user_data);
     if (status != 0)
@@ -411,6 +789,36 @@ int gl_search (const char *term, graph_inst_callback_t callback, /* {{{ */
   return (0);
 } /* }}} int gl_search */
 
+int gl_search_string (const char *term, graph_inst_callback_t callback, /* {{{ */
+    void *user_data)
+{
+  size_t i;
+
+  for (i = 0; i < gl_active_num; i++)
+  {
+    int status;
+
+    status = graph_search_inst_string (gl_active[i], term,
+        /* callback  = */ callback,
+        /* user data = */ user_data);
+    if (status != 0)
+      return (status);
+  }
+
+  for (i = 0; i < gl_dynamic_num; i++)
+  {
+    int status;
+
+    status = graph_search_inst_string (gl_dynamic[i], term,
+        /* callback  = */ callback,
+        /* user data = */ user_data);
+    if (status != 0)
+      return (status);
+  }
+
+  return (0);
+} /* }}} int gl_search_string */
+
 int gl_search_field (graph_ident_field_t field, /* {{{ */
     const char *field_value,
     graph_inst_callback_t callback, void *user_data)
@@ -467,6 +875,7 @@ int gl_update (void) /* {{{ */
 {
   time_t now;
   int status;
+  size_t i;
 
   /*
   printf ("Content-Type: text/plain\n\n");
@@ -492,6 +901,11 @@ int gl_update (void) /* {{{ */
 
   gl_last_update = now;
 
+  for (i = 0; i < gl_active_num; i++)
+    graph_sort_instances (gl_active[i]);
+
+  gl_dump ();
+
   return (status);
 } /* }}} int gl_update */