mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-12-03 08:51:43 +00:00
1cc70943c8
Summary: We need to actually remove the use of the personality function, otherwise we can run into trouble if we want to e.g. delete the personality function because ther's no way to get rid of its uses. Do this by resetting to ConstantPointerNull value that the operands are set to when first allocated. Reviewers: vsk, dexonsmith Subscribers: llvm-commits Differential Revision: http://reviews.llvm.org/D15752 llvm-svn: 256345
121 lines
4.7 KiB
C++
121 lines
4.7 KiB
C++
//===- llvm/unittest/IR/UserTest.cpp - User unit tests --------------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/AsmParser/Parser.h"
|
|
#include "llvm/IR/Function.h"
|
|
#include "llvm/IR/Instructions.h"
|
|
#include "llvm/IR/LLVMContext.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/IR/User.h"
|
|
#include "llvm/Support/SourceMgr.h"
|
|
#include "gtest/gtest.h"
|
|
using namespace llvm;
|
|
|
|
namespace {
|
|
|
|
TEST(UserTest, ValueOpIteration) {
|
|
LLVMContext C;
|
|
|
|
const char *ModuleString = "define void @f(i32 %x, i32 %y) {\n"
|
|
"entry:\n"
|
|
" switch i32 undef, label %s0\n"
|
|
" [ i32 1, label %s1\n"
|
|
" i32 2, label %s2\n"
|
|
" i32 3, label %s3\n"
|
|
" i32 4, label %s4\n"
|
|
" i32 5, label %s5\n"
|
|
" i32 6, label %s6\n"
|
|
" i32 7, label %s7\n"
|
|
" i32 8, label %s8\n"
|
|
" i32 9, label %s9 ]\n"
|
|
"\n"
|
|
"s0:\n"
|
|
" br label %exit\n"
|
|
"s1:\n"
|
|
" br label %exit\n"
|
|
"s2:\n"
|
|
" br label %exit\n"
|
|
"s3:\n"
|
|
" br label %exit\n"
|
|
"s4:\n"
|
|
" br label %exit\n"
|
|
"s5:\n"
|
|
" br label %exit\n"
|
|
"s6:\n"
|
|
" br label %exit\n"
|
|
"s7:\n"
|
|
" br label %exit\n"
|
|
"s8:\n"
|
|
" br label %exit\n"
|
|
"s9:\n"
|
|
" br label %exit\n"
|
|
"\n"
|
|
"exit:\n"
|
|
" %phi = phi i32 [ 0, %s0 ], [ 1, %s1 ],\n"
|
|
" [ 2, %s2 ], [ 3, %s3 ],\n"
|
|
" [ 4, %s4 ], [ 5, %s5 ],\n"
|
|
" [ 6, %s6 ], [ 7, %s7 ],\n"
|
|
" [ 8, %s8 ], [ 9, %s9 ]\n"
|
|
" ret void\n"
|
|
"}\n";
|
|
SMDiagnostic Err;
|
|
std::unique_ptr<Module> M = parseAssemblyString(ModuleString, Err, C);
|
|
|
|
Function *F = M->getFunction("f");
|
|
BasicBlock &ExitBB = F->back();
|
|
PHINode &P = cast<PHINode>(ExitBB.front());
|
|
EXPECT_TRUE(P.value_op_begin() == P.value_op_begin());
|
|
EXPECT_FALSE(P.value_op_begin() == P.value_op_end());
|
|
EXPECT_TRUE(P.value_op_begin() != P.value_op_end());
|
|
EXPECT_FALSE(P.value_op_end() != P.value_op_end());
|
|
EXPECT_TRUE(P.value_op_begin() < P.value_op_end());
|
|
EXPECT_FALSE(P.value_op_begin() < P.value_op_begin());
|
|
EXPECT_TRUE(P.value_op_end() > P.value_op_begin());
|
|
EXPECT_FALSE(P.value_op_begin() > P.value_op_begin());
|
|
EXPECT_TRUE(P.value_op_begin() <= P.value_op_begin());
|
|
EXPECT_FALSE(P.value_op_end() <= P.value_op_begin());
|
|
EXPECT_TRUE(P.value_op_begin() >= P.value_op_begin());
|
|
EXPECT_FALSE(P.value_op_begin() >= P.value_op_end());
|
|
EXPECT_EQ(10, std::distance(P.value_op_begin(), P.value_op_end()));
|
|
|
|
User::value_op_iterator I = P.value_op_begin();
|
|
I += 3;
|
|
EXPECT_EQ(std::next(P.value_op_begin(), 3), I);
|
|
EXPECT_EQ(P.getOperand(3), *I);
|
|
I++;
|
|
EXPECT_EQ(P.getOperand(6), I[2]);
|
|
EXPECT_EQ(P.value_op_end(), (I - 2) + 8);
|
|
}
|
|
|
|
TEST(UserTest, PersonalityUser) {
|
|
Module M("", getGlobalContext());
|
|
FunctionType *RetVoidTy =
|
|
FunctionType::get(Type::getVoidTy(getGlobalContext()), false);
|
|
Function *PersonalityF = Function::Create(
|
|
RetVoidTy, GlobalValue::ExternalLinkage, "PersonalityFn", &M);
|
|
Function *TestF =
|
|
Function::Create(RetVoidTy, GlobalValue::ExternalLinkage, "TestFn", &M);
|
|
|
|
// Set up the personality function
|
|
TestF->setPersonalityFn(PersonalityF);
|
|
auto PersonalityUsers = PersonalityF->user_begin();
|
|
|
|
// One user and that user is the Test function
|
|
EXPECT_EQ(*PersonalityUsers, TestF);
|
|
EXPECT_EQ(++PersonalityUsers, PersonalityF->user_end());
|
|
|
|
// Reset the personality function
|
|
TestF->setPersonalityFn(nullptr);
|
|
|
|
// No users should remain
|
|
EXPECT_TRUE(TestF->user_empty());
|
|
}
|
|
|
|
} // end anonymous namespace
|