Initial commit

2021-10-10 14:39:17 +08:00
commit d25da95e1e
135 changed files with 19184 additions and 0 deletions
--- a/execformat/pe/test_headers.c
+++ b/execformat/pe/test_headers.c
@@ -0,0 +1,118 @@
+
+#include "format.h"
+#include "struct.h"
+#include "reloc.h"
+
+#include "stdio.h"
+#include "sys/types.h"
+#include "sys/stat.h"
+#include "fcntl.h"
+#include "sys/mman.h"
+
+#include <assert.h>
+#include <string.h>
+#include <malloc.h>
+
+const char *caseMachine(uint16_t type) {
+#define CASE(expr)             \
+	case EXECFORMAT_PE_##expr: \
+		return #expr;
+
+	switch (type) {
+		CASE(MACHINE_UNKNOWN)
+		CASE(MACHINE_AMD64)
+		CASE(MACHINE_I386)
+		CASE(MACHINE_EBC)
+		default:
+			return "(Unknown)";
+	}
+
+#undef CASE
+}
+
+void caseFlags(uint32_t flags) {
+#define CASE(expr)                           \
+	if (flags & EXECFORMAT_PE_FLAG_##expr) { \
+		printf("|" #expr);                   \
+	}
+	CASE(RELOCS_STRIPPED)
+	CASE(EXECUTABLE_IMAGE)
+	CASE(LINE_NUMS_STRIPPED)
+	CASE(LOCAL_SYMS_STRIPPED)
+	CASE(AGGRESSIVE_WS_TRIM)
+	CASE(LARGE_ADDRESS_AWARE)
+	CASE(RESERVED_0x0040)
+	CASE(BYTES_REVERSED_LO)
+	CASE(32BIT_MACHINE)
+	CASE(DEBUG_STRIPPED)
+	CASE(REMOVABLE_RUN_FROM_SWAP)
+	CASE(NET_RUN_FROM_SWAP)
+	CASE(SYSTEM)
+	CASE(DLL)
+	CASE(UP_SYSTEM_ONLY)
+	CASE(BYTES_REVERSED_HI)
+	printf("|\n");
+#undef CASE
+}
+void caseSectionFlags(uint32_t flags) {
+#define CASE(expr)                                  \
+	if (flags & EXECFORMAT_PE_SECTIONFLAG_##expr) { \
+		printf("|" #expr);                          \
+	}
+	CASE(NO_PAD)
+	CASE(CNT_CODE)
+	CASE(CNT_INITIALIZED_DATA)
+	CASE(CNT_UNINITIALIZED_DATA)
+	CASE(LNK_INFO)
+	CASE(LNK_REMOVE)
+	CASE(LNK_COMDAT)
+	CASE(GPREL)
+	CASE(LNK_NRELOC_OVFL)
+	CASE(MEM_DISCARDABLE)
+	CASE(MEM_NOT_CACHED)
+	CASE(MEM_NOT_PAGED)
+	CASE(MEM_SHARED)
+	CASE(MEM_EXECUTE)
+	CASE(MEM_READ)
+	CASE(MEM_WRITE)
+	printf("|\n");
+#undef CASE
+}
+
+
+int main(int argc, char *argv[]) {
+	execformat_pe_OptionalHeader_CheckPacking();
+
+	int f = open("../../Main.efi", O_RDONLY);
+	assert(f != -1 && "File open failed");
+
+	struct stat st;
+	fstat(f, &st);
+	void *maddr = mmap(0, st.st_size, PROT_READ, MAP_SHARED, f, 0);
+	void *addr  = malloc(st.st_size);
+	memcpy(addr, maddr, st.st_size);
+
+	execformat_pe_PortableExecutable pe;
+	execformat_pe_LoadMemory(&pe, addr, st.st_size);
+
+	printf("%s Executable, Machine Type=0x%x (%s), Flags=", pe.isPE32P ? "PE32+" : "PE32", pe.header->machineType, caseMachine(pe.header->machineType));
+	caseFlags(pe.header->flags);
+	printf("\nImageBase=%d (0x%x)\n", ((execformat_pe_OptionalHeader_PE32P *)pe.optional)->win.imageBase, ((execformat_pe_OptionalHeader_PE32P *)pe.optional)->win.imageBase);
+
+	printf("numRVAandSizes=%d (want %d)\n\n", pe.numDataDir, EXECFORMAT_PE_DATADIR_INDEX_CLR_RUNTIME_HEADER + 2);
+
+	printf("numSections=%d\n", pe.numSections);
+	char name[9];
+	for (int i = 0; i < pe.numSections; i++) {
+		name[8] = 0;
+		strncpy(name, pe.sections[i].name, 8);
+
+		printf("   %8s Rva0x%08x File0x%08x Len0x%08x ", name, pe.sections[i].virtualAddr, pe.sections[i].pointerToRawData, pe.sections[i].virtualSize);
+		caseSectionFlags(pe.sections[i].flags);
+	}
+
+
+	execformat_pe_BaseRelocate(&pe, 0, 0xFFFFFFFFC0000000ull);
+
+	return 0;
+}