LinkModules.cpp revision 1f6efa3996dd1929fbc129203ce5009b620e6969
1c834bbf55832c45f90b1163f920472fefbf77bc1Mikhail Glushenkov//===- lib/Linker/LinkModules.cpp - Module Linker Implementation ----------===//
2f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman//
3b576c94c15af9a440f69d9d03c2afead7971118cJohn Criswell//                     The LLVM Compiler Infrastructure
4b576c94c15af9a440f69d9d03c2afead7971118cJohn Criswell//
54ee451de366474b9c228b4e5fa573795a715216dChris Lattner// This file is distributed under the University of Illinois Open Source
64ee451de366474b9c228b4e5fa573795a715216dChris Lattner// License. See LICENSE.TXT for details.
7f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman//
8b576c94c15af9a440f69d9d03c2afead7971118cJohn Criswell//===----------------------------------------------------------------------===//
952f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner//
1052f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// This file implements the LLVM module linker.
1152f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner//
1252f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// Specifically, this:
138d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner//  * Merges global variables between the two modules
148d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner//    * Uninit + Uninit = Init, Init + Uninit = Init, Init + Init = Error if !=
15c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner//  * Merges functions between two modules
1652f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner//
1752f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner//===----------------------------------------------------------------------===//
1852f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner
197cc371a7958c7a049c679a59023c3c8c83f6cc83Reid Spencer#include "llvm/Linker.h"
20adbc0b5287bf36893cdcae2440d48b3cb3489e38Chris Lattner#include "llvm/Constants.h"
21adbc0b5287bf36893cdcae2440d48b3cb3489e38Chris Lattner#include "llvm/DerivedTypes.h"
22c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson#include "llvm/LLVMContext.h"
235c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner#include "llvm/Module.h"
2478d033e086e19e016273de014f9214aa6f3f844bReid Spencer#include "llvm/TypeSymbolTable.h"
25ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer#include "llvm/ValueSymbolTable.h"
26933f9bdb00a8fe48833c4ea3ded20a6a90c405b0Bill Wendling#include "llvm/Instructions.h"
27933f9bdb00a8fe48833c4ea3ded20a6a90c405b0Bill Wendling#include "llvm/Assembly/Writer.h"
280fbf0e3d5e8f45925fc879b705ae0320f8bc76f8David Greene#include "llvm/Support/Debug.h"
29c25e7581b9b8088910da31702d4ca21c4734c6d7Torok Edwin#include "llvm/Support/ErrorHandling.h"
3074382b7c699120fbec5cb5603c9cf4212eb37f06Chris Lattner#include "llvm/Support/raw_ostream.h"
311f6efa3996dd1929fbc129203ce5009b620e6969Michael J. Spencer#include "llvm/Support/Path.h"
3205ea54e8869a81b8dd846397175f218f97968907Dan Gohman#include "llvm/Transforms/Utils/ValueMapper.h"
33933f9bdb00a8fe48833c4ea3ded20a6a90c405b0Bill Wendling#include "llvm/ADT/DenseMap.h"
34f7703df4968084c18c248c1feea9961c19a32e6aChris Lattnerusing namespace llvm;
35d0fde30ce850b78371fd1386338350591f9ff494Brian Gaeke
365c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// Error - Simple wrapper function to conditionally assign to E and return true.
375c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// This just makes error return conditions a little bit simpler...
386e0d1cb30957a636c53158d3089e6fb88348a57aDaniel Dunbarstatic inline bool Error(std::string *E, const Twine &Message) {
396e0d1cb30957a636c53158d3089e6fb88348a57aDaniel Dunbar  if (E) *E = Message.str();
405c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return true;
415c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
425c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
43700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Function: ResolveTypes()
44700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//
45700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Description:
46700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  Attempt to link the two specified types together.
47700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//
48700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Inputs:
49700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  DestTy - The type to which we wish to resolve.
50700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  SrcTy  - The original type which we want to resolve.
51700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//
52700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Outputs:
53700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  DestST - The symbol table in which the new type should be placed.
54700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//
55700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Return value:
56700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  true  - There is an error and the types cannot yet be linked.
57700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  false - No errors.
584c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner//
59bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattnerstatic bool ResolveTypes(const Type *DestTy, const Type *SrcTy) {
60e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  if (DestTy == SrcTy) return false;       // If already equal, noop
61bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner  assert(DestTy && SrcTy && "Can't handle null types");
62e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
63bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner  if (const OpaqueType *OT = dyn_cast<OpaqueType>(DestTy)) {
64bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner    // Type _is_ in module, just opaque...
65bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner    const_cast<OpaqueType*>(OT)->refineAbstractTypeTo(SrcTy);
66bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner  } else if (const OpaqueType *OT = dyn_cast<OpaqueType>(SrcTy)) {
67bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner    const_cast<OpaqueType*>(OT)->refineAbstractTypeTo(DestTy);
68bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner  } else {
69bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner    return true;  // Cannot link types... not-equal and neither is opaque.
704c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  }
714c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  return false;
724c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner}
734c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
7462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner/// LinkerTypeMap - This implements a map of types that is stable
7562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner/// even if types are resolved/refined to other types.  This is not a general
7662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner/// purpose map, it is specific to the linker's use.
7762a81a1eba019ab570b002f8e1686494139785a1Chris Lattnernamespace {
7862a81a1eba019ab570b002f8e1686494139785a1Chris Lattnerclass LinkerTypeMap : public AbstractTypeUser {
7962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  typedef DenseMap<const Type*, PATypeHolder> TheMapTy;
8062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  TheMapTy TheMap;
81eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
828c8b9ee8c8646aa1a79c782d4da9abdf3eb4eefeArgyrios Kyrtzidis  LinkerTypeMap(const LinkerTypeMap&); // DO NOT IMPLEMENT
838c8b9ee8c8646aa1a79c782d4da9abdf3eb4eefeArgyrios Kyrtzidis  void operator=(const LinkerTypeMap&); // DO NOT IMPLEMENT
84fc196f9ee909ac35837f6f771a42aa0dec817584Chris Lattnerpublic:
85fc196f9ee909ac35837f6f771a42aa0dec817584Chris Lattner  LinkerTypeMap() {}
86fc196f9ee909ac35837f6f771a42aa0dec817584Chris Lattner  ~LinkerTypeMap() {
8762a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    for (DenseMap<const Type*, PATypeHolder>::iterator I = TheMap.begin(),
8862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner         E = TheMap.end(); I != E; ++I)
8962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      I->first->removeAbstractTypeUser(this);
9062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  }
91eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
9262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// lookup - Return the value for the specified type or null if it doesn't
9362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// exist.
9462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  const Type *lookup(const Type *Ty) const {
9562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    TheMapTy::const_iterator I = TheMap.find(Ty);
9662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    if (I != TheMap.end()) return I->second;
9762a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    return 0;
9862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  }
99eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
10062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// insert - This returns true if the pointer was new to the set, false if it
10162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// was already in the set.
10262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  bool insert(const Type *Src, const Type *Dst) {
1036b345ee9b2833cf1b2f79dc16d06d4060bec36efDan Gohman    if (!TheMap.insert(std::make_pair(Src, PATypeHolder(Dst))).second)
10462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      return false;  // Already in map.
10562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    if (Src->isAbstract())
10662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      Src->addAbstractTypeUser(this);
10762a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    return true;
10862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  }
109eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
11062a81a1eba019ab570b002f8e1686494139785a1Chris Lattnerprotected:
11162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// refineAbstractType - The callback method invoked when an abstract type is
11262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// resolved to another type.  An object must override this method to update
11362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// its internal state to reference NewType instead of OldType.
11462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  ///
11562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  virtual void refineAbstractType(const DerivedType *OldTy,
11662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner                                  const Type *NewTy) {
11762a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    TheMapTy::iterator I = TheMap.find(OldTy);
11862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    const Type *DstTy = I->second;
119eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
12062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    TheMap.erase(I);
12162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    if (OldTy->isAbstract())
12262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      OldTy->removeAbstractTypeUser(this);
12362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner
12462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    // Don't reinsert into the map if the key is concrete now.
12562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    if (NewTy->isAbstract())
12662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      insert(NewTy, DstTy);
12762a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  }
128eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
12962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// The other case which AbstractTypeUsers must be aware of is when a type
13062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// makes the transition from being abstract (where it has clients on it's
13162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// AbstractTypeUsers list) to concrete (where it does not).  This method
13262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// notifies ATU's when this occurs for a type.
13362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  virtual void typeBecameConcrete(const DerivedType *AbsTy) {
13462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    TheMap.erase(AbsTy);
13562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    AbsTy->removeAbstractTypeUser(this);
13662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  }
137eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
13862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  // for debugging...
13962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  virtual void dump() const {
1400fbf0e3d5e8f45925fc879b705ae0320f8bc76f8David Greene    dbgs() << "AbstractTypeSet!\n";
14162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  }
14262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner};
14362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner}
14462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner
14562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner
146e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner// RecursiveResolveTypes - This is just like ResolveTypes, except that it
147e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner// recurses down into derived types, merging the used types if the parent types
148e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner// are compatible.
149a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattnerstatic bool RecursiveResolveTypesI(const Type *DstTy, const Type *SrcTy,
15062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner                                   LinkerTypeMap &Pointers) {
151a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner  if (DstTy == SrcTy) return false;       // If already equal, noop
152f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
153e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  // If we found our opaque type, resolve it now!
15447c5188789bc40671504ed1fa3a44765cefba44fDuncan Sands  if (DstTy->isOpaqueTy() || SrcTy->isOpaqueTy())
155a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    return ResolveTypes(DstTy, SrcTy);
156f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
157e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  // Two types cannot be resolved together if they are of different primitive
158e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  // type.  For example, we cannot resolve an int to a float.
159a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner  if (DstTy->getTypeID() != SrcTy->getTypeID()) return true;
160e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
16156539659eb1b192d493aa333b60213889129b9f1Chris Lattner  // If neither type is abstract, then they really are just different types.
162a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner  if (!DstTy->isAbstract() && !SrcTy->isAbstract())
16356539659eb1b192d493aa333b60213889129b9f1Chris Lattner    return true;
164eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
165e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  // Otherwise, resolve the used type used by this derived type...
166a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner  switch (DstTy->getTypeID()) {
167f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner  default:
168f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner    return true;
169e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  case Type::FunctionTyID: {
170a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const FunctionType *DstFT = cast<FunctionType>(DstTy);
171a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const FunctionType *SrcFT = cast<FunctionType>(SrcTy);
1729ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner    if (DstFT->isVarArg() != SrcFT->isVarArg() ||
1739ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner        DstFT->getNumContainedTypes() != SrcFT->getNumContainedTypes())
17443f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner      return true;
175eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
176a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    // Use TypeHolder's so recursive resolution won't break us.
177a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    PATypeHolder ST(SrcFT), DT(DstFT);
178a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    for (unsigned i = 0, e = DstFT->getNumContainedTypes(); i != e; ++i) {
179a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner      const Type *SE = ST->getContainedType(i), *DE = DT->getContainedType(i);
180a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner      if (SE != DE && RecursiveResolveTypesI(DE, SE, Pointers))
181e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner        return true;
182a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    }
183e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner    return false;
184e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  }
185e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  case Type::StructTyID: {
186a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const StructType *DstST = cast<StructType>(DstTy);
187a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const StructType *SrcST = cast<StructType>(SrcTy);
1889ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner    if (DstST->getNumContainedTypes() != SrcST->getNumContainedTypes())
189f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner      return true;
190eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
191a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    PATypeHolder ST(SrcST), DT(DstST);
192a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    for (unsigned i = 0, e = DstST->getNumContainedTypes(); i != e; ++i) {
193a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner      const Type *SE = ST->getContainedType(i), *DE = DT->getContainedType(i);
194a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner      if (SE != DE && RecursiveResolveTypesI(DE, SE, Pointers))
195e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner        return true;
196a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    }
197e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner    return false;
198e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  }
199e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  case Type::ArrayTyID: {
200a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const ArrayType *DAT = cast<ArrayType>(DstTy);
201a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const ArrayType *SAT = cast<ArrayType>(SrcTy);
202e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner    if (DAT->getNumElements() != SAT->getNumElements()) return true;
203e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    return RecursiveResolveTypesI(DAT->getElementType(), SAT->getElementType(),
204bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner                                  Pointers);
205e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner  }
206f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner  case Type::VectorTyID: {
207a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const VectorType *DVT = cast<VectorType>(DstTy);
208a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const VectorType *SVT = cast<VectorType>(SrcTy);
209f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner    if (DVT->getNumElements() != SVT->getNumElements()) return true;
210f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner    return RecursiveResolveTypesI(DVT->getElementType(), SVT->getElementType(),
211f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner                                  Pointers);
212f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner  }
213e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner  case Type::PointerTyID: {
214a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const PointerType *DstPT = cast<PointerType>(DstTy);
215a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const PointerType *SrcPT = cast<PointerType>(SrcTy);
216eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
2179ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner    if (DstPT->getAddressSpace() != SrcPT->getAddressSpace())
2189ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner      return true;
219eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
220e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // If this is a pointer type, check to see if we have already seen it.  If
221e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // so, we are in a recursive branch.  Cut off the search now.  We cannot use
222e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // an associative container for this search, because the type pointers (keys
22362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    // in the container) change whenever types get resolved.
22462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    if (SrcPT->isAbstract())
22562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      if (const Type *ExistingDestTy = Pointers.lookup(SrcPT))
22662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner        return ExistingDestTy != DstPT;
227eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
22862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    if (DstPT->isAbstract())
22962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      if (const Type *ExistingSrcTy = Pointers.lookup(DstPT))
23062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner        return ExistingSrcTy != SrcPT;
231e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // Otherwise, add the current pointers to the vector to stop recursion on
232e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // this pair.
23362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    if (DstPT->isAbstract())
23462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      Pointers.insert(DstPT, SrcPT);
23562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    if (SrcPT->isAbstract())
23662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      Pointers.insert(SrcPT, DstPT);
237eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
2389ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner    return RecursiveResolveTypesI(DstPT->getElementType(),
2399ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner                                  SrcPT->getElementType(), Pointers);
240e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  }
241f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman  }
242e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner}
243e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
244a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattnerstatic bool RecursiveResolveTypes(const Type *DestTy, const Type *SrcTy) {
24562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  LinkerTypeMap PointerTypes;
246bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner  return RecursiveResolveTypesI(DestTy, SrcTy, PointerTypes);
247e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner}
248e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner
249e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
2502c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner// LinkTypes - Go through the symbol table of the Src module and see if any
2512c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner// types are named in the src module that are not named in the Dst module.
2522c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner// Make sure there are no type name conflicts.
2535c2d335d86579941b71816196a95a64b277e8963Chris Lattnerstatic bool LinkTypes(Module *Dest, const Module *Src, std::string *Err) {
25478d033e086e19e016273de014f9214aa6f3f844bReid Spencer        TypeSymbolTable *DestST = &Dest->getTypeSymbolTable();
25578d033e086e19e016273de014f9214aa6f3f844bReid Spencer  const TypeSymbolTable *SrcST  = &Src->getTypeSymbolTable();
2562c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
2572c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  // Look for a type plane for Type's...
25878d033e086e19e016273de014f9214aa6f3f844bReid Spencer  TypeSymbolTable::const_iterator TI = SrcST->begin();
25978d033e086e19e016273de014f9214aa6f3f844bReid Spencer  TypeSymbolTable::const_iterator TE = SrcST->end();
260567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer  if (TI == TE) return false;  // No named types, do nothing.
2612c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
262cf00c4ab3ba308d45d98c5ccab87362cf802facbMisha Brukman  // Some types cannot be resolved immediately because they depend on other
263cf00c4ab3ba308d45d98c5ccab87362cf802facbMisha Brukman  // types being resolved to each other first.  This contains a list of types we
264cf00c4ab3ba308d45d98c5ccab87362cf802facbMisha Brukman  // are waiting to recheck.
2654c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  std::vector<std::string> DelayedTypesToResolve;
2664c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
267567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer  for ( ; TI != TE; ++TI ) {
268567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer    const std::string &Name = TI->first;
269c28a224c68adf0464c275778af4bce762ad25d37Reid Spencer    const Type *RHS = TI->second;
2702c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
271bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner    // Check to see if this type name is already in the dest module.
27278d033e086e19e016273de014f9214aa6f3f844bReid Spencer    Type *Entry = DestST->lookup(Name);
2734c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
274bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner    // If the name is just in the source module, bring it over to the dest.
275bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner    if (Entry == 0) {
276bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner      if (!Name.empty())
277bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner        DestST->insert(Name, const_cast<Type*>(RHS));
278bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner    } else if (ResolveTypes(Entry, RHS)) {
2794c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      // They look different, save the types 'till later to resolve.
2804c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      DelayedTypesToResolve.push_back(Name);
2814c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    }
2824c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  }
2834c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
2844c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  // Iteratively resolve types while we can...
2854c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  while (!DelayedTypesToResolve.empty()) {
2864c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    // Loop over all of the types, attempting to resolve them if possible...
2874c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    unsigned OldSize = DelayedTypesToResolve.size();
2884c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
289e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner    // Try direct resolution by name...
2904c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    for (unsigned i = 0; i != DelayedTypesToResolve.size(); ++i) {
2914c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      const std::string &Name = DelayedTypesToResolve[i];
29278d033e086e19e016273de014f9214aa6f3f844bReid Spencer      Type *T1 = SrcST->lookup(Name);
29378d033e086e19e016273de014f9214aa6f3f844bReid Spencer      Type *T2 = DestST->lookup(Name);
294bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner      if (!ResolveTypes(T2, T1)) {
2954c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner        // We are making progress!
2964c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner        DelayedTypesToResolve.erase(DelayedTypesToResolve.begin()+i);
2974c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner        --i;
2982f6bb2bce13084dfc4c8acf47f9eae0578267aaaChris Lattner      }
2994c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    }
3002f6bb2bce13084dfc4c8acf47f9eae0578267aaaChris Lattner
3014c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    // Did we not eliminate any types?
3024c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    if (DelayedTypesToResolve.size() == OldSize) {
303e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      // Attempt to resolve subelements of types.  This allows us to merge these
304e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      // two types: { int* } and { opaque* }
3054c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      for (unsigned i = 0, e = DelayedTypesToResolve.size(); i != e; ++i) {
3064c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner        const std::string &Name = DelayedTypesToResolve[i];
307a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner        if (!RecursiveResolveTypes(SrcST->lookup(Name), DestST->lookup(Name))) {
308e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          // We are making progress!
309e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          DelayedTypesToResolve.erase(DelayedTypesToResolve.begin()+i);
310f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
311e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          // Go back to the main loop, perhaps we can resolve directly by name
312e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          // now...
313e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          break;
314e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner        }
315e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      }
316e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
317e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      // If we STILL cannot resolve the types, then there is something wrong.
318e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      if (DelayedTypesToResolve.size() == OldSize) {
319aeb18ce9c012f8554f48bae9bb32351c6eeaa26cChris Lattner        // Remove the symbol name from the destination.
320aeb18ce9c012f8554f48bae9bb32351c6eeaa26cChris Lattner        DelayedTypesToResolve.pop_back();
3214c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      }
3222c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner    }
3232c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  }
3244c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
3254c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
3262c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  return false;
3272c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner}
3282c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
3298bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer/// ForceRenaming - The LLVM SymbolTable class autorenames globals that conflict
3308bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer/// in the symbol table.  This is good for all clients except for us.  Go
3318bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer/// through the trouble to force this back.
332c003628a612d3687fb77088a5894314210a65385Chris Lattnerstatic void ForceRenaming(GlobalValue *GV, const std::string &Name) {
333c003628a612d3687fb77088a5894314210a65385Chris Lattner  assert(GV->getName() != Name && "Can't force rename to self");
334ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  ValueSymbolTable &ST = GV->getParent()->getValueSymbolTable();
335c003628a612d3687fb77088a5894314210a65385Chris Lattner
336c003628a612d3687fb77088a5894314210a65385Chris Lattner  // If there is a conflict, rename the conflict.
33733f294930e0affa51c3818405741abd80cc90316Chris Lattner  if (GlobalValue *ConflictGV = cast_or_null<GlobalValue>(ST.lookup(Name))) {
338bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola    assert(ConflictGV->hasLocalLinkage() &&
339ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer           "Not conflicting with a static global, should link instead!");
34033f294930e0affa51c3818405741abd80cc90316Chris Lattner    GV->takeName(ConflictGV);
34133f294930e0affa51c3818405741abd80cc90316Chris Lattner    ConflictGV->setName(Name);    // This will cause ConflictGV to get renamed
342ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer    assert(ConflictGV->getName() != Name && "ForceRenaming didn't work");
34333f294930e0affa51c3818405741abd80cc90316Chris Lattner  } else {
34433f294930e0affa51c3818405741abd80cc90316Chris Lattner    GV->setName(Name);              // Force the name back
345ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  }
346ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer}
347ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer
348ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer/// CopyGVAttributes - copy additional attributes (those not needed to construct
349eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov/// a GlobalValue) from the SrcGV to the DestGV.
350ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencerstatic void CopyGVAttributes(GlobalValue *DestGV, const GlobalValue *SrcGV) {
35128c3cff8250b3fe2adc6479306fe7dbdb48a1bdbDuncan Sands  // Use the maximum alignment, rather than just copying the alignment of SrcGV.
35228c3cff8250b3fe2adc6479306fe7dbdb48a1bdbDuncan Sands  unsigned Alignment = std::max(DestGV->getAlignment(), SrcGV->getAlignment());
35328c3cff8250b3fe2adc6479306fe7dbdb48a1bdbDuncan Sands  DestGV->copyAttributesFrom(SrcGV);
35428c3cff8250b3fe2adc6479306fe7dbdb48a1bdbDuncan Sands  DestGV->setAlignment(Alignment);
355c003628a612d3687fb77088a5894314210a65385Chris Lattner}
356c003628a612d3687fb77088a5894314210a65385Chris Lattner
357aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// GetLinkageResult - This analyzes the two global values and determines what
358aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// the result will look like in the destination module.  In particular, it
359aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// computes the resultant linkage type, computes whether the global in the
360aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// source should be copied over to the destination (replacing the existing
3619cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov/// one), and computes whether this linkage is an error or not. It also performs
3629cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov/// visibility checks: we cannot link together two symbols with different
3639cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov/// visibilities.
364968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikovstatic bool GetLinkageResult(GlobalValue *Dest, const GlobalValue *Src,
365aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner                             GlobalValue::LinkageTypes &LT, bool &LinkFromSrc,
366aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner                             std::string *Err) {
367bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola  assert((!Dest || !Src->hasLocalLinkage()) &&
368aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner         "If Src has internal linkage, Dest shouldn't be set!");
369aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  if (!Dest) {
370aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    // Linking something to nothing.
371aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LinkFromSrc = true;
372aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LT = Src->getLinkage();
3735cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer  } else if (Src->isDeclaration()) {
3742b48ef0450944c2c46633aec9baf6be835a3b503Anton Korobeynikov    // If Src is external or if both Src & Dest are external..  Just link the
375aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    // external globals, we aren't adding anything.
376b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov    if (Src->hasDLLImportLinkage()) {
37778ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      // If one of GVs has DLLImport linkage, result should be dllimport'ed.
3785cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer      if (Dest->isDeclaration()) {
379b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov        LinkFromSrc = true;
380b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov        LT = Src->getLinkage();
381eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov      }
3828753c447a5f8eec1421331c40ffbb72c6d411f78Andrew Lenharth    } else if (Dest->hasExternalWeakLinkage()) {
383667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands      // If the Dest is weak, use the source linkage.
3848753c447a5f8eec1421331c40ffbb72c6d411f78Andrew Lenharth      LinkFromSrc = true;
3858753c447a5f8eec1421331c40ffbb72c6d411f78Andrew Lenharth      LT = Src->getLinkage();
386b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov    } else {
387b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov      LinkFromSrc = false;
388b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov      LT = Dest->getLinkage();
389b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov    }
3905cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer  } else if (Dest->isDeclaration() && !Dest->hasDLLImportLinkage()) {
391aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    // If Dest is external but Src is not:
392aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LinkFromSrc = true;
393aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LT = Src->getLinkage();
394aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  } else if (Src->hasAppendingLinkage() || Dest->hasAppendingLinkage()) {
395aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    if (Src->getLinkage() != Dest->getLinkage())
396aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      return Error(Err, "Linking globals named '" + Src->getName() +
397aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner            "': can only link appending global with another appending global!");
398aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LinkFromSrc = true; // Special cased.
399aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LT = Src->getLinkage();
400a05ef5e107943478ad02bc0cd5170a3894076bc4Duncan Sands  } else if (Src->isWeakForLinker()) {
401aafce77b17d340aace52bcd49d1944109d82f14aDale Johannesen    // At this point we know that Dest has LinkOnce, External*, Weak, Common,
402aafce77b17d340aace52bcd49d1944109d82f14aDale Johannesen    // or DLL* linkage.
403266c7bbbbcc4b326dea82e577de1a415d6acc23eChris Lattner    if (Dest->hasExternalWeakLinkage() ||
404266c7bbbbcc4b326dea82e577de1a415d6acc23eChris Lattner        Dest->hasAvailableExternallyLinkage() ||
405266c7bbbbcc4b326dea82e577de1a415d6acc23eChris Lattner        (Dest->hasLinkOnceLinkage() &&
406266c7bbbbcc4b326dea82e577de1a415d6acc23eChris Lattner         (Src->hasWeakLinkage() || Src->hasCommonLinkage()))) {
407aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      LinkFromSrc = true;
408aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      LT = Src->getLinkage();
409aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    } else {
410aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      LinkFromSrc = false;
411aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      LT = Dest->getLinkage();
412aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    }
413a05ef5e107943478ad02bc0cd5170a3894076bc4Duncan Sands  } else if (Dest->isWeakForLinker()) {
41478ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov    // At this point we know that Src has External* or DLL* linkage.
41578ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov    if (Src->hasExternalWeakLinkage()) {
41678ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      LinkFromSrc = false;
41778ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      LT = Dest->getLinkage();
41878ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov    } else {
41978ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      LinkFromSrc = true;
42078ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      LT = GlobalValue::ExternalLinkage;
42178ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov    }
422aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  } else {
423b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov    assert((Dest->hasExternalLinkage() ||
424b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov            Dest->hasDLLImportLinkage() ||
42578ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov            Dest->hasDLLExportLinkage() ||
42678ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov            Dest->hasExternalWeakLinkage()) &&
427b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov           (Src->hasExternalLinkage() ||
428b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov            Src->hasDLLImportLinkage() ||
42978ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov            Src->hasDLLExportLinkage() ||
43078ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov            Src->hasExternalWeakLinkage()) &&
431aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner           "Unexpected linkage type!");
432f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman    return Error(Err, "Linking globals named '" + Src->getName() +
433aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner                 "': symbol multiply defined!");
434aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  }
4359cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov
4369cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov  // Check visibility
4379cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov  if (Dest && Src->getVisibility() != Dest->getVisibility())
43897f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner    if (!Src->isDeclaration() && !Dest->isDeclaration())
43997f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner      return Error(Err, "Linking globals named '" + Src->getName() +
44097f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner                   "': symbols have different visibilities!");
441aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  return false;
442aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner}
4435c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
444ab67e705f59d567afded845465f358b8a66ab62eDevang Patel// Insert all of the named mdnoes in Src into the Dest module.
445e5835fbe7fb3d87b8cd110b24f0f9772384a5212Dan Gohmanstatic void LinkNamedMDNodes(Module *Dest, Module *Src,
446e5835fbe7fb3d87b8cd110b24f0f9772384a5212Dan Gohman                             ValueToValueMapTy &ValueMap) {
447ab67e705f59d567afded845465f358b8a66ab62eDevang Patel  for (Module::const_named_metadata_iterator I = Src->named_metadata_begin(),
448ab67e705f59d567afded845465f358b8a66ab62eDevang Patel         E = Src->named_metadata_end(); I != E; ++I) {
449ab67e705f59d567afded845465f358b8a66ab62eDevang Patel    const NamedMDNode *SrcNMD = I;
45017aa92c92a925b4a674440c7ef088c223990e854Dan Gohman    NamedMDNode *DestNMD = Dest->getOrInsertNamedMetadata(SrcNMD->getName());
45117aa92c92a925b4a674440c7ef088c223990e854Dan Gohman    // Add Src elements into Dest node.
45217aa92c92a925b4a674440c7ef088c223990e854Dan Gohman    for (unsigned i = 0, e = SrcNMD->getNumOperands(); i != e; ++i)
453e5835fbe7fb3d87b8cd110b24f0f9772384a5212Dan Gohman      DestNMD->addOperand(cast<MDNode>(MapValue(SrcNMD->getOperand(i),
4546cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman                                                ValueMap,
4556cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman                                                true)));
456ab67e705f59d567afded845465f358b8a66ab62eDevang Patel  }
457ab67e705f59d567afded845465f358b8a66ab62eDevang Patel}
458ab67e705f59d567afded845465f358b8a66ab62eDevang Patel
4595c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// LinkGlobals - Loop through the global variables in the src module and merge
4608166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner// them into the dest module.
461968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikovstatic bool LinkGlobals(Module *Dest, const Module *Src,
46205ea54e8869a81b8dd846397175f218f97968907Dan Gohman                        ValueToValueMapTy &ValueMap,
4638166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                    std::multimap<std::string, GlobalVariable *> &AppendingVars,
4645c2d335d86579941b71816196a95a64b277e8963Chris Lattner                        std::string *Err) {
465d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner  ValueSymbolTable &DestSymTab = Dest->getValueSymbolTable();
466eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
4675c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  // Loop over all of the globals in the src module, mapping them over as we go
4680bb8757997392ee18ca9de0f1fab29463f8e7aebChris Lattner  for (Module::const_global_iterator I = Src->global_begin(),
4690bb8757997392ee18ca9de0f1fab29463f8e7aebChris Lattner       E = Src->global_end(); I != E; ++I) {
470968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov    const GlobalVariable *SGV = I;
47101f69399352380532d480f6749daed660f50b15cAnton Korobeynikov    GlobalValue *DGV = 0;
47201f69399352380532d480f6749daed660f50b15cAnton Korobeynikov
473933f9bdb00a8fe48833c4ea3ded20a6a90c405b0Bill Wendling    // Check to see if may have to link the global with the global, alias or
474d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner    // function.
475bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola    if (SGV->hasName() && !SGV->hasLocalLinkage())
476933f9bdb00a8fe48833c4ea3ded20a6a90c405b0Bill Wendling      DGV = cast_or_null<GlobalValue>(DestSymTab.lookup(SGV->getName()));
477eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
478ae1132d2b8ae07afd2fe9a7cb434d849f884bfa0Chris Lattner    // If we found a global with the same name in the dest module, but it has
479ae1132d2b8ae07afd2fe9a7cb434d849f884bfa0Chris Lattner    // internal linkage, we are really not doing any linkage here.
480bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola    if (DGV && DGV->hasLocalLinkage())
481ae1132d2b8ae07afd2fe9a7cb434d849f884bfa0Chris Lattner      DGV = 0;
482eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
483d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner    // If types don't agree due to opaque types, try to resolve them.
484d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner    if (DGV && DGV->getType() != SGV->getType())
485d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner      RecursiveResolveTypes(SGV->getType(), DGV->getType());
48601f69399352380532d480f6749daed660f50b15cAnton Korobeynikov
487c318329a1098a19137398af4d880f4d8cdd9cd8dDan Gohman    assert((SGV->hasInitializer() || SGV->hasExternalWeakLinkage() ||
488c318329a1098a19137398af4d880f4d8cdd9cd8dDan Gohman            SGV->hasExternalLinkage() || SGV->hasDLLImportLinkage()) &&
4894ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner           "Global must either be external or have an initializer!");
4905c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
491b324bd735ff5c99a52cd3c3f5d01c0e1398a2d3aChris Lattner    GlobalValue::LinkageTypes NewLinkage = GlobalValue::InternalLinkage;
492b324bd735ff5c99a52cd3c3f5d01c0e1398a2d3aChris Lattner    bool LinkFromSrc = false;
493aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    if (GetLinkageResult(DGV, SGV, NewLinkage, LinkFromSrc, Err))
494aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      return true;
4950fec08eb58dd9fffeb72c584aa61a59d71111c8dChris Lattner
4966157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    if (DGV == 0) {
4975c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner      // No linking to be performed, simply create an identical version of the
4988d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      // symbol over in the dest module... the initializer will be filled in
499d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner      // later by LinkGlobalInits.
5002719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      GlobalVariable *NewDGV =
501e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson        new GlobalVariable(*Dest, SGV->getType()->getElementType(),
5022719bac90525985f4ebb97324b4c087d47b54384Chris Lattner                           SGV->isConstant(), SGV->getLinkage(), /*init*/0,
503e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson                           SGV->getName(), 0, false,
504a534b0f7bf0248bf90afe97110a34777e972f325Chris Lattner                           SGV->getType()->getAddressSpace());
505471feac1b297b9458de18d973af371bf7f10296bReid Spencer      // Propagate alignment, visibility and section info.
506ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      CopyGVAttributes(NewDGV, SGV);
5075dfbaf1fe3fa6914095cc55cde20b1ed25187b56Andrew Lenharth
5082719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      // If the LLVM runtime renamed the global, but it is an externally visible
5092719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      // symbol, DGV must be an existing global with internal linkage.  Rename
5102719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      // it.
511bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola      if (!NewDGV->hasLocalLinkage() && NewDGV->getName() != SGV->getName())
512c003628a612d3687fb77088a5894314210a65385Chris Lattner        ForceRenaming(NewDGV, SGV->getName());
5135c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
5146157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      // Make sure to remember this mapping.
515817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov      ValueMap[SGV] = NewDGV;
516817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov
517d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner      // Keep track that this is an appending variable.
5188166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      if (SGV->hasAppendingLinkage())
5198166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        AppendingVars.insert(std::make_pair(SGV->getName(), NewDGV));
5206157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      continue;
5216157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    }
522eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
5236157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    // If the visibilities of the symbols disagree and the destination is a
5246157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    // prototype, take the visibility of its input.
5256157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    if (DGV->isDeclaration())
5266157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      DGV->setVisibility(SGV->getVisibility());
527eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
5286157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    if (DGV->hasAppendingLinkage()) {
5298166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // No linking is performed yet.  Just insert a new copy of the global, and
5308166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // keep track of the fact that it is an appending variable in the
5318166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // AppendingVars map.  The name is cleared out so that no linkage is
5328166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // performed.
5338166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      GlobalVariable *NewDGV =
534e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson        new GlobalVariable(*Dest, SGV->getType()->getElementType(),
5358166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                           SGV->isConstant(), SGV->getLinkage(), /*init*/0,
536e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson                           "", 0, false,
537a534b0f7bf0248bf90afe97110a34777e972f325Chris Lattner                           SGV->getType()->getAddressSpace());
5388166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
53975c7915c3a24f8063c274ccffe2dc9b7bc8bce1fAnton Korobeynikov      // Set alignment allowing CopyGVAttributes merge it with alignment of SGV.
540ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      NewDGV->setAlignment(DGV->getAlignment());
54175c7915c3a24f8063c274ccffe2dc9b7bc8bce1fAnton Korobeynikov      // Propagate alignment, section and visibility info.
542ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      CopyGVAttributes(NewDGV, SGV);
5435dfbaf1fe3fa6914095cc55cde20b1ed25187b56Andrew Lenharth
5448166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Make sure to remember this mapping...
545817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov      ValueMap[SGV] = NewDGV;
5468166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
5478166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Keep track that this is an appending variable...
5488166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      AppendingVars.insert(std::make_pair(SGV->getName(), NewDGV));
5496157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      continue;
5506157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    }
551eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
5526157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    if (LinkFromSrc) {
5536157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      if (isa<GlobalAlias>(DGV))
5541438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov        return Error(Err, "Global-Alias Collision on '" + SGV->getName() +
5551438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov                     "': symbol multiple defined");
556eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
557d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner      // If the types don't match, and if we are to link from the source, nuke
558d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner      // DGV and create a new one of the appropriate type.  Note that the thing
559d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner      // we are replacing may be a function (if a prototype, weak, etc) or a
560d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner      // global variable.
561d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner      GlobalVariable *NewDGV =
562e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson        new GlobalVariable(*Dest, SGV->getType()->getElementType(),
5633d29df3e8a203b167d8071ea6f805b21db18a5afOwen Anderson                           SGV->isConstant(), NewLinkage, /*init*/0,
564e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson                           DGV->getName(), 0, false,
565d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner                           SGV->getType()->getAddressSpace());
566eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
567d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner      // Propagate alignment, section, and visibility info.
568d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner      CopyGVAttributes(NewDGV, SGV);
569baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson      DGV->replaceAllUsesWith(ConstantExpr::getBitCast(NewDGV,
570c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson                                                              DGV->getType()));
571eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
572d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner      // DGV will conflict with NewDGV because they both had the same
573d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner      // name. We must erase this now so ForceRenaming doesn't assert
574d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner      // because DGV might not have internal linkage.
575d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner      if (GlobalVariable *Var = dyn_cast<GlobalVariable>(DGV))
576d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner        Var->eraseFromParent();
577d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner      else
578d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner        cast<Function>(DGV)->eraseFromParent();
579d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner
580d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner      // If the symbol table renamed the global, but it is an externally visible
581d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner      // symbol, DGV must be an existing global with internal linkage.  Rename.
582bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola      if (NewDGV->getName() != SGV->getName() && !NewDGV->hasLocalLinkage())
583d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner        ForceRenaming(NewDGV, SGV->getName());
584eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
5856157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      // Inherit const as appropriate.
586d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner      NewDGV->setConstant(SGV->isConstant());
587eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
5886157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      // Make sure to remember this mapping.
5896157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      ValueMap[SGV] = NewDGV;
5906157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      continue;
5915c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner    }
592eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
5936157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    // Not "link from source", keep the one in the DestModule and remap the
5946157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    // input onto it.
595eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
5966157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    // Special case for const propagation.
5976157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    if (GlobalVariable *DGVar = dyn_cast<GlobalVariable>(DGV))
5986157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      if (DGVar->isDeclaration() && SGV->isConstant() && !DGVar->isConstant())
5996157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner        DGVar->setConstant(true);
6006157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner
601d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov    // SGV is global, but DGV is alias.
602d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov    if (isa<GlobalAlias>(DGV)) {
603d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov      // The only valid mappings are:
604d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov      // - SGV is external declaration, which is effectively a no-op.
605d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov      // - SGV is weak, when we just need to throw SGV out.
606a05ef5e107943478ad02bc0cd5170a3894076bc4Duncan Sands      if (!SGV->isDeclaration() && !SGV->isWeakForLinker())
607d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov        return Error(Err, "Global-Alias Collision on '" + SGV->getName() +
608d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov                     "': symbol multiple defined");
609d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov    }
610eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
6116157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    // Set calculated linkage
6126157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    DGV->setLinkage(NewLinkage);
613eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
6146157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    // Make sure to remember this mapping...
615baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson    ValueMap[SGV] = ConstantExpr::getBitCast(DGV, SGV->getType());
6165c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  }
6175c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return false;
6185c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
6195c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
62058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikovstatic GlobalValue::LinkageTypes
62158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton KorobeynikovCalculateAliasLinkage(const GlobalValue *SGV, const GlobalValue *DGV) {
622667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands  GlobalValue::LinkageTypes SL = SGV->getLinkage();
623667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands  GlobalValue::LinkageTypes DL = DGV->getLinkage();
624667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands  if (SL == GlobalValue::ExternalLinkage || DL == GlobalValue::ExternalLinkage)
62558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    return GlobalValue::ExternalLinkage;
626667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands  else if (SL == GlobalValue::WeakAnyLinkage ||
627667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands           DL == GlobalValue::WeakAnyLinkage)
628667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands    return GlobalValue::WeakAnyLinkage;
629667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands  else if (SL == GlobalValue::WeakODRLinkage ||
630667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands           DL == GlobalValue::WeakODRLinkage)
631667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands    return GlobalValue::WeakODRLinkage;
632667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands  else if (SL == GlobalValue::InternalLinkage &&
633667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands           DL == GlobalValue::InternalLinkage)
63458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    return GlobalValue::InternalLinkage;
6353d10a5a75794356a0a568ce283713adc3a963200Bill Wendling  else if (SL == GlobalValue::LinkerPrivateLinkage &&
6363d10a5a75794356a0a568ce283713adc3a963200Bill Wendling           DL == GlobalValue::LinkerPrivateLinkage)
6373d10a5a75794356a0a568ce283713adc3a963200Bill Wendling    return GlobalValue::LinkerPrivateLinkage;
6384e34d502727df36cc2caa59307953444b1ee9914Bill Wendling  else if (SL == GlobalValue::LinkerPrivateWeakLinkage &&
6394e34d502727df36cc2caa59307953444b1ee9914Bill Wendling           DL == GlobalValue::LinkerPrivateWeakLinkage)
6404e34d502727df36cc2caa59307953444b1ee9914Bill Wendling    return GlobalValue::LinkerPrivateWeakLinkage;
6414e34d502727df36cc2caa59307953444b1ee9914Bill Wendling  else if (SL == GlobalValue::LinkerPrivateWeakDefAutoLinkage &&
6424e34d502727df36cc2caa59307953444b1ee9914Bill Wendling           DL == GlobalValue::LinkerPrivateWeakDefAutoLinkage)
6434e34d502727df36cc2caa59307953444b1ee9914Bill Wendling    return GlobalValue::LinkerPrivateWeakDefAutoLinkage;
644bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola  else {
645667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands    assert (SL == GlobalValue::PrivateLinkage &&
646667d4b8de6dea70195ff12ef39a4deebffa2f5c7Duncan Sands            DL == GlobalValue::PrivateLinkage && "Unexpected linkage type");
647bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola    return GlobalValue::PrivateLinkage;
64858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov  }
64958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov}
65058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
65131ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio// LinkAlias - Loop through the alias in the src module and link them into the
65258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov// dest module. We're assuming, that all functions/global variables were already
65358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov// linked in.
6544fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikovstatic bool LinkAlias(Module *Dest, const Module *Src,
65505ea54e8869a81b8dd846397175f218f97968907Dan Gohman                      ValueToValueMapTy &ValueMap,
6564fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov                      std::string *Err) {
65731ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio  // Loop over all alias in the src module
65831ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio  for (Module::const_alias_iterator I = Src->alias_begin(),
65931ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio         E = Src->alias_end(); I != E; ++I) {
66058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    const GlobalAlias *SGA = I;
66158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    const GlobalValue *SAliasee = SGA->getAliasedGlobal();
66258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    GlobalAlias *NewGA = NULL;
66358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
66458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    // Globals were already linked, thus we can just query ValueMap for variant
665caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    // of SAliasee in Dest.
66605ea54e8869a81b8dd846397175f218f97968907Dan Gohman    ValueToValueMapTy::const_iterator VMI = ValueMap.find(SAliasee);
66758d5e051574339106ce4139697ebadd21e34dbc0Ted Kremenek    assert(VMI != ValueMap.end() && "Aliasee not linked");
66858d5e051574339106ce4139697ebadd21e34dbc0Ted Kremenek    GlobalValue* DAliasee = cast<GlobalValue>(VMI->second);
669caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    GlobalValue* DGV = NULL;
67058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
67119785995f60c6980677888985884d65dc8975b44Rafael Espindola    // Fixup aliases to bitcasts.  Note that aliases to GEPs are still broken
67219785995f60c6980677888985884d65dc8975b44Rafael Espindola    // by this, but aliases to GEPs are broken to a lot of other things, so
67319785995f60c6980677888985884d65dc8975b44Rafael Espindola    // it's less important.
67419785995f60c6980677888985884d65dc8975b44Rafael Espindola    Constant *DAliaseeConst = DAliasee;
67519785995f60c6980677888985884d65dc8975b44Rafael Espindola    if (SGA->getType() != DAliasee->getType())
67619785995f60c6980677888985884d65dc8975b44Rafael Espindola      DAliaseeConst = ConstantExpr::getBitCast(DAliasee, SGA->getType());
67719785995f60c6980677888985884d65dc8975b44Rafael Espindola
67858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    // Try to find something 'similar' to SGA in destination module.
679bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola    if (!DGV && !SGA->hasLocalLinkage()) {
680caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      DGV = Dest->getNamedAlias(SGA->getName());
681caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov
682caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      // If types don't agree due to opaque types, try to resolve them.
683caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      if (DGV && DGV->getType() != SGA->getType())
6845ed2ba26680f06846e7602ce65432b0a723b1273Chris Lattner        RecursiveResolveTypes(SGA->getType(), DGV->getType());
685caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    }
686caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov
687bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola    if (!DGV && !SGA->hasLocalLinkage()) {
688caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      DGV = Dest->getGlobalVariable(SGA->getName());
689caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov
69058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      // If types don't agree due to opaque types, try to resolve them.
691caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      if (DGV && DGV->getType() != SGA->getType())
6925ed2ba26680f06846e7602ce65432b0a723b1273Chris Lattner        RecursiveResolveTypes(SGA->getType(), DGV->getType());
693caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    }
69458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
695bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola    if (!DGV && !SGA->hasLocalLinkage()) {
696caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      DGV = Dest->getFunction(SGA->getName());
697caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov
698caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      // If types don't agree due to opaque types, try to resolve them.
699caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      if (DGV && DGV->getType() != SGA->getType())
7005ed2ba26680f06846e7602ce65432b0a723b1273Chris Lattner        RecursiveResolveTypes(SGA->getType(), DGV->getType());
701caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    }
702caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov
703caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    // No linking to be performed on internal stuff.
704bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola    if (DGV && DGV->hasLocalLinkage())
705caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      DGV = NULL;
706caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov
707caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    if (GlobalAlias *DGA = dyn_cast_or_null<GlobalAlias>(DGV)) {
708caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      // Types are known to be the same, check whether aliasees equal. As
70958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      // globals are already linked we just need query ValueMap to find the
71058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      // mapping.
71158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      if (DAliasee == DGA->getAliasedGlobal()) {
71258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // This is just two copies of the same alias. Propagate linkage, if
71358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // necessary.
71458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        DGA->setLinkage(CalculateAliasLinkage(SGA, DGA));
71558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
71658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        NewGA = DGA;
71758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // Proceed to 'common' steps
71858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      } else
7191438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov        return Error(Err, "Alias Collision on '"  + SGA->getName()+
7201438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov                     "': aliases have different aliasees");
721caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    } else if (GlobalVariable *DGVar = dyn_cast_or_null<GlobalVariable>(DGV)) {
722f88bc65932e5cf41ba6ca706cd1b1b1f65300248Anton Korobeynikov      // The only allowed way is to link alias with external declaration or weak
723f88bc65932e5cf41ba6ca706cd1b1b1f65300248Anton Korobeynikov      // symbol..
724a05ef5e107943478ad02bc0cd5170a3894076bc4Duncan Sands      if (DGVar->isDeclaration() || DGVar->isWeakForLinker()) {
725ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov        // But only if aliasee is global too...
726ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov        if (!isa<GlobalVariable>(DAliasee))
727caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov          return Error(Err, "Global-Alias Collision on '" + SGA->getName() +
728caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov                       "': aliasee is not global variable");
729ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov
73058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        NewGA = new GlobalAlias(SGA->getType(), SGA->getLinkage(),
73119785995f60c6980677888985884d65dc8975b44Rafael Espindola                                SGA->getName(), DAliaseeConst, Dest);
73258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        CopyGVAttributes(NewGA, SGA);
73358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
73458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // Any uses of DGV need to change to NewGA, with cast, if needed.
735caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov        if (SGA->getType() != DGVar->getType())
736baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson          DGVar->replaceAllUsesWith(ConstantExpr::getBitCast(NewGA,
737caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov                                                             DGVar->getType()));
73858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        else
739caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov          DGVar->replaceAllUsesWith(NewGA);
74058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
741caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov        // DGVar will conflict with NewGA because they both had the same
74258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // name. We must erase this now so ForceRenaming doesn't assert
74358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // because DGV might not have internal linkage.
744caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov        DGVar->eraseFromParent();
74558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
74658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // Proceed to 'common' steps
74758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      } else
7481438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov        return Error(Err, "Global-Alias Collision on '" + SGA->getName() +
7491438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov                     "': symbol multiple defined");
750caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    } else if (Function *DF = dyn_cast_or_null<Function>(DGV)) {
751f88bc65932e5cf41ba6ca706cd1b1b1f65300248Anton Korobeynikov      // The only allowed way is to link alias with external declaration or weak
752f88bc65932e5cf41ba6ca706cd1b1b1f65300248Anton Korobeynikov      // symbol...
753a05ef5e107943478ad02bc0cd5170a3894076bc4Duncan Sands      if (DF->isDeclaration() || DF->isWeakForLinker()) {
754ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov        // But only if aliasee is function too...
755ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov        if (!isa<Function>(DAliasee))
756caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov          return Error(Err, "Function-Alias Collision on '" + SGA->getName() +
757caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov                       "': aliasee is not function");
758ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov
759b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        NewGA = new GlobalAlias(SGA->getType(), SGA->getLinkage(),
76019785995f60c6980677888985884d65dc8975b44Rafael Espindola                                SGA->getName(), DAliaseeConst, Dest);
761b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        CopyGVAttributes(NewGA, SGA);
762b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov
763b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        // Any uses of DF need to change to NewGA, with cast, if needed.
764b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        if (SGA->getType() != DF->getType())
765baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson          DF->replaceAllUsesWith(ConstantExpr::getBitCast(NewGA,
766b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov                                                          DF->getType()));
767b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        else
768b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov          DF->replaceAllUsesWith(NewGA);
769b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov
770b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        // DF will conflict with NewGA because they both had the same
771b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        // name. We must erase this now so ForceRenaming doesn't assert
772b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        // because DF might not have internal linkage.
773b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        DF->eraseFromParent();
774b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov
775b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        // Proceed to 'common' steps
776b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov      } else
7771438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov        return Error(Err, "Function-Alias Collision on '" + SGA->getName() +
7781438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov                     "': symbol multiple defined");
77958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    } else {
780caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      // No linking to be performed, simply create an identical version of the
781caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      // alias over in the dest module...
78258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      NewGA = new GlobalAlias(SGA->getType(), SGA->getLinkage(),
78319785995f60c6980677888985884d65dc8975b44Rafael Espindola                              SGA->getName(), DAliaseeConst, Dest);
78458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      CopyGVAttributes(NewGA, SGA);
78558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
78658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      // Proceed to 'common' steps
78758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    }
78858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
78958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    assert(NewGA && "No alias was created in destination module!");
79058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
791b8cdaf7099898d7c0eef4769bf6a89f39e7decc2Anton Korobeynikov    // If the symbol table renamed the alias, but it is an externally visible
792caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    // symbol, DGA must be an global value with internal linkage. Rename it.
79358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    if (NewGA->getName() != SGA->getName() &&
794bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola        !NewGA->hasLocalLinkage())
79558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      ForceRenaming(NewGA, SGA->getName());
79658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
79758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    // Remember this mapping so uses in the source module get remapped
79805ea54e8869a81b8dd846397175f218f97968907Dan Gohman    // later by MapValue.
799817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov    ValueMap[SGA] = NewGA;
80031ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio  }
80158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
80231ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio  return false;
80331ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio}
80431ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio
8055c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
8068d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner// LinkGlobalInits - Update the initializers in the Dest module now that all
8078d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner// globals that may be referenced are in Dest.
8088d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattnerstatic bool LinkGlobalInits(Module *Dest, const Module *Src,
80905ea54e8869a81b8dd846397175f218f97968907Dan Gohman                            ValueToValueMapTy &ValueMap,
8105c2d335d86579941b71816196a95a64b277e8963Chris Lattner                            std::string *Err) {
8118d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  // Loop over all of the globals in the src module, mapping them over as we go
8121127315562b221010040925980040fd4f65bdb1cChris Lattner  for (Module::const_global_iterator I = Src->global_begin(),
8131127315562b221010040925980040fd4f65bdb1cChris Lattner       E = Src->global_end(); I != E; ++I) {
81418961504fc2b299578dba817900a0696cf3ccc4dChris Lattner    const GlobalVariable *SGV = I;
8158d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner
8168d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner    if (SGV->hasInitializer()) {      // Only process initialized GV's
8178d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      // Figure out what the initializer looks like in the dest module...
8184ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner      Constant *SInit =
8196cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman        cast<Constant>(MapValue(SGV->getInitializer(), ValueMap, true));
820d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov      // Grab destination global variable or alias.
821d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov      GlobalValue *DGV = cast<GlobalValue>(ValueMap[SGV]->stripPointerCasts());
822d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov
823933f9bdb00a8fe48833c4ea3ded20a6a90c405b0Bill Wendling      // If dest if global variable, check that initializers match.
824d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov      if (GlobalVariable *DGVar = dyn_cast<GlobalVariable>(DGV)) {
825d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov        if (DGVar->hasInitializer()) {
826933f9bdb00a8fe48833c4ea3ded20a6a90c405b0Bill Wendling          if (SGV->hasExternalLinkage()) {
827d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov            if (DGVar->getInitializer() != SInit)
828d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov              return Error(Err, "Global Variable Collision on '" +
829d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov                           SGV->getName() +
830d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov                           "': global variables have different initializers");
831a05ef5e107943478ad02bc0cd5170a3894076bc4Duncan Sands          } else if (DGVar->isWeakForLinker()) {
832d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov            // Nothing is required, mapped values will take the new global
833d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov            // automatically.
834a05ef5e107943478ad02bc0cd5170a3894076bc4Duncan Sands          } else if (SGV->isWeakForLinker()) {
835d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov            // Nothing is required, mapped values will take the new global
836d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov            // automatically.
837d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov          } else if (DGVar->hasAppendingLinkage()) {
838c23197a26f34f559ea9797de51e187087c039c42Torok Edwin            llvm_unreachable("Appending linkage unimplemented!");
839d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov          } else {
840c23197a26f34f559ea9797de51e187087c039c42Torok Edwin            llvm_unreachable("Unknown linkage!");
841d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov          }
8424ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner        } else {
843d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov          // Copy the initializer over now...
844d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov          DGVar->setInitializer(SInit);
8454ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner        }
8468d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      } else {
847d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov        // Destination is alias, the only valid situation is when source is
848d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov        // weak. Also, note, that we already checked linkage in LinkGlobals(),
849d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov        // thus we assert here.
850d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov        // FIXME: Should we weaken this assumption, 'dereference' alias and
851d13726f1e91218bc3f7103adf6cd98676bc477e2Anton Korobeynikov        // check for initializer of aliasee?
852a05ef5e107943478ad02bc0cd5170a3894076bc4Duncan Sands        assert(SGV->isWeakForLinker());
8538d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      }
8548d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner    }
8558d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  }
8568d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  return false;
8578d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner}
8585c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
85979df7c0aaa18129e55968c8783ef8346807bd4afChris Lattner// LinkFunctionProtos - Link the functions together between the two modules,
860c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// without doing function bodies... this just adds external function prototypes
861c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// to the Dest function...
8625c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner//
86379df7c0aaa18129e55968c8783ef8346807bd4afChris Lattnerstatic bool LinkFunctionProtos(Module *Dest, const Module *Src,
86405ea54e8869a81b8dd846397175f218f97968907Dan Gohman                               ValueToValueMapTy &ValueMap,
8655c2d335d86579941b71816196a95a64b277e8963Chris Lattner                               std::string *Err) {
866d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner  ValueSymbolTable &DestSymTab = Dest->getValueSymbolTable();
867eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
868619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer  // Loop over all of the functions in the src module, mapping them over
8695c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  for (Module::const_iterator I = Src->begin(), E = Src->end(); I != E; ++I) {
87018961504fc2b299578dba817900a0696cf3ccc4dChris Lattner    const Function *SF = I;   // SrcFunction
871194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov    GlobalValue *DGV = 0;
872eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
873d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner    // Check to see if may have to link the function with the global, alias or
874d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner    // function.
875bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola    if (SF->hasName() && !SF->hasLocalLinkage())
87603d7651c3652e1f0cc86e79b26585d86818da9cfDaniel Dunbar      DGV = cast_or_null<GlobalValue>(DestSymTab.lookup(SF->getName()));
877eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
878ae1132d2b8ae07afd2fe9a7cb434d849f884bfa0Chris Lattner    // If we found a global with the same name in the dest module, but it has
879ae1132d2b8ae07afd2fe9a7cb434d849f884bfa0Chris Lattner    // internal linkage, we are really not doing any linkage here.
880bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola    if (DGV && DGV->hasLocalLinkage())
881ae1132d2b8ae07afd2fe9a7cb434d849f884bfa0Chris Lattner      DGV = 0;
882ae1132d2b8ae07afd2fe9a7cb434d849f884bfa0Chris Lattner
883d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner    // If types don't agree due to opaque types, try to resolve them.
884d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner    if (DGV && DGV->getType() != SF->getType())
885d1ec48c641f76d4c0e5b7cb357a1333d4fb75c4aChris Lattner      RecursiveResolveTypes(SF->getType(), DGV->getType());
886194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov
8876157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    GlobalValue::LinkageTypes NewLinkage = GlobalValue::InternalLinkage;
8886157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    bool LinkFromSrc = false;
8896157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    if (GetLinkageResult(DGV, SF, NewLinkage, LinkFromSrc, Err))
8906157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      return true;
891eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
892824684982968244050e56357e736b9940c23e77dChris Lattner    // If there is no linkage to be performed, just bring over SF without
893824684982968244050e56357e736b9940c23e77dChris Lattner    // modifying it.
894194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov    if (DGV == 0) {
895824684982968244050e56357e736b9940c23e77dChris Lattner      // Function does not already exist, simply insert an function signature
896824684982968244050e56357e736b9940c23e77dChris Lattner      // identical to SF into the dest module.
897824684982968244050e56357e736b9940c23e77dChris Lattner      Function *NewDF = Function::Create(SF->getFunctionType(),
898824684982968244050e56357e736b9940c23e77dChris Lattner                                         SF->getLinkage(),
899824684982968244050e56357e736b9940c23e77dChris Lattner                                         SF->getName(), Dest);
900824684982968244050e56357e736b9940c23e77dChris Lattner      CopyGVAttributes(NewDF, SF);
901eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
902824684982968244050e56357e736b9940c23e77dChris Lattner      // If the LLVM runtime renamed the function, but it is an externally
903824684982968244050e56357e736b9940c23e77dChris Lattner      // visible symbol, DF must be an existing function with internal linkage.
904824684982968244050e56357e736b9940c23e77dChris Lattner      // Rename it.
905bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola      if (!NewDF->hasLocalLinkage() && NewDF->getName() != SF->getName())
906824684982968244050e56357e736b9940c23e77dChris Lattner        ForceRenaming(NewDF, SF->getName());
907eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
908824684982968244050e56357e736b9940c23e77dChris Lattner      // ... and remember this mapping...
909824684982968244050e56357e736b9940c23e77dChris Lattner      ValueMap[SF] = NewDF;
910824684982968244050e56357e736b9940c23e77dChris Lattner      continue;
91197f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner    }
912eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
9136157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    // If the visibilities of the symbols disagree and the destination is a
9146157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    // prototype, take the visibility of its input.
9156157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    if (DGV->isDeclaration())
9166157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      DGV->setVisibility(SF->getVisibility());
917eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
9186157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    if (LinkFromSrc) {
9196157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      if (isa<GlobalAlias>(DGV))
9206157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner        return Error(Err, "Function-Alias Collision on '" + SF->getName() +
9216157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner                     "': symbol multiple defined");
922eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
9236157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      // We have a definition of the same name but different type in the
9246157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      // source module. Copy the prototype to the destination and replace
9256157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      // uses of the destination's prototype with the new prototype.
9266157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      Function *NewDF = Function::Create(SF->getFunctionType(), NewLinkage,
9276157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner                                         SF->getName(), Dest);
9286157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      CopyGVAttributes(NewDF, SF);
929eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
9306157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      // Any uses of DF need to change to NewDF, with cast
931baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson      DGV->replaceAllUsesWith(ConstantExpr::getBitCast(NewDF,
932c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson                                                              DGV->getType()));
933eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
9346157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      // DF will conflict with NewDF because they both had the same. We must
9356157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      // erase this now so ForceRenaming doesn't assert because DF might
936eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov      // not have internal linkage.
9376157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      if (GlobalVariable *Var = dyn_cast<GlobalVariable>(DGV))
9386157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner        Var->eraseFromParent();
9396157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      else
9406157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner        cast<Function>(DGV)->eraseFromParent();
941eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
9426157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      // If the symbol table renamed the function, but it is an externally
943eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov      // visible symbol, DF must be an existing function with internal
9446157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      // linkage.  Rename it.
945bb46f52027416598a662dc1c58f48d9d56b1a65bRafael Espindola      if (NewDF->getName() != SF->getName() && !NewDF->hasLocalLinkage())
9466157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner        ForceRenaming(NewDF, SF->getName());
947eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
9486157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      // Remember this mapping so uses in the source module get remapped
94905ea54e8869a81b8dd846397175f218f97968907Dan Gohman      // later by MapValue.
9506157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      ValueMap[SF] = NewDF;
951822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner      continue;
952822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner    }
953eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
9546157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    // Not "link from source", keep the one in the DestModule and remap the
9556157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    // input onto it.
956eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
9576157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    if (isa<GlobalAlias>(DGV)) {
9586157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      // The only valid mappings are:
9596157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      // - SF is external declaration, which is effectively a no-op.
9606157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner      // - SF is weak, when we just need to throw SF out.
961a05ef5e107943478ad02bc0cd5170a3894076bc4Duncan Sands      if (!SF->isDeclaration() && !SF->isWeakForLinker())
9626157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner        return Error(Err, "Function-Alias Collision on '" + SF->getName() +
9636157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner                     "': symbol multiple defined");
964822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner    }
9656157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner
9666157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    // Set calculated linkage
9676157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    DGV->setLinkage(NewLinkage);
9686157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner
9696157e383c4a15ffb6e7f27cdb153daa9caa15ca5Chris Lattner    // Make sure to remember this mapping.
970baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson    ValueMap[SF] = ConstantExpr::getBitCast(DGV, SF->getType());
9715c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  }
9725c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return false;
9735c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
9745c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
975c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// LinkFunctionBody - Copy the source function over into the dest function and
976c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// fix up references to values.  At this point we know that Dest is an external
977c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// function, and that Src is not.
9784bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattnerstatic bool LinkFunctionBody(Function *Dest, Function *Src,
97905ea54e8869a81b8dd846397175f218f97968907Dan Gohman                             ValueToValueMapTy &ValueMap,
9805c2d335d86579941b71816196a95a64b277e8963Chris Lattner                             std::string *Err) {
9815cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer  assert(Src && Dest && Dest->isDeclaration() && !Src->isDeclaration());
9825c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
9830033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner  // Go through and convert function arguments over, remembering the mapping.
984e4d5c441e04bdc00ccf1804744af670655123b07Chris Lattner  Function::arg_iterator DI = Dest->arg_begin();
985e4d5c441e04bdc00ccf1804744af670655123b07Chris Lattner  for (Function::arg_iterator I = Src->arg_begin(), E = Src->arg_end();
98669da5cf26143e4542d4bf8c78ffac6d079efe5c9Chris Lattner       I != E; ++I, ++DI) {
9876bc41e8a74d1756da0003641bfebd02a3d6d9586Owen Anderson    DI->setName(I->getName());  // Copy the name information over...
9885c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
9895c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner    // Add a mapping to our local map
990817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov    ValueMap[I] = DI;
9915c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  }
9925c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
9934bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner  // Splice the body of the source function into the dest function.
9944bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner  Dest->getBasicBlockList().splice(Dest->end(), Src->getBasicBlockList());
9955c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
996c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // At this point, all of the instructions and values of the function are now
997c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // copied over.  The only problem is that they are still referencing values in
998c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // the Source function as operands.  Loop through all of the operands of the
999c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // functions and patch them up to point to the local versions...
10005c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  //
10016cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman  // This is the same as RemapInstruction, except that it avoids remapping
10026cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman  // instruction and basic block operands.
10036cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman  //
100418961504fc2b299578dba817900a0696cf3ccc4dChris Lattner  for (Function::iterator BB = Dest->begin(), BE = Dest->end(); BB != BE; ++BB)
10056cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman    for (BasicBlock::iterator I = BB->begin(), E = BB->end(); I != E; ++I) {
10066cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman      // Remap operands.
100718961504fc2b299578dba817900a0696cf3ccc4dChris Lattner      for (Instruction::op_iterator OI = I->op_begin(), OE = I->op_end();
1008221d688a5ef21a22c2368c9fff0e92d7966c95e5Chris Lattner           OI != OE; ++OI)
10094bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner        if (!isa<Instruction>(*OI) && !isa<BasicBlock>(*OI))
10106cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman          *OI = MapValue(*OI, ValueMap, true);
10116cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman
10126cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman      // Remap attached metadata.
10136cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman      SmallVector<std::pair<unsigned, MDNode *>, 4> MDs;
10146cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman      I->getAllMetadata(MDs);
10156cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman      for (SmallVectorImpl<std::pair<unsigned, MDNode *> >::iterator
10166cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman           MI = MDs.begin(), ME = MDs.end(); MI != ME; ++MI) {
10176cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman        Value *Old = MI->second;
10186cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman        if (!isa<Instruction>(Old) && !isa<BasicBlock>(Old)) {
10196cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman          Value *New = MapValue(Old, ValueMap, true);
10206cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman          if (New != Old)
10216cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman            I->setMetadata(MI->first, cast<MDNode>(New));
10226cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman        }
10236cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman      }
10246cb8c23db1c3becdce6dfbf1b7f1677faca4251eDan Gohman    }
10250033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner
10260033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner  // There is no need to map the arguments anymore.
10271127315562b221010040925980040fd4f65bdb1cChris Lattner  for (Function::arg_iterator I = Src->arg_begin(), E = Src->arg_end();
10281127315562b221010040925980040fd4f65bdb1cChris Lattner       I != E; ++I)
1029ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer    ValueMap.erase(I);
10305c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
10315c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return false;
10325c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
10335c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
10345c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
1035c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// LinkFunctionBodies - Link in the function bodies that are defined in the
1036c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// source module into the DestModule.  This consists basically of copying the
1037c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// function over and fixing up references to values.
10384bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattnerstatic bool LinkFunctionBodies(Module *Dest, Module *Src,
103905ea54e8869a81b8dd846397175f218f97968907Dan Gohman                               ValueToValueMapTy &ValueMap,
10405c2d335d86579941b71816196a95a64b277e8963Chris Lattner                               std::string *Err) {
10415c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
10428bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer  // Loop over all of the functions in the src module, mapping them over as we
10438bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer  // go
10444bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner  for (Module::iterator SF = Src->begin(), E = Src->end(); SF != E; ++SF) {
1045619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer    if (!SF->isDeclaration()) {               // No body if function is external
1046ec91ccba3ca92d5e1f19fb0cb6e57a7d4a3f3195Chris Lattner      Function *DF = dyn_cast<Function>(ValueMap[SF]); // Destination function
1047c2d774b6c1d38421c435b6d3cfaa10402c900aebChris Lattner
104818961504fc2b299578dba817900a0696cf3ccc4dChris Lattner      // DF not external SF external?
1049ec91ccba3ca92d5e1f19fb0cb6e57a7d4a3f3195Chris Lattner      if (DF && DF->isDeclaration())
105035956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner        // Only provide the function body if there isn't one already.
105135956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner        if (LinkFunctionBody(DF, SF, ValueMap, Err))
105235956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner          return true;
1053c2d774b6c1d38421c435b6d3cfaa10402c900aebChris Lattner    }
10545c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  }
10555c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return false;
10565c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
10575c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
10588166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner// LinkAppendingVars - If there were any appending global variables, link them
10598166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner// together now.  Return true on error.
10608166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattnerstatic bool LinkAppendingVars(Module *M,
10618166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                  std::multimap<std::string, GlobalVariable *> &AppendingVars,
10628166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                              std::string *ErrorMsg) {
10638166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  if (AppendingVars.empty()) return false; // Nothing to do.
1064f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
10658166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // Loop over the multimap of appending vars, processing any variables with the
10668166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // same name, forming a new appending global variable with both of the
10678166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // initializers merged together, then rewrite references to the old variables
10688166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // and delete them.
10698166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  std::vector<Constant*> Inits;
10708166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  while (AppendingVars.size() > 1) {
10718166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    // Get the first two elements in the map...
10728166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    std::multimap<std::string,
10738166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      GlobalVariable*>::iterator Second = AppendingVars.begin(), First=Second++;
10748166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
10758166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    // If the first two elements are for different names, there is no pair...
10768166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    // Otherwise there is a pair, so link them together...
10778166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    if (First->first == Second->first) {
10788166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      GlobalVariable *G1 = First->second, *G2 = Second->second;
10798166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      const ArrayType *T1 = cast<ArrayType>(G1->getType()->getElementType());
10808166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      const ArrayType *T2 = cast<ArrayType>(G2->getType()->getElementType());
1081f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
10828166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Check to see that they two arrays agree on type...
10838166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      if (T1->getElementType() != T2->getElementType())
10848166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        return Error(ErrorMsg,
10858166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner         "Appending variables with different element types need to be linked!");
10868166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      if (G1->isConstant() != G2->isConstant())
10878166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        return Error(ErrorMsg,
10888166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                     "Appending variables linked with different const'ness!");
10898166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
10909613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio      if (G1->getAlignment() != G2->getAlignment())
10919613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio        return Error(ErrorMsg,
10929613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio         "Appending variables with different alignment need to be linked!");
10939613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio
10949613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio      if (G1->getVisibility() != G2->getVisibility())
10959613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio        return Error(ErrorMsg,
10969613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio         "Appending variables with different visibility need to be linked!");
10979613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio
10989613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio      if (G1->getSection() != G2->getSection())
10999613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio        return Error(ErrorMsg,
11009613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio         "Appending variables with different section name need to be linked!");
1101eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
11028166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      unsigned NewSize = T1->getNumElements() + T2->getNumElements();
1103debcb01b0f0a15f568ca69e8f288fade4bfc7297Owen Anderson      ArrayType *NewType = ArrayType::get(T1->getElementType(),
1104c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson                                                         NewSize);
11058166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
1106ed74a4ef3b813bed0c0f4bb5af42229d50133da2Chris Lattner      G1->setName("");   // Clear G1's name in case of a conflict!
1107eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
11088166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Create the new global variable...
11098166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      GlobalVariable *NG =
1110e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson        new GlobalVariable(*M, NewType, G1->isConstant(), G1->getLinkage(),
1111e9b11b431308f4766b73cda93e38ec930c912122Owen Anderson                           /*init*/0, First->first, 0, G1->isThreadLocal(),
1112a534b0f7bf0248bf90afe97110a34777e972f325Chris Lattner                           G1->getType()->getAddressSpace());
11138166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
11149613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio      // Propagate alignment, visibility and section info.
11159613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio      CopyGVAttributes(NG, G1);
11169613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio
11178166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Merge the initializer...
11188166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      Inits.reserve(NewSize);
1119de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      if (ConstantArray *I = dyn_cast<ConstantArray>(G1->getInitializer())) {
1120de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        for (unsigned i = 0, e = T1->getNumElements(); i != e; ++i)
1121cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos          Inits.push_back(I->getOperand(i));
1122de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      } else {
1123de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        assert(isa<ConstantAggregateZero>(G1->getInitializer()));
1124a7235ea7245028a0723e8ab7fd011386b3900777Owen Anderson        Constant *CV = Constant::getNullValue(T1->getElementType());
1125de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        for (unsigned i = 0, e = T1->getNumElements(); i != e; ++i)
1126de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner          Inits.push_back(CV);
1127de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      }
1128de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      if (ConstantArray *I = dyn_cast<ConstantArray>(G2->getInitializer())) {
1129de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        for (unsigned i = 0, e = T2->getNumElements(); i != e; ++i)
1130cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos          Inits.push_back(I->getOperand(i));
1131de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      } else {
1132de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        assert(isa<ConstantAggregateZero>(G2->getInitializer()));
1133a7235ea7245028a0723e8ab7fd011386b3900777Owen Anderson        Constant *CV = Constant::getNullValue(T2->getElementType());
1134de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        for (unsigned i = 0, e = T2->getNumElements(); i != e; ++i)
1135de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner          Inits.push_back(CV);
1136de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      }
11371fd7096407d5e598ed3366a1141548e71273f1c5Owen Anderson      NG->setInitializer(ConstantArray::get(NewType, Inits));
11388166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      Inits.clear();
11398166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
11408166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Replace any uses of the two global variables with uses of the new
11418166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // global...
11428166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
11438166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // FIXME: This should rewrite simple/straight-forward uses such as
11448166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // getelementptr instructions to not use the Cast!
1145baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson      G1->replaceAllUsesWith(ConstantExpr::getBitCast(NG,
1146c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson                             G1->getType()));
1147baf3c404409d5e47b13984a7f95bfbd6d1f2e79eOwen Anderson      G2->replaceAllUsesWith(ConstantExpr::getBitCast(NG,
1148c9ab7bf98f6a0261220e4b2380db2df8ee29829bOwen Anderson                             G2->getType()));
11498166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
11508166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Remove the two globals from the module now...
11518166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      M->getGlobalList().erase(G1);
11528166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      M->getGlobalList().erase(G2);
11538166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
11548166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Put the new global into the AppendingVars map so that we can handle
11558166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // linking of more than two vars...
11568166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      Second->second = NG;
11578166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    }
11588166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    AppendingVars.erase(First);
11598166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  }
11608166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
11618166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  return false;
11628166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner}
116352f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner
11649f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikovstatic bool ResolveAliases(Module *Dest) {
11659f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov  for (Module::alias_iterator I = Dest->alias_begin(), E = Dest->alias_end();
11665241957b1f6876452705f94e0ed60dd2c33a5c84Anton Korobeynikov       I != E; ++I)
11673472246edee4e120e3eb1ec4ad443ba884bc7ac7David Chisnall    // We can't sue resolveGlobalAlias here because we need to preserve
11683472246edee4e120e3eb1ec4ad443ba884bc7ac7David Chisnall    // bitcasts and GEPs.
11693472246edee4e120e3eb1ec4ad443ba884bc7ac7David Chisnall    if (const Constant *C = I->getAliasee()) {
11703472246edee4e120e3eb1ec4ad443ba884bc7ac7David Chisnall      while (dyn_cast<GlobalAlias>(C))
11713472246edee4e120e3eb1ec4ad443ba884bc7ac7David Chisnall        C = cast<GlobalAlias>(C)->getAliasee();
11723472246edee4e120e3eb1ec4ad443ba884bc7ac7David Chisnall      const GlobalValue *GV = dyn_cast<GlobalValue>(C);
11733472246edee4e120e3eb1ec4ad443ba884bc7ac7David Chisnall      if (C != I && !(GV && GV->isDeclaration()))
11743472246edee4e120e3eb1ec4ad443ba884bc7ac7David Chisnall        I->replaceAllUsesWith(const_cast<Constant*>(C));
11753472246edee4e120e3eb1ec4ad443ba884bc7ac7David Chisnall    }
11769f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov
11779f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov  return false;
11789f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov}
117952f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner
118052f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// LinkModules - This function links two modules together, with the resulting
118152f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// left module modified to be the composite of the two input modules.  If an
118252f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// error occurs, true is returned and ErrorMsg (if not null) is set to indicate
11835c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// the problem.  Upon failure, the Dest module could be in a modified state, and
11845c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// shouldn't be relied on to be consistent.
1185f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukmanbool
11860ba9e211bed7813949aaca97e6b15579c8188e05Reid SpencerLinker::LinkModules(Module *Dest, Module *Src, std::string *ErrorMsg) {
118757a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  assert(Dest != 0 && "Invalid Destination module");
118857a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  assert(Src  != 0 && "Invalid Source Module");
118957a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer
1190c36357c406911ded5a0e2924998d885815370fdbChris Lattner  if (Dest->getDataLayout().empty()) {
1191c36357c406911ded5a0e2924998d885815370fdbChris Lattner    if (!Src->getDataLayout().empty()) {
1192ec9bfdc152a870d3c62377ed6855fb3bff69511cChris Lattner      Dest->setDataLayout(Src->getDataLayout());
1193c36357c406911ded5a0e2924998d885815370fdbChris Lattner    } else {
1194c36357c406911ded5a0e2924998d885815370fdbChris Lattner      std::string DataLayout;
1195c36357c406911ded5a0e2924998d885815370fdbChris Lattner
1196a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov      if (Dest->getEndianness() == Module::AnyEndianness) {
1197c36357c406911ded5a0e2924998d885815370fdbChris Lattner        if (Src->getEndianness() == Module::BigEndian)
1198c36357c406911ded5a0e2924998d885815370fdbChris Lattner          DataLayout.append("E");
1199c36357c406911ded5a0e2924998d885815370fdbChris Lattner        else if (Src->getEndianness() == Module::LittleEndian)
1200c36357c406911ded5a0e2924998d885815370fdbChris Lattner          DataLayout.append("e");
1201a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov      }
1202a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov
1203a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov      if (Dest->getPointerSize() == Module::AnyPointerSize) {
1204c36357c406911ded5a0e2924998d885815370fdbChris Lattner        if (Src->getPointerSize() == Module::Pointer64)
1205c36357c406911ded5a0e2924998d885815370fdbChris Lattner          DataLayout.append(DataLayout.length() == 0 ? "p:64:64" : "-p:64:64");
1206c36357c406911ded5a0e2924998d885815370fdbChris Lattner        else if (Src->getPointerSize() == Module::Pointer32)
1207c36357c406911ded5a0e2924998d885815370fdbChris Lattner          DataLayout.append(DataLayout.length() == 0 ? "p:32:32" : "-p:32:32");
1208a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov      }
1209c36357c406911ded5a0e2924998d885815370fdbChris Lattner      Dest->setDataLayout(DataLayout);
1210c36357c406911ded5a0e2924998d885815370fdbChris Lattner    }
1211c36357c406911ded5a0e2924998d885815370fdbChris Lattner  }
1212873c5e7859c1534bf92c9e2747f2b70685059598Chris Lattner
1213f27dfcb210fec571d79bc881f5463e0b405707c0Chris Lattner  // Copy the target triple from the source to dest if the dest's is empty.
1214c36357c406911ded5a0e2924998d885815370fdbChris Lattner  if (Dest->getTargetTriple().empty() && !Src->getTargetTriple().empty())
1215c36357c406911ded5a0e2924998d885815370fdbChris Lattner    Dest->setTargetTriple(Src->getTargetTriple());
1216eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
1217c36357c406911ded5a0e2924998d885815370fdbChris Lattner  if (!Src->getDataLayout().empty() && !Dest->getDataLayout().empty() &&
1218c36357c406911ded5a0e2924998d885815370fdbChris Lattner      Src->getDataLayout() != Dest->getDataLayout())
1219bdff548e4dd577a72094d57b282de4e765643b96Chris Lattner    errs() << "WARNING: Linking two modules of different data layouts!\n";
1220152f19ac6514ec29e183ddcdf2ea71599638dc4cChris Lattner  if (!Src->getTargetTriple().empty() &&
1221152f19ac6514ec29e183ddcdf2ea71599638dc4cChris Lattner      Dest->getTargetTriple() != Src->getTargetTriple())
1222bdff548e4dd577a72094d57b282de4e765643b96Chris Lattner    errs() << "WARNING: Linking two modules of different target triples!\n";
1223f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
1224f27dfcb210fec571d79bc881f5463e0b405707c0Chris Lattner  // Append the module inline asm string.
12256631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner  if (!Src->getModuleInlineAsm().empty()) {
12266631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner    if (Dest->getModuleInlineAsm().empty())
12276631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner      Dest->setModuleInlineAsm(Src->getModuleInlineAsm());
1228e1b2e14f92e8e9da1eb4854511a881cf9c22af3cChris Lattner    else
12296631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner      Dest->setModuleInlineAsm(Dest->getModuleInlineAsm()+"\n"+
12306631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner                               Src->getModuleInlineAsm());
1231e1b2e14f92e8e9da1eb4854511a881cf9c22af3cChris Lattner  }
1232eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov
1233719012d6254ee425464f034250176283fb96793bReid Spencer  // Update the destination module's dependent libraries list with the libraries
123457a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  // from the source module. There's no opportunity for duplicates here as the
123557a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  // Module ensures that duplicate insertions are discarded.
1236f27dfcb210fec571d79bc881f5463e0b405707c0Chris Lattner  for (Module::lib_iterator SI = Src->lib_begin(), SE = Src->lib_end();
1237eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov       SI != SE; ++SI)
123857a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer    Dest->addLibrary(*SI);
123957a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer
12402c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  // LinkTypes - Go through the symbol table of the Src module and see if any
12412c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  // types are named in the src module that are not named in the Dst module.
12422c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  // Make sure there are no type name conflicts.
1243eba2cb0d5aef5a8229facae5624c65d9fd65e9d1Mikhail Glushenkov  if (LinkTypes(Dest, Src, ErrorMsg))
1244619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer    return true;
12452c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
12465c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  // ValueMap - Mapping of values from what they used to be in Src, to what they
124705ea54e8869a81b8dd846397175f218f97968907Dan Gohman  // are now in Dest.  ValueToValueMapTy is a ValueMap, which involves some
124805ea54e8869a81b8dd846397175f218f97968907Dan Gohman  // overhead due to the use of Value handles which the Linker doesn't actually
124905ea54e8869a81b8dd846397175f218f97968907Dan Gohman  // need, but this allows us to reuse the ValueMapper code.
125005ea54e8869a81b8dd846397175f218f97968907Dan Gohman  ValueToValueMapTy ValueMap;
12515c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
12528166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // AppendingVars - Keep track of global variables in the destination module
12538166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // with appending linkage.  After the module is linked together, they are
12548166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // appended and the module is rewritten.
12558166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  std::multimap<std::string, GlobalVariable *> AppendingVars;
12561127315562b221010040925980040fd4f65bdb1cChris Lattner  for (Module::global_iterator I = Dest->global_begin(), E = Dest->global_end();
12571127315562b221010040925980040fd4f65bdb1cChris Lattner       I != E; ++I) {
12585a837defff581accc09048bea1cd839009e374a7Chris Lattner    // Add all of the appending globals already in the Dest module to
12595a837defff581accc09048bea1cd839009e374a7Chris Lattner    // AppendingVars.
1260f41464618f41b92cf0a0bd14ccc57bce727fb33dChris Lattner    if (I->hasAppendingLinkage())
1261f41464618f41b92cf0a0bd14ccc57bce727fb33dChris Lattner      AppendingVars.insert(std::make_pair(I->getName(), I));
12625a837defff581accc09048bea1cd839009e374a7Chris Lattner  }
12635a837defff581accc09048bea1cd839009e374a7Chris Lattner
12648166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // Insert all of the globals in src into the Dest module... without linking
12658166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // initializers (which could refer to functions not yet mapped over).
1266ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  if (LinkGlobals(Dest, Src, ValueMap, AppendingVars, ErrorMsg))
12675a837defff581accc09048bea1cd839009e374a7Chris Lattner    return true;
12685c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
1269c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // Link the functions together between the two modules, without doing function
1270c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // bodies... this just adds external function prototypes to the Dest
1271c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // function...  We do this so that when we begin processing function bodies,
1272c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // all of the global values that may be referenced are available in our
1273c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // ValueMap.
1274ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  if (LinkFunctionProtos(Dest, Src, ValueMap, ErrorMsg))
12755a837defff581accc09048bea1cd839009e374a7Chris Lattner    return true;
12765c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
12774fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov  // If there were any alias, link them now. We really need to do this now,
12784fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov  // because all of the aliases that may be referenced need to be available in
12794fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov  // ValueMap
12804fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov  if (LinkAlias(Dest, Src, ValueMap, ErrorMsg)) return true;
12814fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov
12826cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner  // Update the initializers in the Dest module now that all globals that may
12836cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner  // be referenced are in Dest.
12846cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner  if (LinkGlobalInits(Dest, Src, ValueMap, ErrorMsg)) return true;
12856cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner
1286c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // Link in the function bodies that are defined in the source module into the
1287c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // DestModule.  This consists basically of copying the function over and
1288c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // fixing up references to values.
128979df7c0aaa18129e55968c8783ef8346807bd4afChris Lattner  if (LinkFunctionBodies(Dest, Src, ValueMap, ErrorMsg)) return true;
129052f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner
12918166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // If there were any appending global variables, link them together now.
12928166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  if (LinkAppendingVars(Dest, AppendingVars, ErrorMsg)) return true;
12938166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
12943db9191baf5f5229a2f4d204c6332ba27fff81d0Anton Korobeynikov  // Resolve all uses of aliases with aliasees
12953db9191baf5f5229a2f4d204c6332ba27fff81d0Anton Korobeynikov  if (ResolveAliases(Dest)) return true;
12963db9191baf5f5229a2f4d204c6332ba27fff81d0Anton Korobeynikov
1297e422d1b16238ad4d9d7ef3b074715e795ab3ddc0Dan Gohman  // Remap all of the named mdnoes in Src into the Dest module. We do this
1298e422d1b16238ad4d9d7ef3b074715e795ab3ddc0Dan Gohman  // after linking GlobalValues so that MDNodes that reference GlobalValues
1299e422d1b16238ad4d9d7ef3b074715e795ab3ddc0Dan Gohman  // are properly remapped.
1300e422d1b16238ad4d9d7ef3b074715e795ab3ddc0Dan Gohman  LinkNamedMDNodes(Dest, Src, ValueMap);
1301e422d1b16238ad4d9d7ef3b074715e795ab3ddc0Dan Gohman
130257a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  // If the source library's module id is in the dependent library list of the
130357a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  // destination library, remove it since that module is now linked in.
130457a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  sys::Path modId;
1305dd04df0ec33a903ee7fc747701bafde622f77d8bReid Spencer  modId.set(Src->getModuleIdentifier());
130607adb2836b8aa7a3872e33c285958f5937662b50Reid Spencer  if (!modId.isEmpty())
130707adb2836b8aa7a3872e33c285958f5937662b50Reid Spencer    Dest->removeLibrary(modId.getBasename());
130857a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer
130952f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner  return false;
131052f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner}
13119466f5113ba859677a28887c3c7143065e702019Vikram S. Adve
1312567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer// vim: sw=2
1313