contrib/llvm/tools/llvm-ranlib/llvm-ranlib.cpp

   1 //===-- llvm-ranlib.cpp - LLVM archive index generator --------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // Adds or updates an index (symbol table) for an LLVM archive file.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "llvm/IR/LLVMContext.h"
  15 #include "llvm/Bitcode/Archive.h"
  16 #include "llvm/IR/Module.h"
  17 #include "llvm/Support/CommandLine.h"
  18 #include "llvm/Support/FileSystem.h"
  19 #include "llvm/Support/Format.h"
  20 #include "llvm/Support/ManagedStatic.h"
  21 #include "llvm/Support/PrettyStackTrace.h"
  22 #include "llvm/Support/Signals.h"
  23 #include "llvm/Support/raw_ostream.h"
  24 #include <memory>
  25 using namespace llvm;
  26
  27 // llvm-ar operation code and modifier flags
  28 static cl::opt<std::string>
  29 ArchiveName(cl::Positional, cl::Optional, cl::desc("<archive-file>"));
  30
  31 static cl::opt<bool>
  32 Verbose("verbose",cl::Optional,cl::init(false),
  33         cl::desc("Print the symbol table"));
  34
  35 // printSymbolTable - print out the archive's symbol table.
  36 void printSymbolTable(Archive* TheArchive) {
  37   outs() << "\nArchive Symbol Table:\n";
  38   const Archive::SymTabType& symtab = TheArchive->getSymbolTable();
  39   for (Archive::SymTabType::const_iterator I=symtab.begin(), E=symtab.end();
  40        I != E; ++I ) {
  41     unsigned offset = TheArchive->getFirstFileOffset() + I->second;
  42     outs() << " " << format("%9u", offset) << "\t" << I->first <<"\n";
  43   }
  44 }
  45
  46 int main(int argc, char **argv) {
  47   // Print a stack trace if we signal out.
  48   llvm::sys::PrintStackTraceOnErrorSignal();
  49   llvm::PrettyStackTraceProgram X(argc, argv);
  50
  51   LLVMContext &Context = getGlobalContext();
  52   llvm_shutdown_obj Y;  // Call llvm_shutdown() on exit.
  53
  54   // Have the command line options parsed and handle things
  55   // like --help and --version.
  56   cl::ParseCommandLineOptions(argc, argv,
  57     "LLVM Archive Index Generator (llvm-ranlib)\n\n"
  58     "  This program adds or updates an index of bitcode symbols\n"
  59     "  to an LLVM archive file."
  60   );
  61
  62   int exitCode = 0;
  63
  64   // Check the path name of the archive
  65   sys::Path ArchivePath;
  66   if (!ArchivePath.set(ArchiveName)) {
  67     errs() << argv[0] << ": " << "Archive name invalid: " << ArchiveName <<
  68       "\n";
  69     return 1;
  70   }
  71
  72   // Make sure it exists, we don't create empty archives
  73   bool Exists;
  74   if (llvm::sys::fs::exists(ArchivePath.str(), Exists) || !Exists) {
  75     errs() << argv[0] << ": " << "Archive file does not exist" <<
  76       ArchivePath.str() << "\n";
  77     return 1;
  78   }
  79
  80   std::string err_msg;
  81   OwningPtr<Archive>
  82     AutoArchive(Archive::OpenAndLoad(ArchivePath, Context, &err_msg));
  83   Archive* TheArchive = AutoArchive.get();
  84   if (!TheArchive) {
  85     errs() << argv[0] << ": " << err_msg << "\n";
  86     return 1;
  87   }
  88
  89   if (TheArchive->writeToDisk(true, false, &err_msg )) {
  90     errs() << argv[0] << ": " << err_msg << "\n";
  91     return 1;
  92   }
  93
  94   if (Verbose)
  95     printSymbolTable(TheArchive);
  96
  97   return exitCode;
  98 }