diff options
| author | Fangrui Song <maskray@google.com> | 2018-10-26 17:38:27 +0000 | 
|---|---|---|
| committer | Fangrui Song <maskray@google.com> | 2018-10-26 17:38:27 +0000 | 
| commit | f0333f0b07aabbdfb490677fdb2695483f585d3b (patch) | |
| tree | 3fdb2176918a3b53e10571b2cbd2d0d08664feb9 /llvm/tools/llvm-ar/llvm-ar.cpp | |
| parent | 8315d9990c1f98efa069f9d27275f0fbfc3cc0fd (diff) | |
| download | bcm5719-llvm-f0333f0b07aabbdfb490677fdb2695483f585d3b.tar.gz bcm5719-llvm-f0333f0b07aabbdfb490677fdb2695483f585d3b.zip | |
[llvm-ar] Strip trailing \r and format
Reviewers: mstorsjo, rupprecht, gbreynoo
Reviewed By: rupprecht
Subscribers: llvm-commits
Differential Revision: https://reviews.llvm.org/D53769
llvm-svn: 345410
Diffstat (limited to 'llvm/tools/llvm-ar/llvm-ar.cpp')
| -rw-r--r-- | llvm/tools/llvm-ar/llvm-ar.cpp | 2064 | 
1 files changed, 1046 insertions, 1018 deletions
| diff --git a/llvm/tools/llvm-ar/llvm-ar.cpp b/llvm/tools/llvm-ar/llvm-ar.cpp index 2c6dc8fad92..5ab8ae13d3e 100644 --- a/llvm/tools/llvm-ar/llvm-ar.cpp +++ b/llvm/tools/llvm-ar/llvm-ar.cpp @@ -1,1018 +1,1046 @@ -//===-- llvm-ar.cpp - LLVM archive librarian utility ----------------------===//
 -//
 -//                     The LLVM Compiler Infrastructure
 -//
 -// This file is distributed under the University of Illinois Open Source
 -// License. See LICENSE.TXT for details.
 -//
 -//===----------------------------------------------------------------------===//
 -//
 -// Builds up (relatively) standard unix archive files (.a) containing LLVM
 -// bitcode or other files.
 -//
 -//===----------------------------------------------------------------------===//
 -
 -#include "llvm/ADT/StringSwitch.h"
 -#include "llvm/ADT/Triple.h"
 -#include "llvm/IR/LLVMContext.h"
 -#include "llvm/Object/Archive.h"
 -#include "llvm/Object/ArchiveWriter.h"
 -#include "llvm/Object/MachO.h"
 -#include "llvm/Object/ObjectFile.h"
 -#include "llvm/Support/Chrono.h"
 -#include "llvm/Support/CommandLine.h"
 -#include "llvm/Support/Errc.h"
 -#include "llvm/Support/FileSystem.h"
 -#include "llvm/Support/Format.h"
 -#include "llvm/Support/FormatVariadic.h"
 -#include "llvm/Support/InitLLVM.h"
 -#include "llvm/Support/LineIterator.h"
 -#include "llvm/Support/MemoryBuffer.h"
 -#include "llvm/Support/Path.h"
 -#include "llvm/Support/Process.h"
 -#include "llvm/Support/StringSaver.h"
 -#include "llvm/Support/TargetSelect.h"
 -#include "llvm/Support/ToolOutputFile.h"
 -#include "llvm/Support/raw_ostream.h"
 -#include "llvm/ToolDrivers/llvm-dlltool/DlltoolDriver.h"
 -#include "llvm/ToolDrivers/llvm-lib/LibDriver.h"
 -
 -#if !defined(_MSC_VER) && !defined(__MINGW32__)
 -#include <unistd.h>
 -#else
 -#include <io.h>
 -#endif
 -
 -using namespace llvm;
 -
 -// The name this program was invoked as.
 -static StringRef ToolName;
 -
 -// The basename of this program.
 -static StringRef Stem;
 -
 -const char RanlibHelp[] = R"(
 -OVERVIEW: LLVM Ranlib (llvm-ranlib)
 -
 -  This program generates an index to speed access to archives
 -
 -USAGE: llvm-ranlib <archive-file>
 -
 -OPTIONS:
 -  -help                             - Display available options
 -  -version                          - Display the version of this program
 -)";
 -
 -const char ArHelp[] = R"(
 -OVERVIEW: LLVM Archiver
 -
 -USAGE: llvm-ar [options] [-]<operation>[modifiers] [relpos] <archive> [files]
 -       llvm-ar -M [<mri-script]
 -
 -OPTIONS:
 -  --format              - Archive format to create
 -    =default            -   default
 -    =gnu                -   gnu
 -    =darwin             -   darwin
 -    =bsd                -   bsd
 -  --plugin=<string>     - Ignored for compatibility
 -  --help                - Display available options
 -  --version             - Display the version of this program
 -
 -OPERATIONS:
 -  d - delete [files] from the archive
 -  m - move [files] in the archive
 -  p - print [files] found in the archive
 -  q - quick append [files] to the archive
 -  r - replace or insert [files] into the archive
 -  s - act as ranlib
 -  t - display contents of archive
 -  x - extract [files] from the archive
 -
 -MODIFIERS:
 -  [a] - put [files] after [relpos]
 -  [b] - put [files] before [relpos] (same as [i])
 -  [c] - do not warn if archive had to be created
 -  [D] - use zero for timestamps and uids/gids (default)
 -  [i] - put [files] before [relpos] (same as [b])
 -  [l] - ignored for compatibility
 -  [L] - add archive's contents
 -  [o] - preserve original dates
 -  [s] - create an archive index (cf. ranlib)
 -  [S] - do not build a symbol table
 -  [T] - create a thin archive
 -  [u] - update only [files] newer than archive contents
 -  [U] - use actual timestamps and uids/gids
 -  [v] - be verbose about actions taken
 -)";
 -
 -void printHelpMessage() {
 -  if (Stem.contains_lower("ranlib"))
 -    outs() << RanlibHelp;
 -  else if (Stem.contains_lower("ar"))
 -    outs() << ArHelp;
 -}
 -
 -// Show the error message and exit.
 -LLVM_ATTRIBUTE_NORETURN static void fail(Twine Error) {
 -  errs() << ToolName << ": " << Error << ".\n";
 -  printHelpMessage();
 -  exit(1);
 -}
 -
 -static void failIfError(std::error_code EC, Twine Context = "") {
 -  if (!EC)
 -    return;
 -
 -  std::string ContextStr = Context.str();
 -  if (ContextStr == "")
 -    fail(EC.message());
 -  fail(Context + ": " + EC.message());
 -}
 -
 -static void failIfError(Error E, Twine Context = "") {
 -  if (!E)
 -    return;
 -
 -  handleAllErrors(std::move(E), [&](const llvm::ErrorInfoBase &EIB) {
 -    std::string ContextStr = Context.str();
 -    if (ContextStr == "")
 -      fail(EIB.message());
 -    fail(Context + ": " + EIB.message());
 -  });
 -}
 -
 -static SmallVector<const char *, 256> PositionalArgs;
 -
 -static bool MRI;
 -
 -namespace {
 -enum Format { Default, GNU, BSD, DARWIN, Unknown };
 -}
 -
 -static Format FormatType = Default;
 -
 -static std::string Options;
 -
 -// This enumeration delineates the kinds of operations on an archive
 -// that are permitted.
 -enum ArchiveOperation {
 -  Print,            ///< Print the contents of the archive
 -  Delete,           ///< Delete the specified members
 -  Move,             ///< Move members to end or as given by {a,b,i} modifiers
 -  QuickAppend,      ///< Quickly append to end of archive
 -  ReplaceOrInsert,  ///< Replace or Insert members
 -  DisplayTable,     ///< Display the table of contents
 -  Extract,          ///< Extract files back to file system
 -  CreateSymTab      ///< Create a symbol table in an existing archive
 -};
 -
 -// Modifiers to follow operation to vary behavior
 -static bool AddAfter = false;      ///< 'a' modifier
 -static bool AddBefore = false;     ///< 'b' modifier
 -static bool Create = false;        ///< 'c' modifier
 -static bool OriginalDates = false; ///< 'o' modifier
 -static bool OnlyUpdate = false;    ///< 'u' modifier
 -static bool Verbose = false;       ///< 'v' modifier
 -static bool Symtab = true;         ///< 's' modifier
 -static bool Deterministic = true;  ///< 'D' and 'U' modifiers
 -static bool Thin = false;          ///< 'T' modifier
 -static bool AddLibrary = false;    ///< 'L' modifier
 -
 -// Relative Positional Argument (for insert/move). This variable holds
 -// the name of the archive member to which the 'a', 'b' or 'i' modifier
 -// refers. Only one of 'a', 'b' or 'i' can be specified so we only need
 -// one variable.
 -static std::string RelPos;
 -
 -// This variable holds the name of the archive file as given on the
 -// command line.
 -static std::string ArchiveName;
 -
 -// This variable holds the list of member files to proecess, as given
 -// on the command line.
 -static std::vector<StringRef> Members;
 -
 -// Extract the member filename from the command line for the [relpos] argument
 -// associated with a, b, and i modifiers
 -static void getRelPos() {
 -  if (PositionalArgs.size() == 0)
 -    fail("Expected [relpos] for a, b, or i modifier");
 -  RelPos = PositionalArgs[0];
 -  PositionalArgs.erase(PositionalArgs.begin());
 -}
 -
 -// Get the archive file name from the command line
 -static void getArchive() {
 -  if (PositionalArgs.size() == 0)
 -    fail("An archive name must be specified");
 -  ArchiveName = PositionalArgs[0];
 -  PositionalArgs.erase(PositionalArgs.begin());
 -}
 -
 -// Copy over remaining items in PositionalArgs to our Members vector
 -static void getMembers() {
 -  for (auto &Arg : PositionalArgs)
 -    Members.push_back(Arg);
 -}
 -
 -std::vector<std::unique_ptr<MemoryBuffer>> ArchiveBuffers;
 -std::vector<std::unique_ptr<object::Archive>> Archives;
 -
 -static object::Archive &readLibrary(const Twine &Library) {
 -  auto BufOrErr = MemoryBuffer::getFile(Library, -1, false);
 -  failIfError(BufOrErr.getError(), "Could not open library");
 -  ArchiveBuffers.push_back(std::move(*BufOrErr));
 -  auto LibOrErr =
 -      object::Archive::create(ArchiveBuffers.back()->getMemBufferRef());
 -  failIfError(errorToErrorCode(LibOrErr.takeError()),
 -              "Could not parse library");
 -  Archives.push_back(std::move(*LibOrErr));
 -  return *Archives.back();
 -}
 -
 -static void runMRIScript();
 -
 -// Parse the command line options as presented and return the operation
 -// specified. Process all modifiers and check to make sure that constraints on
 -// modifier/operation pairs have not been violated.
 -static ArchiveOperation parseCommandLine() {
 -  if (MRI) {
 -    if (!PositionalArgs.empty() || !Options.empty())
 -      fail("Cannot mix -M and other options");
 -    runMRIScript();
 -  }
 -
 -  // Keep track of number of operations. We can only specify one
 -  // per execution.
 -  unsigned NumOperations = 0;
 -
 -  // Keep track of the number of positional modifiers (a,b,i). Only
 -  // one can be specified.
 -  unsigned NumPositional = 0;
 -
 -  // Keep track of which operation was requested
 -  ArchiveOperation Operation;
 -
 -  bool MaybeJustCreateSymTab = false;
 -
 -  for(unsigned i=0; i<Options.size(); ++i) {
 -    switch(Options[i]) {
 -    case 'd': ++NumOperations; Operation = Delete; break;
 -    case 'm': ++NumOperations; Operation = Move ; break;
 -    case 'p': ++NumOperations; Operation = Print; break;
 -    case 'q': ++NumOperations; Operation = QuickAppend; break;
 -    case 'r': ++NumOperations; Operation = ReplaceOrInsert; break;
 -    case 't': ++NumOperations; Operation = DisplayTable; break;
 -    case 'x': ++NumOperations; Operation = Extract; break;
 -    case 'c': Create = true; break;
 -    case 'l': /* accepted but unused */ break;
 -    case 'o': OriginalDates = true; break;
 -    case 's':
 -      Symtab = true;
 -      MaybeJustCreateSymTab = true;
 -      break;
 -    case 'S':
 -      Symtab = false;
 -      break;
 -    case 'u': OnlyUpdate = true; break;
 -    case 'v': Verbose = true; break;
 -    case 'a':
 -      getRelPos();
 -      AddAfter = true;
 -      NumPositional++;
 -      break;
 -    case 'b':
 -      getRelPos();
 -      AddBefore = true;
 -      NumPositional++;
 -      break;
 -    case 'i':
 -      getRelPos();
 -      AddBefore = true;
 -      NumPositional++;
 -      break;
 -    case 'D':
 -      Deterministic = true;
 -      break;
 -    case 'U':
 -      Deterministic = false;
 -      break;
 -    case 'T':
 -      Thin = true;
 -      break;
 -    case 'L':
 -      AddLibrary = true;
 -      break;
 -    default:
 -      fail(std::string("unknown option ") + Options[i]);
 -    }
 -  }
 -
 -  // At this point, the next thing on the command line must be
 -  // the archive name.
 -  getArchive();
 -
 -  // Everything on the command line at this point is a member.
 -  getMembers();
 -
 - if (NumOperations == 0 && MaybeJustCreateSymTab) {
 -    NumOperations = 1;
 -    Operation = CreateSymTab;
 -    if (!Members.empty())
 -      fail("The s operation takes only an archive as argument");
 -  }
 -
 -  // Perform various checks on the operation/modifier specification
 -  // to make sure we are dealing with a legal request.
 -  if (NumOperations == 0)
 -    fail("You must specify at least one of the operations");
 -  if (NumOperations > 1)
 -    fail("Only one operation may be specified");
 -  if (NumPositional > 1)
 -    fail("You may only specify one of a, b, and i modifiers");
 -  if (AddAfter || AddBefore) {
 -    if (Operation != Move && Operation != ReplaceOrInsert)
 -      fail("The 'a', 'b' and 'i' modifiers can only be specified with "
 -           "the 'm' or 'r' operations");
 -  }
 -  if (OriginalDates && Operation != Extract)
 -    fail("The 'o' modifier is only applicable to the 'x' operation");
 -  if (OnlyUpdate && Operation != ReplaceOrInsert)
 -    fail("The 'u' modifier is only applicable to the 'r' operation");
 -  if (AddLibrary && Operation != QuickAppend)
 -    fail("The 'L' modifier is only applicable to the 'q' operation");
 -
 -  // Return the parsed operation to the caller
 -  return Operation;
 -}
 -
 -// Implements the 'p' operation. This function traverses the archive
 -// looking for members that match the path list.
 -static void doPrint(StringRef Name, const object::Archive::Child &C) {
 -  if (Verbose)
 -    outs() << "Printing " << Name << "\n";
 -
 -  Expected<StringRef> DataOrErr = C.getBuffer();
 -  failIfError(DataOrErr.takeError());
 -  StringRef Data = *DataOrErr;
 -  outs().write(Data.data(), Data.size());
 -}
 -
 -// Utility function for printing out the file mode when the 't' operation is in
 -// verbose mode.
 -static void printMode(unsigned mode) {
 -  outs() << ((mode & 004) ? "r" : "-");
 -  outs() << ((mode & 002) ? "w" : "-");
 -  outs() << ((mode & 001) ? "x" : "-");
 -}
 -
 -// Implement the 't' operation. This function prints out just
 -// the file names of each of the members. However, if verbose mode is requested
 -// ('v' modifier) then the file type, permission mode, user, group, size, and
 -// modification time are also printed.
 -static void doDisplayTable(StringRef Name, const object::Archive::Child &C) {
 -  if (Verbose) {
 -    Expected<sys::fs::perms> ModeOrErr = C.getAccessMode();
 -    failIfError(ModeOrErr.takeError());
 -    sys::fs::perms Mode = ModeOrErr.get();
 -    printMode((Mode >> 6) & 007);
 -    printMode((Mode >> 3) & 007);
 -    printMode(Mode & 007);
 -    Expected<unsigned> UIDOrErr = C.getUID();
 -    failIfError(UIDOrErr.takeError());
 -    outs() << ' ' << UIDOrErr.get();
 -    Expected<unsigned> GIDOrErr = C.getGID();
 -    failIfError(GIDOrErr.takeError());
 -    outs() << '/' << GIDOrErr.get();
 -    Expected<uint64_t> Size = C.getSize();
 -    failIfError(Size.takeError());
 -    outs() << ' ' << format("%6llu", Size.get());
 -    auto ModTimeOrErr = C.getLastModified();
 -    failIfError(ModTimeOrErr.takeError());
 -    // Note: formatv() only handles the default TimePoint<>, which is in
 -    // nanoseconds.
 -    // TODO: fix format_provider<TimePoint<>> to allow other units.
 -    sys::TimePoint<> ModTimeInNs = ModTimeOrErr.get();
 -    outs() << ' ' << formatv("{0:%b %e %H:%M %Y}", ModTimeInNs);
 -    outs() << ' ';
 -  }
 -
 -  if (C.getParent()->isThin()) {
 -    outs() << sys::path::parent_path(ArchiveName);
 -    outs() << '/';
 -  }
 -  outs() << Name << "\n";
 -}
 -
 -// Implement the 'x' operation. This function extracts files back to the file
 -// system.
 -static void doExtract(StringRef Name, const object::Archive::Child &C) {
 -  // Retain the original mode.
 -  Expected<sys::fs::perms> ModeOrErr = C.getAccessMode();
 -  failIfError(ModeOrErr.takeError());
 -  sys::fs::perms Mode = ModeOrErr.get();
 -
 -  int FD;
 -  failIfError(sys::fs::openFileForWrite(sys::path::filename(Name), FD,
 -                                        sys::fs::CD_CreateAlways,
 -                                        sys::fs::F_None, Mode),
 -              Name);
 -
 -  {
 -    raw_fd_ostream file(FD, false);
 -
 -    // Get the data and its length
 -    Expected<StringRef> BufOrErr = C.getBuffer();
 -    failIfError(BufOrErr.takeError());
 -    StringRef Data = BufOrErr.get();
 -
 -    // Write the data.
 -    file.write(Data.data(), Data.size());
 -  }
 -
 -  // If we're supposed to retain the original modification times, etc. do so
 -  // now.
 -  if (OriginalDates) {
 -    auto ModTimeOrErr = C.getLastModified();
 -    failIfError(ModTimeOrErr.takeError());
 -    failIfError(
 -        sys::fs::setLastAccessAndModificationTime(FD, ModTimeOrErr.get()));
 -  }
 -
 -  if (close(FD))
 -    fail("Could not close the file");
 -}
 -
 -static bool shouldCreateArchive(ArchiveOperation Op) {
 -  switch (Op) {
 -  case Print:
 -  case Delete:
 -  case Move:
 -  case DisplayTable:
 -  case Extract:
 -  case CreateSymTab:
 -    return false;
 -
 -  case QuickAppend:
 -  case ReplaceOrInsert:
 -    return true;
 -  }
 -
 -  llvm_unreachable("Missing entry in covered switch.");
 -}
 -
 -static void performReadOperation(ArchiveOperation Operation,
 -                                 object::Archive *OldArchive) {
 -  if (Operation == Extract && OldArchive->isThin())
 -    fail("extracting from a thin archive is not supported");
 -
 -  bool Filter = !Members.empty();
 -  {
 -    Error Err = Error::success();
 -    for (auto &C : OldArchive->children(Err)) {
 -      Expected<StringRef> NameOrErr = C.getName();
 -      failIfError(NameOrErr.takeError());
 -      StringRef Name = NameOrErr.get();
 -
 -      if (Filter) {
 -        auto I = find(Members, Name);
 -        if (I == Members.end())
 -          continue;
 -        Members.erase(I);
 -      }
 -
 -      switch (Operation) {
 -      default:
 -        llvm_unreachable("Not a read operation");
 -      case Print:
 -        doPrint(Name, C);
 -        break;
 -      case DisplayTable:
 -        doDisplayTable(Name, C);
 -        break;
 -      case Extract:
 -        doExtract(Name, C);
 -        break;
 -      }
 -    }
 -    failIfError(std::move(Err));
 -  }
 -
 -  if (Members.empty())
 -    return;
 -  for (StringRef Name : Members)
 -    errs() << Name << " was not found\n";
 -  exit(1);
 -}
 -
 -static void addMember(std::vector<NewArchiveMember> &Members,
 -                      StringRef FileName, int Pos = -1) {
 -  Expected<NewArchiveMember> NMOrErr =
 -      NewArchiveMember::getFile(FileName, Deterministic);
 -  failIfError(NMOrErr.takeError(), FileName);
 -
 -  // Use the basename of the object path for the member name.
 -  NMOrErr->MemberName = sys::path::filename(NMOrErr->MemberName);
 -
 -  if (Pos == -1)
 -    Members.push_back(std::move(*NMOrErr));
 -  else
 -    Members[Pos] = std::move(*NMOrErr);
 -}
 -
 -static void addMember(std::vector<NewArchiveMember> &Members,
 -                      const object::Archive::Child &M, int Pos = -1) {
 -  if (Thin && !M.getParent()->isThin())
 -    fail("Cannot convert a regular archive to a thin one");
 -  Expected<NewArchiveMember> NMOrErr =
 -      NewArchiveMember::getOldMember(M, Deterministic);
 -  failIfError(NMOrErr.takeError());
 -  if (Pos == -1)
 -    Members.push_back(std::move(*NMOrErr));
 -  else
 -    Members[Pos] = std::move(*NMOrErr);
 -}
 -
 -static void addLibMember(std::vector<NewArchiveMember> &Members,
 -                         StringRef FileName) {
 -  Expected<NewArchiveMember> NMOrErr =
 -      NewArchiveMember::getFile(FileName, Deterministic);
 -  failIfError(NMOrErr.takeError(), FileName);
 -  if (identify_magic(NMOrErr->Buf->getBuffer()) == file_magic::archive) {
 -    object::Archive &Lib = readLibrary(FileName);
 -    Error Err = Error::success();
 -
 -    for (auto &Child : Lib.children(Err))
 -      addMember(Members, Child);
 -
 -    failIfError(std::move(Err));
 -  } else {
 -    // Use the basename of the object path for the member name.
 -    NMOrErr->MemberName = sys::path::filename(NMOrErr->MemberName);
 -    Members.push_back(std::move(*NMOrErr));
 -  }
 -}
 -
 -enum InsertAction {
 -  IA_AddOldMember,
 -  IA_AddNewMember,
 -  IA_Delete,
 -  IA_MoveOldMember,
 -  IA_MoveNewMember
 -};
 -
 -static InsertAction computeInsertAction(ArchiveOperation Operation,
 -                                        const object::Archive::Child &Member,
 -                                        StringRef Name,
 -                                        std::vector<StringRef>::iterator &Pos) {
 -  if (Operation == QuickAppend || Members.empty())
 -    return IA_AddOldMember;
 -
 -  auto MI = find_if(Members, [Name](StringRef Path) {
 -    return Name == sys::path::filename(Path);
 -  });
 -
 -  if (MI == Members.end())
 -    return IA_AddOldMember;
 -
 -  Pos = MI;
 -
 -  if (Operation == Delete)
 -    return IA_Delete;
 -
 -  if (Operation == Move)
 -    return IA_MoveOldMember;
 -
 -  if (Operation == ReplaceOrInsert) {
 -    StringRef PosName = sys::path::filename(RelPos);
 -    if (!OnlyUpdate) {
 -      if (PosName.empty())
 -        return IA_AddNewMember;
 -      return IA_MoveNewMember;
 -    }
 -
 -    // We could try to optimize this to a fstat, but it is not a common
 -    // operation.
 -    sys::fs::file_status Status;
 -    failIfError(sys::fs::status(*MI, Status), *MI);
 -    auto ModTimeOrErr = Member.getLastModified();
 -    failIfError(ModTimeOrErr.takeError());
 -    if (Status.getLastModificationTime() < ModTimeOrErr.get()) {
 -      if (PosName.empty())
 -        return IA_AddOldMember;
 -      return IA_MoveOldMember;
 -    }
 -
 -    if (PosName.empty())
 -      return IA_AddNewMember;
 -    return IA_MoveNewMember;
 -  }
 -  llvm_unreachable("No such operation");
 -}
 -
 -// We have to walk this twice and computing it is not trivial, so creating an
 -// explicit std::vector is actually fairly efficient.
 -static std::vector<NewArchiveMember>
 -computeNewArchiveMembers(ArchiveOperation Operation,
 -                         object::Archive *OldArchive) {
 -  std::vector<NewArchiveMember> Ret;
 -  std::vector<NewArchiveMember> Moved;
 -  int InsertPos = -1;
 -  StringRef PosName = sys::path::filename(RelPos);
 -  if (OldArchive) {
 -    Error Err = Error::success();
 -    for (auto &Child : OldArchive->children(Err)) {
 -      int Pos = Ret.size();
 -      Expected<StringRef> NameOrErr = Child.getName();
 -      failIfError(NameOrErr.takeError());
 -      StringRef Name = NameOrErr.get();
 -      if (Name == PosName) {
 -        assert(AddAfter || AddBefore);
 -        if (AddBefore)
 -          InsertPos = Pos;
 -        else
 -          InsertPos = Pos + 1;
 -      }
 -
 -      std::vector<StringRef>::iterator MemberI = Members.end();
 -      InsertAction Action =
 -          computeInsertAction(Operation, Child, Name, MemberI);
 -      switch (Action) {
 -      case IA_AddOldMember:
 -        addMember(Ret, Child);
 -        break;
 -      case IA_AddNewMember:
 -        addMember(Ret, *MemberI);
 -        break;
 -      case IA_Delete:
 -        break;
 -      case IA_MoveOldMember:
 -        addMember(Moved, Child);
 -        break;
 -      case IA_MoveNewMember:
 -        addMember(Moved, *MemberI);
 -        break;
 -      }
 -      if (MemberI != Members.end())
 -        Members.erase(MemberI);
 -    }
 -    failIfError(std::move(Err));
 -  }
 -
 -  if (Operation == Delete)
 -    return Ret;
 -
 -  if (!RelPos.empty() && InsertPos == -1)
 -    fail("Insertion point not found");
 -
 -  if (RelPos.empty())
 -    InsertPos = Ret.size();
 -
 -  assert(unsigned(InsertPos) <= Ret.size());
 -  int Pos = InsertPos;
 -  for (auto &M : Moved) {
 -    Ret.insert(Ret.begin() + Pos, std::move(M));
 -    ++Pos;
 -  }
 -
 -  if (AddLibrary) {
 -    assert(Operation == QuickAppend);
 -    for (auto &Member : Members)
 -      addLibMember(Ret, Member);
 -    return Ret;
 -  }
 -
 -  for (unsigned I = 0; I != Members.size(); ++I)
 -    Ret.insert(Ret.begin() + InsertPos, NewArchiveMember());
 -  Pos = InsertPos;
 -  for (auto &Member : Members) {
 -    addMember(Ret, Member, Pos);
 -    ++Pos;
 -  }
 -
 -  return Ret;
 -}
 -
 -static object::Archive::Kind getDefaultForHost() {
 -  return Triple(sys::getProcessTriple()).isOSDarwin()
 -             ? object::Archive::K_DARWIN
 -             : object::Archive::K_GNU;
 -}
 -
 -static object::Archive::Kind getKindFromMember(const NewArchiveMember &Member) {
 -  Expected<std::unique_ptr<object::ObjectFile>> OptionalObject =
 -      object::ObjectFile::createObjectFile(Member.Buf->getMemBufferRef());
 -
 -  if (OptionalObject)
 -    return isa<object::MachOObjectFile>(**OptionalObject)
 -               ? object::Archive::K_DARWIN
 -               : object::Archive::K_GNU;
 -
 -  // squelch the error in case we had a non-object file
 -  consumeError(OptionalObject.takeError());
 -  return getDefaultForHost();
 -}
 -
 -static void
 -performWriteOperation(ArchiveOperation Operation,
 -                      object::Archive *OldArchive,
 -                      std::unique_ptr<MemoryBuffer> OldArchiveBuf,
 -                      std::vector<NewArchiveMember> *NewMembersP) {
 -  std::vector<NewArchiveMember> NewMembers;
 -  if (!NewMembersP)
 -    NewMembers = computeNewArchiveMembers(Operation, OldArchive);
 -
 -  object::Archive::Kind Kind;
 -  switch (FormatType) {
 -  case Default:
 -    if (Thin)
 -      Kind = object::Archive::K_GNU;
 -    else if (OldArchive)
 -      Kind = OldArchive->kind();
 -    else if (NewMembersP)
 -      Kind = NewMembersP->size() ? getKindFromMember(NewMembersP->front())
 -                                 : getDefaultForHost();
 -    else
 -      Kind = NewMembers.size() ? getKindFromMember(NewMembers.front())
 -                               : getDefaultForHost();
 -    break;
 -  case GNU:
 -    Kind = object::Archive::K_GNU;
 -    break;
 -  case BSD:
 -    if (Thin)
 -      fail("Only the gnu format has a thin mode");
 -    Kind = object::Archive::K_BSD;
 -    break;
 -  case DARWIN:
 -    if (Thin)
 -      fail("Only the gnu format has a thin mode");
 -    Kind = object::Archive::K_DARWIN;
 -    break;
 -  case Unknown:
 -    llvm_unreachable("");
 -  }
 -
 -  Error E =
 -      writeArchive(ArchiveName, NewMembersP ? *NewMembersP : NewMembers, Symtab,
 -                   Kind, Deterministic, Thin, std::move(OldArchiveBuf));
 -  failIfError(std::move(E), ArchiveName);
 -}
 -
 -static void createSymbolTable(object::Archive *OldArchive) {
 -  // When an archive is created or modified, if the s option is given, the
 -  // resulting archive will have a current symbol table. If the S option
 -  // is given, it will have no symbol table.
 -  // In summary, we only need to update the symbol table if we have none.
 -  // This is actually very common because of broken build systems that think
 -  // they have to run ranlib.
 -  if (OldArchive->hasSymbolTable())
 -    return;
 -
 -  performWriteOperation(CreateSymTab, OldArchive, nullptr, nullptr);
 -}
 -
 -static void performOperation(ArchiveOperation Operation,
 -                             object::Archive *OldArchive,
 -                             std::unique_ptr<MemoryBuffer> OldArchiveBuf,
 -                             std::vector<NewArchiveMember> *NewMembers) {
 -  switch (Operation) {
 -  case Print:
 -  case DisplayTable:
 -  case Extract:
 -    performReadOperation(Operation, OldArchive);
 -    return;
 -
 -  case Delete:
 -  case Move:
 -  case QuickAppend:
 -  case ReplaceOrInsert:
 -    performWriteOperation(Operation, OldArchive, std::move(OldArchiveBuf),
 -                          NewMembers);
 -    return;
 -  case CreateSymTab:
 -    createSymbolTable(OldArchive);
 -    return;
 -  }
 -  llvm_unreachable("Unknown operation.");
 -}
 -
 -static int performOperation(ArchiveOperation Operation,
 -                            std::vector<NewArchiveMember> *NewMembers) {
 -  // Create or open the archive object.
 -  ErrorOr<std::unique_ptr<MemoryBuffer>> Buf =
 -      MemoryBuffer::getFile(ArchiveName, -1, false);
 -  std::error_code EC = Buf.getError();
 -  if (EC && EC != errc::no_such_file_or_directory)
 -    fail("error opening '" + ArchiveName + "': " + EC.message() + "!");
 -
 -  if (!EC) {
 -    Error Err = Error::success();
 -    object::Archive Archive(Buf.get()->getMemBufferRef(), Err);
 -    EC = errorToErrorCode(std::move(Err));
 -    failIfError(EC,
 -                "error loading '" + ArchiveName + "': " + EC.message() + "!");
 -    performOperation(Operation, &Archive, std::move(Buf.get()), NewMembers);
 -    return 0;
 -  }
 -
 -  assert(EC == errc::no_such_file_or_directory);
 -
 -  if (!shouldCreateArchive(Operation)) {
 -    failIfError(EC, Twine("error loading '") + ArchiveName + "'");
 -  } else {
 -    if (!Create) {
 -      // Produce a warning if we should and we're creating the archive
 -      errs() << ToolName << ": creating " << ArchiveName << "\n";
 -    }
 -  }
 -
 -  performOperation(Operation, nullptr, nullptr, NewMembers);
 -  return 0;
 -}
 -
 -static void runMRIScript() {
 -  enum class MRICommand { AddLib, AddMod, Create, Delete, Save, End, Invalid };
 -
 -  ErrorOr<std::unique_ptr<MemoryBuffer>> Buf = MemoryBuffer::getSTDIN();
 -  failIfError(Buf.getError());
 -  const MemoryBuffer &Ref = *Buf.get();
 -  bool Saved = false;
 -  std::vector<NewArchiveMember> NewMembers;
 -
 -  for (line_iterator I(Ref, /*SkipBlanks*/ false), E; I != E; ++I) {
 -    StringRef Line = *I;
 -    Line = Line.split(';').first;
 -    Line = Line.split('*').first;
 -    Line = Line.trim();
 -    if (Line.empty())
 -      continue;
 -    StringRef CommandStr, Rest;
 -    std::tie(CommandStr, Rest) = Line.split(' ');
 -    Rest = Rest.trim();
 -    if (!Rest.empty() && Rest.front() == '"' && Rest.back() == '"')
 -      Rest = Rest.drop_front().drop_back();
 -    auto Command = StringSwitch<MRICommand>(CommandStr.lower())
 -                       .Case("addlib", MRICommand::AddLib)
 -                       .Case("addmod", MRICommand::AddMod)
 -                       .Case("create", MRICommand::Create)
 -                       .Case("delete", MRICommand::Delete)
 -                       .Case("save", MRICommand::Save)
 -                       .Case("end", MRICommand::End)
 -                       .Default(MRICommand::Invalid);
 -
 -    switch (Command) {
 -    case MRICommand::AddLib: {
 -      object::Archive &Lib = readLibrary(Rest);
 -      {
 -        Error Err = Error::success();
 -        for (auto &Member : Lib.children(Err))
 -          addMember(NewMembers, Member);
 -        failIfError(std::move(Err));
 -      }
 -      break;
 -    }
 -    case MRICommand::AddMod:
 -      addMember(NewMembers, Rest);
 -      break;
 -    case MRICommand::Create:
 -      Create = true;
 -      if (!ArchiveName.empty())
 -        fail("Editing multiple archives not supported");
 -      if (Saved)
 -        fail("File already saved");
 -      ArchiveName = Rest;
 -      break;
 -    case MRICommand::Delete: {
 -      StringRef Name = sys::path::filename(Rest);
 -      llvm::erase_if(NewMembers,
 -                     [=](NewArchiveMember &M) { return M.MemberName == Name; });
 -      break;
 -    }
 -    case MRICommand::Save:
 -      Saved = true;
 -      break;
 -    case MRICommand::End:
 -      break;
 -    case MRICommand::Invalid:
 -      fail("Unknown command: " + CommandStr);
 -    }
 -  }
 -
 -  // Nothing to do if not saved.
 -  if (Saved)
 -    performOperation(ReplaceOrInsert, &NewMembers);
 -  exit(0);
 -}
 -
 -static bool handleGenericOption(StringRef arg) {
 -  if (arg == "-help" || arg == "--help") {
 -    printHelpMessage();
 -    return true;
 -  }
 -  if (arg == "-version" || arg == "--version") {
 -    cl::PrintVersionMessage();
 -    return true;
 -  }
 -  return false;
 -}
 -
 -static int ar_main(int argc, char **argv) {
 -  SmallVector<const char *, 0> Argv(argv, argv + argc);
 -  BumpPtrAllocator Alloc;
 -  StringSaver Saver(Alloc);
 -  cl::ExpandResponseFiles(Saver, cl::TokenizeGNUCommandLine, Argv);
 -  for(size_t i = 1; i < Argv.size(); ++i) {
 -    StringRef Arg = Argv[i];
 -    const char *match;
 -    auto MatchFlagWithArg = [&](const char *expected) {
 -      size_t len = strlen(expected);
 -      if (Arg == expected) {
 -        if (++i >= Argv.size())
 -          fail(std::string(expected) + " requires an argument");
 -        match = Argv[i];
 -        return true;
 -      }
 -      if (Arg.startswith(expected) && Arg.size() > len &&
 -                 Arg[len] == '=') {
 -        match = Arg.data() + len + 1;
 -        return true;
 -      }
 -      return false;
 -    };
 -    if (handleGenericOption(Argv[i]))
 -      return 0;
 -    if (Arg == "--") {
 -      for(; i < Argv.size(); ++i)
 -        PositionalArgs.push_back(Argv[i]);
 -      break;
 -    }
 -    if (Arg[0] == '-') {
 -      if (Arg.startswith("--"))
 -        Arg = Argv[i] + 2;
 -      else
 -        Arg = Argv[i] + 1;
 -      if (Arg == "M") {
 -        MRI = true;
 -      } else if (MatchFlagWithArg("format")) {
 -        FormatType = StringSwitch<Format>(match)
 -            .Case("default", Default)
 -            .Case("gnu", GNU)
 -            .Case("darwin", DARWIN)
 -            .Case("bsd", BSD)
 -            .Default(Unknown);
 -        if (FormatType == Unknown)
 -          fail(std::string("Invalid format ") + match);
 -      } else if (MatchFlagWithArg("plugin")) {
 -        // Ignored.
 -      } else {
 -        Options += Argv[i] + 1;
 -      }
 -    } else if (Options.empty()) {
 -      Options += Argv[i];
 -    } else {
 -      PositionalArgs.push_back(Argv[i]);
 -    }
 -  }
 -  ArchiveOperation Operation = parseCommandLine();
 -  return performOperation(Operation, nullptr);
 -}
 -
 -static int ranlib_main(int argc, char **argv) {
 -  bool ArchiveSpecified = false;
 -  for(int i = 1; i < argc; ++i) {
 -    if (handleGenericOption(argv[i])) {
 -      return 0;
 -    } else {
 -      if (ArchiveSpecified)
 -        fail("Exactly one archive should be specified");
 -      ArchiveSpecified = true;
 -      ArchiveName = argv[i];
 -    }
 -  }
 -  return performOperation(CreateSymTab, nullptr);
 -}
 -
 -int main(int argc, char **argv) {
 -  InitLLVM X(argc, argv);
 -  ToolName = argv[0];
 -
 -  llvm::InitializeAllTargetInfos();
 -  llvm::InitializeAllTargetMCs();
 -  llvm::InitializeAllAsmParsers();
 -
 -  Stem = sys::path::stem(ToolName);
 -  if (Stem.contains_lower("dlltool"))
 -    return dlltoolDriverMain(makeArrayRef(argv, argc));
 -
 -  if (Stem.contains_lower("ranlib"))
 -    return ranlib_main(argc, argv);
 -
 -  if (Stem.contains_lower("lib"))
 -    return libDriverMain(makeArrayRef(argv, argc));
 -
 -  if (Stem.contains_lower("ar"))
 -    return ar_main(argc, argv);
 -  fail("Not ranlib, ar, lib or dlltool!");
 -}
 +//===-- llvm-ar.cpp - LLVM archive librarian utility ----------------------===// +// +//                     The LLVM Compiler Infrastructure +// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// +// +// Builds up (relatively) standard unix archive files (.a) containing LLVM +// bitcode or other files. +// +//===----------------------------------------------------------------------===// + +#include "llvm/ADT/StringSwitch.h" +#include "llvm/ADT/Triple.h" +#include "llvm/IR/LLVMContext.h" +#include "llvm/Object/Archive.h" +#include "llvm/Object/ArchiveWriter.h" +#include "llvm/Object/MachO.h" +#include "llvm/Object/ObjectFile.h" +#include "llvm/Support/Chrono.h" +#include "llvm/Support/CommandLine.h" +#include "llvm/Support/Errc.h" +#include "llvm/Support/FileSystem.h" +#include "llvm/Support/Format.h" +#include "llvm/Support/FormatVariadic.h" +#include "llvm/Support/InitLLVM.h" +#include "llvm/Support/LineIterator.h" +#include "llvm/Support/MemoryBuffer.h" +#include "llvm/Support/Path.h" +#include "llvm/Support/Process.h" +#include "llvm/Support/StringSaver.h" +#include "llvm/Support/TargetSelect.h" +#include "llvm/Support/ToolOutputFile.h" +#include "llvm/Support/raw_ostream.h" +#include "llvm/ToolDrivers/llvm-dlltool/DlltoolDriver.h" +#include "llvm/ToolDrivers/llvm-lib/LibDriver.h" + +#if !defined(_MSC_VER) && !defined(__MINGW32__) +#include <unistd.h> +#else +#include <io.h> +#endif + +using namespace llvm; + +// The name this program was invoked as. +static StringRef ToolName; + +// The basename of this program. +static StringRef Stem; + +const char RanlibHelp[] = R"( +OVERVIEW: LLVM Ranlib (llvm-ranlib) + +  This program generates an index to speed access to archives + +USAGE: llvm-ranlib <archive-file> + +OPTIONS: +  -help                             - Display available options +  -version                          - Display the version of this program +)"; + +const char ArHelp[] = R"( +OVERVIEW: LLVM Archiver + +USAGE: llvm-ar [options] [-]<operation>[modifiers] [relpos] <archive> [files] +       llvm-ar -M [<mri-script] + +OPTIONS: +  --format              - Archive format to create +    =default            -   default +    =gnu                -   gnu +    =darwin             -   darwin +    =bsd                -   bsd +  --plugin=<string>     - Ignored for compatibility +  --help                - Display available options +  --version             - Display the version of this program + +OPERATIONS: +  d - delete [files] from the archive +  m - move [files] in the archive +  p - print [files] found in the archive +  q - quick append [files] to the archive +  r - replace or insert [files] into the archive +  s - act as ranlib +  t - display contents of archive +  x - extract [files] from the archive + +MODIFIERS: +  [a] - put [files] after [relpos] +  [b] - put [files] before [relpos] (same as [i]) +  [c] - do not warn if archive had to be created +  [D] - use zero for timestamps and uids/gids (default) +  [i] - put [files] before [relpos] (same as [b]) +  [l] - ignored for compatibility +  [L] - add archive's contents +  [o] - preserve original dates +  [s] - create an archive index (cf. ranlib) +  [S] - do not build a symbol table +  [T] - create a thin archive +  [u] - update only [files] newer than archive contents +  [U] - use actual timestamps and uids/gids +  [v] - be verbose about actions taken +)"; + +void printHelpMessage() { +  if (Stem.contains_lower("ranlib")) +    outs() << RanlibHelp; +  else if (Stem.contains_lower("ar")) +    outs() << ArHelp; +} + +// Show the error message and exit. +LLVM_ATTRIBUTE_NORETURN static void fail(Twine Error) { +  errs() << ToolName << ": " << Error << ".\n"; +  printHelpMessage(); +  exit(1); +} + +static void failIfError(std::error_code EC, Twine Context = "") { +  if (!EC) +    return; + +  std::string ContextStr = Context.str(); +  if (ContextStr == "") +    fail(EC.message()); +  fail(Context + ": " + EC.message()); +} + +static void failIfError(Error E, Twine Context = "") { +  if (!E) +    return; + +  handleAllErrors(std::move(E), [&](const llvm::ErrorInfoBase &EIB) { +    std::string ContextStr = Context.str(); +    if (ContextStr == "") +      fail(EIB.message()); +    fail(Context + ": " + EIB.message()); +  }); +} + +static SmallVector<const char *, 256> PositionalArgs; + +static bool MRI; + +namespace { +enum Format { Default, GNU, BSD, DARWIN, Unknown }; +} + +static Format FormatType = Default; + +static std::string Options; + +// This enumeration delineates the kinds of operations on an archive +// that are permitted. +enum ArchiveOperation { +  Print,           ///< Print the contents of the archive +  Delete,          ///< Delete the specified members +  Move,            ///< Move members to end or as given by {a,b,i} modifiers +  QuickAppend,     ///< Quickly append to end of archive +  ReplaceOrInsert, ///< Replace or Insert members +  DisplayTable,    ///< Display the table of contents +  Extract,         ///< Extract files back to file system +  CreateSymTab     ///< Create a symbol table in an existing archive +}; + +// Modifiers to follow operation to vary behavior +static bool AddAfter = false;      ///< 'a' modifier +static bool AddBefore = false;     ///< 'b' modifier +static bool Create = false;        ///< 'c' modifier +static bool OriginalDates = false; ///< 'o' modifier +static bool OnlyUpdate = false;    ///< 'u' modifier +static bool Verbose = false;       ///< 'v' modifier +static bool Symtab = true;         ///< 's' modifier +static bool Deterministic = true;  ///< 'D' and 'U' modifiers +static bool Thin = false;          ///< 'T' modifier +static bool AddLibrary = false;    ///< 'L' modifier + +// Relative Positional Argument (for insert/move). This variable holds +// the name of the archive member to which the 'a', 'b' or 'i' modifier +// refers. Only one of 'a', 'b' or 'i' can be specified so we only need +// one variable. +static std::string RelPos; + +// This variable holds the name of the archive file as given on the +// command line. +static std::string ArchiveName; + +// This variable holds the list of member files to proecess, as given +// on the command line. +static std::vector<StringRef> Members; + +// Extract the member filename from the command line for the [relpos] argument +// associated with a, b, and i modifiers +static void getRelPos() { +  if (PositionalArgs.size() == 0) +    fail("Expected [relpos] for a, b, or i modifier"); +  RelPos = PositionalArgs[0]; +  PositionalArgs.erase(PositionalArgs.begin()); +} + +// Get the archive file name from the command line +static void getArchive() { +  if (PositionalArgs.size() == 0) +    fail("An archive name must be specified"); +  ArchiveName = PositionalArgs[0]; +  PositionalArgs.erase(PositionalArgs.begin()); +} + +// Copy over remaining items in PositionalArgs to our Members vector +static void getMembers() { +  for (auto &Arg : PositionalArgs) +    Members.push_back(Arg); +} + +std::vector<std::unique_ptr<MemoryBuffer>> ArchiveBuffers; +std::vector<std::unique_ptr<object::Archive>> Archives; + +static object::Archive &readLibrary(const Twine &Library) { +  auto BufOrErr = MemoryBuffer::getFile(Library, -1, false); +  failIfError(BufOrErr.getError(), "Could not open library"); +  ArchiveBuffers.push_back(std::move(*BufOrErr)); +  auto LibOrErr = +      object::Archive::create(ArchiveBuffers.back()->getMemBufferRef()); +  failIfError(errorToErrorCode(LibOrErr.takeError()), +              "Could not parse library"); +  Archives.push_back(std::move(*LibOrErr)); +  return *Archives.back(); +} + +static void runMRIScript(); + +// Parse the command line options as presented and return the operation +// specified. Process all modifiers and check to make sure that constraints on +// modifier/operation pairs have not been violated. +static ArchiveOperation parseCommandLine() { +  if (MRI) { +    if (!PositionalArgs.empty() || !Options.empty()) +      fail("Cannot mix -M and other options"); +    runMRIScript(); +  } + +  // Keep track of number of operations. We can only specify one +  // per execution. +  unsigned NumOperations = 0; + +  // Keep track of the number of positional modifiers (a,b,i). Only +  // one can be specified. +  unsigned NumPositional = 0; + +  // Keep track of which operation was requested +  ArchiveOperation Operation; + +  bool MaybeJustCreateSymTab = false; + +  for (unsigned i = 0; i < Options.size(); ++i) { +    switch (Options[i]) { +    case 'd': +      ++NumOperations; +      Operation = Delete; +      break; +    case 'm': +      ++NumOperations; +      Operation = Move; +      break; +    case 'p': +      ++NumOperations; +      Operation = Print; +      break; +    case 'q': +      ++NumOperations; +      Operation = QuickAppend; +      break; +    case 'r': +      ++NumOperations; +      Operation = ReplaceOrInsert; +      break; +    case 't': +      ++NumOperations; +      Operation = DisplayTable; +      break; +    case 'x': +      ++NumOperations; +      Operation = Extract; +      break; +    case 'c': +      Create = true; +      break; +    case 'l': /* accepted but unused */ +      break; +    case 'o': +      OriginalDates = true; +      break; +    case 's': +      Symtab = true; +      MaybeJustCreateSymTab = true; +      break; +    case 'S': +      Symtab = false; +      break; +    case 'u': +      OnlyUpdate = true; +      break; +    case 'v': +      Verbose = true; +      break; +    case 'a': +      getRelPos(); +      AddAfter = true; +      NumPositional++; +      break; +    case 'b': +      getRelPos(); +      AddBefore = true; +      NumPositional++; +      break; +    case 'i': +      getRelPos(); +      AddBefore = true; +      NumPositional++; +      break; +    case 'D': +      Deterministic = true; +      break; +    case 'U': +      Deterministic = false; +      break; +    case 'T': +      Thin = true; +      break; +    case 'L': +      AddLibrary = true; +      break; +    default: +      fail(std::string("unknown option ") + Options[i]); +    } +  } + +  // At this point, the next thing on the command line must be +  // the archive name. +  getArchive(); + +  // Everything on the command line at this point is a member. +  getMembers(); + +  if (NumOperations == 0 && MaybeJustCreateSymTab) { +    NumOperations = 1; +    Operation = CreateSymTab; +    if (!Members.empty()) +      fail("The s operation takes only an archive as argument"); +  } + +  // Perform various checks on the operation/modifier specification +  // to make sure we are dealing with a legal request. +  if (NumOperations == 0) +    fail("You must specify at least one of the operations"); +  if (NumOperations > 1) +    fail("Only one operation may be specified"); +  if (NumPositional > 1) +    fail("You may only specify one of a, b, and i modifiers"); +  if (AddAfter || AddBefore) { +    if (Operation != Move && Operation != ReplaceOrInsert) +      fail("The 'a', 'b' and 'i' modifiers can only be specified with " +           "the 'm' or 'r' operations"); +  } +  if (OriginalDates && Operation != Extract) +    fail("The 'o' modifier is only applicable to the 'x' operation"); +  if (OnlyUpdate && Operation != ReplaceOrInsert) +    fail("The 'u' modifier is only applicable to the 'r' operation"); +  if (AddLibrary && Operation != QuickAppend) +    fail("The 'L' modifier is only applicable to the 'q' operation"); + +  // Return the parsed operation to the caller +  return Operation; +} + +// Implements the 'p' operation. This function traverses the archive +// looking for members that match the path list. +static void doPrint(StringRef Name, const object::Archive::Child &C) { +  if (Verbose) +    outs() << "Printing " << Name << "\n"; + +  Expected<StringRef> DataOrErr = C.getBuffer(); +  failIfError(DataOrErr.takeError()); +  StringRef Data = *DataOrErr; +  outs().write(Data.data(), Data.size()); +} + +// Utility function for printing out the file mode when the 't' operation is in +// verbose mode. +static void printMode(unsigned mode) { +  outs() << ((mode & 004) ? "r" : "-"); +  outs() << ((mode & 002) ? "w" : "-"); +  outs() << ((mode & 001) ? "x" : "-"); +} + +// Implement the 't' operation. This function prints out just +// the file names of each of the members. However, if verbose mode is requested +// ('v' modifier) then the file type, permission mode, user, group, size, and +// modification time are also printed. +static void doDisplayTable(StringRef Name, const object::Archive::Child &C) { +  if (Verbose) { +    Expected<sys::fs::perms> ModeOrErr = C.getAccessMode(); +    failIfError(ModeOrErr.takeError()); +    sys::fs::perms Mode = ModeOrErr.get(); +    printMode((Mode >> 6) & 007); +    printMode((Mode >> 3) & 007); +    printMode(Mode & 007); +    Expected<unsigned> UIDOrErr = C.getUID(); +    failIfError(UIDOrErr.takeError()); +    outs() << ' ' << UIDOrErr.get(); +    Expected<unsigned> GIDOrErr = C.getGID(); +    failIfError(GIDOrErr.takeError()); +    outs() << '/' << GIDOrErr.get(); +    Expected<uint64_t> Size = C.getSize(); +    failIfError(Size.takeError()); +    outs() << ' ' << format("%6llu", Size.get()); +    auto ModTimeOrErr = C.getLastModified(); +    failIfError(ModTimeOrErr.takeError()); +    // Note: formatv() only handles the default TimePoint<>, which is in +    // nanoseconds. +    // TODO: fix format_provider<TimePoint<>> to allow other units. +    sys::TimePoint<> ModTimeInNs = ModTimeOrErr.get(); +    outs() << ' ' << formatv("{0:%b %e %H:%M %Y}", ModTimeInNs); +    outs() << ' '; +  } + +  if (C.getParent()->isThin()) { +    outs() << sys::path::parent_path(ArchiveName); +    outs() << '/'; +  } +  outs() << Name << "\n"; +} + +// Implement the 'x' operation. This function extracts files back to the file +// system. +static void doExtract(StringRef Name, const object::Archive::Child &C) { +  // Retain the original mode. +  Expected<sys::fs::perms> ModeOrErr = C.getAccessMode(); +  failIfError(ModeOrErr.takeError()); +  sys::fs::perms Mode = ModeOrErr.get(); + +  int FD; +  failIfError(sys::fs::openFileForWrite(sys::path::filename(Name), FD, +                                        sys::fs::CD_CreateAlways, +                                        sys::fs::F_None, Mode), +              Name); + +  { +    raw_fd_ostream file(FD, false); + +    // Get the data and its length +    Expected<StringRef> BufOrErr = C.getBuffer(); +    failIfError(BufOrErr.takeError()); +    StringRef Data = BufOrErr.get(); + +    // Write the data. +    file.write(Data.data(), Data.size()); +  } + +  // If we're supposed to retain the original modification times, etc. do so +  // now. +  if (OriginalDates) { +    auto ModTimeOrErr = C.getLastModified(); +    failIfError(ModTimeOrErr.takeError()); +    failIfError( +        sys::fs::setLastAccessAndModificationTime(FD, ModTimeOrErr.get())); +  } + +  if (close(FD)) +    fail("Could not close the file"); +} + +static bool shouldCreateArchive(ArchiveOperation Op) { +  switch (Op) { +  case Print: +  case Delete: +  case Move: +  case DisplayTable: +  case Extract: +  case CreateSymTab: +    return false; + +  case QuickAppend: +  case ReplaceOrInsert: +    return true; +  } + +  llvm_unreachable("Missing entry in covered switch."); +} + +static void performReadOperation(ArchiveOperation Operation, +                                 object::Archive *OldArchive) { +  if (Operation == Extract && OldArchive->isThin()) +    fail("extracting from a thin archive is not supported"); + +  bool Filter = !Members.empty(); +  { +    Error Err = Error::success(); +    for (auto &C : OldArchive->children(Err)) { +      Expected<StringRef> NameOrErr = C.getName(); +      failIfError(NameOrErr.takeError()); +      StringRef Name = NameOrErr.get(); + +      if (Filter) { +        auto I = find(Members, Name); +        if (I == Members.end()) +          continue; +        Members.erase(I); +      } + +      switch (Operation) { +      default: +        llvm_unreachable("Not a read operation"); +      case Print: +        doPrint(Name, C); +        break; +      case DisplayTable: +        doDisplayTable(Name, C); +        break; +      case Extract: +        doExtract(Name, C); +        break; +      } +    } +    failIfError(std::move(Err)); +  } + +  if (Members.empty()) +    return; +  for (StringRef Name : Members) +    errs() << Name << " was not found\n"; +  exit(1); +} + +static void addMember(std::vector<NewArchiveMember> &Members, +                      StringRef FileName, int Pos = -1) { +  Expected<NewArchiveMember> NMOrErr = +      NewArchiveMember::getFile(FileName, Deterministic); +  failIfError(NMOrErr.takeError(), FileName); + +  // Use the basename of the object path for the member name. +  NMOrErr->MemberName = sys::path::filename(NMOrErr->MemberName); + +  if (Pos == -1) +    Members.push_back(std::move(*NMOrErr)); +  else +    Members[Pos] = std::move(*NMOrErr); +} + +static void addMember(std::vector<NewArchiveMember> &Members, +                      const object::Archive::Child &M, int Pos = -1) { +  if (Thin && !M.getParent()->isThin()) +    fail("Cannot convert a regular archive to a thin one"); +  Expected<NewArchiveMember> NMOrErr = +      NewArchiveMember::getOldMember(M, Deterministic); +  failIfError(NMOrErr.takeError()); +  if (Pos == -1) +    Members.push_back(std::move(*NMOrErr)); +  else +    Members[Pos] = std::move(*NMOrErr); +} + +static void addLibMember(std::vector<NewArchiveMember> &Members, +                         StringRef FileName) { +  Expected<NewArchiveMember> NMOrErr = +      NewArchiveMember::getFile(FileName, Deterministic); +  failIfError(NMOrErr.takeError(), FileName); +  if (identify_magic(NMOrErr->Buf->getBuffer()) == file_magic::archive) { +    object::Archive &Lib = readLibrary(FileName); +    Error Err = Error::success(); + +    for (auto &Child : Lib.children(Err)) +      addMember(Members, Child); + +    failIfError(std::move(Err)); +  } else { +    // Use the basename of the object path for the member name. +    NMOrErr->MemberName = sys::path::filename(NMOrErr->MemberName); +    Members.push_back(std::move(*NMOrErr)); +  } +} + +enum InsertAction { +  IA_AddOldMember, +  IA_AddNewMember, +  IA_Delete, +  IA_MoveOldMember, +  IA_MoveNewMember +}; + +static InsertAction computeInsertAction(ArchiveOperation Operation, +                                        const object::Archive::Child &Member, +                                        StringRef Name, +                                        std::vector<StringRef>::iterator &Pos) { +  if (Operation == QuickAppend || Members.empty()) +    return IA_AddOldMember; + +  auto MI = find_if(Members, [Name](StringRef Path) { +    return Name == sys::path::filename(Path); +  }); + +  if (MI == Members.end()) +    return IA_AddOldMember; + +  Pos = MI; + +  if (Operation == Delete) +    return IA_Delete; + +  if (Operation == Move) +    return IA_MoveOldMember; + +  if (Operation == ReplaceOrInsert) { +    StringRef PosName = sys::path::filename(RelPos); +    if (!OnlyUpdate) { +      if (PosName.empty()) +        return IA_AddNewMember; +      return IA_MoveNewMember; +    } + +    // We could try to optimize this to a fstat, but it is not a common +    // operation. +    sys::fs::file_status Status; +    failIfError(sys::fs::status(*MI, Status), *MI); +    auto ModTimeOrErr = Member.getLastModified(); +    failIfError(ModTimeOrErr.takeError()); +    if (Status.getLastModificationTime() < ModTimeOrErr.get()) { +      if (PosName.empty()) +        return IA_AddOldMember; +      return IA_MoveOldMember; +    } + +    if (PosName.empty()) +      return IA_AddNewMember; +    return IA_MoveNewMember; +  } +  llvm_unreachable("No such operation"); +} + +// We have to walk this twice and computing it is not trivial, so creating an +// explicit std::vector is actually fairly efficient. +static std::vector<NewArchiveMember> +computeNewArchiveMembers(ArchiveOperation Operation, +                         object::Archive *OldArchive) { +  std::vector<NewArchiveMember> Ret; +  std::vector<NewArchiveMember> Moved; +  int InsertPos = -1; +  StringRef PosName = sys::path::filename(RelPos); +  if (OldArchive) { +    Error Err = Error::success(); +    for (auto &Child : OldArchive->children(Err)) { +      int Pos = Ret.size(); +      Expected<StringRef> NameOrErr = Child.getName(); +      failIfError(NameOrErr.takeError()); +      StringRef Name = NameOrErr.get(); +      if (Name == PosName) { +        assert(AddAfter || AddBefore); +        if (AddBefore) +          InsertPos = Pos; +        else +          InsertPos = Pos + 1; +      } + +      std::vector<StringRef>::iterator MemberI = Members.end(); +      InsertAction Action = +          computeInsertAction(Operation, Child, Name, MemberI); +      switch (Action) { +      case IA_AddOldMember: +        addMember(Ret, Child); +        break; +      case IA_AddNewMember: +        addMember(Ret, *MemberI); +        break; +      case IA_Delete: +        break; +      case IA_MoveOldMember: +        addMember(Moved, Child); +        break; +      case IA_MoveNewMember: +        addMember(Moved, *MemberI); +        break; +      } +      if (MemberI != Members.end()) +        Members.erase(MemberI); +    } +    failIfError(std::move(Err)); +  } + +  if (Operation == Delete) +    return Ret; + +  if (!RelPos.empty() && InsertPos == -1) +    fail("Insertion point not found"); + +  if (RelPos.empty()) +    InsertPos = Ret.size(); + +  assert(unsigned(InsertPos) <= Ret.size()); +  int Pos = InsertPos; +  for (auto &M : Moved) { +    Ret.insert(Ret.begin() + Pos, std::move(M)); +    ++Pos; +  } + +  if (AddLibrary) { +    assert(Operation == QuickAppend); +    for (auto &Member : Members) +      addLibMember(Ret, Member); +    return Ret; +  } + +  for (unsigned I = 0; I != Members.size(); ++I) +    Ret.insert(Ret.begin() + InsertPos, NewArchiveMember()); +  Pos = InsertPos; +  for (auto &Member : Members) { +    addMember(Ret, Member, Pos); +    ++Pos; +  } + +  return Ret; +} + +static object::Archive::Kind getDefaultForHost() { +  return Triple(sys::getProcessTriple()).isOSDarwin() +             ? object::Archive::K_DARWIN +             : object::Archive::K_GNU; +} + +static object::Archive::Kind getKindFromMember(const NewArchiveMember &Member) { +  Expected<std::unique_ptr<object::ObjectFile>> OptionalObject = +      object::ObjectFile::createObjectFile(Member.Buf->getMemBufferRef()); + +  if (OptionalObject) +    return isa<object::MachOObjectFile>(**OptionalObject) +               ? object::Archive::K_DARWIN +               : object::Archive::K_GNU; + +  // squelch the error in case we had a non-object file +  consumeError(OptionalObject.takeError()); +  return getDefaultForHost(); +} + +static void performWriteOperation(ArchiveOperation Operation, +                                  object::Archive *OldArchive, +                                  std::unique_ptr<MemoryBuffer> OldArchiveBuf, +                                  std::vector<NewArchiveMember> *NewMembersP) { +  std::vector<NewArchiveMember> NewMembers; +  if (!NewMembersP) +    NewMembers = computeNewArchiveMembers(Operation, OldArchive); + +  object::Archive::Kind Kind; +  switch (FormatType) { +  case Default: +    if (Thin) +      Kind = object::Archive::K_GNU; +    else if (OldArchive) +      Kind = OldArchive->kind(); +    else if (NewMembersP) +      Kind = NewMembersP->size() ? getKindFromMember(NewMembersP->front()) +                                 : getDefaultForHost(); +    else +      Kind = NewMembers.size() ? getKindFromMember(NewMembers.front()) +                               : getDefaultForHost(); +    break; +  case GNU: +    Kind = object::Archive::K_GNU; +    break; +  case BSD: +    if (Thin) +      fail("Only the gnu format has a thin mode"); +    Kind = object::Archive::K_BSD; +    break; +  case DARWIN: +    if (Thin) +      fail("Only the gnu format has a thin mode"); +    Kind = object::Archive::K_DARWIN; +    break; +  case Unknown: +    llvm_unreachable(""); +  } + +  Error E = +      writeArchive(ArchiveName, NewMembersP ? *NewMembersP : NewMembers, Symtab, +                   Kind, Deterministic, Thin, std::move(OldArchiveBuf)); +  failIfError(std::move(E), ArchiveName); +} + +static void createSymbolTable(object::Archive *OldArchive) { +  // When an archive is created or modified, if the s option is given, the +  // resulting archive will have a current symbol table. If the S option +  // is given, it will have no symbol table. +  // In summary, we only need to update the symbol table if we have none. +  // This is actually very common because of broken build systems that think +  // they have to run ranlib. +  if (OldArchive->hasSymbolTable()) +    return; + +  performWriteOperation(CreateSymTab, OldArchive, nullptr, nullptr); +} + +static void performOperation(ArchiveOperation Operation, +                             object::Archive *OldArchive, +                             std::unique_ptr<MemoryBuffer> OldArchiveBuf, +                             std::vector<NewArchiveMember> *NewMembers) { +  switch (Operation) { +  case Print: +  case DisplayTable: +  case Extract: +    performReadOperation(Operation, OldArchive); +    return; + +  case Delete: +  case Move: +  case QuickAppend: +  case ReplaceOrInsert: +    performWriteOperation(Operation, OldArchive, std::move(OldArchiveBuf), +                          NewMembers); +    return; +  case CreateSymTab: +    createSymbolTable(OldArchive); +    return; +  } +  llvm_unreachable("Unknown operation."); +} + +static int performOperation(ArchiveOperation Operation, +                            std::vector<NewArchiveMember> *NewMembers) { +  // Create or open the archive object. +  ErrorOr<std::unique_ptr<MemoryBuffer>> Buf = +      MemoryBuffer::getFile(ArchiveName, -1, false); +  std::error_code EC = Buf.getError(); +  if (EC && EC != errc::no_such_file_or_directory) +    fail("error opening '" + ArchiveName + "': " + EC.message() + "!"); + +  if (!EC) { +    Error Err = Error::success(); +    object::Archive Archive(Buf.get()->getMemBufferRef(), Err); +    EC = errorToErrorCode(std::move(Err)); +    failIfError(EC, +                "error loading '" + ArchiveName + "': " + EC.message() + "!"); +    performOperation(Operation, &Archive, std::move(Buf.get()), NewMembers); +    return 0; +  } + +  assert(EC == errc::no_such_file_or_directory); + +  if (!shouldCreateArchive(Operation)) { +    failIfError(EC, Twine("error loading '") + ArchiveName + "'"); +  } else { +    if (!Create) { +      // Produce a warning if we should and we're creating the archive +      errs() << ToolName << ": creating " << ArchiveName << "\n"; +    } +  } + +  performOperation(Operation, nullptr, nullptr, NewMembers); +  return 0; +} + +static void runMRIScript() { +  enum class MRICommand { AddLib, AddMod, Create, Delete, Save, End, Invalid }; + +  ErrorOr<std::unique_ptr<MemoryBuffer>> Buf = MemoryBuffer::getSTDIN(); +  failIfError(Buf.getError()); +  const MemoryBuffer &Ref = *Buf.get(); +  bool Saved = false; +  std::vector<NewArchiveMember> NewMembers; + +  for (line_iterator I(Ref, /*SkipBlanks*/ false), E; I != E; ++I) { +    StringRef Line = *I; +    Line = Line.split(';').first; +    Line = Line.split('*').first; +    Line = Line.trim(); +    if (Line.empty()) +      continue; +    StringRef CommandStr, Rest; +    std::tie(CommandStr, Rest) = Line.split(' '); +    Rest = Rest.trim(); +    if (!Rest.empty() && Rest.front() == '"' && Rest.back() == '"') +      Rest = Rest.drop_front().drop_back(); +    auto Command = StringSwitch<MRICommand>(CommandStr.lower()) +                       .Case("addlib", MRICommand::AddLib) +                       .Case("addmod", MRICommand::AddMod) +                       .Case("create", MRICommand::Create) +                       .Case("delete", MRICommand::Delete) +                       .Case("save", MRICommand::Save) +                       .Case("end", MRICommand::End) +                       .Default(MRICommand::Invalid); + +    switch (Command) { +    case MRICommand::AddLib: { +      object::Archive &Lib = readLibrary(Rest); +      { +        Error Err = Error::success(); +        for (auto &Member : Lib.children(Err)) +          addMember(NewMembers, Member); +        failIfError(std::move(Err)); +      } +      break; +    } +    case MRICommand::AddMod: +      addMember(NewMembers, Rest); +      break; +    case MRICommand::Create: +      Create = true; +      if (!ArchiveName.empty()) +        fail("Editing multiple archives not supported"); +      if (Saved) +        fail("File already saved"); +      ArchiveName = Rest; +      break; +    case MRICommand::Delete: { +      StringRef Name = sys::path::filename(Rest); +      llvm::erase_if(NewMembers, +                     [=](NewArchiveMember &M) { return M.MemberName == Name; }); +      break; +    } +    case MRICommand::Save: +      Saved = true; +      break; +    case MRICommand::End: +      break; +    case MRICommand::Invalid: +      fail("Unknown command: " + CommandStr); +    } +  } + +  // Nothing to do if not saved. +  if (Saved) +    performOperation(ReplaceOrInsert, &NewMembers); +  exit(0); +} + +static bool handleGenericOption(StringRef arg) { +  if (arg == "-help" || arg == "--help") { +    printHelpMessage(); +    return true; +  } +  if (arg == "-version" || arg == "--version") { +    cl::PrintVersionMessage(); +    return true; +  } +  return false; +} + +static int ar_main(int argc, char **argv) { +  SmallVector<const char *, 0> Argv(argv, argv + argc); +  BumpPtrAllocator Alloc; +  StringSaver Saver(Alloc); +  cl::ExpandResponseFiles(Saver, cl::TokenizeGNUCommandLine, Argv); +  for (size_t i = 1; i < Argv.size(); ++i) { +    StringRef Arg = Argv[i]; +    const char *match; +    auto MatchFlagWithArg = [&](const char *expected) { +      size_t len = strlen(expected); +      if (Arg == expected) { +        if (++i >= Argv.size()) +          fail(std::string(expected) + " requires an argument"); +        match = Argv[i]; +        return true; +      } +      if (Arg.startswith(expected) && Arg.size() > len && Arg[len] == '=') { +        match = Arg.data() + len + 1; +        return true; +      } +      return false; +    }; +    if (handleGenericOption(Argv[i])) +      return 0; +    if (Arg == "--") { +      for (; i < Argv.size(); ++i) +        PositionalArgs.push_back(Argv[i]); +      break; +    } +    if (Arg[0] == '-') { +      if (Arg.startswith("--")) +        Arg = Argv[i] + 2; +      else +        Arg = Argv[i] + 1; +      if (Arg == "M") { +        MRI = true; +      } else if (MatchFlagWithArg("format")) { +        FormatType = StringSwitch<Format>(match) +                         .Case("default", Default) +                         .Case("gnu", GNU) +                         .Case("darwin", DARWIN) +                         .Case("bsd", BSD) +                         .Default(Unknown); +        if (FormatType == Unknown) +          fail(std::string("Invalid format ") + match); +      } else if (MatchFlagWithArg("plugin")) { +        // Ignored. +      } else { +        Options += Argv[i] + 1; +      } +    } else if (Options.empty()) { +      Options += Argv[i]; +    } else { +      PositionalArgs.push_back(Argv[i]); +    } +  } +  ArchiveOperation Operation = parseCommandLine(); +  return performOperation(Operation, nullptr); +} + +static int ranlib_main(int argc, char **argv) { +  bool ArchiveSpecified = false; +  for (int i = 1; i < argc; ++i) { +    if (handleGenericOption(argv[i])) { +      return 0; +    } else { +      if (ArchiveSpecified) +        fail("Exactly one archive should be specified"); +      ArchiveSpecified = true; +      ArchiveName = argv[i]; +    } +  } +  return performOperation(CreateSymTab, nullptr); +} + +int main(int argc, char **argv) { +  InitLLVM X(argc, argv); +  ToolName = argv[0]; + +  llvm::InitializeAllTargetInfos(); +  llvm::InitializeAllTargetMCs(); +  llvm::InitializeAllAsmParsers(); + +  Stem = sys::path::stem(ToolName); +  if (Stem.contains_lower("dlltool")) +    return dlltoolDriverMain(makeArrayRef(argv, argc)); + +  if (Stem.contains_lower("ranlib")) +    return ranlib_main(argc, argv); + +  if (Stem.contains_lower("lib")) +    return libDriverMain(makeArrayRef(argv, argc)); + +  if (Stem.contains_lower("ar")) +    return ar_main(argc, argv); +  fail("Not ranlib, ar, lib or dlltool!"); +} | 

