Chris Lattner | 51cbcbf | 2002-11-20 20:47:41 +0000 | [diff] [blame] | 1 | //===- CloneModule.cpp - Clone an entire module ---------------------------===// |
John Criswell | b576c94 | 2003-10-20 19:43:21 +0000 | [diff] [blame] | 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file was developed by the LLVM research group and is distributed under |
| 6 | // the University of Illinois Open Source License. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
Chris Lattner | 51cbcbf | 2002-11-20 20:47:41 +0000 | [diff] [blame] | 9 | // |
| 10 | // This file implements the CloneModule interface which makes a copy of an |
| 11 | // entire module. |
| 12 | // |
| 13 | //===----------------------------------------------------------------------===// |
| 14 | |
| 15 | #include "llvm/Transforms/Utils/Cloning.h" |
| 16 | #include "llvm/Module.h" |
| 17 | #include "llvm/DerivedTypes.h" |
Chris Lattner | 892310e | 2003-04-24 17:15:33 +0000 | [diff] [blame] | 18 | #include "llvm/SymbolTable.h" |
Chris Lattner | 51cbcbf | 2002-11-20 20:47:41 +0000 | [diff] [blame] | 19 | #include "llvm/Constant.h" |
| 20 | #include "ValueMapper.h" |
Chris Lattner | f7703df | 2004-01-09 06:12:26 +0000 | [diff] [blame] | 21 | using namespace llvm; |
Brian Gaeke | d0fde30 | 2003-11-11 22:41:34 +0000 | [diff] [blame] | 22 | |
Chris Lattner | 51cbcbf | 2002-11-20 20:47:41 +0000 | [diff] [blame] | 23 | /// CloneModule - Return an exact copy of the specified module. This is not as |
| 24 | /// easy as it might seem because we have to worry about making copies of global |
Misha Brukman | cf00c4a | 2003-10-10 17:57:28 +0000 | [diff] [blame] | 25 | /// variables and functions, and making their (initializers and references, |
Chris Lattner | 51cbcbf | 2002-11-20 20:47:41 +0000 | [diff] [blame] | 26 | /// respectively) refer to the right globals. |
| 27 | /// |
Chris Lattner | f7703df | 2004-01-09 06:12:26 +0000 | [diff] [blame] | 28 | Module *llvm::CloneModule(const Module *M) { |
Chris Lattner | 51cbcbf | 2002-11-20 20:47:41 +0000 | [diff] [blame] | 29 | // First off, we need to create the new module... |
Chris Lattner | 37f5941 | 2003-04-22 18:02:26 +0000 | [diff] [blame] | 30 | Module *New = new Module(M->getModuleIdentifier()); |
Chris Lattner | 7899b74 | 2003-04-24 15:54:40 +0000 | [diff] [blame] | 31 | New->setEndianness(M->getEndianness()); |
| 32 | New->setPointerSize(M->getPointerSize()); |
Chris Lattner | 51cbcbf | 2002-11-20 20:47:41 +0000 | [diff] [blame] | 33 | |
Chris Lattner | 892310e | 2003-04-24 17:15:33 +0000 | [diff] [blame] | 34 | // Copy all of the type symbol table entries over... |
| 35 | const SymbolTable &SymTab = M->getSymbolTable(); |
Reid Spencer | 9231ac8 | 2004-05-25 08:53:40 +0000 | [diff] [blame] | 36 | SymbolTable::type_const_iterator TypeI = SymTab.type_begin(); |
| 37 | SymbolTable::type_const_iterator TypeE = SymTab.type_end(); |
| 38 | for ( ; TypeI != TypeE; ++TypeI ) { |
| 39 | New->addTypeName(TypeI->first, TypeI->second); |
| 40 | } |
Chris Lattner | 892310e | 2003-04-24 17:15:33 +0000 | [diff] [blame] | 41 | |
Chris Lattner | 51cbcbf | 2002-11-20 20:47:41 +0000 | [diff] [blame] | 42 | // Create the value map that maps things from the old module over to the new |
| 43 | // module. |
| 44 | std::map<const Value*, Value*> ValueMap; |
| 45 | |
| 46 | // Loop over all of the global variables, making corresponding globals in the |
| 47 | // new module. Here we add them to the ValueMap and to the new Module. We |
| 48 | // don't worry about attributes or initializers, they will come later. |
| 49 | // |
Chris Lattner | e4d5c44 | 2005-03-15 04:54:21 +0000 | [diff] [blame^] | 50 | for (Module::const_global_iterator I = M->global_begin(), E = M->global_end(); I != E; ++I) |
Chris Lattner | 4ad02e7 | 2003-04-16 20:28:45 +0000 | [diff] [blame] | 51 | ValueMap[I] = new GlobalVariable(I->getType()->getElementType(), false, |
| 52 | GlobalValue::ExternalLinkage, 0, |
| 53 | I->getName(), New); |
Chris Lattner | 51cbcbf | 2002-11-20 20:47:41 +0000 | [diff] [blame] | 54 | |
| 55 | // Loop over the functions in the module, making external functions as before |
| 56 | for (Module::const_iterator I = M->begin(), E = M->end(); I != E; ++I) |
| 57 | ValueMap[I]=new Function(cast<FunctionType>(I->getType()->getElementType()), |
Chris Lattner | 4ad02e7 | 2003-04-16 20:28:45 +0000 | [diff] [blame] | 58 | GlobalValue::ExternalLinkage, I->getName(), New); |
Chris Lattner | 51cbcbf | 2002-11-20 20:47:41 +0000 | [diff] [blame] | 59 | |
| 60 | // Now that all of the things that global variable initializer can refer to |
| 61 | // have been created, loop through and copy the global variable referrers |
| 62 | // over... We also set the attributes on the global now. |
| 63 | // |
Chris Lattner | e4d5c44 | 2005-03-15 04:54:21 +0000 | [diff] [blame^] | 64 | for (Module::const_global_iterator I = M->global_begin(), E = M->global_end(); I != E; ++I) { |
Chris Lattner | 51cbcbf | 2002-11-20 20:47:41 +0000 | [diff] [blame] | 65 | GlobalVariable *GV = cast<GlobalVariable>(ValueMap[I]); |
| 66 | if (I->hasInitializer()) |
| 67 | GV->setInitializer(cast<Constant>(MapValue(I->getInitializer(), |
| 68 | ValueMap))); |
Chris Lattner | 4ad02e7 | 2003-04-16 20:28:45 +0000 | [diff] [blame] | 69 | GV->setLinkage(I->getLinkage()); |
Chris Lattner | 51cbcbf | 2002-11-20 20:47:41 +0000 | [diff] [blame] | 70 | } |
| 71 | |
| 72 | // Similarly, copy over function bodies now... |
| 73 | // |
| 74 | for (Module::const_iterator I = M->begin(), E = M->end(); I != E; ++I) { |
| 75 | Function *F = cast<Function>(ValueMap[I]); |
| 76 | if (!I->isExternal()) { |
Chris Lattner | e4d5c44 | 2005-03-15 04:54:21 +0000 | [diff] [blame^] | 77 | Function::arg_iterator DestI = F->arg_begin(); |
| 78 | for (Function::const_arg_iterator J = I->arg_begin(); J != I->arg_end(); ++J) { |
Chris Lattner | 51cbcbf | 2002-11-20 20:47:41 +0000 | [diff] [blame] | 79 | DestI->setName(J->getName()); |
| 80 | ValueMap[J] = DestI++; |
| 81 | } |
| 82 | |
| 83 | std::vector<ReturnInst*> Returns; // Ignore returns cloned... |
| 84 | CloneFunctionInto(F, I, ValueMap, Returns); |
| 85 | } |
| 86 | |
Chris Lattner | 4ad02e7 | 2003-04-16 20:28:45 +0000 | [diff] [blame] | 87 | F->setLinkage(I->getLinkage()); |
Chris Lattner | 51cbcbf | 2002-11-20 20:47:41 +0000 | [diff] [blame] | 88 | } |
| 89 | |
| 90 | return New; |
| 91 | } |
Reid Spencer | 9231ac8 | 2004-05-25 08:53:40 +0000 | [diff] [blame] | 92 | |
| 93 | // vim: sw=2 |