spike_main/extensions.cc

   1 #include "extension.h"
   2 #include <string>
   3 #include <map>
   4 #include <dlfcn.h>
   5
   6 static std::map<std::string, std::function<extension_t*()>>& extensions()
   7 {
   8   static std::map<std::string, std::function<extension_t*()>> v;
   9   return v;
  10 }
  11
  12 void register_extension(const char* name, std::function<extension_t*()> f)
  13 {
  14   extensions()[name] = f;
  15 }
  16
  17 std::function<extension_t*()> find_extension(const char* name)
  18 {
  19   if (!extensions().count(name)) {
  20     // try to find extension xyz by loading libxyz.so
  21     std::string libname = std::string("lib") + name + ".so";
  22     if (!dlopen(libname.c_str(), RTLD_LAZY)) {
  23       fprintf(stderr, "couldn't find extension '%s' (or library '%s')\n",
  24               name, libname.c_str());
  25       exit(-1);
  26     }
  27     if (!extensions().count(name)) {
  28       fprintf(stderr, "couldn't find extension '%s' in shared library '%s'\n",
  29               name, libname.c_str());
  30       exit(-1);
  31     }
  32   }
  33
  34   return extensions()[name];
  35 }