LinkModules.cpp revision ed61c0bc76a7b9001d7639a908c955d0fae35bac
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"
291a097e30d39e60303ae2b19f7a56e813f3e3c18eBill Wendling#include <sstream>
30f7703df4968084c18c248c1feea9961c19a32e6aChris Lattnerusing namespace llvm;
31d0fde30ce850b78371fd1386338350591f9ff494Brian Gaeke
325c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// Error - Simple wrapper function to conditionally assign to E and return true.
335c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// This just makes error return conditions a little bit simpler...
348166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattnerstatic inline bool Error(std::string *E, const std::string &Message) {
355c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  if (E) *E = Message;
365c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return true;
375c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
385c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
39719012d6254ee425464f034250176283fb96793bReid Spencer// ToStr - Simple wrapper function to convert a type to a string.
4072cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattnerstatic std::string ToStr(const Type *Ty, const Module *M) {
4172cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattner  std::ostringstream OS;
4272cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattner  WriteTypeSymbolic(OS, Ty, M);
4372cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattner  return OS.str();
4472cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattner}
4572cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattner
46700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//
47700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Function: ResolveTypes()
48700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//
49700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Description:
50700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  Attempt to link the two specified types together.
51700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//
52700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Inputs:
53700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  DestTy - The type to which we wish to resolve.
54700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  SrcTy  - The original type which we want to resolve.
55700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  Name   - The name of the type.
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//
64e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattnerstatic bool ResolveTypes(const Type *DestTy, const Type *SrcTy,
6578d033e086e19e016273de014f9214aa6f3f844bReid Spencer                         TypeSymbolTable *DestST, const std::string &Name) {
66e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  if (DestTy == SrcTy) return false;       // If already equal, noop
67e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
684c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  // Does the type already exist in the module?
694c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  if (DestTy && !isa<OpaqueType>(DestTy)) {  // Yup, the type already exists...
70e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner    if (const OpaqueType *OT = dyn_cast<OpaqueType>(SrcTy)) {
71e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      const_cast<OpaqueType*>(OT)->refineAbstractTypeTo(DestTy);
724c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    } else {
734c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      return true;  // Cannot link types... neither is opaque and not-equal
744c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    }
754c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  } else {                       // Type not in dest module.  Add it now.
764c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    if (DestTy)                  // Type _is_ in module, just opaque...
77e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      const_cast<OpaqueType*>(cast<OpaqueType>(DestTy))
78e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner                           ->refineAbstractTypeTo(SrcTy);
79fcd0234de65fc07fb06f3245b60e82bdba78c745Chris Lattner    else if (!Name.empty())
80e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      DestST->insert(Name, const_cast<Type*>(SrcTy));
814c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  }
824c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  return false;
834c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner}
844c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
8543f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattnerstatic const FunctionType *getFT(const PATypeHolder &TH) {
8643f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner  return cast<FunctionType>(TH.get());
8743f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner}
889732be769e57f1dec03f046ab957053048f60e40Chris Lattnerstatic const StructType *getST(const PATypeHolder &TH) {
8943f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner  return cast<StructType>(TH.get());
9043f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner}
91e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
92e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner// RecursiveResolveTypes - This is just like ResolveTypes, except that it
93e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner// recurses down into derived types, merging the used types if the parent types
94e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner// are compatible.
95e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattnerstatic bool RecursiveResolveTypesI(const PATypeHolder &DestTy,
96e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                                   const PATypeHolder &SrcTy,
9778d033e086e19e016273de014f9214aa6f3f844bReid Spencer                                   TypeSymbolTable *DestST,
9878d033e086e19e016273de014f9214aa6f3f844bReid Spencer                                   const std::string &Name,
99e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                std::vector<std::pair<PATypeHolder, PATypeHolder> > &Pointers) {
10043f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner  const Type *SrcTyT = SrcTy.get();
10143f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner  const Type *DestTyT = DestTy.get();
10243f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner  if (DestTyT == SrcTyT) return false;       // If already equal, noop
103f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
104e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  // If we found our opaque type, resolve it now!
10543f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner  if (isa<OpaqueType>(DestTyT) || isa<OpaqueType>(SrcTyT))
10643f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner    return ResolveTypes(DestTyT, SrcTyT, DestST, Name);
107f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
108e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  // Two types cannot be resolved together if they are of different primitive
109e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  // type.  For example, we cannot resolve an int to a float.
110f70c22b019494723d0e706f93d6542dfaa6e73a5Chris Lattner  if (DestTyT->getTypeID() != SrcTyT->getTypeID()) return true;
111e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
112e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  // Otherwise, resolve the used type used by this derived type...
113f70c22b019494723d0e706f93d6542dfaa6e73a5Chris Lattner  switch (DestTyT->getTypeID()) {
114a54b7cbd452b3adb2f51346140d996b29c2cdb30Reid Spencer  case Type::IntegerTyID: {
115a54b7cbd452b3adb2f51346140d996b29c2cdb30Reid Spencer    if (cast<IntegerType>(DestTyT)->getBitWidth() !=
116a54b7cbd452b3adb2f51346140d996b29c2cdb30Reid Spencer        cast<IntegerType>(SrcTyT)->getBitWidth())
117a54b7cbd452b3adb2f51346140d996b29c2cdb30Reid Spencer      return true;
118a54b7cbd452b3adb2f51346140d996b29c2cdb30Reid Spencer    return false;
119a54b7cbd452b3adb2f51346140d996b29c2cdb30Reid Spencer  }
120e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  case Type::FunctionTyID: {
12143f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner    if (cast<FunctionType>(DestTyT)->isVarArg() !=
122841e00b96295a2b66cb7573e961656d28a6cb12bChris Lattner        cast<FunctionType>(SrcTyT)->isVarArg() ||
123841e00b96295a2b66cb7573e961656d28a6cb12bChris Lattner        cast<FunctionType>(DestTyT)->getNumContainedTypes() !=
124841e00b96295a2b66cb7573e961656d28a6cb12bChris Lattner        cast<FunctionType>(SrcTyT)->getNumContainedTypes())
12543f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner      return true;
12643f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner    for (unsigned i = 0, e = getFT(DestTy)->getNumContainedTypes(); i != e; ++i)
127e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner      if (RecursiveResolveTypesI(getFT(DestTy)->getContainedType(i),
128e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                                 getFT(SrcTy)->getContainedType(i), DestST, "",
129e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                                 Pointers))
130e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner        return true;
131e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner    return false;
132e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  }
133e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  case Type::StructTyID: {
134f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman    if (getST(DestTy)->getNumContainedTypes() !=
13543f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner        getST(SrcTy)->getNumContainedTypes()) return 1;
13643f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner    for (unsigned i = 0, e = getST(DestTy)->getNumContainedTypes(); i != e; ++i)
137e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner      if (RecursiveResolveTypesI(getST(DestTy)->getContainedType(i),
138e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                                 getST(SrcTy)->getContainedType(i), DestST, "",
139e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                                 Pointers))
140e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner        return true;
141e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner    return false;
142e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  }
143e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  case Type::ArrayTyID: {
14443f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner    const ArrayType *DAT = cast<ArrayType>(DestTy.get());
14543f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner    const ArrayType *SAT = cast<ArrayType>(SrcTy.get());
146e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner    if (DAT->getNumElements() != SAT->getNumElements()) return true;
147e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    return RecursiveResolveTypesI(DAT->getElementType(), SAT->getElementType(),
148e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                                  DestST, "", Pointers);
149e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner  }
150e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner  case Type::PointerTyID: {
151e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // If this is a pointer type, check to see if we have already seen it.  If
152e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // so, we are in a recursive branch.  Cut off the search now.  We cannot use
153e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // an associative container for this search, because the type pointers (keys
154e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // in the container) change whenever types get resolved...
155e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    for (unsigned i = 0, e = Pointers.size(); i != e; ++i)
156e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner      if (Pointers[i].first == DestTy)
157e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner        return Pointers[i].second != SrcTy;
158e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner
159e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // Otherwise, add the current pointers to the vector to stop recursion on
160e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // this pair.
161e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    Pointers.push_back(std::make_pair(DestTyT, SrcTyT));
162e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    bool Result =
163e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner      RecursiveResolveTypesI(cast<PointerType>(DestTy.get())->getElementType(),
164e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                             cast<PointerType>(SrcTy.get())->getElementType(),
165e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                             DestST, "", Pointers);
166e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    Pointers.pop_back();
167e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    return Result;
168e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  }
169e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  default: assert(0 && "Unexpected type!"); return true;
170f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman  }
171e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner}
172e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
173e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattnerstatic bool RecursiveResolveTypes(const PATypeHolder &DestTy,
174e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                                  const PATypeHolder &SrcTy,
17578d033e086e19e016273de014f9214aa6f3f844bReid Spencer                                  TypeSymbolTable *DestST,
17678d033e086e19e016273de014f9214aa6f3f844bReid Spencer                                  const std::string &Name){
177e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner  std::vector<std::pair<PATypeHolder, PATypeHolder> > PointerTypes;
178e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner  return RecursiveResolveTypesI(DestTy, SrcTy, DestST, Name, PointerTypes);
179e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner}
180e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner
181e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
1822c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner// LinkTypes - Go through the symbol table of the Src module and see if any
1832c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner// types are named in the src module that are not named in the Dst module.
1842c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner// Make sure there are no type name conflicts.
1855c2d335d86579941b71816196a95a64b277e8963Chris Lattnerstatic bool LinkTypes(Module *Dest, const Module *Src, std::string *Err) {
18678d033e086e19e016273de014f9214aa6f3f844bReid Spencer        TypeSymbolTable *DestST = &Dest->getTypeSymbolTable();
18778d033e086e19e016273de014f9214aa6f3f844bReid Spencer  const TypeSymbolTable *SrcST  = &Src->getTypeSymbolTable();
1882c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
1892c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  // Look for a type plane for Type's...
19078d033e086e19e016273de014f9214aa6f3f844bReid Spencer  TypeSymbolTable::const_iterator TI = SrcST->begin();
19178d033e086e19e016273de014f9214aa6f3f844bReid Spencer  TypeSymbolTable::const_iterator TE = SrcST->end();
192567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer  if (TI == TE) return false;  // No named types, do nothing.
1932c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
194cf00c4ab3ba308d45d98c5ccab87362cf802facbMisha Brukman  // Some types cannot be resolved immediately because they depend on other
195cf00c4ab3ba308d45d98c5ccab87362cf802facbMisha Brukman  // types being resolved to each other first.  This contains a list of types we
196cf00c4ab3ba308d45d98c5ccab87362cf802facbMisha Brukman  // are waiting to recheck.
1974c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  std::vector<std::string> DelayedTypesToResolve;
1984c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
199567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer  for ( ; TI != TE; ++TI ) {
200567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer    const std::string &Name = TI->first;
201c28a224c68adf0464c275778af4bce762ad25d37Reid Spencer    const Type *RHS = TI->second;
2022c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
2032c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner    // Check to see if this type name is already in the dest module...
20478d033e086e19e016273de014f9214aa6f3f844bReid Spencer    Type *Entry = DestST->lookup(Name);
2054c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
2064c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    if (ResolveTypes(Entry, RHS, DestST, Name)) {
2074c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      // They look different, save the types 'till later to resolve.
2084c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      DelayedTypesToResolve.push_back(Name);
2094c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    }
2104c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  }
2114c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
2124c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  // Iteratively resolve types while we can...
2134c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  while (!DelayedTypesToResolve.empty()) {
2144c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    // Loop over all of the types, attempting to resolve them if possible...
2154c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    unsigned OldSize = DelayedTypesToResolve.size();
2164c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
217e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner    // Try direct resolution by name...
2184c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    for (unsigned i = 0; i != DelayedTypesToResolve.size(); ++i) {
2194c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      const std::string &Name = DelayedTypesToResolve[i];
22078d033e086e19e016273de014f9214aa6f3f844bReid Spencer      Type *T1 = SrcST->lookup(Name);
22178d033e086e19e016273de014f9214aa6f3f844bReid Spencer      Type *T2 = DestST->lookup(Name);
2224c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      if (!ResolveTypes(T2, T1, DestST, Name)) {
2234c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner        // We are making progress!
2244c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner        DelayedTypesToResolve.erase(DelayedTypesToResolve.begin()+i);
2254c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner        --i;
2262f6bb2bce13084dfc4c8acf47f9eae0578267aaaChris Lattner      }
2274c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    }
2282f6bb2bce13084dfc4c8acf47f9eae0578267aaaChris Lattner
2294c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    // Did we not eliminate any types?
2304c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    if (DelayedTypesToResolve.size() == OldSize) {
231e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      // Attempt to resolve subelements of types.  This allows us to merge these
232e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      // two types: { int* } and { opaque* }
2334c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      for (unsigned i = 0, e = DelayedTypesToResolve.size(); i != e; ++i) {
2344c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner        const std::string &Name = DelayedTypesToResolve[i];
23578d033e086e19e016273de014f9214aa6f3f844bReid Spencer        PATypeHolder T1(SrcST->lookup(Name));
23678d033e086e19e016273de014f9214aa6f3f844bReid Spencer        PATypeHolder T2(DestST->lookup(Name));
237e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
238e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner        if (!RecursiveResolveTypes(T2, T1, DestST, Name)) {
239e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          // We are making progress!
240e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          DelayedTypesToResolve.erase(DelayedTypesToResolve.begin()+i);
241f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
242e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          // Go back to the main loop, perhaps we can resolve directly by name
243e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          // now...
244e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          break;
245e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner        }
246e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      }
247e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
248e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      // If we STILL cannot resolve the types, then there is something wrong.
249e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      if (DelayedTypesToResolve.size() == OldSize) {
250aeb18ce9c012f8554f48bae9bb32351c6eeaa26cChris Lattner        // Remove the symbol name from the destination.
251aeb18ce9c012f8554f48bae9bb32351c6eeaa26cChris Lattner        DelayedTypesToResolve.pop_back();
2524c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      }
2532c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner    }
2542c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  }
2554c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
2564c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
2572c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  return false;
2582c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner}
2592c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
2605c2d335d86579941b71816196a95a64b277e8963Chris Lattnerstatic void PrintMap(const std::map<const Value*, Value*> &M) {
2615c2d335d86579941b71816196a95a64b277e8963Chris Lattner  for (std::map<const Value*, Value*>::const_iterator I = M.begin(), E =M.end();
2622d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner       I != E; ++I) {
263e81561909d128c6e2d8033cb5465a49b2596b26aBill Wendling    cerr << " Fr: " << (void*)I->first << " ";
26487182ae6baf28ca6da6ee4bc00cf1857f293acfeChris Lattner    I->first->dump();
265e81561909d128c6e2d8033cb5465a49b2596b26aBill Wendling    cerr << " To: " << (void*)I->second << " ";
26687182ae6baf28ca6da6ee4bc00cf1857f293acfeChris Lattner    I->second->dump();
267e81561909d128c6e2d8033cb5465a49b2596b26aBill Wendling    cerr << "\n";
2682d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner  }
2692d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner}
2702d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner
2712d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner
272619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer// RemapOperand - Use ValueMap to convert constants from one module to another.
2735c2d335d86579941b71816196a95a64b277e8963Chris Lattnerstatic Value *RemapOperand(const Value *In,
2740033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner                           std::map<const Value*, Value*> &ValueMap) {
2750033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner  std::map<const Value*,Value*>::const_iterator I = ValueMap.find(In);
276ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  if (I != ValueMap.end())
277ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer    return I->second;
2785c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
279619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer  // Check to see if it's a constant that we are interested in transforming.
280620fd68b1086b4d99b23951357299bee357aa62aChris Lattner  Value *Result = 0;
28118961504fc2b299578dba817900a0696cf3ccc4dChris Lattner  if (const Constant *CPV = dyn_cast<Constant>(In)) {
282de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner    if ((!isa<DerivedType>(CPV->getType()) && !isa<ConstantExpr>(CPV)) ||
283a54b7cbd452b3adb2f51346140d996b29c2cdb30Reid Spencer        isa<ConstantInt>(CPV) || isa<ConstantAggregateZero>(CPV))
2840033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner      return const_cast<Constant*>(CPV);   // Simple constants stay identical.
2858d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner
28618961504fc2b299578dba817900a0696cf3ccc4dChris Lattner    if (const ConstantArray *CPA = dyn_cast<ConstantArray>(CPV)) {
287cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos      std::vector<Constant*> Operands(CPA->getNumOperands());
288cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos      for (unsigned i = 0, e = CPA->getNumOperands(); i != e; ++i)
2890033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner        Operands[i] =cast<Constant>(RemapOperand(CPA->getOperand(i), ValueMap));
290e9bb2df410f7a22decad9a883f7139d5857c1520Chris Lattner      Result = ConstantArray::get(cast<ArrayType>(CPA->getType()), Operands);
29118961504fc2b299578dba817900a0696cf3ccc4dChris Lattner    } else if (const ConstantStruct *CPS = dyn_cast<ConstantStruct>(CPV)) {
292cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos      std::vector<Constant*> Operands(CPS->getNumOperands());
293cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos      for (unsigned i = 0, e = CPS->getNumOperands(); i != e; ++i)
2940033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner        Operands[i] =cast<Constant>(RemapOperand(CPS->getOperand(i), ValueMap));
295e9bb2df410f7a22decad9a883f7139d5857c1520Chris Lattner      Result = ConstantStruct::get(cast<StructType>(CPS->getType()), Operands);
296b976e668165e1875a8f1eb7af800e33bb1e4393dChris Lattner    } else if (isa<ConstantPointerNull>(CPV) || isa<UndefValue>(CPV)) {
29718961504fc2b299578dba817900a0696cf3ccc4dChris Lattner      Result = const_cast<Constant*>(CPV);
2989d6565a5b1fbc4286d6ee638d8f47a3171a9ed7eReid Spencer    } else if (const ConstantVector *CP = dyn_cast<ConstantVector>(CPV)) {
299a88eb9287eb14ba62ca6324b2b249e0ae9fe3309Chris Lattner      std::vector<Constant*> Operands(CP->getNumOperands());
300a88eb9287eb14ba62ca6324b2b249e0ae9fe3309Chris Lattner      for (unsigned i = 0, e = CP->getNumOperands(); i != e; ++i)
301a88eb9287eb14ba62ca6324b2b249e0ae9fe3309Chris Lattner        Operands[i] = cast<Constant>(RemapOperand(CP->getOperand(i), ValueMap));
3029d6565a5b1fbc4286d6ee638d8f47a3171a9ed7eReid Spencer      Result = ConstantVector::get(Operands);
3036cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner    } else if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(CPV)) {
30427d672136a57519095a3ef20161697749cffb57cChris Lattner      std::vector<Constant*> Ops;
30527d672136a57519095a3ef20161697749cffb57cChris Lattner      for (unsigned i = 0, e = CE->getNumOperands(); i != e; ++i)
30627d672136a57519095a3ef20161697749cffb57cChris Lattner        Ops.push_back(cast<Constant>(RemapOperand(CE->getOperand(i),ValueMap)));
30727d672136a57519095a3ef20161697749cffb57cChris Lattner      Result = CE->getWithOperands(Ops);
308619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer    } else if (isa<GlobalValue>(CPV)) {
309619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer      assert(0 && "Unmapped global?");
3108d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner    } else {
3118d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      assert(0 && "Unknown type of derived type constant value!");
3128d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner    }
313620fd68b1086b4d99b23951357299bee357aa62aChris Lattner  } else if (isa<InlineAsm>(In)) {
314620fd68b1086b4d99b23951357299bee357aa62aChris Lattner    Result = const_cast<Value*>(In);
315620fd68b1086b4d99b23951357299bee357aa62aChris Lattner  }
316620fd68b1086b4d99b23951357299bee357aa62aChris Lattner
317619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer  // Cache the mapping in our local map structure
318620fd68b1086b4d99b23951357299bee357aa62aChris Lattner  if (Result) {
319817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov    ValueMap[In] = Result;
3208d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner    return Result;
3218d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  }
3228bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer
3232d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner
324e81561909d128c6e2d8033cb5465a49b2596b26aBill Wendling  cerr << "LinkModules ValueMap: \n";
3250033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner  PrintMap(ValueMap);
3262d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner
327e81561909d128c6e2d8033cb5465a49b2596b26aBill Wendling  cerr << "Couldn't remap value: " << (void*)In << " " << *In << "\n";
3288d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  assert(0 && "Couldn't remap value!");
3298d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  return 0;
3305c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
3315c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
3328bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer/// ForceRenaming - The LLVM SymbolTable class autorenames globals that conflict
3338bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer/// in the symbol table.  This is good for all clients except for us.  Go
3348bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer/// through the trouble to force this back.
335c003628a612d3687fb77088a5894314210a65385Chris Lattnerstatic void ForceRenaming(GlobalValue *GV, const std::string &Name) {
336c003628a612d3687fb77088a5894314210a65385Chris Lattner  assert(GV->getName() != Name && "Can't force rename to self");
337ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  ValueSymbolTable &ST = GV->getParent()->getValueSymbolTable();
338c003628a612d3687fb77088a5894314210a65385Chris Lattner
339c003628a612d3687fb77088a5894314210a65385Chris Lattner  // If there is a conflict, rename the conflict.
34033f294930e0affa51c3818405741abd80cc90316Chris Lattner  if (GlobalValue *ConflictGV = cast_or_null<GlobalValue>(ST.lookup(Name))) {
341ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer    assert(ConflictGV->hasInternalLinkage() &&
342ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer           "Not conflicting with a static global, should link instead!");
34333f294930e0affa51c3818405741abd80cc90316Chris Lattner    GV->takeName(ConflictGV);
34433f294930e0affa51c3818405741abd80cc90316Chris Lattner    ConflictGV->setName(Name);    // This will cause ConflictGV to get renamed
345ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer    assert(ConflictGV->getName() != Name && "ForceRenaming didn't work");
34633f294930e0affa51c3818405741abd80cc90316Chris Lattner  } else {
34733f294930e0affa51c3818405741abd80cc90316Chris Lattner    GV->setName(Name);              // Force the name back
348ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  }
349ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer}
350ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer
351ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer/// CopyGVAttributes - copy additional attributes (those not needed to construct
352ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer/// a GlobalValue) from the SrcGV to the DestGV.
353ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencerstatic void CopyGVAttributes(GlobalValue *DestGV, const GlobalValue *SrcGV) {
354ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  // Propagate alignment, visibility and section info.
355ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  DestGV->setAlignment(std::max(DestGV->getAlignment(), SrcGV->getAlignment()));
356ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  DestGV->setSection(SrcGV->getSection());
357ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  DestGV->setVisibility(SrcGV->getVisibility());
358ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  if (const Function *SrcF = dyn_cast<Function>(SrcGV)) {
359ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer    Function *DestF = cast<Function>(DestGV);
360ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer    DestF->setCallingConv(SrcF->getCallingConv());
361dc024674ff96820d6020757b48d47f46d4c07db2Duncan Sands    DestF->setParamAttrs(SrcF->getParamAttrs());
362194c90ed2a8e74b5a1c5184835f84c572d524dadGordon Henriksen    if (SrcF->hasCollector())
363194c90ed2a8e74b5a1c5184835f84c572d524dadGordon Henriksen      DestF->setCollector(SrcF->getCollector());
364e20c814518e3322f5bf5f83e9ca42bd6bdd40745Anton Korobeynikov  } else if (const GlobalVariable *SrcVar = dyn_cast<GlobalVariable>(SrcGV)) {
365e20c814518e3322f5bf5f83e9ca42bd6bdd40745Anton Korobeynikov    GlobalVariable *DestVar = cast<GlobalVariable>(DestGV);
366e20c814518e3322f5bf5f83e9ca42bd6bdd40745Anton Korobeynikov    DestVar->setThreadLocal(SrcVar->isThreadLocal());
367ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  }
368c003628a612d3687fb77088a5894314210a65385Chris Lattner}
369c003628a612d3687fb77088a5894314210a65385Chris Lattner
370aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// GetLinkageResult - This analyzes the two global values and determines what
371aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// the result will look like in the destination module.  In particular, it
372aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// computes the resultant linkage type, computes whether the global in the
373aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// source should be copied over to the destination (replacing the existing
3749cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov/// one), and computes whether this linkage is an error or not. It also performs
3759cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov/// visibility checks: we cannot link together two symbols with different
3769cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov/// visibilities.
377968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikovstatic bool GetLinkageResult(GlobalValue *Dest, const GlobalValue *Src,
378aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner                             GlobalValue::LinkageTypes &LT, bool &LinkFromSrc,
379aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner                             std::string *Err) {
380aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  assert((!Dest || !Src->hasInternalLinkage()) &&
381aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner         "If Src has internal linkage, Dest shouldn't be set!");
382aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  if (!Dest) {
383aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    // Linking something to nothing.
384aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LinkFromSrc = true;
385aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LT = Src->getLinkage();
3865cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer  } else if (Src->isDeclaration()) {
3872b48ef0450944c2c46633aec9baf6be835a3b503Anton Korobeynikov    // If Src is external or if both Src & Dest are external..  Just link the
388aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    // external globals, we aren't adding anything.
389b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov    if (Src->hasDLLImportLinkage()) {
39078ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      // If one of GVs has DLLImport linkage, result should be dllimport'ed.
3915cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer      if (Dest->isDeclaration()) {
392b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov        LinkFromSrc = true;
393b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov        LT = Src->getLinkage();
394b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov      }
3958753c447a5f8eec1421331c40ffbb72c6d411f78Andrew Lenharth    } else if (Dest->hasExternalWeakLinkage()) {
3968753c447a5f8eec1421331c40ffbb72c6d411f78Andrew Lenharth      //If the Dest is weak, use the source linkage
3978753c447a5f8eec1421331c40ffbb72c6d411f78Andrew Lenharth      LinkFromSrc = true;
3988753c447a5f8eec1421331c40ffbb72c6d411f78Andrew Lenharth      LT = Src->getLinkage();
399b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov    } else {
400b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov      LinkFromSrc = false;
401b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov      LT = Dest->getLinkage();
402b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov    }
4035cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer  } else if (Dest->isDeclaration() && !Dest->hasDLLImportLinkage()) {
404aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    // If Dest is external but Src is not:
405aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LinkFromSrc = true;
406aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LT = Src->getLinkage();
407aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  } else if (Src->hasAppendingLinkage() || Dest->hasAppendingLinkage()) {
408aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    if (Src->getLinkage() != Dest->getLinkage())
409aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      return Error(Err, "Linking globals named '" + Src->getName() +
410aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner            "': can only link appending global with another appending global!");
411aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LinkFromSrc = true; // Special cased.
412aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LT = Src->getLinkage();
413aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  } else if (Src->hasWeakLinkage() || Src->hasLinkOnceLinkage()) {
414619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer    // At this point we know that Dest has LinkOnce, External*, Weak, or
415619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer    // DLL* linkage.
41678ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov    if ((Dest->hasLinkOnceLinkage() && Src->hasWeakLinkage()) ||
41778ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov        Dest->hasExternalWeakLinkage()) {
418aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      LinkFromSrc = true;
419aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      LT = Src->getLinkage();
420aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    } else {
421aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      LinkFromSrc = false;
422aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      LT = Dest->getLinkage();
423aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    }
424aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  } else if (Dest->hasWeakLinkage() || Dest->hasLinkOnceLinkage()) {
42578ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov    // At this point we know that Src has External* or DLL* linkage.
42678ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov    if (Src->hasExternalWeakLinkage()) {
42778ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      LinkFromSrc = false;
42878ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      LT = Dest->getLinkage();
42978ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov    } else {
43078ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      LinkFromSrc = true;
43178ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      LT = GlobalValue::ExternalLinkage;
43278ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov    }
433aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  } else {
434b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov    assert((Dest->hasExternalLinkage() ||
435b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov            Dest->hasDLLImportLinkage() ||
43678ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov            Dest->hasDLLExportLinkage() ||
43778ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov            Dest->hasExternalWeakLinkage()) &&
438b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov           (Src->hasExternalLinkage() ||
439b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov            Src->hasDLLImportLinkage() ||
44078ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov            Src->hasDLLExportLinkage() ||
44178ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov            Src->hasExternalWeakLinkage()) &&
442aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner           "Unexpected linkage type!");
443f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman    return Error(Err, "Linking globals named '" + Src->getName() +
444aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner                 "': symbol multiply defined!");
445aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  }
4469cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov
4479cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov  // Check visibility
4489cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov  if (Dest && Src->getVisibility() != Dest->getVisibility())
44997f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner    if (!Src->isDeclaration() && !Dest->isDeclaration())
45097f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner      return Error(Err, "Linking globals named '" + Src->getName() +
45197f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner                   "': symbols have different visibilities!");
452aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  return false;
453aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner}
4545c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
4555c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// LinkGlobals - Loop through the global variables in the src module and merge
4568166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner// them into the dest module.
457968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikovstatic bool LinkGlobals(Module *Dest, const Module *Src,
4585c2d335d86579941b71816196a95a64b277e8963Chris Lattner                        std::map<const Value*, Value*> &ValueMap,
4598166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                    std::multimap<std::string, GlobalVariable *> &AppendingVars,
4605c2d335d86579941b71816196a95a64b277e8963Chris Lattner                        std::string *Err) {
4615c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  // Loop over all of the globals in the src module, mapping them over as we go
462968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov  for (Module::const_global_iterator I = Src->global_begin(), E = Src->global_end();
4631127315562b221010040925980040fd4f65bdb1cChris Lattner       I != E; ++I) {
464968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov    const GlobalVariable *SGV = I;
46501f69399352380532d480f6749daed660f50b15cAnton Korobeynikov    GlobalValue *DGV = 0;
46601f69399352380532d480f6749daed660f50b15cAnton Korobeynikov
46701f69399352380532d480f6749daed660f50b15cAnton Korobeynikov    // Check to see if may have to link the global with the global
468ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer    if (SGV->hasName() && !SGV->hasInternalLinkage()) {
469ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      DGV = Dest->getGlobalVariable(SGV->getName());
470ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      if (DGV && DGV->getType() != SGV->getType())
471ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // If types don't agree due to opaque types, try to resolve them.
472ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        RecursiveResolveTypes(SGV->getType(), DGV->getType(),
473ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer                              &Dest->getTypeSymbolTable(), "");
474ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer    }
4755c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
47601f69399352380532d480f6749daed660f50b15cAnton Korobeynikov    // Check to see if may have to link the global with the alias
477aeb09967fdb2ff074c91ce8039ebd1917a2f6060Anton Korobeynikov    if (!DGV && SGV->hasName() && !SGV->hasInternalLinkage()) {
47801f69399352380532d480f6749daed660f50b15cAnton Korobeynikov      DGV = Dest->getNamedAlias(SGV->getName());
47901f69399352380532d480f6749daed660f50b15cAnton Korobeynikov      if (DGV && DGV->getType() != SGV->getType())
48001f69399352380532d480f6749daed660f50b15cAnton Korobeynikov        // If types don't agree due to opaque types, try to resolve them.
48101f69399352380532d480f6749daed660f50b15cAnton Korobeynikov        RecursiveResolveTypes(SGV->getType(), DGV->getType(),
48201f69399352380532d480f6749daed660f50b15cAnton Korobeynikov                              &Dest->getTypeSymbolTable(), "");
48301f69399352380532d480f6749daed660f50b15cAnton Korobeynikov    }
48401f69399352380532d480f6749daed660f50b15cAnton Korobeynikov
485aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    if (DGV && DGV->hasInternalLinkage())
486aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      DGV = 0;
487aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner
488c318329a1098a19137398af4d880f4d8cdd9cd8dDan Gohman    assert((SGV->hasInitializer() || SGV->hasExternalWeakLinkage() ||
489c318329a1098a19137398af4d880f4d8cdd9cd8dDan Gohman            SGV->hasExternalLinkage() || SGV->hasDLLImportLinkage()) &&
4904ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner           "Global must either be external or have an initializer!");
4915c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
492b324bd735ff5c99a52cd3c3f5d01c0e1398a2d3aChris Lattner    GlobalValue::LinkageTypes NewLinkage = GlobalValue::InternalLinkage;
493b324bd735ff5c99a52cd3c3f5d01c0e1398a2d3aChris Lattner    bool LinkFromSrc = false;
494aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    if (GetLinkageResult(DGV, SGV, NewLinkage, LinkFromSrc, Err))
495aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      return true;
4960fec08eb58dd9fffeb72c584aa61a59d71111c8dChris Lattner
497aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    if (!DGV) {
4985c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner      // No linking to be performed, simply create an identical version of the
4998d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      // symbol over in the dest module... the initializer will be filled in
5008d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      // later by LinkGlobalInits...
5012719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      GlobalVariable *NewDGV =
5022719bac90525985f4ebb97324b4c087d47b54384Chris Lattner        new GlobalVariable(SGV->getType()->getElementType(),
5032719bac90525985f4ebb97324b4c087d47b54384Chris Lattner                           SGV->isConstant(), SGV->getLinkage(), /*init*/0,
504e20c814518e3322f5bf5f83e9ca42bd6bdd40745Anton Korobeynikov                           SGV->getName(), Dest);
505471feac1b297b9458de18d973af371bf7f10296bReid Spencer      // Propagate alignment, visibility and section info.
506ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      CopyGVAttributes(NewDGV, SGV);
5075dfbaf1fe3fa6914095cc55cde20b1ed25187b56Andrew Lenharth
5082719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      // If the LLVM runtime renamed the global, but it is an externally visible
5092719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      // symbol, DGV must be an existing global with internal linkage.  Rename
5102719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      // it.
511c003628a612d3687fb77088a5894314210a65385Chris Lattner      if (NewDGV->getName() != SGV->getName() && !NewDGV->hasInternalLinkage())
512c003628a612d3687fb77088a5894314210a65385Chris Lattner        ForceRenaming(NewDGV, SGV->getName());
5135c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
5145c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner      // Make sure to remember this mapping...
515817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov      ValueMap[SGV] = NewDGV;
516817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov
5178166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      if (SGV->hasAppendingLinkage())
5188166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        // Keep track that this is an appending variable...
5198166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        AppendingVars.insert(std::make_pair(SGV->getName(), NewDGV));
520aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    } else if (DGV->hasAppendingLinkage()) {
5218166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // No linking is performed yet.  Just insert a new copy of the global, and
5228166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // keep track of the fact that it is an appending variable in the
5238166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // AppendingVars map.  The name is cleared out so that no linkage is
5248166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // performed.
5258166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      GlobalVariable *NewDGV =
5268166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        new GlobalVariable(SGV->getType()->getElementType(),
5278166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                           SGV->isConstant(), SGV->getLinkage(), /*init*/0,
528e20c814518e3322f5bf5f83e9ca42bd6bdd40745Anton Korobeynikov                           "", Dest);
5298166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
53075c7915c3a24f8063c274ccffe2dc9b7bc8bce1fAnton Korobeynikov      // Set alignment allowing CopyGVAttributes merge it with alignment of SGV.
531ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      NewDGV->setAlignment(DGV->getAlignment());
53275c7915c3a24f8063c274ccffe2dc9b7bc8bce1fAnton Korobeynikov      // Propagate alignment, section and visibility info.
533ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      CopyGVAttributes(NewDGV, SGV);
5345dfbaf1fe3fa6914095cc55cde20b1ed25187b56Andrew Lenharth
5358166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Make sure to remember this mapping...
536817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov      ValueMap[SGV] = NewDGV;
5378166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
5388166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Keep track that this is an appending variable...
5398166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      AppendingVars.insert(std::make_pair(SGV->getName(), NewDGV));
54001f69399352380532d480f6749daed660f50b15cAnton Korobeynikov    } else if (GlobalAlias *DGA = dyn_cast<GlobalAlias>(DGV)) {
54101f69399352380532d480f6749daed660f50b15cAnton Korobeynikov      // SGV is global, but DGV is alias. The only valid mapping is when SGV is
54201f69399352380532d480f6749daed660f50b15cAnton Korobeynikov      // external declaration, which is effectively a no-op. Also make sure
543817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov      // linkage calculation was correct.
54401f69399352380532d480f6749daed660f50b15cAnton Korobeynikov      if (SGV->isDeclaration() && !LinkFromSrc) {
54501f69399352380532d480f6749daed660f50b15cAnton Korobeynikov        // Make sure to remember this mapping...
546817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov        ValueMap[SGV] = DGA;
54701f69399352380532d480f6749daed660f50b15cAnton Korobeynikov      } else
5481438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov        return Error(Err, "Global-Alias Collision on '" + SGV->getName() +
5491438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov                     "': symbol multiple defined");
55001f69399352380532d480f6749daed660f50b15cAnton Korobeynikov    } else if (GlobalVariable *DGVar = dyn_cast<GlobalVariable>(DGV)) {
551817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov      // Otherwise, perform the global-global mapping as instructed by
552817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov      // GetLinkageResult.
553aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      if (LinkFromSrc) {
554968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov        // Propagate alignment, section, and visibility info.
55501f69399352380532d480f6749daed660f50b15cAnton Korobeynikov        CopyGVAttributes(DGVar, SGV);
556968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov
557968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov        // If the types don't match, and if we are to link from the source, nuke
558968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov        // DGV and create a new one of the appropriate type.
55901f69399352380532d480f6749daed660f50b15cAnton Korobeynikov        if (SGV->getType() != DGVar->getType()) {
560968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov          GlobalVariable *NewDGV =
561968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov            new GlobalVariable(SGV->getType()->getElementType(),
56201f69399352380532d480f6749daed660f50b15cAnton Korobeynikov                               DGVar->isConstant(), DGVar->getLinkage(),
56301f69399352380532d480f6749daed660f50b15cAnton Korobeynikov                               /*init*/0, DGVar->getName(), Dest);
56401f69399352380532d480f6749daed660f50b15cAnton Korobeynikov          CopyGVAttributes(NewDGV, DGVar);
565968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov          DGV->replaceAllUsesWith(ConstantExpr::getBitCast(NewDGV,
56601f69399352380532d480f6749daed660f50b15cAnton Korobeynikov                                                           DGVar->getType()));
56701f69399352380532d480f6749daed660f50b15cAnton Korobeynikov          // DGVar will conflict with NewDGV because they both had the same
568968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov          // name. We must erase this now so ForceRenaming doesn't assert
569968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov          // because DGV might not have internal linkage.
57001f69399352380532d480f6749daed660f50b15cAnton Korobeynikov          DGVar->eraseFromParent();
571968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov
572968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov          // If the symbol table renamed the global, but it is an externally
573968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov          // visible symbol, DGV must be an existing global with internal
574968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov          // linkage. Rename it.
575968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov          if (NewDGV->getName() != SGV->getName() &&
576968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov              !NewDGV->hasInternalLinkage())
577968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov            ForceRenaming(NewDGV, SGV->getName());
578968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov
57901f69399352380532d480f6749daed660f50b15cAnton Korobeynikov          DGVar = NewDGV;
580968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov        }
581968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov
582aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner        // Inherit const as appropriate
58301f69399352380532d480f6749daed660f50b15cAnton Korobeynikov        DGVar->setConstant(SGV->isConstant());
584968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov
585968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov        // Set initializer to zero, so we can link the stuff later
58601f69399352380532d480f6749daed660f50b15cAnton Korobeynikov        DGVar->setInitializer(0);
587aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      } else {
588968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov        // Special case for const propagation
58901f69399352380532d480f6749daed660f50b15cAnton Korobeynikov        if (DGVar->isDeclaration() && SGV->isConstant() && !DGVar->isConstant())
59001f69399352380532d480f6749daed660f50b15cAnton Korobeynikov          DGVar->setConstant(true);
591aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      }
592aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner
593968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov      // Set calculated linkage
59401f69399352380532d480f6749daed660f50b15cAnton Korobeynikov      DGVar->setLinkage(NewLinkage);
595968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov
596968e39a5aba998a126278425da8287aae358e7a7Anton Korobeynikov      // Make sure to remember this mapping...
597817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov      ValueMap[SGV] = ConstantExpr::getBitCast(DGVar, SGV->getType());
5985c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner    }
5995c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  }
6005c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return false;
6015c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
6025c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
60358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikovstatic GlobalValue::LinkageTypes
60458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton KorobeynikovCalculateAliasLinkage(const GlobalValue *SGV, const GlobalValue *DGV) {
60558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov  if (SGV->hasExternalLinkage() || DGV->hasExternalLinkage())
60658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    return GlobalValue::ExternalLinkage;
60758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov  else if (SGV->hasWeakLinkage() || DGV->hasWeakLinkage())
60858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    return GlobalValue::WeakLinkage;
60958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov  else {
61058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    assert(SGV->hasInternalLinkage() && DGV->hasInternalLinkage() &&
61158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov           "Unexpected linkage type");
61258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    return GlobalValue::InternalLinkage;
61358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov  }
61458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov}
61558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
61631ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio// LinkAlias - Loop through the alias in the src module and link them into the
61758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov// dest module. We're assuming, that all functions/global variables were already
61858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov// linked in.
6194fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikovstatic bool LinkAlias(Module *Dest, const Module *Src,
6204fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov                      std::map<const Value*, Value*> &ValueMap,
6214fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov                      std::string *Err) {
62231ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio  // Loop over all alias in the src module
62331ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio  for (Module::const_alias_iterator I = Src->alias_begin(),
62431ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio         E = Src->alias_end(); I != E; ++I) {
62558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    const GlobalAlias *SGA = I;
62658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    const GlobalValue *SAliasee = SGA->getAliasedGlobal();
62758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    GlobalAlias *NewGA = NULL;
62858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
62958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    // Globals were already linked, thus we can just query ValueMap for variant
63058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    // of SAliasee in Dest
63158d5e051574339106ce4139697ebadd21e34dbc0Ted Kremenek    std::map<const Value*,Value*>::const_iterator VMI = ValueMap.find(SAliasee);
63258d5e051574339106ce4139697ebadd21e34dbc0Ted Kremenek    assert(VMI != ValueMap.end() && "Aliasee not linked");
63358d5e051574339106ce4139697ebadd21e34dbc0Ted Kremenek    GlobalValue* DAliasee = cast<GlobalValue>(VMI->second);
63458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
63558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    // Try to find something 'similar' to SGA in destination module.
63658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    if (GlobalAlias *DGA = Dest->getNamedAlias(SGA->getName())) {
63758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      // If types don't agree due to opaque types, try to resolve them.
63858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      if (RecursiveResolveTypes(SGA->getType(), DGA->getType(),
63958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov                                &Dest->getTypeSymbolTable(), ""))
6401438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov        return Error(Err, "Alias Collision on '" + SGA->getName()+
6411438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov                     "': aliases have different types");
64258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
64358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      // Now types are known to be the same, check whether aliasees equal. As
64458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      // globals are already linked we just need query ValueMap to find the
64558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      // mapping.
64658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      if (DAliasee == DGA->getAliasedGlobal()) {
64758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // This is just two copies of the same alias. Propagate linkage, if
64858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // necessary.
64958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        DGA->setLinkage(CalculateAliasLinkage(SGA, DGA));
65058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
65158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        NewGA = DGA;
65258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // Proceed to 'common' steps
65358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      } else
6541438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov        return Error(Err, "Alias Collision on '"  + SGA->getName()+
6551438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov                     "': aliases have different aliasees");
65658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    } else if (GlobalVariable *DGV = Dest->getGlobalVariable(SGA->getName())) {
65758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      RecursiveResolveTypes(SGA->getType(), DGV->getType(),
65858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov                            &Dest->getTypeSymbolTable(), "");
659b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov
66058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      // The only allowed way is to link alias with external declaration.
66158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      if (DGV->isDeclaration()) {
662ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov        // But only if aliasee is global too...
663ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov        if (!isa<GlobalVariable>(DAliasee))
664ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov            return Error(Err, "Global-Alias Collision on '" + SGA->getName() +
665ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov                         "': aliasee is not global variable");
666ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov
66758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        NewGA = new GlobalAlias(SGA->getType(), SGA->getLinkage(),
66858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov                                SGA->getName(), DAliasee, Dest);
66958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        CopyGVAttributes(NewGA, SGA);
67058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
67158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // Any uses of DGV need to change to NewGA, with cast, if needed.
67258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        if (SGA->getType() != DGV->getType())
67358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov          DGV->replaceAllUsesWith(ConstantExpr::getBitCast(NewGA,
67458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov                                                           DGV->getType()));
67558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        else
67658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov          DGV->replaceAllUsesWith(NewGA);
67758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
67858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // DGV will conflict with NewGA because they both had the same
67958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // name. We must erase this now so ForceRenaming doesn't assert
68058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // because DGV might not have internal linkage.
68158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        DGV->eraseFromParent();
68258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
68358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        // Proceed to 'common' steps
68458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      } else
6851438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov        return Error(Err, "Global-Alias Collision on '" + SGA->getName() +
6861438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov                     "': symbol multiple defined");
68758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    } else if (Function *DF = Dest->getFunction(SGA->getName())) {
68858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      RecursiveResolveTypes(SGA->getType(), DF->getType(),
68958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov                            &Dest->getTypeSymbolTable(), "");
690b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov
691b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov      // The only allowed way is to link alias with external declaration.
692b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov      if (DF->isDeclaration()) {
693ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov        // But only if aliasee is function too...
694ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov        if (!isa<Function>(DAliasee))
695ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov            return Error(Err, "Function-Alias Collision on '" + SGA->getName() +
696ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov                         "': aliasee is not function");
697ed61c0bc76a7b9001d7639a908c955d0fae35bacAnton Korobeynikov
698b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        NewGA = new GlobalAlias(SGA->getType(), SGA->getLinkage(),
699b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov                                SGA->getName(), DAliasee, Dest);
700b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        CopyGVAttributes(NewGA, SGA);
701b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov
702b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        // Any uses of DF need to change to NewGA, with cast, if needed.
703b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        if (SGA->getType() != DF->getType())
704b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov          DF->replaceAllUsesWith(ConstantExpr::getBitCast(NewGA,
705b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov                                                          DF->getType()));
706b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        else
707b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov          DF->replaceAllUsesWith(NewGA);
708b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov
709b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        // DF will conflict with NewGA because they both had the same
710b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        // name. We must erase this now so ForceRenaming doesn't assert
711b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        // because DF might not have internal linkage.
712b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        DF->eraseFromParent();
713b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov
714b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov        // Proceed to 'common' steps
715b5a4bd86df527c7d164197906861d2ca6f48678dAnton Korobeynikov      } else
7161438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov        return Error(Err, "Function-Alias Collision on '" + SGA->getName() +
7171438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov                     "': symbol multiple defined");
71858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    } else {
71958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      // Nothing similar found, just copy alias into destination module.
72058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
72158887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      NewGA = new GlobalAlias(SGA->getType(), SGA->getLinkage(),
72258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov                              SGA->getName(), DAliasee, Dest);
72358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      CopyGVAttributes(NewGA, SGA);
72458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
72558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      // Proceed to 'common' steps
72658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    }
72758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
72858887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    assert(NewGA && "No alias was created in destination module!");
72958887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
730b8cdaf7099898d7c0eef4769bf6a89f39e7decc2Anton Korobeynikov    // If the symbol table renamed the alias, but it is an externally visible
731b8cdaf7099898d7c0eef4769bf6a89f39e7decc2Anton Korobeynikov    // symbol, DGV must be an global value with internal linkage. Rename it.
73258887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    if (NewGA->getName() != SGA->getName() &&
73358887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov        !NewGA->hasInternalLinkage())
73458887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov      ForceRenaming(NewGA, SGA->getName());
73558887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
73658887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    // Remember this mapping so uses in the source module get remapped
73758887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov    // later by RemapOperand.
738817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov    ValueMap[SGA] = NewGA;
73931ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio  }
74058887bc59316bcaf0c0675a45ddecd6994f3fbc6Anton Korobeynikov
74131ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio  return false;
74231ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio}
74331ed0fb804dd86fb15093e114701932119914400Lauro Ramos Venancio
7445c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
7458d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner// LinkGlobalInits - Update the initializers in the Dest module now that all
7468d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner// globals that may be referenced are in Dest.
7478d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattnerstatic bool LinkGlobalInits(Module *Dest, const Module *Src,
7485c2d335d86579941b71816196a95a64b277e8963Chris Lattner                            std::map<const Value*, Value*> &ValueMap,
7495c2d335d86579941b71816196a95a64b277e8963Chris Lattner                            std::string *Err) {
7508d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner
7518d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  // Loop over all of the globals in the src module, mapping them over as we go
7521127315562b221010040925980040fd4f65bdb1cChris Lattner  for (Module::const_global_iterator I = Src->global_begin(),
7531127315562b221010040925980040fd4f65bdb1cChris Lattner       E = Src->global_end(); I != E; ++I) {
75418961504fc2b299578dba817900a0696cf3ccc4dChris Lattner    const GlobalVariable *SGV = I;
7558d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner
7568d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner    if (SGV->hasInitializer()) {      // Only process initialized GV's
7578d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      // Figure out what the initializer looks like in the dest module...
7584ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner      Constant *SInit =
7590033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner        cast<Constant>(RemapOperand(SGV->getInitializer(), ValueMap));
7608d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner
761f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman      GlobalVariable *DGV = cast<GlobalVariable>(ValueMap[SGV]);
7624ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner      if (DGV->hasInitializer()) {
7634ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner        if (SGV->hasExternalLinkage()) {
7644ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner          if (DGV->getInitializer() != SInit)
7651438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov            return Error(Err, "Global Variable Collision on '" + SGV->getName() +
7661438b9dd82887b8faa2c4670dca45965609a43c6Anton Korobeynikov                         "': global variables have different initializers");
76772ac148d4964fbff950fe11a0eeeda73973b849bChris Lattner        } else if (DGV->hasLinkOnceLinkage() || DGV->hasWeakLinkage()) {
7684ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner          // Nothing is required, mapped values will take the new global
7694ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner          // automatically.
77057cb9883aa4c82357594d4186858f175135be3b1Chris Lattner        } else if (SGV->hasLinkOnceLinkage() || SGV->hasWeakLinkage()) {
77157cb9883aa4c82357594d4186858f175135be3b1Chris Lattner          // Nothing is required, mapped values will take the new global
77257cb9883aa4c82357594d4186858f175135be3b1Chris Lattner          // automatically.
7734ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner        } else if (DGV->hasAppendingLinkage()) {
7744ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner          assert(0 && "Appending linkage unimplemented!");
7754ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner        } else {
7764ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner          assert(0 && "Unknown linkage!");
7774ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner        }
7788d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      } else {
7798d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner        // Copy the initializer over now...
7804ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner        DGV->setInitializer(SInit);
7818d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      }
7828d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner    }
7838d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  }
7848d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  return false;
7858d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner}
7865c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
78779df7c0aaa18129e55968c8783ef8346807bd4afChris Lattner// LinkFunctionProtos - Link the functions together between the two modules,
788c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// without doing function bodies... this just adds external function prototypes
789c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// to the Dest function...
7905c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner//
79179df7c0aaa18129e55968c8783ef8346807bd4afChris Lattnerstatic bool LinkFunctionProtos(Module *Dest, const Module *Src,
7925c2d335d86579941b71816196a95a64b277e8963Chris Lattner                               std::map<const Value*, Value*> &ValueMap,
7935c2d335d86579941b71816196a95a64b277e8963Chris Lattner                               std::string *Err) {
794619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer  // Loop over all of the functions in the src module, mapping them over
7955c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  for (Module::const_iterator I = Src->begin(), E = Src->end(); I != E; ++I) {
79618961504fc2b299578dba817900a0696cf3ccc4dChris Lattner    const Function *SF = I;   // SrcFunction
7974ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner    Function *DF = 0;
7988bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer    if (SF->hasName() && !SF->hasInternalLinkage()) {
7998bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer      // Check to see if may have to link the function.
800ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      DF = Dest->getFunction(SF->getName());
801ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      if (DF && SF->getType() != DF->getType())
802ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // If types don't agree because of opaque, try to resolve them
803ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        RecursiveResolveTypes(SF->getType(), DF->getType(),
804ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer                              &Dest->getTypeSymbolTable(), "");
8058bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer    }
8069cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov
8079cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov    // Check visibility
8089cd3ccf5065a8a139e458d016c88a8512471598bAnton Korobeynikov    if (DF && !DF->hasInternalLinkage() &&
80997f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner        SF->getVisibility() != DF->getVisibility()) {
81097f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner      // If one is a prototype, ignore its visibility.  Prototypes are always
81197f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner      // overridden by the definition.
81297f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner      if (!SF->isDeclaration() && !DF->isDeclaration())
81397f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner        return Error(Err, "Linking functions named '" + SF->getName() +
81497f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner                     "': symbols have different visibilities!");
81597f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner    }
816ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer
817650d5053bfb388bc81aecd202f16d2e128e1f23eAnton Korobeynikov    if (DF && DF->hasInternalLinkage())
818650d5053bfb388bc81aecd202f16d2e128e1f23eAnton Korobeynikov      DF = NULL;
819650d5053bfb388bc81aecd202f16d2e128e1f23eAnton Korobeynikov
820ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer    if (DF && DF->getType() != SF->getType()) {
821ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      if (DF->isDeclaration() && !SF->isDeclaration()) {
822ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // We have a definition of the same name but different type in the
823ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // source module. Copy the prototype to the destination and replace
824ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // uses of the destination's prototype with the new prototype.
825ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        Function *NewDF = new Function(SF->getFunctionType(), SF->getLinkage(),
826ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer                                       SF->getName(), Dest);
827ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        CopyGVAttributes(NewDF, SF);
828ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer
829ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // Any uses of DF need to change to NewDF, with cast
830ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        DF->replaceAllUsesWith(ConstantExpr::getBitCast(NewDF, DF->getType()));
831ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer
832ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // DF will conflict with NewDF because they both had the same. We must
833ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // erase this now so ForceRenaming doesn't assert because DF might
834ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // not have internal linkage.
835ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        DF->eraseFromParent();
836ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer
837ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // If the symbol table renamed the function, but it is an externally
838ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // visible symbol, DF must be an existing function with internal
839ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // linkage.  Rename it.
840ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        if (NewDF->getName() != SF->getName() && !NewDF->hasInternalLinkage())
841ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer          ForceRenaming(NewDF, SF->getName());
842ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer
843ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // Remember this mapping so uses in the source module get remapped
844ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // later by RemapOperand.
845ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        ValueMap[SF] = NewDF;
846ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      } else if (SF->isDeclaration()) {
847ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // We have two functions of the same name but different type and the
848ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // source is a declaration while the destination is not. Any use of
849ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // the source must be mapped to the destination, with a cast.
850ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        ValueMap[SF] = ConstantExpr::getBitCast(DF, SF->getType());
851ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      } else {
852ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // We have two functions of the same name but different types and they
853ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        // are both definitions. This is an error.
854ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        return Error(Err, "Function '" + DF->getName() + "' defined as both '" +
855ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer                     ToStr(SF->getFunctionType(), Src) + "' and '" +
856ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer                     ToStr(DF->getFunctionType(), Dest) + "'");
857ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      }
858ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer    } else if (!DF || SF->hasInternalLinkage() || DF->hasInternalLinkage()) {
8590fec08eb58dd9fffeb72c584aa61a59d71111c8dChris Lattner      // Function does not already exist, simply insert an function signature
86097f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner      // identical to SF into the dest module.
8612719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      Function *NewDF = new Function(SF->getFunctionType(), SF->getLinkage(),
8622719bac90525985f4ebb97324b4c087d47b54384Chris Lattner                                     SF->getName(), Dest);
863ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      CopyGVAttributes(NewDF, SF);
8642719bac90525985f4ebb97324b4c087d47b54384Chris Lattner
8652719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      // If the LLVM runtime renamed the function, but it is an externally
8662719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      // visible symbol, DF must be an existing function with internal linkage.
8672719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      // Rename it.
868c003628a612d3687fb77088a5894314210a65385Chris Lattner      if (NewDF->getName() != SF->getName() && !NewDF->hasInternalLinkage())
86982b5b219dd3b9035442389ddcd58e6f859aae039Chris Lattner        ForceRenaming(NewDF, SF->getName());
8704ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner
8714ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner      // ... and remember this mapping...
872817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov      ValueMap[SF] = NewDF;
8735cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer    } else if (SF->isDeclaration()) {
874ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      // If SF is a declaration or if both SF & DF are declarations, just link
875ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      // the declarations, we aren't adding anything.
876b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov      if (SF->hasDLLImportLinkage()) {
8775cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer        if (DF->isDeclaration()) {
878b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov          ValueMap.insert(std::make_pair(SF, DF));
879b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov          DF->setLinkage(SF->getLinkage());
880b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov        }
881b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov      } else {
882817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov        ValueMap[SF] = DF;
8838bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer      }
8845cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer    } else if (DF->isDeclaration() && !DF->hasDLLImportLinkage()) {
885b74ed07bfd3af42331b1964c24c39912610a08f4Anton Korobeynikov      // If DF is external but SF is not...
886c2b97d43a074445c6856d69538a95de09450c26eChris Lattner      // Link the external functions, update linkage qualifiers
887c2b97d43a074445c6856d69538a95de09450c26eChris Lattner      ValueMap.insert(std::make_pair(SF, DF));
888c2b97d43a074445c6856d69538a95de09450c26eChris Lattner      DF->setLinkage(SF->getLinkage());
88997f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner      // Visibility of prototype is overridden by vis of definition.
89097f8b0900abddfa72fd59e777399280adb4ca0a6Chris Lattner      DF->setVisibility(SF->getVisibility());
89135956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner    } else if (SF->hasWeakLinkage() || SF->hasLinkOnceLinkage()) {
89278ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      // At this point we know that DF has LinkOnce, Weak, or External* linkage.
893817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov      ValueMap[SF] = DF;
89472ac148d4964fbff950fe11a0eeeda73973b849bChris Lattner
89535956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner      // Linkonce+Weak = Weak
89678ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      // *+External Weak = *
89778ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      if ((DF->hasLinkOnceLinkage() && SF->hasWeakLinkage()) ||
89878ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov          DF->hasExternalWeakLinkage())
89935956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner        DF->setLinkage(SF->getLinkage());
90035956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner    } else if (DF->hasWeakLinkage() || DF->hasLinkOnceLinkage()) {
90178ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      // At this point we know that SF has LinkOnce or External* linkage.
902817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov      ValueMap[SF] = DF;
90378ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov      if (!SF->hasLinkOnceLinkage() && !SF->hasExternalWeakLinkage())
90478ee7b78c3c47b71c4b7a1475438d6574216a64bAnton Korobeynikov        // Don't inherit linkonce & external weak linkage
90572ac148d4964fbff950fe11a0eeeda73973b849bChris Lattner        DF->setLinkage(SF->getLinkage());
906c2b97d43a074445c6856d69538a95de09450c26eChris Lattner    } else if (SF->getLinkage() != DF->getLinkage()) {
907ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer        return Error(Err, "Functions named '" + SF->getName() +
908ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer                     "' have different linkage specifiers!");
909c2b97d43a074445c6856d69538a95de09450c26eChris Lattner    } else if (SF->hasExternalLinkage()) {
910ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      // The function is defined identically in both modules!!
911f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman      return Error(Err, "Function '" +
912f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman                   ToStr(SF->getFunctionType(), Src) + "':\"" +
913c2b97d43a074445c6856d69538a95de09450c26eChris Lattner                   SF->getName() + "\" - Function is already defined!");
914c2b97d43a074445c6856d69538a95de09450c26eChris Lattner    } else {
915c2b97d43a074445c6856d69538a95de09450c26eChris Lattner      assert(0 && "Unknown linkage configuration found!");
9165c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner    }
9175c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  }
9185c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return false;
9195c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
9205c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
921c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// LinkFunctionBody - Copy the source function over into the dest function and
922c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// fix up references to values.  At this point we know that Dest is an external
923c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// function, and that Src is not.
9244bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattnerstatic bool LinkFunctionBody(Function *Dest, Function *Src,
925ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer                             std::map<const Value*, Value*> &ValueMap,
9265c2d335d86579941b71816196a95a64b277e8963Chris Lattner                             std::string *Err) {
9275cbf985dcbc89fba3208e7baf8b6f488b06d3ec9Reid Spencer  assert(Src && Dest && Dest->isDeclaration() && !Src->isDeclaration());
9285c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
9290033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner  // Go through and convert function arguments over, remembering the mapping.
930e4d5c441e04bdc00ccf1804744af670655123b07Chris Lattner  Function::arg_iterator DI = Dest->arg_begin();
931e4d5c441e04bdc00ccf1804744af670655123b07Chris Lattner  for (Function::arg_iterator I = Src->arg_begin(), E = Src->arg_end();
93269da5cf26143e4542d4bf8c78ffac6d079efe5c9Chris Lattner       I != E; ++I, ++DI) {
93369da5cf26143e4542d4bf8c78ffac6d079efe5c9Chris Lattner    DI->setName(I->getName());  // Copy the name information over...
9345c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
9355c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner    // Add a mapping to our local map
936817bf2aeb22db13d03beb15e9bf17c425d0c694dAnton Korobeynikov    ValueMap[I] = DI;
9375c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  }
9385c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
9394bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner  // Splice the body of the source function into the dest function.
9404bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner  Dest->getBasicBlockList().splice(Dest->end(), Src->getBasicBlockList());
9415c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
942c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // At this point, all of the instructions and values of the function are now
943c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // copied over.  The only problem is that they are still referencing values in
944c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // the Source function as operands.  Loop through all of the operands of the
945c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // functions and patch them up to point to the local versions...
9465c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  //
94718961504fc2b299578dba817900a0696cf3ccc4dChris Lattner  for (Function::iterator BB = Dest->begin(), BE = Dest->end(); BB != BE; ++BB)
94818961504fc2b299578dba817900a0696cf3ccc4dChris Lattner    for (BasicBlock::iterator I = BB->begin(), E = BB->end(); I != E; ++I)
94918961504fc2b299578dba817900a0696cf3ccc4dChris Lattner      for (Instruction::op_iterator OI = I->op_begin(), OE = I->op_end();
950221d688a5ef21a22c2368c9fff0e92d7966c95e5Chris Lattner           OI != OE; ++OI)
9514bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner        if (!isa<Instruction>(*OI) && !isa<BasicBlock>(*OI))
952ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer          *OI = RemapOperand(*OI, ValueMap);
9530033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner
9540033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner  // There is no need to map the arguments anymore.
9551127315562b221010040925980040fd4f65bdb1cChris Lattner  for (Function::arg_iterator I = Src->arg_begin(), E = Src->arg_end();
9561127315562b221010040925980040fd4f65bdb1cChris Lattner       I != E; ++I)
957ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer    ValueMap.erase(I);
9585c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
9595c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return false;
9605c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
9615c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
9625c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
963c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// LinkFunctionBodies - Link in the function bodies that are defined in the
964c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// source module into the DestModule.  This consists basically of copying the
965c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// function over and fixing up references to values.
9664bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattnerstatic bool LinkFunctionBodies(Module *Dest, Module *Src,
9675c2d335d86579941b71816196a95a64b277e8963Chris Lattner                               std::map<const Value*, Value*> &ValueMap,
9685c2d335d86579941b71816196a95a64b277e8963Chris Lattner                               std::string *Err) {
9695c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
9708bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer  // Loop over all of the functions in the src module, mapping them over as we
9718bef0373f1b8fab20b9acd277377ca01d72bac7eReid Spencer  // go
9724bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner  for (Module::iterator SF = Src->begin(), E = Src->end(); SF != E; ++SF) {
973619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer    if (!SF->isDeclaration()) {               // No body if function is external
97418961504fc2b299578dba817900a0696cf3ccc4dChris Lattner      Function *DF = cast<Function>(ValueMap[SF]); // Destination function
975c2d774b6c1d38421c435b6d3cfaa10402c900aebChris Lattner
97618961504fc2b299578dba817900a0696cf3ccc4dChris Lattner      // DF not external SF external?
977ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer      if (DF->isDeclaration())
97835956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner        // Only provide the function body if there isn't one already.
97935956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner        if (LinkFunctionBody(DF, SF, ValueMap, Err))
98035956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner          return true;
981c2d774b6c1d38421c435b6d3cfaa10402c900aebChris Lattner    }
9825c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  }
9835c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return false;
9845c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
9855c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
9868166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner// LinkAppendingVars - If there were any appending global variables, link them
9878166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner// together now.  Return true on error.
9888166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattnerstatic bool LinkAppendingVars(Module *M,
9898166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                  std::multimap<std::string, GlobalVariable *> &AppendingVars,
9908166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                              std::string *ErrorMsg) {
9918166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  if (AppendingVars.empty()) return false; // Nothing to do.
992f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
9938166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // Loop over the multimap of appending vars, processing any variables with the
9948166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // same name, forming a new appending global variable with both of the
9958166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // initializers merged together, then rewrite references to the old variables
9968166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // and delete them.
9978166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  std::vector<Constant*> Inits;
9988166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  while (AppendingVars.size() > 1) {
9998166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    // Get the first two elements in the map...
10008166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    std::multimap<std::string,
10018166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      GlobalVariable*>::iterator Second = AppendingVars.begin(), First=Second++;
10028166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
10038166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    // If the first two elements are for different names, there is no pair...
10048166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    // Otherwise there is a pair, so link them together...
10058166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    if (First->first == Second->first) {
10068166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      GlobalVariable *G1 = First->second, *G2 = Second->second;
10078166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      const ArrayType *T1 = cast<ArrayType>(G1->getType()->getElementType());
10088166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      const ArrayType *T2 = cast<ArrayType>(G2->getType()->getElementType());
1009f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
10108166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Check to see that they two arrays agree on type...
10118166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      if (T1->getElementType() != T2->getElementType())
10128166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        return Error(ErrorMsg,
10138166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner         "Appending variables with different element types need to be linked!");
10148166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      if (G1->isConstant() != G2->isConstant())
10158166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        return Error(ErrorMsg,
10168166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                     "Appending variables linked with different const'ness!");
10178166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
10189613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio      if (G1->getAlignment() != G2->getAlignment())
10199613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio        return Error(ErrorMsg,
10209613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio         "Appending variables with different alignment need to be linked!");
10219613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio
10229613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio      if (G1->getVisibility() != G2->getVisibility())
10239613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio        return Error(ErrorMsg,
10249613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio         "Appending variables with different visibility need to be linked!");
10259613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio
10269613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio      if (G1->getSection() != G2->getSection())
10279613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio        return Error(ErrorMsg,
10289613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio         "Appending variables with different section name need to be linked!");
10299613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio
10308166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      unsigned NewSize = T1->getNumElements() + T2->getNumElements();
10318166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      ArrayType *NewType = ArrayType::get(T1->getElementType(), NewSize);
10328166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
1033ed74a4ef3b813bed0c0f4bb5af42229d50133da2Chris Lattner      G1->setName("");   // Clear G1's name in case of a conflict!
1034ed74a4ef3b813bed0c0f4bb5af42229d50133da2Chris Lattner
10358166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Create the new global variable...
10368166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      GlobalVariable *NG =
10378166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        new GlobalVariable(NewType, G1->isConstant(), G1->getLinkage(),
1038c763552299165b88d34a7d4f2d76ff413cbc7f67Lauro Ramos Venancio                           /*init*/0, First->first, M, G1->isThreadLocal());
10398166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
10409613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio      // Propagate alignment, visibility and section info.
10419613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio      CopyGVAttributes(NG, G1);
10429613e871d26f123789d98c61879b4dbe944a6d31Lauro Ramos Venancio
10438166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Merge the initializer...
10448166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      Inits.reserve(NewSize);
1045de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      if (ConstantArray *I = dyn_cast<ConstantArray>(G1->getInitializer())) {
1046de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        for (unsigned i = 0, e = T1->getNumElements(); i != e; ++i)
1047cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos          Inits.push_back(I->getOperand(i));
1048de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      } else {
1049de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        assert(isa<ConstantAggregateZero>(G1->getInitializer()));
1050de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        Constant *CV = Constant::getNullValue(T1->getElementType());
1051de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        for (unsigned i = 0, e = T1->getNumElements(); i != e; ++i)
1052de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner          Inits.push_back(CV);
1053de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      }
1054de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      if (ConstantArray *I = dyn_cast<ConstantArray>(G2->getInitializer())) {
1055de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        for (unsigned i = 0, e = T2->getNumElements(); i != e; ++i)
1056cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos          Inits.push_back(I->getOperand(i));
1057de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      } else {
1058de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        assert(isa<ConstantAggregateZero>(G2->getInitializer()));
1059de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        Constant *CV = Constant::getNullValue(T2->getElementType());
1060de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        for (unsigned i = 0, e = T2->getNumElements(); i != e; ++i)
1061de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner          Inits.push_back(CV);
1062de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      }
10638166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      NG->setInitializer(ConstantArray::get(NewType, Inits));
10648166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      Inits.clear();
10658166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
10668166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Replace any uses of the two global variables with uses of the new
10678166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // global...
10688166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
10698166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // FIXME: This should rewrite simple/straight-forward uses such as
10708166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // getelementptr instructions to not use the Cast!
10714da49122f3f3c8da68a52723d846b88c72166a68Reid Spencer      G1->replaceAllUsesWith(ConstantExpr::getBitCast(NG, G1->getType()));
10724da49122f3f3c8da68a52723d846b88c72166a68Reid Spencer      G2->replaceAllUsesWith(ConstantExpr::getBitCast(NG, G2->getType()));
10738166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
10748166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Remove the two globals from the module now...
10758166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      M->getGlobalList().erase(G1);
10768166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      M->getGlobalList().erase(G2);
10778166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
10788166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Put the new global into the AppendingVars map so that we can handle
10798166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // linking of more than two vars...
10808166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      Second->second = NG;
10818166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    }
10828166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    AppendingVars.erase(First);
10838166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  }
10848166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
10858166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  return false;
10868166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner}
108752f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner
10889f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikovstatic bool ResolveAliases(Module *Dest) {
10899f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov  for (Module::alias_iterator I = Dest->alias_begin(), E = Dest->alias_end();
10909f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov       I != E; ++I) {
10919f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov    GlobalValue* GV = const_cast<GlobalValue*>(I->getAliasedGlobal());
10929f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov    if (!GV->isDeclaration())
10939f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov      I->replaceAllUsesWith(GV);
10949f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov  }
10959f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov
10969f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov  return false;
10979f2ee703663d8ca7d91cfdaf773fc70273e9e482Anton Korobeynikov}
109852f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner
109952f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// LinkModules - This function links two modules together, with the resulting
110052f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// left module modified to be the composite of the two input modules.  If an
110152f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// error occurs, true is returned and ErrorMsg (if not null) is set to indicate
11025c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// the problem.  Upon failure, the Dest module could be in a modified state, and
11035c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// shouldn't be relied on to be consistent.
1104f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukmanbool
11050ba9e211bed7813949aaca97e6b15579c8188e05Reid SpencerLinker::LinkModules(Module *Dest, Module *Src, std::string *ErrorMsg) {
110657a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  assert(Dest != 0 && "Invalid Destination module");
110757a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  assert(Src  != 0 && "Invalid Source Module");
110857a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer
1109c36357c406911ded5a0e2924998d885815370fdbChris Lattner  if (Dest->getDataLayout().empty()) {
1110c36357c406911ded5a0e2924998d885815370fdbChris Lattner    if (!Src->getDataLayout().empty()) {
1111ec9bfdc152a870d3c62377ed6855fb3bff69511cChris Lattner      Dest->setDataLayout(Src->getDataLayout());
1112c36357c406911ded5a0e2924998d885815370fdbChris Lattner    } else {
1113c36357c406911ded5a0e2924998d885815370fdbChris Lattner      std::string DataLayout;
1114c36357c406911ded5a0e2924998d885815370fdbChris Lattner
1115a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov      if (Dest->getEndianness() == Module::AnyEndianness) {
1116c36357c406911ded5a0e2924998d885815370fdbChris Lattner        if (Src->getEndianness() == Module::BigEndian)
1117c36357c406911ded5a0e2924998d885815370fdbChris Lattner          DataLayout.append("E");
1118c36357c406911ded5a0e2924998d885815370fdbChris Lattner        else if (Src->getEndianness() == Module::LittleEndian)
1119c36357c406911ded5a0e2924998d885815370fdbChris Lattner          DataLayout.append("e");
1120a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov      }
1121a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov
1122a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov      if (Dest->getPointerSize() == Module::AnyPointerSize) {
1123c36357c406911ded5a0e2924998d885815370fdbChris Lattner        if (Src->getPointerSize() == Module::Pointer64)
1124c36357c406911ded5a0e2924998d885815370fdbChris Lattner          DataLayout.append(DataLayout.length() == 0 ? "p:64:64" : "-p:64:64");
1125c36357c406911ded5a0e2924998d885815370fdbChris Lattner        else if (Src->getPointerSize() == Module::Pointer32)
1126c36357c406911ded5a0e2924998d885815370fdbChris Lattner          DataLayout.append(DataLayout.length() == 0 ? "p:32:32" : "-p:32:32");
1127a27694d7aac1bfcec2b776df460719b4a165a79aAnton Korobeynikov      }
1128c36357c406911ded5a0e2924998d885815370fdbChris Lattner      Dest->setDataLayout(DataLayout);
1129c36357c406911ded5a0e2924998d885815370fdbChris Lattner    }
1130c36357c406911ded5a0e2924998d885815370fdbChris Lattner  }
1131873c5e7859c1534bf92c9e2747f2b70685059598Chris Lattner
1132f27dfcb210fec571d79bc881f5463e0b405707c0Chris Lattner  // Copy the target triple from the source to dest if the dest's is empty.
1133c36357c406911ded5a0e2924998d885815370fdbChris Lattner  if (Dest->getTargetTriple().empty() && !Src->getTargetTriple().empty())
1134c36357c406911ded5a0e2924998d885815370fdbChris Lattner    Dest->setTargetTriple(Src->getTargetTriple());
1135c36357c406911ded5a0e2924998d885815370fdbChris Lattner
1136c36357c406911ded5a0e2924998d885815370fdbChris Lattner  if (!Src->getDataLayout().empty() && !Dest->getDataLayout().empty() &&
1137c36357c406911ded5a0e2924998d885815370fdbChris Lattner      Src->getDataLayout() != Dest->getDataLayout())
113826f238589f9bb372d24b6fb2bc32edbf046fd9eeReid Spencer    cerr << "WARNING: Linking two modules of different data layouts!\n";
1139152f19ac6514ec29e183ddcdf2ea71599638dc4cChris Lattner  if (!Src->getTargetTriple().empty() &&
1140152f19ac6514ec29e183ddcdf2ea71599638dc4cChris Lattner      Dest->getTargetTriple() != Src->getTargetTriple())
1141e81561909d128c6e2d8033cb5465a49b2596b26aBill Wendling    cerr << "WARNING: Linking two modules of different target triples!\n";
1142f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
1143f27dfcb210fec571d79bc881f5463e0b405707c0Chris Lattner  // Append the module inline asm string.
11446631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner  if (!Src->getModuleInlineAsm().empty()) {
11456631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner    if (Dest->getModuleInlineAsm().empty())
11466631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner      Dest->setModuleInlineAsm(Src->getModuleInlineAsm());
1147e1b2e14f92e8e9da1eb4854511a881cf9c22af3cChris Lattner    else
11486631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner      Dest->setModuleInlineAsm(Dest->getModuleInlineAsm()+"\n"+
11496631601eeda2300abdce11fbd7a7a3daa2694d39Chris Lattner                               Src->getModuleInlineAsm());
1150e1b2e14f92e8e9da1eb4854511a881cf9c22af3cChris Lattner  }
1151e1b2e14f92e8e9da1eb4854511a881cf9c22af3cChris Lattner
1152719012d6254ee425464f034250176283fb96793bReid Spencer  // Update the destination module's dependent libraries list with the libraries
115357a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  // from the source module. There's no opportunity for duplicates here as the
115457a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  // Module ensures that duplicate insertions are discarded.
1155f27dfcb210fec571d79bc881f5463e0b405707c0Chris Lattner  for (Module::lib_iterator SI = Src->lib_begin(), SE = Src->lib_end();
1156f27dfcb210fec571d79bc881f5463e0b405707c0Chris Lattner       SI != SE; ++SI)
115757a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer    Dest->addLibrary(*SI);
115857a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer
11592c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  // LinkTypes - Go through the symbol table of the Src module and see if any
11602c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  // types are named in the src module that are not named in the Dst module.
11612c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  // Make sure there are no type name conflicts.
1162619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer  if (LinkTypes(Dest, Src, ErrorMsg))
1163619f024cc315b9edcc6bc71ebea68fd44d85aeafReid Spencer    return true;
11642c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
11655c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  // ValueMap - Mapping of values from what they used to be in Src, to what they
11665c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  // are now in Dest.
11675c2d335d86579941b71816196a95a64b277e8963Chris Lattner  std::map<const Value*, Value*> ValueMap;
11685c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
11698166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // AppendingVars - Keep track of global variables in the destination module
11708166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // with appending linkage.  After the module is linked together, they are
11718166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // appended and the module is rewritten.
11728166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  std::multimap<std::string, GlobalVariable *> AppendingVars;
11731127315562b221010040925980040fd4f65bdb1cChris Lattner  for (Module::global_iterator I = Dest->global_begin(), E = Dest->global_end();
11741127315562b221010040925980040fd4f65bdb1cChris Lattner       I != E; ++I) {
11755a837defff581accc09048bea1cd839009e374a7Chris Lattner    // Add all of the appending globals already in the Dest module to
11765a837defff581accc09048bea1cd839009e374a7Chris Lattner    // AppendingVars.
1177f41464618f41b92cf0a0bd14ccc57bce727fb33dChris Lattner    if (I->hasAppendingLinkage())
1178f41464618f41b92cf0a0bd14ccc57bce727fb33dChris Lattner      AppendingVars.insert(std::make_pair(I->getName(), I));
11795a837defff581accc09048bea1cd839009e374a7Chris Lattner  }
11805a837defff581accc09048bea1cd839009e374a7Chris Lattner
11818166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // Insert all of the globals in src into the Dest module... without linking
11828166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // initializers (which could refer to functions not yet mapped over).
1183ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  if (LinkGlobals(Dest, Src, ValueMap, AppendingVars, ErrorMsg))
11845a837defff581accc09048bea1cd839009e374a7Chris Lattner    return true;
11855c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
1186c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // Link the functions together between the two modules, without doing function
1187c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // bodies... this just adds external function prototypes to the Dest
1188c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // function...  We do this so that when we begin processing function bodies,
1189c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // all of the global values that may be referenced are available in our
1190c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // ValueMap.
1191ef9b9a793949469cdaa4ab6d0173136229dcab7bReid Spencer  if (LinkFunctionProtos(Dest, Src, ValueMap, ErrorMsg))
11925a837defff581accc09048bea1cd839009e374a7Chris Lattner    return true;
11935c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
11944fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov  // If there were any alias, link them now. We really need to do this now,
11954fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov  // because all of the aliases that may be referenced need to be available in
11964fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov  // ValueMap
11974fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov  if (LinkAlias(Dest, Src, ValueMap, ErrorMsg)) return true;
11984fb2873bb46330da18f413cec661ce2fd0f9f37cAnton Korobeynikov
11996cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner  // Update the initializers in the Dest module now that all globals that may
12006cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner  // be referenced are in Dest.
12016cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner  if (LinkGlobalInits(Dest, Src, ValueMap, ErrorMsg)) return true;
12026cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner
1203c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // Link in the function bodies that are defined in the source module into the
1204c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // DestModule.  This consists basically of copying the function over and
1205c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // fixing up references to values.
120679df7c0aaa18129e55968c8783ef8346807bd4afChris Lattner  if (LinkFunctionBodies(Dest, Src, ValueMap, ErrorMsg)) return true;
120752f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner
12088166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // If there were any appending global variables, link them together now.
12098166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  if (LinkAppendingVars(Dest, AppendingVars, ErrorMsg)) return true;
12108166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
12113db9191baf5f5229a2f4d204c6332ba27fff81d0Anton Korobeynikov  // Resolve all uses of aliases with aliasees
12123db9191baf5f5229a2f4d204c6332ba27fff81d0Anton Korobeynikov  if (ResolveAliases(Dest)) return true;
12133db9191baf5f5229a2f4d204c6332ba27fff81d0Anton Korobeynikov
121457a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  // If the source library's module id is in the dependent library list of the
121557a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  // destination library, remove it since that module is now linked in.
121657a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  sys::Path modId;
1217dd04df0ec33a903ee7fc747701bafde622f77d8bReid Spencer  modId.set(Src->getModuleIdentifier());
121807adb2836b8aa7a3872e33c285958f5937662b50Reid Spencer  if (!modId.isEmpty())
121907adb2836b8aa7a3872e33c285958f5937662b50Reid Spencer    Dest->removeLibrary(modId.getBasename());
122057a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer
122152f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner  return false;
122252f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner}
12239466f5113ba859677a28887c3c7143065e702019Vikram S. Adve
1224567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer// vim: sw=2
1225