LinkModules.cpp revision ed74a4ef3b813bed0c0f4bb5af42229d50133da2
17f49602e5ac7186c9505a0496393dd624515b459Reid Spencer//===- lib/Linker/LinkModules.cpp - Module Linker Implementation ----------===//
2f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman//
3b576c94c15af9a440f69d9d03c2afead7971118cJohn Criswell//                     The LLVM Compiler Infrastructure
4b576c94c15af9a440f69d9d03c2afead7971118cJohn Criswell//
5b576c94c15af9a440f69d9d03c2afead7971118cJohn Criswell// This file was developed by the LLVM research group and is distributed under
6b576c94c15af9a440f69d9d03c2afead7971118cJohn Criswell// the University of Illinois Open Source 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"
235c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner#include "llvm/SymbolTable.h"
2447b14a4a6a455c7be169cfd312fcbe796f0ad426Misha Brukman#include "llvm/Instructions.h"
25adbc0b5287bf36893cdcae2440d48b3cb3489e38Chris Lattner#include "llvm/Assembly/Writer.h"
2657a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer#include "llvm/System/Path.h"
27c28a224c68adf0464c275778af4bce762ad25d37Reid Spencer#include <iostream>
2872cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattner#include <sstream>
29f7703df4968084c18c248c1feea9961c19a32e6aChris Lattnerusing namespace llvm;
30d0fde30ce850b78371fd1386338350591f9ff494Brian Gaeke
315c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// Error - Simple wrapper function to conditionally assign to E and return true.
325c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// This just makes error return conditions a little bit simpler...
338166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattnerstatic inline bool Error(std::string *E, const std::string &Message) {
345c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  if (E) *E = Message;
355c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return true;
365c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
375c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
38719012d6254ee425464f034250176283fb96793bReid Spencer// ToStr - Simple wrapper function to convert a type to a string.
3972cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattnerstatic std::string ToStr(const Type *Ty, const Module *M) {
4072cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattner  std::ostringstream OS;
4172cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattner  WriteTypeSymbolic(OS, Ty, M);
4272cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattner  return OS.str();
4372cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattner}
4472cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattner
45700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//
46700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Function: ResolveTypes()
47700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//
48700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Description:
49700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  Attempt to link the two specified types together.
50700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//
51700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Inputs:
52700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  DestTy - The type to which we wish to resolve.
53700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  SrcTy  - The original type which we want to resolve.
54700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  Name   - The name of the type.
55700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//
56700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Outputs:
57700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  DestST - The symbol table in which the new type should be placed.
58700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//
59700867bb69ae1469f2b20005a79fd133e83fd552John Criswell// Return value:
60700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  true  - There is an error and the types cannot yet be linked.
61700867bb69ae1469f2b20005a79fd133e83fd552John Criswell//  false - No errors.
624c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner//
63e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattnerstatic bool ResolveTypes(const Type *DestTy, const Type *SrcTy,
64e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner                         SymbolTable *DestST, const std::string &Name) {
65e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  if (DestTy == SrcTy) return false;       // If already equal, noop
66e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
674c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  // Does the type already exist in the module?
684c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  if (DestTy && !isa<OpaqueType>(DestTy)) {  // Yup, the type already exists...
69e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner    if (const OpaqueType *OT = dyn_cast<OpaqueType>(SrcTy)) {
70e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      const_cast<OpaqueType*>(OT)->refineAbstractTypeTo(DestTy);
714c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    } else {
724c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      return true;  // Cannot link types... neither is opaque and not-equal
734c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    }
744c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  } else {                       // Type not in dest module.  Add it now.
754c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    if (DestTy)                  // Type _is_ in module, just opaque...
76e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      const_cast<OpaqueType*>(cast<OpaqueType>(DestTy))
77e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner                           ->refineAbstractTypeTo(SrcTy);
78fcd0234de65fc07fb06f3245b60e82bdba78c745Chris Lattner    else if (!Name.empty())
79e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      DestST->insert(Name, const_cast<Type*>(SrcTy));
804c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  }
814c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  return false;
824c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner}
834c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
8443f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattnerstatic const FunctionType *getFT(const PATypeHolder &TH) {
8543f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner  return cast<FunctionType>(TH.get());
8643f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner}
879732be769e57f1dec03f046ab957053048f60e40Chris Lattnerstatic const StructType *getST(const PATypeHolder &TH) {
8843f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner  return cast<StructType>(TH.get());
8943f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner}
90e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
91e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner// RecursiveResolveTypes - This is just like ResolveTypes, except that it
92e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner// recurses down into derived types, merging the used types if the parent types
93e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner// are compatible.
94e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattnerstatic bool RecursiveResolveTypesI(const PATypeHolder &DestTy,
95e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                                   const PATypeHolder &SrcTy,
96e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                                   SymbolTable *DestST, const std::string &Name,
97e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                std::vector<std::pair<PATypeHolder, PATypeHolder> > &Pointers) {
9843f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner  const Type *SrcTyT = SrcTy.get();
9943f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner  const Type *DestTyT = DestTy.get();
10043f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner  if (DestTyT == SrcTyT) return false;       // If already equal, noop
101f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
102e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  // If we found our opaque type, resolve it now!
10343f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner  if (isa<OpaqueType>(DestTyT) || isa<OpaqueType>(SrcTyT))
10443f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner    return ResolveTypes(DestTyT, SrcTyT, DestST, Name);
105f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
106e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  // Two types cannot be resolved together if they are of different primitive
107e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  // type.  For example, we cannot resolve an int to a float.
108f70c22b019494723d0e706f93d6542dfaa6e73a5Chris Lattner  if (DestTyT->getTypeID() != SrcTyT->getTypeID()) return true;
109e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
110e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  // Otherwise, resolve the used type used by this derived type...
111f70c22b019494723d0e706f93d6542dfaa6e73a5Chris Lattner  switch (DestTyT->getTypeID()) {
112e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  case Type::FunctionTyID: {
11343f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner    if (cast<FunctionType>(DestTyT)->isVarArg() !=
114841e00b96295a2b66cb7573e961656d28a6cb12bChris Lattner        cast<FunctionType>(SrcTyT)->isVarArg() ||
115841e00b96295a2b66cb7573e961656d28a6cb12bChris Lattner        cast<FunctionType>(DestTyT)->getNumContainedTypes() !=
116841e00b96295a2b66cb7573e961656d28a6cb12bChris Lattner        cast<FunctionType>(SrcTyT)->getNumContainedTypes())
11743f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner      return true;
11843f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner    for (unsigned i = 0, e = getFT(DestTy)->getNumContainedTypes(); i != e; ++i)
119e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner      if (RecursiveResolveTypesI(getFT(DestTy)->getContainedType(i),
120e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                                 getFT(SrcTy)->getContainedType(i), DestST, "",
121e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                                 Pointers))
122e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner        return true;
123e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner    return false;
124e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  }
125e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  case Type::StructTyID: {
126f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman    if (getST(DestTy)->getNumContainedTypes() !=
12743f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner        getST(SrcTy)->getNumContainedTypes()) return 1;
12843f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner    for (unsigned i = 0, e = getST(DestTy)->getNumContainedTypes(); i != e; ++i)
129e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner      if (RecursiveResolveTypesI(getST(DestTy)->getContainedType(i),
130e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                                 getST(SrcTy)->getContainedType(i), DestST, "",
131e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                                 Pointers))
132e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner        return true;
133e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner    return false;
134e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  }
135e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  case Type::ArrayTyID: {
13643f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner    const ArrayType *DAT = cast<ArrayType>(DestTy.get());
13743f4ba8c7d19664bb5de73a7b391b67061f25d22Chris Lattner    const ArrayType *SAT = cast<ArrayType>(SrcTy.get());
138e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner    if (DAT->getNumElements() != SAT->getNumElements()) return true;
139e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    return RecursiveResolveTypesI(DAT->getElementType(), SAT->getElementType(),
140e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                                  DestST, "", Pointers);
141e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner  }
142e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner  case Type::PointerTyID: {
143e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // If this is a pointer type, check to see if we have already seen it.  If
144e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // so, we are in a recursive branch.  Cut off the search now.  We cannot use
145e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // an associative container for this search, because the type pointers (keys
146e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // in the container) change whenever types get resolved...
147e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    for (unsigned i = 0, e = Pointers.size(); i != e; ++i)
148e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner      if (Pointers[i].first == DestTy)
149e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner        return Pointers[i].second != SrcTy;
150e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner
151e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // Otherwise, add the current pointers to the vector to stop recursion on
152e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    // this pair.
153e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    Pointers.push_back(std::make_pair(DestTyT, SrcTyT));
154e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    bool Result =
155e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner      RecursiveResolveTypesI(cast<PointerType>(DestTy.get())->getElementType(),
156e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                             cast<PointerType>(SrcTy.get())->getElementType(),
157e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                             DestST, "", Pointers);
158e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    Pointers.pop_back();
159e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner    return Result;
160e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  }
161e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner  default: assert(0 && "Unexpected type!"); return true;
162f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman  }
163e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner}
164e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
165e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattnerstatic bool RecursiveResolveTypes(const PATypeHolder &DestTy,
166e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                                  const PATypeHolder &SrcTy,
167e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner                                  SymbolTable *DestST, const std::string &Name){
168e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner  std::vector<std::pair<PATypeHolder, PATypeHolder> > PointerTypes;
169e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner  return RecursiveResolveTypesI(DestTy, SrcTy, DestST, Name, PointerTypes);
170e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner}
171e3092c94ad2e3af96f37a0a8186149acbbd9700aChris Lattner
172e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
1732c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner// LinkTypes - Go through the symbol table of the Src module and see if any
1742c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner// types are named in the src module that are not named in the Dst module.
1752c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner// Make sure there are no type name conflicts.
1765c2d335d86579941b71816196a95a64b277e8963Chris Lattnerstatic bool LinkTypes(Module *Dest, const Module *Src, std::string *Err) {
1776e6026b46569b01f8f6d4dcdb6c899c3a9c76b3eChris Lattner  SymbolTable       *DestST = &Dest->getSymbolTable();
1786e6026b46569b01f8f6d4dcdb6c899c3a9c76b3eChris Lattner  const SymbolTable *SrcST  = &Src->getSymbolTable();
1792c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
1802c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  // Look for a type plane for Type's...
181567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer  SymbolTable::type_const_iterator TI = SrcST->type_begin();
182567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer  SymbolTable::type_const_iterator TE = SrcST->type_end();
183567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer  if (TI == TE) return false;  // No named types, do nothing.
1842c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
185cf00c4ab3ba308d45d98c5ccab87362cf802facbMisha Brukman  // Some types cannot be resolved immediately because they depend on other
186cf00c4ab3ba308d45d98c5ccab87362cf802facbMisha Brukman  // types being resolved to each other first.  This contains a list of types we
187cf00c4ab3ba308d45d98c5ccab87362cf802facbMisha Brukman  // are waiting to recheck.
1884c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  std::vector<std::string> DelayedTypesToResolve;
1894c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
190567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer  for ( ; TI != TE; ++TI ) {
191567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer    const std::string &Name = TI->first;
192c28a224c68adf0464c275778af4bce762ad25d37Reid Spencer    const Type *RHS = TI->second;
1932c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
1942c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner    // Check to see if this type name is already in the dest module...
195567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer    Type *Entry = DestST->lookupType(Name);
1964c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
1974c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    if (ResolveTypes(Entry, RHS, DestST, Name)) {
1984c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      // They look different, save the types 'till later to resolve.
1994c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      DelayedTypesToResolve.push_back(Name);
2004c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    }
2014c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  }
2024c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
2034c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  // Iteratively resolve types while we can...
2044c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner  while (!DelayedTypesToResolve.empty()) {
2054c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    // Loop over all of the types, attempting to resolve them if possible...
2064c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    unsigned OldSize = DelayedTypesToResolve.size();
2074c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
208e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner    // Try direct resolution by name...
2094c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    for (unsigned i = 0; i != DelayedTypesToResolve.size(); ++i) {
2104c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      const std::string &Name = DelayedTypesToResolve[i];
211567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer      Type *T1 = SrcST->lookupType(Name);
212567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer      Type *T2 = DestST->lookupType(Name);
2134c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      if (!ResolveTypes(T2, T1, DestST, Name)) {
2144c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner        // We are making progress!
2154c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner        DelayedTypesToResolve.erase(DelayedTypesToResolve.begin()+i);
2164c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner        --i;
2172f6bb2bce13084dfc4c8acf47f9eae0578267aaaChris Lattner      }
2184c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    }
2192f6bb2bce13084dfc4c8acf47f9eae0578267aaaChris Lattner
2204c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    // Did we not eliminate any types?
2214c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner    if (DelayedTypesToResolve.size() == OldSize) {
222e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      // Attempt to resolve subelements of types.  This allows us to merge these
223e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      // two types: { int* } and { opaque* }
2244c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      for (unsigned i = 0, e = DelayedTypesToResolve.size(); i != e; ++i) {
2254c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner        const std::string &Name = DelayedTypesToResolve[i];
226567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer        PATypeHolder T1(SrcST->lookupType(Name));
227567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer        PATypeHolder T2(DestST->lookupType(Name));
228e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
229e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner        if (!RecursiveResolveTypes(T2, T1, DestST, Name)) {
230e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          // We are making progress!
231e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          DelayedTypesToResolve.erase(DelayedTypesToResolve.begin()+i);
232f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
233e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          // Go back to the main loop, perhaps we can resolve directly by name
234e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          // now...
235e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner          break;
236e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner        }
237e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      }
238e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner
239e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      // If we STILL cannot resolve the types, then there is something wrong.
240e76c57ad467ed57f09f2a3ef51628d4b6eb2b304Chris Lattner      if (DelayedTypesToResolve.size() == OldSize) {
241aeb18ce9c012f8554f48bae9bb32351c6eeaa26cChris Lattner        // Remove the symbol name from the destination.
242aeb18ce9c012f8554f48bae9bb32351c6eeaa26cChris Lattner        DelayedTypesToResolve.pop_back();
2434c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner      }
2442c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner    }
2452c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  }
2464c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
2474c00e53b81de81ecf4ba0c4e287ea230c79e82aeChris Lattner
2482c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  return false;
2492c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner}
2502c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
2515c2d335d86579941b71816196a95a64b277e8963Chris Lattnerstatic void PrintMap(const std::map<const Value*, Value*> &M) {
2525c2d335d86579941b71816196a95a64b277e8963Chris Lattner  for (std::map<const Value*, Value*>::const_iterator I = M.begin(), E =M.end();
2532d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner       I != E; ++I) {
2545c2d335d86579941b71816196a95a64b277e8963Chris Lattner    std::cerr << " Fr: " << (void*)I->first << " ";
25587182ae6baf28ca6da6ee4bc00cf1857f293acfeChris Lattner    I->first->dump();
2565c2d335d86579941b71816196a95a64b277e8963Chris Lattner    std::cerr << " To: " << (void*)I->second << " ";
25787182ae6baf28ca6da6ee4bc00cf1857f293acfeChris Lattner    I->second->dump();
2585c2d335d86579941b71816196a95a64b277e8963Chris Lattner    std::cerr << "\n";
2592d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner  }
2602d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner}
2612d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner
2622d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner
2630033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner// RemapOperand - Use ValueMap to convert references from one module to another.
2640033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner// This is somewhat sophisticated in that it can automatically handle constant
2650033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner// references correctly as well...
2665c2d335d86579941b71816196a95a64b277e8963Chris Lattnerstatic Value *RemapOperand(const Value *In,
2670033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner                           std::map<const Value*, Value*> &ValueMap) {
2680033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner  std::map<const Value*,Value*>::const_iterator I = ValueMap.find(In);
2690033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner  if (I != ValueMap.end()) return I->second;
2705c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
2710033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner  // Check to see if it's a constant that we are interesting in transforming.
27218961504fc2b299578dba817900a0696cf3ccc4dChris Lattner  if (const Constant *CPV = dyn_cast<Constant>(In)) {
273de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner    if ((!isa<DerivedType>(CPV->getType()) && !isa<ConstantExpr>(CPV)) ||
274de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        isa<ConstantAggregateZero>(CPV))
2750033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner      return const_cast<Constant*>(CPV);   // Simple constants stay identical.
2768d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner
277e9bb2df410f7a22decad9a883f7139d5857c1520Chris Lattner    Constant *Result = 0;
2788d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner
27918961504fc2b299578dba817900a0696cf3ccc4dChris Lattner    if (const ConstantArray *CPA = dyn_cast<ConstantArray>(CPV)) {
280cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos      std::vector<Constant*> Operands(CPA->getNumOperands());
281cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos      for (unsigned i = 0, e = CPA->getNumOperands(); i != e; ++i)
2820033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner        Operands[i] =cast<Constant>(RemapOperand(CPA->getOperand(i), ValueMap));
283e9bb2df410f7a22decad9a883f7139d5857c1520Chris Lattner      Result = ConstantArray::get(cast<ArrayType>(CPA->getType()), Operands);
28418961504fc2b299578dba817900a0696cf3ccc4dChris Lattner    } else if (const ConstantStruct *CPS = dyn_cast<ConstantStruct>(CPV)) {
285cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos      std::vector<Constant*> Operands(CPS->getNumOperands());
286cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos      for (unsigned i = 0, e = CPS->getNumOperands(); i != e; ++i)
2870033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner        Operands[i] =cast<Constant>(RemapOperand(CPS->getOperand(i), ValueMap));
288e9bb2df410f7a22decad9a883f7139d5857c1520Chris Lattner      Result = ConstantStruct::get(cast<StructType>(CPS->getType()), Operands);
289b976e668165e1875a8f1eb7af800e33bb1e4393dChris Lattner    } else if (isa<ConstantPointerNull>(CPV) || isa<UndefValue>(CPV)) {
29018961504fc2b299578dba817900a0696cf3ccc4dChris Lattner      Result = const_cast<Constant*>(CPV);
29100dc4796132933e76fe2b834ca087e42621c175aReid Spencer    } else if (isa<GlobalValue>(CPV)) {
2920033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner      Result = cast<Constant>(RemapOperand(CPV, ValueMap));
2936cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner    } else if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(CPV)) {
294b319faff77a5bb12e753c45662f6c532b77f0b91Chris Lattner      if (CE->getOpcode() == Instruction::GetElementPtr) {
2950033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner        Value *Ptr = RemapOperand(CE->getOperand(0), ValueMap);
296b319faff77a5bb12e753c45662f6c532b77f0b91Chris Lattner        std::vector<Constant*> Indices;
297b319faff77a5bb12e753c45662f6c532b77f0b91Chris Lattner        Indices.reserve(CE->getNumOperands()-1);
298b319faff77a5bb12e753c45662f6c532b77f0b91Chris Lattner        for (unsigned i = 1, e = CE->getNumOperands(); i != e; ++i)
299b319faff77a5bb12e753c45662f6c532b77f0b91Chris Lattner          Indices.push_back(cast<Constant>(RemapOperand(CE->getOperand(i),
3000033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner                                                        ValueMap)));
301b319faff77a5bb12e753c45662f6c532b77f0b91Chris Lattner
302b319faff77a5bb12e753c45662f6c532b77f0b91Chris Lattner        Result = ConstantExpr::getGetElementPtr(cast<Constant>(Ptr), Indices);
303b319faff77a5bb12e753c45662f6c532b77f0b91Chris Lattner      } else if (CE->getNumOperands() == 1) {
304ad333484ea5ae976b83e35ccd3f6cfa6e71290e2Chris Lattner        // Cast instruction
305ad333484ea5ae976b83e35ccd3f6cfa6e71290e2Chris Lattner        assert(CE->getOpcode() == Instruction::Cast);
3060033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner        Value *V = RemapOperand(CE->getOperand(0), ValueMap);
307ad333484ea5ae976b83e35ccd3f6cfa6e71290e2Chris Lattner        Result = ConstantExpr::getCast(cast<Constant>(V), CE->getType());
3081438102576b94d6dac9bb18e8df291c4d632572dChris Lattner      } else if (CE->getNumOperands() == 3) {
3091438102576b94d6dac9bb18e8df291c4d632572dChris Lattner        // Select instruction
3101438102576b94d6dac9bb18e8df291c4d632572dChris Lattner        assert(CE->getOpcode() == Instruction::Select);
3110033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner        Value *V1 = RemapOperand(CE->getOperand(0), ValueMap);
3120033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner        Value *V2 = RemapOperand(CE->getOperand(1), ValueMap);
3130033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner        Value *V3 = RemapOperand(CE->getOperand(2), ValueMap);
3141438102576b94d6dac9bb18e8df291c4d632572dChris Lattner        Result = ConstantExpr::getSelect(cast<Constant>(V1), cast<Constant>(V2),
3151438102576b94d6dac9bb18e8df291c4d632572dChris Lattner                                         cast<Constant>(V3));
3166cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner      } else if (CE->getNumOperands() == 2) {
3176cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner        // Binary operator...
3180033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner        Value *V1 = RemapOperand(CE->getOperand(0), ValueMap);
3190033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner        Value *V2 = RemapOperand(CE->getOperand(1), ValueMap);
3206cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner
3216cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner        Result = ConstantExpr::get(CE->getOpcode(), cast<Constant>(V1),
322d981f8af79be9ea647d593531091740a3a27a85fChris Lattner                                   cast<Constant>(V2));
3236cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner      } else {
324b319faff77a5bb12e753c45662f6c532b77f0b91Chris Lattner        assert(0 && "Unknown constant expr type!");
3256cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner      }
3266cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner
3278d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner    } else {
3288d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      assert(0 && "Unknown type of derived type constant value!");
3298d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner    }
3308d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner
3318d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner    // Cache the mapping in our local map structure...
3320033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner    ValueMap.insert(std::make_pair(In, Result));
3338d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner    return Result;
3348d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  }
3352d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner
3360033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner  std::cerr << "LinkModules ValueMap: \n";
3370033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner  PrintMap(ValueMap);
3382d3e8bba628843bf2dffa21f69d9b45098d3bfc4Chris Lattner
3395c2d335d86579941b71816196a95a64b277e8963Chris Lattner  std::cerr << "Couldn't remap value: " << (void*)In << " " << *In << "\n";
3408d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  assert(0 && "Couldn't remap value!");
3418d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  return 0;
3425c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
3435c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
344c003628a612d3687fb77088a5894314210a65385Chris Lattner/// ForceRenaming - The LLVM SymbolTable class autorenames globals that conflict
345c003628a612d3687fb77088a5894314210a65385Chris Lattner/// in the symbol table.  This is good for all clients except for us.  Go
346c003628a612d3687fb77088a5894314210a65385Chris Lattner/// through the trouble to force this back.
347c003628a612d3687fb77088a5894314210a65385Chris Lattnerstatic void ForceRenaming(GlobalValue *GV, const std::string &Name) {
348c003628a612d3687fb77088a5894314210a65385Chris Lattner  assert(GV->getName() != Name && "Can't force rename to self");
349c003628a612d3687fb77088a5894314210a65385Chris Lattner  SymbolTable &ST = GV->getParent()->getSymbolTable();
350c003628a612d3687fb77088a5894314210a65385Chris Lattner
351c003628a612d3687fb77088a5894314210a65385Chris Lattner  // If there is a conflict, rename the conflict.
352c003628a612d3687fb77088a5894314210a65385Chris Lattner  Value *ConflictVal = ST.lookup(GV->getType(), Name);
353c003628a612d3687fb77088a5894314210a65385Chris Lattner  assert(ConflictVal&&"Why do we have to force rename if there is no conflic?");
354c003628a612d3687fb77088a5894314210a65385Chris Lattner  GlobalValue *ConflictGV = cast<GlobalValue>(ConflictVal);
355c003628a612d3687fb77088a5894314210a65385Chris Lattner  assert(ConflictGV->hasInternalLinkage() &&
356c003628a612d3687fb77088a5894314210a65385Chris Lattner         "Not conflicting with a static global, should link instead!");
357c003628a612d3687fb77088a5894314210a65385Chris Lattner
358c003628a612d3687fb77088a5894314210a65385Chris Lattner  ConflictGV->setName("");          // Eliminate the conflict
359c003628a612d3687fb77088a5894314210a65385Chris Lattner  GV->setName(Name);                // Force the name back
360c003628a612d3687fb77088a5894314210a65385Chris Lattner  ConflictGV->setName(Name);        // This will cause ConflictGV to get renamed
3617b0c84dcf59b6ac3a27843fd8162ab36928fba35Chris Lattner  assert(GV->getName() == Name && ConflictGV->getName() != Name &&
362c003628a612d3687fb77088a5894314210a65385Chris Lattner         "ForceRenaming didn't work");
363c003628a612d3687fb77088a5894314210a65385Chris Lattner}
364c003628a612d3687fb77088a5894314210a65385Chris Lattner
365aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// GetLinkageResult - This analyzes the two global values and determines what
366aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// the result will look like in the destination module.  In particular, it
367aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// computes the resultant linkage type, computes whether the global in the
368aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// source should be copied over to the destination (replacing the existing
369aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner/// one), and computes whether this linkage is an error or not.
370aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattnerstatic bool GetLinkageResult(GlobalValue *Dest, GlobalValue *Src,
371aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner                             GlobalValue::LinkageTypes &LT, bool &LinkFromSrc,
372aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner                             std::string *Err) {
373aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  assert((!Dest || !Src->hasInternalLinkage()) &&
374aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner         "If Src has internal linkage, Dest shouldn't be set!");
375aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  if (!Dest) {
376aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    // Linking something to nothing.
377aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LinkFromSrc = true;
378aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LT = Src->getLinkage();
379aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  } else if (Src->isExternal()) {
380aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    // If Src is external or if both Src & Drc are external..  Just link the
381aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    // external globals, we aren't adding anything.
382aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LinkFromSrc = false;
383aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LT = Dest->getLinkage();
384aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  } else if (Dest->isExternal()) {
385aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    // If Dest is external but Src is not:
386aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LinkFromSrc = true;
387aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LT = Src->getLinkage();
388aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  } else if (Src->hasAppendingLinkage() || Dest->hasAppendingLinkage()) {
389aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    if (Src->getLinkage() != Dest->getLinkage())
390aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      return Error(Err, "Linking globals named '" + Src->getName() +
391aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner            "': can only link appending global with another appending global!");
392aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LinkFromSrc = true; // Special cased.
393aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LT = Src->getLinkage();
394aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  } else if (Src->hasWeakLinkage() || Src->hasLinkOnceLinkage()) {
395aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    // At this point we know that Dest has LinkOnce, External or Weak linkage.
396aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    if (Dest->hasLinkOnceLinkage() && Src->hasWeakLinkage()) {
397aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      LinkFromSrc = true;
398aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      LT = Src->getLinkage();
399aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    } else {
400aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      LinkFromSrc = false;
401aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      LT = Dest->getLinkage();
402aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    }
403aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  } else if (Dest->hasWeakLinkage() || Dest->hasLinkOnceLinkage()) {
404aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    // At this point we know that Src has External linkage.
405aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LinkFromSrc = true;
406aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    LT = GlobalValue::ExternalLinkage;
407aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  } else {
408aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    assert(Dest->hasExternalLinkage() && Src->hasExternalLinkage() &&
409aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner           "Unexpected linkage type!");
410f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman    return Error(Err, "Linking globals named '" + Src->getName() +
411aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner                 "': symbol multiply defined!");
412aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  }
413aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner  return false;
414aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner}
4155c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
4165c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// LinkGlobals - Loop through the global variables in the src module and merge
4178166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner// them into the dest module.
418aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattnerstatic bool LinkGlobals(Module *Dest, Module *Src,
4195c2d335d86579941b71816196a95a64b277e8963Chris Lattner                        std::map<const Value*, Value*> &ValueMap,
4208166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                    std::multimap<std::string, GlobalVariable *> &AppendingVars,
4215a837defff581accc09048bea1cd839009e374a7Chris Lattner                        std::map<std::string, GlobalValue*> &GlobalsByName,
4225c2d335d86579941b71816196a95a64b277e8963Chris Lattner                        std::string *Err) {
4235c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  // We will need a module level symbol table if the src module has a module
4245c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  // level symbol table...
425b91b657e026e86082b665cc875c2881ed075b68dChris Lattner  SymbolTable *ST = (SymbolTable*)&Dest->getSymbolTable();
426f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
4275c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  // Loop over all of the globals in the src module, mapping them over as we go
428e4d5c441e04bdc00ccf1804744af670655123b07Chris Lattner  for (Module::global_iterator I = Src->global_begin(), E = Src->global_end(); I != E; ++I) {
429aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    GlobalVariable *SGV = I;
4304ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner    GlobalVariable *DGV = 0;
4315a837defff581accc09048bea1cd839009e374a7Chris Lattner    // Check to see if may have to link the global.
432aad2deb17c3e624cae4ebc1ed51fc5264fa1c0b8Chris Lattner    if (SGV->hasName() && !SGV->hasInternalLinkage())
433aad2deb17c3e624cae4ebc1ed51fc5264fa1c0b8Chris Lattner      if (!(DGV = Dest->getGlobalVariable(SGV->getName(),
434aad2deb17c3e624cae4ebc1ed51fc5264fa1c0b8Chris Lattner                                          SGV->getType()->getElementType()))) {
435aad2deb17c3e624cae4ebc1ed51fc5264fa1c0b8Chris Lattner        std::map<std::string, GlobalValue*>::iterator EGV =
436aad2deb17c3e624cae4ebc1ed51fc5264fa1c0b8Chris Lattner          GlobalsByName.find(SGV->getName());
437aad2deb17c3e624cae4ebc1ed51fc5264fa1c0b8Chris Lattner        if (EGV != GlobalsByName.end())
438aad2deb17c3e624cae4ebc1ed51fc5264fa1c0b8Chris Lattner          DGV = dyn_cast<GlobalVariable>(EGV->second);
439aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner        if (DGV)
440aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner          // If types don't agree due to opaque types, try to resolve them.
441aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner          RecursiveResolveTypes(SGV->getType(), DGV->getType(),ST, "");
442aad2deb17c3e624cae4ebc1ed51fc5264fa1c0b8Chris Lattner      }
4435c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
444aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    if (DGV && DGV->hasInternalLinkage())
445aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      DGV = 0;
446aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner
4474ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner    assert(SGV->hasInitializer() || SGV->hasExternalLinkage() &&
4484ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner           "Global must either be external or have an initializer!");
4495c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
450aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    GlobalValue::LinkageTypes NewLinkage;
451aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    bool LinkFromSrc;
452aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    if (GetLinkageResult(DGV, SGV, NewLinkage, LinkFromSrc, Err))
453aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      return true;
4540fec08eb58dd9fffeb72c584aa61a59d71111c8dChris Lattner
455aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    if (!DGV) {
4565c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner      // No linking to be performed, simply create an identical version of the
4578d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      // symbol over in the dest module... the initializer will be filled in
4588d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      // later by LinkGlobalInits...
4592719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      GlobalVariable *NewDGV =
4602719bac90525985f4ebb97324b4c087d47b54384Chris Lattner        new GlobalVariable(SGV->getType()->getElementType(),
4612719bac90525985f4ebb97324b4c087d47b54384Chris Lattner                           SGV->isConstant(), SGV->getLinkage(), /*init*/0,
4622719bac90525985f4ebb97324b4c087d47b54384Chris Lattner                           SGV->getName(), Dest);
4632719bac90525985f4ebb97324b4c087d47b54384Chris Lattner
4642719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      // If the LLVM runtime renamed the global, but it is an externally visible
4652719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      // symbol, DGV must be an existing global with internal linkage.  Rename
4662719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      // it.
467c003628a612d3687fb77088a5894314210a65385Chris Lattner      if (NewDGV->getName() != SGV->getName() && !NewDGV->hasInternalLinkage())
468c003628a612d3687fb77088a5894314210a65385Chris Lattner        ForceRenaming(NewDGV, SGV->getName());
4695c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
4705c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner      // Make sure to remember this mapping...
4712719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      ValueMap.insert(std::make_pair(SGV, NewDGV));
4728166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      if (SGV->hasAppendingLinkage())
4738166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        // Keep track that this is an appending variable...
4748166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        AppendingVars.insert(std::make_pair(SGV->getName(), NewDGV));
475aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner    } else if (DGV->hasAppendingLinkage()) {
4768166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // No linking is performed yet.  Just insert a new copy of the global, and
4778166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // keep track of the fact that it is an appending variable in the
4788166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // AppendingVars map.  The name is cleared out so that no linkage is
4798166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // performed.
4808166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      GlobalVariable *NewDGV =
4818166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        new GlobalVariable(SGV->getType()->getElementType(),
4828166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                           SGV->isConstant(), SGV->getLinkage(), /*init*/0,
4838166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                           "", Dest);
4848166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
4858166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Make sure to remember this mapping...
4868166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      ValueMap.insert(std::make_pair(SGV, NewDGV));
4878166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
4888166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Keep track that this is an appending variable...
4898166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      AppendingVars.insert(std::make_pair(SGV->getName(), NewDGV));
4904ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner    } else {
491aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      // Otherwise, perform the mapping as instructed by GetLinkageResult.  If
492aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      // the types don't match, and if we are to link from the source, nuke DGV
493aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      // and create a new one of the appropriate type.
494aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      if (SGV->getType() != DGV->getType() && LinkFromSrc) {
495aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner        GlobalVariable *NewDGV =
496aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner          new GlobalVariable(SGV->getType()->getElementType(),
497aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner                             DGV->isConstant(), DGV->getLinkage());
498aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner        Dest->getGlobalList().insert(DGV, NewDGV);
499aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner        DGV->replaceAllUsesWith(ConstantExpr::getCast(NewDGV, DGV->getType()));
500aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner        DGV->eraseFromParent();
501aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner        NewDGV->setName(SGV->getName());
502aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner        DGV = NewDGV;
503aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      }
504aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner
505aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      DGV->setLinkage(NewLinkage);
506aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner
507aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      if (LinkFromSrc) {
508aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner        // Inherit const as appropriate
509e6f8c5a71682ffcc33d3c83a40821ded8f2c8033Chris Lattner        DGV->setConstant(SGV->isConstant());
510aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner        DGV->setInitializer(0);
511aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      } else {
512aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner        if (SGV->isConstant() && !DGV->isConstant()) {
513e6f8c5a71682ffcc33d3c83a40821ded8f2c8033Chris Lattner          if (DGV->isExternal())
514aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner            DGV->setConstant(true);
515aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner        }
516c8ef1edc63508d21bf8e38112fc125cde01fdbb9Chris Lattner        SGV->setLinkage(GlobalValue::ExternalLinkage);
517c8ef1edc63508d21bf8e38112fc125cde01fdbb9Chris Lattner        SGV->setInitializer(0);
518aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      }
519aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner
520aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner      ValueMap.insert(std::make_pair(SGV,
521aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner                                     ConstantExpr::getCast(DGV,
522aee38ea5698fd770bb41d01d06542cb849e131a4Chris Lattner                                                           SGV->getType())));
5235c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner    }
5245c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  }
5255c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return false;
5265c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
5275c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
5285c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
5298d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner// LinkGlobalInits - Update the initializers in the Dest module now that all
5308d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner// globals that may be referenced are in Dest.
5318d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattnerstatic bool LinkGlobalInits(Module *Dest, const Module *Src,
5325c2d335d86579941b71816196a95a64b277e8963Chris Lattner                            std::map<const Value*, Value*> &ValueMap,
5335c2d335d86579941b71816196a95a64b277e8963Chris Lattner                            std::string *Err) {
5348d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner
5358d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  // Loop over all of the globals in the src module, mapping them over as we go
536e4d5c441e04bdc00ccf1804744af670655123b07Chris Lattner  for (Module::const_global_iterator I = Src->global_begin(), E = Src->global_end(); I != E; ++I){
53718961504fc2b299578dba817900a0696cf3ccc4dChris Lattner    const GlobalVariable *SGV = I;
5388d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner
5398d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner    if (SGV->hasInitializer()) {      // Only process initialized GV's
5408d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      // Figure out what the initializer looks like in the dest module...
5414ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner      Constant *SInit =
5420033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner        cast<Constant>(RemapOperand(SGV->getInitializer(), ValueMap));
5438d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner
544f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman      GlobalVariable *DGV = cast<GlobalVariable>(ValueMap[SGV]);
5454ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner      if (DGV->hasInitializer()) {
5464ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner        if (SGV->hasExternalLinkage()) {
5474ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner          if (DGV->getInitializer() != SInit)
548f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman            return Error(Err, "Global Variable Collision on '" +
54972cf7df6a364c213a94d01e1f0f8d918940c8416Chris Lattner                         ToStr(SGV->getType(), Src) +"':%"+SGV->getName()+
5504ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner                         " - Global variables have different initializers");
55172ac148d4964fbff950fe11a0eeeda73973b849bChris Lattner        } else if (DGV->hasLinkOnceLinkage() || DGV->hasWeakLinkage()) {
5524ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner          // Nothing is required, mapped values will take the new global
5534ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner          // automatically.
55457cb9883aa4c82357594d4186858f175135be3b1Chris Lattner        } else if (SGV->hasLinkOnceLinkage() || SGV->hasWeakLinkage()) {
55557cb9883aa4c82357594d4186858f175135be3b1Chris Lattner          // Nothing is required, mapped values will take the new global
55657cb9883aa4c82357594d4186858f175135be3b1Chris Lattner          // automatically.
5574ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner        } else if (DGV->hasAppendingLinkage()) {
5584ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner          assert(0 && "Appending linkage unimplemented!");
5594ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner        } else {
5604ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner          assert(0 && "Unknown linkage!");
5614ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner        }
5628d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      } else {
5638d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner        // Copy the initializer over now...
5644ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner        DGV->setInitializer(SInit);
5658d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner      }
5668d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner    }
5678d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  }
5688d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner  return false;
5698d2de8a82cce67513debee7a3fa5aca0189b4105Chris Lattner}
5705c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
57179df7c0aaa18129e55968c8783ef8346807bd4afChris Lattner// LinkFunctionProtos - Link the functions together between the two modules,
572c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// without doing function bodies... this just adds external function prototypes
573c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// to the Dest function...
5745c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner//
57579df7c0aaa18129e55968c8783ef8346807bd4afChris Lattnerstatic bool LinkFunctionProtos(Module *Dest, const Module *Src,
5765c2d335d86579941b71816196a95a64b277e8963Chris Lattner                               std::map<const Value*, Value*> &ValueMap,
5775a837defff581accc09048bea1cd839009e374a7Chris Lattner                             std::map<std::string, GlobalValue*> &GlobalsByName,
5785c2d335d86579941b71816196a95a64b277e8963Chris Lattner                               std::string *Err) {
579b91b657e026e86082b665cc875c2881ed075b68dChris Lattner  SymbolTable *ST = (SymbolTable*)&Dest->getSymbolTable();
580f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
581c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // Loop over all of the functions in the src module, mapping them over as we
582c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // go
5835c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  for (Module::const_iterator I = Src->begin(), E = Src->end(); I != E; ++I) {
58418961504fc2b299578dba817900a0696cf3ccc4dChris Lattner    const Function *SF = I;   // SrcFunction
5854ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner    Function *DF = 0;
5865a837defff581accc09048bea1cd839009e374a7Chris Lattner    if (SF->hasName() && !SF->hasInternalLinkage()) {
5875a837defff581accc09048bea1cd839009e374a7Chris Lattner      // Check to see if may have to link the function.
588aad2deb17c3e624cae4ebc1ed51fc5264fa1c0b8Chris Lattner      if (!(DF = Dest->getFunction(SF->getName(), SF->getFunctionType()))) {
589aad2deb17c3e624cae4ebc1ed51fc5264fa1c0b8Chris Lattner        std::map<std::string, GlobalValue*>::iterator EF =
590aad2deb17c3e624cae4ebc1ed51fc5264fa1c0b8Chris Lattner          GlobalsByName.find(SF->getName());
591aad2deb17c3e624cae4ebc1ed51fc5264fa1c0b8Chris Lattner        if (EF != GlobalsByName.end())
592aad2deb17c3e624cae4ebc1ed51fc5264fa1c0b8Chris Lattner          DF = dyn_cast<Function>(EF->second);
593aad2deb17c3e624cae4ebc1ed51fc5264fa1c0b8Chris Lattner        if (DF && RecursiveResolveTypes(SF->getType(), DF->getType(), ST, ""))
594aad2deb17c3e624cae4ebc1ed51fc5264fa1c0b8Chris Lattner          DF = 0;  // FIXME: gross.
595aad2deb17c3e624cae4ebc1ed51fc5264fa1c0b8Chris Lattner      }
5965a837defff581accc09048bea1cd839009e374a7Chris Lattner    }
5975c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
5984ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner    if (!DF || SF->hasInternalLinkage() || DF->hasInternalLinkage()) {
5990fec08eb58dd9fffeb72c584aa61a59d71111c8dChris Lattner      // Function does not already exist, simply insert an function signature
6000fec08eb58dd9fffeb72c584aa61a59d71111c8dChris Lattner      // identical to SF into the dest module...
6012719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      Function *NewDF = new Function(SF->getFunctionType(), SF->getLinkage(),
6022719bac90525985f4ebb97324b4c087d47b54384Chris Lattner                                     SF->getName(), Dest);
603cb9048a6e813ca0c89f59375f3aa15ae49e153b4Chris Lattner      NewDF->setCallingConv(SF->getCallingConv());
6042719bac90525985f4ebb97324b4c087d47b54384Chris Lattner
6052719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      // If the LLVM runtime renamed the function, but it is an externally
6062719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      // visible symbol, DF must be an existing function with internal linkage.
6072719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      // Rename it.
608c003628a612d3687fb77088a5894314210a65385Chris Lattner      if (NewDF->getName() != SF->getName() && !NewDF->hasInternalLinkage())
60982b5b219dd3b9035442389ddcd58e6f859aae039Chris Lattner        ForceRenaming(NewDF, SF->getName());
6104ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner
6114ad02e726d9b634372b037d4b352d8b63bb9e849Chris Lattner      // ... and remember this mapping...
6122719bac90525985f4ebb97324b4c087d47b54384Chris Lattner      ValueMap.insert(std::make_pair(SF, NewDF));
613c2b97d43a074445c6856d69538a95de09450c26eChris Lattner    } else if (SF->isExternal()) {
614c2b97d43a074445c6856d69538a95de09450c26eChris Lattner      // If SF is external or if both SF & DF are external..  Just link the
615c2b97d43a074445c6856d69538a95de09450c26eChris Lattner      // external functions, we aren't adding anything.
616c2b97d43a074445c6856d69538a95de09450c26eChris Lattner      ValueMap.insert(std::make_pair(SF, DF));
617c2b97d43a074445c6856d69538a95de09450c26eChris Lattner    } else if (DF->isExternal()) {   // If DF is external but SF is not...
618c2b97d43a074445c6856d69538a95de09450c26eChris Lattner      // Link the external functions, update linkage qualifiers
619c2b97d43a074445c6856d69538a95de09450c26eChris Lattner      ValueMap.insert(std::make_pair(SF, DF));
620c2b97d43a074445c6856d69538a95de09450c26eChris Lattner      DF->setLinkage(SF->getLinkage());
621c2b97d43a074445c6856d69538a95de09450c26eChris Lattner
62235956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner    } else if (SF->hasWeakLinkage() || SF->hasLinkOnceLinkage()) {
62372ac148d4964fbff950fe11a0eeeda73973b849bChris Lattner      // At this point we know that DF has LinkOnce, Weak, or External linkage.
62472ac148d4964fbff950fe11a0eeeda73973b849bChris Lattner      ValueMap.insert(std::make_pair(SF, DF));
62572ac148d4964fbff950fe11a0eeeda73973b849bChris Lattner
62635956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner      // Linkonce+Weak = Weak
62735956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner      if (DF->hasLinkOnceLinkage() && SF->hasWeakLinkage())
62835956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner        DF->setLinkage(SF->getLinkage());
62935956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner
63035956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner    } else if (DF->hasWeakLinkage() || DF->hasLinkOnceLinkage()) {
63172ac148d4964fbff950fe11a0eeeda73973b849bChris Lattner      // At this point we know that SF has LinkOnce or External linkage.
63272ac148d4964fbff950fe11a0eeeda73973b849bChris Lattner      ValueMap.insert(std::make_pair(SF, DF));
63372ac148d4964fbff950fe11a0eeeda73973b849bChris Lattner      if (!SF->hasLinkOnceLinkage())   // Don't inherit linkonce linkage
63472ac148d4964fbff950fe11a0eeeda73973b849bChris Lattner        DF->setLinkage(SF->getLinkage());
63572ac148d4964fbff950fe11a0eeeda73973b849bChris Lattner
636c2b97d43a074445c6856d69538a95de09450c26eChris Lattner    } else if (SF->getLinkage() != DF->getLinkage()) {
6370fec08eb58dd9fffeb72c584aa61a59d71111c8dChris Lattner      return Error(Err, "Functions named '" + SF->getName() +
6380fec08eb58dd9fffeb72c584aa61a59d71111c8dChris Lattner                   "' have different linkage specifiers!");
639c2b97d43a074445c6856d69538a95de09450c26eChris Lattner    } else if (SF->hasExternalLinkage()) {
640c2b97d43a074445c6856d69538a95de09450c26eChris Lattner      // The function is defined in both modules!!
641f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman      return Error(Err, "Function '" +
642f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman                   ToStr(SF->getFunctionType(), Src) + "':\"" +
643c2b97d43a074445c6856d69538a95de09450c26eChris Lattner                   SF->getName() + "\" - Function is already defined!");
644c2b97d43a074445c6856d69538a95de09450c26eChris Lattner    } else {
645c2b97d43a074445c6856d69538a95de09450c26eChris Lattner      assert(0 && "Unknown linkage configuration found!");
6465c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner    }
6475c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  }
6485c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return false;
6495c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
6505c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
651c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// LinkFunctionBody - Copy the source function over into the dest function and
652c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// fix up references to values.  At this point we know that Dest is an external
653c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// function, and that Src is not.
6544bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattnerstatic bool LinkFunctionBody(Function *Dest, Function *Src,
6555c2d335d86579941b71816196a95a64b277e8963Chris Lattner                             std::map<const Value*, Value*> &GlobalMap,
6565c2d335d86579941b71816196a95a64b277e8963Chris Lattner                             std::string *Err) {
6575c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  assert(Src && Dest && Dest->isExternal() && !Src->isExternal());
6585c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
6590033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner  // Go through and convert function arguments over, remembering the mapping.
660e4d5c441e04bdc00ccf1804744af670655123b07Chris Lattner  Function::arg_iterator DI = Dest->arg_begin();
661e4d5c441e04bdc00ccf1804744af670655123b07Chris Lattner  for (Function::arg_iterator I = Src->arg_begin(), E = Src->arg_end();
66269da5cf26143e4542d4bf8c78ffac6d079efe5c9Chris Lattner       I != E; ++I, ++DI) {
66369da5cf26143e4542d4bf8c78ffac6d079efe5c9Chris Lattner    DI->setName(I->getName());  // Copy the name information over...
6645c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
6655c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner    // Add a mapping to our local map
6660033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner    GlobalMap.insert(std::make_pair(I, DI));
6675c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  }
6685c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
6694bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner  // Splice the body of the source function into the dest function.
6704bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner  Dest->getBasicBlockList().splice(Dest->end(), Src->getBasicBlockList());
6715c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
672c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // At this point, all of the instructions and values of the function are now
673c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // copied over.  The only problem is that they are still referencing values in
674c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // the Source function as operands.  Loop through all of the operands of the
675c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // functions and patch them up to point to the local versions...
6765c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  //
67718961504fc2b299578dba817900a0696cf3ccc4dChris Lattner  for (Function::iterator BB = Dest->begin(), BE = Dest->end(); BB != BE; ++BB)
67818961504fc2b299578dba817900a0696cf3ccc4dChris Lattner    for (BasicBlock::iterator I = BB->begin(), E = BB->end(); I != E; ++I)
67918961504fc2b299578dba817900a0696cf3ccc4dChris Lattner      for (Instruction::op_iterator OI = I->op_begin(), OE = I->op_end();
680221d688a5ef21a22c2368c9fff0e92d7966c95e5Chris Lattner           OI != OE; ++OI)
6814bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner        if (!isa<Instruction>(*OI) && !isa<BasicBlock>(*OI))
6820033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner          *OI = RemapOperand(*OI, GlobalMap);
6830033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner
6840033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner  // There is no need to map the arguments anymore.
685e4d5c441e04bdc00ccf1804744af670655123b07Chris Lattner  for (Function::arg_iterator I = Src->arg_begin(), E = Src->arg_end(); I != E; ++I)
6860033baf94e049e075f3ba4700c109779e2d7c131Chris Lattner    GlobalMap.erase(I);
6875c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
6885c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return false;
6895c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
6905c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
6915c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
692c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// LinkFunctionBodies - Link in the function bodies that are defined in the
693c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// source module into the DestModule.  This consists basically of copying the
694c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner// function over and fixing up references to values.
6954bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattnerstatic bool LinkFunctionBodies(Module *Dest, Module *Src,
6965c2d335d86579941b71816196a95a64b277e8963Chris Lattner                               std::map<const Value*, Value*> &ValueMap,
6975c2d335d86579941b71816196a95a64b277e8963Chris Lattner                               std::string *Err) {
6985c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
699c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // Loop over all of the functions in the src module, mapping them over as we
700c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // go
7014bbfbff119acab90fc67cd3d01b156bf88832286Chris Lattner  for (Module::iterator SF = Src->begin(), E = Src->end(); SF != E; ++SF) {
70218961504fc2b299578dba817900a0696cf3ccc4dChris Lattner    if (!SF->isExternal()) {                  // No body if function is external
70318961504fc2b299578dba817900a0696cf3ccc4dChris Lattner      Function *DF = cast<Function>(ValueMap[SF]); // Destination function
704c2d774b6c1d38421c435b6d3cfaa10402c900aebChris Lattner
70518961504fc2b299578dba817900a0696cf3ccc4dChris Lattner      // DF not external SF external?
70635956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner      if (DF->isExternal()) {
70735956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner        // Only provide the function body if there isn't one already.
70835956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner        if (LinkFunctionBody(DF, SF, ValueMap, Err))
70935956558e26e655aaa2e8ad6ee405365271869a4Chris Lattner          return true;
710c2d774b6c1d38421c435b6d3cfaa10402c900aebChris Lattner      }
711c2d774b6c1d38421c435b6d3cfaa10402c900aebChris Lattner    }
7125c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  }
7135c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  return false;
7145c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner}
7155c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
7168166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner// LinkAppendingVars - If there were any appending global variables, link them
7178166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner// together now.  Return true on error.
7188166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattnerstatic bool LinkAppendingVars(Module *M,
7198166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                  std::multimap<std::string, GlobalVariable *> &AppendingVars,
7208166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                              std::string *ErrorMsg) {
7218166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  if (AppendingVars.empty()) return false; // Nothing to do.
722f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
7238166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // Loop over the multimap of appending vars, processing any variables with the
7248166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // same name, forming a new appending global variable with both of the
7258166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // initializers merged together, then rewrite references to the old variables
7268166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // and delete them.
7278166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  std::vector<Constant*> Inits;
7288166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  while (AppendingVars.size() > 1) {
7298166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    // Get the first two elements in the map...
7308166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    std::multimap<std::string,
7318166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      GlobalVariable*>::iterator Second = AppendingVars.begin(), First=Second++;
7328166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
7338166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    // If the first two elements are for different names, there is no pair...
7348166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    // Otherwise there is a pair, so link them together...
7358166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    if (First->first == Second->first) {
7368166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      GlobalVariable *G1 = First->second, *G2 = Second->second;
7378166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      const ArrayType *T1 = cast<ArrayType>(G1->getType()->getElementType());
7388166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      const ArrayType *T2 = cast<ArrayType>(G2->getType()->getElementType());
739f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
7408166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Check to see that they two arrays agree on type...
7418166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      if (T1->getElementType() != T2->getElementType())
7428166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        return Error(ErrorMsg,
7438166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner         "Appending variables with different element types need to be linked!");
7448166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      if (G1->isConstant() != G2->isConstant())
7458166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        return Error(ErrorMsg,
7468166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                     "Appending variables linked with different const'ness!");
7478166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
7488166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      unsigned NewSize = T1->getNumElements() + T2->getNumElements();
7498166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      ArrayType *NewType = ArrayType::get(T1->getElementType(), NewSize);
7508166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
751ed74a4ef3b813bed0c0f4bb5af42229d50133da2Chris Lattner      G1->setName("");   // Clear G1's name in case of a conflict!
752ed74a4ef3b813bed0c0f4bb5af42229d50133da2Chris Lattner
7538166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Create the new global variable...
7548166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      GlobalVariable *NG =
7558166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner        new GlobalVariable(NewType, G1->isConstant(), G1->getLinkage(),
7568166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner                           /*init*/0, First->first, M);
7578166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
7588166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Merge the initializer...
7598166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      Inits.reserve(NewSize);
760de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      if (ConstantArray *I = dyn_cast<ConstantArray>(G1->getInitializer())) {
761de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        for (unsigned i = 0, e = T1->getNumElements(); i != e; ++i)
762cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos          Inits.push_back(I->getOperand(i));
763de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      } else {
764de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        assert(isa<ConstantAggregateZero>(G1->getInitializer()));
765de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        Constant *CV = Constant::getNullValue(T1->getElementType());
766de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        for (unsigned i = 0, e = T1->getNumElements(); i != e; ++i)
767de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner          Inits.push_back(CV);
768de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      }
769de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      if (ConstantArray *I = dyn_cast<ConstantArray>(G2->getInitializer())) {
770de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        for (unsigned i = 0, e = T2->getNumElements(); i != e; ++i)
771cc7ba49d85b66955516ff0d9574a70e1637dea36Alkis Evlogimenos          Inits.push_back(I->getOperand(i));
772de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      } else {
773de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        assert(isa<ConstantAggregateZero>(G2->getInitializer()));
774de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        Constant *CV = Constant::getNullValue(T2->getElementType());
775de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner        for (unsigned i = 0, e = T2->getNumElements(); i != e; ++i)
776de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner          Inits.push_back(CV);
777de512b5b2edebe9c9021a92c7c7a9ae9fbc380d6Chris Lattner      }
7788166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      NG->setInitializer(ConstantArray::get(NewType, Inits));
7798166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      Inits.clear();
7808166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
7818166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Replace any uses of the two global variables with uses of the new
7828166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // global...
7838166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
7848166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // FIXME: This should rewrite simple/straight-forward uses such as
7858166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // getelementptr instructions to not use the Cast!
78600dc4796132933e76fe2b834ca087e42621c175aReid Spencer      G1->replaceAllUsesWith(ConstantExpr::getCast(NG, G1->getType()));
78700dc4796132933e76fe2b834ca087e42621c175aReid Spencer      G2->replaceAllUsesWith(ConstantExpr::getCast(NG, G2->getType()));
7888166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
7898166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Remove the two globals from the module now...
7908166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      M->getGlobalList().erase(G1);
7918166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      M->getGlobalList().erase(G2);
7928166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
7938166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // Put the new global into the AppendingVars map so that we can handle
7948166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      // linking of more than two vars...
7958166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner      Second->second = NG;
7968166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    }
7978166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner    AppendingVars.erase(First);
7988166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  }
7998166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
8008166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  return false;
8018166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner}
80252f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner
80352f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner
80452f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// LinkModules - This function links two modules together, with the resulting
80552f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// left module modified to be the composite of the two input modules.  If an
80652f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner// error occurs, true is returned and ErrorMsg (if not null) is set to indicate
8075c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// the problem.  Upon failure, the Dest module could be in a modified state, and
8085c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner// shouldn't be relied on to be consistent.
809f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukmanbool
8100ba9e211bed7813949aaca97e6b15579c8188e05Reid SpencerLinker::LinkModules(Module *Dest, Module *Src, std::string *ErrorMsg) {
81157a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  assert(Dest != 0 && "Invalid Destination module");
81257a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  assert(Src  != 0 && "Invalid Source Module");
81357a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer
814873c5e7859c1534bf92c9e2747f2b70685059598Chris Lattner  if (Dest->getEndianness() == Module::AnyEndianness)
815873c5e7859c1534bf92c9e2747f2b70685059598Chris Lattner    Dest->setEndianness(Src->getEndianness());
816873c5e7859c1534bf92c9e2747f2b70685059598Chris Lattner  if (Dest->getPointerSize() == Module::AnyPointerSize)
817873c5e7859c1534bf92c9e2747f2b70685059598Chris Lattner    Dest->setPointerSize(Src->getPointerSize());
818152f19ac6514ec29e183ddcdf2ea71599638dc4cChris Lattner  if (Dest->getTargetTriple().empty())
819152f19ac6514ec29e183ddcdf2ea71599638dc4cChris Lattner    Dest->setTargetTriple(Src->getTargetTriple());
820873c5e7859c1534bf92c9e2747f2b70685059598Chris Lattner
821873c5e7859c1534bf92c9e2747f2b70685059598Chris Lattner  if (Src->getEndianness() != Module::AnyEndianness &&
822873c5e7859c1534bf92c9e2747f2b70685059598Chris Lattner      Dest->getEndianness() != Src->getEndianness())
82343a9994c0164848d44527060da72985c75183a48Chris Lattner    std::cerr << "WARNING: Linking two modules of different endianness!\n";
824873c5e7859c1534bf92c9e2747f2b70685059598Chris Lattner  if (Src->getPointerSize() != Module::AnyPointerSize &&
825873c5e7859c1534bf92c9e2747f2b70685059598Chris Lattner      Dest->getPointerSize() != Src->getPointerSize())
82643a9994c0164848d44527060da72985c75183a48Chris Lattner    std::cerr << "WARNING: Linking two modules of different pointer size!\n";
827152f19ac6514ec29e183ddcdf2ea71599638dc4cChris Lattner  if (!Src->getTargetTriple().empty() &&
828152f19ac6514ec29e183ddcdf2ea71599638dc4cChris Lattner      Dest->getTargetTriple() != Src->getTargetTriple())
829152f19ac6514ec29e183ddcdf2ea71599638dc4cChris Lattner    std::cerr << "WARNING: Linking two modules of different target triples!\n";
830f976c856fcc5055f3fc7d9f070d72c2d027c1d9dMisha Brukman
831719012d6254ee425464f034250176283fb96793bReid Spencer  // Update the destination module's dependent libraries list with the libraries
83257a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  // from the source module. There's no opportunity for duplicates here as the
83357a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  // Module ensures that duplicate insertions are discarded.
83457a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  Module::lib_iterator SI = Src->lib_begin();
83557a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  Module::lib_iterator SE = Src->lib_end();
83657a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  while ( SI != SE ) {
83757a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer    Dest->addLibrary(*SI);
83857a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer    ++SI;
83957a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  }
84057a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer
8412c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  // LinkTypes - Go through the symbol table of the Src module and see if any
8422c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  // types are named in the src module that are not named in the Dst module.
8432c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  // Make sure there are no type name conflicts.
8442c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner  if (LinkTypes(Dest, Src, ErrorMsg)) return true;
8452c236f3e20a0fb84b5948efa6bb7bb60d759cb32Chris Lattner
8465c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  // ValueMap - Mapping of values from what they used to be in Src, to what they
8475c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner  // are now in Dest.
8485c2d335d86579941b71816196a95a64b277e8963Chris Lattner  std::map<const Value*, Value*> ValueMap;
8495c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
8508166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // AppendingVars - Keep track of global variables in the destination module
8518166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // with appending linkage.  After the module is linked together, they are
8528166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // appended and the module is rewritten.
8538166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  std::multimap<std::string, GlobalVariable *> AppendingVars;
8548166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
8555a837defff581accc09048bea1cd839009e374a7Chris Lattner  // GlobalsByName - The LLVM SymbolTable class fights our best efforts at
8565a837defff581accc09048bea1cd839009e374a7Chris Lattner  // linking by separating globals by type.  Until PR411 is fixed, we replicate
8575a837defff581accc09048bea1cd839009e374a7Chris Lattner  // it's functionality here.
8585a837defff581accc09048bea1cd839009e374a7Chris Lattner  std::map<std::string, GlobalValue*> GlobalsByName;
8595a837defff581accc09048bea1cd839009e374a7Chris Lattner
860e4d5c441e04bdc00ccf1804744af670655123b07Chris Lattner  for (Module::global_iterator I = Dest->global_begin(), E = Dest->global_end(); I != E; ++I) {
8615a837defff581accc09048bea1cd839009e374a7Chris Lattner    // Add all of the appending globals already in the Dest module to
8625a837defff581accc09048bea1cd839009e374a7Chris Lattner    // AppendingVars.
863f41464618f41b92cf0a0bd14ccc57bce727fb33dChris Lattner    if (I->hasAppendingLinkage())
864f41464618f41b92cf0a0bd14ccc57bce727fb33dChris Lattner      AppendingVars.insert(std::make_pair(I->getName(), I));
8658166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
8665a837defff581accc09048bea1cd839009e374a7Chris Lattner    // Keep track of all globals by name.
8675a837defff581accc09048bea1cd839009e374a7Chris Lattner    if (!I->hasInternalLinkage() && I->hasName())
8685a837defff581accc09048bea1cd839009e374a7Chris Lattner      GlobalsByName[I->getName()] = I;
8695a837defff581accc09048bea1cd839009e374a7Chris Lattner  }
8705a837defff581accc09048bea1cd839009e374a7Chris Lattner
8715a837defff581accc09048bea1cd839009e374a7Chris Lattner  // Keep track of all globals by name.
8725a837defff581accc09048bea1cd839009e374a7Chris Lattner  for (Module::iterator I = Dest->begin(), E = Dest->end(); I != E; ++I)
8735a837defff581accc09048bea1cd839009e374a7Chris Lattner    if (!I->hasInternalLinkage() && I->hasName())
8745a837defff581accc09048bea1cd839009e374a7Chris Lattner      GlobalsByName[I->getName()] = I;
8755a837defff581accc09048bea1cd839009e374a7Chris Lattner
8768166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // Insert all of the globals in src into the Dest module... without linking
8778166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // initializers (which could refer to functions not yet mapped over).
8785a837defff581accc09048bea1cd839009e374a7Chris Lattner  if (LinkGlobals(Dest, Src, ValueMap, AppendingVars, GlobalsByName, ErrorMsg))
8795a837defff581accc09048bea1cd839009e374a7Chris Lattner    return true;
8805c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
881c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // Link the functions together between the two modules, without doing function
882c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // bodies... this just adds external function prototypes to the Dest
883c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // function...  We do this so that when we begin processing function bodies,
884c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // all of the global values that may be referenced are available in our
885c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // ValueMap.
8865a837defff581accc09048bea1cd839009e374a7Chris Lattner  if (LinkFunctionProtos(Dest, Src, ValueMap, GlobalsByName, ErrorMsg))
8875a837defff581accc09048bea1cd839009e374a7Chris Lattner    return true;
8885c377c524a6929cdaa683f3f034b3fc01526b264Chris Lattner
8896cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner  // Update the initializers in the Dest module now that all globals that may
8906cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner  // be referenced are in Dest.
8916cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner  if (LinkGlobalInits(Dest, Src, ValueMap, ErrorMsg)) return true;
8926cdf1971bdf88ddd9a7d46b5f5f975497d68c38eChris Lattner
893c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // Link in the function bodies that are defined in the source module into the
894c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // DestModule.  This consists basically of copying the function over and
895c8cc4cb03bd90f89be7fe1649542a2d5ae689632Chris Lattner  // fixing up references to values.
89679df7c0aaa18129e55968c8783ef8346807bd4afChris Lattner  if (LinkFunctionBodies(Dest, Src, ValueMap, ErrorMsg)) return true;
89752f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner
8988166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  // If there were any appending global variables, link them together now.
8998166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner  if (LinkAppendingVars(Dest, AppendingVars, ErrorMsg)) return true;
9008166e6eef6518da6f1f805e0d5e2bb22c15dd49cChris Lattner
90157a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  // If the source library's module id is in the dependent library list of the
90257a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  // destination library, remove it since that module is now linked in.
90357a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer  sys::Path modId;
904dd04df0ec33a903ee7fc747701bafde622f77d8bReid Spencer  modId.set(Src->getModuleIdentifier());
90507adb2836b8aa7a3872e33c285958f5937662b50Reid Spencer  if (!modId.isEmpty())
90607adb2836b8aa7a3872e33c285958f5937662b50Reid Spencer    Dest->removeLibrary(modId.getBasename());
90757a0efa32240ef03ea318f9ba7680fd2b8609c6eReid Spencer
90852f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner  return false;
90952f7e908cb634f9b9b539d5c6670b8a065478915Chris Lattner}
9109466f5113ba859677a28887c3c7143065e702019Vikram S. Adve
911567bc2cc1ed4d610daceecbb4ff7f92ebde1b530Reid Spencer// vim: sw=2
912