This file contains basic Linker facilities needed by all tools

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@18860 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 166 insertions, 0 deletions

diff --git a/lib/Linker/Linker.cpp b/lib/Linker/Linker.cpp
new file mode 100644
index 0000000000..e72f5b9f6a
--- /dev/null
+++ b/lib/Linker/Linker.cpp
@@ -0,0 +1,166 @@
+//===- lib/Linker/LinkItems.cpp - Link LLVM objects and libraries ---------===//
+// 
+//                     The LLVM Compiler Infrastructure
+//
+// This file was developed by Reid Spencer and is distributed under the 
+// University of Illinois Open Source License. See LICENSE.TXT for details.
+// 
+//===----------------------------------------------------------------------===//
+//
+// This file contains routines to handle linking together LLVM bytecode files,
+// and to handle annoying things like static libraries.
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/Linker.h"
+#include "llvm/Module.h"
+#include "llvm/Bytecode/Reader.h"
+#include <iostream>
+
+using namespace llvm;
+
+Linker::Linker(const std::string& progname, unsigned flags)
+  : Composite(0)
+  , LibPaths()
+  , Flags(flags)
+  , Error()
+  , ProgramName(progname)
+{
+  Composite = new Module(progname);
+}
+
+Linker::Linker(const std::string& progname, Module* aModule, unsigned flags)
+  : Composite(aModule)
+  , LibPaths()
+  , Flags(flags)
+  , Error()
+  , ProgramName(progname)
+{
+}
+
+Linker::~Linker() {
+  delete Composite;
+}
+
+bool 
+Linker::error(const std::string& message) {
+  Error = message;
+  if (!(Flags&QuietErrors)) {
+    std::cerr << ProgramName << ": error: " << message << "\n";
+  }
+  return true;
+}
+
+bool
+Linker::warning(const std::string& message) {
+  Error = message;
+  if (!(Flags&QuietErrors)) {
+    std::cerr << ProgramName << ": warning: " << message << "\n";
+  }
+  return false;
+}
+
+void
+Linker::verbose(const std::string& message) {
+  if (Flags&Verbose) {
+    std::cerr << "  " << message << "\n";
+  }
+}
+
+void
+Linker::addPath(const sys::Path& path) {
+  assert(path.isDirectory() && "Can only insert directories into the path");
+  LibPaths.push_back(path);
+}
+
+void
+Linker::addPaths(const std::vector<std::string>& paths) {
+  for (unsigned i = 0; i < paths.size(); i++) {
+    sys::Path aPath;
+    aPath.setDirectory(paths[i]);
+    LibPaths.push_back(aPath);
+  }
+}
+
+void
+Linker::addSystemPaths() {
+  sys::Path::GetBytecodeLibraryPaths(LibPaths);
+  LibPaths.insert(LibPaths.begin(),sys::Path("./"));
+}
+
+Module*
+Linker::releaseModule() {
+  Module* result = Composite;
+  Composite = new Module(ProgramName);
+  LibPaths.clear();
+  Error.clear();
+  Flags = 0;
+  return result;
+}
+
+// LoadObject - Read in and parse the bytecode file named by FN and return the
+// module it contains (wrapped in an auto_ptr), or auto_ptr<Module>() and set 
+// Error if an error occurs.
+std::auto_ptr<Module> 
+Linker::LoadObject(const sys::Path &FN) {
+  std::string ParseErrorMessage;
+  Module *Result = ParseBytecodeFile(FN.toString(), &ParseErrorMessage);
+  if (Result) 
+    return std::auto_ptr<Module>(Result);
+  Error = "Bytecode file '" + FN.toString() + "' could not be loaded";
+  if (ParseErrorMessage.size()) 
+    Error += ": " + ParseErrorMessage;
+  return std::auto_ptr<Module>();
+}
+
+static inline sys::Path IsLibrary(const std::string& Name, 
+                                  const sys::Path& Directory) {
+
+  assert(Directory.isDirectory() && "Need to specify a directory");
+  sys::Path FullPath(Directory);
+  FullPath.appendFile("lib" + Name);
+
+  FullPath.appendSuffix("a");
+  if (FullPath.isArchive())
+    return FullPath;
+
+  FullPath.elideSuffix();
+  FullPath.appendSuffix("bca");
+  if (FullPath.isArchive())
+    return FullPath;
+
+  FullPath.elideSuffix();
+  FullPath.appendSuffix(&(LTDL_SHLIB_EXT[1]));
+  if (FullPath.isDynamicLibrary())
+    return FullPath;
+
+  FullPath.clear();
+  return FullPath;
+}
+
+/// FindLib - Try to convert Filename into the name of a file that we can open,
+/// if it does not already name a file we can open, by first trying to open
+/// Filename, then libFilename.[suffix] for each of a set of several common
+/// library suffixes, in each of the directories in Paths and the directory
+/// named by the value of the environment variable LLVM_LIB_SEARCH_PATH. Returns
+/// an empty string if no matching file can be found.
+///
+sys::Path 
+Linker::FindLib(const std::string &Filename) 
+{
+  // Determine if the pathname can be found as it stands.
+  sys::Path FilePath(Filename);
+  if (FilePath.readable() && 
+      (FilePath.isArchive() || FilePath.isDynamicLibrary()))
+    return FilePath;
+
+  // Iterate over the directories in Paths to see if we can find the library
+  // there.
+  for (unsigned Index = 0; Index != LibPaths.size(); ++Index) {
+    sys::Path Directory(LibPaths[Index]);
+    sys::Path FullPath = IsLibrary(Filename,Directory);
+    if (!FullPath.isEmpty())
+      return FullPath;
+  }
+  return sys::Path();
+}