+/*\r
+ * Acess v1\r
+ * Portable Executable Loader\r
+ */\r
+#define DEBUG 1\r
+#include <acess.h>\r
+#include <binary.h>\r
+#include <modules.h>\r
+#include "pe.h"\r
+\r
+// === PROTOTYPES ===\r
+ int PE_Install(char **Arguments);\r
+tBinary *PE_Load(int fp);\r
+tBinary *MZ_Open(int fp);\r
+ int PE_Relocate(void *Base);\r
+ int PE_GetSymbol(void *Base, char *Name, Uint *Ret);\r
+\r
+// === GLOBALS ===\r
+MODULE_DEFINE(0, 0x0032, BinPE, PE_Install, NULL, NULL);\r
+char *gsPE_DefaultInterpreter = "/Acess/Libs/ld-acess.so";\r
+tBinaryType gPE_Loader = {\r
+ NULL,\r
+ ('M'|('Z'<<8)), 0xFFFF, // 'MZ'\r
+ "PE/DOS",\r
+ PE_Load, PE_Relocate, PE_GetSymbol\r
+ };\r
+\r
+// === CODE ===\r
+int PE_Install(char **Arguments)\r
+{\r
+ Binary_RegisterType(&gPE_Loader);\r
+ return 1;\r
+}\r
+\r
+/**\r
+ * \brief Loads a PE Binary\r
+ */\r
+tBinary *PE_Load(int FP)\r
+{\r
+ int count, i, j, k;\r
+ int iPageCount;\r
+ tBinary *ret;\r
+ tPE_DOS_HEADER dosHdr;\r
+ tPE_IMAGE_HEADERS peHeaders;\r
+ tPE_SECTION_HEADER *peSections;\r
+ char namebuf[9] = {0};\r
+ Uint iFlags, iVA;\r
+ \r
+ ENTER("xFP", FP);\r
+ \r
+ // Read DOS header and check\r
+ VFS_Read(FP, sizeof(tPE_DOS_HEADER), &dosHdr);\r
+ if( dosHdr.Ident != ('M'|('Z'<<8)) ) {\r
+ LEAVE('n');\r
+ return NULL;\r
+ }\r
+ \r
+ // - Read PE Header\r
+ VFS_Seek(FP, dosHdr.PeHdrOffs, SEEK_SET);\r
+ if( VFS_Tell(FP) != dosHdr.PeHdrOffs ) {\r
+ ret = MZ_Open(FP);\r
+ LEAVE('p', ret);\r
+ return ret;\r
+ }\r
+ VFS_Read(FP, sizeof(tPE_IMAGE_HEADERS), &peHeaders);\r
+ \r
+ // - Check PE Signature and pass on to the MZ Loader if invalid\r
+ if( peHeaders.Signature != (('P')|('E'<<8)) ) {\r
+ ret = MZ_Open(FP);\r
+ LEAVE('p', ret);\r
+ return ret;\r
+ }\r
+ \r
+ // Read Sections (Uses `count` as a temp variable)\r
+ count = sizeof(tPE_SECTION_HEADER) * peHeaders.FileHeader.SectionCount;\r
+ peSections = malloc( count );\r
+ if(!peSections)\r
+ {\r
+ Warning("PE_Load - Unable to allocate `peSections`, 0x%x bytes", count);\r
+ LEAVE('n');\r
+ return NULL;\r
+ }\r
+ VFS_Read(FP, count, peSections);\r
+ \r
+ // Count Pages\r
+ iPageCount = 1; // 1st page is headers\r
+ for( i = 0; i < peHeaders.FileHeader.SectionCount; i++ )\r
+ {\r
+ // Check if the section is loadable\r
+ // (VA is zero in non-loadable sections)\r
+ if(peSections[i].RVA + peHeaders.OptHeader.ImageBase == 0) continue;\r
+ \r
+ // Moar pages\r
+ iPageCount += (peSections[i].VirtualSize + 0xFFF) >> 12;\r
+ }\r
+ \r
+ LOG("%i Pages", iPageCount);\r
+ \r
+ // Initialise Executable Information\r
+ ret = malloc(sizeof(tBinary) + sizeof(tBinaryPage)*iPageCount);\r
+ \r
+ ret->Entry = peHeaders.OptHeader.EntryPoint + peHeaders.OptHeader.ImageBase;\r
+ ret->Base = peHeaders.OptHeader.ImageBase;\r
+ ret->Interpreter = gsPE_DefaultInterpreter;\r
+ ret->NumPages = iPageCount;\r
+ \r
+ LOG("Entry=%p, BaseAddress=0x%x\n", ret->Entry, ret->Base);\r
+ \r
+ ret->Pages[0].Virtual = peHeaders.OptHeader.ImageBase;\r
+ ret->Pages[0].Physical = 0;\r
+ ret->Pages[0].Size = 4096;\r
+ ret->Pages[0].Flags = 0;\r
+ \r
+ // Parse Sections\r
+ j = 1; // Page Index\r
+ for( i = 0; i < peHeaders.FileHeader.SectionCount; i++ )\r
+ {\r
+ iVA = peSections[i].RVA + peHeaders.OptHeader.ImageBase;\r
+ \r
+ // Skip non-loadable sections\r
+ if(iVA == 0) continue;\r
+ \r
+ // Create Name Buffer\r
+ memcpy(namebuf, peSections[i].Name, 8);\r
+ LOG("Section %i '%s', iVA = %p", i, namebuf, iVA);\r
+ \r
+ // Create Flags\r
+ iFlags = 0;\r
+ if(peSections[i].Flags & PE_SECTION_FLAG_MEM_EXECUTE)\r
+ iFlags |= BIN_PAGEFLAG_EXEC;\r
+ if( !(peSections[i].Flags & PE_SECTION_FLAG_MEM_WRITE) )\r
+ iFlags |= BIN_PAGEFLAG_RO;\r
+ \r
+ // Create Page Listing\r
+ count = (peSections[i].RawSize + 0xFFF) >> 12;\r
+ for(k=0;k<count;k++)\r
+ {\r
+ ret->Pages[j+k].Virtual = iVA + (k<<12);\r
+ ret->Pages[j+k].Physical = peSections[i].RawOffs + (k<<12); // Store the offset in the physical address\r
+ if(k == count-1 && (peSections[i].RawSize & 0xFFF))\r
+ ret->Pages[j+k].Size = peSections[i].RawSize & 0xFFF; // Byte count in page\r
+ else\r
+ ret->Pages[j+k].Size = 4096;\r
+ ret->Pages[j+k].Flags = iFlags;\r
+ }\r
+ count = (peSections[i].VirtualSize + 0xFFF) >> 12;\r
+ for(;k<count;k++)\r
+ {\r
+ ret->Pages[j+k].Virtual = iVA + (k<<12);\r
+ ret->Pages[j+k].Physical = -1; // -1 = Fill with zeros\r
+ if(k == count-1 && (peSections[i].VirtualSize & 0xFFF))\r
+ ret->Pages[j+k].Size = peSections[i].VirtualSize & 0xFFF; // Byte count in page\r
+ else\r
+ ret->Pages[j+k].Size = 4096;\r
+ ret->Pages[j+k].Flags = iFlags;\r
+ }\r
+ j += count;\r
+ \r
+ LOG("%i Name:'%s', RVA: 0x%x, Size: 0x%x (0x%x), Ofs: 0x%x, Flags: 0x%x",\r
+ i, namebuf, \r
+ iVA,\r
+ peSections[i].VirtualSize, peSections[i].RawSize, peSections[i].RawOffs,\r
+ peSections[i].Flags\r
+ );\r
+ \r
+ }\r
+ // Free Executable Memory\r
+ free(peSections);\r
+ \r
+ LEAVE('p', ret);\r
+ return ret;\r
+}\r
+\r
+/**\r
+ */\r
+tBinary *MZ_Open(int FP)\r
+{\r
+ ENTER("xFP", FP);\r
+ UNIMPLEMENTED();\r
+ LEAVE('n');\r
+ return NULL;\r
+}\r
+\r
+int PE_Relocate(void *Base)\r
+{\r
+ tPE_DOS_HEADER *dosHdr;\r
+ tPE_IMAGE_HEADERS *peHeaders;\r
+ tPE_SECTION_HEADER *peSections;\r
+ tPE_DATA_DIR *directory;\r
+ tPE_IMPORT_DIR *impDir;\r
+ int i;\r
+ Uint iBase = (Uint)Base;\r
+ #if 0\r
+ void *hLibrary;\r
+ char *libPath;\r
+ #endif\r
+ \r
+ ENTER("pBase", Base);\r
+ dosHdr = Base;\r
+ peHeaders = (void*)( iBase + dosHdr->PeHdrOffs );\r
+ peSections = (void*)( iBase + sizeof(tPE_IMAGE_HEADERS) );\r
+ \r
+ directory = (void*)(peSections[0].RVA + iBase);\r
+ \r
+ // === Load Import Tables\r
+ impDir = (void*)( directory[PE_DIR_IMPORT].RVA + iBase );\r
+ for( i = 0; impDir[i].DLLName != NULL; i++ )\r
+ {\r
+ impDir[i].DLLName += iBase;\r
+ impDir[i].ImportLookupTable += iBase/4;\r
+ impDir[i].ImportAddressTable += iBase/4;\r
+ LOG("DLL Required '%s'(0x%x)", impDir[i].DLLName, impDir[i].DLLName);\r
+ #if 0\r
+ libPath = FindLibrary(impDir[i].DLLName);\r
+ if(libPath == NULL)\r
+ {\r
+ Warning("PE_Relocate - Unable to find library '%s'");\r
+ LEAVE('i', -1);\r
+ return -1;\r
+ }\r
+ LOG("DLL Path = '%s'", libPath);\r
+ hLibrary = DynLib_Load(libPath, 0);\r
+ #endif\r
+ }\r
+ \r
+ for(i=0;i<PE_DIR_LAST;i++)\r
+ LOG("directory[%i] = {RVA=0x%x,Size=0x%x}", i, directory[i].RVA, directory[i].Size);\r
+ \r
+ LEAVE('i', 0);\r
+ return 0;\r
+}\r
+\r
+int PE_GetSymbol(void *Base, char *Name, Uint *Ret)\r
+{\r
+ return 0;\r
+}\r