WebSVN – Kolibri OS – Diff – /programs/develop/ktcc/trunk/source/tccpe.c

 /*
- *  TCCPE.C - PE file output for the TinyC Compiler
+ *  TCCPE.C - PE file output for the Tiny C Compiler
  *
- *  Copyright (c) 2005 grischka
+ *  Copyright (c) 2005-2007 grischka
+ *
  * This library is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Lesser General Public
  * License as published by the Free Software Foundation; either
  * version 2 of the License, or (at your option) any later version.
  * You should have received a copy of the GNU Lesser General Public
  * License along with this library; if not, write to the Free Software
  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  */
-typedef unsigned char BYTE;
-typedef unsigned short WORD;
-typedef unsigned long DWORD;
+#include "tcc.h"
-#define ST static
+#ifndef _WIN32
+#define stricmp strcasecmp
+#define strnicmp strncasecmp
+#endif
+#ifndef MAX_PATH
+#define MAX_PATH 260
+#endif
+#define PE_MERGE_DATA
+/* #define PE_PRINT_SECTIONS */
+#ifdef TCC_TARGET_X86_64
+# define ADDR3264 ULONGLONG
+# define REL_TYPE_DIRECT R_X86_64_64
+# define R_XXX_THUNKFIX R_X86_64_PC32
+# define R_XXX_RELATIVE R_X86_64_RELATIVE
+# define IMAGE_FILE_MACHINE 0x8664
+# define RSRC_RELTYPE 3
+#elif defined TCC_TARGET_ARM
+# define ADDR3264 DWORD
+# define REL_TYPE_DIRECT R_ARM_ABS32
+# define R_XXX_THUNKFIX R_ARM_ABS32
+# define R_XXX_RELATIVE R_ARM_RELATIVE
+# define IMAGE_FILE_MACHINE 0x01C0
+# define RSRC_RELTYPE 7 /* ??? (not tested) */
+#elif defined TCC_TARGET_I386
+# define ADDR3264 DWORD
+# define REL_TYPE_DIRECT R_386_32
-/* XXX: move that to TCC ? */
+# define R_XXX_THUNKFIX R_386_32
-int verbose = 0;
+# define R_XXX_RELATIVE R_386_RELATIVE
+# define IMAGE_FILE_MACHINE 0x014C
+# define RSRC_RELTYPE 7 /* DIR32NB */
-/* definitions below are from winnt.h */
+#endif
-typedef struct _IMAGE_DOS_HEADER {	/* DOS .EXE header */
+#ifdef _WIN32
-    WORD e_magic;		/* Magic number */
-    WORD e_cblp;		/* Bytes on last page of file */
-    WORD e_cp;			/* Pages in file */
-    WORD e_crlc;		/* Relocations */
-    WORD e_cparhdr;		/* Size of header in paragraphs */
+void dbg_printf (const char *fmt, ...)
-    WORD e_minalloc;		/* Minimum extra paragraphs needed */
+{
-    WORD e_maxalloc;		/* Maximum extra paragraphs needed */
-    WORD e_ss;			/* Initial (relative) SS value */
-    WORD e_sp;			/* Initial SP value */
-    WORD e_csum;		/* Checksum */
+    char buffer[4000];
-    WORD e_ip;			/* Initial IP value */
+    va_list arg;
-    WORD e_cs;			/* Initial (relative) CS value */
-    WORD e_lfarlc;		/* File address of relocation table */
-    WORD e_ovno;		/* Overlay number */
+    int x;
-    WORD e_res[4];		/* Reserved words */
+    va_start(arg, fmt);
-    WORD e_oemid;		/* OEM identifier (for e_oeminfo) */
+    x = vsprintf (buffer, fmt, arg);
+    strcpy(buffer+x, "\n");
+    OutputDebugString(buffer);
+}
+#endif
+/* ----------------------------------------------------------- */
+#ifndef IMAGE_NT_SIGNATURE
+/* ----------------------------------------------------------- */
+/* definitions below are from winnt.h */
+typedef unsigned char BYTE;
+typedef unsigned short WORD;
+typedef unsigned int DWORD;
+typedef unsigned long long ULONGLONG;
+#pragma pack(push, 1)
+typedef struct _IMAGE_DOS_HEADER {  /* DOS .EXE header */
+    WORD e_magic;         /* Magic number */
+    WORD e_cblp;          /* Bytes on last page of file */
+    WORD e_cp;            /* Pages in file */
+    WORD e_crlc;          /* Relocations */
+    WORD e_cparhdr;       /* Size of header in paragraphs */
+    WORD e_minalloc;      /* Minimum extra paragraphs needed */
+    WORD e_maxalloc;      /* Maximum extra paragraphs needed */
+    WORD e_ss;            /* Initial (relative) SS value */
+    WORD e_sp;            /* Initial SP value */
+    WORD e_csum;          /* Checksum */
+    WORD e_ip;            /* Initial IP value */
+    WORD e_cs;            /* Initial (relative) CS value */
+    WORD e_lfarlc;        /* File address of relocation table */
+    WORD e_ovno;          /* Overlay number */
-    WORD e_oeminfo;		/* OEM information; e_oemid specific */
+    WORD e_res[4];        /* Reserved words */
-    WORD e_res2[10];		/* Reserved words */
+    WORD e_oemid;         /* OEM identifier (for e_oeminfo) */
-    DWORD e_lfanew;		/* File address of new exe header */
+    WORD e_oeminfo;       /* OEM information; e_oemid specific */
-    BYTE e_code[0x40];
+    WORD e_res2[10];      /* Reserved words */
+    DWORD e_lfanew;        /* File address of new exe header */
 } IMAGE_DOS_HEADER, *PIMAGE_DOS_HEADER;
-#define IMAGE_NT_SIGNATURE  0x00004550	/* PE00 */
+#define IMAGE_NT_SIGNATURE  0x00004550  /* PE00 */
 #define SIZE_OF_NT_SIGNATURE 4
 typedef struct _IMAGE_FILE_HEADER {
-    WORD Machine;
+    WORD    Machine;
-    WORD NumberOfSections;
+    WORD    NumberOfSections;
-    DWORD TimeDateStamp;
+    DWORD   TimeDateStamp;
-    DWORD PointerToSymbolTable;
+    DWORD   PointerToSymbolTable;
-    DWORD NumberOfSymbols;
+    DWORD   NumberOfSymbols;
-    WORD SizeOfOptionalHeader;
+    WORD    SizeOfOptionalHeader;
-    WORD Characteristics;
+    WORD    Characteristics;
 } IMAGE_FILE_HEADER, *PIMAGE_FILE_HEADER;
 #define IMAGE_SIZEOF_FILE_HEADER 20
 typedef struct _IMAGE_DATA_DIRECTORY {
-    DWORD VirtualAddress;
+    DWORD   VirtualAddress;
-    DWORD Size;
+    DWORD   Size;
 } IMAGE_DATA_DIRECTORY, *PIMAGE_DATA_DIRECTORY;
+typedef struct _IMAGE_OPTIONAL_HEADER {
-typedef struct _IMAGE_OPTIONAL_HEADER {
+    /* Standard fields. */
-    /* Standard fields. */
+    WORD    Magic;
-    WORD Magic;
+    BYTE    MajorLinkerVersion;
-    BYTE MajorLinkerVersion;
+    BYTE    MinorLinkerVersion;
-    BYTE MinorLinkerVersion;
+    DWORD   SizeOfCode;
-    DWORD SizeOfCode;
+    DWORD   SizeOfInitializedData;
-    DWORD SizeOfInitializedData;
+    DWORD   SizeOfUninitializedData;
-    DWORD SizeOfUninitializedData;
+    DWORD   AddressOfEntryPoint;
-    DWORD AddressOfEntryPoint;
+    DWORD   BaseOfCode;
-    DWORD BaseOfCode;
+#ifndef TCC_TARGET_X86_64
-    DWORD BaseOfData;
+    DWORD   BaseOfData;
+#endif
     /* NT additional fields. */
-    DWORD ImageBase;
+    ADDR3264 ImageBase;
-    DWORD SectionAlignment;
+    DWORD   SectionAlignment;
-    DWORD FileAlignment;
+    DWORD   FileAlignment;
-    WORD MajorOperatingSystemVersion;
+    WORD    MajorOperatingSystemVersion;
-    WORD MinorOperatingSystemVersion;
+    WORD    MinorOperatingSystemVersion;
-    WORD MajorImageVersion;
+    WORD    MajorImageVersion;
-    WORD MinorImageVersion;
+    WORD    MinorImageVersion;
-    WORD MajorSubsystemVersion;
+    WORD    MajorSubsystemVersion;
-    WORD MinorSubsystemVersion;
+    WORD    MinorSubsystemVersion;
-    DWORD Win32VersionValue;
+    DWORD   Win32VersionValue;
+    DWORD   SizeOfImage;
-    DWORD SizeOfImage;
-    DWORD SizeOfHeaders;
-    DWORD CheckSum;
-    WORD Subsystem;
+    DWORD   SizeOfHeaders;
-    WORD DllCharacteristics;
+    DWORD   CheckSum;
-    DWORD SizeOfStackReserve;
+    WORD    Subsystem;
-    DWORD SizeOfStackCommit;
+    WORD    DllCharacteristics;
-    DWORD SizeOfHeapReserve;
+    ADDR3264 SizeOfStackReserve;
-    DWORD SizeOfHeapCommit;
+    ADDR3264 SizeOfStackCommit;
-    DWORD LoaderFlags;
+    ADDR3264 SizeOfHeapReserve;
-    DWORD NumberOfRvaAndSizes;
+    ADDR3264 SizeOfHeapCommit;
-    IMAGE_DATA_DIRECTORY DataDirectory[16];
+    DWORD   LoaderFlags;
+    DWORD   NumberOfRvaAndSizes;
-} IMAGE_OPTIONAL_HEADER32, *PIMAGE_OPTIONAL_HEADER32;
+    IMAGE_DATA_DIRECTORY DataDirectory[16];
+} IMAGE_OPTIONAL_HEADER32, IMAGE_OPTIONAL_HEADER64, IMAGE_OPTIONAL_HEADER;
-#define IMAGE_DIRECTORY_ENTRY_EXPORT          0	/* Export Directory */
+#define IMAGE_DIRECTORY_ENTRY_EXPORT          0   /* Export Directory */
-#define IMAGE_DIRECTORY_ENTRY_IMPORT          1	/* Import Directory */
+#define IMAGE_DIRECTORY_ENTRY_IMPORT          1   /* Import Directory */
-#define IMAGE_DIRECTORY_ENTRY_RESOURCE        2	/* Resource Directory */
+#define IMAGE_DIRECTORY_ENTRY_RESOURCE        2   /* Resource Directory */
-#define IMAGE_DIRECTORY_ENTRY_EXCEPTION       3	/* Exception Directory */
+#define IMAGE_DIRECTORY_ENTRY_EXCEPTION       3   /* Exception Directory */
-#define IMAGE_DIRECTORY_ENTRY_SECURITY        4	/* Security Directory */
+#define IMAGE_DIRECTORY_ENTRY_SECURITY        4   /* Security Directory */
-#define IMAGE_DIRECTORY_ENTRY_BASERELOC       5	/* Base Relocation Table */
+#define IMAGE_DIRECTORY_ENTRY_BASERELOC       5   /* Base Relocation Table */
-#define IMAGE_DIRECTORY_ENTRY_DEBUG           6	/* Debug Directory */
+#define IMAGE_DIRECTORY_ENTRY_DEBUG           6   /* Debug Directory */
 /*      IMAGE_DIRECTORY_ENTRY_COPYRIGHT       7      (X86 usage) */
-#define IMAGE_DIRECTORY_ENTRY_ARCHITECTURE    7	/* Architecture Specific Data */
+#define IMAGE_DIRECTORY_ENTRY_ARCHITECTURE    7   /* Architecture Specific Data */
-#define IMAGE_DIRECTORY_ENTRY_GLOBALPTR       8	/* RVA of GP */
+#define IMAGE_DIRECTORY_ENTRY_GLOBALPTR       8   /* RVA of GP */
-#define IMAGE_DIRECTORY_ENTRY_TLS             9	/* TLS Directory */
+#define IMAGE_DIRECTORY_ENTRY_TLS             9   /* TLS Directory */
-#define IMAGE_DIRECTORY_ENTRY_LOAD_CONFIG    10	/* Load Configuration Directory */
+#define IMAGE_DIRECTORY_ENTRY_LOAD_CONFIG    10   /* Load Configuration Directory */
-#define IMAGE_DIRECTORY_ENTRY_BOUND_IMPORT   11	/* Bound Import Directory in headers */
+#define IMAGE_DIRECTORY_ENTRY_BOUND_IMPORT   11   /* Bound Import Directory in headers */
-#define IMAGE_DIRECTORY_ENTRY_IAT            12	/* Import Address Table */
+#define IMAGE_DIRECTORY_ENTRY_IAT            12   /* Import Address Table */
-#define IMAGE_DIRECTORY_ENTRY_DELAY_IMPORT   13	/* Delay Load Import Descriptors */
+#define IMAGE_DIRECTORY_ENTRY_DELAY_IMPORT   13   /* Delay Load Import Descriptors */
-#define IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR 14	/* COM Runtime descriptor */
+#define IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR 14   /* COM Runtime descriptor */
 /* Section header format. */
-#define IMAGE_SIZEOF_SHORT_NAME              8
+#define IMAGE_SIZEOF_SHORT_NAME         8
 typedef struct _IMAGE_SECTION_HEADER {
+    BYTE    Name[IMAGE_SIZEOF_SHORT_NAME];
+    union {
+            DWORD   PhysicalAddress;
+            DWORD   VirtualSize;
+    } Misc;
+    DWORD   VirtualAddress;
+    DWORD   SizeOfRawData;
+    DWORD   PointerToRawData;
+    DWORD   PointerToRelocations;
+    DWORD   PointerToLinenumbers;
+    WORD    NumberOfRelocations;
+    WORD    NumberOfLinenumbers;
+    DWORD   Characteristics;
+} IMAGE_SECTION_HEADER, *PIMAGE_SECTION_HEADER;
+#define IMAGE_SIZEOF_SECTION_HEADER     40
+typedef struct _IMAGE_EXPORT_DIRECTORY {
+    DWORD Characteristics;
+    DWORD TimeDateStamp;
+    WORD MajorVersion;
+    WORD MinorVersion;
+    DWORD Name;
+    DWORD Base;
+    DWORD NumberOfFunctions;
-    BYTE Name[IMAGE_SIZEOF_SHORT_NAME];
-    union {
+    DWORD NumberOfNames;
-	DWORD PhysicalAddress;
+    DWORD AddressOfFunctions;
-	DWORD VirtualSize;
+    DWORD AddressOfNames;
-    } Misc;
+    DWORD AddressOfNameOrdinals;
-    DWORD VirtualAddress;
+} IMAGE_EXPORT_DIRECTORY,*PIMAGE_EXPORT_DIRECTORY;
-    DWORD SizeOfRawData;
-    DWORD PointerToRawData;
+typedef struct _IMAGE_IMPORT_DESCRIPTOR {
-    DWORD PointerToRelocations;
+    union {
-    DWORD PointerToLinenumbers;
+        DWORD Characteristics;
-    WORD NumberOfRelocations;
+        DWORD OriginalFirstThunk;
-    WORD NumberOfLinenumbers;
+    };
-    DWORD Characteristics;
+    DWORD TimeDateStamp;
-} IMAGE_SECTION_HEADER, *PIMAGE_SECTION_HEADER;
+    DWORD ForwarderChain;
+    DWORD Name;
-#define IMAGE_SIZEOF_SECTION_HEADER          40
+    DWORD FirstThunk;
 } IMAGE_IMPORT_DESCRIPTOR;
-/* ----------------------------------------------------------- */
-typedef struct _IMAGE_BASE_RELOCATION {
-    DWORD VirtualAddress;
-    DWORD SizeOfBlock;
-//  WORD    TypeOffset[1];
-} IMAGE_BASE_RELOCATION;
-#define IMAGE_SIZEOF_BASE_RELOCATION         8
-#define IMAGE_REL_BASED_ABSOLUTE              0
-#define IMAGE_REL_BASED_HIGH                  1
-#define IMAGE_REL_BASED_LOW                   2
-#define IMAGE_REL_BASED_HIGHLOW               3
-#define IMAGE_REL_BASED_HIGHADJ               4
-#define IMAGE_REL_BASED_MIPS_JMPADDR          5
-#define IMAGE_REL_BASED_SECTION               6
-#define IMAGE_REL_BASED_REL32                 7
-/* ----------------------------------------------------------- */
-/* ----------------------------------------------------------- */
-IMAGE_DOS_HEADER pe_dos_hdr = {
-x5A4D,			/*WORD e_magic;         Magic number */
-x0090,			/*WORD e_cblp;          Bytes on last page of file */
-x0003,			/*WORD e_cp;            Pages in file */
-x0000,			/*WORD e_crlc;          Relocations */
-x0004,			/*WORD e_cparhdr;       Size of header in paragraphs */
-x0000,			/*WORD e_minalloc;      Minimum extra paragraphs needed */
-xFFFF,			/*WORD e_maxalloc;      Maximum extra paragraphs needed */
-x0000,			/*WORD e_ss;            Initial (relative) SS value */
-x00B8,			/*WORD e_sp;            Initial SP value */
-x0000,			/*WORD e_csum;          Checksum */
-x0000,			/*WORD e_ip;            Initial IP value */
-x0000,			/*WORD e_cs;            Initial (relative) CS value */
-x0040,			/*WORD e_lfarlc;        File address of relocation table */
-x0000,			/*WORD e_ovno;          Overlay number */
-    {0, 0, 0, 0},		/*WORD e_res[4];     Reserved words */
-x0000,			/*WORD e_oemid;         OEM identifier (for e_oeminfo) */
-x0000,			/*WORD e_oeminfo;       OEM information; e_oemid specific */
-    {0, 0, 0, 0, 0, 0, 0, 0, 0, 0},	/*WORD e_res2[10];      Reserved words */
+typedef struct _IMAGE_BASE_RELOCATION {
-x00000080,			/*DWORD   e_lfanew;        File address of new exe header */
-    {				/* 14 code bytes + "This program cannot be run in DOS mode.\r\r\n$" + 6 * 0x00 */
-     /*0040 */ 0x0e, 0x1f, 0xba, 0x0e, 0x00, 0xb4, 0x09, 0xcd, 0x21, 0xb8,
-x01, 0x4c, 0xcd, 0x21, 0x54, 0x68,
-     /*0050 */ 0x69, 0x73, 0x20, 0x70, 0x72, 0x6f, 0x67, 0x72, 0x61, 0x6d,
-x20, 0x63, 0x61, 0x6e, 0x6e, 0x6f,
-     /*0060 */ 0x74, 0x20, 0x62, 0x65, 0x20, 0x72, 0x75, 0x6e, 0x20, 0x69,
+    DWORD   VirtualAddress;
-x6e, 0x20, 0x44, 0x4f, 0x53, 0x20,
-     /*0070 */ 0x6d, 0x6f, 0x64, 0x65, 0x2e, 0x0d, 0x0d, 0x0a, 0x24, 0x00,
-x00, 0x00, 0x00, 0x00, 0x00, 0x00,
-     /*0080 */
-     }
-};
-DWORD pe_magic = IMAGE_NT_SIGNATURE;
-IMAGE_FILE_HEADER pe_file_hdr = {
-x014C,			/*WORD    Machine; */
-x0003,			/*WORD    NumberOfSections; */
-x00000000,			/*DWORD   TimeDateStamp; */
-x00000000,			/*DWORD   PointerToSymbolTable; */
-x00000000,			/*DWORD   NumberOfSymbols; */
-x00E0,			/*WORD    SizeOfOptionalHeader; */
-x030F			/*WORD    Characteristics; */
-};
-IMAGE_OPTIONAL_HEADER32 pe_opt_hdr = {
-    /* Standard fields. */
-x010B,			/*WORD    Magic; */
-x06,			/*BYTE    MajorLinkerVersion; */
-x00,			/*BYTE    MinorLinkerVersion; */
-x00000000,			/*DWORD   SizeOfCode; */
-x00000000,			/*DWORD   SizeOfInitializedData; */
-x00000000,			/*DWORD   SizeOfUninitializedData; */
-x00000000,			/*DWORD   AddressOfEntryPoint; */
-x00000000,			/*DWORD   BaseOfCode; */
-x00000000,			/*DWORD   BaseOfData; */
-    /* NT additional fields. */
+    DWORD   SizeOfBlock;
+//  WORD    TypeOffset[1];
+} IMAGE_BASE_RELOCATION;
+#define IMAGE_SIZEOF_BASE_RELOCATION     8
-x00400000,			/*DWORD   ImageBase; */
+#define IMAGE_REL_BASED_ABSOLUTE         0
-x00001000,			/*DWORD   SectionAlignment; */
+#define IMAGE_REL_BASED_HIGH             1
+#define IMAGE_REL_BASED_LOW              2
+#define IMAGE_REL_BASED_HIGHLOW          3
-x00000200,			/*DWORD   FileAlignment; */
+#define IMAGE_REL_BASED_HIGHADJ          4
+#define IMAGE_REL_BASED_MIPS_JMPADDR     5
+#define IMAGE_REL_BASED_SECTION          6
+#define IMAGE_REL_BASED_REL32            7
-x0004,			/*WORD    MajorOperatingSystemVersion; */
+#pragma pack(pop)
-x0000,			/*WORD    MinorOperatingSystemVersion; */
-x0000,			/*WORD    MajorImageVersion; */
-x0000,			/*WORD    MinorImageVersion; */
-x0004,			/*WORD    MajorSubsystemVersion; */
-x0000,			/*WORD    MinorSubsystemVersion; */
-x00000000,			/*DWORD   Win32VersionValue; */
+/* ----------------------------------------------------------- */
-x00000000,			/*DWORD   SizeOfImage; */
+#endif /* ndef IMAGE_NT_SIGNATURE */
-x00000200,			/*DWORD   SizeOfHeaders; */
-x00000000,			/*DWORD   CheckSum; */
-x0002,			/*WORD    Subsystem; */
-x0000,			/*WORD    DllCharacteristics; */
+/* ----------------------------------------------------------- */
-x00100000,			/*DWORD   SizeOfStackReserve; */
+#pragma pack(push, 1)
-x00001000,			/*DWORD   SizeOfStackCommit; */
-x00100000,			/*DWORD   SizeOfHeapReserve; */
-x00001000,			/*DWORD   SizeOfHeapCommit; */
-x00000000,			/*DWORD   LoaderFlags; */
-x00000010,			/*DWORD   NumberOfRvaAndSizes; */
-    /* IMAGE_DATA_DIRECTORY DataDirectory[16]; */
-    {{0, 0}, {0, 0}, {0, 0}, {0, 0}, {0, 0}, {0, 0}, {0, 0}, {0, 0},
+struct pe_header
-     {0, 0}, {0, 0}, {0, 0}, {0, 0}, {0, 0}, {0, 0}, {0, 0}, {0, 0}}
-};
+{
-/*----------------------------------------------------------------------------*/
+    IMAGE_DOS_HEADER doshdr;
     BYTE dosstub[0x40];
-/*----------------------------------------------------------------------------*/
+    DWORD nt_sig;
+    IMAGE_FILE_HEADER filehdr;
 #ifdef TCC_TARGET_X86_64
+    IMAGE_OPTIONAL_HEADER64 opthdr;
+#else
-struct pe_import_header {
+#ifdef _WIN64
-    DWORD first_entry;
+    IMAGE_OPTIONAL_HEADER32 opthdr;
+#else
-    DWORD time_date;
+    IMAGE_OPTIONAL_HEADER opthdr;
-    DWORD forwarder;
+#endif
-    DWORD lib_name_offset;
+#endif
-    DWORD first_thunk;
+};
-};
+struct pe_reloc_header {
-struct pe_export_header {
+    DWORD offset;
-    DWORD Characteristics;
+    DWORD size;
-    DWORD TimeDateStamp;
+};
     DWORD Version;
-    DWORD Name;
+struct pe_rsrc_header {
-    DWORD Base;
+    struct _IMAGE_FILE_HEADER filehdr;
-    DWORD NumberOfFunctions;
+    struct _IMAGE_SECTION_HEADER sectionhdr;
+};
-    DWORD NumberOfNames;
+struct pe_rsrc_reloc {
-    DWORD AddressOfFunctions;
+    DWORD offset;
-    DWORD AddressOfNames;
+    DWORD size;
-    DWORD AddressOfNameOrdinals;
+    WORD type;
 };
-struct pe_reloc_header {
+#pragma pack(pop)
-    DWORD offset;
-    DWORD size;
+/* ------------------------------------------------------------- */
-};
+/* internal temporary structures */
+/*
+#define IMAGE_SCN_CNT_CODE                  0x00000020
-/* ------------------------------------------------------------- */
+#define IMAGE_SCN_CNT_INITIALIZED_DATA      0x00000040
-/* internal temporary structures */
+#define IMAGE_SCN_CNT_UNINITIALIZED_DATA    0x00000080
+#define IMAGE_SCN_MEM_DISCARDABLE           0x02000000
-ST const char *pe_sec_names[] = {
+#define IMAGE_SCN_MEM_SHARED                0x10000000
-    ".text",
+#define IMAGE_SCN_MEM_EXECUTE               0x20000000
-    ".data",
+#define IMAGE_SCN_MEM_READ                  0x40000000
-    ".bss",
+#define IMAGE_SCN_MEM_WRITE                 0x80000000
-    ".rsrc",
+*/
     ".reloc",
+enum {
-    ".stab",
+    sec_text = 0,
-    ".stabstr"
+    sec_data ,
+    sec_bss ,
-};
+    sec_idata ,
     sec_pdata ,
-enum {
+    sec_other ,
+    sec_rsrc ,
-    sec_text = 0,
+    sec_stab ,
-    sec_data,
+    sec_reloc ,
-    sec_bss,
+    sec_last
-    sec_rsrc,
+};
     sec_reloc,
-    sec_stab,
+static const DWORD pe_sec_flags[] = {
-    sec_stabstr,
+x60000020, /* ".text"     , */
-    pe_sec_number
+xC0000040, /* ".data"     , */
+xC0000080, /* ".bss"      , */
+x40000040, /* ".idata"    , */
+x40000040, /* ".pdata"    , */
-};
+xE0000060, /* < other >   , */
+x40000040, /* ".rsrc"     , */
 x42000802, /* ".stab"     , */
-ST DWORD pe_flags[] = {
+x42000040, /* ".reloc"    , */
-x60000020,			/* ".text", */
+};
 xC0000040,			/* ".data", */
-xC0000080,			/* ".bss", */
+struct section_info {
-x40000040,			/* ".rsrc", */
+    int cls, ord;
-x42000040,			/* ".reloc", */
+    char name[32];
-x42000802,			/* ".stab", */
+    DWORD sh_addr;
-x42000802			/* ".stabstr", */
+    DWORD sh_size;
+    DWORD sh_flags;
+    unsigned char *data;
+    DWORD data_size;
-};
+    IMAGE_SECTION_HEADER ish;
 };
 struct section_info {
-    struct section_info *next;
+struct import_symbol {
-    int id;
-    DWORD sh_addr;
-    DWORD sh_size;
-    unsigned char *data;
+    int sym_index;
-    DWORD data_size;
-};
-struct import_symbol {
-    int sym_index;
-    int offset;
-};
-struct pe_import_info {
-    int dll_index;
-    int sym_count;
-    struct import_symbol **symbols;
-};
+    int iat_index;
     int thk_offset;
-struct pe_info {
+};
     const char *filename;
+struct pe_import_info {
-    DWORD sizeofheaders;
-    DWORD imagebase;
-    DWORD start_addr;
-    DWORD imp_offs;
-    DWORD imp_size;
-    DWORD iat_offs;
-    DWORD iat_size;
-    DWORD exp_offs;
-    DWORD exp_size;
-    struct section_info sh_info[pe_sec_number];
-    int sec_count;
-    struct pe_import_info **imp_info;
-    int imp_count;
-    Section *reloc;
-    Section *thunk;
-    TCCState *s1;
-};
-/* ------------------------------------------------------------- */
-#define PE_MERGE_DATA
-// #define PE_PRINT_SECTIONS
-#ifndef MAX_PATH
-#define MAX_PATH 260
-#endif
+    int dll_index;
-void error_noabort(const char *, ...);
-ST char pe_type;
+    int sym_count;
+    struct import_symbol **symbols;
-#define PE_NUL 0
+};
+struct pe_info {
-#define PE_DLL 1
+    TCCState *s1;
+    Section *reloc;
+    Section *thunk;
-#define PE_GUI 2
+    const char *filename;
+    int type;
-#define PE_EXE 3
+    DWORD sizeofheaders;
+    ADDR3264 imagebase;
+    DWORD start_addr;
+    DWORD imp_offs;
+    DWORD imp_size;
-ST int pe_find_import(TCCState * s1, const char *symbol, char *ret)
+    DWORD iat_offs;
-{
+    DWORD iat_size;
-    int sym_index = find_elf_sym(s1->dynsymtab_section, symbol);
+    DWORD exp_offs;
-    if (0 == sym_index) {
+    DWORD exp_size;
-	/* Hm, maybe it's '_symbol' instead of 'symbol' or '__imp__symbol' */
+    int subsystem;
+    DWORD section_align;
+    DWORD file_align;
+    struct section_info *sec_info;
-	char buffer[100];
+    int sec_count;
+    struct pe_import_info **imp_info;
-	if (0 == memcmp(symbol, "__imp__", 7))
+    int imp_count;
+};
+#define PE_NUL 0
+#define PE_DLL 1
+#define PE_GUI 2
-	    symbol += 6;
+#define PE_EXE 3
+#define PE_RUN 4
-	else
+/* --------------------------------------------*/
+static const char *pe_export_name(TCCState *s1, ElfW(Sym) *sym)
-	    buffer[0] = '_', strcpy(buffer + 1, symbol), symbol = buffer;
+{
-	sym_index = find_elf_sym(s1->dynsymtab_section, symbol);
+    const char *name = symtab_section->link->data + sym->st_name;
     if (s1->leading_underscore && name[0] == '_' && !(sym->st_other & ST_PE_STDCALL))
-    if (ret)
-	strcpy(ret, symbol);
-    return sym_index;
-}
-#ifdef WIN32
-ST void **pe_imp;
-ST int nb_pe_imp;
-void *resolve_sym(struct TCCState *s1, const char *symbol, int type)
-{
-    char buffer[100], *p = buffer;
+        return name + 1;
-    void *a = NULL;
-    int sym_index = pe_find_import(s1, symbol, p);
-    int dll_index;
-    const char *dll_name;
-    void *hm;
-    if (sym_index) {
-	dll_index = ((Elf32_Sym *) s1->dynsymtab_section->data)[sym_index].
-	    st_other;
-        dll_name = s1->loaded_dlls[dll_index]->name;
+    return name;
 	hm = GetModuleHandleA(dll_name);
         if (NULL == hm)
-            hm = LoadLibraryA(dll_name);
+static int pe_find_import(TCCState * s1, ElfW(Sym) *sym)
-        if (hm) {
+{
-            a = GetProcAddress(hm, buffer);
+    char buffer[200];
-            if (a && STT_OBJECT == type) {
+    const char *s, *p;
-                // need to return a pointer to the address for data objects
+    int sym_index = 0, n = 0;
                 dynarray_add(&pe_imp, &nb_pe_imp, a);
-                a = &pe_imp[nb_pe_imp - 1];
+    do {
         s = pe_export_name(s1, sym);
         if (n) {
             /* second try: */
-    return a;
+	    if (sym->st_other & ST_PE_STDCALL) {
-}
+                /* try w/0 stdcall deco (windows API convention) */
-#endif
+	        p = strrchr(s, '@');
 	        if (!p || s[0] != '_')
-#define for_sym_in_symtab(sym) \
+	            break;
-for (sym = (Elf32_Sym *)symtab_section->data + 1; \
+	        strcpy(buffer, s+1)[p-s-1] = 0;
-	 sym < (Elf32_Sym *)(symtab_section->data + \
+	    } else if (s[0] != '_') { /* try non-ansi function */
-		symtab_section->data_offset); \
-	 ++sym)
+	        buffer[0] = '_', strcpy(buffer + 1, s);
-#define pe_set_datadir(dir,addr,size) \
+	    } else if (0 == memcmp(s, "__imp__", 7)) { /* mingw 2.0 */
-	pe_opt_hdr.DataDirectory[dir].VirtualAddress = addr, \
+	        strcpy(buffer, s + 6);
-	pe_opt_hdr.DataDirectory[dir].Size = size
+	    } else if (0 == memcmp(s, "_imp___", 7)) { /* mingw 3.7 */
+	        strcpy(buffer, s + 6);
-/*----------------------------------------------------------------------------*/
+	    } else {
-ST void dynarray_reset(void ***pp, int *n)
+	        break;
+	    }
-    int i;
+	    s = buffer;
+        }
+        sym_index = find_elf_sym(s1->dynsymtab_section, s);
-    for (i = 0; i < *n; ++i)
+        // printf("find (%d) %d %s\n", n, sym_index, s);
-	tcc_free((*pp)[i]);
+    } while (0 == sym_index && ++n < 2);
-    tcc_free(*pp);
+    return sym_index;
-    *pp = NULL;
+}
-    *n = 0;
-}
 /*----------------------------------------------------------------------------*/
+static int dynarray_assoc(void **pp, int n, int key)
+{
+    int i;
+    for (i = 0; i < n; ++i, ++pp)
+    if (key == **(int **) pp)
+        return i;
+    return -1;
+}
+#if 0
+ST_FN DWORD umin(DWORD a, DWORD b)
+{
+    return a < b ? a : b;
+}
+#endif
+static DWORD umax(DWORD a, DWORD b)
+{
+    return a < b ? b : a;
+}
 ST int dynarray_assoc(void **pp, int n, int key)
-{
+static DWORD pe_file_align(struct pe_info *pe, DWORD n)
     int i;
+    return (n + (pe->file_align - 1)) & ~(pe->file_align - 1);
+}
+static DWORD pe_virtual_align(struct pe_info *pe, DWORD n)
+{
+    return (n + (pe->section_align - 1)) & ~(pe->section_align - 1);
+}
+static void pe_align_section(Section *s, int a)
+{
+    int i = s->data_offset & (a-1);
+    if (i)
+        section_ptr_add(s, a - i);
+}
+static void pe_set_datadir(struct pe_header *hdr, int dir, DWORD addr, DWORD size)
+{
+    hdr->opthdr.DataDirectory[dir].VirtualAddress = addr;
+    hdr->opthdr.DataDirectory[dir].Size = size;
+}
+static int pe_fwrite(void *data, unsigned len, FILE *fp, DWORD *psum)
+{
+    if (psum) {
+        DWORD sum = *psum;
+        WORD *p = data;
+        int i;
+        for (i = len; i > 0; i -= 2) {
+            sum += (i >= 2) ? *p++ : *(BYTE*)p;
+            sum = (sum + (sum >> 16)) & 0xFFFF;
+        }
+        *psum = sum;
+    }
+    return len == fwrite(data, 1, len, fp) ? 0 : -1;
+}
+static void pe_fpad(FILE *fp, DWORD new_pos)
+{
+    DWORD pos = ftell(fp);
+    while (++pos <= new_pos)
+        fputc(0, fp);
+}
+/*----------------------------------------------------------------------------*/
+static int pe_write(struct pe_info *pe)
+{
+    static const struct pe_header pe_template = {
+    {
+    /* IMAGE_DOS_HEADER doshdr */
+x5A4D, /*WORD e_magic;         Magic number */
+x0090, /*WORD e_cblp;          Bytes on last page of file */
+x0003, /*WORD e_cp;            Pages in file */
+x0000, /*WORD e_crlc;          Relocations */
+x0004, /*WORD e_cparhdr;       Size of header in paragraphs */
+x0000, /*WORD e_minalloc;      Minimum extra paragraphs needed */
+xFFFF, /*WORD e_maxalloc;      Maximum extra paragraphs needed */
+x0000, /*WORD e_ss;            Initial (relative) SS value */
+x00B8, /*WORD e_sp;            Initial SP value */
+x0000, /*WORD e_csum;          Checksum */
+x0000, /*WORD e_ip;            Initial IP value */
+x0000, /*WORD e_cs;            Initial (relative) CS value */
+x0040, /*WORD e_lfarlc;        File address of relocation table */
+x0000, /*WORD e_ovno;          Overlay number */
+    {0,0,0,0}, /*WORD e_res[4];     Reserved words */
+x0000, /*WORD e_oemid;         OEM identifier (for e_oeminfo) */
+x0000, /*WORD e_oeminfo;       OEM information; e_oemid specific */
+    {0,0,0,0,0,0,0,0,0,0}, /*WORD e_res2[10];      Reserved words */
+x00000080  /*DWORD   e_lfanew;        File address of new exe header */
+    },{
+    /* BYTE dosstub[0x40] */
+    /* 14 code bytes + "This program cannot be run in DOS mode.\r\r\n$" + 6 * 0x00 */
+x0e,0x1f,0xba,0x0e,0x00,0xb4,0x09,0xcd,0x21,0xb8,0x01,0x4c,0xcd,0x21,0x54,0x68,
+x69,0x73,0x20,0x70,0x72,0x6f,0x67,0x72,0x61,0x6d,0x20,0x63,0x61,0x6e,0x6e,0x6f,
+x74,0x20,0x62,0x65,0x20,0x72,0x75,0x6e,0x20,0x69,0x6e,0x20,0x44,0x4f,0x53,0x20,
+x6d,0x6f,0x64,0x65,0x2e,0x0d,0x0d,0x0a,0x24,0x00,0x00,0x00,0x00,0x00,0x00,0x00,
+    },
+x00004550, /* DWORD nt_sig = IMAGE_NT_SIGNATURE */
+    {
+    /* IMAGE_FILE_HEADER filehdr */
+    IMAGE_FILE_MACHINE, /*WORD    Machine; */
+x0003, /*WORD    NumberOfSections; */
+x00000000, /*DWORD   TimeDateStamp; */
+x00000000, /*DWORD   PointerToSymbolTable; */
+x00000000, /*DWORD   NumberOfSymbols; */
+#if defined(TCC_TARGET_X86_64)
+x00F0, /*WORD    SizeOfOptionalHeader; */
+x022F  /*WORD    Characteristics; */
+#define CHARACTERISTICS_DLL 0x222E
+#elif defined(TCC_TARGET_I386)
+x00E0, /*WORD    SizeOfOptionalHeader; */
+x030F  /*WORD    Characteristics; */
+#define CHARACTERISTICS_DLL 0x230E
+#elif defined(TCC_TARGET_ARM)
+x00E0, /*WORD    SizeOfOptionalHeader; */
+x010F, /*WORD    Characteristics; */
+#define CHARACTERISTICS_DLL 0x230F
+#endif
+},{
+    /* IMAGE_OPTIONAL_HEADER opthdr */
+    /* Standard fields. */
+#ifdef TCC_TARGET_X86_64
+x020B, /*WORD    Magic; */
+#else
-    for (i = 0; i < n; ++i, ++pp)
+x010B, /*WORD    Magic; */
-	if (key == **(int **) pp)
+#endif
-	    return i;
+x06, /*BYTE    MajorLinkerVersion; */
-    return -1;
+x00, /*BYTE    MinorLinkerVersion; */
-}
+x00000000, /*DWORD   SizeOfCode; */
 x00000000, /*DWORD   SizeOfInitializedData; */
-#if 0
+x00000000, /*DWORD   SizeOfUninitializedData; */
-ST DWORD umin(DWORD a, DWORD b)
+x00000000, /*DWORD   AddressOfEntryPoint; */
-{
+x00000000, /*DWORD   BaseOfCode; */
-    return a < b ? a : b;
+#ifndef TCC_TARGET_X86_64
-}
-#endif
+x00000000, /*DWORD   BaseOfData; */
-ST DWORD umax(DWORD a, DWORD b)
-{
+#endif
-    return a < b ? b : a;
+    /* NT additional fields. */
-}
+#if defined(TCC_TARGET_ARM)
-ST void pe_fpad(FILE * fp, DWORD new_pos)
-{
-    DWORD pos = ftell(fp);
+x00100000,	    /*DWORD   ImageBase; */
-    while (++pos <= new_pos)
-	fputc(0, fp);
-}
-ST DWORD pe_file_align(DWORD n)
+#else
+x00400000,	    /*DWORD   ImageBase; */
+#endif
 x00001000, /*DWORD   SectionAlignment; */
-    return (n + (0x200 - 1)) & ~(0x200 - 1);
-}
-ST DWORD pe_virtual_align(DWORD n)
-{
-    return (n + (0x1000 - 1)) & ~(0x1000 - 1);
+x00000200, /*DWORD   FileAlignment; */
-}
+x0004, /*WORD    MajorOperatingSystemVersion; */
-ST void pe_align_section(Section * s, int a)
-{
-    int i = s->data_offset & (a - 1);
-    if (i)
-	section_ptr_add(s, a - i);
-}
-/*----------------------------------------------------------------------------*/
-ST int pe_write_pe(struct pe_info *pe)
-{
-    int i;
-    FILE *op;
-    DWORD file_offset;
-    IMAGE_SECTION_HEADER ish[pe_sec_number], *psh;
-    int sec_index = 0;
-    op = fopen(pe->filename, "wb");
-    if (NULL == op) {
-	error_noabort("could not create file: %s", pe->filename);
+x0000, /*WORD    MinorOperatingSystemVersion; */
+x0000, /*WORD    MajorImageVersion; */
+x0000, /*WORD    MinorImageVersion; */
+x0004, /*WORD    MajorSubsystemVersion; */
+x0000, /*WORD    MinorSubsystemVersion; */
+x00000000, /*DWORD   Win32VersionValue; */
+x00000000, /*DWORD   SizeOfImage; */
+x00000200, /*DWORD   SizeOfHeaders; */
+x00000000, /*DWORD   CheckSum; */
+x0002, /*WORD    Subsystem; */
+x0000, /*WORD    DllCharacteristics; */
-	return 1;
+x00100000, /*DWORD   SizeOfStackReserve; */
+x00001000, /*DWORD   SizeOfStackCommit; */
+x00100000, /*DWORD   SizeOfHeapReserve; */
+x00001000, /*DWORD   SizeOfHeapCommit; */
+x00000000, /*DWORD   LoaderFlags; */
+x00000010, /*DWORD   NumberOfRvaAndSizes; */
+    /* IMAGE_DATA_DIRECTORY DataDirectory[16]; */
+    {{0,0}, {0,0}, {0,0}, {0,0}, {0,0}, {0,0}, {0,0}, {0,0},
-    }
+     {0,0}, {0,0}, {0,0}, {0,0}, {0,0}, {0,0}, {0,0}, {0,0}}
+    }};
-    memset(&ish, 0, sizeof ish);
-    pe->sizeofheaders = pe_file_align(sizeof pe_dos_hdr
+    struct pe_header pe_header = pe_template;
-				      + sizeof pe_magic
-				      + sizeof pe_file_hdr
+    int i;
-				      + sizeof pe_opt_hdr
+    FILE *op;
+    DWORD file_offset, sum;
-				      +
+    struct section_info *si;
+    IMAGE_SECTION_HEADER *psh;
+    op = fopen(pe->filename, "wb");
+    if (NULL == op) {
-				      pe->sec_count *
+        tcc_error_noabort("could not write '%s': %s", pe->filename, strerror(errno));
+        return -1;
-				      sizeof(IMAGE_SECTION_HEADER)
+    }
-	);
+    pe->sizeofheaders = pe_file_align(pe,
+        sizeof (struct pe_header)
+        + pe->sec_count * sizeof (IMAGE_SECTION_HEADER)
+        );
+    file_offset = pe->sizeofheaders;
+    if (2 == pe->s1->verbose)
+        printf("-------------------------------"
+               "\n  virt   file   size  section" "\n");
+    for (i = 0; i < pe->sec_count; ++i) {
+        DWORD addr, size;
-    file_offset = pe->sizeofheaders;
+        const char *sh_name;
-    pe_fpad(op, file_offset);
+        si = pe->sec_info + i;
-    if (2 == verbose)
+        sh_name = si->name;
-	printf("-------------------------------"
+        addr = si->sh_addr - pe->imagebase;
-	       "\n  virt   file   size  section" "\n");
+        size = si->sh_size;
-    for (i = 0; i < pe->sec_count; ++i) {
+        psh = &si->ish;
-	struct section_info *si = pe->sh_info + i;
-	const char *sh_name = pe_sec_names[si->id];
+        if (2 == pe->s1->verbose)
-	unsigned long addr = si->sh_addr - pe->imagebase;
-	unsigned long size = si->sh_size;
+            printf("%6x %6x %6x  %s\n",
+                (unsigned)addr, (unsigned)file_offset, (unsigned)size, sh_name);
-	if (2 == verbose)
-	    printf("%6lx %6lx %6lx  %s\n",
-		   addr, file_offset, size, sh_name);
+        switch (si->cls) {
+            case sec_text:
+                pe_header.opthdr.BaseOfCode = addr;
+                pe_header.opthdr.AddressOfEntryPoint = addr + pe->start_addr;
+                break;
+            case sec_data:
+#ifndef TCC_TARGET_X86_64
+                pe_header.opthdr.BaseOfData = addr;
+#endif
+                break;
+            case sec_bss:
+                break;
+            case sec_reloc:
+                pe_set_datadir(&pe_header, IMAGE_DIRECTORY_ENTRY_BASERELOC, addr, size);
+                break;
+            case sec_rsrc:
+                pe_set_datadir(&pe_header, IMAGE_DIRECTORY_ENTRY_RESOURCE, addr, size);
+                break;
+            case sec_pdata:
+                pe_set_datadir(&pe_header, IMAGE_DIRECTORY_ENTRY_EXCEPTION, addr, size);
-	switch (si->id) {
-	case sec_text:
-	    pe_opt_hdr.BaseOfCode = addr;
+                break;
-	    pe_opt_hdr.AddressOfEntryPoint = addr + pe->start_addr;
-	    break;
-	case sec_data:
-	    pe_opt_hdr.BaseOfData = addr;
-	    if (pe->imp_size) {
-		pe_set_datadir(IMAGE_DIRECTORY_ENTRY_IMPORT,
-			       pe->imp_offs + addr, pe->imp_size);
-		pe_set_datadir(IMAGE_DIRECTORY_ENTRY_IAT,
-			       pe->iat_offs + addr, pe->iat_size);
+            case sec_stab:
-	    }
-	    if (pe->exp_size) {
-		pe_set_datadir(IMAGE_DIRECTORY_ENTRY_EXPORT,
+                break;
-			       pe->exp_offs + addr, pe->exp_size);
+        }
-	    }
-	    break;
+        if (pe->thunk == pe->s1->sections[si->ord]) {
+            if (pe->imp_size) {
-	case sec_bss:
+                pe_set_datadir(&pe_header, IMAGE_DIRECTORY_ENTRY_IMPORT,
-	    break;
+                    pe->imp_offs + addr, pe->imp_size);
                 pe_set_datadir(&pe_header, IMAGE_DIRECTORY_ENTRY_IAT,
-	case sec_reloc:
+                    pe->iat_offs + addr, pe->iat_size);
+            }
-	    pe_set_datadir(IMAGE_DIRECTORY_ENTRY_BASERELOC, addr, size);
+            if (pe->exp_size) {
-	    break;
+                pe_set_datadir(&pe_header, IMAGE_DIRECTORY_ENTRY_EXPORT,
                     pe->exp_offs + addr, pe->exp_size);
 	case sec_rsrc:
 	    pe_set_datadir(IMAGE_DIRECTORY_ENTRY_RESOURCE, addr, size);
 	    break;
+        strncpy((char*)psh->Name, sh_name, sizeof psh->Name);
-	case sec_stab:
-	    break;
+        psh->Characteristics = pe_sec_flags[si->cls];
+        psh->VirtualAddress = addr;
+        psh->Misc.VirtualSize = size;
-	case sec_stabstr:
+        pe_header.opthdr.SizeOfImage =
-	    break;
+            umax(pe_virtual_align(pe, size + addr), pe_header.opthdr.SizeOfImage);
-	}
         if (si->data_size) {
-	psh = &ish[sec_index++];
+            psh->PointerToRawData = file_offset;
-	strcpy((char *) psh->Name, sh_name);
+            file_offset = pe_file_align(pe, file_offset + si->data_size);
+            psh->SizeOfRawData = file_offset - psh->PointerToRawData;
-	psh->Characteristics = pe_flags[si->id];
+        }
-	psh->VirtualAddress = addr;
+    }
 	psh->Misc.VirtualSize = size;
-	pe_opt_hdr.SizeOfImage =
+    //pe_header.filehdr.TimeDateStamp = time(NULL);
+    pe_header.filehdr.NumberOfSections = pe->sec_count;
-	    umax(psh->VirtualAddress + psh->Misc.VirtualSize,
+    pe_header.opthdr.SizeOfHeaders = pe->sizeofheaders;
+    pe_header.opthdr.ImageBase = pe->imagebase;
-		 pe_opt_hdr.SizeOfImage);
+    pe_header.opthdr.Subsystem = pe->subsystem;
-	if (si->data_size) {
-	    psh->PointerToRawData = file_offset;
-	    fwrite(si->data, 1, si->data_size, op);
-	    file_offset = pe_file_align(file_offset + si->data_size);
+    if (pe->s1->pe_stack_size)
-	    psh->SizeOfRawData = file_offset - psh->PointerToRawData;
+        pe_header.opthdr.SizeOfStackReserve = pe->s1->pe_stack_size;
-	    pe_fpad(op, file_offset);
+    if (PE_DLL == pe->type)
-	}
+        pe_header.filehdr.Characteristics = CHARACTERISTICS_DLL;
     sum = 0;
-	/*----------------------------------------------------- */
+    pe_fwrite(&pe_header, sizeof pe_header, op, &sum);
     for (i = 0; i < pe->sec_count; ++i)
-    pe_file_hdr.NumberOfSections = sec_index;
+        pe_fwrite(&pe->sec_info[i].ish, sizeof(IMAGE_SECTION_HEADER), op, &sum);
-    pe_opt_hdr.SizeOfHeaders = pe->sizeofheaders;
+    pe_fpad(op, pe->sizeofheaders);
-    pe_opt_hdr.ImageBase = pe->imagebase;
+    for (i = 0; i < pe->sec_count; ++i) {
-    if (PE_DLL == pe_type)
+        si = pe->sec_info + i;
-	pe_file_hdr.Characteristics = 0x230E;
+        psh = &si->ish;
-    else if (PE_GUI != pe_type)
+        if (si->data_size) {
-	pe_opt_hdr.Subsystem = 3;
+            pe_fwrite(si->data, si->data_size, op, &sum);
             file_offset = psh->PointerToRawData + psh->SizeOfRawData;
-    fseek(op, SEEK_SET, 0);
+            pe_fpad(op, file_offset);
-    fwrite(&pe_dos_hdr, 1, sizeof pe_dos_hdr, op);
+        }
     fwrite(&pe_magic, 1, sizeof pe_magic, op);
     fwrite(&pe_file_hdr, 1, sizeof pe_file_hdr, op);
+    pe_header.opthdr.CheckSum = sum + file_offset;
-    fwrite(&pe_opt_hdr, 1, sizeof pe_opt_hdr, op);
+    fseek(op, offsetof(struct pe_header, opthdr.CheckSum), SEEK_SET);
-    for (i = 0; i < sec_index; ++i)
+    pe_fwrite(&pe_header.opthdr.CheckSum, sizeof pe_header.opthdr.CheckSum, op, NULL);
-	fwrite(&ish[i], 1, sizeof(IMAGE_SECTION_HEADER), op);
+    fclose (op);
-    fclose(op);
+    if (2 == pe->s1->verbose)
+        printf("-------------------------------\n");
+    if (pe->s1->verbose)
-    if (2 == verbose)
+        printf("<- %s (%u bytes)\n", pe->filename, (unsigned)file_offset);
-	printf("-------------------------------\n");
-    if (verbose)
+    return 0;
+}
-	printf("<-- %s (%lu bytes)\n", pe->filename, file_offset);
+/*----------------------------------------------------------------------------*/
-    return 0;
+static struct import_symbol *pe_add_import(struct pe_info *pe, int sym_index)
+{
-}
+    int i;
+    int dll_index;
-/*----------------------------------------------------------------------------*/
+    struct pe_import_info *p;
-ST int pe_add_import(struct pe_info *pe, int sym_index, DWORD offset)
+    struct import_symbol *s;
-{
+    ElfW(Sym) *isym;
-    int i;
-    int dll_index;
+    isym = (ElfW(Sym) *)pe->s1->dynsymtab_section->data + sym_index;
-    struct pe_import_info *p;
+    dll_index = isym->st_size;
-    struct import_symbol *s;
-    dll_index =
+    i = dynarray_assoc ((void**)pe->imp_info, pe->imp_count, dll_index);
-	((Elf32_Sym *) pe->s1->dynsymtab_section->data)[sym_index].
+    if (-1 != i) {
-	st_other;
-    i = dynarray_assoc((void **) pe->imp_info, pe->imp_count, dll_index);
+        p = pe->imp_info[i];
+        goto found_dll;
-    if (-1 != i) {
+    }
-	p = pe->imp_info[i];
+    p = tcc_mallocz(sizeof *p);
-	goto found_dll;
-    }
+    p->dll_index = dll_index;
+    dynarray_add((void***)&pe->imp_info, &pe->imp_count, p);
-    p = tcc_mallocz(sizeof *p);
-    p->dll_index = dll_index;
+found_dll:
-    dynarray_add((void ***) &pe->imp_info, &pe->imp_count, p);
+    i = dynarray_assoc ((void**)p->symbols, p->sym_count, sym_index);
+    if (-1 != i)
-  found_dll:
+        return p->symbols[i];
-    i = dynarray_assoc((void **) p->symbols, p->sym_count, sym_index);
-    if (-1 != i)
+    s = tcc_mallocz(sizeof *s);
-	goto found_sym;
+    dynarray_add((void***)&p->symbols, &p->sym_count, s);
-    s = tcc_mallocz(sizeof *s);
+    s->sym_index = sym_index;
-    s->sym_index = sym_index;
+    return s;
+}
-    s->offset = offset;
+/*----------------------------------------------------------------------------*/
-    dynarray_add((void ***) &p->symbols, &p->sym_count, s);
+static void pe_build_imports(struct pe_info *pe)
-  found_sym:
+{
-    return 1;
+    int thk_ptr, ent_ptr, dll_ptr, sym_cnt, i;
+    DWORD rva_base = pe->thunk->sh_addr - pe->imagebase;
+    int ndlls = pe->imp_count;
-}
+    for (sym_cnt = i = 0; i < ndlls; ++i)
+        sym_cnt += pe->imp_info[i]->sym_count;
-/*----------------------------------------------------------------------------*/
+    if (0 == sym_cnt)
+        return;
-ST void pe_build_imports(struct pe_info *pe)
-{
+    pe_align_section(pe->thunk, 16);
-    int thk_ptr, ent_ptr, dll_ptr, sym_cnt, i;
+    pe->imp_offs = dll_ptr = pe->thunk->data_offset;
-    DWORD voffset = pe->thunk->sh_addr - pe->imagebase;
+    pe->imp_size = (ndlls + 1) * sizeof(IMAGE_IMPORT_DESCRIPTOR);
-    int ndlls = pe->imp_count;
+    pe->iat_offs = dll_ptr + pe->imp_size;
+    pe->iat_size = (sym_cnt + ndlls) * sizeof(ADDR3264);
-    for (sym_cnt = i = 0; i < ndlls; ++i)
+    section_ptr_add(pe->thunk, pe->imp_size + 2*pe->iat_size);
-	sym_cnt += pe->imp_info[i]->sym_count;
+    thk_ptr = pe->iat_offs;
-    if (0 == sym_cnt)
+    ent_ptr = pe->iat_offs + pe->iat_size;
 	return;
+    for (i = 0; i < pe->imp_count; ++i) {
-    pe_align_section(pe->thunk, 16);
+        IMAGE_IMPORT_DESCRIPTOR *hdr;
         int k, n, dllindex;
+        ADDR3264 v;
+        struct pe_import_info *p = pe->imp_info[i];
+        const char *name;
+        DLLReference *dllref;
+        dllindex = p->dll_index;
+        if (dllindex)
-    pe->imp_offs = dll_ptr = pe->thunk->data_offset;
+            name = (dllref = pe->s1->loaded_dlls[dllindex-1])->name;
-    pe->imp_size = (ndlls + 1) * sizeof(struct pe_import_header);
+        else
-    pe->iat_offs = dll_ptr + pe->imp_size;
-    pe->iat_size = (sym_cnt + ndlls) * sizeof(DWORD);
-    section_ptr_add(pe->thunk, pe->imp_size + 2 * pe->iat_size);
+            name = "", dllref = NULL;
-    thk_ptr = pe->iat_offs;
+        /* put the dll name into the import header */
-    ent_ptr = pe->iat_offs + pe->iat_size;
+        v = put_elf_str(pe->thunk, name);
-    for (i = 0; i < pe->imp_count; ++i) {
+        hdr = (IMAGE_IMPORT_DESCRIPTOR*)(pe->thunk->data + dll_ptr);
-	struct pe_import_header *hdr;
+        hdr->FirstThunk = thk_ptr + rva_base;
-	int k, n, v;
+        hdr->OriginalFirstThunk = ent_ptr + rva_base;
+        hdr->Name = v + rva_base;
-	struct pe_import_info *p = pe->imp_info[i];
-	const char *name = pe->s1->loaded_dlls[p->dll_index]->name;
+        for (k = 0, n = p->sym_count; k <= n; ++k) {
+            if (k < n) {
-	/* put the dll name into the import header */
-	if (0 == strncmp(name, "lib", 3))
-	    name += 3;
+                int iat_index = p->symbols[k]->iat_index;
+                int sym_index = p->symbols[k]->sym_index;
+                ElfW(Sym) *imp_sym = (ElfW(Sym) *)pe->s1->dynsymtab_section->data + sym_index;
-	v = put_elf_str(pe->thunk, name);
+                ElfW(Sym) *org_sym = (ElfW(Sym) *)symtab_section->data + iat_index;
+                const char *name = pe->s1->dynsymtab_section->link->data + imp_sym->st_name;
+                int ordinal;
+                org_sym->st_value = thk_ptr;
-	hdr = (struct pe_import_header *) (pe->thunk->data + dll_ptr);
+                org_sym->st_shndx = pe->thunk->sh_num;
+                if (dllref)
+                    v = 0, ordinal = imp_sym->st_value; /* ordinal from pe_load_def */
+                else
-	hdr->first_thunk = thk_ptr + voffset;
+                    ordinal = 0, v = imp_sym->st_value; /* address from tcc_add_symbol() */
-	hdr->first_entry = ent_ptr + voffset;
+#ifdef TCC_IS_NATIVE
+                if (pe->type == PE_RUN) {
+                    if (dllref) {
+                        if ( !dllref->handle )
-	hdr->lib_name_offset = v + voffset;
+                            dllref->handle = LoadLibrary(dllref->name);
+                        v = (ADDR3264)GetProcAddress(dllref->handle, ordinal?(LPCSTR)NULL+ordinal:name);
+                    }
+                    if (!v)
+                        tcc_error_noabort("can't build symbol '%s'", name);
+                } else
+#endif
+                if (ordinal) {
-	for (k = 0, n = p->sym_count; k <= n; ++k) {
+                    v = ordinal | (ADDR3264)1 << (sizeof(ADDR3264)*8 - 1);
-	    if (k < n) {
+                } else {
-		DWORD offset = p->symbols[k]->offset;
+                    v = pe->thunk->data_offset + rva_base;
+                    section_ptr_add(pe->thunk, sizeof(WORD)); /* hint, not used */
+                    put_elf_str(pe->thunk, name);
-		int sym_index = p->symbols[k]->sym_index;
 		Elf32_Sym *sym =
-		    (Elf32_Sym *) pe->s1->dynsymtab_section->data +
+            } else {
+                v = 0; /* last entry is zero */
-		    sym_index;
+            }
-		const char *name =
-		    pe->s1->dynsymtab_section->link->data + sym->st_name;
+            *(ADDR3264*)(pe->thunk->data+thk_ptr) =
+            *(ADDR3264*)(pe->thunk->data+ent_ptr) = v;
-		if (offset & 0x80000000) {	/* ref to data */
+            thk_ptr += sizeof (ADDR3264);
-		    Elf32_Sym *sym =
-			&((Elf32_Sym *) symtab_section->
+            ent_ptr += sizeof (ADDR3264);
-			  data)[offset & 0x7FFFFFFF];
+        }
-		    sym->st_value = thk_ptr;
+        dll_ptr += sizeof(IMAGE_IMPORT_DESCRIPTOR);
-		    sym->st_shndx = pe->thunk->sh_num;
+        dynarray_reset(&p->symbols, &p->sym_count);
-		} else {	/* ref to function */
+    }
+    dynarray_reset(&pe->imp_info, &pe->imp_count);
-		    char buffer[100];
+}
-		    sprintf(buffer, "IAT.%s", name);
-		    sym_index =
+/* ------------------------------------------------------------- */
+struct pe_sort_sym
+{
+    int index;
+    const char *name;
+};
+static int sym_cmp(const void *va, const void *vb)
+{
+    const char *ca = (*(struct pe_sort_sym**)va)->name;
+    const char *cb = (*(struct pe_sort_sym**)vb)->name;
+    return strcmp(ca, cb);
+}
-			put_elf_sym(symtab_section, thk_ptr, sizeof(DWORD),
+static void pe_build_exports(struct pe_info *pe)
 				    ELF32_ST_INFO(STB_GLOBAL, STT_OBJECT),
-, pe->thunk->sh_num, buffer);
+    ElfW(Sym) *sym;
+    int sym_index, sym_end;
+    DWORD rva_base, func_o, name_o, ord_o, str_o;
-		    put_elf_reloc(symtab_section, text_section, offset, R_386_32,	/*R_JMP_SLOT, */
+    IMAGE_EXPORT_DIRECTORY *hdr;
-				  sym_index);
+    int sym_count, ord;
-		}
+    struct pe_sort_sym **sorted, *p;
-		v = pe->thunk->data_offset + voffset;
-		section_ptr_add(pe->thunk, sizeof(WORD));	/* hint, not used */
+    FILE *op;
-		put_elf_str(pe->thunk, name);
+    char buf[MAX_PATH];
-	    } else {
+    const char *dllname;
-		v = 0;		// last entry is zero
+    const char *name;
-	    }
+    rva_base = pe->thunk->sh_addr - pe->imagebase;
-	    *(DWORD *) (pe->thunk->data + thk_ptr) =
+    sym_count = 0, sorted = NULL, op = NULL;
 		*(DWORD *) (pe->thunk->data + ent_ptr) = v;
-	    thk_ptr += sizeof(DWORD);
+    sym_end = symtab_section->data_offset / sizeof(ElfW(Sym));
+    for (sym_index = 1; sym_index < sym_end; ++sym_index) {
+        sym = (ElfW(Sym)*)symtab_section->data + sym_index;
-	    ent_ptr += sizeof(DWORD);
+        name = pe_export_name(pe->s1, sym);
         if ((sym->st_other & ST_PE_EXPORT)
-	dll_ptr += sizeof(struct pe_import_header);
+            /* export only symbols from actually written sections */
-	dynarray_reset((void ***) &p->symbols, &p->sym_count);
+            && pe->s1->sections[sym->st_shndx]->sh_addr) {
-    }
-    dynarray_reset((void ***) &pe->imp_info, &pe->imp_count);
+            p = tcc_malloc(sizeof *p);
             p->index = sym_index;
             p->name = name;
-/* ------------------------------------------------------------- */
+            dynarray_add((void***)&sorted, &sym_count, p);
 ST int sym_cmp(const void *va, const void *vb)
+#if 0
         if (sym->st_other & ST_PE_EXPORT)
-    Elf32_Sym *sa = (Elf32_Sym *)symtab_section->data + *(int*)va;
+            printf("export: %s\n", name);
-    Elf32_Sym *sb = (Elf32_Sym *)symtab_section->data + *(int*)vb;
-    const char *ca = symtab_section->link->data + sa->st_name;
-    const char *cb = symtab_section->link->data + sb->st_name;
-    return strcmp(ca, cb);
-}
-ST void pe_build_exports(struct pe_info *pe)
-{
-    Elf32_Sym *sym;
-    DWORD func_offset, voffset;
-    struct pe_export_header *hdr;
-    int sym_count, n, ord, *sorted;
-    voffset = pe->thunk->sh_addr - pe->imagebase;
-    sym_count = 0, n = 1, sorted = NULL;
-    // for simplicity only functions are exported
-    for_sym_in_symtab(sym)
-    {
-        if ((sym->st_other & 1)
-            && sym->st_shndx == text_section->sh_num)
-            dynarray_add((void***)&sorted, &sym_count, (void*)n);
-        ++n;
-    }
-    if (0 == sym_count)
-	return;
+        if (sym->st_other & ST_PE_STDCALL)
+            printf("stdcall: %s\n", name);
+#endif
+    }
+    if (0 == sym_count)
+        return;
+    qsort (sorted, sym_count, sizeof *sorted, sym_cmp);
+    pe_align_section(pe->thunk, 16);
+    dllname = tcc_basename(pe->filename);
+    pe->exp_offs = pe->thunk->data_offset;
+    func_o = pe->exp_offs + sizeof(IMAGE_EXPORT_DIRECTORY);
+    name_o = func_o + sym_count * sizeof (DWORD);
+    ord_o = name_o + sym_count * sizeof (DWORD);
+    str_o = ord_o + sym_count * sizeof(WORD);
+    hdr = section_ptr_add(pe->thunk, str_o - pe->exp_offs);
+    hdr->Characteristics        = 0;
+    hdr->Base                   = 1;
+    hdr->NumberOfFunctions      = sym_count;
+    hdr->NumberOfNames          = sym_count;
+    hdr->AddressOfFunctions     = func_o + rva_base;
+    hdr->AddressOfNames         = name_o + rva_base;
+    hdr->AddressOfNameOrdinals  = ord_o + rva_base;
+    hdr->Name                   = str_o + rva_base;
+    put_elf_str(pe->thunk, dllname);
+#if 1
-    qsort (sorted, sym_count, sizeof sorted[0], sym_cmp);
+    /* automatically write exports to .def */
-    pe_align_section(pe->thunk, 16);
+    pstrcpy(buf, sizeof buf, pe->filename);
+    strcpy(tcc_fileextension(buf), ".def");
-    pe->exp_offs = pe->thunk->data_offset;
+    op = fopen(buf, "w");
-    hdr = section_ptr_add(pe->thunk,
+    if (NULL == op) {
-			  sizeof(struct pe_export_header) +
-			  sym_count * (2 * sizeof(DWORD) + sizeof(WORD)));
+        tcc_error_noabort("could not create '%s': %s", buf, strerror(errno));
+    } else {
+        fprintf(op, "LIBRARY %s\n\nEXPORTS\n", dllname);
-    func_offset = pe->exp_offs + sizeof(struct pe_export_header);
+        if (pe->s1->verbose)
+            printf("<- %s (%d symbols)\n", buf, sym_count);
-    hdr->Characteristics = 0;
+    }
-    hdr->Base = 1;
+#endif
     hdr->NumberOfFunctions = sym_count;
-    hdr->NumberOfNames = sym_count;
+    for (ord = 0; ord < sym_count; ++ord)
+    {
+        p = sorted[ord], sym_index = p->index, name = p->name;
+        /* insert actual address later in pe_relocate_rva */
+        put_elf_reloc(symtab_section, pe->thunk,
+            func_o, R_XXX_RELATIVE, sym_index);
+        *(DWORD*)(pe->thunk->data + name_o)
+            = pe->thunk->data_offset + rva_base;
+        *(WORD*)(pe->thunk->data + ord_o)
+            = ord;
+        put_elf_str(pe->thunk, name);
+        func_o += sizeof (DWORD);
+        name_o += sizeof (DWORD);
+        ord_o += sizeof (WORD);
+        if (op)
+            fprintf(op, "%s\n", name);
+    }
+    pe->exp_size = pe->thunk->data_offset - pe->exp_offs;
+    dynarray_reset(&sorted, &sym_count);
+    if (op)
+        fclose(op);
+}
+/* ------------------------------------------------------------- */
+static void pe_build_reloc (struct pe_info *pe)
+{
+    DWORD offset, block_ptr, addr;
+    int count, i;
+    ElfW_Rel *rel, *rel_end;
+    Section *s = NULL, *sr;
+    offset = addr = block_ptr = count = i = 0;
+    rel = rel_end = NULL;
+    for(;;) {
-    hdr->AddressOfFunctions = func_offset + voffset;
+        if (rel < rel_end) {
-    hdr->AddressOfNames         = hdr->AddressOfFunctions + sym_count * sizeof(DWORD);
+            int type = ELFW(R_TYPE)(rel->r_info);
-    hdr->AddressOfNameOrdinals  = hdr->AddressOfNames + sym_count * sizeof(DWORD);
+            addr = rel->r_offset + s->sh_addr;
-    hdr->Name = pe->thunk->data_offset + voffset;
+            ++ rel;
-    put_elf_str(pe->thunk, tcc_basename(pe->filename));
+            if (type != REL_TYPE_DIRECT)
+                continue;
+            if (count == 0) { /* new block */
+                block_ptr = pe->reloc->data_offset;
+                section_ptr_add(pe->reloc, sizeof(struct pe_reloc_header));
+                offset = addr & 0xFFFFFFFF<<12;
+            }
+            if ((addr -= offset)  < (1<<12)) { /* one block spans 4k addresses */
+                WORD *wp = section_ptr_add(pe->reloc, sizeof (WORD));
+                *wp = addr | IMAGE_REL_BASED_HIGHLOW<<12;
+                ++count;
+                continue;
+            }
+            -- rel;
+        } else if (i < pe->sec_count) {
+            sr = (s = pe->s1->sections[pe->sec_info[i++].ord])->reloc;
-    for (ord = 0; ord < sym_count; ++ord)
+            if (sr) {
                 rel = (ElfW_Rel *)sr->data;
-        char *name; DWORD *p, *pfunc, *pname; WORD *pord;
+                rel_end = (ElfW_Rel *)(sr->data + sr->data_offset);
+            }
+            continue;
+        }
-        sym = (Elf32_Sym *)symtab_section->data + sorted[ord];
+        if (count) {
-        name = symtab_section->link->data + sym->st_name;
-        p = (DWORD*)(pe->thunk->data + func_offset);
-        pfunc = p + ord;
-        pname = p + sym_count + ord;
-        pord = (WORD *)(p + 2*sym_count) + ord;
-        *pfunc = sym->st_value + pe->s1->sections[sym->st_shndx]->sh_addr - pe->imagebase;
+            /* store the last block and ready for a new one */
-	*pname = pe->thunk->data_offset + voffset;
+            struct pe_reloc_header *hdr;
-        *pord  = ord;
+            if (count & 1) /* align for DWORDS */
-	put_elf_str(pe->thunk, name);
+                section_ptr_add(pe->reloc, sizeof(WORD)), ++count;
-	/* printf("export: %s\n", name); */
+            hdr = (struct pe_reloc_header *)(pe->reloc->data + block_ptr);
-    }
+            hdr -> offset = offset - pe->imagebase;
+            hdr -> size = count * sizeof(WORD) + sizeof(struct pe_reloc_header);
-    pe->exp_size = pe->thunk->data_offset - pe->exp_offs;
+            count = 0;
-    tcc_free(sorted);
+        }
+        if (rel >= rel_end)
-}
+            break;
+    }
-/* ------------------------------------------------------------- */
+}
-ST void pe_build_reloc(struct pe_info *pe, int *section_order,
+/* ------------------------------------------------------------- */
+static int pe_section_class(Section *s)
-		       int section_count)
+{
-{
+    int type, flags;
-    DWORD offset, block_ptr, addr;
+    const char *name;
-    int count, i;
+    type = s->sh_type;
-    Elf32_Rel *rel, *rel_end;
+    flags = s->sh_flags;
-    Section *s = NULL, *sr;
+    name = s->name;
+    if (flags & SHF_ALLOC) {
+        if (type == SHT_PROGBITS) {
-    offset = addr = block_ptr = count = i = 0;
+            if (flags & SHF_EXECINSTR)
-    rel = rel_end = NULL;
+                return sec_text;
-    for (;;) {
-	if (rel < rel_end) {
+            if (flags & SHF_WRITE)
-	    int type = ELF32_R_TYPE(rel->r_info);
+                return sec_data;
+            if (0 == strcmp(name, ".rsrc"))
-	    addr = rel->r_offset + s->sh_addr;
+                return sec_rsrc;
-	    ++rel;
+            if (0 == strcmp(name, ".iedat"))
-	    if (type != R_386_32)
+                return sec_idata;
-		continue;
+            if (0 == strcmp(name, ".pdata"))
-	    if (count == 0) {	/* new block */
+                return sec_pdata;
-		block_ptr = pe->reloc->data_offset;
+            return sec_other;
-		section_ptr_add(pe->reloc, sizeof(struct pe_reloc_header));
+        } else if (type == SHT_NOBITS) {
-		offset = addr & 0xFFFFFFFF << 12;
+            if (flags & SHF_WRITE)
-	    }
+                return sec_bss;
-	    if ((addr -= offset) < (1 << 12)) {	/* one block spans 4k addresses */
+        }
-		WORD *wp = section_ptr_add(pe->reloc, sizeof(WORD));
+    } else {
+        if (0 == strcmp(name, ".reloc"))
+            return sec_reloc;
+        if (0 == strncmp(name, ".stab", 5)) /* .stab and .stabstr */
+            return sec_stab;
-		*wp = addr | IMAGE_REL_BASED_HIGHLOW << 12;
+    }
+    return -1;
+}
+static int pe_assign_addresses (struct pe_info *pe)
+{
+    int i, k, o, c;
+    DWORD addr;
+    int *section_order;
+    struct section_info *si;
+    Section *s;
+    // pe->thunk = new_section(pe->s1, ".iedat", SHT_PROGBITS, SHF_ALLOC);
-		++count;
+    section_order = tcc_malloc(pe->s1->nb_sections * sizeof (int));
+    for (o = k = 0 ; k < sec_last; ++k) {
+        for (i = 1; i < pe->s1->nb_sections; ++i) {
+            s = pe->s1->sections[i];
+            if (k == pe_section_class(s)) {
+                // printf("%s %d\n", s->name, k);
-		continue;
+                s->sh_addr = pe->imagebase;
+                section_order[o++] = i;
+            }
+        }
+    }
+    pe->sec_info = tcc_mallocz(o * sizeof (struct section_info));
+    addr = pe->imagebase + 1;
+    for (i = 0; i < o; ++i)
+    {
+        k = section_order[i];
+        s = pe->s1->sections[k];
+        c = pe_section_class(s);
+        si = &pe->sec_info[pe->sec_count];
+#ifdef PE_MERGE_DATA
+        if (c == sec_bss && pe->sec_count && si[-1].cls == sec_data) {
+            /* append .bss to .data */
             s->sh_addr = addr = ((addr-1) | (s->sh_addralign-1)) + 1;
+            addr += s->data_offset;
+            si[-1].sh_size = addr - si[-1].sh_addr;
+            continue;
+        }
+#endif
+        if (c == sec_stab && 0 == pe->s1->do_debug)
+            continue;
+        strcpy(si->name, s->name);
+        si->cls = c;
+        si->ord = k;
+        si->sh_addr = s->sh_addr = addr = pe_virtual_align(pe, addr);
+        si->sh_flags = s->sh_flags;
+        if (c == sec_data && NULL == pe->thunk)
-	    --rel;
+            pe->thunk = s;
 	} else if (i < section_count) {
-	    sr = (s = pe->s1->sections[section_order[i++]])->reloc;
+        if (s == pe->thunk) {
-	    if (sr) {
+            pe_build_imports(pe);
-		rel = (Elf32_Rel *) sr->data;
+            pe_build_exports(pe);
-		rel_end = (Elf32_Rel *) (sr->data + sr->data_offset);
+        }
         if (c == sec_reloc)
-	    continue;
+            pe_build_reloc (pe);
-	}
+        if (s->data_offset)
+        {
+            if (s->sh_type != SHT_NOBITS) {
+                si->data = s->data;
+                si->data_size = s->data_offset;
-	if (count) {		/* store the last block and ready for a new one */
+            }
-	    struct pe_reloc_header *hdr;
-	    if (count & 1)
+            addr += s->data_offset;
+            si->sh_size = s->data_offset;
+            ++pe->sec_count;
+        }
+        // printf("%08x %05x %s\n", si->sh_addr, si->sh_size, si->name);
+    }
+#if 0
+    for (i = 1; i < pe->s1->nb_sections; ++i) {
+        Section *s = pe->s1->sections[i];
-		section_ptr_add(pe->reloc, 2), ++count;
+        int type = s->sh_type;
+        int flags = s->sh_flags;
+        printf("section %-16s %-10s %5x %s,%s,%s\n",
+            s->name,
+            type == SHT_PROGBITS ? "progbits" :
+            type == SHT_NOBITS ? "nobits" :
-	    hdr = (struct pe_reloc_header *) (pe->reloc->data + block_ptr);
+            type == SHT_SYMTAB ? "symtab" :
-	    hdr->offset = offset - pe->imagebase;
+            type == SHT_STRTAB ? "strtab" :
+            type == SHT_RELX ? "rel" : "???",
+            s->data_offset,
+            flags & SHF_ALLOC ? "alloc" : "",
+            flags & SHF_WRITE ? "write" : "",
+            flags & SHF_EXECINSTR ? "exec" : ""
+            );
+    }
-	    hdr->size =
+    pe->s1->verbose = 2;
-		count * sizeof(WORD) + sizeof(struct pe_reloc_header);
+#endif
+    tcc_free(section_order);
+    return 0;
+}
-	    count = 0;
+/* ------------------------------------------------------------- */
+static void pe_relocate_rva (struct pe_info *pe, Section *s)
+{
+    Section *sr = s->reloc;
+    ElfW_Rel *rel, *rel_end;
+    rel_end = (ElfW_Rel *)(sr->data + sr->data_offset);
+    for(rel = (ElfW_Rel *)sr->data; rel < rel_end; rel++) {
+        if (ELFW(R_TYPE)(rel->r_info) == R_XXX_RELATIVE) {
+            int sym_index = ELFW(R_SYM)(rel->r_info);
+            DWORD addr = s->sh_addr;
+            if (sym_index) {
+                ElfW(Sym) *sym = (ElfW(Sym) *)symtab_section->data + sym_index;
+                addr = sym->st_value;
+            }
+            // printf("reloc rva %08x %08x %s\n", (DWORD)rel->r_offset, addr, s->name);
+            *(DWORD*)(s->data + rel->r_offset) += addr - pe->imagebase;
+        }
+    }
+}
+/*----------------------------------------------------------------------------*/
-	}
+static int pe_isafunc(int sym_index)
+{
+    Section *sr = text_section->reloc;
+    ElfW_Rel *rel, *rel_end;
-	if (rel >= rel_end)
+    Elf32_Word info = ELF32_R_INFO(sym_index, R_386_PC32);
+    if (!sr)
-	    break;
+        return 0;
-    }
+    rel_end = (ElfW_Rel *)(sr->data + sr->data_offset);
-}
+    for (rel = (ElfW_Rel *)sr->data; rel < rel_end; rel++)
+        if (rel->r_info == info)
+            return 1;
-/* ------------------------------------------------------------- */
+    return 0;
-ST int pe_assign_addresses(struct pe_info *pe)
+}
-{
-    int i, k, n;
+/*----------------------------------------------------------------------------*/
-    DWORD addr;
+static int pe_check_symbols(struct pe_info *pe)
+{
-    int section_order[pe_sec_number];
+    ElfW(Sym) *sym;
-    struct section_info *si_data = NULL;
+    int sym_index, sym_end;
+    int ret = 0;
-    pe->imagebase = PE_DLL == pe_type ? 0x10000000 : 0x00400000;
-    addr = pe->imagebase + 1;
+    pe_align_section(text_section, 8);
-    if (PE_DLL == pe_type)
+    sym_end = symtab_section->data_offset / sizeof(ElfW(Sym));
-	pe->reloc = new_section(pe->s1, ".reloc", SHT_DYNAMIC, SHF_ALLOC);
+    for (sym_index = 1; sym_index < sym_end; ++sym_index) {
-    for (n = k = 0; n < pe_sec_number; ++n) {
+        sym = (ElfW(Sym) *)symtab_section->data + sym_index;
-	for (i = 1; i < pe->s1->nb_sections; ++i) {
+        if (sym->st_shndx == SHN_UNDEF) {
 	    Section *s = pe->s1->sections[i];
-	    if (0 == strcmp(s->name, pe_sec_names[n])) {
+            const char *name = symtab_section->link->data + sym->st_name;
-		struct section_info *si = &pe->sh_info[pe->sec_count];
+            unsigned type = ELFW(ST_TYPE)(sym->st_info);
-#ifdef PE_MERGE_DATA
+            int imp_sym = pe_find_import(pe->s1, sym);
-		if (n == sec_bss && si_data) {
+            struct import_symbol *is;
-		    /* append .bss to .data */
+            if (0 == imp_sym)
-		    s->sh_addr = addr = ((addr - 1) | 15) + 1;
+                goto not_found;
 		    addr += s->data_offset;
-		    si_data->sh_size = addr - si_data->sh_addr;
+            if (type == STT_NOTYPE) {
-		} else
+                /* symbols from assembler have no type, find out which */
-#endif
+                if (pe_isafunc(sym_index))
                     type = STT_FUNC;
-		    si->sh_addr = s->sh_addr = addr =
+                else
-			pe_virtual_align(addr);
+                    type = STT_OBJECT;
-		    si->id = n;
+            }
-		    if (n == sec_data) {
+            is = pe_add_import(pe, imp_sym);
-			pe->thunk = s;
-			si_data = si;
+            if (type == STT_FUNC) {
-			pe_build_imports(pe);
+                unsigned long offset = is->thk_offset;
-			pe_build_exports(pe);
+                if (offset) {
+                    /* got aliased symbol, like stricmp and _stricmp */
+                } else {
-		    } else if (n == sec_reloc) {
+                    char buffer[100];
-			pe_build_reloc(pe, section_order, k);
+                    WORD *p;
-		    }
+                    offset = text_section->data_offset;
-		    if (s->data_offset) {
+                    /* add the 'jmp IAT[x]' instruction */
-			if (n != sec_bss) {
+#ifdef TCC_TARGET_ARM
-			    si->data = s->data;
+                    p = section_ptr_add(text_section, 8+4); // room for code and address
+                    (*(DWORD*)(p)) = 0xE59FC000; // arm code ldr ip, [pc] ; PC+8+0 = 0001xxxx
-			    si->data_size = s->data_offset;
+                    (*(DWORD*)(p+2)) = 0xE59CF000; // arm code ldr pc, [ip]
-			}
+#else
+                    p = section_ptr_add(text_section, 8);
+                    *p = 0x25FF;
+#ifdef TCC_TARGET_X86_64
+                    *(DWORD*)(p+1) = (DWORD)-4;
+#endif
+#endif
+                    /* add a helper symbol, will be patched later in
+                       pe_build_imports */
+                    sprintf(buffer, "IAT.%s", name);
+                    is->iat_index = put_elf_sym(
+                        symtab_section, 0, sizeof(DWORD),
+                        ELFW(ST_INFO)(STB_GLOBAL, STT_OBJECT),
+, SHN_UNDEF, buffer);
+#ifdef TCC_TARGET_ARM
+                    put_elf_reloc(symtab_section, text_section,
+                        offset + 8, R_XXX_THUNKFIX, is->iat_index); // offset to IAT position
+#else
+                    put_elf_reloc(symtab_section, text_section,
-			addr += s->data_offset;
+                        offset + 2, R_XXX_THUNKFIX, is->iat_index);
+#endif
+                    is->thk_offset = offset;
+                }
+                /* tcc_realloc might have altered sym's address */
-			si->sh_size = s->data_offset;
+                sym = (ElfW(Sym) *)symtab_section->data + sym_index;
-			++pe->sec_count;
+                /* patch the original symbol */
+                sym->st_value = offset;
+                sym->st_shndx = text_section->sh_num;
+                sym->st_other &= ~ST_PE_EXPORT; /* do not export */
-		    }
+                continue;
-		    //printf("Section %08X %04X %s\n", si->sh_addr, si->data_size, s->name);
+            }
-		}
-		section_order[k] = i, ++k;
+            if (type == STT_OBJECT) { /* data, ptr to that should be */
-	    }
+                if (0 == is->iat_index) {
-	}
+                    /* original symbol will be patched later in pe_build_imports */
+                    is->iat_index = sym_index;
-    }
+                    continue;
+                }
-    return 0;
+            }
-}
+        not_found:
-/*----------------------------------------------------------------------------*/
+            tcc_error_noabort("undefined symbol '%s'", name);
-ST int pe_check_symbols(struct pe_info *pe)
+            ret = -1;
-{
-    Elf32_Sym *sym;
+        } else if (pe->s1->rdynamic
+                   && ELFW(ST_BIND)(sym->st_info) != STB_LOCAL) {
+            /* if -rdynamic option, then export all non local symbols */
+            sym->st_other |= ST_PE_EXPORT;
+        }
+    }
-    int ret = 0;
+    return ret;
+}
-    pe_align_section(text_section, 8);
+/*----------------------------------------------------------------------------*/
-    for_sym_in_symtab(sym) {
+#ifdef PE_PRINT_SECTIONS
+static void pe_print_section(FILE * f, Section * s)
-	if (sym->st_shndx == SHN_UNDEF) {
+{
+    /* just if you'r curious */
+    BYTE *p, *e, b;
-	    const char *symbol = symtab_section->link->data + sym->st_name;
+    int i, n, l, m;
-	    unsigned type = ELF32_ST_TYPE(sym->st_info);
+    p = s->data;
-	    int sym_index = pe_find_import(pe->s1, symbol, NULL);
+    e = s->data + s->data_offset;
-	    if (sym_index) {
+    l = e - p;
-		if (type == STT_FUNC) {
-		    unsigned long offset = text_section->data_offset;
-		    if (pe_add_import(pe, sym_index, offset + 2)) {
+    fprintf(f, "section  \"%s\"", s->name);
-			/* add the 'jmp IAT[x]' instruction */
-			*(WORD *) section_ptr_add(text_section, 8) =
+    if (s->link)
-x25FF;
-			/* patch the symbol */
+        fprintf(f, "\nlink     \"%s\"", s->link->name);
-			sym->st_shndx = text_section->sh_num;
+    if (s->reloc)
-			sym->st_value = offset;
+        fprintf(f, "\nreloc    \"%s\"", s->reloc->name);
-			continue;
+    fprintf(f, "\nv_addr   %08X", (unsigned)s->sh_addr);
-		    }
+    fprintf(f, "\ncontents %08X", (unsigned)l);
-		} else if (type == STT_OBJECT) {	/* data, ptr to that should be */
+    fprintf(f, "\n\n");
-		    if (pe_add_import(pe, sym_index,
-				      (sym -
+    if (s->sh_type == SHT_NOBITS)
-				      (Elf32_Sym *) symtab_section->data) |
+        return;
-x80000000))
-			continue;
+    if (0 == l)
-		}
+        return;
-	    }
-	    error_noabort("undefined symbol '%s'", symbol);
+    if (s->sh_type == SHT_SYMTAB)
-	    ret = 1;
+        m = sizeof(ElfW(Sym));
-	} else
+    else if (s->sh_type == SHT_RELX)
-	    if (pe->s1->rdynamic
+        m = sizeof(ElfW_Rel);
-		&& ELF32_ST_BIND(sym->st_info) != STB_LOCAL) {
+    else
-	    /* if -rdynamic option, then export all non local symbols */
+        m = 16;
+    fprintf(f, "%-8s", "offset");
+    for (i = 0; i < m; ++i)
+        fprintf(f, " %02x", i);
+    n = 56;
+    if (s->sh_type == SHT_SYMTAB || s->sh_type == SHT_RELX) {
+        const char *fields1[] = {
+            "name",
+            "value",
+            "size",
+            "bind",
+            "type",
+            "other",
-	    sym->st_other |= 1;
+            "shndx",
+            NULL
+        };
-	}
+        const char *fields2[] = {
             "offs",
+            "type",
+            "symb",
-    return ret;
+            NULL
+        };
+        const char **p;
+        if (s->sh_type == SHT_SYMTAB)
+            p = fields1, n = 106;
+        else
+            p = fields2, n = 58;
+        for (i = 0; p[i]; ++i)
+            fprintf(f, "%6s", p[i]);
+        fprintf(f, "  symbol");
+    }
+    fprintf(f, "\n");
+    for (i = 0; i < n; ++i)
+        fprintf(f, "-");
+    fprintf(f, "\n");
+    for (i = 0; i < l;)
-}
+    {
+        fprintf(f, "%08X", i);
+        for (n = 0; n < m; ++n) {
+            if (n + i < l)
+                fprintf(f, " %02X", p[i + n]);
+            else
+                fprintf(f, "   ");
+        }
+        if (s->sh_type == SHT_SYMTAB) {
+            ElfW(Sym) *sym = (ElfW(Sym) *) (p + i);
+            const char *name = s->link->data + sym->st_name;
+            fprintf(f, "  %04X  %04X  %04X   %02X    %02X    %02X   %04X  \"%s\"",
+                    (unsigned)sym->st_name,
+                    (unsigned)sym->st_value,
+                    (unsigned)sym->st_size,
+                    (unsigned)ELFW(ST_BIND)(sym->st_info),
+                    (unsigned)ELFW(ST_TYPE)(sym->st_info),
+                    (unsigned)sym->st_other,
+                    (unsigned)sym->st_shndx,
+                    name);
+        } else if (s->sh_type == SHT_RELX) {
+            ElfW_Rel *rel = (ElfW_Rel *) (p + i);
+            ElfW(Sym) *sym =
+                (ElfW(Sym) *) s->link->data + ELFW(R_SYM)(rel->r_info);
+            const char *name = s->link->link->data + sym->st_name;
+            fprintf(f, "  %04X   %02X   %04X  \"%s\"",
+                    (unsigned)rel->r_offset,
+                    (unsigned)ELFW(R_TYPE)(rel->r_info),
+                    (unsigned)ELFW(R_SYM)(rel->r_info),
+                    name);
+        } else {
+            fprintf(f, "   ");
+            for (n = 0; n < m; ++n) {
+                if (n + i < l) {
+                    b = p[i + n];
+                    if (b < 32 || b >= 127)
+                        b = '.';
+                    fprintf(f, "%c", b);
+                }
+            }
+        }
+        i += m;
+        fprintf(f, "\n");
+    }
+    fprintf(f, "\n\n");
+}
+static void pe_print_sections(TCCState *s1, const char *fname)
+{
+    Section *s;
+    FILE *f;
+    int i;
+    f = fopen(fname, "w");
+    for (i = 1; i < s1->nb_sections; ++i) {
+        s = s1->sections[i];
+        pe_print_section(f, s);
-/*----------------------------------------------------------------------------*/
+    }
+    pe_print_section(f, s1->dynsymtab_section);
+    fclose(f);
+}
-#ifdef PE_PRINT_SECTIONS
+#endif
-ST void pe_print_section(FILE * f, Section * s)
+/* ------------------------------------------------------------- */
-{				/* just if you'r curious */
+/* helper function for load/store to insert one more indirection */
-    BYTE *p, *e, b;
+ST_FUNC SValue *pe_getimport(SValue *sv, SValue *v2)
+{
+    Sym *sym;
+    ElfW(Sym) *esym;
+    int r2;
-    int i, n, l, m;
+    if ((sv->r & (VT_VALMASK|VT_SYM)) != (VT_CONST|VT_SYM) || (sv->r2 != VT_CONST))
+        return sv;
+    sym = sv->sym;
-    p = s->data;
+    if ((sym->type.t & (VT_EXTERN|VT_STATIC)) != VT_EXTERN)
+        return sv;
+    if (!sym->c)
+        put_extern_sym(sym, NULL, 0, 0);
+    esym = &((ElfW(Sym) *)symtab_section->data)[sym->c];
+    if (!(esym->st_other & ST_PE_IMPORT))
-    e = s->data + s->data_offset;
+        return sv;
+    // printf("import %04x %04x %04x %s\n", sv->type.t, sym->type.t, sv->r, get_tok_str(sv->sym->v, NULL));
-    l = e - p;
+    memset(v2, 0, sizeof *v2);
+    v2->type.t = VT_PTR;
+    v2->r = VT_CONST | VT_SYM | VT_LVAL;
-    fprintf(f, "section  \"%s\"", s->name);
+    v2->sym = sv->sym;
     if (s->link)
-	fprintf(f, "\nlink     \"%s\"", s->link->name);
+    r2 = get_reg(RC_INT);
+    load(r2, v2);
-    if (s->reloc)
+    v2->r = r2;
 	fprintf(f, "\nreloc    \"%s\"", s->reloc->name);
-    fprintf(f, "\nv_addr   %08X", s->sh_addr);
-    fprintf(f, "\ncontents %08X", l);
+    if ((uint32_t)sv->c.i) {
-    fprintf(f, "\n\n");
+        vpushv(v2);
+        vpushi(sv->c.i);
+        gen_opi('+');
-    if (s->sh_type == SHT_NOBITS)
+        *v2 = *vtop--;
-	return;
+    }
-    if (s->sh_type == SHT_SYMTAB)
-	m = sizeof(Elf32_Sym);
+    v2->type.t = sv->type.t;
-    if (s->sh_type == SHT_REL)
+    v2->r |= sv->r & VT_LVAL;
-	m = sizeof(Elf32_Rel);
+    return v2;
-    else
+}
-	m = 16;
-    for (i = 0; i < l;) {
+ST_FUNC int pe_putimport(TCCState *s1, int dllindex, const char *name, addr_t value)
-	fprintf(f, "%08X", i);
+{
-	for (n = 0; n < m; ++n) {
-	    if (n + i < l)
-		fprintf(f, " %02X", p[i + n]);
+    return add_elf_sym(
-	    else
+        s1->dynsymtab_section,
-		fprintf(f, "   ");
+        value,
-	}
+        dllindex, /* st_size */
+        ELFW(ST_INFO)(STB_GLOBAL, STT_NOTYPE),
-	if (s->sh_type == SHT_SYMTAB) {
+,
-	    Elf32_Sym *sym = (Elf32_Sym *) (p + i);
+        value ? SHN_ABS : SHN_UNDEF,
-	    const char *name = s->link->data + sym->st_name;
+        name
-	    fprintf(f,
+        );
-		    "  name:%04X"
+}
-		    "  value:%04X"
-		    "  size:%04X"
+static int add_dllref(TCCState *s1, const char *dllname)
-		    "  bind:%02X"
-		    "  type:%02X"
+{
+    DLLReference *dllref;
-		    "  other:%02X"
+    int i;
-		    "  shndx:%04X"
+    for (i = 0; i < s1->nb_loaded_dlls; ++i)
-		    "  \"%s\"",
+        if (0 == strcmp(s1->loaded_dlls[i]->name, dllname))
+            return i + 1;
-		    sym->st_name,
+    dllref = tcc_mallocz(sizeof(DLLReference) + strlen(dllname));
-		    sym->st_value,
-		    sym->st_size,
-		    ELF32_ST_BIND(sym->st_info),
+    strcpy(dllref->name, dllname);
-		    ELF32_ST_TYPE(sym->st_info),
+    dynarray_add((void ***) &s1->loaded_dlls, &s1->nb_loaded_dlls, dllref);
-		    sym->st_other, sym->st_shndx, name);
+    return s1->nb_loaded_dlls;
-	} else if (s->sh_type == SHT_REL) {
+}
-	    Elf32_Rel *rel = (Elf32_Rel *) (p + i);
-	    Elf32_Sym *sym =
+/* ------------------------------------------------------------- */
-		(Elf32_Sym *) s->link->data + ELF32_R_SYM(rel->r_info);
+static int read_mem(int fd, unsigned offset, void *buffer, unsigned len)
+{
-	    const char *name = s->link->link->data + sym->st_name;
+    lseek(fd, offset, SEEK_SET);
-	    fprintf(f,
+    return len == read(fd, buffer, len);
-		    "  offset:%04X"
+}
 		    "  type:%02X"
+/* -------------------------------------------------------------
+ *  This is for compiled windows resources in 'coff' format
+ *  as generated by 'windres.exe -O coff ...'.
+ */
+static int pe_load_res(TCCState *s1, int fd)
+{
+    struct pe_rsrc_header hdr;
+    Section *rsrc_section;
+    int i, ret = -1;
+    BYTE *ptr;
+    unsigned offs;
+    if (!read_mem(fd, 0, &hdr, sizeof hdr))
+        goto quit;
+    if (hdr.filehdr.Machine != IMAGE_FILE_MACHINE
+        || hdr.filehdr.NumberOfSections != 1
+        || strcmp(hdr.sectionhdr.Name, ".rsrc") != 0)
+        goto quit;
+    rsrc_section = new_section(s1, ".rsrc", SHT_PROGBITS, SHF_ALLOC);
+    ptr = section_ptr_add(rsrc_section, hdr.sectionhdr.SizeOfRawData);
+    offs = hdr.sectionhdr.PointerToRawData;
+    if (!read_mem(fd, offs, ptr, hdr.sectionhdr.SizeOfRawData))
-		    "  symbol:%04X"
+        goto quit;
-		    "  \"%s\"",
+    offs = hdr.sectionhdr.PointerToRelocations;
-		    rel->r_offset,
+    for (i = 0; i < hdr.sectionhdr.NumberOfRelocations; ++i)
-		    ELF32_R_TYPE(rel->r_info),
+    {
-		    ELF32_R_SYM(rel->r_info), name);
+        struct pe_rsrc_reloc rel;
-	} else {
-	    fprintf(f, "   ");
+        if (!read_mem(fd, offs, &rel, sizeof rel))
-	    for (n = 0; n < m; ++n) {
+            goto quit;
-		if (n + i < l) {
+        // printf("rsrc_reloc: %x %x %x\n", rel.offset, rel.size, rel.type);
-		    b = p[i + n];
+        if (rel.type != RSRC_RELTYPE)
-		    if (b < 32 || b >= 127)
-			b = '.';
+            goto quit;
-		    fprintf(f, "%c", b);
+        put_elf_reloc(symtab_section, rsrc_section,
-		}
-	    }
+            rel.offset, R_XXX_RELATIVE, 0);
-	}
+        offs += sizeof rel;
-	i += m;
+    }
-	fprintf(f, "\n");
     ret = 0;
-    fprintf(f, "\n\n");
+quit:
+    return ret;
 }
+/* ------------------------------------------------------------- */
+#ifndef TCC_TARGET_MEOS
+static int pe_load_def(TCCState *s1, int fd)
+{
+    int state = 0, ret = -1, dllindex = 0, ord;
+    char line[400], dllname[80], *p, *x;
+    FILE *fp;
+    fp = fdopen(dup(fd), "rb");
+    while (fgets(line, sizeof line, fp))
+    {
+        p = trimfront(trimback(line, strchr(line, 0)));
+        if (0 == *p || ';' == *p)
+            continue;
+        switch (state) {
+        case 0:
+            if (0 != strnicmp(p, "LIBRARY", 7))
+                goto quit;
+            pstrcpy(dllname, sizeof dllname, trimfront(p+7));
+            ++state;
+            continue;
+        case 1:
+            if (0 != stricmp(p, "EXPORTS"))
+                goto quit;
+            ++state;
+            continue;
+        case 2:
+            dllindex = add_dllref(s1, dllname);
+            ++state;
+            /* fall through */
+        default:
+            /* get ordinal and will store in sym->st_value */
+            ord = 0;
+            x = strchr(p, ' ');
+            if (x) {
+                *x = 0, x = strrchr(x + 1, '@');
+                if (x) {
+                    char *d;
+                    ord = (int)strtol(x + 1, &d, 10);
+                    if (*d)
+                        ord = 0;
+                }
+            }
+            pe_putimport(s1, dllindex, p, ord);
+            continue;
+        }
+    }
+    ret = 0;
+quit:
+    fclose(fp);
+    return ret;
+}
+#else
+static int pe_load_def(TCCState *s1, int fd)
+{ return 0; }
+#endif
+/* ------------------------------------------------------------- */
+#define TINY_IMPDEF_GET_EXPORT_NAMES_ONLY
+#include "win32/tools/tiny_impdef.c"
+static int pe_load_dll(TCCState *s1, const char *dllname, int fd)
 #endif
     char *p, *q;
-static int pe_test_cmd(const char **pp, const char *cmd)
+    int index;
-{
+    p = get_export_names(fd);
-    const char *p;
+    if (!p)
-    char *q, buf[16];
+        return -1;
-    int ret;
+    index = add_dllref(s1, dllname);
-    p = *pp;
+    for (q = p; *q; q += 1 + strlen(q))
-    q = buf;
+        pe_putimport(s1, index, q, 0);
-    while (*p != '\0' && !is_space(*p)) {
+    tcc_free(p);
-        if ((q - buf) < sizeof(buf) - 1)
+    return 0;
+}
-            *q++ = toup(*p);
+/* ------------------------------------------------------------- */
-        p++;
+ST_FUNC int pe_load_file(struct TCCState *s1, const char *filename, int fd)
-    }
+{
-    *q = '\0';
+    int ret = -1;
-    ret = !strcmp(buf, cmd);
+    char buf[10];
+    if (0 == strcmp(tcc_fileextension(filename), ".def"))
+        ret = pe_load_def(s1, fd);
+    else if (pe_load_res(s1, fd) == 0)
+        ret = 0;
-    *pp = p;
+    else if (read_mem(fd, 0, buf, sizeof buf) && 0 == strncmp(buf, "MZ", 2))
-    return ret;
+        ret = pe_load_dll(s1, tcc_basename(filename), fd);
-}
+    return ret;
+}
-/* ------------------------------------------------------------- */
-int pe_load_def_file(TCCState * s1, FILE * fp)
+/* ------------------------------------------------------------- */
+#ifdef TCC_TARGET_X86_64
+static unsigned pe_add_uwwind_info(TCCState *s1)
+{
+    if (NULL == s1->uw_pdata) {
+        s1->uw_pdata = find_section(tcc_state, ".pdata");
+        s1->uw_pdata->sh_addralign = 4;
-    DLLReference *dllref;
+        s1->uw_sym = put_elf_sym(symtab_section, 0, 0, 0, 0, text_section->sh_num, NULL);
+    }
-    int f = 0, sym_index;
-    char *p, line[120], dllname[40];
+    if (0 == s1->uw_offs) {
-    while (fgets(line, sizeof line, fp)) {
+        /* As our functions all have the same stackframe, we use one entry for all */
-	//p = strchr(line, 0);
+        static const unsigned char uw_info[] = {
-	while (p > line && p[-1] <= ' ')
+x01, // UBYTE: 3 Version , UBYTE: 5 Flags
+x04, // UBYTE Size of prolog
-	    --p;
+x02, // UBYTE Count of unwind codes
-	*p = 0;
+x05, // UBYTE: 4 Frame Register (rbp), UBYTE: 4 Frame Register offset (scaled)
-	p = line;
-	while (*p && *p <= ' ')
-	    ++p;
+            // USHORT * n Unwind codes array
-	if (*p && ';' != *p)
-	    switch (f) {
+            // 0x0b, 0x01, 0xff, 0xff, // stack size
-	    case 0:
+x04, 0x03, // set frame ptr (mov rsp -> rbp)
-                if (!pe_test_cmd((const char **)&p, "LIBRARY"))
-		    return -1;
+x01, 0x50  // push reg (rbp)
-                while (is_space(*p))
+        };
-                    p++;
+        Section *s = text_section;
-		pstrcpy(dllname, sizeof(dllname), p);
+        unsigned char *p;
-		++f;
+        section_ptr_add(s, -s->data_offset & 3); /* align */
-		continue;
+        s1->uw_offs = s->data_offset;
+        p = section_ptr_add(s, sizeof uw_info);
-	    case 1:
+        memcpy(p, uw_info, sizeof uw_info);
                 if (!pe_test_cmd((const char **)&p, "EXPORTS"))
 		    return -1;
-		++f;
+    return s1->uw_offs;
+}
 		continue;
 ST_FUNC void pe_add_unwind_data(unsigned start, unsigned end, unsigned stack)
 	    case 2:
-		dllref =
+    TCCState *s1 = tcc_state;
+    Section *pd;
-		    tcc_malloc(sizeof(DLLReference) + strlen(dllname));
+    unsigned o, n, d;
+    struct /* _RUNTIME_FUNCTION */ {
+      DWORD BeginAddress;
-		strcpy(dllref->name, dllname);
+      DWORD EndAddress;
-		dllref->level = 0;
+      DWORD UnwindData;
+    } *p;
-		dynarray_add((void ***) &s1->loaded_dlls,
+    d = pe_add_uwwind_info(s1);
+    pd = s1->uw_pdata;
+    o = pd->data_offset;
+    p = section_ptr_add(pd, sizeof *p);
+    /* record this function */
+    p->BeginAddress = start;
+    p->EndAddress = end;
+    p->UnwindData = d;
+    /* put relocations on it */
+    for (n = o + sizeof *p; o < n; o += sizeof p->BeginAddress)
+        put_elf_reloc(symtab_section, pd, o,  R_X86_64_RELATIVE, s1->uw_sym);
+}
-			     &s1->nb_loaded_dlls, dllref);
+#endif
+/* ------------------------------------------------------------- */
+#ifdef TCC_TARGET_X86_64
+#define PE_STDSYM(n,s) n
+#else
+#define PE_STDSYM(n,s) "_" n s
+#endif
-		++f;
+static void pe_add_runtime(TCCState *s1, struct pe_info *pe)
+{
+    const char *start_symbol;
+    int pe_type = 0;
-	    default:
-		/* tccpe needs to know from what dll it should import
+    if (find_elf_sym(symtab_section, PE_STDSYM("WinMain","@16")))
+        pe_type = PE_GUI;
+    else
+    if (TCC_OUTPUT_DLL == s1->output_type) {
+        pe_type = PE_DLL;
-                   the sym */
+        /* need this for 'tccelf.c:relocate_section()' */
+        s1->output_type = TCC_OUTPUT_EXE;
+    }
+    else
+        pe_type = PE_EXE;
+    start_symbol =
+        TCC_OUTPUT_MEMORY == s1->output_type
+        ? PE_GUI == pe_type ? "__runwinmain" : "_main"
+        : PE_DLL == pe_type ? PE_STDSYM("__dllstart","@12")
+        : PE_GUI == pe_type ? "__winstart" : "__start"
+        ;
+    if (!s1->leading_underscore || strchr(start_symbol, '@'))
+        ++start_symbol;
+    /* grab the startup code from libtcc1 */
+    if (TCC_OUTPUT_MEMORY != s1->output_type || PE_GUI == pe_type)
+        add_elf_sym(symtab_section,
+, 0,
+            ELFW(ST_INFO)(STB_GLOBAL, STT_NOTYPE), 0,
+            SHN_UNDEF, start_symbol);
+    tcc_add_pragma_libs(s1);
-		sym_index = add_elf_sym(s1->dynsymtab_section,
+    if (0 == s1->nostdlib) {
+        static const char *libs[] = {
+            "tcc1", "msvcrt", "kernel32", "", "user32", "gdi32", NULL
+        };
+        const char **pp, *p;
+        for (pp = libs; 0 != (p = *pp); ++pp) {
+            if (0 == *p) {
-, 0, ELF32_ST_INFO(STB_GLOBAL,
+                if (PE_DLL != pe_type && PE_GUI != pe_type)
+                    break;
-							    STT_FUNC),
+            } else if (tcc_add_library_err(s1, p) < 0) {
-                                        s1->nb_loaded_dlls - 1,
-					text_section->sh_num, p);
-		continue;
-	    }
-    }
-    return 0;
-}
+                break;
-/* ------------------------------------------------------------- */
-void pe_guess_outfile(char *objfilename, int output_type)
-{
     char *ext = strrchr(objfilename, '.');
     if (NULL == ext)
         //ext = strchr(objfilename, 0);
-    if (output_type == TCC_OUTPUT_DLL)

Subversion Repositories Kolibri OS

(root)/programs/develop/ktcc/trunk/source/tccpe.c – Rev 647 → 6429