llvm/unittests/ExecutionEngine/Orc/IndirectionUtilsTest.cpp

   1 //===- LazyEmittingLayerTest.cpp - Unit tests for the lazy emitting layer -===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "llvm/ExecutionEngine/Orc/IndirectionUtils.h"
  10 #include "OrcTestCommon.h"
  11 #include "llvm/ADT/SmallVector.h"
  12 #include "gtest/gtest.h"
  13
  14 using namespace llvm;
  15
  16 namespace {
  17
  18 TEST(IndirectionUtilsTest, MakeStub) {
  19   LLVMContext Context;
  20   ModuleBuilder MB(Context, "x86_64-apple-macosx10.10", "");
  21   StructType *ArgTy = getDummyStructTy(Context);
  22   Type *ArgPtrTy = PointerType::getUnqual(ArgTy);
  23   FunctionType *FTy = FunctionType::get(
  24       Type::getVoidTy(Context), {ArgPtrTy, ArgPtrTy}, false);
  25   Function *F = MB.createFunctionDecl(FTy, "");
  26   AttributeSet FnAttrs = AttributeSet::get(
  27       Context, AttrBuilder(Context).addAttribute(Attribute::NoUnwind));
  28   AttributeSet RetAttrs; // None
  29   AttributeSet ArgAttrs[2] = {
  30       AttributeSet::get(Context, AttrBuilder(Context).addStructRetAttr(ArgTy)),
  31       AttributeSet::get(Context, AttrBuilder(Context).addByValAttr(ArgTy)),
  32   };
  33   F->setAttributes(AttributeList::get(Context, FnAttrs, RetAttrs, ArgAttrs));
  34
  35   auto ImplPtr = orc::createImplPointer(*F->getType(), *MB.getModule(), "", nullptr);
  36   orc::makeStub(*F, *ImplPtr);
  37
  38   auto II = F->getEntryBlock().begin();
  39   EXPECT_TRUE(isa<LoadInst>(*II)) << "First instruction of stub should be a load.";
  40   auto *Call = dyn_cast<CallInst>(std::next(II));
  41   EXPECT_TRUE(Call != nullptr) << "Second instruction of stub should be a call.";
  42   EXPECT_TRUE(Call->isTailCall()) << "Indirect call from stub should be tail call.";
  43   EXPECT_TRUE(Call->hasStructRetAttr())
  44     << "makeStub should propagate sret attr on 1st argument.";
  45   EXPECT_TRUE(Call->paramHasAttr(1U, Attribute::ByVal))
  46     << "makeStub should propagate byval attr on 2nd argument.";
  47 }
  48
  49 }