GlibDynamicLibraryLoader.cc

// -*- tab-width: 2; indent-tabs-mode: nil; coding: utf-8-with-signature -*-
//-----------------------------------------------------------------------------
// Copyright 2000-2025 CEA (www.cea.fr) IFPEN (www.ifpenergiesnouvelles.com)
// See the top-level COPYRIGHT file for details.
// SPDX-License-Identifier: Apache-2.0
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* GlibDynamicLibraryLoader.cc                                 (C) 2000-2025 */
/*                                                                           */
/* Chargeur dynamique de bibliothèque avec Glib (utilise gmodule).           */
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arccore/base/String.h"
#include "arccore/base/PlatformUtils.h"
#include "arccore/base/ArrayView.h"
#include "arccore/base/internal/IDynamicLibraryLoader.h"
 
#include "gmodule.h"
 
#include <iostream>
#include <set>
#include <vector>
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane
{
 
class GlibDynamicLibraryLoader;
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
class GlibDynamicLibrary
: public IDynamicLibrary
{
 public:
 
  GlibDynamicLibrary(GlibDynamicLibraryLoader* mng, GModule* gmodule)
  : m_manager(mng)
  , m_gmodule(gmodule)
  {}
 
 public:
 
  void close() override;
  void* getSymbolAddress(const String& symbol_name, bool* is_found) override;
 
 private:
 
  GlibDynamicLibraryLoader* m_manager = nullptr;
  GModule* m_gmodule = nullptr;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
class GlibDynamicLibraryLoader
: public IDynamicLibraryLoader
{
 public:
 
  GlibDynamicLibraryLoader()
  {
    String s = platform::getEnvironmentVariable("ARCANE_VERBOSE_DYNAMICLIBRARY");
    if (s == "1" || s == "true")
      m_is_verbose = true;
  }
 
  void build() override {}
 
  IDynamicLibrary* open(const String& directory, const String& name) override
  {
    IDynamicLibrary* dl = _tryOpen(directory, name);
    if (!dl) {
      // Si on ne trouve pas, essaie avec l'extension '.dll' ou '.dylib' car sous
      // windows ou macos, certaines version de la GLIB prefixent automatiquement le
      // nom de la bibliothèque par 'lib' si elle ne finit pas par '.dll' ou '.dylib'.
#ifdef ARCCORE_OS_WIN32
      dl = _tryOpen(directory, name + ".dll");
#endif
#ifdef ARCCORE_OS_MACOS
      dl = _tryOpen(directory, "lib" + name + ".dylib");
#endif
    }
    if (!dl) {
      // Si on ne trouve pas, essaie en cherchant à côté du binaire
      dl = _tryOpen(".", name);
    }
    if (!dl) {
      // Si on ne trouve pas, essaie en cherchant à côté du binaire
      // et avec l'extension dll ou dylib
#ifdef ARCCORE_OS_WIN32
      dl = _tryOpen(".", name + ".dll");
#endif
#ifdef ARCCORE_OS_MACOS
      dl = _tryOpen(".", "lib" + name + ".dylib");
#endif
    }
    return dl;
  }
 
  IDynamicLibrary* _tryOpen(const String& directory, const String& name)
  {
    const gchar* gdirectory = reinterpret_cast<const gchar*>(directory.utf8().data());
    const gchar* gname = reinterpret_cast<const gchar*>(name.utf8().data());
    gchar* full_path = g_module_build_path(gdirectory, gname);
    if (m_is_verbose) {
      std::cout << "** Load Dynamic Library '" << full_path << "'...";
    }
    GModule* gmodule = g_module_open(full_path, GModuleFlags());
    g_free(full_path);
    if (m_is_verbose) {
      if (!gmodule) {
        std::cout << " NOT FOUND\n";
      }
      else {
        std::cout << " OK\n";
      }
    }
    if (!gmodule)
      return nullptr;
    auto lib = new GlibDynamicLibrary(this, gmodule);
    m_opened_libraries.insert(lib);
    return lib;
  }
 
  void closeLibraries() override
  {
    // Cette méthode va modifier m opened libraries donc il faut le copier avant.
    std::vector<GlibDynamicLibrary*> libs(m_opened_libraries.begin(), m_opened_libraries.end());
    for (auto lib : libs) {
      lib->close();
      delete lib;
    }
  }
 
  void removeInstance(GlibDynamicLibrary* lib)
  {
    auto iter = m_opened_libraries.find(lib);
    if (iter != m_opened_libraries.end())
      m_opened_libraries.erase(iter);
  }
 
 private:
 
  bool m_is_verbose = false;
  std::set<GlibDynamicLibrary*> m_opened_libraries;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
void GlibDynamicLibrary::
close()
{
  if (!m_gmodule)
    return;
  bool is_ok = g_module_close(m_gmodule);
  m_gmodule = 0;
  if (!is_ok)
    std::cerr << "WARNING: can not unload module\n";
  m_manager->removeInstance(this);
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
void* GlibDynamicLibrary::
getSymbolAddress(const String& symbol_name, bool* is_found)
{
  if (is_found)
    *is_found = false;
  if (!m_gmodule)
    return nullptr;
  const gchar* gname = reinterpret_cast<const gchar*>(symbol_name.utf8().data());
  void* symbol_addr = nullptr;
  bool r = ::g_module_symbol(m_gmodule, gname, &symbol_addr);
  if (is_found)
    (*is_found) = r;
  return symbol_addr;
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
extern "C++" ARCCORE_BASE_EXPORT IDynamicLibraryLoader*
createGlibDynamicLibraryLoader()
{
  IDynamicLibraryLoader* idll = new GlibDynamicLibraryLoader();
  idll->build();
  return idll;
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
namespace
{
  GlibDynamicLibraryLoader glib_dynamic_loader;
  IDynamicLibraryLoader* global_default_loader = &glib_dynamic_loader;
}
 
IDynamicLibraryLoader* IDynamicLibraryLoader::getDefault()
{
  return global_default_loader;
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // namespace Arcane
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/