lib/Linker/LinkItems.cpp

   1 //===- lib/Linker/LinkItems.cpp - Link LLVM objects and libraries ---------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file was developed by Reid Spencer and is distributed under the
   6 // University of Illinois Open Source License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file contains routines to handle linking together LLVM bytecode files,
  11 // and to handle annoying things like static libraries.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "llvm/Linker.h"
  16 #include "llvm/Module.h"
  17
  18 using namespace llvm;
  19
  20 // LinkItems - This function is the main entry point into linking. It takes a
  21 // list of LinkItem which indicates the order the files should be linked and
  22 // how each file should be treated (plain file or with library search). The
  23 // function only links bytecode and produces a result list of items that are
  24 // native objects.
  25 bool
  26 Linker::LinkInItems(const ItemList& Items, ItemList& NativeItems) {
  27   // Clear the NativeItems just in case
  28   NativeItems.clear();
  29
  30   // For each linkage item ...
  31   for (ItemList::const_iterator I = Items.begin(), E = Items.end();
  32        I != E; ++I) {
  33     if (I->second) {
  34       // Link in the library suggested.
  35       bool is_native = false;
  36       if (LinkInLibrary(I->first, is_native))
  37         return true;
  38       if (is_native)
  39         NativeItems.push_back(*I);
  40     } else {
  41       // Link in the file suggested
  42       bool is_native = false;
  43       if (LinkInFile(sys::Path(I->first), is_native))
  44         return true;
  45       if (is_native)
  46         NativeItems.push_back(*I);
  47     }
  48   }
  49
  50   // At this point we have processed all the link items provided to us. Since
  51   // we have an aggregated module at this point, the dependent libraries in
  52   // that module should also be aggregated with duplicates eliminated. This is
  53   // now the time to process the dependent libraries to resolve any remaining
  54   // symbols.
  55   bool is_native;
  56   for (Module::lib_iterator I = Composite->lib_begin(),
  57          E = Composite->lib_end(); I != E; ++I) {
  58     if(LinkInLibrary(*I, is_native))
  59       return true;
  60     if (is_native)
  61       NativeItems.push_back(std::make_pair(*I, true));
  62   }
  63
  64   return false;
  65 }
  66
  67
  68 /// LinkInLibrary - links one library into the HeadModule.
  69 ///
  70 bool Linker::LinkInLibrary(const std::string& Lib, bool& is_native) {
  71   is_native = false;
  72   // Determine where this library lives.
  73   sys::Path Pathname = FindLib(Lib);
  74   if (Pathname.isEmpty())
  75     return warning("Cannot find library '" + Lib + "'");
  76
  77   // If its an archive, try to link it in
  78   std::string Magic;
  79   Pathname.getMagicNumber(Magic, 64);
  80   switch (sys::IdentifyFileType(Magic.c_str(), 64)) {
  81     default: assert(0 && "Bad file type identification");
  82     case sys::Unknown_FileType:
  83       return warning("Supposed library '" + Lib + "' isn't a library.");
  84
  85     case sys::Bytecode_FileType:
  86     case sys::CompressedBytecode_FileType:
  87       // LLVM ".so" file.
  88       if (LinkInFile(Pathname, is_native))
  89         return error("Cannot link file '" + Pathname.toString() + "'");
  90       break;
  91
  92     case sys::Archive_FileType:
  93       if (LinkInArchive(Pathname, is_native))
  94         return error("Cannot link archive '" + Pathname.toString() + "'");
  95       break;
  96
  97     case sys::ELF_Relocatable_FileType:
  98     case sys::ELF_SharedObject_FileType:
  99     case sys::Mach_O_Object_FileType:
 100     case sys::Mach_O_FixedVirtualMemorySharedLib_FileType:
 101     case sys::Mach_O_DynamicallyLinkedSharedLib_FileType:
 102     case sys::Mach_O_DynamicallyLinkedSharedLibStub_FileType:
 103     case sys::COFF_FileType:
 104       is_native = true;
 105       break;
 106   }
 107   return false;
 108 }
 109
 110 /// LinkLibraries - takes the specified library files and links them into the
 111 /// main bytecode object file.
 112 ///
 113 /// Inputs:
 114 ///  Libraries  - The list of libraries to link into the module.
 115 ///
 116 /// Return value:
 117 ///  FALSE - No error.
 118 ///  TRUE  - Error.
 119 ///
 120 bool Linker::LinkInLibraries(const std::vector<std::string> &Libraries) {
 121
 122   // Process the set of libraries we've been provided.
 123   bool is_native = false;
 124   for (unsigned i = 0; i < Libraries.size(); ++i)
 125     if (LinkInLibrary(Libraries[i], is_native))
 126       return true;
 127
 128   // At this point we have processed all the libraries provided to us. Since
 129   // we have an aggregated module at this point, the dependent libraries in
 130   // that module should also be aggregated with duplicates eliminated. This is
 131   // now the time to process the dependent libraries to resolve any remaining
 132   // symbols.
 133   const Module::LibraryListType& DepLibs = Composite->getLibraries();
 134   for (Module::LibraryListType::const_iterator I = DepLibs.begin(),
 135          E = DepLibs.end(); I != E; ++I)
 136     if (LinkInLibrary(*I, is_native))
 137       return true;
 138
 139   return false;
 140 }
 141
 142 /// LinkInFile - opens a bytecode file and links in all objects which
 143 /// provide symbols that are currently undefined.
 144 ///
 145 /// Inputs:
 146 ///  File - The pathname of the bytecode file.
 147 ///
 148 /// Outputs:
 149 ///  ErrorMessage - A C++ string detailing what error occurred, if any.
 150 ///
 151 /// Return Value:
 152 ///  TRUE  - An error occurred.
 153 ///  FALSE - No errors.
 154 ///
 155 bool Linker::LinkInFile(const sys::Path &File, bool &is_native) {
 156   is_native = false;
 157   // Make sure we can at least read the file
 158   if (!File.canRead())
 159     return error("Cannot find linker input '" + File.toString() + "'");
 160
 161   // If its an archive, try to link it in
 162   std::string Magic;
 163   File.getMagicNumber(Magic, 64);
 164   switch (sys::IdentifyFileType(Magic.c_str(), 64)) {
 165     default: assert(0 && "Bad file type identification");
 166     case sys::Unknown_FileType:
 167       return warning("Supposed object file '" + File.toString() +
 168                      "' not recognized as such");
 169
 170     case sys::Archive_FileType:
 171       // A user may specify an ar archive without -l, perhaps because it
 172       // is not installed as a library. Detect that and link the archive.
 173       verbose("Linking archive file '" + File.toString() + "'");
 174       if (LinkInArchive(File, is_native))
 175         return error("Cannot link archive '" + File.toString() + "'");
 176       break;
 177
 178     case sys::Bytecode_FileType:
 179     case sys::CompressedBytecode_FileType: {
 180       verbose("Linking bytecode file '" + File.toString() + "'");
 181       std::auto_ptr<Module> M(LoadObject(File));
 182       if (M.get() == 0)
 183         return error("Cannot load file '" + File.toString() + "'" + Error);
 184       if (LinkInModule(M.get()))
 185         return error("Cannot link file '" + File.toString() + "'" + Error);
 186
 187       verbose("Linked in file '" + File.toString() + "'");
 188       break;
 189     }
 190
 191     case sys::ELF_Relocatable_FileType:
 192     case sys::ELF_SharedObject_FileType:
 193     case sys::Mach_O_Object_FileType:
 194     case sys::Mach_O_FixedVirtualMemorySharedLib_FileType:
 195     case sys::Mach_O_DynamicallyLinkedSharedLib_FileType:
 196     case sys::Mach_O_DynamicallyLinkedSharedLibStub_FileType:
 197     case sys::COFF_FileType:
 198       is_native = true;
 199       break;
 200   }
 201   return false;
 202 }
 203
 204 /// LinkFiles - takes a module and a list of files and links them all together.
 205 /// It locates the file either in the current directory, as its absolute
 206 /// or relative pathname, or as a file somewhere in LLVM_LIB_SEARCH_PATH.
 207 ///
 208 /// Inputs:
 209 ///  Files      - A vector of sys::Path indicating the LLVM bytecode filenames
 210 ///               to be linked.  The names can refer to a mixture of pure LLVM
 211 ///               bytecode files and archive (ar) formatted files.
 212 ///
 213 /// Return value:
 214 ///  FALSE - No errors.
 215 ///  TRUE  - Some error occurred.
 216 ///
 217 bool Linker::LinkInFiles(const std::vector<sys::Path> &Files) {
 218   bool is_native;
 219   for (unsigned i = 0; i < Files.size(); ++i)
 220     if (LinkInFile(Files[i], is_native))
 221       return true;
 222   return false;
 223 }