diff options
Diffstat (limited to 'llvm/tools')
| -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!"); +} |
