diff options
Diffstat (limited to 'src/cmd/ld/pe.c')
| -rw-r--r-- | src/cmd/ld/pe.c | 157 |
1 files changed, 157 insertions, 0 deletions
diff --git a/src/cmd/ld/pe.c b/src/cmd/ld/pe.c new file mode 100644 index 000000000..c225d0cd4 --- /dev/null +++ b/src/cmd/ld/pe.c @@ -0,0 +1,157 @@ +// Copyright 2009 The Go Authors. All rights reserved. +// Use of this source code is governed by a BSD-style +// license that can be found in the LICENSE file. + +// PE (Portable Executable) file writing +// http://www.microsoft.com/whdc/system/platform/firmware/PECOFF.mspx + +#include <time.h> + +#include "l.h" +#include "../ld/lib.h" +#include "../ld/pe.h" + +static int pe64; +static int nsect; +static int sect_virt_begin; +static int sect_raw_begin = PERESERVE; + +static IMAGE_FILE_HEADER fh; +static IMAGE_OPTIONAL_HEADER oh; +static IMAGE_SECTION_HEADER sh[16]; +static IMAGE_SECTION_HEADER *textsect, *datsect, *bsssect; + +static IMAGE_SECTION_HEADER* +new_section(char *name, int size, int noraw) +{ + IMAGE_SECTION_HEADER *h; + + if(nsect == 16) { + diag("too many sections"); + errorexit(); + } + h = &sh[nsect++]; + strncpy(h->Name, name, sizeof(h->Name)); + h->VirtualSize = size; + if(!sect_virt_begin) + sect_virt_begin = 0x1000; + h->VirtualAddress = sect_virt_begin; + sect_virt_begin = rnd(sect_virt_begin+size, 0x1000); + if(!noraw) { + h->SizeOfRawData = rnd(size, PEALIGN); + h->PointerToRawData = sect_raw_begin; + sect_raw_begin += h->SizeOfRawData; + } + return h; +} + +void +peinit(void) +{ + switch(thechar) { + // 64-bit architectures + case '6': + pe64 = 1; + break; + // 32-bit architectures + default: + break; + } +} + +static void +pewrite(void) +{ + int i, j; + + strnput("MZ", 0x3c); + LPUT(0x40); // file offset to PE header + strnput("PE", 4); + + for (i=0; i<sizeof(fh); i++) + cput(((char*)&fh)[i]); + for (i=0; i<sizeof(oh); i++) + cput(((char*)&oh)[i]); + for (i=0; i<nsect; i++) + for (j=0; j<sizeof(sh[i]); j++) + cput(((char*)&sh[i])[j]); +} + +void +dope(void) +{ + textsect = new_section(".text", textsize, 0); + textsect->Characteristics = IMAGE_SCN_CNT_CODE| + IMAGE_SCN_CNT_INITIALIZED_DATA| + IMAGE_SCN_MEM_EXECUTE|IMAGE_SCN_MEM_READ; + + datsect = new_section(".data", datsize, 0); + datsect->Characteristics = IMAGE_SCN_CNT_INITIALIZED_DATA| + IMAGE_SCN_MEM_READ|IMAGE_SCN_MEM_WRITE; + INITDAT = PEBASE+datsect->VirtualAddress; + + bsssect = new_section(".bss", bsssize, 1); + bsssect->Characteristics = IMAGE_SCN_CNT_UNINITIALIZED_DATA| + IMAGE_SCN_MEM_READ|IMAGE_SCN_MEM_WRITE; +} + +void +asmbpe(void) +{ + switch(thechar) { + default: + diag("unknown PE architecture"); + errorexit(); + case '6': + fh.Machine = IMAGE_FILE_MACHINE_AMD64; + break; + case '8': + fh.Machine = IMAGE_FILE_MACHINE_I386; + break; + } + + if(!debug['s']) { + IMAGE_SECTION_HEADER *symsect; + symsect = new_section(".symdat", 8+symsize+lcsize, 0); + symsect->Characteristics = IMAGE_SCN_MEM_READ| + IMAGE_SCN_CNT_INITIALIZED_DATA; + } + + fh.NumberOfSections = nsect; + fh.TimeDateStamp = time(0); + fh.SizeOfOptionalHeader = sizeof(oh); + fh.Characteristics = IMAGE_FILE_RELOCS_STRIPPED| + IMAGE_FILE_EXECUTABLE_IMAGE|IMAGE_FILE_DEBUG_STRIPPED; + if(thechar == '8') + fh.Characteristics |= IMAGE_FILE_32BIT_MACHINE; + + oh.Magic = 0x10b; // PE32 + oh.MajorLinkerVersion = 1; + oh.MinorLinkerVersion = 0; + oh.SizeOfCode = textsect->SizeOfRawData; + oh.SizeOfInitializedData = datsect->SizeOfRawData; + oh.SizeOfUninitializedData = bsssect->SizeOfRawData; + oh.AddressOfEntryPoint = entryvalue()-PEBASE; + oh.BaseOfCode = textsect->VirtualAddress; + oh.BaseOfData = datsect->VirtualAddress; + + oh.ImageBase = PEBASE; + oh.SectionAlignment = 0x00001000; + oh.FileAlignment = PEALIGN; + oh.MajorOperatingSystemVersion = 4; + oh.MinorOperatingSystemVersion = 0; + oh.MajorImageVersion = 1; + oh.MinorImageVersion = 0; + oh.MajorSubsystemVersion = 4; + oh.MinorSubsystemVersion = 0; + oh.SizeOfImage = sect_virt_begin; + oh.SizeOfHeaders = PERESERVE; + oh.Subsystem = 3; // WINDOWS_CUI + oh.SizeOfStackReserve = 0x00200000; + oh.SizeOfStackCommit = 0x00001000; + oh.SizeOfHeapReserve = 0x00100000; + oh.SizeOfHeapCommit = 0x00001000; + oh.NumberOfRvaAndSizes = 16; + + pewrite(); +} |