diff --git a/clang/include/clang/Basic/Attr.td b/clang/include/clang/Basic/Attr.td --- a/clang/include/clang/Basic/Attr.td +++ b/clang/include/clang/Basic/Attr.td @@ -1979,6 +1979,13 @@ let Subjects = SubjectList<[Function], ErrorDiag>; } +def WebAssemblyAsync : InheritableAttr, + TargetSpecificAttr { + let Spellings = [Clang<"wasm_async">]; + let Documentation = [WebAssemblyAsyncDocs]; + let Subjects = SubjectList<[Function], ErrorDiag>; +} + def NoSplitStack : InheritableAttr { let Spellings = [GCC<"no_split_stack">]; let Subjects = SubjectList<[Function], ErrorDiag>; diff --git a/clang/include/clang/Basic/AttrDocs.td b/clang/include/clang/Basic/AttrDocs.td --- a/clang/include/clang/Basic/AttrDocs.td +++ b/clang/include/clang/Basic/AttrDocs.td @@ -5602,6 +5602,17 @@ }]; } +def WebAssemblyAsyncDocs : Documentation { + let Category = DocCatFunction; + let Content = [{ +Clang supports the ``__attribute__((wasm_async))`` +attribute for the WebAssembly target. This attribute may be attached to a +function definition, which indicates the function will be used with JavaScript +promise integration (JSPI). The attribute will cause the creation of a custom +section named "async" that contains each wasm_async function's index value. + }]; +} + def ArtificialDocs : Documentation { let Category = DocCatFunction; let Content = [{ diff --git a/clang/lib/CodeGen/TargetInfo.cpp b/clang/lib/CodeGen/TargetInfo.cpp --- a/clang/lib/CodeGen/TargetInfo.cpp +++ b/clang/lib/CodeGen/TargetInfo.cpp @@ -894,6 +894,12 @@ B.addAttribute("wasm-export-name", Attr->getExportName()); Fn->addFnAttrs(B); } + if (FD->getAttr()) { + llvm::Function *Fn = cast(GV); + llvm::AttrBuilder B(GV->getContext()); + B.addAttribute("wasm-async"); + Fn->addFnAttrs(B); + } } if (auto *FD = dyn_cast_or_null(D)) { diff --git a/clang/lib/Sema/SemaDeclAttr.cpp b/clang/lib/Sema/SemaDeclAttr.cpp --- a/clang/lib/Sema/SemaDeclAttr.cpp +++ b/clang/lib/Sema/SemaDeclAttr.cpp @@ -7633,6 +7633,23 @@ FD->addAttr(::new (S.Context) WebAssemblyImportNameAttr(S.Context, AL, Str)); } +static void handleWebAssemblyAsyncAttr(Sema &S, Decl *D, const ParsedAttr &AL) { + if (!isFunctionOrMethod(D)) { + S.Diag(D->getLocation(), diag::warn_attribute_wrong_decl_type) + << "'wasm_async'" << ExpectedFunction; + return; + } + + auto *FD = cast(D); + if (FD->isThisDeclarationADefinition()) { + S.Diag(D->getLocation(), diag::err_alias_is_definition) << FD << 0; + return; + } + + D->addAttr(::new (S.Context) WebAssemblyAsyncAttr(S.Context, AL)); + D->addAttr(UsedAttr::CreateImplicit(S.Context)); +} + static void handleRISCVInterruptAttr(Sema &S, Decl *D, const ParsedAttr &AL) { // Warn about repeated attributes. @@ -8739,6 +8756,9 @@ case ParsedAttr::AT_WebAssemblyImportName: handleWebAssemblyImportNameAttr(S, D, AL); break; + case ParsedAttr::AT_WebAssemblyAsync: + handleWebAssemblyAsyncAttr(S, D, AL); + break; case ParsedAttr::AT_IBOutlet: handleIBOutlet(S, D, AL); break; diff --git a/clang/test/CodeGen/WebAssembly/wasm-async.c b/clang/test/CodeGen/WebAssembly/wasm-async.c new file mode 100644 --- /dev/null +++ b/clang/test/CodeGen/WebAssembly/wasm-async.c @@ -0,0 +1,9 @@ +// RUN: %clang_cc1 -triple wasm32-unknown-unknown-wasm -emit-llvm -o - %s | FileCheck %s + +__attribute__((wasm_async)) int foo(void) { + return 43; +} + +// CHECK: define i32 @foo() [[A:#[0-9]+]] + +// CHECK: attributes [[A]] = {{{.*}} "wasm-async" {{.*}}} diff --git a/clang/test/Misc/pragma-attribute-supported-attributes-list.test b/clang/test/Misc/pragma-attribute-supported-attributes-list.test --- a/clang/test/Misc/pragma-attribute-supported-attributes-list.test +++ b/clang/test/Misc/pragma-attribute-supported-attributes-list.test @@ -196,6 +196,7 @@ // CHECK-NEXT: WarnUnusedResult (SubjectMatchRule_objc_method, SubjectMatchRule_enum, SubjectMatchRule_record, SubjectMatchRule_hasType_functionType, SubjectMatchRule_type_alias) // CHECK-NEXT: Weak (SubjectMatchRule_variable, SubjectMatchRule_function, SubjectMatchRule_record) // CHECK-NEXT: WeakRef (SubjectMatchRule_variable, SubjectMatchRule_function) +// CHECK-NEXT: WebAssemblyAsync (SubjectMatchRule_function) // CHECK-NEXT: WebAssemblyExportName (SubjectMatchRule_function) // CHECK-NEXT: WebAssemblyImportModule (SubjectMatchRule_function) // CHECK-NEXT: WebAssemblyImportName (SubjectMatchRule_function) diff --git a/lld/test/wasm/async.ll b/lld/test/wasm/async.ll new file mode 100644 --- /dev/null +++ b/lld/test/wasm/async.ll @@ -0,0 +1,25 @@ +; RUN: llc -filetype=obj %s -o %t.o +; RUN: wasm-ld -o %t.wasm %t.o +; RUN: obj2yaml %t.wasm | FileCheck %s + +target triple = "wasm32-unknown-unknown" + +define void @foo() #0 { + ret void +} + +define void @bar() #0 { + ret void +} + +define void @_start() { + call void @foo() + call void @bar() + ret void +} + +attributes #0 = { "wasm-async" } + +; CHECK: - Type: CUSTOM +; CHECK-NEXT: Name: async +; CHECK-NEXT: Payload: '0000000001000000' diff --git a/lld/test/wasm/merge-async-section.ll b/lld/test/wasm/merge-async-section.ll new file mode 100644 --- /dev/null +++ b/lld/test/wasm/merge-async-section.ll @@ -0,0 +1,48 @@ +; RUN: split-file %s %t +; RUN: llc -filetype=obj %t/a.ll -o %t1.o +; RUN: llc -filetype=obj %t/b.ll -o %t2.o +; RUN: wasm-ld -o %t.wasm %t1.o %t2.o +; RUN: obj2yaml %t.wasm | FileCheck %s + +; Ensure two async sections are concatenated together. + +; CHECK: - Type: CUSTOM +; CHECK-NEXT: Name: async +; CHECK-NEXT: Payload: '000000000100000003000000' + +;--- a.ll + +target triple = "wasm32-unknown-unknown" + +; Function index 3 (after linking) +declare void @baz() + +; Function index 0 +define void @foo() #0 { + ret void +} + +; Function index 1 +define void @bar() #0 { + ret void +} + +; Function index 2 +define void @_start() { + call void @foo() + call void @bar() + call void @baz() + ret void +} + +attributes #0 = { "wasm-async" } + +;--- b.ll + +target triple = "wasm32-unknown-unknown" + +define void @baz() #0 { + ret void +} + +attributes #0 = { "wasm-async" } diff --git a/lld/wasm/InputChunks.cpp b/lld/wasm/InputChunks.cpp --- a/lld/wasm/InputChunks.cpp +++ b/lld/wasm/InputChunks.cpp @@ -150,6 +150,7 @@ case R_WASM_TABLE_INDEX_I32: case R_WASM_MEMORY_ADDR_I32: case R_WASM_FUNCTION_OFFSET_I32: + case R_WASM_FUNCTION_INDEX_I32: case R_WASM_SECTION_OFFSET_I32: case R_WASM_GLOBAL_INDEX_I32: case R_WASM_MEMORY_ADDR_LOCREL_I32: diff --git a/lld/wasm/InputFiles.cpp b/lld/wasm/InputFiles.cpp --- a/lld/wasm/InputFiles.cpp +++ b/lld/wasm/InputFiles.cpp @@ -193,6 +193,7 @@ case R_WASM_TYPE_INDEX_LEB: return typeMap[reloc.Index]; case R_WASM_FUNCTION_INDEX_LEB: + case R_WASM_FUNCTION_INDEX_I32: return getFunctionSymbol(reloc.Index)->getFunctionIndex(); case R_WASM_GLOBAL_INDEX_LEB: case R_WASM_GLOBAL_INDEX_I32: diff --git a/llvm/include/llvm/BinaryFormat/WasmRelocs.def b/llvm/include/llvm/BinaryFormat/WasmRelocs.def --- a/llvm/include/llvm/BinaryFormat/WasmRelocs.def +++ b/llvm/include/llvm/BinaryFormat/WasmRelocs.def @@ -28,3 +28,4 @@ WASM_RELOC(R_WASM_MEMORY_ADDR_LOCREL_I32, 23) WASM_RELOC(R_WASM_TABLE_INDEX_REL_SLEB64, 24) WASM_RELOC(R_WASM_MEMORY_ADDR_TLS_SLEB64, 25) +WASM_RELOC(R_WASM_FUNCTION_INDEX_I32, 26) diff --git a/llvm/include/llvm/MC/MCExpr.h b/llvm/include/llvm/MC/MCExpr.h --- a/llvm/include/llvm/MC/MCExpr.h +++ b/llvm/include/llvm/MC/MCExpr.h @@ -330,6 +330,7 @@ VK_WASM_MBREL, // Memory address relative to __memory_base VK_WASM_TBREL, // Table index relative to __table_base VK_WASM_GOT_TLS, // Wasm global index of TLS symbol. + VK_WASM_FUNCINDEX, // Wasm function index. VK_AMDGPU_GOTPCREL32_LO, // symbol@gotpcrel32@lo VK_AMDGPU_GOTPCREL32_HI, // symbol@gotpcrel32@hi diff --git a/llvm/lib/MC/MCExpr.cpp b/llvm/lib/MC/MCExpr.cpp --- a/llvm/lib/MC/MCExpr.cpp +++ b/llvm/lib/MC/MCExpr.cpp @@ -360,6 +360,7 @@ case VK_WASM_TLSREL: return "TLSREL"; case VK_WASM_TBREL: return "TBREL"; case VK_WASM_GOT_TLS: return "GOT@TLS"; + case VK_WASM_FUNCINDEX: return "FUNCINDEX"; case VK_AMDGPU_GOTPCREL32_LO: return "gotpcrel32@lo"; case VK_AMDGPU_GOTPCREL32_HI: return "gotpcrel32@hi"; case VK_AMDGPU_REL32_LO: return "rel32@lo"; @@ -503,6 +504,7 @@ .Case("mbrel", VK_WASM_MBREL) .Case("tlsrel", VK_WASM_TLSREL) .Case("got@tls", VK_WASM_GOT_TLS) + .Case("funcindex", VK_WASM_FUNCINDEX) .Case("gotpcrel32@lo", VK_AMDGPU_GOTPCREL32_LO) .Case("gotpcrel32@hi", VK_AMDGPU_GOTPCREL32_HI) .Case("rel32@lo", VK_AMDGPU_REL32_LO) diff --git a/llvm/lib/MC/WasmObjectWriter.cpp b/llvm/lib/MC/WasmObjectWriter.cpp --- a/llvm/lib/MC/WasmObjectWriter.cpp +++ b/llvm/lib/MC/WasmObjectWriter.cpp @@ -671,6 +671,7 @@ // Provisional value is same as the index return getRelocationIndexValue(RelEntry); case wasm::R_WASM_FUNCTION_INDEX_LEB: + case wasm::R_WASM_FUNCTION_INDEX_I32: case wasm::R_WASM_GLOBAL_INDEX_LEB: case wasm::R_WASM_GLOBAL_INDEX_I32: case wasm::R_WASM_TAG_INDEX_LEB: @@ -791,6 +792,7 @@ case wasm::R_WASM_TABLE_INDEX_I32: case wasm::R_WASM_MEMORY_ADDR_I32: case wasm::R_WASM_FUNCTION_OFFSET_I32: + case wasm::R_WASM_FUNCTION_INDEX_I32: case wasm::R_WASM_SECTION_OFFSET_I32: case wasm::R_WASM_GLOBAL_INDEX_I32: case wasm::R_WASM_MEMORY_ADDR_LOCREL_I32: diff --git a/llvm/lib/Object/WasmObjectFile.cpp b/llvm/lib/Object/WasmObjectFile.cpp --- a/llvm/lib/Object/WasmObjectFile.cpp +++ b/llvm/lib/Object/WasmObjectFile.cpp @@ -948,6 +948,7 @@ Reloc.Index = readVaruint32(Ctx); switch (type) { case wasm::R_WASM_FUNCTION_INDEX_LEB: + case wasm::R_WASM_FUNCTION_INDEX_I32: case wasm::R_WASM_TABLE_INDEX_SLEB: case wasm::R_WASM_TABLE_INDEX_SLEB64: case wasm::R_WASM_TABLE_INDEX_I32: @@ -1045,6 +1046,7 @@ Reloc.Type == wasm::R_WASM_MEMORY_ADDR_LOCREL_I32 || Reloc.Type == wasm::R_WASM_SECTION_OFFSET_I32 || Reloc.Type == wasm::R_WASM_FUNCTION_OFFSET_I32 || + Reloc.Type == wasm::R_WASM_FUNCTION_INDEX_I32 || Reloc.Type == wasm::R_WASM_GLOBAL_INDEX_I32) Size = 4; if (Reloc.Type == wasm::R_WASM_TABLE_INDEX_I64 || diff --git a/llvm/lib/Target/WebAssembly/MCTargetDesc/WebAssemblyWasmObjectWriter.cpp b/llvm/lib/Target/WebAssembly/MCTargetDesc/WebAssemblyWasmObjectWriter.cpp --- a/llvm/lib/Target/WebAssembly/MCTargetDesc/WebAssemblyWasmObjectWriter.cpp +++ b/llvm/lib/Target/WebAssembly/MCTargetDesc/WebAssemblyWasmObjectWriter.cpp @@ -91,6 +91,8 @@ return wasm::R_WASM_TYPE_INDEX_LEB; case MCSymbolRefExpr::VK_None: break; + case MCSymbolRefExpr::VK_WASM_FUNCINDEX: + return wasm::R_WASM_FUNCTION_INDEX_I32; default: report_fatal_error("unknown VariantKind"); break; diff --git a/llvm/lib/Target/WebAssembly/WebAssemblyAsmPrinter.h b/llvm/lib/Target/WebAssembly/WebAssemblyAsmPrinter.h --- a/llvm/lib/Target/WebAssembly/WebAssemblyAsmPrinter.h +++ b/llvm/lib/Target/WebAssembly/WebAssemblyAsmPrinter.h @@ -66,6 +66,7 @@ void emitEndOfAsmFile(Module &M) override; void EmitProducerInfo(Module &M); void EmitTargetFeatures(Module &M); + void EmitAsync(Module &M); void emitSymbolType(const MCSymbolWasm *Sym); void emitGlobalVariable(const GlobalVariable *GV) override; void emitJumpTableInfo() override; diff --git a/llvm/lib/Target/WebAssembly/WebAssemblyAsmPrinter.cpp b/llvm/lib/Target/WebAssembly/WebAssemblyAsmPrinter.cpp --- a/llvm/lib/Target/WebAssembly/WebAssemblyAsmPrinter.cpp +++ b/llvm/lib/Target/WebAssembly/WebAssemblyAsmPrinter.cpp @@ -438,6 +438,7 @@ EmitProducerInfo(M); EmitTargetFeatures(M); + EmitAsync(M); } void WebAssemblyAsmPrinter::EmitProducerInfo(Module &M) { @@ -556,6 +557,33 @@ OutStreamer->popSection(); } +void WebAssemblyAsmPrinter::EmitAsync(Module &M) { + SmallVector EmittedAsyncs; + + for (const auto &F : M) { + auto *Sym = cast(getSymbol(&F)); + if (F.hasFnAttribute("wasm-async")) { + EmittedAsyncs.push_back(Sym); + } + } + if (EmittedAsyncs.size() == 0) + return; + + MCSectionWasm *AsyncSection = OutContext.getWasmSection( + ".custom_section.async", SectionKind::getMetadata()); + OutStreamer->pushSection(); + OutStreamer->switchSection(AsyncSection); + + for (auto &Sym : EmittedAsyncs) { + OutStreamer->emitValue( + MCSymbolRefExpr::create(Sym, MCSymbolRefExpr::VK_WASM_FUNCINDEX, + OutContext), + 4); + } + + OutStreamer->popSection(); +} + void WebAssemblyAsmPrinter::emitConstantPool() { emitDecls(*MMI->getModule()); assert(MF->getConstantPool()->getConstants().empty() && diff --git a/llvm/test/CodeGen/WebAssembly/async.ll b/llvm/test/CodeGen/WebAssembly/async.ll new file mode 100644 --- /dev/null +++ b/llvm/test/CodeGen/WebAssembly/async.ll @@ -0,0 +1,17 @@ +; RUN: llc < %s -asm-verbose=false -wasm-keep-registers | FileCheck %s + +target triple = "wasm32-unknown-unknown" + +define void @test0() #0 { + ret void +} + +define void @test1() #0 { + ret void +} + +attributes #0 = { "wasm-async" } + +; CHECK: .section .custom_section.async,"",@ +; CHECK-NEXT: .int32 test0@FUNCINDEX +; CHECK-NEXT: .int32 test1@FUNCINDEX diff --git a/llvm/test/MC/WebAssembly/async.s b/llvm/test/MC/WebAssembly/async.s new file mode 100644 --- /dev/null +++ b/llvm/test/MC/WebAssembly/async.s @@ -0,0 +1,21 @@ +# RUN: llvm-mc -triple=wasm32-unknown-unknown < %s | FileCheck %s +# Check that it also comiled to object for format. +# RUN: llvm-mc -triple=wasm32-unknown-unknown -filetype=obj -o - < %s | obj2yaml | FileCheck -check-prefix=CHECK-OBJ %s + +foo: + .globl foo + .functype foo () -> () + end_function + +.section .custom_section.async,"",@ + .int32 foo@FUNCINDEX + +# CHECK: .section .custom_section.async,"",@ +# CHECK-NEXT: .int32 foo@FUNCINDEX + +# CHECK-OBJ: - Type: CUSTOM +# CHECK-OBJ-NEXT: Relocations: +# CHECK-OBJ-NEXT: - Type: R_WASM_FUNCTION_INDEX_I32 +# CHECK-OBJ-NEXT: Index: 0 +# CHECK-OBJ-NEXT: Offset: 0x0 +# CHECK-OBJ-NEXT: Name: async