1 files changed, 48 insertions, 0 deletions
diff --git a/unittests/ExecutionEngine/Orc/IndirectionUtilsTest.cpp b/unittests/ExecutionEngine/Orc/IndirectionUtilsTest.cpp
new file mode 100644
index 0000000000000..1a533b05838d9
--- /dev/null
+++ b/unittests/ExecutionEngine/Orc/IndirectionUtilsTest.cpp
@@ -0,0 +1,48 @@
+//===- LazyEmittingLayerTest.cpp - Unit tests for the lazy emitting layer -===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+
+#include "OrcTestCommon.h"
+#include "llvm/ADT/SmallVector.h"
+#include "llvm/ExecutionEngine/Orc/IndirectionUtils.h"
+#include "gtest/gtest.h"
+
+using namespace llvm;
+
+namespace {
+
+TEST(IndirectionUtilsTest, MakeStub) {
+  ModuleBuilder MB(getGlobalContext(), "x86_64-apple-macosx10.10", "");
+  Function *F = MB.createFunctionDecl<void(DummyStruct, DummyStruct)>(MB.getModule(), "");
+  SmallVector<AttributeSet, 4> Attrs;
+  Attrs.push_back(
+    AttributeSet::get(MB.getModule()->getContext(), 1U,
+                      AttrBuilder().addAttribute(Attribute::StructRet)));
+  Attrs.push_back(
+    AttributeSet::get(MB.getModule()->getContext(), 2U,
+                      AttrBuilder().addAttribute(Attribute::ByVal)));
+  Attrs.push_back(
+    AttributeSet::get(MB.getModule()->getContext(), ~0U,
+                      AttrBuilder().addAttribute(Attribute::NoUnwind)));
+  F->setAttributes(AttributeSet::get(MB.getModule()->getContext(), Attrs));
+
+  auto ImplPtr = orc::createImplPointer(*F->getType(), *MB.getModule(), "", nullptr);
+  orc::makeStub(*F, *ImplPtr);
+
+  auto II = F->getEntryBlock().begin();
+  EXPECT_TRUE(isa<LoadInst>(*II)) << "First instruction of stub should be a load.";
+  auto *Call = dyn_cast<CallInst>(std::next(II));
+  EXPECT_TRUE(Call != nullptr) << "Second instruction of stub should be a call.";
+  EXPECT_TRUE(Call->isTailCall()) << "Indirect call from stub should be tail call.";
+  EXPECT_TRUE(Call->hasStructRetAttr())
+    << "makeStub should propagate sret attr on 1st argument.";
+  EXPECT_TRUE(Call->paramHasAttr(2U, Attribute::ByVal))
+    << "makeStub should propagate byval attr on 2nd argument.";
+}
+
+}