Mercurial > hg > CbC > CbC_llvm

annotate tools/bugpoint/ExtractFunction.cpp @ 147:c2174574ed3a

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
LLVM 10
author Shinji KONO <kono@ie.u-ryukyu.ac.jp>
date Wed, 14 Aug 2019 16:55:33 +0900
parents 3a76565eade5
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
1 //===- ExtractFunction.cpp - Extract a function from Program --------------===//
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
2 //
147
c2174574ed3a LLVM 10
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 134
diff changeset
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
c2174574ed3a LLVM 10
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 134
diff changeset
4 // See https://llvm.org/LICENSE.txt for license information.
c2174574ed3a LLVM 10
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 134
diff changeset
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
6 //
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
7 //===----------------------------------------------------------------------===//
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
8 //
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
9 // This file implements several methods that are used to extract functions,
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
10 // loops, or portions of a module from the rest of the module.
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
11 //
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
12 //===----------------------------------------------------------------------===//
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
13
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
14 #include "BugDriver.h"
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
15 #include "llvm/IR/Constants.h"
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
16 #include "llvm/IR/DataLayout.h"
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
17 #include "llvm/IR/DerivedTypes.h"
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
18 #include "llvm/IR/LLVMContext.h"
83
60c9769439b8 LLVM 3.7
Tatsuki IHA <e125716@ie.u-ryukyu.ac.jp>
parents: 77
diff changeset
19 #include "llvm/IR/LegacyPassManager.h"
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
20 #include "llvm/IR/Module.h"
77
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
21 #include "llvm/IR/Verifier.h"
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
22 #include "llvm/Pass.h"
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
23 #include "llvm/Support/CommandLine.h"
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
24 #include "llvm/Support/Debug.h"
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
25 #include "llvm/Support/FileUtilities.h"
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
26 #include "llvm/Support/Path.h"
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
27 #include "llvm/Support/Signals.h"
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
28 #include "llvm/Support/ToolOutputFile.h"
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
29 #include "llvm/Transforms/IPO.h"
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
30 #include "llvm/Transforms/Scalar.h"
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
31 #include "llvm/Transforms/Utils/Cloning.h"
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
32 #include "llvm/Transforms/Utils/CodeExtractor.h"
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
33 #include <set>
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
34 using namespace llvm;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
35
77
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
36 #define DEBUG_TYPE "bugpoint"
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
37
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
38 namespace llvm {
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
39 bool DisableSimplifyCFG = false;
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
40 extern cl::opt<std::string> OutputPrefix;
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
41 } // End llvm namespace
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
42
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
43 namespace {
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
44 cl::opt<bool> NoDCE("disable-dce",
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
45 cl::desc("Do not use the -dce pass to reduce testcases"));
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
46 cl::opt<bool, true>
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
47 NoSCFG("disable-simplifycfg", cl::location(DisableSimplifyCFG),
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
48 cl::desc("Do not use the -simplifycfg pass to reduce testcases"));
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
49
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
50 Function *globalInitUsesExternalBA(GlobalVariable *GV) {
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
51 if (!GV->hasInitializer())
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
52 return nullptr;
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
53
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
54 Constant *I = GV->getInitializer();
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
55
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
56 // walk the values used by the initializer
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
57 // (and recurse into things like ConstantExpr)
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
58 std::vector<Constant *> Todo;
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
59 std::set<Constant *> Done;
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
60 Todo.push_back(I);
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
61
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
62 while (!Todo.empty()) {
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
63 Constant *V = Todo.back();
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
64 Todo.pop_back();
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
65 Done.insert(V);
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
66
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
67 if (BlockAddress *BA = dyn_cast<BlockAddress>(V)) {
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
68 Function *F = BA->getFunction();
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
69 if (F->isDeclaration())
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
70 return F;
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
71 }
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
72
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
73 for (User::op_iterator i = V->op_begin(), e = V->op_end(); i != e; ++i) {
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
74 Constant *C = dyn_cast<Constant>(*i);
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
75 if (C && !isa<GlobalValue>(C) && !Done.count(C))
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
76 Todo.push_back(C);
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
77 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
78 }
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
79 return nullptr;
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
80 }
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
81 } // end anonymous namespace
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
82
77
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
83 std::unique_ptr<Module>
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
84 BugDriver::deleteInstructionFromProgram(const Instruction *I,
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
85 unsigned Simplification) {
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
86 // FIXME, use vmap?
134
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
87 std::unique_ptr<Module> Clone = CloneModule(*Program);
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
88
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
89 const BasicBlock *PBB = I->getParent();
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
90 const Function *PF = PBB->getParent();
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
91
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
92 Module::iterator RFI = Clone->begin(); // Get iterator to corresponding fn
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
93 std::advance(
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
94 RFI, std::distance(PF->getParent()->begin(), Module::const_iterator(PF)));
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
95
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
96 Function::iterator RBI = RFI->begin(); // Get iterator to corresponding BB
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
97 std::advance(RBI, std::distance(PF->begin(), Function::const_iterator(PBB)));
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
98
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
99 BasicBlock::iterator RI = RBI->begin(); // Get iterator to corresponding inst
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
100 std::advance(RI, std::distance(PBB->begin(), BasicBlock::const_iterator(I)));
100
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
101 Instruction *TheInst = &*RI; // Got the corresponding instruction!
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
102
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
103 // If this instruction produces a value, replace any users with null values
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
104 if (!TheInst->getType()->isVoidTy())
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
105 TheInst->replaceAllUsesWith(Constant::getNullValue(TheInst->getType()));
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
106
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
107 // Remove the instruction from the program.
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
108 TheInst->getParent()->getInstList().erase(TheInst);
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
109
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
110 // Spiff up the output a little bit.
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
111 std::vector<std::string> Passes;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
112
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
113 /// Can we get rid of the -disable-* options?
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
114 if (Simplification > 1 && !NoDCE)
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
115 Passes.push_back("dce");
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
116 if (Simplification && !DisableSimplifyCFG)
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
117 Passes.push_back("simplifycfg"); // Delete dead control flow
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
118
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
119 Passes.push_back("verify");
134
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
120 std::unique_ptr<Module> New = runPassesOn(Clone.get(), Passes);
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
121 if (!New) {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
122 errs() << "Instruction removal failed. Sorry. :( Please report a bug!\n";
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
123 exit(1);
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
124 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
125 return New;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
126 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
127
77
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
128 std::unique_ptr<Module>
147
c2174574ed3a LLVM 10
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 134
diff changeset
129 BugDriver::performFinalCleanups(std::unique_ptr<Module> M,
c2174574ed3a LLVM 10
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 134
diff changeset
130 bool MayModifySemantics) {
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
131 // Make all functions external, so GlobalDCE doesn't delete them...
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
132 for (Module::iterator I = M->begin(), E = M->end(); I != E; ++I)
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
133 I->setLinkage(GlobalValue::ExternalLinkage);
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
134
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
135 std::vector<std::string> CleanupPasses;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
136 CleanupPasses.push_back("globaldce");
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
137
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
138 if (MayModifySemantics)
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
139 CleanupPasses.push_back("deadarghaX0r");
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
140 else
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
141 CleanupPasses.push_back("deadargelim");
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
142
147
c2174574ed3a LLVM 10
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 134
diff changeset
143 std::unique_ptr<Module> New = runPassesOn(M.get(), CleanupPasses);
77
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
144 if (!New) {
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
145 errs() << "Final cleanups failed. Sorry. :( Please report a bug!\n";
77
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
146 return nullptr;
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
147 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
148 return New;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
149 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
150
77
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
151 std::unique_ptr<Module> BugDriver::extractLoop(Module *M) {
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
152 std::vector<std::string> LoopExtractPasses;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
153 LoopExtractPasses.push_back("loop-extract-single");
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
154
77
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
155 std::unique_ptr<Module> NewM = runPassesOn(M, LoopExtractPasses);
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
156 if (!NewM) {
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
157 outs() << "*** Loop extraction failed: ";
134
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
158 EmitProgressBitcode(*M, "loopextraction", true);
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
159 outs() << "*** Sorry. :( Please report a bug!\n";
77
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
160 return nullptr;
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
161 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
162
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
163 // Check to see if we created any new functions. If not, no loops were
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
164 // extracted and we should return null. Limit the number of loops we extract
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
165 // to avoid taking forever.
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
166 static unsigned NumExtracted = 32;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
167 if (M->size() == NewM->size() || --NumExtracted == 0) {
77
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
168 return nullptr;
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
169 } else {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
170 assert(M->size() < NewM->size() && "Loop extract removed functions?");
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
171 Module::iterator MI = NewM->begin();
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
172 for (unsigned i = 0, e = M->size(); i != e; ++i)
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
173 ++MI;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
174 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
175
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
176 return NewM;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
177 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
178
100
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
179 static void eliminateAliases(GlobalValue *GV) {
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
180 // First, check whether a GlobalAlias references this definition.
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
181 // GlobalAlias MAY NOT reference declarations.
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
182 for (;;) {
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
183 // 1. Find aliases
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
184 SmallVector<GlobalAlias *, 1> aliases;
100
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
185 Module *M = GV->getParent();
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
186 for (Module::alias_iterator I = M->alias_begin(), E = M->alias_end();
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
187 I != E; ++I)
100
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
188 if (I->getAliasee()->stripPointerCasts() == GV)
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
189 aliases.push_back(&*I);
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
190 if (aliases.empty())
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
191 break;
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
192 // 2. Resolve aliases
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
193 for (unsigned i = 0, e = aliases.size(); i < e; ++i) {
100
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
194 aliases[i]->replaceAllUsesWith(aliases[i]->getAliasee());
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
195 aliases[i]->eraseFromParent();
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
196 }
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
197 // 3. Repeat until no more aliases found; there might
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
198 // be an alias to an alias...
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
199 }
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
200 }
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
201
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
202 //
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
203 // DeleteGlobalInitializer - "Remove" the global variable by deleting its
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
204 // initializer,
100
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
205 // making it external.
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
206 //
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
207 void llvm::DeleteGlobalInitializer(GlobalVariable *GV) {
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
208 eliminateAliases(GV);
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
209 GV->setInitializer(nullptr);
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
210 GV->setComdat(nullptr);
100
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
211 }
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
212
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
213 // DeleteFunctionBody - "Remove" the function by deleting all of its basic
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
214 // blocks, making it external.
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
215 //
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
216 void llvm::DeleteFunctionBody(Function *F) {
100
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
217 eliminateAliases(F);
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
218 // Function declarations can't have comdats.
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
219 F->setComdat(nullptr);
100
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
220
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
221 // delete the body of the function...
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
222 F->deleteBody();
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
223 assert(F->isDeclaration() && "This didn't make the function external!");
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
224 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
225
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
226 /// GetTorInit - Given a list of entries for static ctors/dtors, return them
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
227 /// as a constant array.
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
228 static Constant *GetTorInit(std::vector<std::pair<Function *, int>> &TorList) {
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
229 assert(!TorList.empty() && "Don't create empty tor list!");
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
230 std::vector<Constant *> ArrayElts;
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
231 Type *Int32Ty = Type::getInt32Ty(TorList[0].first->getContext());
95
afa8332a0e37 LLVM 3.8
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 83
diff changeset
232
121
803732b1fca8 LLVM 5.0
kono
parents: 120
diff changeset
233 StructType *STy = StructType::get(Int32Ty, TorList[0].first->getType());
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
234 for (unsigned i = 0, e = TorList.size(); i != e; ++i) {
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
235 Constant *Elts[] = {ConstantInt::get(Int32Ty, TorList[i].second),
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
236 TorList[i].first};
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
237 ArrayElts.push_back(ConstantStruct::get(STy, Elts));
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
238 }
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
239 return ConstantArray::get(
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
240 ArrayType::get(ArrayElts[0]->getType(), ArrayElts.size()), ArrayElts);
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
241 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
242
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
243 /// SplitStaticCtorDtor - A module was recently split into two parts, M1/M2, and
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
244 /// M1 has all of the global variables. If M2 contains any functions that are
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
245 /// static ctors/dtors, we need to add an llvm.global_[cd]tors global to M2, and
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
246 /// prune appropriate entries out of M1s list.
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
247 static void SplitStaticCtorDtor(const char *GlobalName, Module *M1, Module *M2,
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
248 ValueToValueMapTy &VMap) {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
249 GlobalVariable *GV = M1->getNamedGlobal(GlobalName);
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
250 if (!GV || GV->isDeclaration() || GV->hasLocalLinkage() || !GV->use_empty())
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
251 return;
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
252
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
253 std::vector<std::pair<Function *, int>> M1Tors, M2Tors;
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
254 ConstantArray *InitList = dyn_cast<ConstantArray>(GV->getInitializer());
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
255 if (!InitList)
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
256 return;
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
257
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
258 for (unsigned i = 0, e = InitList->getNumOperands(); i != e; ++i) {
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
259 if (ConstantStruct *CS =
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
260 dyn_cast<ConstantStruct>(InitList->getOperand(i))) {
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
261 if (CS->getNumOperands() != 2)
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
262 return; // Not array of 2-element structs.
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
263
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
264 if (CS->getOperand(1)->isNullValue())
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
265 break; // Found a null terminator, stop here.
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
266
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
267 ConstantInt *CI = dyn_cast<ConstantInt>(CS->getOperand(0));
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
268 int Priority = CI ? CI->getSExtValue() : 0;
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
269
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
270 Constant *FP = CS->getOperand(1);
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
271 if (ConstantExpr *CE = dyn_cast<ConstantExpr>(FP))
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
272 if (CE->isCast())
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
273 FP = CE->getOperand(0);
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
274 if (Function *F = dyn_cast<Function>(FP)) {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
275 if (!F->isDeclaration())
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
276 M1Tors.push_back(std::make_pair(F, Priority));
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
277 else {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
278 // Map to M2's version of the function.
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
279 F = cast<Function>(VMap[F]);
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
280 M2Tors.push_back(std::make_pair(F, Priority));
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
281 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
282 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
283 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
284 }
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
285
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
286 GV->eraseFromParent();
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
287 if (!M1Tors.empty()) {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
288 Constant *M1Init = GetTorInit(M1Tors);
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
289 new GlobalVariable(*M1, M1Init->getType(), false,
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
290 GlobalValue::AppendingLinkage, M1Init, GlobalName);
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
291 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
292
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
293 GV = M2->getNamedGlobal(GlobalName);
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
294 assert(GV && "Not a clone of M1?");
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
295 assert(GV->use_empty() && "llvm.ctors shouldn't have uses!");
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
296
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
297 GV->eraseFromParent();
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
298 if (!M2Tors.empty()) {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
299 Constant *M2Init = GetTorInit(M2Tors);
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
300 new GlobalVariable(*M2, M2Init->getType(), false,
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
301 GlobalValue::AppendingLinkage, M2Init, GlobalName);
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
302 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
303 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
304
100
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
305 std::unique_ptr<Module>
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
306 llvm::SplitFunctionsOutOfModule(Module *M, const std::vector<Function *> &F,
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
307 ValueToValueMapTy &VMap) {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
308 // Make sure functions & globals are all external so that linkage
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
309 // between the two modules will work.
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
310 for (Module::iterator I = M->begin(), E = M->end(); I != E; ++I)
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
311 I->setLinkage(GlobalValue::ExternalLinkage);
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
312 for (Module::global_iterator I = M->global_begin(), E = M->global_end();
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
313 I != E; ++I) {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
314 if (I->hasName() && I->getName()[0] == '\01')
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
315 I->setName(I->getName().substr(1));
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
316 I->setLinkage(GlobalValue::ExternalLinkage);
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
317 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
318
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
319 ValueToValueMapTy NewVMap;
134
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
320 std::unique_ptr<Module> New = CloneModule(*M, NewVMap);
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
321
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
322 // Remove the Test functions from the Safe module
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
323 std::set<Function *> TestFunctions;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
324 for (unsigned i = 0, e = F.size(); i != e; ++i) {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
325 Function *TNOF = cast<Function>(VMap[F[i]]);
147
c2174574ed3a LLVM 10
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 134
diff changeset
326 LLVM_DEBUG(errs() << "Removing function ");
c2174574ed3a LLVM 10
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 134
diff changeset
327 LLVM_DEBUG(TNOF->printAsOperand(errs(), false));
c2174574ed3a LLVM 10
Shinji KONO <kono@ie.u-ryukyu.ac.jp>
parents: 134
diff changeset
328 LLVM_DEBUG(errs() << "\n");
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
329 TestFunctions.insert(cast<Function>(NewVMap[TNOF]));
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
330 DeleteFunctionBody(TNOF); // Function is now external in this module!
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
331 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
332
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
333 // Remove the Safe functions from the Test module
100
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
334 for (Function &I : *New)
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
335 if (!TestFunctions.count(&I))
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
336 DeleteFunctionBody(&I);
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
337
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
338 // Try to split the global initializers evenly
100
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
339 for (GlobalVariable &I : M->globals()) {
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
340 GlobalVariable *GV = cast<GlobalVariable>(NewVMap[&I]);
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
341 if (Function *TestFn = globalInitUsesExternalBA(&I)) {
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
342 if (Function *SafeFn = globalInitUsesExternalBA(GV)) {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
343 errs() << "*** Error: when reducing functions, encountered "
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
344 "the global '";
77
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
345 GV->printAsOperand(errs(), false);
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
346 errs() << "' with an initializer that references blockaddresses "
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
347 "from safe function '"
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
348 << SafeFn->getName() << "' and from test function '"
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
349 << TestFn->getName() << "'.\n";
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
350 exit(1);
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
351 }
100
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
352 DeleteGlobalInitializer(&I); // Delete the initializer to make it external
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
353 } else {
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
354 // If we keep it in the safe module, then delete it in the test module
100
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
355 DeleteGlobalInitializer(GV);
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
356 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
357 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
358
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
359 // Make sure that there is a global ctor/dtor array in both halves of the
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
360 // module if they both have static ctor/dtor functions.
100
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
361 SplitStaticCtorDtor("llvm.global_ctors", M, New.get(), NewVMap);
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
362 SplitStaticCtorDtor("llvm.global_dtors", M, New.get(), NewVMap);
7d135dc70f03 LLVM 3.9
Miyagi Mitsuki <e135756@ie.u-ryukyu.ac.jp>
parents: 95
diff changeset
363
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
364 return New;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
365 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
366
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
367 //===----------------------------------------------------------------------===//
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
368 // Basic Block Extraction Code
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
369 //===----------------------------------------------------------------------===//
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
370
77
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
371 std::unique_ptr<Module>
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
372 BugDriver::extractMappedBlocksFromModule(const std::vector<BasicBlock *> &BBs,
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
373 Module *M) {
134
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
374 auto Temp = sys::fs::TempFile::create(OutputPrefix + "-extractblocks%%%%%%%");
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
375 if (!Temp) {
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
376 outs() << "*** Basic Block extraction failed!\n";
134
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
377 errs() << "Error creating temporary file: " << toString(Temp.takeError())
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
378 << "\n";
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
379 EmitProgressBitcode(*M, "basicblockextractfail", true);
77
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
380 return nullptr;
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
381 }
134
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
382 DiscardTemp Discard{*Temp};
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
383
134
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
384 // Extract all of the blocks except the ones in BBs.
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
385 SmallVector<BasicBlock *, 32> BlocksToExtract;
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
386 for (Function &F : *M)
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
387 for (BasicBlock &BB : F)
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
388 // Check if this block is going to be extracted.
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
389 if (std::find(BBs.begin(), BBs.end(), &BB) == BBs.end())
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
390 BlocksToExtract.push_back(&BB);
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
391
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
392 raw_fd_ostream OS(Temp->FD, /*shouldClose*/ false);
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
393 for (BasicBlock *BB : BBs) {
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
394 // If the BB doesn't have a name, give it one so we have something to key
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
395 // off of.
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
396 if (!BB->hasName())
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
397 BB->setName("tmpbb");
134
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
398 OS << BB->getParent()->getName() << " " << BB->getName() << "\n";
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
399 }
134
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
400 OS.flush();
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
401 if (OS.has_error()) {
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
402 errs() << "Error writing list of blocks to not extract\n";
134
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
403 EmitProgressBitcode(*M, "basicblockextractfail", true);
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
404 OS.clear_error();
77
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
405 return nullptr;
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
406 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
407
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
408 std::string uniqueFN = "--extract-blocks-file=";
134
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
409 uniqueFN += Temp->TmpName;
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
410 const char *ExtraArg = uniqueFN.c_str();
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
411
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
412 std::vector<std::string> PI;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
413 PI.push_back("extract-blocks");
120
1172e4bd9c6f update 4.0.0
mir3636
parents: 100
diff changeset
414 std::unique_ptr<Module> Ret = runPassesOn(M, PI, 1, &ExtraArg);
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
415
77
54457678186b LLVM 3.6
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents: 0
diff changeset
416 if (!Ret) {
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
417 outs() << "*** Basic Block extraction failed, please report a bug!\n";
134
3a76565eade5 update 5.0.1
mir3636
parents: 121
diff changeset
418 EmitProgressBitcode(*M, "basicblockextractfail", true);
0
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
419 }
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
420 return Ret;
95c75e76d11b LLVM 3.4
Kaito Tokumori <e105711@ie.u-ryukyu.ac.jp>
parents:
diff changeset
421 }