[lld] r243532 - COFF: Set load config table entry on non-x86.

[Rui Ueyama]

Author: ruiu
Date: Wed Jul 29 11:30:31 2015
New Revision: 243532

URL: http://llvm.org/viewvc/llvm-project?rev=243532&view=rev
Log:
COFF: Set load config table entry on non-x86.

Added:
    lld/trunk/test/COFF/loadcfg.test
Modified:
    lld/trunk/COFF/Config.h
    lld/trunk/COFF/Driver.cpp
    lld/trunk/COFF/Writer.cpp

Modified: lld/trunk/COFF/Config.h
URL: http://llvm.org/viewvc/llvm-project/lld/trunk/COFF/Config.h?rev=243532&r1=243531&r2=243532&view=diff
==============================================================================
--- lld/trunk/COFF/Config.h (original)
+++ lld/trunk/COFF/Config.h Wed Jul 29 11:30:31 2015
@@ -79,6 +79,7 @@ struct Configuration {
   std::vector<Export> Exports;
   std::set<std::string> DelayLoads;
   Undefined *DelayLoadHelper = nullptr;
+  StringRef LoadConfigUsed;
 
   // Used for SafeSEH.
   DefinedRelative *SEHTable = nullptr;

Modified: lld/trunk/COFF/Driver.cpp
URL: http://llvm.org/viewvc/llvm-project/lld/trunk/COFF/Driver.cpp?rev=243532&r1=243531&r2=243532&view=diff
==============================================================================
--- lld/trunk/COFF/Driver.cpp (original)
+++ lld/trunk/COFF/Driver.cpp Wed Jul 29 11:30:31 2015
@@ -622,6 +622,7 @@ bool LinkerDriver::link(llvm::ArrayRef<c
     Config->SEHTable = Symtab.addRelative("___safe_se_handler_table", 0);
     Config->SEHCount = Symtab.addAbsolute("___safe_se_handler_count", 0);
   }
+  Config->LoadConfigUsed = mangle("_load_config_used");
 
   // Read as much files as we can from directives sections.
   if (auto EC = Symtab.run()) {
@@ -659,10 +660,9 @@ bool LinkerDriver::link(llvm::ArrayRef<c
     }
 
     // Windows specific -- if __load_config_used can be resolved, resolve it.
-    if (Config->Machine == I386)
-      if (Symbol *Sym = Symtab.find("__load_config_used"))
-        if (isa<Lazy>(Sym->Body))
-          Symtab.addUndefined("__load_config_used");
+    if (Symbol *Sym = Symtab.find(Config->LoadConfigUsed))
+      if (isa<Lazy>(Sym->Body))
+        Symtab.addUndefined(Config->LoadConfigUsed);
 
     if (Symtab.queueEmpty())
       break;

Modified: lld/trunk/COFF/Writer.cpp
URL: http://llvm.org/viewvc/llvm-project/lld/trunk/COFF/Writer.cpp?rev=243532&r1=243531&r2=243532&view=diff
==============================================================================
--- lld/trunk/COFF/Writer.cpp (original)
+++ lld/trunk/COFF/Writer.cpp Wed Jul 29 11:30:31 2015
@@ -526,10 +526,10 @@ template <typename PEHeaderTy> void Writ
       Dir[TLS_TABLE].Size = 40;
     }
   }
-  if (Symbol *Sym = Symtab->findUnderscore("_load_config_used")) {
+  if (Symbol *Sym = Symtab->find(Config->LoadConfigUsed)) {
     if (Defined *B = dyn_cast<Defined>(Sym->Body)) {
       Dir[LOAD_CONFIG_TABLE].RelativeVirtualAddress = B->getRVA();
-      Dir[LOAD_CONFIG_TABLE].Size = 64;
+      Dir[LOAD_CONFIG_TABLE].Size = Config->is64() ? 112 : 64;
     }
   }
 

Added: lld/trunk/test/COFF/loadcfg.test
URL: http://llvm.org/viewvc/llvm-project/lld/trunk/test/COFF/loadcfg.test?rev=243532&view=auto
==============================================================================
--- lld/trunk/test/COFF/loadcfg.test (added)
+++ lld/trunk/test/COFF/loadcfg.test Wed Jul 29 11:30:31 2015
@@ -0,0 +1,42 @@
+# RUN: yaml2obj < %s > %t.obj
+# RUN: lld -flavor link2 /out:%t.exe %t.obj /entry:main /subsystem:console
+# RUN: llvm-readobj -file-headers %t.exe | FileCheck %s
+
+# CHECK: LoadConfigTableRVA: 0x1000
+# CHECK: LoadConfigTableSize: 0x70
+
+---
+header:
+  Machine:         IMAGE_FILE_MACHINE_AMD64
+  Characteristics: []
+sections:
+  - Name:            .text
+    Characteristics: [ IMAGE_SCN_CNT_CODE, IMAGE_SCN_MEM_EXECUTE, IMAGE_SCN_MEM_READ ]
+    Alignment:       4
+    SectionData:     B82A000000C3
+symbols:
+  - Name:            .text
+    Value:           0
+    SectionNumber:   1
+    SimpleType:      IMAGE_SYM_TYPE_NULL
+    ComplexType:     IMAGE_SYM_DTYPE_NULL
+    StorageClass:    IMAGE_SYM_CLASS_STATIC
+    SectionDefinition:
+      Length:          6
+      NumberOfRelocations: 0
+      NumberOfLinenumbers: 0
+      CheckSum:        0
+      Number:          0
+  - Name:            main
+    Value:           0
+    SectionNumber:   1
+    SimpleType:      IMAGE_SYM_TYPE_NULL
+    ComplexType:     IMAGE_SYM_DTYPE_NULL
+    StorageClass:    IMAGE_SYM_CLASS_EXTERNAL
+  - Name:            _load_config_used
+    Value:           0
+    SectionNumber:   1
+    SimpleType:      IMAGE_SYM_TYPE_NULL
+    ComplexType:     IMAGE_SYM_DTYPE_NULL
+    StorageClass:    IMAGE_SYM_CLASS_EXTERNAL
+...