Various Changes

[tpg/acess2.git] / Usermode / Libraries / ld-acess.so_src / loadlib.c

diff --git a/Usermode/Libraries/ld-acess.so_src/loadlib.c b/Usermode/Libraries/ld-acess.so_src/loadlib.c
index 8583141..b5fb0a1 100644 (file)
--- a/Usermode/Libraries/ld-acess.so_src/loadlib.c
+++ b/Usermode/Libraries/ld-acess.so_src/loadlib.c
@@ -12,13 +12,16 @@
 # define DEBUGS(v...)  \r
 #endif\r
 \r
+// === CONSTANTS ===\r
+#define        MAX_LOADED_LIBRARIES    64\r
+#define        MAX_STRINGS_BYTES       4096\r
+#define        SYSTEM_LIB_DIR  "/Acess/Libs/"\r
+\r
 // === PROTOTYPES ===\r
 Uint   IsFileLoaded(char *file);
  int   GetSymbolFromBase(Uint base, char *name, Uint *ret);\r
 \r
 // === GLOABLS ===\r
-#define        MAX_LOADED_LIBRARIES    64\r
-#define        MAX_STRINGS_BYTES       4096\r
 struct {\r
        Uint    Base;\r
        char    *Name;\r
@@ -28,23 +31,48 @@ char        *gsNextAvailString = gsLoadedStrings;
 //tLoadLib     *gpLoadedLibraries = NULL;\r
 \r
 // === CODE ===\r
-Uint LoadLibrary(char *filename, char *SearchDir, char **envp)\r
+char *FindLibrary(char *DestBuf, char *SoName, char *ExtraSearchDir)\r
+{      \r
+       // -- #1: Executable Specified\r
+       if(ExtraSearchDir)\r
+       {\r
+               strcpy(DestBuf, ExtraSearchDir);\r
+               strcat(DestBuf, "/");\r
+               strcat(DestBuf, SoName);\r
+               if(file_exists(DestBuf))        return DestBuf;\r
+       }\r
+       \r
+       // -- #2: System\r
+       strcpy(DestBuf, SYSTEM_LIB_DIR);\r
+       strcat(DestBuf, SoName);\r
+       if(file_exists(DestBuf))        return DestBuf;\r
+       \r
+       // -- #3: Current Directory\r
+       if(file_exists(SoName)) return SoName;\r
+       \r
+       return NULL;\r
+}\r
+\r
+/**\r
+ */\r
+Uint LoadLibrary(char *SoName, char *SearchDir, char **envp)\r
 {\r
-       char    sTmpName[1024] = "/Acess/Libs/";\r
+       char    sTmpName[1024];\r
+       char    *filename;\r
        Uint    iArg;\r
        void    (*fEntry)(int, int, char *[], char**);\r
        \r
        DEBUGS("LoadLibrary: (filename='%s', envp=0x%x)\n", filename, envp);\r
        \r
        // Create Temp Name\r
-       strcpy(&sTmpName[12], filename);\r
-       DEBUGS(" LoadLibrary: sTmpName='%s'\n", sTmpName);\r
+       filename = FindLibrary(sTmpName, SoName, SearchDir);\r
+       DEBUGS(" LoadLibrary: filename='%s'\n", filename);\r
        \r
-       if( (iArg = IsFileLoaded(sTmpName)) )\r
+       if( (iArg = IsFileLoaded(filename)) )\r
                return iArg;\r
        \r
        // Load Library\r
-       iArg = SysLoadBin(sTmpName, (Uint*)&fEntry);\r
+       iArg = SysLoadBin(filename, (Uint*)&fEntry);\r
        if(iArg == 0) {\r
                DEBUGS("LoadLibrary: RETURN 0\n");\r
                return 0;\r
@@ -53,10 +81,10 @@ Uint LoadLibrary(char *filename, char *SearchDir, char **envp)
        DEBUGS(" LoadLibrary: iArg=0x%x, iEntry=0x%x\n", iArg, fEntry);\r
        \r
        // Load Symbols\r
-       fEntry = (void*)DoRelocate( iArg, envp, sTmpName );\r
+       fEntry = (void*)DoRelocate( iArg, envp, filename );\r
        \r
        // Call Entrypoint\r
-       DEBUGS(" LoadLibrary: '%s' Entry 0x%x\n", filename, fEntry);\r
+       DEBUGS(" LoadLibrary: '%s' Entry 0x%x\n", SoName, fEntry);\r
        fEntry(iArg, 0, NULL, envp);\r
        \r
        DEBUGS("LoadLibrary: RETURN 1\n");\r