DsoUtils.h

Go to the documentation of this file.

//  note: This, I believe, should be part of Reflex::SharedLibrary

#ifndef GAUDIKERNEL_DSOUTILS_H
#define GAUDIKERNEL_DSOUTILS_H

// STL includes
#include <string>

#include "GaudiKernel/System.h"

namespace DsoUtils
{

  inline std::string libNativeName( const std::string& libName )
  {
#if defined( _WIN32 )
    return libName + ".dll";
#elif defined( __linux ) || defined( __APPLE__ )
    return "lib" + libName + ".so";
#else
    // variant of the GIGO design pattern
    return libName;
#endif
  }

#ifdef _GNU_SOURCE
#include <dlfcn.h>
  static std::string dsoName( void* addr )
  {
    Dl_info info;
    if ( dladdr( addr, &info ) == 0 ) return "";

    const char* pos = strrchr( info.dli_fname, '/' );
    if ( pos )
      ++pos;
    else
      pos = info.dli_fname;
    return pos;
  }
#elif defined( _WIN32 )
#include <windows.h>

  static std::string dsoName( void* addr )
  {
    if ( addr ) {
      MEMORY_BASIC_INFORMATION mbi;
      if ( VirtualQuery( addr, &mbi, sizeof( mbi ) ) ) {
        HMODULE h_module = (HMODULE)mbi.AllocationBase;
        char mod[1024];
        if ( GetModuleFileName( h_module, mod, sizeof( mod ) ) ) {
          const char* pos = strrchr( mod, '\\' );
          if ( pos )
            ++pos;
          else
            pos = mod;
          return pos;
        }
      }
    }
    return "";
  }

#else // dummy implementation for unknown platforms
  static std::string dsoName( const ROOT::Reflex::Member& ) { return ""; }

#endif

  static bool inDso( void* addr, const std::string& dsoname )
  {
#ifdef _WIN32
    char sep = '\\';
#else
    char sep = '/';
#endif

    std::string srcname = dsoName( addr );
    if ( srcname.empty() ) {
      // we do not know the name of the library, let's guess it's OK
      return true;
    }

    std::string::size_type pos = dsoname.find_last_of( sep );
    std::string curname;
    if ( std::string::npos == pos ) {
      curname = dsoname;
    } else {
      curname = dsoname.substr( pos + 1 );
    }

    return srcname == curname;
  }

} // end namespace DsoUtils

#endif // not GAUDIKERNEL_DSOUTILS_H