blob: 7395bab803096850e2f41cd9acd318aa010e7ab9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
//===-- GlobalDCE.cpp - DCE unreachable internal methods ---------*- C++ -*--=//
//
// This transform is designed to eliminate unreachable internal globals
//
//===----------------------------------------------------------------------===//
#include "llvm/Transforms/IPO/GlobalDCE.h"
#include "llvm/Analysis/CallGraph.h"
#include "llvm/Module.h"
#include "llvm/Method.h"
#include "Support/DepthFirstIterator.h"
#include <set>
static bool RemoveUnreachableMethods(Module *M, cfg::CallGraph *CG) {
// Create a call graph if one is not already available...
cfg::CallGraph &CallGraph = CG ? *CG : *new cfg::CallGraph(M);
// Calculate which methods are reachable from the external methods in the call
// graph.
//
set<cfg::CallGraphNode*> ReachableNodes(df_begin(&CallGraph),
df_end(&CallGraph));
// Loop over the methods in the module twice. The first time is used to drop
// references that methods have to each other before they are deleted. The
// second pass removes the methods that need to be removed.
//
vector<cfg::CallGraphNode*> MethodsToDelete; // Track unused methods
for (Module::iterator I = M->begin(), E = M->end(); I != E; ++I) {
cfg::CallGraphNode *N = CallGraph[*I];
if (!ReachableNodes.count(N)) { // Not reachable??
(*I)->dropAllReferences();
N->removeAllCalledMethods();
MethodsToDelete.push_back(N);
}
}
// Nothing to do if no unreachable methods have been found...
if (MethodsToDelete.empty()) {
// Free the created call graph if it was not passed in
if (&CallGraph != CG) delete &CallGraph;
return false;
}
// Unreachables methods have been found and should have no references to them,
// delete them now.
//
for (vector<cfg::CallGraphNode*>::iterator I = MethodsToDelete.begin(),
E = MethodsToDelete.end(); I != E; ++I)
delete CallGraph.removeMethodFromModule(*I);
// Free the created call graph if it was not passed in
if (&CallGraph != CG) delete &CallGraph;
return true;
}
bool GlobalDCE::run(Module *M, cfg::CallGraph *CG = 0) {
return RemoveUnreachableMethods(M, CG);
}
|
