blob: db95674d36ad8126f559df86419dd468ccff9cda [file] [log] [blame]
Chris Lattnerce8781c2003-12-30 07:45:46 +00001//===-- llvm-ar.cpp - LLVM archive librarian utility ----------------------===//
Misha Brukman650ba8e2005-04-22 00:00:37 +00002//
John Criswell09344dc2003-10-20 17:47:21 +00003// The LLVM Compiler Infrastructure
4//
Chris Lattner345353d2007-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 Brukman650ba8e2005-04-22 00:00:37 +00007//
John Criswell09344dc2003-10-20 17:47:21 +00008//===----------------------------------------------------------------------===//
Tanya Lattner57c70a62003-08-28 15:22:38 +00009//
Misha Brukman650ba8e2005-04-22 00:00:37 +000010// Builds up (relatively) standard unix archive files (.a) containing LLVM
Gabor Greife16561c2007-07-05 17:07:56 +000011// bitcode or other files.
Tanya Lattner57c70a62003-08-28 15:22:38 +000012//
13//===----------------------------------------------------------------------===//
Brian Gaeke81d153e2003-10-10 18:47:08 +000014
Chandler Carruth9fb823b2013-01-02 11:36:10 +000015#include "llvm/IR/LLVMContext.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000016#include "llvm/IR/Module.h"
Rafael Espindola3e2b21c2013-07-12 20:21:39 +000017#include "llvm/Object/Archive.h"
Rafael Espindolace7f52d2013-07-23 10:47:01 +000018#include "llvm/Object/ObjectFile.h"
Reid Spencer7c16caa2004-09-01 22:55:40 +000019#include "llvm/Support/CommandLine.h"
Michael J. Spencer58df2e02011-01-10 02:34:23 +000020#include "llvm/Support/FileSystem.h"
Chandler Carruth4d88a1c2012-12-04 10:44:52 +000021#include "llvm/Support/Format.h"
Chris Lattner76d46322006-12-06 01:18:01 +000022#include "llvm/Support/ManagedStatic.h"
Rafael Espindola3e2b21c2013-07-12 20:21:39 +000023#include "llvm/Support/MemoryBuffer.h"
Chris Lattnere3fc2d12009-03-06 05:34:10 +000024#include "llvm/Support/PrettyStackTrace.h"
Michael J. Spencer447762d2010-11-29 18:16:10 +000025#include "llvm/Support/Signals.h"
Rafael Espindola3e2b21c2013-07-12 20:21:39 +000026#include "llvm/Support/ToolOutputFile.h"
Chandler Carruth4d88a1c2012-12-04 10:44:52 +000027#include "llvm/Support/raw_ostream.h"
Reid Spencer84a12bf2004-11-14 22:20:07 +000028#include <algorithm>
Joerg Sonnenberger4a0f7be2012-10-26 10:49:15 +000029#include <cstdlib>
Chandler Carruth4d88a1c2012-12-04 10:44:52 +000030#include <memory>
Rafael Espindola4a3365c2013-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 Gaeke960707c2003-11-11 22:41:34 +000038using namespace llvm;
39
Rafael Espindola3e2b21c2013-07-12 20:21:39 +000040// The name this program was invoked as.
41static StringRef ToolName;
42
43static const char *TemporaryOutput;
Rafael Espindola7303af32013-07-16 16:00:32 +000044static int TmpArchiveFD = -1;
Rafael Espindola3e2b21c2013-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 Espindola7303af32013-07-16 16:00:32 +000049 if (TmpArchiveFD != -1)
50 close(TmpArchiveFD);
Rafael Espindola3e2b21c2013-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
Rafael Espindola0c8a3522013-08-28 16:22:16 +000066// llvm-ar/llvm-ranlib remaining positional arguments.
Misha Brukman650ba8e2005-04-22 00:00:37 +000067static cl::list<std::string>
68RestOfArgs(cl::Positional, cl::OneOrMore,
Reid Spencer84a12bf2004-11-14 22:20:07 +000069 cl::desc("[relpos] [count] <archive-file> [members]..."));
Tanya Lattner57c70a62003-08-28 15:22:38 +000070
Rafael Espindola0c8a3522013-08-28 16:22:16 +000071std::string Options;
72
Reid Spencer9fc38b12004-11-16 06:41:09 +000073// MoreHelp - Provide additional help output explaining the operations and
74// modifiers of llvm-ar. This object instructs the CommandLine library
75// to print the text of the constructor when the --help option is given.
76static cl::extrahelp MoreHelp(
Misha Brukman650ba8e2005-04-22 00:00:37 +000077 "\nOPERATIONS:\n"
Reid Spencer9fc38b12004-11-16 06:41:09 +000078 " d[NsS] - delete file(s) from the archive\n"
79 " m[abiSs] - move file(s) in the archive\n"
80 " p[kN] - print file(s) found in the archive\n"
81 " q[ufsS] - quick append file(s) to the archive\n"
Rafael Espindola740a6bc2012-08-10 01:57:52 +000082 " r[abfiuRsS] - replace or insert file(s) into the archive\n"
Reid Spencer9fc38b12004-11-16 06:41:09 +000083 " t - display contents of archive\n"
84 " x[No] - extract file(s) from the archive\n"
85 "\nMODIFIERS (operation specific):\n"
86 " [a] - put file(s) after [relpos]\n"
87 " [b] - put file(s) before [relpos] (same as [i])\n"
Reid Spencer9fc38b12004-11-16 06:41:09 +000088 " [i] - put file(s) before [relpos] (same as [b])\n"
Reid Spencer9fc38b12004-11-16 06:41:09 +000089 " [N] - use instance [count] of name\n"
90 " [o] - preserve original dates\n"
Reid Spencer9fc38b12004-11-16 06:41:09 +000091 " [s] - create an archive index (cf. ranlib)\n"
92 " [S] - do not build a symbol table\n"
93 " [u] - update only files newer than archive contents\n"
Reid Spencer9fc38b12004-11-16 06:41:09 +000094 "\nMODIFIERS (generic):\n"
95 " [c] - do not warn if the library had to be created\n"
96 " [v] - be verbose about actions taken\n"
Reid Spencer9fc38b12004-11-16 06:41:09 +000097);
98
Reid Spencer84a12bf2004-11-14 22:20:07 +000099// This enumeration delineates the kinds of operations on an archive
100// that are permitted.
101enum ArchiveOperation {
Reid Spencer84a12bf2004-11-14 22:20:07 +0000102 Print, ///< Print the contents of the archive
103 Delete, ///< Delete the specified members
104 Move, ///< Move members to end or as given by {a,b,i} modifiers
105 QuickAppend, ///< Quickly append to end of archive
106 ReplaceOrInsert, ///< Replace or Insert members
107 DisplayTable, ///< Display the table of contents
Rafael Espindolab6b5f52e2013-07-29 12:40:31 +0000108 Extract, ///< Extract files back to file system
109 CreateSymTab ///< Create a symbol table in an existing archive
Tanya Lattnerc970a382003-12-06 23:01:25 +0000110};
Tanya Lattner57c70a62003-08-28 15:22:38 +0000111
Reid Spencer84a12bf2004-11-14 22:20:07 +0000112// Modifiers to follow operation to vary behavior
Rafael Espindola05571532013-07-12 16:29:27 +0000113static bool AddAfter = false; ///< 'a' modifier
114static bool AddBefore = false; ///< 'b' modifier
115static bool Create = false; ///< 'c' modifier
116static bool OriginalDates = false; ///< 'o' modifier
117static bool OnlyUpdate = false; ///< 'u' modifier
118static bool Verbose = false; ///< 'v' modifier
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000119static bool Symtab = true; ///< 's' modifier
Tanya Lattner57c70a62003-08-28 15:22:38 +0000120
Reid Spencer84a12bf2004-11-14 22:20:07 +0000121// Relative Positional Argument (for insert/move). This variable holds
122// the name of the archive member to which the 'a', 'b' or 'i' modifier
123// refers. Only one of 'a', 'b' or 'i' can be specified so we only need
124// one variable.
Rafael Espindola05571532013-07-12 16:29:27 +0000125static std::string RelPos;
Tanya Lattnerc970a382003-12-06 23:01:25 +0000126
Reid Spencer84a12bf2004-11-14 22:20:07 +0000127// This variable holds the name of the archive file as given on the
128// command line.
Rafael Espindola05571532013-07-12 16:29:27 +0000129static std::string ArchiveName;
Tanya Lattnerc970a382003-12-06 23:01:25 +0000130
Reid Spencer84a12bf2004-11-14 22:20:07 +0000131// This variable holds the list of member files to proecess, as given
132// on the command line.
Rafael Espindola05571532013-07-12 16:29:27 +0000133static std::vector<std::string> Members;
Tanya Lattnerc970a382003-12-06 23:01:25 +0000134
Joerg Sonnenberger4a0f7be2012-10-26 10:49:15 +0000135// show_help - Show the error message, the help message and exit.
136LLVM_ATTRIBUTE_NORETURN static void
137show_help(const std::string &msg) {
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000138 errs() << ToolName << ": " << msg << "\n\n";
Joerg Sonnenberger4a0f7be2012-10-26 10:49:15 +0000139 cl::PrintHelpMessage();
Joerg Sonnenberger4a0f7be2012-10-26 10:49:15 +0000140 std::exit(1);
141}
142
Reid Spencer84a12bf2004-11-14 22:20:07 +0000143// getRelPos - Extract the member filename from the command line for
144// the [relpos] argument associated with a, b, and i modifiers
Rafael Espindola05571532013-07-12 16:29:27 +0000145static void getRelPos() {
Joerg Sonnenberger4a0f7be2012-10-26 10:49:15 +0000146 if(RestOfArgs.size() == 0)
147 show_help("Expected [relpos] for a, b, or i modifier");
148 RelPos = RestOfArgs[0];
149 RestOfArgs.erase(RestOfArgs.begin());
Tanya Lattnerc970a382003-12-06 23:01:25 +0000150}
151
Rafael Espindola0c8a3522013-08-28 16:22:16 +0000152static void getOptions() {
153 if(RestOfArgs.size() == 0)
154 show_help("Expected options");
155 Options = RestOfArgs[0];
156 RestOfArgs.erase(RestOfArgs.begin());
157}
158
Reid Spencer84a12bf2004-11-14 22:20:07 +0000159// getArchive - Get the archive file name from the command line
Rafael Espindola05571532013-07-12 16:29:27 +0000160static void getArchive() {
Joerg Sonnenberger4a0f7be2012-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 Lattnerc970a382003-12-06 23:01:25 +0000165}
166
Reid Spencer84a12bf2004-11-14 22:20:07 +0000167// getMembers - Copy over remaining items in RestOfArgs to our Members vector
168// This is just for clarity.
Rafael Espindola05571532013-07-12 16:29:27 +0000169static void getMembers() {
Reid Spencer84a12bf2004-11-14 22:20:07 +0000170 if(RestOfArgs.size() > 0)
Misha Brukman650ba8e2005-04-22 00:00:37 +0000171 Members = std::vector<std::string>(RestOfArgs);
Tanya Lattnerc970a382003-12-06 23:01:25 +0000172}
173
Reid Spencer84a12bf2004-11-14 22:20:07 +0000174// parseCommandLine - Parse the command line options as presented and return the
Misha Brukman650ba8e2005-04-22 00:00:37 +0000175// operation specified. Process all modifiers and check to make sure that
Reid Spencer84a12bf2004-11-14 22:20:07 +0000176// constraints on modifier/operation pairs have not been violated.
Rafael Espindola05571532013-07-12 16:29:27 +0000177static ArchiveOperation parseCommandLine() {
Rafael Espindola0c8a3522013-08-28 16:22:16 +0000178 getOptions();
Tanya Lattnerc970a382003-12-06 23:01:25 +0000179
Reid Spencer84a12bf2004-11-14 22:20:07 +0000180 // Keep track of number of operations. We can only specify one
181 // per execution.
Tanya Lattnerc970a382003-12-06 23:01:25 +0000182 unsigned NumOperations = 0;
183
Reid Spencer84a12bf2004-11-14 22:20:07 +0000184 // Keep track of the number of positional modifiers (a,b,i). Only
185 // one can be specified.
186 unsigned NumPositional = 0;
187
188 // Keep track of which operation was requested
Rafael Espindola544615f2013-07-05 12:12:43 +0000189 ArchiveOperation Operation;
Reid Spencer84a12bf2004-11-14 22:20:07 +0000190
Rafael Espindolab6b5f52e2013-07-29 12:40:31 +0000191 bool MaybeJustCreateSymTab = false;
192
Tanya Lattnerc970a382003-12-06 23:01:25 +0000193 for(unsigned i=0; i<Options.size(); ++i) {
194 switch(Options[i]) {
Reid Spencer84a12bf2004-11-14 22:20:07 +0000195 case 'd': ++NumOperations; Operation = Delete; break;
196 case 'm': ++NumOperations; Operation = Move ; break;
197 case 'p': ++NumOperations; Operation = Print; break;
Seo Sanghyeond42a60b2007-12-25 13:53:47 +0000198 case 'q': ++NumOperations; Operation = QuickAppend; break;
Misha Brukman650ba8e2005-04-22 00:00:37 +0000199 case 'r': ++NumOperations; Operation = ReplaceOrInsert; break;
Reid Spencer84a12bf2004-11-14 22:20:07 +0000200 case 't': ++NumOperations; Operation = DisplayTable; break;
201 case 'x': ++NumOperations; Operation = Extract; break;
202 case 'c': Create = true; break;
Reid Spencer84a12bf2004-11-14 22:20:07 +0000203 case 'l': /* accepted but unused */ break;
204 case 'o': OriginalDates = true; break;
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000205 case 's':
206 Symtab = true;
Rafael Espindolab6b5f52e2013-07-29 12:40:31 +0000207 MaybeJustCreateSymTab = true;
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000208 break;
209 case 'S':
210 Symtab = false;
211 break;
Reid Spencer84a12bf2004-11-14 22:20:07 +0000212 case 'u': OnlyUpdate = true; break;
213 case 'v': Verbose = true; break;
Tanya Lattnerc970a382003-12-06 23:01:25 +0000214 case 'a':
Tanya Lattnerc970a382003-12-06 23:01:25 +0000215 getRelPos();
Reid Spencer84a12bf2004-11-14 22:20:07 +0000216 AddAfter = true;
217 NumPositional++;
Tanya Lattnerc970a382003-12-06 23:01:25 +0000218 break;
219 case 'b':
Tanya Lattnerc970a382003-12-06 23:01:25 +0000220 getRelPos();
Reid Spencer84a12bf2004-11-14 22:20:07 +0000221 AddBefore = true;
222 NumPositional++;
Tanya Lattnerc970a382003-12-06 23:01:25 +0000223 break;
224 case 'i':
Tanya Lattnerc970a382003-12-06 23:01:25 +0000225 getRelPos();
Rafael Espindolace2c84e2013-07-11 15:54:53 +0000226 AddBefore = true;
Reid Spencer84a12bf2004-11-14 22:20:07 +0000227 NumPositional++;
Tanya Lattnerc970a382003-12-06 23:01:25 +0000228 break;
Tanya Lattnerc970a382003-12-06 23:01:25 +0000229 default:
Reid Spencer9fc38b12004-11-16 06:41:09 +0000230 cl::PrintHelpMessage();
Tanya Lattnerc970a382003-12-06 23:01:25 +0000231 }
232 }
233
Misha Brukman650ba8e2005-04-22 00:00:37 +0000234 // At this point, the next thing on the command line must be
Reid Spencer84a12bf2004-11-14 22:20:07 +0000235 // the archive name.
Tanya Lattnerc970a382003-12-06 23:01:25 +0000236 getArchive();
Reid Spencer84a12bf2004-11-14 22:20:07 +0000237
238 // Everything on the command line at this point is a member.
Tanya Lattnerc970a382003-12-06 23:01:25 +0000239 getMembers();
240
Rafael Espindolab6b5f52e2013-07-29 12:40:31 +0000241 if (NumOperations == 0 && MaybeJustCreateSymTab) {
242 NumOperations = 1;
243 Operation = CreateSymTab;
244 if (!Members.empty())
245 show_help("The s operation takes only an archive as argument");
246 }
247
Reid Spencer84a12bf2004-11-14 22:20:07 +0000248 // Perform various checks on the operation/modifier specification
249 // to make sure we are dealing with a legal request.
250 if (NumOperations == 0)
Joerg Sonnenberger4a0f7be2012-10-26 10:49:15 +0000251 show_help("You must specify at least one of the operations");
Reid Spencer84a12bf2004-11-14 22:20:07 +0000252 if (NumOperations > 1)
Joerg Sonnenberger4a0f7be2012-10-26 10:49:15 +0000253 show_help("Only one operation may be specified");
Reid Spencer84a12bf2004-11-14 22:20:07 +0000254 if (NumPositional > 1)
Joerg Sonnenberger4a0f7be2012-10-26 10:49:15 +0000255 show_help("You may only specify one of a, b, and i modifiers");
Rafael Espindolace2c84e2013-07-11 15:54:53 +0000256 if (AddAfter || AddBefore) {
Reid Spencer84a12bf2004-11-14 22:20:07 +0000257 if (Operation != Move && Operation != ReplaceOrInsert)
Joerg Sonnenberger4a0f7be2012-10-26 10:49:15 +0000258 show_help("The 'a', 'b' and 'i' modifiers can only be specified with "
259 "the 'm' or 'r' operations");
260 }
Reid Spencer84a12bf2004-11-14 22:20:07 +0000261 if (OriginalDates && Operation != Extract)
Joerg Sonnenberger4a0f7be2012-10-26 10:49:15 +0000262 show_help("The 'o' modifier is only applicable to the 'x' operation");
Reid Spencer84a12bf2004-11-14 22:20:07 +0000263 if (OnlyUpdate && Operation != ReplaceOrInsert)
Joerg Sonnenberger4a0f7be2012-10-26 10:49:15 +0000264 show_help("The 'u' modifier is only applicable to the 'r' operation");
Reid Spencer84a12bf2004-11-14 22:20:07 +0000265
266 // Return the parsed operation to the caller
267 return Operation;
Tanya Lattnerc970a382003-12-06 23:01:25 +0000268}
Tanya Lattner57c70a62003-08-28 15:22:38 +0000269
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000270// Implements the 'p' operation. This function traverses the archive
271// looking for members that match the path list.
272static void doPrint(StringRef Name, object::Archive::child_iterator I) {
273 if (Verbose)
274 outs() << "Printing " << Name << "\n";
Rafael Espindola3703fd02013-06-19 14:58:16 +0000275
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000276 StringRef Data = I->getBuffer();
277 outs().write(Data.data(), Data.size());
Reid Spencer84a12bf2004-11-14 22:20:07 +0000278}
279
280// putMode - utility function for printing out the file mode when the 't'
281// operation is in verbose mode.
Rafael Espindola05571532013-07-12 16:29:27 +0000282static void printMode(unsigned mode) {
Misha Brukman650ba8e2005-04-22 00:00:37 +0000283 if (mode & 004)
Chris Lattner2907f442010-11-29 23:02:20 +0000284 outs() << "r";
Reid Spencer84a12bf2004-11-14 22:20:07 +0000285 else
Chris Lattner2907f442010-11-29 23:02:20 +0000286 outs() << "-";
Reid Spencer84a12bf2004-11-14 22:20:07 +0000287 if (mode & 002)
Chris Lattner2907f442010-11-29 23:02:20 +0000288 outs() << "w";
Reid Spencer84a12bf2004-11-14 22:20:07 +0000289 else
Chris Lattner2907f442010-11-29 23:02:20 +0000290 outs() << "-";
Reid Spencer84a12bf2004-11-14 22:20:07 +0000291 if (mode & 001)
Chris Lattner2907f442010-11-29 23:02:20 +0000292 outs() << "x";
Reid Spencer84a12bf2004-11-14 22:20:07 +0000293 else
Chris Lattner2907f442010-11-29 23:02:20 +0000294 outs() << "-";
Reid Spencer84a12bf2004-11-14 22:20:07 +0000295}
296
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000297// Implement the 't' operation. This function prints out just
Reid Spencer84a12bf2004-11-14 22:20:07 +0000298// the file names of each of the members. However, if verbose mode is requested
299// ('v' modifier) then the file type, permission mode, user, group, size, and
300// modification time are also printed.
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000301static void doDisplayTable(StringRef Name, object::Archive::child_iterator I) {
302 if (Verbose) {
303 sys::fs::perms Mode = I->getAccessMode();
304 printMode((Mode >> 6) & 007);
305 printMode((Mode >> 3) & 007);
306 printMode(Mode & 007);
307 outs() << ' ' << I->getUID();
308 outs() << '/' << I->getGID();
309 outs() << ' ' << format("%6llu", I->getSize());
310 outs() << ' ' << I->getLastModified().str();
311 outs() << ' ';
Reid Spencer84a12bf2004-11-14 22:20:07 +0000312 }
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000313 outs() << Name << "\n";
Reid Spencer84a12bf2004-11-14 22:20:07 +0000314}
315
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000316// Implement the 'x' operation. This function extracts files back to the file
317// system.
318static void doExtract(StringRef Name, object::Archive::child_iterator I) {
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000319 // Retain the original mode.
320 sys::fs::perms Mode = I->getAccessMode();
Rafael Espindola6d354812013-07-16 19:44:17 +0000321 SmallString<128> Storage = Name;
Rafael Espindolabb625bb2013-07-08 16:16:51 +0000322
Rafael Espindola6d354812013-07-16 19:44:17 +0000323 int FD;
324 failIfError(
Rafael Espindola90c7f1c2014-02-24 18:20:12 +0000325 sys::fs::openFileForWrite(Storage.c_str(), FD, sys::fs::F_None, Mode),
Rafael Espindola6d354812013-07-16 19:44:17 +0000326 Storage.c_str());
Reid Spencer84a12bf2004-11-14 22:20:07 +0000327
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000328 {
329 raw_fd_ostream file(FD, false);
Reid Spencer84a12bf2004-11-14 22:20:07 +0000330
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000331 // Get the data and its length
332 StringRef Data = I->getBuffer();
Rafael Espindola4a3365c2013-06-20 20:56:14 +0000333
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000334 // Write the data.
335 file.write(Data.data(), Data.size());
Reid Spencer84a12bf2004-11-14 22:20:07 +0000336 }
337
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000338 // If we're supposed to retain the original modification times, etc. do so
339 // now.
340 if (OriginalDates)
341 failIfError(
342 sys::fs::setLastModificationAndAccessTime(FD, I->getLastModified()));
Reid Spencer84a12bf2004-11-14 22:20:07 +0000343
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000344 if (close(FD))
345 fail("Could not close the file");
Reid Spencer84a12bf2004-11-14 22:20:07 +0000346}
347
Rafael Espindola05571532013-07-12 16:29:27 +0000348static bool shouldCreateArchive(ArchiveOperation Op) {
Rafael Espindola8ef843f2013-07-05 13:03:07 +0000349 switch (Op) {
350 case Print:
351 case Delete:
352 case Move:
353 case DisplayTable:
354 case Extract:
Rafael Espindolab6b5f52e2013-07-29 12:40:31 +0000355 case CreateSymTab:
Rafael Espindola8ef843f2013-07-05 13:03:07 +0000356 return false;
357
358 case QuickAppend:
359 case ReplaceOrInsert:
360 return true;
361 }
Michael Gottesman11dd1d02013-07-06 02:39:51 +0000362
363 llvm_unreachable("Missing entry in covered switch.");
Rafael Espindola8ef843f2013-07-05 13:03:07 +0000364}
365
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000366static void performReadOperation(ArchiveOperation Operation,
367 object::Archive *OldArchive) {
Rafael Espindola23a97502014-01-21 16:09:45 +0000368 for (object::Archive::child_iterator I = OldArchive->child_begin(),
369 E = OldArchive->child_end();
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000370 I != E; ++I) {
371 StringRef Name;
372 failIfError(I->getName(Name));
373
374 if (!Members.empty() &&
375 std::find(Members.begin(), Members.end(), Name) == Members.end())
376 continue;
377
378 switch (Operation) {
379 default:
380 llvm_unreachable("Not a read operation");
381 case Print:
382 doPrint(Name, I);
383 break;
384 case DisplayTable:
385 doDisplayTable(Name, I);
386 break;
387 case Extract:
388 doExtract(Name, I);
389 break;
390 }
391 }
392}
393
394namespace {
395class NewArchiveIterator {
396 bool IsNewMember;
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000397 StringRef Name;
Rafael Espindolab6082a32014-01-22 16:43:45 +0000398
Rafael Espindola023e6562013-07-12 20:28:02 +0000399 object::Archive::child_iterator OldI;
Rafael Espindolab6082a32014-01-22 16:43:45 +0000400
Rafael Espindola623c3d82013-07-22 15:11:51 +0000401 std::string NewFilename;
Rafael Espindolab6082a32014-01-22 16:43:45 +0000402 mutable int NewFD;
403 mutable sys::fs::file_status NewStatus;
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000404
405public:
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000406 NewArchiveIterator(object::Archive::child_iterator I, StringRef Name);
407 NewArchiveIterator(std::string *I, StringRef Name);
Rafael Espindolafcc3a1a2013-07-19 21:23:28 +0000408 NewArchiveIterator();
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000409 bool isNewMember() const;
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000410 StringRef getName() const;
Rafael Espindolab6082a32014-01-22 16:43:45 +0000411
412 object::Archive::child_iterator getOld() const;
413
414 const char *getNew() const;
415 int getFD() const;
416 const sys::fs::file_status &getStatus() const;
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000417};
418}
419
Rafael Espindolafcc3a1a2013-07-19 21:23:28 +0000420NewArchiveIterator::NewArchiveIterator() {}
421
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000422NewArchiveIterator::NewArchiveIterator(object::Archive::child_iterator I,
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000423 StringRef Name)
424 : IsNewMember(false), Name(Name), OldI(I) {}
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000425
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000426NewArchiveIterator::NewArchiveIterator(std::string *NewFilename, StringRef Name)
Rafael Espindolab6082a32014-01-22 16:43:45 +0000427 : IsNewMember(true), Name(Name), NewFilename(*NewFilename), NewFD(-1) {}
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000428
429StringRef NewArchiveIterator::getName() const { return Name; }
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000430
431bool NewArchiveIterator::isNewMember() const { return IsNewMember; }
432
433object::Archive::child_iterator NewArchiveIterator::getOld() const {
434 assert(!IsNewMember);
435 return OldI;
436}
437
Rafael Espindola8c1ee472013-07-16 03:30:10 +0000438const char *NewArchiveIterator::getNew() const {
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000439 assert(IsNewMember);
Rafael Espindola623c3d82013-07-22 15:11:51 +0000440 return NewFilename.c_str();
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000441}
442
Rafael Espindolab6082a32014-01-22 16:43:45 +0000443int NewArchiveIterator::getFD() const {
444 assert(IsNewMember);
445 if (NewFD != -1)
446 return NewFD;
447 failIfError(sys::fs::openFileForRead(NewFilename, NewFD), NewFilename);
448 assert(NewFD != -1);
449
450 failIfError(sys::fs::status(NewFD, NewStatus), NewFilename);
451
452 // Opening a directory doesn't make sense. Let it fail.
453 // Linux cannot open directories with open(2), although
454 // cygwin and *bsd can.
455 if (NewStatus.type() == sys::fs::file_type::directory_file)
456 failIfError(error_code(errc::is_a_directory, posix_category()),
457 NewFilename);
458
459 return NewFD;
460}
461
462const sys::fs::file_status &NewArchiveIterator::getStatus() const {
463 assert(IsNewMember);
464 assert(NewFD != -1 && "Must call getFD first");
465 return NewStatus;
466}
467
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000468template <typename T>
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000469void addMember(std::vector<NewArchiveIterator> &Members, T I, StringRef Name,
470 int Pos = -1) {
471 NewArchiveIterator NI(I, Name);
472 if (Pos == -1)
473 Members.push_back(NI);
474 else
Rafael Espindola623c3d82013-07-22 15:11:51 +0000475 Members[Pos] = NI;
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000476}
477
Rafael Espindola623c3d82013-07-22 15:11:51 +0000478enum InsertAction {
479 IA_AddOldMember,
480 IA_AddNewMeber,
481 IA_Delete,
482 IA_MoveOldMember,
483 IA_MoveNewMember
484};
485
486static InsertAction
487computeInsertAction(ArchiveOperation Operation,
488 object::Archive::child_iterator I, StringRef Name,
489 std::vector<std::string>::iterator &Pos) {
490 if (Operation == QuickAppend || Members.empty())
491 return IA_AddOldMember;
492
Benjamin Kramer3a377bc2014-03-01 11:47:00 +0000493 std::vector<std::string>::iterator MI = std::find_if(
494 Members.begin(), Members.end(),
495 [Name](StringRef Path) { return Name == sys::path::filename(Path); });
Rafael Espindola623c3d82013-07-22 15:11:51 +0000496
497 if (MI == Members.end())
498 return IA_AddOldMember;
499
500 Pos = MI;
501
502 if (Operation == Delete)
503 return IA_Delete;
504
505 if (Operation == Move)
506 return IA_MoveOldMember;
507
508 if (Operation == ReplaceOrInsert) {
509 StringRef PosName = sys::path::filename(RelPos);
510 if (!OnlyUpdate) {
511 if (PosName.empty())
512 return IA_AddNewMeber;
513 return IA_MoveNewMember;
514 }
515
516 // We could try to optimize this to a fstat, but it is not a common
517 // operation.
518 sys::fs::file_status Status;
519 failIfError(sys::fs::status(*MI, Status));
520 if (Status.getLastModificationTime() < I->getLastModified()) {
521 if (PosName.empty())
522 return IA_AddOldMember;
523 return IA_MoveOldMember;
524 }
525
526 if (PosName.empty())
527 return IA_AddNewMeber;
528 return IA_MoveNewMember;
529 }
530 llvm_unreachable("No such operation");
531}
532
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000533// We have to walk this twice and computing it is not trivial, so creating an
534// explicit std::vector is actually fairly efficient.
535static std::vector<NewArchiveIterator>
536computeNewArchiveMembers(ArchiveOperation Operation,
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000537 object::Archive *OldArchive) {
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000538 std::vector<NewArchiveIterator> Ret;
539 std::vector<NewArchiveIterator> Moved;
540 int InsertPos = -1;
541 StringRef PosName = sys::path::filename(RelPos);
542 if (OldArchive) {
Rafael Espindola23a97502014-01-21 16:09:45 +0000543 for (object::Archive::child_iterator I = OldArchive->child_begin(),
544 E = OldArchive->child_end();
Rafael Espindola9cd24352013-07-21 12:58:07 +0000545 I != E; ++I) {
546 int Pos = Ret.size();
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000547 StringRef Name;
548 failIfError(I->getName(Name));
549 if (Name == PosName) {
550 assert(AddAfter || AddBefore);
551 if (AddBefore)
552 InsertPos = Pos;
553 else
554 InsertPos = Pos + 1;
555 }
Rafael Espindola623c3d82013-07-22 15:11:51 +0000556
557 std::vector<std::string>::iterator MemberI = Members.end();
558 InsertAction Action = computeInsertAction(Operation, I, Name, MemberI);
559 switch (Action) {
560 case IA_AddOldMember:
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000561 addMember(Ret, I, Name);
Rafael Espindola623c3d82013-07-22 15:11:51 +0000562 break;
563 case IA_AddNewMeber:
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000564 addMember(Ret, &*MemberI, Name);
Rafael Espindola623c3d82013-07-22 15:11:51 +0000565 break;
566 case IA_Delete:
567 break;
568 case IA_MoveOldMember:
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000569 addMember(Moved, I, Name);
Rafael Espindola623c3d82013-07-22 15:11:51 +0000570 break;
571 case IA_MoveNewMember:
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000572 addMember(Moved, &*MemberI, Name);
Rafael Espindola623c3d82013-07-22 15:11:51 +0000573 break;
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000574 }
Rafael Espindola623c3d82013-07-22 15:11:51 +0000575 if (MemberI != Members.end())
576 Members.erase(MemberI);
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000577 }
578 }
579
580 if (Operation == Delete)
581 return Ret;
582
Rafael Espindolafcc3a1a2013-07-19 21:23:28 +0000583 if (!RelPos.empty() && InsertPos == -1)
584 fail("Insertion point not found");
585
Rafael Espindola623c3d82013-07-22 15:11:51 +0000586 if (RelPos.empty())
587 InsertPos = Ret.size();
Rafael Espindolafcc3a1a2013-07-19 21:23:28 +0000588
Rafael Espindola623c3d82013-07-22 15:11:51 +0000589 assert(unsigned(InsertPos) <= Ret.size());
590 Ret.insert(Ret.begin() + InsertPos, Moved.begin(), Moved.end());
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000591
Rafael Espindola623c3d82013-07-22 15:11:51 +0000592 Ret.insert(Ret.begin() + InsertPos, Members.size(), NewArchiveIterator());
Rafael Espindolafcc3a1a2013-07-19 21:23:28 +0000593 int Pos = InsertPos;
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000594 for (std::vector<std::string>::iterator I = Members.begin(),
Rafael Espindola623c3d82013-07-22 15:11:51 +0000595 E = Members.end();
596 I != E; ++I, ++Pos) {
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000597 StringRef Name = sys::path::filename(*I);
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000598 addMember(Ret, &*I, Name, Pos);
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000599 }
600
601 return Ret;
602}
603
604template <typename T>
Rafael Espindolaed900302014-01-14 17:02:09 +0000605static void printWithSpacePadding(raw_fd_ostream &OS, T Data, unsigned Size,
606 bool MayTruncate = false) {
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000607 uint64_t OldPos = OS.tell();
608 OS << Data;
609 unsigned SizeSoFar = OS.tell() - OldPos;
Rafael Espindolaed900302014-01-14 17:02:09 +0000610 if (Size > SizeSoFar) {
611 unsigned Remaining = Size - SizeSoFar;
612 for (unsigned I = 0; I < Remaining; ++I)
613 OS << ' ';
614 } else if (Size < SizeSoFar) {
615 assert(MayTruncate && "Data doesn't fit in Size");
616 // Some of the data this is used for (like UID) can be larger than the
617 // space available in the archive format. Truncate in that case.
618 OS.seek(OldPos + Size);
619 }
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000620}
621
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000622static void print32BE(raw_fd_ostream &Out, unsigned Val) {
623 for (int I = 3; I >= 0; --I) {
624 char V = (Val >> (8 * I)) & 0xff;
625 Out << V;
626 }
627}
628
629static void printRestOfMemberHeader(raw_fd_ostream &Out,
630 const sys::TimeValue &ModTime, unsigned UID,
631 unsigned GID, unsigned Perms,
632 unsigned Size) {
633 printWithSpacePadding(Out, ModTime.toEpochTime(), 12);
Rafael Espindolaed900302014-01-14 17:02:09 +0000634 printWithSpacePadding(Out, UID, 6, true);
635 printWithSpacePadding(Out, GID, 6, true);
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000636 printWithSpacePadding(Out, format("%o", Perms), 8);
637 printWithSpacePadding(Out, Size, 10);
638 Out << "`\n";
639}
640
641static void printMemberHeader(raw_fd_ostream &Out, StringRef Name,
642 const sys::TimeValue &ModTime, unsigned UID,
643 unsigned GID, unsigned Perms, unsigned Size) {
644 printWithSpacePadding(Out, Twine(Name) + "/", 16);
645 printRestOfMemberHeader(Out, ModTime, UID, GID, Perms, Size);
646}
647
648static void printMemberHeader(raw_fd_ostream &Out, unsigned NameOffset,
649 const sys::TimeValue &ModTime, unsigned UID,
650 unsigned GID, unsigned Perms, unsigned Size) {
651 Out << '/';
652 printWithSpacePadding(Out, NameOffset, 15);
653 printRestOfMemberHeader(Out, ModTime, UID, GID, Perms, Size);
654}
655
656static void writeStringTable(raw_fd_ostream &Out,
657 ArrayRef<NewArchiveIterator> Members,
658 std::vector<unsigned> &StringMapIndexes) {
659 unsigned StartOffset = 0;
660 for (ArrayRef<NewArchiveIterator>::iterator I = Members.begin(),
661 E = Members.end();
662 I != E; ++I) {
663 StringRef Name = I->getName();
664 if (Name.size() < 16)
665 continue;
666 if (StartOffset == 0) {
667 printWithSpacePadding(Out, "//", 58);
668 Out << "`\n";
669 StartOffset = Out.tell();
670 }
671 StringMapIndexes.push_back(Out.tell() - StartOffset);
672 Out << Name << "/\n";
673 }
674 if (StartOffset == 0)
675 return;
676 if (Out.tell() % 2)
677 Out << '\n';
678 int Pos = Out.tell();
679 Out.seek(StartOffset - 12);
680 printWithSpacePadding(Out, Pos - StartOffset, 10);
681 Out.seek(Pos);
682}
683
684static void writeSymbolTable(
685 raw_fd_ostream &Out, ArrayRef<NewArchiveIterator> Members,
Rafael Espindola09419ad2014-01-24 21:52:44 +0000686 ArrayRef<MemoryBuffer *> Buffers,
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000687 std::vector<std::pair<unsigned, unsigned> > &MemberOffsetRefs) {
688 unsigned StartOffset = 0;
689 unsigned MemberNum = 0;
Rafael Espindolaf12b8282014-02-21 20:10:59 +0000690 std::string NameBuf;
691 raw_string_ostream NameOS(NameBuf);
692 unsigned NumSyms = 0;
693 std::vector<object::SymbolicFile *> DeleteIt;
694 LLVMContext &Context = getGlobalContext();
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000695 for (ArrayRef<NewArchiveIterator>::iterator I = Members.begin(),
696 E = Members.end();
697 I != E; ++I, ++MemberNum) {
Rafael Espindola09419ad2014-01-24 21:52:44 +0000698 MemoryBuffer *MemberBuffer = Buffers[MemberNum];
Rafael Espindolaf12b8282014-02-21 20:10:59 +0000699 ErrorOr<object::SymbolicFile *> ObjOrErr =
700 object::SymbolicFile::createSymbolicFile(
701 MemberBuffer, false, sys::fs::file_magic::unknown, &Context);
Rafael Espindolaafcc3df2014-01-24 21:32:21 +0000702 if (!ObjOrErr)
703 continue; // FIXME: check only for "not an object file" errors.
Rafael Espindolaf12b8282014-02-21 20:10:59 +0000704 object::SymbolicFile *Obj = ObjOrErr.get();
Rafael Espindolab6082a32014-01-22 16:43:45 +0000705
Rafael Espindola188bfd22013-07-23 11:02:16 +0000706 DeleteIt.push_back(Obj);
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000707 if (!StartOffset) {
708 printMemberHeader(Out, "", sys::TimeValue::now(), 0, 0, 0, 0);
709 StartOffset = Out.tell();
710 print32BE(Out, 0);
711 }
712
Rafael Espindolaf12b8282014-02-21 20:10:59 +0000713 for (object::basic_symbol_iterator I = Obj->symbol_begin(),
714 E = Obj->symbol_end();
Rafael Espindola5e812af2014-01-30 02:49:50 +0000715 I != E; ++I) {
Rafael Espindolaf12b8282014-02-21 20:10:59 +0000716 uint32_t Symflags = I->getFlags();
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000717 if (Symflags & object::SymbolRef::SF_FormatSpecific)
718 continue;
719 if (!(Symflags & object::SymbolRef::SF_Global))
720 continue;
721 if (Symflags & object::SymbolRef::SF_Undefined)
722 continue;
Rafael Espindolaf12b8282014-02-21 20:10:59 +0000723 failIfError(I->printName(NameOS));
724 NameOS << '\0';
725 ++NumSyms;
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000726 MemberOffsetRefs.push_back(std::make_pair(Out.tell(), MemberNum));
727 print32BE(Out, 0);
728 }
729 }
Rafael Espindolaf12b8282014-02-21 20:10:59 +0000730 Out << NameOS.str();
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000731
Rafael Espindolaf12b8282014-02-21 20:10:59 +0000732 for (std::vector<object::SymbolicFile *>::iterator I = DeleteIt.begin(),
733 E = DeleteIt.end();
Rafael Espindola188bfd22013-07-23 11:02:16 +0000734 I != E; ++I) {
Rafael Espindolaf12b8282014-02-21 20:10:59 +0000735 object::SymbolicFile *O = *I;
Rafael Espindola188bfd22013-07-23 11:02:16 +0000736 delete O;
737 }
738
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000739 if (StartOffset == 0)
740 return;
741
742 if (Out.tell() % 2)
743 Out << '\0';
744
745 unsigned Pos = Out.tell();
746 Out.seek(StartOffset - 12);
747 printWithSpacePadding(Out, Pos - StartOffset, 10);
748 Out.seek(StartOffset);
Rafael Espindolaf12b8282014-02-21 20:10:59 +0000749 print32BE(Out, NumSyms);
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000750 Out.seek(Pos);
751}
752
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000753static void performWriteOperation(ArchiveOperation Operation,
754 object::Archive *OldArchive) {
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000755 SmallString<128> TmpArchive;
756 failIfError(sys::fs::createUniqueFile(ArchiveName + ".temp-archive-%%%%%%%.a",
757 TmpArchiveFD, TmpArchive));
758
759 TemporaryOutput = TmpArchive.c_str();
760 tool_output_file Output(TemporaryOutput, TmpArchiveFD);
761 raw_fd_ostream &Out = Output.os();
762 Out << "!<arch>\n";
763
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000764 std::vector<NewArchiveIterator> NewMembers =
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000765 computeNewArchiveMembers(Operation, OldArchive);
766
767 std::vector<std::pair<unsigned, unsigned> > MemberOffsetRefs;
768
Rafael Espindola09419ad2014-01-24 21:52:44 +0000769 std::vector<MemoryBuffer *> MemberBuffers;
Rafael Espindolaafcc3df2014-01-24 21:32:21 +0000770 MemberBuffers.resize(NewMembers.size());
771
772 for (unsigned I = 0, N = NewMembers.size(); I < N; ++I) {
Ahmed Charles56440fd2014-03-06 05:51:42 +0000773 std::unique_ptr<MemoryBuffer> MemberBuffer;
Rafael Espindolaafcc3df2014-01-24 21:32:21 +0000774 NewArchiveIterator &Member = NewMembers[I];
775
776 if (Member.isNewMember()) {
777 const char *Filename = Member.getNew();
778 int FD = Member.getFD();
779 const sys::fs::file_status &Status = Member.getStatus();
780 failIfError(MemoryBuffer::getOpenFile(FD, Filename, MemberBuffer,
781 Status.getSize(), false),
782 Filename);
783
784 } else {
785 object::Archive::child_iterator OldMember = Member.getOld();
786 failIfError(OldMember->getMemoryBuffer(MemberBuffer));
787 }
Ahmed Charles96c9d952014-03-05 10:19:29 +0000788 MemberBuffers[I] = MemberBuffer.release();
Rafael Espindolaafcc3df2014-01-24 21:32:21 +0000789 }
790
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000791 if (Symtab) {
Rafael Espindolaafcc3df2014-01-24 21:32:21 +0000792 writeSymbolTable(Out, NewMembers, MemberBuffers, MemberOffsetRefs);
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000793 }
794
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000795 std::vector<unsigned> StringMapIndexes;
796 writeStringTable(Out, NewMembers, StringMapIndexes);
797
798 std::vector<std::pair<unsigned, unsigned> >::iterator MemberRefsI =
799 MemberOffsetRefs.begin();
800
801 unsigned MemberNum = 0;
802 unsigned LongNameMemberNum = 0;
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000803 for (std::vector<NewArchiveIterator>::iterator I = NewMembers.begin(),
804 E = NewMembers.end();
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000805 I != E; ++I, ++MemberNum) {
806
807 unsigned Pos = Out.tell();
808 while (MemberRefsI != MemberOffsetRefs.end() &&
809 MemberRefsI->second == MemberNum) {
810 Out.seek(MemberRefsI->first);
811 print32BE(Out, Pos);
812 ++MemberRefsI;
813 }
814 Out.seek(Pos);
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000815
Rafael Espindola09419ad2014-01-24 21:52:44 +0000816 const MemoryBuffer *File = MemberBuffers[MemberNum];
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000817 if (I->isNewMember()) {
Rafael Espindola8c1ee472013-07-16 03:30:10 +0000818 const char *FileName = I->getNew();
Rafael Espindolab6082a32014-01-22 16:43:45 +0000819 const sys::fs::file_status &Status = I->getStatus();
NAKAMURA Takumi0d82bac2013-11-08 12:35:56 +0000820
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000821 StringRef Name = sys::path::filename(FileName);
822 if (Name.size() < 16)
823 printMemberHeader(Out, Name, Status.getLastModificationTime(),
824 Status.getUser(), Status.getGroup(),
825 Status.permissions(), Status.getSize());
826 else
827 printMemberHeader(Out, StringMapIndexes[LongNameMemberNum++],
828 Status.getLastModificationTime(), Status.getUser(),
829 Status.getGroup(), Status.permissions(),
830 Status.getSize());
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000831 } else {
832 object::Archive::child_iterator OldMember = I->getOld();
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000833 StringRef Name = I->getName();
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000834
Rafael Espindolace7f52d2013-07-23 10:47:01 +0000835 if (Name.size() < 16)
836 printMemberHeader(Out, Name, OldMember->getLastModified(),
837 OldMember->getUID(), OldMember->getGID(),
838 OldMember->getAccessMode(), OldMember->getSize());
839 else
840 printMemberHeader(Out, StringMapIndexes[LongNameMemberNum++],
841 OldMember->getLastModified(), OldMember->getUID(),
842 OldMember->getGID(), OldMember->getAccessMode(),
843 OldMember->getSize());
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000844 }
845
Rafael Espindolaafcc3df2014-01-24 21:32:21 +0000846 Out << File->getBuffer();
847
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000848 if (Out.tell() % 2)
849 Out << '\n';
850 }
Rafael Espindola09419ad2014-01-24 21:52:44 +0000851
852 for (unsigned I = 0, N = MemberBuffers.size(); I < N; ++I) {
853 delete MemberBuffers[I];
854 }
855
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000856 Output.keep();
857 Out.close();
858 sys::fs::rename(TemporaryOutput, ArchiveName);
Craig Toppere6cb63e2014-04-25 04:24:47 +0000859 TemporaryOutput = nullptr;
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000860}
861
Rafael Espindolab6b5f52e2013-07-29 12:40:31 +0000862static void createSymbolTable(object::Archive *OldArchive) {
863 // When an archive is created or modified, if the s option is given, the
864 // resulting archive will have a current symbol table. If the S option
865 // is given, it will have no symbol table.
866 // In summary, we only need to update the symbol table if we have none.
867 // This is actually very common because of broken build systems that think
868 // they have to run ranlib.
869 if (OldArchive->hasSymbolTable())
870 return;
871
872 performWriteOperation(CreateSymTab, OldArchive);
873}
874
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000875static void performOperation(ArchiveOperation Operation,
876 object::Archive *OldArchive) {
877 switch (Operation) {
878 case Print:
879 case DisplayTable:
880 case Extract:
881 performReadOperation(Operation, OldArchive);
882 return;
883
884 case Delete:
885 case Move:
886 case QuickAppend:
887 case ReplaceOrInsert:
888 performWriteOperation(Operation, OldArchive);
889 return;
Rafael Espindolab6b5f52e2013-07-29 12:40:31 +0000890 case CreateSymTab:
891 createSymbolTable(OldArchive);
892 return;
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000893 }
894 llvm_unreachable("Unknown operation.");
895}
896
Rafael Espindola0c8a3522013-08-28 16:22:16 +0000897static int ar_main(char **argv);
898static int ranlib_main();
899
Reid Spencer84a12bf2004-11-14 22:20:07 +0000900// main - main program for llvm-ar .. see comments in the code
Tanya Lattner57c70a62003-08-28 15:22:38 +0000901int main(int argc, char **argv) {
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000902 ToolName = argv[0];
Chris Lattnere3fc2d12009-03-06 05:34:10 +0000903 // Print a stack trace if we signal out.
904 sys::PrintStackTraceOnErrorSignal();
905 PrettyStackTraceProgram X(argc, argv);
906 llvm_shutdown_obj Y; // Call llvm_shutdown() on exit.
Reid Spencer84a12bf2004-11-14 22:20:07 +0000907
Reid Spencer84a12bf2004-11-14 22:20:07 +0000908 // Have the command line options parsed and handle things
909 // like --help and --version.
910 cl::ParseCommandLineOptions(argc, argv,
Dan Gohman2c6a8212007-10-08 15:45:12 +0000911 "LLVM Archiver (llvm-ar)\n\n"
Gabor Greife16561c2007-07-05 17:07:56 +0000912 " This program archives bitcode files into single libraries\n"
Reid Spencer84a12bf2004-11-14 22:20:07 +0000913 );
914
Rafael Espindola0642df32013-08-28 21:00:03 +0000915 StringRef Stem = sys::path::stem(ToolName);
Rafael Espindolae804b1a2013-10-29 14:25:43 +0000916 if (Stem.find("ar") != StringRef::npos)
Rafael Espindola0c8a3522013-08-28 16:22:16 +0000917 return ar_main(argv);
Rafael Espindolae804b1a2013-10-29 14:25:43 +0000918 if (Stem.find("ranlib") != StringRef::npos)
Rafael Espindola0c8a3522013-08-28 16:22:16 +0000919 return ranlib_main();
920 fail("Not ranlib or ar!");
921}
922
923static int performOperation(ArchiveOperation Operation);
924
925int ranlib_main() {
926 if (RestOfArgs.size() != 1)
927 fail(ToolName + "takes just one archive as argument");
928 ArchiveName = RestOfArgs[0];
929 return performOperation(CreateSymTab);
930}
931
932int ar_main(char **argv) {
Joerg Sonnenberger4a0f7be2012-10-26 10:49:15 +0000933 // Do our own parsing of the command line because the CommandLine utility
934 // can't handle the grouped positional parameters without a dash.
935 ArchiveOperation Operation = parseCommandLine();
Rafael Espindola0c8a3522013-08-28 16:22:16 +0000936 return performOperation(Operation);
937}
Tanya Lattner57c70a62003-08-28 15:22:38 +0000938
Rafael Espindola0c8a3522013-08-28 16:22:16 +0000939static int performOperation(ArchiveOperation Operation) {
Joerg Sonnenberger4a0f7be2012-10-26 10:49:15 +0000940 // Create or open the archive object.
Ahmed Charles56440fd2014-03-06 05:51:42 +0000941 std::unique_ptr<MemoryBuffer> Buf;
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000942 error_code EC = MemoryBuffer::getFile(ArchiveName, Buf, -1, false);
943 if (EC && EC != llvm::errc::no_such_file_or_directory) {
Rafael Espindola0c8a3522013-08-28 16:22:16 +0000944 errs() << ToolName << ": error opening '" << ArchiveName
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000945 << "': " << EC.message() << "!\n";
Joerg Sonnenberger4a0f7be2012-10-26 10:49:15 +0000946 return 1;
947 }
948
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000949 if (!EC) {
Ahmed Charles96c9d952014-03-05 10:19:29 +0000950 object::Archive Archive(Buf.release(), EC);
Joerg Sonnenberger4a0f7be2012-10-26 10:49:15 +0000951
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000952 if (EC) {
Rafael Espindola0c8a3522013-08-28 16:22:16 +0000953 errs() << ToolName << ": error loading '" << ArchiveName
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000954 << "': " << EC.message() << "!\n";
955 return 1;
956 }
957 performOperation(Operation, &Archive);
958 return 0;
959 }
960
961 assert(EC == llvm::errc::no_such_file_or_directory);
962
963 if (!shouldCreateArchive(Operation)) {
964 failIfError(EC, Twine("error loading '") + ArchiveName + "'");
965 } else {
966 if (!Create) {
967 // Produce a warning if we should and we're creating the archive
Rafael Espindola0c8a3522013-08-28 16:22:16 +0000968 errs() << ToolName << ": creating " << ArchiveName << "\n";
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000969 }
970 }
971
Craig Toppere6cb63e2014-04-25 04:24:47 +0000972 performOperation(Operation, nullptr);
Rafael Espindola3e2b21c2013-07-12 20:21:39 +0000973 return 0;
Tanya Lattner57c70a62003-08-28 15:22:38 +0000974}