[WebAssembly] Improve support for "needed" list in dylink section

[sbc100]

Summary

This change adds basic support for shared library dependencies
via the dylink section.

See https://github.com/WebAssembly/tool-conventions/pull/77

[sbc100]

• move to mono-repo

[ruiu]

Is this ready for review? It doesn't seem to create a valid executable though, as there's no code to emit filenames of shared object files.

[sbc100]

• add test

[sbc100]

• remove assert

[sbc100]

• clang-format

[sbc100]

This change is now ready for review. Please note that this is only the most simple support for .so file. All it does is add each library to the "needed" section, it doesn't do any symbol resolution against the shared libraries yet.

[ruiu]

lld/wasm/Writer.cpp

504 ↗

(On Diff #190211)

Maybe it is better to avoid auto and make it explicit as SharedFile *

505 ↗

(On Diff #190211)

nit: you can omit llvm::.

llvm/lib/Object/WasmObjectFile.cpp

327 ↗

(On Diff #190211)

Is this dead code?

[sbc100]

llvm/lib/Object/WasmObjectFile.cpp

327 ↗

(On Diff #190211)

Its fixing a bug that isSharedLibrary(), which is only ever used by lld, was always returning false.

I guess at least some other tool such as objdump should probably be calling that and displaying different output for shared libs.

[ruiu]

LGTM with the previous comments fixed.

llvm/lib/Object/WasmObjectFile.cpp

327 ↗

(On Diff #190211)

Got it. We eventually need a test, but it doesn't have to be at this time.

test/wasm/shared-needed.ll

+; RUN: llc -O0 -filetype=obj %s -o %t.o
+; RUN: wasm-ld -shared -o %t.so %t.o
+; RUN: obj2yaml %t.so | FileCheck %s
+
+target triple = "wasm32-unknown-unknown"
+
+@data = global i32 2, align 4
+
+define default void @foo() {
+entry:
+  ret void
+}
+
+; CHECK:      Sections:
+; CHECK-NEXT:   - Type:            CUSTOM
+; CHECK-NEXT:     Name:            dylink
+; CHECK-NEXT:     MemorySize:      4
+; CHECK-NEXT:     MemoryAlignment: 2
+; CHECK-NEXT:     TableSize:       0
+; CHECK-NEXT:     TableAlignment:  0
+; CHECK-NEXT:     Needed:          []

wasm/InputFiles.h

 class InputGlobal;
 class InputEvent;
 class InputSection;
 
 class InputFile {
 public:
   enum Kind {
     ObjectKind,
+    SharedKind,
     ArchiveKind,
     BitcodeKind,
   };
 
   virtual ~InputFile() {}
 
   // Returns the filename.
   StringRef getName() const { return MB.getBufferIdentifier(); }
private:
   Symbol *createDefined(const WasmSymbol &Sym);
   Symbol *createUndefined(const WasmSymbol &Sym);
 
   bool isExcludedByComdat(InputChunk *Chunk) const;
 
   std::unique_ptr<WasmObjectFile> WasmObj;
 };
 
+// .so file.
+class SharedFile : public InputFile {
+public:
+  explicit SharedFile(MemoryBufferRef M) : InputFile(SharedKind, M) {}
+  static bool classof(const InputFile *F) { return F->kind() == SharedKind; }
+
+  void parse() override {}
+};
+
+// .bc file
 class BitcodeFile : public InputFile {
 public:
   explicit BitcodeFile(MemoryBufferRef M) : InputFile(BitcodeKind, M) {}
   static bool classof(const InputFile *F) { return F->kind() == BitcodeKind; }
 
   void parse() override;
   std::unique_ptr<llvm::lto::InputFile> Obj;
 };

wasm/InputFiles.cpp

Optional<MemoryBufferRef> lld::wasm::readFile(StringRef Path) {
   MemoryBufferRef MBRef = MB->getMemBufferRef();
   make<std::unique_ptr<MemoryBuffer>>(std::move(MB)); // take MB ownership
 
   return MBRef;
 }
 
 InputFile *lld::wasm::createObjectFile(MemoryBufferRef MB) {
   file_magic Magic = identify_magic(MB.getBuffer());
-  if (Magic == file_magic::wasm_object)
+  if (Magic == file_magic::wasm_object) {
+    std::unique_ptr<Binary> Bin = check(createBinary(MB));
+    auto *Obj = dyn_cast<WasmObjectFile>(Bin.get());
+    assert(Obj);
+    if (Obj->isSharedObject())
+      return make<SharedFile>(MB);
     return make<ObjFile>(MB);
+  }
 
   if (Magic == file_magic::bitcode)
     return make<BitcodeFile>(MB);
 
   fatal("unknown file type: " + MB.getBufferIdentifier());
 }
 
 void ObjFile::dumpInfo() const {

wasm/SymbolTable.h

 // add*() functions, which are called by input files as they are parsed.
 // There is one add* function per symbol type.
 class SymbolTable {
 public:
   void addFile(InputFile *File);
   void addCombinedLTOObject();
 
   std::vector<ObjFile *> ObjectFiles;
+  std::vector<InputFile *> SharedFiles;
   std::vector<BitcodeFile *> BitcodeFiles;
   std::vector<InputFunction *> SyntheticFunctions;
   std::vector<InputGlobal *> SyntheticGlobals;
 
   void reportRemainingUndefines();
 
   ArrayRef<Symbol *> getSymbols() const { return SymVector; }