blob: 6026fa7cfea6f06fec41cf8ffcb8d40c0056cc32 [file] [log] [blame]
Chris Lattner97f752f2003-12-30 07:45:46 +00001//===-- llvm-ar.cpp - LLVM archive librarian utility ----------------------===//
Misha Brukman3da94ae2005-04-22 00:00:37 +00002//
John Criswell7c0e0222003-10-20 17:47:21 +00003// The LLVM Compiler Infrastructure
4//
Chris Lattner21c62da2007-12-29 20:44:31 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Misha Brukman3da94ae2005-04-22 00:00:37 +00007//
John Criswell7c0e0222003-10-20 17:47:21 +00008//===----------------------------------------------------------------------===//
Tanya Lattner14baebf2003-08-28 15:22:38 +00009//
Misha Brukman3da94ae2005-04-22 00:00:37 +000010// Builds up (relatively) standard unix archive files (.a) containing LLVM
Gabor Greifa99be512007-07-05 17:07:56 +000011// bitcode or other files.
Tanya Lattner14baebf2003-08-28 15:22:38 +000012//
13//===----------------------------------------------------------------------===//
Brian Gaeke4fa9fd32003-10-10 18:47:08 +000014
Chandler Carruth0b8c9a82013-01-02 11:36:10 +000015#include "llvm/IR/LLVMContext.h"
Chandler Carruth0b8c9a82013-01-02 11:36:10 +000016#include "llvm/IR/Module.h"
Rafael Espindola34ac52d2013-07-12 20:21:39 +000017#include "llvm/Object/Archive.h"
Rafael Espindola26b10da2013-07-23 10:47:01 +000018#include "llvm/Object/ObjectFile.h"
Reid Spencer551ccae2004-09-01 22:55:40 +000019#include "llvm/Support/CommandLine.h"
Michael J. Spencer54453f22011-01-10 02:34:23 +000020#include "llvm/Support/FileSystem.h"
Chandler Carruthf010c462012-12-04 10:44:52 +000021#include "llvm/Support/Format.h"
Chris Lattnerc30598b2006-12-06 01:18:01 +000022#include "llvm/Support/ManagedStatic.h"
Rafael Espindola34ac52d2013-07-12 20:21:39 +000023#include "llvm/Support/MemoryBuffer.h"
Chris Lattnercc14d252009-03-06 05:34:10 +000024#include "llvm/Support/PrettyStackTrace.h"
Michael J. Spencer1f6efa32010-11-29 18:16:10 +000025#include "llvm/Support/Signals.h"
Rafael Espindola34ac52d2013-07-12 20:21:39 +000026#include "llvm/Support/ToolOutputFile.h"
Chandler Carruthf010c462012-12-04 10:44:52 +000027#include "llvm/Support/raw_ostream.h"
Reid Spencer3a1582b2004-11-14 22:20:07 +000028#include <algorithm>
Joerg Sonnenberger975bc072012-10-26 10:49:15 +000029#include <cstdlib>
Chandler Carruthf010c462012-12-04 10:44:52 +000030#include <memory>
Rafael Espindola11ca2e52013-06-20 20:56:14 +000031
32#if !defined(_MSC_VER) && !defined(__MINGW32__)
33#include <unistd.h>
34#else
35#include <io.h>
36#endif
37
Brian Gaeked0fde302003-11-11 22:41:34 +000038using namespace llvm;
39
Rafael Espindola34ac52d2013-07-12 20:21:39 +000040// The name this program was invoked as.
41static StringRef ToolName;
42
43static const char *TemporaryOutput;
Rafael Espindola95779b62013-07-16 16:00:32 +000044static int TmpArchiveFD = -1;
Rafael Espindola34ac52d2013-07-12 20:21:39 +000045
46// fail - Show the error message and exit.
47LLVM_ATTRIBUTE_NORETURN static void fail(Twine Error) {
48 outs() << ToolName << ": " << Error << ".\n";
Rafael Espindola95779b62013-07-16 16:00:32 +000049 if (TmpArchiveFD != -1)
50 close(TmpArchiveFD);
Rafael Espindola34ac52d2013-07-12 20:21:39 +000051 if (TemporaryOutput)
52 sys::fs::remove(TemporaryOutput);
53 exit(1);
54}
55
56static void failIfError(error_code EC, Twine Context = "") {
57 if (!EC)
58 return;
59
60 std::string ContextStr = Context.str();
61 if (ContextStr == "")
62 fail(EC.message());
63 fail(Context + ": " + EC.message());
64}
65
Dan Gohman82f209e2007-10-15 21:10:03 +000066// Option for compatibility with AIX, not used but must allow it to be present.
Misha Brukman3da94ae2005-04-22 00:00:37 +000067static cl::opt<bool>
68X32Option ("X32_64", cl::Hidden,
Reid Spencerbede5832004-11-16 06:41:09 +000069 cl::desc("Ignored option for compatibility with AIX"));
Tanya Lattner14baebf2003-08-28 15:22:38 +000070
Reid Spencerbede5832004-11-16 06:41:09 +000071// llvm-ar operation code and modifier flags. This must come first.
Misha Brukman3da94ae2005-04-22 00:00:37 +000072static cl::opt<std::string>
Reid Spencer3a1582b2004-11-14 22:20:07 +000073Options(cl::Positional, cl::Required, cl::desc("{operation}[modifiers]..."));
Tanya Lattner14baebf2003-08-28 15:22:38 +000074
Reid Spencerbede5832004-11-16 06:41:09 +000075// llvm-ar remaining positional arguments.
Misha Brukman3da94ae2005-04-22 00:00:37 +000076static cl::list<std::string>
77RestOfArgs(cl::Positional, cl::OneOrMore,
Reid Spencer3a1582b2004-11-14 22:20:07 +000078 cl::desc("[relpos] [count] <archive-file> [members]..."));
Tanya Lattner14baebf2003-08-28 15:22:38 +000079
Reid Spencerbede5832004-11-16 06:41:09 +000080// MoreHelp - Provide additional help output explaining the operations and
81// modifiers of llvm-ar. This object instructs the CommandLine library
82// to print the text of the constructor when the --help option is given.
83static cl::extrahelp MoreHelp(
Misha Brukman3da94ae2005-04-22 00:00:37 +000084 "\nOPERATIONS:\n"
Reid Spencerbede5832004-11-16 06:41:09 +000085 " d[NsS] - delete file(s) from the archive\n"
86 " m[abiSs] - move file(s) in the archive\n"
87 " p[kN] - print file(s) found in the archive\n"
88 " q[ufsS] - quick append file(s) to the archive\n"
Rafael Espindola94bc2462012-08-10 01:57:52 +000089 " r[abfiuRsS] - replace or insert file(s) into the archive\n"
Reid Spencerbede5832004-11-16 06:41:09 +000090 " t - display contents of archive\n"
91 " x[No] - extract file(s) from the archive\n"
92 "\nMODIFIERS (operation specific):\n"
93 " [a] - put file(s) after [relpos]\n"
94 " [b] - put file(s) before [relpos] (same as [i])\n"
Reid Spencerbede5832004-11-16 06:41:09 +000095 " [i] - put file(s) before [relpos] (same as [b])\n"
Reid Spencerbede5832004-11-16 06:41:09 +000096 " [N] - use instance [count] of name\n"
97 " [o] - preserve original dates\n"
Reid Spencerbede5832004-11-16 06:41:09 +000098 " [s] - create an archive index (cf. ranlib)\n"
99 " [S] - do not build a symbol table\n"
100 " [u] - update only files newer than archive contents\n"
Reid Spencerbede5832004-11-16 06:41:09 +0000101 "\nMODIFIERS (generic):\n"
102 " [c] - do not warn if the library had to be created\n"
103 " [v] - be verbose about actions taken\n"
Reid Spencerbede5832004-11-16 06:41:09 +0000104);
105
Reid Spencer3a1582b2004-11-14 22:20:07 +0000106// This enumeration delineates the kinds of operations on an archive
107// that are permitted.
108enum ArchiveOperation {
Reid Spencer3a1582b2004-11-14 22:20:07 +0000109 Print, ///< Print the contents of the archive
110 Delete, ///< Delete the specified members
111 Move, ///< Move members to end or as given by {a,b,i} modifiers
112 QuickAppend, ///< Quickly append to end of archive
113 ReplaceOrInsert, ///< Replace or Insert members
114 DisplayTable, ///< Display the table of contents
Rafael Espindolacf48cf22013-07-29 12:40:31 +0000115 Extract, ///< Extract files back to file system
116 CreateSymTab ///< Create a symbol table in an existing archive
Tanya Lattner57bd7962003-12-06 23:01:25 +0000117};
Tanya Lattner14baebf2003-08-28 15:22:38 +0000118
Reid Spencer3a1582b2004-11-14 22:20:07 +0000119// Modifiers to follow operation to vary behavior
Rafael Espindola2494dfc2013-07-12 16:29:27 +0000120static bool AddAfter = false; ///< 'a' modifier
121static bool AddBefore = false; ///< 'b' modifier
122static bool Create = false; ///< 'c' modifier
123static bool OriginalDates = false; ///< 'o' modifier
124static bool OnlyUpdate = false; ///< 'u' modifier
125static bool Verbose = false; ///< 'v' modifier
Rafael Espindola26b10da2013-07-23 10:47:01 +0000126static bool Symtab = true; ///< 's' modifier
Tanya Lattner14baebf2003-08-28 15:22:38 +0000127
Reid Spencer3a1582b2004-11-14 22:20:07 +0000128// Relative Positional Argument (for insert/move). This variable holds
129// the name of the archive member to which the 'a', 'b' or 'i' modifier
130// refers. Only one of 'a', 'b' or 'i' can be specified so we only need
131// one variable.
Rafael Espindola2494dfc2013-07-12 16:29:27 +0000132static std::string RelPos;
Tanya Lattner57bd7962003-12-06 23:01:25 +0000133
Reid Spencer3a1582b2004-11-14 22:20:07 +0000134// This variable holds the name of the archive file as given on the
135// command line.
Rafael Espindola2494dfc2013-07-12 16:29:27 +0000136static std::string ArchiveName;
Tanya Lattner57bd7962003-12-06 23:01:25 +0000137
Reid Spencer3a1582b2004-11-14 22:20:07 +0000138// This variable holds the list of member files to proecess, as given
139// on the command line.
Rafael Espindola2494dfc2013-07-12 16:29:27 +0000140static std::vector<std::string> Members;
Tanya Lattner57bd7962003-12-06 23:01:25 +0000141
Joerg Sonnenberger975bc072012-10-26 10:49:15 +0000142// show_help - Show the error message, the help message and exit.
143LLVM_ATTRIBUTE_NORETURN static void
144show_help(const std::string &msg) {
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000145 errs() << ToolName << ": " << msg << "\n\n";
Joerg Sonnenberger975bc072012-10-26 10:49:15 +0000146 cl::PrintHelpMessage();
Joerg Sonnenberger975bc072012-10-26 10:49:15 +0000147 std::exit(1);
148}
149
Reid Spencer3a1582b2004-11-14 22:20:07 +0000150// getRelPos - Extract the member filename from the command line for
151// the [relpos] argument associated with a, b, and i modifiers
Rafael Espindola2494dfc2013-07-12 16:29:27 +0000152static void getRelPos() {
Joerg Sonnenberger975bc072012-10-26 10:49:15 +0000153 if(RestOfArgs.size() == 0)
154 show_help("Expected [relpos] for a, b, or i modifier");
155 RelPos = RestOfArgs[0];
156 RestOfArgs.erase(RestOfArgs.begin());
Tanya Lattner57bd7962003-12-06 23:01:25 +0000157}
158
Reid Spencer3a1582b2004-11-14 22:20:07 +0000159// getArchive - Get the archive file name from the command line
Rafael Espindola2494dfc2013-07-12 16:29:27 +0000160static void getArchive() {
Joerg Sonnenberger975bc072012-10-26 10:49:15 +0000161 if(RestOfArgs.size() == 0)
162 show_help("An archive name must be specified");
163 ArchiveName = RestOfArgs[0];
164 RestOfArgs.erase(RestOfArgs.begin());
Tanya Lattner57bd7962003-12-06 23:01:25 +0000165}
166
Reid Spencer3a1582b2004-11-14 22:20:07 +0000167// getMembers - Copy over remaining items in RestOfArgs to our Members vector
168// This is just for clarity.
Rafael Espindola2494dfc2013-07-12 16:29:27 +0000169static void getMembers() {
Reid Spencer3a1582b2004-11-14 22:20:07 +0000170 if(RestOfArgs.size() > 0)
Misha Brukman3da94ae2005-04-22 00:00:37 +0000171 Members = std::vector<std::string>(RestOfArgs);
Tanya Lattner57bd7962003-12-06 23:01:25 +0000172}
173
Reid Spencer3a1582b2004-11-14 22:20:07 +0000174// parseCommandLine - Parse the command line options as presented and return the
Misha Brukman3da94ae2005-04-22 00:00:37 +0000175// operation specified. Process all modifiers and check to make sure that
Reid Spencer3a1582b2004-11-14 22:20:07 +0000176// constraints on modifier/operation pairs have not been violated.
Rafael Espindola2494dfc2013-07-12 16:29:27 +0000177static ArchiveOperation parseCommandLine() {
Tanya Lattner57bd7962003-12-06 23:01:25 +0000178
Reid Spencer3a1582b2004-11-14 22:20:07 +0000179 // Keep track of number of operations. We can only specify one
180 // per execution.
Tanya Lattner57bd7962003-12-06 23:01:25 +0000181 unsigned NumOperations = 0;
182
Reid Spencer3a1582b2004-11-14 22:20:07 +0000183 // Keep track of the number of positional modifiers (a,b,i). Only
184 // one can be specified.
185 unsigned NumPositional = 0;
186
187 // Keep track of which operation was requested
Rafael Espindola87b8a7f92013-07-05 12:12:43 +0000188 ArchiveOperation Operation;
Reid Spencer3a1582b2004-11-14 22:20:07 +0000189
Rafael Espindolacf48cf22013-07-29 12:40:31 +0000190 bool MaybeJustCreateSymTab = false;
191
Tanya Lattner57bd7962003-12-06 23:01:25 +0000192 for(unsigned i=0; i<Options.size(); ++i) {
193 switch(Options[i]) {
Reid Spencer3a1582b2004-11-14 22:20:07 +0000194 case 'd': ++NumOperations; Operation = Delete; break;
195 case 'm': ++NumOperations; Operation = Move ; break;
196 case 'p': ++NumOperations; Operation = Print; break;
Seo Sanghyeondc32d192007-12-25 13:53:47 +0000197 case 'q': ++NumOperations; Operation = QuickAppend; break;
Misha Brukman3da94ae2005-04-22 00:00:37 +0000198 case 'r': ++NumOperations; Operation = ReplaceOrInsert; break;
Reid Spencer3a1582b2004-11-14 22:20:07 +0000199 case 't': ++NumOperations; Operation = DisplayTable; break;
200 case 'x': ++NumOperations; Operation = Extract; break;
201 case 'c': Create = true; break;
Reid Spencer3a1582b2004-11-14 22:20:07 +0000202 case 'l': /* accepted but unused */ break;
203 case 'o': OriginalDates = true; break;
Rafael Espindola26b10da2013-07-23 10:47:01 +0000204 case 's':
205 Symtab = true;
Rafael Espindolacf48cf22013-07-29 12:40:31 +0000206 MaybeJustCreateSymTab = true;
Rafael Espindola26b10da2013-07-23 10:47:01 +0000207 break;
208 case 'S':
209 Symtab = false;
210 break;
Reid Spencer3a1582b2004-11-14 22:20:07 +0000211 case 'u': OnlyUpdate = true; break;
212 case 'v': Verbose = true; break;
Tanya Lattner57bd7962003-12-06 23:01:25 +0000213 case 'a':
Tanya Lattner57bd7962003-12-06 23:01:25 +0000214 getRelPos();
Reid Spencer3a1582b2004-11-14 22:20:07 +0000215 AddAfter = true;
216 NumPositional++;
Tanya Lattner57bd7962003-12-06 23:01:25 +0000217 break;
218 case 'b':
Tanya Lattner57bd7962003-12-06 23:01:25 +0000219 getRelPos();
Reid Spencer3a1582b2004-11-14 22:20:07 +0000220 AddBefore = true;
221 NumPositional++;
Tanya Lattner57bd7962003-12-06 23:01:25 +0000222 break;
223 case 'i':
Tanya Lattner57bd7962003-12-06 23:01:25 +0000224 getRelPos();
Rafael Espindola1b3e3ee2013-07-11 15:54:53 +0000225 AddBefore = true;
Reid Spencer3a1582b2004-11-14 22:20:07 +0000226 NumPositional++;
Tanya Lattner57bd7962003-12-06 23:01:25 +0000227 break;
Tanya Lattner57bd7962003-12-06 23:01:25 +0000228 default:
Reid Spencerbede5832004-11-16 06:41:09 +0000229 cl::PrintHelpMessage();
Tanya Lattner57bd7962003-12-06 23:01:25 +0000230 }
231 }
232
Misha Brukman3da94ae2005-04-22 00:00:37 +0000233 // At this point, the next thing on the command line must be
Reid Spencer3a1582b2004-11-14 22:20:07 +0000234 // the archive name.
Tanya Lattner57bd7962003-12-06 23:01:25 +0000235 getArchive();
Reid Spencer3a1582b2004-11-14 22:20:07 +0000236
237 // Everything on the command line at this point is a member.
Tanya Lattner57bd7962003-12-06 23:01:25 +0000238 getMembers();
239
Rafael Espindolacf48cf22013-07-29 12:40:31 +0000240 if (NumOperations == 0 && MaybeJustCreateSymTab) {
241 NumOperations = 1;
242 Operation = CreateSymTab;
243 if (!Members.empty())
244 show_help("The s operation takes only an archive as argument");
245 }
246
Reid Spencer3a1582b2004-11-14 22:20:07 +0000247 // Perform various checks on the operation/modifier specification
248 // to make sure we are dealing with a legal request.
249 if (NumOperations == 0)
Joerg Sonnenberger975bc072012-10-26 10:49:15 +0000250 show_help("You must specify at least one of the operations");
Reid Spencer3a1582b2004-11-14 22:20:07 +0000251 if (NumOperations > 1)
Joerg Sonnenberger975bc072012-10-26 10:49:15 +0000252 show_help("Only one operation may be specified");
Reid Spencer3a1582b2004-11-14 22:20:07 +0000253 if (NumPositional > 1)
Joerg Sonnenberger975bc072012-10-26 10:49:15 +0000254 show_help("You may only specify one of a, b, and i modifiers");
Rafael Espindola1b3e3ee2013-07-11 15:54:53 +0000255 if (AddAfter || AddBefore) {
Reid Spencer3a1582b2004-11-14 22:20:07 +0000256 if (Operation != Move && Operation != ReplaceOrInsert)
Joerg Sonnenberger975bc072012-10-26 10:49:15 +0000257 show_help("The 'a', 'b' and 'i' modifiers can only be specified with "
258 "the 'm' or 'r' operations");
259 }
Reid Spencer3a1582b2004-11-14 22:20:07 +0000260 if (OriginalDates && Operation != Extract)
Joerg Sonnenberger975bc072012-10-26 10:49:15 +0000261 show_help("The 'o' modifier is only applicable to the 'x' operation");
Reid Spencer3a1582b2004-11-14 22:20:07 +0000262 if (OnlyUpdate && Operation != ReplaceOrInsert)
Joerg Sonnenberger975bc072012-10-26 10:49:15 +0000263 show_help("The 'u' modifier is only applicable to the 'r' operation");
Reid Spencer3a1582b2004-11-14 22:20:07 +0000264
265 // Return the parsed operation to the caller
266 return Operation;
Tanya Lattner57bd7962003-12-06 23:01:25 +0000267}
Tanya Lattner14baebf2003-08-28 15:22:38 +0000268
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000269// Implements the 'p' operation. This function traverses the archive
270// looking for members that match the path list.
271static void doPrint(StringRef Name, object::Archive::child_iterator I) {
272 if (Verbose)
273 outs() << "Printing " << Name << "\n";
Rafael Espindola8fe960e2013-06-19 14:58:16 +0000274
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000275 StringRef Data = I->getBuffer();
276 outs().write(Data.data(), Data.size());
Reid Spencer3a1582b2004-11-14 22:20:07 +0000277}
278
279// putMode - utility function for printing out the file mode when the 't'
280// operation is in verbose mode.
Rafael Espindola2494dfc2013-07-12 16:29:27 +0000281static void printMode(unsigned mode) {
Misha Brukman3da94ae2005-04-22 00:00:37 +0000282 if (mode & 004)
Chris Lattner424a04e2010-11-29 23:02:20 +0000283 outs() << "r";
Reid Spencer3a1582b2004-11-14 22:20:07 +0000284 else
Chris Lattner424a04e2010-11-29 23:02:20 +0000285 outs() << "-";
Reid Spencer3a1582b2004-11-14 22:20:07 +0000286 if (mode & 002)
Chris Lattner424a04e2010-11-29 23:02:20 +0000287 outs() << "w";
Reid Spencer3a1582b2004-11-14 22:20:07 +0000288 else
Chris Lattner424a04e2010-11-29 23:02:20 +0000289 outs() << "-";
Reid Spencer3a1582b2004-11-14 22:20:07 +0000290 if (mode & 001)
Chris Lattner424a04e2010-11-29 23:02:20 +0000291 outs() << "x";
Reid Spencer3a1582b2004-11-14 22:20:07 +0000292 else
Chris Lattner424a04e2010-11-29 23:02:20 +0000293 outs() << "-";
Reid Spencer3a1582b2004-11-14 22:20:07 +0000294}
295
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000296// Implement the 't' operation. This function prints out just
Reid Spencer3a1582b2004-11-14 22:20:07 +0000297// the file names of each of the members. However, if verbose mode is requested
298// ('v' modifier) then the file type, permission mode, user, group, size, and
299// modification time are also printed.
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000300static void doDisplayTable(StringRef Name, object::Archive::child_iterator I) {
301 if (Verbose) {
302 sys::fs::perms Mode = I->getAccessMode();
303 printMode((Mode >> 6) & 007);
304 printMode((Mode >> 3) & 007);
305 printMode(Mode & 007);
306 outs() << ' ' << I->getUID();
307 outs() << '/' << I->getGID();
308 outs() << ' ' << format("%6llu", I->getSize());
309 outs() << ' ' << I->getLastModified().str();
310 outs() << ' ';
Reid Spencer3a1582b2004-11-14 22:20:07 +0000311 }
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000312 outs() << Name << "\n";
Reid Spencer3a1582b2004-11-14 22:20:07 +0000313}
314
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000315// Implement the 'x' operation. This function extracts files back to the file
316// system.
317static void doExtract(StringRef Name, object::Archive::child_iterator I) {
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000318 // Retain the original mode.
319 sys::fs::perms Mode = I->getAccessMode();
Rafael Espindolac1b49b52013-07-16 19:44:17 +0000320 SmallString<128> Storage = Name;
Rafael Espindola35637fc2013-07-08 16:16:51 +0000321
Rafael Espindolac1b49b52013-07-16 19:44:17 +0000322 int FD;
323 failIfError(
NAKAMURA Takumi946dbd82013-07-17 12:31:50 +0000324 sys::fs::openFileForWrite(Storage.c_str(), FD, sys::fs::F_Binary, Mode),
Rafael Espindolac1b49b52013-07-16 19:44:17 +0000325 Storage.c_str());
Reid Spencer3a1582b2004-11-14 22:20:07 +0000326
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000327 {
328 raw_fd_ostream file(FD, false);
Reid Spencer3a1582b2004-11-14 22:20:07 +0000329
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000330 // Get the data and its length
331 StringRef Data = I->getBuffer();
Rafael Espindola11ca2e52013-06-20 20:56:14 +0000332
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000333 // Write the data.
334 file.write(Data.data(), Data.size());
Reid Spencer3a1582b2004-11-14 22:20:07 +0000335 }
336
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000337 // If we're supposed to retain the original modification times, etc. do so
338 // now.
339 if (OriginalDates)
340 failIfError(
341 sys::fs::setLastModificationAndAccessTime(FD, I->getLastModified()));
Reid Spencer3a1582b2004-11-14 22:20:07 +0000342
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000343 if (close(FD))
344 fail("Could not close the file");
Reid Spencer3a1582b2004-11-14 22:20:07 +0000345}
346
Rafael Espindola2494dfc2013-07-12 16:29:27 +0000347static bool shouldCreateArchive(ArchiveOperation Op) {
Rafael Espindola61de1422013-07-05 13:03:07 +0000348 switch (Op) {
349 case Print:
350 case Delete:
351 case Move:
352 case DisplayTable:
353 case Extract:
Rafael Espindolacf48cf22013-07-29 12:40:31 +0000354 case CreateSymTab:
Rafael Espindola61de1422013-07-05 13:03:07 +0000355 return false;
356
357 case QuickAppend:
358 case ReplaceOrInsert:
359 return true;
360 }
Michael Gottesman2be430d2013-07-06 02:39:51 +0000361
362 llvm_unreachable("Missing entry in covered switch.");
Rafael Espindola61de1422013-07-05 13:03:07 +0000363}
364
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000365static void performReadOperation(ArchiveOperation Operation,
366 object::Archive *OldArchive) {
367 for (object::Archive::child_iterator I = OldArchive->begin_children(),
368 E = OldArchive->end_children();
369 I != E; ++I) {
370 StringRef Name;
371 failIfError(I->getName(Name));
372
373 if (!Members.empty() &&
374 std::find(Members.begin(), Members.end(), Name) == Members.end())
375 continue;
376
377 switch (Operation) {
378 default:
379 llvm_unreachable("Not a read operation");
380 case Print:
381 doPrint(Name, I);
382 break;
383 case DisplayTable:
384 doDisplayTable(Name, I);
385 break;
386 case Extract:
387 doExtract(Name, I);
388 break;
389 }
390 }
391}
392
393namespace {
394class NewArchiveIterator {
395 bool IsNewMember;
Rafael Espindola26b10da2013-07-23 10:47:01 +0000396 StringRef Name;
Rafael Espindola33ccfb22013-07-12 20:28:02 +0000397 object::Archive::child_iterator OldI;
Rafael Espindola51392a02013-07-22 15:11:51 +0000398 std::string NewFilename;
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000399
400public:
Rafael Espindola26b10da2013-07-23 10:47:01 +0000401 NewArchiveIterator(object::Archive::child_iterator I, StringRef Name);
402 NewArchiveIterator(std::string *I, StringRef Name);
Rafael Espindolab55dcfe2013-07-19 21:23:28 +0000403 NewArchiveIterator();
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000404 bool isNewMember() const;
405 object::Archive::child_iterator getOld() const;
Rafael Espindola289e2412013-07-16 03:30:10 +0000406 const char *getNew() const;
Rafael Espindola26b10da2013-07-23 10:47:01 +0000407 StringRef getName() const;
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000408};
409}
410
Rafael Espindolab55dcfe2013-07-19 21:23:28 +0000411NewArchiveIterator::NewArchiveIterator() {}
412
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000413NewArchiveIterator::NewArchiveIterator(object::Archive::child_iterator I,
Rafael Espindola26b10da2013-07-23 10:47:01 +0000414 StringRef Name)
415 : IsNewMember(false), Name(Name), OldI(I) {}
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000416
Rafael Espindola26b10da2013-07-23 10:47:01 +0000417NewArchiveIterator::NewArchiveIterator(std::string *NewFilename, StringRef Name)
418 : IsNewMember(true), Name(Name), NewFilename(*NewFilename) {}
419
420StringRef NewArchiveIterator::getName() const { return Name; }
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000421
422bool NewArchiveIterator::isNewMember() const { return IsNewMember; }
423
424object::Archive::child_iterator NewArchiveIterator::getOld() const {
425 assert(!IsNewMember);
426 return OldI;
427}
428
Rafael Espindola289e2412013-07-16 03:30:10 +0000429const char *NewArchiveIterator::getNew() const {
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000430 assert(IsNewMember);
Rafael Espindola51392a02013-07-22 15:11:51 +0000431 return NewFilename.c_str();
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000432}
433
434template <typename T>
Rafael Espindola26b10da2013-07-23 10:47:01 +0000435void addMember(std::vector<NewArchiveIterator> &Members, T I, StringRef Name,
436 int Pos = -1) {
437 NewArchiveIterator NI(I, Name);
438 if (Pos == -1)
439 Members.push_back(NI);
440 else
Rafael Espindola51392a02013-07-22 15:11:51 +0000441 Members[Pos] = NI;
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000442}
443
444namespace {
445class HasName {
446 StringRef Name;
447
448public:
449 HasName(StringRef Name) : Name(Name) {}
450 bool operator()(StringRef Path) { return Name == sys::path::filename(Path); }
451};
452}
453
Rafael Espindola51392a02013-07-22 15:11:51 +0000454enum InsertAction {
455 IA_AddOldMember,
456 IA_AddNewMeber,
457 IA_Delete,
458 IA_MoveOldMember,
459 IA_MoveNewMember
460};
461
462static InsertAction
463computeInsertAction(ArchiveOperation Operation,
464 object::Archive::child_iterator I, StringRef Name,
465 std::vector<std::string>::iterator &Pos) {
466 if (Operation == QuickAppend || Members.empty())
467 return IA_AddOldMember;
468
469 std::vector<std::string>::iterator MI =
470 std::find_if(Members.begin(), Members.end(), HasName(Name));
471
472 if (MI == Members.end())
473 return IA_AddOldMember;
474
475 Pos = MI;
476
477 if (Operation == Delete)
478 return IA_Delete;
479
480 if (Operation == Move)
481 return IA_MoveOldMember;
482
483 if (Operation == ReplaceOrInsert) {
484 StringRef PosName = sys::path::filename(RelPos);
485 if (!OnlyUpdate) {
486 if (PosName.empty())
487 return IA_AddNewMeber;
488 return IA_MoveNewMember;
489 }
490
491 // We could try to optimize this to a fstat, but it is not a common
492 // operation.
493 sys::fs::file_status Status;
494 failIfError(sys::fs::status(*MI, Status));
495 if (Status.getLastModificationTime() < I->getLastModified()) {
496 if (PosName.empty())
497 return IA_AddOldMember;
498 return IA_MoveOldMember;
499 }
500
501 if (PosName.empty())
502 return IA_AddNewMeber;
503 return IA_MoveNewMember;
504 }
505 llvm_unreachable("No such operation");
506}
507
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000508// We have to walk this twice and computing it is not trivial, so creating an
509// explicit std::vector is actually fairly efficient.
510static std::vector<NewArchiveIterator>
511computeNewArchiveMembers(ArchiveOperation Operation,
Rafael Espindola26b10da2013-07-23 10:47:01 +0000512 object::Archive *OldArchive) {
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000513 std::vector<NewArchiveIterator> Ret;
514 std::vector<NewArchiveIterator> Moved;
515 int InsertPos = -1;
516 StringRef PosName = sys::path::filename(RelPos);
517 if (OldArchive) {
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000518 for (object::Archive::child_iterator I = OldArchive->begin_children(),
519 E = OldArchive->end_children();
Rafael Espindola02a13642013-07-21 12:58:07 +0000520 I != E; ++I) {
521 int Pos = Ret.size();
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000522 StringRef Name;
523 failIfError(I->getName(Name));
524 if (Name == PosName) {
525 assert(AddAfter || AddBefore);
526 if (AddBefore)
527 InsertPos = Pos;
528 else
529 InsertPos = Pos + 1;
530 }
Rafael Espindola51392a02013-07-22 15:11:51 +0000531
532 std::vector<std::string>::iterator MemberI = Members.end();
533 InsertAction Action = computeInsertAction(Operation, I, Name, MemberI);
534 switch (Action) {
535 case IA_AddOldMember:
Rafael Espindola26b10da2013-07-23 10:47:01 +0000536 addMember(Ret, I, Name);
Rafael Espindola51392a02013-07-22 15:11:51 +0000537 break;
538 case IA_AddNewMeber:
Rafael Espindola26b10da2013-07-23 10:47:01 +0000539 addMember(Ret, &*MemberI, Name);
Rafael Espindola51392a02013-07-22 15:11:51 +0000540 break;
541 case IA_Delete:
542 break;
543 case IA_MoveOldMember:
Rafael Espindola26b10da2013-07-23 10:47:01 +0000544 addMember(Moved, I, Name);
Rafael Espindola51392a02013-07-22 15:11:51 +0000545 break;
546 case IA_MoveNewMember:
Rafael Espindola26b10da2013-07-23 10:47:01 +0000547 addMember(Moved, &*MemberI, Name);
Rafael Espindola51392a02013-07-22 15:11:51 +0000548 break;
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000549 }
Rafael Espindola51392a02013-07-22 15:11:51 +0000550 if (MemberI != Members.end())
551 Members.erase(MemberI);
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000552 }
553 }
554
555 if (Operation == Delete)
556 return Ret;
557
Rafael Espindolab55dcfe2013-07-19 21:23:28 +0000558 if (!RelPos.empty() && InsertPos == -1)
559 fail("Insertion point not found");
560
Rafael Espindola51392a02013-07-22 15:11:51 +0000561 if (RelPos.empty())
562 InsertPos = Ret.size();
Rafael Espindolab55dcfe2013-07-19 21:23:28 +0000563
Rafael Espindola51392a02013-07-22 15:11:51 +0000564 assert(unsigned(InsertPos) <= Ret.size());
565 Ret.insert(Ret.begin() + InsertPos, Moved.begin(), Moved.end());
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000566
Rafael Espindola51392a02013-07-22 15:11:51 +0000567 Ret.insert(Ret.begin() + InsertPos, Members.size(), NewArchiveIterator());
Rafael Espindolab55dcfe2013-07-19 21:23:28 +0000568 int Pos = InsertPos;
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000569 for (std::vector<std::string>::iterator I = Members.begin(),
Rafael Espindola51392a02013-07-22 15:11:51 +0000570 E = Members.end();
571 I != E; ++I, ++Pos) {
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000572 StringRef Name = sys::path::filename(*I);
Rafael Espindola26b10da2013-07-23 10:47:01 +0000573 addMember(Ret, &*I, Name, Pos);
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000574 }
575
576 return Ret;
577}
578
579template <typename T>
580static void printWithSpacePadding(raw_ostream &OS, T Data, unsigned Size) {
581 uint64_t OldPos = OS.tell();
582 OS << Data;
583 unsigned SizeSoFar = OS.tell() - OldPos;
584 assert(Size >= SizeSoFar && "Data doesn't fit in Size");
585 unsigned Remaining = Size - SizeSoFar;
586 for (unsigned I = 0; I < Remaining; ++I)
587 OS << ' ';
588}
589
Rafael Espindola26b10da2013-07-23 10:47:01 +0000590static void print32BE(raw_fd_ostream &Out, unsigned Val) {
591 for (int I = 3; I >= 0; --I) {
592 char V = (Val >> (8 * I)) & 0xff;
593 Out << V;
594 }
595}
596
597static void printRestOfMemberHeader(raw_fd_ostream &Out,
598 const sys::TimeValue &ModTime, unsigned UID,
599 unsigned GID, unsigned Perms,
600 unsigned Size) {
601 printWithSpacePadding(Out, ModTime.toEpochTime(), 12);
602 printWithSpacePadding(Out, UID, 6);
603 printWithSpacePadding(Out, GID, 6);
604 printWithSpacePadding(Out, format("%o", Perms), 8);
605 printWithSpacePadding(Out, Size, 10);
606 Out << "`\n";
607}
608
609static void printMemberHeader(raw_fd_ostream &Out, StringRef Name,
610 const sys::TimeValue &ModTime, unsigned UID,
611 unsigned GID, unsigned Perms, unsigned Size) {
612 printWithSpacePadding(Out, Twine(Name) + "/", 16);
613 printRestOfMemberHeader(Out, ModTime, UID, GID, Perms, Size);
614}
615
616static void printMemberHeader(raw_fd_ostream &Out, unsigned NameOffset,
617 const sys::TimeValue &ModTime, unsigned UID,
618 unsigned GID, unsigned Perms, unsigned Size) {
619 Out << '/';
620 printWithSpacePadding(Out, NameOffset, 15);
621 printRestOfMemberHeader(Out, ModTime, UID, GID, Perms, Size);
622}
623
624static void writeStringTable(raw_fd_ostream &Out,
625 ArrayRef<NewArchiveIterator> Members,
626 std::vector<unsigned> &StringMapIndexes) {
627 unsigned StartOffset = 0;
628 for (ArrayRef<NewArchiveIterator>::iterator I = Members.begin(),
629 E = Members.end();
630 I != E; ++I) {
631 StringRef Name = I->getName();
632 if (Name.size() < 16)
633 continue;
634 if (StartOffset == 0) {
635 printWithSpacePadding(Out, "//", 58);
636 Out << "`\n";
637 StartOffset = Out.tell();
638 }
639 StringMapIndexes.push_back(Out.tell() - StartOffset);
640 Out << Name << "/\n";
641 }
642 if (StartOffset == 0)
643 return;
644 if (Out.tell() % 2)
645 Out << '\n';
646 int Pos = Out.tell();
647 Out.seek(StartOffset - 12);
648 printWithSpacePadding(Out, Pos - StartOffset, 10);
649 Out.seek(Pos);
650}
651
652static void writeSymbolTable(
653 raw_fd_ostream &Out, ArrayRef<NewArchiveIterator> Members,
654 std::vector<std::pair<unsigned, unsigned> > &MemberOffsetRefs) {
655 unsigned StartOffset = 0;
656 unsigned MemberNum = 0;
657 std::vector<StringRef> SymNames;
Rafael Espindolaaf02bf42013-07-23 11:02:16 +0000658 std::vector<object::ObjectFile *> DeleteIt;
Rafael Espindola26b10da2013-07-23 10:47:01 +0000659 for (ArrayRef<NewArchiveIterator>::iterator I = Members.begin(),
660 E = Members.end();
661 I != E; ++I, ++MemberNum) {
662 object::ObjectFile *Obj;
663 if (I->isNewMember()) {
664 const char *Filename = I->getNew();
665 Obj = object::ObjectFile::createObjectFile(Filename);
666 } else {
667 object::Archive::child_iterator OldMember = I->getOld();
668 OwningPtr<object::Binary> Binary;
669 error_code EC = OldMember->getAsBinary(Binary);
670 if (EC) { // FIXME: check only for "not an object file" errors.
671 Obj = NULL;
672 } else {
673 Obj = dyn_cast<object::ObjectFile>(Binary.get());
674 if (Obj)
675 Binary.take();
676 }
677 }
678 if (!Obj)
679 continue;
Rafael Espindolaaf02bf42013-07-23 11:02:16 +0000680 DeleteIt.push_back(Obj);
Rafael Espindola26b10da2013-07-23 10:47:01 +0000681 if (!StartOffset) {
682 printMemberHeader(Out, "", sys::TimeValue::now(), 0, 0, 0, 0);
683 StartOffset = Out.tell();
684 print32BE(Out, 0);
685 }
686
687 error_code Err;
688 for (object::symbol_iterator I = Obj->begin_symbols(),
689 E = Obj->end_symbols();
690 I != E; I.increment(Err), failIfError(Err)) {
691 uint32_t Symflags;
692 failIfError(I->getFlags(Symflags));
693 if (Symflags & object::SymbolRef::SF_FormatSpecific)
694 continue;
695 if (!(Symflags & object::SymbolRef::SF_Global))
696 continue;
697 if (Symflags & object::SymbolRef::SF_Undefined)
698 continue;
699 StringRef Name;
700 failIfError(I->getName(Name));
701 SymNames.push_back(Name);
702 MemberOffsetRefs.push_back(std::make_pair(Out.tell(), MemberNum));
703 print32BE(Out, 0);
704 }
705 }
706 for (std::vector<StringRef>::iterator I = SymNames.begin(),
707 E = SymNames.end();
708 I != E; ++I) {
709 Out << *I;
710 Out << '\0';
711 }
712
Rafael Espindolaaf02bf42013-07-23 11:02:16 +0000713 for (std::vector<object::ObjectFile *>::iterator I = DeleteIt.begin(),
714 E = DeleteIt.end();
715 I != E; ++I) {
716 object::ObjectFile *O = *I;
717 delete O;
718 }
719
Rafael Espindola26b10da2013-07-23 10:47:01 +0000720 if (StartOffset == 0)
721 return;
722
723 if (Out.tell() % 2)
724 Out << '\0';
725
726 unsigned Pos = Out.tell();
727 Out.seek(StartOffset - 12);
728 printWithSpacePadding(Out, Pos - StartOffset, 10);
729 Out.seek(StartOffset);
730 print32BE(Out, SymNames.size());
731 Out.seek(Pos);
732}
733
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000734static void performWriteOperation(ArchiveOperation Operation,
735 object::Archive *OldArchive) {
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000736 SmallString<128> TmpArchive;
737 failIfError(sys::fs::createUniqueFile(ArchiveName + ".temp-archive-%%%%%%%.a",
738 TmpArchiveFD, TmpArchive));
739
740 TemporaryOutput = TmpArchive.c_str();
741 tool_output_file Output(TemporaryOutput, TmpArchiveFD);
742 raw_fd_ostream &Out = Output.os();
743 Out << "!<arch>\n";
744
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000745 std::vector<NewArchiveIterator> NewMembers =
Rafael Espindola26b10da2013-07-23 10:47:01 +0000746 computeNewArchiveMembers(Operation, OldArchive);
747
748 std::vector<std::pair<unsigned, unsigned> > MemberOffsetRefs;
749
750 if (Symtab) {
751 writeSymbolTable(Out, NewMembers, MemberOffsetRefs);
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000752 }
753
Rafael Espindola26b10da2013-07-23 10:47:01 +0000754 std::vector<unsigned> StringMapIndexes;
755 writeStringTable(Out, NewMembers, StringMapIndexes);
756
757 std::vector<std::pair<unsigned, unsigned> >::iterator MemberRefsI =
758 MemberOffsetRefs.begin();
759
760 unsigned MemberNum = 0;
761 unsigned LongNameMemberNum = 0;
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000762 for (std::vector<NewArchiveIterator>::iterator I = NewMembers.begin(),
763 E = NewMembers.end();
Rafael Espindola26b10da2013-07-23 10:47:01 +0000764 I != E; ++I, ++MemberNum) {
765
766 unsigned Pos = Out.tell();
767 while (MemberRefsI != MemberOffsetRefs.end() &&
768 MemberRefsI->second == MemberNum) {
769 Out.seek(MemberRefsI->first);
770 print32BE(Out, Pos);
771 ++MemberRefsI;
772 }
773 Out.seek(Pos);
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000774
775 if (I->isNewMember()) {
Rafael Espindola289e2412013-07-16 03:30:10 +0000776 const char *FileName = I->getNew();
Rafael Espindolabd6cb262013-07-16 03:34:31 +0000777
Rafael Espindolac1b49b52013-07-16 19:44:17 +0000778 int FD;
779 failIfError(sys::fs::openFileForRead(FileName, FD), FileName);
Rafael Espindolabd6cb262013-07-16 03:34:31 +0000780
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000781 sys::fs::file_status Status;
Rafael Espindolabd6cb262013-07-16 03:34:31 +0000782 failIfError(sys::fs::status(FD, Status), FileName);
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000783
Rafael Espindola718af782013-07-13 05:07:22 +0000784 OwningPtr<MemoryBuffer> File;
Rafael Espindola70c7e482013-07-23 20:25:01 +0000785 failIfError(MemoryBuffer::getOpenFile(FD, FileName, File,
786 Status.getSize(), false),
787 FileName);
Rafael Espindola718af782013-07-13 05:07:22 +0000788
Rafael Espindola26b10da2013-07-23 10:47:01 +0000789 StringRef Name = sys::path::filename(FileName);
790 if (Name.size() < 16)
791 printMemberHeader(Out, Name, Status.getLastModificationTime(),
792 Status.getUser(), Status.getGroup(),
793 Status.permissions(), Status.getSize());
794 else
795 printMemberHeader(Out, StringMapIndexes[LongNameMemberNum++],
796 Status.getLastModificationTime(), Status.getUser(),
797 Status.getGroup(), Status.permissions(),
798 Status.getSize());
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000799 Out << File->getBuffer();
800 } else {
801 object::Archive::child_iterator OldMember = I->getOld();
Rafael Espindola26b10da2013-07-23 10:47:01 +0000802 StringRef Name = I->getName();
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000803
Rafael Espindola26b10da2013-07-23 10:47:01 +0000804 if (Name.size() < 16)
805 printMemberHeader(Out, Name, OldMember->getLastModified(),
806 OldMember->getUID(), OldMember->getGID(),
807 OldMember->getAccessMode(), OldMember->getSize());
808 else
809 printMemberHeader(Out, StringMapIndexes[LongNameMemberNum++],
810 OldMember->getLastModified(), OldMember->getUID(),
811 OldMember->getGID(), OldMember->getAccessMode(),
812 OldMember->getSize());
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000813 Out << OldMember->getBuffer();
814 }
815
816 if (Out.tell() % 2)
817 Out << '\n';
818 }
819 Output.keep();
820 Out.close();
821 sys::fs::rename(TemporaryOutput, ArchiveName);
822 TemporaryOutput = NULL;
823}
824
Rafael Espindolacf48cf22013-07-29 12:40:31 +0000825static void createSymbolTable(object::Archive *OldArchive) {
826 // When an archive is created or modified, if the s option is given, the
827 // resulting archive will have a current symbol table. If the S option
828 // is given, it will have no symbol table.
829 // In summary, we only need to update the symbol table if we have none.
830 // This is actually very common because of broken build systems that think
831 // they have to run ranlib.
832 if (OldArchive->hasSymbolTable())
833 return;
834
835 performWriteOperation(CreateSymTab, OldArchive);
836}
837
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000838static void performOperation(ArchiveOperation Operation,
839 object::Archive *OldArchive) {
840 switch (Operation) {
841 case Print:
842 case DisplayTable:
843 case Extract:
844 performReadOperation(Operation, OldArchive);
845 return;
846
847 case Delete:
848 case Move:
849 case QuickAppend:
850 case ReplaceOrInsert:
851 performWriteOperation(Operation, OldArchive);
852 return;
Rafael Espindolacf48cf22013-07-29 12:40:31 +0000853 case CreateSymTab:
854 createSymbolTable(OldArchive);
855 return;
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000856 }
857 llvm_unreachable("Unknown operation.");
858}
859
Reid Spencer3a1582b2004-11-14 22:20:07 +0000860// main - main program for llvm-ar .. see comments in the code
Tanya Lattner14baebf2003-08-28 15:22:38 +0000861int main(int argc, char **argv) {
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000862 ToolName = argv[0];
Chris Lattnercc14d252009-03-06 05:34:10 +0000863 // Print a stack trace if we signal out.
864 sys::PrintStackTraceOnErrorSignal();
865 PrettyStackTraceProgram X(argc, argv);
866 llvm_shutdown_obj Y; // Call llvm_shutdown() on exit.
Reid Spencer3a1582b2004-11-14 22:20:07 +0000867
Reid Spencer3a1582b2004-11-14 22:20:07 +0000868 // Have the command line options parsed and handle things
869 // like --help and --version.
870 cl::ParseCommandLineOptions(argc, argv,
Dan Gohman82a13c92007-10-08 15:45:12 +0000871 "LLVM Archiver (llvm-ar)\n\n"
Gabor Greifa99be512007-07-05 17:07:56 +0000872 " This program archives bitcode files into single libraries\n"
Reid Spencer3a1582b2004-11-14 22:20:07 +0000873 );
874
Joerg Sonnenberger975bc072012-10-26 10:49:15 +0000875 // Do our own parsing of the command line because the CommandLine utility
876 // can't handle the grouped positional parameters without a dash.
877 ArchiveOperation Operation = parseCommandLine();
Tanya Lattner14baebf2003-08-28 15:22:38 +0000878
Joerg Sonnenberger975bc072012-10-26 10:49:15 +0000879 // Create or open the archive object.
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000880 OwningPtr<MemoryBuffer> Buf;
881 error_code EC = MemoryBuffer::getFile(ArchiveName, Buf, -1, false);
882 if (EC && EC != llvm::errc::no_such_file_or_directory) {
883 errs() << argv[0] << ": error opening '" << ArchiveName
884 << "': " << EC.message() << "!\n";
Joerg Sonnenberger975bc072012-10-26 10:49:15 +0000885 return 1;
886 }
887
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000888 if (!EC) {
889 object::Archive Archive(Buf.take(), EC);
Joerg Sonnenberger975bc072012-10-26 10:49:15 +0000890
Rafael Espindola34ac52d2013-07-12 20:21:39 +0000891 if (EC) {
892 errs() << argv[0] << ": error loading '" << ArchiveName
893 << "': " << EC.message() << "!\n";
894 return 1;
895 }
896 performOperation(Operation, &Archive);
897 return 0;
898 }
899
900 assert(EC == llvm::errc::no_such_file_or_directory);
901
902 if (!shouldCreateArchive(Operation)) {
903 failIfError(EC, Twine("error loading '") + ArchiveName + "'");
904 } else {
905 if (!Create) {
906 // Produce a warning if we should and we're creating the archive
907 errs() << argv[0] << ": creating " << ArchiveName << "\n";
908 }
909 }
910
911 performOperation(Operation, NULL);
912 return 0;
Tanya Lattner14baebf2003-08-28 15:22:38 +0000913}