]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - HLT/BASE/AliHLTHOMERLibManager.cxx
Ignore
[u/mrichter/AliRoot.git] / HLT / BASE / AliHLTHOMERLibManager.cxx
index 3ec3698f836cff9cde911b94d611bd374c821f0f..44ddbcce0564f442c8b094a09d24902d5872a7b7 100644 (file)
 #include "AliHLTHOMERLibManager.h"
 #include "AliHLTHOMERReader.h"
 #include "AliHLTHOMERWriter.h"
+#include "AliHLTLogging.h"
 #include "TString.h"
 #include "TSystem.h"
 
 /** ROOT macro for the implementation of ROOT specific class methods */
 ClassImp(AliHLTHOMERLibManager)
 
+// This list must be NULL terminated, since we use it as a marker to identify
+// the end of the list.
+const char* AliHLTHOMERLibManager::fgkLibraries[] = {"libAliHLTHOMER.so", "libHOMER.so", NULL};
+// The size of the list of reference counts must be one less than fgkLibraries.
+int AliHLTHOMERLibManager::fgkLibRefCount[] = {0, 0};
+
 AliHLTHOMERLibManager::AliHLTHOMERLibManager()
   :
   fLibraryStatus(0),
@@ -46,7 +53,8 @@ AliHLTHOMERLibManager::AliHLTHOMERLibManager()
   fFctCreateReaderFromBuffer(NULL),
   fFctDeleteReader(NULL),
   fFctCreateWriter(NULL),
-  fFctDeleteWriter(NULL)
+  fFctDeleteWriter(NULL),
+  fLoadedLib(NULL)
 {
   // see header file for class documentation
   // or
@@ -58,6 +66,7 @@ AliHLTHOMERLibManager::AliHLTHOMERLibManager()
 AliHLTHOMERLibManager::~AliHLTHOMERLibManager()
 {
   // see header file for class documentation
+  UnloadHOMERLibrary();
 }
 
 AliHLTHOMERReader* AliHLTHOMERLibManager::OpenReader(const char* hostname, unsigned short port )
@@ -71,7 +80,7 @@ AliHLTHOMERReader* AliHLTHOMERLibManager::OpenReader(const char* hostname, unsig
   
   AliHLTHOMERReader* pReader=NULL;
   if (fFctCreateReaderFromTCPPort!=NULL && (pReader=(((AliHLTHOMERReaderCreateFromTCPPort_t)fFctCreateReaderFromTCPPort)(hostname, port)))==NULL) {
-    HLTError("can not create instance of HOMER reader (function %p)", fFctCreateReaderFromTCPPort);
+    cout <<"can not create instance of HOMER reader from ports" << endl;
   }
   
   return pReader;
@@ -88,13 +97,14 @@ AliHLTHOMERReader* AliHLTHOMERLibManager::OpenReader(unsigned int tcpCnt, const
   
   AliHLTHOMERReader* pReader=NULL;
   if (fFctCreateReaderFromTCPPorts!=NULL && (pReader=(((AliHLTHOMERReaderCreateFromTCPPorts_t)fFctCreateReaderFromTCPPorts)(tcpCnt, hostnames, ports)))==NULL) {
-    HLTError("can not create instance of HOMER reader (function %p)", fFctCreateReaderFromTCPPorts);
+    //HLTError("can not create instance of HOMER reader (function %p)", fFctCreateReaderFromTCPPorts);
+    cout << "can not create instance of HOMER reader from port"<<endl;
   }
   
   return pReader;
 }
 
-AliHLTHOMERReader* AliHLTHOMERLibManager::OpenReader(const AliHLTUInt8_t* pBuffer, int size)
+AliHLTHOMERReader* AliHLTHOMERLibManager::OpenReaderBuffer(const AliHLTUInt8_t* pBuffer, int size)
 {
   // see header file for class documentation
   if (fLibraryStatus<0) return NULL;
@@ -105,7 +115,7 @@ AliHLTHOMERReader* AliHLTHOMERLibManager::OpenReader(const AliHLTUInt8_t* pBuffe
   
   AliHLTHOMERReader* pReader=NULL;
   if (fFctCreateReaderFromBuffer!=NULL && (pReader=(((AliHLTHOMERReaderCreateFromBuffer_t)fFctCreateReaderFromBuffer)(pBuffer, size)))==NULL) {
-    HLTError("can not create instance of HOMER reader (function %p)", fFctCreateReaderFromBuffer);
+    //HLTError("can not create instance of HOMER reader (function %p)", fFctCreateReaderFromBuffer);
   }
   
   return pReader;
@@ -137,9 +147,9 @@ AliHLTHOMERWriter* AliHLTHOMERLibManager::OpenWriter()
   }
   
   AliHLTHOMERWriter* pWriter=NULL;
-//   if (fFctCreateWriter!=NULL && (pWriter=(((AliHLTHOMERWriterCreate_t)fFctCreateWriter)()))==NULL) {
+  if (fFctCreateWriter!=NULL && (pWriter=(((AliHLTHOMERWriterCreate_t)fFctCreateWriter)()))==NULL) {
 //     HLTError("can not create instance of HOMER writer (function %p)", fFctCreateWriter);
-//   }
+  }
   
   return pWriter;
 }
@@ -164,38 +174,45 @@ int AliHLTHOMERLibManager::LoadHOMERLibrary()
 {
   // see header file for class documentation
   int iResult=-EBADF;
-  const char* libraries[]={"libAliHLTHOMER.so", "libHOMER.so", NULL};
-  const char** library=&libraries[0];
+  const char** library=&fgkLibraries[0];
+  int* refcount = &fgkLibRefCount[0];
   do {
     TString libs = gSystem->GetLibraries();
-    if (libs.Contains(*library) ||
-       (gSystem->Load(*library)) >= 0) {
+    if (libs.Contains(*library)) {
+      iResult=1;
+      break;
+    }
+    if ((gSystem->Load(*library)) >= 0) {
+      ++(*refcount);
+      fLoadedLib = *library;
       iResult=1;
       break;
     }
-  } while (*(++library)!=NULL);
+    ++library;
+    ++refcount;
+  } while ((*library)!=NULL);
 
   if (iResult>0 && *library!=NULL) {
     // print compile info
     typedef void (*CompileInfo)( char*& date, char*& time);
     CompileInfo fctInfo=(CompileInfo)gSystem->DynFindSymbol(*library, "CompileInfo");
     if (fctInfo) {
-      char* date="";
-      char* time="";
+      char* date=0;
+      char* time=0;
       (*fctInfo)(date, time);
-      if (!date) date="unknown";
-      if (!time) time="unknown";
-      HLTInfo("%s build on %s (%s)", *library, date, time);
+      if (!date) {date=new Char_t[8]; strcpy(date,"unknown");}
+      if (!time) {time=new Char_t[8]; strcpy(time,"unknown");}
+      //HLTInfo("%s build on %s (%s)", *library, date, time);
     } else {
-      HLTInfo("no build info available for %s", *library);
+      //HLTInfo("no build info available for %s", *library);
     }
 
-    fFctCreateReaderFromTCPPort=gSystem->DynFindSymbol(*library, ALIHLTHOMERREADER_CREATE_FROM_TCPPORT);
-    fFctCreateReaderFromTCPPorts=gSystem->DynFindSymbol(*library, ALIHLTHOMERREADER_CREATE_FROM_TCPPORTS);
-    fFctCreateReaderFromBuffer=gSystem->DynFindSymbol(*library, ALIHLTHOMERREADER_CREATE_FROM_BUFFER);
-    fFctDeleteReader=gSystem->DynFindSymbol(*library, ALIHLTHOMERREADER_DELETE);
-    fFctCreateWriter=gSystem->DynFindSymbol(*library, ALIHLTHOMERWRITER_CREATE);
-    fFctDeleteWriter=gSystem->DynFindSymbol(*library, ALIHLTHOMERWRITER_DELETE);
+    fFctCreateReaderFromTCPPort=(void (*)())gSystem->DynFindSymbol(*library, ALIHLTHOMERREADER_CREATE_FROM_TCPPORT);
+    fFctCreateReaderFromTCPPorts=(void (*)())gSystem->DynFindSymbol(*library, ALIHLTHOMERREADER_CREATE_FROM_TCPPORTS);
+    fFctCreateReaderFromBuffer=(void (*)())gSystem->DynFindSymbol(*library, ALIHLTHOMERREADER_CREATE_FROM_BUFFER);
+    fFctDeleteReader=(void (*)())gSystem->DynFindSymbol(*library, ALIHLTHOMERREADER_DELETE);
+    fFctCreateWriter=(void (*)())gSystem->DynFindSymbol(*library, ALIHLTHOMERWRITER_CREATE);
+    fFctDeleteWriter=(void (*)())gSystem->DynFindSymbol(*library, ALIHLTHOMERWRITER_DELETE);
     if (fFctCreateReaderFromTCPPort==NULL ||
        fFctCreateReaderFromTCPPorts==NULL ||
        fFctCreateReaderFromBuffer==NULL || 
@@ -204,7 +221,6 @@ int AliHLTHOMERLibManager::LoadHOMERLibrary()
        fFctDeleteWriter==NULL) {
       iResult=-ENOSYS;
     } else {
-      //HLTDebug("%s: entries found", *library);
     }
   }
   if (iResult<0 || *library==NULL) {
@@ -218,3 +234,71 @@ int AliHLTHOMERLibManager::LoadHOMERLibrary()
 
   return iResult;
 }
+
+int AliHLTHOMERLibManager::UnloadHOMERLibrary()
+{
+  // see header file for class documentation
+  int iResult=0;
+  
+  if (fLoadedLib != NULL)
+  {
+    // Find the corresponding reference count.
+    const char** library=&fgkLibraries[0];
+    int* refcount = &fgkLibRefCount[0];
+    while (*library != NULL)
+    {
+      if (strcmp(*library, fLoadedLib) == 0) break;
+      ++library;
+      ++refcount;
+    }
+    
+    // Decrease the reference count and remove the library if it is zero.
+    if (*refcount >= 0) --(*refcount);
+    if (*refcount == 0)
+    {
+      // Check that the library we are trying to unload is actually the last library
+      // in the gSystem->GetLibraries() list. If not then we must abort the removal.
+      // This is because of a ROOT bug/feature/limitation. If we try unload the library
+      // then ROOT will also wipe all libraries in the gSystem->GetLibraries() list
+      // following the library we want to unload.
+      TString libstring = gSystem->GetLibraries();
+      TString token, lastlib;
+      Ssiz_t from = 0;
+      Int_t numOfLibs = 0, posOfLib = -1;
+      while (libstring.Tokenize(token, from, " "))
+      {
+        ++numOfLibs;
+        lastlib = token;
+        if (token.Contains(fLoadedLib)) posOfLib = numOfLibs;
+      }
+      if (numOfLibs == posOfLib)
+      {
+        gSystem->Unload(fLoadedLib);
+
+        // Check that the library is gone, since Unload() does not return a status code.
+        libstring = gSystem->GetLibraries();
+        if (libstring.Contains(fLoadedLib)) iResult = -EBADF;
+      }
+      else
+      {
+        AliHLTLogging log;
+        log.LoggingVarargs(kHLTLogWarning, Class_Name(), FUNCTIONNAME(), __FILE__, __LINE__,
+          Form("ROOT limitation! Cannot properly cleanup and unload the shared"
+            " library '%s' since another library '%s' was loaded afterwards. Trying to"
+            " unload this library will remove the others and lead to serious memory faults.",
+            fLoadedLib, lastlib.Data()
+        ));
+      }
+    }
+  }
+
+  // Clear the function pointers.
+  fFctCreateReaderFromTCPPort = NULL;
+  fFctCreateReaderFromTCPPorts = NULL;
+  fFctCreateReaderFromBuffer = NULL;
+  fFctDeleteReader = NULL;
+  fFctCreateWriter = NULL;
+  fFctDeleteWriter = NULL;
+
+  return iResult;
+}