LinkModules.cpp revision debcb01b0f0a15f568ca69e8f288fade4bfc7297
1c834bbf55832c45f90b1163f920472fefbf77bc1Mikhail Glushenkov//===- lib/Linker/LinkModules.cpp - Module Linker Implementation ----------===// 2f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman// 3b576c94c15af9a440f69d9d03c2afead7971118cJohn Criswell// The LLVM Compiler Infrastructure 4b576c94c15af9a440f69d9d03c2afead7971118cJohn Criswell// 54ee451de366474b9c228b4e5fa573795a715216dChris Lattner// This file is distributed under the University of Illinois Open Source 64ee451de366474b9c228b4e5fa573795a715216dChris Lattner// License. See LICENSE.TXT for details. 7f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman// 8b576c94c15af9a440f69d9d03c2afead7971118cJohn Criswell//===----------------------------------------------------------------------===// 952f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// 1052f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// This file implements the LLVM module linker. 1152f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// 1252f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// Specifically, this: 138d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner// * Merges global variables between the two modules 148d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner// * Uninit + Uninit = Init, Init + Uninit = Init, Init + Init = Error if != 15c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// * Merges functions between two modules 1652f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// 1752f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner//===----------------------------------------------------------------------===// 1852f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner 197cc371a7958c7a049c679a59023c3c8c83f6cc83Reid Spencer#include "llvm/Linker.h" 20adbc0b5287bf36893cdcae2440d48b3cb3489e38Chris Lattner#include "llvm/Constants.h" 21adbc0b5287bf36893cdcae2440d48b3cb3489e38Chris Lattner#include "llvm/DerivedTypes.h" 22c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson#include "llvm/LLVMContext.h" 235c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner#include "llvm/Module.h" 2478d033e086e19e016273de014f9214aa6f3f844bReid Spencer#include "llvm/TypeSymbolTable.h" 25ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer#include "llvm/ValueSymbolTable.h" 2647b14a4a6a455c7be169cfd312fcbe796f0ad426Misha Brukman#include "llvm/Instructions.h" 27adbc0b5287bf36893cdcae2440d48b3cb3489e38Chris Lattner#include "llvm/Assembly/Writer.h" 2841edad7e4b8872414564dea5fb41fa6194375355Bill Wendling#include "llvm/Support/Streams.h" 29c25e7581b9b8088910da31702d4ca21c4734c6d7Torok Edwin#include "llvm/Support/ErrorHandling.h" 3057a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer#include "llvm/System/Path.h" 3162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner#include "llvm/ADT/DenseMap.h" 321a097e30d39e60303ae2b19f7a56e813f3e3c18eBill Wendling#include <sstream> 33f7703df4968084c18c248c1feea9961c19a32e6aChris Lattnerusing namespace llvm; 34d0fde30ce850b78371fd1386338350591f9ff494Brian Gaeke 355c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// Error - Simple wrapper function to conditionally assign to E and return true. 365c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// This just makes error return conditions a little bit simpler... 376e0d1cb30957a636c53158d3089e6fb88348a57aDaniel Dunbarstatic inline bool Error(std::string *E, const Twine &Message) { 386e0d1cb30957a636c53158d3089e6fb88348a57aDaniel Dunbar if (E) *E = Message.str(); 395c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner return true; 405c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner} 415c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 42700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Function: ResolveTypes() 43700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// 44700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Description: 45700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Attempt to link the two specified types together. 46700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// 47700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Inputs: 48700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// DestTy - The type to which we wish to resolve. 49700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// SrcTy - The original type which we want to resolve. 50700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// 51700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Outputs: 52700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// DestST - The symbol table in which the new type should be placed. 53700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// 54700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Return value: 55700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// true - There is an error and the types cannot yet be linked. 56700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// false - No errors. 574c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner// 58bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattnerstatic bool ResolveTypes(const Type *DestTy, const Type *SrcTy) { 59e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner if (DestTy == SrcTy) return false; // If already equal, noop 60bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner assert(DestTy && SrcTy && "Can't handle null types"); 61e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner 62bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner if (const OpaqueType *OT = dyn_cast<OpaqueType>(DestTy)) { 63bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner // Type _is_ in module, just opaque... 64bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner const_cast<OpaqueType*>(OT)->refineAbstractTypeTo(SrcTy); 65bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner } else if (const OpaqueType *OT = dyn_cast<OpaqueType>(SrcTy)) { 66bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner const_cast<OpaqueType*>(OT)->refineAbstractTypeTo(DestTy); 67bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner } else { 68bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner return true; // Cannot link types... not-equal and neither is opaque. 694c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner } 704c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner return false; 714c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner} 724c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner 7362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner/// LinkerTypeMap - This implements a map of types that is stable 7462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner/// even if types are resolved/refined to other types. This is not a general 7562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner/// purpose map, it is specific to the linker's use. 7662a81a1eba019ab570b002f8e1686494139785a1Chris Lattnernamespace { 7762a81a1eba019ab570b002f8e1686494139785a1Chris Lattnerclass LinkerTypeMap : public AbstractTypeUser { 7862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner typedef DenseMap<const Type*, PATypeHolder> TheMapTy; 7962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner TheMapTy TheMap; 80eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 81fc196f9ee909ac35837f6f771a42aa0dec817584Chris Lattner LinkerTypeMap(const LinkerTypeMap&); // DO NOT IMPLEMENT 82fc196f9ee909ac35837f6f771a42aa0dec817584Chris Lattner void operator=(const LinkerTypeMap&); // DO NOT IMPLEMENT 83fc196f9ee909ac35837f6f771a42aa0dec817584Chris Lattnerpublic: 84fc196f9ee909ac35837f6f771a42aa0dec817584Chris Lattner LinkerTypeMap() {} 85fc196f9ee909ac35837f6f771a42aa0dec817584Chris Lattner ~LinkerTypeMap() { 8662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner for (DenseMap<const Type*, PATypeHolder>::iterator I = TheMap.begin(), 8762a81a1eba019ab570b002f8e1686494139785a1Chris Lattner E = TheMap.end(); I != E; ++I) 8862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner I->first->removeAbstractTypeUser(this); 8962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner } 90eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 9162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner /// lookup - Return the value for the specified type or null if it doesn't 9262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner /// exist. 9362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner const Type *lookup(const Type *Ty) const { 9462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner TheMapTy::const_iterator I = TheMap.find(Ty); 9562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner if (I != TheMap.end()) return I->second; 9662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner return 0; 9762a81a1eba019ab570b002f8e1686494139785a1Chris Lattner } 98eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 9962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner /// erase - Remove the specified type, returning true if it was in the set. 10062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner bool erase(const Type *Ty) { 10162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner if (!TheMap.erase(Ty)) 10262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner return false; 10362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner if (Ty->isAbstract()) 10462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner Ty->removeAbstractTypeUser(this); 10562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner return true; 10662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner } 107eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 10862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner /// insert - This returns true if the pointer was new to the set, false if it 10962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner /// was already in the set. 11062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner bool insert(const Type *Src, const Type *Dst) { 1116b345ee9b2833cf1b2f79dc16d06d4060bec36efDan Gohman if (!TheMap.insert(std::make_pair(Src, PATypeHolder(Dst))).second) 11262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner return false; // Already in map. 11362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner if (Src->isAbstract()) 11462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner Src->addAbstractTypeUser(this); 11562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner return true; 11662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner } 117eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 11862a81a1eba019ab570b002f8e1686494139785a1Chris Lattnerprotected: 11962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner /// refineAbstractType - The callback method invoked when an abstract type is 12062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner /// resolved to another type. An object must override this method to update 12162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner /// its internal state to reference NewType instead of OldType. 12262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner /// 12362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner virtual void refineAbstractType(const DerivedType *OldTy, 12462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner const Type *NewTy) { 12562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner TheMapTy::iterator I = TheMap.find(OldTy); 12662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner const Type *DstTy = I->second; 127eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 12862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner TheMap.erase(I); 12962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner if (OldTy->isAbstract()) 13062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner OldTy->removeAbstractTypeUser(this); 13162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner 13262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner // Don't reinsert into the map if the key is concrete now. 13362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner if (NewTy->isAbstract()) 13462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner insert(NewTy, DstTy); 13562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner } 136eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 13762a81a1eba019ab570b002f8e1686494139785a1Chris Lattner /// The other case which AbstractTypeUsers must be aware of is when a type 13862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner /// makes the transition from being abstract (where it has clients on it's 13962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner /// AbstractTypeUsers list) to concrete (where it does not). This method 14062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner /// notifies ATU's when this occurs for a type. 14162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner virtual void typeBecameConcrete(const DerivedType *AbsTy) { 14262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner TheMap.erase(AbsTy); 14362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner AbsTy->removeAbstractTypeUser(this); 14462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner } 145eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 14662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner // for debugging... 14762a81a1eba019ab570b002f8e1686494139785a1Chris Lattner virtual void dump() const { 14862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner cerr << "AbstractTypeSet!\n"; 14962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner } 15062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner}; 15162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner} 15262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner 15362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner 154e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner// RecursiveResolveTypes - This is just like ResolveTypes, except that it 155e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner// recurses down into derived types, merging the used types if the parent types 156e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner// are compatible. 157a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattnerstatic bool RecursiveResolveTypesI(const Type *DstTy, const Type *SrcTy, 15862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner LinkerTypeMap &Pointers) { 159a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner if (DstTy == SrcTy) return false; // If already equal, noop 160f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman 161e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner // If we found our opaque type, resolve it now! 162a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner if (isa<OpaqueType>(DstTy) || isa<OpaqueType>(SrcTy)) 163a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner return ResolveTypes(DstTy, SrcTy); 164f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman 165e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner // Two types cannot be resolved together if they are of different primitive 166e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner // type. For example, we cannot resolve an int to a float. 167a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner if (DstTy->getTypeID() != SrcTy->getTypeID()) return true; 168e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner 16956539659eb1b192d493aa333b60213889129b9f1Chris Lattner // If neither type is abstract, then they really are just different types. 170a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner if (!DstTy->isAbstract() && !SrcTy->isAbstract()) 17156539659eb1b192d493aa333b60213889129b9f1Chris Lattner return true; 172eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 173e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner // Otherwise, resolve the used type used by this derived type... 174a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner switch (DstTy->getTypeID()) { 175f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner default: 176f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner return true; 177e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner case Type::FunctionTyID: { 178a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner const FunctionType *DstFT = cast<FunctionType>(DstTy); 179a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner const FunctionType *SrcFT = cast<FunctionType>(SrcTy); 1809ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner if (DstFT->isVarArg() != SrcFT->isVarArg() || 1819ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner DstFT->getNumContainedTypes() != SrcFT->getNumContainedTypes()) 18243f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner return true; 183eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 184a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner // Use TypeHolder's so recursive resolution won't break us. 185a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner PATypeHolder ST(SrcFT), DT(DstFT); 186a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner for (unsigned i = 0, e = DstFT->getNumContainedTypes(); i != e; ++i) { 187a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner const Type *SE = ST->getContainedType(i), *DE = DT->getContainedType(i); 188a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner if (SE != DE && RecursiveResolveTypesI(DE, SE, Pointers)) 189e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner return true; 190a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner } 191e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner return false; 192e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner } 193e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner case Type::StructTyID: { 194a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner const StructType *DstST = cast<StructType>(DstTy); 195a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner const StructType *SrcST = cast<StructType>(SrcTy); 1969ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner if (DstST->getNumContainedTypes() != SrcST->getNumContainedTypes()) 197f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner return true; 198eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 199a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner PATypeHolder ST(SrcST), DT(DstST); 200a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner for (unsigned i = 0, e = DstST->getNumContainedTypes(); i != e; ++i) { 201a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner const Type *SE = ST->getContainedType(i), *DE = DT->getContainedType(i); 202a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner if (SE != DE && RecursiveResolveTypesI(DE, SE, Pointers)) 203e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner return true; 204a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner } 205e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner return false; 206e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner } 207e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner case Type::ArrayTyID: { 208a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner const ArrayType *DAT = cast<ArrayType>(DstTy); 209a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner const ArrayType *SAT = cast<ArrayType>(SrcTy); 210e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner if (DAT->getNumElements() != SAT->getNumElements()) return true; 211e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner return RecursiveResolveTypesI(DAT->getElementType(), SAT->getElementType(), 212bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner Pointers); 213e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner } 214f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner case Type::VectorTyID: { 215a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner const VectorType *DVT = cast<VectorType>(DstTy); 216a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner const VectorType *SVT = cast<VectorType>(SrcTy); 217f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner if (DVT->getNumElements() != SVT->getNumElements()) return true; 218f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner return RecursiveResolveTypesI(DVT->getElementType(), SVT->getElementType(), 219f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner Pointers); 220f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner } 221e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner case Type::PointerTyID: { 222a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner const PointerType *DstPT = cast<PointerType>(DstTy); 223a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner const PointerType *SrcPT = cast<PointerType>(SrcTy); 224eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 2259ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner if (DstPT->getAddressSpace() != SrcPT->getAddressSpace()) 2269ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner return true; 227eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 228e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner // If this is a pointer type, check to see if we have already seen it. If 229e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner // so, we are in a recursive branch. Cut off the search now. We cannot use 230e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner // an associative container for this search, because the type pointers (keys 23162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner // in the container) change whenever types get resolved. 23262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner if (SrcPT->isAbstract()) 23362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner if (const Type *ExistingDestTy = Pointers.lookup(SrcPT)) 23462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner return ExistingDestTy != DstPT; 235eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 23662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner if (DstPT->isAbstract()) 23762a81a1eba019ab570b002f8e1686494139785a1Chris Lattner if (const Type *ExistingSrcTy = Pointers.lookup(DstPT)) 23862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner return ExistingSrcTy != SrcPT; 239e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner // Otherwise, add the current pointers to the vector to stop recursion on 240e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner // this pair. 24162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner if (DstPT->isAbstract()) 24262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner Pointers.insert(DstPT, SrcPT); 24362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner if (SrcPT->isAbstract()) 24462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner Pointers.insert(SrcPT, DstPT); 245eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 2469ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner return RecursiveResolveTypesI(DstPT->getElementType(), 2479ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner SrcPT->getElementType(), Pointers); 248e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner } 249f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman } 250e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner} 251e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner 252a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattnerstatic bool RecursiveResolveTypes(const Type *DestTy, const Type *SrcTy) { 25362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner LinkerTypeMap PointerTypes; 254bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner return RecursiveResolveTypesI(DestTy, SrcTy, PointerTypes); 255e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner} 256e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner 257e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner 2582c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner// LinkTypes - Go through the symbol table of the Src module and see if any 2592c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner// types are named in the src module that are not named in the Dst module. 2602c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner// Make sure there are no type name conflicts. 2615c2d335d86579941b71816196a95a64b277e8963Chris Lattnerstatic bool LinkTypes(Module *Dest, const Module *Src, std::string *Err) { 26278d033e086e19e016273de014f9214aa6f3f844bReid Spencer TypeSymbolTable *DestST = &Dest->getTypeSymbolTable(); 26378d033e086e19e016273de014f9214aa6f3f844bReid Spencer const TypeSymbolTable *SrcST = &Src->getTypeSymbolTable(); 2642c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner 2652c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner // Look for a type plane for Type's... 26678d033e086e19e016273de014f9214aa6f3f844bReid Spencer TypeSymbolTable::const_iterator TI = SrcST->begin(); 26778d033e086e19e016273de014f9214aa6f3f844bReid Spencer TypeSymbolTable::const_iterator TE = SrcST->end(); 268567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer if (TI == TE) return false; // No named types, do nothing. 2692c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner 270cf00c4ab3ba308d45d98c5ccab87362cf802facbMisha Brukman // Some types cannot be resolved immediately because they depend on other 271cf00c4ab3ba308d45d98c5ccab87362cf802facbMisha Brukman // types being resolved to each other first. This contains a list of types we 272cf00c4ab3ba308d45d98c5ccab87362cf802facbMisha Brukman // are waiting to recheck. 2734c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner std::vector<std::string> DelayedTypesToResolve; 2744c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner 275567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer for ( ; TI != TE; ++TI ) { 276567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer const std::string &Name = TI->first; 277c28a224c68adf0464c275778af4bce762ad25d37Reid Spencer const Type *RHS = TI->second; 2782c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner 279bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner // Check to see if this type name is already in the dest module. 28078d033e086e19e016273de014f9214aa6f3f844bReid Spencer Type *Entry = DestST->lookup(Name); 2814c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner 282bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner // If the name is just in the source module, bring it over to the dest. 283bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner if (Entry == 0) { 284bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner if (!Name.empty()) 285bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner DestST->insert(Name, const_cast<Type*>(RHS)); 286bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner } else if (ResolveTypes(Entry, RHS)) { 2874c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner // They look different, save the types 'till later to resolve. 2884c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner DelayedTypesToResolve.push_back(Name); 2894c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner } 2904c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner } 2914c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner 2924c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner // Iteratively resolve types while we can... 2934c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner while (!DelayedTypesToResolve.empty()) { 2944c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner // Loop over all of the types, attempting to resolve them if possible... 2954c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner unsigned OldSize = DelayedTypesToResolve.size(); 2964c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner 297e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner // Try direct resolution by name... 2984c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner for (unsigned i = 0; i != DelayedTypesToResolve.size(); ++i) { 2994c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner const std::string &Name = DelayedTypesToResolve[i]; 30078d033e086e19e016273de014f9214aa6f3f844bReid Spencer Type *T1 = SrcST->lookup(Name); 30178d033e086e19e016273de014f9214aa6f3f844bReid Spencer Type *T2 = DestST->lookup(Name); 302bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner if (!ResolveTypes(T2, T1)) { 3034c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner // We are making progress! 3044c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner DelayedTypesToResolve.erase(DelayedTypesToResolve.begin()+i); 3054c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner --i; 3062f6bb2bce13084dfc4c8acf47f9eae0578267aaaChris Lattner } 3074c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner } 3082f6bb2bce13084dfc4c8acf47f9eae0578267aaaChris Lattner 3094c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner // Did we not eliminate any types? 3104c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner if (DelayedTypesToResolve.size() == OldSize) { 311e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner // Attempt to resolve subelements of types. This allows us to merge these 312e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner // two types: { int* } and { opaque* } 3134c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner for (unsigned i = 0, e = DelayedTypesToResolve.size(); i != e; ++i) { 3144c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner const std::string &Name = DelayedTypesToResolve[i]; 315a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner if (!RecursiveResolveTypes(SrcST->lookup(Name), DestST->lookup(Name))) { 316e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner // We are making progress! 317e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner DelayedTypesToResolve.erase(DelayedTypesToResolve.begin()+i); 318f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman 319e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner // Go back to the main loop, perhaps we can resolve directly by name 320e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner // now... 321e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner break; 322e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner } 323e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner } 324e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner 325e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner // If we STILL cannot resolve the types, then there is something wrong. 326e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner if (DelayedTypesToResolve.size() == OldSize) { 327aeb18ce9c012f8554f48bae9bb32351c6eeaa26cChris Lattner // Remove the symbol name from the destination. 328aeb18ce9c012f8554f48bae9bb32351c6eeaa26cChris Lattner DelayedTypesToResolve.pop_back(); 3294c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner } 3302c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner } 3312c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner } 3324c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner 3334c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner 3342c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner return false; 3352c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner} 3362c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner 3370bb8757997392ee18ca9de0f1fab29463f8e7aebChris Lattner#ifndef NDEBUG 3385c2d335d86579941b71816196a95a64b277e8963Chris Lattnerstatic void PrintMap(const std::map<const Value*, Value*> &M) { 3395c2d335d86579941b71816196a95a64b277e8963Chris Lattner for (std::map<const Value*, Value*>::const_iterator I = M.begin(), E =M.end(); 3402d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner I != E; ++I) { 341e81561909d128c6e2d8033cb5465a49b2596b26aBill Wendling cerr << " Fr: " << (void*)I->first << " "; 34287182ae6baf28ca6da6ee4bc00cf1857f293acfeChris Lattner I->first->dump(); 343e81561909d128c6e2d8033cb5465a49b2596b26aBill Wendling cerr << " To: " << (void*)I->second << " "; 34487182ae6baf28ca6da6ee4bc00cf1857f293acfeChris Lattner I->second->dump(); 345e81561909d128c6e2d8033cb5465a49b2596b26aBill Wendling cerr << "\n"; 3462d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner } 3472d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner} 3480bb8757997392ee18ca9de0f1fab29463f8e7aebChris Lattner#endif 3492d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner 3502d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner 351619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer// RemapOperand - Use ValueMap to convert constants from one module to another. 3525c2d335d86579941b71816196a95a64b277e8963Chris Lattnerstatic Value *RemapOperand(const Value *In, 353c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson std::map<const Value*, Value*> &ValueMap, 354c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson LLVMContext &Context) { 3550033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner std::map<const Value*,Value*>::const_iterator I = ValueMap.find(In); 356eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov if (I != ValueMap.end()) 357ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer return I->second; 3585c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 359619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer // Check to see if it's a constant that we are interested in transforming. 360620fd68b1086b4d99b23951357299bee357aa62aChris Lattner Value *Result = 0; 36118961504fc2b299578dba817900a0696cf3ccc4dChris Lattner if (const Constant *CPV = dyn_cast<Constant>(In)) { 362de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner if ((!isa<DerivedType>(CPV->getType()) && !isa<ConstantExpr>(CPV)) || 363a54b7cbd452b3adb2f51346140d996b29c2cdb30Reid Spencer isa<ConstantInt>(CPV) || isa<ConstantAggregateZero>(CPV)) 3640033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner return const_cast<Constant*>(CPV); // Simple constants stay identical. 3658d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner 36618961504fc2b299578dba817900a0696cf3ccc4dChris Lattner if (const ConstantArray *CPA = dyn_cast<ConstantArray>(CPV)) { 367cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos std::vector<Constant*> Operands(CPA->getNumOperands()); 368cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos for (unsigned i = 0, e = CPA->getNumOperands(); i != e; ++i) 369c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson Operands[i] =cast<Constant>(RemapOperand(CPA->getOperand(i), ValueMap, 370c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson Context)); 371c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson Result = 3721fd7096407d5e598ed3366a1141548e71273f1c5Owen Anderson ConstantArray::get(cast<ArrayType>(CPA->getType()), Operands); 37318961504fc2b299578dba817900a0696cf3ccc4dChris Lattner } else if (const ConstantStruct *CPS = dyn_cast<ConstantStruct>(CPV)) { 374cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos std::vector<Constant*> Operands(CPS->getNumOperands()); 375cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos for (unsigned i = 0, e = CPS->getNumOperands(); i != e; ++i) 376c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson Operands[i] =cast<Constant>(RemapOperand(CPS->getOperand(i), ValueMap, 377c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson Context)); 378c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson Result = 3798fa3338ed2400c1352b137613d2c2c70d1ead695Owen Anderson ConstantStruct::get(cast<StructType>(CPS->getType()), Operands); 380b976e668165e1875a8f1eb7af800e33bb1e4393dChris Lattner } else if (isa<ConstantPointerNull>(CPV) || isa<UndefValue>(CPV)) { 38118961504fc2b299578dba817900a0696cf3ccc4dChris Lattner Result = const_cast<Constant*>(CPV); 3829d6565a5b1fbc4286d6ee638d8f47a3171a9ed7eReid Spencer } else if (const ConstantVector *CP = dyn_cast<ConstantVector>(CPV)) { 383a88eb9287eb14ba62ca6324b2b249e0ae9fe3309Chris Lattner std::vector<Constant*> Operands(CP->getNumOperands()); 384a88eb9287eb14ba62ca6324b2b249e0ae9fe3309Chris Lattner for (unsigned i = 0, e = CP->getNumOperands(); i != e; ++i) 385c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson Operands[i] = cast<Constant>(RemapOperand(CP->getOperand(i), ValueMap, 386c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson Context)); 387af7ec975870f92245f1f1484ac80a1e2db6a0afaOwen Anderson Result = ConstantVector::get(Operands); 3886cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner } else if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(CPV)) { 38927d672136a57519095a3ef20161697749cffb57cChris Lattner std::vector<Constant*> Ops; 39027d672136a57519095a3ef20161697749cffb57cChris Lattner for (unsigned i = 0, e = CE->getNumOperands(); i != e; ++i) 391c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson Ops.push_back(cast<Constant>(RemapOperand(CE->getOperand(i),ValueMap, 392c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson Context))); 39327d672136a57519095a3ef20161697749cffb57cChris Lattner Result = CE->getWithOperands(Ops); 3948d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner } else { 3950bb8757997392ee18ca9de0f1fab29463f8e7aebChris Lattner assert(!isa<GlobalValue>(CPV) && "Unmapped global?"); 396c23197a26f34f559ea9797de51e187087c039c42Torok Edwin llvm_unreachable("Unknown type of derived type constant value!"); 3978d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner } 398620fd68b1086b4d99b23951357299bee357aa62aChris Lattner } else if (isa<InlineAsm>(In)) { 399620fd68b1086b4d99b23951357299bee357aa62aChris Lattner Result = const_cast<Value*>(In); 400620fd68b1086b4d99b23951357299bee357aa62aChris Lattner } 401eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 402619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer // Cache the mapping in our local map structure 403620fd68b1086b4d99b23951357299bee357aa62aChris Lattner if (Result) { 404817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov ValueMap[In] = Result; 4058d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner return Result; 4068d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner } 407eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 4080bb8757997392ee18ca9de0f1fab29463f8e7aebChris Lattner#ifndef NDEBUG 409e81561909d128c6e2d8033cb5465a49b2596b26aBill Wendling cerr << "LinkModules ValueMap: \n"; 4100033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner PrintMap(ValueMap); 4112d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner 412e81561909d128c6e2d8033cb5465a49b2596b26aBill Wendling cerr << "Couldn't remap value: " << (void*)In << " " << *In << "\n"; 413c23197a26f34f559ea9797de51e187087c039c42Torok Edwin llvm_unreachable("Couldn't remap value!"); 4140bb8757997392ee18ca9de0f1fab29463f8e7aebChris Lattner#endif 4158d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner return 0; 4165c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner} 4175c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 4188bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer/// ForceRenaming - The LLVM SymbolTable class autorenames globals that conflict 4198bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer/// in the symbol table. This is good for all clients except for us. Go 4208bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer/// through the trouble to force this back. 421c003628a612d3687fb77088a5894314210a65385Chris Lattnerstatic void ForceRenaming(GlobalValue *GV, const std::string &Name) { 422c003628a612d3687fb77088a5894314210a65385Chris Lattner assert(GV->getName() != Name && "Can't force rename to self"); 423ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer ValueSymbolTable &ST = GV->getParent()->getValueSymbolTable(); 424c003628a612d3687fb77088a5894314210a65385Chris Lattner 425c003628a612d3687fb77088a5894314210a65385Chris Lattner // If there is a conflict, rename the conflict. 42633f294930e0affa51c3818405741abd80cc90316Chris Lattner if (GlobalValue *ConflictGV = cast_or_null<GlobalValue>(ST.lookup(Name))) { 427bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola assert(ConflictGV->hasLocalLinkage() && 428ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer "Not conflicting with a static global, should link instead!"); 42933f294930e0affa51c3818405741abd80cc90316Chris Lattner GV->takeName(ConflictGV); 43033f294930e0affa51c3818405741abd80cc90316Chris Lattner ConflictGV->setName(Name); // This will cause ConflictGV to get renamed 431ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer assert(ConflictGV->getName() != Name && "ForceRenaming didn't work"); 43233f294930e0affa51c3818405741abd80cc90316Chris Lattner } else { 43333f294930e0affa51c3818405741abd80cc90316Chris Lattner GV->setName(Name); // Force the name back 434ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer } 435ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer} 436ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer 437ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer/// CopyGVAttributes - copy additional attributes (those not needed to construct 438eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov/// a GlobalValue) from the SrcGV to the DestGV. 439ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencerstatic void CopyGVAttributes(GlobalValue *DestGV, const GlobalValue *SrcGV) { 44028c3cff8250b3fe2adc6479306fe7dbdb48a1bdbDuncan Sands // Use the maximum alignment, rather than just copying the alignment of SrcGV. 44128c3cff8250b3fe2adc6479306fe7dbdb48a1bdbDuncan Sands unsigned Alignment = std::max(DestGV->getAlignment(), SrcGV->getAlignment()); 44228c3cff8250b3fe2adc6479306fe7dbdb48a1bdbDuncan Sands DestGV->copyAttributesFrom(SrcGV); 44328c3cff8250b3fe2adc6479306fe7dbdb48a1bdbDuncan Sands DestGV->setAlignment(Alignment); 444c003628a612d3687fb77088a5894314210a65385Chris Lattner} 445c003628a612d3687fb77088a5894314210a65385Chris Lattner 446aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// GetLinkageResult - This analyzes the two global values and determines what 447aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// the result will look like in the destination module. In particular, it 448aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// computes the resultant linkage type, computes whether the global in the 449aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// source should be copied over to the destination (replacing the existing 4509cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov/// one), and computes whether this linkage is an error or not. It also performs 4519cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov/// visibility checks: we cannot link together two symbols with different 4529cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov/// visibilities. 453968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikovstatic bool GetLinkageResult(GlobalValue *Dest, const GlobalValue *Src, 454aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner GlobalValue::LinkageTypes <, bool &LinkFromSrc, 455aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner std::string *Err) { 456bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola assert((!Dest || !Src->hasLocalLinkage()) && 457aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner "If Src has internal linkage, Dest shouldn't be set!"); 458aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner if (!Dest) { 459aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner // Linking something to nothing. 460aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner LinkFromSrc = true; 461aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner LT = Src->getLinkage(); 4625cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer } else if (Src->isDeclaration()) { 4632b48ef0450944c2c46633aec9baf6be835a3b503Anton Korobeynikov // If Src is external or if both Src & Dest are external.. Just link the 464aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner // external globals, we aren't adding anything. 465b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov if (Src->hasDLLImportLinkage()) { 46678ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov // If one of GVs has DLLImport linkage, result should be dllimport'ed. 4675cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer if (Dest->isDeclaration()) { 468b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov LinkFromSrc = true; 469b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov LT = Src->getLinkage(); 470eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov } 4718753c447a5f8eec1421331c40ffbb72c6d411f78Andrew Lenharth } else if (Dest->hasExternalWeakLinkage()) { 472667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands // If the Dest is weak, use the source linkage. 4738753c447a5f8eec1421331c40ffbb72c6d411f78Andrew Lenharth LinkFromSrc = true; 4748753c447a5f8eec1421331c40ffbb72c6d411f78Andrew Lenharth LT = Src->getLinkage(); 475b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov } else { 476b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov LinkFromSrc = false; 477b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov LT = Dest->getLinkage(); 478b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov } 4795cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer } else if (Dest->isDeclaration() && !Dest->hasDLLImportLinkage()) { 480aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner // If Dest is external but Src is not: 481aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner LinkFromSrc = true; 482aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner LT = Src->getLinkage(); 483aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner } else if (Src->hasAppendingLinkage() || Dest->hasAppendingLinkage()) { 484aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner if (Src->getLinkage() != Dest->getLinkage()) 485aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner return Error(Err, "Linking globals named '" + Src->getName() + 486aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner "': can only link appending global with another appending global!"); 487aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner LinkFromSrc = true; // Special cased. 488aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner LT = Src->getLinkage(); 489a05ef5e107943478ad02bc0cd5170a3894076bc4Duncan Sands } else if (Src->isWeakForLinker()) { 490aafce77b17d340aace52bcd49d1944109d82f14aDale Johannesen // At this point we know that Dest has LinkOnce, External*, Weak, Common, 491aafce77b17d340aace52bcd49d1944109d82f14aDale Johannesen // or DLL* linkage. 492266c7bbbbcc4b326dea82e577de1a415d6acc23eChris Lattner if (Dest->hasExternalWeakLinkage() || 493266c7bbbbcc4b326dea82e577de1a415d6acc23eChris Lattner Dest->hasAvailableExternallyLinkage() || 494266c7bbbbcc4b326dea82e577de1a415d6acc23eChris Lattner (Dest->hasLinkOnceLinkage() && 495266c7bbbbcc4b326dea82e577de1a415d6acc23eChris Lattner (Src->hasWeakLinkage() || Src->hasCommonLinkage()))) { 496aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner LinkFromSrc = true; 497aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner LT = Src->getLinkage(); 498aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner } else { 499aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner LinkFromSrc = false; 500aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner LT = Dest->getLinkage(); 501aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner } 502a05ef5e107943478ad02bc0cd5170a3894076bc4Duncan Sands } else if (Dest->isWeakForLinker()) { 50378ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov // At this point we know that Src has External* or DLL* linkage. 50478ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov if (Src->hasExternalWeakLinkage()) { 50578ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov LinkFromSrc = false; 50678ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov LT = Dest->getLinkage(); 50778ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov } else { 50878ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov LinkFromSrc = true; 50978ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov LT = GlobalValue::ExternalLinkage; 51078ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov } 511aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner } else { 512b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov assert((Dest->hasExternalLinkage() || 513b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov Dest->hasDLLImportLinkage() || 51478ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov Dest->hasDLLExportLinkage() || 51578ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov Dest->hasExternalWeakLinkage()) && 516b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov (Src->hasExternalLinkage() || 517b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov Src->hasDLLImportLinkage() || 51878ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov Src->hasDLLExportLinkage() || 51978ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov Src->hasExternalWeakLinkage()) && 520aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner "Unexpected linkage type!"); 521f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman return Error(Err, "Linking globals named '" + Src->getName() + 522aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner "': symbol multiply defined!"); 523aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner } 5249cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov 5259cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov // Check visibility 5269cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov if (Dest && Src->getVisibility() != Dest->getVisibility()) 52797f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner if (!Src->isDeclaration() && !Dest->isDeclaration()) 52897f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner return Error(Err, "Linking globals named '" + Src->getName() + 52997f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner "': symbols have different visibilities!"); 530aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner return false; 531aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner} 5325c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 5335c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// LinkGlobals - Loop through the global variables in the src module and merge 5348166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner// them into the dest module. 535968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikovstatic bool LinkGlobals(Module *Dest, const Module *Src, 5365c2d335d86579941b71816196a95a64b277e8963Chris Lattner std::map<const Value*, Value*> &ValueMap, 5378166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner std::multimap<std::string, GlobalVariable *> &AppendingVars, 5385c2d335d86579941b71816196a95a64b277e8963Chris Lattner std::string *Err) { 539d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner ValueSymbolTable &DestSymTab = Dest->getValueSymbolTable(); 540eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 5415c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner // Loop over all of the globals in the src module, mapping them over as we go 5420bb8757997392ee18ca9de0f1fab29463f8e7aebChris Lattner for (Module::const_global_iterator I = Src->global_begin(), 5430bb8757997392ee18ca9de0f1fab29463f8e7aebChris Lattner E = Src->global_end(); I != E; ++I) { 544968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov const GlobalVariable *SGV = I; 54501f69399352380532d480f6749daed660f50b15cAnton Korobeynikov GlobalValue *DGV = 0; 54601f69399352380532d480f6749daed660f50b15cAnton Korobeynikov 547d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner // Check to see if may have to link the global with the global, alias or 548d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner // function. 549bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola if (SGV->hasName() && !SGV->hasLocalLinkage()) 55003d7651c3652e1f0cc86e79b26585d86818da9cfDaniel Dunbar DGV = cast_or_null<GlobalValue>(DestSymTab.lookup(SGV->getName())); 551eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 552ae1132d2b8ae07afd2fe9a7cb434d849f884bfa0Chris Lattner // If we found a global with the same name in the dest module, but it has 553ae1132d2b8ae07afd2fe9a7cb434d849f884bfa0Chris Lattner // internal linkage, we are really not doing any linkage here. 554bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola if (DGV && DGV->hasLocalLinkage()) 555ae1132d2b8ae07afd2fe9a7cb434d849f884bfa0Chris Lattner DGV = 0; 556eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 557d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner // If types don't agree due to opaque types, try to resolve them. 558d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner if (DGV && DGV->getType() != SGV->getType()) 559d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner RecursiveResolveTypes(SGV->getType(), DGV->getType()); 56001f69399352380532d480f6749daed660f50b15cAnton Korobeynikov 561c318329a1098a19137398af4d880f4d8cdd9cd8dDan Gohman assert((SGV->hasInitializer() || SGV->hasExternalWeakLinkage() || 562c318329a1098a19137398af4d880f4d8cdd9cd8dDan Gohman SGV->hasExternalLinkage() || SGV->hasDLLImportLinkage()) && 5634ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner "Global must either be external or have an initializer!"); 5645c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 565b324bd735ff5c99a52cd3c3f5d01c0e1398a2d3aChris Lattner GlobalValue::LinkageTypes NewLinkage = GlobalValue::InternalLinkage; 566b324bd735ff5c99a52cd3c3f5d01c0e1398a2d3aChris Lattner bool LinkFromSrc = false; 567aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner if (GetLinkageResult(DGV, SGV, NewLinkage, LinkFromSrc, Err)) 568aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner return true; 5690fec08eb58dd9fffeb72c584aa61a59d71111c8dChris Lattner 5706157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner if (DGV == 0) { 5715c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner // No linking to be performed, simply create an identical version of the 5728d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner // symbol over in the dest module... the initializer will be filled in 573d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner // later by LinkGlobalInits. 5742719bac90525985f4ebb97324b4c087d47b54384Chris Lattner GlobalVariable *NewDGV = 575e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson new GlobalVariable(*Dest, SGV->getType()->getElementType(), 5762719bac90525985f4ebb97324b4c087d47b54384Chris Lattner SGV->isConstant(), SGV->getLinkage(), /*init*/0, 577e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson SGV->getName(), 0, false, 578a534b0f7bf0248bf90afe97110a34777e972f325Chris Lattner SGV->getType()->getAddressSpace()); 579471feac1b297b9458de18d973af371bf7f10296bReid Spencer // Propagate alignment, visibility and section info. 580ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer CopyGVAttributes(NewDGV, SGV); 5815dfbaf1fe3fa6914095cc55cde20b1ed25187b56Andrew Lenharth 5822719bac90525985f4ebb97324b4c087d47b54384Chris Lattner // If the LLVM runtime renamed the global, but it is an externally visible 5832719bac90525985f4ebb97324b4c087d47b54384Chris Lattner // symbol, DGV must be an existing global with internal linkage. Rename 5842719bac90525985f4ebb97324b4c087d47b54384Chris Lattner // it. 585bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola if (!NewDGV->hasLocalLinkage() && NewDGV->getName() != SGV->getName()) 586c003628a612d3687fb77088a5894314210a65385Chris Lattner ForceRenaming(NewDGV, SGV->getName()); 5875c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 5886157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // Make sure to remember this mapping. 589817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov ValueMap[SGV] = NewDGV; 590817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov 591d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner // Keep track that this is an appending variable. 5928166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner if (SGV->hasAppendingLinkage()) 5938166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner AppendingVars.insert(std::make_pair(SGV->getName(), NewDGV)); 5946157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner continue; 5956157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner } 596eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 5976157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // If the visibilities of the symbols disagree and the destination is a 5986157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // prototype, take the visibility of its input. 5996157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner if (DGV->isDeclaration()) 6006157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner DGV->setVisibility(SGV->getVisibility()); 601eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 6026157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner if (DGV->hasAppendingLinkage()) { 6038166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // No linking is performed yet. Just insert a new copy of the global, and 6048166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // keep track of the fact that it is an appending variable in the 6058166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // AppendingVars map. The name is cleared out so that no linkage is 6068166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // performed. 6078166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner GlobalVariable *NewDGV = 608e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson new GlobalVariable(*Dest, SGV->getType()->getElementType(), 6098166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner SGV->isConstant(), SGV->getLinkage(), /*init*/0, 610e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson "", 0, false, 611a534b0f7bf0248bf90afe97110a34777e972f325Chris Lattner SGV->getType()->getAddressSpace()); 6128166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner 61375c7915c3a24f8063c274ccffe2dc9b7bc8bce1fAnton Korobeynikov // Set alignment allowing CopyGVAttributes merge it with alignment of SGV. 614ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer NewDGV->setAlignment(DGV->getAlignment()); 61575c7915c3a24f8063c274ccffe2dc9b7bc8bce1fAnton Korobeynikov // Propagate alignment, section and visibility info. 616ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer CopyGVAttributes(NewDGV, SGV); 6175dfbaf1fe3fa6914095cc55cde20b1ed25187b56Andrew Lenharth 6188166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // Make sure to remember this mapping... 619817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov ValueMap[SGV] = NewDGV; 6208166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner 6218166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // Keep track that this is an appending variable... 6228166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner AppendingVars.insert(std::make_pair(SGV->getName(), NewDGV)); 6236157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner continue; 6246157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner } 625eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 6266157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner if (LinkFromSrc) { 6276157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner if (isa<GlobalAlias>(DGV)) 6281438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov return Error(Err, "Global-Alias Collision on '" + SGV->getName() + 6291438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov "': symbol multiple defined"); 630eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 631d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner // If the types don't match, and if we are to link from the source, nuke 632d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner // DGV and create a new one of the appropriate type. Note that the thing 633d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner // we are replacing may be a function (if a prototype, weak, etc) or a 634d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner // global variable. 635d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner GlobalVariable *NewDGV = 636e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson new GlobalVariable(*Dest, SGV->getType()->getElementType(), 6373d29df3e8a203b167d8071ea6f805b21db18a5afOwen Anderson SGV->isConstant(), NewLinkage, /*init*/0, 638e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson DGV->getName(), 0, false, 639d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner SGV->getType()->getAddressSpace()); 640eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 641d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner // Propagate alignment, section, and visibility info. 642d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner CopyGVAttributes(NewDGV, SGV); 643baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson DGV->replaceAllUsesWith(ConstantExpr::getBitCast(NewDGV, 644c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson DGV->getType())); 645eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 646d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner // DGV will conflict with NewDGV because they both had the same 647d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner // name. We must erase this now so ForceRenaming doesn't assert 648d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner // because DGV might not have internal linkage. 649d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner if (GlobalVariable *Var = dyn_cast<GlobalVariable>(DGV)) 650d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner Var->eraseFromParent(); 651d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner else 652d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner cast<Function>(DGV)->eraseFromParent(); 653d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner DGV = NewDGV; 654d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner 655d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner // If the symbol table renamed the global, but it is an externally visible 656d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner // symbol, DGV must be an existing global with internal linkage. Rename. 657bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola if (NewDGV->getName() != SGV->getName() && !NewDGV->hasLocalLinkage()) 658d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner ForceRenaming(NewDGV, SGV->getName()); 659eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 6606157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // Inherit const as appropriate. 661d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner NewDGV->setConstant(SGV->isConstant()); 662eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 6636157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // Make sure to remember this mapping. 6646157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner ValueMap[SGV] = NewDGV; 6656157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner continue; 6665c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner } 667eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 6686157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // Not "link from source", keep the one in the DestModule and remap the 6696157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // input onto it. 670eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 6716157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // Special case for const propagation. 6726157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner if (GlobalVariable *DGVar = dyn_cast<GlobalVariable>(DGV)) 6736157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner if (DGVar->isDeclaration() && SGV->isConstant() && !DGVar->isConstant()) 6746157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner DGVar->setConstant(true); 6756157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner 676d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov // SGV is global, but DGV is alias. 677d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov if (isa<GlobalAlias>(DGV)) { 678d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov // The only valid mappings are: 679d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov // - SGV is external declaration, which is effectively a no-op. 680d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov // - SGV is weak, when we just need to throw SGV out. 681a05ef5e107943478ad02bc0cd5170a3894076bc4Duncan Sands if (!SGV->isDeclaration() && !SGV->isWeakForLinker()) 682d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov return Error(Err, "Global-Alias Collision on '" + SGV->getName() + 683d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov "': symbol multiple defined"); 684d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov } 685eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 6866157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // Set calculated linkage 6876157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner DGV->setLinkage(NewLinkage); 688eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 6896157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // Make sure to remember this mapping... 690baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson ValueMap[SGV] = ConstantExpr::getBitCast(DGV, SGV->getType()); 6915c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner } 6925c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner return false; 6935c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner} 6945c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 69558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikovstatic GlobalValue::LinkageTypes 69658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton KorobeynikovCalculateAliasLinkage(const GlobalValue *SGV, const GlobalValue *DGV) { 697667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands GlobalValue::LinkageTypes SL = SGV->getLinkage(); 698667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands GlobalValue::LinkageTypes DL = DGV->getLinkage(); 699667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands if (SL == GlobalValue::ExternalLinkage || DL == GlobalValue::ExternalLinkage) 70058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov return GlobalValue::ExternalLinkage; 701667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands else if (SL == GlobalValue::WeakAnyLinkage || 702667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands DL == GlobalValue::WeakAnyLinkage) 703667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands return GlobalValue::WeakAnyLinkage; 704667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands else if (SL == GlobalValue::WeakODRLinkage || 705667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands DL == GlobalValue::WeakODRLinkage) 706667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands return GlobalValue::WeakODRLinkage; 707667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands else if (SL == GlobalValue::InternalLinkage && 708667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands DL == GlobalValue::InternalLinkage) 70958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov return GlobalValue::InternalLinkage; 7103d10a5a75794356a0a568ce283713adc3a963200Bill Wendling else if (SL == GlobalValue::LinkerPrivateLinkage && 7113d10a5a75794356a0a568ce283713adc3a963200Bill Wendling DL == GlobalValue::LinkerPrivateLinkage) 7123d10a5a75794356a0a568ce283713adc3a963200Bill Wendling return GlobalValue::LinkerPrivateLinkage; 713bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola else { 714667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands assert (SL == GlobalValue::PrivateLinkage && 715667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands DL == GlobalValue::PrivateLinkage && "Unexpected linkage type"); 716bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola return GlobalValue::PrivateLinkage; 71758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov } 71858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov} 71958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov 72031ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio// LinkAlias - Loop through the alias in the src module and link them into the 72158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov// dest module. We're assuming, that all functions/global variables were already 72258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov// linked in. 7234fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikovstatic bool LinkAlias(Module *Dest, const Module *Src, 7244fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov std::map<const Value*, Value*> &ValueMap, 7254fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov std::string *Err) { 72631ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio // Loop over all alias in the src module 72731ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio for (Module::const_alias_iterator I = Src->alias_begin(), 72831ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio E = Src->alias_end(); I != E; ++I) { 72958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov const GlobalAlias *SGA = I; 73058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov const GlobalValue *SAliasee = SGA->getAliasedGlobal(); 73158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov GlobalAlias *NewGA = NULL; 73258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov 73358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov // Globals were already linked, thus we can just query ValueMap for variant 734caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov // of SAliasee in Dest. 73558d5e051574339106ce4139697ebadd21e34dbc0Ted Kremenek std::map<const Value*,Value*>::const_iterator VMI = ValueMap.find(SAliasee); 73658d5e051574339106ce4139697ebadd21e34dbc0Ted Kremenek assert(VMI != ValueMap.end() && "Aliasee not linked"); 73758d5e051574339106ce4139697ebadd21e34dbc0Ted Kremenek GlobalValue* DAliasee = cast<GlobalValue>(VMI->second); 738caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov GlobalValue* DGV = NULL; 73958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov 74058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov // Try to find something 'similar' to SGA in destination module. 741bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola if (!DGV && !SGA->hasLocalLinkage()) { 742caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov DGV = Dest->getNamedAlias(SGA->getName()); 743caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov 744caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov // If types don't agree due to opaque types, try to resolve them. 745caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov if (DGV && DGV->getType() != SGA->getType()) 7465ed2ba26680f06846e7602ce65432b0a723b1273Chris Lattner RecursiveResolveTypes(SGA->getType(), DGV->getType()); 747caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov } 748caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov 749bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola if (!DGV && !SGA->hasLocalLinkage()) { 750caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov DGV = Dest->getGlobalVariable(SGA->getName()); 751caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov 75258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov // If types don't agree due to opaque types, try to resolve them. 753caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov if (DGV && DGV->getType() != SGA->getType()) 7545ed2ba26680f06846e7602ce65432b0a723b1273Chris Lattner RecursiveResolveTypes(SGA->getType(), DGV->getType()); 755caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov } 75658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov 757bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola if (!DGV && !SGA->hasLocalLinkage()) { 758caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov DGV = Dest->getFunction(SGA->getName()); 759caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov 760caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov // If types don't agree due to opaque types, try to resolve them. 761caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov if (DGV && DGV->getType() != SGA->getType()) 7625ed2ba26680f06846e7602ce65432b0a723b1273Chris Lattner RecursiveResolveTypes(SGA->getType(), DGV->getType()); 763caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov } 764caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov 765caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov // No linking to be performed on internal stuff. 766bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola if (DGV && DGV->hasLocalLinkage()) 767caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov DGV = NULL; 768caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov 769caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov if (GlobalAlias *DGA = dyn_cast_or_null<GlobalAlias>(DGV)) { 770caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov // Types are known to be the same, check whether aliasees equal. As 77158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov // globals are already linked we just need query ValueMap to find the 77258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov // mapping. 77358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov if (DAliasee == DGA->getAliasedGlobal()) { 77458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov // This is just two copies of the same alias. Propagate linkage, if 77558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov // necessary. 77658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov DGA->setLinkage(CalculateAliasLinkage(SGA, DGA)); 77758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov 77858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov NewGA = DGA; 77958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov // Proceed to 'common' steps 78058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov } else 7811438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov return Error(Err, "Alias Collision on '" + SGA->getName()+ 7821438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov "': aliases have different aliasees"); 783caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov } else if (GlobalVariable *DGVar = dyn_cast_or_null<GlobalVariable>(DGV)) { 784f88bc65932e5cf41ba6ca706cd1b1b1f65300248Anton Korobeynikov // The only allowed way is to link alias with external declaration or weak 785f88bc65932e5cf41ba6ca706cd1b1b1f65300248Anton Korobeynikov // symbol.. 786a05ef5e107943478ad02bc0cd5170a3894076bc4Duncan Sands if (DGVar->isDeclaration() || DGVar->isWeakForLinker()) { 787ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov // But only if aliasee is global too... 788ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov if (!isa<GlobalVariable>(DAliasee)) 789caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov return Error(Err, "Global-Alias Collision on '" + SGA->getName() + 790caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov "': aliasee is not global variable"); 791ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov 79258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov NewGA = new GlobalAlias(SGA->getType(), SGA->getLinkage(), 79358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov SGA->getName(), DAliasee, Dest); 79458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov CopyGVAttributes(NewGA, SGA); 79558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov 79658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov // Any uses of DGV need to change to NewGA, with cast, if needed. 797caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov if (SGA->getType() != DGVar->getType()) 798baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson DGVar->replaceAllUsesWith(ConstantExpr::getBitCast(NewGA, 799caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov DGVar->getType())); 80058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov else 801caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov DGVar->replaceAllUsesWith(NewGA); 80258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov 803caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov // DGVar will conflict with NewGA because they both had the same 80458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov // name. We must erase this now so ForceRenaming doesn't assert 80558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov // because DGV might not have internal linkage. 806caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov DGVar->eraseFromParent(); 80758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov 80858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov // Proceed to 'common' steps 80958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov } else 8101438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov return Error(Err, "Global-Alias Collision on '" + SGA->getName() + 8111438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov "': symbol multiple defined"); 812caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov } else if (Function *DF = dyn_cast_or_null<Function>(DGV)) { 813f88bc65932e5cf41ba6ca706cd1b1b1f65300248Anton Korobeynikov // The only allowed way is to link alias with external declaration or weak 814f88bc65932e5cf41ba6ca706cd1b1b1f65300248Anton Korobeynikov // symbol... 815a05ef5e107943478ad02bc0cd5170a3894076bc4Duncan Sands if (DF->isDeclaration() || DF->isWeakForLinker()) { 816ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov // But only if aliasee is function too... 817ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov if (!isa<Function>(DAliasee)) 818caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov return Error(Err, "Function-Alias Collision on '" + SGA->getName() + 819caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov "': aliasee is not function"); 820ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov 821b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov NewGA = new GlobalAlias(SGA->getType(), SGA->getLinkage(), 822b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov SGA->getName(), DAliasee, Dest); 823b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov CopyGVAttributes(NewGA, SGA); 824b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov 825b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov // Any uses of DF need to change to NewGA, with cast, if needed. 826b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov if (SGA->getType() != DF->getType()) 827baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson DF->replaceAllUsesWith(ConstantExpr::getBitCast(NewGA, 828b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov DF->getType())); 829b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov else 830b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov DF->replaceAllUsesWith(NewGA); 831b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov 832b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov // DF will conflict with NewGA because they both had the same 833b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov // name. We must erase this now so ForceRenaming doesn't assert 834b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov // because DF might not have internal linkage. 835b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov DF->eraseFromParent(); 836b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov 837b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov // Proceed to 'common' steps 838b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov } else 8391438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov return Error(Err, "Function-Alias Collision on '" + SGA->getName() + 8401438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov "': symbol multiple defined"); 84158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov } else { 842caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov // No linking to be performed, simply create an identical version of the 843caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov // alias over in the dest module... 84458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov 84558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov NewGA = new GlobalAlias(SGA->getType(), SGA->getLinkage(), 84658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov SGA->getName(), DAliasee, Dest); 84758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov CopyGVAttributes(NewGA, SGA); 84858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov 84958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov // Proceed to 'common' steps 85058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov } 85158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov 85258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov assert(NewGA && "No alias was created in destination module!"); 85358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov 854b8cdaf7099898d7c0eef4769bf6a89f39e7decc2Anton Korobeynikov // If the symbol table renamed the alias, but it is an externally visible 855caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov // symbol, DGA must be an global value with internal linkage. Rename it. 85658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov if (NewGA->getName() != SGA->getName() && 857bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola !NewGA->hasLocalLinkage()) 85858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov ForceRenaming(NewGA, SGA->getName()); 85958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov 86058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov // Remember this mapping so uses in the source module get remapped 86158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov // later by RemapOperand. 862817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov ValueMap[SGA] = NewGA; 86331ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio } 86458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov 86531ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio return false; 86631ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio} 86731ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio 8685c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 8698d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner// LinkGlobalInits - Update the initializers in the Dest module now that all 8708d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner// globals that may be referenced are in Dest. 8718d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattnerstatic bool LinkGlobalInits(Module *Dest, const Module *Src, 8725c2d335d86579941b71816196a95a64b277e8963Chris Lattner std::map<const Value*, Value*> &ValueMap, 8735c2d335d86579941b71816196a95a64b277e8963Chris Lattner std::string *Err) { 8748d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner // Loop over all of the globals in the src module, mapping them over as we go 8751127315562b221010040925980040fd4f65bdb1cChris Lattner for (Module::const_global_iterator I = Src->global_begin(), 8761127315562b221010040925980040fd4f65bdb1cChris Lattner E = Src->global_end(); I != E; ++I) { 87718961504fc2b299578dba817900a0696cf3ccc4dChris Lattner const GlobalVariable *SGV = I; 8788d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner 8798d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner if (SGV->hasInitializer()) { // Only process initialized GV's 8808d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner // Figure out what the initializer looks like in the dest module... 8814ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner Constant *SInit = 882c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson cast<Constant>(RemapOperand(SGV->getInitializer(), ValueMap, 883c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson Dest->getContext())); 884d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov // Grab destination global variable or alias. 885d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov GlobalValue *DGV = cast<GlobalValue>(ValueMap[SGV]->stripPointerCasts()); 886d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov 887d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov // If dest if global variable, check that initializers match. 888d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov if (GlobalVariable *DGVar = dyn_cast<GlobalVariable>(DGV)) { 889d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov if (DGVar->hasInitializer()) { 890d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov if (SGV->hasExternalLinkage()) { 891d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov if (DGVar->getInitializer() != SInit) 892d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov return Error(Err, "Global Variable Collision on '" + 893d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov SGV->getName() + 894d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov "': global variables have different initializers"); 895a05ef5e107943478ad02bc0cd5170a3894076bc4Duncan Sands } else if (DGVar->isWeakForLinker()) { 896d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov // Nothing is required, mapped values will take the new global 897d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov // automatically. 898a05ef5e107943478ad02bc0cd5170a3894076bc4Duncan Sands } else if (SGV->isWeakForLinker()) { 899d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov // Nothing is required, mapped values will take the new global 900d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov // automatically. 901d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov } else if (DGVar->hasAppendingLinkage()) { 902c23197a26f34f559ea9797de51e187087c039c42Torok Edwin llvm_unreachable("Appending linkage unimplemented!"); 903d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov } else { 904c23197a26f34f559ea9797de51e187087c039c42Torok Edwin llvm_unreachable("Unknown linkage!"); 905d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov } 9064ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner } else { 907d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov // Copy the initializer over now... 908d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov DGVar->setInitializer(SInit); 9094ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner } 9108d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner } else { 911d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov // Destination is alias, the only valid situation is when source is 912d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov // weak. Also, note, that we already checked linkage in LinkGlobals(), 913d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov // thus we assert here. 914d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov // FIXME: Should we weaken this assumption, 'dereference' alias and 915d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov // check for initializer of aliasee? 916a05ef5e107943478ad02bc0cd5170a3894076bc4Duncan Sands assert(SGV->isWeakForLinker()); 9178d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner } 9188d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner } 9198d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner } 9208d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner return false; 9218d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner} 9225c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 92379df7c0aaa18129e55968c8783ef8346807bd4afChris Lattner// LinkFunctionProtos - Link the functions together between the two modules, 924c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// without doing function bodies... this just adds external function prototypes 925c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// to the Dest function... 9265c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// 92779df7c0aaa18129e55968c8783ef8346807bd4afChris Lattnerstatic bool LinkFunctionProtos(Module *Dest, const Module *Src, 9285c2d335d86579941b71816196a95a64b277e8963Chris Lattner std::map<const Value*, Value*> &ValueMap, 9295c2d335d86579941b71816196a95a64b277e8963Chris Lattner std::string *Err) { 930d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner ValueSymbolTable &DestSymTab = Dest->getValueSymbolTable(); 931eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 932619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer // Loop over all of the functions in the src module, mapping them over 9335c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner for (Module::const_iterator I = Src->begin(), E = Src->end(); I != E; ++I) { 93418961504fc2b299578dba817900a0696cf3ccc4dChris Lattner const Function *SF = I; // SrcFunction 935194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov GlobalValue *DGV = 0; 936eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 937d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner // Check to see if may have to link the function with the global, alias or 938d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner // function. 939bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola if (SF->hasName() && !SF->hasLocalLinkage()) 94003d7651c3652e1f0cc86e79b26585d86818da9cfDaniel Dunbar DGV = cast_or_null<GlobalValue>(DestSymTab.lookup(SF->getName())); 941eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 942ae1132d2b8ae07afd2fe9a7cb434d849f884bfa0Chris Lattner // If we found a global with the same name in the dest module, but it has 943ae1132d2b8ae07afd2fe9a7cb434d849f884bfa0Chris Lattner // internal linkage, we are really not doing any linkage here. 944bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola if (DGV && DGV->hasLocalLinkage()) 945ae1132d2b8ae07afd2fe9a7cb434d849f884bfa0Chris Lattner DGV = 0; 946ae1132d2b8ae07afd2fe9a7cb434d849f884bfa0Chris Lattner 947d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner // If types don't agree due to opaque types, try to resolve them. 948d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner if (DGV && DGV->getType() != SF->getType()) 949d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner RecursiveResolveTypes(SF->getType(), DGV->getType()); 950194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov 9516157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner GlobalValue::LinkageTypes NewLinkage = GlobalValue::InternalLinkage; 9526157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner bool LinkFromSrc = false; 9536157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner if (GetLinkageResult(DGV, SF, NewLinkage, LinkFromSrc, Err)) 9546157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner return true; 955eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 956824684982968244050e56357e736b9940c23e77dChris Lattner // If there is no linkage to be performed, just bring over SF without 957824684982968244050e56357e736b9940c23e77dChris Lattner // modifying it. 958194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov if (DGV == 0) { 959824684982968244050e56357e736b9940c23e77dChris Lattner // Function does not already exist, simply insert an function signature 960824684982968244050e56357e736b9940c23e77dChris Lattner // identical to SF into the dest module. 961824684982968244050e56357e736b9940c23e77dChris Lattner Function *NewDF = Function::Create(SF->getFunctionType(), 962824684982968244050e56357e736b9940c23e77dChris Lattner SF->getLinkage(), 963824684982968244050e56357e736b9940c23e77dChris Lattner SF->getName(), Dest); 964824684982968244050e56357e736b9940c23e77dChris Lattner CopyGVAttributes(NewDF, SF); 965eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 966824684982968244050e56357e736b9940c23e77dChris Lattner // If the LLVM runtime renamed the function, but it is an externally 967824684982968244050e56357e736b9940c23e77dChris Lattner // visible symbol, DF must be an existing function with internal linkage. 968824684982968244050e56357e736b9940c23e77dChris Lattner // Rename it. 969bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola if (!NewDF->hasLocalLinkage() && NewDF->getName() != SF->getName()) 970824684982968244050e56357e736b9940c23e77dChris Lattner ForceRenaming(NewDF, SF->getName()); 971eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 972824684982968244050e56357e736b9940c23e77dChris Lattner // ... and remember this mapping... 973824684982968244050e56357e736b9940c23e77dChris Lattner ValueMap[SF] = NewDF; 974824684982968244050e56357e736b9940c23e77dChris Lattner continue; 97597f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner } 976eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 9776157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // If the visibilities of the symbols disagree and the destination is a 9786157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // prototype, take the visibility of its input. 9796157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner if (DGV->isDeclaration()) 9806157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner DGV->setVisibility(SF->getVisibility()); 981eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 9826157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner if (LinkFromSrc) { 9836157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner if (isa<GlobalAlias>(DGV)) 9846157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner return Error(Err, "Function-Alias Collision on '" + SF->getName() + 9856157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner "': symbol multiple defined"); 986eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 9876157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // We have a definition of the same name but different type in the 9886157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // source module. Copy the prototype to the destination and replace 9896157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // uses of the destination's prototype with the new prototype. 9906157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner Function *NewDF = Function::Create(SF->getFunctionType(), NewLinkage, 9916157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner SF->getName(), Dest); 9926157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner CopyGVAttributes(NewDF, SF); 993eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 9946157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // Any uses of DF need to change to NewDF, with cast 995baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson DGV->replaceAllUsesWith(ConstantExpr::getBitCast(NewDF, 996c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson DGV->getType())); 997eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 9986157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // DF will conflict with NewDF because they both had the same. We must 9996157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // erase this now so ForceRenaming doesn't assert because DF might 1000eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov // not have internal linkage. 10016157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner if (GlobalVariable *Var = dyn_cast<GlobalVariable>(DGV)) 10026157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner Var->eraseFromParent(); 10036157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner else 10046157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner cast<Function>(DGV)->eraseFromParent(); 1005eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 10066157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // If the symbol table renamed the function, but it is an externally 1007eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov // visible symbol, DF must be an existing function with internal 10086157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // linkage. Rename it. 1009bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola if (NewDF->getName() != SF->getName() && !NewDF->hasLocalLinkage()) 10106157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner ForceRenaming(NewDF, SF->getName()); 1011eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 10126157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // Remember this mapping so uses in the source module get remapped 10136157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // later by RemapOperand. 10146157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner ValueMap[SF] = NewDF; 1015822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner continue; 1016822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner } 1017eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 10186157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // Not "link from source", keep the one in the DestModule and remap the 10196157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // input onto it. 1020eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 10216157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner if (isa<GlobalAlias>(DGV)) { 10226157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // The only valid mappings are: 10236157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // - SF is external declaration, which is effectively a no-op. 10246157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // - SF is weak, when we just need to throw SF out. 1025a05ef5e107943478ad02bc0cd5170a3894076bc4Duncan Sands if (!SF->isDeclaration() && !SF->isWeakForLinker()) 10266157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner return Error(Err, "Function-Alias Collision on '" + SF->getName() + 10276157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner "': symbol multiple defined"); 1028822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner } 10296157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner 10306157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // Set calculated linkage 10316157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner DGV->setLinkage(NewLinkage); 10326157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner 10336157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner // Make sure to remember this mapping. 1034baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson ValueMap[SF] = ConstantExpr::getBitCast(DGV, SF->getType()); 10355c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner } 10365c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner return false; 10375c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner} 10385c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 1039c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// LinkFunctionBody - Copy the source function over into the dest function and 1040c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// fix up references to values. At this point we know that Dest is an external 1041c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// function, and that Src is not. 10424bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattnerstatic bool LinkFunctionBody(Function *Dest, Function *Src, 1043ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer std::map<const Value*, Value*> &ValueMap, 10445c2d335d86579941b71816196a95a64b277e8963Chris Lattner std::string *Err) { 10455cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer assert(Src && Dest && Dest->isDeclaration() && !Src->isDeclaration()); 10465c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 10470033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner // Go through and convert function arguments over, remembering the mapping. 1048e4d5c441e04bdc00ccf1804744af670655123b07Chris Lattner Function::arg_iterator DI = Dest->arg_begin(); 1049e4d5c441e04bdc00ccf1804744af670655123b07Chris Lattner for (Function::arg_iterator I = Src->arg_begin(), E = Src->arg_end(); 105069da5cf26143e4542d4bf8c78ffac6d079efe5c9Chris Lattner I != E; ++I, ++DI) { 10516bc41e8a74d1756da0003641bfebd02a3d6d9586Owen Anderson DI->setName(I->getName()); // Copy the name information over... 10525c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 10535c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner // Add a mapping to our local map 1054817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov ValueMap[I] = DI; 10555c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner } 10565c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 10574bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner // Splice the body of the source function into the dest function. 10584bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner Dest->getBasicBlockList().splice(Dest->end(), Src->getBasicBlockList()); 10595c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 1060c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner // At this point, all of the instructions and values of the function are now 1061c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner // copied over. The only problem is that they are still referencing values in 1062c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner // the Source function as operands. Loop through all of the operands of the 1063c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner // functions and patch them up to point to the local versions... 10645c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner // 106518961504fc2b299578dba817900a0696cf3ccc4dChris Lattner for (Function::iterator BB = Dest->begin(), BE = Dest->end(); BB != BE; ++BB) 106618961504fc2b299578dba817900a0696cf3ccc4dChris Lattner for (BasicBlock::iterator I = BB->begin(), E = BB->end(); I != E; ++I) 106718961504fc2b299578dba817900a0696cf3ccc4dChris Lattner for (Instruction::op_iterator OI = I->op_begin(), OE = I->op_end(); 1068221d688a5ef21a22c2368c9fff0e92d7966c95e5Chris Lattner OI != OE; ++OI) 10694bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner if (!isa<Instruction>(*OI) && !isa<BasicBlock>(*OI)) 1070e922c0201916e0b980ab3cfe91e1413e68d55647Owen Anderson *OI = RemapOperand(*OI, ValueMap, Dest->getContext()); 10710033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner 10720033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner // There is no need to map the arguments anymore. 10731127315562b221010040925980040fd4f65bdb1cChris Lattner for (Function::arg_iterator I = Src->arg_begin(), E = Src->arg_end(); 10741127315562b221010040925980040fd4f65bdb1cChris Lattner I != E; ++I) 1075ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer ValueMap.erase(I); 10765c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 10775c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner return false; 10785c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner} 10795c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 10805c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 1081c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// LinkFunctionBodies - Link in the function bodies that are defined in the 1082c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// source module into the DestModule. This consists basically of copying the 1083c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// function over and fixing up references to values. 10844bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattnerstatic bool LinkFunctionBodies(Module *Dest, Module *Src, 10855c2d335d86579941b71816196a95a64b277e8963Chris Lattner std::map<const Value*, Value*> &ValueMap, 10865c2d335d86579941b71816196a95a64b277e8963Chris Lattner std::string *Err) { 10875c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 10888bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer // Loop over all of the functions in the src module, mapping them over as we 10898bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer // go 10904bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner for (Module::iterator SF = Src->begin(), E = Src->end(); SF != E; ++SF) { 1091619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer if (!SF->isDeclaration()) { // No body if function is external 1092ec91ccba3ca92d5e1f19fb0cb6e57a7d4a3f3195Chris Lattner Function *DF = dyn_cast<Function>(ValueMap[SF]); // Destination function 1093c2d774b6c1d38421c435b6d3cfaa10402c900aebChris Lattner 109418961504fc2b299578dba817900a0696cf3ccc4dChris Lattner // DF not external SF external? 1095ec91ccba3ca92d5e1f19fb0cb6e57a7d4a3f3195Chris Lattner if (DF && DF->isDeclaration()) 109635956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner // Only provide the function body if there isn't one already. 109735956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner if (LinkFunctionBody(DF, SF, ValueMap, Err)) 109835956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner return true; 1099c2d774b6c1d38421c435b6d3cfaa10402c900aebChris Lattner } 11005c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner } 11015c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner return false; 11025c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner} 11035c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 11048166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner// LinkAppendingVars - If there were any appending global variables, link them 11058166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner// together now. Return true on error. 11068166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattnerstatic bool LinkAppendingVars(Module *M, 11078166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner std::multimap<std::string, GlobalVariable *> &AppendingVars, 11088166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner std::string *ErrorMsg) { 11098166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner if (AppendingVars.empty()) return false; // Nothing to do. 1110f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman 1111c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson LLVMContext &Context = M->getContext(); 1112c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson 11138166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // Loop over the multimap of appending vars, processing any variables with the 11148166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // same name, forming a new appending global variable with both of the 11158166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // initializers merged together, then rewrite references to the old variables 11168166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // and delete them. 11178166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner std::vector<Constant*> Inits; 11188166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner while (AppendingVars.size() > 1) { 11198166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // Get the first two elements in the map... 11208166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner std::multimap<std::string, 11218166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner GlobalVariable*>::iterator Second = AppendingVars.begin(), First=Second++; 11228166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner 11238166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // If the first two elements are for different names, there is no pair... 11248166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // Otherwise there is a pair, so link them together... 11258166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner if (First->first == Second->first) { 11268166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner GlobalVariable *G1 = First->second, *G2 = Second->second; 11278166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner const ArrayType *T1 = cast<ArrayType>(G1->getType()->getElementType()); 11288166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner const ArrayType *T2 = cast<ArrayType>(G2->getType()->getElementType()); 1129f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman 11308166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // Check to see that they two arrays agree on type... 11318166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner if (T1->getElementType() != T2->getElementType()) 11328166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner return Error(ErrorMsg, 11338166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner "Appending variables with different element types need to be linked!"); 11348166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner if (G1->isConstant() != G2->isConstant()) 11358166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner return Error(ErrorMsg, 11368166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner "Appending variables linked with different const'ness!"); 11378166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner 11389613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio if (G1->getAlignment() != G2->getAlignment()) 11399613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio return Error(ErrorMsg, 11409613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio "Appending variables with different alignment need to be linked!"); 11419613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio 11429613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio if (G1->getVisibility() != G2->getVisibility()) 11439613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio return Error(ErrorMsg, 11449613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio "Appending variables with different visibility need to be linked!"); 11459613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio 11469613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio if (G1->getSection() != G2->getSection()) 11479613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio return Error(ErrorMsg, 11489613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio "Appending variables with different section name need to be linked!"); 1149eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 11508166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner unsigned NewSize = T1->getNumElements() + T2->getNumElements(); 1151debcb01b0f0a15f568ca69e8f288fade4bfc7297Owen Anderson ArrayType *NewType = ArrayType::get(T1->getElementType(), 1152c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson NewSize); 11538166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner 1154ed74a4ef3b813bed0c0f4bb5af42229d50133da2Chris Lattner G1->setName(""); // Clear G1's name in case of a conflict! 1155eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 11568166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // Create the new global variable... 11578166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner GlobalVariable *NG = 1158e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson new GlobalVariable(*M, NewType, G1->isConstant(), G1->getLinkage(), 1159e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson /*init*/0, First->first, 0, G1->isThreadLocal(), 1160a534b0f7bf0248bf90afe97110a34777e972f325Chris Lattner G1->getType()->getAddressSpace()); 11618166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner 11629613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio // Propagate alignment, visibility and section info. 11639613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio CopyGVAttributes(NG, G1); 11649613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio 11658166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // Merge the initializer... 11668166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner Inits.reserve(NewSize); 1167de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner if (ConstantArray *I = dyn_cast<ConstantArray>(G1->getInitializer())) { 1168de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner for (unsigned i = 0, e = T1->getNumElements(); i != e; ++i) 1169cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos Inits.push_back(I->getOperand(i)); 1170de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner } else { 1171de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner assert(isa<ConstantAggregateZero>(G1->getInitializer())); 1172c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson Constant *CV = Context.getNullValue(T1->getElementType()); 1173de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner for (unsigned i = 0, e = T1->getNumElements(); i != e; ++i) 1174de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner Inits.push_back(CV); 1175de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner } 1176de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner if (ConstantArray *I = dyn_cast<ConstantArray>(G2->getInitializer())) { 1177de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner for (unsigned i = 0, e = T2->getNumElements(); i != e; ++i) 1178cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos Inits.push_back(I->getOperand(i)); 1179de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner } else { 1180de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner assert(isa<ConstantAggregateZero>(G2->getInitializer())); 1181c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson Constant *CV = Context.getNullValue(T2->getElementType()); 1182de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner for (unsigned i = 0, e = T2->getNumElements(); i != e; ++i) 1183de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner Inits.push_back(CV); 1184de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner } 11851fd7096407d5e598ed3366a1141548e71273f1c5Owen Anderson NG->setInitializer(ConstantArray::get(NewType, Inits)); 11868166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner Inits.clear(); 11878166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner 11888166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // Replace any uses of the two global variables with uses of the new 11898166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // global... 11908166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner 11918166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // FIXME: This should rewrite simple/straight-forward uses such as 11928166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // getelementptr instructions to not use the Cast! 1193baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson G1->replaceAllUsesWith(ConstantExpr::getBitCast(NG, 1194c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson G1->getType())); 1195baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson G2->replaceAllUsesWith(ConstantExpr::getBitCast(NG, 1196c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson G2->getType())); 11978166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner 11988166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // Remove the two globals from the module now... 11998166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner M->getGlobalList().erase(G1); 12008166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner M->getGlobalList().erase(G2); 12018166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner 12028166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // Put the new global into the AppendingVars map so that we can handle 12038166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // linking of more than two vars... 12048166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner Second->second = NG; 12058166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner } 12068166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner AppendingVars.erase(First); 12078166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner } 12088166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner 12098166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner return false; 12108166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner} 121152f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner 12129f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikovstatic bool ResolveAliases(Module *Dest) { 12139f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov for (Module::alias_iterator I = Dest->alias_begin(), E = Dest->alias_end(); 12145241957b1f6876452705f94e0ed60dd2c33a5c84Anton Korobeynikov I != E; ++I) 121519e861a4ffb896f16a691d5ac869e894df3cd464Anton Korobeynikov if (const GlobalValue *GV = I->resolveAliasedGlobal()) 1216832b2a9cd8870211bf2d347d7b435beacbb06c8dAnton Korobeynikov if (GV != I && !GV->isDeclaration()) 12175241957b1f6876452705f94e0ed60dd2c33a5c84Anton Korobeynikov I->replaceAllUsesWith(const_cast<GlobalValue*>(GV)); 12189f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov 12199f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov return false; 12209f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov} 122152f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner 122252f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// LinkModules - This function links two modules together, with the resulting 122352f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// left module modified to be the composite of the two input modules. If an 122452f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// error occurs, true is returned and ErrorMsg (if not null) is set to indicate 12255c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// the problem. Upon failure, the Dest module could be in a modified state, and 12265c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// shouldn't be relied on to be consistent. 1227f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukmanbool 12280ba9e211bed7813949aaca97e6b15579c8188e05Reid SpencerLinker::LinkModules(Module *Dest, Module *Src, std::string *ErrorMsg) { 122957a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer assert(Dest != 0 && "Invalid Destination module"); 123057a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer assert(Src != 0 && "Invalid Source Module"); 123157a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer 1232c36357c406911ded5a0e2924998d885815370fdbChris Lattner if (Dest->getDataLayout().empty()) { 1233c36357c406911ded5a0e2924998d885815370fdbChris Lattner if (!Src->getDataLayout().empty()) { 1234ec9bfdc152a870d3c62377ed6855fb3bff69511cChris Lattner Dest->setDataLayout(Src->getDataLayout()); 1235c36357c406911ded5a0e2924998d885815370fdbChris Lattner } else { 1236c36357c406911ded5a0e2924998d885815370fdbChris Lattner std::string DataLayout; 1237c36357c406911ded5a0e2924998d885815370fdbChris Lattner 1238a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov if (Dest->getEndianness() == Module::AnyEndianness) { 1239c36357c406911ded5a0e2924998d885815370fdbChris Lattner if (Src->getEndianness() == Module::BigEndian) 1240c36357c406911ded5a0e2924998d885815370fdbChris Lattner DataLayout.append("E"); 1241c36357c406911ded5a0e2924998d885815370fdbChris Lattner else if (Src->getEndianness() == Module::LittleEndian) 1242c36357c406911ded5a0e2924998d885815370fdbChris Lattner DataLayout.append("e"); 1243a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov } 1244a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov 1245a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov if (Dest->getPointerSize() == Module::AnyPointerSize) { 1246c36357c406911ded5a0e2924998d885815370fdbChris Lattner if (Src->getPointerSize() == Module::Pointer64) 1247c36357c406911ded5a0e2924998d885815370fdbChris Lattner DataLayout.append(DataLayout.length() == 0 ? "p:64:64" : "-p:64:64"); 1248c36357c406911ded5a0e2924998d885815370fdbChris Lattner else if (Src->getPointerSize() == Module::Pointer32) 1249c36357c406911ded5a0e2924998d885815370fdbChris Lattner DataLayout.append(DataLayout.length() == 0 ? "p:32:32" : "-p:32:32"); 1250a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov } 1251c36357c406911ded5a0e2924998d885815370fdbChris Lattner Dest->setDataLayout(DataLayout); 1252c36357c406911ded5a0e2924998d885815370fdbChris Lattner } 1253c36357c406911ded5a0e2924998d885815370fdbChris Lattner } 1254873c5e7859c1534bf92c9e2747f2b70685059598Chris Lattner 1255f27dfcb210fec571d79bc881f5463e0b405707c0Chris Lattner // Copy the target triple from the source to dest if the dest's is empty. 1256c36357c406911ded5a0e2924998d885815370fdbChris Lattner if (Dest->getTargetTriple().empty() && !Src->getTargetTriple().empty()) 1257c36357c406911ded5a0e2924998d885815370fdbChris Lattner Dest->setTargetTriple(Src->getTargetTriple()); 1258eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 1259c36357c406911ded5a0e2924998d885815370fdbChris Lattner if (!Src->getDataLayout().empty() && !Dest->getDataLayout().empty() && 1260c36357c406911ded5a0e2924998d885815370fdbChris Lattner Src->getDataLayout() != Dest->getDataLayout()) 126126f238589f9bb372d24b6fb2bc32edbf046fd9eeReid Spencer cerr << "WARNING: Linking two modules of different data layouts!\n"; 1262152f19ac6514ec29e183ddcdf2ea71599638dc4cChris Lattner if (!Src->getTargetTriple().empty() && 1263152f19ac6514ec29e183ddcdf2ea71599638dc4cChris Lattner Dest->getTargetTriple() != Src->getTargetTriple()) 1264e81561909d128c6e2d8033cb5465a49b2596b26aBill Wendling cerr << "WARNING: Linking two modules of different target triples!\n"; 1265f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman 1266f27dfcb210fec571d79bc881f5463e0b405707c0Chris Lattner // Append the module inline asm string. 12676631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner if (!Src->getModuleInlineAsm().empty()) { 12686631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner if (Dest->getModuleInlineAsm().empty()) 12696631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner Dest->setModuleInlineAsm(Src->getModuleInlineAsm()); 1270e1b2e14f92e8e9da1eb4854511a881cf9c22af3cChris Lattner else 12716631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner Dest->setModuleInlineAsm(Dest->getModuleInlineAsm()+"\n"+ 12726631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner Src->getModuleInlineAsm()); 1273e1b2e14f92e8e9da1eb4854511a881cf9c22af3cChris Lattner } 1274eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov 1275719012d6254ee425464f034250176283fb96793bReid Spencer // Update the destination module's dependent libraries list with the libraries 127657a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer // from the source module. There's no opportunity for duplicates here as the 127757a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer // Module ensures that duplicate insertions are discarded. 1278f27dfcb210fec571d79bc881f5463e0b405707c0Chris Lattner for (Module::lib_iterator SI = Src->lib_begin(), SE = Src->lib_end(); 1279eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov SI != SE; ++SI) 128057a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer Dest->addLibrary(*SI); 128157a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer 12822c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner // LinkTypes - Go through the symbol table of the Src module and see if any 12832c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner // types are named in the src module that are not named in the Dst module. 12842c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner // Make sure there are no type name conflicts. 1285eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov if (LinkTypes(Dest, Src, ErrorMsg)) 1286619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer return true; 12872c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner 12885c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner // ValueMap - Mapping of values from what they used to be in Src, to what they 12895c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner // are now in Dest. 12905c2d335d86579941b71816196a95a64b277e8963Chris Lattner std::map<const Value*, Value*> ValueMap; 12915c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 12928166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // AppendingVars - Keep track of global variables in the destination module 12938166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // with appending linkage. After the module is linked together, they are 12948166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // appended and the module is rewritten. 12958166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner std::multimap<std::string, GlobalVariable *> AppendingVars; 12961127315562b221010040925980040fd4f65bdb1cChris Lattner for (Module::global_iterator I = Dest->global_begin(), E = Dest->global_end(); 12971127315562b221010040925980040fd4f65bdb1cChris Lattner I != E; ++I) { 12985a837defff581accc09048bea1cd839009e374a7Chris Lattner // Add all of the appending globals already in the Dest module to 12995a837defff581accc09048bea1cd839009e374a7Chris Lattner // AppendingVars. 1300f41464618f41b92cf0a0bd14ccc57bce727fb33dChris Lattner if (I->hasAppendingLinkage()) 1301f41464618f41b92cf0a0bd14ccc57bce727fb33dChris Lattner AppendingVars.insert(std::make_pair(I->getName(), I)); 13025a837defff581accc09048bea1cd839009e374a7Chris Lattner } 13035a837defff581accc09048bea1cd839009e374a7Chris Lattner 13048166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // Insert all of the globals in src into the Dest module... without linking 13058166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // initializers (which could refer to functions not yet mapped over). 1306ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer if (LinkGlobals(Dest, Src, ValueMap, AppendingVars, ErrorMsg)) 13075a837defff581accc09048bea1cd839009e374a7Chris Lattner return true; 13085c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 1309c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner // Link the functions together between the two modules, without doing function 1310c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner // bodies... this just adds external function prototypes to the Dest 1311c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner // function... We do this so that when we begin processing function bodies, 1312c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner // all of the global values that may be referenced are available in our 1313c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner // ValueMap. 1314ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer if (LinkFunctionProtos(Dest, Src, ValueMap, ErrorMsg)) 13155a837defff581accc09048bea1cd839009e374a7Chris Lattner return true; 13165c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner 13174fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov // If there were any alias, link them now. We really need to do this now, 13184fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov // because all of the aliases that may be referenced need to be available in 13194fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov // ValueMap 13204fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov if (LinkAlias(Dest, Src, ValueMap, ErrorMsg)) return true; 13214fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov 13226cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner // Update the initializers in the Dest module now that all globals that may 13236cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner // be referenced are in Dest. 13246cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner if (LinkGlobalInits(Dest, Src, ValueMap, ErrorMsg)) return true; 13256cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner 1326c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner // Link in the function bodies that are defined in the source module into the 1327c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner // DestModule. This consists basically of copying the function over and 1328c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner // fixing up references to values. 132979df7c0aaa18129e55968c8783ef8346807bd4afChris Lattner if (LinkFunctionBodies(Dest, Src, ValueMap, ErrorMsg)) return true; 133052f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner 13318166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner // If there were any appending global variables, link them together now. 13328166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner if (LinkAppendingVars(Dest, AppendingVars, ErrorMsg)) return true; 13338166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner 13343db9191baf5f5229a2f4d204c6332ba27fff81d0Anton Korobeynikov // Resolve all uses of aliases with aliasees 13353db9191baf5f5229a2f4d204c6332ba27fff81d0Anton Korobeynikov if (ResolveAliases(Dest)) return true; 13363db9191baf5f5229a2f4d204c6332ba27fff81d0Anton Korobeynikov 133757a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer // If the source library's module id is in the dependent library list of the 133857a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer // destination library, remove it since that module is now linked in. 133957a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer sys::Path modId; 1340dd04df0ec33a903ee7fc747701bafde622f77d8bReid Spencer modId.set(Src->getModuleIdentifier()); 134107adb2836b8aa7a3872e33c285958f5937662b50Reid Spencer if (!modId.isEmpty()) 134207adb2836b8aa7a3872e33c285958f5937662b50Reid Spencer Dest->removeLibrary(modId.getBasename()); 134357a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer 134452f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner return false; 134552f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner} 13469466f5113ba859677a28887c3c7143065e702019Vikram S. Adve 1347567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer// vim: sw=2 1348