X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=ghc%2Frts%2FLinker.c;h=308df052ba6fae4af9ed899e4ce14431e2f1cdbd;hb=c7d594318efe7eca8e84ef19ae167cf75a9a3092;hp=7bfc7c610d9e49791e0ad2844fb93b783fb701c3;hpb=7794a5513328199916f5230521b10b874c42f2ed;p=ghc-hetmet.git diff --git a/ghc/rts/Linker.c b/ghc/rts/Linker.c index 7bfc7c6..308df05 100644 --- a/ghc/rts/Linker.c +++ b/ghc/rts/Linker.c @@ -1,5 +1,5 @@ /* ----------------------------------------------------------------------------- - * $Id: Linker.c,v 1.7 2001/01/10 17:21:18 sewardj Exp $ + * $Id: Linker.c,v 1.32 2001/02/28 10:03:42 sewardj Exp $ * * (c) The GHC Team, 2000 * @@ -12,77 +12,36 @@ #include "HsFFI.h" #include "Hash.h" #include "Linker.h" +#include "LinkerInternals.h" #include "RtsUtils.h" +#include "StoragePriv.h" -/* These two are POSIX headers */ +#ifdef HAVE_SYS_TYPES_H #include +#endif + +#ifdef HAVE_SYS_STAT_H #include +#endif -/* ToDo: configure this */ +#ifdef HAVE_DLFCN_H #include +#endif -#ifdef GHCI /* endif is right at end of file */ - -/* A bucket in the symbol hash-table. Primarily, maps symbol names to - * absolute addresses. All symbols from a given module are linked - * together, so they can be freed at the same time. There's also a - * bucket link field for the hash table. - */ -typedef struct _SymbolVal { - char *lbl; - void *addr; -} SymbolVal; - -typedef enum { OBJECT_LOADED, OBJECT_RESOLVED } OStatus; - -/* Indication of section kinds for loaded objects. Needed by - the GC for deciding whether or not a pointer on the stack - is a code pointer. -*/ -typedef enum { SECTIONKIND_CODE_OR_RODATA, - SECTIONKIND_RWDATA, - SECTIONKIND_OTHER, - SECTIONKIND_NOINFOAVAIL } - SectionKind; - -typedef struct { void* start; void* end; SectionKind kind; } - Section; - -/* Top-level structure for an object module. One of these is allocated - * for each object file in use. - */ -typedef struct _ObjectCode { - OStatus status; - char* fileName; - int fileSize; - char* formatName; /* eg "ELF32", "DLL", "COFF", etc. */ - - SymbolVal *symbols; - int n_symbols; - - /* ptr to malloc'd lump of memory holding the obj file */ - void* image; - - /* The section-kind entries for this object module. Dynamically expands. */ - Section* sections; - int n_sections; - - /* Allow a chain of these things */ - struct _ObjectCode * next; -} ObjectCode; - +#if defined(linux_TARGET_OS) || defined(solaris2_TARGET_OS) || defined(freebsd_TARGET_OS) +# define OBJFORMAT_ELF +#elif defined(cygwin32_TARGET_OS) || defined (mingw32_TARGET_OS) +# define OBJFORMAT_PEi386 +#endif /* Hash table mapping symbol names to Symbol */ /*Str*/HashTable *symhash; -/* List of currently loaded objects */ -ObjectCode *objects; - -#if defined(linux_TARGET_OS) || defined(solaris2_TARGET_OS) +#if defined(OBJFORMAT_ELF) static int ocVerifyImage_ELF ( ObjectCode* oc ); static int ocGetNames_ELF ( ObjectCode* oc ); static int ocResolve_ELF ( ObjectCode* oc ); -#elif defined(cygwin32_TARGET_OS) +#elif defined(OBJFORMAT_PEi386) static int ocVerifyImage_PEi386 ( ObjectCode* oc ); static int ocGetNames_PEi386 ( ObjectCode* oc ); static int ocResolve_PEi386 ( ObjectCode* oc ); @@ -92,6 +51,12 @@ static int ocResolve_PEi386 ( ObjectCode* oc ); * Built-in symbols from the RTS */ +typedef struct _RtsSymbolVal { + char *lbl; + void *addr; +} RtsSymbolVal; + + #define RTS_SYMBOLS \ SymX(MainRegTable) \ Sym(stg_gc_enter_1) \ @@ -127,6 +92,9 @@ static int ocResolve_PEi386 ( ObjectCode* oc ); SymX(stg_sel_10_upd_info) \ SymX(stg_sel_11_upd_info) \ SymX(stg_sel_12_upd_info) \ + SymX(stg_sel_13_upd_info) \ + SymX(stg_sel_14_upd_info) \ + SymX(stg_sel_15_upd_info) \ SymX(stg_upd_frame_info) \ SymX(stg_seq_frame_info) \ SymX(stg_CAF_BLACKHOLE_info) \ @@ -136,12 +104,14 @@ static int ocResolve_PEi386 ( ObjectCode* oc ); SymX(stg_WEAK_info) \ SymX(stg_CHARLIKE_closure) \ SymX(stg_INTLIKE_closure) \ - SymX(stg_CAF_UNENTERED_entry) \ SymX(newCAF) \ + SymX(newBCOzh_fast) \ + SymX(mkApUpd0zh_fast) \ SymX(putMVarzh_fast) \ SymX(newMVarzh_fast) \ SymX(takeMVarzh_fast) \ SymX(tryTakeMVarzh_fast) \ + SymX(tryPutMVarzh_fast) \ SymX(catchzh_fast) \ SymX(raisezh_fast) \ SymX(forkzh_fast) \ @@ -193,6 +163,7 @@ static int ocResolve_PEi386 ( ObjectCode* oc ); SymX(__int_encodeFloat) \ SymX(__int_encodeDouble) \ SymX(__gmpz_cmp_si) \ + SymX(__gmpz_cmp_ui) \ SymX(__gmpz_cmp) \ SymX(__gmpn_gcd_1) \ SymX(prog_argv) \ @@ -217,7 +188,16 @@ static int ocResolve_PEi386 ( ObjectCode* oc ); SymX(PatErrorHdrHook) \ SymX(defaultsHook) \ SymX(PreTraceHook) \ - SymX(PostTraceHook) + SymX(PostTraceHook) \ + SymX(stg_sig_install) \ + Sym(nocldstop) \ + SymX(createAdjustor) \ + SymX(rts_mkInt) \ + SymX(rts_mkStablePtr) \ + SymX(rts_apply) \ + SymX(rts_evalIO) \ + SymX(rts_checkSchedStatus) \ + SymX(rts_getInt) #ifndef SUPPORT_LONG_LONGS #define RTS_LONG_LONG_SYMS /* nothing */ @@ -279,7 +259,7 @@ RTS_SYMBOLS (void*)(&(vvv)) }, #define SymX(vvv) Sym(vvv) -static SymbolVal rtsSyms[] = { +static RtsSymbolVal rtsSyms[] = { RTS_SYMBOLS RTS_LONG_LONG_SYMS { 0, 0 } /* sentinel */ @@ -288,21 +268,60 @@ static SymbolVal rtsSyms[] = { /* ----------------------------------------------------------------------------- * initialize the object linker */ +#if defined(OBJFORMAT_ELF) static void *dl_prog_handle; +#endif void initLinker( void ) { - SymbolVal *sym; + RtsSymbolVal *sym; symhash = allocStrHashTable(); /* populate the symbol table with stuff from the RTS */ for (sym = rtsSyms; sym->lbl != NULL; sym++) { - insertStrHashTable(symhash, sym->lbl, sym); + insertStrHashTable(symhash, sym->lbl, sym->addr); } - +# if defined(OBJFORMAT_ELF) dl_prog_handle = dlopen(NULL, RTLD_LAZY); +# endif +} + +/* ----------------------------------------------------------------------------- + * Add a DLL from which symbols may be found. In the ELF case, just + * do RTLD_GLOBAL-style add, so no further messing around needs to + * happen in order that symbols in the loaded .so are findable -- + * lookupSymbol() will subsequently see them by dlsym on the program's + * dl-handle. Returns 0 if fail, 1 if success. + */ +char* +addDLL ( char* dll_name ) +{ +# if defined(OBJFORMAT_ELF) + void *hdl; + char *buf; + char *errmsg; + + buf = stgMallocBytes(strlen(dll_name) + 10, "addDll"); + sprintf(buf, "lib%s.so", dll_name); + hdl = dlopen(buf, RTLD_NOW | RTLD_GLOBAL ); + free(buf); + if (hdl == NULL) { + /* dlopen failed; return a ptr to the error msg. */ + errmsg = dlerror(); + if (errmsg == NULL) errmsg = "addDLL: unknown error"; + return errmsg; + } else { + return NULL; + } + ASSERT(0); /*NOTREACHED*/ +# elif defined(OBJFORMAT_PEi386) + barf("addDLL: not implemented on PEi386 yet"); + return 0; +# else + barf("addDLL: not implemented on this platform"); +# endif } /* ----------------------------------------------------------------------------- @@ -311,16 +330,37 @@ initLinker( void ) void * lookupSymbol( char *lbl ) { - SymbolVal *val; + void *val; + ASSERT(symhash != NULL); val = lookupStrHashTable(symhash, lbl); if (val == NULL) { +# if defined(OBJFORMAT_ELF) return dlsym(dl_prog_handle, lbl); +# elif defined(OBJFORMAT_PEi386) + ASSERT(2+2 == 5); + return NULL; +# endif + } else { + return val; + } +} + +static +void * +lookupLocalSymbol( ObjectCode* oc, char *lbl ) +{ + void *val; + val = lookupStrHashTable(oc->lochash, lbl); + + if (val == NULL) { + return NULL; } else { - return val->addr; + return val; } } + /* ----------------------------------------------------------------------------- * Load an obj (populate the global symbol table, but don't resolve yet) * @@ -334,20 +374,20 @@ loadObj( char *path ) int r, n; FILE *f; -#ifdef DEBUG +# ifdef DEBUG /* assert that we haven't already loaded this object */ { ObjectCode *o; for (o = objects; o; o = o->next) ASSERT(strcmp(o->fileName, path)); } -#endif /* DEBUG */ +# endif /* DEBUG */ oc = stgMallocBytes(sizeof(ObjectCode), "loadObj(oc)"); -# if defined(linux_TARGET_OS) || defined(solaris2_TARGET_OS) +# if defined(OBJFORMAT_ELF) oc->formatName = "ELF"; -# elif defined(cygwin32_TARGET_OS) +# elif defined(OBJFORMAT_PEi386) oc->formatName = "PEi386"; # else free(oc); @@ -357,11 +397,15 @@ loadObj( char *path ) r = stat(path, &st); if (r == -1) { return 0; } - oc->fileName = path; + /* sigh, strdup() isn't a POSIX function, so do it the long way */ + oc->fileName = stgMallocBytes( strlen(path)+1, "loadObj" ); + strcpy(oc->fileName, path); + oc->fileSize = st.st_size; oc->image = stgMallocBytes( st.st_size, "loadObj(image)" ); oc->symbols = NULL; oc->sections = NULL; + oc->lochash = allocStrHashTable(); /* chain it onto the list of objects */ oc->next = objects; @@ -379,9 +423,9 @@ loadObj( char *path ) } /* verify the in-memory image */ -# if defined(linux_TARGET_OS) || defined(solaris2_TARGET_OS) +# if defined(OBJFORMAT_ELF) r = ocVerifyImage_ELF ( oc ); -# elif defined(cygwin32_TARGET_OS) +# elif defined(OBJFORMAT_PEi386) r = ocVerifyImage_PEi386 ( oc ); # else barf("loadObj: no verify method"); @@ -389,9 +433,9 @@ loadObj( char *path ) if (!r) { return r; } /* build the symbol list for this image */ -# if defined(linux_TARGET_OS) || defined(solaris2_TARGET_OS) +# if defined(OBJFORMAT_ELF) r = ocGetNames_ELF ( oc ); -# elif defined(cygwin32_TARGET_OS) +# elif defined(OBJFORMAT_PEi386) r = ocGetNames_PEi386 ( oc ); # else barf("loadObj: no getNames method"); @@ -417,13 +461,13 @@ resolveObjs( void ) for (oc = objects; oc; oc = oc->next) { if (oc->status != OBJECT_RESOLVED) { -# if defined(linux_TARGET_OS) || defined(solaris2_TARGET_OS) +# if defined(OBJFORMAT_ELF) r = ocResolve_ELF ( oc ); -# elif defined(cygwin32_TARGET_OS) +# elif defined(OBJFORMAT_PEi386) r = ocResolve_PEi386 ( oc ); -# else - barf("link: not implemented on this platform"); -# endif +# else + barf("resolveObjs: not implemented on this platform"); +# endif if (!r) { return r; } oc->status = OBJECT_RESOLVED; } @@ -437,37 +481,53 @@ resolveObjs( void ) HsInt unloadObj( char *path ) { - ObjectCode *oc; + ObjectCode *oc, *prev; - for (oc = objects; oc; oc = oc->next) { + ASSERT(symhash != NULL); + ASSERT(objects != NULL); + + prev = NULL; + for (oc = objects; oc; prev = oc, oc = oc->next) { if (!strcmp(oc->fileName,path)) { /* Remove all the mappings for the symbols within this * object.. */ { - SymbolVal *s; - for (s = oc->symbols; s < oc->symbols + oc->n_symbols; s++) { - removeStrHashTable(symhash, s->lbl, NULL); - } + int i; + for (i = 0; i < oc->n_symbols; i++) { + if (oc->symbols[i] != NULL) { + removeStrHashTable(symhash, oc->symbols[i], NULL); + } + } + } + + if (prev == NULL) { + objects = oc->next; + } else { + prev->next = oc->next; } /* We're going to leave this in place, in case there are any pointers from the heap into it: */ /* free(oc->image); */ + free(oc->fileName); free(oc->symbols); free(oc->sections); + /* The local hash table should have been freed at the end + of the ocResolve_ call on it. */ + ASSERT(oc->lochash == NULL); free(oc); return 1; } } - + belch("unloadObj: can't find `%s' to unload", path); return 0; } /* -------------------------------------------------------------------------- - * PEi386 specifics (cygwin32) + * PEi386 specifics (Win32 targets) * ------------------------------------------------------------------------*/ /* The information for this linker comes from @@ -478,7 +538,7 @@ unloadObj( char *path ) */ -#if defined(cygwin32_TARGET_OS) +#if defined(OBJFORMAT_PEi386) @@ -578,7 +638,7 @@ typedef are just plain wrong. Sigh. */ static UChar * -myindex ( int scale, int index, void* base ) +myindex ( int scale, void* base, int index ) { return ((UChar*)base) + scale * index; @@ -639,11 +699,10 @@ cstring_from_COFF_symbol_name ( UChar* name, UChar* strtab ) /* The annoying case: 8 bytes. Copy into a temporary (which is never freed ...) */ - newstr = malloc(9); - if (newstr) { - strncpy(newstr,name,8); - newstr[8] = 0; - } + newstr = stgMallocBytes(9, "cstring_from_COFF_symbol_name"); + ASSERT(newstr); + strncpy(newstr,name,8); + newstr[8] = 0; return newstr; } @@ -665,7 +724,7 @@ findPEi386SectionCalled ( ObjectCode* oc, char* name ) UChar* n2; COFF_section* section_i = (COFF_section*) - myindex ( sizeof_COFF_section, i, sectab ); + myindex ( sizeof_COFF_section, sectab, i ); n1 = (UChar*) &(section_i->Name); n2 = name; if (n1[0]==n2[0] && n1[1]==n2[1] && n1[2]==n2[2] && @@ -681,14 +740,16 @@ findPEi386SectionCalled ( ObjectCode* oc, char* name ) static void zapTrailingAtSign ( UChar* sym ) { +# define my_isdigit(c) ((c) >= '0' && (c) <= '9') int i, j; if (sym[0] == 0) return; i = 0; while (sym[i] != 0) i++; i--; j = i; - while (j > 0 && isdigit(sym[j])) j--; + while (j > 0 && my_isdigit(sym[j])) j--; if (j > 0 && sym[j] == '@' && j != i) sym[j] = 0; +# undef my_isdigit } @@ -715,28 +776,30 @@ ocVerifyImage_PEi386 ( ObjectCode* oc ) + hdr->NumberOfSymbols * sizeof_COFF_symbol; if (hdr->Machine != 0x14c) { - oc->errMsg("Not x86 PEi386"); - return FALSE; + belch("Not x86 PEi386"); + return 0; } if (hdr->SizeOfOptionalHeader != 0) { - oc->errMsg("PEi386 with nonempty optional header"); - return FALSE; + belch("PEi386 with nonempty optional header"); + return 0; } if ( /* (hdr->Characteristics & IMAGE_FILE_RELOCS_STRIPPED) || */ (hdr->Characteristics & IMAGE_FILE_EXECUTABLE_IMAGE) || (hdr->Characteristics & IMAGE_FILE_DLL) || (hdr->Characteristics & IMAGE_FILE_SYSTEM) ) { - oc->errMsg("Not a PEi386 object file"); - return FALSE; + belch("Not a PEi386 object file"); + return 0; } if ( (hdr->Characteristics & IMAGE_FILE_BYTES_REVERSED_HI) || !(hdr->Characteristics & IMAGE_FILE_32BIT_MACHINE) ) { - oc->errMsg("Invalid PEi386 word size or endiannness"); - return FALSE; + belch("Invalid PEi386 word size or endiannness"); + return 0; } - if (!verb) return TRUE; /* No further verification after this point; only debug printing. */ + i = 0; + IF_DEBUG(linker, i=1); + if (i == 0) return 1; fprintf ( stderr, "sectab offset = %d\n", ((UChar*)sectab) - ((UChar*)hdr) ); @@ -764,7 +827,7 @@ ocVerifyImage_PEi386 ( ObjectCode* oc ) fprintf ( stderr, "\n" ); fprintf ( stderr, "string table has size 0x%x\n", * (UInt32*)strtab ); fprintf ( stderr, "---START of string table---\n"); - for (i = 4; i < *(UInt32*)strtab; i++) { + for (i = 4; i < *(Int32*)strtab; i++) { if (strtab[i] == 0) fprintf ( stderr, "\n"); else fprintf( stderr, "%c", strtab[i] ); @@ -776,7 +839,7 @@ ocVerifyImage_PEi386 ( ObjectCode* oc ) COFF_reloc* reltab; COFF_section* sectab_i = (COFF_section*) - myindex ( sizeof_COFF_section, i, sectab ); + myindex ( sizeof_COFF_section, sectab, i ); fprintf ( stderr, "\n" "section %d\n" @@ -805,13 +868,13 @@ ocVerifyImage_PEi386 ( ObjectCode* oc ) for (j = 0; j < sectab_i->NumberOfRelocations; j++) { COFF_symbol* sym; COFF_reloc* rel = (COFF_reloc*) - myindex ( sizeof_COFF_reloc, j, reltab ); + myindex ( sizeof_COFF_reloc, reltab, j ); fprintf ( stderr, " type 0x%-4x vaddr 0x%-8x name `", (UInt32)rel->Type, rel->VirtualAddress ); sym = (COFF_symbol*) - myindex ( sizeof_COFF_symbol, rel->SymbolTableIndex, symtab ); + myindex ( sizeof_COFF_symbol, symtab, rel->SymbolTableIndex ); printName ( sym->Name, strtab ); fprintf ( stderr, "'\n" ); } @@ -823,9 +886,9 @@ ocVerifyImage_PEi386 ( ObjectCode* oc ) i = 0; while (1) { COFF_symbol* symtab_i; - if (i >= hdr->NumberOfSymbols) break; + if (i >= (Int32)(hdr->NumberOfSymbols)) break; symtab_i = (COFF_symbol*) - myindex ( sizeof_COFF_symbol, i, symtab ); + myindex ( sizeof_COFF_symbol, symtab, i ); fprintf ( stderr, "symbol %d\n" " name `", @@ -851,7 +914,7 @@ ocVerifyImage_PEi386 ( ObjectCode* oc ) fprintf ( stderr, "\n" ); - return TRUE; + return 1; } @@ -881,12 +944,20 @@ ocGetNames_PEi386 ( ObjectCode* oc ) + hdr->NumberOfSymbols * sizeof_COFF_symbol; /* Copy exported symbols into the ObjectCode. */ + + oc->n_symbols = hdr->NumberOfSymbols; + oc->symbols = stgMallocBytes(oc->n_symbols * sizeof(char*), + "ocGetNames_PEi386(oc->symbols)"); + /* Call me paranoid; I don't care. */ + for (i = 0; i < oc->n_symbols; i++) + oc->symbols[i] = NULL; + i = 0; while (1) { COFF_symbol* symtab_i; - if (i >= hdr->NumberOfSymbols) break; + if (i >= (Int32)(hdr->NumberOfSymbols)) break; symtab_i = (COFF_symbol*) - myindex ( sizeof_COFF_symbol, i, symtab ); + myindex ( sizeof_COFF_symbol, symtab, i ); if (symtab_i->StorageClass == IMAGE_SYM_CLASS_EXTERNAL && symtab_i->SectionNumber != IMAGE_SYM_UNDEFINED) { @@ -894,13 +965,8 @@ ocGetNames_PEi386 ( ObjectCode* oc ) /* This symbol is global and defined, viz, exported */ COFF_section* sectabent; - sname = cstring_from_COFF_symbol_name ( - symtab_i->Name, strtab - ); - if (!sname) { - oc->errMsg("Out of memory when copying PEi386 symbol"); - return FALSE; - } + /* cstring_from_COFF_symbol_name always succeeds. */ + sname = cstring_from_COFF_symbol_name ( symtab_i->Name, strtab ); /* for IMAGE_SYMCLASS_EXTERNAL && !IMAGE_SYM_UNDEFINED, @@ -909,23 +975,27 @@ ocGetNames_PEi386 ( ObjectCode* oc ) */ sectabent = (COFF_section*) myindex ( sizeof_COFF_section, - symtab_i->SectionNumber-1, - sectab ); + sectab, + symtab_i->SectionNumber-1 ); addr = ((UChar*)(oc->image)) + (sectabent->PointerToRawData + symtab_i->Value); - /* fprintf ( stderr, "addSymbol %p `%s'\n", addr,sname); */ - if (!addSymbol(oc,sname,addr)) return FALSE; + IF_DEBUG(linker, belch("addSymbol %p `%s'\n", addr,sname);) + ASSERT(i >= 0 && i < oc->n_symbols); + oc->symbols[i] = sname; + insertStrHashTable(symhash, sname, addr); } i += symtab_i->NumberOfAuxSymbols; i++; } - oc->sections = stgMallocBytes( NumberOfSections * sizeof(Section), - "ocGetNamesPEi386" ); - /* Copy section information into the ObjectCode. */ - for (i = 0; i < hdr->NumberOfSections; i++) { + + oc->n_sections = hdr->NumberOfSections; + oc->sections = stgMallocBytes( oc->n_sections * sizeof(Section), + "ocGetNamesPEi386" ); + + for (i = 0; i < oc->n_sections; i++) { UChar* start; UChar* end; @@ -933,8 +1003,8 @@ ocGetNames_PEi386 ( ObjectCode* oc ) = SECTIONKIND_OTHER; COFF_section* sectab_i = (COFF_section*) - myindex ( sizeof_COFF_section, i, sectab ); - /* fprintf ( stderr, "section name = %s\n", sectab_i->Name ); */ + myindex ( sizeof_COFF_section, sectab, i ); + IF_DEBUG(linker, belchf("section name = %s\n", sectab_i->Name )); #if 0 /* I'm sure this is the Right Way to do it. However, the @@ -957,22 +1027,22 @@ ocGetNames_PEi386 ( ObjectCode* oc ) end = start + sectab_i->SizeOfRawData - 1; - if (kind != SECTIONKIND_OTHER) { - addSection ( oc, start, end, kind ); - } else { - fprintf ( stderr, "unknown section name = `%s'\n", - sectab_i->Name); - oc->errMsg("Unknown PEi386 section name"); - return FALSE; + if (kind == SECTIONKIND_OTHER) { + belch("Unknown PEi386 section name `%s'", sectab_i->Name); + return 0; } + + oc->sections[i].start = start; + oc->sections[i].end = end; + oc->sections[i].kind = kind; } - return TRUE; + return 1; } static int -ocResolve_PEi386 ( ObjectCode* oc, int verb ) +ocResolve_PEi386 ( ObjectCode* oc ) { COFF_header* hdr; COFF_section* sectab; @@ -984,7 +1054,10 @@ ocResolve_PEi386 ( ObjectCode* oc, int verb ) UInt32* pP; int i, j; - char symbol[1000]; // ToDo + + /* ToDo: should be variable-sized? But is at least safe in the + sense of buffer-overrun-proof. */ + char symbol[1000]; hdr = (COFF_header*)(oc->image); sectab = (COFF_section*) ( @@ -1002,7 +1075,7 @@ ocResolve_PEi386 ( ObjectCode* oc, int verb ) for (i = 0; i < hdr->NumberOfSections; i++) { COFF_section* sectab_i = (COFF_section*) - myindex ( sizeof_COFF_section, i, sectab ); + myindex ( sizeof_COFF_section, sectab, i ); COFF_reloc* reltab = (COFF_reloc*) ( ((UChar*)(oc->image)) + sectab_i->PointerToRelocations @@ -1011,7 +1084,7 @@ ocResolve_PEi386 ( ObjectCode* oc, int verb ) COFF_symbol* sym; COFF_reloc* reltab_j = (COFF_reloc*) - myindex ( sizeof_COFF_reloc, j, reltab ); + myindex ( sizeof_COFF_reloc, reltab, j ); /* the location to patch */ pP = (UInt32*)( @@ -1024,39 +1097,39 @@ ocResolve_PEi386 ( ObjectCode* oc, int verb ) /* the symbol to connect to */ sym = (COFF_symbol*) myindex ( sizeof_COFF_symbol, - reltab_j->SymbolTableIndex, symtab ); - if (verb) { - fprintf ( stderr, - "reloc sec %2d num %3d: type 0x%-4x " - "vaddr 0x%-8x name `", - i, j, - (UInt32)reltab_j->Type, - reltab_j->VirtualAddress ); - printName ( sym->Name, strtab ); - fprintf ( stderr, "'\n" ); - } + symtab, reltab_j->SymbolTableIndex ); + IF_DEBUG(linker, + fprintf ( stderr, + "reloc sec %2d num %3d: type 0x%-4x " + "vaddr 0x%-8x name `", + i, j, + (UInt32)reltab_j->Type, + reltab_j->VirtualAddress ); + printName ( sym->Name, strtab ); + fprintf ( stderr, "'\n" )); if (sym->StorageClass == IMAGE_SYM_CLASS_STATIC) { COFF_section* section_sym = findPEi386SectionCalled ( oc, sym->Name ); if (!section_sym) { fprintf ( stderr, "bad section = `%s'\n", sym->Name ); - oc->errMsg("Can't find abovementioned PEi386 section"); - return FALSE; + barf("Can't find abovementioned PEi386 section"); + return 0; } S = ((UInt32)(oc->image)) + (section_sym->PointerToRawData + sym->Value); } else { - copyName ( sym->Name, strtab, symbol, 1000 ); - zapTrailingAtSign ( symbol ); - S = (UInt32) ocLookupSym ( oc, symbol ); - if (S == 0) - S = (UInt32)(oc->clientLookup ( symbol )); - if (S == 0) { - belch("%s: unresolvable reference to `%s'", oc->fileName, symbol); - return FALSE; - } + copyName ( sym->Name, strtab, symbol, 1000 ); + zapTrailingAtSign ( symbol ); + (void*)S = lookupLocalSymbol( oc, symbol ); + if ((void*)S == NULL) + (void*)S = lookupSymbol( symbol ); + if (S == 0) { + belch("ocResolve_PEi386: %s: unknown symbol `%s'", + oc->fileName, symbol); + return 0; + } } switch (reltab_j->Type) { @@ -1082,28 +1155,32 @@ ocResolve_PEi386 ( ObjectCode* oc, int verb ) fprintf(stderr, "unhandled PEi386 relocation type %d\n", reltab_j->Type); - oc->errMsg("unhandled PEi386 relocation type"); - return FALSE; + barf("unhandled PEi386 relocation type"); + return 0; } } } - return TRUE; + return 1; } -#endif /* defined(cygwin32_TARGET_OS) */ +#endif /* defined(OBJFORMAT_PEi386) */ /* -------------------------------------------------------------------------- - * ELF specifics (Linux, Solaris) + * ELF specifics * ------------------------------------------------------------------------*/ -#if defined(linux_TARGET_OS) || defined(solaris2_TARGET_OS) +#if defined(OBJFORMAT_ELF) #define FALSE 0 #define TRUE 1 +#if defined(sparc_TARGET_ARCH) +# define ELF_TARGET_SPARC /* Used inside */ +#endif + #include static char * @@ -1197,9 +1274,9 @@ ocVerifyImage_ELF ( ObjectCode* oc ) for (i = 0; i < ehdr->e_shnum; i++) { IF_DEBUG(linker,fprintf(stderr, "%2d: ", i )); - IF_DEBUG(linker,fprintf(stderr, "type=%2d ", shdr[i].sh_type )); - IF_DEBUG(linker,fprintf(stderr, "size=%4d ", shdr[i].sh_size )); - IF_DEBUG(linker,fprintf(stderr, "offs=%4d ", shdr[i].sh_offset )); + IF_DEBUG(linker,fprintf(stderr, "type=%2d ", (int)shdr[i].sh_type )); + IF_DEBUG(linker,fprintf(stderr, "size=%4d ", (int)shdr[i].sh_size )); + IF_DEBUG(linker,fprintf(stderr, "offs=%4d ", (int)shdr[i].sh_offset )); IF_DEBUG(linker,fprintf(stderr, " (%p .. %p) ", ehdrC + shdr[i].sh_offset, ehdrC + shdr[i].sh_offset + shdr[i].sh_size - 1)); @@ -1300,17 +1377,19 @@ ocGetNames_ELF ( ObjectCode* oc ) Elf32_Shdr* shdr = (Elf32_Shdr*) (ehdrC + ehdr->e_shoff); char* sh_strtab = ehdrC + shdr[ehdr->e_shstrndx].sh_offset; + ASSERT(symhash != NULL); + if (!strtab) { belch("ocGetNames_ELF: no strtab"); return 0; } k = 0; - oc->sections = stgMallocBytes( ehdr->e_shnum * sizeof(Section), - "ocGetNames_ELF" ); oc->n_sections = ehdr->e_shnum; + oc->sections = stgMallocBytes( oc->n_sections * sizeof(Section), + "ocGetNames_ELF(oc->sections)" ); - for (i = 0; i < ehdr->e_shnum; i++) { + for (i = 0; i < oc->n_sections; i++) { /* make a section entry for relevant sections */ SectionKind kind = SECTIONKIND_OTHER; @@ -1332,37 +1411,57 @@ ocGetNames_ELF ( ObjectCode* oc ) /* copy stuff into this module's object symbol table */ stab = (Elf32_Sym*) (ehdrC + shdr[i].sh_offset); nent = shdr[i].sh_size / sizeof(Elf32_Sym); - oc->symbols = malloc(nent * sizeof(SymbolVal)); + + oc->n_symbols = nent; + oc->symbols = stgMallocBytes(oc->n_symbols * sizeof(char*), + "ocGetNames_ELF(oc->symbols)"); + for (j = 0; j < nent; j++) { - if ( ( ELF32_ST_BIND(stab[j].st_info)==STB_GLOBAL /* || - ELF32_ST_BIND(stab[j].st_info)==STB_LOCAL */ + if ( ( ELF32_ST_BIND(stab[j].st_info)==STB_GLOBAL + || ELF32_ST_BIND(stab[j].st_info)==STB_LOCAL ) - /* and not an undefined symbol */ - && stab[j].st_shndx != SHN_UNDEF - && + /* and not an undefined symbol */ + && stab[j].st_shndx != SHN_UNDEF + /* and not in a "special section" */ + && stab[j].st_shndx < SHN_LORESERVE + && /* and it's a not a section or string table or anything silly */ ( ELF32_ST_TYPE(stab[j].st_info)==STT_FUNC || ELF32_ST_TYPE(stab[j].st_info)==STT_OBJECT || - ELF32_ST_TYPE(stab[j].st_info)==STT_NOTYPE ) - ) { + ELF32_ST_TYPE(stab[j].st_info)==STT_NOTYPE + ) + ) { char* nm = strtab + stab[j].st_name; char* ad = ehdrC + shdr[ stab[j].st_shndx ].sh_offset + stab[j].st_value; ASSERT(nm != NULL); ASSERT(ad != NULL); - IF_DEBUG(linker,belch( "addOTabName: %10p %s %s", - ad, oc->fileName, nm )); - oc->symbols[j].lbl = nm; - oc->symbols[j].addr = ad; - insertStrHashTable(symhash, nm, &(oc->symbols[j])); + oc->symbols[j] = nm; + if (ELF32_ST_BIND(stab[j].st_info)==STB_LOCAL) { + IF_DEBUG(linker,belch( "addOTabName(LOCL): %10p %s %s", + ad, oc->fileName, nm )); + insertStrHashTable(oc->lochash, nm, ad); + } else { + IF_DEBUG(linker,belch( "addOTabName(GLOB): %10p %s %s", + ad, oc->fileName, nm )); + insertStrHashTable(symhash, nm, ad); + } + } + else { + IF_DEBUG(linker,belch( "skipping `%s'", + strtab + stab[j].st_name )); + /* + fprintf(stderr, + "skipping bind = %d, type = %d, shndx = %d `%s'\n", + (int)ELF32_ST_BIND(stab[j].st_info), + (int)ELF32_ST_TYPE(stab[j].st_info), + (int)stab[j].st_shndx, + strtab + stab[j].st_name + ); + */ + oc->symbols[j] = NULL; } - else { - IF_DEBUG(linker,belch( "skipping `%s'", strtab + - stab[j].st_name )); - oc->symbols[j].lbl = NULL; - oc->symbols[j].addr = NULL; - } } } @@ -1370,16 +1469,190 @@ ocGetNames_ELF ( ObjectCode* oc ) } +/* Do ELF relocations which lack an explicit addend. All x86-linux + relocations appear to be of this form. */ +static int do_Elf32_Rel_relocations ( ObjectCode* oc, char* ehdrC, + Elf32_Shdr* shdr, int shnum, + Elf32_Sym* stab, char* strtab ) +{ + int j; + char *symbol; + Elf32_Word* targ; + Elf32_Rel* rtab = (Elf32_Rel*) (ehdrC + shdr[shnum].sh_offset); + int nent = shdr[shnum].sh_size / sizeof(Elf32_Rel); + int target_shndx = shdr[shnum].sh_info; + int symtab_shndx = shdr[shnum].sh_link; + stab = (Elf32_Sym*) (ehdrC + shdr[ symtab_shndx ].sh_offset); + targ = (Elf32_Word*)(ehdrC + shdr[ target_shndx ].sh_offset); + IF_DEBUG(linker,belch( "relocations for section %d using symtab %d", + target_shndx, symtab_shndx )); + for (j = 0; j < nent; j++) { + Elf32_Addr offset = rtab[j].r_offset; + Elf32_Word info = rtab[j].r_info; + + Elf32_Addr P = ((Elf32_Addr)targ) + offset; + Elf32_Word* pP = (Elf32_Word*)P; + Elf32_Addr A = *pP; + Elf32_Addr S; + + IF_DEBUG(linker,belch( "Rel entry %3d is raw(%6p %6p)", + j, (void*)offset, (void*)info )); + if (!info) { + IF_DEBUG(linker,belch( " ZERO" )); + S = 0; + } else { + /* First see if it is a nameless local symbol. */ + if (stab[ ELF32_R_SYM(info)].st_name == 0) { + symbol = "(noname)"; + S = (Elf32_Addr) + (ehdrC + shdr[stab[ELF32_R_SYM(info)].st_shndx ].sh_offset + + stab[ELF32_R_SYM(info)].st_value); + } else { + /* No? Should be in a symbol table then; first try the + local one. */ + symbol = strtab+stab[ ELF32_R_SYM(info)].st_name; + (void*)S = lookupLocalSymbol( oc, symbol ); + if ((void*)S == NULL) + (void*)S = lookupSymbol( symbol ); + } + if (!S) { + barf("do_Elf32_Rel_relocations: %s: unknown symbol `%s'", + oc->fileName, symbol); + } + IF_DEBUG(linker,belch( "`%s' resolves to %p", symbol, (void*)S )); + } + IF_DEBUG(linker,belch( "Reloc: P = %p S = %p A = %p", + (void*)P, (void*)S, (void*)A )); + switch (ELF32_R_TYPE(info)) { +# ifdef i386_TARGET_ARCH + case R_386_32: *pP = S + A; break; + case R_386_PC32: *pP = S + A - P; break; +# endif + default: + fprintf(stderr, "unhandled ELF relocation(Rel) type %d\n", + ELF32_R_TYPE(info)); + barf("do_Elf32_Rel_relocations: unhandled ELF relocation type"); + return 0; + } + + } + return 1; +} + + +/* Do ELF relocations for which explicit addends are supplied. + sparc-solaris relocations appear to be of this form. */ +static int do_Elf32_Rela_relocations ( ObjectCode* oc, char* ehdrC, + Elf32_Shdr* shdr, int shnum, + Elf32_Sym* stab, char* strtab ) +{ + int j; + char *symbol; + Elf32_Word* targ; + Elf32_Rela* rtab = (Elf32_Rela*) (ehdrC + shdr[shnum].sh_offset); + int nent = shdr[shnum].sh_size / sizeof(Elf32_Rela); + int target_shndx = shdr[shnum].sh_info; + int symtab_shndx = shdr[shnum].sh_link; + stab = (Elf32_Sym*) (ehdrC + shdr[ symtab_shndx ].sh_offset); + targ = (Elf32_Word*)(ehdrC + shdr[ target_shndx ].sh_offset); + IF_DEBUG(linker,belch( "relocations for section %d using symtab %d", + target_shndx, symtab_shndx )); + for (j = 0; j < nent; j++) { + Elf32_Addr offset = rtab[j].r_offset; + Elf32_Word info = rtab[j].r_info; + Elf32_Sword addend = rtab[j].r_addend; + + Elf32_Addr P = ((Elf32_Addr)targ) + offset; + Elf32_Addr A = addend; + Elf32_Addr S; +# if defined(sparc_TARGET_ARCH) + /* This #ifdef only serves to avoid unused-var warnings. */ + Elf32_Word* pP = (Elf32_Word*)P; + Elf32_Word w1, w2; +# endif + + IF_DEBUG(linker,belch( "Rel entry %3d is raw(%6p %6p %6p) ", + j, (void*)offset, (void*)info, + (void*)addend )); + if (!info) { + IF_DEBUG(linker,belch( " ZERO" )); + S = 0; + } else { + /* First see if it is a nameless local symbol. */ + if (stab[ ELF32_R_SYM(info)].st_name == 0) { + symbol = "(noname)"; + S = (Elf32_Addr) + (ehdrC + shdr[stab[ELF32_R_SYM(info)].st_shndx ].sh_offset + + stab[ELF32_R_SYM(info)].st_value); + } else { + /* No? Should be in a symbol table then; first try the + local one. */ + symbol = strtab+stab[ ELF32_R_SYM(info)].st_name; + (void*)S = lookupLocalSymbol( oc, symbol ); + if ((void*)S == NULL) + (void*)S = lookupSymbol( symbol ); + } + if (!S) { + barf("do_Elf32_Rela_relocations: %s: unknown symbol `%s'", + oc->fileName, symbol); + /* + S = 0x11223344; + fprintf ( stderr, "S %p A %p S+A %p S+A-P %p\n",S,A,S+A,S+A-P); + */ + } + IF_DEBUG(linker,belch( "`%s' resolves to %p", symbol, (void*)S )); + } + IF_DEBUG(linker,fprintf ( stderr, "Reloc: P = %p S = %p A = %p\n", + (void*)P, (void*)S, (void*)A )); + switch (ELF32_R_TYPE(info)) { +# if defined(sparc_TARGET_ARCH) + case R_SPARC_WDISP30: + w1 = *pP & 0xC0000000; + w2 = (Elf32_Word)((S + A - P) >> 2); + ASSERT((w2 & 0xC0000000) == 0); + w1 |= w2; + *pP = w1; + break; + case R_SPARC_HI22: + w1 = *pP & 0xFFC00000; + w2 = (Elf32_Word)((S + A) >> 10); + ASSERT((w2 & 0xFFC00000) == 0); + w1 |= w2; + *pP = w1; + break; + case R_SPARC_LO10: + w1 = *pP & ~0x3FF; + w2 = (Elf32_Word)((S + A) & 0x3FF); + ASSERT((w2 & ~0x3FF) == 0); + w1 |= w2; + *pP = w1; + break; + case R_SPARC_32: + w2 = (Elf32_Word)(S + A); + *pP = w2; + break; +# endif + default: + fprintf(stderr, "unhandled ELF relocation(RelA) type %d\n", + ELF32_R_TYPE(info)); + barf("do_Elf32_Rela_relocations: unhandled ELF relocation type"); + return 0; + } + + } + return 1; +} + + static int ocResolve_ELF ( ObjectCode* oc ) { - char *strtab, *symbol; - int i, j; + char *strtab; + int shnum, ok; Elf32_Sym* stab = NULL; char* ehdrC = (char*)(oc->image); Elf32_Ehdr* ehdr = (Elf32_Ehdr*) ehdrC; Elf32_Shdr* shdr = (Elf32_Shdr*) (ehdrC + ehdr->e_shoff); - Elf32_Word* targ; /* first find "the" symbol table */ stab = (Elf32_Sym*) findElfSection ( ehdrC, SHT_SYMTAB ); @@ -1392,121 +1665,27 @@ ocResolve_ELF ( ObjectCode* oc ) return 0; } - for (i = 0; i < ehdr->e_shnum; i++) { - if (shdr[i].sh_type == SHT_REL ) { - Elf32_Rel* rtab = (Elf32_Rel*) (ehdrC + shdr[i].sh_offset); - int nent = shdr[i].sh_size / sizeof(Elf32_Rel); - int target_shndx = shdr[i].sh_info; - int symtab_shndx = shdr[i].sh_link; - stab = (Elf32_Sym*) (ehdrC + shdr[ symtab_shndx ].sh_offset); - targ = (Elf32_Word*)(ehdrC + shdr[ target_shndx ].sh_offset); - IF_DEBUG(linker,belch( "relocations for section %d using symtab %d", - target_shndx, symtab_shndx )); - for (j = 0; j < nent; j++) { - Elf32_Addr offset = rtab[j].r_offset; - Elf32_Word info = rtab[j].r_info; - - Elf32_Addr P = ((Elf32_Addr)targ) + offset; - Elf32_Word* pP = (Elf32_Word*)P; - Elf32_Addr A = *pP; - Elf32_Addr S; - - IF_DEBUG(linker,belch( "Rel entry %3d is raw(%6p %6p) ", - j, (void*)offset, (void*)info )); - if (!info) { - IF_DEBUG(linker,belch( " ZERO" )); - S = 0; - } else { - /* First see if it is a nameless local symbol. */ - if (stab[ ELF32_R_SYM(info)].st_name == 0) { - symbol = "(noname)"; - S = (Elf32_Addr)(ehdrC - + shdr[stab[ELF32_R_SYM(info)].st_shndx ].sh_offset - + stab[ELF32_R_SYM(info)].st_value - ); - } else { - /* No? Should be in the symbol table then. */ - symbol = strtab+stab[ ELF32_R_SYM(info)].st_name; - (void *)S = lookupSymbol( symbol ); - } - if (!S) { - barf("ocResolve_ELF: %s: unknown symbol `%s'", - oc->fileName, symbol); - } - IF_DEBUG(linker,belch( "`%s' resolves to %p", symbol, (void*)S )); - } - IF_DEBUG(linker,fprintf ( stderr, "Reloc: P = %p S = %p A = %p\n", (void*)P, (void*)S, (void*)A )); - switch (ELF32_R_TYPE(info)) { - case R_386_32: *pP = S + A; break; - case R_386_PC32: *pP = S + A - P; break; - default: fprintf(stderr, - "unhandled ELF relocation type %d", - ELF32_R_TYPE(info)); - belch("ocResolve_ELF: unhandled ELF relocation type"); - return 0; - } - - } + /* Process the relocation sections. */ + for (shnum = 0; shnum < ehdr->e_shnum; shnum++) { + if (shdr[shnum].sh_type == SHT_REL ) { + ok = do_Elf32_Rel_relocations ( oc, ehdrC, shdr, + shnum, stab, strtab ); + if (!ok) return ok; } else - if (shdr[i].sh_type == SHT_RELA) { - belch("ocResolve_ELF: RelA style reloc table -- not yet done"); - return 0; + if (shdr[shnum].sh_type == SHT_RELA) { + ok = do_Elf32_Rela_relocations ( oc, ehdrC, shdr, + shnum, stab, strtab ); + if (!ok) return ok; } } - return 1; -} - - -#endif /* defined(linux_TARGET_OS) || defined(solaris2_TARGET_OS) */ + /* Free the local symbol table; we won't need it again. */ + freeHashTable(oc->lochash, NULL); + oc->lochash = NULL; -/* ----------------------------------------------------------------------------- - * Look up an address to discover whether it is in text or data space. - * - * Used by the garbage collector when walking the stack. - * -------------------------------------------------------------------------- */ - -static __inline__ SectionKind -lookupSection ( void* addr ) -{ - int i; - ObjectCode* oc; - - for ( oc = objects; oc; oc = oc->next ) { - for (i = 0; i < oc->n_sections; i++) { - if (oc->sections[i].start <= addr - && addr <= oc->sections[i].end) - return oc->sections[i].kind; - } - } - return SECTIONKIND_OTHER; -} - -int -is_dynamically_loaded_code_or_rodata_ptr ( char* p ) -{ - SectionKind sk = lookupSection(p); - ASSERT (sk != SECTIONKIND_NOINFOAVAIL); - return (sk == SECTIONKIND_CODE_OR_RODATA); -} - - -int -is_dynamically_loaded_rwdata_ptr ( char* p ) -{ - SectionKind sk = lookupSection(p); - ASSERT (sk != SECTIONKIND_NOINFOAVAIL); - return (sk == SECTIONKIND_RWDATA); + return 1; } -int -is_not_dynamically_loaded_ptr ( char* p ) -{ - SectionKind sk = lookupSection(p); - ASSERT (sk != SECTIONKIND_NOINFOAVAIL); - return (sk == SECTIONKIND_OTHER); -} - -#endif /* GHCI */ +#endif /* ELF */