xref: /freebsd/contrib/llvm-project/llvm/tools/llvm-ar/llvm-ar.cpp (revision 0fca6ea1d4eea4c934cfff25ac9ee8ad6fe95583)
1 //===-- llvm-ar.cpp - LLVM archive librarian utility ----------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // Builds up (relatively) standard unix archive files (.a) containing LLVM
10 // bitcode or other files.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #include "llvm/ADT/StringExtras.h"
15 #include "llvm/ADT/StringSwitch.h"
16 #include "llvm/BinaryFormat/Magic.h"
17 #include "llvm/IR/LLVMContext.h"
18 #include "llvm/Object/Archive.h"
19 #include "llvm/Object/ArchiveWriter.h"
20 #include "llvm/Object/SymbolicFile.h"
21 #include "llvm/Support/Chrono.h"
22 #include "llvm/Support/CommandLine.h"
23 #include "llvm/Support/ConvertUTF.h"
24 #include "llvm/Support/Errc.h"
25 #include "llvm/Support/FileSystem.h"
26 #include "llvm/Support/Format.h"
27 #include "llvm/Support/FormatVariadic.h"
28 #include "llvm/Support/LLVMDriver.h"
29 #include "llvm/Support/LineIterator.h"
30 #include "llvm/Support/MemoryBuffer.h"
31 #include "llvm/Support/Path.h"
32 #include "llvm/Support/Process.h"
33 #include "llvm/Support/StringSaver.h"
34 #include "llvm/Support/TargetSelect.h"
35 #include "llvm/Support/ToolOutputFile.h"
36 #include "llvm/Support/WithColor.h"
37 #include "llvm/Support/raw_ostream.h"
38 #include "llvm/TargetParser/Host.h"
39 #include "llvm/TargetParser/Triple.h"
40 #include "llvm/ToolDrivers/llvm-dlltool/DlltoolDriver.h"
41 #include "llvm/ToolDrivers/llvm-lib/LibDriver.h"
42 
43 #if !defined(_MSC_VER) && !defined(__MINGW32__)
44 #include <unistd.h>
45 #else
46 #include <io.h>
47 #endif
48 
49 #ifdef _WIN32
50 #include "llvm/Support/Windows/WindowsSupport.h"
51 #endif
52 
53 using namespace llvm;
54 using namespace llvm::object;
55 
56 // The name this program was invoked as.
57 static StringRef ToolName;
58 
59 // The basename of this program.
60 static StringRef Stem;
61 
printRanLibHelp(StringRef ToolName)62 static void printRanLibHelp(StringRef ToolName) {
63   outs() << "OVERVIEW: LLVM ranlib\n\n"
64          << "Generate an index for archives\n\n"
65          << "USAGE: " + ToolName + " archive...\n\n"
66          << "OPTIONS:\n"
67          << "  -h --help             - Display available options\n"
68          << "  -V --version          - Display the version of this program\n"
69          << "  -D                    - Use zero for timestamps and uids/gids "
70             "(default)\n"
71          << "  -U                    - Use actual timestamps and uids/gids\n"
72          << "  -X{32|64|32_64|any}   - Specify which archive symbol tables "
73             "should be generated if they do not already exist (AIX OS only)\n";
74 }
75 
printArHelp(StringRef ToolName)76 static void printArHelp(StringRef ToolName) {
77   const char ArOptions[] =
78       R"(OPTIONS:
79   --format              - archive format to create
80     =default            -   default
81     =gnu                -   gnu
82     =darwin             -   darwin
83     =bsd                -   bsd
84     =bigarchive         -   big archive (AIX OS)
85     =coff               -   coff
86   --plugin=<string>     - ignored for compatibility
87   -h --help             - display this help and exit
88   --output              - the directory to extract archive members to
89   --rsp-quoting         - quoting style for response files
90     =posix              -   posix
91     =windows            -   windows
92   --thin                - create a thin archive
93   --version             - print the version and exit
94   -X{32|64|32_64|any}   - object mode (only for AIX OS)
95   @<file>               - read options from <file>
96 
97 OPERATIONS:
98   d - delete [files] from the archive
99   m - move [files] in the archive
100   p - print contents of [files] found in the archive
101   q - quick append [files] to the archive
102   r - replace or insert [files] into the archive
103   s - act as ranlib
104   t - display list of files in archive
105   x - extract [files] from the archive
106 
107 MODIFIERS:
108   [a] - put [files] after [relpos]
109   [b] - put [files] before [relpos] (same as [i])
110   [c] - do not warn if archive had to be created
111   [D] - use zero for timestamps and uids/gids (default)
112   [h] - display this help and exit
113   [i] - put [files] before [relpos] (same as [b])
114   [l] - ignored for compatibility
115   [L] - add archive's contents
116   [N] - use instance [count] of name
117   [o] - preserve original dates
118   [O] - display member offsets
119   [P] - use full names when matching (implied for thin archives)
120   [s] - create an archive index (cf. ranlib)
121   [S] - do not build a symbol table
122   [T] - deprecated, use --thin instead
123   [u] - update only [files] newer than archive contents
124   [U] - use actual timestamps and uids/gids
125   [v] - be verbose about actions taken
126   [V] - display the version and exit
127 )";
128 
129   outs() << "OVERVIEW: LLVM Archiver\n\n"
130          << "USAGE: " + ToolName +
131                 " [options] [-]<operation>[modifiers] [relpos] "
132                 "[count] <archive> [files]\n"
133          << "       " + ToolName + " -M [<mri-script]\n\n";
134 
135   outs() << ArOptions;
136 }
137 
printHelpMessage()138 static void printHelpMessage() {
139   if (Stem.contains_insensitive("ranlib"))
140     printRanLibHelp(Stem);
141   else if (Stem.contains_insensitive("ar"))
142     printArHelp(Stem);
143 }
144 
145 static unsigned MRILineNumber;
146 static bool ParsingMRIScript;
147 
148 // Show the error plus the usage message, and exit.
badUsage(Twine Error)149 [[noreturn]] static void badUsage(Twine Error) {
150   WithColor::error(errs(), ToolName) << Error << "\n";
151   printHelpMessage();
152   exit(1);
153 }
154 
155 // Show the error message and exit.
fail(Twine Error)156 [[noreturn]] static void fail(Twine Error) {
157   if (ParsingMRIScript) {
158     WithColor::error(errs(), ToolName)
159         << "script line " << MRILineNumber << ": " << Error << "\n";
160   } else {
161     WithColor::error(errs(), ToolName) << Error << "\n";
162   }
163   exit(1);
164 }
165 
failIfError(std::error_code EC,Twine Context="")166 static void failIfError(std::error_code EC, Twine Context = "") {
167   if (!EC)
168     return;
169 
170   std::string ContextStr = Context.str();
171   if (ContextStr.empty())
172     fail(EC.message());
173   fail(Context + ": " + EC.message());
174 }
175 
failIfError(Error E,Twine Context="")176 static void failIfError(Error E, Twine Context = "") {
177   if (!E)
178     return;
179 
180   handleAllErrors(std::move(E), [&](const llvm::ErrorInfoBase &EIB) {
181     std::string ContextStr = Context.str();
182     if (ContextStr.empty())
183       fail(EIB.message());
184     fail(Context + ": " + EIB.message());
185   });
186 }
187 
warn(Twine Message)188 static void warn(Twine Message) {
189   WithColor::warning(errs(), ToolName) << Message << "\n";
190 }
191 
192 static SmallVector<const char *, 256> PositionalArgs;
193 
194 static bool MRI;
195 
196 namespace {
197 enum Format { Default, GNU, COFF, BSD, DARWIN, BIGARCHIVE, Unknown };
198 }
199 
200 static Format FormatType = Default;
201 
202 static std::string Options;
203 
204 // This enumeration delineates the kinds of operations on an archive
205 // that are permitted.
206 enum ArchiveOperation {
207   Print,           ///< Print the contents of the archive
208   Delete,          ///< Delete the specified members
209   Move,            ///< Move members to end or as given by {a,b,i} modifiers
210   QuickAppend,     ///< Quickly append to end of archive
211   ReplaceOrInsert, ///< Replace or Insert members
212   DisplayTable,    ///< Display the table of contents
213   Extract,         ///< Extract files back to file system
214   CreateSymTab     ///< Create a symbol table in an existing archive
215 };
216 
217 enum class BitModeTy { Bit32, Bit64, Bit32_64, Any, Unknown };
218 
219 static BitModeTy BitMode = BitModeTy::Bit32;
220 
221 // Modifiers to follow operation to vary behavior
222 static bool AddAfter = false;             ///< 'a' modifier
223 static bool AddBefore = false;            ///< 'b' modifier
224 static bool Create = false;               ///< 'c' modifier
225 static bool OriginalDates = false;        ///< 'o' modifier
226 static bool DisplayMemberOffsets = false; ///< 'O' modifier
227 static bool CompareFullPath = false;      ///< 'P' modifier
228 static bool OnlyUpdate = false;           ///< 'u' modifier
229 static bool Verbose = false;              ///< 'v' modifier
230 static SymtabWritingMode Symtab =
231     SymtabWritingMode::NormalSymtab;      ///< 's' modifier
232 static bool Deterministic = true;         ///< 'D' and 'U' modifiers
233 static bool Thin = false;                 ///< 'T' modifier
234 static bool AddLibrary = false;           ///< 'L' modifier
235 
236 // Relative Positional Argument (for insert/move). This variable holds
237 // the name of the archive member to which the 'a', 'b' or 'i' modifier
238 // refers. Only one of 'a', 'b' or 'i' can be specified so we only need
239 // one variable.
240 static std::string RelPos;
241 
242 // Count parameter for 'N' modifier. This variable specifies which file should
243 // match for extract/delete operations when there are multiple matches. This is
244 // 1-indexed. A value of 0 is invalid, and implies 'N' is not used.
245 static int CountParam = 0;
246 
247 // This variable holds the name of the archive file as given on the
248 // command line.
249 static std::string ArchiveName;
250 
251 // Output directory specified by --output.
252 static std::string OutputDir;
253 
254 static std::vector<std::unique_ptr<MemoryBuffer>> ArchiveBuffers;
255 static std::vector<std::unique_ptr<object::Archive>> Archives;
256 
257 // This variable holds the list of member files to proecess, as given
258 // on the command line.
259 static std::vector<StringRef> Members;
260 
261 // Static buffer to hold StringRefs.
262 static BumpPtrAllocator Alloc;
263 
264 // Extract the member filename from the command line for the [relpos] argument
265 // associated with a, b, and i modifiers
getRelPos()266 static void getRelPos() {
267   if (PositionalArgs.empty())
268     fail("expected [relpos] for 'a', 'b', or 'i' modifier");
269   RelPos = PositionalArgs[0];
270   PositionalArgs.erase(PositionalArgs.begin());
271 }
272 
273 // Extract the parameter from the command line for the [count] argument
274 // associated with the N modifier
getCountParam()275 static void getCountParam() {
276   if (PositionalArgs.empty())
277     badUsage("expected [count] for 'N' modifier");
278   auto CountParamArg = StringRef(PositionalArgs[0]);
279   if (CountParamArg.getAsInteger(10, CountParam))
280     badUsage("value for [count] must be numeric, got: " + CountParamArg);
281   if (CountParam < 1)
282     badUsage("value for [count] must be positive, got: " + CountParamArg);
283   PositionalArgs.erase(PositionalArgs.begin());
284 }
285 
286 // Get the archive file name from the command line
getArchive()287 static void getArchive() {
288   if (PositionalArgs.empty())
289     badUsage("an archive name must be specified");
290   ArchiveName = PositionalArgs[0];
291   PositionalArgs.erase(PositionalArgs.begin());
292 }
293 
readLibrary(const Twine & Library)294 static object::Archive &readLibrary(const Twine &Library) {
295   auto BufOrErr = MemoryBuffer::getFile(Library, /*IsText=*/false,
296                                         /*RequiresNullTerminator=*/false);
297   failIfError(BufOrErr.getError(), "could not open library " + Library);
298   ArchiveBuffers.push_back(std::move(*BufOrErr));
299   auto LibOrErr =
300       object::Archive::create(ArchiveBuffers.back()->getMemBufferRef());
301   failIfError(errorToErrorCode(LibOrErr.takeError()),
302               "could not parse library");
303   Archives.push_back(std::move(*LibOrErr));
304   return *Archives.back();
305 }
306 
307 static void runMRIScript();
308 
309 // Parse the command line options as presented and return the operation
310 // specified. Process all modifiers and check to make sure that constraints on
311 // modifier/operation pairs have not been violated.
parseCommandLine()312 static ArchiveOperation parseCommandLine() {
313   if (MRI) {
314     if (!PositionalArgs.empty() || !Options.empty())
315       badUsage("cannot mix -M and other options");
316     runMRIScript();
317   }
318 
319   // Keep track of number of operations. We can only specify one
320   // per execution.
321   unsigned NumOperations = 0;
322 
323   // Keep track of the number of positional modifiers (a,b,i). Only
324   // one can be specified.
325   unsigned NumPositional = 0;
326 
327   // Keep track of which operation was requested
328   ArchiveOperation Operation;
329 
330   bool MaybeJustCreateSymTab = false;
331 
332   for (unsigned i = 0; i < Options.size(); ++i) {
333     switch (Options[i]) {
334     case 'd':
335       ++NumOperations;
336       Operation = Delete;
337       break;
338     case 'm':
339       ++NumOperations;
340       Operation = Move;
341       break;
342     case 'p':
343       ++NumOperations;
344       Operation = Print;
345       break;
346     case 'q':
347       ++NumOperations;
348       Operation = QuickAppend;
349       break;
350     case 'r':
351       ++NumOperations;
352       Operation = ReplaceOrInsert;
353       break;
354     case 't':
355       ++NumOperations;
356       Operation = DisplayTable;
357       break;
358     case 'x':
359       ++NumOperations;
360       Operation = Extract;
361       break;
362     case 'c':
363       Create = true;
364       break;
365     case 'l': /* accepted but unused */
366       break;
367     case 'o':
368       OriginalDates = true;
369       break;
370     case 'O':
371       DisplayMemberOffsets = true;
372       break;
373     case 'P':
374       CompareFullPath = true;
375       break;
376     case 's':
377       Symtab = SymtabWritingMode::NormalSymtab;
378       MaybeJustCreateSymTab = true;
379       break;
380     case 'S':
381       Symtab = SymtabWritingMode::NoSymtab;
382       break;
383     case 'u':
384       OnlyUpdate = true;
385       break;
386     case 'v':
387       Verbose = true;
388       break;
389     case 'a':
390       getRelPos();
391       AddAfter = true;
392       NumPositional++;
393       break;
394     case 'b':
395       getRelPos();
396       AddBefore = true;
397       NumPositional++;
398       break;
399     case 'i':
400       getRelPos();
401       AddBefore = true;
402       NumPositional++;
403       break;
404     case 'D':
405       Deterministic = true;
406       break;
407     case 'U':
408       Deterministic = false;
409       break;
410     case 'N':
411       getCountParam();
412       break;
413     case 'T':
414       Thin = true;
415       break;
416     case 'L':
417       AddLibrary = true;
418       break;
419     case 'V':
420       cl::PrintVersionMessage();
421       exit(0);
422     case 'h':
423       printHelpMessage();
424       exit(0);
425     default:
426       badUsage(std::string("unknown option ") + Options[i]);
427     }
428   }
429 
430   // Thin archives store path names, so P should be forced.
431   if (Thin)
432     CompareFullPath = true;
433 
434   // At this point, the next thing on the command line must be
435   // the archive name.
436   getArchive();
437 
438   // Everything on the command line at this point is a member.
439   Members.assign(PositionalArgs.begin(), PositionalArgs.end());
440 
441   if (NumOperations == 0 && MaybeJustCreateSymTab) {
442     NumOperations = 1;
443     Operation = CreateSymTab;
444     if (!Members.empty())
445       badUsage("the 's' operation takes only an archive as argument");
446   }
447 
448   // Perform various checks on the operation/modifier specification
449   // to make sure we are dealing with a legal request.
450   if (NumOperations == 0)
451     badUsage("you must specify at least one of the operations");
452   if (NumOperations > 1)
453     badUsage("only one operation may be specified");
454   if (NumPositional > 1)
455     badUsage("you may only specify one of 'a', 'b', and 'i' modifiers");
456   if (AddAfter || AddBefore)
457     if (Operation != Move && Operation != ReplaceOrInsert)
458       badUsage("the 'a', 'b' and 'i' modifiers can only be specified with "
459                "the 'm' or 'r' operations");
460   if (CountParam)
461     if (Operation != Extract && Operation != Delete)
462       badUsage("the 'N' modifier can only be specified with the 'x' or 'd' "
463                "operations");
464   if (OriginalDates && Operation != Extract)
465     badUsage("the 'o' modifier is only applicable to the 'x' operation");
466   if (OnlyUpdate && Operation != ReplaceOrInsert)
467     badUsage("the 'u' modifier is only applicable to the 'r' operation");
468   if (AddLibrary && Operation != QuickAppend)
469     badUsage("the 'L' modifier is only applicable to the 'q' operation");
470 
471   if (!OutputDir.empty()) {
472     if (Operation != Extract)
473       badUsage("--output is only applicable to the 'x' operation");
474     bool IsDir = false;
475     // If OutputDir is not a directory, create_directories may still succeed if
476     // all components of the path prefix are directories. Test is_directory as
477     // well.
478     if (!sys::fs::create_directories(OutputDir))
479       sys::fs::is_directory(OutputDir, IsDir);
480     if (!IsDir)
481       fail("'" + OutputDir + "' is not a directory");
482   }
483 
484   // Return the parsed operation to the caller
485   return Operation;
486 }
487 
488 // Implements the 'p' operation. This function traverses the archive
489 // looking for members that match the path list.
doPrint(StringRef Name,const object::Archive::Child & C)490 static void doPrint(StringRef Name, const object::Archive::Child &C) {
491   if (Verbose)
492     outs() << "Printing " << Name << "\n";
493 
494   Expected<StringRef> DataOrErr = C.getBuffer();
495   failIfError(DataOrErr.takeError());
496   StringRef Data = *DataOrErr;
497   outs().write(Data.data(), Data.size());
498 }
499 
500 // Utility function for printing out the file mode when the 't' operation is in
501 // verbose mode.
printMode(unsigned mode)502 static void printMode(unsigned mode) {
503   outs() << ((mode & 004) ? "r" : "-");
504   outs() << ((mode & 002) ? "w" : "-");
505   outs() << ((mode & 001) ? "x" : "-");
506 }
507 
508 // Implement the 't' operation. This function prints out just
509 // the file names of each of the members. However, if verbose mode is requested
510 // ('v' modifier) then the file type, permission mode, user, group, size, and
511 // modification time are also printed.
doDisplayTable(StringRef Name,const object::Archive::Child & C)512 static void doDisplayTable(StringRef Name, const object::Archive::Child &C) {
513   if (Verbose) {
514     Expected<sys::fs::perms> ModeOrErr = C.getAccessMode();
515     failIfError(ModeOrErr.takeError());
516     sys::fs::perms Mode = ModeOrErr.get();
517     printMode((Mode >> 6) & 007);
518     printMode((Mode >> 3) & 007);
519     printMode(Mode & 007);
520     Expected<unsigned> UIDOrErr = C.getUID();
521     failIfError(UIDOrErr.takeError());
522     outs() << ' ' << UIDOrErr.get();
523     Expected<unsigned> GIDOrErr = C.getGID();
524     failIfError(GIDOrErr.takeError());
525     outs() << '/' << GIDOrErr.get();
526     Expected<uint64_t> Size = C.getSize();
527     failIfError(Size.takeError());
528     outs() << ' ' << format("%6llu", Size.get());
529     auto ModTimeOrErr = C.getLastModified();
530     failIfError(ModTimeOrErr.takeError());
531     // Note: formatv() only handles the default TimePoint<>, which is in
532     // nanoseconds.
533     // TODO: fix format_provider<TimePoint<>> to allow other units.
534     sys::TimePoint<> ModTimeInNs = ModTimeOrErr.get();
535     outs() << ' ' << formatv("{0:%b %e %H:%M %Y}", ModTimeInNs);
536     outs() << ' ';
537   }
538 
539   if (C.getParent()->isThin()) {
540     if (!sys::path::is_absolute(Name)) {
541       StringRef ParentDir = sys::path::parent_path(ArchiveName);
542       if (!ParentDir.empty())
543         outs() << sys::path::convert_to_slash(ParentDir) << '/';
544     }
545     outs() << Name;
546   } else {
547     outs() << Name;
548     if (DisplayMemberOffsets)
549       outs() << " 0x" << utohexstr(C.getDataOffset(), true);
550   }
551   outs() << '\n';
552 }
553 
normalizePath(StringRef Path)554 static std::string normalizePath(StringRef Path) {
555   return CompareFullPath ? sys::path::convert_to_slash(Path)
556                          : std::string(sys::path::filename(Path));
557 }
558 
comparePaths(StringRef Path1,StringRef Path2)559 static bool comparePaths(StringRef Path1, StringRef Path2) {
560 // When on Windows this function calls CompareStringOrdinal
561 // as Windows file paths are case-insensitive.
562 // CompareStringOrdinal compares two Unicode strings for
563 // binary equivalence and allows for case insensitivity.
564 #ifdef _WIN32
565   SmallVector<wchar_t, 128> WPath1, WPath2;
566   failIfError(sys::windows::UTF8ToUTF16(normalizePath(Path1), WPath1));
567   failIfError(sys::windows::UTF8ToUTF16(normalizePath(Path2), WPath2));
568 
569   return CompareStringOrdinal(WPath1.data(), WPath1.size(), WPath2.data(),
570                               WPath2.size(), true) == CSTR_EQUAL;
571 #else
572   return normalizePath(Path1) == normalizePath(Path2);
573 #endif
574 }
575 
576 // Implement the 'x' operation. This function extracts files back to the file
577 // system.
doExtract(StringRef Name,const object::Archive::Child & C)578 static void doExtract(StringRef Name, const object::Archive::Child &C) {
579   // Retain the original mode.
580   Expected<sys::fs::perms> ModeOrErr = C.getAccessMode();
581   failIfError(ModeOrErr.takeError());
582   sys::fs::perms Mode = ModeOrErr.get();
583 
584   StringRef outputFilePath;
585   SmallString<128> path;
586   if (OutputDir.empty()) {
587     outputFilePath = sys::path::filename(Name);
588   } else {
589     sys::path::append(path, OutputDir, sys::path::filename(Name));
590     outputFilePath = path.str();
591   }
592 
593   if (Verbose)
594     outs() << "x - " << outputFilePath << '\n';
595 
596   int FD;
597   failIfError(sys::fs::openFileForWrite(outputFilePath, FD,
598                                         sys::fs::CD_CreateAlways,
599                                         sys::fs::OF_None, Mode),
600               Name);
601 
602   {
603     raw_fd_ostream file(FD, false);
604 
605     // Get the data and its length
606     Expected<StringRef> BufOrErr = C.getBuffer();
607     failIfError(BufOrErr.takeError());
608     StringRef Data = BufOrErr.get();
609 
610     // Write the data.
611     file.write(Data.data(), Data.size());
612   }
613 
614   // If we're supposed to retain the original modification times, etc. do so
615   // now.
616   if (OriginalDates) {
617     auto ModTimeOrErr = C.getLastModified();
618     failIfError(ModTimeOrErr.takeError());
619     failIfError(
620         sys::fs::setLastAccessAndModificationTime(FD, ModTimeOrErr.get()));
621   }
622 
623   if (close(FD))
624     fail("Could not close the file");
625 }
626 
shouldCreateArchive(ArchiveOperation Op)627 static bool shouldCreateArchive(ArchiveOperation Op) {
628   switch (Op) {
629   case Print:
630   case Delete:
631   case Move:
632   case DisplayTable:
633   case Extract:
634   case CreateSymTab:
635     return false;
636 
637   case QuickAppend:
638   case ReplaceOrInsert:
639     return true;
640   }
641 
642   llvm_unreachable("Missing entry in covered switch.");
643 }
644 
isValidInBitMode(Binary & Bin)645 static bool isValidInBitMode(Binary &Bin) {
646   if (BitMode == BitModeTy::Bit32_64 || BitMode == BitModeTy::Any)
647     return true;
648 
649   if (SymbolicFile *SymFile = dyn_cast<SymbolicFile>(&Bin)) {
650     bool Is64Bit = SymFile->is64Bit();
651     if ((Is64Bit && (BitMode == BitModeTy::Bit32)) ||
652         (!Is64Bit && (BitMode == BitModeTy::Bit64)))
653       return false;
654   }
655   // In AIX "ar", non-object files are always considered to have a valid bit
656   // mode.
657   return true;
658 }
659 
getAsBinary(const NewArchiveMember & NM,LLVMContext * Context)660 Expected<std::unique_ptr<Binary>> getAsBinary(const NewArchiveMember &NM,
661                                               LLVMContext *Context) {
662   auto BinaryOrErr = createBinary(NM.Buf->getMemBufferRef(), Context);
663   if (BinaryOrErr)
664     return std::move(*BinaryOrErr);
665   return BinaryOrErr.takeError();
666 }
667 
getAsBinary(const Archive::Child & C,LLVMContext * Context)668 Expected<std::unique_ptr<Binary>> getAsBinary(const Archive::Child &C,
669                                               LLVMContext *Context) {
670   return C.getAsBinary(Context);
671 }
672 
isValidInBitMode(const A & Member)673 template <class A> static bool isValidInBitMode(const A &Member) {
674   if (object::Archive::getDefaultKind() != object::Archive::K_AIXBIG)
675     return true;
676   LLVMContext Context;
677   Expected<std::unique_ptr<Binary>> BinOrErr = getAsBinary(Member, &Context);
678   // In AIX "ar", if there is a non-object file member, it is never ignored due
679   // to the bit mode setting.
680   if (!BinOrErr) {
681     consumeError(BinOrErr.takeError());
682     return true;
683   }
684   return isValidInBitMode(*BinOrErr.get());
685 }
686 
warnInvalidObjectForFileMode(Twine Name)687 static void warnInvalidObjectForFileMode(Twine Name) {
688   warn("'" + Name + "' is not valid with the current object file mode");
689 }
690 
performReadOperation(ArchiveOperation Operation,object::Archive * OldArchive)691 static void performReadOperation(ArchiveOperation Operation,
692                                  object::Archive *OldArchive) {
693   if (Operation == Extract && OldArchive->isThin())
694     fail("extracting from a thin archive is not supported");
695 
696   bool Filter = !Members.empty();
697   StringMap<int> MemberCount;
698   {
699     Error Err = Error::success();
700     for (auto &C : OldArchive->children(Err)) {
701       Expected<StringRef> NameOrErr = C.getName();
702       failIfError(NameOrErr.takeError());
703       StringRef Name = NameOrErr.get();
704 
705       // Check whether to ignore this object due to its bitness.
706       if (!isValidInBitMode(C))
707         continue;
708 
709       if (Filter) {
710         auto I = find_if(Members, [Name](StringRef Path) {
711           return comparePaths(Name, Path);
712         });
713         if (I == Members.end())
714           continue;
715         if (CountParam && ++MemberCount[Name] != CountParam)
716           continue;
717         Members.erase(I);
718       }
719 
720       switch (Operation) {
721       default:
722         llvm_unreachable("Not a read operation");
723       case Print:
724         doPrint(Name, C);
725         break;
726       case DisplayTable:
727         doDisplayTable(Name, C);
728         break;
729       case Extract:
730         doExtract(Name, C);
731         break;
732       }
733     }
734     failIfError(std::move(Err));
735   }
736 
737   if (Members.empty())
738     return;
739   for (StringRef Name : Members)
740     WithColor::error(errs(), ToolName) << "'" << Name << "' was not found\n";
741   exit(1);
742 }
743 
addChildMember(std::vector<NewArchiveMember> & Members,const object::Archive::Child & M,bool FlattenArchive=false)744 static void addChildMember(std::vector<NewArchiveMember> &Members,
745                            const object::Archive::Child &M,
746                            bool FlattenArchive = false) {
747   Expected<NewArchiveMember> NMOrErr =
748       NewArchiveMember::getOldMember(M, Deterministic);
749   failIfError(NMOrErr.takeError());
750   // If the child member we're trying to add is thin, use the path relative to
751   // the archive it's in, so the file resolves correctly.
752   if (Thin && FlattenArchive) {
753     StringSaver Saver(Alloc);
754     Expected<std::string> FileNameOrErr(M.getName());
755     failIfError(FileNameOrErr.takeError());
756     if (sys::path::is_absolute(*FileNameOrErr)) {
757       NMOrErr->MemberName = Saver.save(sys::path::convert_to_slash(*FileNameOrErr));
758     } else {
759       FileNameOrErr = M.getFullName();
760       failIfError(FileNameOrErr.takeError());
761       Expected<std::string> PathOrErr =
762           computeArchiveRelativePath(ArchiveName, *FileNameOrErr);
763       NMOrErr->MemberName = Saver.save(
764           PathOrErr ? *PathOrErr : sys::path::convert_to_slash(*FileNameOrErr));
765     }
766   }
767   if (FlattenArchive &&
768       identify_magic(NMOrErr->Buf->getBuffer()) == file_magic::archive) {
769     Expected<std::string> FileNameOrErr = M.getFullName();
770     failIfError(FileNameOrErr.takeError());
771     object::Archive &Lib = readLibrary(*FileNameOrErr);
772     // When creating thin archives, only flatten if the member is also thin.
773     if (!Thin || Lib.isThin()) {
774       Error Err = Error::success();
775       // Only Thin archives are recursively flattened.
776       for (auto &Child : Lib.children(Err))
777         addChildMember(Members, Child, /*FlattenArchive=*/Thin);
778       failIfError(std::move(Err));
779       return;
780     }
781   }
782   Members.push_back(std::move(*NMOrErr));
783 }
784 
getArchiveMember(StringRef FileName)785 static NewArchiveMember getArchiveMember(StringRef FileName) {
786   Expected<NewArchiveMember> NMOrErr =
787       NewArchiveMember::getFile(FileName, Deterministic);
788   failIfError(NMOrErr.takeError(), FileName);
789   StringSaver Saver(Alloc);
790   // For regular archives, use the basename of the object path for the member
791   // name. For thin archives, use the full relative paths so the file resolves
792   // correctly.
793   if (!Thin) {
794     NMOrErr->MemberName = sys::path::filename(NMOrErr->MemberName);
795   } else {
796     if (sys::path::is_absolute(FileName))
797       NMOrErr->MemberName = Saver.save(sys::path::convert_to_slash(FileName));
798     else {
799       Expected<std::string> PathOrErr =
800           computeArchiveRelativePath(ArchiveName, FileName);
801       NMOrErr->MemberName = Saver.save(
802           PathOrErr ? *PathOrErr : sys::path::convert_to_slash(FileName));
803     }
804   }
805   return std::move(*NMOrErr);
806 }
807 
addMember(std::vector<NewArchiveMember> & Members,NewArchiveMember & NM)808 static void addMember(std::vector<NewArchiveMember> &Members,
809                       NewArchiveMember &NM) {
810   Members.push_back(std::move(NM));
811 }
812 
addMember(std::vector<NewArchiveMember> & Members,StringRef FileName,bool FlattenArchive=false)813 static void addMember(std::vector<NewArchiveMember> &Members,
814                       StringRef FileName, bool FlattenArchive = false) {
815   NewArchiveMember NM = getArchiveMember(FileName);
816   if (!isValidInBitMode(NM)) {
817     warnInvalidObjectForFileMode(FileName);
818     return;
819   }
820 
821   if (FlattenArchive &&
822       identify_magic(NM.Buf->getBuffer()) == file_magic::archive) {
823     object::Archive &Lib = readLibrary(FileName);
824     // When creating thin archives, only flatten if the member is also thin.
825     if (!Thin || Lib.isThin()) {
826       Error Err = Error::success();
827       // Only Thin archives are recursively flattened.
828       for (auto &Child : Lib.children(Err))
829         addChildMember(Members, Child, /*FlattenArchive=*/Thin);
830       failIfError(std::move(Err));
831       return;
832     }
833   }
834   Members.push_back(std::move(NM));
835 }
836 
837 enum InsertAction {
838   IA_AddOldMember,
839   IA_AddNewMember,
840   IA_Delete,
841   IA_MoveOldMember,
842   IA_MoveNewMember
843 };
844 
computeInsertAction(ArchiveOperation Operation,const object::Archive::Child & Member,StringRef Name,std::vector<StringRef>::iterator & Pos,StringMap<int> & MemberCount)845 static InsertAction computeInsertAction(ArchiveOperation Operation,
846                                         const object::Archive::Child &Member,
847                                         StringRef Name,
848                                         std::vector<StringRef>::iterator &Pos,
849                                         StringMap<int> &MemberCount) {
850   if (!isValidInBitMode(Member))
851     return IA_AddOldMember;
852 
853   if (Operation == QuickAppend || Members.empty())
854     return IA_AddOldMember;
855 
856   auto MI = find_if(Members, [Name](StringRef Path) {
857     if (Thin && !sys::path::is_absolute(Path)) {
858       Expected<std::string> PathOrErr =
859           computeArchiveRelativePath(ArchiveName, Path);
860       return comparePaths(Name, PathOrErr ? *PathOrErr : Path);
861     } else {
862       return comparePaths(Name, Path);
863     }
864   });
865 
866   if (MI == Members.end())
867     return IA_AddOldMember;
868 
869   Pos = MI;
870 
871   if (Operation == Delete) {
872     if (CountParam && ++MemberCount[Name] != CountParam)
873       return IA_AddOldMember;
874     return IA_Delete;
875   }
876 
877   if (Operation == Move)
878     return IA_MoveOldMember;
879 
880   if (Operation == ReplaceOrInsert) {
881     if (!OnlyUpdate) {
882       if (RelPos.empty())
883         return IA_AddNewMember;
884       return IA_MoveNewMember;
885     }
886 
887     // We could try to optimize this to a fstat, but it is not a common
888     // operation.
889     sys::fs::file_status Status;
890     failIfError(sys::fs::status(*MI, Status), *MI);
891     auto ModTimeOrErr = Member.getLastModified();
892     failIfError(ModTimeOrErr.takeError());
893     if (Status.getLastModificationTime() < ModTimeOrErr.get()) {
894       if (RelPos.empty())
895         return IA_AddOldMember;
896       return IA_MoveOldMember;
897     }
898 
899     if (RelPos.empty())
900       return IA_AddNewMember;
901     return IA_MoveNewMember;
902   }
903   llvm_unreachable("No such operation");
904 }
905 
906 // We have to walk this twice and computing it is not trivial, so creating an
907 // explicit std::vector is actually fairly efficient.
908 static std::vector<NewArchiveMember>
computeNewArchiveMembers(ArchiveOperation Operation,object::Archive * OldArchive)909 computeNewArchiveMembers(ArchiveOperation Operation,
910                          object::Archive *OldArchive) {
911   std::vector<NewArchiveMember> Ret;
912   std::vector<NewArchiveMember> Moved;
913   int InsertPos = -1;
914   if (OldArchive) {
915     Error Err = Error::success();
916     StringMap<int> MemberCount;
917     for (auto &Child : OldArchive->children(Err)) {
918       int Pos = Ret.size();
919       Expected<StringRef> NameOrErr = Child.getName();
920       failIfError(NameOrErr.takeError());
921       std::string Name = std::string(NameOrErr.get());
922       if (comparePaths(Name, RelPos) && isValidInBitMode(Child)) {
923         assert(AddAfter || AddBefore);
924         if (AddBefore)
925           InsertPos = Pos;
926         else
927           InsertPos = Pos + 1;
928       }
929 
930       std::vector<StringRef>::iterator MemberI = Members.end();
931       InsertAction Action =
932           computeInsertAction(Operation, Child, Name, MemberI, MemberCount);
933 
934       auto HandleNewMember = [](auto Member, auto &Members, auto &Child) {
935         NewArchiveMember NM = getArchiveMember(*Member);
936         if (isValidInBitMode(NM))
937           addMember(Members, NM);
938         else {
939           // If a new member is not a valid object for the bit mode, add
940           // the old member back.
941           warnInvalidObjectForFileMode(*Member);
942           addChildMember(Members, Child, /*FlattenArchive=*/Thin);
943         }
944       };
945 
946       switch (Action) {
947       case IA_AddOldMember:
948         addChildMember(Ret, Child, /*FlattenArchive=*/Thin);
949         break;
950       case IA_AddNewMember:
951         HandleNewMember(MemberI, Ret, Child);
952         break;
953       case IA_Delete:
954         break;
955       case IA_MoveOldMember:
956         addChildMember(Moved, Child, /*FlattenArchive=*/Thin);
957         break;
958       case IA_MoveNewMember:
959         HandleNewMember(MemberI, Moved, Child);
960         break;
961       }
962       // When processing elements with the count param, we need to preserve the
963       // full members list when iterating over all archive members. For
964       // instance, "llvm-ar dN 2 archive.a member.o" should delete the second
965       // file named member.o it sees; we are not done with member.o the first
966       // time we see it in the archive.
967       if (MemberI != Members.end() && !CountParam)
968         Members.erase(MemberI);
969     }
970     failIfError(std::move(Err));
971   }
972 
973   if (Operation == Delete)
974     return Ret;
975 
976   if (!RelPos.empty() && InsertPos == -1)
977     fail("insertion point not found");
978 
979   if (RelPos.empty())
980     InsertPos = Ret.size();
981 
982   assert(unsigned(InsertPos) <= Ret.size());
983   int Pos = InsertPos;
984   for (auto &M : Moved) {
985     Ret.insert(Ret.begin() + Pos, std::move(M));
986     ++Pos;
987   }
988 
989   if (AddLibrary) {
990     assert(Operation == QuickAppend);
991     for (auto &Member : Members)
992       addMember(Ret, Member, /*FlattenArchive=*/true);
993     return Ret;
994   }
995 
996   std::vector<NewArchiveMember> NewMembers;
997   for (auto &Member : Members)
998     addMember(NewMembers, Member, /*FlattenArchive=*/Thin);
999   Ret.reserve(Ret.size() + NewMembers.size());
1000   std::move(NewMembers.begin(), NewMembers.end(),
1001             std::inserter(Ret, std::next(Ret.begin(), InsertPos)));
1002 
1003   return Ret;
1004 }
1005 
performWriteOperation(ArchiveOperation Operation,object::Archive * OldArchive,std::unique_ptr<MemoryBuffer> OldArchiveBuf,std::vector<NewArchiveMember> * NewMembersP)1006 static void performWriteOperation(ArchiveOperation Operation,
1007                                   object::Archive *OldArchive,
1008                                   std::unique_ptr<MemoryBuffer> OldArchiveBuf,
1009                                   std::vector<NewArchiveMember> *NewMembersP) {
1010   if (OldArchive) {
1011     if (Thin && !OldArchive->isThin())
1012       fail("cannot convert a regular archive to a thin one");
1013 
1014     if (OldArchive->isThin())
1015       Thin = true;
1016   }
1017 
1018   std::vector<NewArchiveMember> NewMembers;
1019   if (!NewMembersP)
1020     NewMembers = computeNewArchiveMembers(Operation, OldArchive);
1021 
1022   object::Archive::Kind Kind;
1023   switch (FormatType) {
1024   case Default:
1025     if (Thin)
1026       Kind = object::Archive::K_GNU;
1027     else if (OldArchive) {
1028       Kind = OldArchive->kind();
1029       std::optional<object::Archive::Kind> AltKind;
1030       if (Kind == object::Archive::K_BSD)
1031         AltKind = object::Archive::K_DARWIN;
1032       else if (Kind == object::Archive::K_GNU && !OldArchive->hasSymbolTable())
1033         // If there is no symbol table, we can't tell GNU from COFF format
1034         // from the old archive type.
1035         AltKind = object::Archive::K_COFF;
1036       if (AltKind) {
1037         auto InferredKind = Kind;
1038         if (NewMembersP && !NewMembersP->empty())
1039           InferredKind = NewMembersP->front().detectKindFromObject();
1040         else if (!NewMembers.empty())
1041           InferredKind = NewMembers.front().detectKindFromObject();
1042         if (InferredKind == AltKind)
1043           Kind = *AltKind;
1044       }
1045     } else if (NewMembersP)
1046       Kind = !NewMembersP->empty() ? NewMembersP->front().detectKindFromObject()
1047                                    : object::Archive::getDefaultKind();
1048     else
1049       Kind = !NewMembers.empty() ? NewMembers.front().detectKindFromObject()
1050                                  : object::Archive::getDefaultKind();
1051     break;
1052   case GNU:
1053     Kind = object::Archive::K_GNU;
1054     break;
1055   case COFF:
1056     Kind = object::Archive::K_COFF;
1057     break;
1058   case BSD:
1059     if (Thin)
1060       fail("only the gnu format has a thin mode");
1061     Kind = object::Archive::K_BSD;
1062     break;
1063   case DARWIN:
1064     if (Thin)
1065       fail("only the gnu format has a thin mode");
1066     Kind = object::Archive::K_DARWIN;
1067     break;
1068   case BIGARCHIVE:
1069     if (Thin)
1070       fail("only the gnu format has a thin mode");
1071     Kind = object::Archive::K_AIXBIG;
1072     break;
1073   case Unknown:
1074     llvm_unreachable("");
1075   }
1076 
1077   Error E =
1078       writeArchive(ArchiveName, NewMembersP ? *NewMembersP : NewMembers, Symtab,
1079                    Kind, Deterministic, Thin, std::move(OldArchiveBuf));
1080   failIfError(std::move(E), ArchiveName);
1081 }
1082 
createSymbolTable(object::Archive * OldArchive)1083 static void createSymbolTable(object::Archive *OldArchive) {
1084   // When an archive is created or modified, if the s option is given, the
1085   // resulting archive will have a current symbol table. If the S option
1086   // is given, it will have no symbol table.
1087   // In summary, we only need to update the symbol table if we have none.
1088   // This is actually very common because of broken build systems that think
1089   // they have to run ranlib.
1090   if (OldArchive->hasSymbolTable()) {
1091     if (OldArchive->kind() != object::Archive::K_AIXBIG)
1092       return;
1093 
1094     // For archives in the Big Archive format, the bit mode option specifies
1095     // which symbol table to generate. The presence of a symbol table that does
1096     // not match the specified bit mode does not prevent creation of the symbol
1097     // table that has been requested.
1098     if (OldArchive->kind() == object::Archive::K_AIXBIG) {
1099       BigArchive *BigArc = dyn_cast<BigArchive>(OldArchive);
1100       if (BigArc->has32BitGlobalSymtab() &&
1101           Symtab == SymtabWritingMode::BigArchive32)
1102         return;
1103 
1104       if (BigArc->has64BitGlobalSymtab() &&
1105           Symtab == SymtabWritingMode::BigArchive64)
1106         return;
1107 
1108       if (BigArc->has32BitGlobalSymtab() && BigArc->has64BitGlobalSymtab() &&
1109           Symtab == SymtabWritingMode::NormalSymtab)
1110         return;
1111 
1112       Symtab = SymtabWritingMode::NormalSymtab;
1113     }
1114   }
1115   if (OldArchive->isThin())
1116     Thin = true;
1117   performWriteOperation(CreateSymTab, OldArchive, nullptr, nullptr);
1118 }
1119 
performOperation(ArchiveOperation Operation,object::Archive * OldArchive,std::unique_ptr<MemoryBuffer> OldArchiveBuf,std::vector<NewArchiveMember> * NewMembers)1120 static void performOperation(ArchiveOperation Operation,
1121                              object::Archive *OldArchive,
1122                              std::unique_ptr<MemoryBuffer> OldArchiveBuf,
1123                              std::vector<NewArchiveMember> *NewMembers) {
1124   switch (Operation) {
1125   case Print:
1126   case DisplayTable:
1127   case Extract:
1128     performReadOperation(Operation, OldArchive);
1129     return;
1130 
1131   case Delete:
1132   case Move:
1133   case QuickAppend:
1134   case ReplaceOrInsert:
1135     performWriteOperation(Operation, OldArchive, std::move(OldArchiveBuf),
1136                           NewMembers);
1137     return;
1138   case CreateSymTab:
1139     createSymbolTable(OldArchive);
1140     return;
1141   }
1142   llvm_unreachable("Unknown operation.");
1143 }
1144 
performOperation(ArchiveOperation Operation)1145 static int performOperation(ArchiveOperation Operation) {
1146   // Create or open the archive object.
1147   ErrorOr<std::unique_ptr<MemoryBuffer>> Buf = MemoryBuffer::getFile(
1148       ArchiveName, /*IsText=*/false, /*RequiresNullTerminator=*/false);
1149   std::error_code EC = Buf.getError();
1150   if (EC && EC != errc::no_such_file_or_directory)
1151     fail("unable to open '" + ArchiveName + "': " + EC.message());
1152 
1153   if (!EC) {
1154     Expected<std::unique_ptr<object::Archive>> ArchiveOrError =
1155         object::Archive::create(Buf.get()->getMemBufferRef());
1156     if (!ArchiveOrError)
1157       failIfError(ArchiveOrError.takeError(),
1158                   "unable to load '" + ArchiveName + "'");
1159 
1160     std::unique_ptr<object::Archive> Archive = std::move(ArchiveOrError.get());
1161     if (Archive->isThin())
1162       CompareFullPath = true;
1163     performOperation(Operation, Archive.get(), std::move(Buf.get()),
1164                      /*NewMembers=*/nullptr);
1165     return 0;
1166   }
1167 
1168   assert(EC == errc::no_such_file_or_directory);
1169 
1170   if (!shouldCreateArchive(Operation)) {
1171     failIfError(EC, Twine("unable to load '") + ArchiveName + "'");
1172   } else {
1173     if (!Create) {
1174       // Produce a warning if we should and we're creating the archive
1175       warn("creating " + ArchiveName);
1176     }
1177   }
1178 
1179   performOperation(Operation, nullptr, nullptr, /*NewMembers=*/nullptr);
1180   return 0;
1181 }
1182 
runMRIScript()1183 static void runMRIScript() {
1184   enum class MRICommand { AddLib, AddMod, Create, CreateThin, Delete, Save, End, Invalid };
1185 
1186   ErrorOr<std::unique_ptr<MemoryBuffer>> Buf = MemoryBuffer::getSTDIN();
1187   failIfError(Buf.getError());
1188   const MemoryBuffer &Ref = *Buf.get();
1189   bool Saved = false;
1190   std::vector<NewArchiveMember> NewMembers;
1191   ParsingMRIScript = true;
1192 
1193   for (line_iterator I(Ref, /*SkipBlanks*/ false), E; I != E; ++I) {
1194     ++MRILineNumber;
1195     StringRef Line = *I;
1196     Line = Line.split(';').first;
1197     Line = Line.split('*').first;
1198     Line = Line.trim();
1199     if (Line.empty())
1200       continue;
1201     StringRef CommandStr, Rest;
1202     std::tie(CommandStr, Rest) = Line.split(' ');
1203     Rest = Rest.trim();
1204     if (!Rest.empty() && Rest.front() == '"' && Rest.back() == '"')
1205       Rest = Rest.drop_front().drop_back();
1206     auto Command = StringSwitch<MRICommand>(CommandStr.lower())
1207                        .Case("addlib", MRICommand::AddLib)
1208                        .Case("addmod", MRICommand::AddMod)
1209                        .Case("create", MRICommand::Create)
1210                        .Case("createthin", MRICommand::CreateThin)
1211                        .Case("delete", MRICommand::Delete)
1212                        .Case("save", MRICommand::Save)
1213                        .Case("end", MRICommand::End)
1214                        .Default(MRICommand::Invalid);
1215 
1216     switch (Command) {
1217     case MRICommand::AddLib: {
1218       if (!Create)
1219         fail("no output archive has been opened");
1220       object::Archive &Lib = readLibrary(Rest);
1221       {
1222         if (Thin && !Lib.isThin())
1223           fail("cannot add a regular archive's contents to a thin archive");
1224         Error Err = Error::success();
1225         for (auto &Member : Lib.children(Err))
1226           addChildMember(NewMembers, Member, /*FlattenArchive=*/Thin);
1227         failIfError(std::move(Err));
1228       }
1229       break;
1230     }
1231     case MRICommand::AddMod:
1232       if (!Create)
1233         fail("no output archive has been opened");
1234       addMember(NewMembers, Rest);
1235       break;
1236     case MRICommand::CreateThin:
1237       Thin = true;
1238       [[fallthrough]];
1239     case MRICommand::Create:
1240       Create = true;
1241       if (!ArchiveName.empty())
1242         fail("editing multiple archives not supported");
1243       if (Saved)
1244         fail("file already saved");
1245       ArchiveName = std::string(Rest);
1246       if (ArchiveName.empty())
1247         fail("missing archive name");
1248       break;
1249     case MRICommand::Delete: {
1250       llvm::erase_if(NewMembers, [=](NewArchiveMember &M) {
1251         return comparePaths(M.MemberName, Rest);
1252       });
1253       break;
1254     }
1255     case MRICommand::Save:
1256       Saved = true;
1257       break;
1258     case MRICommand::End:
1259       break;
1260     case MRICommand::Invalid:
1261       fail("unknown command: " + CommandStr);
1262     }
1263   }
1264 
1265   ParsingMRIScript = false;
1266 
1267   // Nothing to do if not saved.
1268   if (Saved)
1269     performOperation(ReplaceOrInsert, /*OldArchive=*/nullptr,
1270                      /*OldArchiveBuf=*/nullptr, &NewMembers);
1271   exit(0);
1272 }
1273 
handleGenericOption(StringRef arg)1274 static bool handleGenericOption(StringRef arg) {
1275   if (arg == "--help" || arg == "-h") {
1276     printHelpMessage();
1277     return true;
1278   }
1279   if (arg == "--version") {
1280     cl::PrintVersionMessage();
1281     return true;
1282   }
1283   return false;
1284 }
1285 
getBitMode(const char * RawBitMode)1286 static BitModeTy getBitMode(const char *RawBitMode) {
1287   return StringSwitch<BitModeTy>(RawBitMode)
1288       .Case("32", BitModeTy::Bit32)
1289       .Case("64", BitModeTy::Bit64)
1290       .Case("32_64", BitModeTy::Bit32_64)
1291       .Case("any", BitModeTy::Any)
1292       .Default(BitModeTy::Unknown);
1293 }
1294 
matchFlagWithArg(StringRef Expected,ArrayRef<const char * >::iterator & ArgIt,ArrayRef<const char * > Args)1295 static const char *matchFlagWithArg(StringRef Expected,
1296                                     ArrayRef<const char *>::iterator &ArgIt,
1297                                     ArrayRef<const char *> Args) {
1298   StringRef Arg = *ArgIt;
1299 
1300   Arg.consume_front("--");
1301 
1302   size_t len = Expected.size();
1303   if (Arg == Expected) {
1304     if (++ArgIt == Args.end())
1305       fail(std::string(Expected) + " requires an argument");
1306 
1307     return *ArgIt;
1308   }
1309   if (Arg.starts_with(Expected) && Arg.size() > len && Arg[len] == '=')
1310     return Arg.data() + len + 1;
1311 
1312   return nullptr;
1313 }
1314 
getRspQuoting(ArrayRef<const char * > ArgsArr)1315 static cl::TokenizerCallback getRspQuoting(ArrayRef<const char *> ArgsArr) {
1316   cl::TokenizerCallback Ret =
1317       Triple(sys::getProcessTriple()).getOS() == Triple::Win32
1318           ? cl::TokenizeWindowsCommandLine
1319           : cl::TokenizeGNUCommandLine;
1320 
1321   for (ArrayRef<const char *>::iterator ArgIt = ArgsArr.begin();
1322        ArgIt != ArgsArr.end(); ++ArgIt) {
1323     if (const char *Match = matchFlagWithArg("rsp-quoting", ArgIt, ArgsArr)) {
1324       StringRef MatchRef = Match;
1325       if (MatchRef == "posix")
1326         Ret = cl::TokenizeGNUCommandLine;
1327       else if (MatchRef == "windows")
1328         Ret = cl::TokenizeWindowsCommandLine;
1329       else
1330         fail(std::string("Invalid response file quoting style ") + Match);
1331     }
1332   }
1333 
1334   return Ret;
1335 }
1336 
ar_main(int argc,char ** argv)1337 static int ar_main(int argc, char **argv) {
1338   SmallVector<const char *, 0> Argv(argv + 1, argv + argc);
1339   StringSaver Saver(Alloc);
1340 
1341   cl::ExpandResponseFiles(Saver, getRspQuoting(ArrayRef(argv, argc)), Argv);
1342 
1343   // Get BitMode from enviorment variable "OBJECT_MODE" for AIX OS, if
1344   // specified.
1345   if (object::Archive::getDefaultKind() == object::Archive::K_AIXBIG) {
1346     BitMode = getBitMode(getenv("OBJECT_MODE"));
1347     if (BitMode == BitModeTy::Unknown)
1348       BitMode = BitModeTy::Bit32;
1349   }
1350 
1351   for (ArrayRef<const char *>::iterator ArgIt = Argv.begin();
1352        ArgIt != Argv.end(); ++ArgIt) {
1353     const char *Match = nullptr;
1354 
1355     if (handleGenericOption(*ArgIt))
1356       return 0;
1357     if (strcmp(*ArgIt, "--") == 0) {
1358       ++ArgIt;
1359       for (; ArgIt != Argv.end(); ++ArgIt)
1360         PositionalArgs.push_back(*ArgIt);
1361       break;
1362     }
1363 
1364     if (*ArgIt[0] != '-') {
1365       if (Options.empty())
1366         Options += *ArgIt;
1367       else
1368         PositionalArgs.push_back(*ArgIt);
1369       continue;
1370     }
1371 
1372     if (strcmp(*ArgIt, "-M") == 0) {
1373       MRI = true;
1374       continue;
1375     }
1376 
1377     if (strcmp(*ArgIt, "--thin") == 0) {
1378       Thin = true;
1379       continue;
1380     }
1381 
1382     Match = matchFlagWithArg("format", ArgIt, Argv);
1383     if (Match) {
1384       FormatType = StringSwitch<Format>(Match)
1385                        .Case("default", Default)
1386                        .Case("gnu", GNU)
1387                        .Case("darwin", DARWIN)
1388                        .Case("bsd", BSD)
1389                        .Case("bigarchive", BIGARCHIVE)
1390                        .Case("coff", COFF)
1391                        .Default(Unknown);
1392       if (FormatType == Unknown)
1393         fail(std::string("Invalid format ") + Match);
1394       continue;
1395     }
1396 
1397     if ((Match = matchFlagWithArg("output", ArgIt, Argv))) {
1398       OutputDir = Match;
1399       continue;
1400     }
1401 
1402     if (matchFlagWithArg("plugin", ArgIt, Argv) ||
1403         matchFlagWithArg("rsp-quoting", ArgIt, Argv))
1404       continue;
1405 
1406     if (strncmp(*ArgIt, "-X", 2) == 0) {
1407       if (object::Archive::getDefaultKind() == object::Archive::K_AIXBIG) {
1408         Match = *(*ArgIt + 2) != '\0' ? *ArgIt + 2 : *(++ArgIt);
1409         BitMode = getBitMode(Match);
1410         if (BitMode == BitModeTy::Unknown)
1411           fail(Twine("invalid bit mode: ") + Match);
1412         continue;
1413       } else {
1414         fail(Twine(*ArgIt) + " option not supported on non AIX OS");
1415       }
1416     }
1417 
1418     Options += *ArgIt + 1;
1419   }
1420 
1421   return performOperation(parseCommandLine());
1422 }
1423 
ranlib_main(int argc,char ** argv)1424 static int ranlib_main(int argc, char **argv) {
1425   std::vector<StringRef> Archives;
1426   bool HasAIXXOption = false;
1427 
1428   for (int i = 1; i < argc; ++i) {
1429     StringRef arg(argv[i]);
1430     if (handleGenericOption(arg)) {
1431       return 0;
1432     } else if (arg.consume_front("-")) {
1433       // Handle the -D/-U flag
1434       while (!arg.empty()) {
1435         if (arg.front() == 'D') {
1436           Deterministic = true;
1437         } else if (arg.front() == 'U') {
1438           Deterministic = false;
1439         } else if (arg.front() == 'h') {
1440           printHelpMessage();
1441           return 0;
1442         } else if (arg.front() == 'V') {
1443           cl::PrintVersionMessage();
1444           return 0;
1445         } else if (arg.front() == 'X') {
1446           if (object::Archive::getDefaultKind() == object::Archive::K_AIXBIG) {
1447             HasAIXXOption = true;
1448             arg.consume_front("X");
1449             const char *Xarg = arg.data();
1450             if (Xarg[0] == '\0') {
1451               if (argv[i + 1][0] != '-')
1452                 BitMode = getBitMode(argv[++i]);
1453               else
1454                 BitMode = BitModeTy::Unknown;
1455             } else
1456               BitMode = getBitMode(arg.data());
1457 
1458             if (BitMode == BitModeTy::Unknown)
1459               fail("the specified object mode is not valid. Specify -X32, "
1460                    "-X64, -X32_64, or -Xany");
1461           } else {
1462             fail(Twine("-") + Twine(arg) +
1463                  " option not supported on non AIX OS");
1464           }
1465           break;
1466         } else {
1467           // TODO: GNU ranlib also supports a -t flag
1468           fail("Invalid option: '-" + arg + "'");
1469         }
1470         arg = arg.drop_front(1);
1471       }
1472     } else {
1473       Archives.push_back(arg);
1474     }
1475   }
1476 
1477   if (object::Archive::getDefaultKind() == object::Archive::K_AIXBIG) {
1478     // If not specify -X option, get BitMode from enviorment variable
1479     // "OBJECT_MODE" for AIX OS if specify.
1480     if (!HasAIXXOption) {
1481       if (char *EnvObjectMode = getenv("OBJECT_MODE")) {
1482         BitMode = getBitMode(EnvObjectMode);
1483         if (BitMode == BitModeTy::Unknown)
1484           fail("the OBJECT_MODE environment variable has an invalid value. "
1485                "OBJECT_MODE must be 32, 64, 32_64, or any");
1486       }
1487     }
1488 
1489     switch (BitMode) {
1490     case BitModeTy::Bit32:
1491       Symtab = SymtabWritingMode::BigArchive32;
1492       break;
1493     case BitModeTy::Bit64:
1494       Symtab = SymtabWritingMode::BigArchive64;
1495       break;
1496     default:
1497       Symtab = SymtabWritingMode::NormalSymtab;
1498       break;
1499     }
1500   }
1501 
1502   for (StringRef Archive : Archives) {
1503     ArchiveName = Archive.str();
1504     performOperation(CreateSymTab);
1505   }
1506   if (Archives.empty())
1507     badUsage("an archive name must be specified");
1508   return 0;
1509 }
1510 
llvm_ar_main(int argc,char ** argv,const llvm::ToolContext &)1511 int llvm_ar_main(int argc, char **argv, const llvm::ToolContext &) {
1512   ToolName = argv[0];
1513 
1514   llvm::InitializeAllTargetInfos();
1515   llvm::InitializeAllTargetMCs();
1516   llvm::InitializeAllAsmParsers();
1517 
1518   Stem = sys::path::stem(ToolName);
1519   auto Is = [](StringRef Tool) {
1520     // We need to recognize the following filenames.
1521     //
1522     // Lib.exe -> lib (see D44808, MSBuild runs Lib.exe)
1523     // dlltool.exe -> dlltool
1524     // arm-pokymllib32-linux-gnueabi-llvm-ar-10 -> ar
1525     auto I = Stem.rfind_insensitive(Tool);
1526     return I != StringRef::npos &&
1527            (I + Tool.size() == Stem.size() || !isAlnum(Stem[I + Tool.size()]));
1528   };
1529 
1530   if (Is("dlltool"))
1531     return dlltoolDriverMain(ArrayRef(argv, argc));
1532   if (Is("ranlib"))
1533     return ranlib_main(argc, argv);
1534   if (Is("lib"))
1535     return libDriverMain(ArrayRef(argv, argc));
1536   if (Is("ar"))
1537     return ar_main(argc, argv);
1538 
1539   fail("not ranlib, ar, lib or dlltool");
1540 }
1541