| [658] | 1 | // This may look like C code, but it is really -*- C++ -*-
 | 
|---|
 | 2 | // Gestionnaire de lien dynamique   -   R. Ansari 12/98
 | 
|---|
 | 3 | // LAL (Orsay) / IN2P3-CNRS  DAPNIA/SPP (Saclay) / CEA
 | 
|---|
 | 4 | 
 | 
|---|
 | 5 | #include <stdio.h>
 | 
|---|
 | 6 | #include <stdlib.h>
 | 
|---|
 | 7 | 
 | 
|---|
 | 8 | #include <iostream.h>
 | 
|---|
 | 9 | 
 | 
|---|
 | 10 | #include "pdlmgr.h"
 | 
|---|
 | 11 | 
 | 
|---|
 | 12 | // Extension de noms de fichiers Shared libs
 | 
|---|
 | 13 | static const char* sofext = ".so";
 | 
|---|
 | 14 | static const char* sofext_HPUX = ".sl";
 | 
|---|
 | 15 | 
 | 
|---|
 | 16 | // Variables et methodes static
 | 
|---|
 | 17 | int PDynLinkMgr::numSO = 0;
 | 
|---|
 | 18 | string*  PDynLinkMgr::tmpDir = NULL;
 | 
|---|
 | 19 | 
 | 
|---|
 | 20 | /* --Methode-Static-- */
 | 
|---|
 | 21 | void PDynLinkMgr::SetTmpDir(string const & path)
 | 
|---|
 | 22 | {
 | 
|---|
 | 23 | if ( (path.length() > 0) && (path[path.length()] != '/')  ) GetTmpDir() = path + '/';
 | 
|---|
 | 24 | else GetTmpDir() = path;
 | 
|---|
 | 25 | #if defined(OSF1) || defined(Linux) || defined(SunOS)
 | 
|---|
 | 26 |   string cmd = "LD_LIBRARY_PATH=";
 | 
|---|
 | 27 |   char* varenv=NULL;
 | 
|---|
 | 28 |   varenv=getenv("LD_LIBRARY_PATH");
 | 
|---|
 | 29 | #elif defined(IRIX64)
 | 
|---|
 | 30 |   string cmd = "LD_LIBRARYN32_PATH=";
 | 
|---|
 | 31 |   char* varenv=NULL;
 | 
|---|
 | 32 |   varenv=getenv("LD_LIBRARYN32_PATH");
 | 
|---|
 | 33 | 
 | 
|---|
 | 34 |   if (varenv == NULL) { 
 | 
|---|
 | 35 |     cmd += '.';  
 | 
|---|
 | 36 |     if (path.length() > 0)  cmd += ':' + path; 
 | 
|---|
 | 37 |     }
 | 
|---|
 | 38 |   else { 
 | 
|---|
 | 39 |     if (varenv[0] != '.') cmd += ".:";
 | 
|---|
 | 40 |     if (path.length() > 0)  cmd += path + ':';
 | 
|---|
 | 41 |     cmd += varenv;
 | 
|---|
 | 42 |     putenv(cmd.c_str());
 | 
|---|
 | 43 |     }
 | 
|---|
 | 44 | #elif defined(AIX)
 | 
|---|
 | 45 |   string cmd = "LIBPATH=";
 | 
|---|
 | 46 |   char* varenv=NULL;
 | 
|---|
 | 47 |   varenv=getenv("LIBPATH");
 | 
|---|
 | 48 |   if (varenv == NULL) { 
 | 
|---|
 | 49 |     cmd += '.';  
 | 
|---|
 | 50 |     if (path.length() > 0)  cmd += ':' + path; 
 | 
|---|
 | 51 |     cmd += ":/usr/lib:/lib";
 | 
|---|
 | 52 |     }
 | 
|---|
 | 53 |   else { 
 | 
|---|
 | 54 |     if (varenv[0] != '.') cmd += ".:";
 | 
|---|
 | 55 |     if (path.length() > 0)  cmd += path + ':';
 | 
|---|
 | 56 |     cmd += varenv;
 | 
|---|
 | 57 |     putenv(const_cast<char *>(cmd.c_str()));
 | 
|---|
 | 58 |     }
 | 
|---|
 | 59 | 
 | 
|---|
 | 60 | #endif
 | 
|---|
 | 61 | return;
 | 
|---|
 | 62 | }
 | 
|---|
 | 63 | 
 | 
|---|
 | 64 | /* --Methode-Static-- */
 | 
|---|
 | 65 | string& PDynLinkMgr::GetTmpDir()
 | 
|---|
 | 66 | {
 | 
|---|
 | 67 | if (tmpDir == NULL) { 
 | 
|---|
 | 68 |   tmpDir = new string("");
 | 
|---|
 | 69 |   char* varenv;
 | 
|---|
 | 70 |   if ( (varenv=getenv("PEIDA_TMP")) != NULL )  *tmpDir = varenv; 
 | 
|---|
 | 71 |   else if ( (varenv=getenv("TMPDIR")) != NULL )  *tmpDir = varenv; 
 | 
|---|
 | 72 |   }
 | 
|---|
 | 73 | return(*tmpDir);
 | 
|---|
 | 74 | }
 | 
|---|
 | 75 | 
 | 
|---|
 | 76 | /* --Methode-Static-- */
 | 
|---|
 | 77 | PDynLinkMgr* PDynLinkMgr::BuildFromCFile(string const & fname)
 | 
|---|
 | 78 | {
 | 
|---|
 | 79 | size_t l = fname.length();
 | 
|---|
 | 80 | if (l < 1)  return(NULL);
 | 
|---|
 | 81 | string fnameobj = GetTmpDir()+"tmp_pdl.o";
 | 
|---|
 | 82 | 
 | 
|---|
 | 83 | string cmd;
 | 
|---|
 | 84 | int rc;
 | 
|---|
 | 85 | 
 | 
|---|
 | 86 | // Compilation du fichier
 | 
|---|
 | 87 | #ifndef __mac__
 | 
|---|
 | 88 | cmd = "cc -c -o " + fnameobj + " " + fname;
 | 
|---|
 | 89 | #else 
 | 
|---|
 | 90 | cmd = "Il faut compiler !!!" + fnameobj + " " + fname;
 | 
|---|
 | 91 | #endif
 | 
|---|
 | 92 | rc = system(cmd.c_str());
 | 
|---|
 | 93 | if (rc != 0)  { 
 | 
|---|
 | 94 |   cerr << "PDynLinkMgr::BuildFromCFile() Error Rc(" << cmd <<")= "<< rc << endl; 
 | 
|---|
 | 95 |   return(NULL);
 | 
|---|
 | 96 |   }
 | 
|---|
 | 97 | 
 | 
|---|
 | 98 | char buff[32];
 | 
|---|
 | 99 | numSO++;
 | 
|---|
 | 100 | #ifndef HPUX
 | 
|---|
 | 101 | sprintf(buff,"pdlmgr%d%s", numSO,sofext);
 | 
|---|
 | 102 | #endif
 | 
|---|
 | 103 | string fnameso = GetTmpDir()+buff;
 | 
|---|
 | 104 | 
 | 
|---|
 | 105 | // Creation du shared-lib
 | 
|---|
 | 106 | #if defined(OSF1) 
 | 
|---|
 | 107 |   cmd = "ld -shared -o " + fnameso + " -all " + fnameobj + " -none -lm -lc";
 | 
|---|
 | 108 | #elif defined(Linux)
 | 
|---|
 | 109 |   cmd = "ld -shared -o " + fnameso + " " + fnameobj + " -lm -lc";
 | 
|---|
 | 110 | #elif defined(SunOS)
 | 
|---|
 | 111 |   cmd = "ld -G -o " + fnameso + " " + fnameobj + " -lm -lc";
 | 
|---|
 | 112 | #elif defined(IRIX64)
 | 
|---|
 | 113 |   cmd = "ld -shared -o " + fnameso + " " + fnameobj + " -lm -lc";
 | 
|---|
 | 114 | #elif defined(AIX)
 | 
|---|
 | 115 |   cmd = "ld -G -bnogc -bexpall -bM:1L -o " + fnameso + " " + fnameobj;
 | 
|---|
 | 116 | #elif defined(HPUX)
 | 
|---|
 | 117 |   cmd = "ld -b -o " + fnameso + " " + fnameobj + " -lm -lc";
 | 
|---|
 | 118 | #else 
 | 
|---|
 | 119 |   cmd = "ld -o " + fnameso + " " + fnameobj + " -lm -lc";
 | 
|---|
 | 120 | #endif
 | 
|---|
 | 121 | rc = system(cmd.c_str());
 | 
|---|
 | 122 | if (rc != 0)  { 
 | 
|---|
 | 123 |   cerr << "PDynLinkMgr::BuildFromCFile() Error Rc(" << cmd <<")= "<< rc << endl; 
 | 
|---|
 | 124 |   return(NULL);
 | 
|---|
 | 125 |   }
 | 
|---|
 | 126 | PDynLinkMgr* rdyn = new PDynLinkMgr(fnameso, false);
 | 
|---|
 | 127 | rdyn->copy = true;
 | 
|---|
 | 128 | return(rdyn);
 | 
|---|
 | 129 | 
 | 
|---|
 | 130 | }
 | 
|---|
 | 131 | 
 | 
|---|
 | 132 | /* --Methode-- */
 | 
|---|
 | 133 | PDynLinkMgr::PDynLinkMgr(string& soname, bool cp)
 | 
|---|
 | 134 | {
 | 
|---|
 | 135 | dlhandle = NULL;
 | 
|---|
 | 136 | soName = "";
 | 
|---|
 | 137 | 
 | 
|---|
 | 138 | if (soname.find_last_of(".") > soname.length())  
 | 
|---|
 | 139 | #ifdef HPUX
 | 
|---|
 | 140 |   soname += sofext_HPUX;
 | 
|---|
 | 141 | #else
 | 
|---|
 | 142 |   soname += sofext;
 | 
|---|
 | 143 | #endif
 | 
|---|
 | 144 | 
 | 
|---|
 | 145 | string fnameso;
 | 
|---|
 | 146 | if (cp) {
 | 
|---|
 | 147 |   numSO++;
 | 
|---|
 | 148 |   char buff[32];
 | 
|---|
 | 149 | #ifndef HPUX
 | 
|---|
 | 150 |   sprintf(buff,"pdlmgr%d%s", numSO,sofext);
 | 
|---|
 | 151 | #else
 | 
|---|
 | 152 |   sprintf(buff,"pdlmgr%d%s", numSO,sofext_HPUX);
 | 
|---|
 | 153 | #endif
 | 
|---|
 | 154 |   fnameso = GetTmpDir()+buff;
 | 
|---|
 | 155 |   string cmd = "cp " + soname + " " + fnameso;
 | 
|---|
 | 156 |   int rc = system(cmd.c_str());
 | 
|---|
 | 157 |   if (rc != 0)  { 
 | 
|---|
 | 158 |     cerr << "PDynLinkMgr::PDynLinkMgr() Error Rc(" << cmd <<")= "<< rc << endl; 
 | 
|---|
 | 159 |     return;
 | 
|---|
 | 160 |     }
 | 
|---|
 | 161 |   } 
 | 
|---|
 | 162 | else fnameso = soname;
 | 
|---|
 | 163 | copy = cp;
 | 
|---|
 | 164 | soName = fnameso;
 | 
|---|
 | 165 | 
 | 
|---|
 | 166 | #if defined(HPUX)
 | 
|---|
 | 167 | dlhandle = NULL;
 | 
|---|
 | 168 | cerr << "PDynLinkMgr::PDynLinkMgr() Not yet available on HP-UX " << endl;
 | 
|---|
 | 169 | return;
 | 
|---|
 | 170 | #else
 | 
|---|
 | 171 | dlhandle =  dlopen(fnameso.c_str(), RTLD_NOW);
 | 
|---|
 | 172 | if (dlhandle == NULL) { 
 | 
|---|
 | 173 |   cerr << "PDynLinkMgr::PDynLinkMgr(): Error opening SO " << fnameso 
 | 
|---|
 | 174 |        << " (" << soname << ")" << endl; 
 | 
|---|
 | 175 |   string sn = dlerror();
 | 
|---|
 | 176 |   cerr << "Loader Error (dlerror()) :" << sn << endl;
 | 
|---|
 | 177 |   return; 
 | 
|---|
 | 178 |   }
 | 
|---|
 | 179 | #endif
 | 
|---|
 | 180 | }
 | 
|---|
 | 181 | 
 | 
|---|
 | 182 | /* --Methode-- */
 | 
|---|
 | 183 | PDynLinkMgr::~PDynLinkMgr()
 | 
|---|
 | 184 | {
 | 
|---|
 | 185 | #if defined(HPUX)
 | 
|---|
 | 186 | cerr << "PDynLinkMgr::~PDynLinkMgr() Not yet available on HP-UX " << endl;
 | 
|---|
 | 187 | return;
 | 
|---|
 | 188 | #else
 | 
|---|
 | 189 | if (dlhandle) dlclose(dlhandle);    dlhandle = NULL;
 | 
|---|
 | 190 | if (copy) {
 | 
|---|
 | 191 |   string cmd = "rm -f " + soName;
 | 
|---|
 | 192 |   system(cmd.c_str());
 | 
|---|
 | 193 |   }
 | 
|---|
 | 194 | #endif
 | 
|---|
 | 195 | }
 | 
|---|
 | 196 | 
 | 
|---|
 | 197 | /* --Methode-- */
 | 
|---|
 | 198 | DlFunction PDynLinkMgr::GetFunction(string const & funcname)
 | 
|---|
 | 199 | {
 | 
|---|
 | 200 | DlFunction f = NULL;
 | 
|---|
 | 201 | #if defined(HPUX)
 | 
|---|
 | 202 | cerr << "PDynLinkMgr::GetFunction() Not yet available on HP-UX " << endl;
 | 
|---|
 | 203 | return f;
 | 
|---|
 | 204 | #else
 | 
|---|
 | 205 | if (dlhandle != NULL)  
 | 
|---|
 | 206 |   f = (DlFunction)dlsym(dlhandle, funcname.c_str());
 | 
|---|
 | 207 | if (f == NULL) cerr << "PDynLinkMgr::GetFunction(): Error linking " << funcname << endl;
 | 
|---|
 | 208 | return(f);
 | 
|---|
 | 209 | #endif
 | 
|---|
 | 210 | }
 | 
|---|
 | 211 | 
 | 
|---|