llvm-libtool-darwin.cpp 26 KB

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