ExtractFunction.cpp revision 866aa0d742b7bc9811fd1b45507af999c605205a
1afade9294af43c6b947b9aeaa1555883d5f853e3Chris Lattner//===- ExtractFunction.cpp - Extract a function from Program --------------===// 23da94aec4d429b2ba0f65fa040c33650cade196bMisha Brukman// 37c0e022c5c4be4b11e199a53f73bbdd84e34aa80John Criswell// The LLVM Compiler Infrastructure 47c0e022c5c4be4b11e199a53f73bbdd84e34aa80John Criswell// 521c62da287237d39d0d95004881ea4baae3be6daChris Lattner// This file is distributed under the University of Illinois Open Source 621c62da287237d39d0d95004881ea4baae3be6daChris Lattner// License. See LICENSE.TXT for details. 73da94aec4d429b2ba0f65fa040c33650cade196bMisha Brukman// 87c0e022c5c4be4b11e199a53f73bbdd84e34aa80John Criswell//===----------------------------------------------------------------------===// 9afade9294af43c6b947b9aeaa1555883d5f853e3Chris Lattner// 10efdc0b505712d1ca4460def27e51c430f033d58dChris Lattner// This file implements several methods that are used to extract functions, 11efdc0b505712d1ca4460def27e51c430f033d58dChris Lattner// loops, or portions of a module from the rest of the module. 12afade9294af43c6b947b9aeaa1555883d5f853e3Chris Lattner// 13afade9294af43c6b947b9aeaa1555883d5f853e3Chris Lattner//===----------------------------------------------------------------------===// 14afade9294af43c6b947b9aeaa1555883d5f853e3Chris Lattner 15afade9294af43c6b947b9aeaa1555883d5f853e3Chris Lattner#include "BugDriver.h" 165a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner#include "llvm/Constants.h" 175a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner#include "llvm/DerivedTypes.h" 18db1cd5eba551bdd3d7767207891b578f61f8b39dOwen Anderson#include "llvm/LLVMContext.h" 19afade9294af43c6b947b9aeaa1555883d5f853e3Chris Lattner#include "llvm/Module.h" 20afade9294af43c6b947b9aeaa1555883d5f853e3Chris Lattner#include "llvm/PassManager.h" 21d1a85a744cc1001c2b7fc37cf37aca266964f519Brian Gaeke#include "llvm/Pass.h" 22e49603d79d220a795bd50684c8b1f503ee40f97fMisha Brukman#include "llvm/Analysis/Verifier.h" 23e860dcb6e4eeb1ffae2299e5157a36c025260912Dan Gohman#include "llvm/Assembly/Writer.h" 24afade9294af43c6b947b9aeaa1555883d5f853e3Chris Lattner#include "llvm/Transforms/IPO.h" 256520785dcd22012535934098942d57c07c7631c2Chris Lattner#include "llvm/Transforms/Scalar.h" 26afade9294af43c6b947b9aeaa1555883d5f853e3Chris Lattner#include "llvm/Transforms/Utils/Cloning.h" 275e783ab0b5fc3407ec59f1a598fdb9ef3b96b287Chris Lattner#include "llvm/Transforms/Utils/FunctionUtils.h" 285da69c79f9c4490c6657c207430dfeb1060fc4ddChris Lattner#include "llvm/Target/TargetData.h" 29551ccae044b0ff658fe629dd67edd5ffe75d10e8Reid Spencer#include "llvm/Support/CommandLine.h" 30551ccae044b0ff658fe629dd67edd5ffe75d10e8Reid Spencer#include "llvm/Support/Debug.h" 31551ccae044b0ff658fe629dd67edd5ffe75d10e8Reid Spencer#include "llvm/Support/FileUtilities.h" 3274382b7c699120fbec5cb5603c9cf4212eb37f06Chris Lattner#include "llvm/Support/raw_ostream.h" 336fa98b13206583e6eb90b8304758b35548914944Nick Lewycky#include "llvm/System/Path.h" 346fa98b13206583e6eb90b8304758b35548914944Nick Lewycky#include "llvm/System/Signals.h" 35fb4b96e77e2930bf3d0c148f1c3685b6a4434666Chris Lattner#include <set> 36c6b519d64ef55d39e66a49510d4703a49bf228ccChris Lattnerusing namespace llvm; 37d0fde30ce850b78371fd1386338350591f9ff494Brian Gaeke 38d0fde30ce850b78371fd1386338350591f9ff494Brian Gaekenamespace llvm { 39c6b519d64ef55d39e66a49510d4703a49bf228ccChris Lattner bool DisableSimplifyCFG = false; 4068ccdaa84909108c42417a8091c771598e26456eDaniel Dunbar extern cl::opt<std::string> OutputPrefix; 41d0fde30ce850b78371fd1386338350591f9ff494Brian Gaeke} // End llvm namespace 42d0fde30ce850b78371fd1386338350591f9ff494Brian Gaeke 436db70ef879916e6115ac97eb76e4fea973652e2cChris Lattnernamespace { 446db70ef879916e6115ac97eb76e4fea973652e2cChris Lattner cl::opt<bool> 456db70ef879916e6115ac97eb76e4fea973652e2cChris Lattner NoDCE ("disable-dce", 466db70ef879916e6115ac97eb76e4fea973652e2cChris Lattner cl::desc("Do not use the -dce pass to reduce testcases")); 4747ae4a1cee5eec5767a11403c0fac7c91ec45461Chris Lattner cl::opt<bool, true> 4847ae4a1cee5eec5767a11403c0fac7c91ec45461Chris Lattner NoSCFG("disable-simplifycfg", cl::location(DisableSimplifyCFG), 496db70ef879916e6115ac97eb76e4fea973652e2cChris Lattner cl::desc("Do not use the -simplifycfg pass to reduce testcases")); 506db70ef879916e6115ac97eb76e4fea973652e2cChris Lattner} 51afade9294af43c6b947b9aeaa1555883d5f853e3Chris Lattner 526520785dcd22012535934098942d57c07c7631c2Chris Lattner/// deleteInstructionFromProgram - This method clones the current Program and 536520785dcd22012535934098942d57c07c7631c2Chris Lattner/// deletes the specified instruction from the cloned module. It then runs a 546520785dcd22012535934098942d57c07c7631c2Chris Lattner/// series of cleanup passes (ADCE and SimplifyCFG) to eliminate any code which 556520785dcd22012535934098942d57c07c7631c2Chris Lattner/// depends on the value. The modified module is then returned. 566520785dcd22012535934098942d57c07c7631c2Chris Lattner/// 570cc8807029f577996a442b96d24c3346ed6de091Chris LattnerModule *BugDriver::deleteInstructionFromProgram(const Instruction *I, 58866aa0d742b7bc9811fd1b45507af999c605205aRafael Espindola unsigned Simplification) { 59866aa0d742b7bc9811fd1b45507af999c605205aRafael Espindola // FIXME, use vmap? 60866aa0d742b7bc9811fd1b45507af999c605205aRafael Espindola Module *Clone = CloneModule(Program); 616520785dcd22012535934098942d57c07c7631c2Chris Lattner 620cc8807029f577996a442b96d24c3346ed6de091Chris Lattner const BasicBlock *PBB = I->getParent(); 630cc8807029f577996a442b96d24c3346ed6de091Chris Lattner const Function *PF = PBB->getParent(); 646520785dcd22012535934098942d57c07c7631c2Chris Lattner 65866aa0d742b7bc9811fd1b45507af999c605205aRafael Espindola Module::iterator RFI = Clone->begin(); // Get iterator to corresponding fn 660cc8807029f577996a442b96d24c3346ed6de091Chris Lattner std::advance(RFI, std::distance(PF->getParent()->begin(), 670cc8807029f577996a442b96d24c3346ed6de091Chris Lattner Module::const_iterator(PF))); 686520785dcd22012535934098942d57c07c7631c2Chris Lattner 696520785dcd22012535934098942d57c07c7631c2Chris Lattner Function::iterator RBI = RFI->begin(); // Get iterator to corresponding BB 700cc8807029f577996a442b96d24c3346ed6de091Chris Lattner std::advance(RBI, std::distance(PF->begin(), Function::const_iterator(PBB))); 716520785dcd22012535934098942d57c07c7631c2Chris Lattner 726520785dcd22012535934098942d57c07c7631c2Chris Lattner BasicBlock::iterator RI = RBI->begin(); // Get iterator to corresponding inst 730cc8807029f577996a442b96d24c3346ed6de091Chris Lattner std::advance(RI, std::distance(PBB->begin(), BasicBlock::const_iterator(I))); 740cc8807029f577996a442b96d24c3346ed6de091Chris Lattner Instruction *TheInst = RI; // Got the corresponding instruction! 756520785dcd22012535934098942d57c07c7631c2Chris Lattner 766520785dcd22012535934098942d57c07c7631c2Chris Lattner // If this instruction produces a value, replace any users with null values 77e49a13e7260b83ce56d01446f2a165cc9f35da7fDan Gohman if (!TheInst->getType()->isVoidTy()) 78a7235ea7245028a0723e8ab7fd011386b3900777Owen Anderson TheInst->replaceAllUsesWith(Constant::getNullValue(TheInst->getType())); 796520785dcd22012535934098942d57c07c7631c2Chris Lattner 806520785dcd22012535934098942d57c07c7631c2Chris Lattner // Remove the instruction from the program. 810cc8807029f577996a442b96d24c3346ed6de091Chris Lattner TheInst->getParent()->getInstList().erase(TheInst); 826520785dcd22012535934098942d57c07c7631c2Chris Lattner 8344be25716628941b4cccccf56a28ee0ba2606850Chris Lattner // Spiff up the output a little bit. 84866aa0d742b7bc9811fd1b45507af999c605205aRafael Espindola std::vector<std::string> Passes; 855da69c79f9c4490c6657c207430dfeb1060fc4ddChris Lattner 86866aa0d742b7bc9811fd1b45507af999c605205aRafael Espindola /// Can we get rid of the -disable-* options? 876db70ef879916e6115ac97eb76e4fea973652e2cChris Lattner if (Simplification > 1 && !NoDCE) 88866aa0d742b7bc9811fd1b45507af999c605205aRafael Espindola Passes.push_back("dce"); 8947ae4a1cee5eec5767a11403c0fac7c91ec45461Chris Lattner if (Simplification && !DisableSimplifyCFG) 90866aa0d742b7bc9811fd1b45507af999c605205aRafael Espindola Passes.push_back("simplifycfg"); // Delete dead control flow 91866aa0d742b7bc9811fd1b45507af999c605205aRafael Espindola 92866aa0d742b7bc9811fd1b45507af999c605205aRafael Espindola Passes.push_back("verify"); 93866aa0d742b7bc9811fd1b45507af999c605205aRafael Espindola Module *New = runPassesOn(Clone, Passes); 94866aa0d742b7bc9811fd1b45507af999c605205aRafael Espindola delete Clone; 95866aa0d742b7bc9811fd1b45507af999c605205aRafael Espindola if (!New) { 96866aa0d742b7bc9811fd1b45507af999c605205aRafael Espindola errs() << "Instruction removal failed. Sorry. :( Please report a bug!\n"; 97866aa0d742b7bc9811fd1b45507af999c605205aRafael Espindola exit(1); 98866aa0d742b7bc9811fd1b45507af999c605205aRafael Espindola } 99866aa0d742b7bc9811fd1b45507af999c605205aRafael Espindola return New; 1006520785dcd22012535934098942d57c07c7631c2Chris Lattner} 101ba386d943f4a83095d9c625cb0d46c1afe45ed1fChris Lattner 102ba386d943f4a83095d9c625cb0d46c1afe45ed1fChris Lattner/// performFinalCleanups - This method clones the current Program and performs 103ba386d943f4a83095d9c625cb0d46c1afe45ed1fChris Lattner/// a series of cleanups intended to get rid of extra cruft on the module 1049b5b1905dbad19de374e7617f615ba522ffe0468Chris Lattner/// before handing it to the user. 105ba386d943f4a83095d9c625cb0d46c1afe45ed1fChris Lattner/// 106fcb6ec0c7e37c2d15ddb04878f05cbd69d1da036Chris LattnerModule *BugDriver::performFinalCleanups(Module *M, bool MayModifySemantics) { 10728b8ed90c75ce6c271500fa778fef252f267a5ffChris Lattner // Make all functions external, so GlobalDCE doesn't delete them... 10828b8ed90c75ce6c271500fa778fef252f267a5ffChris Lattner for (Module::iterator I = M->begin(), E = M->end(); I != E; ++I) 10928b8ed90c75ce6c271500fa778fef252f267a5ffChris Lattner I->setLinkage(GlobalValue::ExternalLinkage); 1103da94aec4d429b2ba0f65fa040c33650cade196bMisha Brukman 1118261dfed05e32302469ef707cc881fed2c31f85fRafael Espindola std::vector<std::string> CleanupPasses; 1128261dfed05e32302469ef707cc881fed2c31f85fRafael Espindola CleanupPasses.push_back("globaldce"); 113fcb6ec0c7e37c2d15ddb04878f05cbd69d1da036Chris Lattner 114c6b519d64ef55d39e66a49510d4703a49bf228ccChris Lattner if (MayModifySemantics) 1158261dfed05e32302469ef707cc881fed2c31f85fRafael Espindola CleanupPasses.push_back("deadarghaX0r"); 116c6b519d64ef55d39e66a49510d4703a49bf228ccChris Lattner else 1178261dfed05e32302469ef707cc881fed2c31f85fRafael Espindola CleanupPasses.push_back("deadargelim"); 118fcb6ec0c7e37c2d15ddb04878f05cbd69d1da036Chris Lattner 1198261dfed05e32302469ef707cc881fed2c31f85fRafael Espindola CleanupPasses.push_back("deadtypeelim"); 120d0a6a8ca31e507ab3852f156867c439a4cd015f0Dan Gohman 121a75766a6c14b364b74b30546802e26d4b4b36a9bChris Lattner Module *New = runPassesOn(M, CleanupPasses); 122a75766a6c14b364b74b30546802e26d4b4b36a9bChris Lattner if (New == 0) { 12365f57c233cd4499e2e8b52a503201e64edfd6a9eDan Gohman errs() << "Final cleanups failed. Sorry. :( Please report a bug!\n"; 1249b5b1905dbad19de374e7617f615ba522ffe0468Chris Lattner return M; 125fcb6ec0c7e37c2d15ddb04878f05cbd69d1da036Chris Lattner } 126a75766a6c14b364b74b30546802e26d4b4b36a9bChris Lattner delete M; 127a75766a6c14b364b74b30546802e26d4b4b36a9bChris Lattner return New; 128ba386d943f4a83095d9c625cb0d46c1afe45ed1fChris Lattner} 129be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner 130be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner 1317546c3884a400b72d10fc19f120c6798b294a39dChris Lattner/// ExtractLoop - Given a module, extract up to one loop from it into a new 1327546c3884a400b72d10fc19f120c6798b294a39dChris Lattner/// function. This returns null if there are no extractable loops in the 1337546c3884a400b72d10fc19f120c6798b294a39dChris Lattner/// program or if the loop extractor crashes. 1347546c3884a400b72d10fc19f120c6798b294a39dChris LattnerModule *BugDriver::ExtractLoop(Module *M) { 1358261dfed05e32302469ef707cc881fed2c31f85fRafael Espindola std::vector<std::string> LoopExtractPasses; 1368261dfed05e32302469ef707cc881fed2c31f85fRafael Espindola LoopExtractPasses.push_back("loop-extract-single"); 1377546c3884a400b72d10fc19f120c6798b294a39dChris Lattner 138a75766a6c14b364b74b30546802e26d4b4b36a9bChris Lattner Module *NewM = runPassesOn(M, LoopExtractPasses); 139a75766a6c14b364b74b30546802e26d4b4b36a9bChris Lattner if (NewM == 0) { 140ac95cc79ac0b899d566cc29c0f646f39c2fa35c0Dan Gohman outs() << "*** Loop extraction failed: "; 141bae1b71cbb930e419df03db209ebc547a0e4ec72Rafael Espindola EmitProgressBitcode(M, "loopextraction", true); 142ac95cc79ac0b899d566cc29c0f646f39c2fa35c0Dan Gohman outs() << "*** Sorry. :( Please report a bug!\n"; 1437546c3884a400b72d10fc19f120c6798b294a39dChris Lattner return 0; 1447546c3884a400b72d10fc19f120c6798b294a39dChris Lattner } 145a75766a6c14b364b74b30546802e26d4b4b36a9bChris Lattner 146a75766a6c14b364b74b30546802e26d4b4b36a9bChris Lattner // Check to see if we created any new functions. If not, no loops were 147a269ec7b0a83d3b20730fd2d9f7c3ed5a552da90Chris Lattner // extracted and we should return null. Limit the number of loops we extract 148a269ec7b0a83d3b20730fd2d9f7c3ed5a552da90Chris Lattner // to avoid taking forever. 149a269ec7b0a83d3b20730fd2d9f7c3ed5a552da90Chris Lattner static unsigned NumExtracted = 32; 15090c18c5c69d9c451e5fdca1e4b4b95e8ed13291aChris Lattner if (M->size() == NewM->size() || --NumExtracted == 0) { 151a75766a6c14b364b74b30546802e26d4b4b36a9bChris Lattner delete NewM; 152a75766a6c14b364b74b30546802e26d4b4b36a9bChris Lattner return 0; 15390c18c5c69d9c451e5fdca1e4b4b95e8ed13291aChris Lattner } else { 15490c18c5c69d9c451e5fdca1e4b4b95e8ed13291aChris Lattner assert(M->size() < NewM->size() && "Loop extract removed functions?"); 15590c18c5c69d9c451e5fdca1e4b4b95e8ed13291aChris Lattner Module::iterator MI = NewM->begin(); 15690c18c5c69d9c451e5fdca1e4b4b95e8ed13291aChris Lattner for (unsigned i = 0, e = M->size(); i != e; ++i) 15790c18c5c69d9c451e5fdca1e4b4b95e8ed13291aChris Lattner ++MI; 158a75766a6c14b364b74b30546802e26d4b4b36a9bChris Lattner } 1593da94aec4d429b2ba0f65fa040c33650cade196bMisha Brukman 160a75766a6c14b364b74b30546802e26d4b4b36a9bChris Lattner return NewM; 1617546c3884a400b72d10fc19f120c6798b294a39dChris Lattner} 1627546c3884a400b72d10fc19f120c6798b294a39dChris Lattner 1637546c3884a400b72d10fc19f120c6798b294a39dChris Lattner 164be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner// DeleteFunctionBody - "Remove" the function by deleting all of its basic 165be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner// blocks, making it external. 166be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner// 167be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattnervoid llvm::DeleteFunctionBody(Function *F) { 168be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner // delete the body of the function... 169be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner F->deleteBody(); 1705cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer assert(F->isDeclaration() && "This didn't make the function external!"); 171be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner} 172be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner 1735a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner/// GetTorInit - Given a list of entries for static ctors/dtors, return them 1745a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner/// as a constant array. 1755a7a9e53861188ad84cb0e08e34562da744a907aChris Lattnerstatic Constant *GetTorInit(std::vector<std::pair<Function*, int> > &TorList) { 1765a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner assert(!TorList.empty() && "Don't create empty tor list!"); 1775a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner std::vector<Constant*> ArrayElts; 1785a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner for (unsigned i = 0, e = TorList.size(); i != e; ++i) { 1795a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner std::vector<Constant*> Elts; 1801d0be15f89cb5056e20e2d24faa8d6afb1573bcaOwen Anderson Elts.push_back(ConstantInt::get( 1811d0be15f89cb5056e20e2d24faa8d6afb1573bcaOwen Anderson Type::getInt32Ty(TorList[i].first->getContext()), TorList[i].second)); 1825a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner Elts.push_back(TorList[i].first); 183c332fba8285e35a5a11463c34795af84f3960759Nick Lewycky ArrayElts.push_back(ConstantStruct::get(TorList[i].first->getContext(), 184c332fba8285e35a5a11463c34795af84f3960759Nick Lewycky Elts, false)); 1855a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner } 186debcb01b0f0a15f568ca69e8f288fade4bfc7297Owen Anderson return ConstantArray::get(ArrayType::get(ArrayElts[0]->getType(), 1875a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner ArrayElts.size()), 1885a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner ArrayElts); 1895a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner} 1905a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner 1915a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner/// SplitStaticCtorDtor - A module was recently split into two parts, M1/M2, and 1925a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner/// M1 has all of the global variables. If M2 contains any functions that are 1935a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner/// static ctors/dtors, we need to add an llvm.global_[cd]tors global to M2, and 1945a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner/// prune appropriate entries out of M1s list. 195d50330cd02b00c8e3de40e8544c45701b9891d87Dan Gohmanstatic void SplitStaticCtorDtor(const char *GlobalName, Module *M1, Module *M2, 19680a098583b3a9754fd83bf93322a8be679ce0036Duncan Sands ValueMap<const Value*, Value*> &VMap) { 1975a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner GlobalVariable *GV = M1->getNamedGlobal(GlobalName); 198bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola if (!GV || GV->isDeclaration() || GV->hasLocalLinkage() || 1995a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner !GV->use_empty()) return; 2005a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner 2015a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner std::vector<std::pair<Function*, int> > M1Tors, M2Tors; 2025a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner ConstantArray *InitList = dyn_cast<ConstantArray>(GV->getInitializer()); 2035a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner if (!InitList) return; 2045a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner 2055a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner for (unsigned i = 0, e = InitList->getNumOperands(); i != e; ++i) { 2065a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner if (ConstantStruct *CS = dyn_cast<ConstantStruct>(InitList->getOperand(i))){ 2075a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner if (CS->getNumOperands() != 2) return; // Not array of 2-element structs. 2085a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner 2095a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner if (CS->getOperand(1)->isNullValue()) 2105a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner break; // Found a null terminator, stop here. 2115a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner 212b83eb6447ba155342598f0fabe1f08f5baa9164aReid Spencer ConstantInt *CI = dyn_cast<ConstantInt>(CS->getOperand(0)); 213b83eb6447ba155342598f0fabe1f08f5baa9164aReid Spencer int Priority = CI ? CI->getSExtValue() : 0; 2145a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner 2155a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner Constant *FP = CS->getOperand(1); 2165a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner if (ConstantExpr *CE = dyn_cast<ConstantExpr>(FP)) 2173da59db637a887474c1b1346c1f3ccf53b6c4663Reid Spencer if (CE->isCast()) 2185a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner FP = CE->getOperand(0); 2195a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner if (Function *F = dyn_cast<Function>(FP)) { 2205cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer if (!F->isDeclaration()) 2215a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner M1Tors.push_back(std::make_pair(F, Priority)); 2225a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner else { 2235a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner // Map to M2's version of the function. 224e9916a302f1bacad234d7dafc1df3dc968a6ba0fDevang Patel F = cast<Function>(VMap[F]); 2255a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner M2Tors.push_back(std::make_pair(F, Priority)); 2265a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner } 2275a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner } 2285a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner } 2295a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner } 2305a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner 2315a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner GV->eraseFromParent(); 2325a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner if (!M1Tors.empty()) { 2335a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner Constant *M1Init = GetTorInit(M1Tors); 234e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson new GlobalVariable(*M1, M1Init->getType(), false, 2353d29df3e8a203b167d8071ea6f805b21db18a5afOwen Anderson GlobalValue::AppendingLinkage, 236e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson M1Init, GlobalName); 2375a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner } 2385a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner 2395a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner GV = M2->getNamedGlobal(GlobalName); 2405a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner assert(GV && "Not a clone of M1?"); 2415a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner assert(GV->use_empty() && "llvm.ctors shouldn't have uses!"); 2425a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner 2435a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner GV->eraseFromParent(); 2445a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner if (!M2Tors.empty()) { 2455a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner Constant *M2Init = GetTorInit(M2Tors); 246e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson new GlobalVariable(*M2, M2Init->getType(), false, 2473d29df3e8a203b167d8071ea6f805b21db18a5afOwen Anderson GlobalValue::AppendingLinkage, 248e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson M2Init, GlobalName); 2495a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner } 2505a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner} 2515a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner 252e47863e212e0af6239f6f40b2496ede29847bbcePatrick Jenkins 253be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner/// SplitFunctionsOutOfModule - Given a module and a list of functions in the 254be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner/// module, split the functions OUT of the specified module, and place them in 255be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner/// the new module. 256d50330cd02b00c8e3de40e8544c45701b9891d87Dan GohmanModule * 257d50330cd02b00c8e3de40e8544c45701b9891d87Dan Gohmanllvm::SplitFunctionsOutOfModule(Module *M, 258d50330cd02b00c8e3de40e8544c45701b9891d87Dan Gohman const std::vector<Function*> &F, 259e9916a302f1bacad234d7dafc1df3dc968a6ba0fDevang Patel ValueMap<const Value*, Value*> &VMap) { 260be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner // Make sure functions & globals are all external so that linkage 261be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner // between the two modules will work. 262be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner for (Module::iterator I = M->begin(), E = M->end(); I != E; ++I) 263be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner I->setLinkage(GlobalValue::ExternalLinkage); 2645a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner for (Module::global_iterator I = M->global_begin(), E = M->global_end(); 2657220b814ed6f432d1d31d444b65e0c3b120942dfOwen Anderson I != E; ++I) { 2663f53fa9a51c4ce7ba81170ca7ab2e49bd37281b0Daniel Dunbar if (I->hasName() && I->getName()[0] == '\01') 2673f53fa9a51c4ce7ba81170ca7ab2e49bd37281b0Daniel Dunbar I->setName(I->getName().substr(1)); 268be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner I->setLinkage(GlobalValue::ExternalLinkage); 2697220b814ed6f432d1d31d444b65e0c3b120942dfOwen Anderson } 270be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner 271e9916a302f1bacad234d7dafc1df3dc968a6ba0fDevang Patel ValueMap<const Value*, Value*> NewVMap; 272e9916a302f1bacad234d7dafc1df3dc968a6ba0fDevang Patel Module *New = CloneModule(M, NewVMap); 273be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner 274fef02427cda8dc0b3e6aedd6d78c4ba3e1e48069Chris Lattner // Make sure global initializers exist only in the safe module (CBE->.so) 275fef02427cda8dc0b3e6aedd6d78c4ba3e1e48069Chris Lattner for (Module::global_iterator I = New->global_begin(), E = New->global_end(); 276fef02427cda8dc0b3e6aedd6d78c4ba3e1e48069Chris Lattner I != E; ++I) 277fef02427cda8dc0b3e6aedd6d78c4ba3e1e48069Chris Lattner I->setInitializer(0); // Delete the initializer to make it external 278be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner 279fef02427cda8dc0b3e6aedd6d78c4ba3e1e48069Chris Lattner // Remove the Test functions from the Safe module 280d50330cd02b00c8e3de40e8544c45701b9891d87Dan Gohman std::set<Function *> TestFunctions; 281be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner for (unsigned i = 0, e = F.size(); i != e; ++i) { 282e9916a302f1bacad234d7dafc1df3dc968a6ba0fDevang Patel Function *TNOF = cast<Function>(VMap[F[i]]); 28365f57c233cd4499e2e8b52a503201e64edfd6a9eDan Gohman DEBUG(errs() << "Removing function "); 28465f57c233cd4499e2e8b52a503201e64edfd6a9eDan Gohman DEBUG(WriteAsOperand(errs(), TNOF, false)); 28565f57c233cd4499e2e8b52a503201e64edfd6a9eDan Gohman DEBUG(errs() << "\n"); 286e9916a302f1bacad234d7dafc1df3dc968a6ba0fDevang Patel TestFunctions.insert(cast<Function>(NewVMap[TNOF])); 287fef02427cda8dc0b3e6aedd6d78c4ba3e1e48069Chris Lattner DeleteFunctionBody(TNOF); // Function is now external in this module! 288e47863e212e0af6239f6f40b2496ede29847bbcePatrick Jenkins } 289e47863e212e0af6239f6f40b2496ede29847bbcePatrick Jenkins 2905a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner 291fef02427cda8dc0b3e6aedd6d78c4ba3e1e48069Chris Lattner // Remove the Safe functions from the Test module 292fef02427cda8dc0b3e6aedd6d78c4ba3e1e48069Chris Lattner for (Module::iterator I = New->begin(), E = New->end(); I != E; ++I) 293d50330cd02b00c8e3de40e8544c45701b9891d87Dan Gohman if (!TestFunctions.count(I)) 294fef02427cda8dc0b3e6aedd6d78c4ba3e1e48069Chris Lattner DeleteFunctionBody(I); 295fef02427cda8dc0b3e6aedd6d78c4ba3e1e48069Chris Lattner 296e47863e212e0af6239f6f40b2496ede29847bbcePatrick Jenkins 2975a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner // Make sure that there is a global ctor/dtor array in both halves of the 2985a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner // module if they both have static ctor/dtor functions. 299e9916a302f1bacad234d7dafc1df3dc968a6ba0fDevang Patel SplitStaticCtorDtor("llvm.global_ctors", M, New, NewVMap); 300e9916a302f1bacad234d7dafc1df3dc968a6ba0fDevang Patel SplitStaticCtorDtor("llvm.global_dtors", M, New, NewVMap); 3015a7a9e53861188ad84cb0e08e34562da744a907aChris Lattner 302be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner return New; 303be21ca54e08339ede5dd4bbb882182d22e274988Chris Lattner} 3045e783ab0b5fc3407ec59f1a598fdb9ef3b96b287Chris Lattner 3055e783ab0b5fc3407ec59f1a598fdb9ef3b96b287Chris Lattner//===----------------------------------------------------------------------===// 3065e783ab0b5fc3407ec59f1a598fdb9ef3b96b287Chris Lattner// Basic Block Extraction Code 3075e783ab0b5fc3407ec59f1a598fdb9ef3b96b287Chris Lattner//===----------------------------------------------------------------------===// 3085e783ab0b5fc3407ec59f1a598fdb9ef3b96b287Chris Lattner 3095e783ab0b5fc3407ec59f1a598fdb9ef3b96b287Chris Lattner/// ExtractMappedBlocksFromModule - Extract all but the specified basic blocks 3105e783ab0b5fc3407ec59f1a598fdb9ef3b96b287Chris Lattner/// into their own functions. The only detail is that M is actually a module 3115e783ab0b5fc3407ec59f1a598fdb9ef3b96b287Chris Lattner/// cloned from the one the BBs are in, so some mapping needs to be performed. 3125e783ab0b5fc3407ec59f1a598fdb9ef3b96b287Chris Lattner/// If this operation fails for some reason (ie the implementation is buggy), 3135e783ab0b5fc3407ec59f1a598fdb9ef3b96b287Chris Lattner/// this function should return null, otherwise it returns a new Module. 3145e783ab0b5fc3407ec59f1a598fdb9ef3b96b287Chris LattnerModule *BugDriver::ExtractMappedBlocksFromModule(const 3155e783ab0b5fc3407ec59f1a598fdb9ef3b96b287Chris Lattner std::vector<BasicBlock*> &BBs, 3165e783ab0b5fc3407ec59f1a598fdb9ef3b96b287Chris Lattner Module *M) { 31768ccdaa84909108c42417a8091c771598e26456eDaniel Dunbar sys::Path uniqueFilename(OutputPrefix + "-extractblocks"); 3186fa98b13206583e6eb90b8304758b35548914944Nick Lewycky std::string ErrMsg; 3196fa98b13206583e6eb90b8304758b35548914944Nick Lewycky if (uniqueFilename.createTemporaryFileOnDisk(true, &ErrMsg)) { 320ac95cc79ac0b899d566cc29c0f646f39c2fa35c0Dan Gohman outs() << "*** Basic Block extraction failed!\n"; 32165f57c233cd4499e2e8b52a503201e64edfd6a9eDan Gohman errs() << "Error creating temporary file: " << ErrMsg << "\n"; 322bae1b71cbb930e419df03db209ebc547a0e4ec72Rafael Espindola EmitProgressBitcode(M, "basicblockextractfail", true); 3236fa98b13206583e6eb90b8304758b35548914944Nick Lewycky return 0; 3246fa98b13206583e6eb90b8304758b35548914944Nick Lewycky } 3256fa98b13206583e6eb90b8304758b35548914944Nick Lewycky sys::RemoveFileOnSignal(uniqueFilename); 3266fa98b13206583e6eb90b8304758b35548914944Nick Lewycky 327ac95cc79ac0b899d566cc29c0f646f39c2fa35c0Dan Gohman std::string ErrorInfo; 328baa26395ccf17fc988bb9cf62d6659ca8415ece9Dan Gohman raw_fd_ostream BlocksToNotExtractFile(uniqueFilename.c_str(), ErrorInfo); 329ac95cc79ac0b899d566cc29c0f646f39c2fa35c0Dan Gohman if (!ErrorInfo.empty()) { 330ac95cc79ac0b899d566cc29c0f646f39c2fa35c0Dan Gohman outs() << "*** Basic Block extraction failed!\n"; 331ac95cc79ac0b899d566cc29c0f646f39c2fa35c0Dan Gohman errs() << "Error writing list of blocks to not extract: " << ErrorInfo 33265f57c233cd4499e2e8b52a503201e64edfd6a9eDan Gohman << "\n"; 333bae1b71cbb930e419df03db209ebc547a0e4ec72Rafael Espindola EmitProgressBitcode(M, "basicblockextractfail", true); 3346fa98b13206583e6eb90b8304758b35548914944Nick Lewycky return 0; 3356fa98b13206583e6eb90b8304758b35548914944Nick Lewycky } 3366fa98b13206583e6eb90b8304758b35548914944Nick Lewycky for (std::vector<BasicBlock*>::const_iterator I = BBs.begin(), E = BBs.end(); 3376fa98b13206583e6eb90b8304758b35548914944Nick Lewycky I != E; ++I) { 3386fa98b13206583e6eb90b8304758b35548914944Nick Lewycky BasicBlock *BB = *I; 3394a6a6f21f3342428e5dcf28b0ddfbe0039a040e5Chris Lattner // If the BB doesn't have a name, give it one so we have something to key 3404a6a6f21f3342428e5dcf28b0ddfbe0039a040e5Chris Lattner // off of. 3414a6a6f21f3342428e5dcf28b0ddfbe0039a040e5Chris Lattner if (!BB->hasName()) BB->setName("tmpbb"); 342f6ccee5a9d2b9573f679bca6266ade3eb8cd3f88Daniel Dunbar BlocksToNotExtractFile << BB->getParent()->getNameStr() << " " 3436fa98b13206583e6eb90b8304758b35548914944Nick Lewycky << BB->getName() << "\n"; 3446fa98b13206583e6eb90b8304758b35548914944Nick Lewycky } 3456fa98b13206583e6eb90b8304758b35548914944Nick Lewycky BlocksToNotExtractFile.close(); 3466fa98b13206583e6eb90b8304758b35548914944Nick Lewycky 34712ea66a7277240c5b045ed14c140f94d453eea0eBenjamin Kramer std::string uniqueFN = "--extract-blocks-file=" + uniqueFilename.str(); 34812ea66a7277240c5b045ed14c140f94d453eea0eBenjamin Kramer const char *ExtraArg = uniqueFN.c_str(); 3496fa98b13206583e6eb90b8304758b35548914944Nick Lewycky 3508261dfed05e32302469ef707cc881fed2c31f85fRafael Espindola std::vector<std::string> PI; 3518261dfed05e32302469ef707cc881fed2c31f85fRafael Espindola PI.push_back("extract-blocks"); 3526fa98b13206583e6eb90b8304758b35548914944Nick Lewycky Module *Ret = runPassesOn(M, PI, false, 1, &ExtraArg); 3536fa98b13206583e6eb90b8304758b35548914944Nick Lewycky 354d27047f87dae7c6eb59069b1e5d1368a0b2db5ffDan Gohman uniqueFilename.eraseFromDisk(); // Free disk space 3556fa98b13206583e6eb90b8304758b35548914944Nick Lewycky 356891150f0b2a124a75c9f31516182ce864ae69ef8Chris Lattner if (Ret == 0) { 357ac95cc79ac0b899d566cc29c0f646f39c2fa35c0Dan Gohman outs() << "*** Basic Block extraction failed, please report a bug!\n"; 358bae1b71cbb930e419df03db209ebc547a0e4ec72Rafael Espindola EmitProgressBitcode(M, "basicblockextractfail", true); 359891150f0b2a124a75c9f31516182ce864ae69ef8Chris Lattner } 3605e783ab0b5fc3407ec59f1a598fdb9ef3b96b287Chris Lattner return Ret; 3615e783ab0b5fc3407ec59f1a598fdb9ef3b96b287Chris Lattner} 362