[lld] r375077 - [lld][WebAssembly] Fix for weak references to data symbols in archives

[Sam Clegg via llvm-commits]

Author: sbc
Date: Wed Oct 16 20:21:02 2019
New Revision: 375077

URL: http://llvm.org/viewvc/llvm-project?rev=375077&view=rev
Log:
[lld][WebAssembly] Fix for weak references to data symbols in archives

Fix a bug where were not handling relocations against weakly undefined
data symbol.  Add a test for this case.  Also ensure that the weak
references to data symbols are not pulled in from archive files by
default (but are if `-u <name>` is added to the command line).

Fixes: PR43696

Differential Revision: https://reviews.llvm.org/D69073

Modified:
    lld/trunk/test/wasm/archive-weak-undefined.ll
    lld/trunk/wasm/InputFiles.cpp

Modified: lld/trunk/test/wasm/archive-weak-undefined.ll
URL: http://llvm.org/viewvc/llvm-project/lld/trunk/test/wasm/archive-weak-undefined.ll?rev=375077&r1=375076&r2=375077&view=diff
==============================================================================
--- lld/trunk/test/wasm/archive-weak-undefined.ll (original)
+++ lld/trunk/test/wasm/archive-weak-undefined.ll Wed Oct 16 20:21:02 2019
@@ -1,19 +1,39 @@
+; Test that weak undefined symbols do not fetch members from archive files.
 ; RUN: llc -filetype=obj %s -o %t.o
-; RUN: llc -filetype=obj %S/Inputs/ret32.ll -o %t.a1.o
+; RUN: llc -filetype=obj %S/Inputs/ret32.ll -o %t.ret32.o
+; RUN: llc -filetype=obj %S/Inputs/hello.ll -o %t.hello.o
 ; RUN: rm -f %t.a
-; RUN: llvm-ar rcs %t.a %t.a1.o
+; RUN: llvm-ar rcs %t.a %t.ret32.o %t.hello.o
+
 ; RUN: wasm-ld %t.o %t.a -o %t.wasm
 ; RUN: obj2yaml %t.wasm | FileCheck %s
 
+; RUN: wasm-ld -u hello_str %t.o %t.a -o %t2.wasm
+; RUN: obj2yaml %t2.wasm | FileCheck %s -check-prefix=CHECK-DATA
+
 target triple = "wasm32-unknown-unknown"
 
+; Weak external function symbol
 declare extern_weak i32 @ret32()
 
+; Weak external data symbol
+ at hello_str = extern_weak global i8*, align 4
+
 define void @_start() {
-entry:
+  br i1 icmp ne (i8** @hello_str, i8** null), label %if.then, label %if.end
+
+if.then:
   %call1 = call i32 @ret32()
+  br label %if.end
+
+if.end:
   ret void
 }
 
+; Ensure we have no data section.  If we do, would mean that hello_str was
+; pulled out of the library.
+; CHECK-NOT:  Type:            DATA
+; CHECK-DATA: Type:            DATA
+
 ; CHECK: Name: 'undefined:ret32'
 ; CHECK-NOT: Name: ret32

Modified: lld/trunk/wasm/InputFiles.cpp
URL: http://llvm.org/viewvc/llvm-project/lld/trunk/wasm/InputFiles.cpp?rev=375077&r1=375076&r2=375077&view=diff
==============================================================================
--- lld/trunk/wasm/InputFiles.cpp (original)
+++ lld/trunk/wasm/InputFiles.cpp Wed Oct 16 20:21:02 2019
@@ -189,7 +189,7 @@ uint32_t ObjFile::calcNewValue(const Was
   case R_WASM_MEMORY_ADDR_I32:
   case R_WASM_MEMORY_ADDR_LEB:
   case R_WASM_MEMORY_ADDR_REL_SLEB:
-    if (isa<UndefinedData>(sym))
+    if (isa<UndefinedData>(sym) || sym->isUndefWeak())
       return 0;
     return cast<DefinedData>(sym)->getVirtualAddress() + reloc.Addend;
   case R_WASM_TYPE_INDEX_LEB: