Line data    Source code

       1             : /* +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
       2             :    Copyright (c) 2011-2023 The plumed team
       3             :    (see the PEOPLE file at the root of the distribution for a list of names)
       4             : 
       5             :    See http://www.plumed.org for more information.
       6             : 
       7             :    This file is part of plumed, version 2.
       8             : 
       9             :    plumed is free software: you can redistribute it and/or modify
      10             :    it under the terms of the GNU Lesser General Public License as published by
      11             :    the Free Software Foundation, either version 3 of the License, or
      12             :    (at your option) any later version.
      13             : 
      14             :    plumed is distributed in the hope that it will be useful,
      15             :    but WITHOUT ANY WARRANTY; without even the implied warranty of
      16             :    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
      17             :    GNU Lesser General Public License for more details.
      18             : 
      19             :    You should have received a copy of the GNU Lesser General Public License
      20             :    along with plumed.  If not, see <http://www.gnu.org/licenses/>.
      21             : +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ */
      22             : #include "DLLoader.h"
      23             : 
      24             : #include "Exception.h"
      25             : #include <cstdlib>
      26             : #include <iostream>
      27             : #include "core/ActionRegister.h"
      28             : #include "core/CLToolRegister.h"
      29             : 
      30             : #ifdef __PLUMED_HAS_DLOPEN
      31             : #include <dlfcn.h>
      32             : #endif
      33             : 
      34             : namespace PLMD {
      35             : 
      36          42 : bool DLLoader::installed() {
      37             : #ifdef __PLUMED_HAS_DLOPEN
      38          42 :   return true;
      39             : #else
      40             :   return false;
      41             : #endif
      42             : }
      43             : 
      44             : 
      45          51 : void* DLLoader::load(const std::string&s) {
      46             : #ifdef __PLUMED_HAS_DLOPEN
      47          51 :   auto lockerAction=Register::registrationLock(s);
      48          51 :   void* p=dlopen(s.c_str(),RTLD_NOW|RTLD_LOCAL);
      49          51 :   if(!p) {
      50           0 :     plumed_error()<<"Could not load library "<<s<<"\n"<<dlerror();
      51             :   }
      52          51 :   handles.push_back(p);
      53          51 :   Register::completeAllRegistrations(p);
      54          51 :   return p;
      55             : #else
      56             :   plumed_error()<<"you are trying to use dlopen but it's not configured on your system";
      57             : #endif
      58          51 : }
      59             : 
      60      812284 : DLLoader::~DLLoader() {
      61      812284 :   auto debug=std::getenv("PLUMED_LOAD_DEBUG");
      62             : #ifdef __PLUMED_HAS_DLOPEN
      63      812284 :   if(debug) {
      64           0 :     std::fprintf(stderr,"delete dlloader\n");
      65             :   }
      66      812335 :   while(!handles.empty()) {
      67          51 :     int ret=dlclose(handles.back());
      68          51 :     if(ret) {
      69           0 :       std::fprintf(stderr,"+++ error reported by dlclose: %s\n",dlerror());
      70             :     }
      71             :     handles.pop_back();
      72             :   }
      73      812284 :   if(debug) {
      74           0 :     std::fprintf(stderr,"end delete dlloader\n");
      75             :   }
      76             : #endif
      77      812284 : }
      78             : 
      79      812284 : DLLoader::DLLoader() {
      80             :   // do nothing
      81      812284 : }
      82             : 
      83       55012 : const std::vector<void*> & DLLoader::getHandles() const noexcept {
      84       55012 :   return handles;
      85             : }
      86             : 
      87           0 : DLLoader::EnsureGlobalDLOpen::EnsureGlobalDLOpen(const void *symbol) noexcept {
      88             : #ifdef __PLUMED_HAS_DLOPEN
      89             : #ifdef __PLUMED_HAS_DLADDR
      90             :   Dl_info info;
      91             :   // from the manual:
      92             :   // If the address specified in addr could not be matched to a shared
      93             :   //      object, then these functions return 0.  In this case, an error
      94             :   //      message is not available via dlerror(3).
      95           0 :   int zeroIsError=dladdr(symbol, &info);
      96           0 :   if(zeroIsError!=0) {
      97             :     //This "promotes" to GLOBAL the object with the symbol pointed by ptr
      98           0 :     handle_ = dlopen(info.dli_fname, RTLD_GLOBAL | RTLD_NOW);
      99             :   } else {
     100           0 :     std::fprintf(stderr,
     101             :                  "+++WARNING+++"
     102             :                  "Failure in finding any object that contains the symbol %p.\n",
     103             :                  symbol);
     104             :   }
     105             : #else
     106             :   std::fprintf(stderr,
     107             :                "+++WARNING+++"
     108             :                "I can't use dladdr for promoting the library containing the symbol %p.\n"
     109             :                "This system seems not to support dladdr",
     110             :                symbol);
     111             : #endif //__PLUMED_HAS_DLADDR
     112             : #endif //__PLUMED_HAS_DLOPEN
     113           0 : }
     114             : 
     115           0 : DLLoader::EnsureGlobalDLOpen::~EnsureGlobalDLOpen() {
     116             : #ifdef __PLUMED_HAS_DLOPEN
     117           0 :   if (handle_) {
     118           0 :     dlclose(handle_);
     119             :   }
     120             : #endif //__PLUMED_HAS_DLOPEN
     121           0 : }
     122             : 
     123           0 : bool DLLoader::isPlumedGlobal() {
     124             : #if defined(__APPLE__)
     125             :   bool result;
     126             : #ifdef __PLUMED_HAS_DLOPEN
     127             :   void* handle;
     128             : #if defined(__PLUMED_HAS_RTLD_DEFAULT)
     129             :   handle=RTLD_DEFAULT;
     130             : #else
     131             :   handle=dlopen(NULL,RTLD_LOCAL);
     132             : #endif
     133             :   // we check for two variants, see wrapper/Plumed.h for an explanation:
     134             :   result=dlsym(handle,"plumed_plumedmain_create") || dlsym(handle,"plumedmain_create");
     135             :   if(handle) {
     136             :     dlclose(handle);
     137             :   }
     138             : #else
     139             :   // if a system cannot use dlopen, we assume plumed is globally available
     140             :   result=true;
     141             : #endif //__PLUMED_HAS_DLOPEN
     142             :   return result;
     143             : #else
     144           0 :   plumed_error()<<"DLLoader::isPlumedGlobal() is only functional with APPLE dlsym";
     145             : #endif
     146             : }
     147             : 
     148             : } // namespace PLMD