diff options
Diffstat (limited to 'lib/Analysis/DataStructure/DataStructure.cpp')
-rw-r--r-- | lib/Analysis/DataStructure/DataStructure.cpp | 19 |
1 files changed, 15 insertions, 4 deletions
diff --git a/lib/Analysis/DataStructure/DataStructure.cpp b/lib/Analysis/DataStructure/DataStructure.cpp index 2418b0c657..2313cd09e8 100644 --- a/lib/Analysis/DataStructure/DataStructure.cpp +++ b/lib/Analysis/DataStructure/DataStructure.cpp @@ -439,14 +439,25 @@ DSNodeHandle DSGraph::cloneInto(const DSGraph &G, for (unsigned i = FN, e = Nodes.size(); i != e; ++i) Nodes[i]->NodeType &= ~StripBits; - // Copy the value map... + // Copy the value map... and merge all of the global nodes... for (std::map<Value*, DSNodeHandle>::const_iterator I = G.ValueMap.begin(), - E = G.ValueMap.end(); I != E; ++I) - OldValMap[I->first] = DSNodeHandle(OldNodeMap[I->second.getNode()], - I->second.getOffset()); + E = G.ValueMap.end(); I != E; ++I) { + DSNodeHandle &H = OldValMap[I->first]; + H = DSNodeHandle(OldNodeMap[I->second.getNode()], I->second.getOffset()); + + if (isa<GlobalValue>(I->first)) { // Is this a global? + std::map<Value*, DSNodeHandle>::iterator GVI = ValueMap.find(I->first); + if (GVI != ValueMap.end()) { // Is the global value in this fun already? + GVI->second.mergeWith(H); + } else { + ValueMap[I->first] = H; // Add global pointer to this graph + } + } + } // Copy the function calls list... CopyFunctionCallsList(G.FunctionCalls, FunctionCalls, OldNodeMap); + // Return the returned node pointer... return DSNodeHandle(OldNodeMap[G.RetNode.getNode()], G.RetNode.getOffset()); } |