Kernel/VTerm - "Fix" wrapping issue in VTerm (why was old behavior there?)

[tpg/acess2.git] / AcessNative / ld-acess_src / binary.c

diff --git a/AcessNative/ld-acess_src/binary.c b/AcessNative/ld-acess_src/binary.c
index 0ce9fac..738c6e8 100644 (file)
--- a/AcessNative/ld-acess_src/binary.c
+++ b/AcessNative/ld-acess_src/binary.c
@@ -1,12 +1,18 @@
 /*
- * AcessNative
+ * AcessNative Dynamic Linker
+ * - By John Hodge (thePowersGang)
+ * 
+ * binary.c
+ * - Provides binary loading and type abstraction
  */
+#define DEBUG  0
+#define _POSIX_C_SOURCE        200809L // needed for strdup
 #include "common.h"
 #include <stdint.h>
 #include <stdio.h>
 #include <string.h>
 
-#define LIBRARY_PATH   "../Usermode/Output/i386/Libs"
+#define LIBRARY_PATH   "$$$$../Usermode/Output/x86_64/Libs"
 
 // === TYPES ===
 typedef struct sBinary {
@@ -18,11 +24,12 @@ typedef struct sBinary {
 }      tBinary;
 
 // === IMPORTS ===
-extern void    *Elf_Load(FILE *FP);
-extern uintptr_t       Elf_Relocate(void *Base);
-extern int     Elf_GetSymbol(void *Base, char *Name, uintptr_t *ret);
+extern void    *Elf_Load(int fd);
+extern uintptr_t       ElfRelocate(void *Base);
+extern int     ElfGetSymbol(void *Base, char *Name, uintptr_t *ret, size_t *size);
 extern int     ciNumBuiltinSymbols;
 extern tSym    caBuiltinSymbols[];
+extern char    **gEnvP;
 
 // === PROTOTYPES ===
 void   Binary_AddToList(const char *Filename, void *Base, tBinFmt *Format);
@@ -33,8 +40,8 @@ tBinFmt       gElf_FormatDef = {
 //     .Magic = "\x7F""ELF",
        .Name = "ELF32",
        .Load = Elf_Load,
-       .Relocate = Elf_Relocate,
-       .GetSymbol = Elf_GetSymbol
+       .Relocate = ElfRelocate,
+       .GetSymbol = ElfGetSymbol
        };
 tBinary        *gLoadedBinaries;
 
@@ -43,7 +50,7 @@ char *Binary_LocateLibrary(const char *Name)
 {
        char    *envPath = getenv("ACESS_LIBRARY_PATH");
         int    nameLen = strlen(Name);
-       FILE    *fp;
+        int    fd;
        
        if( strcmp(Name, "libld-acess.so") == 0 ) {
                return strdup("libld-acess.so");
@@ -59,9 +66,9 @@ char *Binary_LocateLibrary(const char *Name)
                strcat(tmp, "/");
                strcat(tmp, Name);
                
-               fp = fopen(tmp, "r");
-               if(fp) {
-                       fclose(fp);
+               fd = acess__SysOpen(tmp, 4);    // OPENFLAG_EXEC
+               if(fd != -1) {
+                       acess__SysClose(fd);
                        return strdup(tmp);
                }
        }               
@@ -74,16 +81,20 @@ char *Binary_LocateLibrary(const char *Name)
                strcat(tmp, "/");
                strcat(tmp, Name);
                
+               #if DEBUG
                printf("Binary_LocateLibrary: tmp = '%s'\n", tmp);
+               #endif
 
-               fp = fopen(tmp, "r");
-               if(fp) {
-                       fclose(fp);
+               fd = acess__SysOpen(tmp, 4);    // OPENFLAG_EXEC
+               if(fd != -1) {
+                       acess__SysClose(fd);
                        return strdup(tmp);
                }
        }               
 
+       #if DEBUG
        fprintf(stderr, "Unable to locate library '%s'\n", Name);
+       #endif
 
        return NULL;
 }
@@ -92,23 +103,31 @@ void *Binary_LoadLibrary(const char *Name)
 {
        char    *path;
        void    *ret;
-        int    (*entry)(int,char*[],char**) = NULL;
+        int    (*entry)(void *,int,char*[],char**) = NULL;
 
        // Find File
        path = Binary_LocateLibrary(Name);
+       #if DEBUG
        printf("Binary_LoadLibrary: path = '%s'\n", path);
+       #endif
        if( !path ) {
                return NULL;
        }
 
        ret = Binary_Load(path, (uintptr_t*)&entry);
+       if( ret != (void*)-1 )
+               Debug("LOADED '%s' to %p (Entry=%p)", path, ret, entry);
        free(path);
        
+       #if DEBUG
        printf("Binary_LoadLibrary: ret = %p, entry = %p\n", ret, entry);
+       #endif
        if( entry ) {
                char    *argv[] = {NULL};
+               #if DEBUG
                printf("Calling '%s' entry point %p\n", Name, entry);
-               entry(0, argv, NULL);
+               #endif
+               entry(ret, 0, argv, gEnvP);
        }
 
        return ret;
@@ -116,8 +135,8 @@ void *Binary_LoadLibrary(const char *Name)
 
 void *Binary_Load(const char *Filename, uintptr_t *EntryPoint)
 {
-       FILE    *fp;
-       uint32_t        dword;
+        int    fd;
+       uint32_t        dword = 0xFA17FA17;
        void    *ret;
        uintptr_t       entry = 0;
        tBinFmt *fmt;
@@ -138,36 +157,43 @@ void *Binary_Load(const char *Filename, uintptr_t *EntryPoint)
                }
        }
 
-       fp = fopen(Filename, "r");
-       if( !fp ) {
+       fd = acess__SysOpen(Filename, 2|1);     // Execute and Read
+       if( fd == -1 ) {
                // TODO: Handle libary directories
                perror("Opening binary");
                return NULL;
        }
 
-       fread(&dword, 1, 4, fp);
-       fseek(fp, 0, SEEK_SET);
+       acess__SysRead(fd, &dword, 4);
+       acess__SysSeek(fd, 0, ACESS_SEEK_SET);
        
        if( memcmp(&dword, "\x7F""ELF", 4) == 0 ) {
                fmt = &gElf_FormatDef;
        }
        else {
-               fclose(fp);
+               fprintf(stderr, "Unknown executable format (0x%08x)\n", dword);
+               acess__SysClose(fd);
                return NULL;
        }
        
-       printf("fmt->Load(%p)...\n", fp);
-       ret = fmt->Load(fp);
-       printf("fmt->Load(%p): %p\n", fp, ret);
+       #if DEBUG
+       printf("fmt->Load(0x%x)...\n", fd);
+       #endif
+       ret = fmt->Load(fd);
+       acess__SysClose(fd);
+       #if DEBUG
+       printf("fmt->Load(0x%x): %p\n", fd, ret);
+       #endif
        if( !ret ) {
-               fclose(fp);
                return NULL;
        }
        
        Binary_AddToList(Filename, ret, fmt);
 
        entry = fmt->Relocate(ret);
+       #if DEBUG
        printf("fmt->Relocate(%p): %p\n", ret, (void*)entry);
+       #endif
        if( !entry ) {
                // TODO: Clean up
                return NULL;
@@ -176,8 +202,6 @@ void *Binary_Load(const char *Filename, uintptr_t *EntryPoint)
        if( EntryPoint )
                *EntryPoint = entry;
 
-       fclose(fp);
-
        Binary_SetReadyToUse(ret);
 
        return ret;
@@ -205,7 +229,7 @@ void Binary_SetReadyToUse(void *Base)
        }
 }
 
-int Binary_GetSymbol(const char *SymbolName, uintptr_t *Value)
+int Binary_GetSymbol(const char *SymbolName, uintptr_t *Value, size_t *Size, void *IgnoreBase)
 {
         int    i;
        tBinary *bin;
@@ -220,20 +244,24 @@ int Binary_GetSymbol(const char *SymbolName, uintptr_t *Value)
        {
                if( strcmp(caBuiltinSymbols[i].Name, SymbolName) == 0 ) {
                        *Value = (uintptr_t)caBuiltinSymbols[i].Value;
+                       if(Size)        *Size = 0;
                        return 1;
                }
        }
        
-       // TODO: Search list of loaded binaries
+       // Search list of loaded binaries
        for(bin = gLoadedBinaries; bin; bin = bin->Next)
        {
+               if( bin->Base == IgnoreBase )   continue ;
                if( !bin->Ready )       continue;
                //printf(" Binary_GetSymbol: bin = %p{%p, %s}\n", bin, bin->Base, bin->Path);
-               if( bin->Format->GetSymbol(bin->Base, (char*)SymbolName, Value) )
+               if( bin->Format->GetSymbol(bin->Base, (char*)SymbolName, Value, Size) )
                        return 1;
        }
 
        //printf("Binary_GetSymbol: RETURN 0, not found\n");
-       
+       printf("--- ERROR: Unable to find symbol '%s'\n", SymbolName);
+
+       exit( -1 );
        return 0;
 }