Initial commit of vpp code.

Change-Id: Ib246f1fbfce93274020ee93ce461e3d8bd8b9f17
Signed-off-by: Ed Warnicke <eaw@cisco.com>

1 files changed, 191 insertions, 0 deletions

diff --git a/vpp-api-test/vat/plugin.c b/vpp-api-test/vat/plugin.c
new file mode 100644
index 00000000..dca46953
--- /dev/null
+++ b/vpp-api-test/vat/plugin.c
@@ -0,0 +1,191 @@
+/*
+ * Copyright (c) 2015 Cisco and/or its affiliates.
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+/*
+ * plugin.c: plugin handling
+ */
+
+#include <vat/vat.h>
+#include <vat/plugin.h>
+#include <dlfcn.h>
+#include <dirent.h>
+
+plugin_main_t vat_plugin_main;
+
+static int 
+load_one_plugin (plugin_main_t *pm, plugin_info_t *pi)
+{
+  void *handle, *register_handle;
+  clib_error_t * (*fp)(vat_main_t *);
+  clib_error_t * error;
+  
+  handle = dlopen ((char *)pi->name, RTLD_LAZY);
+
+  /* 
+   * Note: this can happen if the plugin has an undefined symbol reference,
+   * so print a warning. Otherwise, the poor slob won't know what happened.
+   * Ask me how I know that...
+   */
+  if (handle == 0)
+    {
+      clib_warning ("%s", dlerror());
+      return -1;
+    }
+  
+  pi->handle = handle;
+
+  register_handle = dlsym (pi->handle, "vat_plugin_register");
+  if (register_handle == 0)
+    return 0;
+
+  fp = register_handle;
+
+  error = (*fp)(pm->vat_main);
+
+  if (error)
+    {
+      clib_error_report (error);
+      dlclose (handle);
+      return 1;
+    }
+
+  clib_warning ("Loaded plugin: %s", pi->name);
+
+  return 0;
+}
+
+static u8 **split_plugin_path (plugin_main_t *pm)
+{
+  int i;
+  u8 **rv = 0;
+  u8 *path = pm->plugin_path;
+  u8 *this = 0;
+
+  for (i = 0; i < vec_len (pm->plugin_path); i++)
+    {
+      if (path[i] != ':')
+        {
+          vec_add1(this, path[i]);
+          continue;
+        }
+      vec_add1(this, 0);
+      vec_add1 (rv, this);
+      this = 0;
+    }
+  if (this)
+    {
+      vec_add1 (this, 0);
+      vec_add1 (rv, this);
+    }
+  return rv;
+}
+
+int vat_load_new_plugins (plugin_main_t *pm)
+{
+  DIR *dp;
+  struct dirent *entry;
+  struct stat statb;
+  uword *p;
+  plugin_info_t *pi;
+  u8 **plugin_path;
+  int i;
+
+  plugin_path = split_plugin_path (pm);
+  
+  for (i = 0; i < vec_len (plugin_path); i++)
+    {
+      dp = opendir ((char *)plugin_path[i]);
+  
+      if (dp == 0)
+        continue;
+      
+      while ((entry = readdir (dp)))
+        {
+          u8 *plugin_name;
+          
+          if (pm->plugin_name_filter)
+            {
+              int j;
+              for (j = 0; j < vec_len (pm->plugin_name_filter); j++)
+                if (entry->d_name[j] != pm->plugin_name_filter[j])
+                  goto next;
+            }
+
+          plugin_name = format (0, "%s/%s%c", plugin_path[i],
+                                entry->d_name, 0);
+          
+          /* unreadable */
+          if (stat ((char *)plugin_name, &statb) < 0)
+            {
+            ignore:
+              vec_free (plugin_name);
+              continue;
+            }
+          
+          /* a dir or other things which aren't plugins */
+          if (!S_ISREG(statb.st_mode))
+            goto ignore;
+          
+          p = hash_get_mem (pm->plugin_by_name_hash, plugin_name);
+          if (p == 0) 
+            {
+              vec_add2 (pm->plugin_info, pi, 1);
+              pi->name = plugin_name;
+              pi->file_info = statb;
+              
+              if (load_one_plugin (pm, pi))
+                {
+                  vec_free (plugin_name);
+                  _vec_len (pm->plugin_info) = vec_len (pm->plugin_info) - 1;
+                  continue;
+                }
+              memset (pi, 0, sizeof (*pi));
+              hash_set_mem (pm->plugin_by_name_hash, plugin_name, 
+                            pi - pm->plugin_info);
+            }
+        next:
+          ;
+        }
+      closedir (dp);
+      vec_free (plugin_path[i]);
+    }
+  vec_free (plugin_path);
+  return 0;
+}
+
+#define QUOTE_(x) #x
+#define QUOTE(x) QUOTE_(x)
+
+/* 
+ * Load plugins from /usr/lib/vpp_api_test_plugins by default
+ */
+char *vat_plugin_path = "/usr/lib/vpp_api_test_plugins";
+                         
+char *vat_plugin_name_filter = 0;
+
+int vat_plugin_init (vat_main_t * vam)
+{
+  plugin_main_t *pm = &vat_plugin_main;
+  
+  clib_warning ("plugin-path %s", vat_plugin_path);
+
+  pm->plugin_path = format (0, "%s%c", vat_plugin_path, 0);
+  if (vat_plugin_name_filter)
+    pm->plugin_name_filter = format (0, "%s%c", vat_plugin_name_filter, 0);
+
+  pm->plugin_by_name_hash = hash_create_string (0, sizeof (uword));
+  pm->vat_main = vam;
+  
+  return vat_load_new_plugins (pm);
+}