1919#include " llvm/Option/ArgList.h"
2020#include " llvm/Support/CommandLine.h"
2121#include " llvm/Support/Error.h"
22+ #include " llvm/Support/FileOutputBuffer.h"
2223#include " llvm/Support/InitLLVM.h"
2324#include " llvm/Support/WithColor.h"
2425
@@ -78,23 +79,36 @@ class LipoOptTable : public opt::OptTable {
7879enum class LipoAction {
7980 PrintArchs,
8081 VerifyArch,
82+ ThinArch,
8183};
8284
8385struct Config {
8486 SmallVector<std::string, 1 > InputFiles;
8587 SmallVector<std::string, 1 > VerifyArchList;
88+ std::string ThinArchType;
89+ std::string OutputFile;
8690 LipoAction ActionToPerform;
8791};
8892
8993} // end namespace
9094
95+ static void validateArchitectureName (StringRef ArchitectureName) {
96+ if (Triple (ArchitectureName).getArch () == Triple::ArchType::UnknownArch)
97+ reportError (" Invalid architecture: "
98+ }
99+
91100static Config parseLipoOptions (ArrayRef<const char *> ArgsArr) {
92101 Config C;
93102 LipoOptTable T;
94103 unsigned MissingArgumentIndex, MissingArgumentCount;
95104 llvm::opt::InputArgList InputArgs =
96105 T.ParseArgs (ArgsArr, MissingArgumentIndex, MissingArgumentCount);
97106
107+ if (MissingArgumentCount)
108+ reportError (" missing argument to "
109+ StringRef (InputArgs.getArgString (MissingArgumentIndex)) +
110+ " option"
111+
98112 if (InputArgs.size () == 0 ) {
99113 // PrintHelp does not accept Twine.
100114 T.PrintHelp (errs (), " llvm-lipo input[s] option[s]" " llvm-lipo"
@@ -121,6 +135,9 @@ static Config parseLipoOptions(ArrayRef<const char *> ArgsArr) {
121135 if (C.InputFiles .empty ())
122136 reportError (" at least one input file should be specified"
123137
138+ if (InputArgs.hasArg (LIPO_output))
139+ C.OutputFile = InputArgs.getLastArgValue (LIPO_output);
140+
124141 SmallVector<opt::Arg *, 1 > ActionArgs (InputArgs.filtered (LIPO_action_group));
125142 if (ActionArgs.empty ())
126143 reportError (" at least one action should be specified"
@@ -151,6 +168,17 @@ static Config parseLipoOptions(ArrayRef<const char *> ArgsArr) {
151168 C.ActionToPerform = LipoAction::PrintArchs;
152169 return C;
153170
171+ case LIPO_thin:
172+ if (C.InputFiles .size () > 1 )
173+ reportError (" thin expects a single input file"
174+ C.ThinArchType = ActionArgs[0 ]->getValue ();
175+ validateArchitectureName (C.ThinArchType );
176+ if (C.OutputFile .empty ())
177+ reportError (" thin expects a single output file"
178+
179+ C.ActionToPerform = LipoAction::ThinArch;
180+ return C;
181+
154182 default :
155183 reportError (" llvm-lipo action unspecified"
156184 }
@@ -164,8 +192,12 @@ readInputBinaries(ArrayRef<std::string> InputFiles) {
164192 createBinary (InputFile);
165193 if (!BinaryOrErr)
166194 reportError (InputFile, BinaryOrErr.takeError ());
167- if (!isa<MachOObjectFile>(BinaryOrErr->getBinary ()) &&
168- !isa<MachOUniversalBinary>(BinaryOrErr->getBinary ()))
195+ // TODO: Add compatibility for archive files
196+ if (BinaryOrErr->getBinary ()->isArchive ())
197+ reportError (" File "
198+ " is an archive file and is not yet supported."
199+ if (!BinaryOrErr->getBinary ()->isMachO () &&
200+ !BinaryOrErr->getBinary ()->isMachOUniversalBinary ())
169201 reportError (" File " " has unsupported binary format"
170202 InputBinaries.push_back (std::move (*BinaryOrErr));
171203 }
@@ -180,8 +212,7 @@ static void verifyArch(ArrayRef<OwningBinary<Binary>> InputBinaries,
180212 assert (InputBinaries.size () == 1 && " Incorrect number of input binaries"
181213
182214 for (StringRef Arch : VerifyArchList)
183- if (Triple (Arch).getArch () == Triple::ArchType::UnknownArch)
184- reportError (" Invalid architecture: "
215+ validateArchitectureName (Arch);
185216
186217 if (auto UO =
187218 dyn_cast<MachOUniversalBinary>(InputBinaries.front ().getBinary ())) {
@@ -238,6 +269,44 @@ static void printArchs(ArrayRef<OwningBinary<Binary>> InputBinaries) {
238269 exit (EXIT_SUCCESS);
239270}
240271
272+ LLVM_ATTRIBUTE_NORETURN
273+ static void extractSlice (ArrayRef<OwningBinary<Binary>> InputBinaries,
274+ StringRef ThinArchType, StringRef OutputFileName) {
275+ assert (!ThinArchType.empty () && " The architecture type should be non-empty"
276+ assert (InputBinaries.size () == 1 && " Incorrect number of input binaries"
277+ assert (!OutputFileName.empty () && " Thin expects a single output file"
278+
279+ if (InputBinaries.front ().getBinary ()->isMachO ()) {
280+ reportError (" input file "
281+ InputBinaries.front ().getBinary ()->getFileName () +
282+ " must be a fat file when the -thin option is specified"
283+ exit (EXIT_FAILURE);
284+ }
285+
286+ auto *UO = cast<MachOUniversalBinary>(InputBinaries.front ().getBinary ());
287+ Expected<std::unique_ptr<MachOObjectFile>> Obj =
288+ UO->getObjectForArch (ThinArchType);
289+ if (!Obj)
290+ reportError (" fat input file " getFileName () +
291+ " does not contain the specified architecture "
292+ " to thin it to"
293+
294+ Expected<std::unique_ptr<FileOutputBuffer>> OutFileOrError =
295+ FileOutputBuffer::create (OutputFileName,
296+ Obj.get ()->getMemoryBufferRef ().getBufferSize (),
297+ sys::fs::can_execute (UO->getFileName ())
298+ ? FileOutputBuffer::F_executable
299+ : 0 );
300+ if (!OutFileOrError)
301+ reportError (OutputFileName, OutFileOrError.takeError ());
302+ std::copy (Obj.get ()->getMemoryBufferRef ().getBufferStart (),
303+ Obj.get ()->getMemoryBufferRef ().getBufferEnd (),
304+ OutFileOrError.get ()->getBufferStart ());
305+ if (Error E = OutFileOrError.get ()->commit ())
306+ reportError (OutputFileName, std::move (E));
307+ exit (EXIT_SUCCESS);
308+ }
309+
241310int main (int argc, char **argv) {
242311 InitLLVM X (argc, argv);
243312 Config C = parseLipoOptions (makeArrayRef (argv + 1 , argc));
@@ -251,6 +320,9 @@ int main(int argc, char **argv) {
251320 case LipoAction::PrintArchs:
252321 printArchs (InputBinaries);
253322 break ;
323+ case LipoAction::ThinArch:
324+ extractSlice (InputBinaries, C.ThinArchType , C.OutputFile );
325+ break ;
254326 }
255327 return EXIT_SUCCESS;
256328}
0 commit comments