159a5afa9SMikhail Glushenkov //===- lib/Linker/LinkModules.cpp - Module Linker Implementation ----------===// 2361e513dSReid Spencer // 3361e513dSReid Spencer // The LLVM Compiler Infrastructure 4361e513dSReid Spencer // 5f3ebc3f3SChris Lattner // This file is distributed under the University of Illinois Open Source 6f3ebc3f3SChris Lattner // License. See LICENSE.TXT for details. 7361e513dSReid Spencer // 8361e513dSReid Spencer //===----------------------------------------------------------------------===// 9361e513dSReid Spencer // 10361e513dSReid Spencer // This file implements the LLVM module linker. 11361e513dSReid Spencer // 12361e513dSReid Spencer //===----------------------------------------------------------------------===// 13361e513dSReid Spencer 146cc07df4SChandler Carruth #include "llvm/Linker/Linker.h" 15ed0881b2SChandler Carruth #include "llvm-c/Linker.h" 1623f8d64bSRafael Espindola #include "llvm/ADT/Optional.h" 1766f02413SBill Wendling #include "llvm/ADT/SetVector.h" 180f7fd36fSEli Bendersky #include "llvm/ADT/SmallString.h" 199fb823bbSChandler Carruth #include "llvm/IR/Constants.h" 209fb823bbSChandler Carruth #include "llvm/IR/Module.h" 21dcb603feSChandler Carruth #include "llvm/IR/TypeFinder.h" 22e17f3708SEli Bendersky #include "llvm/Support/CommandLine.h" 23b6af2f36SBill Wendling #include "llvm/Support/Debug.h" 24b6af2f36SBill Wendling #include "llvm/Support/raw_ostream.h" 25cbb91408STanya Lattner #include "llvm/Transforms/Utils/Cloning.h" 26981af002SWill Dietz #include <cctype> 27361e513dSReid Spencer using namespace llvm; 28361e513dSReid Spencer 29e17f3708SEli Bendersky 30b1ed91f3SChris Lattner //===----------------------------------------------------------------------===// 31b1ed91f3SChris Lattner // TypeMap implementation. 32b1ed91f3SChris Lattner //===----------------------------------------------------------------------===// 33b1ed91f3SChris Lattner 34b1ed91f3SChris Lattner namespace { 35aa9918aaSRafael Espindola typedef SmallPtrSet<StructType*, 32> TypeSet; 36aa9918aaSRafael Espindola 37b1ed91f3SChris Lattner class TypeMapTy : public ValueMapTypeRemapper { 38b1ed91f3SChris Lattner /// MappedTypes - This is a mapping from a source type to a destination type 39b1ed91f3SChris Lattner /// to use. 40b1ed91f3SChris Lattner DenseMap<Type*, Type*> MappedTypes; 41b1ed91f3SChris Lattner 42b1ed91f3SChris Lattner /// SpeculativeTypes - When checking to see if two subgraphs are isomorphic, 43b1ed91f3SChris Lattner /// we speculatively add types to MappedTypes, but keep track of them here in 44b1ed91f3SChris Lattner /// case we need to roll back. 45b1ed91f3SChris Lattner SmallVector<Type*, 16> SpeculativeTypes; 46b1ed91f3SChris Lattner 475e3bd972SChris Lattner /// SrcDefinitionsToResolve - This is a list of non-opaque structs in the 485e3bd972SChris Lattner /// source module that are mapped to an opaque struct in the destination 495e3bd972SChris Lattner /// module. 505e3bd972SChris Lattner SmallVector<StructType*, 16> SrcDefinitionsToResolve; 515e3bd972SChris Lattner 525e3bd972SChris Lattner /// DstResolvedOpaqueTypes - This is the set of opaque types in the 535e3bd972SChris Lattner /// destination modules who are getting a body from the source module. 545e3bd972SChris Lattner SmallPtrSet<StructType*, 16> DstResolvedOpaqueTypes; 55d48b7783SBill Wendling 568c2cc416SBill Wendling public: 57aa9918aaSRafael Espindola TypeMapTy(TypeSet &Set) : DstStructTypesSet(Set) {} 58aa9918aaSRafael Espindola 59aa9918aaSRafael Espindola TypeSet &DstStructTypesSet; 60b1ed91f3SChris Lattner /// addTypeMapping - Indicate that the specified type in the destination 61b1ed91f3SChris Lattner /// module is conceptually equivalent to the specified type in the source 62b1ed91f3SChris Lattner /// module. 63b1ed91f3SChris Lattner void addTypeMapping(Type *DstTy, Type *SrcTy); 64b1ed91f3SChris Lattner 65b1ed91f3SChris Lattner /// linkDefinedTypeBodies - Produce a body for an opaque type in the dest 66b1ed91f3SChris Lattner /// module from a type definition in the source module. 67b1ed91f3SChris Lattner void linkDefinedTypeBodies(); 68b1ed91f3SChris Lattner 69b1ed91f3SChris Lattner /// get - Return the mapped type to use for the specified input type from the 70b1ed91f3SChris Lattner /// source module. 71b1ed91f3SChris Lattner Type *get(Type *SrcTy); 72b1ed91f3SChris Lattner 73b1ed91f3SChris Lattner FunctionType *get(FunctionType *T) {return cast<FunctionType>(get((Type*)T));} 74b1ed91f3SChris Lattner 75b6af2f36SBill Wendling /// dump - Dump out the type map for debugging purposes. 76b6af2f36SBill Wendling void dump() const { 77b6af2f36SBill Wendling for (DenseMap<Type*, Type*>::const_iterator 78b6af2f36SBill Wendling I = MappedTypes.begin(), E = MappedTypes.end(); I != E; ++I) { 79b6af2f36SBill Wendling dbgs() << "TypeMap: "; 80b6af2f36SBill Wendling I->first->dump(); 81b6af2f36SBill Wendling dbgs() << " => "; 82b6af2f36SBill Wendling I->second->dump(); 83b6af2f36SBill Wendling dbgs() << '\n'; 84b6af2f36SBill Wendling } 85b6af2f36SBill Wendling } 86b6af2f36SBill Wendling 87b1ed91f3SChris Lattner private: 88b1ed91f3SChris Lattner Type *getImpl(Type *T); 89b1ed91f3SChris Lattner /// remapType - Implement the ValueMapTypeRemapper interface. 908548299aSCraig Topper Type *remapType(Type *SrcTy) override { 91b1ed91f3SChris Lattner return get(SrcTy); 92b1ed91f3SChris Lattner } 93b1ed91f3SChris Lattner 94b1ed91f3SChris Lattner bool areTypesIsomorphic(Type *DstTy, Type *SrcTy); 95b1ed91f3SChris Lattner }; 96b1ed91f3SChris Lattner } 97b1ed91f3SChris Lattner 98b1ed91f3SChris Lattner void TypeMapTy::addTypeMapping(Type *DstTy, Type *SrcTy) { 99b1ed91f3SChris Lattner Type *&Entry = MappedTypes[SrcTy]; 100b1ed91f3SChris Lattner if (Entry) return; 101b1ed91f3SChris Lattner 102b1ed91f3SChris Lattner if (DstTy == SrcTy) { 103b1ed91f3SChris Lattner Entry = DstTy; 104b1ed91f3SChris Lattner return; 105b1ed91f3SChris Lattner } 106b1ed91f3SChris Lattner 107b1ed91f3SChris Lattner // Check to see if these types are recursively isomorphic and establish a 108b1ed91f3SChris Lattner // mapping between them if so. 109d48b7783SBill Wendling if (!areTypesIsomorphic(DstTy, SrcTy)) { 110b1ed91f3SChris Lattner // Oops, they aren't isomorphic. Just discard this request by rolling out 111b1ed91f3SChris Lattner // any speculative mappings we've established. 112b1ed91f3SChris Lattner for (unsigned i = 0, e = SpeculativeTypes.size(); i != e; ++i) 113b1ed91f3SChris Lattner MappedTypes.erase(SpeculativeTypes[i]); 114d48b7783SBill Wendling } 115b1ed91f3SChris Lattner SpeculativeTypes.clear(); 116b1ed91f3SChris Lattner } 117b1ed91f3SChris Lattner 118b1ed91f3SChris Lattner /// areTypesIsomorphic - Recursively walk this pair of types, returning true 119b1ed91f3SChris Lattner /// if they are isomorphic, false if they are not. 120b1ed91f3SChris Lattner bool TypeMapTy::areTypesIsomorphic(Type *DstTy, Type *SrcTy) { 121b1ed91f3SChris Lattner // Two types with differing kinds are clearly not isomorphic. 122b1ed91f3SChris Lattner if (DstTy->getTypeID() != SrcTy->getTypeID()) return false; 123b1ed91f3SChris Lattner 124b1ed91f3SChris Lattner // If we have an entry in the MappedTypes table, then we have our answer. 125b1ed91f3SChris Lattner Type *&Entry = MappedTypes[SrcTy]; 126b1ed91f3SChris Lattner if (Entry) 127b1ed91f3SChris Lattner return Entry == DstTy; 128b1ed91f3SChris Lattner 129b1ed91f3SChris Lattner // Two identical types are clearly isomorphic. Remember this 130b1ed91f3SChris Lattner // non-speculatively. 131b1ed91f3SChris Lattner if (DstTy == SrcTy) { 132b1ed91f3SChris Lattner Entry = DstTy; 133361e513dSReid Spencer return true; 134361e513dSReid Spencer } 135361e513dSReid Spencer 136b1ed91f3SChris Lattner // Okay, we have two types with identical kinds that we haven't seen before. 137361e513dSReid Spencer 138b1ed91f3SChris Lattner // If this is an opaque struct type, special case it. 139b1ed91f3SChris Lattner if (StructType *SSTy = dyn_cast<StructType>(SrcTy)) { 140b1ed91f3SChris Lattner // Mapping an opaque type to any struct, just keep the dest struct. 141b1ed91f3SChris Lattner if (SSTy->isOpaque()) { 142b1ed91f3SChris Lattner Entry = DstTy; 143b1ed91f3SChris Lattner SpeculativeTypes.push_back(SrcTy); 144b1ed91f3SChris Lattner return true; 145361e513dSReid Spencer } 146b1ed91f3SChris Lattner 1475e3bd972SChris Lattner // Mapping a non-opaque source type to an opaque dest. If this is the first 1485e3bd972SChris Lattner // type that we're mapping onto this destination type then we succeed. Keep 1495e3bd972SChris Lattner // the dest, but fill it in later. This doesn't need to be speculative. If 1505e3bd972SChris Lattner // this is the second (different) type that we're trying to map onto the 1515e3bd972SChris Lattner // same opaque type then we fail. 152b1ed91f3SChris Lattner if (cast<StructType>(DstTy)->isOpaque()) { 1535e3bd972SChris Lattner // We can only map one source type onto the opaque destination type. 1545e3bd972SChris Lattner if (!DstResolvedOpaqueTypes.insert(cast<StructType>(DstTy))) 1555e3bd972SChris Lattner return false; 1565e3bd972SChris Lattner SrcDefinitionsToResolve.push_back(SSTy); 157b1ed91f3SChris Lattner Entry = DstTy; 158b1ed91f3SChris Lattner return true; 159b1ed91f3SChris Lattner } 160b1ed91f3SChris Lattner } 161b1ed91f3SChris Lattner 162b1ed91f3SChris Lattner // If the number of subtypes disagree between the two types, then we fail. 163b1ed91f3SChris Lattner if (SrcTy->getNumContainedTypes() != DstTy->getNumContainedTypes()) 164b1ed91f3SChris Lattner return false; 165b1ed91f3SChris Lattner 166b1ed91f3SChris Lattner // Fail if any of the extra properties (e.g. array size) of the type disagree. 167b1ed91f3SChris Lattner if (isa<IntegerType>(DstTy)) 168b1ed91f3SChris Lattner return false; // bitwidth disagrees. 169b1ed91f3SChris Lattner if (PointerType *PT = dyn_cast<PointerType>(DstTy)) { 170b1ed91f3SChris Lattner if (PT->getAddressSpace() != cast<PointerType>(SrcTy)->getAddressSpace()) 171b1ed91f3SChris Lattner return false; 172eaf9b762SChris Lattner 173b1ed91f3SChris Lattner } else if (FunctionType *FT = dyn_cast<FunctionType>(DstTy)) { 174b1ed91f3SChris Lattner if (FT->isVarArg() != cast<FunctionType>(SrcTy)->isVarArg()) 175b1ed91f3SChris Lattner return false; 176b1ed91f3SChris Lattner } else if (StructType *DSTy = dyn_cast<StructType>(DstTy)) { 177b1ed91f3SChris Lattner StructType *SSTy = cast<StructType>(SrcTy); 17844f7ab45SChris Lattner if (DSTy->isLiteral() != SSTy->isLiteral() || 179b1ed91f3SChris Lattner DSTy->isPacked() != SSTy->isPacked()) 180b1ed91f3SChris Lattner return false; 181b1ed91f3SChris Lattner } else if (ArrayType *DATy = dyn_cast<ArrayType>(DstTy)) { 182b1ed91f3SChris Lattner if (DATy->getNumElements() != cast<ArrayType>(SrcTy)->getNumElements()) 183b1ed91f3SChris Lattner return false; 184b1ed91f3SChris Lattner } else if (VectorType *DVTy = dyn_cast<VectorType>(DstTy)) { 1855fad3e9aSJoey Gouly if (DVTy->getNumElements() != cast<VectorType>(SrcTy)->getNumElements()) 186361e513dSReid Spencer return false; 187361e513dSReid Spencer } 188361e513dSReid Spencer 189b1ed91f3SChris Lattner // Otherwise, we speculate that these two types will line up and recursively 190b1ed91f3SChris Lattner // check the subelements. 191b1ed91f3SChris Lattner Entry = DstTy; 192b1ed91f3SChris Lattner SpeculativeTypes.push_back(SrcTy); 193eee6f995SChris Lattner 194d48b7783SBill Wendling for (unsigned i = 0, e = SrcTy->getNumContainedTypes(); i != e; ++i) 195d48b7783SBill Wendling if (!areTypesIsomorphic(DstTy->getContainedType(i), 196d48b7783SBill Wendling SrcTy->getContainedType(i))) 197b1ed91f3SChris Lattner return false; 198eee6f995SChris Lattner 199b1ed91f3SChris Lattner // If everything seems to have lined up, then everything is great. 200eee6f995SChris Lattner return true; 201eee6f995SChris Lattner } 202eee6f995SChris Lattner 203b1ed91f3SChris Lattner /// linkDefinedTypeBodies - Produce a body for an opaque type in the dest 204b1ed91f3SChris Lattner /// module from a type definition in the source module. 205b1ed91f3SChris Lattner void TypeMapTy::linkDefinedTypeBodies() { 206b1ed91f3SChris Lattner SmallVector<Type*, 16> Elements; 207b1ed91f3SChris Lattner SmallString<16> TmpName; 208eee6f995SChris Lattner 209b1ed91f3SChris Lattner // Note that processing entries in this loop (calling 'get') can add new 2105e3bd972SChris Lattner // entries to the SrcDefinitionsToResolve vector. 2115e3bd972SChris Lattner while (!SrcDefinitionsToResolve.empty()) { 2125e3bd972SChris Lattner StructType *SrcSTy = SrcDefinitionsToResolve.pop_back_val(); 213b1ed91f3SChris Lattner StructType *DstSTy = cast<StructType>(MappedTypes[SrcSTy]); 214eee6f995SChris Lattner 215b1ed91f3SChris Lattner // TypeMap is a many-to-one mapping, if there were multiple types that 216b1ed91f3SChris Lattner // provide a body for DstSTy then previous iterations of this loop may have 217b1ed91f3SChris Lattner // already handled it. Just ignore this case. 218b1ed91f3SChris Lattner if (!DstSTy->isOpaque()) continue; 219b1ed91f3SChris Lattner assert(!SrcSTy->isOpaque() && "Not resolving a definition?"); 220b1ed91f3SChris Lattner 221b1ed91f3SChris Lattner // Map the body of the source type over to a new body for the dest type. 222b1ed91f3SChris Lattner Elements.resize(SrcSTy->getNumElements()); 223b1ed91f3SChris Lattner for (unsigned i = 0, e = Elements.size(); i != e; ++i) 224b1ed91f3SChris Lattner Elements[i] = getImpl(SrcSTy->getElementType(i)); 225b1ed91f3SChris Lattner 226b1ed91f3SChris Lattner DstSTy->setBody(Elements, SrcSTy->isPacked()); 227b1ed91f3SChris Lattner 228b1ed91f3SChris Lattner // If DstSTy has no name or has a longer name than STy, then viciously steal 229b1ed91f3SChris Lattner // STy's name. 230b1ed91f3SChris Lattner if (!SrcSTy->hasName()) continue; 231b1ed91f3SChris Lattner StringRef SrcName = SrcSTy->getName(); 232b1ed91f3SChris Lattner 233b1ed91f3SChris Lattner if (!DstSTy->hasName() || DstSTy->getName().size() > SrcName.size()) { 234b1ed91f3SChris Lattner TmpName.insert(TmpName.end(), SrcName.begin(), SrcName.end()); 235b1ed91f3SChris Lattner SrcSTy->setName(""); 236b1ed91f3SChris Lattner DstSTy->setName(TmpName.str()); 237b1ed91f3SChris Lattner TmpName.clear(); 238b1ed91f3SChris Lattner } 239b1ed91f3SChris Lattner } 2405e3bd972SChris Lattner 2415e3bd972SChris Lattner DstResolvedOpaqueTypes.clear(); 242eee6f995SChris Lattner } 243eee6f995SChris Lattner 244b1ed91f3SChris Lattner /// get - Return the mapped type to use for the specified input type from the 245b1ed91f3SChris Lattner /// source module. 246b1ed91f3SChris Lattner Type *TypeMapTy::get(Type *Ty) { 247b1ed91f3SChris Lattner Type *Result = getImpl(Ty); 248b1ed91f3SChris Lattner 249b1ed91f3SChris Lattner // If this caused a reference to any struct type, resolve it before returning. 2505e3bd972SChris Lattner if (!SrcDefinitionsToResolve.empty()) 251b1ed91f3SChris Lattner linkDefinedTypeBodies(); 252b1ed91f3SChris Lattner return Result; 253eee6f995SChris Lattner } 254eee6f995SChris Lattner 255b1ed91f3SChris Lattner /// getImpl - This is the recursive version of get(). 256b1ed91f3SChris Lattner Type *TypeMapTy::getImpl(Type *Ty) { 257b1ed91f3SChris Lattner // If we already have an entry for this type, return it. 258b1ed91f3SChris Lattner Type **Entry = &MappedTypes[Ty]; 259b1ed91f3SChris Lattner if (*Entry) return *Entry; 260b1ed91f3SChris Lattner 261b1ed91f3SChris Lattner // If this is not a named struct type, then just map all of the elements and 262b1ed91f3SChris Lattner // then rebuild the type from inside out. 26344f7ab45SChris Lattner if (!isa<StructType>(Ty) || cast<StructType>(Ty)->isLiteral()) { 264b1ed91f3SChris Lattner // If there are no element types to map, then the type is itself. This is 265b1ed91f3SChris Lattner // true for the anonymous {} struct, things like 'float', integers, etc. 266b1ed91f3SChris Lattner if (Ty->getNumContainedTypes() == 0) 267b1ed91f3SChris Lattner return *Entry = Ty; 268b1ed91f3SChris Lattner 269b1ed91f3SChris Lattner // Remap all of the elements, keeping track of whether any of them change. 270b1ed91f3SChris Lattner bool AnyChange = false; 271b1ed91f3SChris Lattner SmallVector<Type*, 4> ElementTypes; 272b1ed91f3SChris Lattner ElementTypes.resize(Ty->getNumContainedTypes()); 273b1ed91f3SChris Lattner for (unsigned i = 0, e = Ty->getNumContainedTypes(); i != e; ++i) { 274b1ed91f3SChris Lattner ElementTypes[i] = getImpl(Ty->getContainedType(i)); 275b1ed91f3SChris Lattner AnyChange |= ElementTypes[i] != Ty->getContainedType(i); 276eee6f995SChris Lattner } 277b1ed91f3SChris Lattner 278b1ed91f3SChris Lattner // If we found our type while recursively processing stuff, just use it. 279b1ed91f3SChris Lattner Entry = &MappedTypes[Ty]; 280b1ed91f3SChris Lattner if (*Entry) return *Entry; 281b1ed91f3SChris Lattner 282b1ed91f3SChris Lattner // If all of the element types mapped directly over, then the type is usable 283b1ed91f3SChris Lattner // as-is. 284b1ed91f3SChris Lattner if (!AnyChange) 285b1ed91f3SChris Lattner return *Entry = Ty; 286b1ed91f3SChris Lattner 287b1ed91f3SChris Lattner // Otherwise, rebuild a modified type. 288b1ed91f3SChris Lattner switch (Ty->getTypeID()) { 289a2886c21SCraig Topper default: llvm_unreachable("unknown derived type to remap"); 290b1ed91f3SChris Lattner case Type::ArrayTyID: 291b1ed91f3SChris Lattner return *Entry = ArrayType::get(ElementTypes[0], 292b1ed91f3SChris Lattner cast<ArrayType>(Ty)->getNumElements()); 293b1ed91f3SChris Lattner case Type::VectorTyID: 294b1ed91f3SChris Lattner return *Entry = VectorType::get(ElementTypes[0], 295b1ed91f3SChris Lattner cast<VectorType>(Ty)->getNumElements()); 296b1ed91f3SChris Lattner case Type::PointerTyID: 297b1ed91f3SChris Lattner return *Entry = PointerType::get(ElementTypes[0], 298b1ed91f3SChris Lattner cast<PointerType>(Ty)->getAddressSpace()); 299b1ed91f3SChris Lattner case Type::FunctionTyID: 300b1ed91f3SChris Lattner return *Entry = FunctionType::get(ElementTypes[0], 301717d7eddSFrits van Bommel makeArrayRef(ElementTypes).slice(1), 302b1ed91f3SChris Lattner cast<FunctionType>(Ty)->isVarArg()); 303b1ed91f3SChris Lattner case Type::StructTyID: 304b1ed91f3SChris Lattner // Note that this is only reached for anonymous structs. 305b1ed91f3SChris Lattner return *Entry = StructType::get(Ty->getContext(), ElementTypes, 306b1ed91f3SChris Lattner cast<StructType>(Ty)->isPacked()); 307b1ed91f3SChris Lattner } 308b1ed91f3SChris Lattner } 309b1ed91f3SChris Lattner 310b1ed91f3SChris Lattner // Otherwise, this is an unmapped named struct. If the struct can be directly 311b1ed91f3SChris Lattner // mapped over, just use it as-is. This happens in a case when the linked-in 312b1ed91f3SChris Lattner // module has something like: 313b1ed91f3SChris Lattner // %T = type {%T*, i32} 314b1ed91f3SChris Lattner // @GV = global %T* null 315b1ed91f3SChris Lattner // where T does not exist at all in the destination module. 316b1ed91f3SChris Lattner // 317b1ed91f3SChris Lattner // The other case we watch for is when the type is not in the destination 318b1ed91f3SChris Lattner // module, but that it has to be rebuilt because it refers to something that 319b1ed91f3SChris Lattner // is already mapped. For example, if the destination module has: 320b1ed91f3SChris Lattner // %A = type { i32 } 321b1ed91f3SChris Lattner // and the source module has something like 322b1ed91f3SChris Lattner // %A' = type { i32 } 323b1ed91f3SChris Lattner // %B = type { %A'* } 324b1ed91f3SChris Lattner // @GV = global %B* null 325b1ed91f3SChris Lattner // then we want to create a new type: "%B = type { %A*}" and have it take the 326b1ed91f3SChris Lattner // pristine "%B" name from the source module. 327b1ed91f3SChris Lattner // 328b1ed91f3SChris Lattner // To determine which case this is, we have to recursively walk the type graph 329b1ed91f3SChris Lattner // speculating that we'll be able to reuse it unmodified. Only if this is 330b1ed91f3SChris Lattner // safe would we map the entire thing over. Because this is an optimization, 331b1ed91f3SChris Lattner // and is not required for the prettiness of the linked module, we just skip 332b1ed91f3SChris Lattner // it and always rebuild a type here. 333b1ed91f3SChris Lattner StructType *STy = cast<StructType>(Ty); 334b1ed91f3SChris Lattner 335b1ed91f3SChris Lattner // If the type is opaque, we can just use it directly. 336aa9918aaSRafael Espindola if (STy->isOpaque()) { 337aa9918aaSRafael Espindola // A named structure type from src module is used. Add it to the Set of 338aa9918aaSRafael Espindola // identified structs in the destination module. 339aa9918aaSRafael Espindola DstStructTypesSet.insert(STy); 340b1ed91f3SChris Lattner return *Entry = STy; 341aa9918aaSRafael Espindola } 342b1ed91f3SChris Lattner 343b1ed91f3SChris Lattner // Otherwise we create a new type and resolve its body later. This will be 344b1ed91f3SChris Lattner // resolved by the top level of get(). 3455e3bd972SChris Lattner SrcDefinitionsToResolve.push_back(STy); 3465e3bd972SChris Lattner StructType *DTy = StructType::create(STy->getContext()); 347aa9918aaSRafael Espindola // A new identified structure type was created. Add it to the set of 348aa9918aaSRafael Espindola // identified structs in the destination module. 349aa9918aaSRafael Espindola DstStructTypesSet.insert(DTy); 3505e3bd972SChris Lattner DstResolvedOpaqueTypes.insert(DTy); 3515e3bd972SChris Lattner return *Entry = DTy; 352b1ed91f3SChris Lattner } 353b1ed91f3SChris Lattner 354b1ed91f3SChris Lattner //===----------------------------------------------------------------------===// 355b1ed91f3SChris Lattner // ModuleLinker implementation. 356b1ed91f3SChris Lattner //===----------------------------------------------------------------------===// 357b1ed91f3SChris Lattner 358b1ed91f3SChris Lattner namespace { 359f6f121e2SJames Molloy class ModuleLinker; 360f6f121e2SJames Molloy 361f6f121e2SJames Molloy /// ValueMaterializerTy - Creates prototypes for functions that are lazily 362f6f121e2SJames Molloy /// linked on the fly. This speeds up linking for modules with many 363f6f121e2SJames Molloy /// lazily linked functions of which few get used. 364f6f121e2SJames Molloy class ValueMaterializerTy : public ValueMaterializer { 365f6f121e2SJames Molloy TypeMapTy &TypeMap; 366f6f121e2SJames Molloy Module *DstM; 367f6f121e2SJames Molloy std::vector<Function*> &LazilyLinkFunctions; 368f6f121e2SJames Molloy public: 369f6f121e2SJames Molloy ValueMaterializerTy(TypeMapTy &TypeMap, Module *DstM, 370f6f121e2SJames Molloy std::vector<Function*> &LazilyLinkFunctions) : 371f6f121e2SJames Molloy ValueMaterializer(), TypeMap(TypeMap), DstM(DstM), 372f6f121e2SJames Molloy LazilyLinkFunctions(LazilyLinkFunctions) { 373f6f121e2SJames Molloy } 374f6f121e2SJames Molloy 3758548299aSCraig Topper Value *materializeValueFor(Value *V) override; 376f6f121e2SJames Molloy }; 377f6f121e2SJames Molloy 378b1ed91f3SChris Lattner /// ModuleLinker - This is an implementation class for the LinkModules 379b1ed91f3SChris Lattner /// function, which is the entrypoint for this file. 380b1ed91f3SChris Lattner class ModuleLinker { 381b1ed91f3SChris Lattner Module *DstM, *SrcM; 382b1ed91f3SChris Lattner 383b1ed91f3SChris Lattner TypeMapTy TypeMap; 384f6f121e2SJames Molloy ValueMaterializerTy ValMaterializer; 385b1ed91f3SChris Lattner 386b1ed91f3SChris Lattner /// ValueMap - Mapping of values from what they used to be in Src, to what 387b1ed91f3SChris Lattner /// they are now in DstM. ValueToValueMapTy is a ValueMap, which involves 388b1ed91f3SChris Lattner /// some overhead due to the use of Value handles which the Linker doesn't 389b1ed91f3SChris Lattner /// actually need, but this allows us to reuse the ValueMapper code. 390b1ed91f3SChris Lattner ValueToValueMapTy ValueMap; 391b1ed91f3SChris Lattner 392b1ed91f3SChris Lattner struct AppendingVarInfo { 393b1ed91f3SChris Lattner GlobalVariable *NewGV; // New aggregate global in dest module. 394b1ed91f3SChris Lattner Constant *DstInit; // Old initializer from dest module. 395b1ed91f3SChris Lattner Constant *SrcInit; // Old initializer from src module. 396b1ed91f3SChris Lattner }; 397b1ed91f3SChris Lattner 398b1ed91f3SChris Lattner std::vector<AppendingVarInfo> AppendingVars; 399b1ed91f3SChris Lattner 400cbb91408STanya Lattner unsigned Mode; // Mode to treat source module. 401cbb91408STanya Lattner 402cbb91408STanya Lattner // Set of items not to link in from source. 403cbb91408STanya Lattner SmallPtrSet<const Value*, 16> DoNotLinkFromSource; 404cbb91408STanya Lattner 4050a48b877STanya Lattner // Vector of functions to lazily link in. 406fa228782SBill Wendling std::vector<Function*> LazilyLinkFunctions; 4070a48b877STanya Lattner 4087da92ed8SEli Bendersky bool SuppressWarnings; 4097da92ed8SEli Bendersky 410b1ed91f3SChris Lattner public: 411b1ed91f3SChris Lattner std::string ErrorMsg; 412b1ed91f3SChris Lattner 4137da92ed8SEli Bendersky ModuleLinker(Module *dstM, TypeSet &Set, Module *srcM, unsigned mode, 4147da92ed8SEli Bendersky bool SuppressWarnings=false) 415f6f121e2SJames Molloy : DstM(dstM), SrcM(srcM), TypeMap(Set), 4167da92ed8SEli Bendersky ValMaterializer(TypeMap, DstM, LazilyLinkFunctions), Mode(mode), 4177da92ed8SEli Bendersky SuppressWarnings(SuppressWarnings) {} 418b1ed91f3SChris Lattner 419b1ed91f3SChris Lattner bool run(); 420b1ed91f3SChris Lattner 421b1ed91f3SChris Lattner private: 422b1ed91f3SChris Lattner /// emitError - Helper method for setting a message and returning an error 423b1ed91f3SChris Lattner /// code. 424b1ed91f3SChris Lattner bool emitError(const Twine &Message) { 425b1ed91f3SChris Lattner ErrorMsg = Message.str(); 426b1ed91f3SChris Lattner return true; 427b1ed91f3SChris Lattner } 428b1ed91f3SChris Lattner 429*dad0a645SDavid Majnemer bool getComdatLeader(Module *M, StringRef ComdatName, 430*dad0a645SDavid Majnemer const GlobalVariable *&GVar); 431*dad0a645SDavid Majnemer bool computeResultingSelectionKind(StringRef ComdatName, 432*dad0a645SDavid Majnemer Comdat::SelectionKind Src, 433*dad0a645SDavid Majnemer Comdat::SelectionKind Dst, 434*dad0a645SDavid Majnemer Comdat::SelectionKind &Result, 435*dad0a645SDavid Majnemer bool &LinkFromSrc); 436*dad0a645SDavid Majnemer std::map<const Comdat *, std::pair<Comdat::SelectionKind, bool>> 437*dad0a645SDavid Majnemer ComdatsChosen; 438*dad0a645SDavid Majnemer bool getComdatResult(const Comdat *SrcC, Comdat::SelectionKind &SK, 439*dad0a645SDavid Majnemer bool &LinkFromSrc); 440*dad0a645SDavid Majnemer 441b1ed91f3SChris Lattner /// getLinkageResult - This analyzes the two global values and determines 442b1ed91f3SChris Lattner /// what the result will look like in the destination module. 443b1ed91f3SChris Lattner bool getLinkageResult(GlobalValue *Dest, const GlobalValue *Src, 44423f8d64bSRafael Espindola GlobalValue::LinkageTypes <, 44523f8d64bSRafael Espindola GlobalValue::VisibilityTypes &Vis, 44623f8d64bSRafael Espindola bool &LinkFromSrc); 447b1ed91f3SChris Lattner 448b1ed91f3SChris Lattner /// getLinkedToGlobal - Given a global in the source module, return the 449b1ed91f3SChris Lattner /// global in the destination module that is being linked to, if any. 450b1ed91f3SChris Lattner GlobalValue *getLinkedToGlobal(GlobalValue *SrcGV) { 451b1ed91f3SChris Lattner // If the source has no name it can't link. If it has local linkage, 452b1ed91f3SChris Lattner // there is no name match-up going on. 453b1ed91f3SChris Lattner if (!SrcGV->hasName() || SrcGV->hasLocalLinkage()) 4542617dcceSCraig Topper return nullptr; 455b1ed91f3SChris Lattner 456b1ed91f3SChris Lattner // Otherwise see if we have a match in the destination module's symtab. 457b1ed91f3SChris Lattner GlobalValue *DGV = DstM->getNamedValue(SrcGV->getName()); 4582617dcceSCraig Topper if (!DGV) return nullptr; 459b1ed91f3SChris Lattner 460b1ed91f3SChris Lattner // If we found a global with the same name in the dest module, but it has 461b1ed91f3SChris Lattner // internal linkage, we are really not doing any linkage here. 462b1ed91f3SChris Lattner if (DGV->hasLocalLinkage()) 4632617dcceSCraig Topper return nullptr; 464b1ed91f3SChris Lattner 465b1ed91f3SChris Lattner // Otherwise, we do in fact link to the destination global. 466b1ed91f3SChris Lattner return DGV; 467b1ed91f3SChris Lattner } 468b1ed91f3SChris Lattner 469b1ed91f3SChris Lattner void computeTypeMapping(); 470b1ed91f3SChris Lattner 471b1ed91f3SChris Lattner bool linkAppendingVarProto(GlobalVariable *DstGV, GlobalVariable *SrcGV); 472b1ed91f3SChris Lattner bool linkGlobalProto(GlobalVariable *SrcGV); 473b1ed91f3SChris Lattner bool linkFunctionProto(Function *SrcF); 474b1ed91f3SChris Lattner bool linkAliasProto(GlobalAlias *SrcA); 47566f02413SBill Wendling bool linkModuleFlagsMetadata(); 476b1ed91f3SChris Lattner 477b1ed91f3SChris Lattner void linkAppendingVarInit(const AppendingVarInfo &AVI); 478b1ed91f3SChris Lattner void linkGlobalInits(); 479b1ed91f3SChris Lattner void linkFunctionBody(Function *Dst, Function *Src); 480b1ed91f3SChris Lattner void linkAliasBodies(); 481b1ed91f3SChris Lattner void linkNamedMDNodes(); 482eee6f995SChris Lattner }; 483d48b7783SBill Wendling } 484d48b7783SBill Wendling 485b1ed91f3SChris Lattner /// forceRenaming - The LLVM SymbolTable class autorenames globals that conflict 48690246aa7SReid Spencer /// in the symbol table. This is good for all clients except for us. Go 48790246aa7SReid Spencer /// through the trouble to force this back. 488b1ed91f3SChris Lattner static void forceRenaming(GlobalValue *GV, StringRef Name) { 489b1ed91f3SChris Lattner // If the global doesn't force its name or if it already has the right name, 490b1ed91f3SChris Lattner // there is nothing for us to do. 491b1ed91f3SChris Lattner if (GV->hasLocalLinkage() || GV->getName() == Name) 492b1ed91f3SChris Lattner return; 493b1ed91f3SChris Lattner 494b1ed91f3SChris Lattner Module *M = GV->getParent(); 495361e513dSReid Spencer 496361e513dSReid Spencer // If there is a conflict, rename the conflict. 497b1ed91f3SChris Lattner if (GlobalValue *ConflictGV = M->getNamedValue(Name)) { 4982a8d2e0eSChris Lattner GV->takeName(ConflictGV); 499361e513dSReid Spencer ConflictGV->setName(Name); // This will cause ConflictGV to get renamed 500b1ed91f3SChris Lattner assert(ConflictGV->getName() != Name && "forceRenaming didn't work"); 5012a8d2e0eSChris Lattner } else { 5022a8d2e0eSChris Lattner GV->setName(Name); // Force the name back 5033aaaa0b2SReid Spencer } 5043aaaa0b2SReid Spencer } 5053aaaa0b2SReid Spencer 506b6af2f36SBill Wendling /// copyGVAttributes - copy additional attributes (those not needed to construct 5073aaaa0b2SReid Spencer /// a GlobalValue) from the SrcGV to the DestGV. 508b6af2f36SBill Wendling static void copyGVAttributes(GlobalValue *DestGV, const GlobalValue *SrcGV) { 509dd7daee8SDuncan Sands // Use the maximum alignment, rather than just copying the alignment of SrcGV. 51099e05cf1SRafael Espindola auto *DestGO = dyn_cast<GlobalObject>(DestGV); 511a7d9c69cSRafael Espindola unsigned Alignment; 51299e05cf1SRafael Espindola if (DestGO) 51399e05cf1SRafael Espindola Alignment = std::max(DestGO->getAlignment(), SrcGV->getAlignment()); 514a7d9c69cSRafael Espindola 515dd7daee8SDuncan Sands DestGV->copyAttributesFrom(SrcGV); 516a7d9c69cSRafael Espindola 51799e05cf1SRafael Espindola if (DestGO) 51899e05cf1SRafael Espindola DestGO->setAlignment(Alignment); 519b1ed91f3SChris Lattner 520b1ed91f3SChris Lattner forceRenaming(DestGV, SrcGV->getName()); 521361e513dSReid Spencer } 522361e513dSReid Spencer 52323f8d64bSRafael Espindola static bool isLessConstraining(GlobalValue::VisibilityTypes a, 52423f8d64bSRafael Espindola GlobalValue::VisibilityTypes b) { 52523f8d64bSRafael Espindola if (a == GlobalValue::HiddenVisibility) 52623f8d64bSRafael Espindola return false; 52723f8d64bSRafael Espindola if (b == GlobalValue::HiddenVisibility) 52823f8d64bSRafael Espindola return true; 52923f8d64bSRafael Espindola if (a == GlobalValue::ProtectedVisibility) 53023f8d64bSRafael Espindola return false; 53123f8d64bSRafael Espindola if (b == GlobalValue::ProtectedVisibility) 53223f8d64bSRafael Espindola return true; 53323f8d64bSRafael Espindola return false; 53423f8d64bSRafael Espindola } 53523f8d64bSRafael Espindola 536f6f121e2SJames Molloy Value *ValueMaterializerTy::materializeValueFor(Value *V) { 537f6f121e2SJames Molloy Function *SF = dyn_cast<Function>(V); 538f6f121e2SJames Molloy if (!SF) 5392617dcceSCraig Topper return nullptr; 540f6f121e2SJames Molloy 541f6f121e2SJames Molloy Function *DF = Function::Create(TypeMap.get(SF->getFunctionType()), 542f6f121e2SJames Molloy SF->getLinkage(), SF->getName(), DstM); 543f6f121e2SJames Molloy copyGVAttributes(DF, SF); 544f6f121e2SJames Molloy 545f6f121e2SJames Molloy LazilyLinkFunctions.push_back(SF); 546f6f121e2SJames Molloy return DF; 547f6f121e2SJames Molloy } 548f6f121e2SJames Molloy 549*dad0a645SDavid Majnemer bool ModuleLinker::getComdatLeader(Module *M, StringRef ComdatName, 550*dad0a645SDavid Majnemer const GlobalVariable *&GVar) { 551*dad0a645SDavid Majnemer const GlobalValue *GVal = M->getNamedValue(ComdatName); 552*dad0a645SDavid Majnemer if (const auto *GA = dyn_cast_or_null<GlobalAlias>(GVal)) { 553*dad0a645SDavid Majnemer GVal = GA->getBaseObject(); 554*dad0a645SDavid Majnemer if (!GVal) 555*dad0a645SDavid Majnemer // We cannot resolve the size of the aliasee yet. 556*dad0a645SDavid Majnemer return emitError("Linking COMDATs named '" + ComdatName + 557*dad0a645SDavid Majnemer "': COMDAT key involves incomputable alias size."); 558*dad0a645SDavid Majnemer } 559*dad0a645SDavid Majnemer 560*dad0a645SDavid Majnemer GVar = dyn_cast_or_null<GlobalVariable>(GVal); 561*dad0a645SDavid Majnemer if (!GVar) 562*dad0a645SDavid Majnemer return emitError( 563*dad0a645SDavid Majnemer "Linking COMDATs named '" + ComdatName + 564*dad0a645SDavid Majnemer "': GlobalVariable required for data dependent selection!"); 565*dad0a645SDavid Majnemer 566*dad0a645SDavid Majnemer return false; 567*dad0a645SDavid Majnemer } 568*dad0a645SDavid Majnemer 569*dad0a645SDavid Majnemer bool ModuleLinker::computeResultingSelectionKind(StringRef ComdatName, 570*dad0a645SDavid Majnemer Comdat::SelectionKind Src, 571*dad0a645SDavid Majnemer Comdat::SelectionKind Dst, 572*dad0a645SDavid Majnemer Comdat::SelectionKind &Result, 573*dad0a645SDavid Majnemer bool &LinkFromSrc) { 574*dad0a645SDavid Majnemer // The ability to mix Comdat::SelectionKind::Any with 575*dad0a645SDavid Majnemer // Comdat::SelectionKind::Largest is a behavior that comes from COFF. 576*dad0a645SDavid Majnemer bool DstAnyOrLargest = Dst == Comdat::SelectionKind::Any || 577*dad0a645SDavid Majnemer Dst == Comdat::SelectionKind::Largest; 578*dad0a645SDavid Majnemer bool SrcAnyOrLargest = Src == Comdat::SelectionKind::Any || 579*dad0a645SDavid Majnemer Src == Comdat::SelectionKind::Largest; 580*dad0a645SDavid Majnemer if (DstAnyOrLargest && SrcAnyOrLargest) { 581*dad0a645SDavid Majnemer if (Dst == Comdat::SelectionKind::Largest || 582*dad0a645SDavid Majnemer Src == Comdat::SelectionKind::Largest) 583*dad0a645SDavid Majnemer Result = Comdat::SelectionKind::Largest; 584*dad0a645SDavid Majnemer else 585*dad0a645SDavid Majnemer Result = Comdat::SelectionKind::Any; 586*dad0a645SDavid Majnemer } else if (Src == Dst) { 587*dad0a645SDavid Majnemer Result = Dst; 588*dad0a645SDavid Majnemer } else { 589*dad0a645SDavid Majnemer return emitError("Linking COMDATs named '" + ComdatName + 590*dad0a645SDavid Majnemer "': invalid selection kinds!"); 591*dad0a645SDavid Majnemer } 592*dad0a645SDavid Majnemer 593*dad0a645SDavid Majnemer switch (Result) { 594*dad0a645SDavid Majnemer case Comdat::SelectionKind::Any: 595*dad0a645SDavid Majnemer // Go with Dst. 596*dad0a645SDavid Majnemer LinkFromSrc = false; 597*dad0a645SDavid Majnemer break; 598*dad0a645SDavid Majnemer case Comdat::SelectionKind::NoDuplicates: 599*dad0a645SDavid Majnemer return emitError("Linking COMDATs named '" + ComdatName + 600*dad0a645SDavid Majnemer "': noduplicates has been violated!"); 601*dad0a645SDavid Majnemer case Comdat::SelectionKind::ExactMatch: 602*dad0a645SDavid Majnemer case Comdat::SelectionKind::Largest: 603*dad0a645SDavid Majnemer case Comdat::SelectionKind::SameSize: { 604*dad0a645SDavid Majnemer const GlobalVariable *DstGV; 605*dad0a645SDavid Majnemer const GlobalVariable *SrcGV; 606*dad0a645SDavid Majnemer if (getComdatLeader(DstM, ComdatName, DstGV) || 607*dad0a645SDavid Majnemer getComdatLeader(SrcM, ComdatName, SrcGV)) 608*dad0a645SDavid Majnemer return true; 609*dad0a645SDavid Majnemer 610*dad0a645SDavid Majnemer const DataLayout *DstDL = DstM->getDataLayout(); 611*dad0a645SDavid Majnemer const DataLayout *SrcDL = SrcM->getDataLayout(); 612*dad0a645SDavid Majnemer if (!DstDL || !SrcDL) { 613*dad0a645SDavid Majnemer return emitError( 614*dad0a645SDavid Majnemer "Linking COMDATs named '" + ComdatName + 615*dad0a645SDavid Majnemer "': can't do size dependent selection without DataLayout!"); 616*dad0a645SDavid Majnemer } 617*dad0a645SDavid Majnemer uint64_t DstSize = 618*dad0a645SDavid Majnemer DstDL->getTypeAllocSize(DstGV->getType()->getPointerElementType()); 619*dad0a645SDavid Majnemer uint64_t SrcSize = 620*dad0a645SDavid Majnemer SrcDL->getTypeAllocSize(SrcGV->getType()->getPointerElementType()); 621*dad0a645SDavid Majnemer if (Result == Comdat::SelectionKind::ExactMatch) { 622*dad0a645SDavid Majnemer if (SrcGV->getInitializer() != DstGV->getInitializer()) 623*dad0a645SDavid Majnemer return emitError("Linking COMDATs named '" + ComdatName + 624*dad0a645SDavid Majnemer "': ExactMatch violated!"); 625*dad0a645SDavid Majnemer LinkFromSrc = false; 626*dad0a645SDavid Majnemer } else if (Result == Comdat::SelectionKind::Largest) { 627*dad0a645SDavid Majnemer LinkFromSrc = SrcSize > DstSize; 628*dad0a645SDavid Majnemer } else if (Result == Comdat::SelectionKind::SameSize) { 629*dad0a645SDavid Majnemer if (SrcSize != DstSize) 630*dad0a645SDavid Majnemer return emitError("Linking COMDATs named '" + ComdatName + 631*dad0a645SDavid Majnemer "': SameSize violated!"); 632*dad0a645SDavid Majnemer LinkFromSrc = false; 633*dad0a645SDavid Majnemer } else { 634*dad0a645SDavid Majnemer llvm_unreachable("unknown selection kind"); 635*dad0a645SDavid Majnemer } 636*dad0a645SDavid Majnemer break; 637*dad0a645SDavid Majnemer } 638*dad0a645SDavid Majnemer } 639*dad0a645SDavid Majnemer 640*dad0a645SDavid Majnemer return false; 641*dad0a645SDavid Majnemer } 642*dad0a645SDavid Majnemer 643*dad0a645SDavid Majnemer bool ModuleLinker::getComdatResult(const Comdat *SrcC, 644*dad0a645SDavid Majnemer Comdat::SelectionKind &Result, 645*dad0a645SDavid Majnemer bool &LinkFromSrc) { 646*dad0a645SDavid Majnemer StringRef ComdatName = SrcC->getName(); 647*dad0a645SDavid Majnemer Module::ComdatSymTabType &ComdatSymTab = DstM->getComdatSymbolTable(); 648*dad0a645SDavid Majnemer Module::ComdatSymTabType::iterator DstCI = ComdatSymTab.find(ComdatName); 649*dad0a645SDavid Majnemer if (DstCI != ComdatSymTab.end()) { 650*dad0a645SDavid Majnemer const Comdat *DstC = &DstCI->second; 651*dad0a645SDavid Majnemer Comdat::SelectionKind SSK = SrcC->getSelectionKind(); 652*dad0a645SDavid Majnemer Comdat::SelectionKind DSK = DstC->getSelectionKind(); 653*dad0a645SDavid Majnemer if (computeResultingSelectionKind(ComdatName, SSK, DSK, Result, LinkFromSrc)) 654*dad0a645SDavid Majnemer return true; 655*dad0a645SDavid Majnemer } 656*dad0a645SDavid Majnemer return false; 657*dad0a645SDavid Majnemer } 658f6f121e2SJames Molloy 659b1ed91f3SChris Lattner /// getLinkageResult - This analyzes the two global values and determines what 660fc61de39SChris Lattner /// the result will look like in the destination module. In particular, it 66123f8d64bSRafael Espindola /// computes the resultant linkage type and visibility, computes whether the 66223f8d64bSRafael Espindola /// global in the source should be copied over to the destination (replacing 66323f8d64bSRafael Espindola /// the existing one), and computes whether this linkage is an error or not. 664b1ed91f3SChris Lattner bool ModuleLinker::getLinkageResult(GlobalValue *Dest, const GlobalValue *Src, 665b1ed91f3SChris Lattner GlobalValue::LinkageTypes <, 66623f8d64bSRafael Espindola GlobalValue::VisibilityTypes &Vis, 667b1ed91f3SChris Lattner bool &LinkFromSrc) { 668b1ed91f3SChris Lattner assert(Dest && "Must have two globals being queried"); 669b1ed91f3SChris Lattner assert(!Src->hasLocalLinkage() && 670fc61de39SChris Lattner "If Src has internal linkage, Dest shouldn't be set!"); 671b1ed91f3SChris Lattner 6728bb15d82SPeter Collingbourne bool SrcIsDeclaration = Src->isDeclaration() && !Src->isMaterializable(); 6730c134b52SChris Lattner bool DestIsDeclaration = Dest->isDeclaration(); 674b1ed91f3SChris Lattner 675b1ed91f3SChris Lattner if (SrcIsDeclaration) { 6761f93c50aSAnton Korobeynikov // If Src is external or if both Src & Dest are external.. Just link the 677fc61de39SChris Lattner // external globals, we aren't adding anything. 6787157bb76SNico Rieck if (Src->hasDLLImportStorageClass()) { 6797157bb76SNico Rieck // If one of GVs is marked as DLLImport, result should be dllimport'ed. 680b1ed91f3SChris Lattner if (DestIsDeclaration) { 681d61d39ecSAnton Korobeynikov LinkFromSrc = true; 682d61d39ecSAnton Korobeynikov LT = Src->getLinkage(); 683d61d39ecSAnton Korobeynikov } 684e06036d2SAndrew Lenharth } else if (Dest->hasExternalWeakLinkage()) { 68512da8ce3SDuncan Sands // If the Dest is weak, use the source linkage. 686e06036d2SAndrew Lenharth LinkFromSrc = true; 687e06036d2SAndrew Lenharth LT = Src->getLinkage(); 688d61d39ecSAnton Korobeynikov } else { 689fc61de39SChris Lattner LinkFromSrc = false; 690fc61de39SChris Lattner LT = Dest->getLinkage(); 691d61d39ecSAnton Korobeynikov } 6927157bb76SNico Rieck } else if (DestIsDeclaration && !Dest->hasDLLImportStorageClass()) { 693fc61de39SChris Lattner // If Dest is external but Src is not: 694fc61de39SChris Lattner LinkFromSrc = true; 695fc61de39SChris Lattner LT = Src->getLinkage(); 696d725c991SDuncan Sands } else if (Src->isWeakForLinker()) { 697ce4396bcSDale Johannesen // At this point we know that Dest has LinkOnce, External*, Weak, Common, 698ce4396bcSDale Johannesen // or DLL* linkage. 699184f1be4SChris Lattner if (Dest->hasExternalWeakLinkage() || 700184f1be4SChris Lattner Dest->hasAvailableExternallyLinkage() || 701184f1be4SChris Lattner (Dest->hasLinkOnceLinkage() && 702184f1be4SChris Lattner (Src->hasWeakLinkage() || Src->hasCommonLinkage()))) { 703fc61de39SChris Lattner LinkFromSrc = true; 704fc61de39SChris Lattner LT = Src->getLinkage(); 705fc61de39SChris Lattner } else { 706fc61de39SChris Lattner LinkFromSrc = false; 707fc61de39SChris Lattner LT = Dest->getLinkage(); 708fc61de39SChris Lattner } 709d725c991SDuncan Sands } else if (Dest->isWeakForLinker()) { 71012c94949SAnton Korobeynikov // At this point we know that Src has External* or DLL* linkage. 71112c94949SAnton Korobeynikov if (Src->hasExternalWeakLinkage()) { 71212c94949SAnton Korobeynikov LinkFromSrc = false; 71312c94949SAnton Korobeynikov LT = Dest->getLinkage(); 71412c94949SAnton Korobeynikov } else { 715fc61de39SChris Lattner LinkFromSrc = true; 716fc61de39SChris Lattner LT = GlobalValue::ExternalLinkage; 71712c94949SAnton Korobeynikov } 718fc61de39SChris Lattner } else { 7197157bb76SNico Rieck assert((Dest->hasExternalLinkage() || Dest->hasExternalWeakLinkage()) && 7207157bb76SNico Rieck (Src->hasExternalLinkage() || Src->hasExternalWeakLinkage()) && 721fc61de39SChris Lattner "Unexpected linkage type!"); 722b1ed91f3SChris Lattner return emitError("Linking globals named '" + Src->getName() + 723fc61de39SChris Lattner "': symbol multiply defined!"); 724fc61de39SChris Lattner } 72531fc4f95SAnton Korobeynikov 72623f8d64bSRafael Espindola // Compute the visibility. We follow the rules in the System V Application 72723f8d64bSRafael Espindola // Binary Interface. 728b2becfdbSDuncan P. N. Exon Smith assert(!GlobalValue::isLocalLinkage(LT) && 729b2becfdbSDuncan P. N. Exon Smith "Symbols with local linkage should not be merged"); 73023f8d64bSRafael Espindola Vis = isLessConstraining(Src->getVisibility(), Dest->getVisibility()) ? 73123f8d64bSRafael Espindola Dest->getVisibility() : Src->getVisibility(); 732fc61de39SChris Lattner return false; 733fc61de39SChris Lattner } 734361e513dSReid Spencer 735b1ed91f3SChris Lattner /// computeTypeMapping - Loop over all of the linked values to compute type 736b1ed91f3SChris Lattner /// mappings. For example, if we link "extern Foo *x" and "Foo *x = NULL", then 737b1ed91f3SChris Lattner /// we have two struct types 'Foo' but one got renamed when the module was 738b1ed91f3SChris Lattner /// loaded into the same LLVMContext. 739b1ed91f3SChris Lattner void ModuleLinker::computeTypeMapping() { 740b1ed91f3SChris Lattner // Incorporate globals. 741b1ed91f3SChris Lattner for (Module::global_iterator I = SrcM->global_begin(), 742b1ed91f3SChris Lattner E = SrcM->global_end(); I != E; ++I) { 743b1ed91f3SChris Lattner GlobalValue *DGV = getLinkedToGlobal(I); 7442617dcceSCraig Topper if (!DGV) continue; 745b1ed91f3SChris Lattner 746b1ed91f3SChris Lattner if (!DGV->hasAppendingLinkage() || !I->hasAppendingLinkage()) { 747b1ed91f3SChris Lattner TypeMap.addTypeMapping(DGV->getType(), I->getType()); 748b1ed91f3SChris Lattner continue; 7495c310be0SDevang Patel } 7505c310be0SDevang Patel 751b1ed91f3SChris Lattner // Unify the element type of appending arrays. 752b1ed91f3SChris Lattner ArrayType *DAT = cast<ArrayType>(DGV->getType()->getElementType()); 753b1ed91f3SChris Lattner ArrayType *SAT = cast<ArrayType>(I->getType()->getElementType()); 754b1ed91f3SChris Lattner TypeMap.addTypeMapping(DAT->getElementType(), SAT->getElementType()); 755b1ed91f3SChris Lattner } 75634fc5a0eSChris Lattner 757b1ed91f3SChris Lattner // Incorporate functions. 758b1ed91f3SChris Lattner for (Module::iterator I = SrcM->begin(), E = SrcM->end(); I != E; ++I) { 759b1ed91f3SChris Lattner if (GlobalValue *DGV = getLinkedToGlobal(I)) 760b1ed91f3SChris Lattner TypeMap.addTypeMapping(DGV->getType(), I->getType()); 761b1ed91f3SChris Lattner } 762e79f4c77SAnton Korobeynikov 763d48b7783SBill Wendling // Incorporate types by name, scanning all the types in the source module. 764d48b7783SBill Wendling // At this point, the destination module may have a type "%foo = { i32 }" for 7652b3f61afSBill Wendling // example. When the source module got loaded into the same LLVMContext, if 7662b3f61afSBill Wendling // it had the same type, it would have been renamed to "%foo.42 = { i32 }". 7678555a37cSBill Wendling TypeFinder SrcStructTypes; 7688555a37cSBill Wendling SrcStructTypes.run(*SrcM, true); 7692b3f61afSBill Wendling SmallPtrSet<StructType*, 32> SrcStructTypesSet(SrcStructTypes.begin(), 7702b3f61afSBill Wendling SrcStructTypes.end()); 7712b3f61afSBill Wendling 7722b3f61afSBill Wendling for (unsigned i = 0, e = SrcStructTypes.size(); i != e; ++i) { 7732b3f61afSBill Wendling StructType *ST = SrcStructTypes[i]; 7742b3f61afSBill Wendling if (!ST->hasName()) continue; 7752b3f61afSBill Wendling 7762b3f61afSBill Wendling // Check to see if there is a dot in the name followed by a digit. 777d48b7783SBill Wendling size_t DotPos = ST->getName().rfind('.'); 778d48b7783SBill Wendling if (DotPos == 0 || DotPos == StringRef::npos || 77983c74e9fSGuy Benyei ST->getName().back() == '.' || 78083c74e9fSGuy Benyei !isdigit(static_cast<unsigned char>(ST->getName()[DotPos+1]))) 781d48b7783SBill Wendling continue; 7822b3f61afSBill Wendling 7832b3f61afSBill Wendling // Check to see if the destination module has a struct with the prefix name. 784d48b7783SBill Wendling if (StructType *DST = DstM->getTypeByName(ST->getName().substr(0, DotPos))) 7852b3f61afSBill Wendling // Don't use it if this actually came from the source module. They're in 7868737480dSBill Wendling // the same LLVMContext after all. Also don't use it unless the type is 7878737480dSBill Wendling // actually used in the destination module. This can happen in situations 7888737480dSBill Wendling // like this: 7898737480dSBill Wendling // 7908737480dSBill Wendling // Module A Module B 7918737480dSBill Wendling // -------- -------- 7928737480dSBill Wendling // %Z = type { %A } %B = type { %C.1 } 7938737480dSBill Wendling // %A = type { %B.1, [7 x i8] } %C.1 = type { i8* } 7948737480dSBill Wendling // %B.1 = type { %C } %A.2 = type { %B.3, [5 x i8] } 7958737480dSBill Wendling // %C = type { i8* } %B.3 = type { %C.1 } 7968737480dSBill Wendling // 7978737480dSBill Wendling // When we link Module B with Module A, the '%B' in Module B is 7988737480dSBill Wendling // used. However, that would then use '%C.1'. But when we process '%C.1', 7998737480dSBill Wendling // we prefer to take the '%C' version. So we are then left with both 8008737480dSBill Wendling // '%C.1' and '%C' being used for the same types. This leads to some 8018737480dSBill Wendling // variables using one type and some using the other. 802aa9918aaSRafael Espindola if (!SrcStructTypesSet.count(DST) && TypeMap.DstStructTypesSet.count(DST)) 8032b3f61afSBill Wendling TypeMap.addTypeMapping(DST, ST); 8042b3f61afSBill Wendling } 8052b3f61afSBill Wendling 806b1ed91f3SChris Lattner // Don't bother incorporating aliases, they aren't generally typed well. 807361e513dSReid Spencer 808b1ed91f3SChris Lattner // Now that we have discovered all of the type equivalences, get a body for 809b1ed91f3SChris Lattner // any 'opaque' types in the dest module that are now resolved. 810b1ed91f3SChris Lattner TypeMap.linkDefinedTypeBodies(); 811b1ed91f3SChris Lattner } 812c08e7a00SChris Lattner 813b1ed91f3SChris Lattner /// linkAppendingVarProto - If there were any appending global variables, link 814b1ed91f3SChris Lattner /// them together now. Return true on error. 815b1ed91f3SChris Lattner bool ModuleLinker::linkAppendingVarProto(GlobalVariable *DstGV, 816b1ed91f3SChris Lattner GlobalVariable *SrcGV) { 817d48b7783SBill Wendling 818b1ed91f3SChris Lattner if (!SrcGV->hasAppendingLinkage() || !DstGV->hasAppendingLinkage()) 819b1ed91f3SChris Lattner return emitError("Linking globals named '" + SrcGV->getName() + 820b1ed91f3SChris Lattner "': can only link appending global with another appending global!"); 821361e513dSReid Spencer 822b1ed91f3SChris Lattner ArrayType *DstTy = cast<ArrayType>(DstGV->getType()->getElementType()); 823b1ed91f3SChris Lattner ArrayType *SrcTy = 824b1ed91f3SChris Lattner cast<ArrayType>(TypeMap.get(SrcGV->getType()->getElementType())); 825b1ed91f3SChris Lattner Type *EltTy = DstTy->getElementType(); 826b1ed91f3SChris Lattner 827b1ed91f3SChris Lattner // Check to see that they two arrays agree on type. 828b1ed91f3SChris Lattner if (EltTy != SrcTy->getElementType()) 829b1ed91f3SChris Lattner return emitError("Appending variables with different element types!"); 830b1ed91f3SChris Lattner if (DstGV->isConstant() != SrcGV->isConstant()) 831b1ed91f3SChris Lattner return emitError("Appending variables linked with different const'ness!"); 832b1ed91f3SChris Lattner 833b1ed91f3SChris Lattner if (DstGV->getAlignment() != SrcGV->getAlignment()) 834b1ed91f3SChris Lattner return emitError( 835b1ed91f3SChris Lattner "Appending variables with different alignment need to be linked!"); 836b1ed91f3SChris Lattner 837b1ed91f3SChris Lattner if (DstGV->getVisibility() != SrcGV->getVisibility()) 838b1ed91f3SChris Lattner return emitError( 839b1ed91f3SChris Lattner "Appending variables with different visibility need to be linked!"); 840b1ed91f3SChris Lattner 841fac3a018SRafael Espindola if (DstGV->hasUnnamedAddr() != SrcGV->hasUnnamedAddr()) 842fac3a018SRafael Espindola return emitError( 843fac3a018SRafael Espindola "Appending variables with different unnamed_addr need to be linked!"); 844fac3a018SRafael Espindola 84564c1e180SRafael Espindola if (StringRef(DstGV->getSection()) != SrcGV->getSection()) 846b1ed91f3SChris Lattner return emitError( 847b1ed91f3SChris Lattner "Appending variables with different section name need to be linked!"); 848b1ed91f3SChris Lattner 849b1ed91f3SChris Lattner uint64_t NewSize = DstTy->getNumElements() + SrcTy->getNumElements(); 850b1ed91f3SChris Lattner ArrayType *NewType = ArrayType::get(EltTy, NewSize); 851b1ed91f3SChris Lattner 852b1ed91f3SChris Lattner // Create the new global variable. 853b1ed91f3SChris Lattner GlobalVariable *NG = 854b1ed91f3SChris Lattner new GlobalVariable(*DstGV->getParent(), NewType, SrcGV->isConstant(), 8552617dcceSCraig Topper DstGV->getLinkage(), /*init*/nullptr, /*name*/"", DstGV, 856cbe34b4cSHans Wennborg DstGV->getThreadLocalMode(), 857b1ed91f3SChris Lattner DstGV->getType()->getAddressSpace()); 858b1ed91f3SChris Lattner 859b1ed91f3SChris Lattner // Propagate alignment, visibility and section info. 860b6af2f36SBill Wendling copyGVAttributes(NG, DstGV); 861b1ed91f3SChris Lattner 862b1ed91f3SChris Lattner AppendingVarInfo AVI; 863b1ed91f3SChris Lattner AVI.NewGV = NG; 864b1ed91f3SChris Lattner AVI.DstInit = DstGV->getInitializer(); 865b1ed91f3SChris Lattner AVI.SrcInit = SrcGV->getInitializer(); 866b1ed91f3SChris Lattner AppendingVars.push_back(AVI); 867b1ed91f3SChris Lattner 868b1ed91f3SChris Lattner // Replace any uses of the two global variables with uses of the new 869b1ed91f3SChris Lattner // global. 870b1ed91f3SChris Lattner ValueMap[SrcGV] = ConstantExpr::getBitCast(NG, TypeMap.get(SrcGV->getType())); 871b1ed91f3SChris Lattner 872b1ed91f3SChris Lattner DstGV->replaceAllUsesWith(ConstantExpr::getBitCast(NG, DstGV->getType())); 873b1ed91f3SChris Lattner DstGV->eraseFromParent(); 874b1ed91f3SChris Lattner 875cbb91408STanya Lattner // Track the source variable so we don't try to link it. 876cbb91408STanya Lattner DoNotLinkFromSource.insert(SrcGV); 877cbb91408STanya Lattner 878b1ed91f3SChris Lattner return false; 879b1ed91f3SChris Lattner } 880b1ed91f3SChris Lattner 881b1ed91f3SChris Lattner /// linkGlobalProto - Loop through the global variables in the src module and 882b1ed91f3SChris Lattner /// merge them into the dest module. 883b1ed91f3SChris Lattner bool ModuleLinker::linkGlobalProto(GlobalVariable *SGV) { 884b1ed91f3SChris Lattner GlobalValue *DGV = getLinkedToGlobal(SGV); 88523f8d64bSRafael Espindola llvm::Optional<GlobalValue::VisibilityTypes> NewVisibility; 886d4885daeSRafael Espindola bool HasUnnamedAddr = SGV->hasUnnamedAddr(); 887b1ed91f3SChris Lattner 888*dad0a645SDavid Majnemer bool LinkFromSrc = false; 889*dad0a645SDavid Majnemer Comdat *DC = nullptr; 890*dad0a645SDavid Majnemer if (const Comdat *SC = SGV->getComdat()) { 891*dad0a645SDavid Majnemer Comdat::SelectionKind SK; 892*dad0a645SDavid Majnemer std::tie(SK, LinkFromSrc) = ComdatsChosen[SC]; 893*dad0a645SDavid Majnemer DC = DstM->getOrInsertComdat(SC->getName()); 894*dad0a645SDavid Majnemer DC->setSelectionKind(SK); 895*dad0a645SDavid Majnemer } 896*dad0a645SDavid Majnemer 897b1ed91f3SChris Lattner if (DGV) { 898*dad0a645SDavid Majnemer if (!DC) { 899b1ed91f3SChris Lattner // Concatenation of appending linkage variables is magic and handled later. 900b1ed91f3SChris Lattner if (DGV->hasAppendingLinkage() || SGV->hasAppendingLinkage()) 901b1ed91f3SChris Lattner return linkAppendingVarProto(cast<GlobalVariable>(DGV), SGV); 902b1ed91f3SChris Lattner 903b1ed91f3SChris Lattner // Determine whether linkage of these two globals follows the source 904b1ed91f3SChris Lattner // module's definition or the destination module's definition. 9051b9633d7SChris Lattner GlobalValue::LinkageTypes NewLinkage = GlobalValue::InternalLinkage; 90623f8d64bSRafael Espindola GlobalValue::VisibilityTypes NV; 90723f8d64bSRafael Espindola if (getLinkageResult(DGV, SGV, NewLinkage, NV, LinkFromSrc)) 908fc61de39SChris Lattner return true; 90923f8d64bSRafael Espindola NewVisibility = NV; 910fd9a9415SRafael Espindola HasUnnamedAddr = HasUnnamedAddr && DGV->hasUnnamedAddr(); 911361e513dSReid Spencer 912b1ed91f3SChris Lattner // If we're not linking from the source, then keep the definition that we 913b1ed91f3SChris Lattner // have. 914b1ed91f3SChris Lattner if (!LinkFromSrc) { 91534fc5a0eSChris Lattner // Special case for const propagation. 91634fc5a0eSChris Lattner if (GlobalVariable *DGVar = dyn_cast<GlobalVariable>(DGV)) 917*dad0a645SDavid Majnemer if (DGVar->isDeclaration() && SGV->isConstant() && 918*dad0a645SDavid Majnemer !DGVar->isConstant()) 91934fc5a0eSChris Lattner DGVar->setConstant(true); 92034fc5a0eSChris Lattner 921d4885daeSRafael Espindola // Set calculated linkage, visibility and unnamed_addr. 92234fc5a0eSChris Lattner DGV->setLinkage(NewLinkage); 92323f8d64bSRafael Espindola DGV->setVisibility(*NewVisibility); 924d4885daeSRafael Espindola DGV->setUnnamedAddr(HasUnnamedAddr); 925*dad0a645SDavid Majnemer } 926*dad0a645SDavid Majnemer } 92734fc5a0eSChris Lattner 928*dad0a645SDavid Majnemer if (!LinkFromSrc) { 929b1ed91f3SChris Lattner // Make sure to remember this mapping. 930b1ed91f3SChris Lattner ValueMap[SGV] = ConstantExpr::getBitCast(DGV,TypeMap.get(SGV->getType())); 931b1ed91f3SChris Lattner 932cbb91408STanya Lattner // Track the source global so that we don't attempt to copy it over when 933cbb91408STanya Lattner // processing global initializers. 934cbb91408STanya Lattner DoNotLinkFromSource.insert(SGV); 935cbb91408STanya Lattner 936b1ed91f3SChris Lattner return false; 937361e513dSReid Spencer } 938b1ed91f3SChris Lattner } 939b1ed91f3SChris Lattner 940*dad0a645SDavid Majnemer // If the Comdat this variable was inside of wasn't selected, skip it. 941*dad0a645SDavid Majnemer if (DC && !DGV && !LinkFromSrc) { 942*dad0a645SDavid Majnemer DoNotLinkFromSource.insert(SGV); 943*dad0a645SDavid Majnemer return false; 944*dad0a645SDavid Majnemer } 945*dad0a645SDavid Majnemer 946b1ed91f3SChris Lattner // No linking to be performed or linking from the source: simply create an 947b1ed91f3SChris Lattner // identical version of the symbol over in the dest module... the 948b1ed91f3SChris Lattner // initializer will be filled in later by LinkGlobalInits. 949b1ed91f3SChris Lattner GlobalVariable *NewDGV = 950b1ed91f3SChris Lattner new GlobalVariable(*DstM, TypeMap.get(SGV->getType()->getElementType()), 9512617dcceSCraig Topper SGV->isConstant(), SGV->getLinkage(), /*init*/nullptr, 9522617dcceSCraig Topper SGV->getName(), /*insertbefore*/nullptr, 953cbe34b4cSHans Wennborg SGV->getThreadLocalMode(), 954b1ed91f3SChris Lattner SGV->getType()->getAddressSpace()); 955b1ed91f3SChris Lattner // Propagate alignment, visibility and section info. 956b6af2f36SBill Wendling copyGVAttributes(NewDGV, SGV); 95723f8d64bSRafael Espindola if (NewVisibility) 95823f8d64bSRafael Espindola NewDGV->setVisibility(*NewVisibility); 959d4885daeSRafael Espindola NewDGV->setUnnamedAddr(HasUnnamedAddr); 960b1ed91f3SChris Lattner 961*dad0a645SDavid Majnemer if (DC) 962*dad0a645SDavid Majnemer NewDGV->setComdat(DC); 963*dad0a645SDavid Majnemer 964b1ed91f3SChris Lattner if (DGV) { 965b1ed91f3SChris Lattner DGV->replaceAllUsesWith(ConstantExpr::getBitCast(NewDGV, DGV->getType())); 966b1ed91f3SChris Lattner DGV->eraseFromParent(); 967b1ed91f3SChris Lattner } 968b1ed91f3SChris Lattner 969b1ed91f3SChris Lattner // Make sure to remember this mapping. 970b1ed91f3SChris Lattner ValueMap[SGV] = NewDGV; 971361e513dSReid Spencer return false; 972361e513dSReid Spencer } 973361e513dSReid Spencer 974b1ed91f3SChris Lattner /// linkFunctionProto - Link the function in the source module into the 975b1ed91f3SChris Lattner /// destination module if needed, setting up mapping information. 976b1ed91f3SChris Lattner bool ModuleLinker::linkFunctionProto(Function *SF) { 977b1ed91f3SChris Lattner GlobalValue *DGV = getLinkedToGlobal(SF); 97823f8d64bSRafael Espindola llvm::Optional<GlobalValue::VisibilityTypes> NewVisibility; 979fd9a9415SRafael Espindola bool HasUnnamedAddr = SF->hasUnnamedAddr(); 980dac5fa94SAnton Korobeynikov 9810ead7a50SChris Lattner bool LinkFromSrc = false; 982*dad0a645SDavid Majnemer Comdat *DC = nullptr; 983*dad0a645SDavid Majnemer if (const Comdat *SC = SF->getComdat()) { 984*dad0a645SDavid Majnemer Comdat::SelectionKind SK; 985*dad0a645SDavid Majnemer std::tie(SK, LinkFromSrc) = ComdatsChosen[SC]; 986*dad0a645SDavid Majnemer DC = DstM->getOrInsertComdat(SC->getName()); 987*dad0a645SDavid Majnemer DC->setSelectionKind(SK); 988*dad0a645SDavid Majnemer } 989*dad0a645SDavid Majnemer 990*dad0a645SDavid Majnemer if (DGV) { 991*dad0a645SDavid Majnemer if (!DC) { 992*dad0a645SDavid Majnemer GlobalValue::LinkageTypes NewLinkage = GlobalValue::InternalLinkage; 99323f8d64bSRafael Espindola GlobalValue::VisibilityTypes NV; 99423f8d64bSRafael Espindola if (getLinkageResult(DGV, SF, NewLinkage, NV, LinkFromSrc)) 9950ead7a50SChris Lattner return true; 99623f8d64bSRafael Espindola NewVisibility = NV; 997fd9a9415SRafael Espindola HasUnnamedAddr = HasUnnamedAddr && DGV->hasUnnamedAddr(); 9980ead7a50SChris Lattner 999b1ed91f3SChris Lattner if (!LinkFromSrc) { 10000ead7a50SChris Lattner // Set calculated linkage 10010ead7a50SChris Lattner DGV->setLinkage(NewLinkage); 100223f8d64bSRafael Espindola DGV->setVisibility(*NewVisibility); 1003fd9a9415SRafael Espindola DGV->setUnnamedAddr(HasUnnamedAddr); 1004*dad0a645SDavid Majnemer } 1005*dad0a645SDavid Majnemer } 100677e8ac4dSChris Lattner 1007*dad0a645SDavid Majnemer if (!LinkFromSrc) { 10080ead7a50SChris Lattner // Make sure to remember this mapping. 1009b1ed91f3SChris Lattner ValueMap[SF] = ConstantExpr::getBitCast(DGV, TypeMap.get(SF->getType())); 1010b1ed91f3SChris Lattner 1011cbb91408STanya Lattner // Track the function from the source module so we don't attempt to remap 1012cbb91408STanya Lattner // it. 1013cbb91408STanya Lattner DoNotLinkFromSource.insert(SF); 1014cbb91408STanya Lattner 1015b1ed91f3SChris Lattner return false; 1016361e513dSReid Spencer } 1017b1ed91f3SChris Lattner } 1018b1ed91f3SChris Lattner 1019f6f121e2SJames Molloy // If the function is to be lazily linked, don't create it just yet. 1020f6f121e2SJames Molloy // The ValueMaterializerTy will deal with creating it if it's used. 1021f6f121e2SJames Molloy if (!DGV && (SF->hasLocalLinkage() || SF->hasLinkOnceLinkage() || 1022f6f121e2SJames Molloy SF->hasAvailableExternallyLinkage())) { 1023f6f121e2SJames Molloy DoNotLinkFromSource.insert(SF); 1024f6f121e2SJames Molloy return false; 1025f6f121e2SJames Molloy } 1026f6f121e2SJames Molloy 1027*dad0a645SDavid Majnemer // If the Comdat this function was inside of wasn't selected, skip it. 1028*dad0a645SDavid Majnemer if (DC && !DGV && !LinkFromSrc) { 1029*dad0a645SDavid Majnemer DoNotLinkFromSource.insert(SF); 1030*dad0a645SDavid Majnemer return false; 1031*dad0a645SDavid Majnemer } 1032*dad0a645SDavid Majnemer 1033b1ed91f3SChris Lattner // If there is no linkage to be performed or we are linking from the source, 1034b1ed91f3SChris Lattner // bring SF over. 1035b1ed91f3SChris Lattner Function *NewDF = Function::Create(TypeMap.get(SF->getFunctionType()), 1036b1ed91f3SChris Lattner SF->getLinkage(), SF->getName(), DstM); 1037b6af2f36SBill Wendling copyGVAttributes(NewDF, SF); 103823f8d64bSRafael Espindola if (NewVisibility) 103923f8d64bSRafael Espindola NewDF->setVisibility(*NewVisibility); 1040fd9a9415SRafael Espindola NewDF->setUnnamedAddr(HasUnnamedAddr); 1041b1ed91f3SChris Lattner 1042*dad0a645SDavid Majnemer if (DC) 1043*dad0a645SDavid Majnemer NewDF->setComdat(DC); 1044*dad0a645SDavid Majnemer 1045b1ed91f3SChris Lattner if (DGV) { 1046b1ed91f3SChris Lattner // Any uses of DF need to change to NewDF, with cast. 1047b1ed91f3SChris Lattner DGV->replaceAllUsesWith(ConstantExpr::getBitCast(NewDF, DGV->getType())); 1048b1ed91f3SChris Lattner DGV->eraseFromParent(); 1049b1ed91f3SChris Lattner } 1050b1ed91f3SChris Lattner 1051b1ed91f3SChris Lattner ValueMap[SF] = NewDF; 1052361e513dSReid Spencer return false; 1053361e513dSReid Spencer } 1054361e513dSReid Spencer 1055b1ed91f3SChris Lattner /// LinkAliasProto - Set up prototypes for any aliases that come over from the 1056b1ed91f3SChris Lattner /// source module. 1057b1ed91f3SChris Lattner bool ModuleLinker::linkAliasProto(GlobalAlias *SGA) { 1058b1ed91f3SChris Lattner GlobalValue *DGV = getLinkedToGlobal(SGA); 105923f8d64bSRafael Espindola llvm::Optional<GlobalValue::VisibilityTypes> NewVisibility; 106042a4c9f9SRafael Espindola bool HasUnnamedAddr = SGA->hasUnnamedAddr(); 1061b1ed91f3SChris Lattner 1062*dad0a645SDavid Majnemer bool LinkFromSrc = false; 1063*dad0a645SDavid Majnemer Comdat *DC = nullptr; 1064*dad0a645SDavid Majnemer if (const Comdat *SC = SGA->getComdat()) { 1065*dad0a645SDavid Majnemer Comdat::SelectionKind SK; 1066*dad0a645SDavid Majnemer std::tie(SK, LinkFromSrc) = ComdatsChosen[SC]; 1067*dad0a645SDavid Majnemer DC = DstM->getOrInsertComdat(SC->getName()); 1068*dad0a645SDavid Majnemer DC->setSelectionKind(SK); 1069*dad0a645SDavid Majnemer } 1070*dad0a645SDavid Majnemer 1071b1ed91f3SChris Lattner if (DGV) { 1072*dad0a645SDavid Majnemer if (!DC) { 1073b1ed91f3SChris Lattner GlobalValue::LinkageTypes NewLinkage = GlobalValue::InternalLinkage; 107423f8d64bSRafael Espindola GlobalValue::VisibilityTypes NV; 107523f8d64bSRafael Espindola if (getLinkageResult(DGV, SGA, NewLinkage, NV, LinkFromSrc)) 1076b1ed91f3SChris Lattner return true; 107723f8d64bSRafael Espindola NewVisibility = NV; 107842a4c9f9SRafael Espindola HasUnnamedAddr = HasUnnamedAddr && DGV->hasUnnamedAddr(); 1079b1ed91f3SChris Lattner 1080b1ed91f3SChris Lattner if (!LinkFromSrc) { 1081b1ed91f3SChris Lattner // Set calculated linkage. 1082b1ed91f3SChris Lattner DGV->setLinkage(NewLinkage); 108323f8d64bSRafael Espindola DGV->setVisibility(*NewVisibility); 108442a4c9f9SRafael Espindola DGV->setUnnamedAddr(HasUnnamedAddr); 1085*dad0a645SDavid Majnemer } 1086*dad0a645SDavid Majnemer } 1087b1ed91f3SChris Lattner 1088*dad0a645SDavid Majnemer if (!LinkFromSrc) { 1089b1ed91f3SChris Lattner // Make sure to remember this mapping. 1090b1ed91f3SChris Lattner ValueMap[SGA] = ConstantExpr::getBitCast(DGV,TypeMap.get(SGA->getType())); 1091b1ed91f3SChris Lattner 1092cbb91408STanya Lattner // Track the alias from the source module so we don't attempt to remap it. 1093cbb91408STanya Lattner DoNotLinkFromSource.insert(SGA); 1094cbb91408STanya Lattner 1095b1ed91f3SChris Lattner return false; 1096b1ed91f3SChris Lattner } 1097b1ed91f3SChris Lattner } 1098b1ed91f3SChris Lattner 1099*dad0a645SDavid Majnemer // If the Comdat this alias was inside of wasn't selected, skip it. 1100*dad0a645SDavid Majnemer if (DC && !DGV && !LinkFromSrc) { 1101*dad0a645SDavid Majnemer DoNotLinkFromSource.insert(SGA); 1102*dad0a645SDavid Majnemer return false; 1103*dad0a645SDavid Majnemer } 1104*dad0a645SDavid Majnemer 1105b1ed91f3SChris Lattner // If there is no linkage to be performed or we're linking from the source, 1106b1ed91f3SChris Lattner // bring over SGA. 11074fe0094fSRafael Espindola auto *PTy = cast<PointerType>(TypeMap.get(SGA->getType())); 1108f1bedd37SRafael Espindola auto *NewDA = 1109f1bedd37SRafael Espindola GlobalAlias::create(PTy->getElementType(), PTy->getAddressSpace(), 111083705658SRafael Espindola SGA->getLinkage(), SGA->getName(), DstM); 1111b6af2f36SBill Wendling copyGVAttributes(NewDA, SGA); 111223f8d64bSRafael Espindola if (NewVisibility) 111323f8d64bSRafael Espindola NewDA->setVisibility(*NewVisibility); 111442a4c9f9SRafael Espindola NewDA->setUnnamedAddr(HasUnnamedAddr); 1115b1ed91f3SChris Lattner 111664c1e180SRafael Espindola if (DGV) { 111764c1e180SRafael Espindola // Any uses of DGV need to change to NewDA, with cast. 111864c1e180SRafael Espindola DGV->replaceAllUsesWith(ConstantExpr::getBitCast(NewDA, DGV->getType())); 111964c1e180SRafael Espindola DGV->eraseFromParent(); 112064c1e180SRafael Espindola } 1121b1ed91f3SChris Lattner 1122b1ed91f3SChris Lattner ValueMap[SGA] = NewDA; 1123b1ed91f3SChris Lattner return false; 1124b1ed91f3SChris Lattner } 1125b1ed91f3SChris Lattner 112600245f42SChris Lattner static void getArrayElements(Constant *C, SmallVectorImpl<Constant*> &Dest) { 11276705883aSChris Lattner unsigned NumElements = cast<ArrayType>(C->getType())->getNumElements(); 112800245f42SChris Lattner 11296705883aSChris Lattner for (unsigned i = 0; i != NumElements; ++i) 11306705883aSChris Lattner Dest.push_back(C->getAggregateElement(i)); 113100245f42SChris Lattner } 113200245f42SChris Lattner 1133b1ed91f3SChris Lattner void ModuleLinker::linkAppendingVarInit(const AppendingVarInfo &AVI) { 1134b1ed91f3SChris Lattner // Merge the initializer. 1135b1ed91f3SChris Lattner SmallVector<Constant*, 16> Elements; 113600245f42SChris Lattner getArrayElements(AVI.DstInit, Elements); 1137b1ed91f3SChris Lattner 1138f6f121e2SJames Molloy Constant *SrcInit = MapValue(AVI.SrcInit, ValueMap, RF_None, &TypeMap, &ValMaterializer); 113900245f42SChris Lattner getArrayElements(SrcInit, Elements); 114000245f42SChris Lattner 1141b1ed91f3SChris Lattner ArrayType *NewType = cast<ArrayType>(AVI.NewGV->getType()->getElementType()); 1142b1ed91f3SChris Lattner AVI.NewGV->setInitializer(ConstantArray::get(NewType, Elements)); 1143b1ed91f3SChris Lattner } 1144b1ed91f3SChris Lattner 1145b6af2f36SBill Wendling /// linkGlobalInits - Update the initializers in the Dest module now that all 1146b6af2f36SBill Wendling /// globals that may be referenced are in Dest. 1147b1ed91f3SChris Lattner void ModuleLinker::linkGlobalInits() { 1148b1ed91f3SChris Lattner // Loop over all of the globals in the src module, mapping them over as we go 1149b1ed91f3SChris Lattner for (Module::const_global_iterator I = SrcM->global_begin(), 1150b1ed91f3SChris Lattner E = SrcM->global_end(); I != E; ++I) { 1151cbb91408STanya Lattner 1152cbb91408STanya Lattner // Only process initialized GV's or ones not already in dest. 1153cbb91408STanya Lattner if (!I->hasInitializer() || DoNotLinkFromSource.count(I)) continue; 1154b1ed91f3SChris Lattner 1155b1ed91f3SChris Lattner // Grab destination global variable. 1156b1ed91f3SChris Lattner GlobalVariable *DGV = cast<GlobalVariable>(ValueMap[I]); 1157b1ed91f3SChris Lattner // Figure out what the initializer looks like in the dest module. 1158b1ed91f3SChris Lattner DGV->setInitializer(MapValue(I->getInitializer(), ValueMap, 1159f6f121e2SJames Molloy RF_None, &TypeMap, &ValMaterializer)); 1160b1ed91f3SChris Lattner } 1161b1ed91f3SChris Lattner } 1162b1ed91f3SChris Lattner 1163b6af2f36SBill Wendling /// linkFunctionBody - Copy the source function over into the dest function and 1164b6af2f36SBill Wendling /// fix up references to values. At this point we know that Dest is an external 1165b6af2f36SBill Wendling /// function, and that Src is not. 1166b1ed91f3SChris Lattner void ModuleLinker::linkFunctionBody(Function *Dst, Function *Src) { 1167b1ed91f3SChris Lattner assert(Src && Dst && Dst->isDeclaration() && !Src->isDeclaration()); 1168361e513dSReid Spencer 11697391dde1SChris Lattner // Go through and convert function arguments over, remembering the mapping. 1170b1ed91f3SChris Lattner Function::arg_iterator DI = Dst->arg_begin(); 1171531f9e92SChris Lattner for (Function::arg_iterator I = Src->arg_begin(), E = Src->arg_end(); 1172361e513dSReid Spencer I != E; ++I, ++DI) { 1173b1ed91f3SChris Lattner DI->setName(I->getName()); // Copy the name over. 1174361e513dSReid Spencer 1175b1ed91f3SChris Lattner // Add a mapping to our mapping. 117666a6271cSAnton Korobeynikov ValueMap[I] = DI; 1177361e513dSReid Spencer } 1178361e513dSReid Spencer 1179cbb91408STanya Lattner if (Mode == Linker::DestroySource) { 11802f0557d0SChris Lattner // Splice the body of the source function into the dest function. 1181b1ed91f3SChris Lattner Dst->getBasicBlockList().splice(Dst->end(), Src->getBasicBlockList()); 1182361e513dSReid Spencer 1183361e513dSReid Spencer // At this point, all of the instructions and values of the function are now 1184361e513dSReid Spencer // copied over. The only problem is that they are still referencing values in 1185361e513dSReid Spencer // the Source function as operands. Loop through all of the operands of the 118643f8d164SChris Lattner // functions and patch them up to point to the local versions. 1187b1ed91f3SChris Lattner for (Function::iterator BB = Dst->begin(), BE = Dst->end(); BB != BE; ++BB) 118843f8d164SChris Lattner for (BasicBlock::iterator I = BB->begin(), E = BB->end(); I != E; ++I) 1189f6f121e2SJames Molloy RemapInstruction(I, ValueMap, RF_IgnoreMissingEntries, 1190f6f121e2SJames Molloy &TypeMap, &ValMaterializer); 11917391dde1SChris Lattner 1192cbb91408STanya Lattner } else { 1193cbb91408STanya Lattner // Clone the body of the function into the dest function. 1194cbb91408STanya Lattner SmallVector<ReturnInst*, 8> Returns; // Ignore returns. 11952617dcceSCraig Topper CloneFunctionInto(Dst, Src, ValueMap, false, Returns, "", nullptr, 1196f6f121e2SJames Molloy &TypeMap, &ValMaterializer); 1197cbb91408STanya Lattner } 1198cbb91408STanya Lattner 11997391dde1SChris Lattner // There is no need to map the arguments anymore. 120044ab8aefSChris Lattner for (Function::arg_iterator I = Src->arg_begin(), E = Src->arg_end(); 120144ab8aefSChris Lattner I != E; ++I) 12023aaaa0b2SReid Spencer ValueMap.erase(I); 1203cbb91408STanya Lattner 1204361e513dSReid Spencer } 1205361e513dSReid Spencer 1206b6af2f36SBill Wendling /// linkAliasBodies - Insert all of the aliases in Src into the Dest module. 1207b1ed91f3SChris Lattner void ModuleLinker::linkAliasBodies() { 1208b1ed91f3SChris Lattner for (Module::alias_iterator I = SrcM->alias_begin(), E = SrcM->alias_end(); 1209cbb91408STanya Lattner I != E; ++I) { 1210cbb91408STanya Lattner if (DoNotLinkFromSource.count(I)) 1211cbb91408STanya Lattner continue; 1212b1ed91f3SChris Lattner if (Constant *Aliasee = I->getAliasee()) { 1213b1ed91f3SChris Lattner GlobalAlias *DA = cast<GlobalAlias>(ValueMap[I]); 12146b238633SRafael Espindola Constant *Val = 12156b238633SRafael Espindola MapValue(Aliasee, ValueMap, RF_None, &TypeMap, &ValMaterializer); 121664c1e180SRafael Espindola DA->setAliasee(Val); 12172c4a34aeSDavid Chisnall } 1218b1ed91f3SChris Lattner } 1219cbb91408STanya Lattner } 1220b1ed91f3SChris Lattner 1221b6af2f36SBill Wendling /// linkNamedMDNodes - Insert all of the named MDNodes in Src into the Dest 1222b1ed91f3SChris Lattner /// module. 1223b1ed91f3SChris Lattner void ModuleLinker::linkNamedMDNodes() { 122466f02413SBill Wendling const NamedMDNode *SrcModFlags = SrcM->getModuleFlagsMetadata(); 1225b1ed91f3SChris Lattner for (Module::const_named_metadata_iterator I = SrcM->named_metadata_begin(), 1226b1ed91f3SChris Lattner E = SrcM->named_metadata_end(); I != E; ++I) { 122766f02413SBill Wendling // Don't link module flags here. Do them separately. 122866f02413SBill Wendling if (&*I == SrcModFlags) continue; 1229b1ed91f3SChris Lattner NamedMDNode *DestNMD = DstM->getOrInsertNamedMetadata(I->getName()); 1230b1ed91f3SChris Lattner // Add Src elements into Dest node. 1231b1ed91f3SChris Lattner for (unsigned i = 0, e = I->getNumOperands(); i != e; ++i) 1232b1ed91f3SChris Lattner DestNMD->addOperand(MapValue(I->getOperand(i), ValueMap, 1233f6f121e2SJames Molloy RF_None, &TypeMap, &ValMaterializer)); 1234b1ed91f3SChris Lattner } 1235b1ed91f3SChris Lattner } 1236b1ed91f3SChris Lattner 123766f02413SBill Wendling /// linkModuleFlagsMetadata - Merge the linker flags in Src into the Dest 123866f02413SBill Wendling /// module. 123966f02413SBill Wendling bool ModuleLinker::linkModuleFlagsMetadata() { 12400ec72bbcSDaniel Dunbar // If the source module has no module flags, we are done. 124166f02413SBill Wendling const NamedMDNode *SrcModFlags = SrcM->getModuleFlagsMetadata(); 124266f02413SBill Wendling if (!SrcModFlags) return false; 124366f02413SBill Wendling 124466f02413SBill Wendling // If the destination module doesn't have module flags yet, then just copy 124566f02413SBill Wendling // over the source module's flags. 12460ec72bbcSDaniel Dunbar NamedMDNode *DstModFlags = DstM->getOrInsertModuleFlagsMetadata(); 124766f02413SBill Wendling if (DstModFlags->getNumOperands() == 0) { 124866f02413SBill Wendling for (unsigned I = 0, E = SrcModFlags->getNumOperands(); I != E; ++I) 124966f02413SBill Wendling DstModFlags->addOperand(SrcModFlags->getOperand(I)); 125066f02413SBill Wendling 125166f02413SBill Wendling return false; 125266f02413SBill Wendling } 125366f02413SBill Wendling 12540ec72bbcSDaniel Dunbar // First build a map of the existing module flags and requirements. 12550ec72bbcSDaniel Dunbar DenseMap<MDString*, MDNode*> Flags; 12560ec72bbcSDaniel Dunbar SmallSetVector<MDNode*, 16> Requirements; 12570ec72bbcSDaniel Dunbar for (unsigned I = 0, E = DstModFlags->getNumOperands(); I != E; ++I) { 12580ec72bbcSDaniel Dunbar MDNode *Op = DstModFlags->getOperand(I); 12590ec72bbcSDaniel Dunbar ConstantInt *Behavior = cast<ConstantInt>(Op->getOperand(0)); 12600ec72bbcSDaniel Dunbar MDString *ID = cast<MDString>(Op->getOperand(1)); 12610ec72bbcSDaniel Dunbar 12620ec72bbcSDaniel Dunbar if (Behavior->getZExtValue() == Module::Require) { 12630ec72bbcSDaniel Dunbar Requirements.insert(cast<MDNode>(Op->getOperand(2))); 12640ec72bbcSDaniel Dunbar } else { 12650ec72bbcSDaniel Dunbar Flags[ID] = Op; 12660ec72bbcSDaniel Dunbar } 12670ec72bbcSDaniel Dunbar } 12680ec72bbcSDaniel Dunbar 12690ec72bbcSDaniel Dunbar // Merge in the flags from the source module, and also collect its set of 12700ec72bbcSDaniel Dunbar // requirements. 127166f02413SBill Wendling bool HasErr = false; 12720ec72bbcSDaniel Dunbar for (unsigned I = 0, E = SrcModFlags->getNumOperands(); I != E; ++I) { 12730ec72bbcSDaniel Dunbar MDNode *SrcOp = SrcModFlags->getOperand(I); 12740ec72bbcSDaniel Dunbar ConstantInt *SrcBehavior = cast<ConstantInt>(SrcOp->getOperand(0)); 12750ec72bbcSDaniel Dunbar MDString *ID = cast<MDString>(SrcOp->getOperand(1)); 12760ec72bbcSDaniel Dunbar MDNode *DstOp = Flags.lookup(ID); 12770ec72bbcSDaniel Dunbar unsigned SrcBehaviorValue = SrcBehavior->getZExtValue(); 127866f02413SBill Wendling 12790ec72bbcSDaniel Dunbar // If this is a requirement, add it and continue. 12800ec72bbcSDaniel Dunbar if (SrcBehaviorValue == Module::Require) { 12810ec72bbcSDaniel Dunbar // If the destination module does not already have this requirement, add 12820ec72bbcSDaniel Dunbar // it. 12830ec72bbcSDaniel Dunbar if (Requirements.insert(cast<MDNode>(SrcOp->getOperand(2)))) { 12840ec72bbcSDaniel Dunbar DstModFlags->addOperand(SrcOp); 12850ec72bbcSDaniel Dunbar } 12860ec72bbcSDaniel Dunbar continue; 12870ec72bbcSDaniel Dunbar } 128866f02413SBill Wendling 12890ec72bbcSDaniel Dunbar // If there is no existing flag with this ID, just add it. 12900ec72bbcSDaniel Dunbar if (!DstOp) { 12910ec72bbcSDaniel Dunbar Flags[ID] = SrcOp; 12920ec72bbcSDaniel Dunbar DstModFlags->addOperand(SrcOp); 12930ec72bbcSDaniel Dunbar continue; 12940ec72bbcSDaniel Dunbar } 129566f02413SBill Wendling 12960ec72bbcSDaniel Dunbar // Otherwise, perform a merge. 12970ec72bbcSDaniel Dunbar ConstantInt *DstBehavior = cast<ConstantInt>(DstOp->getOperand(0)); 12980ec72bbcSDaniel Dunbar unsigned DstBehaviorValue = DstBehavior->getZExtValue(); 12990ec72bbcSDaniel Dunbar 13000ec72bbcSDaniel Dunbar // If either flag has override behavior, handle it first. 13010ec72bbcSDaniel Dunbar if (DstBehaviorValue == Module::Override) { 13020ec72bbcSDaniel Dunbar // Diagnose inconsistent flags which both have override behavior. 13030ec72bbcSDaniel Dunbar if (SrcBehaviorValue == Module::Override && 13040ec72bbcSDaniel Dunbar SrcOp->getOperand(2) != DstOp->getOperand(2)) { 13050ec72bbcSDaniel Dunbar HasErr |= emitError("linking module flags '" + ID->getString() + 13060ec72bbcSDaniel Dunbar "': IDs have conflicting override values"); 13070ec72bbcSDaniel Dunbar } 13080ec72bbcSDaniel Dunbar continue; 13090ec72bbcSDaniel Dunbar } else if (SrcBehaviorValue == Module::Override) { 13100ec72bbcSDaniel Dunbar // Update the destination flag to that of the source. 13110ec72bbcSDaniel Dunbar DstOp->replaceOperandWith(0, SrcBehavior); 13120ec72bbcSDaniel Dunbar DstOp->replaceOperandWith(2, SrcOp->getOperand(2)); 13130ec72bbcSDaniel Dunbar continue; 13140ec72bbcSDaniel Dunbar } 13150ec72bbcSDaniel Dunbar 13160ec72bbcSDaniel Dunbar // Diagnose inconsistent merge behavior types. 13170ec72bbcSDaniel Dunbar if (SrcBehaviorValue != DstBehaviorValue) { 13180ec72bbcSDaniel Dunbar HasErr |= emitError("linking module flags '" + ID->getString() + 131966f02413SBill Wendling "': IDs have conflicting behaviors"); 13200ec72bbcSDaniel Dunbar continue; 132166f02413SBill Wendling } 132266f02413SBill Wendling 13230ec72bbcSDaniel Dunbar // Perform the merge for standard behavior types. 13240ec72bbcSDaniel Dunbar switch (SrcBehaviorValue) { 13250ec72bbcSDaniel Dunbar case Module::Require: 13262a30d788SCraig Topper case Module::Override: llvm_unreachable("not possible"); 13270ec72bbcSDaniel Dunbar case Module::Error: { 13280ec72bbcSDaniel Dunbar // Emit an error if the values differ. 13290ec72bbcSDaniel Dunbar if (SrcOp->getOperand(2) != DstOp->getOperand(2)) { 13300ec72bbcSDaniel Dunbar HasErr |= emitError("linking module flags '" + ID->getString() + 13310ec72bbcSDaniel Dunbar "': IDs have conflicting values"); 133266f02413SBill Wendling } 13330ec72bbcSDaniel Dunbar continue; 133466f02413SBill Wendling } 13350ec72bbcSDaniel Dunbar case Module::Warning: { 13360ec72bbcSDaniel Dunbar // Emit a warning if the values differ. 13370ec72bbcSDaniel Dunbar if (SrcOp->getOperand(2) != DstOp->getOperand(2)) { 1338e17f3708SEli Bendersky if (!SuppressWarnings) { 13390ec72bbcSDaniel Dunbar errs() << "WARNING: linking module flags '" << ID->getString() 13400ec72bbcSDaniel Dunbar << "': IDs have conflicting values"; 13410ec72bbcSDaniel Dunbar } 1342e17f3708SEli Bendersky } 13430ec72bbcSDaniel Dunbar continue; 13440ec72bbcSDaniel Dunbar } 1345d77d9fb0SDaniel Dunbar case Module::Append: { 1346d77d9fb0SDaniel Dunbar MDNode *DstValue = cast<MDNode>(DstOp->getOperand(2)); 1347d77d9fb0SDaniel Dunbar MDNode *SrcValue = cast<MDNode>(SrcOp->getOperand(2)); 1348d77d9fb0SDaniel Dunbar unsigned NumOps = DstValue->getNumOperands() + SrcValue->getNumOperands(); 1349d77d9fb0SDaniel Dunbar Value **VP, **Values = VP = new Value*[NumOps]; 1350d77d9fb0SDaniel Dunbar for (unsigned i = 0, e = DstValue->getNumOperands(); i != e; ++i, ++VP) 1351d77d9fb0SDaniel Dunbar *VP = DstValue->getOperand(i); 1352d77d9fb0SDaniel Dunbar for (unsigned i = 0, e = SrcValue->getNumOperands(); i != e; ++i, ++VP) 1353d77d9fb0SDaniel Dunbar *VP = SrcValue->getOperand(i); 1354d77d9fb0SDaniel Dunbar DstOp->replaceOperandWith(2, MDNode::get(DstM->getContext(), 1355d77d9fb0SDaniel Dunbar ArrayRef<Value*>(Values, 1356d77d9fb0SDaniel Dunbar NumOps))); 1357d77d9fb0SDaniel Dunbar delete[] Values; 1358d77d9fb0SDaniel Dunbar break; 1359d77d9fb0SDaniel Dunbar } 1360d77d9fb0SDaniel Dunbar case Module::AppendUnique: { 1361d77d9fb0SDaniel Dunbar SmallSetVector<Value*, 16> Elts; 1362d77d9fb0SDaniel Dunbar MDNode *DstValue = cast<MDNode>(DstOp->getOperand(2)); 1363d77d9fb0SDaniel Dunbar MDNode *SrcValue = cast<MDNode>(SrcOp->getOperand(2)); 1364d77d9fb0SDaniel Dunbar for (unsigned i = 0, e = DstValue->getNumOperands(); i != e; ++i) 1365d77d9fb0SDaniel Dunbar Elts.insert(DstValue->getOperand(i)); 1366d77d9fb0SDaniel Dunbar for (unsigned i = 0, e = SrcValue->getNumOperands(); i != e; ++i) 1367d77d9fb0SDaniel Dunbar Elts.insert(SrcValue->getOperand(i)); 1368d77d9fb0SDaniel Dunbar DstOp->replaceOperandWith(2, MDNode::get(DstM->getContext(), 1369d77d9fb0SDaniel Dunbar ArrayRef<Value*>(Elts.begin(), 1370d77d9fb0SDaniel Dunbar Elts.end()))); 1371d77d9fb0SDaniel Dunbar break; 1372d77d9fb0SDaniel Dunbar } 137366f02413SBill Wendling } 137466f02413SBill Wendling } 137566f02413SBill Wendling 13760ec72bbcSDaniel Dunbar // Check all of the requirements. 13770ec72bbcSDaniel Dunbar for (unsigned I = 0, E = Requirements.size(); I != E; ++I) { 13780ec72bbcSDaniel Dunbar MDNode *Requirement = Requirements[I]; 13790ec72bbcSDaniel Dunbar MDString *Flag = cast<MDString>(Requirement->getOperand(0)); 13800ec72bbcSDaniel Dunbar Value *ReqValue = Requirement->getOperand(1); 13810ec72bbcSDaniel Dunbar 13820ec72bbcSDaniel Dunbar MDNode *Op = Flags[Flag]; 13830ec72bbcSDaniel Dunbar if (!Op || Op->getOperand(2) != ReqValue) { 13840ec72bbcSDaniel Dunbar HasErr |= emitError("linking module flags '" + Flag->getString() + 13859273b539SBill Wendling "': does not have the required value"); 13860ec72bbcSDaniel Dunbar continue; 138766f02413SBill Wendling } 138866f02413SBill Wendling } 138966f02413SBill Wendling 139066f02413SBill Wendling return HasErr; 139166f02413SBill Wendling } 139266f02413SBill Wendling 1393b1ed91f3SChris Lattner bool ModuleLinker::run() { 139466f02413SBill Wendling assert(DstM && "Null destination module"); 139566f02413SBill Wendling assert(SrcM && "Null source module"); 1396b1ed91f3SChris Lattner 1397b1ed91f3SChris Lattner // Inherit the target data from the source module if the destination module 1398b1ed91f3SChris Lattner // doesn't have one already. 1399f863ee29SRafael Espindola if (!DstM->getDataLayout() && SrcM->getDataLayout()) 1400b1ed91f3SChris Lattner DstM->setDataLayout(SrcM->getDataLayout()); 1401b1ed91f3SChris Lattner 1402b1ed91f3SChris Lattner // Copy the target triple from the source to dest if the dest's is empty. 1403b1ed91f3SChris Lattner if (DstM->getTargetTriple().empty() && !SrcM->getTargetTriple().empty()) 1404b1ed91f3SChris Lattner DstM->setTargetTriple(SrcM->getTargetTriple()); 1405b1ed91f3SChris Lattner 1406f863ee29SRafael Espindola if (SrcM->getDataLayout() && DstM->getDataLayout() && 1407ae593f15SRafael Espindola *SrcM->getDataLayout() != *DstM->getDataLayout()) { 1408e17f3708SEli Bendersky if (!SuppressWarnings) { 1409026fc5f6SJF Bastien errs() << "WARNING: Linking two modules of different data layouts: '" 1410026fc5f6SJF Bastien << SrcM->getModuleIdentifier() << "' is '" 1411026fc5f6SJF Bastien << SrcM->getDataLayoutStr() << "' whereas '" 1412026fc5f6SJF Bastien << DstM->getModuleIdentifier() << "' is '" 1413026fc5f6SJF Bastien << DstM->getDataLayoutStr() << "'\n"; 1414e17f3708SEli Bendersky } 1415e17f3708SEli Bendersky } 1416b1ed91f3SChris Lattner if (!SrcM->getTargetTriple().empty() && 1417b1ed91f3SChris Lattner DstM->getTargetTriple() != SrcM->getTargetTriple()) { 1418e17f3708SEli Bendersky if (!SuppressWarnings) { 1419026fc5f6SJF Bastien errs() << "WARNING: Linking two modules of different target triples: " 1420026fc5f6SJF Bastien << SrcM->getModuleIdentifier() << "' is '" 1421026fc5f6SJF Bastien << SrcM->getTargetTriple() << "' whereas '" 1422026fc5f6SJF Bastien << DstM->getModuleIdentifier() << "' is '" 1423b1ed91f3SChris Lattner << DstM->getTargetTriple() << "'\n"; 1424b1ed91f3SChris Lattner } 1425e17f3708SEli Bendersky } 1426b1ed91f3SChris Lattner 1427b1ed91f3SChris Lattner // Append the module inline asm string. 1428b1ed91f3SChris Lattner if (!SrcM->getModuleInlineAsm().empty()) { 1429b1ed91f3SChris Lattner if (DstM->getModuleInlineAsm().empty()) 1430b1ed91f3SChris Lattner DstM->setModuleInlineAsm(SrcM->getModuleInlineAsm()); 1431b1ed91f3SChris Lattner else 1432b1ed91f3SChris Lattner DstM->setModuleInlineAsm(DstM->getModuleInlineAsm()+"\n"+ 1433b1ed91f3SChris Lattner SrcM->getModuleInlineAsm()); 1434b1ed91f3SChris Lattner } 1435b1ed91f3SChris Lattner 1436b1ed91f3SChris Lattner // Loop over all of the linked values to compute type mappings. 1437b1ed91f3SChris Lattner computeTypeMapping(); 1438b1ed91f3SChris Lattner 1439*dad0a645SDavid Majnemer ComdatsChosen.clear(); 1440*dad0a645SDavid Majnemer for (const StringMapEntry<llvm::Comdat> &SMEC : SrcM->getComdatSymbolTable()) { 1441*dad0a645SDavid Majnemer const Comdat &C = SMEC.getValue(); 1442*dad0a645SDavid Majnemer if (ComdatsChosen.count(&C)) 1443*dad0a645SDavid Majnemer continue; 1444*dad0a645SDavid Majnemer Comdat::SelectionKind SK; 1445*dad0a645SDavid Majnemer bool LinkFromSrc; 1446*dad0a645SDavid Majnemer if (getComdatResult(&C, SK, LinkFromSrc)) 1447*dad0a645SDavid Majnemer return true; 1448*dad0a645SDavid Majnemer ComdatsChosen[&C] = std::make_pair(SK, LinkFromSrc); 1449*dad0a645SDavid Majnemer } 1450*dad0a645SDavid Majnemer 1451b1ed91f3SChris Lattner // Insert all of the globals in src into the DstM module... without linking 1452b1ed91f3SChris Lattner // initializers (which could refer to functions not yet mapped over). 1453b1ed91f3SChris Lattner for (Module::global_iterator I = SrcM->global_begin(), 1454b1ed91f3SChris Lattner E = SrcM->global_end(); I != E; ++I) 1455b1ed91f3SChris Lattner if (linkGlobalProto(I)) 1456b1ed91f3SChris Lattner return true; 1457b1ed91f3SChris Lattner 1458b1ed91f3SChris Lattner // Link the functions together between the two modules, without doing function 1459b1ed91f3SChris Lattner // bodies... this just adds external function prototypes to the DstM 1460b1ed91f3SChris Lattner // function... We do this so that when we begin processing function bodies, 1461b1ed91f3SChris Lattner // all of the global values that may be referenced are available in our 1462b1ed91f3SChris Lattner // ValueMap. 1463b1ed91f3SChris Lattner for (Module::iterator I = SrcM->begin(), E = SrcM->end(); I != E; ++I) 1464b1ed91f3SChris Lattner if (linkFunctionProto(I)) 1465b1ed91f3SChris Lattner return true; 1466b1ed91f3SChris Lattner 1467b1ed91f3SChris Lattner // If there were any aliases, link them now. 1468b1ed91f3SChris Lattner for (Module::alias_iterator I = SrcM->alias_begin(), 1469b1ed91f3SChris Lattner E = SrcM->alias_end(); I != E; ++I) 1470b1ed91f3SChris Lattner if (linkAliasProto(I)) 1471b1ed91f3SChris Lattner return true; 1472b1ed91f3SChris Lattner 1473b1ed91f3SChris Lattner for (unsigned i = 0, e = AppendingVars.size(); i != e; ++i) 1474b1ed91f3SChris Lattner linkAppendingVarInit(AppendingVars[i]); 1475b1ed91f3SChris Lattner 1476b1ed91f3SChris Lattner // Link in the function bodies that are defined in the source module into 1477b1ed91f3SChris Lattner // DstM. 1478b1ed91f3SChris Lattner for (Module::iterator SF = SrcM->begin(), E = SrcM->end(); SF != E; ++SF) { 1479ea166d44STanya Lattner // Skip if not linking from source. 1480ea166d44STanya Lattner if (DoNotLinkFromSource.count(SF)) continue; 1481ea166d44STanya Lattner 14823fa50f9bSPeter Collingbourne Function *DF = cast<Function>(ValueMap[SF]); 14833fa50f9bSPeter Collingbourne if (SF->hasPrefixData()) { 14843fa50f9bSPeter Collingbourne // Link in the prefix data. 14853fa50f9bSPeter Collingbourne DF->setPrefixData(MapValue( 14863fa50f9bSPeter Collingbourne SF->getPrefixData(), ValueMap, RF_None, &TypeMap, &ValMaterializer)); 14873fa50f9bSPeter Collingbourne } 14883fa50f9bSPeter Collingbourne 1489ea166d44STanya Lattner // Skip if no body (function is external) or materialize. 1490ea166d44STanya Lattner if (SF->isDeclaration()) { 1491ea166d44STanya Lattner if (!SF->isMaterializable()) 1492ea166d44STanya Lattner continue; 1493ea166d44STanya Lattner if (SF->Materialize(&ErrorMsg)) 1494ea166d44STanya Lattner return true; 1495ea166d44STanya Lattner } 1496b1ed91f3SChris Lattner 14973fa50f9bSPeter Collingbourne linkFunctionBody(DF, SF); 149800623787SBill Wendling SF->Dematerialize(); 1499b1ed91f3SChris Lattner } 1500b1ed91f3SChris Lattner 1501b1ed91f3SChris Lattner // Resolve all uses of aliases with aliasees. 1502b1ed91f3SChris Lattner linkAliasBodies(); 1503b1ed91f3SChris Lattner 150466f02413SBill Wendling // Remap all of the named MDNodes in Src into the DstM module. We do this 15056ddbb2e2SDevang Patel // after linking GlobalValues so that MDNodes that reference GlobalValues 15066ddbb2e2SDevang Patel // are properly remapped. 15076ddbb2e2SDevang Patel linkNamedMDNodes(); 15086ddbb2e2SDevang Patel 150966f02413SBill Wendling // Merge the module flags into the DstM module. 151066f02413SBill Wendling if (linkModuleFlagsMetadata()) 151166f02413SBill Wendling return true; 151266f02413SBill Wendling 151391686d6dSBill Wendling // Update the initializers in the DstM module now that all globals that may 151491686d6dSBill Wendling // be referenced are in DstM. 151591686d6dSBill Wendling linkGlobalInits(); 151691686d6dSBill Wendling 15170a48b877STanya Lattner // Process vector of lazily linked in functions. 15180a48b877STanya Lattner bool LinkedInAnyFunctions; 15190a48b877STanya Lattner do { 15200a48b877STanya Lattner LinkedInAnyFunctions = false; 15210a48b877STanya Lattner 1522fa228782SBill Wendling for(std::vector<Function*>::iterator I = LazilyLinkFunctions.begin(), 15230a48b877STanya Lattner E = LazilyLinkFunctions.end(); I != E; ++I) { 1524f6f121e2SJames Molloy Function *SF = *I; 1525f6f121e2SJames Molloy if (!SF) 15260a48b877STanya Lattner continue; 15270a48b877STanya Lattner 1528fa228782SBill Wendling Function *DF = cast<Function>(ValueMap[SF]); 15293fa50f9bSPeter Collingbourne if (SF->hasPrefixData()) { 15303fa50f9bSPeter Collingbourne // Link in the prefix data. 15313fa50f9bSPeter Collingbourne DF->setPrefixData(MapValue(SF->getPrefixData(), 15323fa50f9bSPeter Collingbourne ValueMap, 15333fa50f9bSPeter Collingbourne RF_None, 15343fa50f9bSPeter Collingbourne &TypeMap, 15353fa50f9bSPeter Collingbourne &ValMaterializer)); 15363fa50f9bSPeter Collingbourne } 15370a48b877STanya Lattner 15380a48b877STanya Lattner // Materialize if necessary. 15390a48b877STanya Lattner if (SF->isDeclaration()) { 15400a48b877STanya Lattner if (!SF->isMaterializable()) 15410a48b877STanya Lattner continue; 15420a48b877STanya Lattner if (SF->Materialize(&ErrorMsg)) 15430a48b877STanya Lattner return true; 15440a48b877STanya Lattner } 15450a48b877STanya Lattner 1546f6f121e2SJames Molloy // Erase from vector *before* the function body is linked - linkFunctionBody could 1547f6f121e2SJames Molloy // invalidate I. 1548f6f121e2SJames Molloy LazilyLinkFunctions.erase(I); 1549f6f121e2SJames Molloy 15500a48b877STanya Lattner // Link in function body. 15510a48b877STanya Lattner linkFunctionBody(DF, SF); 155200623787SBill Wendling SF->Dematerialize(); 15530a48b877STanya Lattner 15540a48b877STanya Lattner // Set flag to indicate we may have more functions to lazily link in 15550a48b877STanya Lattner // since we linked in a function. 15560a48b877STanya Lattner LinkedInAnyFunctions = true; 1557f6f121e2SJames Molloy break; 15580a48b877STanya Lattner } 15590a48b877STanya Lattner } while (LinkedInAnyFunctions); 15600a48b877STanya Lattner 1561b1ed91f3SChris Lattner // Now that all of the types from the source are used, resolve any structs 1562b1ed91f3SChris Lattner // copied over to the dest that didn't exist there. 1563b1ed91f3SChris Lattner TypeMap.linkDefinedTypeBodies(); 15642609888fSAnton Korobeynikov 15652609888fSAnton Korobeynikov return false; 15662609888fSAnton Korobeynikov } 1567361e513dSReid Spencer 15687da92ed8SEli Bendersky Linker::Linker(Module *M, bool SuppressWarnings) 15697da92ed8SEli Bendersky : Composite(M), SuppressWarnings(SuppressWarnings) { 1570aa9918aaSRafael Espindola TypeFinder StructTypes; 1571aa9918aaSRafael Espindola StructTypes.run(*M, true); 1572aa9918aaSRafael Espindola IdentifiedStructTypes.insert(StructTypes.begin(), StructTypes.end()); 1573aa9918aaSRafael Espindola } 15743df61b7bSRafael Espindola 15753df61b7bSRafael Espindola Linker::~Linker() { 15763df61b7bSRafael Espindola } 15773df61b7bSRafael Espindola 157891e6f6e1SBill Wendling void Linker::deleteModule() { 157991e6f6e1SBill Wendling delete Composite; 15802617dcceSCraig Topper Composite = nullptr; 158191e6f6e1SBill Wendling } 158291e6f6e1SBill Wendling 15833df61b7bSRafael Espindola bool Linker::linkInModule(Module *Src, unsigned Mode, std::string *ErrorMsg) { 15847da92ed8SEli Bendersky ModuleLinker TheLinker(Composite, IdentifiedStructTypes, Src, Mode, 15857da92ed8SEli Bendersky SuppressWarnings); 1586287f18b4SRafael Espindola if (TheLinker.run()) { 1587287f18b4SRafael Espindola if (ErrorMsg) 1588287f18b4SRafael Espindola *ErrorMsg = TheLinker.ErrorMsg; 1589287f18b4SRafael Espindola return true; 1590287f18b4SRafael Espindola } 1591287f18b4SRafael Espindola return false; 15923df61b7bSRafael Espindola } 15933df61b7bSRafael Espindola 1594b1ed91f3SChris Lattner //===----------------------------------------------------------------------===// 1595b1ed91f3SChris Lattner // LinkModules entrypoint. 1596b1ed91f3SChris Lattner //===----------------------------------------------------------------------===// 1597b1ed91f3SChris Lattner 1598b6af2f36SBill Wendling /// LinkModules - This function links two modules together, with the resulting 1599970cc634SEli Bendersky /// Dest module modified to be the composite of the two input modules. If an 1600b6af2f36SBill Wendling /// error occurs, true is returned and ErrorMsg (if not null) is set to indicate 1601b6af2f36SBill Wendling /// the problem. Upon failure, the Dest module could be in a modified state, 1602b6af2f36SBill Wendling /// and shouldn't be relied on to be consistent. 1603cbb91408STanya Lattner bool Linker::LinkModules(Module *Dest, Module *Src, unsigned Mode, 1604cbb91408STanya Lattner std::string *ErrorMsg) { 1605287f18b4SRafael Espindola Linker L(Dest); 1606287f18b4SRafael Espindola return L.linkInModule(Src, Mode, ErrorMsg); 1607361e513dSReid Spencer } 1608a3aeb980SBill Wendling 1609a3aeb980SBill Wendling //===----------------------------------------------------------------------===// 1610a3aeb980SBill Wendling // C API. 1611a3aeb980SBill Wendling //===----------------------------------------------------------------------===// 1612a3aeb980SBill Wendling 1613a3aeb980SBill Wendling LLVMBool LLVMLinkModules(LLVMModuleRef Dest, LLVMModuleRef Src, 1614a3aeb980SBill Wendling LLVMLinkerMode Mode, char **OutMessages) { 1615a3aeb980SBill Wendling std::string Messages; 1616a3aeb980SBill Wendling LLVMBool Result = Linker::LinkModules(unwrap(Dest), unwrap(Src), 16172617dcceSCraig Topper Mode, OutMessages? &Messages : nullptr); 1618a3aeb980SBill Wendling if (OutMessages) 1619a3aeb980SBill Wendling *OutMessages = strdup(Messages.c_str()); 1620a3aeb980SBill Wendling return Result; 1621a3aeb980SBill Wendling } 1622