Module.Linux.cpp

#include "Module.h"
#include "Logging/Logger.h"
 
#include <array>
#include <span>
#include <string_view>
 
#include <dlfcn.h>
 
namespace {
 
  const Cogs::Logging::Log logger = Cogs::Logging::getLogger("Module");
 
  std::span<const char*> dllVersions() {
#if defined( DEBUG )
    static std::array paths = { ".x64.Debug.so" };
    return std::span<const char*>(paths);
#elif defined( OSOMP_EnableProfiler )
    static std::array paths = { ".x64.ReleaseMemProfile.so", ".x64.Release.so" };
    return std::span<const char*>(paths);
#elif defined( ASAN )
    static std::array paths = { ".x64.ReleaseASan.so", ".x64.Release.so" };
    return std::span<const char*>(paths);
#elif defined( TSAN )
    static std::array paths = { ".x64.ReleaseTSan.so", ".x64.Release.so" };
    return std::span<const char*>(paths);
#else
    static std::array paths = { ".x64.Release.so" };
    return std::span<const char*>(paths);
#endif
  }
}
 
void Cogs::Module::load(const std::string& path, LoadStatus& status, void*& handle) {
  if (path.find(".so") == std::string::npos) {
 
    std::span<const char*> versions = dllVersions();
    std::string libraryPath;
    libraryPath.reserve(path.size() + 40);
 
    for (const char* version : versions) {
      libraryPath.clear();
      libraryPath += "lib";
      libraryPath += path;
      libraryPath += version;
 
      handle = ::dlopen(libraryPath.c_str(), RTLD_NOW | RTLD_LOCAL);
      if (handle) break;
    }
  }
  else {
    handle = ::dlopen(path.c_str(), RTLD_NOW | RTLD_LOCAL);
  }
  if(handle == nullptr) {
    LOG_ERROR(logger, "dlopen failed: %s", dlerror());
  }
  status = handle ? LoadStatus::Loaded : LoadStatus::Failed;
}
 
void Cogs::Module::unload(void* handle) {
  if (handle) {
    dlclose(handle);
  }
}
 
void* Cogs::Module::getProcAddress(void* handle, const char* procName) {
  return dlsym(handle, procName);
}