SemaWasm.cpp

Go to the documentation of this file.

//===------ SemaWasm.cpp ---- WebAssembly target-specific routines --------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
//  This file implements semantic analysis functions specific to WebAssembly.
//
//===----------------------------------------------------------------------===//
 
#include "clang/Sema/SemaWasm.h"
#include "clang/AST/ASTContext.h"
#include "clang/AST/Decl.h"
#include "clang/AST/Type.h"
#include "clang/Basic/AddressSpaces.h"
#include "clang/Basic/DiagnosticSema.h"
#include "clang/Basic/TargetBuiltins.h"
#include "clang/Basic/TargetInfo.h"
#include "clang/Sema/Attr.h"
#include "clang/Sema/Sema.h"
 
namespace clang {
 
SemaWasm::SemaWasm(Sema &S) : SemaBase(S) {}
 
/// Checks the argument at the given index is a WebAssembly table and if it
/// is, sets ElTy to the element type.
static bool CheckWasmBuiltinArgIsTable(Sema &S, CallExpr *E, unsigned ArgIndex,
                                       QualType &ElTy) {
  Expr *ArgExpr = E->getArg(ArgIndex);
  const auto *ATy = dyn_cast<ArrayType>(ArgExpr->getType());
  if (!ATy || !ATy->getElementType().isWebAssemblyReferenceType()) {
    return S.Diag(ArgExpr->getBeginLoc(),
                  diag::err_wasm_builtin_arg_must_be_table_type)
           << ArgIndex + 1 << ArgExpr->getSourceRange();
  }
  ElTy = ATy->getElementType();
  return false;
}
 
static bool CheckWasmTableElement(Sema &S, QualType &ElTy, CallExpr *E,
                                  unsigned TableIndex, unsigned ArgIndex) {
  Expr *NewElemArg = E->getArg(ArgIndex);
  QualType QT = NewElemArg->getType();
  // Compare the types after removing insignificant qualifiers
  if (!S.getASTContext().hasSameType(ElTy.getTypePtr(), QT.getTypePtr())) {
    return S.Diag(NewElemArg->getBeginLoc(),
                  diag::err_wasm_builtin_arg_must_match_table_element_type)
           << (ArgIndex + 1) << (TableIndex + 1)
           << NewElemArg->getSourceRange();
  }
  return false;
}
 
/// Checks the argument at the given index is an integer.
static bool CheckWasmBuiltinArgIsInteger(Sema &S, CallExpr *E,
                                         unsigned ArgIndex) {
  Expr *ArgExpr = E->getArg(ArgIndex);
  if (!ArgExpr->getType()->isIntegerType()) {
    return S.Diag(ArgExpr->getBeginLoc(),
                  diag::err_wasm_builtin_arg_must_be_integer_type)
           << ArgIndex + 1 << ArgExpr->getSourceRange();
  }
  return false;
}
 
bool SemaWasm::BuiltinWasmRefNullExtern(CallExpr *TheCall) {
  if (SemaRef.checkArgCount(TheCall, /*DesiredArgCount=*/0))
    return true;
  TheCall->setType(getASTContext().getWebAssemblyExternrefType());
 
  return false;
}
 
bool SemaWasm::BuiltinWasmRefIsNullExtern(CallExpr *TheCall) {
  if (SemaRef.checkArgCount(TheCall, 1)) {
    return true;
  }
 
  Expr *ArgExpr = TheCall->getArg(0);
  if (!ArgExpr->getType().isWebAssemblyExternrefType()) {
    SemaRef.Diag(ArgExpr->getBeginLoc(),
                 diag::err_wasm_builtin_arg_must_be_externref_type)
        << 1 << ArgExpr->getSourceRange();
    return true;
  }
 
  return false;
}
 
bool SemaWasm::BuiltinWasmRefNullFunc(CallExpr *TheCall) {
  ASTContext &Context = getASTContext();
  if (SemaRef.checkArgCount(TheCall, /*DesiredArgCount=*/0))
    return true;
 
  // This custom type checking code ensures that the nodes are as expected
  // in order to later on generate the necessary builtin.
  QualType Pointee = Context.getFunctionType(Context.VoidTy, {}, {});
  QualType Type = Context.getPointerType(Pointee);
  Pointee = Context.getAddrSpaceQualType(Pointee, LangAS::wasm_funcref);
  Type = Context.getAttributedType(attr::WebAssemblyFuncref, Type,
                                   Context.getPointerType(Pointee));
  TheCall->setType(Type);
 
  return false;
}
 
/// Check that the first argument is a WebAssembly table, and the second
/// is an index to use as index into the table.
bool SemaWasm::BuiltinWasmTableGet(CallExpr *TheCall) {
  if (SemaRef.checkArgCount(TheCall, 2))
    return true;
 
  QualType ElTy;
  if (CheckWasmBuiltinArgIsTable(SemaRef, TheCall, 0, ElTy))
    return true;
 
  if (CheckWasmBuiltinArgIsInteger(SemaRef, TheCall, 1))
    return true;
 
  // If all is well, we set the type of TheCall to be the type of the
  // element of the table.
  // i.e. a table.get on an externref table has type externref,
  // or whatever the type of the table element is.
  TheCall->setType(ElTy);
 
  return false;
}
 
/// Check that the first argument is a WebAssembly table, the second is
/// an index to use as index into the table and the third is the reference
/// type to set into the table.
bool SemaWasm::BuiltinWasmTableSet(CallExpr *TheCall) {
  if (SemaRef.checkArgCount(TheCall, 3))
    return true;
 
  QualType ElTy;
  if (CheckWasmBuiltinArgIsTable(SemaRef, TheCall, 0, ElTy))
    return true;
 
  if (CheckWasmBuiltinArgIsInteger(SemaRef, TheCall, 1))
    return true;
 
  if (CheckWasmTableElement(SemaRef, ElTy, TheCall, 0, 2))
    return true;
 
  return false;
}
 
/// Check that the argument is a WebAssembly table.
bool SemaWasm::BuiltinWasmTableSize(CallExpr *TheCall) {
  if (SemaRef.checkArgCount(TheCall, 1))
    return true;
 
  QualType ElTy;
  if (CheckWasmBuiltinArgIsTable(SemaRef, TheCall, 0, ElTy))
    return true;
 
  return false;
}
 
/// Check that the first argument is a WebAssembly table, the second is the
/// value to use for new elements (of a type matching the table type), the
/// third value is an integer.
bool SemaWasm::BuiltinWasmTableGrow(CallExpr *TheCall) {
  if (SemaRef.checkArgCount(TheCall, 3))
    return true;
 
  QualType ElTy;
  if (CheckWasmBuiltinArgIsTable(SemaRef, TheCall, 0, ElTy))
    return true;
 
  if (CheckWasmTableElement(SemaRef, ElTy, TheCall, 0, 1))
    return true;
 
  if (CheckWasmBuiltinArgIsInteger(SemaRef, TheCall, 2))
    return true;
 
  return false;
}
 
/// Check that the first argument is a WebAssembly table, the second is an
/// integer, the third is the value to use to fill the table (of a type
/// matching the table type), and the fourth is an integer.
bool SemaWasm::BuiltinWasmTableFill(CallExpr *TheCall) {
  if (SemaRef.checkArgCount(TheCall, 4))
    return true;
 
  QualType ElTy;
  if (CheckWasmBuiltinArgIsTable(SemaRef, TheCall, 0, ElTy))
    return true;
 
  if (CheckWasmBuiltinArgIsInteger(SemaRef, TheCall, 1))
    return true;
 
  if (CheckWasmTableElement(SemaRef, ElTy, TheCall, 0, 2))
    return true;
 
  if (CheckWasmBuiltinArgIsInteger(SemaRef, TheCall, 3))
    return true;
 
  return false;
}
 
/// Check that the first argument is a WebAssembly table, the second is also a
/// WebAssembly table (of the same element type), and the third to fifth
/// arguments are integers.
bool SemaWasm::BuiltinWasmTableCopy(CallExpr *TheCall) {
  if (SemaRef.checkArgCount(TheCall, 5))
    return true;
 
  QualType XElTy;
  if (CheckWasmBuiltinArgIsTable(SemaRef, TheCall, 0, XElTy))
    return true;
 
  QualType YElTy;
  if (CheckWasmBuiltinArgIsTable(SemaRef, TheCall, 1, YElTy))
    return true;
 
  Expr *TableYArg = TheCall->getArg(1);
  if (!getASTContext().hasSameType(XElTy.getTypePtr(), YElTy.getTypePtr())) {
    return Diag(TableYArg->getBeginLoc(),
                diag::err_wasm_builtin_arg_must_match_table_element_type)
           << 2 << 1 << TableYArg->getSourceRange();
  }
 
  for (int I = 2; I <= 4; I++) {
    if (CheckWasmBuiltinArgIsInteger(SemaRef, TheCall, I))
      return true;
  }
 
  return false;
}
 
bool SemaWasm::BuiltinWasmTestFunctionPointerSignature(const TargetInfo &TI,
                                                       CallExpr *TheCall) {
  if (SemaRef.checkArgCount(TheCall, 1))
    return true;
 
  Expr *FuncPtrArg = TheCall->getArg(0);
  QualType ArgType = FuncPtrArg->getType();
 
  // Check that the argument is a function pointer
  const PointerType *PtrTy = ArgType->getAs<PointerType>();
  if (!PtrTy) {
    return Diag(FuncPtrArg->getBeginLoc(),
                diag::err_typecheck_expect_function_pointer)
           << ArgType << FuncPtrArg->getSourceRange();
  }
 
  const FunctionProtoType *FuncTy =
      PtrTy->getPointeeType()->getAs<FunctionProtoType>();
  if (!FuncTy) {
    return Diag(FuncPtrArg->getBeginLoc(),
                diag::err_typecheck_expect_function_pointer)
           << ArgType << FuncPtrArg->getSourceRange();
  }
 
  if (TI.getABI() == "experimental-mv") {
    auto isStructOrUnion = [](QualType T) {
      return T->isUnionType() || T->isStructureType();
    };
    if (isStructOrUnion(FuncTy->getReturnType())) {
      return Diag(
                 FuncPtrArg->getBeginLoc(),
                 diag::
                     err_wasm_builtin_test_fp_sig_cannot_include_struct_or_union)
             << 0 << FuncTy->getReturnType() << FuncPtrArg->getSourceRange();
    }
    auto NParams = FuncTy->getNumParams();
    for (unsigned I = 0; I < NParams; I++) {
      if (isStructOrUnion(FuncTy->getParamType(I))) {
        return Diag(
                   FuncPtrArg->getBeginLoc(),
                   diag::
                       err_wasm_builtin_test_fp_sig_cannot_include_struct_or_union)
               << 1 << FuncPtrArg->getSourceRange();
      }
    }
  }
 
  // Set return type to int (the result of the test)
  TheCall->setType(getASTContext().IntTy);
  return false;
}
 
bool SemaWasm::CheckWebAssemblyBuiltinFunctionCall(const TargetInfo &TI,
                                                   unsigned BuiltinID,
                                                   CallExpr *TheCall) {
  switch (BuiltinID) {
  case WebAssembly::BI__builtin_wasm_ref_null_extern:
    return BuiltinWasmRefNullExtern(TheCall);
  case WebAssembly::BI__builtin_wasm_ref_null_func:
    return BuiltinWasmRefNullFunc(TheCall);
  case WebAssembly::BI__builtin_wasm_ref_is_null_extern:
    return BuiltinWasmRefIsNullExtern(TheCall);
  case WebAssembly::BI__builtin_wasm_table_get:
    return BuiltinWasmTableGet(TheCall);
  case WebAssembly::BI__builtin_wasm_table_set:
    return BuiltinWasmTableSet(TheCall);
  case WebAssembly::BI__builtin_wasm_table_size:
    return BuiltinWasmTableSize(TheCall);
  case WebAssembly::BI__builtin_wasm_table_grow:
    return BuiltinWasmTableGrow(TheCall);
  case WebAssembly::BI__builtin_wasm_table_fill:
    return BuiltinWasmTableFill(TheCall);
  case WebAssembly::BI__builtin_wasm_table_copy:
    return BuiltinWasmTableCopy(TheCall);
  case WebAssembly::BI__builtin_wasm_test_function_pointer_signature:
    return BuiltinWasmTestFunctionPointerSignature(TI, TheCall);
  }
 
  return false;
}
 
WebAssemblyImportModuleAttr *
SemaWasm::mergeImportModuleAttr(Decl *D,
                                const WebAssemblyImportModuleAttr &AL) {
  if (auto *FD = dyn_cast<FunctionDecl>(D)) {
    if (const auto *ExistingAttr = FD->getAttr<WebAssemblyImportModuleAttr>()) {
      if (ExistingAttr->getImportModule() == AL.getImportModule())
        return nullptr;
      Diag(ExistingAttr->getLocation(), diag::warn_mismatched_import)
          << 0 << ExistingAttr->getImportModule() << AL.getImportModule();
      Diag(AL.getLoc(), diag::note_previous_attribute);
      return nullptr;
    }
    if (FD->hasBody()) {
      Diag(AL.getLoc(), diag::warn_import_on_definition) << 0 << 0;
      return nullptr;
    }
  } else if (auto *VD = dyn_cast<VarDecl>(D)) {
    if (const auto *ExistingAttr = VD->getAttr<WebAssemblyImportModuleAttr>()) {
      if (ExistingAttr->getImportModule() == AL.getImportModule())
        return nullptr;
      Diag(ExistingAttr->getLocation(), diag::warn_mismatched_import)
          << 0 << ExistingAttr->getImportModule() << AL.getImportModule();
      Diag(AL.getLoc(), diag::note_previous_attribute);
      return nullptr;
    }
    if (VD->isThisDeclarationADefinition() != VarDecl::DeclarationOnly) {
      Diag(AL.getLoc(), diag::warn_import_on_definition) << 0 << 1;
      return nullptr;
    }
  }
  return ::new (getASTContext())
      WebAssemblyImportModuleAttr(getASTContext(), AL, AL.getImportModule());
}
 
WebAssemblyImportNameAttr *
SemaWasm::mergeImportNameAttr(Decl *D, const WebAssemblyImportNameAttr &AL) {
  if (auto *FD = dyn_cast<FunctionDecl>(D)) {
    if (const auto *ExistingAttr = FD->getAttr<WebAssemblyImportNameAttr>()) {
      if (ExistingAttr->getImportName() == AL.getImportName())
        return nullptr;
      Diag(ExistingAttr->getLocation(), diag::warn_mismatched_import)
          << 1 << ExistingAttr->getImportName() << AL.getImportName();
      Diag(AL.getLoc(), diag::note_previous_attribute);
      return nullptr;
    }
    if (FD->hasBody()) {
      Diag(AL.getLoc(), diag::warn_import_on_definition) << 1 << 0;
      return nullptr;
    }
  } else if (auto *VD = dyn_cast<VarDecl>(D)) {
    if (const auto *ExistingAttr = VD->getAttr<WebAssemblyImportNameAttr>()) {
      if (ExistingAttr->getImportName() == AL.getImportName())
        return nullptr;
      Diag(ExistingAttr->getLocation(), diag::warn_mismatched_import)
          << 1 << ExistingAttr->getImportName() << AL.getImportName();
      Diag(AL.getLoc(), diag::note_previous_attribute);
      return nullptr;
    }
    if (VD->isThisDeclarationADefinition() != VarDecl::DeclarationOnly) {
      Diag(AL.getLoc(), diag::warn_import_on_definition) << 1 << 1;
      return nullptr;
    }
  }
  return ::new (getASTContext())
      WebAssemblyImportNameAttr(getASTContext(), AL, AL.getImportName());
}
 
WebAssemblyExportNameAttr *
SemaWasm::mergeExportNameAttr(Decl *D, const WebAssemblyExportNameAttr &AL) {
  if (const auto *ExistingAttr = D->getAttr<WebAssemblyExportNameAttr>()) {
    if (ExistingAttr->getExportName() == AL.getExportName())
      return nullptr;
    Diag(ExistingAttr->getLocation(), diag::warn_mismatched_import)
        << 2 << ExistingAttr->getExportName() << AL.getExportName();
    Diag(AL.getLoc(), diag::note_previous_attribute);
    return nullptr;
  }
  return ::new (getASTContext())
      WebAssemblyExportNameAttr(getASTContext(), AL, AL.getExportName());
}
 
void SemaWasm::handleWebAssemblyImportModuleAttr(Decl *D,
                                                 const ParsedAttr &AL) {
  StringRef Str;
  SourceLocation ArgLoc;
  if (!SemaRef.checkStringLiteralArgumentAttr(AL, 0, Str, &ArgLoc))
    return;
 
  if (auto *FD = dyn_cast<FunctionDecl>(D)) {
    if (FD->hasBody()) {
      Diag(AL.getLoc(), diag::warn_import_on_definition) << 0 << 0;
      return;
    }
    FD->addAttr(::new (getASTContext())
                    WebAssemblyImportModuleAttr(getASTContext(), AL, Str));
  } else {
    auto *VD = cast<VarDecl>(D);
    if (VD->isThisDeclarationADefinition() != VarDecl::DeclarationOnly) {
      Diag(AL.getLoc(), diag::warn_import_on_definition) << 0 << 1;
      return;
    }
    VD->addAttr(::new (getASTContext())
                    WebAssemblyImportModuleAttr(getASTContext(), AL, Str));
  }
}
 
void SemaWasm::handleWebAssemblyImportNameAttr(Decl *D, const ParsedAttr &AL) {
  StringRef Str;
  SourceLocation ArgLoc;
  if (!SemaRef.checkStringLiteralArgumentAttr(AL, 0, Str, &ArgLoc))
    return;
 
  if (auto *FD = dyn_cast<FunctionDecl>(D)) {
    if (FD->hasBody()) {
      Diag(AL.getLoc(), diag::warn_import_on_definition) << 1 << 0;
      return;
    }
    FD->addAttr(::new (getASTContext())
                    WebAssemblyImportNameAttr(getASTContext(), AL, Str));
  } else {
    auto *VD = cast<VarDecl>(D);
    if (VD->isThisDeclarationADefinition() != VarDecl::DeclarationOnly) {
      Diag(AL.getLoc(), diag::warn_import_on_definition) << 1 << 1;
      return;
    }
    VD->addAttr(::new (getASTContext())
                    WebAssemblyImportNameAttr(getASTContext(), AL, Str));
  }
}
 
void SemaWasm::handleWebAssemblyExportNameAttr(Decl *D, const ParsedAttr &AL) {
  ASTContext &Context = getASTContext();
 
  StringRef Str;
  SourceLocation ArgLoc;
  if (!SemaRef.checkStringLiteralArgumentAttr(AL, 0, Str, &ArgLoc))
    return;
 
  D->addAttr(::new (Context) WebAssemblyExportNameAttr(Context, AL, Str));
  D->addAttr(UsedAttr::CreateImplicit(Context));
}
 
} // namespace clang