77
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
1 //===- llvm/unittest/IR/UserTest.cpp - User unit tests --------------------===//
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
2 //
|
147
|
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
|
77
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
6 //
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
7 //===----------------------------------------------------------------------===//
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
8
|
121
|
9 #include "llvm/IR/User.h"
|
77
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
10 #include "llvm/AsmParser/Parser.h"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
11 #include "llvm/IR/Function.h"
|
83
|
12 #include "llvm/IR/Instructions.h"
|
77
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
13 #include "llvm/IR/LLVMContext.h"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
14 #include "llvm/IR/Module.h"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
15 #include "llvm/Support/SourceMgr.h"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
16 #include "gtest/gtest.h"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
17 using namespace llvm;
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
18
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
19 namespace {
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
20
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
21 TEST(UserTest, ValueOpIteration) {
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
22 LLVMContext C;
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
23
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
24 const char *ModuleString = "define void @f(i32 %x, i32 %y) {\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
25 "entry:\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
26 " switch i32 undef, label %s0\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
27 " [ i32 1, label %s1\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
28 " i32 2, label %s2\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
29 " i32 3, label %s3\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
30 " i32 4, label %s4\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
31 " i32 5, label %s5\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
32 " i32 6, label %s6\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
33 " i32 7, label %s7\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
34 " i32 8, label %s8\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
35 " i32 9, label %s9 ]\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
36 "\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
37 "s0:\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
38 " br label %exit\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
39 "s1:\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
40 " br label %exit\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
41 "s2:\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
42 " br label %exit\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
43 "s3:\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
44 " br label %exit\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
45 "s4:\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
46 " br label %exit\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
47 "s5:\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
48 " br label %exit\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
49 "s6:\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
50 " br label %exit\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
51 "s7:\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
52 " br label %exit\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
53 "s8:\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
54 " br label %exit\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
55 "s9:\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
56 " br label %exit\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
57 "\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
58 "exit:\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
59 " %phi = phi i32 [ 0, %s0 ], [ 1, %s1 ],\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
60 " [ 2, %s2 ], [ 3, %s3 ],\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
61 " [ 4, %s4 ], [ 5, %s5 ],\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
62 " [ 6, %s6 ], [ 7, %s7 ],\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
63 " [ 8, %s8 ], [ 9, %s9 ]\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
64 " ret void\n"
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
65 "}\n";
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
66 SMDiagnostic Err;
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
67 std::unique_ptr<Module> M = parseAssemblyString(ModuleString, Err, C);
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
68
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
69 Function *F = M->getFunction("f");
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
70 BasicBlock &ExitBB = F->back();
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
71 PHINode &P = cast<PHINode>(ExitBB.front());
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
72 EXPECT_TRUE(P.value_op_begin() == P.value_op_begin());
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
73 EXPECT_FALSE(P.value_op_begin() == P.value_op_end());
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
74 EXPECT_TRUE(P.value_op_begin() != P.value_op_end());
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
75 EXPECT_FALSE(P.value_op_end() != P.value_op_end());
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
76 EXPECT_TRUE(P.value_op_begin() < P.value_op_end());
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
77 EXPECT_FALSE(P.value_op_begin() < P.value_op_begin());
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
78 EXPECT_TRUE(P.value_op_end() > P.value_op_begin());
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
79 EXPECT_FALSE(P.value_op_begin() > P.value_op_begin());
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
80 EXPECT_TRUE(P.value_op_begin() <= P.value_op_begin());
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
81 EXPECT_FALSE(P.value_op_end() <= P.value_op_begin());
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
82 EXPECT_TRUE(P.value_op_begin() >= P.value_op_begin());
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
83 EXPECT_FALSE(P.value_op_begin() >= P.value_op_end());
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
84 EXPECT_EQ(10, std::distance(P.value_op_begin(), P.value_op_end()));
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
85
|
121
|
86 // const value op iteration
|
|
87 const PHINode *IP = &P;
|
|
88 EXPECT_TRUE(IP->value_op_begin() == IP->value_op_begin());
|
|
89 EXPECT_FALSE(IP->value_op_begin() == IP->value_op_end());
|
|
90 EXPECT_TRUE(IP->value_op_begin() != IP->value_op_end());
|
|
91 EXPECT_FALSE(IP->value_op_end() != IP->value_op_end());
|
|
92 EXPECT_TRUE(IP->value_op_begin() < IP->value_op_end());
|
|
93 EXPECT_FALSE(IP->value_op_begin() < IP->value_op_begin());
|
|
94 EXPECT_TRUE(IP->value_op_end() > IP->value_op_begin());
|
|
95 EXPECT_FALSE(IP->value_op_begin() > IP->value_op_begin());
|
|
96 EXPECT_TRUE(IP->value_op_begin() <= IP->value_op_begin());
|
|
97 EXPECT_FALSE(IP->value_op_end() <= IP->value_op_begin());
|
|
98 EXPECT_TRUE(IP->value_op_begin() >= IP->value_op_begin());
|
|
99 EXPECT_FALSE(IP->value_op_begin() >= IP->value_op_end());
|
|
100 EXPECT_EQ(10, std::distance(IP->value_op_begin(), IP->value_op_end()));
|
|
101
|
77
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
102 User::value_op_iterator I = P.value_op_begin();
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
103 I += 3;
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
104 EXPECT_EQ(std::next(P.value_op_begin(), 3), I);
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
105 EXPECT_EQ(P.getOperand(3), *I);
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
106 I++;
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
107 EXPECT_EQ(P.getOperand(6), I[2]);
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
108 EXPECT_EQ(P.value_op_end(), (I - 2) + 8);
|
121
|
109
|
|
110 // const value op
|
|
111 User::const_value_op_iterator CI = IP->value_op_begin();
|
|
112 CI += 3;
|
|
113 EXPECT_EQ(std::next(IP->value_op_begin(), 3), CI);
|
|
114 EXPECT_EQ(IP->getOperand(3), *CI);
|
|
115 CI++;
|
|
116 EXPECT_EQ(IP->getOperand(6), CI[2]);
|
|
117 EXPECT_EQ(IP->value_op_end(), (CI - 2) + 8);
|
77
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
118 }
|
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
119
|
100
|
120 TEST(UserTest, PersonalityUser) {
|
120
|
121 LLVMContext Context;
|
|
122 Module M("", Context);
|
|
123 FunctionType *RetVoidTy = FunctionType::get(Type::getVoidTy(Context), false);
|
100
|
124 Function *PersonalityF = Function::Create(
|
|
125 RetVoidTy, GlobalValue::ExternalLinkage, "PersonalityFn", &M);
|
|
126 Function *TestF =
|
|
127 Function::Create(RetVoidTy, GlobalValue::ExternalLinkage, "TestFn", &M);
|
|
128
|
|
129 // Set up the personality function
|
|
130 TestF->setPersonalityFn(PersonalityF);
|
|
131 auto PersonalityUsers = PersonalityF->user_begin();
|
|
132
|
|
133 // One user and that user is the Test function
|
|
134 EXPECT_EQ(*PersonalityUsers, TestF);
|
|
135 EXPECT_EQ(++PersonalityUsers, PersonalityF->user_end());
|
|
136
|
|
137 // Reset the personality function
|
|
138 TestF->setPersonalityFn(nullptr);
|
|
139
|
|
140 // No users should remain
|
|
141 EXPECT_TRUE(TestF->user_empty());
|
|
142 }
|
|
143
|
77
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff
changeset
|
144 } // end anonymous namespace
|