X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=Usermode%2FLibraries%2Fld-acess.so_src%2Floadlib.c;h=3f8e674260b3a988554a9ae724d216b75c1e57a5;hb=30c11f371da98d13344e8bed3766d3e574a65ac9;hp=da74b824fcd52d16173a893eb0ce891150c8dfe7;hpb=3d6a345b39afa454f0f33fc8d48b96229971e6ed;p=tpg%2Facess2.git

diff --git a/Usermode/Libraries/ld-acess.so_src/loadlib.c b/Usermode/Libraries/ld-acess.so_src/loadlib.c
index da74b824..3f8e6742 100644
--- a/Usermode/Libraries/ld-acess.so_src/loadlib.c
+++ b/Usermode/Libraries/ld-acess.so_src/loadlib.c
@@ -3,6 +3,7 @@
  By thePowersGang
 */
 #include "common.h"
+#include <stdint.h>
 
 #define DEBUG	0
 
@@ -13,8 +14,8 @@
 #endif
 
 // === PROTOTYPES ===
-void	*IsFileLoaded(char *file);
- int	GetSymbolFromBase(void *base, char *name, void **ret);
+void	*IsFileLoaded(const char *file);
+ int	GetSymbolFromBase(void *base, const char *name, void **ret, size_t *size);
 
 // === IMPORTS ===
 extern const struct {
@@ -22,6 +23,7 @@ extern const struct {
 	char	*Name;
 }	caLocalExports[];
 extern const int	ciNumLocalExports;
+extern char	**gEnvP;
 
 // === GLOABLS ===
 tLoadedLib	gLoadedLibraries[MAX_LOADED_LIBRARIES];
@@ -30,7 +32,7 @@ char	*gsNextAvailString = gsLoadedStrings;
 //tLoadLib	*gpLoadedLibraries = NULL;
 
 // === CODE ===
-char *FindLibrary(char *DestBuf, char *SoName, char *ExtraSearchDir)
+const char *FindLibrary(char *DestBuf, const char *SoName, const char *ExtraSearchDir)
 {	
 	// -- #1: Executable Specified
 	if(ExtraSearchDir)
@@ -54,43 +56,44 @@ char *FindLibrary(char *DestBuf, char *SoName, char *ExtraSearchDir)
 
 /**
  */
-void *LoadLibrary(char *SoName, char *SearchDir, char **envp)
+void *LoadLibrary(const char *SoName, const char *SearchDir, char **envp)
 {
 	char	sTmpName[1024];
-	char	*filename;
+	const char	*filename;
 	void	*base;
 	void	(*fEntry)(void *, int, char *[], char**);
 	
-	DEBUGS("LoadLibrary: (filename='%s', envp=0x%x)\n", filename, envp);
+	DEBUGS("LoadLibrary: (SoName='%s', SearchDir='%s', envp=%p)", SoName, SearchDir, envp);
 	
 	// Create Temp Name
 	filename = FindLibrary(sTmpName, SoName, SearchDir);
 	if(filename == NULL) {
-		DEBUGS("LoadLibrary: RETURN 0\n");
+		DEBUGS("LoadLibrary: RETURN 0");
 		return 0;
 	}
-	DEBUGS(" LoadLibrary: filename='%s'\n", filename);
+	DEBUGS(" LoadLibrary: filename='%s'", filename);
 	
 	if( (base = IsFileLoaded(filename)) )
 		return base;
-	
+
+	DEBUGS(" LoadLibrary: SysLoadBin()");	
 	// Load Library
 	base = SysLoadBin(filename, (void**)&fEntry);
 	if(!base) {
-		DEBUGS("LoadLibrary: RETURN 0\n");
+		DEBUGS("LoadLibrary: RETURN 0");
 		return 0;
 	}
 	
-	DEBUGS(" LoadLibrary: iArg=%p, iEntry=0x%x\n", base, fEntry);
+	DEBUGS(" LoadLibrary: iArg=%p, fEntry=%p", base, fEntry);
 	
 	// Load Symbols
 	fEntry = DoRelocate( base, envp, filename );
 	
 	// Call Entrypoint
-	DEBUGS(" LoadLibrary: '%s' Entry 0x%x\n", SoName, fEntry);
-	fEntry(base, 0, NULL, envp);
+	DEBUGS(" LoadLibrary: '%s' Entry %p", SoName, fEntry);
+	fEntry(base, 0, NULL, gEnvP);
 	
-	DEBUGS("LoadLibrary: RETURN 1\n");
+	DEBUGS("LoadLibrary: RETURN 1");
 	return base;
 }
 
@@ -98,7 +101,7 @@ void *LoadLibrary(char *SoName, char *SearchDir, char **envp)
  * \fn Uint IsFileLoaded(char *file)
  * \brief Determine if a file is already loaded
  */
-void *IsFileLoaded(char *file)
+void *IsFileLoaded(const char *file)
 {
 	 int	i;
 	DEBUGS("IsFileLoaded: (file='%s')", file);
@@ -107,7 +110,7 @@ void *IsFileLoaded(char *file)
 		if(gLoadedLibraries[i].Base == 0)	break;	// Last entry has Base set to NULL
 		DEBUGS(" strcmp('%s', '%s')", gLoadedLibraries[i].Name, file);
 		if(strcmp(gLoadedLibraries[i].Name, file) == 0) {
-			DEBUGS("IsFileLoaded: Found %i (0x%x)", i, gLoadedLibraries[i].Base);
+			DEBUGS("IsFileLoaded: Found %i (%p)", i, gLoadedLibraries[i].Base);
 			return gLoadedLibraries[i].Base;
 		}
 	}
@@ -119,12 +122,12 @@ void *IsFileLoaded(char *file)
  * \fn void AddLoaded(char *File, Uint base)
  * \brief Add a file to the loaded list
  */
-void AddLoaded(char *File, void *base)
+void AddLoaded(const char *File, void *base)
 {
 	 int	i, length;
 	char	*name = gsNextAvailString;
 	
-	DEBUGS("AddLoaded: (File='%s', base=0x%x)", File, base);
+	DEBUGS("AddLoaded: (File='%s', base=%p)", File, base);
 	
 	// Find a free slot
 	for( i = 0; i < MAX_LOADED_LIBRARIES; i ++ )
@@ -148,7 +151,7 @@ void AddLoaded(char *File, void *base)
 	strcpy(name, File);
 	gLoadedLibraries[i].Name = name;
 	gsNextAvailString = &name[length+1];
-	DEBUGS("'%s' (0x%x) loaded as %i\n", name, base, i);
+	DEBUGS("'%s' (%p) loaded as %i", name, base, i);
 	return;
 }
 
@@ -192,10 +195,10 @@ void Unload(void *Base)
 }
 
 /**
- \fn Uint GetSymbol(char *name)
+ \fn Uint GetSymbol(const char *name)
  \brief Gets a symbol value from a loaded library
 */
-void *GetSymbol(char *name)
+void *GetSymbol(const char *name, size_t *Size)
 {
 	 int	i;
 	void	*ret;
@@ -207,13 +210,14 @@ void *GetSymbol(char *name)
 			return caLocalExports[i].Value;
 	}
 	
-	for(i=0;i<sizeof(gLoadedLibraries)/sizeof(gLoadedLibraries[0]);i++)
+	// Entry 0 is ld-acess, ignore it
+	for(i = 1; i < MAX_LOADED_LIBRARIES; i ++)
 	{
 		if(gLoadedLibraries[i].Base == 0)	break;
 		
 		//SysDebug(" GetSymbol: Trying 0x%x, '%s'",
 		//	gLoadedLibraries[i].Base, gLoadedLibraries[i].Name);
-		if(GetSymbolFromBase(gLoadedLibraries[i].Base, name, &ret))	return ret;
+		if(GetSymbolFromBase(gLoadedLibraries[i].Base, name, &ret, Size))	return ret;
 	}
 	SysDebug("GetSymbol: === Symbol '%s' not found ===", name);
 	return 0;
@@ -223,13 +227,15 @@ void *GetSymbol(char *name)
  \fn int GetSymbolFromBase(Uint base, char *name, Uint *ret)
  \breif Gets a symbol from a specified library
 */
-int GetSymbolFromBase(void *base, char *name, void **ret)
+int GetSymbolFromBase(void *base, const char *name, void **ret, size_t *Size)
 {
-	if(*(Uint32*)base == (0x7F|('E'<<8)|('L'<<16)|('F'<<24)))
-		return ElfGetSymbol(base, name, ret);
-	if(*(Uint16*)base == ('M'|('Z'<<8)))
-		return PE_GetSymbol(base, name, ret);
-	SysDebug("Unknown type at %p", base);
+	uint8_t	*hdr = base;
+	if(hdr[0] == 0x7F && hdr[1] == 'E' && hdr[2] == 'L' && hdr[3] == 'F')
+		return ElfGetSymbol(base, name, ret, Size);
+	if(hdr[0] == 'M' && hdr[1] == 'Z')
+		return PE_GetSymbol(base, name, ret, Size);
+	SysDebug("Unknown type at %p (%02x %02x %02x %02x)", base,
+		hdr[0], hdr[1], hdr[2], hdr[3]);
 	return 0;
 }