LinkModules.cpp revision 0bb8757997392ee18ca9de0f1fab29463f8e7aeb
17f49602e5ac7186c9505a0496393dd624515b459Reid Spencer//===- 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"
225c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner#include "llvm/Module.h"
2378d033e086e19e016273de014f9214aa6f3f844bReid Spencer#include "llvm/TypeSymbolTable.h"
24ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer#include "llvm/ValueSymbolTable.h"
2547b14a4a6a455c7be169cfd312fcbe796f0ad426Misha Brukman#include "llvm/Instructions.h"
26adbc0b5287bf36893cdcae2440d48b3cb3489e38Chris Lattner#include "llvm/Assembly/Writer.h"
2741edad7e4b8872414564dea5fb41fa6194375355Bill Wendling#include "llvm/Support/Streams.h"
2857a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer#include "llvm/System/Path.h"
2962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner#include "llvm/ADT/DenseMap.h"
301a097e30d39e60303ae2b19f7a56e813f3e3c18eBill Wendling#include <sstream>
31f7703df4968084c18c248c1feea9961c19a32e6aChris Lattnerusing namespace llvm;
32d0fde30ce850b78371fd1386338350591f9ff494Brian Gaeke
335c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// Error - Simple wrapper function to conditionally assign to E and return true.
345c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// This just makes error return conditions a little bit simpler...
358166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattnerstatic inline bool Error(std::string *E, const std::string &Message) {
365c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  if (E) *E = Message;
375c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return true;
385c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
395c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
40719012d6254ee425464f034250176283fb96793bReid Spencer// ToStr - Simple wrapper function to convert a type to a string.
4172cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattnerstatic std::string ToStr(const Type *Ty, const Module *M) {
4272cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattner  std::ostringstream OS;
4372cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattner  WriteTypeSymbolic(OS, Ty, M);
4472cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattner  return OS.str();
4572cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattner}
4672cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattner
47700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//
48700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Function: ResolveTypes()
49700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//
50700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Description:
51700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  Attempt to link the two specified types together.
52700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//
53700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Inputs:
54700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  DestTy - The type to which we wish to resolve.
55700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  SrcTy  - The original type which we want to resolve.
56700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//
57700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Outputs:
58700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  DestST - The symbol table in which the new type should be placed.
59700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//
60700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Return value:
61700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  true  - There is an error and the types cannot yet be linked.
62700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  false - No errors.
634c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner//
64bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattnerstatic bool ResolveTypes(const Type *DestTy, const Type *SrcTy) {
65e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  if (DestTy == SrcTy) return false;       // If already equal, noop
66bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner  assert(DestTy && SrcTy && "Can't handle null types");
67e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
68bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner  if (const OpaqueType *OT = dyn_cast<OpaqueType>(DestTy)) {
69bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner    // Type _is_ in module, just opaque...
70bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner    const_cast<OpaqueType*>(OT)->refineAbstractTypeTo(SrcTy);
71bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner  } else if (const OpaqueType *OT = dyn_cast<OpaqueType>(SrcTy)) {
72bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner    const_cast<OpaqueType*>(OT)->refineAbstractTypeTo(DestTy);
73bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner  } else {
74bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner    return true;  // Cannot link types... not-equal and neither is opaque.
754c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  }
764c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  return false;
774c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner}
784c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
7962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner/// LinkerTypeMap - This implements a map of types that is stable
8062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner/// even if types are resolved/refined to other types.  This is not a general
8162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner/// purpose map, it is specific to the linker's use.
8262a81a1eba019ab570b002f8e1686494139785a1Chris Lattnernamespace {
8362a81a1eba019ab570b002f8e1686494139785a1Chris Lattnerclass LinkerTypeMap : public AbstractTypeUser {
8462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  typedef DenseMap<const Type*, PATypeHolder> TheMapTy;
8562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  TheMapTy TheMap;
8662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner
87fc196f9ee909ac35837f6f771a42aa0dec817584Chris Lattner  LinkerTypeMap(const LinkerTypeMap&); // DO NOT IMPLEMENT
88fc196f9ee909ac35837f6f771a42aa0dec817584Chris Lattner  void operator=(const LinkerTypeMap&); // DO NOT IMPLEMENT
89fc196f9ee909ac35837f6f771a42aa0dec817584Chris Lattnerpublic:
90fc196f9ee909ac35837f6f771a42aa0dec817584Chris Lattner  LinkerTypeMap() {}
91fc196f9ee909ac35837f6f771a42aa0dec817584Chris Lattner  ~LinkerTypeMap() {
9262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    for (DenseMap<const Type*, PATypeHolder>::iterator I = TheMap.begin(),
9362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner         E = TheMap.end(); I != E; ++I)
9462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      I->first->removeAbstractTypeUser(this);
9562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  }
9662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner
9762a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// lookup - Return the value for the specified type or null if it doesn't
9862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// exist.
9962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  const Type *lookup(const Type *Ty) const {
10062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    TheMapTy::const_iterator I = TheMap.find(Ty);
10162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    if (I != TheMap.end()) return I->second;
10262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    return 0;
10362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  }
10462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner
10562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// erase - Remove the specified type, returning true if it was in the set.
10662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  bool erase(const Type *Ty) {
10762a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    if (!TheMap.erase(Ty))
10862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      return false;
10962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    if (Ty->isAbstract())
11062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      Ty->removeAbstractTypeUser(this);
11162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    return true;
11262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  }
11362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner
11462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// insert - This returns true if the pointer was new to the set, false if it
11562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// was already in the set.
11662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  bool insert(const Type *Src, const Type *Dst) {
1176b345ee9b2833cf1b2f79dc16d06d4060bec36efDan Gohman    if (!TheMap.insert(std::make_pair(Src, PATypeHolder(Dst))).second)
11862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      return false;  // Already in map.
11962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    if (Src->isAbstract())
12062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      Src->addAbstractTypeUser(this);
12162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    return true;
12262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  }
12362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner
12462a81a1eba019ab570b002f8e1686494139785a1Chris Lattnerprotected:
12562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// refineAbstractType - The callback method invoked when an abstract type is
12662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// resolved to another type.  An object must override this method to update
12762a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// its internal state to reference NewType instead of OldType.
12862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  ///
12962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  virtual void refineAbstractType(const DerivedType *OldTy,
13062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner                                  const Type *NewTy) {
13162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    TheMapTy::iterator I = TheMap.find(OldTy);
13262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    const Type *DstTy = I->second;
13362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner
13462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    TheMap.erase(I);
13562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    if (OldTy->isAbstract())
13662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      OldTy->removeAbstractTypeUser(this);
13762a81a1eba019ab570b002f8e1686494139785a1Chris Lattner
13862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    // Don't reinsert into the map if the key is concrete now.
13962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    if (NewTy->isAbstract())
14062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      insert(NewTy, DstTy);
14162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  }
14262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner
14362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// The other case which AbstractTypeUsers must be aware of is when a type
14462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// makes the transition from being abstract (where it has clients on it's
14562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// AbstractTypeUsers list) to concrete (where it does not).  This method
14662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  /// notifies ATU's when this occurs for a type.
14762a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  virtual void typeBecameConcrete(const DerivedType *AbsTy) {
14862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    TheMap.erase(AbsTy);
14962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    AbsTy->removeAbstractTypeUser(this);
15062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  }
15162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner
15262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  // for debugging...
15362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  virtual void dump() const {
15462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    cerr << "AbstractTypeSet!\n";
15562a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  }
15662a81a1eba019ab570b002f8e1686494139785a1Chris Lattner};
15762a81a1eba019ab570b002f8e1686494139785a1Chris Lattner}
15862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner
15962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner
160e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner// RecursiveResolveTypes - This is just like ResolveTypes, except that it
161e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner// recurses down into derived types, merging the used types if the parent types
162e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner// are compatible.
163a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattnerstatic bool RecursiveResolveTypesI(const Type *DstTy, const Type *SrcTy,
16462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner                                   LinkerTypeMap &Pointers) {
165a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner  if (DstTy == SrcTy) return false;       // If already equal, noop
166f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
167e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  // If we found our opaque type, resolve it now!
168a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner  if (isa<OpaqueType>(DstTy) || isa<OpaqueType>(SrcTy))
169a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    return ResolveTypes(DstTy, SrcTy);
170f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
171e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  // Two types cannot be resolved together if they are of different primitive
172e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  // type.  For example, we cannot resolve an int to a float.
173a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner  if (DstTy->getTypeID() != SrcTy->getTypeID()) return true;
174e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
17556539659eb1b192d493aa333b60213889129b9f1Chris Lattner  // If neither type is abstract, then they really are just different types.
176a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner  if (!DstTy->isAbstract() && !SrcTy->isAbstract())
17756539659eb1b192d493aa333b60213889129b9f1Chris Lattner    return true;
17856539659eb1b192d493aa333b60213889129b9f1Chris Lattner
179e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  // Otherwise, resolve the used type used by this derived type...
180a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner  switch (DstTy->getTypeID()) {
181f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner  default:
182f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner    return true;
183e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  case Type::FunctionTyID: {
184a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const FunctionType *DstFT = cast<FunctionType>(DstTy);
185a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const FunctionType *SrcFT = cast<FunctionType>(SrcTy);
1869ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner    if (DstFT->isVarArg() != SrcFT->isVarArg() ||
1879ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner        DstFT->getNumContainedTypes() != SrcFT->getNumContainedTypes())
18843f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner      return true;
189a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner
190a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    // Use TypeHolder's so recursive resolution won't break us.
191a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    PATypeHolder ST(SrcFT), DT(DstFT);
192a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    for (unsigned i = 0, e = DstFT->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::StructTyID: {
200a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const StructType *DstST = cast<StructType>(DstTy);
201a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const StructType *SrcST = cast<StructType>(SrcTy);
2029ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner    if (DstST->getNumContainedTypes() != SrcST->getNumContainedTypes())
203f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner      return true;
204a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner
205a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    PATypeHolder ST(SrcST), DT(DstST);
206a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    for (unsigned i = 0, e = DstST->getNumContainedTypes(); i != e; ++i) {
207a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner      const Type *SE = ST->getContainedType(i), *DE = DT->getContainedType(i);
208a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner      if (SE != DE && RecursiveResolveTypesI(DE, SE, Pointers))
209e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner        return true;
210a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    }
211e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner    return false;
212e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  }
213e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  case Type::ArrayTyID: {
214a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const ArrayType *DAT = cast<ArrayType>(DstTy);
215a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const ArrayType *SAT = cast<ArrayType>(SrcTy);
216e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner    if (DAT->getNumElements() != SAT->getNumElements()) return true;
217e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    return RecursiveResolveTypesI(DAT->getElementType(), SAT->getElementType(),
218bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner                                  Pointers);
219e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner  }
220f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner  case Type::VectorTyID: {
221a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const VectorType *DVT = cast<VectorType>(DstTy);
222a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const VectorType *SVT = cast<VectorType>(SrcTy);
223f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner    if (DVT->getNumElements() != SVT->getNumElements()) return true;
224f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner    return RecursiveResolveTypesI(DVT->getElementType(), SVT->getElementType(),
225f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner                                  Pointers);
226f6f4f7a149e2864cc0441afcbed5fd99ff4b9587Chris Lattner  }
227e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner  case Type::PointerTyID: {
228a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const PointerType *DstPT = cast<PointerType>(DstTy);
229a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner    const PointerType *SrcPT = cast<PointerType>(SrcTy);
2309ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner
2319ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner    if (DstPT->getAddressSpace() != SrcPT->getAddressSpace())
2329ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner      return true;
2339ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner
234e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // If this is a pointer type, check to see if we have already seen it.  If
235e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // so, we are in a recursive branch.  Cut off the search now.  We cannot use
236e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // an associative container for this search, because the type pointers (keys
23762a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    // in the container) change whenever types get resolved.
23862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    if (SrcPT->isAbstract())
23962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      if (const Type *ExistingDestTy = Pointers.lookup(SrcPT))
24062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner        return ExistingDestTy != DstPT;
24162a81a1eba019ab570b002f8e1686494139785a1Chris Lattner
24262a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    if (DstPT->isAbstract())
24362a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      if (const Type *ExistingSrcTy = Pointers.lookup(DstPT))
24462a81a1eba019ab570b002f8e1686494139785a1Chris Lattner        return ExistingSrcTy != SrcPT;
245e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // Otherwise, add the current pointers to the vector to stop recursion on
246e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // this pair.
24762a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    if (DstPT->isAbstract())
24862a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      Pointers.insert(DstPT, SrcPT);
24962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner    if (SrcPT->isAbstract())
25062a81a1eba019ab570b002f8e1686494139785a1Chris Lattner      Pointers.insert(SrcPT, DstPT);
251a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner
2529ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner    return RecursiveResolveTypesI(DstPT->getElementType(),
2539ddf2c898f5227ca137aef7abc3051e16bd60025Chris Lattner                                  SrcPT->getElementType(), Pointers);
254e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  }
255f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman  }
256e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner}
257e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
258a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattnerstatic bool RecursiveResolveTypes(const Type *DestTy, const Type *SrcTy) {
25962a81a1eba019ab570b002f8e1686494139785a1Chris Lattner  LinkerTypeMap PointerTypes;
260bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner  return RecursiveResolveTypesI(DestTy, SrcTy, PointerTypes);
261e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner}
262e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner
263e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
2642c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner// LinkTypes - Go through the symbol table of the Src module and see if any
2652c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner// types are named in the src module that are not named in the Dst module.
2662c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner// Make sure there are no type name conflicts.
2675c2d335d86579941b71816196a95a64b277e8963Chris Lattnerstatic bool LinkTypes(Module *Dest, const Module *Src, std::string *Err) {
26878d033e086e19e016273de014f9214aa6f3f844bReid Spencer        TypeSymbolTable *DestST = &Dest->getTypeSymbolTable();
26978d033e086e19e016273de014f9214aa6f3f844bReid Spencer  const TypeSymbolTable *SrcST  = &Src->getTypeSymbolTable();
2702c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
2712c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  // Look for a type plane for Type's...
27278d033e086e19e016273de014f9214aa6f3f844bReid Spencer  TypeSymbolTable::const_iterator TI = SrcST->begin();
27378d033e086e19e016273de014f9214aa6f3f844bReid Spencer  TypeSymbolTable::const_iterator TE = SrcST->end();
274567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer  if (TI == TE) return false;  // No named types, do nothing.
2752c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
276cf00c4ab3ba308d45d98c5ccab87362cf802facbMisha Brukman  // Some types cannot be resolved immediately because they depend on other
277cf00c4ab3ba308d45d98c5ccab87362cf802facbMisha Brukman  // types being resolved to each other first.  This contains a list of types we
278cf00c4ab3ba308d45d98c5ccab87362cf802facbMisha Brukman  // are waiting to recheck.
2794c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  std::vector<std::string> DelayedTypesToResolve;
2804c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
281567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer  for ( ; TI != TE; ++TI ) {
282567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer    const std::string &Name = TI->first;
283c28a224c68adf0464c275778af4bce762ad25d37Reid Spencer    const Type *RHS = TI->second;
2842c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
285bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner    // Check to see if this type name is already in the dest module.
28678d033e086e19e016273de014f9214aa6f3f844bReid Spencer    Type *Entry = DestST->lookup(Name);
2874c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
288bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner    // If the name is just in the source module, bring it over to the dest.
289bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner    if (Entry == 0) {
290bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner      if (!Name.empty())
291bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner        DestST->insert(Name, const_cast<Type*>(RHS));
292bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner    } else if (ResolveTypes(Entry, RHS)) {
2934c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      // They look different, save the types 'till later to resolve.
2944c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      DelayedTypesToResolve.push_back(Name);
2954c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    }
2964c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  }
2974c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
2984c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  // Iteratively resolve types while we can...
2994c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  while (!DelayedTypesToResolve.empty()) {
3004c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    // Loop over all of the types, attempting to resolve them if possible...
3014c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    unsigned OldSize = DelayedTypesToResolve.size();
3024c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
303e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner    // Try direct resolution by name...
3044c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    for (unsigned i = 0; i != DelayedTypesToResolve.size(); ++i) {
3054c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      const std::string &Name = DelayedTypesToResolve[i];
30678d033e086e19e016273de014f9214aa6f3f844bReid Spencer      Type *T1 = SrcST->lookup(Name);
30778d033e086e19e016273de014f9214aa6f3f844bReid Spencer      Type *T2 = DestST->lookup(Name);
308bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner      if (!ResolveTypes(T2, T1)) {
3094c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner        // We are making progress!
3104c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner        DelayedTypesToResolve.erase(DelayedTypesToResolve.begin()+i);
3114c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner        --i;
3122f6bb2bce13084dfc4c8acf47f9eae0578267aaaChris Lattner      }
3134c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    }
3142f6bb2bce13084dfc4c8acf47f9eae0578267aaaChris Lattner
3154c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    // Did we not eliminate any types?
3164c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    if (DelayedTypesToResolve.size() == OldSize) {
317e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      // Attempt to resolve subelements of types.  This allows us to merge these
318e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      // two types: { int* } and { opaque* }
3194c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      for (unsigned i = 0, e = DelayedTypesToResolve.size(); i != e; ++i) {
3204c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner        const std::string &Name = DelayedTypesToResolve[i];
321a4477f9b31ce0b6fadc5365ff9355679c1ecb954Chris Lattner        if (!RecursiveResolveTypes(SrcST->lookup(Name), DestST->lookup(Name))) {
322e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          // We are making progress!
323e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          DelayedTypesToResolve.erase(DelayedTypesToResolve.begin()+i);
324f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
325e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          // Go back to the main loop, perhaps we can resolve directly by name
326e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          // now...
327e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          break;
328e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner        }
329e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      }
330e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
331e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      // If we STILL cannot resolve the types, then there is something wrong.
332e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      if (DelayedTypesToResolve.size() == OldSize) {
333aeb18ce9c012f8554f48bae9bb32351c6eeaa26cChris Lattner        // Remove the symbol name from the destination.
334aeb18ce9c012f8554f48bae9bb32351c6eeaa26cChris Lattner        DelayedTypesToResolve.pop_back();
3354c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      }
3362c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner    }
3372c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  }
3384c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
3394c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
3402c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  return false;
3412c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner}
3422c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
3430bb8757997392ee18ca9de0f1fab29463f8e7aebChris Lattner#ifndef NDEBUG
3445c2d335d86579941b71816196a95a64b277e8963Chris Lattnerstatic void PrintMap(const std::map<const Value*, Value*> &M) {
3455c2d335d86579941b71816196a95a64b277e8963Chris Lattner  for (std::map<const Value*, Value*>::const_iterator I = M.begin(), E =M.end();
3462d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner       I != E; ++I) {
347e81561909d128c6e2d8033cb5465a49b2596b26aBill Wendling    cerr << " Fr: " << (void*)I->first << " ";
34887182ae6baf28ca6da6ee4bc00cf1857f293acfeChris Lattner    I->first->dump();
349e81561909d128c6e2d8033cb5465a49b2596b26aBill Wendling    cerr << " To: " << (void*)I->second << " ";
35087182ae6baf28ca6da6ee4bc00cf1857f293acfeChris Lattner    I->second->dump();
351e81561909d128c6e2d8033cb5465a49b2596b26aBill Wendling    cerr << "\n";
3522d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner  }
3532d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner}
3540bb8757997392ee18ca9de0f1fab29463f8e7aebChris Lattner#endif
3552d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner
3562d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner
357619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer// RemapOperand - Use ValueMap to convert constants from one module to another.
3585c2d335d86579941b71816196a95a64b277e8963Chris Lattnerstatic Value *RemapOperand(const Value *In,
3590033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner                           std::map<const Value*, Value*> &ValueMap) {
3600033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner  std::map<const Value*,Value*>::const_iterator I = ValueMap.find(In);
361ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  if (I != ValueMap.end())
362ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer    return I->second;
3635c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
364619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer  // Check to see if it's a constant that we are interested in transforming.
365620fd68b1086b4d99b23951357299bee357aa62aChris Lattner  Value *Result = 0;
36618961504fc2b299578dba817900a0696cf3ccc4dChris Lattner  if (const Constant *CPV = dyn_cast<Constant>(In)) {
367de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner    if ((!isa<DerivedType>(CPV->getType()) && !isa<ConstantExpr>(CPV)) ||
368a54b7cbd452b3adb2f51346140d996b29c2cdb30Reid Spencer        isa<ConstantInt>(CPV) || isa<ConstantAggregateZero>(CPV))
3690033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner      return const_cast<Constant*>(CPV);   // Simple constants stay identical.
3708d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner
37118961504fc2b299578dba817900a0696cf3ccc4dChris Lattner    if (const ConstantArray *CPA = dyn_cast<ConstantArray>(CPV)) {
372cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos      std::vector<Constant*> Operands(CPA->getNumOperands());
373cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos      for (unsigned i = 0, e = CPA->getNumOperands(); i != e; ++i)
3740033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner        Operands[i] =cast<Constant>(RemapOperand(CPA->getOperand(i), ValueMap));
375e9bb2df410f7a22decad9a883f7139d5857c1520Chris Lattner      Result = ConstantArray::get(cast<ArrayType>(CPA->getType()), Operands);
37618961504fc2b299578dba817900a0696cf3ccc4dChris Lattner    } else if (const ConstantStruct *CPS = dyn_cast<ConstantStruct>(CPV)) {
377cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos      std::vector<Constant*> Operands(CPS->getNumOperands());
378cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos      for (unsigned i = 0, e = CPS->getNumOperands(); i != e; ++i)
3790033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner        Operands[i] =cast<Constant>(RemapOperand(CPS->getOperand(i), ValueMap));
380e9bb2df410f7a22decad9a883f7139d5857c1520Chris Lattner      Result = ConstantStruct::get(cast<StructType>(CPS->getType()), Operands);
381b976e668165e1875a8f1eb7af800e33bb1e4393dChris Lattner    } else if (isa<ConstantPointerNull>(CPV) || isa<UndefValue>(CPV)) {
38218961504fc2b299578dba817900a0696cf3ccc4dChris Lattner      Result = const_cast<Constant*>(CPV);
3839d6565a5b1fbc4286d6ee638d8f47a3171a9ed7eReid Spencer    } else if (const ConstantVector *CP = dyn_cast<ConstantVector>(CPV)) {
384a88eb9287eb14ba62ca6324b2b249e0ae9fe3309Chris Lattner      std::vector<Constant*> Operands(CP->getNumOperands());
385a88eb9287eb14ba62ca6324b2b249e0ae9fe3309Chris Lattner      for (unsigned i = 0, e = CP->getNumOperands(); i != e; ++i)
386a88eb9287eb14ba62ca6324b2b249e0ae9fe3309Chris Lattner        Operands[i] = cast<Constant>(RemapOperand(CP->getOperand(i), ValueMap));
3879d6565a5b1fbc4286d6ee638d8f47a3171a9ed7eReid Spencer      Result = ConstantVector::get(Operands);
3886cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner    } else if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(CPV)) {
38927d672136a57519095a3ef20161697749cffb57cChris Lattner      std::vector<Constant*> Ops;
39027d672136a57519095a3ef20161697749cffb57cChris Lattner      for (unsigned i = 0, e = CE->getNumOperands(); i != e; ++i)
39127d672136a57519095a3ef20161697749cffb57cChris Lattner        Ops.push_back(cast<Constant>(RemapOperand(CE->getOperand(i),ValueMap)));
39227d672136a57519095a3ef20161697749cffb57cChris Lattner      Result = CE->getWithOperands(Ops);
3938d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner    } else {
3940bb8757997392ee18ca9de0f1fab29463f8e7aebChris Lattner      assert(!isa<GlobalValue>(CPV) && "Unmapped global?");
3958d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      assert(0 && "Unknown type of derived type constant value!");
3968d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner    }
397620fd68b1086b4d99b23951357299bee357aa62aChris Lattner  } else if (isa<InlineAsm>(In)) {
398620fd68b1086b4d99b23951357299bee357aa62aChris Lattner    Result = const_cast<Value*>(In);
399620fd68b1086b4d99b23951357299bee357aa62aChris Lattner  }
400620fd68b1086b4d99b23951357299bee357aa62aChris Lattner
401619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer  // Cache the mapping in our local map structure
402620fd68b1086b4d99b23951357299bee357aa62aChris Lattner  if (Result) {
403817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov    ValueMap[In] = Result;
4048d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner    return Result;
4058d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  }
4068bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer
4070bb8757997392ee18ca9de0f1fab29463f8e7aebChris Lattner#ifndef NDEBUG
408e81561909d128c6e2d8033cb5465a49b2596b26aBill Wendling  cerr << "LinkModules ValueMap: \n";
4090033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner  PrintMap(ValueMap);
4102d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner
411e81561909d128c6e2d8033cb5465a49b2596b26aBill Wendling  cerr << "Couldn't remap value: " << (void*)In << " " << *In << "\n";
4128d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  assert(0 && "Couldn't remap value!");
4130bb8757997392ee18ca9de0f1fab29463f8e7aebChris Lattner#endif
4148d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  return 0;
4155c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
4165c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
4178bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer/// ForceRenaming - The LLVM SymbolTable class autorenames globals that conflict
4188bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer/// in the symbol table.  This is good for all clients except for us.  Go
4198bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer/// through the trouble to force this back.
420c003628a612d3687fb77088a5894314210a65385Chris Lattnerstatic void ForceRenaming(GlobalValue *GV, const std::string &Name) {
421c003628a612d3687fb77088a5894314210a65385Chris Lattner  assert(GV->getName() != Name && "Can't force rename to self");
422ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  ValueSymbolTable &ST = GV->getParent()->getValueSymbolTable();
423c003628a612d3687fb77088a5894314210a65385Chris Lattner
424c003628a612d3687fb77088a5894314210a65385Chris Lattner  // If there is a conflict, rename the conflict.
42533f294930e0affa51c3818405741abd80cc90316Chris Lattner  if (GlobalValue *ConflictGV = cast_or_null<GlobalValue>(ST.lookup(Name))) {
426ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer    assert(ConflictGV->hasInternalLinkage() &&
427ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer           "Not conflicting with a static global, should link instead!");
42833f294930e0affa51c3818405741abd80cc90316Chris Lattner    GV->takeName(ConflictGV);
42933f294930e0affa51c3818405741abd80cc90316Chris Lattner    ConflictGV->setName(Name);    // This will cause ConflictGV to get renamed
430ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer    assert(ConflictGV->getName() != Name && "ForceRenaming didn't work");
43133f294930e0affa51c3818405741abd80cc90316Chris Lattner  } else {
43233f294930e0affa51c3818405741abd80cc90316Chris Lattner    GV->setName(Name);              // Force the name back
433ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  }
434ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer}
435ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer
436ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer/// CopyGVAttributes - copy additional attributes (those not needed to construct
437ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer/// a GlobalValue) from the SrcGV to the DestGV.
438ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencerstatic void CopyGVAttributes(GlobalValue *DestGV, const GlobalValue *SrcGV) {
43928c3cff8250b3fe2adc6479306fe7dbdb48a1bdbDuncan Sands  // Use the maximum alignment, rather than just copying the alignment of SrcGV.
44028c3cff8250b3fe2adc6479306fe7dbdb48a1bdbDuncan Sands  unsigned Alignment = std::max(DestGV->getAlignment(), SrcGV->getAlignment());
44128c3cff8250b3fe2adc6479306fe7dbdb48a1bdbDuncan Sands  DestGV->copyAttributesFrom(SrcGV);
44228c3cff8250b3fe2adc6479306fe7dbdb48a1bdbDuncan Sands  DestGV->setAlignment(Alignment);
443c003628a612d3687fb77088a5894314210a65385Chris Lattner}
444c003628a612d3687fb77088a5894314210a65385Chris Lattner
445aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// GetLinkageResult - This analyzes the two global values and determines what
446aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// the result will look like in the destination module.  In particular, it
447aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// computes the resultant linkage type, computes whether the global in the
448aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// source should be copied over to the destination (replacing the existing
4499cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov/// one), and computes whether this linkage is an error or not. It also performs
4509cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov/// visibility checks: we cannot link together two symbols with different
4519cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov/// visibilities.
452968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikovstatic bool GetLinkageResult(GlobalValue *Dest, const GlobalValue *Src,
453aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner                             GlobalValue::LinkageTypes &LT, bool &LinkFromSrc,
454aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner                             std::string *Err) {
455aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  assert((!Dest || !Src->hasInternalLinkage()) &&
456aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner         "If Src has internal linkage, Dest shouldn't be set!");
457aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  if (!Dest) {
458aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    // Linking something to nothing.
459aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LinkFromSrc = true;
460aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LT = Src->getLinkage();
4615cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer  } else if (Src->isDeclaration()) {
4622b48ef0450944c2c46633aec9baf6be835a3b503Anton Korobeynikov    // If Src is external or if both Src & Dest are external..  Just link the
463aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    // external globals, we aren't adding anything.
464b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov    if (Src->hasDLLImportLinkage()) {
46578ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      // If one of GVs has DLLImport linkage, result should be dllimport'ed.
4665cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer      if (Dest->isDeclaration()) {
467b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov        LinkFromSrc = true;
468b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov        LT = Src->getLinkage();
469b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov      }
4708753c447a5f8eec1421331c40ffbb72c6d411f78Andrew Lenharth    } else if (Dest->hasExternalWeakLinkage()) {
4718753c447a5f8eec1421331c40ffbb72c6d411f78Andrew Lenharth      //If the Dest is weak, use the source linkage
4728753c447a5f8eec1421331c40ffbb72c6d411f78Andrew Lenharth      LinkFromSrc = true;
4738753c447a5f8eec1421331c40ffbb72c6d411f78Andrew Lenharth      LT = Src->getLinkage();
474b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov    } else {
475b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov      LinkFromSrc = false;
476b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov      LT = Dest->getLinkage();
477b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov    }
4785cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer  } else if (Dest->isDeclaration() && !Dest->hasDLLImportLinkage()) {
479aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    // If Dest is external but Src is not:
480aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LinkFromSrc = true;
481aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LT = Src->getLinkage();
482aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  } else if (Src->hasAppendingLinkage() || Dest->hasAppendingLinkage()) {
483aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    if (Src->getLinkage() != Dest->getLinkage())
484aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      return Error(Err, "Linking globals named '" + Src->getName() +
485aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner            "': can only link appending global with another appending global!");
486aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LinkFromSrc = true; // Special cased.
487aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LT = Src->getLinkage();
48880585f1daac3570f19cd255c5006dfcf2ff65a89Anton Korobeynikov  } else if (Src->isWeakForLinker()) {
489aafce77b17d340aace52bcd49d1944109d82f14aDale Johannesen    // At this point we know that Dest has LinkOnce, External*, Weak, Common,
490aafce77b17d340aace52bcd49d1944109d82f14aDale Johannesen    // or DLL* linkage.
49180585f1daac3570f19cd255c5006dfcf2ff65a89Anton Korobeynikov    if ((Dest->hasLinkOnceLinkage() &&
492aafce77b17d340aace52bcd49d1944109d82f14aDale Johannesen          (Src->hasWeakLinkage() || Src->hasCommonLinkage())) ||
49378ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov        Dest->hasExternalWeakLinkage()) {
494aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      LinkFromSrc = true;
495aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      LT = Src->getLinkage();
496aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    } else {
497aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      LinkFromSrc = false;
498aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      LT = Dest->getLinkage();
499aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    }
50080585f1daac3570f19cd255c5006dfcf2ff65a89Anton Korobeynikov  } else if (Dest->isWeakForLinker()) {
50178ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov    // At this point we know that Src has External* or DLL* linkage.
50278ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov    if (Src->hasExternalWeakLinkage()) {
50378ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      LinkFromSrc = false;
50478ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      LT = Dest->getLinkage();
50578ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov    } else {
50678ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      LinkFromSrc = true;
50778ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      LT = GlobalValue::ExternalLinkage;
50878ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov    }
509aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  } else {
510b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov    assert((Dest->hasExternalLinkage() ||
511b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov            Dest->hasDLLImportLinkage() ||
51278ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov            Dest->hasDLLExportLinkage() ||
51378ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov            Dest->hasExternalWeakLinkage()) &&
514b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov           (Src->hasExternalLinkage() ||
515b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov            Src->hasDLLImportLinkage() ||
51678ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov            Src->hasDLLExportLinkage() ||
51778ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov            Src->hasExternalWeakLinkage()) &&
518aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner           "Unexpected linkage type!");
519f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman    return Error(Err, "Linking globals named '" + Src->getName() +
520aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner                 "': symbol multiply defined!");
521aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  }
5229cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov
5239cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov  // Check visibility
5249cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov  if (Dest && Src->getVisibility() != Dest->getVisibility())
52597f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner    if (!Src->isDeclaration() && !Dest->isDeclaration())
52697f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner      return Error(Err, "Linking globals named '" + Src->getName() +
52797f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner                   "': symbols have different visibilities!");
528aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  return false;
529aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner}
5305c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
5315c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// LinkGlobals - Loop through the global variables in the src module and merge
5328166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner// them into the dest module.
533968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikovstatic bool LinkGlobals(Module *Dest, const Module *Src,
5345c2d335d86579941b71816196a95a64b277e8963Chris Lattner                        std::map<const Value*, Value*> &ValueMap,
5358166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                    std::multimap<std::string, GlobalVariable *> &AppendingVars,
5365c2d335d86579941b71816196a95a64b277e8963Chris Lattner                        std::string *Err) {
5375c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  // Loop over all of the globals in the src module, mapping them over as we go
5380bb8757997392ee18ca9de0f1fab29463f8e7aebChris Lattner  for (Module::const_global_iterator I = Src->global_begin(),
5390bb8757997392ee18ca9de0f1fab29463f8e7aebChris Lattner       E = Src->global_end(); I != E; ++I) {
540968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov    const GlobalVariable *SGV = I;
54101f69399352380532d480f6749daed660f50b15cAnton Korobeynikov    GlobalValue *DGV = 0;
54201f69399352380532d480f6749daed660f50b15cAnton Korobeynikov
54301f69399352380532d480f6749daed660f50b15cAnton Korobeynikov    // Check to see if may have to link the global with the global
544ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer    if (SGV->hasName() && !SGV->hasInternalLinkage()) {
545ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      DGV = Dest->getGlobalVariable(SGV->getName());
546ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      if (DGV && DGV->getType() != SGV->getType())
547ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // If types don't agree due to opaque types, try to resolve them.
548bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner        RecursiveResolveTypes(SGV->getType(), DGV->getType());
549ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer    }
5505c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
55101f69399352380532d480f6749daed660f50b15cAnton Korobeynikov    // Check to see if may have to link the global with the alias
552aeb09967fdb2ff074c91ce8039ebd1917a2f6060Anton Korobeynikov    if (!DGV && SGV->hasName() && !SGV->hasInternalLinkage()) {
55301f69399352380532d480f6749daed660f50b15cAnton Korobeynikov      DGV = Dest->getNamedAlias(SGV->getName());
55401f69399352380532d480f6749daed660f50b15cAnton Korobeynikov      if (DGV && DGV->getType() != SGV->getType())
55501f69399352380532d480f6749daed660f50b15cAnton Korobeynikov        // If types don't agree due to opaque types, try to resolve them.
556bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner        RecursiveResolveTypes(SGV->getType(), DGV->getType());
55701f69399352380532d480f6749daed660f50b15cAnton Korobeynikov    }
55801f69399352380532d480f6749daed660f50b15cAnton Korobeynikov
559aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    if (DGV && DGV->hasInternalLinkage())
560aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      DGV = 0;
561aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner
562c318329a1098a19137398af4d880f4d8cdd9cd8dDan Gohman    assert((SGV->hasInitializer() || SGV->hasExternalWeakLinkage() ||
563c318329a1098a19137398af4d880f4d8cdd9cd8dDan Gohman            SGV->hasExternalLinkage() || SGV->hasDLLImportLinkage()) &&
5644ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner           "Global must either be external or have an initializer!");
5655c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
566b324bd735ff5c99a52cd3c3f5d01c0e1398a2d3aChris Lattner    GlobalValue::LinkageTypes NewLinkage = GlobalValue::InternalLinkage;
567b324bd735ff5c99a52cd3c3f5d01c0e1398a2d3aChris Lattner    bool LinkFromSrc = false;
568aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    if (GetLinkageResult(DGV, SGV, NewLinkage, LinkFromSrc, Err))
569aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      return true;
5700fec08eb58dd9fffeb72c584aa61a59d71111c8dChris Lattner
571aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    if (!DGV) {
5725c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner      // No linking to be performed, simply create an identical version of the
5738d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      // symbol over in the dest module... the initializer will be filled in
5748d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      // later by LinkGlobalInits...
5752719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      GlobalVariable *NewDGV =
5762719bac90525985f4ebb97324b4c087d47b54384Chris Lattner        new GlobalVariable(SGV->getType()->getElementType(),
5772719bac90525985f4ebb97324b4c087d47b54384Chris Lattner                           SGV->isConstant(), SGV->getLinkage(), /*init*/0,
578a534b0f7bf0248bf90afe97110a34777e972f325Chris Lattner                           SGV->getName(), Dest, false,
579a534b0f7bf0248bf90afe97110a34777e972f325Chris Lattner                           SGV->getType()->getAddressSpace());
580471feac1b297b9458de18d973af371bf7f10296bReid Spencer      // Propagate alignment, visibility and section info.
581ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      CopyGVAttributes(NewDGV, SGV);
5825dfbaf1fe3fa6914095cc55cde20b1ed25187b56Andrew Lenharth
5832719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      // If the LLVM runtime renamed the global, but it is an externally visible
5842719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      // symbol, DGV must be an existing global with internal linkage.  Rename
5852719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      // it.
586c003628a612d3687fb77088a5894314210a65385Chris Lattner      if (NewDGV->getName() != SGV->getName() && !NewDGV->hasInternalLinkage())
587c003628a612d3687fb77088a5894314210a65385Chris Lattner        ForceRenaming(NewDGV, SGV->getName());
5885c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
5895c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner      // Make sure to remember this mapping...
590817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov      ValueMap[SGV] = NewDGV;
591817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov
5928166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      if (SGV->hasAppendingLinkage())
5938166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        // Keep track that this is an appending variable...
5948166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        AppendingVars.insert(std::make_pair(SGV->getName(), NewDGV));
595aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    } else if (DGV->hasAppendingLinkage()) {
5968166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // No linking is performed yet.  Just insert a new copy of the global, and
5978166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // keep track of the fact that it is an appending variable in the
5988166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // AppendingVars map.  The name is cleared out so that no linkage is
5998166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // performed.
6008166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      GlobalVariable *NewDGV =
6018166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        new GlobalVariable(SGV->getType()->getElementType(),
6028166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                           SGV->isConstant(), SGV->getLinkage(), /*init*/0,
603a534b0f7bf0248bf90afe97110a34777e972f325Chris Lattner                           "", Dest, false,
604a534b0f7bf0248bf90afe97110a34777e972f325Chris Lattner                           SGV->getType()->getAddressSpace());
6058166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
60675c7915c3a24f8063c274ccffe2dc9b7bc8bce1fAnton Korobeynikov      // Set alignment allowing CopyGVAttributes merge it with alignment of SGV.
607ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      NewDGV->setAlignment(DGV->getAlignment());
60875c7915c3a24f8063c274ccffe2dc9b7bc8bce1fAnton Korobeynikov      // Propagate alignment, section and visibility info.
609ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      CopyGVAttributes(NewDGV, SGV);
6105dfbaf1fe3fa6914095cc55cde20b1ed25187b56Andrew Lenharth
6118166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Make sure to remember this mapping...
612817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov      ValueMap[SGV] = NewDGV;
6138166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
6148166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Keep track that this is an appending variable...
6158166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      AppendingVars.insert(std::make_pair(SGV->getName(), NewDGV));
61601f69399352380532d480f6749daed660f50b15cAnton Korobeynikov    } else if (GlobalAlias *DGA = dyn_cast<GlobalAlias>(DGV)) {
61701f69399352380532d480f6749daed660f50b15cAnton Korobeynikov      // SGV is global, but DGV is alias. The only valid mapping is when SGV is
61801f69399352380532d480f6749daed660f50b15cAnton Korobeynikov      // external declaration, which is effectively a no-op. Also make sure
619817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov      // linkage calculation was correct.
62001f69399352380532d480f6749daed660f50b15cAnton Korobeynikov      if (SGV->isDeclaration() && !LinkFromSrc) {
62101f69399352380532d480f6749daed660f50b15cAnton Korobeynikov        // Make sure to remember this mapping...
622817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov        ValueMap[SGV] = DGA;
62301f69399352380532d480f6749daed660f50b15cAnton Korobeynikov      } else
6241438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov        return Error(Err, "Global-Alias Collision on '" + SGV->getName() +
6251438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov                     "': symbol multiple defined");
62601f69399352380532d480f6749daed660f50b15cAnton Korobeynikov    } else if (GlobalVariable *DGVar = dyn_cast<GlobalVariable>(DGV)) {
627817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov      // Otherwise, perform the global-global mapping as instructed by
628817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov      // GetLinkageResult.
629aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      if (LinkFromSrc) {
630968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov        // Propagate alignment, section, and visibility info.
63101f69399352380532d480f6749daed660f50b15cAnton Korobeynikov        CopyGVAttributes(DGVar, SGV);
632968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov
633968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov        // If the types don't match, and if we are to link from the source, nuke
634968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov        // DGV and create a new one of the appropriate type.
63501f69399352380532d480f6749daed660f50b15cAnton Korobeynikov        if (SGV->getType() != DGVar->getType()) {
636968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov          GlobalVariable *NewDGV =
637968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov            new GlobalVariable(SGV->getType()->getElementType(),
63801f69399352380532d480f6749daed660f50b15cAnton Korobeynikov                               DGVar->isConstant(), DGVar->getLinkage(),
639a534b0f7bf0248bf90afe97110a34777e972f325Chris Lattner                               /*init*/0, DGVar->getName(), Dest, false,
640a534b0f7bf0248bf90afe97110a34777e972f325Chris Lattner                               SGV->getType()->getAddressSpace());
64101f69399352380532d480f6749daed660f50b15cAnton Korobeynikov          CopyGVAttributes(NewDGV, DGVar);
642968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov          DGV->replaceAllUsesWith(ConstantExpr::getBitCast(NewDGV,
64301f69399352380532d480f6749daed660f50b15cAnton Korobeynikov                                                           DGVar->getType()));
64401f69399352380532d480f6749daed660f50b15cAnton Korobeynikov          // DGVar will conflict with NewDGV because they both had the same
645968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov          // name. We must erase this now so ForceRenaming doesn't assert
646968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov          // because DGV might not have internal linkage.
64701f69399352380532d480f6749daed660f50b15cAnton Korobeynikov          DGVar->eraseFromParent();
648968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov
649968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov          // If the symbol table renamed the global, but it is an externally
650968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov          // visible symbol, DGV must be an existing global with internal
651968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov          // linkage. Rename it.
652968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov          if (NewDGV->getName() != SGV->getName() &&
653968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov              !NewDGV->hasInternalLinkage())
654968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov            ForceRenaming(NewDGV, SGV->getName());
655968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov
65601f69399352380532d480f6749daed660f50b15cAnton Korobeynikov          DGVar = NewDGV;
657968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov        }
658968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov
659aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner        // Inherit const as appropriate
66001f69399352380532d480f6749daed660f50b15cAnton Korobeynikov        DGVar->setConstant(SGV->isConstant());
661968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov
662968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov        // Set initializer to zero, so we can link the stuff later
66301f69399352380532d480f6749daed660f50b15cAnton Korobeynikov        DGVar->setInitializer(0);
664aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      } else {
665968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov        // Special case for const propagation
66601f69399352380532d480f6749daed660f50b15cAnton Korobeynikov        if (DGVar->isDeclaration() && SGV->isConstant() && !DGVar->isConstant())
66701f69399352380532d480f6749daed660f50b15cAnton Korobeynikov          DGVar->setConstant(true);
668aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      }
669aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner
670968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov      // Set calculated linkage
67101f69399352380532d480f6749daed660f50b15cAnton Korobeynikov      DGVar->setLinkage(NewLinkage);
672968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov
673968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov      // Make sure to remember this mapping...
674817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov      ValueMap[SGV] = ConstantExpr::getBitCast(DGVar, SGV->getType());
6755c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner    }
6765c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  }
6775c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return false;
6785c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
6795c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
68058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikovstatic GlobalValue::LinkageTypes
68158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton KorobeynikovCalculateAliasLinkage(const GlobalValue *SGV, const GlobalValue *DGV) {
68258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov  if (SGV->hasExternalLinkage() || DGV->hasExternalLinkage())
68358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    return GlobalValue::ExternalLinkage;
68458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov  else if (SGV->hasWeakLinkage() || DGV->hasWeakLinkage())
68558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    return GlobalValue::WeakLinkage;
68658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov  else {
68758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    assert(SGV->hasInternalLinkage() && DGV->hasInternalLinkage() &&
68858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov           "Unexpected linkage type");
68958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    return GlobalValue::InternalLinkage;
69058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov  }
69158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov}
69258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
69331ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio// LinkAlias - Loop through the alias in the src module and link them into the
69458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov// dest module. We're assuming, that all functions/global variables were already
69558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov// linked in.
6964fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikovstatic bool LinkAlias(Module *Dest, const Module *Src,
6974fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov                      std::map<const Value*, Value*> &ValueMap,
6984fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov                      std::string *Err) {
69931ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio  // Loop over all alias in the src module
70031ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio  for (Module::const_alias_iterator I = Src->alias_begin(),
70131ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio         E = Src->alias_end(); I != E; ++I) {
70258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    const GlobalAlias *SGA = I;
70358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    const GlobalValue *SAliasee = SGA->getAliasedGlobal();
70458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    GlobalAlias *NewGA = NULL;
70558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
70658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    // Globals were already linked, thus we can just query ValueMap for variant
707caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    // of SAliasee in Dest.
70858d5e051574339106ce4139697ebadd21e34dbc0Ted Kremenek    std::map<const Value*,Value*>::const_iterator VMI = ValueMap.find(SAliasee);
70958d5e051574339106ce4139697ebadd21e34dbc0Ted Kremenek    assert(VMI != ValueMap.end() && "Aliasee not linked");
71058d5e051574339106ce4139697ebadd21e34dbc0Ted Kremenek    GlobalValue* DAliasee = cast<GlobalValue>(VMI->second);
711caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    GlobalValue* DGV = NULL;
71258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
71358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    // Try to find something 'similar' to SGA in destination module.
714caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    if (!DGV && !SGA->hasInternalLinkage()) {
715caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      DGV = Dest->getNamedAlias(SGA->getName());
716caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov
717caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      // If types don't agree due to opaque types, try to resolve them.
718caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      if (DGV && DGV->getType() != SGA->getType())
7195ed2ba26680f06846e7602ce65432b0a723b1273Chris Lattner        RecursiveResolveTypes(SGA->getType(), DGV->getType());
720caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    }
721caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov
722caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    if (!DGV && !SGA->hasInternalLinkage()) {
723caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      DGV = Dest->getGlobalVariable(SGA->getName());
724caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov
72558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      // If types don't agree due to opaque types, try to resolve them.
726caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      if (DGV && DGV->getType() != SGA->getType())
7275ed2ba26680f06846e7602ce65432b0a723b1273Chris Lattner        RecursiveResolveTypes(SGA->getType(), DGV->getType());
728caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    }
72958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
730caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    if (!DGV && !SGA->hasInternalLinkage()) {
731caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      DGV = Dest->getFunction(SGA->getName());
732caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov
733caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      // If types don't agree due to opaque types, try to resolve them.
734caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      if (DGV && DGV->getType() != SGA->getType())
7355ed2ba26680f06846e7602ce65432b0a723b1273Chris Lattner        RecursiveResolveTypes(SGA->getType(), DGV->getType());
736caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    }
737caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov
738caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    // No linking to be performed on internal stuff.
739caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    if (DGV && DGV->hasInternalLinkage())
740caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      DGV = NULL;
741caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov
742caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    if (GlobalAlias *DGA = dyn_cast_or_null<GlobalAlias>(DGV)) {
743caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      // Types are known to be the same, check whether aliasees equal. As
74458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      // globals are already linked we just need query ValueMap to find the
74558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      // mapping.
74658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      if (DAliasee == DGA->getAliasedGlobal()) {
74758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // This is just two copies of the same alias. Propagate linkage, if
74858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // necessary.
74958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        DGA->setLinkage(CalculateAliasLinkage(SGA, DGA));
75058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
75158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        NewGA = DGA;
75258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // Proceed to 'common' steps
75358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      } else
7541438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov        return Error(Err, "Alias Collision on '"  + SGA->getName()+
7551438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov                     "': aliases have different aliasees");
756caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    } else if (GlobalVariable *DGVar = dyn_cast_or_null<GlobalVariable>(DGV)) {
757f88bc65932e5cf41ba6ca706cd1b1b1f65300248Anton Korobeynikov      // The only allowed way is to link alias with external declaration or weak
758f88bc65932e5cf41ba6ca706cd1b1b1f65300248Anton Korobeynikov      // symbol..
75980585f1daac3570f19cd255c5006dfcf2ff65a89Anton Korobeynikov      if (DGVar->isDeclaration() || DGVar->isWeakForLinker()) {
760ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov        // But only if aliasee is global too...
761ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov        if (!isa<GlobalVariable>(DAliasee))
762caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov          return Error(Err, "Global-Alias Collision on '" + SGA->getName() +
763caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov                       "': aliasee is not global variable");
764ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov
76558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        NewGA = new GlobalAlias(SGA->getType(), SGA->getLinkage(),
76658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov                                SGA->getName(), DAliasee, Dest);
76758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        CopyGVAttributes(NewGA, SGA);
76858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
76958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // Any uses of DGV need to change to NewGA, with cast, if needed.
770caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov        if (SGA->getType() != DGVar->getType())
771caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov          DGVar->replaceAllUsesWith(ConstantExpr::getBitCast(NewGA,
772caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov                                                             DGVar->getType()));
77358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        else
774caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov          DGVar->replaceAllUsesWith(NewGA);
77558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
776caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov        // DGVar will conflict with NewGA because they both had the same
77758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // name. We must erase this now so ForceRenaming doesn't assert
77858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // because DGV might not have internal linkage.
779caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov        DGVar->eraseFromParent();
78058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
78158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // Proceed to 'common' steps
78258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      } else
7831438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov        return Error(Err, "Global-Alias Collision on '" + SGA->getName() +
7841438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov                     "': symbol multiple defined");
785caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    } else if (Function *DF = dyn_cast_or_null<Function>(DGV)) {
786f88bc65932e5cf41ba6ca706cd1b1b1f65300248Anton Korobeynikov      // The only allowed way is to link alias with external declaration or weak
787f88bc65932e5cf41ba6ca706cd1b1b1f65300248Anton Korobeynikov      // symbol...
78880585f1daac3570f19cd255c5006dfcf2ff65a89Anton Korobeynikov      if (DF->isDeclaration() || DF->isWeakForLinker()) {
789ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov        // But only if aliasee is function too...
790ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov        if (!isa<Function>(DAliasee))
791caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov          return Error(Err, "Function-Alias Collision on '" + SGA->getName() +
792caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov                       "': aliasee is not function");
793ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov
794b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        NewGA = new GlobalAlias(SGA->getType(), SGA->getLinkage(),
795b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov                                SGA->getName(), DAliasee, Dest);
796b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        CopyGVAttributes(NewGA, SGA);
797b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov
798b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        // Any uses of DF need to change to NewGA, with cast, if needed.
799b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        if (SGA->getType() != DF->getType())
800b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov          DF->replaceAllUsesWith(ConstantExpr::getBitCast(NewGA,
801b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov                                                          DF->getType()));
802b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        else
803b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov          DF->replaceAllUsesWith(NewGA);
804b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov
805b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        // DF will conflict with NewGA because they both had the same
806b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        // name. We must erase this now so ForceRenaming doesn't assert
807b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        // because DF might not have internal linkage.
808b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        DF->eraseFromParent();
809b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov
810b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        // Proceed to 'common' steps
811b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov      } else
8121438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov        return Error(Err, "Function-Alias Collision on '" + SGA->getName() +
8131438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov                     "': symbol multiple defined");
81458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    } else {
815caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      // No linking to be performed, simply create an identical version of the
816caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov      // alias over in the dest module...
81758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
81858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      NewGA = new GlobalAlias(SGA->getType(), SGA->getLinkage(),
81958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov                              SGA->getName(), DAliasee, Dest);
82058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      CopyGVAttributes(NewGA, SGA);
82158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
82258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      // Proceed to 'common' steps
82358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    }
82458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
82558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    assert(NewGA && "No alias was created in destination module!");
82658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
827b8cdaf7099898d7c0eef4769bf6a89f39e7decc2Anton Korobeynikov    // If the symbol table renamed the alias, but it is an externally visible
828caa8ae8125124822b01ac21a8de186c13fcdd8f3Anton Korobeynikov    // symbol, DGA must be an global value with internal linkage. Rename it.
82958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    if (NewGA->getName() != SGA->getName() &&
83058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        !NewGA->hasInternalLinkage())
83158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      ForceRenaming(NewGA, SGA->getName());
83258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
83358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    // Remember this mapping so uses in the source module get remapped
83458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    // later by RemapOperand.
835817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov    ValueMap[SGA] = NewGA;
83631ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio  }
83758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
83831ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio  return false;
83931ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio}
84031ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio
8415c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
8428d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner// LinkGlobalInits - Update the initializers in the Dest module now that all
8438d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner// globals that may be referenced are in Dest.
8448d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattnerstatic bool LinkGlobalInits(Module *Dest, const Module *Src,
8455c2d335d86579941b71816196a95a64b277e8963Chris Lattner                            std::map<const Value*, Value*> &ValueMap,
8465c2d335d86579941b71816196a95a64b277e8963Chris Lattner                            std::string *Err) {
8478d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner
8488d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  // Loop over all of the globals in the src module, mapping them over as we go
8491127315562b221010040925980040fd4f65bdb1cChris Lattner  for (Module::const_global_iterator I = Src->global_begin(),
8501127315562b221010040925980040fd4f65bdb1cChris Lattner       E = Src->global_end(); I != E; ++I) {
85118961504fc2b299578dba817900a0696cf3ccc4dChris Lattner    const GlobalVariable *SGV = I;
8528d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner
8538d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner    if (SGV->hasInitializer()) {      // Only process initialized GV's
8548d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      // Figure out what the initializer looks like in the dest module...
8554ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner      Constant *SInit =
8560033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner        cast<Constant>(RemapOperand(SGV->getInitializer(), ValueMap));
8578d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner
8580b12ecf6ff6b5d3a144178257b6206f0c4788792Anton Korobeynikov      GlobalVariable *DGV =
8590b12ecf6ff6b5d3a144178257b6206f0c4788792Anton Korobeynikov        cast<GlobalVariable>(ValueMap[SGV]->stripPointerCasts());
8604ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner      if (DGV->hasInitializer()) {
8614ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner        if (SGV->hasExternalLinkage()) {
8624ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner          if (DGV->getInitializer() != SInit)
8631438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov            return Error(Err, "Global Variable Collision on '" + SGV->getName() +
8641438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov                         "': global variables have different initializers");
86580585f1daac3570f19cd255c5006dfcf2ff65a89Anton Korobeynikov        } else if (DGV->isWeakForLinker()) {
8664ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner          // Nothing is required, mapped values will take the new global
8674ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner          // automatically.
86880585f1daac3570f19cd255c5006dfcf2ff65a89Anton Korobeynikov        } else if (SGV->isWeakForLinker()) {
86957cb9883aa4c82357594d4186858f175135be3b1Chris Lattner          // Nothing is required, mapped values will take the new global
87057cb9883aa4c82357594d4186858f175135be3b1Chris Lattner          // automatically.
8714ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner        } else if (DGV->hasAppendingLinkage()) {
8724ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner          assert(0 && "Appending linkage unimplemented!");
8734ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner        } else {
8744ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner          assert(0 && "Unknown linkage!");
8754ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner        }
8768d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      } else {
8778d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner        // Copy the initializer over now...
8784ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner        DGV->setInitializer(SInit);
8798d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      }
8808d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner    }
8818d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  }
8828d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  return false;
8838d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner}
8845c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
88579df7c0aaa18129e55968c8783ef8346807bd4afChris Lattner// LinkFunctionProtos - Link the functions together between the two modules,
886c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// without doing function bodies... this just adds external function prototypes
887c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// to the Dest function...
8885c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner//
88979df7c0aaa18129e55968c8783ef8346807bd4afChris Lattnerstatic bool LinkFunctionProtos(Module *Dest, const Module *Src,
8905c2d335d86579941b71816196a95a64b277e8963Chris Lattner                               std::map<const Value*, Value*> &ValueMap,
8915c2d335d86579941b71816196a95a64b277e8963Chris Lattner                               std::string *Err) {
892619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer  // Loop over all of the functions in the src module, mapping them over
8935c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  for (Module::const_iterator I = Src->begin(), E = Src->end(); I != E; ++I) {
89418961504fc2b299578dba817900a0696cf3ccc4dChris Lattner    const Function *SF = I;   // SrcFunction
895824684982968244050e56357e736b9940c23e77dChris Lattner
896194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov    GlobalValue *DGV = 0;
897ec91ccba3ca92d5e1f19fb0cb6e57a7d4a3f3195Chris Lattner    Value *MappedDF;
898824684982968244050e56357e736b9940c23e77dChris Lattner
899824684982968244050e56357e736b9940c23e77dChris Lattner    // If this function is internal or has no name, it doesn't participate in
900824684982968244050e56357e736b9940c23e77dChris Lattner    // linkage.
9018bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer    if (SF->hasName() && !SF->hasInternalLinkage()) {
9028bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer      // Check to see if may have to link the function.
903194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov      DGV = Dest->getFunction(SF->getName());
9048bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer    }
905194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov
906194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov    // Check to see if may have to link the function with the alias
907194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov    if (!DGV && SF->hasName() && !SF->hasInternalLinkage()) {
908194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov      DGV = Dest->getNamedAlias(SF->getName());
909194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov      if (DGV && DGV->getType() != SF->getType())
910194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov        // If types don't agree due to opaque types, try to resolve them.
911194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov        RecursiveResolveTypes(SF->getType(), DGV->getType());
912194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov    }
913194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov
914194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov    if (DGV && DGV->hasInternalLinkage())
915194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov      DGV = 0;
916194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov
917824684982968244050e56357e736b9940c23e77dChris Lattner    // If there is no linkage to be performed, just bring over SF without
918824684982968244050e56357e736b9940c23e77dChris Lattner    // modifying it.
919194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov    if (DGV == 0) {
920824684982968244050e56357e736b9940c23e77dChris Lattner      // Function does not already exist, simply insert an function signature
921824684982968244050e56357e736b9940c23e77dChris Lattner      // identical to SF into the dest module.
922824684982968244050e56357e736b9940c23e77dChris Lattner      Function *NewDF = Function::Create(SF->getFunctionType(),
923824684982968244050e56357e736b9940c23e77dChris Lattner                                         SF->getLinkage(),
924824684982968244050e56357e736b9940c23e77dChris Lattner                                         SF->getName(), Dest);
925824684982968244050e56357e736b9940c23e77dChris Lattner      CopyGVAttributes(NewDF, SF);
926824684982968244050e56357e736b9940c23e77dChris Lattner
927824684982968244050e56357e736b9940c23e77dChris Lattner      // If the LLVM runtime renamed the function, but it is an externally
928824684982968244050e56357e736b9940c23e77dChris Lattner      // visible symbol, DF must be an existing function with internal linkage.
929824684982968244050e56357e736b9940c23e77dChris Lattner      // Rename it.
930824684982968244050e56357e736b9940c23e77dChris Lattner      if (!NewDF->hasInternalLinkage() && NewDF->getName() != SF->getName())
931824684982968244050e56357e736b9940c23e77dChris Lattner        ForceRenaming(NewDF, SF->getName());
932824684982968244050e56357e736b9940c23e77dChris Lattner
933824684982968244050e56357e736b9940c23e77dChris Lattner      // ... and remember this mapping...
934824684982968244050e56357e736b9940c23e77dChris Lattner      ValueMap[SF] = NewDF;
935824684982968244050e56357e736b9940c23e77dChris Lattner      continue;
936194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov    } else if (GlobalAlias *DGA = dyn_cast<GlobalAlias>(DGV)) {
937f88bc65932e5cf41ba6ca706cd1b1b1f65300248Anton Korobeynikov      // SF is function, but DF is alias.
938f88bc65932e5cf41ba6ca706cd1b1b1f65300248Anton Korobeynikov      // The only valid mappings are:
939f88bc65932e5cf41ba6ca706cd1b1b1f65300248Anton Korobeynikov      // - SF is external declaration, which is effectively a no-op.
940f88bc65932e5cf41ba6ca706cd1b1b1f65300248Anton Korobeynikov      // - SF is weak, when we just need to throw SF out.
94180585f1daac3570f19cd255c5006dfcf2ff65a89Anton Korobeynikov      if (!SF->isDeclaration() && !SF->isWeakForLinker())
942194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov        return Error(Err, "Function-Alias Collision on '" + SF->getName() +
943194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov                     "': symbol multiple defined");
944194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov
945194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov      // Make sure to remember this mapping...
946f88bc65932e5cf41ba6ca706cd1b1b1f65300248Anton Korobeynikov      ValueMap[SF] = ConstantExpr::getBitCast(DGA, SF->getType());
947194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov      continue;
948824684982968244050e56357e736b9940c23e77dChris Lattner    }
949194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov
950194c2cef8a0036dd4e6295d048f37d6130f92389Anton Korobeynikov    Function* DF = cast<Function>(DGV);
951824684982968244050e56357e736b9940c23e77dChris Lattner    // If types don't agree because of opaque, try to resolve them.
952824684982968244050e56357e736b9940c23e77dChris Lattner    if (SF->getType() != DF->getType())
953bc1c82a0f73eda86cf24416cebc83b4690fcd76aChris Lattner      RecursiveResolveTypes(SF->getType(), DF->getType());
954824684982968244050e56357e736b9940c23e77dChris Lattner
955824684982968244050e56357e736b9940c23e77dChris Lattner    // Check visibility, merging if a definition overrides a prototype.
956824684982968244050e56357e736b9940c23e77dChris Lattner    if (SF->getVisibility() != DF->getVisibility()) {
95797f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner      // If one is a prototype, ignore its visibility.  Prototypes are always
95897f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner      // overridden by the definition.
95997f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner      if (!SF->isDeclaration() && !DF->isDeclaration())
96097f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner        return Error(Err, "Linking functions named '" + SF->getName() +
96197f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner                     "': symbols have different visibilities!");
962bc3d1c7e4c290a35df801000222b8f4b2d51ef20Chris Lattner
963bc3d1c7e4c290a35df801000222b8f4b2d51ef20Chris Lattner      // Otherwise, replace the visibility of DF if DF is a prototype.
964bc3d1c7e4c290a35df801000222b8f4b2d51ef20Chris Lattner      if (DF->isDeclaration())
965bc3d1c7e4c290a35df801000222b8f4b2d51ef20Chris Lattner        DF->setVisibility(SF->getVisibility());
96697f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner    }
967ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer
968824684982968244050e56357e736b9940c23e77dChris Lattner    if (DF->getType() != SF->getType()) {
969ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      if (DF->isDeclaration() && !SF->isDeclaration()) {
970ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // We have a definition of the same name but different type in the
971ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // source module. Copy the prototype to the destination and replace
972ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // uses of the destination's prototype with the new prototype.
973b1dbcd886a4b5597a839f299054b78b33fb2d6dfGabor Greif        Function *NewDF = Function::Create(SF->getFunctionType(),
974b1dbcd886a4b5597a839f299054b78b33fb2d6dfGabor Greif                                           SF->getLinkage(),
975051a950000e21935165db56695e35bade668193bGabor Greif                                           SF->getName(), Dest);
976ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        CopyGVAttributes(NewDF, SF);
977ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer
978ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // Any uses of DF need to change to NewDF, with cast
979ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        DF->replaceAllUsesWith(ConstantExpr::getBitCast(NewDF, DF->getType()));
980ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer
981ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // DF will conflict with NewDF because they both had the same. We must
982ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // erase this now so ForceRenaming doesn't assert because DF might
983ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // not have internal linkage.
984ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        DF->eraseFromParent();
985ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer
986ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // If the symbol table renamed the function, but it is an externally
987ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // visible symbol, DF must be an existing function with internal
988ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // linkage.  Rename it.
989ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        if (NewDF->getName() != SF->getName() && !NewDF->hasInternalLinkage())
990ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer          ForceRenaming(NewDF, SF->getName());
991ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer
992ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // Remember this mapping so uses in the source module get remapped
993ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // later by RemapOperand.
994ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        ValueMap[SF] = NewDF;
995ec91ccba3ca92d5e1f19fb0cb6e57a7d4a3f3195Chris Lattner        continue;
996ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      } else {
997ec91ccba3ca92d5e1f19fb0cb6e57a7d4a3f3195Chris Lattner        // We have two functions of the same name but different type. Any use
998ec91ccba3ca92d5e1f19fb0cb6e57a7d4a3f3195Chris Lattner        // of the source must be mapped to the destination, with a cast.
999ec91ccba3ca92d5e1f19fb0cb6e57a7d4a3f3195Chris Lattner        MappedDF = ConstantExpr::getBitCast(DF, SF->getType());
1000ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      }
1001ec91ccba3ca92d5e1f19fb0cb6e57a7d4a3f3195Chris Lattner    } else {
1002ec91ccba3ca92d5e1f19fb0cb6e57a7d4a3f3195Chris Lattner       MappedDF = DF;
1003824684982968244050e56357e736b9940c23e77dChris Lattner    }
1004824684982968244050e56357e736b9940c23e77dChris Lattner
1005824684982968244050e56357e736b9940c23e77dChris Lattner    if (SF->isDeclaration()) {
1006ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      // If SF is a declaration or if both SF & DF are declarations, just link
1007ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      // the declarations, we aren't adding anything.
1008b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov      if (SF->hasDLLImportLinkage()) {
10095cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer        if (DF->isDeclaration()) {
1010ec91ccba3ca92d5e1f19fb0cb6e57a7d4a3f3195Chris Lattner          ValueMap[SF] = MappedDF;
1011b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov          DF->setLinkage(SF->getLinkage());
1012822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner        }
1013b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov      } else {
1014ec91ccba3ca92d5e1f19fb0cb6e57a7d4a3f3195Chris Lattner        ValueMap[SF] = MappedDF;
1015822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner      }
1016822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner      continue;
1017822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner    }
1018822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner
1019822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner    // If DF is external but SF is not, link the external functions, update
1020822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner    // linkage qualifiers.
1021822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner    if (DF->isDeclaration() && !DF->hasDLLImportLinkage()) {
1022ec91ccba3ca92d5e1f19fb0cb6e57a7d4a3f3195Chris Lattner      ValueMap.insert(std::make_pair(SF, MappedDF));
1023c2b97d43a074445c6856d69538a95de09450c26eChris Lattner      DF->setLinkage(SF->getLinkage());
1024822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner      continue;
1025822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner    }
1026822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner
1027822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner    // At this point we know that DF has LinkOnce, Weak, or External* linkage.
102880585f1daac3570f19cd255c5006dfcf2ff65a89Anton Korobeynikov    if (SF->isWeakForLinker()) {
1029ec91ccba3ca92d5e1f19fb0cb6e57a7d4a3f3195Chris Lattner      ValueMap[SF] = MappedDF;
103072ac148d4964fbff950fe11a0eeeda73973b849bChris Lattner
103135956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner      // Linkonce+Weak = Weak
103278ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      // *+External Weak = *
103380585f1daac3570f19cd255c5006dfcf2ff65a89Anton Korobeynikov      if ((DF->hasLinkOnceLinkage() &&
1034aafce77b17d340aace52bcd49d1944109d82f14aDale Johannesen              (SF->hasWeakLinkage() || SF->hasCommonLinkage())) ||
103578ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov          DF->hasExternalWeakLinkage())
103635956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner        DF->setLinkage(SF->getLinkage());
1037822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner      continue;
1038822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner    }
1039822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner
104080585f1daac3570f19cd255c5006dfcf2ff65a89Anton Korobeynikov    if (DF->isWeakForLinker()) {
104178ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      // At this point we know that SF has LinkOnce or External* linkage.
1042ec91ccba3ca92d5e1f19fb0cb6e57a7d4a3f3195Chris Lattner      ValueMap[SF] = MappedDF;
1043822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner
1044822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner      // If the source function has stronger linkage than the destination,
1045822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner      // its body and linkage should override ours.
1046822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner      if (!SF->hasLinkOnceLinkage() && !SF->hasExternalWeakLinkage()) {
1047822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner        // Don't inherit linkonce & external weak linkage.
104872ac148d4964fbff950fe11a0eeeda73973b849bChris Lattner        DF->setLinkage(SF->getLinkage());
1049822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner        DF->deleteBody();
1050822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner      }
1051822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner      continue;
1052822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner    }
1053822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner
1054822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner    if (SF->getLinkage() != DF->getLinkage())
1055822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner      return Error(Err, "Functions named '" + SF->getName() +
1056822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner                   "' have different linkage specifiers!");
1057822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner
1058822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner    // The function is defined identically in both modules!
1059822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner    if (SF->hasExternalLinkage())
1060f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman      return Error(Err, "Function '" +
1061f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman                   ToStr(SF->getFunctionType(), Src) + "':\"" +
1062c2b97d43a074445c6856d69538a95de09450c26eChris Lattner                   SF->getName() + "\" - Function is already defined!");
1063822143e6f7271c6546434d2d0b3da7a29711020dChris Lattner    assert(0 && "Unknown linkage configuration found!");
10645c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  }
10655c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return false;
10665c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
10675c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
1068c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// LinkFunctionBody - Copy the source function over into the dest function and
1069c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// fix up references to values.  At this point we know that Dest is an external
1070c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// function, and that Src is not.
10714bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattnerstatic bool LinkFunctionBody(Function *Dest, Function *Src,
1072ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer                             std::map<const Value*, Value*> &ValueMap,
10735c2d335d86579941b71816196a95a64b277e8963Chris Lattner                             std::string *Err) {
10745cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer  assert(Src && Dest && Dest->isDeclaration() && !Src->isDeclaration());
10755c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
10760033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner  // Go through and convert function arguments over, remembering the mapping.
1077e4d5c441e04bdc00ccf1804744af670655123b07Chris Lattner  Function::arg_iterator DI = Dest->arg_begin();
1078e4d5c441e04bdc00ccf1804744af670655123b07Chris Lattner  for (Function::arg_iterator I = Src->arg_begin(), E = Src->arg_end();
107969da5cf26143e4542d4bf8c78ffac6d079efe5c9Chris Lattner       I != E; ++I, ++DI) {
10806bc41e8a74d1756da0003641bfebd02a3d6d9586Owen Anderson    DI->setName(I->getName());  // Copy the name information over...
10815c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
10825c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner    // Add a mapping to our local map
1083817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov    ValueMap[I] = DI;
10845c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  }
10855c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
10864bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner  // Splice the body of the source function into the dest function.
10874bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner  Dest->getBasicBlockList().splice(Dest->end(), Src->getBasicBlockList());
10885c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
1089c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // At this point, all of the instructions and values of the function are now
1090c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // copied over.  The only problem is that they are still referencing values in
1091c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // the Source function as operands.  Loop through all of the operands of the
1092c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // functions and patch them up to point to the local versions...
10935c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  //
109418961504fc2b299578dba817900a0696cf3ccc4dChris Lattner  for (Function::iterator BB = Dest->begin(), BE = Dest->end(); BB != BE; ++BB)
109518961504fc2b299578dba817900a0696cf3ccc4dChris Lattner    for (BasicBlock::iterator I = BB->begin(), E = BB->end(); I != E; ++I)
109618961504fc2b299578dba817900a0696cf3ccc4dChris Lattner      for (Instruction::op_iterator OI = I->op_begin(), OE = I->op_end();
1097221d688a5ef21a22c2368c9fff0e92d7966c95e5Chris Lattner           OI != OE; ++OI)
10984bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner        if (!isa<Instruction>(*OI) && !isa<BasicBlock>(*OI))
1099ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer          *OI = RemapOperand(*OI, ValueMap);
11000033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner
11010033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner  // There is no need to map the arguments anymore.
11021127315562b221010040925980040fd4f65bdb1cChris Lattner  for (Function::arg_iterator I = Src->arg_begin(), E = Src->arg_end();
11031127315562b221010040925980040fd4f65bdb1cChris Lattner       I != E; ++I)
1104ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer    ValueMap.erase(I);
11055c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
11065c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return false;
11075c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
11085c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
11095c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
1110c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// LinkFunctionBodies - Link in the function bodies that are defined in the
1111c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// source module into the DestModule.  This consists basically of copying the
1112c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// function over and fixing up references to values.
11134bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattnerstatic bool LinkFunctionBodies(Module *Dest, Module *Src,
11145c2d335d86579941b71816196a95a64b277e8963Chris Lattner                               std::map<const Value*, Value*> &ValueMap,
11155c2d335d86579941b71816196a95a64b277e8963Chris Lattner                               std::string *Err) {
11165c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
11178bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer  // Loop over all of the functions in the src module, mapping them over as we
11188bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer  // go
11194bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner  for (Module::iterator SF = Src->begin(), E = Src->end(); SF != E; ++SF) {
1120619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer    if (!SF->isDeclaration()) {               // No body if function is external
1121ec91ccba3ca92d5e1f19fb0cb6e57a7d4a3f3195Chris Lattner      Function *DF = dyn_cast<Function>(ValueMap[SF]); // Destination function
1122c2d774b6c1d38421c435b6d3cfaa10402c900aebChris Lattner
112318961504fc2b299578dba817900a0696cf3ccc4dChris Lattner      // DF not external SF external?
1124ec91ccba3ca92d5e1f19fb0cb6e57a7d4a3f3195Chris Lattner      if (DF && DF->isDeclaration())
112535956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner        // Only provide the function body if there isn't one already.
112635956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner        if (LinkFunctionBody(DF, SF, ValueMap, Err))
112735956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner          return true;
1128c2d774b6c1d38421c435b6d3cfaa10402c900aebChris Lattner    }
11295c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  }
11305c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return false;
11315c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
11325c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
11338166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner// LinkAppendingVars - If there were any appending global variables, link them
11348166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner// together now.  Return true on error.
11358166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattnerstatic bool LinkAppendingVars(Module *M,
11368166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                  std::multimap<std::string, GlobalVariable *> &AppendingVars,
11378166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                              std::string *ErrorMsg) {
11388166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  if (AppendingVars.empty()) return false; // Nothing to do.
1139f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
11408166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // Loop over the multimap of appending vars, processing any variables with the
11418166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // same name, forming a new appending global variable with both of the
11428166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // initializers merged together, then rewrite references to the old variables
11438166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // and delete them.
11448166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  std::vector<Constant*> Inits;
11458166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  while (AppendingVars.size() > 1) {
11468166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    // Get the first two elements in the map...
11478166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    std::multimap<std::string,
11488166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      GlobalVariable*>::iterator Second = AppendingVars.begin(), First=Second++;
11498166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
11508166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    // If the first two elements are for different names, there is no pair...
11518166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    // Otherwise there is a pair, so link them together...
11528166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    if (First->first == Second->first) {
11538166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      GlobalVariable *G1 = First->second, *G2 = Second->second;
11548166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      const ArrayType *T1 = cast<ArrayType>(G1->getType()->getElementType());
11558166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      const ArrayType *T2 = cast<ArrayType>(G2->getType()->getElementType());
1156f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
11578166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Check to see that they two arrays agree on type...
11588166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      if (T1->getElementType() != T2->getElementType())
11598166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        return Error(ErrorMsg,
11608166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner         "Appending variables with different element types need to be linked!");
11618166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      if (G1->isConstant() != G2->isConstant())
11628166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        return Error(ErrorMsg,
11638166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                     "Appending variables linked with different const'ness!");
11648166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
11659613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio      if (G1->getAlignment() != G2->getAlignment())
11669613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio        return Error(ErrorMsg,
11679613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio         "Appending variables with different alignment need to be linked!");
11689613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio
11699613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio      if (G1->getVisibility() != G2->getVisibility())
11709613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio        return Error(ErrorMsg,
11719613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio         "Appending variables with different visibility need to be linked!");
11729613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio
11739613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio      if (G1->getSection() != G2->getSection())
11749613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio        return Error(ErrorMsg,
11759613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio         "Appending variables with different section name need to be linked!");
11769613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio
11778166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      unsigned NewSize = T1->getNumElements() + T2->getNumElements();
11788166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      ArrayType *NewType = ArrayType::get(T1->getElementType(), NewSize);
11798166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
1180ed74a4ef3b813bed0c0f4bb5af42229d50133da2Chris Lattner      G1->setName("");   // Clear G1's name in case of a conflict!
1181ed74a4ef3b813bed0c0f4bb5af42229d50133da2Chris Lattner
11828166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Create the new global variable...
11838166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      GlobalVariable *NG =
11848166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        new GlobalVariable(NewType, G1->isConstant(), G1->getLinkage(),
1185a534b0f7bf0248bf90afe97110a34777e972f325Chris Lattner                           /*init*/0, First->first, M, G1->isThreadLocal(),
1186a534b0f7bf0248bf90afe97110a34777e972f325Chris Lattner                           G1->getType()->getAddressSpace());
11878166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
11889613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio      // Propagate alignment, visibility and section info.
11899613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio      CopyGVAttributes(NG, G1);
11909613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio
11918166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Merge the initializer...
11928166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      Inits.reserve(NewSize);
1193de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      if (ConstantArray *I = dyn_cast<ConstantArray>(G1->getInitializer())) {
1194de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        for (unsigned i = 0, e = T1->getNumElements(); i != e; ++i)
1195cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos          Inits.push_back(I->getOperand(i));
1196de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      } else {
1197de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        assert(isa<ConstantAggregateZero>(G1->getInitializer()));
1198de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        Constant *CV = Constant::getNullValue(T1->getElementType());
1199de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        for (unsigned i = 0, e = T1->getNumElements(); i != e; ++i)
1200de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner          Inits.push_back(CV);
1201de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      }
1202de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      if (ConstantArray *I = dyn_cast<ConstantArray>(G2->getInitializer())) {
1203de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        for (unsigned i = 0, e = T2->getNumElements(); i != e; ++i)
1204cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos          Inits.push_back(I->getOperand(i));
1205de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      } else {
1206de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        assert(isa<ConstantAggregateZero>(G2->getInitializer()));
1207de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        Constant *CV = Constant::getNullValue(T2->getElementType());
1208de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        for (unsigned i = 0, e = T2->getNumElements(); i != e; ++i)
1209de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner          Inits.push_back(CV);
1210de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      }
12118166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      NG->setInitializer(ConstantArray::get(NewType, Inits));
12128166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      Inits.clear();
12138166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
12148166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Replace any uses of the two global variables with uses of the new
12158166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // global...
12168166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
12178166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // FIXME: This should rewrite simple/straight-forward uses such as
12188166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // getelementptr instructions to not use the Cast!
12194da49122f3f3c8da68a52723d846b88c72166a68Reid Spencer      G1->replaceAllUsesWith(ConstantExpr::getBitCast(NG, G1->getType()));
12204da49122f3f3c8da68a52723d846b88c72166a68Reid Spencer      G2->replaceAllUsesWith(ConstantExpr::getBitCast(NG, G2->getType()));
12218166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
12228166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Remove the two globals from the module now...
12238166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      M->getGlobalList().erase(G1);
12248166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      M->getGlobalList().erase(G2);
12258166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
12268166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Put the new global into the AppendingVars map so that we can handle
12278166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // linking of more than two vars...
12288166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      Second->second = NG;
12298166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    }
12308166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    AppendingVars.erase(First);
12318166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  }
12328166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
12338166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  return false;
12348166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner}
123552f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner
12369f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikovstatic bool ResolveAliases(Module *Dest) {
12379f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov  for (Module::alias_iterator I = Dest->alias_begin(), E = Dest->alias_end();
12385241957b1f6876452705f94e0ed60dd2c33a5c84Anton Korobeynikov       I != E; ++I)
12395241957b1f6876452705f94e0ed60dd2c33a5c84Anton Korobeynikov    if (const GlobalValue *GV = I->resolveAliasedGlobal())
12405241957b1f6876452705f94e0ed60dd2c33a5c84Anton Korobeynikov      if (!GV->isDeclaration())
12415241957b1f6876452705f94e0ed60dd2c33a5c84Anton Korobeynikov        I->replaceAllUsesWith(const_cast<GlobalValue*>(GV));
12429f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov
12439f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov  return false;
12449f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov}
124552f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner
124652f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// LinkModules - This function links two modules together, with the resulting
124752f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// left module modified to be the composite of the two input modules.  If an
124852f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// error occurs, true is returned and ErrorMsg (if not null) is set to indicate
12495c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// the problem.  Upon failure, the Dest module could be in a modified state, and
12505c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// shouldn't be relied on to be consistent.
1251f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukmanbool
12520ba9e211bed7813949aaca97e6b15579c8188e05Reid SpencerLinker::LinkModules(Module *Dest, Module *Src, std::string *ErrorMsg) {
125357a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  assert(Dest != 0 && "Invalid Destination module");
125457a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  assert(Src  != 0 && "Invalid Source Module");
125557a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer
1256c36357c406911ded5a0e2924998d885815370fdbChris Lattner  if (Dest->getDataLayout().empty()) {
1257c36357c406911ded5a0e2924998d885815370fdbChris Lattner    if (!Src->getDataLayout().empty()) {
1258ec9bfdc152a870d3c62377ed6855fb3bff69511cChris Lattner      Dest->setDataLayout(Src->getDataLayout());
1259c36357c406911ded5a0e2924998d885815370fdbChris Lattner    } else {
1260c36357c406911ded5a0e2924998d885815370fdbChris Lattner      std::string DataLayout;
1261c36357c406911ded5a0e2924998d885815370fdbChris Lattner
1262a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov      if (Dest->getEndianness() == Module::AnyEndianness) {
1263c36357c406911ded5a0e2924998d885815370fdbChris Lattner        if (Src->getEndianness() == Module::BigEndian)
1264c36357c406911ded5a0e2924998d885815370fdbChris Lattner          DataLayout.append("E");
1265c36357c406911ded5a0e2924998d885815370fdbChris Lattner        else if (Src->getEndianness() == Module::LittleEndian)
1266c36357c406911ded5a0e2924998d885815370fdbChris Lattner          DataLayout.append("e");
1267a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov      }
1268a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov
1269a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov      if (Dest->getPointerSize() == Module::AnyPointerSize) {
1270c36357c406911ded5a0e2924998d885815370fdbChris Lattner        if (Src->getPointerSize() == Module::Pointer64)
1271c36357c406911ded5a0e2924998d885815370fdbChris Lattner          DataLayout.append(DataLayout.length() == 0 ? "p:64:64" : "-p:64:64");
1272c36357c406911ded5a0e2924998d885815370fdbChris Lattner        else if (Src->getPointerSize() == Module::Pointer32)
1273c36357c406911ded5a0e2924998d885815370fdbChris Lattner          DataLayout.append(DataLayout.length() == 0 ? "p:32:32" : "-p:32:32");
1274a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov      }
1275c36357c406911ded5a0e2924998d885815370fdbChris Lattner      Dest->setDataLayout(DataLayout);
1276c36357c406911ded5a0e2924998d885815370fdbChris Lattner    }
1277c36357c406911ded5a0e2924998d885815370fdbChris Lattner  }
1278873c5e7859c1534bf92c9e2747f2b70685059598Chris Lattner
1279f27dfcb210fec571d79bc881f5463e0b405707c0Chris Lattner  // Copy the target triple from the source to dest if the dest's is empty.
1280c36357c406911ded5a0e2924998d885815370fdbChris Lattner  if (Dest->getTargetTriple().empty() && !Src->getTargetTriple().empty())
1281c36357c406911ded5a0e2924998d885815370fdbChris Lattner    Dest->setTargetTriple(Src->getTargetTriple());
1282c36357c406911ded5a0e2924998d885815370fdbChris Lattner
1283c36357c406911ded5a0e2924998d885815370fdbChris Lattner  if (!Src->getDataLayout().empty() && !Dest->getDataLayout().empty() &&
1284c36357c406911ded5a0e2924998d885815370fdbChris Lattner      Src->getDataLayout() != Dest->getDataLayout())
128526f238589f9bb372d24b6fb2bc32edbf046fd9eeReid Spencer    cerr << "WARNING: Linking two modules of different data layouts!\n";
1286152f19ac6514ec29e183ddcdf2ea71599638dc4cChris Lattner  if (!Src->getTargetTriple().empty() &&
1287152f19ac6514ec29e183ddcdf2ea71599638dc4cChris Lattner      Dest->getTargetTriple() != Src->getTargetTriple())
1288e81561909d128c6e2d8033cb5465a49b2596b26aBill Wendling    cerr << "WARNING: Linking two modules of different target triples!\n";
1289f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
1290f27dfcb210fec571d79bc881f5463e0b405707c0Chris Lattner  // Append the module inline asm string.
12916631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner  if (!Src->getModuleInlineAsm().empty()) {
12926631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner    if (Dest->getModuleInlineAsm().empty())
12936631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner      Dest->setModuleInlineAsm(Src->getModuleInlineAsm());
1294e1b2e14f92e8e9da1eb4854511a881cf9c22af3cChris Lattner    else
12956631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner      Dest->setModuleInlineAsm(Dest->getModuleInlineAsm()+"\n"+
12966631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner                               Src->getModuleInlineAsm());
1297e1b2e14f92e8e9da1eb4854511a881cf9c22af3cChris Lattner  }
1298e1b2e14f92e8e9da1eb4854511a881cf9c22af3cChris Lattner
1299719012d6254ee425464f034250176283fb96793bReid Spencer  // Update the destination module's dependent libraries list with the libraries
130057a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  // from the source module. There's no opportunity for duplicates here as the
130157a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  // Module ensures that duplicate insertions are discarded.
1302f27dfcb210fec571d79bc881f5463e0b405707c0Chris Lattner  for (Module::lib_iterator SI = Src->lib_begin(), SE = Src->lib_end();
1303f27dfcb210fec571d79bc881f5463e0b405707c0Chris Lattner       SI != SE; ++SI)
130457a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer    Dest->addLibrary(*SI);
130557a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer
13062c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  // LinkTypes - Go through the symbol table of the Src module and see if any
13072c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  // types are named in the src module that are not named in the Dst module.
13082c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  // Make sure there are no type name conflicts.
1309619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer  if (LinkTypes(Dest, Src, ErrorMsg))
1310619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer    return true;
13112c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
13125c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  // ValueMap - Mapping of values from what they used to be in Src, to what they
13135c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  // are now in Dest.
13145c2d335d86579941b71816196a95a64b277e8963Chris Lattner  std::map<const Value*, Value*> ValueMap;
13155c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
13168166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // AppendingVars - Keep track of global variables in the destination module
13178166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // with appending linkage.  After the module is linked together, they are
13188166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // appended and the module is rewritten.
13198166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  std::multimap<std::string, GlobalVariable *> AppendingVars;
13201127315562b221010040925980040fd4f65bdb1cChris Lattner  for (Module::global_iterator I = Dest->global_begin(), E = Dest->global_end();
13211127315562b221010040925980040fd4f65bdb1cChris Lattner       I != E; ++I) {
13225a837defff581accc09048bea1cd839009e374a7Chris Lattner    // Add all of the appending globals already in the Dest module to
13235a837defff581accc09048bea1cd839009e374a7Chris Lattner    // AppendingVars.
1324f41464618f41b92cf0a0bd14ccc57bce727fb33dChris Lattner    if (I->hasAppendingLinkage())
1325f41464618f41b92cf0a0bd14ccc57bce727fb33dChris Lattner      AppendingVars.insert(std::make_pair(I->getName(), I));
13265a837defff581accc09048bea1cd839009e374a7Chris Lattner  }
13275a837defff581accc09048bea1cd839009e374a7Chris Lattner
13288166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // Insert all of the globals in src into the Dest module... without linking
13298166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // initializers (which could refer to functions not yet mapped over).
1330ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  if (LinkGlobals(Dest, Src, ValueMap, AppendingVars, ErrorMsg))
13315a837defff581accc09048bea1cd839009e374a7Chris Lattner    return true;
13325c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
1333c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // Link the functions together between the two modules, without doing function
1334c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // bodies... this just adds external function prototypes to the Dest
1335c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // function...  We do this so that when we begin processing function bodies,
1336c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // all of the global values that may be referenced are available in our
1337c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // ValueMap.
1338ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  if (LinkFunctionProtos(Dest, Src, ValueMap, ErrorMsg))
13395a837defff581accc09048bea1cd839009e374a7Chris Lattner    return true;
13405c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
13414fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov  // If there were any alias, link them now. We really need to do this now,
13424fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov  // because all of the aliases that may be referenced need to be available in
13434fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov  // ValueMap
13444fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov  if (LinkAlias(Dest, Src, ValueMap, ErrorMsg)) return true;
13454fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov
13466cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner  // Update the initializers in the Dest module now that all globals that may
13476cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner  // be referenced are in Dest.
13486cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner  if (LinkGlobalInits(Dest, Src, ValueMap, ErrorMsg)) return true;
13496cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner
1350c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // Link in the function bodies that are defined in the source module into the
1351c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // DestModule.  This consists basically of copying the function over and
1352c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // fixing up references to values.
135379df7c0aaa18129e55968c8783ef8346807bd4afChris Lattner  if (LinkFunctionBodies(Dest, Src, ValueMap, ErrorMsg)) return true;
135452f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner
13558166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // If there were any appending global variables, link them together now.
13568166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  if (LinkAppendingVars(Dest, AppendingVars, ErrorMsg)) return true;
13578166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
13583db9191baf5f5229a2f4d204c6332ba27fff81d0Anton Korobeynikov  // Resolve all uses of aliases with aliasees
13593db9191baf5f5229a2f4d204c6332ba27fff81d0Anton Korobeynikov  if (ResolveAliases(Dest)) return true;
13603db9191baf5f5229a2f4d204c6332ba27fff81d0Anton Korobeynikov
136157a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  // If the source library's module id is in the dependent library list of the
136257a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  // destination library, remove it since that module is now linked in.
136357a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  sys::Path modId;
1364dd04df0ec33a903ee7fc747701bafde622f77d8bReid Spencer  modId.set(Src->getModuleIdentifier());
136507adb2836b8aa7a3872e33c285958f5937662b50Reid Spencer  if (!modId.isEmpty())
136607adb2836b8aa7a3872e33c285958f5937662b50Reid Spencer    Dest->removeLibrary(modId.getBasename());
136757a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer
136852f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner  return false;
136952f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner}
13709466f5113ba859677a28887c3c7143065e702019Vikram S. Adve
1371567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer// vim: sw=2
1372