| [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 |  | 
|---|