AcessNative - Work on the ELF Loader
[tpg/acess2.git] / AcessNative / ld-acess.so_src / elf.c
1 /*\r
2  * Acess v0.1\r
3  * ELF Executable Loader Code\r
4  */\r
5 #define DEBUG   0\r
6 #include <stdlib.h>\r
7 #include <stdio.h>\r
8 #include <string.h>\r
9 #include <unistd.h>\r
10 #include "common.h"\r
11 #include "elf.h"\r
12 \r
13 #define DEBUG_WARN      1\r
14 \r
15 #define MKPTR(_type,_val)       ((_type*)(uintptr_t)(_val))\r
16 #define PTRMK(_type,_val)       MKPTR(_type,_val)\r
17 #define PTR(_val)       ((void*)(uintptr_t)(_val))\r
18 \r
19 #if 0\r
20 # define ENTER(...)\r
21 # define LOG(s, ...)    printf("%s: " s, __func__, __VA_ARGS__)\r
22 # define LOGS(s)        printf("%s: " s, __func__)\r
23 # define LEAVE(...)\r
24 #else\r
25 #define ENTER(...)\r
26 #define LOG(...)\r
27 #define LOGS(...)\r
28 #define LEAVE(...)\r
29 #endif\r
30 \r
31 // === PROTOTYPES ===\r
32 void    *Elf_Load(FILE *FP);\r
33 uintptr_t       Elf_Relocate(void *Base);\r
34  int    Elf_GetSymbol(void *Base, char *Name, uintptr_t *ret);\r
35  int    Elf_Int_DoRelocate(uint32_t r_info, uint32_t *ptr, uint32_t addend, Elf32_Sym *symtab, void *Base);\r
36 uint32_t        Elf_Int_HashString(char *str);\r
37 \r
38 // === CODE ===\r
39 void *Elf_Load(FILE *FP)\r
40 {\r
41         Elf32_Ehdr      hdr;\r
42         Elf32_Phdr      *phtab;\r
43          int    i, j;\r
44          int    iPageCount;\r
45         uint32_t        max, base = -1;\r
46         uint32_t        addr;\r
47         uint32_t        baseDiff = 0;\r
48         \r
49         ENTER("pFP", FP);\r
50         \r
51         // Read ELF Header\r
52         fread(&hdr, sizeof(hdr), 1, FP);\r
53         \r
54         // Check the file type\r
55         if(hdr.ident[0] != 0x7F || hdr.ident[1] != 'E' || hdr.ident[2] != 'L' || hdr.ident[3] != 'F') {\r
56                 Warning("Non-ELF File was passed to the ELF loader\n");\r
57                 LEAVE('n');\r
58                 return NULL;\r
59         }\r
60         \r
61         // Check for a program header\r
62         if(hdr.phoff == 0) {\r
63                 #if DEBUG_WARN\r
64                 Warning("ELF File does not contain a program header\n");\r
65                 #endif\r
66                 LEAVE('n');\r
67                 return NULL;\r
68         }\r
69         \r
70         // Read Program Header Table\r
71         phtab = malloc( sizeof(Elf32_Phdr) * hdr.phentcount );\r
72         if( !phtab ) {\r
73                 LEAVE('n');\r
74                 return NULL;\r
75         }\r
76         LOG("hdr.phoff = 0x%08x\n", hdr.phoff);\r
77         fseek(FP, hdr.phoff, SEEK_SET);\r
78         fread(phtab, sizeof(Elf32_Phdr), hdr.phentcount, FP);\r
79         \r
80         // Count Pages\r
81         iPageCount = 0;\r
82         LOG("hdr.phentcount = %i\n", hdr.phentcount);\r
83         for( i = 0; i < hdr.phentcount; i++ )\r
84         {\r
85                 // Ignore Non-LOAD types\r
86                 if(phtab[i].Type != PT_LOAD)\r
87                         continue;\r
88                 iPageCount += ((phtab[i].VAddr&0xFFF) + phtab[i].MemSize + 0xFFF) >> 12;\r
89                 LOG("phtab[%i] = {VAddr:0x%x, MemSize:0x%x}\n", i, phtab[i].VAddr, phtab[i].MemSize);\r
90         }\r
91         \r
92         LOG("iPageCount = %i\n", iPageCount);\r
93         \r
94         // Allocate Information Structure\r
95         //ret = malloc( sizeof(tBinary) + sizeof(tBinaryPage)*iPageCount );\r
96         // Fill Info Struct\r
97         //ret->Entry = hdr.entrypoint;\r
98         //ret->Base = -1;               // Set Base to maximum value\r
99         //ret->NumPages = iPageCount;\r
100         //ret->Interpreter = NULL;\r
101 \r
102         // Prescan for base and size\r
103         for( i = 0; i < hdr.phentcount; i ++)\r
104         {\r
105                 if( phtab[i].Type != PT_LOAD )\r
106                         continue;\r
107                 if( phtab[i].VAddr < base )\r
108                         base = phtab[i].VAddr;\r
109                 if( phtab[i].VAddr > max )\r
110                         max = phtab[i].VAddr;\r
111         }\r
112 \r
113         LOG("base = %08x, max = %08x\n", base, max);\r
114 \r
115         if( base == 0 ) {\r
116                 // Find a nice space (31 address bits allowed)\r
117                 base = FindFreeRange( max, 31 );\r
118                 LOG("new base = %08x\n", base);\r
119                 if( base == 0 ) return NULL;\r
120                 baseDiff = base;\r
121         }\r
122         \r
123         // Load Pages\r
124         j = 0;\r
125         for( i = 0; i < hdr.phentcount; i++ )\r
126         {\r
127                 //LOG("phtab[%i].Type = 0x%x", i, phtab[i].Type);\r
128                 LOG("phtab[%i] = {\n", i);\r
129                 LOG(" .Type = 0x%08x\n", phtab[i].Type);\r
130                 LOG(" .Offset = 0x%08x\n", phtab[i].Offset);\r
131                 LOG(" .VAddr = 0x%08x\n", phtab[i].VAddr);\r
132                 LOG(" .PAddr = 0x%08x\n", phtab[i].PAddr);\r
133                 LOG(" .FileSize = 0x%08x\n", phtab[i].FileSize);\r
134                 LOG(" .MemSize = 0x%08x\n", phtab[i].MemSize);\r
135                 LOG(" .Flags = 0x%08x\n", phtab[i].Flags);\r
136                 LOG(" .Align = 0x%08x\n", phtab[i].Align);\r
137                 LOGS(" }\n");\r
138                 // Get Interpreter Name\r
139                 if( phtab[i].Type == PT_INTERP )\r
140                 {\r
141                         char *tmp;\r
142                         //if(ret->Interpreter)  continue;\r
143                         tmp = malloc(phtab[i].FileSize);\r
144                         fseek(FP, phtab[i].Offset, SEEK_SET);\r
145                         fread(tmp, phtab[i].FileSize, 1, FP);\r
146                         //ret->Interpreter = Binary_RegInterp(tmp);\r
147                         LOG("Interpreter '%s'\n", tmp);\r
148                         free(tmp);\r
149                         continue;\r
150                 }\r
151                 // Ignore non-LOAD types\r
152                 if(phtab[i].Type != PT_LOAD)    continue;\r
153                 \r
154                 LOG("phtab[%i] = {VAddr:0x%x,Offset:0x%x,FileSize:0x%x}\n",\r
155                         i, phtab[i].VAddr+baseDiff, phtab[i].Offset, phtab[i].FileSize);\r
156                 \r
157                 addr = phtab[i].VAddr + baseDiff;\r
158 \r
159                 if( AllocateMemory( addr, phtab[i].MemSize ) ) {\r
160                         return NULL;\r
161                 }\r
162                 \r
163                 fseek(FP, phtab[i].Offset, SEEK_SET);\r
164                 fread( PTRMK(void, addr), phtab[i].FileSize, 1, FP );\r
165                 memset( PTRMK(char, addr) + phtab[i].FileSize, 0, phtab[i].MemSize - phtab[i].FileSize );\r
166         }\r
167         \r
168         // Clean Up\r
169         free(phtab);\r
170         // Return\r
171         LEAVE('p', base);\r
172         return PTRMK(void, base);\r
173 }\r
174 \r
175 // --- ELF RELOCATION ---\r
176 /**\r
177  * \brief Relocates a loaded ELF Executable\r
178  */\r
179 uintptr_t Elf_Relocate(void *Base)\r
180 {\r
181         Elf32_Ehdr      *hdr = Base;\r
182         Elf32_Phdr      *phtab;\r
183          int    i, j;   // Counters\r
184         char    *libPath;\r
185         uint32_t        iRealBase = -1;\r
186         uintptr_t       iBaseDiff;\r
187          int    iSegmentCount;\r
188          int    iSymCount = 0;\r
189         Elf32_Rel       *rel = NULL;\r
190         Elf32_Rela      *rela = NULL;\r
191         uint32_t        *pltgot = NULL;\r
192         void    *plt = NULL;\r
193         uint32_t        *ptr;\r
194          int    relSz=0, relEntSz=8;\r
195          int    relaSz=0, relaEntSz=8;\r
196          int    pltSz=0, pltType=0;\r
197         Elf32_Dyn       *dynamicTab = NULL;     // Dynamic Table Pointer\r
198         char    *dynstrtab = NULL;      // .dynamic String Table\r
199         Elf32_Sym       *dynsymtab = NULL;\r
200          int    bFailed = 0;\r
201         \r
202         ENTER("pBase", Base);\r
203         LOG("Base = %p\n", Base);\r
204         \r
205         // Parse Program Header to get Dynamic Table\r
206         phtab = Base + hdr->phoff;\r
207         iSegmentCount = hdr->phentcount;\r
208         for(i = 0; i < iSegmentCount; i ++ )\r
209         {\r
210                 // Determine linked base address\r
211                 if(phtab[i].Type == PT_LOAD && iRealBase > phtab[i].VAddr)\r
212                         iRealBase = phtab[i].VAddr;\r
213                 \r
214                 // Find Dynamic Section\r
215                 if(phtab[i].Type == PT_DYNAMIC) {\r
216                         if(dynamicTab) {\r
217                                 Warning("Elf_Relocate - Multiple PT_DYNAMIC segments\n");\r
218                                 continue;\r
219                         }\r
220                         dynamicTab = MKPTR(void, phtab[i].VAddr);\r
221                         j = i;  // Save Dynamic Table ID\r
222                         break;\r
223                 }\r
224         }\r
225         \r
226         // Check if a PT_DYNAMIC segement was found\r
227         if(!dynamicTab) {\r
228                 Warning("Elf_Relocate: No PT_DYNAMIC segment in image, returning\n");\r
229                 LEAVE('x', hdr->entrypoint);\r
230                 return hdr->entrypoint;\r
231         }\r
232         \r
233         // Page Align real base\r
234         iRealBase &= ~0xFFF;\r
235         \r
236         LOG("dynamicTab = %p\n", dynamicTab);\r
237         // Adjust "Real" Base\r
238         iBaseDiff = (uintptr_t)Base - iRealBase;\r
239         LOG("iBaseDiff = %p\n", (void*)iBaseDiff);\r
240         // Adjust Dynamic Table\r
241         dynamicTab = PTR( (uintptr_t)dynamicTab + iBaseDiff);\r
242         LOG("dynamicTab = %p\n", dynamicTab);\r
243 \r
244         hdr->entrypoint += iBaseDiff;\r
245         \r
246         // === Get Symbol table and String Table ===\r
247         for( j = 0; dynamicTab[j].d_tag != DT_NULL; j++)\r
248         {\r
249                 switch(dynamicTab[j].d_tag)\r
250                 {\r
251                 // --- Symbol Table ---\r
252                 case DT_SYMTAB:\r
253                         dynamicTab[j].d_val += iBaseDiff;\r
254                         dynsymtab = PTRMK(void, dynamicTab[j].d_val);\r
255                         hdr->misc.SymTable = dynamicTab[j].d_val;       // Saved in unused bytes of ident\r
256                         break;\r
257                 \r
258                 // --- String Table ---\r
259                 case DT_STRTAB:\r
260                         dynamicTab[j].d_val += iBaseDiff;\r
261                         dynstrtab = PTRMK(void, dynamicTab[j].d_val);\r
262                         break;\r
263                 \r
264                 // --- Hash Table --\r
265                 case DT_HASH:\r
266                         dynamicTab[j].d_val += iBaseDiff;\r
267                         iSymCount = (PTRMK(uint32_t, dynamicTab[j].d_val))[1];\r
268                         hdr->misc.HashTable = dynamicTab[j].d_val;      // Saved in unused bytes of ident\r
269                         break;\r
270                 }\r
271         }\r
272 \r
273 \r
274         // Alter Symbols to true base\r
275         for(i = 0; i < iSymCount; i ++)\r
276         {\r
277                 dynsymtab[i].value += iBaseDiff;\r
278                 dynsymtab[i].nameOfs += (uintptr_t)dynstrtab;\r
279                 LOG("Sym '%s' = 0x%x (relocated)\n", MKPTR(char,dynsymtab[i].name), dynsymtab[i].value);\r
280         }\r
281         \r
282         // === Add to loaded list (can be imported now) ===\r
283         Binary_SetReadyToUse( Base );\r
284 \r
285         // === Parse Relocation Data ===\r
286         for( j = 0; dynamicTab[j].d_tag != DT_NULL; j++)\r
287         {\r
288                 switch(dynamicTab[j].d_tag)\r
289                 {\r
290                 // --- Shared Library Name ---\r
291                 case DT_SONAME:\r
292                         LOG(".so Name '%s'\n", dynstrtab + dynamicTab[j].d_val);\r
293                         break;\r
294                 // --- Needed Library ---\r
295                 case DT_NEEDED:\r
296                         libPath = dynstrtab + dynamicTab[j].d_val;\r
297                         Binary_LoadLibrary(libPath);\r
298                         break;\r
299                 // --- PLT/GOT ---\r
300                 case DT_PLTGOT: pltgot = (void*)(iBaseDiff+dynamicTab[j].d_val);        break;\r
301                 case DT_JMPREL: plt = (void*)(iBaseDiff+dynamicTab[j].d_val);   break;\r
302                 case DT_PLTREL: pltType = dynamicTab[j].d_val;  break;\r
303                 case DT_PLTRELSZ:       pltSz = dynamicTab[j].d_val;    break;\r
304                 \r
305                 // --- Relocation ---\r
306                 case DT_REL:    rel = (void*)(iBaseDiff + dynamicTab[j].d_val); break;\r
307                 case DT_RELSZ:  relSz = dynamicTab[j].d_val;    break;\r
308                 case DT_RELENT: relEntSz = dynamicTab[j].d_val; break;\r
309                 \r
310                 case DT_RELA:   rela = (void*)(iBaseDiff + dynamicTab[j].d_val);        break;\r
311                 case DT_RELASZ: relaSz = dynamicTab[j].d_val;   break;\r
312                 case DT_RELAENT:        relaEntSz = dynamicTab[j].d_val;        break;\r
313                 }\r
314         }\r
315         \r
316         // Parse Relocation Entries\r
317         if(rel && relSz)\r
318         {\r
319                 j = relSz / relEntSz;\r
320                 for( i = 0; i < j; i++ )\r
321                 {\r
322                         ptr = (void*)(iBaseDiff + rel[i].r_offset);\r
323                         if( !Elf_Int_DoRelocate(rel[i].r_info, ptr, *ptr, dynsymtab, Base) ) {\r
324                                 bFailed = 1;\r
325                         }\r
326                 }\r
327         }\r
328         // Parse Relocation Entries\r
329         if(rela && relaSz)\r
330         {\r
331                 j = relaSz / relaEntSz;\r
332                 for( i = 0; i < j; i++ )\r
333                 {\r
334                         ptr = (void*)(iBaseDiff + rela[i].r_offset);\r
335                         if( !Elf_Int_DoRelocate(rel[i].r_info, ptr, rela[i].r_addend, dynsymtab, Base) ) {\r
336                                 bFailed = 1;\r
337                         }\r
338                 }\r
339         }\r
340         \r
341         // === Process PLT (Procedure Linkage Table) ===\r
342         if(plt && pltSz)\r
343         {\r
344                 if(pltType == DT_REL)\r
345                 {\r
346                         Elf32_Rel       *pltRel = plt;\r
347                         j = pltSz / sizeof(Elf32_Rel);\r
348                         LOG("PLT Rel - plt = %p, pltSz = %i (%i ents)\n", plt, pltSz, j);\r
349                         for(i = 0; i < j; i++)\r
350                         {\r
351                                 ptr = (void*)(iBaseDiff + pltRel[i].r_offset);\r
352                                 if( !Elf_Int_DoRelocate(pltRel[i].r_info, ptr, *ptr, dynsymtab, Base) ) {\r
353                                         bFailed = 1;\r
354                                 }\r
355                         }\r
356                 }\r
357                 else\r
358                 {\r
359                         Elf32_Rela      *pltRela = plt;\r
360                         j = pltSz / sizeof(Elf32_Rela);\r
361                         LOG("PLT RelA - plt = %p, pltSz = %i (%i ents)\n", plt, pltSz, j);\r
362                         for(i=0;i<j;i++)\r
363                         {\r
364                                 ptr = (void*)(iBaseDiff + pltRela[i].r_offset);\r
365                                 if( !Elf_Int_DoRelocate(pltRela[i].r_info, ptr, pltRela[i].r_addend, dynsymtab, Base) ) {\r
366                                         bFailed = 1;\r
367                                 }\r
368                         }\r
369                 }\r
370         }\r
371         \r
372         if(bFailed) {\r
373                 LEAVE('i', 0);\r
374                 return 0;\r
375         }\r
376         \r
377         LEAVE('x', hdr->entrypoint);\r
378         return hdr->entrypoint;\r
379 }\r
380 \r
381 /**\r
382  * \fn void Elf_Int_DoRelocate(uint32_t r_info, uint32_t *ptr, uint32_t addend, Elf32_Sym *symtab, void *base)\r
383  * \brief Performs a relocation\r
384  * \param r_info        Field from relocation entry\r
385  * \param ptr   Pointer to location of relocation\r
386  * \param addend        Value to add to symbol\r
387  * \param symtab        Symbol Table\r
388  * \param base  Base of loaded binary\r
389  */\r
390 int Elf_Int_DoRelocate(uint32_t r_info, uint32_t *ptr, uint32_t addend, Elf32_Sym *symtab, void *base)\r
391 {\r
392         uintptr_t       val;\r
393          int    type = ELF32_R_TYPE(r_info);\r
394          int    sym = ELF32_R_SYM(r_info);\r
395         char    *sSymName = PTRMK(char, symtab[sym].name);\r
396         \r
397         //LogF("Elf_Int_DoRelocate: (r_info=0x%x, ptr=0x%x, addend=0x%x, .., base=0x%x)\n",\r
398         //      r_info, ptr, addend, base);\r
399         \r
400         switch( type )\r
401         {\r
402         // Standard 32 Bit Relocation (S+A)\r
403         case R_386_32:\r
404                 if( !Elf_GetSymbol( base, sSymName, &val ) && !Binary_GetSymbol( sSymName, &val ) ) {\r
405                         Warning("Unable to find symbol '%s'", sSymName);\r
406                         return 0;\r
407                 }\r
408                 LOG("%08x R_386_32 *%p += %p('%s')\n", r_info, ptr, (void*)val, sSymName);\r
409                 *ptr = val + addend;\r
410                 break;\r
411                 \r
412         // 32 Bit Relocation wrt. Offset (S+A-P)\r
413         case R_386_PC32:\r
414                 if( !Elf_GetSymbol( base, sSymName, &val ) && !Binary_GetSymbol( sSymName, &val ) ) {\r
415                         Warning("Unable to find symbol '%s'", sSymName);\r
416                         return 0;\r
417                 }\r
418                 LOG("%08x R_386_PC32 *%p = 0x%x + %p('%s') - %p\n", r_info, ptr, *ptr, (void*)val, sSymName, ptr );\r
419                 // TODO: Check if it needs the true value of ptr or the compiled value\r
420                 // NOTE: Testing using true value\r
421                 *ptr = val + addend - (uintptr_t)ptr;\r
422                 break;\r
423 \r
424         // Absolute Value of a symbol (S)\r
425         case R_386_GLOB_DAT:\r
426                 if( !Elf_GetSymbol( base, sSymName, &val ) && !Binary_GetSymbol( sSymName, &val ) ) {\r
427                         Warning("Unable to find symbol '%s'", sSymName);\r
428                         return 0; \r
429                 }\r
430                 LOG("%08x R_386_GLOB_DAT *%p = 0x%x(%s)\n", r_info, ptr, (unsigned int)val, sSymName);\r
431                 *ptr = val;\r
432                 break;\r
433         \r
434         // Absolute Value of a symbol (S)\r
435         case R_386_JMP_SLOT:\r
436                 if( !Elf_GetSymbol( base, sSymName, &val ) && !Binary_GetSymbol( sSymName, &val ) ) {\r
437                         Warning("Unable to find symbol '%s'", sSymName);\r
438                         return 0;\r
439                 }\r
440                 LOG("%08x R_386_JMP_SLOT *%p = 0x%x (%s)\n", r_info, ptr, (unsigned int)val, sSymName);\r
441                 *ptr = val;\r
442                 break;\r
443 \r
444         // Base Address (B+A)\r
445         case R_386_RELATIVE:\r
446                 LOG("%08x R_386_RELATIVE *%p = %p + 0x%x\n", r_info, ptr, base, addend);\r
447                 *ptr = (uintptr_t)base + addend;\r
448                 break;\r
449                 \r
450         default:\r
451                 LOG("Rel %p: 0x%x,%i\n", ptr, sym, type);\r
452                 break;\r
453         }\r
454         return 1;\r
455 }\r
456 \r
457 /**\r
458  * \fn int Elf_GetSymbol(void *Base, char *name, uintptr_t *ret)\r
459  * \brief Get a symbol from the loaded binary\r
460  */\r
461 int Elf_GetSymbol(void *Base, char *Name, uintptr_t *ret)\r
462 {\r
463         Elf32_Ehdr      *hdr = (void*)Base;\r
464         Elf32_Sym       *symtab;\r
465          int    nbuckets = 0;\r
466          int    iSymCount = 0;\r
467          int    i;\r
468         uint32_t        *pBuckets;\r
469         uint32_t        *pChains;\r
470         uint32_t        iNameHash;\r
471 \r
472         if(!Base)       return 0;\r
473 \r
474         pBuckets = PTR(hdr->misc.HashTable);\r
475         symtab = PTR(hdr->misc.SymTable);\r
476         \r
477         nbuckets = pBuckets[0];\r
478         iSymCount = pBuckets[1];\r
479         pBuckets = &pBuckets[2];\r
480         pChains = &pBuckets[ nbuckets ];\r
481         \r
482         // Get hash\r
483         iNameHash = Elf_Int_HashString(Name);\r
484         iNameHash %= nbuckets;\r
485 \r
486         // Check Bucket\r
487         i = pBuckets[ iNameHash ];\r
488         if(symtab[i].shndx != SHN_UNDEF && strcmp(MKPTR(char,symtab[i].name), Name) == 0) {\r
489                 if(ret) *ret = symtab[ i ].value;\r
490                 return 1;\r
491         }\r
492         \r
493         // Walk Chain\r
494         while(pChains[i] != STN_UNDEF)\r
495         {\r
496                 i = pChains[i];\r
497                 if(symtab[i].shndx != SHN_UNDEF && strcmp(MKPTR(char,symtab[i].name), Name) == 0) {\r
498                         if(ret) *ret = symtab[ i ].value;\r
499                         return 1;\r
500                 }\r
501         }\r
502         return 0;\r
503 }\r
504 \r
505 /**\r
506  * \fn uint32_t Elf_Int_HashString(char *str)\r
507  * \brief Hash a string in the ELF format\r
508  * \param str   String to hash\r
509  * \return Hash value\r
510  */\r
511 uint32_t Elf_Int_HashString(char *str)\r
512 {\r
513         uint32_t        h = 0, g;\r
514         while(*str)\r
515         {\r
516                 h = (h << 4) + *str++;\r
517                 if( (g = h & 0xf0000000) )\r
518                         h ^= g >> 24;\r
519                 h &= ~g;\r
520         }\r
521         return h;\r
522 }\r

UCC git Repository :: git.ucc.asn.au