llvm-libtool-darwin.cpp 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694
  1. //===-- llvm-libtool-darwin.cpp - a tool for creating libraries -----------===//
  2. //
  3. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
  4. // See https://llvm.org/LICENSE.txt for license information.
  5. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
  6. //
  7. //===----------------------------------------------------------------------===//
  8. //
  9. // A utility for creating static and dynamic libraries for Darwin.
  10. //
  11. //===----------------------------------------------------------------------===//
  12. #include "llvm/BinaryFormat/Magic.h"
  13. #include "llvm/IR/LLVMContext.h"
  14. #include "llvm/Object/ArchiveWriter.h"
  15. #include "llvm/Object/IRObjectFile.h"
  16. #include "llvm/Object/MachO.h"
  17. #include "llvm/Object/MachOUniversal.h"
  18. #include "llvm/Object/MachOUniversalWriter.h"
  19. #include "llvm/Object/ObjectFile.h"
  20. #include "llvm/Support/CommandLine.h"
  21. #include "llvm/Support/InitLLVM.h"
  22. #include "llvm/Support/LineIterator.h"
  23. #include "llvm/Support/VirtualFileSystem.h"
  24. #include "llvm/Support/WithColor.h"
  25. #include "llvm/Support/raw_ostream.h"
  26. #include "llvm/TextAPI/Architecture.h"
  27. #include <map>
  28. #include <type_traits>
  29. using namespace llvm;
  30. using namespace llvm::object;
  31. static LLVMContext LLVMCtx;
  32. class NewArchiveMemberList;
  33. typedef std::map<uint64_t, NewArchiveMemberList> MembersPerArchitectureMap;
  34. cl::OptionCategory LibtoolCategory("llvm-libtool-darwin Options");
  35. static cl::opt<std::string> OutputFile("o", cl::desc("Specify output filename"),
  36. cl::value_desc("filename"),
  37. cl::cat(LibtoolCategory));
  38. static cl::list<std::string> InputFiles(cl::Positional,
  39. cl::desc("<input files>"),
  40. cl::ZeroOrMore,
  41. cl::cat(LibtoolCategory));
  42. static cl::opt<std::string> ArchType(
  43. "arch_only", cl::desc("Specify architecture type for output library"),
  44. cl::value_desc("arch_type"), cl::ZeroOrMore, cl::cat(LibtoolCategory));
  45. enum class Operation { None, Static };
  46. static cl::opt<Operation> LibraryOperation(
  47. cl::desc("Library Type: "),
  48. cl::values(
  49. clEnumValN(Operation::Static, "static",
  50. "Produce a statically linked library from the input files")),
  51. cl::init(Operation::None), cl::cat(LibtoolCategory));
  52. static cl::opt<bool> DeterministicOption(
  53. "D", cl::desc("Use zero for timestamps and UIDs/GIDs (Default)"),
  54. cl::init(false), cl::cat(LibtoolCategory));
  55. static cl::opt<bool>
  56. NonDeterministicOption("U", cl::desc("Use actual timestamps and UIDs/GIDs"),
  57. cl::init(false), cl::cat(LibtoolCategory));
  58. static cl::opt<std::string>
  59. FileList("filelist",
  60. cl::desc("Pass in file containing a list of filenames"),
  61. cl::value_desc("listfile[,dirname]"), cl::cat(LibtoolCategory));
  62. static cl::list<std::string> Libraries(
  63. "l",
  64. cl::desc(
  65. "l<x> searches for the library libx.a in the library search path. If"
  66. " the string 'x' ends with '.o', then the library 'x' is searched for"
  67. " without prepending 'lib' or appending '.a'"),
  68. cl::ZeroOrMore, cl::Prefix, cl::cat(LibtoolCategory));
  69. static cl::list<std::string> LibrarySearchDirs(
  70. "L",
  71. cl::desc(
  72. "L<dir> adds <dir> to the list of directories in which to search for"
  73. " libraries"),
  74. cl::ZeroOrMore, cl::Prefix, cl::cat(LibtoolCategory));
  75. static cl::opt<bool>
  76. VersionOption("V", cl::desc("Print the version number and exit"),
  77. cl::cat(LibtoolCategory));
  78. static cl::opt<bool> NoWarningForNoSymbols(
  79. "no_warning_for_no_symbols",
  80. cl::desc("Do not warn about files that have no symbols"),
  81. cl::cat(LibtoolCategory), cl::init(false));
  82. static const std::array<std::string, 3> StandardSearchDirs{
  83. "/lib",
  84. "/usr/lib",
  85. "/usr/local/lib",
  86. };
  87. struct Config {
  88. bool Deterministic = true; // Updated by 'D' and 'U' modifiers.
  89. uint32_t ArchCPUType;
  90. uint32_t ArchCPUSubtype;
  91. };
  92. static Expected<std::string> searchForFile(const Twine &FileName) {
  93. auto FindLib =
  94. [FileName](ArrayRef<std::string> SearchDirs) -> Optional<std::string> {
  95. for (StringRef Dir : SearchDirs) {
  96. SmallString<128> Path;
  97. sys::path::append(Path, Dir, FileName);
  98. if (sys::fs::exists(Path))
  99. return std::string(Path);
  100. }
  101. return None;
  102. };
  103. Optional<std::string> Found = FindLib(LibrarySearchDirs);
  104. if (!Found)
  105. Found = FindLib(StandardSearchDirs);
  106. if (Found)
  107. return *Found;
  108. return createStringError(std::errc::invalid_argument,
  109. "cannot locate file '%s'", FileName.str().c_str());
  110. }
  111. static Error processCommandLineLibraries() {
  112. for (StringRef BaseName : Libraries) {
  113. Expected<std::string> FullPath = searchForFile(
  114. BaseName.endswith(".o") ? BaseName.str() : "lib" + BaseName + ".a");
  115. if (!FullPath)
  116. return FullPath.takeError();
  117. InputFiles.push_back(FullPath.get());
  118. }
  119. return Error::success();
  120. }
  121. static Error processFileList() {
  122. StringRef FileName, DirName;
  123. std::tie(FileName, DirName) = StringRef(FileList).rsplit(",");
  124. ErrorOr<std::unique_ptr<MemoryBuffer>> FileOrErr =
  125. MemoryBuffer::getFileOrSTDIN(FileName, /*IsText=*/false,
  126. /*RequiresNullTerminator=*/false);
  127. if (std::error_code EC = FileOrErr.getError())
  128. return createFileError(FileName, errorCodeToError(EC));
  129. const MemoryBuffer &Ref = *FileOrErr.get();
  130. line_iterator I(Ref, /*SkipBlanks=*/false);
  131. if (I.is_at_eof())
  132. return createStringError(std::errc::invalid_argument,
  133. "file list file: '%s' is empty",
  134. FileName.str().c_str());
  135. for (; !I.is_at_eof(); ++I) {
  136. StringRef Line = *I;
  137. if (Line.empty())
  138. return createStringError(std::errc::invalid_argument,
  139. "file list file: '%s': filename cannot be empty",
  140. FileName.str().c_str());
  141. SmallString<128> Path;
  142. if (!DirName.empty())
  143. sys::path::append(Path, DirName, Line);
  144. else
  145. sys::path::append(Path, Line);
  146. InputFiles.push_back(static_cast<std::string>(Path));
  147. }
  148. return Error::success();
  149. }
  150. static Error validateArchitectureName(StringRef ArchitectureName) {
  151. if (!MachOObjectFile::isValidArch(ArchitectureName)) {
  152. std::string Buf;
  153. raw_string_ostream OS(Buf);
  154. for (StringRef Arch : MachOObjectFile::getValidArchs())
  155. OS << Arch << " ";
  156. return createStringError(
  157. std::errc::invalid_argument,
  158. "invalid architecture '%s': valid architecture names are %s",
  159. ArchitectureName.str().c_str(), OS.str().c_str());
  160. }
  161. return Error::success();
  162. }
  163. static uint64_t getCPUID(uint32_t CPUType, uint32_t CPUSubtype) {
  164. switch (CPUType) {
  165. case MachO::CPU_TYPE_ARM:
  166. case MachO::CPU_TYPE_ARM64:
  167. case MachO::CPU_TYPE_ARM64_32:
  168. case MachO::CPU_TYPE_X86_64:
  169. // We consider CPUSubtype only for the above 4 CPUTypes to match cctools'
  170. // libtool behavior.
  171. return static_cast<uint64_t>(CPUType) << 32 | CPUSubtype;
  172. default:
  173. return CPUType;
  174. }
  175. }
  176. // MembersData is an organized collection of members.
  177. struct MembersData {
  178. // MembersPerArchitectureMap is a mapping from CPU architecture to a list of
  179. // members.
  180. MembersPerArchitectureMap MembersPerArchitecture;
  181. std::vector<std::unique_ptr<MemoryBuffer>> FileBuffers;
  182. };
  183. // NewArchiveMemberList instances serve as collections of archive members and
  184. // information about those members.
  185. class NewArchiveMemberList {
  186. std::vector<NewArchiveMember> Members;
  187. // This vector contains the file that each NewArchiveMember from Members came
  188. // from. Therefore, it has the same size as Members.
  189. std::vector<StringRef> Files;
  190. public:
  191. // Add a NewArchiveMember and the file it came from to the list.
  192. void push_back(NewArchiveMember &&Member, StringRef File) {
  193. Members.push_back(std::move(Member));
  194. Files.push_back(File);
  195. }
  196. ArrayRef<NewArchiveMember> getMembers() const { return Members; }
  197. ArrayRef<StringRef> getFiles() const { return Files; }
  198. static_assert(
  199. std::is_same<decltype(MembersData::MembersPerArchitecture)::mapped_type,
  200. NewArchiveMemberList>(),
  201. "This test makes sure NewArchiveMemberList is used by MembersData since "
  202. "the following asserts test invariants required for MembersData.");
  203. static_assert(
  204. !std::is_copy_constructible<
  205. decltype(NewArchiveMemberList::Members)::value_type>::value,
  206. "MembersData::MembersPerArchitecture has a dependency on "
  207. "MembersData::FileBuffers so it should not be able to "
  208. "be copied on its own without FileBuffers. Unfortunately, "
  209. "is_copy_constructible does not detect whether the container (ie vector) "
  210. "of a non-copyable type is itself non-copyable so we have to test the "
  211. "actual type of the stored data (ie, value_type).");
  212. static_assert(
  213. !std::is_copy_assignable<
  214. decltype(NewArchiveMemberList::Members)::value_type>::value,
  215. "MembersData::MembersPerArchitecture has a dependency on "
  216. "MembersData::FileBuffers so it should not be able to "
  217. "be copied on its own without FileBuffers. Unfortunately, "
  218. "is_copy_constructible does not detect whether the container (ie vector) "
  219. "of a non-copyable type is itself non-copyable so we have to test the "
  220. "actual type of the stored data (ie, value_type).");
  221. };
  222. // MembersBuilder collects and organizes all members from the files provided by
  223. // the user.
  224. class MembersBuilder {
  225. public:
  226. MembersBuilder(const Config &C) : C(C) {}
  227. Expected<MembersData> build() {
  228. for (StringRef FileName : InputFiles)
  229. if (Error E = AddMember(*this, FileName)())
  230. return std::move(E);
  231. if (!ArchType.empty()) {
  232. uint64_t ArchCPUID = getCPUID(C.ArchCPUType, C.ArchCPUSubtype);
  233. if (Data.MembersPerArchitecture.find(ArchCPUID) ==
  234. Data.MembersPerArchitecture.end())
  235. return createStringError(std::errc::invalid_argument,
  236. "no library created (no object files in input "
  237. "files matching -arch_only %s)",
  238. ArchType.c_str());
  239. }
  240. return std::move(Data);
  241. }
  242. private:
  243. class AddMember {
  244. MembersBuilder &Builder;
  245. StringRef FileName;
  246. public:
  247. AddMember(MembersBuilder &Builder, StringRef FileName)
  248. : Builder(Builder), FileName(FileName) {}
  249. Error operator()() {
  250. Expected<NewArchiveMember> NewMemberOrErr =
  251. NewArchiveMember::getFile(FileName, Builder.C.Deterministic);
  252. if (!NewMemberOrErr)
  253. return createFileError(FileName, NewMemberOrErr.takeError());
  254. auto &NewMember = *NewMemberOrErr;
  255. // For regular archives, use the basename of the object path for the
  256. // member name.
  257. NewMember.MemberName = sys::path::filename(NewMember.MemberName);
  258. file_magic Magic = identify_magic(NewMember.Buf->getBuffer());
  259. // Flatten archives.
  260. if (Magic == file_magic::archive)
  261. return addArchiveMembers(std::move(NewMember));
  262. // Flatten universal files.
  263. if (Magic == file_magic::macho_universal_binary)
  264. return addUniversalMembers(std::move(NewMember));
  265. // Bitcode files.
  266. if (Magic == file_magic::bitcode)
  267. return verifyAndAddIRObject(std::move(NewMember));
  268. return verifyAndAddMachOObject(std::move(NewMember));
  269. }
  270. private:
  271. // Check that a file's architecture [FileCPUType, FileCPUSubtype]
  272. // matches the architecture specified under -arch_only flag.
  273. bool acceptFileArch(uint32_t FileCPUType, uint32_t FileCPUSubtype) {
  274. if (Builder.C.ArchCPUType != FileCPUType)
  275. return false;
  276. switch (Builder.C.ArchCPUType) {
  277. case MachO::CPU_TYPE_ARM:
  278. case MachO::CPU_TYPE_ARM64_32:
  279. case MachO::CPU_TYPE_X86_64:
  280. return Builder.C.ArchCPUSubtype == FileCPUSubtype;
  281. case MachO::CPU_TYPE_ARM64:
  282. if (Builder.C.ArchCPUSubtype == MachO::CPU_SUBTYPE_ARM64_ALL)
  283. return FileCPUSubtype == MachO::CPU_SUBTYPE_ARM64_ALL ||
  284. FileCPUSubtype == MachO::CPU_SUBTYPE_ARM64_V8;
  285. else
  286. return Builder.C.ArchCPUSubtype == FileCPUSubtype;
  287. default:
  288. return true;
  289. }
  290. }
  291. Error verifyAndAddMachOObject(NewArchiveMember Member) {
  292. auto MBRef = Member.Buf->getMemBufferRef();
  293. Expected<std::unique_ptr<object::ObjectFile>> ObjOrErr =
  294. object::ObjectFile::createObjectFile(MBRef);
  295. // Throw error if not a valid object file.
  296. if (!ObjOrErr)
  297. return createFileError(Member.MemberName, ObjOrErr.takeError());
  298. // Throw error if not in Mach-O format.
  299. if (!isa<object::MachOObjectFile>(**ObjOrErr))
  300. return createStringError(std::errc::invalid_argument,
  301. "'%s': format not supported",
  302. Member.MemberName.data());
  303. auto *O = dyn_cast<MachOObjectFile>(ObjOrErr->get());
  304. uint32_t FileCPUType, FileCPUSubtype;
  305. std::tie(FileCPUType, FileCPUSubtype) = MachO::getCPUTypeFromArchitecture(
  306. MachO::getArchitectureFromName(O->getArchTriple().getArchName()));
  307. // If -arch_only is specified then skip this file if it doesn't match
  308. // the architecture specified.
  309. if (!ArchType.empty() && !acceptFileArch(FileCPUType, FileCPUSubtype)) {
  310. return Error::success();
  311. }
  312. if (!NoWarningForNoSymbols && O->symbols().empty())
  313. WithColor::warning() << Member.MemberName + " has no symbols\n";
  314. uint64_t FileCPUID = getCPUID(FileCPUType, FileCPUSubtype);
  315. Builder.Data.MembersPerArchitecture[FileCPUID].push_back(
  316. std::move(Member), FileName);
  317. return Error::success();
  318. }
  319. Error verifyAndAddIRObject(NewArchiveMember Member) {
  320. auto MBRef = Member.Buf->getMemBufferRef();
  321. Expected<std::unique_ptr<object::IRObjectFile>> IROrErr =
  322. object::IRObjectFile::create(MBRef, LLVMCtx);
  323. // Throw error if not a valid IR object file.
  324. if (!IROrErr)
  325. return createFileError(Member.MemberName, IROrErr.takeError());
  326. Triple TT = Triple(IROrErr->get()->getTargetTriple());
  327. Expected<uint32_t> FileCPUTypeOrErr = MachO::getCPUType(TT);
  328. if (!FileCPUTypeOrErr)
  329. return FileCPUTypeOrErr.takeError();
  330. Expected<uint32_t> FileCPUSubTypeOrErr = MachO::getCPUSubType(TT);
  331. if (!FileCPUSubTypeOrErr)
  332. return FileCPUSubTypeOrErr.takeError();
  333. // If -arch_only is specified then skip this file if it doesn't match
  334. // the architecture specified.
  335. if (!ArchType.empty() &&
  336. !acceptFileArch(*FileCPUTypeOrErr, *FileCPUSubTypeOrErr)) {
  337. return Error::success();
  338. }
  339. uint64_t FileCPUID = getCPUID(*FileCPUTypeOrErr, *FileCPUSubTypeOrErr);
  340. Builder.Data.MembersPerArchitecture[FileCPUID].push_back(
  341. std::move(Member), FileName);
  342. return Error::success();
  343. }
  344. Error addChildMember(const object::Archive::Child &M) {
  345. Expected<NewArchiveMember> NewMemberOrErr =
  346. NewArchiveMember::getOldMember(M, Builder.C.Deterministic);
  347. if (!NewMemberOrErr)
  348. return NewMemberOrErr.takeError();
  349. auto &NewMember = *NewMemberOrErr;
  350. file_magic Magic = identify_magic(NewMember.Buf->getBuffer());
  351. if (Magic == file_magic::bitcode)
  352. return verifyAndAddIRObject(std::move(NewMember));
  353. return verifyAndAddMachOObject(std::move(NewMember));
  354. }
  355. Error processArchive(object::Archive &Lib) {
  356. Error Err = Error::success();
  357. for (const object::Archive::Child &Child : Lib.children(Err))
  358. if (Error E = addChildMember(Child))
  359. return createFileError(FileName, std::move(E));
  360. if (Err)
  361. return createFileError(FileName, std::move(Err));
  362. return Error::success();
  363. }
  364. Error addArchiveMembers(NewArchiveMember NewMember) {
  365. Expected<std::unique_ptr<Archive>> LibOrErr =
  366. object::Archive::create(NewMember.Buf->getMemBufferRef());
  367. if (!LibOrErr)
  368. return createFileError(FileName, LibOrErr.takeError());
  369. if (Error E = processArchive(**LibOrErr))
  370. return E;
  371. // Update vector FileBuffers with the MemoryBuffers to transfer
  372. // ownership.
  373. Builder.Data.FileBuffers.push_back(std::move(NewMember.Buf));
  374. return Error::success();
  375. }
  376. Error addUniversalMembers(NewArchiveMember NewMember) {
  377. Expected<std::unique_ptr<MachOUniversalBinary>> BinaryOrErr =
  378. MachOUniversalBinary::create(NewMember.Buf->getMemBufferRef());
  379. if (!BinaryOrErr)
  380. return createFileError(FileName, BinaryOrErr.takeError());
  381. auto *UO = BinaryOrErr->get();
  382. for (const MachOUniversalBinary::ObjectForArch &O : UO->objects()) {
  383. Expected<std::unique_ptr<MachOObjectFile>> MachOObjOrErr =
  384. O.getAsObjectFile();
  385. if (MachOObjOrErr) {
  386. NewArchiveMember NewMember =
  387. NewArchiveMember(MachOObjOrErr->get()->getMemoryBufferRef());
  388. NewMember.MemberName = sys::path::filename(NewMember.MemberName);
  389. if (Error E = verifyAndAddMachOObject(std::move(NewMember)))
  390. return E;
  391. continue;
  392. }
  393. Expected<std::unique_ptr<IRObjectFile>> IRObjectOrError =
  394. O.getAsIRObject(LLVMCtx);
  395. if (IRObjectOrError) {
  396. // A universal file member can be a MachOObjectFile, an IRObject or an
  397. // Archive. In case we can successfully cast the member as an
  398. // IRObject, it is safe to throw away the error generated due to
  399. // casting the object as a MachOObjectFile.
  400. consumeError(MachOObjOrErr.takeError());
  401. NewArchiveMember NewMember =
  402. NewArchiveMember(IRObjectOrError->get()->getMemoryBufferRef());
  403. NewMember.MemberName = sys::path::filename(NewMember.MemberName);
  404. if (Error E = verifyAndAddIRObject(std::move(NewMember)))
  405. return E;
  406. continue;
  407. }
  408. Expected<std::unique_ptr<Archive>> ArchiveOrError = O.getAsArchive();
  409. if (ArchiveOrError) {
  410. // A universal file member can be a MachOObjectFile, an IRObject or an
  411. // Archive. In case we can successfully cast the member as an Archive,
  412. // it is safe to throw away the error generated due to casting the
  413. // object as a MachOObjectFile.
  414. consumeError(MachOObjOrErr.takeError());
  415. consumeError(IRObjectOrError.takeError());
  416. if (Error E = processArchive(**ArchiveOrError))
  417. return E;
  418. continue;
  419. }
  420. Error CombinedError = joinErrors(
  421. ArchiveOrError.takeError(),
  422. joinErrors(IRObjectOrError.takeError(), MachOObjOrErr.takeError()));
  423. return createFileError(FileName, std::move(CombinedError));
  424. }
  425. // Update vector FileBuffers with the MemoryBuffers to transfer
  426. // ownership.
  427. Builder.Data.FileBuffers.push_back(std::move(NewMember.Buf));
  428. return Error::success();
  429. }
  430. };
  431. MembersData Data;
  432. const Config &C;
  433. };
  434. static Expected<SmallVector<Slice, 2>>
  435. buildSlices(ArrayRef<OwningBinary<Archive>> OutputBinaries) {
  436. SmallVector<Slice, 2> Slices;
  437. for (const auto &OB : OutputBinaries) {
  438. const Archive &A = *OB.getBinary();
  439. Expected<Slice> ArchiveSlice = Slice::create(A, &LLVMCtx);
  440. if (!ArchiveSlice)
  441. return ArchiveSlice.takeError();
  442. Slices.push_back(*ArchiveSlice);
  443. }
  444. return Slices;
  445. }
  446. static Error
  447. checkForDuplicates(const MembersPerArchitectureMap &MembersPerArch) {
  448. for (const auto &M : MembersPerArch) {
  449. ArrayRef<NewArchiveMember> Members = M.second.getMembers();
  450. ArrayRef<StringRef> Files = M.second.getFiles();
  451. StringMap<std::vector<StringRef>> MembersToFiles;
  452. for (auto Iterators = std::make_pair(Members.begin(), Files.begin());
  453. Iterators.first != Members.end();
  454. ++Iterators.first, ++Iterators.second) {
  455. assert(Iterators.second != Files.end() &&
  456. "Files should be the same size as Members.");
  457. MembersToFiles[Iterators.first->MemberName].push_back(*Iterators.second);
  458. }
  459. std::string ErrorData;
  460. raw_string_ostream ErrorStream(ErrorData);
  461. for (const auto &MemberToFile : MembersToFiles) {
  462. if (MemberToFile.getValue().size() > 1) {
  463. ErrorStream << "file '" << MemberToFile.getKey().str()
  464. << "' was specified multiple times.\n";
  465. for (StringRef OriginalFile : MemberToFile.getValue())
  466. ErrorStream << "in: " << OriginalFile.str() << '\n';
  467. ErrorStream << '\n';
  468. }
  469. }
  470. ErrorStream.flush();
  471. if (ErrorData.size() > 0)
  472. return createStringError(std::errc::invalid_argument, ErrorData.c_str());
  473. }
  474. return Error::success();
  475. }
  476. static Error createStaticLibrary(const Config &C) {
  477. MembersBuilder Builder(C);
  478. auto DataOrError = Builder.build();
  479. if (auto Error = DataOrError.takeError())
  480. return Error;
  481. const auto &NewMembers = DataOrError->MembersPerArchitecture;
  482. if (Error E = checkForDuplicates(NewMembers))
  483. WithColor::defaultWarningHandler(std::move(E));
  484. if (NewMembers.size() == 1)
  485. return writeArchive(OutputFile, NewMembers.begin()->second.getMembers(),
  486. /*WriteSymtab=*/true,
  487. /*Kind=*/object::Archive::K_DARWIN, C.Deterministic,
  488. /*Thin=*/false);
  489. SmallVector<OwningBinary<Archive>, 2> OutputBinaries;
  490. for (const std::pair<const uint64_t, NewArchiveMemberList> &M : NewMembers) {
  491. Expected<std::unique_ptr<MemoryBuffer>> OutputBufferOrErr =
  492. writeArchiveToBuffer(M.second.getMembers(),
  493. /*WriteSymtab=*/true,
  494. /*Kind=*/object::Archive::K_DARWIN,
  495. C.Deterministic,
  496. /*Thin=*/false);
  497. if (!OutputBufferOrErr)
  498. return OutputBufferOrErr.takeError();
  499. std::unique_ptr<MemoryBuffer> &OutputBuffer = OutputBufferOrErr.get();
  500. Expected<std::unique_ptr<Archive>> ArchiveOrError =
  501. Archive::create(OutputBuffer->getMemBufferRef());
  502. if (!ArchiveOrError)
  503. return ArchiveOrError.takeError();
  504. std::unique_ptr<Archive> &A = ArchiveOrError.get();
  505. OutputBinaries.push_back(
  506. OwningBinary<Archive>(std::move(A), std::move(OutputBuffer)));
  507. }
  508. Expected<SmallVector<Slice, 2>> Slices = buildSlices(OutputBinaries);
  509. if (!Slices)
  510. return Slices.takeError();
  511. llvm::stable_sort(*Slices);
  512. return writeUniversalBinary(*Slices, OutputFile);
  513. }
  514. static Expected<Config> parseCommandLine(int Argc, char **Argv) {
  515. Config C;
  516. cl::ParseCommandLineOptions(Argc, Argv, "llvm-libtool-darwin\n");
  517. if (LibraryOperation == Operation::None) {
  518. if (!VersionOption) {
  519. std::string Error;
  520. raw_string_ostream Stream(Error);
  521. LibraryOperation.error("must be specified", "", Stream);
  522. return createStringError(std::errc::invalid_argument, Error.c_str());
  523. }
  524. return C;
  525. }
  526. if (OutputFile.empty()) {
  527. std::string Error;
  528. raw_string_ostream Stream(Error);
  529. OutputFile.error("must be specified", "o", Stream);
  530. return createStringError(std::errc::invalid_argument, Error.c_str());
  531. }
  532. if (DeterministicOption && NonDeterministicOption)
  533. return createStringError(std::errc::invalid_argument,
  534. "cannot specify both -D and -U flags");
  535. else if (NonDeterministicOption)
  536. C.Deterministic = false;
  537. if (!Libraries.empty())
  538. if (Error E = processCommandLineLibraries())
  539. return std::move(E);
  540. if (!FileList.empty())
  541. if (Error E = processFileList())
  542. return std::move(E);
  543. if (InputFiles.empty())
  544. return createStringError(std::errc::invalid_argument,
  545. "no input files specified");
  546. if (ArchType.getNumOccurrences()) {
  547. if (Error E = validateArchitectureName(ArchType))
  548. return std::move(E);
  549. std::tie(C.ArchCPUType, C.ArchCPUSubtype) =
  550. MachO::getCPUTypeFromArchitecture(
  551. MachO::getArchitectureFromName(ArchType));
  552. }
  553. return C;
  554. }
  555. int main(int Argc, char **Argv) {
  556. InitLLVM X(Argc, Argv);
  557. cl::HideUnrelatedOptions({&LibtoolCategory, &getColorCategory()});
  558. Expected<Config> ConfigOrErr = parseCommandLine(Argc, Argv);
  559. if (!ConfigOrErr) {
  560. WithColor::defaultErrorHandler(ConfigOrErr.takeError());
  561. return EXIT_FAILURE;
  562. }
  563. if (VersionOption)
  564. cl::PrintVersionMessage();
  565. Config C = *ConfigOrErr;
  566. switch (LibraryOperation) {
  567. case Operation::None:
  568. break;
  569. case Operation::Static:
  570. if (Error E = createStaticLibrary(C)) {
  571. WithColor::defaultErrorHandler(std::move(E));
  572. return EXIT_FAILURE;
  573. }
  574. break;
  575. }
  576. }