commit ee981ac8996577d688e639bfbae9777bc342ca7c
author Sam Clegg <sbc@chromium.org> Wed Dec 01 18:10:37 2021 -0800
committer Copybara-Service <copybara-worker@google.com> Thu Dec 02 08:45:58 2021 -0800
tree 67910c0f3f2bfedf685ca38047790494840e9cc0
parent 59a15a6399986a359d16109b202f50b30b5c859e

[lld][WebAssembly] Fix for debug relocations against undefined function symbols

This is very similar to https://reviews.llvm.org/D103557 but applies to
symbols which are undefined at link time rather than compile time.

We already have code that handles symbols which were defined at link
time but dead stripped by `--gc-sections` (See
`test/wasm/debug-removed-fn.ll`). In that case the symbols are not live
(!isLive()).  However, we can also have live symbols (which are
references by the program) but which are undefined at link time and are
imported by the linker.

In the test case here the symbol `undef` is used but is not defined
in the program but is imported by the linker due to the
`--import-undefined` flag.

Fixes: https://github.com/emscripten-core/emscripten/issues/15528

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

GitOrigin-RevId: 6f5c5cbe5f8209de3f4d91de75d7f3a716e9f131

test/wasm/debug-undefined-fs.s

diff --git a/test/wasm/debug-undefined-fs.s b/test/wasm/debug-undefined-fs.s
new file mode 100644
index 0000000..4426e84
--- /dev/null
+++ b/test/wasm/debug-undefined-fs.s
@@ -0,0 +1,37 @@
+# Verify that we can handle R_WASM_FUNCTION_OFFSET relocations against live but
+# undefined symbols.  Test that the .debug_info and .debug_int sections are
+# generated without error
+#
+# Based on llvm/test/MC/WebAssembly/debuginfo-relocs.s
+#
+# RUN: llvm-mc -filetype=obj -triple=wasm32-unknown-unknown -o %t.o %s
+# RUN: wasm-ld --import-undefined %t.o -o %t.wasm
+# RUN: obj2yaml %t.wasm | FileCheck %s
+
+.functype undef () -> ()
+
+bar:
+    .functype bar () -> ()
+    end_function
+
+    .globl _start
+_start:
+    .functype _start () -> ()
+    call bar
+    call undef
+    end_function
+
+.section .debug_int,"",@
+.Ld:
+  .int32 1
+.size .Ld, 4
+
+.section .debug_info,"",@
+    .int32 bar
+    .int32 undef
+    .int32 .Ld
+
+# CHECK:          Name:            .debug_info
+# CHECK-NEXT:     Payload:         02000000FFFFFFFF00000000
+# CHECK:          Name:            .debug_int
+# CHECK-NEXT:     Payload:         '01000000'

wasm/InputFiles.cpp

diff --git a/wasm/InputFiles.cpp b/wasm/InputFiles.cpp
index 3fe8846..e3a7f56 100644
--- a/wasm/InputFiles.cpp
+++ b/wasm/InputFiles.cpp
@@ -197,6 +197,9 @@
     return getTagSymbol(reloc.Index)->getTagIndex();
   case R_WASM_FUNCTION_OFFSET_I32:
   case R_WASM_FUNCTION_OFFSET_I64: {
+    if (isa<UndefinedFunction>(sym)) {
+      return tombstone ? tombstone : reloc.Addend;
+    }
     auto *f = cast<DefinedFunction>(sym);
     return f->function->getOffset(f->function->getFunctionCodeOffset() +
                                   reloc.Addend);