2 files changed, 248 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();
+}
diff --git a/src/cmd/ld/pe.h b/src/cmd/ld/pe.h
new file mode 100644
index 000000000..e7e2f9f08
--- /dev/null
+++ b/src/cmd/ld/pe.h
@@ -0,0 +1,91 @@
+// 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.
+
+typedef struct {
+	uint16 Machine;
+	uint16 NumberOfSections;
+	uint32 TimeDateStamp;
+	uint32 PointerToSymbolTable;
+	uint32 NumberOfSymbols;
+	uint16 SizeOfOptionalHeader;
+	uint16 Characteristics;
+} IMAGE_FILE_HEADER;
+
+typedef struct {
+	uint32 VirtualAddress;
+	uint32 Size;
+} IMAGE_DATA_DIRECTORY;
+
+typedef struct {
+	uint16 Magic;
+	uint8  MajorLinkerVersion;
+	uint8  MinorLinkerVersion;
+	uint32 SizeOfCode;
+	uint32 SizeOfInitializedData;
+	uint32 SizeOfUninitializedData;
+	uint32 AddressOfEntryPoint;
+	uint32 BaseOfCode;
+	uint32 BaseOfData;
+	uint32 ImageBase;
+	uint32 SectionAlignment;
+	uint32 FileAlignment;
+	uint16 MajorOperatingSystemVersion;
+	uint16 MinorOperatingSystemVersion;
+	uint16 MajorImageVersion;
+	uint16 MinorImageVersion;
+	uint16 MajorSubsystemVersion;
+	uint16 MinorSubsystemVersion;
+	uint32 Win32VersionValue;
+	uint32 SizeOfImage;
+	uint32 SizeOfHeaders;
+	uint32 CheckSum;
+	uint16 Subsystem;
+	uint16 DllCharacteristics;
+	uint32 SizeOfStackReserve;
+	uint32 SizeOfStackCommit;
+	uint32 SizeOfHeapReserve;
+	uint32 SizeOfHeapCommit;
+	uint32 LoaderFlags;
+	uint32 NumberOfRvaAndSizes;
+	IMAGE_DATA_DIRECTORY DataDirectory[16];
+} IMAGE_OPTIONAL_HEADER;
+
+typedef struct {
+	uint8  Name[8];
+	uint32 VirtualSize;
+	uint32 VirtualAddress;
+	uint32 SizeOfRawData;
+	uint32 PointerToRawData;
+	uint32 PointerToRelocations;
+	uint32 PointerToLineNumbers;
+	uint16 NumberOfRelocations;
+	uint16 NumberOfLineNumbers;
+	uint32 Characteristics;
+} IMAGE_SECTION_HEADER;
+
+#define PERESERVE	0x400
+#define PEALIGN		0x200
+#define PEBASE		0x00400000
+
+enum {
+	IMAGE_FILE_MACHINE_I386 = 0x14c,
+	IMAGE_FILE_MACHINE_AMD64 = 0x8664,
+
+	IMAGE_FILE_RELOCS_STRIPPED = 0x0001,
+	IMAGE_FILE_EXECUTABLE_IMAGE = 0x0002,
+	IMAGE_FILE_LARGE_ADDRESS_AWARE = 0x0020,
+	IMAGE_FILE_32BIT_MACHINE = 0x0100,
+	IMAGE_FILE_DEBUG_STRIPPED = 0x0200,
+
+	IMAGE_SCN_CNT_CODE = 0x00000020,
+	IMAGE_SCN_CNT_INITIALIZED_DATA = 0x00000040,
+	IMAGE_SCN_CNT_UNINITIALIZED_DATA = 0x00000080,
+	IMAGE_SCN_MEM_EXECUTE = 0x20000000,
+	IMAGE_SCN_MEM_READ = 0x40000000,
+	IMAGE_SCN_MEM_WRITE = 0x80000000,
+};
+
+void peinit(void);
+void dope(void);
+void asmbpe(void);