1 //===- ExecutionEngineTest.cpp - Unit tests for ExecutionEngine -----------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "llvm/DerivedTypes.h"
11 #include "llvm/GlobalVariable.h"
12 #include "llvm/LLVMContext.h"
13 #include "llvm/Module.h"
14 #include "llvm/ADT/OwningPtr.h"
15 #include "llvm/ExecutionEngine/Interpreter.h"
16 #include "gtest/gtest.h"
17 
18 using namespace llvm;
19 
20 namespace {
21 
22 class ExecutionEngineTest : public testing::Test {
23 protected:
24   ExecutionEngineTest()
25     : M(new Module("<main>", getGlobalContext())),
26       Engine(EngineBuilder(M).create()) {
27   }
28 
29   virtual void SetUp() {
30     ASSERT_TRUE(Engine.get() != NULL);
31   }
32 
33   GlobalVariable *NewExtGlobal(const Type *T, const Twine &Name) {
34     return new GlobalVariable(*M, T, false,  // Not constant.
35                               GlobalValue::ExternalLinkage, NULL, Name);
36   }
37 
38   Module *const M;
39   const OwningPtr<ExecutionEngine> Engine;
40 };
41 
42 TEST_F(ExecutionEngineTest, ForwardGlobalMapping) {
43   GlobalVariable *G1 = NewExtGlobal(Type::Int32Ty, "Global1");
44   int32_t Mem1 = 3;
45   Engine->addGlobalMapping(G1, &Mem1);
46   EXPECT_EQ(&Mem1, Engine->getPointerToGlobalIfAvailable(G1));
47   int32_t Mem2 = 4;
48   Engine->updateGlobalMapping(G1, &Mem2);
49   EXPECT_EQ(&Mem2, Engine->getPointerToGlobalIfAvailable(G1));
50   Engine->updateGlobalMapping(G1, NULL);
51   EXPECT_EQ(NULL, Engine->getPointerToGlobalIfAvailable(G1));
52   Engine->updateGlobalMapping(G1, &Mem2);
53   EXPECT_EQ(&Mem2, Engine->getPointerToGlobalIfAvailable(G1));
54 
55   GlobalVariable *G2 = NewExtGlobal(Type::Int32Ty, "Global1");
56   EXPECT_EQ(NULL, Engine->getPointerToGlobalIfAvailable(G2))
57     << "The NULL return shouldn't depend on having called"
58     << " updateGlobalMapping(..., NULL)";
59   // Check that update...() can be called before add...().
60   Engine->updateGlobalMapping(G2, &Mem1);
61   EXPECT_EQ(&Mem1, Engine->getPointerToGlobalIfAvailable(G2));
62   EXPECT_EQ(&Mem2, Engine->getPointerToGlobalIfAvailable(G1))
63     << "A second mapping shouldn't affect the first.";
64 }
65 
66 TEST_F(ExecutionEngineTest, ReverseGlobalMapping) {
67   GlobalVariable *G1 = NewExtGlobal(Type::Int32Ty, "Global1");
68 
69   int32_t Mem1 = 3;
70   Engine->addGlobalMapping(G1, &Mem1);
71   EXPECT_EQ(G1, Engine->getGlobalValueAtAddress(&Mem1));
72   int32_t Mem2 = 4;
73   Engine->updateGlobalMapping(G1, &Mem2);
74   EXPECT_EQ(NULL, Engine->getGlobalValueAtAddress(&Mem1));
75   EXPECT_EQ(G1, Engine->getGlobalValueAtAddress(&Mem2));
76 
77   GlobalVariable *G2 = NewExtGlobal(Type::Int32Ty, "Global2");
78   Engine->updateGlobalMapping(G2, &Mem1);
79   EXPECT_EQ(G2, Engine->getGlobalValueAtAddress(&Mem1));
80   EXPECT_EQ(G1, Engine->getGlobalValueAtAddress(&Mem2));
81   Engine->updateGlobalMapping(G1, NULL);
82   EXPECT_EQ(G2, Engine->getGlobalValueAtAddress(&Mem1))
83     << "Removing one mapping doesn't affect a different one.";
84   EXPECT_EQ(NULL, Engine->getGlobalValueAtAddress(&Mem2));
85   Engine->updateGlobalMapping(G2, &Mem2);
86   EXPECT_EQ(NULL, Engine->getGlobalValueAtAddress(&Mem1));
87   EXPECT_EQ(G2, Engine->getGlobalValueAtAddress(&Mem2))
88     << "Once a mapping is removed, we can point another GV at the"
89     << " now-free address.";
90 }
91 
92 }
93