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