diff options
| -rw-r--r-- | llvm/test/tools/llvm-ar/Inputs/add-lib1.yaml | 30 | ||||
| -rw-r--r-- | llvm/test/tools/llvm-ar/Inputs/add-lib2.yaml | 30 | ||||
| -rw-r--r-- | llvm/test/tools/llvm-ar/Inputs/add-lib3.yaml | 30 | ||||
| -rw-r--r-- | llvm/test/tools/llvm-ar/add-library.test | 43 | ||||
| -rw-r--r-- | llvm/tools/llvm-ar/llvm-ar.cpp | 1997 |
5 files changed, 1151 insertions, 979 deletions
diff --git a/llvm/test/tools/llvm-ar/Inputs/add-lib1.yaml b/llvm/test/tools/llvm-ar/Inputs/add-lib1.yaml new file mode 100644 index 00000000000..7ae9fd95a62 --- /dev/null +++ b/llvm/test/tools/llvm-ar/Inputs/add-lib1.yaml @@ -0,0 +1,30 @@ +--- !ELF +FileHeader: + Class: ELFCLASS64 + Data: ELFDATA2LSB + Type: ET_REL + Machine: EM_X86_64 +Sections: + - Name: .text + Type: SHT_PROGBITS + Flags: [ SHF_ALLOC, SHF_EXECINSTR ] + AddressAlign: 0x0000000000000004 + Content: '' + - Name: .comment + Type: SHT_PROGBITS + Flags: [ SHF_MERGE, SHF_STRINGS ] + AddressAlign: 0x0000000000000001 + Content: 00636C616E672076657273696F6E20332E392E3020287472756E6B203237333632342920286C6C766D2F7472756E6B203237333633362900 + - Name: .note.GNU-stack + Type: SHT_PROGBITS + AddressAlign: 0x0000000000000001 + Content: '' +Symbols: + Global: + - Name: lib1 + Index: SHN_ABS + Value: 0x1234 + Local: + - Name: '-' + Type: STT_FILE +... diff --git a/llvm/test/tools/llvm-ar/Inputs/add-lib2.yaml b/llvm/test/tools/llvm-ar/Inputs/add-lib2.yaml new file mode 100644 index 00000000000..8d224b95a4d --- /dev/null +++ b/llvm/test/tools/llvm-ar/Inputs/add-lib2.yaml @@ -0,0 +1,30 @@ +--- !ELF +FileHeader: + Class: ELFCLASS64 + Data: ELFDATA2LSB + Type: ET_REL + Machine: EM_X86_64 +Sections: + - Name: .text + Type: SHT_PROGBITS + Flags: [ SHF_ALLOC, SHF_EXECINSTR ] + AddressAlign: 0x0000000000000004 + Content: '' + - Name: .comment + Type: SHT_PROGBITS + Flags: [ SHF_MERGE, SHF_STRINGS ] + AddressAlign: 0x0000000000000001 + Content: 00636C616E672076657273696F6E20332E392E3020287472756E6B203237333632342920286C6C766D2F7472756E6B203237333633362900 + - Name: .note.GNU-stack + Type: SHT_PROGBITS + AddressAlign: 0x0000000000000001 + Content: '' +Symbols: + Global: + - Name: lib2 + Index: SHN_ABS + Value: 0x1234 + Local: + - Name: '-' + Type: STT_FILE +... diff --git a/llvm/test/tools/llvm-ar/Inputs/add-lib3.yaml b/llvm/test/tools/llvm-ar/Inputs/add-lib3.yaml new file mode 100644 index 00000000000..0f1cfe7d806 --- /dev/null +++ b/llvm/test/tools/llvm-ar/Inputs/add-lib3.yaml @@ -0,0 +1,30 @@ +--- !ELF +FileHeader: + Class: ELFCLASS64 + Data: ELFDATA2LSB + Type: ET_REL + Machine: EM_X86_64 +Sections: + - Name: .text + Type: SHT_PROGBITS + Flags: [ SHF_ALLOC, SHF_EXECINSTR ] + AddressAlign: 0x0000000000000004 + Content: '' + - Name: .comment + Type: SHT_PROGBITS + Flags: [ SHF_MERGE, SHF_STRINGS ] + AddressAlign: 0x0000000000000001 + Content: 00636C616E672076657273696F6E20332E392E3020287472756E6B203237333632342920286C6C766D2F7472756E6B203237333633362900 + - Name: .note.GNU-stack + Type: SHT_PROGBITS + AddressAlign: 0x0000000000000001 + Content: '' +Symbols: + Global: + - Name: lib3 + Index: SHN_ABS + Value: 0x1234 + Local: + - Name: '-' + Type: STT_FILE +... diff --git a/llvm/test/tools/llvm-ar/add-library.test b/llvm/test/tools/llvm-ar/add-library.test new file mode 100644 index 00000000000..bd44a7e9a4a --- /dev/null +++ b/llvm/test/tools/llvm-ar/add-library.test @@ -0,0 +1,43 @@ +RUN: yaml2obj %S/Inputs/add-lib1.yaml -o %t-add-lib1.o +RUN: yaml2obj %S/Inputs/add-lib2.yaml -o %t-add-lib2.o +RUN: yaml2obj %S/Inputs/add-lib2.yaml -o %t-add-lib3.o + +RUN: rm -f %t.ar +RUN: llvm-ar crs %t.ar %t-add-lib1.o +RUN: llvm-ar cqs %t.ar %t-add-lib2.o + +RUN: llvm-ar tv %t.ar | FileCheck %s --check-prefix=CHECK-NAMES-NO-ADDLIB +CHECK-NAMES-NO-ADDLIB: add-library.test.tmp-add-lib1.o +CHECK-NAMES-NO-ADDLIB: add-library.test.tmp-add-lib2.o + +RUN: llvm-nm %t.ar | FileCheck %s --check-prefix=CHECK-SYMBOLS-NO-ADDLIB +CHECK-SYMBOLS-NO-ADDLIB: add-lib1 +CHECK-SYMBOLS-NO-ADDLIB: add-lib2 + +RUN: rm -f %t1.ar +RUN: llvm-ar crs %t1.ar %t-add-lib3.o +RUN: llvm-ar cqLs %t1.ar %t.ar + +RUN: llvm-ar tv %t1.ar | FileCheck %s --check-prefix=CHECK-NAMES-ADDLIB +CHECK-NAMES-ADDLIB: add-library.test.tmp-add-lib3.o +CHECK-NAMES-ADDLIB: add-library.test.tmp-add-lib1.o +CHECK-NAMES-ADDLIB: add-library.test.tmp-add-lib2.o + +RUN: llvm-nm %t1.ar | FileCheck %s --check-prefix=CHECK-SYMBOLS-ADDLIB +CHECK-SYMBOLS-ADDLIB: add-lib3 +CHECK-SYMBOLS-ADDLIB: add-lib1 +CHECK-SYMBOLS-ADDLIB: add-lib2 + +RUN: llvm-ar cqLs %t1.ar %t-add-lib1.o + +RUN: llvm-ar tv %t1.ar | FileCheck %s --check-prefix=CHECK-NAMES-DUPLICATE +CHECK-NAMES-DUPLICATE: add-library.test.tmp-add-lib3.o +CHECK-NAMES-DUPLICATE: add-library.test.tmp-add-lib1.o +CHECK-NAMES-DUPLICATE: add-library.test.tmp-add-lib2.o +CHECK-NAMES-DUPLICATE: add-library.test.tmp-add-lib1.o + +RUN: llvm-nm %t1.ar | FileCheck %s --check-prefix=CHECK-SYMBOLS-DUPLICATE +CHECK-SYMBOLS-DUPLICATE: add-lib3 +CHECK-SYMBOLS-DUPLICATE: add-lib1 +CHECK-SYMBOLS-DUPLICATE: add-lib2 +CHECK-SYMBOLS-DUPLICATE: add-lib1 diff --git a/llvm/tools/llvm-ar/llvm-ar.cpp b/llvm/tools/llvm-ar/llvm-ar.cpp index 454b3971d28..2c6dc8fad92 100644 --- a/llvm/tools/llvm-ar/llvm-ar.cpp +++ b/llvm/tools/llvm-ar/llvm-ar.cpp @@ -1,979 +1,1018 @@ -//===-- 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 - [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 - -// 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); -} - -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; - 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"); - - // 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); -} - -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; - } - - 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; - std::vector<std::unique_ptr<MemoryBuffer>> ArchiveBuffers; - std::vector<std::unique_ptr<object::Archive>> Archives; - - 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: { - auto BufOrErr = MemoryBuffer::getFile(Rest, -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)); - object::Archive &Lib = *Archives.back(); - { - 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!");
+}
|
