xref: /freebsd/contrib/llvm-project/lld/MachO/Driver.cpp (revision 0fca6ea1d4eea4c934cfff25ac9ee8ad6fe95583)
15ffd83dbSDimitry Andric //===- Driver.cpp ---------------------------------------------------------===//
25ffd83dbSDimitry Andric //
35ffd83dbSDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
45ffd83dbSDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
55ffd83dbSDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
65ffd83dbSDimitry Andric //
75ffd83dbSDimitry Andric //===----------------------------------------------------------------------===//
85ffd83dbSDimitry Andric 
95ffd83dbSDimitry Andric #include "Driver.h"
105ffd83dbSDimitry Andric #include "Config.h"
11fe6060f1SDimitry Andric #include "ICF.h"
125ffd83dbSDimitry Andric #include "InputFiles.h"
13e8d8bef9SDimitry Andric #include "LTO.h"
14fe6060f1SDimitry Andric #include "MarkLive.h"
15e8d8bef9SDimitry Andric #include "ObjC.h"
165ffd83dbSDimitry Andric #include "OutputSection.h"
175ffd83dbSDimitry Andric #include "OutputSegment.h"
1804eeddc0SDimitry Andric #include "SectionPriorities.h"
195ffd83dbSDimitry Andric #include "SymbolTable.h"
205ffd83dbSDimitry Andric #include "Symbols.h"
21e8d8bef9SDimitry Andric #include "SyntheticSections.h"
225ffd83dbSDimitry Andric #include "Target.h"
23fe6060f1SDimitry Andric #include "UnwindInfoSection.h"
245ffd83dbSDimitry Andric #include "Writer.h"
255ffd83dbSDimitry Andric 
265ffd83dbSDimitry Andric #include "lld/Common/Args.h"
27bdd1243dSDimitry Andric #include "lld/Common/CommonLinkerContext.h"
285ffd83dbSDimitry Andric #include "lld/Common/Driver.h"
295ffd83dbSDimitry Andric #include "lld/Common/ErrorHandler.h"
305ffd83dbSDimitry Andric #include "lld/Common/LLVM.h"
315ffd83dbSDimitry Andric #include "lld/Common/Memory.h"
32e8d8bef9SDimitry Andric #include "lld/Common/Reproduce.h"
335ffd83dbSDimitry Andric #include "lld/Common/Version.h"
345ffd83dbSDimitry Andric #include "llvm/ADT/DenseSet.h"
355ffd83dbSDimitry Andric #include "llvm/ADT/StringExtras.h"
365ffd83dbSDimitry Andric #include "llvm/ADT/StringRef.h"
375ffd83dbSDimitry Andric #include "llvm/BinaryFormat/MachO.h"
385ffd83dbSDimitry Andric #include "llvm/BinaryFormat/Magic.h"
39fe6060f1SDimitry Andric #include "llvm/Config/llvm-config.h"
40e8d8bef9SDimitry Andric #include "llvm/LTO/LTO.h"
415ffd83dbSDimitry Andric #include "llvm/Object/Archive.h"
425ffd83dbSDimitry Andric #include "llvm/Option/ArgList.h"
43e8d8bef9SDimitry Andric #include "llvm/Support/CommandLine.h"
44e8d8bef9SDimitry Andric #include "llvm/Support/FileSystem.h"
455ffd83dbSDimitry Andric #include "llvm/Support/MemoryBuffer.h"
46fe6060f1SDimitry Andric #include "llvm/Support/Parallel.h"
475ffd83dbSDimitry Andric #include "llvm/Support/Path.h"
48e8d8bef9SDimitry Andric #include "llvm/Support/TarWriter.h"
49e8d8bef9SDimitry Andric #include "llvm/Support/TargetSelect.h"
50fe6060f1SDimitry Andric #include "llvm/Support/TimeProfiler.h"
5106c3fb27SDimitry Andric #include "llvm/TargetParser/Host.h"
52*0fca6ea1SDimitry Andric #include "llvm/TextAPI/Architecture.h"
53fe6060f1SDimitry Andric #include "llvm/TextAPI/PackedVersion.h"
54e8d8bef9SDimitry Andric 
55e8d8bef9SDimitry Andric #include <algorithm>
565ffd83dbSDimitry Andric 
575ffd83dbSDimitry Andric using namespace llvm;
585ffd83dbSDimitry Andric using namespace llvm::MachO;
59e8d8bef9SDimitry Andric using namespace llvm::object;
605ffd83dbSDimitry Andric using namespace llvm::opt;
61e8d8bef9SDimitry Andric using namespace llvm::sys;
625ffd83dbSDimitry Andric using namespace lld;
635ffd83dbSDimitry Andric using namespace lld::macho;
645ffd83dbSDimitry Andric 
6504eeddc0SDimitry Andric std::unique_ptr<Configuration> macho::config;
6604eeddc0SDimitry Andric std::unique_ptr<DependencyTracker> macho::depTracker;
675ffd83dbSDimitry Andric 
getOutputType(const InputArgList & args)68fe6060f1SDimitry Andric static HeaderFileType getOutputType(const InputArgList &args) {
69e8d8bef9SDimitry Andric   // TODO: -r, -dylinker, -preload...
70fe6060f1SDimitry Andric   Arg *outputArg = args.getLastArg(OPT_bundle, OPT_dylib, OPT_execute);
71e8d8bef9SDimitry Andric   if (outputArg == nullptr)
72e8d8bef9SDimitry Andric     return MH_EXECUTE;
735ffd83dbSDimitry Andric 
74e8d8bef9SDimitry Andric   switch (outputArg->getOption().getID()) {
75e8d8bef9SDimitry Andric   case OPT_bundle:
76e8d8bef9SDimitry Andric     return MH_BUNDLE;
77e8d8bef9SDimitry Andric   case OPT_dylib:
78e8d8bef9SDimitry Andric     return MH_DYLIB;
79e8d8bef9SDimitry Andric   case OPT_execute:
80e8d8bef9SDimitry Andric     return MH_EXECUTE;
81e8d8bef9SDimitry Andric   default:
82e8d8bef9SDimitry Andric     llvm_unreachable("internal error");
83e8d8bef9SDimitry Andric   }
845ffd83dbSDimitry Andric }
855ffd83dbSDimitry Andric 
86349cc55cSDimitry Andric static DenseMap<CachedHashStringRef, StringRef> resolvedLibraries;
findLibrary(StringRef name)87bdd1243dSDimitry Andric static std::optional<StringRef> findLibrary(StringRef name) {
88349cc55cSDimitry Andric   CachedHashStringRef key(name);
89349cc55cSDimitry Andric   auto entry = resolvedLibraries.find(key);
90349cc55cSDimitry Andric   if (entry != resolvedLibraries.end())
91349cc55cSDimitry Andric     return entry->second;
92349cc55cSDimitry Andric 
93349cc55cSDimitry Andric   auto doFind = [&] {
947a6dacacSDimitry Andric     // Special case for Csu support files required for Mac OS X 10.7 and older
957a6dacacSDimitry Andric     // (crt1.o)
967a6dacacSDimitry Andric     if (name.ends_with(".o"))
977a6dacacSDimitry Andric       return findPathCombination(name, config->librarySearchPaths, {""});
98e8d8bef9SDimitry Andric     if (config->searchDylibsFirst) {
9906c3fb27SDimitry Andric       if (std::optional<StringRef> path =
10006c3fb27SDimitry Andric               findPathCombination("lib" + name, config->librarySearchPaths,
10106c3fb27SDimitry Andric                                   {".tbd", ".dylib", ".so"}))
102e8d8bef9SDimitry Andric         return path;
103fe6060f1SDimitry Andric       return findPathCombination("lib" + name, config->librarySearchPaths,
104fe6060f1SDimitry Andric                                  {".a"});
1055ffd83dbSDimitry Andric     }
106fe6060f1SDimitry Andric     return findPathCombination("lib" + name, config->librarySearchPaths,
10706c3fb27SDimitry Andric                                {".tbd", ".dylib", ".so", ".a"});
108349cc55cSDimitry Andric   };
109349cc55cSDimitry Andric 
110bdd1243dSDimitry Andric   std::optional<StringRef> path = doFind();
111349cc55cSDimitry Andric   if (path)
112349cc55cSDimitry Andric     resolvedLibraries[key] = *path;
113349cc55cSDimitry Andric 
114349cc55cSDimitry Andric   return path;
115e8d8bef9SDimitry Andric }
116e8d8bef9SDimitry Andric 
117349cc55cSDimitry Andric static DenseMap<CachedHashStringRef, StringRef> resolvedFrameworks;
findFramework(StringRef name)118bdd1243dSDimitry Andric static std::optional<StringRef> findFramework(StringRef name) {
119349cc55cSDimitry Andric   CachedHashStringRef key(name);
120349cc55cSDimitry Andric   auto entry = resolvedFrameworks.find(key);
121349cc55cSDimitry Andric   if (entry != resolvedFrameworks.end())
122349cc55cSDimitry Andric     return entry->second;
123349cc55cSDimitry Andric 
124e8d8bef9SDimitry Andric   SmallString<260> symlink;
125e8d8bef9SDimitry Andric   StringRef suffix;
126e8d8bef9SDimitry Andric   std::tie(name, suffix) = name.split(",");
127e8d8bef9SDimitry Andric   for (StringRef dir : config->frameworkSearchPaths) {
128e8d8bef9SDimitry Andric     symlink = dir;
129e8d8bef9SDimitry Andric     path::append(symlink, name + ".framework", name);
130e8d8bef9SDimitry Andric 
131e8d8bef9SDimitry Andric     if (!suffix.empty()) {
132e8d8bef9SDimitry Andric       // NOTE: we must resolve the symlink before trying the suffixes, because
133e8d8bef9SDimitry Andric       // there are no symlinks for the suffixed paths.
134e8d8bef9SDimitry Andric       SmallString<260> location;
135e8d8bef9SDimitry Andric       if (!fs::real_path(symlink, location)) {
136e8d8bef9SDimitry Andric         // only append suffix if realpath() succeeds
137e8d8bef9SDimitry Andric         Twine suffixed = location + suffix;
138e8d8bef9SDimitry Andric         if (fs::exists(suffixed))
13904eeddc0SDimitry Andric           return resolvedFrameworks[key] = saver().save(suffixed.str());
140e8d8bef9SDimitry Andric       }
141e8d8bef9SDimitry Andric       // Suffix lookup failed, fall through to the no-suffix case.
142e8d8bef9SDimitry Andric     }
143e8d8bef9SDimitry Andric 
144bdd1243dSDimitry Andric     if (std::optional<StringRef> path = resolveDylibPath(symlink.str()))
145349cc55cSDimitry Andric       return resolvedFrameworks[key] = *path;
1465ffd83dbSDimitry Andric   }
1475ffd83dbSDimitry Andric   return {};
1485ffd83dbSDimitry Andric }
1495ffd83dbSDimitry Andric 
warnIfNotDirectory(StringRef option,StringRef path)150e8d8bef9SDimitry Andric static bool warnIfNotDirectory(StringRef option, StringRef path) {
1515ffd83dbSDimitry Andric   if (!fs::exists(path)) {
1525ffd83dbSDimitry Andric     warn("directory not found for option -" + option + path);
1535ffd83dbSDimitry Andric     return false;
1545ffd83dbSDimitry Andric   } else if (!fs::is_directory(path)) {
1555ffd83dbSDimitry Andric     warn("option -" + option + path + " references a non-directory path");
1565ffd83dbSDimitry Andric     return false;
1575ffd83dbSDimitry Andric   }
1585ffd83dbSDimitry Andric   return true;
1595ffd83dbSDimitry Andric }
1605ffd83dbSDimitry Andric 
161e8d8bef9SDimitry Andric static std::vector<StringRef>
getSearchPaths(unsigned optionCode,InputArgList & args,const std::vector<StringRef> & roots,const SmallVector<StringRef,2> & systemPaths)162fe6060f1SDimitry Andric getSearchPaths(unsigned optionCode, InputArgList &args,
163e8d8bef9SDimitry Andric                const std::vector<StringRef> &roots,
1645ffd83dbSDimitry Andric                const SmallVector<StringRef, 2> &systemPaths) {
165e8d8bef9SDimitry Andric   std::vector<StringRef> paths;
166e8d8bef9SDimitry Andric   StringRef optionLetter{optionCode == OPT_F ? "F" : "L"};
167e8d8bef9SDimitry Andric   for (StringRef path : args::getStrings(args, optionCode)) {
168e8d8bef9SDimitry Andric     // NOTE: only absolute paths are re-rooted to syslibroot(s)
169e8d8bef9SDimitry Andric     bool found = false;
170e8d8bef9SDimitry Andric     if (path::is_absolute(path, path::Style::posix)) {
171e8d8bef9SDimitry Andric       for (StringRef root : roots) {
172e8d8bef9SDimitry Andric         SmallString<261> buffer(root);
173e8d8bef9SDimitry Andric         path::append(buffer, path);
174e8d8bef9SDimitry Andric         // Do not warn about paths that are computed via the syslib roots
175e8d8bef9SDimitry Andric         if (fs::is_directory(buffer)) {
17604eeddc0SDimitry Andric           paths.push_back(saver().save(buffer.str()));
177e8d8bef9SDimitry Andric           found = true;
178e8d8bef9SDimitry Andric         }
179e8d8bef9SDimitry Andric       }
180e8d8bef9SDimitry Andric     }
181e8d8bef9SDimitry Andric     if (!found && warnIfNotDirectory(optionLetter, path))
1825ffd83dbSDimitry Andric       paths.push_back(path);
1835ffd83dbSDimitry Andric   }
184e8d8bef9SDimitry Andric 
185e8d8bef9SDimitry Andric   // `-Z` suppresses the standard "system" search paths.
186e8d8bef9SDimitry Andric   if (args.hasArg(OPT_Z))
187e8d8bef9SDimitry Andric     return paths;
188e8d8bef9SDimitry Andric 
189fe6060f1SDimitry Andric   for (const StringRef &path : systemPaths) {
190fe6060f1SDimitry Andric     for (const StringRef &root : roots) {
191e8d8bef9SDimitry Andric       SmallString<261> buffer(root);
192e8d8bef9SDimitry Andric       path::append(buffer, path);
193e8d8bef9SDimitry Andric       if (fs::is_directory(buffer))
19404eeddc0SDimitry Andric         paths.push_back(saver().save(buffer.str()));
1955ffd83dbSDimitry Andric     }
1965ffd83dbSDimitry Andric   }
197e8d8bef9SDimitry Andric   return paths;
1985ffd83dbSDimitry Andric }
1995ffd83dbSDimitry Andric 
getSystemLibraryRoots(InputArgList & args)200fe6060f1SDimitry Andric static std::vector<StringRef> getSystemLibraryRoots(InputArgList &args) {
201e8d8bef9SDimitry Andric   std::vector<StringRef> roots;
202e8d8bef9SDimitry Andric   for (const Arg *arg : args.filtered(OPT_syslibroot))
203e8d8bef9SDimitry Andric     roots.push_back(arg->getValue());
204e8d8bef9SDimitry Andric   // NOTE: the final `-syslibroot` being `/` will ignore all roots
205349cc55cSDimitry Andric   if (!roots.empty() && roots.back() == "/")
206e8d8bef9SDimitry Andric     roots.clear();
207e8d8bef9SDimitry Andric   // NOTE: roots can never be empty - add an empty root to simplify the library
208e8d8bef9SDimitry Andric   // and framework search path computation.
209e8d8bef9SDimitry Andric   if (roots.empty())
210e8d8bef9SDimitry Andric     roots.emplace_back("");
211e8d8bef9SDimitry Andric   return roots;
2125ffd83dbSDimitry Andric }
2135ffd83dbSDimitry Andric 
214e8d8bef9SDimitry Andric static std::vector<StringRef>
getLibrarySearchPaths(InputArgList & args,const std::vector<StringRef> & roots)215fe6060f1SDimitry Andric getLibrarySearchPaths(InputArgList &args, const std::vector<StringRef> &roots) {
216e8d8bef9SDimitry Andric   return getSearchPaths(OPT_L, args, roots, {"/usr/lib", "/usr/local/lib"});
217e8d8bef9SDimitry Andric }
218e8d8bef9SDimitry Andric 
219e8d8bef9SDimitry Andric static std::vector<StringRef>
getFrameworkSearchPaths(InputArgList & args,const std::vector<StringRef> & roots)220fe6060f1SDimitry Andric getFrameworkSearchPaths(InputArgList &args,
221e8d8bef9SDimitry Andric                         const std::vector<StringRef> &roots) {
222e8d8bef9SDimitry Andric   return getSearchPaths(OPT_F, args, roots,
2235ffd83dbSDimitry Andric                         {"/Library/Frameworks", "/System/Library/Frameworks"});
2245ffd83dbSDimitry Andric }
2255ffd83dbSDimitry Andric 
getLTOCachePolicy(InputArgList & args)226fe6060f1SDimitry Andric static llvm::CachePruningPolicy getLTOCachePolicy(InputArgList &args) {
227fe6060f1SDimitry Andric   SmallString<128> ltoPolicy;
228fe6060f1SDimitry Andric   auto add = [&ltoPolicy](Twine val) {
229fe6060f1SDimitry Andric     if (!ltoPolicy.empty())
230fe6060f1SDimitry Andric       ltoPolicy += ":";
231fe6060f1SDimitry Andric     val.toVector(ltoPolicy);
232fe6060f1SDimitry Andric   };
233fe6060f1SDimitry Andric   for (const Arg *arg :
234bdd1243dSDimitry Andric        args.filtered(OPT_thinlto_cache_policy_eq, OPT_prune_interval_lto,
235fe6060f1SDimitry Andric                      OPT_prune_after_lto, OPT_max_relative_cache_size_lto)) {
236fe6060f1SDimitry Andric     switch (arg->getOption().getID()) {
237bdd1243dSDimitry Andric     case OPT_thinlto_cache_policy_eq:
238349cc55cSDimitry Andric       add(arg->getValue());
239349cc55cSDimitry Andric       break;
240fe6060f1SDimitry Andric     case OPT_prune_interval_lto:
241fe6060f1SDimitry Andric       if (!strcmp("-1", arg->getValue()))
242fe6060f1SDimitry Andric         add("prune_interval=87600h"); // 10 years
243fe6060f1SDimitry Andric       else
244fe6060f1SDimitry Andric         add(Twine("prune_interval=") + arg->getValue() + "s");
245fe6060f1SDimitry Andric       break;
246fe6060f1SDimitry Andric     case OPT_prune_after_lto:
247fe6060f1SDimitry Andric       add(Twine("prune_after=") + arg->getValue() + "s");
248fe6060f1SDimitry Andric       break;
249fe6060f1SDimitry Andric     case OPT_max_relative_cache_size_lto:
250fe6060f1SDimitry Andric       add(Twine("cache_size=") + arg->getValue() + "%");
251fe6060f1SDimitry Andric       break;
252fe6060f1SDimitry Andric     }
253fe6060f1SDimitry Andric   }
254fe6060f1SDimitry Andric   return CHECK(parseCachePruningPolicy(ltoPolicy), "invalid LTO cache policy");
255fe6060f1SDimitry Andric }
256fe6060f1SDimitry Andric 
257fcaf7f86SDimitry Andric // What caused a given library to be loaded. Only relevant for archives.
258fcaf7f86SDimitry Andric // Note that this does not tell us *how* we should load the library, i.e.
259fcaf7f86SDimitry Andric // whether we should do it lazily or eagerly (AKA force loading). The "how" is
260fcaf7f86SDimitry Andric // decided within addFile().
261fcaf7f86SDimitry Andric enum class LoadType {
262fcaf7f86SDimitry Andric   CommandLine,      // Library was passed as a regular CLI argument
263fcaf7f86SDimitry Andric   CommandLineForce, // Library was passed via `-force_load`
264fcaf7f86SDimitry Andric   LCLinkerOption,   // Library was passed via LC_LINKER_OPTIONS
265fcaf7f86SDimitry Andric };
266fe6060f1SDimitry Andric 
267fcaf7f86SDimitry Andric struct ArchiveFileInfo {
268fcaf7f86SDimitry Andric   ArchiveFile *file;
269fcaf7f86SDimitry Andric   bool isCommandLineLoad;
270fcaf7f86SDimitry Andric };
271fcaf7f86SDimitry Andric 
272fcaf7f86SDimitry Andric static DenseMap<StringRef, ArchiveFileInfo> loadedArchives;
273fcaf7f86SDimitry Andric 
saveThinArchiveToRepro(ArchiveFile const * file)274*0fca6ea1SDimitry Andric static void saveThinArchiveToRepro(ArchiveFile const *file) {
275*0fca6ea1SDimitry Andric   assert(tar && file->getArchive().isThin());
276*0fca6ea1SDimitry Andric 
277*0fca6ea1SDimitry Andric   Error e = Error::success();
278*0fca6ea1SDimitry Andric   for (const object::Archive::Child &c : file->getArchive().children(e)) {
279*0fca6ea1SDimitry Andric     MemoryBufferRef mb = CHECK(c.getMemoryBufferRef(),
280*0fca6ea1SDimitry Andric                                toString(file) + ": failed to get buffer");
281*0fca6ea1SDimitry Andric     tar->append(relativeToRoot(CHECK(c.getFullName(), file)), mb.getBuffer());
282*0fca6ea1SDimitry Andric   }
283*0fca6ea1SDimitry Andric   if (e)
284*0fca6ea1SDimitry Andric     error(toString(file) +
285*0fca6ea1SDimitry Andric           ": Archive::children failed: " + toString(std::move(e)));
286*0fca6ea1SDimitry Andric }
287*0fca6ea1SDimitry Andric 
addFile(StringRef path,LoadType loadType,bool isLazy=false,bool isExplicit=true,bool isBundleLoader=false,bool isForceHidden=false)288fcaf7f86SDimitry Andric static InputFile *addFile(StringRef path, LoadType loadType,
28904eeddc0SDimitry Andric                           bool isLazy = false, bool isExplicit = true,
290972a253aSDimitry Andric                           bool isBundleLoader = false,
291972a253aSDimitry Andric                           bool isForceHidden = false) {
292bdd1243dSDimitry Andric   std::optional<MemoryBufferRef> buffer = readFile(path);
2935ffd83dbSDimitry Andric   if (!buffer)
294e8d8bef9SDimitry Andric     return nullptr;
2955ffd83dbSDimitry Andric   MemoryBufferRef mbref = *buffer;
296e8d8bef9SDimitry Andric   InputFile *newFile = nullptr;
2975ffd83dbSDimitry Andric 
298fe6060f1SDimitry Andric   file_magic magic = identify_magic(mbref.getBuffer());
299e8d8bef9SDimitry Andric   switch (magic) {
3005ffd83dbSDimitry Andric   case file_magic::archive: {
301fcaf7f86SDimitry Andric     bool isCommandLineLoad = loadType != LoadType::LCLinkerOption;
302fe6060f1SDimitry Andric     // Avoid loading archives twice. If the archives are being force-loaded,
303fe6060f1SDimitry Andric     // loading them twice would create duplicate symbol errors. In the
304fe6060f1SDimitry Andric     // non-force-loading case, this is just a minor performance optimization.
305fe6060f1SDimitry Andric     // We don't take a reference to cachedFile here because the
306fe6060f1SDimitry Andric     // loadArchiveMember() call below may recursively call addFile() and
307fe6060f1SDimitry Andric     // invalidate this reference.
30881ad6265SDimitry Andric     auto entry = loadedArchives.find(path);
309fe6060f1SDimitry Andric 
310fcaf7f86SDimitry Andric     ArchiveFile *file;
311fcaf7f86SDimitry Andric     if (entry == loadedArchives.end()) {
312fcaf7f86SDimitry Andric       // No cached archive, we need to create a new one
313349cc55cSDimitry Andric       std::unique_ptr<object::Archive> archive = CHECK(
3145ffd83dbSDimitry Andric           object::Archive::create(mbref), path + ": failed to parse archive");
3155ffd83dbSDimitry Andric 
316349cc55cSDimitry Andric       if (!archive->isEmpty() && !archive->hasSymbolTable())
3175ffd83dbSDimitry Andric         error(path + ": archive has no index; run ranlib to add one");
318972a253aSDimitry Andric       file = make<ArchiveFile>(std::move(archive), isForceHidden);
319*0fca6ea1SDimitry Andric 
320*0fca6ea1SDimitry Andric       if (tar && file->getArchive().isThin())
321*0fca6ea1SDimitry Andric         saveThinArchiveToRepro(file);
322fcaf7f86SDimitry Andric     } else {
323fcaf7f86SDimitry Andric       file = entry->second.file;
324fcaf7f86SDimitry Andric       // Command-line loads take precedence. If file is previously loaded via
325fcaf7f86SDimitry Andric       // command line, or is loaded via LC_LINKER_OPTION and being loaded via
326fcaf7f86SDimitry Andric       // LC_LINKER_OPTION again, using the cached archive is enough.
327fcaf7f86SDimitry Andric       if (entry->second.isCommandLineLoad || !isCommandLineLoad)
328fcaf7f86SDimitry Andric         return file;
329fcaf7f86SDimitry Andric     }
3305ffd83dbSDimitry Andric 
331fcaf7f86SDimitry Andric     bool isLCLinkerForceLoad = loadType == LoadType::LCLinkerOption &&
332fcaf7f86SDimitry Andric                                config->forceLoadSwift &&
33306c3fb27SDimitry Andric                                path::filename(path).starts_with("libswift");
334fcaf7f86SDimitry Andric     if ((isCommandLineLoad && config->allLoad) ||
335fcaf7f86SDimitry Andric         loadType == LoadType::CommandLineForce || isLCLinkerForceLoad) {
3365f757f3fSDimitry Andric       if (readFile(path)) {
337349cc55cSDimitry Andric         Error e = Error::success();
338349cc55cSDimitry Andric         for (const object::Archive::Child &c : file->getArchive().children(e)) {
339fcaf7f86SDimitry Andric           StringRef reason;
340fcaf7f86SDimitry Andric           switch (loadType) {
341fcaf7f86SDimitry Andric             case LoadType::LCLinkerOption:
342fcaf7f86SDimitry Andric               reason = "LC_LINKER_OPTION";
343fcaf7f86SDimitry Andric               break;
344fcaf7f86SDimitry Andric             case LoadType::CommandLineForce:
345fcaf7f86SDimitry Andric               reason = "-force_load";
346fcaf7f86SDimitry Andric               break;
347fcaf7f86SDimitry Andric             case LoadType::CommandLine:
348fcaf7f86SDimitry Andric               reason = "-all_load";
349fcaf7f86SDimitry Andric               break;
350fcaf7f86SDimitry Andric           }
351*0fca6ea1SDimitry Andric           if (Error e = file->fetch(c, reason)) {
352*0fca6ea1SDimitry Andric             if (config->warnThinArchiveMissingMembers)
353*0fca6ea1SDimitry Andric               warn(toString(file) + ": " + reason +
354349cc55cSDimitry Andric                    " failed to load archive member: " + toString(std::move(e)));
355*0fca6ea1SDimitry Andric             else
356*0fca6ea1SDimitry Andric               llvm::consumeError(std::move(e));
357*0fca6ea1SDimitry Andric           }
358e8d8bef9SDimitry Andric         }
359349cc55cSDimitry Andric         if (e)
360349cc55cSDimitry Andric           error(toString(file) +
361349cc55cSDimitry Andric                 ": Archive::children failed: " + toString(std::move(e)));
362e8d8bef9SDimitry Andric       }
363fcaf7f86SDimitry Andric     } else if (isCommandLineLoad && config->forceLoadObjC) {
364349cc55cSDimitry Andric       for (const object::Archive::Symbol &sym : file->getArchive().symbols())
365*0fca6ea1SDimitry Andric         if (sym.getName().starts_with(objc::symbol_names::klass))
366349cc55cSDimitry Andric           file->fetch(sym);
367e8d8bef9SDimitry Andric 
368e8d8bef9SDimitry Andric       // TODO: no need to look for ObjC sections for a given archive member if
369349cc55cSDimitry Andric       // we already found that it contains an ObjC symbol.
3705f757f3fSDimitry Andric       if (readFile(path)) {
371349cc55cSDimitry Andric         Error e = Error::success();
372349cc55cSDimitry Andric         for (const object::Archive::Child &c : file->getArchive().children(e)) {
373349cc55cSDimitry Andric           Expected<MemoryBufferRef> mb = c.getMemoryBufferRef();
374*0fca6ea1SDimitry Andric           if (!mb) {
375*0fca6ea1SDimitry Andric             // We used to create broken repro tarballs that only included those
376*0fca6ea1SDimitry Andric             // object files from thin archives that ended up being used.
377*0fca6ea1SDimitry Andric             if (config->warnThinArchiveMissingMembers)
378*0fca6ea1SDimitry Andric               warn(toString(file) + ": -ObjC failed to open archive member: " +
379*0fca6ea1SDimitry Andric                    toString(mb.takeError()));
380*0fca6ea1SDimitry Andric             else
381*0fca6ea1SDimitry Andric               llvm::consumeError(mb.takeError());
382*0fca6ea1SDimitry Andric             continue;
383*0fca6ea1SDimitry Andric           }
384*0fca6ea1SDimitry Andric 
385*0fca6ea1SDimitry Andric           if (!hasObjCSection(*mb))
386349cc55cSDimitry Andric             continue;
387349cc55cSDimitry Andric           if (Error e = file->fetch(c, "-ObjC"))
388349cc55cSDimitry Andric             error(toString(file) + ": -ObjC failed to load archive member: " +
389349cc55cSDimitry Andric                   toString(std::move(e)));
390e8d8bef9SDimitry Andric         }
391349cc55cSDimitry Andric         if (e)
392349cc55cSDimitry Andric           error(toString(file) +
393349cc55cSDimitry Andric                 ": Archive::children failed: " + toString(std::move(e)));
394e8d8bef9SDimitry Andric       }
395e8d8bef9SDimitry Andric     }
396e8d8bef9SDimitry Andric 
397349cc55cSDimitry Andric     file->addLazySymbols();
398fcaf7f86SDimitry Andric     loadedArchives[path] = ArchiveFileInfo{file, isCommandLineLoad};
399fcaf7f86SDimitry Andric     newFile = file;
4005ffd83dbSDimitry Andric     break;
4015ffd83dbSDimitry Andric   }
4025ffd83dbSDimitry Andric   case file_magic::macho_object:
40304eeddc0SDimitry Andric     newFile = make<ObjFile>(mbref, getModTime(path), "", isLazy);
4045ffd83dbSDimitry Andric     break;
4055ffd83dbSDimitry Andric   case file_magic::macho_dynamically_linked_shared_lib:
406e8d8bef9SDimitry Andric   case file_magic::macho_dynamically_linked_shared_lib_stub:
407fe6060f1SDimitry Andric   case file_magic::tapi_file:
40881ad6265SDimitry Andric     if (DylibFile *dylibFile =
40981ad6265SDimitry Andric             loadDylib(mbref, nullptr, /*isBundleLoader=*/false, isExplicit))
410fe6060f1SDimitry Andric       newFile = dylibFile;
411fe6060f1SDimitry Andric     break;
412e8d8bef9SDimitry Andric   case file_magic::bitcode:
41304eeddc0SDimitry Andric     newFile = make<BitcodeFile>(mbref, "", 0, isLazy);
414fe6060f1SDimitry Andric     break;
415fe6060f1SDimitry Andric   case file_magic::macho_executable:
416fe6060f1SDimitry Andric   case file_magic::macho_bundle:
417fe6060f1SDimitry Andric     // We only allow executable and bundle type here if it is used
418fe6060f1SDimitry Andric     // as a bundle loader.
419fe6060f1SDimitry Andric     if (!isBundleLoader)
420fe6060f1SDimitry Andric       error(path + ": unhandled file type");
421fe6060f1SDimitry Andric     if (DylibFile *dylibFile = loadDylib(mbref, nullptr, isBundleLoader))
422fe6060f1SDimitry Andric       newFile = dylibFile;
423e8d8bef9SDimitry Andric     break;
4245ffd83dbSDimitry Andric   default:
4255ffd83dbSDimitry Andric     error(path + ": unhandled file type");
4265ffd83dbSDimitry Andric   }
427fe6060f1SDimitry Andric   if (newFile && !isa<DylibFile>(newFile)) {
42804eeddc0SDimitry Andric     if ((isa<ObjFile>(newFile) || isa<BitcodeFile>(newFile)) && newFile->lazy &&
42904eeddc0SDimitry Andric         config->forceLoadObjC) {
43004eeddc0SDimitry Andric       for (Symbol *sym : newFile->symbols)
431*0fca6ea1SDimitry Andric         if (sym && sym->getName().starts_with(objc::symbol_names::klass)) {
43204eeddc0SDimitry Andric           extract(*newFile, "-ObjC");
43304eeddc0SDimitry Andric           break;
43404eeddc0SDimitry Andric         }
43504eeddc0SDimitry Andric       if (newFile->lazy && hasObjCSection(mbref))
43604eeddc0SDimitry Andric         extract(*newFile, "-ObjC");
43704eeddc0SDimitry Andric     }
43804eeddc0SDimitry Andric 
439e8d8bef9SDimitry Andric     // printArchiveMemberLoad() prints both .a and .o names, so no need to
44004eeddc0SDimitry Andric     // print the .a name here. Similarly skip lazy files.
44104eeddc0SDimitry Andric     if (config->printEachFile && magic != file_magic::archive && !isLazy)
442e8d8bef9SDimitry Andric       message(toString(newFile));
443e8d8bef9SDimitry Andric     inputFiles.insert(newFile);
444e8d8bef9SDimitry Andric   }
445e8d8bef9SDimitry Andric   return newFile;
4465ffd83dbSDimitry Andric }
4475ffd83dbSDimitry Andric 
448bdd1243dSDimitry Andric static std::vector<StringRef> missingAutolinkWarnings;
addLibrary(StringRef name,bool isNeeded,bool isWeak,bool isReexport,bool isHidden,bool isExplicit,LoadType loadType)449fe6060f1SDimitry Andric static void addLibrary(StringRef name, bool isNeeded, bool isWeak,
450972a253aSDimitry Andric                        bool isReexport, bool isHidden, bool isExplicit,
4515f757f3fSDimitry Andric                        LoadType loadType) {
452bdd1243dSDimitry Andric   if (std::optional<StringRef> path = findLibrary(name)) {
453fe6060f1SDimitry Andric     if (auto *dylibFile = dyn_cast_or_null<DylibFile>(
454972a253aSDimitry Andric             addFile(*path, loadType, /*isLazy=*/false, isExplicit,
455972a253aSDimitry Andric                     /*isBundleLoader=*/false, isHidden))) {
456fe6060f1SDimitry Andric       if (isNeeded)
457fe6060f1SDimitry Andric         dylibFile->forceNeeded = true;
458fe6060f1SDimitry Andric       if (isWeak)
459e8d8bef9SDimitry Andric         dylibFile->forceWeakImport = true;
460fe6060f1SDimitry Andric       if (isReexport) {
461fe6060f1SDimitry Andric         config->hasReexports = true;
462fe6060f1SDimitry Andric         dylibFile->reexport = true;
463fe6060f1SDimitry Andric       }
464fe6060f1SDimitry Andric     }
465e8d8bef9SDimitry Andric     return;
466e8d8bef9SDimitry Andric   }
467bdd1243dSDimitry Andric   if (loadType == LoadType::LCLinkerOption) {
468bdd1243dSDimitry Andric     missingAutolinkWarnings.push_back(
4695f757f3fSDimitry Andric         saver().save("auto-linked library not found for -l" + name));
470bdd1243dSDimitry Andric     return;
471bdd1243dSDimitry Andric   }
472e8d8bef9SDimitry Andric   error("library not found for -l" + name);
473e8d8bef9SDimitry Andric }
474e8d8bef9SDimitry Andric 
47581ad6265SDimitry Andric static DenseSet<StringRef> loadedObjectFrameworks;
addFramework(StringRef name,bool isNeeded,bool isWeak,bool isReexport,bool isExplicit,LoadType loadType)476fe6060f1SDimitry Andric static void addFramework(StringRef name, bool isNeeded, bool isWeak,
4775f757f3fSDimitry Andric                          bool isReexport, bool isExplicit, LoadType loadType) {
478bdd1243dSDimitry Andric   if (std::optional<StringRef> path = findFramework(name)) {
47981ad6265SDimitry Andric     if (loadedObjectFrameworks.contains(*path))
48081ad6265SDimitry Andric       return;
48181ad6265SDimitry Andric 
48281ad6265SDimitry Andric     InputFile *file =
483fcaf7f86SDimitry Andric         addFile(*path, loadType, /*isLazy=*/false, isExplicit, false);
48481ad6265SDimitry Andric     if (auto *dylibFile = dyn_cast_or_null<DylibFile>(file)) {
485fe6060f1SDimitry Andric       if (isNeeded)
486fe6060f1SDimitry Andric         dylibFile->forceNeeded = true;
487fe6060f1SDimitry Andric       if (isWeak)
488e8d8bef9SDimitry Andric         dylibFile->forceWeakImport = true;
489fe6060f1SDimitry Andric       if (isReexport) {
490fe6060f1SDimitry Andric         config->hasReexports = true;
491fe6060f1SDimitry Andric         dylibFile->reexport = true;
492fe6060f1SDimitry Andric       }
49381ad6265SDimitry Andric     } else if (isa_and_nonnull<ObjFile>(file) ||
49481ad6265SDimitry Andric                isa_and_nonnull<BitcodeFile>(file)) {
49581ad6265SDimitry Andric       // Cache frameworks containing object or bitcode files to avoid duplicate
49681ad6265SDimitry Andric       // symbols. Frameworks containing static archives are cached separately
49781ad6265SDimitry Andric       // in addFile() to share caching with libraries, and frameworks
49881ad6265SDimitry Andric       // containing dylibs should allow overwriting of attributes such as
49981ad6265SDimitry Andric       // forceNeeded by subsequent loads
50081ad6265SDimitry Andric       loadedObjectFrameworks.insert(*path);
501fe6060f1SDimitry Andric     }
502e8d8bef9SDimitry Andric     return;
503e8d8bef9SDimitry Andric   }
504bdd1243dSDimitry Andric   if (loadType == LoadType::LCLinkerOption) {
5055f757f3fSDimitry Andric     missingAutolinkWarnings.push_back(
5065f757f3fSDimitry Andric         saver().save("auto-linked framework not found for -framework " + name));
507bdd1243dSDimitry Andric     return;
508bdd1243dSDimitry Andric   }
509e8d8bef9SDimitry Andric   error("framework not found for -framework " + name);
510e8d8bef9SDimitry Andric }
511e8d8bef9SDimitry Andric 
512fe6060f1SDimitry Andric // Parses LC_LINKER_OPTION contents, which can add additional command line
513349cc55cSDimitry Andric // flags. This directly parses the flags instead of using the standard argument
514349cc55cSDimitry Andric // parser to improve performance.
parseLCLinkerOption(llvm::SmallVectorImpl<StringRef> & LCLinkerOptions,InputFile * f,unsigned argc,StringRef data)5155f757f3fSDimitry Andric void macho::parseLCLinkerOption(
5165f757f3fSDimitry Andric     llvm::SmallVectorImpl<StringRef> &LCLinkerOptions, InputFile *f,
5175f757f3fSDimitry Andric     unsigned argc, StringRef data) {
518bdd1243dSDimitry Andric   if (config->ignoreAutoLink)
519bdd1243dSDimitry Andric     return;
520bdd1243dSDimitry Andric 
521349cc55cSDimitry Andric   SmallVector<StringRef, 4> argv;
522e8d8bef9SDimitry Andric   size_t offset = 0;
523e8d8bef9SDimitry Andric   for (unsigned i = 0; i < argc && offset < data.size(); ++i) {
524e8d8bef9SDimitry Andric     argv.push_back(data.data() + offset);
525e8d8bef9SDimitry Andric     offset += strlen(data.data() + offset) + 1;
526e8d8bef9SDimitry Andric   }
527e8d8bef9SDimitry Andric   if (argv.size() != argc || offset > data.size())
528e8d8bef9SDimitry Andric     fatal(toString(f) + ": invalid LC_LINKER_OPTION");
529e8d8bef9SDimitry Andric 
530349cc55cSDimitry Andric   unsigned i = 0;
531349cc55cSDimitry Andric   StringRef arg = argv[i];
532349cc55cSDimitry Andric   if (arg.consume_front("-l")) {
533bdd1243dSDimitry Andric     if (config->ignoreAutoLinkOptions.contains(arg))
534bdd1243dSDimitry Andric       return;
535349cc55cSDimitry Andric   } else if (arg == "-framework") {
536349cc55cSDimitry Andric     StringRef name = argv[++i];
537bdd1243dSDimitry Andric     if (config->ignoreAutoLinkOptions.contains(name))
538bdd1243dSDimitry Andric       return;
539349cc55cSDimitry Andric   } else {
540349cc55cSDimitry Andric     error(arg + " is not allowed in LC_LINKER_OPTION");
541e8d8bef9SDimitry Andric   }
5425f757f3fSDimitry Andric 
5435f757f3fSDimitry Andric   LCLinkerOptions.append(argv);
5445f757f3fSDimitry Andric }
5455f757f3fSDimitry Andric 
resolveLCLinkerOptions()5465f757f3fSDimitry Andric void macho::resolveLCLinkerOptions() {
5475f757f3fSDimitry Andric   while (!unprocessedLCLinkerOptions.empty()) {
5485f757f3fSDimitry Andric     SmallVector<StringRef> LCLinkerOptions(unprocessedLCLinkerOptions);
5495f757f3fSDimitry Andric     unprocessedLCLinkerOptions.clear();
5505f757f3fSDimitry Andric 
5515f757f3fSDimitry Andric     for (unsigned i = 0; i < LCLinkerOptions.size(); ++i) {
5525f757f3fSDimitry Andric       StringRef arg = LCLinkerOptions[i];
5535f757f3fSDimitry Andric       if (arg.consume_front("-l")) {
5545f757f3fSDimitry Andric         assert(!config->ignoreAutoLinkOptions.contains(arg));
5555f757f3fSDimitry Andric         addLibrary(arg, /*isNeeded=*/false, /*isWeak=*/false,
5565f757f3fSDimitry Andric                    /*isReexport=*/false, /*isHidden=*/false,
5575f757f3fSDimitry Andric                    /*isExplicit=*/false, LoadType::LCLinkerOption);
5585f757f3fSDimitry Andric       } else if (arg == "-framework") {
5595f757f3fSDimitry Andric         StringRef name = LCLinkerOptions[++i];
5605f757f3fSDimitry Andric         assert(!config->ignoreAutoLinkOptions.contains(name));
5615f757f3fSDimitry Andric         addFramework(name, /*isNeeded=*/false, /*isWeak=*/false,
5625f757f3fSDimitry Andric                      /*isReexport=*/false, /*isExplicit=*/false,
5635f757f3fSDimitry Andric                      LoadType::LCLinkerOption);
5645f757f3fSDimitry Andric       } else {
5655f757f3fSDimitry Andric         error(arg + " is not allowed in LC_LINKER_OPTION");
5665f757f3fSDimitry Andric       }
5675f757f3fSDimitry Andric     }
5685f757f3fSDimitry Andric   }
569e8d8bef9SDimitry Andric }
570e8d8bef9SDimitry Andric 
addFileList(StringRef path,bool isLazy)57104eeddc0SDimitry Andric static void addFileList(StringRef path, bool isLazy) {
572bdd1243dSDimitry Andric   std::optional<MemoryBufferRef> buffer = readFile(path);
573e8d8bef9SDimitry Andric   if (!buffer)
574e8d8bef9SDimitry Andric     return;
575e8d8bef9SDimitry Andric   MemoryBufferRef mbref = *buffer;
576e8d8bef9SDimitry Andric   for (StringRef path : args::getLines(mbref))
577fcaf7f86SDimitry Andric     addFile(rerootPath(path), LoadType::CommandLine, isLazy);
5785ffd83dbSDimitry Andric }
5795ffd83dbSDimitry Andric 
5805ffd83dbSDimitry Andric // We expect sub-library names of the form "libfoo", which will match a dylib
581e8d8bef9SDimitry Andric // with a path of .*/libfoo.{dylib, tbd}.
582e8d8bef9SDimitry Andric // XXX ld64 seems to ignore the extension entirely when matching sub-libraries;
583e8d8bef9SDimitry Andric // I'm not sure what the use case for that is.
markReexport(StringRef searchName,ArrayRef<StringRef> extensions)584e8d8bef9SDimitry Andric static bool markReexport(StringRef searchName, ArrayRef<StringRef> extensions) {
5855ffd83dbSDimitry Andric   for (InputFile *file : inputFiles) {
5865ffd83dbSDimitry Andric     if (auto *dylibFile = dyn_cast<DylibFile>(file)) {
5875ffd83dbSDimitry Andric       StringRef filename = path::filename(dylibFile->getName());
588e8d8bef9SDimitry Andric       if (filename.consume_front(searchName) &&
589fcaf7f86SDimitry Andric           (filename.empty() || llvm::is_contained(extensions, filename))) {
5905ffd83dbSDimitry Andric         dylibFile->reexport = true;
5915ffd83dbSDimitry Andric         return true;
5925ffd83dbSDimitry Andric       }
5935ffd83dbSDimitry Andric     }
5945ffd83dbSDimitry Andric   }
5955ffd83dbSDimitry Andric   return false;
5965ffd83dbSDimitry Andric }
5975ffd83dbSDimitry Andric 
598e8d8bef9SDimitry Andric // This function is called on startup. We need this for LTO since
599e8d8bef9SDimitry Andric // LTO calls LLVM functions to compile bitcode files to native code.
600e8d8bef9SDimitry Andric // Technically this can be delayed until we read bitcode files, but
601e8d8bef9SDimitry Andric // we don't bother to do lazily because the initialization is fast.
initLLVM()602e8d8bef9SDimitry Andric static void initLLVM() {
603e8d8bef9SDimitry Andric   InitializeAllTargets();
604e8d8bef9SDimitry Andric   InitializeAllTargetMCs();
605e8d8bef9SDimitry Andric   InitializeAllAsmPrinters();
606e8d8bef9SDimitry Andric   InitializeAllAsmParsers();
607e8d8bef9SDimitry Andric }
608e8d8bef9SDimitry Andric 
compileBitcodeFiles()609bdd1243dSDimitry Andric static bool compileBitcodeFiles() {
610fe6060f1SDimitry Andric   TimeTraceScope timeScope("LTO");
611fe6060f1SDimitry Andric   auto *lto = make<BitcodeCompiler>();
612e8d8bef9SDimitry Andric   for (InputFile *file : inputFiles)
613e8d8bef9SDimitry Andric     if (auto *bitcodeFile = dyn_cast<BitcodeFile>(file))
61404eeddc0SDimitry Andric       if (!file->lazy)
615e8d8bef9SDimitry Andric         lto->add(*bitcodeFile);
616e8d8bef9SDimitry Andric 
617bdd1243dSDimitry Andric   std::vector<ObjFile *> compiled = lto->compile();
618bdd1243dSDimitry Andric   for (ObjFile *file : compiled)
619e8d8bef9SDimitry Andric     inputFiles.insert(file);
620bdd1243dSDimitry Andric 
621bdd1243dSDimitry Andric   return !compiled.empty();
622e8d8bef9SDimitry Andric }
623e8d8bef9SDimitry Andric 
624e8d8bef9SDimitry Andric // Replaces common symbols with defined symbols residing in __common sections.
625e8d8bef9SDimitry Andric // This function must be called after all symbol names are resolved (i.e. after
626e8d8bef9SDimitry Andric // all InputFiles have been loaded.) As a result, later operations won't see
627e8d8bef9SDimitry Andric // any CommonSymbols.
replaceCommonSymbols()628e8d8bef9SDimitry Andric static void replaceCommonSymbols() {
629fe6060f1SDimitry Andric   TimeTraceScope timeScope("Replace common symbols");
630fe6060f1SDimitry Andric   ConcatOutputSection *osec = nullptr;
631fe6060f1SDimitry Andric   for (Symbol *sym : symtab->getSymbols()) {
632e8d8bef9SDimitry Andric     auto *common = dyn_cast<CommonSymbol>(sym);
633e8d8bef9SDimitry Andric     if (common == nullptr)
634e8d8bef9SDimitry Andric       continue;
635e8d8bef9SDimitry Andric 
636e8d8bef9SDimitry Andric     // Casting to size_t will truncate large values on 32-bit architectures,
637e8d8bef9SDimitry Andric     // but it's not really worth supporting the linking of 64-bit programs on
638e8d8bef9SDimitry Andric     // 32-bit archs.
639fe6060f1SDimitry Andric     ArrayRef<uint8_t> data = {nullptr, static_cast<size_t>(common->size)};
64081ad6265SDimitry Andric     // FIXME avoid creating one Section per symbol?
64181ad6265SDimitry Andric     auto *section =
64281ad6265SDimitry Andric         make<Section>(common->getFile(), segment_names::data,
64381ad6265SDimitry Andric                       section_names::common, S_ZEROFILL, /*addr=*/0);
64481ad6265SDimitry Andric     auto *isec = make<ConcatInputSection>(*section, data, common->align);
645fe6060f1SDimitry Andric     if (!osec)
646fe6060f1SDimitry Andric       osec = ConcatOutputSection::getOrCreateForInput(isec);
647fe6060f1SDimitry Andric     isec->parent = osec;
648*0fca6ea1SDimitry Andric     addInputSection(isec);
649e8d8bef9SDimitry Andric 
650fe6060f1SDimitry Andric     // FIXME: CommonSymbol should store isReferencedDynamically, noDeadStrip
651fe6060f1SDimitry Andric     // and pass them on here.
65281ad6265SDimitry Andric     replaceSymbol<Defined>(
653bdd1243dSDimitry Andric         sym, sym->getName(), common->getFile(), isec, /*value=*/0, common->size,
65481ad6265SDimitry Andric         /*isWeakDef=*/false, /*isExternal=*/true, common->privateExtern,
65506c3fb27SDimitry Andric         /*includeInSymtab=*/true, /*isReferencedDynamically=*/false,
65606c3fb27SDimitry Andric         /*noDeadStrip=*/false);
657e8d8bef9SDimitry Andric   }
658e8d8bef9SDimitry Andric }
659e8d8bef9SDimitry Andric 
initializeSectionRenameMap()660fe6060f1SDimitry Andric static void initializeSectionRenameMap() {
661fe6060f1SDimitry Andric   if (config->dataConst) {
662fe6060f1SDimitry Andric     SmallVector<StringRef> v{section_names::got,
663fe6060f1SDimitry Andric                              section_names::authGot,
664fe6060f1SDimitry Andric                              section_names::authPtr,
665fe6060f1SDimitry Andric                              section_names::nonLazySymbolPtr,
666fe6060f1SDimitry Andric                              section_names::const_,
667fe6060f1SDimitry Andric                              section_names::cfString,
668fe6060f1SDimitry Andric                              section_names::moduleInitFunc,
669fe6060f1SDimitry Andric                              section_names::moduleTermFunc,
670fe6060f1SDimitry Andric                              section_names::objcClassList,
671fe6060f1SDimitry Andric                              section_names::objcNonLazyClassList,
672fe6060f1SDimitry Andric                              section_names::objcCatList,
673fe6060f1SDimitry Andric                              section_names::objcNonLazyCatList,
674fe6060f1SDimitry Andric                              section_names::objcProtoList,
675fcaf7f86SDimitry Andric                              section_names::objCImageInfo};
676fe6060f1SDimitry Andric     for (StringRef s : v)
677fe6060f1SDimitry Andric       config->sectionRenameMap[{segment_names::data, s}] = {
678fe6060f1SDimitry Andric           segment_names::dataConst, s};
679fe6060f1SDimitry Andric   }
680fe6060f1SDimitry Andric   config->sectionRenameMap[{segment_names::text, section_names::staticInit}] = {
681fe6060f1SDimitry Andric       segment_names::text, section_names::text};
682fe6060f1SDimitry Andric   config->sectionRenameMap[{segment_names::import, section_names::pointers}] = {
683fe6060f1SDimitry Andric       config->dataConst ? segment_names::dataConst : segment_names::data,
684fe6060f1SDimitry Andric       section_names::nonLazySymbolPtr};
685fe6060f1SDimitry Andric }
686fe6060f1SDimitry Andric 
toLowerDash(char x)687e8d8bef9SDimitry Andric static inline char toLowerDash(char x) {
688e8d8bef9SDimitry Andric   if (x >= 'A' && x <= 'Z')
689e8d8bef9SDimitry Andric     return x - 'A' + 'a';
690e8d8bef9SDimitry Andric   else if (x == ' ')
691e8d8bef9SDimitry Andric     return '-';
692e8d8bef9SDimitry Andric   return x;
693e8d8bef9SDimitry Andric }
694e8d8bef9SDimitry Andric 
lowerDash(StringRef s)695e8d8bef9SDimitry Andric static std::string lowerDash(StringRef s) {
696e8d8bef9SDimitry Andric   return std::string(map_iterator(s.begin(), toLowerDash),
697e8d8bef9SDimitry Andric                      map_iterator(s.end(), toLowerDash));
698e8d8bef9SDimitry Andric }
699e8d8bef9SDimitry Andric 
70081ad6265SDimitry Andric struct PlatformVersion {
70181ad6265SDimitry Andric   PlatformType platform = PLATFORM_UNKNOWN;
70281ad6265SDimitry Andric   llvm::VersionTuple minimum;
70381ad6265SDimitry Andric   llvm::VersionTuple sdk;
70481ad6265SDimitry Andric };
705fe6060f1SDimitry Andric 
parsePlatformVersion(const Arg * arg)70681ad6265SDimitry Andric static PlatformVersion parsePlatformVersion(const Arg *arg) {
70781ad6265SDimitry Andric   assert(arg->getOption().getID() == OPT_platform_version);
708e8d8bef9SDimitry Andric   StringRef platformStr = arg->getValue(0);
709e8d8bef9SDimitry Andric   StringRef minVersionStr = arg->getValue(1);
710e8d8bef9SDimitry Andric   StringRef sdkVersionStr = arg->getValue(2);
711e8d8bef9SDimitry Andric 
71281ad6265SDimitry Andric   PlatformVersion platformVersion;
71381ad6265SDimitry Andric 
714e8d8bef9SDimitry Andric   // TODO(compnerd) see if we can generate this case list via XMACROS
71581ad6265SDimitry Andric   platformVersion.platform =
71604eeddc0SDimitry Andric       StringSwitch<PlatformType>(lowerDash(platformStr))
71704eeddc0SDimitry Andric           .Cases("macos", "1", PLATFORM_MACOS)
71804eeddc0SDimitry Andric           .Cases("ios", "2", PLATFORM_IOS)
71904eeddc0SDimitry Andric           .Cases("tvos", "3", PLATFORM_TVOS)
72004eeddc0SDimitry Andric           .Cases("watchos", "4", PLATFORM_WATCHOS)
72104eeddc0SDimitry Andric           .Cases("bridgeos", "5", PLATFORM_BRIDGEOS)
72204eeddc0SDimitry Andric           .Cases("mac-catalyst", "6", PLATFORM_MACCATALYST)
72304eeddc0SDimitry Andric           .Cases("ios-simulator", "7", PLATFORM_IOSSIMULATOR)
72404eeddc0SDimitry Andric           .Cases("tvos-simulator", "8", PLATFORM_TVOSSIMULATOR)
72504eeddc0SDimitry Andric           .Cases("watchos-simulator", "9", PLATFORM_WATCHOSSIMULATOR)
72604eeddc0SDimitry Andric           .Cases("driverkit", "10", PLATFORM_DRIVERKIT)
727*0fca6ea1SDimitry Andric           .Cases("xros", "11", PLATFORM_XROS)
728*0fca6ea1SDimitry Andric           .Cases("xros-simulator", "12", PLATFORM_XROS_SIMULATOR)
72904eeddc0SDimitry Andric           .Default(PLATFORM_UNKNOWN);
73081ad6265SDimitry Andric   if (platformVersion.platform == PLATFORM_UNKNOWN)
731e8d8bef9SDimitry Andric     error(Twine("malformed platform: ") + platformStr);
732e8d8bef9SDimitry Andric   // TODO: check validity of version strings, which varies by platform
733e8d8bef9SDimitry Andric   // NOTE: ld64 accepts version strings with 5 components
734e8d8bef9SDimitry Andric   // llvm::VersionTuple accepts no more than 4 components
735e8d8bef9SDimitry Andric   // Has Apple ever published version strings with 5 components?
73681ad6265SDimitry Andric   if (platformVersion.minimum.tryParse(minVersionStr))
737e8d8bef9SDimitry Andric     error(Twine("malformed minimum version: ") + minVersionStr);
73881ad6265SDimitry Andric   if (platformVersion.sdk.tryParse(sdkVersionStr))
739e8d8bef9SDimitry Andric     error(Twine("malformed sdk version: ") + sdkVersionStr);
74081ad6265SDimitry Andric   return platformVersion;
74181ad6265SDimitry Andric }
74281ad6265SDimitry Andric 
74306c3fb27SDimitry Andric // Has the side-effect of setting Config::platformInfo and
74406c3fb27SDimitry Andric // potentially Config::secondaryPlatformInfo.
setPlatformVersions(StringRef archName,const ArgList & args)74506c3fb27SDimitry Andric static void setPlatformVersions(StringRef archName, const ArgList &args) {
74681ad6265SDimitry Andric   std::map<PlatformType, PlatformVersion> platformVersions;
74781ad6265SDimitry Andric   const PlatformVersion *lastVersionInfo = nullptr;
74881ad6265SDimitry Andric   for (const Arg *arg : args.filtered(OPT_platform_version)) {
74981ad6265SDimitry Andric     PlatformVersion version = parsePlatformVersion(arg);
75081ad6265SDimitry Andric 
75181ad6265SDimitry Andric     // For each platform, the last flag wins:
75281ad6265SDimitry Andric     // `-platform_version macos 2 3 -platform_version macos 4 5` has the same
75381ad6265SDimitry Andric     // effect as just passing `-platform_version macos 4 5`.
75481ad6265SDimitry Andric     // FIXME: ld64 warns on multiple flags for one platform. Should we?
75581ad6265SDimitry Andric     platformVersions[version.platform] = version;
75681ad6265SDimitry Andric     lastVersionInfo = &platformVersions[version.platform];
75781ad6265SDimitry Andric   }
75881ad6265SDimitry Andric 
75981ad6265SDimitry Andric   if (platformVersions.empty()) {
76081ad6265SDimitry Andric     error("must specify -platform_version");
76106c3fb27SDimitry Andric     return;
76281ad6265SDimitry Andric   }
76381ad6265SDimitry Andric   if (platformVersions.size() > 2) {
76481ad6265SDimitry Andric     error("must specify -platform_version at most twice");
76506c3fb27SDimitry Andric     return;
76681ad6265SDimitry Andric   }
76781ad6265SDimitry Andric   if (platformVersions.size() == 2) {
76881ad6265SDimitry Andric     bool isZipperedCatalyst = platformVersions.count(PLATFORM_MACOS) &&
76981ad6265SDimitry Andric                               platformVersions.count(PLATFORM_MACCATALYST);
77081ad6265SDimitry Andric 
77181ad6265SDimitry Andric     if (!isZipperedCatalyst) {
77281ad6265SDimitry Andric       error("lld supports writing zippered outputs only for "
77381ad6265SDimitry Andric             "macos and mac-catalyst");
77481ad6265SDimitry Andric     } else if (config->outputType != MH_DYLIB &&
77581ad6265SDimitry Andric                config->outputType != MH_BUNDLE) {
77681ad6265SDimitry Andric       error("writing zippered outputs only valid for -dylib and -bundle");
77781ad6265SDimitry Andric     }
77881ad6265SDimitry Andric 
77906c3fb27SDimitry Andric     config->platformInfo = {
78006c3fb27SDimitry Andric         MachO::Target(getArchitectureFromName(archName), PLATFORM_MACOS,
78106c3fb27SDimitry Andric                       platformVersions[PLATFORM_MACOS].minimum),
78206c3fb27SDimitry Andric         platformVersions[PLATFORM_MACOS].sdk};
78306c3fb27SDimitry Andric     config->secondaryPlatformInfo = {
78406c3fb27SDimitry Andric         MachO::Target(getArchitectureFromName(archName), PLATFORM_MACCATALYST,
78506c3fb27SDimitry Andric                       platformVersions[PLATFORM_MACCATALYST].minimum),
78606c3fb27SDimitry Andric         platformVersions[PLATFORM_MACCATALYST].sdk};
78706c3fb27SDimitry Andric     return;
78806c3fb27SDimitry Andric   }
78906c3fb27SDimitry Andric 
79006c3fb27SDimitry Andric   config->platformInfo = {MachO::Target(getArchitectureFromName(archName),
79106c3fb27SDimitry Andric                                         lastVersionInfo->platform,
79206c3fb27SDimitry Andric                                         lastVersionInfo->minimum),
79306c3fb27SDimitry Andric                           lastVersionInfo->sdk};
794e8d8bef9SDimitry Andric }
795e8d8bef9SDimitry Andric 
796fe6060f1SDimitry Andric // Has the side-effect of setting Config::target.
createTargetInfo(InputArgList & args)797fe6060f1SDimitry Andric static TargetInfo *createTargetInfo(InputArgList &args) {
798fe6060f1SDimitry Andric   StringRef archName = args.getLastArgValue(OPT_arch);
799349cc55cSDimitry Andric   if (archName.empty()) {
800349cc55cSDimitry Andric     error("must specify -arch");
801349cc55cSDimitry Andric     return nullptr;
802349cc55cSDimitry Andric   }
803fe6060f1SDimitry Andric 
80406c3fb27SDimitry Andric   setPlatformVersions(archName, args);
805bdd1243dSDimitry Andric   auto [cpuType, cpuSubtype] = getCPUTypeFromArchitecture(config->arch());
806fe6060f1SDimitry Andric   switch (cpuType) {
807fe6060f1SDimitry Andric   case CPU_TYPE_X86_64:
808fe6060f1SDimitry Andric     return createX86_64TargetInfo();
809fe6060f1SDimitry Andric   case CPU_TYPE_ARM64:
810fe6060f1SDimitry Andric     return createARM64TargetInfo();
811fe6060f1SDimitry Andric   case CPU_TYPE_ARM64_32:
812fe6060f1SDimitry Andric     return createARM64_32TargetInfo();
813fe6060f1SDimitry Andric   default:
814349cc55cSDimitry Andric     error("missing or unsupported -arch " + archName);
815349cc55cSDimitry Andric     return nullptr;
816fe6060f1SDimitry Andric   }
817fe6060f1SDimitry Andric }
818fe6060f1SDimitry Andric 
819fe6060f1SDimitry Andric static UndefinedSymbolTreatment
getUndefinedSymbolTreatment(const ArgList & args)820fe6060f1SDimitry Andric getUndefinedSymbolTreatment(const ArgList &args) {
821fe6060f1SDimitry Andric   StringRef treatmentStr = args.getLastArgValue(OPT_undefined);
822fe6060f1SDimitry Andric   auto treatment =
823e8d8bef9SDimitry Andric       StringSwitch<UndefinedSymbolTreatment>(treatmentStr)
824fe6060f1SDimitry Andric           .Cases("error", "", UndefinedSymbolTreatment::error)
825e8d8bef9SDimitry Andric           .Case("warning", UndefinedSymbolTreatment::warning)
826e8d8bef9SDimitry Andric           .Case("suppress", UndefinedSymbolTreatment::suppress)
827e8d8bef9SDimitry Andric           .Case("dynamic_lookup", UndefinedSymbolTreatment::dynamic_lookup)
828e8d8bef9SDimitry Andric           .Default(UndefinedSymbolTreatment::unknown);
829fe6060f1SDimitry Andric   if (treatment == UndefinedSymbolTreatment::unknown) {
830e8d8bef9SDimitry Andric     warn(Twine("unknown -undefined TREATMENT '") + treatmentStr +
831e8d8bef9SDimitry Andric          "', defaulting to 'error'");
832fe6060f1SDimitry Andric     treatment = UndefinedSymbolTreatment::error;
833fe6060f1SDimitry Andric   } else if (config->namespaceKind == NamespaceKind::twolevel &&
834fe6060f1SDimitry Andric              (treatment == UndefinedSymbolTreatment::warning ||
835fe6060f1SDimitry Andric               treatment == UndefinedSymbolTreatment::suppress)) {
836fe6060f1SDimitry Andric     if (treatment == UndefinedSymbolTreatment::warning)
837bdd1243dSDimitry Andric       fatal("'-undefined warning' only valid with '-flat_namespace'");
838fe6060f1SDimitry Andric     else
839bdd1243dSDimitry Andric       fatal("'-undefined suppress' only valid with '-flat_namespace'");
840fe6060f1SDimitry Andric     treatment = UndefinedSymbolTreatment::error;
841e8d8bef9SDimitry Andric   }
842fe6060f1SDimitry Andric   return treatment;
8435ffd83dbSDimitry Andric }
8445ffd83dbSDimitry Andric 
getICFLevel(const ArgList & args)845fe6060f1SDimitry Andric static ICFLevel getICFLevel(const ArgList &args) {
846fe6060f1SDimitry Andric   StringRef icfLevelStr = args.getLastArgValue(OPT_icf_eq);
847fe6060f1SDimitry Andric   auto icfLevel = StringSwitch<ICFLevel>(icfLevelStr)
848fe6060f1SDimitry Andric                       .Cases("none", "", ICFLevel::none)
849fe6060f1SDimitry Andric                       .Case("safe", ICFLevel::safe)
850fe6060f1SDimitry Andric                       .Case("all", ICFLevel::all)
851fe6060f1SDimitry Andric                       .Default(ICFLevel::unknown);
852fe6060f1SDimitry Andric   if (icfLevel == ICFLevel::unknown) {
853fe6060f1SDimitry Andric     warn(Twine("unknown --icf=OPTION `") + icfLevelStr +
854fe6060f1SDimitry Andric          "', defaulting to `none'");
855fe6060f1SDimitry Andric     icfLevel = ICFLevel::none;
856fe6060f1SDimitry Andric   }
857fe6060f1SDimitry Andric   return icfLevel;
858fe6060f1SDimitry Andric }
859fe6060f1SDimitry Andric 
getObjCStubsMode(const ArgList & args)860bdd1243dSDimitry Andric static ObjCStubsMode getObjCStubsMode(const ArgList &args) {
861bdd1243dSDimitry Andric   const Arg *arg = args.getLastArg(OPT_objc_stubs_fast, OPT_objc_stubs_small);
862bdd1243dSDimitry Andric   if (!arg)
863bdd1243dSDimitry Andric     return ObjCStubsMode::fast;
864bdd1243dSDimitry Andric 
8657a6dacacSDimitry Andric   if (arg->getOption().getID() == OPT_objc_stubs_small) {
8667a6dacacSDimitry Andric     if (is_contained({AK_arm64e, AK_arm64}, config->arch()))
8677a6dacacSDimitry Andric       return ObjCStubsMode::small;
8687a6dacacSDimitry Andric     else
869bdd1243dSDimitry Andric       warn("-objc_stubs_small is not yet implemented, defaulting to "
870bdd1243dSDimitry Andric            "-objc_stubs_fast");
8717a6dacacSDimitry Andric   }
872bdd1243dSDimitry Andric   return ObjCStubsMode::fast;
873bdd1243dSDimitry Andric }
874bdd1243dSDimitry Andric 
warnIfDeprecatedOption(const Option & opt)875fe6060f1SDimitry Andric static void warnIfDeprecatedOption(const Option &opt) {
8765ffd83dbSDimitry Andric   if (!opt.getGroup().isValid())
8775ffd83dbSDimitry Andric     return;
8785ffd83dbSDimitry Andric   if (opt.getGroup().getID() == OPT_grp_deprecated) {
8795ffd83dbSDimitry Andric     warn("Option `" + opt.getPrefixedName() + "' is deprecated in ld64:");
8805ffd83dbSDimitry Andric     warn(opt.getHelpText());
8815ffd83dbSDimitry Andric   }
8825ffd83dbSDimitry Andric }
8835ffd83dbSDimitry Andric 
warnIfUnimplementedOption(const Option & opt)884fe6060f1SDimitry Andric static void warnIfUnimplementedOption(const Option &opt) {
885e8d8bef9SDimitry Andric   if (!opt.getGroup().isValid() || !opt.hasFlag(DriverFlag::HelpHidden))
8865ffd83dbSDimitry Andric     return;
8875ffd83dbSDimitry Andric   switch (opt.getGroup().getID()) {
8885ffd83dbSDimitry Andric   case OPT_grp_deprecated:
8895ffd83dbSDimitry Andric     // warn about deprecated options elsewhere
8905ffd83dbSDimitry Andric     break;
8915ffd83dbSDimitry Andric   case OPT_grp_undocumented:
8925ffd83dbSDimitry Andric     warn("Option `" + opt.getPrefixedName() +
8935ffd83dbSDimitry Andric          "' is undocumented. Should lld implement it?");
8945ffd83dbSDimitry Andric     break;
8955ffd83dbSDimitry Andric   case OPT_grp_obsolete:
8965ffd83dbSDimitry Andric     warn("Option `" + opt.getPrefixedName() +
8975ffd83dbSDimitry Andric          "' is obsolete. Please modernize your usage.");
8985ffd83dbSDimitry Andric     break;
8995ffd83dbSDimitry Andric   case OPT_grp_ignored:
9005ffd83dbSDimitry Andric     warn("Option `" + opt.getPrefixedName() + "' is ignored.");
9015ffd83dbSDimitry Andric     break;
902349cc55cSDimitry Andric   case OPT_grp_ignored_silently:
903349cc55cSDimitry Andric     break;
9045ffd83dbSDimitry Andric   default:
9055ffd83dbSDimitry Andric     warn("Option `" + opt.getPrefixedName() +
9065ffd83dbSDimitry Andric          "' is not yet implemented. Stay tuned...");
9075ffd83dbSDimitry Andric     break;
9085ffd83dbSDimitry Andric   }
9095ffd83dbSDimitry Andric }
9105ffd83dbSDimitry Andric 
getReproduceOption(InputArgList & args)911fe6060f1SDimitry Andric static const char *getReproduceOption(InputArgList &args) {
912fe6060f1SDimitry Andric   if (const Arg *arg = args.getLastArg(OPT_reproduce))
913e8d8bef9SDimitry Andric     return arg->getValue();
914e8d8bef9SDimitry Andric   return getenv("LLD_REPRODUCE");
915e8d8bef9SDimitry Andric }
916e8d8bef9SDimitry Andric 
917bdd1243dSDimitry Andric // Parse options of the form "old;new".
getOldNewOptions(opt::InputArgList & args,unsigned id)918bdd1243dSDimitry Andric static std::pair<StringRef, StringRef> getOldNewOptions(opt::InputArgList &args,
919bdd1243dSDimitry Andric                                                         unsigned id) {
920bdd1243dSDimitry Andric   auto *arg = args.getLastArg(id);
921bdd1243dSDimitry Andric   if (!arg)
922bdd1243dSDimitry Andric     return {"", ""};
923bdd1243dSDimitry Andric 
924bdd1243dSDimitry Andric   StringRef s = arg->getValue();
925bdd1243dSDimitry Andric   std::pair<StringRef, StringRef> ret = s.split(';');
926bdd1243dSDimitry Andric   if (ret.second.empty())
927bdd1243dSDimitry Andric     error(arg->getSpelling() + " expects 'old;new' format, but got " + s);
928bdd1243dSDimitry Andric   return ret;
929bdd1243dSDimitry Andric }
930bdd1243dSDimitry Andric 
93106c3fb27SDimitry Andric // Parse options of the form "old;new[;extra]".
93206c3fb27SDimitry Andric static std::tuple<StringRef, StringRef, StringRef>
getOldNewOptionsExtra(opt::InputArgList & args,unsigned id)93306c3fb27SDimitry Andric getOldNewOptionsExtra(opt::InputArgList &args, unsigned id) {
93406c3fb27SDimitry Andric   auto [oldDir, second] = getOldNewOptions(args, id);
93506c3fb27SDimitry Andric   auto [newDir, extraDir] = second.split(';');
93606c3fb27SDimitry Andric   return {oldDir, newDir, extraDir};
93706c3fb27SDimitry Andric }
93806c3fb27SDimitry Andric 
parseClangOption(StringRef opt,const Twine & msg)939e8d8bef9SDimitry Andric static void parseClangOption(StringRef opt, const Twine &msg) {
940e8d8bef9SDimitry Andric   std::string err;
941e8d8bef9SDimitry Andric   raw_string_ostream os(err);
942e8d8bef9SDimitry Andric 
943e8d8bef9SDimitry Andric   const char *argv[] = {"lld", opt.data()};
944e8d8bef9SDimitry Andric   if (cl::ParseCommandLineOptions(2, argv, "", &os))
945e8d8bef9SDimitry Andric     return;
946e8d8bef9SDimitry Andric   os.flush();
947e8d8bef9SDimitry Andric   error(msg + ": " + StringRef(err).trim());
948e8d8bef9SDimitry Andric }
949e8d8bef9SDimitry Andric 
parseDylibVersion(const ArgList & args,unsigned id)950fe6060f1SDimitry Andric static uint32_t parseDylibVersion(const ArgList &args, unsigned id) {
951fe6060f1SDimitry Andric   const Arg *arg = args.getLastArg(id);
952e8d8bef9SDimitry Andric   if (!arg)
953e8d8bef9SDimitry Andric     return 0;
954e8d8bef9SDimitry Andric 
955e8d8bef9SDimitry Andric   if (config->outputType != MH_DYLIB) {
956e8d8bef9SDimitry Andric     error(arg->getAsString(args) + ": only valid with -dylib");
957e8d8bef9SDimitry Andric     return 0;
958e8d8bef9SDimitry Andric   }
959e8d8bef9SDimitry Andric 
960e8d8bef9SDimitry Andric   PackedVersion version;
961e8d8bef9SDimitry Andric   if (!version.parse32(arg->getValue())) {
962e8d8bef9SDimitry Andric     error(arg->getAsString(args) + ": malformed version");
963e8d8bef9SDimitry Andric     return 0;
964e8d8bef9SDimitry Andric   }
965e8d8bef9SDimitry Andric 
966e8d8bef9SDimitry Andric   return version.rawValue();
967e8d8bef9SDimitry Andric }
968e8d8bef9SDimitry Andric 
parseProtection(StringRef protStr)969fe6060f1SDimitry Andric static uint32_t parseProtection(StringRef protStr) {
970fe6060f1SDimitry Andric   uint32_t prot = 0;
971fe6060f1SDimitry Andric   for (char c : protStr) {
972fe6060f1SDimitry Andric     switch (c) {
973fe6060f1SDimitry Andric     case 'r':
974fe6060f1SDimitry Andric       prot |= VM_PROT_READ;
975fe6060f1SDimitry Andric       break;
976fe6060f1SDimitry Andric     case 'w':
977fe6060f1SDimitry Andric       prot |= VM_PROT_WRITE;
978fe6060f1SDimitry Andric       break;
979fe6060f1SDimitry Andric     case 'x':
980fe6060f1SDimitry Andric       prot |= VM_PROT_EXECUTE;
981fe6060f1SDimitry Andric       break;
982fe6060f1SDimitry Andric     case '-':
983fe6060f1SDimitry Andric       break;
984fe6060f1SDimitry Andric     default:
985fe6060f1SDimitry Andric       error("unknown -segprot letter '" + Twine(c) + "' in " + protStr);
986fe6060f1SDimitry Andric       return 0;
987fe6060f1SDimitry Andric     }
988fe6060f1SDimitry Andric   }
989fe6060f1SDimitry Andric   return prot;
990fe6060f1SDimitry Andric }
991fe6060f1SDimitry Andric 
parseSectAlign(const opt::InputArgList & args)992fe6060f1SDimitry Andric static std::vector<SectionAlign> parseSectAlign(const opt::InputArgList &args) {
993fe6060f1SDimitry Andric   std::vector<SectionAlign> sectAligns;
994fe6060f1SDimitry Andric   for (const Arg *arg : args.filtered(OPT_sectalign)) {
995fe6060f1SDimitry Andric     StringRef segName = arg->getValue(0);
996fe6060f1SDimitry Andric     StringRef sectName = arg->getValue(1);
997fe6060f1SDimitry Andric     StringRef alignStr = arg->getValue(2);
9987a6dacacSDimitry Andric     alignStr.consume_front_insensitive("0x");
999fe6060f1SDimitry Andric     uint32_t align;
1000fe6060f1SDimitry Andric     if (alignStr.getAsInteger(16, align)) {
1001fe6060f1SDimitry Andric       error("-sectalign: failed to parse '" + StringRef(arg->getValue(2)) +
1002fe6060f1SDimitry Andric             "' as number");
1003fe6060f1SDimitry Andric       continue;
1004fe6060f1SDimitry Andric     }
1005fe6060f1SDimitry Andric     if (!isPowerOf2_32(align)) {
1006fe6060f1SDimitry Andric       error("-sectalign: '" + StringRef(arg->getValue(2)) +
1007fe6060f1SDimitry Andric             "' (in base 16) not a power of two");
1008fe6060f1SDimitry Andric       continue;
1009fe6060f1SDimitry Andric     }
1010fe6060f1SDimitry Andric     sectAligns.push_back({segName, sectName, align});
1011fe6060f1SDimitry Andric   }
1012fe6060f1SDimitry Andric   return sectAligns;
1013fe6060f1SDimitry Andric }
1014fe6060f1SDimitry Andric 
removeSimulator(PlatformType platform)101504eeddc0SDimitry Andric PlatformType macho::removeSimulator(PlatformType platform) {
1016fe6060f1SDimitry Andric   switch (platform) {
101704eeddc0SDimitry Andric   case PLATFORM_IOSSIMULATOR:
101804eeddc0SDimitry Andric     return PLATFORM_IOS;
101904eeddc0SDimitry Andric   case PLATFORM_TVOSSIMULATOR:
102004eeddc0SDimitry Andric     return PLATFORM_TVOS;
102104eeddc0SDimitry Andric   case PLATFORM_WATCHOSSIMULATOR:
102204eeddc0SDimitry Andric     return PLATFORM_WATCHOS;
1023*0fca6ea1SDimitry Andric   case PLATFORM_XROS_SIMULATOR:
1024*0fca6ea1SDimitry Andric     return PLATFORM_XROS;
1025fe6060f1SDimitry Andric   default:
1026fe6060f1SDimitry Andric     return platform;
1027fe6060f1SDimitry Andric   }
1028fe6060f1SDimitry Andric }
1029fe6060f1SDimitry Andric 
supportsNoPie()1030bdd1243dSDimitry Andric static bool supportsNoPie() {
1031bdd1243dSDimitry Andric   return !(config->arch() == AK_arm64 || config->arch() == AK_arm64e ||
1032bdd1243dSDimitry Andric            config->arch() == AK_arm64_32);
1033bdd1243dSDimitry Andric }
1034bdd1243dSDimitry Andric 
shouldAdhocSignByDefault(Architecture arch,PlatformType platform)1035bdd1243dSDimitry Andric static bool shouldAdhocSignByDefault(Architecture arch, PlatformType platform) {
1036bdd1243dSDimitry Andric   if (arch != AK_arm64 && arch != AK_arm64e)
1037bdd1243dSDimitry Andric     return false;
1038bdd1243dSDimitry Andric 
1039bdd1243dSDimitry Andric   return platform == PLATFORM_MACOS || platform == PLATFORM_IOSSIMULATOR ||
1040bdd1243dSDimitry Andric          platform == PLATFORM_TVOSSIMULATOR ||
1041*0fca6ea1SDimitry Andric          platform == PLATFORM_WATCHOSSIMULATOR ||
1042*0fca6ea1SDimitry Andric          platform == PLATFORM_XROS_SIMULATOR;
1043bdd1243dSDimitry Andric }
1044bdd1243dSDimitry Andric 
dataConstDefault(const InputArgList & args)1045fe6060f1SDimitry Andric static bool dataConstDefault(const InputArgList &args) {
1046*0fca6ea1SDimitry Andric   static const std::array<std::pair<PlatformType, VersionTuple>, 6> minVersion =
1047bdd1243dSDimitry Andric       {{{PLATFORM_MACOS, VersionTuple(10, 15)},
104804eeddc0SDimitry Andric         {PLATFORM_IOS, VersionTuple(13, 0)},
104904eeddc0SDimitry Andric         {PLATFORM_TVOS, VersionTuple(13, 0)},
105004eeddc0SDimitry Andric         {PLATFORM_WATCHOS, VersionTuple(6, 0)},
1051*0fca6ea1SDimitry Andric         {PLATFORM_XROS, VersionTuple(1, 0)},
1052bdd1243dSDimitry Andric         {PLATFORM_BRIDGEOS, VersionTuple(4, 0)}}};
105304eeddc0SDimitry Andric   PlatformType platform = removeSimulator(config->platformInfo.target.Platform);
1054fe6060f1SDimitry Andric   auto it = llvm::find_if(minVersion,
1055fe6060f1SDimitry Andric                           [&](const auto &p) { return p.first == platform; });
1056fe6060f1SDimitry Andric   if (it != minVersion.end())
105706c3fb27SDimitry Andric     if (config->platformInfo.target.MinDeployment < it->second)
1058fe6060f1SDimitry Andric       return false;
1059fe6060f1SDimitry Andric 
1060fe6060f1SDimitry Andric   switch (config->outputType) {
1061fe6060f1SDimitry Andric   case MH_EXECUTE:
1062bdd1243dSDimitry Andric     return !(args.hasArg(OPT_no_pie) && supportsNoPie());
1063fe6060f1SDimitry Andric   case MH_BUNDLE:
1064fe6060f1SDimitry Andric     // FIXME: return false when -final_name ...
1065fe6060f1SDimitry Andric     // has prefix "/System/Library/UserEventPlugins/"
1066fe6060f1SDimitry Andric     // or matches "/usr/libexec/locationd" "/usr/libexec/terminusd"
1067fe6060f1SDimitry Andric     return true;
1068fe6060f1SDimitry Andric   case MH_DYLIB:
1069fe6060f1SDimitry Andric     return true;
1070fe6060f1SDimitry Andric   case MH_OBJECT:
1071fe6060f1SDimitry Andric     return false;
1072fe6060f1SDimitry Andric   default:
1073fe6060f1SDimitry Andric     llvm_unreachable(
1074fe6060f1SDimitry Andric         "unsupported output type for determining data-const default");
1075fe6060f1SDimitry Andric   }
1076fe6060f1SDimitry Andric   return false;
1077fe6060f1SDimitry Andric }
1078fe6060f1SDimitry Andric 
shouldEmitChainedFixups(const InputArgList & args)1079bdd1243dSDimitry Andric static bool shouldEmitChainedFixups(const InputArgList &args) {
1080bdd1243dSDimitry Andric   const Arg *arg = args.getLastArg(OPT_fixup_chains, OPT_no_fixup_chains);
1081bdd1243dSDimitry Andric   if (arg && arg->getOption().matches(OPT_no_fixup_chains))
1082bdd1243dSDimitry Andric     return false;
1083bdd1243dSDimitry Andric 
1084*0fca6ea1SDimitry Andric   bool requested = arg && arg->getOption().matches(OPT_fixup_chains);
1085*0fca6ea1SDimitry Andric   if (!config->isPic) {
1086*0fca6ea1SDimitry Andric     if (requested)
1087*0fca6ea1SDimitry Andric       error("-fixup_chains is incompatible with -no_pie");
1088bdd1243dSDimitry Andric 
1089bdd1243dSDimitry Andric     return false;
1090bdd1243dSDimitry Andric   }
1091bdd1243dSDimitry Andric 
1092bdd1243dSDimitry Andric   if (!is_contained({AK_x86_64, AK_x86_64h, AK_arm64}, config->arch())) {
1093*0fca6ea1SDimitry Andric     if (requested)
1094bdd1243dSDimitry Andric       error("-fixup_chains is only supported on x86_64 and arm64 targets");
1095*0fca6ea1SDimitry Andric 
1096bdd1243dSDimitry Andric     return false;
1097bdd1243dSDimitry Andric   }
1098bdd1243dSDimitry Andric 
1099*0fca6ea1SDimitry Andric   if (args.hasArg(OPT_preload)) {
1100*0fca6ea1SDimitry Andric     if (requested)
1101*0fca6ea1SDimitry Andric       error("-fixup_chains is incompatible with -preload");
1102*0fca6ea1SDimitry Andric 
1103bdd1243dSDimitry Andric     return false;
1104bdd1243dSDimitry Andric   }
1105bdd1243dSDimitry Andric 
1106*0fca6ea1SDimitry Andric   if (requested)
1107*0fca6ea1SDimitry Andric     return true;
1108*0fca6ea1SDimitry Andric 
1109*0fca6ea1SDimitry Andric   static const std::array<std::pair<PlatformType, VersionTuple>, 9> minVersion =
1110*0fca6ea1SDimitry Andric       {{
1111*0fca6ea1SDimitry Andric           {PLATFORM_IOS, VersionTuple(13, 4)},
1112*0fca6ea1SDimitry Andric           {PLATFORM_IOSSIMULATOR, VersionTuple(16, 0)},
1113*0fca6ea1SDimitry Andric           {PLATFORM_MACOS, VersionTuple(13, 0)},
1114*0fca6ea1SDimitry Andric           {PLATFORM_TVOS, VersionTuple(14, 0)},
1115*0fca6ea1SDimitry Andric           {PLATFORM_TVOSSIMULATOR, VersionTuple(15, 0)},
1116*0fca6ea1SDimitry Andric           {PLATFORM_WATCHOS, VersionTuple(7, 0)},
1117*0fca6ea1SDimitry Andric           {PLATFORM_WATCHOSSIMULATOR, VersionTuple(8, 0)},
1118*0fca6ea1SDimitry Andric           {PLATFORM_XROS, VersionTuple(1, 0)},
1119*0fca6ea1SDimitry Andric           {PLATFORM_XROS_SIMULATOR, VersionTuple(1, 0)},
1120*0fca6ea1SDimitry Andric       }};
1121*0fca6ea1SDimitry Andric   PlatformType platform = config->platformInfo.target.Platform;
1122*0fca6ea1SDimitry Andric   auto it = llvm::find_if(minVersion,
1123*0fca6ea1SDimitry Andric                           [&](const auto &p) { return p.first == platform; });
1124*0fca6ea1SDimitry Andric 
1125*0fca6ea1SDimitry Andric   // We don't know the versions for other platforms, so default to disabled.
1126*0fca6ea1SDimitry Andric   if (it == minVersion.end())
1127*0fca6ea1SDimitry Andric     return false;
1128*0fca6ea1SDimitry Andric 
1129*0fca6ea1SDimitry Andric   if (it->second > config->platformInfo.target.MinDeployment)
1130*0fca6ea1SDimitry Andric     return false;
1131*0fca6ea1SDimitry Andric 
1132*0fca6ea1SDimitry Andric   return true;
1133*0fca6ea1SDimitry Andric }
1134*0fca6ea1SDimitry Andric 
shouldEmitRelativeMethodLists(const InputArgList & args)1135*0fca6ea1SDimitry Andric static bool shouldEmitRelativeMethodLists(const InputArgList &args) {
1136*0fca6ea1SDimitry Andric   const Arg *arg = args.getLastArg(OPT_objc_relative_method_lists,
1137*0fca6ea1SDimitry Andric                                    OPT_no_objc_relative_method_lists);
1138*0fca6ea1SDimitry Andric   if (arg && arg->getOption().getID() == OPT_objc_relative_method_lists)
1139*0fca6ea1SDimitry Andric     return true;
1140*0fca6ea1SDimitry Andric   if (arg && arg->getOption().getID() == OPT_no_objc_relative_method_lists)
1141*0fca6ea1SDimitry Andric     return false;
1142*0fca6ea1SDimitry Andric 
1143*0fca6ea1SDimitry Andric   // TODO: If no flag is specified, don't default to false, but instead:
1144*0fca6ea1SDimitry Andric   //   - default false on   <   ios14
1145*0fca6ea1SDimitry Andric   //   - default true  on   >=  ios14
1146*0fca6ea1SDimitry Andric   // For now, until this feature is confirmed stable, default to false if no
1147*0fca6ea1SDimitry Andric   // flag is explicitly specified
1148*0fca6ea1SDimitry Andric   return false;
1149bdd1243dSDimitry Andric }
1150bdd1243dSDimitry Andric 
clear()1151fe6060f1SDimitry Andric void SymbolPatterns::clear() {
1152fe6060f1SDimitry Andric   literals.clear();
1153fe6060f1SDimitry Andric   globs.clear();
1154fe6060f1SDimitry Andric }
1155fe6060f1SDimitry Andric 
insert(StringRef symbolName)1156fe6060f1SDimitry Andric void SymbolPatterns::insert(StringRef symbolName) {
1157fe6060f1SDimitry Andric   if (symbolName.find_first_of("*?[]") == StringRef::npos)
1158fe6060f1SDimitry Andric     literals.insert(CachedHashStringRef(symbolName));
1159fe6060f1SDimitry Andric   else if (Expected<GlobPattern> pattern = GlobPattern::create(symbolName))
1160fe6060f1SDimitry Andric     globs.emplace_back(*pattern);
1161fe6060f1SDimitry Andric   else
1162fe6060f1SDimitry Andric     error("invalid symbol-name pattern: " + symbolName);
1163fe6060f1SDimitry Andric }
1164fe6060f1SDimitry Andric 
matchLiteral(StringRef symbolName) const1165fe6060f1SDimitry Andric bool SymbolPatterns::matchLiteral(StringRef symbolName) const {
1166fe6060f1SDimitry Andric   return literals.contains(CachedHashStringRef(symbolName));
1167fe6060f1SDimitry Andric }
1168fe6060f1SDimitry Andric 
matchGlob(StringRef symbolName) const1169fe6060f1SDimitry Andric bool SymbolPatterns::matchGlob(StringRef symbolName) const {
1170fe6060f1SDimitry Andric   for (const GlobPattern &glob : globs)
1171fe6060f1SDimitry Andric     if (glob.match(symbolName))
1172fe6060f1SDimitry Andric       return true;
1173fe6060f1SDimitry Andric   return false;
1174fe6060f1SDimitry Andric }
1175fe6060f1SDimitry Andric 
match(StringRef symbolName) const1176fe6060f1SDimitry Andric bool SymbolPatterns::match(StringRef symbolName) const {
1177fe6060f1SDimitry Andric   return matchLiteral(symbolName) || matchGlob(symbolName);
1178fe6060f1SDimitry Andric }
1179fe6060f1SDimitry Andric 
parseSymbolPatternsFile(const Arg * arg,SymbolPatterns & symbolPatterns)118081ad6265SDimitry Andric static void parseSymbolPatternsFile(const Arg *arg,
118181ad6265SDimitry Andric                                     SymbolPatterns &symbolPatterns) {
1182fe6060f1SDimitry Andric   StringRef path = arg->getValue();
1183bdd1243dSDimitry Andric   std::optional<MemoryBufferRef> buffer = readFile(path);
1184fe6060f1SDimitry Andric   if (!buffer) {
1185fe6060f1SDimitry Andric     error("Could not read symbol file: " + path);
118681ad6265SDimitry Andric     return;
1187fe6060f1SDimitry Andric   }
1188fe6060f1SDimitry Andric   MemoryBufferRef mbref = *buffer;
1189fe6060f1SDimitry Andric   for (StringRef line : args::getLines(mbref)) {
1190fe6060f1SDimitry Andric     line = line.take_until([](char c) { return c == '#'; }).trim();
1191fe6060f1SDimitry Andric     if (!line.empty())
1192fe6060f1SDimitry Andric       symbolPatterns.insert(line);
1193fe6060f1SDimitry Andric   }
1194fe6060f1SDimitry Andric }
119581ad6265SDimitry Andric 
handleSymbolPatterns(InputArgList & args,SymbolPatterns & symbolPatterns,unsigned singleOptionCode,unsigned listFileOptionCode)119681ad6265SDimitry Andric static void handleSymbolPatterns(InputArgList &args,
119781ad6265SDimitry Andric                                  SymbolPatterns &symbolPatterns,
119881ad6265SDimitry Andric                                  unsigned singleOptionCode,
119981ad6265SDimitry Andric                                  unsigned listFileOptionCode) {
120081ad6265SDimitry Andric   for (const Arg *arg : args.filtered(singleOptionCode))
120181ad6265SDimitry Andric     symbolPatterns.insert(arg->getValue());
120281ad6265SDimitry Andric   for (const Arg *arg : args.filtered(listFileOptionCode))
120381ad6265SDimitry Andric     parseSymbolPatternsFile(arg, symbolPatterns);
1204fe6060f1SDimitry Andric }
1205fe6060f1SDimitry Andric 
createFiles(const InputArgList & args)1206349cc55cSDimitry Andric static void createFiles(const InputArgList &args) {
1207fe6060f1SDimitry Andric   TimeTraceScope timeScope("Load input files");
1208fe6060f1SDimitry Andric   // This loop should be reserved for options whose exact ordering matters.
1209fe6060f1SDimitry Andric   // Other options should be handled via filtered() and/or getLastArg().
121004eeddc0SDimitry Andric   bool isLazy = false;
1211*0fca6ea1SDimitry Andric   // If we've processed an opening --start-lib, without a matching --end-lib
1212*0fca6ea1SDimitry Andric   bool inLib = false;
1213fe6060f1SDimitry Andric   for (const Arg *arg : args) {
1214fe6060f1SDimitry Andric     const Option &opt = arg->getOption();
1215fe6060f1SDimitry Andric     warnIfDeprecatedOption(opt);
1216fe6060f1SDimitry Andric     warnIfUnimplementedOption(opt);
1217fe6060f1SDimitry Andric 
1218fe6060f1SDimitry Andric     switch (opt.getID()) {
1219fe6060f1SDimitry Andric     case OPT_INPUT:
1220fcaf7f86SDimitry Andric       addFile(rerootPath(arg->getValue()), LoadType::CommandLine, isLazy);
1221fe6060f1SDimitry Andric       break;
1222fe6060f1SDimitry Andric     case OPT_needed_library:
1223fe6060f1SDimitry Andric       if (auto *dylibFile = dyn_cast_or_null<DylibFile>(
1224fcaf7f86SDimitry Andric               addFile(rerootPath(arg->getValue()), LoadType::CommandLine)))
1225fe6060f1SDimitry Andric         dylibFile->forceNeeded = true;
1226fe6060f1SDimitry Andric       break;
1227fe6060f1SDimitry Andric     case OPT_reexport_library:
1228349cc55cSDimitry Andric       if (auto *dylibFile = dyn_cast_or_null<DylibFile>(
1229fcaf7f86SDimitry Andric               addFile(rerootPath(arg->getValue()), LoadType::CommandLine))) {
1230fe6060f1SDimitry Andric         config->hasReexports = true;
1231fe6060f1SDimitry Andric         dylibFile->reexport = true;
1232fe6060f1SDimitry Andric       }
1233fe6060f1SDimitry Andric       break;
1234fe6060f1SDimitry Andric     case OPT_weak_library:
1235fe6060f1SDimitry Andric       if (auto *dylibFile = dyn_cast_or_null<DylibFile>(
1236fcaf7f86SDimitry Andric               addFile(rerootPath(arg->getValue()), LoadType::CommandLine)))
1237fe6060f1SDimitry Andric         dylibFile->forceWeakImport = true;
1238fe6060f1SDimitry Andric       break;
1239fe6060f1SDimitry Andric     case OPT_filelist:
124004eeddc0SDimitry Andric       addFileList(arg->getValue(), isLazy);
1241fe6060f1SDimitry Andric       break;
1242fe6060f1SDimitry Andric     case OPT_force_load:
1243fcaf7f86SDimitry Andric       addFile(rerootPath(arg->getValue()), LoadType::CommandLineForce);
1244fe6060f1SDimitry Andric       break;
1245972a253aSDimitry Andric     case OPT_load_hidden:
1246972a253aSDimitry Andric       addFile(rerootPath(arg->getValue()), LoadType::CommandLine,
1247972a253aSDimitry Andric               /*isLazy=*/false, /*isExplicit=*/true, /*isBundleLoader=*/false,
1248972a253aSDimitry Andric               /*isForceHidden=*/true);
1249972a253aSDimitry Andric       break;
1250fe6060f1SDimitry Andric     case OPT_l:
1251fe6060f1SDimitry Andric     case OPT_needed_l:
1252fe6060f1SDimitry Andric     case OPT_reexport_l:
1253fe6060f1SDimitry Andric     case OPT_weak_l:
1254972a253aSDimitry Andric     case OPT_hidden_l:
1255fe6060f1SDimitry Andric       addLibrary(arg->getValue(), opt.getID() == OPT_needed_l,
1256fe6060f1SDimitry Andric                  opt.getID() == OPT_weak_l, opt.getID() == OPT_reexport_l,
1257972a253aSDimitry Andric                  opt.getID() == OPT_hidden_l,
1258fcaf7f86SDimitry Andric                  /*isExplicit=*/true, LoadType::CommandLine);
1259fe6060f1SDimitry Andric       break;
1260fe6060f1SDimitry Andric     case OPT_framework:
1261fe6060f1SDimitry Andric     case OPT_needed_framework:
1262fe6060f1SDimitry Andric     case OPT_reexport_framework:
1263fe6060f1SDimitry Andric     case OPT_weak_framework:
1264fe6060f1SDimitry Andric       addFramework(arg->getValue(), opt.getID() == OPT_needed_framework,
1265fe6060f1SDimitry Andric                    opt.getID() == OPT_weak_framework,
1266349cc55cSDimitry Andric                    opt.getID() == OPT_reexport_framework, /*isExplicit=*/true,
1267fcaf7f86SDimitry Andric                    LoadType::CommandLine);
1268fe6060f1SDimitry Andric       break;
126904eeddc0SDimitry Andric     case OPT_start_lib:
1270*0fca6ea1SDimitry Andric       if (inLib)
127104eeddc0SDimitry Andric         error("nested --start-lib");
1272*0fca6ea1SDimitry Andric       inLib = true;
1273*0fca6ea1SDimitry Andric       if (!config->allLoad)
127404eeddc0SDimitry Andric         isLazy = true;
127504eeddc0SDimitry Andric       break;
127604eeddc0SDimitry Andric     case OPT_end_lib:
1277*0fca6ea1SDimitry Andric       if (!inLib)
127804eeddc0SDimitry Andric         error("stray --end-lib");
1279*0fca6ea1SDimitry Andric       inLib = false;
128004eeddc0SDimitry Andric       isLazy = false;
128104eeddc0SDimitry Andric       break;
1282fe6060f1SDimitry Andric     default:
1283fe6060f1SDimitry Andric       break;
1284fe6060f1SDimitry Andric     }
1285fe6060f1SDimitry Andric   }
1286fe6060f1SDimitry Andric }
1287fe6060f1SDimitry Andric 
gatherInputSections()1288fe6060f1SDimitry Andric static void gatherInputSections() {
1289fe6060f1SDimitry Andric   TimeTraceScope timeScope("Gathering input sections");
1290fe6060f1SDimitry Andric   for (const InputFile *file : inputFiles) {
129181ad6265SDimitry Andric     for (const Section *section : file->sections) {
129281ad6265SDimitry Andric       // Compact unwind entries require special handling elsewhere. (In
129381ad6265SDimitry Andric       // contrast, EH frames are handled like regular ConcatInputSections.)
129481ad6265SDimitry Andric       if (section->name == section_names::compactUnwind)
1295349cc55cSDimitry Andric         continue;
1296*0fca6ea1SDimitry Andric       // Addrsig sections contain metadata only needed at link time.
1297*0fca6ea1SDimitry Andric       if (section->name == section_names::addrSig)
1298fe6060f1SDimitry Andric         continue;
1299*0fca6ea1SDimitry Andric       for (const Subsection &subsection : section->subsections)
1300*0fca6ea1SDimitry Andric         addInputSection(subsection.isec);
1301fe6060f1SDimitry Andric     }
1302fcaf7f86SDimitry Andric     if (!file->objCImageInfo.empty())
1303fcaf7f86SDimitry Andric       in.objCImageInfo->addFile(file);
1304fe6060f1SDimitry Andric   }
1305fe6060f1SDimitry Andric }
1306fe6060f1SDimitry Andric 
foldIdenticalLiterals()1307fe6060f1SDimitry Andric static void foldIdenticalLiterals() {
130881ad6265SDimitry Andric   TimeTraceScope timeScope("Fold identical literals");
1309fe6060f1SDimitry Andric   // We always create a cStringSection, regardless of whether dedupLiterals is
1310fe6060f1SDimitry Andric   // true. If it isn't, we simply create a non-deduplicating CStringSection.
1311fe6060f1SDimitry Andric   // Either way, we must unconditionally finalize it here.
1312fe6060f1SDimitry Andric   in.cStringSection->finalizeContents();
1313bdd1243dSDimitry Andric   in.objcMethnameSection->finalizeContents();
1314fe6060f1SDimitry Andric   in.wordLiteralSection->finalizeContents();
1315fe6060f1SDimitry Andric }
1316fe6060f1SDimitry Andric 
addSynthenticMethnames()1317bdd1243dSDimitry Andric static void addSynthenticMethnames() {
1318bdd1243dSDimitry Andric   std::string &data = *make<std::string>();
1319bdd1243dSDimitry Andric   llvm::raw_string_ostream os(data);
1320bdd1243dSDimitry Andric   for (Symbol *sym : symtab->getSymbols())
1321bdd1243dSDimitry Andric     if (isa<Undefined>(sym))
1322*0fca6ea1SDimitry Andric       if (ObjCStubsSection::isObjCStubSymbol(sym))
1323*0fca6ea1SDimitry Andric         os << ObjCStubsSection::getMethname(sym) << '\0';
1324bdd1243dSDimitry Andric 
1325bdd1243dSDimitry Andric   if (data.empty())
1326bdd1243dSDimitry Andric     return;
1327bdd1243dSDimitry Andric 
1328bdd1243dSDimitry Andric   const auto *buf = reinterpret_cast<const uint8_t *>(data.c_str());
1329bdd1243dSDimitry Andric   Section &section = *make<Section>(/*file=*/nullptr, segment_names::text,
1330bdd1243dSDimitry Andric                                     section_names::objcMethname,
1331bdd1243dSDimitry Andric                                     S_CSTRING_LITERALS, /*addr=*/0);
1332bdd1243dSDimitry Andric 
1333bdd1243dSDimitry Andric   auto *isec =
1334bdd1243dSDimitry Andric       make<CStringInputSection>(section, ArrayRef<uint8_t>{buf, data.size()},
1335bdd1243dSDimitry Andric                                 /*align=*/1, /*dedupLiterals=*/true);
1336bdd1243dSDimitry Andric   isec->splitIntoPieces();
1337bdd1243dSDimitry Andric   for (auto &piece : isec->pieces)
1338bdd1243dSDimitry Andric     piece.live = true;
1339bdd1243dSDimitry Andric   section.subsections.push_back({0, isec});
1340bdd1243dSDimitry Andric   in.objcMethnameSection->addInput(isec);
1341bdd1243dSDimitry Andric   in.objcMethnameSection->isec->markLive(0);
1342bdd1243dSDimitry Andric }
1343bdd1243dSDimitry Andric 
referenceStubBinder()1344fe6060f1SDimitry Andric static void referenceStubBinder() {
1345fe6060f1SDimitry Andric   bool needsStubHelper = config->outputType == MH_DYLIB ||
1346fe6060f1SDimitry Andric                          config->outputType == MH_EXECUTE ||
1347fe6060f1SDimitry Andric                          config->outputType == MH_BUNDLE;
1348fe6060f1SDimitry Andric   if (!needsStubHelper || !symtab->find("dyld_stub_binder"))
1349fe6060f1SDimitry Andric     return;
1350fe6060f1SDimitry Andric 
1351fe6060f1SDimitry Andric   // dyld_stub_binder is used by dyld to resolve lazy bindings. This code here
1352fe6060f1SDimitry Andric   // adds a opportunistic reference to dyld_stub_binder if it happens to exist.
1353fe6060f1SDimitry Andric   // dyld_stub_binder is in libSystem.dylib, which is usually linked in. This
1354fe6060f1SDimitry Andric   // isn't needed for correctness, but the presence of that symbol suppresses
1355fe6060f1SDimitry Andric   // "no symbols" diagnostics from `nm`.
1356bdd1243dSDimitry Andric   // StubHelperSection::setUp() adds a reference and errors out if
1357fe6060f1SDimitry Andric   // dyld_stub_binder doesn't exist in case it is actually needed.
1358fe6060f1SDimitry Andric   symtab->addUndefined("dyld_stub_binder", /*file=*/nullptr, /*isWeak=*/false);
1359fe6060f1SDimitry Andric }
1360fe6060f1SDimitry Andric 
createAliases()1361bdd1243dSDimitry Andric static void createAliases() {
1362bdd1243dSDimitry Andric   for (const auto &pair : config->aliasedSymbols) {
1363bdd1243dSDimitry Andric     if (const auto &sym = symtab->find(pair.first)) {
1364bdd1243dSDimitry Andric       if (const auto &defined = dyn_cast<Defined>(sym)) {
1365bdd1243dSDimitry Andric         symtab->aliasDefined(defined, pair.second, defined->getFile())
1366bdd1243dSDimitry Andric             ->noDeadStrip = true;
1367bdd1243dSDimitry Andric       } else {
1368bdd1243dSDimitry Andric         error("TODO: support aliasing to symbols of kind " +
1369bdd1243dSDimitry Andric               Twine(sym->kind()));
1370bdd1243dSDimitry Andric       }
1371bdd1243dSDimitry Andric     } else {
1372bdd1243dSDimitry Andric       warn("undefined base symbol '" + pair.first + "' for alias '" +
1373bdd1243dSDimitry Andric            pair.second + "'\n");
1374bdd1243dSDimitry Andric     }
1375bdd1243dSDimitry Andric   }
1376bdd1243dSDimitry Andric 
1377bdd1243dSDimitry Andric   for (const InputFile *file : inputFiles) {
1378bdd1243dSDimitry Andric     if (auto *objFile = dyn_cast<ObjFile>(file)) {
1379bdd1243dSDimitry Andric       for (const AliasSymbol *alias : objFile->aliases) {
1380bdd1243dSDimitry Andric         if (const auto &aliased = symtab->find(alias->getAliasedName())) {
1381bdd1243dSDimitry Andric           if (const auto &defined = dyn_cast<Defined>(aliased)) {
1382bdd1243dSDimitry Andric             symtab->aliasDefined(defined, alias->getName(), alias->getFile(),
1383bdd1243dSDimitry Andric                                  alias->privateExtern);
1384bdd1243dSDimitry Andric           } else {
1385bdd1243dSDimitry Andric             // Common, dylib, and undefined symbols are all valid alias
1386bdd1243dSDimitry Andric             // referents (undefineds can become valid Defined symbols later on
1387bdd1243dSDimitry Andric             // in the link.)
1388bdd1243dSDimitry Andric             error("TODO: support aliasing to symbols of kind " +
1389bdd1243dSDimitry Andric                   Twine(aliased->kind()));
1390bdd1243dSDimitry Andric           }
1391bdd1243dSDimitry Andric         } else {
1392bdd1243dSDimitry Andric           // This shouldn't happen since MC generates undefined symbols to
1393bdd1243dSDimitry Andric           // represent the alias referents. Thus we fatal() instead of just
1394bdd1243dSDimitry Andric           // warning here.
1395bdd1243dSDimitry Andric           fatal("unable to find alias referent " + alias->getAliasedName() +
1396bdd1243dSDimitry Andric                 " for " + alias->getName());
1397bdd1243dSDimitry Andric         }
1398bdd1243dSDimitry Andric       }
1399bdd1243dSDimitry Andric     }
1400bdd1243dSDimitry Andric   }
1401bdd1243dSDimitry Andric }
1402bdd1243dSDimitry Andric 
handleExplicitExports()1403bdd1243dSDimitry Andric static void handleExplicitExports() {
14045f757f3fSDimitry Andric   static constexpr int kMaxWarnings = 3;
1405bdd1243dSDimitry Andric   if (config->hasExplicitExports) {
14065f757f3fSDimitry Andric     std::atomic<uint64_t> warningsCount{0};
14075f757f3fSDimitry Andric     parallelForEach(symtab->getSymbols(), [&warningsCount](Symbol *sym) {
1408bdd1243dSDimitry Andric       if (auto *defined = dyn_cast<Defined>(sym)) {
140906c3fb27SDimitry Andric         if (config->exportedSymbols.match(sym->getName())) {
1410bdd1243dSDimitry Andric           if (defined->privateExtern) {
1411bdd1243dSDimitry Andric             if (defined->weakDefCanBeHidden) {
1412bdd1243dSDimitry Andric               // weak_def_can_be_hidden symbols behave similarly to
1413bdd1243dSDimitry Andric               // private_extern symbols in most cases, except for when
1414bdd1243dSDimitry Andric               // it is explicitly exported.
1415bdd1243dSDimitry Andric               // The former can be exported but the latter cannot.
1416bdd1243dSDimitry Andric               defined->privateExtern = false;
1417bdd1243dSDimitry Andric             } else {
14185f757f3fSDimitry Andric               // Only print the first 3 warnings verbosely, and
14195f757f3fSDimitry Andric               // shorten the rest to avoid crowding logs.
14205f757f3fSDimitry Andric               if (warningsCount.fetch_add(1, std::memory_order_relaxed) <
14215f757f3fSDimitry Andric                   kMaxWarnings)
1422bdd1243dSDimitry Andric                 warn("cannot export hidden symbol " + toString(*defined) +
1423bdd1243dSDimitry Andric                      "\n>>> defined in " + toString(defined->getFile()));
1424bdd1243dSDimitry Andric             }
1425bdd1243dSDimitry Andric           }
1426bdd1243dSDimitry Andric         } else {
1427bdd1243dSDimitry Andric           defined->privateExtern = true;
1428bdd1243dSDimitry Andric         }
142906c3fb27SDimitry Andric       } else if (auto *dysym = dyn_cast<DylibSymbol>(sym)) {
143006c3fb27SDimitry Andric         dysym->shouldReexport = config->exportedSymbols.match(sym->getName());
1431bdd1243dSDimitry Andric       }
1432bdd1243dSDimitry Andric     });
14335f757f3fSDimitry Andric     if (warningsCount > kMaxWarnings)
14345f757f3fSDimitry Andric       warn("<... " + Twine(warningsCount - kMaxWarnings) +
14355f757f3fSDimitry Andric            " more similar warnings...>");
1436bdd1243dSDimitry Andric   } else if (!config->unexportedSymbols.empty()) {
1437bdd1243dSDimitry Andric     parallelForEach(symtab->getSymbols(), [](Symbol *sym) {
1438bdd1243dSDimitry Andric       if (auto *defined = dyn_cast<Defined>(sym))
1439bdd1243dSDimitry Andric         if (config->unexportedSymbols.match(defined->getName()))
1440bdd1243dSDimitry Andric           defined->privateExtern = true;
1441bdd1243dSDimitry Andric     });
1442bdd1243dSDimitry Andric   }
1443bdd1243dSDimitry Andric }
1444bdd1243dSDimitry Andric 
eraseInitializerSymbols()1445*0fca6ea1SDimitry Andric static void eraseInitializerSymbols() {
1446*0fca6ea1SDimitry Andric   for (ConcatInputSection *isec : in.initOffsets->inputs())
1447*0fca6ea1SDimitry Andric     for (Defined *sym : isec->symbols)
1448*0fca6ea1SDimitry Andric       sym->used = false;
1449*0fca6ea1SDimitry Andric }
1450*0fca6ea1SDimitry Andric 
getRuntimePaths(opt::InputArgList & args)1451*0fca6ea1SDimitry Andric static SmallVector<StringRef, 0> getRuntimePaths(opt::InputArgList &args) {
1452*0fca6ea1SDimitry Andric   SmallVector<StringRef, 0> vals;
1453*0fca6ea1SDimitry Andric   DenseSet<StringRef> seen;
1454*0fca6ea1SDimitry Andric   for (const Arg *arg : args.filtered(OPT_rpath)) {
1455*0fca6ea1SDimitry Andric     StringRef val = arg->getValue();
1456*0fca6ea1SDimitry Andric     if (seen.insert(val).second)
1457*0fca6ea1SDimitry Andric       vals.push_back(val);
1458*0fca6ea1SDimitry Andric     else if (config->warnDuplicateRpath)
1459*0fca6ea1SDimitry Andric       warn("duplicate -rpath '" + val + "' ignored [--warn-duplicate-rpath]");
1460*0fca6ea1SDimitry Andric   }
1461*0fca6ea1SDimitry Andric   return vals;
1462*0fca6ea1SDimitry Andric }
1463*0fca6ea1SDimitry Andric 
146406c3fb27SDimitry Andric namespace lld {
146506c3fb27SDimitry Andric namespace macho {
link(ArrayRef<const char * > argsArr,llvm::raw_ostream & stdoutOS,llvm::raw_ostream & stderrOS,bool exitEarly,bool disableOutput)146606c3fb27SDimitry Andric bool link(ArrayRef<const char *> argsArr, llvm::raw_ostream &stdoutOS,
146706c3fb27SDimitry Andric           llvm::raw_ostream &stderrOS, bool exitEarly, bool disableOutput) {
146804eeddc0SDimitry Andric   // This driver-specific context will be freed later by lldMain().
146904eeddc0SDimitry Andric   auto *ctx = new CommonLinkerContext;
14705ffd83dbSDimitry Andric 
147104eeddc0SDimitry Andric   ctx->e.initialize(stdoutOS, stderrOS, exitEarly, disableOutput);
147204eeddc0SDimitry Andric   ctx->e.cleanupCallback = []() {
1473349cc55cSDimitry Andric     resolvedFrameworks.clear();
1474349cc55cSDimitry Andric     resolvedLibraries.clear();
1475349cc55cSDimitry Andric     cachedReads.clear();
1476349cc55cSDimitry Andric     concatOutputSections.clear();
1477349cc55cSDimitry Andric     inputFiles.clear();
1478349cc55cSDimitry Andric     inputSections.clear();
1479*0fca6ea1SDimitry Andric     inputSectionsOrder = 0;
1480349cc55cSDimitry Andric     loadedArchives.clear();
148181ad6265SDimitry Andric     loadedObjectFrameworks.clear();
1482bdd1243dSDimitry Andric     missingAutolinkWarnings.clear();
1483349cc55cSDimitry Andric     syntheticSections.clear();
1484349cc55cSDimitry Andric     thunkMap.clear();
14855f757f3fSDimitry Andric     unprocessedLCLinkerOptions.clear();
1486*0fca6ea1SDimitry Andric     ObjCSelRefsHelper::cleanup();
1487349cc55cSDimitry Andric 
1488349cc55cSDimitry Andric     firstTLVDataSection = nullptr;
1489349cc55cSDimitry Andric     tar = nullptr;
1490349cc55cSDimitry Andric     memset(&in, 0, sizeof(in));
1491349cc55cSDimitry Andric 
1492349cc55cSDimitry Andric     resetLoadedDylibs();
1493349cc55cSDimitry Andric     resetOutputSegments();
1494349cc55cSDimitry Andric     resetWriter();
1495349cc55cSDimitry Andric     InputFile::resetIdCount();
1496*0fca6ea1SDimitry Andric 
1497*0fca6ea1SDimitry Andric     objc::doCleanup();
1498349cc55cSDimitry Andric   };
1499e8d8bef9SDimitry Andric 
150004eeddc0SDimitry Andric   ctx->e.logName = args::getFilenameWithoutExe(argsArr[0]);
1501fe6060f1SDimitry Andric 
15025ffd83dbSDimitry Andric   MachOOptTable parser;
1503fe6060f1SDimitry Andric   InputArgList args = parser.parse(argsArr.slice(1));
1504fe6060f1SDimitry Andric 
150504eeddc0SDimitry Andric   ctx->e.errorLimitExceededMsg = "too many errors emitted, stopping now "
1506fe6060f1SDimitry Andric                                  "(use --error-limit=0 to see all errors)";
150704eeddc0SDimitry Andric   ctx->e.errorLimit = args::getInteger(args, OPT_error_limit_eq, 20);
150804eeddc0SDimitry Andric   ctx->e.verbose = args.hasArg(OPT_verbose);
15095ffd83dbSDimitry Andric 
15105ffd83dbSDimitry Andric   if (args.hasArg(OPT_help_hidden)) {
15115ffd83dbSDimitry Andric     parser.printHelp(argsArr[0], /*showHidden=*/true);
15125ffd83dbSDimitry Andric     return true;
1513e8d8bef9SDimitry Andric   }
1514e8d8bef9SDimitry Andric   if (args.hasArg(OPT_help)) {
15155ffd83dbSDimitry Andric     parser.printHelp(argsArr[0], /*showHidden=*/false);
15165ffd83dbSDimitry Andric     return true;
15175ffd83dbSDimitry Andric   }
1518e8d8bef9SDimitry Andric   if (args.hasArg(OPT_version)) {
1519e8d8bef9SDimitry Andric     message(getLLDVersion());
1520e8d8bef9SDimitry Andric     return true;
1521e8d8bef9SDimitry Andric   }
1522e8d8bef9SDimitry Andric 
152304eeddc0SDimitry Andric   config = std::make_unique<Configuration>();
152404eeddc0SDimitry Andric   symtab = std::make_unique<SymbolTable>();
152581ad6265SDimitry Andric   config->outputType = getOutputType(args);
1526fe6060f1SDimitry Andric   target = createTargetInfo(args);
152704eeddc0SDimitry Andric   depTracker = std::make_unique<DependencyTracker>(
152804eeddc0SDimitry Andric       args.getLastArgValue(OPT_dependency_info));
152906c3fb27SDimitry Andric 
153006c3fb27SDimitry Andric   config->ltoo = args::getInteger(args, OPT_lto_O, 2);
153106c3fb27SDimitry Andric   if (config->ltoo > 3)
153206c3fb27SDimitry Andric     error("--lto-O: invalid optimization level: " + Twine(config->ltoo));
153306c3fb27SDimitry Andric   unsigned ltoCgo =
153406c3fb27SDimitry Andric       args::getInteger(args, OPT_lto_CGO, args::getCGOptLevel(config->ltoo));
153506c3fb27SDimitry Andric   if (auto level = CodeGenOpt::getLevel(ltoCgo))
153606c3fb27SDimitry Andric     config->ltoCgo = *level;
153706c3fb27SDimitry Andric   else
153806c3fb27SDimitry Andric     error("--lto-CGO: invalid codegen optimization level: " + Twine(ltoCgo));
153906c3fb27SDimitry Andric 
1540349cc55cSDimitry Andric   if (errorCount())
1541349cc55cSDimitry Andric     return false;
1542349cc55cSDimitry Andric 
154381ad6265SDimitry Andric   if (args.hasArg(OPT_pagezero_size)) {
154481ad6265SDimitry Andric     uint64_t pagezeroSize = args::getHex(args, OPT_pagezero_size, 0);
154581ad6265SDimitry Andric 
154681ad6265SDimitry Andric     // ld64 does something really weird. It attempts to realign the value to the
1547bdd1243dSDimitry Andric     // page size, but assumes the page size is 4K. This doesn't work with most
1548bdd1243dSDimitry Andric     // of Apple's ARM64 devices, which use a page size of 16K. This means that
1549bdd1243dSDimitry Andric     // it will first 4K align it by rounding down, then round up to 16K.  This
1550bdd1243dSDimitry Andric     // probably only happened because no one using this arg with anything other
1551bdd1243dSDimitry Andric     // then 0, so no one checked if it did what is what it says it does.
155281ad6265SDimitry Andric 
155381ad6265SDimitry Andric     // So we are not copying this weird behavior and doing the it in a logical
155481ad6265SDimitry Andric     // way, by always rounding down to page size.
155581ad6265SDimitry Andric     if (!isAligned(Align(target->getPageSize()), pagezeroSize)) {
155681ad6265SDimitry Andric       pagezeroSize -= pagezeroSize % target->getPageSize();
155781ad6265SDimitry Andric       warn("__PAGEZERO size is not page aligned, rounding down to 0x" +
155881ad6265SDimitry Andric            Twine::utohexstr(pagezeroSize));
155981ad6265SDimitry Andric     }
156081ad6265SDimitry Andric 
156181ad6265SDimitry Andric     target->pageZeroSize = pagezeroSize;
156281ad6265SDimitry Andric   }
156381ad6265SDimitry Andric 
1564349cc55cSDimitry Andric   config->osoPrefix = args.getLastArgValue(OPT_oso_prefix);
1565349cc55cSDimitry Andric   if (!config->osoPrefix.empty()) {
1566349cc55cSDimitry Andric     // Expand special characters, such as ".", "..", or  "~", if present.
1567349cc55cSDimitry Andric     // Note: LD64 only expands "." and not other special characters.
1568349cc55cSDimitry Andric     // That seems silly to imitate so we will not try to follow it, but rather
1569349cc55cSDimitry Andric     // just use real_path() to do it.
1570349cc55cSDimitry Andric 
1571349cc55cSDimitry Andric     // The max path length is 4096, in theory. However that seems quite long
1572349cc55cSDimitry Andric     // and seems unlikely that any one would want to strip everything from the
1573349cc55cSDimitry Andric     // path. Hence we've picked a reasonably large number here.
1574349cc55cSDimitry Andric     SmallString<1024> expanded;
1575349cc55cSDimitry Andric     if (!fs::real_path(config->osoPrefix, expanded,
1576349cc55cSDimitry Andric                        /*expand_tilde=*/true)) {
1577349cc55cSDimitry Andric       // Note: LD64 expands "." to be `<current_dir>/`
1578349cc55cSDimitry Andric       // (ie., it has a slash suffix) whereas real_path() doesn't.
1579349cc55cSDimitry Andric       // So we have to append '/' to be consistent.
1580349cc55cSDimitry Andric       StringRef sep = sys::path::get_separator();
1581349cc55cSDimitry Andric       // real_path removes trailing slashes as part of the normalization, but
1582349cc55cSDimitry Andric       // these are meaningful for our text based stripping
1583*0fca6ea1SDimitry Andric       if (config->osoPrefix == "." || config->osoPrefix.ends_with(sep))
1584349cc55cSDimitry Andric         expanded += sep;
158504eeddc0SDimitry Andric       config->osoPrefix = saver().save(expanded.str());
1586349cc55cSDimitry Andric     }
1587349cc55cSDimitry Andric   }
1588fe6060f1SDimitry Andric 
1589bdd1243dSDimitry Andric   bool pie = args.hasFlag(OPT_pie, OPT_no_pie, true);
1590bdd1243dSDimitry Andric   if (!supportsNoPie() && !pie) {
1591bdd1243dSDimitry Andric     warn("-no_pie ignored for arm64");
1592bdd1243dSDimitry Andric     pie = true;
1593bdd1243dSDimitry Andric   }
1594bdd1243dSDimitry Andric 
1595bdd1243dSDimitry Andric   config->isPic = config->outputType == MH_DYLIB ||
1596bdd1243dSDimitry Andric                   config->outputType == MH_BUNDLE ||
1597bdd1243dSDimitry Andric                   (config->outputType == MH_EXECUTE && pie);
1598bdd1243dSDimitry Andric 
1599fe6060f1SDimitry Andric   // Must be set before any InputSections and Symbols are created.
1600fe6060f1SDimitry Andric   config->deadStrip = args.hasArg(OPT_dead_strip);
1601fe6060f1SDimitry Andric 
1602fe6060f1SDimitry Andric   config->systemLibraryRoots = getSystemLibraryRoots(args);
1603e8d8bef9SDimitry Andric   if (const char *path = getReproduceOption(args)) {
1604e8d8bef9SDimitry Andric     // Note that --reproduce is a debug option so you can ignore it
1605e8d8bef9SDimitry Andric     // if you are trying to understand the whole picture of the code.
1606e8d8bef9SDimitry Andric     Expected<std::unique_ptr<TarWriter>> errOrWriter =
1607e8d8bef9SDimitry Andric         TarWriter::create(path, path::stem(path));
1608e8d8bef9SDimitry Andric     if (errOrWriter) {
1609e8d8bef9SDimitry Andric       tar = std::move(*errOrWriter);
1610e8d8bef9SDimitry Andric       tar->append("response.txt", createResponseFile(args));
1611e8d8bef9SDimitry Andric       tar->append("version.txt", getLLDVersion() + "\n");
1612e8d8bef9SDimitry Andric     } else {
1613e8d8bef9SDimitry Andric       error("--reproduce: " + toString(errOrWriter.takeError()));
1614e8d8bef9SDimitry Andric     }
1615e8d8bef9SDimitry Andric   }
16165ffd83dbSDimitry Andric 
1617fe6060f1SDimitry Andric   if (auto *arg = args.getLastArg(OPT_threads_eq)) {
1618fe6060f1SDimitry Andric     StringRef v(arg->getValue());
1619fe6060f1SDimitry Andric     unsigned threads = 0;
1620fe6060f1SDimitry Andric     if (!llvm::to_integer(v, threads, 0) || threads == 0)
1621fe6060f1SDimitry Andric       error(arg->getSpelling() + ": expected a positive integer, but got '" +
1622fe6060f1SDimitry Andric             arg->getValue() + "'");
1623fe6060f1SDimitry Andric     parallel::strategy = hardware_concurrency(threads);
1624fe6060f1SDimitry Andric     config->thinLTOJobs = v;
1625fe6060f1SDimitry Andric   }
1626fe6060f1SDimitry Andric   if (auto *arg = args.getLastArg(OPT_thinlto_jobs_eq))
1627fe6060f1SDimitry Andric     config->thinLTOJobs = arg->getValue();
1628fe6060f1SDimitry Andric   if (!get_threadpool_strategy(config->thinLTOJobs))
1629fe6060f1SDimitry Andric     error("--thinlto-jobs: invalid job count: " + config->thinLTOJobs);
16305ffd83dbSDimitry Andric 
1631fe6060f1SDimitry Andric   for (const Arg *arg : args.filtered(OPT_u)) {
1632fe6060f1SDimitry Andric     config->explicitUndefineds.push_back(symtab->addUndefined(
1633fe6060f1SDimitry Andric         arg->getValue(), /*file=*/nullptr, /*isWeakRef=*/false));
1634fe6060f1SDimitry Andric   }
1635fe6060f1SDimitry Andric 
1636fe6060f1SDimitry Andric   for (const Arg *arg : args.filtered(OPT_U))
1637fe6060f1SDimitry Andric     config->explicitDynamicLookups.insert(arg->getValue());
1638fe6060f1SDimitry Andric 
1639fe6060f1SDimitry Andric   config->mapFile = args.getLastArgValue(OPT_map);
1640fe6060f1SDimitry Andric   config->optimize = args::getInteger(args, OPT_O, 1);
16415ffd83dbSDimitry Andric   config->outputFile = args.getLastArgValue(OPT_o, "a.out");
1642fe6060f1SDimitry Andric   config->finalOutput =
1643fe6060f1SDimitry Andric       args.getLastArgValue(OPT_final_output, config->outputFile);
1644fe6060f1SDimitry Andric   config->astPaths = args.getAllArgValues(OPT_add_ast_path);
1645e8d8bef9SDimitry Andric   config->headerPad = args::getHex(args, OPT_headerpad, /*Default=*/32);
1646e8d8bef9SDimitry Andric   config->headerPadMaxInstallNames =
1647e8d8bef9SDimitry Andric       args.hasArg(OPT_headerpad_max_install_names);
1648fe6060f1SDimitry Andric   config->printDylibSearch =
1649fe6060f1SDimitry Andric       args.hasArg(OPT_print_dylib_search) || getenv("RC_TRACE_DYLIB_SEARCHING");
1650e8d8bef9SDimitry Andric   config->printEachFile = args.hasArg(OPT_t);
1651e8d8bef9SDimitry Andric   config->printWhyLoad = args.hasArg(OPT_why_load);
1652349cc55cSDimitry Andric   config->omitDebugInfo = args.hasArg(OPT_S);
1653349cc55cSDimitry Andric   config->errorForArchMismatch = args.hasArg(OPT_arch_errors_fatal);
1654fe6060f1SDimitry Andric   if (const Arg *arg = args.getLastArg(OPT_bundle_loader)) {
1655fe6060f1SDimitry Andric     if (config->outputType != MH_BUNDLE)
1656fe6060f1SDimitry Andric       error("-bundle_loader can only be used with MachO bundle output");
1657fcaf7f86SDimitry Andric     addFile(arg->getValue(), LoadType::CommandLine, /*isLazy=*/false,
1658fcaf7f86SDimitry Andric             /*isExplicit=*/false, /*isBundleLoader=*/true);
1659fe6060f1SDimitry Andric   }
1660bdd1243dSDimitry Andric   for (auto *arg : args.filtered(OPT_dyld_env)) {
1661bdd1243dSDimitry Andric     StringRef envPair(arg->getValue());
1662bdd1243dSDimitry Andric     if (!envPair.contains('='))
1663bdd1243dSDimitry Andric       error("-dyld_env's argument is  malformed. Expected "
1664bdd1243dSDimitry Andric             "-dyld_env <ENV_VAR>=<VALUE>, got `" +
1665bdd1243dSDimitry Andric             envPair + "`");
1666bdd1243dSDimitry Andric     config->dyldEnvs.push_back(envPair);
1667bdd1243dSDimitry Andric   }
1668bdd1243dSDimitry Andric   if (!config->dyldEnvs.empty() && config->outputType != MH_EXECUTE)
1669bdd1243dSDimitry Andric     error("-dyld_env can only be used when creating executable output");
1670bdd1243dSDimitry Andric 
1671fe6060f1SDimitry Andric   if (const Arg *arg = args.getLastArg(OPT_umbrella)) {
1672fe6060f1SDimitry Andric     if (config->outputType != MH_DYLIB)
1673fe6060f1SDimitry Andric       warn("-umbrella used, but not creating dylib");
1674fe6060f1SDimitry Andric     config->umbrella = arg->getValue();
1675fe6060f1SDimitry Andric   }
1676e8d8bef9SDimitry Andric   config->ltoObjPath = args.getLastArgValue(OPT_object_path_lto);
1677fe6060f1SDimitry Andric   config->thinLTOCacheDir = args.getLastArgValue(OPT_cache_path_lto);
1678fe6060f1SDimitry Andric   config->thinLTOCachePolicy = getLTOCachePolicy(args);
1679bdd1243dSDimitry Andric   config->thinLTOEmitImportsFiles = args.hasArg(OPT_thinlto_emit_imports_files);
1680bdd1243dSDimitry Andric   config->thinLTOEmitIndexFiles = args.hasArg(OPT_thinlto_emit_index_files) ||
1681bdd1243dSDimitry Andric                                   args.hasArg(OPT_thinlto_index_only) ||
1682bdd1243dSDimitry Andric                                   args.hasArg(OPT_thinlto_index_only_eq);
1683bdd1243dSDimitry Andric   config->thinLTOIndexOnly = args.hasArg(OPT_thinlto_index_only) ||
1684bdd1243dSDimitry Andric                              args.hasArg(OPT_thinlto_index_only_eq);
1685bdd1243dSDimitry Andric   config->thinLTOIndexOnlyArg = args.getLastArgValue(OPT_thinlto_index_only_eq);
1686bdd1243dSDimitry Andric   config->thinLTOObjectSuffixReplace =
1687bdd1243dSDimitry Andric       getOldNewOptions(args, OPT_thinlto_object_suffix_replace_eq);
168806c3fb27SDimitry Andric   std::tie(config->thinLTOPrefixReplaceOld, config->thinLTOPrefixReplaceNew,
168906c3fb27SDimitry Andric            config->thinLTOPrefixReplaceNativeObject) =
169006c3fb27SDimitry Andric       getOldNewOptionsExtra(args, OPT_thinlto_prefix_replace_eq);
1691bdd1243dSDimitry Andric   if (config->thinLTOEmitIndexFiles && !config->thinLTOIndexOnly) {
1692bdd1243dSDimitry Andric     if (args.hasArg(OPT_thinlto_object_suffix_replace_eq))
1693bdd1243dSDimitry Andric       error("--thinlto-object-suffix-replace is not supported with "
1694bdd1243dSDimitry Andric             "--thinlto-emit-index-files");
1695bdd1243dSDimitry Andric     else if (args.hasArg(OPT_thinlto_prefix_replace_eq))
1696bdd1243dSDimitry Andric       error("--thinlto-prefix-replace is not supported with "
1697bdd1243dSDimitry Andric             "--thinlto-emit-index-files");
1698bdd1243dSDimitry Andric   }
169906c3fb27SDimitry Andric   if (!config->thinLTOPrefixReplaceNativeObject.empty() &&
170006c3fb27SDimitry Andric       config->thinLTOIndexOnlyArg.empty()) {
170106c3fb27SDimitry Andric     error("--thinlto-prefix-replace=old_dir;new_dir;obj_dir must be used with "
170206c3fb27SDimitry Andric           "--thinlto-index-only=");
170306c3fb27SDimitry Andric   }
1704*0fca6ea1SDimitry Andric   config->warnDuplicateRpath =
1705*0fca6ea1SDimitry Andric       args.hasFlag(OPT_warn_duplicate_rpath, OPT_no_warn_duplicate_rpath, true);
1706*0fca6ea1SDimitry Andric   config->runtimePaths = getRuntimePaths(args);
170704eeddc0SDimitry Andric   config->allLoad = args.hasFlag(OPT_all_load, OPT_noall_load, false);
1708fe6060f1SDimitry Andric   config->archMultiple = args.hasArg(OPT_arch_multiple);
1709fe6060f1SDimitry Andric   config->applicationExtension = args.hasFlag(
1710fe6060f1SDimitry Andric       OPT_application_extension, OPT_no_application_extension, false);
1711fe6060f1SDimitry Andric   config->exportDynamic = args.hasArg(OPT_export_dynamic);
1712e8d8bef9SDimitry Andric   config->forceLoadObjC = args.hasArg(OPT_ObjC);
1713fe6060f1SDimitry Andric   config->forceLoadSwift = args.hasArg(OPT_force_load_swift_libs);
1714fe6060f1SDimitry Andric   config->deadStripDylibs = args.hasArg(OPT_dead_strip_dylibs);
1715e8d8bef9SDimitry Andric   config->demangle = args.hasArg(OPT_demangle);
1716e8d8bef9SDimitry Andric   config->implicitDylibs = !args.hasArg(OPT_no_implicit_dylibs);
1717fe6060f1SDimitry Andric   config->emitFunctionStarts =
1718fe6060f1SDimitry Andric       args.hasFlag(OPT_function_starts, OPT_no_function_starts, true);
1719fe6060f1SDimitry Andric   config->emitDataInCodeInfo =
1720fe6060f1SDimitry Andric       args.hasFlag(OPT_data_in_code_info, OPT_no_data_in_code_info, true);
1721bdd1243dSDimitry Andric   config->emitChainedFixups = shouldEmitChainedFixups(args);
1722bdd1243dSDimitry Andric   config->emitInitOffsets =
1723bdd1243dSDimitry Andric       config->emitChainedFixups || args.hasArg(OPT_init_offsets);
1724*0fca6ea1SDimitry Andric   config->emitRelativeMethodLists = shouldEmitRelativeMethodLists(args);
1725fe6060f1SDimitry Andric   config->icfLevel = getICFLevel(args);
1726*0fca6ea1SDimitry Andric   config->keepICFStabs = args.hasArg(OPT_keep_icf_stabs);
1727bdd1243dSDimitry Andric   config->dedupStrings =
1728bdd1243dSDimitry Andric       args.hasFlag(OPT_deduplicate_strings, OPT_no_deduplicate_strings, true);
1729bdd1243dSDimitry Andric   config->deadStripDuplicates = args.hasArg(OPT_dead_strip_duplicates);
1730349cc55cSDimitry Andric   config->warnDylibInstallName = args.hasFlag(
1731349cc55cSDimitry Andric       OPT_warn_dylib_install_name, OPT_no_warn_dylib_install_name, false);
173281ad6265SDimitry Andric   config->ignoreOptimizationHints = args.hasArg(OPT_ignore_optimization_hints);
173304eeddc0SDimitry Andric   config->callGraphProfileSort = args.hasFlag(
173404eeddc0SDimitry Andric       OPT_call_graph_profile_sort, OPT_no_call_graph_profile_sort, true);
1735bdd1243dSDimitry Andric   config->printSymbolOrder = args.getLastArgValue(OPT_print_symbol_order_eq);
1736bdd1243dSDimitry Andric   config->forceExactCpuSubtypeMatch =
1737bdd1243dSDimitry Andric       getenv("LD_DYLIB_CPU_SUBTYPES_MUST_MATCH");
1738bdd1243dSDimitry Andric   config->objcStubsMode = getObjCStubsMode(args);
1739bdd1243dSDimitry Andric   config->ignoreAutoLink = args.hasArg(OPT_ignore_auto_link);
1740bdd1243dSDimitry Andric   for (const Arg *arg : args.filtered(OPT_ignore_auto_link_option))
1741bdd1243dSDimitry Andric     config->ignoreAutoLinkOptions.insert(arg->getValue());
1742bdd1243dSDimitry Andric   config->strictAutoLink = args.hasArg(OPT_strict_auto_link);
174306c3fb27SDimitry Andric   config->ltoDebugPassManager = args.hasArg(OPT_lto_debug_pass_manager);
174406c3fb27SDimitry Andric   config->csProfileGenerate = args.hasArg(OPT_cs_profile_generate);
174506c3fb27SDimitry Andric   config->csProfilePath = args.getLastArgValue(OPT_cs_profile_path);
17465f757f3fSDimitry Andric   config->pgoWarnMismatch =
17475f757f3fSDimitry Andric       args.hasFlag(OPT_pgo_warn_mismatch, OPT_no_pgo_warn_mismatch, true);
1748*0fca6ea1SDimitry Andric   config->warnThinArchiveMissingMembers =
1749*0fca6ea1SDimitry Andric       args.hasFlag(OPT_warn_thin_archive_missing_members,
1750*0fca6ea1SDimitry Andric                    OPT_no_warn_thin_archive_missing_members, true);
175106c3fb27SDimitry Andric   config->generateUuid = !args.hasArg(OPT_no_uuid);
1752e8d8bef9SDimitry Andric 
1753fcaf7f86SDimitry Andric   for (const Arg *arg : args.filtered(OPT_alias)) {
1754fcaf7f86SDimitry Andric     config->aliasedSymbols.push_back(
1755fcaf7f86SDimitry Andric         std::make_pair(arg->getValue(0), arg->getValue(1)));
1756fcaf7f86SDimitry Andric   }
1757fcaf7f86SDimitry Andric 
1758bdd1243dSDimitry Andric   if (const char *zero = getenv("ZERO_AR_DATE"))
1759bdd1243dSDimitry Andric     config->zeroModTime = strcmp(zero, "0") != 0;
176006c3fb27SDimitry Andric   if (args.getLastArg(OPT_reproducible))
176106c3fb27SDimitry Andric     config->zeroModTime = true;
1762fe6060f1SDimitry Andric 
1763*0fca6ea1SDimitry Andric   std::array<PlatformType, 4> encryptablePlatforms{
1764*0fca6ea1SDimitry Andric       PLATFORM_IOS, PLATFORM_WATCHOS, PLATFORM_TVOS, PLATFORM_XROS};
1765fe6060f1SDimitry Andric   config->emitEncryptionInfo =
1766fe6060f1SDimitry Andric       args.hasFlag(OPT_encryptable, OPT_no_encryption,
1767fe6060f1SDimitry Andric                    is_contained(encryptablePlatforms, config->platform()));
1768fe6060f1SDimitry Andric 
1769fe6060f1SDimitry Andric   if (const Arg *arg = args.getLastArg(OPT_install_name)) {
1770349cc55cSDimitry Andric     if (config->warnDylibInstallName && config->outputType != MH_DYLIB)
1771349cc55cSDimitry Andric       warn(
1772349cc55cSDimitry Andric           arg->getAsString(args) +
1773349cc55cSDimitry Andric           ": ignored, only has effect with -dylib [--warn-dylib-install-name]");
1774fe6060f1SDimitry Andric     else
1775fe6060f1SDimitry Andric       config->installName = arg->getValue();
1776fe6060f1SDimitry Andric   } else if (config->outputType == MH_DYLIB) {
1777fe6060f1SDimitry Andric     config->installName = config->finalOutput;
1778fe6060f1SDimitry Andric   }
1779fe6060f1SDimitry Andric 
1780fe6060f1SDimitry Andric   if (args.hasArg(OPT_mark_dead_strippable_dylib)) {
1781fe6060f1SDimitry Andric     if (config->outputType != MH_DYLIB)
1782fe6060f1SDimitry Andric       warn("-mark_dead_strippable_dylib: ignored, only has effect with -dylib");
1783fe6060f1SDimitry Andric     else
1784fe6060f1SDimitry Andric       config->markDeadStrippableDylib = true;
1785fe6060f1SDimitry Andric   }
1786fe6060f1SDimitry Andric 
1787fe6060f1SDimitry Andric   if (const Arg *arg = args.getLastArg(OPT_static, OPT_dynamic))
1788e8d8bef9SDimitry Andric     config->staticLink = (arg->getOption().getID() == OPT_static);
1789e8d8bef9SDimitry Andric 
1790fe6060f1SDimitry Andric   if (const Arg *arg =
1791fe6060f1SDimitry Andric           args.getLastArg(OPT_flat_namespace, OPT_twolevel_namespace))
1792fe6060f1SDimitry Andric     config->namespaceKind = arg->getOption().getID() == OPT_twolevel_namespace
1793fe6060f1SDimitry Andric                                 ? NamespaceKind::twolevel
1794fe6060f1SDimitry Andric                                 : NamespaceKind::flat;
1795fe6060f1SDimitry Andric 
1796fe6060f1SDimitry Andric   config->undefinedSymbolTreatment = getUndefinedSymbolTreatment(args);
1797fe6060f1SDimitry Andric 
1798fe6060f1SDimitry Andric   if (config->outputType == MH_EXECUTE)
1799fe6060f1SDimitry Andric     config->entry = symtab->addUndefined(args.getLastArgValue(OPT_e, "_main"),
1800fe6060f1SDimitry Andric                                          /*file=*/nullptr,
1801fe6060f1SDimitry Andric                                          /*isWeakRef=*/false);
1802fe6060f1SDimitry Andric 
1803e8d8bef9SDimitry Andric   config->librarySearchPaths =
1804e8d8bef9SDimitry Andric       getLibrarySearchPaths(args, config->systemLibraryRoots);
1805e8d8bef9SDimitry Andric   config->frameworkSearchPaths =
1806e8d8bef9SDimitry Andric       getFrameworkSearchPaths(args, config->systemLibraryRoots);
1807fe6060f1SDimitry Andric   if (const Arg *arg =
1808e8d8bef9SDimitry Andric           args.getLastArg(OPT_search_paths_first, OPT_search_dylibs_first))
1809e8d8bef9SDimitry Andric     config->searchDylibsFirst =
1810e8d8bef9SDimitry Andric         arg->getOption().getID() == OPT_search_dylibs_first;
1811e8d8bef9SDimitry Andric 
1812e8d8bef9SDimitry Andric   config->dylibCompatibilityVersion =
1813e8d8bef9SDimitry Andric       parseDylibVersion(args, OPT_compatibility_version);
1814e8d8bef9SDimitry Andric   config->dylibCurrentVersion = parseDylibVersion(args, OPT_current_version);
1815e8d8bef9SDimitry Andric 
1816fe6060f1SDimitry Andric   config->dataConst =
1817fe6060f1SDimitry Andric       args.hasFlag(OPT_data_const, OPT_no_data_const, dataConstDefault(args));
1818fe6060f1SDimitry Andric   // Populate config->sectionRenameMap with builtin default renames.
1819fe6060f1SDimitry Andric   // Options -rename_section and -rename_segment are able to override.
1820fe6060f1SDimitry Andric   initializeSectionRenameMap();
1821fe6060f1SDimitry Andric   // Reject every special character except '.' and '$'
1822fe6060f1SDimitry Andric   // TODO(gkm): verify that this is the proper set of invalid chars
1823fe6060f1SDimitry Andric   StringRef invalidNameChars("!\"#%&'()*+,-/:;<=>?@[\\]^`{|}~");
1824fe6060f1SDimitry Andric   auto validName = [invalidNameChars](StringRef s) {
1825fe6060f1SDimitry Andric     if (s.find_first_of(invalidNameChars) != StringRef::npos)
1826fe6060f1SDimitry Andric       error("invalid name for segment or section: " + s);
1827fe6060f1SDimitry Andric     return s;
1828fe6060f1SDimitry Andric   };
1829fe6060f1SDimitry Andric   for (const Arg *arg : args.filtered(OPT_rename_section)) {
1830fe6060f1SDimitry Andric     config->sectionRenameMap[{validName(arg->getValue(0)),
1831fe6060f1SDimitry Andric                               validName(arg->getValue(1))}] = {
1832fe6060f1SDimitry Andric         validName(arg->getValue(2)), validName(arg->getValue(3))};
1833fe6060f1SDimitry Andric   }
1834fe6060f1SDimitry Andric   for (const Arg *arg : args.filtered(OPT_rename_segment)) {
1835fe6060f1SDimitry Andric     config->segmentRenameMap[validName(arg->getValue(0))] =
1836fe6060f1SDimitry Andric         validName(arg->getValue(1));
1837fe6060f1SDimitry Andric   }
1838fe6060f1SDimitry Andric 
1839fe6060f1SDimitry Andric   config->sectionAlignments = parseSectAlign(args);
1840fe6060f1SDimitry Andric 
1841fe6060f1SDimitry Andric   for (const Arg *arg : args.filtered(OPT_segprot)) {
1842fe6060f1SDimitry Andric     StringRef segName = arg->getValue(0);
1843fe6060f1SDimitry Andric     uint32_t maxProt = parseProtection(arg->getValue(1));
1844fe6060f1SDimitry Andric     uint32_t initProt = parseProtection(arg->getValue(2));
1845fe6060f1SDimitry Andric     if (maxProt != initProt && config->arch() != AK_i386)
1846fe6060f1SDimitry Andric       error("invalid argument '" + arg->getAsString(args) +
1847fe6060f1SDimitry Andric             "': max and init must be the same for non-i386 archs");
1848fe6060f1SDimitry Andric     if (segName == segment_names::linkEdit)
1849fe6060f1SDimitry Andric       error("-segprot cannot be used to change __LINKEDIT's protections");
1850fe6060f1SDimitry Andric     config->segmentProtections.push_back({segName, maxProt, initProt});
1851fe6060f1SDimitry Andric   }
1852fe6060f1SDimitry Andric 
185381ad6265SDimitry Andric   config->hasExplicitExports =
185481ad6265SDimitry Andric       args.hasArg(OPT_no_exported_symbols) ||
185581ad6265SDimitry Andric       args.hasArgNoClaim(OPT_exported_symbol, OPT_exported_symbols_list);
1856fe6060f1SDimitry Andric   handleSymbolPatterns(args, config->exportedSymbols, OPT_exported_symbol,
1857fe6060f1SDimitry Andric                        OPT_exported_symbols_list);
1858fe6060f1SDimitry Andric   handleSymbolPatterns(args, config->unexportedSymbols, OPT_unexported_symbol,
1859fe6060f1SDimitry Andric                        OPT_unexported_symbols_list);
186081ad6265SDimitry Andric   if (config->hasExplicitExports && !config->unexportedSymbols.empty())
186181ad6265SDimitry Andric     error("cannot use both -exported_symbol* and -unexported_symbol* options");
186281ad6265SDimitry Andric 
186381ad6265SDimitry Andric   if (args.hasArg(OPT_no_exported_symbols) && !config->exportedSymbols.empty())
186481ad6265SDimitry Andric     error("cannot use both -exported_symbol* and -no_exported_symbols options");
186581ad6265SDimitry Andric 
186681ad6265SDimitry Andric   // Imitating LD64's:
186781ad6265SDimitry Andric   // -non_global_symbols_no_strip_list and -non_global_symbols_strip_list can't
186881ad6265SDimitry Andric   // both be present.
186981ad6265SDimitry Andric   // But -x can be used with either of these two, in which case, the last arg
187081ad6265SDimitry Andric   // takes effect.
187181ad6265SDimitry Andric   // (TODO: This is kind of confusing - considering disallowing using them
187281ad6265SDimitry Andric   // together for a more straightforward behaviour)
187381ad6265SDimitry Andric   {
187481ad6265SDimitry Andric     bool includeLocal = false;
187581ad6265SDimitry Andric     bool excludeLocal = false;
187681ad6265SDimitry Andric     for (const Arg *arg :
187781ad6265SDimitry Andric          args.filtered(OPT_x, OPT_non_global_symbols_no_strip_list,
187881ad6265SDimitry Andric                        OPT_non_global_symbols_strip_list)) {
187981ad6265SDimitry Andric       switch (arg->getOption().getID()) {
188081ad6265SDimitry Andric       case OPT_x:
188181ad6265SDimitry Andric         config->localSymbolsPresence = SymtabPresence::None;
188281ad6265SDimitry Andric         break;
188381ad6265SDimitry Andric       case OPT_non_global_symbols_no_strip_list:
188481ad6265SDimitry Andric         if (excludeLocal) {
188581ad6265SDimitry Andric           error("cannot use both -non_global_symbols_no_strip_list and "
188681ad6265SDimitry Andric                 "-non_global_symbols_strip_list");
188781ad6265SDimitry Andric         } else {
188881ad6265SDimitry Andric           includeLocal = true;
188981ad6265SDimitry Andric           config->localSymbolsPresence = SymtabPresence::SelectivelyIncluded;
189081ad6265SDimitry Andric           parseSymbolPatternsFile(arg, config->localSymbolPatterns);
189181ad6265SDimitry Andric         }
189281ad6265SDimitry Andric         break;
189381ad6265SDimitry Andric       case OPT_non_global_symbols_strip_list:
189481ad6265SDimitry Andric         if (includeLocal) {
189581ad6265SDimitry Andric           error("cannot use both -non_global_symbols_no_strip_list and "
189681ad6265SDimitry Andric                 "-non_global_symbols_strip_list");
189781ad6265SDimitry Andric         } else {
189881ad6265SDimitry Andric           excludeLocal = true;
189981ad6265SDimitry Andric           config->localSymbolsPresence = SymtabPresence::SelectivelyExcluded;
190081ad6265SDimitry Andric           parseSymbolPatternsFile(arg, config->localSymbolPatterns);
190181ad6265SDimitry Andric         }
190281ad6265SDimitry Andric         break;
190381ad6265SDimitry Andric       default:
190481ad6265SDimitry Andric         llvm_unreachable("unexpected option");
190581ad6265SDimitry Andric       }
190681ad6265SDimitry Andric     }
1907fe6060f1SDimitry Andric   }
1908fe6060f1SDimitry Andric   // Explicitly-exported literal symbols must be defined, but might
190981ad6265SDimitry Andric   // languish in an archive if unreferenced elsewhere or if they are in the
191081ad6265SDimitry Andric   // non-global strip list. Light a fire under those lazy symbols!
1911fe6060f1SDimitry Andric   for (const CachedHashStringRef &cachedName : config->exportedSymbols.literals)
1912fe6060f1SDimitry Andric     symtab->addUndefined(cachedName.val(), /*file=*/nullptr,
1913fe6060f1SDimitry Andric                          /*isWeakRef=*/false);
1914fe6060f1SDimitry Andric 
191581ad6265SDimitry Andric   for (const Arg *arg : args.filtered(OPT_why_live))
191681ad6265SDimitry Andric     config->whyLive.insert(arg->getValue());
191781ad6265SDimitry Andric   if (!config->whyLive.empty() && !config->deadStrip)
191881ad6265SDimitry Andric     warn("-why_live has no effect without -dead_strip, ignoring");
191981ad6265SDimitry Andric 
1920e8d8bef9SDimitry Andric   config->saveTemps = args.hasArg(OPT_save_temps);
19215ffd83dbSDimitry Andric 
1922fe6060f1SDimitry Andric   config->adhocCodesign = args.hasFlag(
1923fe6060f1SDimitry Andric       OPT_adhoc_codesign, OPT_no_adhoc_codesign,
1924bdd1243dSDimitry Andric       shouldAdhocSignByDefault(config->arch(), config->platform()));
1925fe6060f1SDimitry Andric 
19265ffd83dbSDimitry Andric   if (args.hasArg(OPT_v)) {
1927349cc55cSDimitry Andric     message(getLLDVersion(), lld::errs());
19285ffd83dbSDimitry Andric     message(StringRef("Library search paths:") +
1929fe6060f1SDimitry Andric                 (config->librarySearchPaths.empty()
1930fe6060f1SDimitry Andric                      ? ""
1931349cc55cSDimitry Andric                      : "\n\t" + join(config->librarySearchPaths, "\n\t")),
1932349cc55cSDimitry Andric             lld::errs());
19335ffd83dbSDimitry Andric     message(StringRef("Framework search paths:") +
1934fe6060f1SDimitry Andric                 (config->frameworkSearchPaths.empty()
1935fe6060f1SDimitry Andric                      ? ""
1936349cc55cSDimitry Andric                      : "\n\t" + join(config->frameworkSearchPaths, "\n\t")),
1937349cc55cSDimitry Andric             lld::errs());
19385ffd83dbSDimitry Andric   }
19395ffd83dbSDimitry Andric 
1940fe6060f1SDimitry Andric   config->progName = argsArr[0];
1941fe6060f1SDimitry Andric 
194281ad6265SDimitry Andric   config->timeTraceEnabled = args.hasArg(OPT_time_trace_eq);
1943fe6060f1SDimitry Andric   config->timeTraceGranularity =
1944fe6060f1SDimitry Andric       args::getInteger(args, OPT_time_trace_granularity_eq, 500);
1945fe6060f1SDimitry Andric 
1946fe6060f1SDimitry Andric   // Initialize time trace profiler.
1947fe6060f1SDimitry Andric   if (config->timeTraceEnabled)
1948fe6060f1SDimitry Andric     timeTraceProfilerInitialize(config->timeTraceGranularity, config->progName);
1949fe6060f1SDimitry Andric 
1950fe6060f1SDimitry Andric   {
1951fe6060f1SDimitry Andric     TimeTraceScope timeScope("ExecuteLinker");
1952fe6060f1SDimitry Andric 
1953e8d8bef9SDimitry Andric     initLLVM(); // must be run before any call to addFile()
1954fe6060f1SDimitry Andric     createFiles(args);
19555ffd83dbSDimitry Andric 
19565ffd83dbSDimitry Andric     // Now that all dylibs have been loaded, search for those that should be
19575ffd83dbSDimitry Andric     // re-exported.
1958fe6060f1SDimitry Andric     {
1959fe6060f1SDimitry Andric       auto reexportHandler = [](const Arg *arg,
1960fe6060f1SDimitry Andric                                 const std::vector<StringRef> &extensions) {
19615ffd83dbSDimitry Andric         config->hasReexports = true;
19625ffd83dbSDimitry Andric         StringRef searchName = arg->getValue();
1963e8d8bef9SDimitry Andric         if (!markReexport(searchName, extensions))
1964e8d8bef9SDimitry Andric           error(arg->getSpelling() + " " + searchName +
1965e8d8bef9SDimitry Andric                 " does not match a supplied dylib");
1966fe6060f1SDimitry Andric       };
1967fe6060f1SDimitry Andric       std::vector<StringRef> extensions = {".tbd"};
1968fe6060f1SDimitry Andric       for (const Arg *arg : args.filtered(OPT_sub_umbrella))
1969fe6060f1SDimitry Andric         reexportHandler(arg, extensions);
1970fe6060f1SDimitry Andric 
1971fe6060f1SDimitry Andric       extensions.push_back(".dylib");
1972fe6060f1SDimitry Andric       for (const Arg *arg : args.filtered(OPT_sub_library))
1973fe6060f1SDimitry Andric         reexportHandler(arg, extensions);
19745ffd83dbSDimitry Andric     }
19755ffd83dbSDimitry Andric 
1976349cc55cSDimitry Andric     cl::ResetAllOptionOccurrences();
1977349cc55cSDimitry Andric 
1978e8d8bef9SDimitry Andric     // Parse LTO options.
1979fe6060f1SDimitry Andric     if (const Arg *arg = args.getLastArg(OPT_mcpu))
198004eeddc0SDimitry Andric       parseClangOption(saver().save("-mcpu=" + StringRef(arg->getValue())),
1981e8d8bef9SDimitry Andric                        arg->getSpelling());
1982e8d8bef9SDimitry Andric 
1983bdd1243dSDimitry Andric     for (const Arg *arg : args.filtered(OPT_mllvm)) {
1984e8d8bef9SDimitry Andric       parseClangOption(arg->getValue(), arg->getSpelling());
1985bdd1243dSDimitry Andric       config->mllvmOpts.emplace_back(arg->getValue());
1986bdd1243dSDimitry Andric     }
1987e8d8bef9SDimitry Andric 
1988bdd1243dSDimitry Andric     createSyntheticSections();
1989bdd1243dSDimitry Andric     createSyntheticSymbols();
1990bdd1243dSDimitry Andric     addSynthenticMethnames();
1991bdd1243dSDimitry Andric 
1992bdd1243dSDimitry Andric     createAliases();
1993bdd1243dSDimitry Andric     // If we are in "explicit exports" mode, hide everything that isn't
1994bdd1243dSDimitry Andric     // explicitly exported. Do this before running LTO so that LTO can better
1995bdd1243dSDimitry Andric     // optimize.
1996bdd1243dSDimitry Andric     handleExplicitExports();
1997bdd1243dSDimitry Andric 
1998bdd1243dSDimitry Andric     bool didCompileBitcodeFiles = compileBitcodeFiles();
1999bdd1243dSDimitry Andric 
20005f757f3fSDimitry Andric     resolveLCLinkerOptions();
20015f757f3fSDimitry Andric 
2002bdd1243dSDimitry Andric     // If --thinlto-index-only is given, we should create only "index
2003bdd1243dSDimitry Andric     // files" and not object files. Index file creation is already done
2004bdd1243dSDimitry Andric     // in compileBitcodeFiles, so we are done if that's the case.
2005bdd1243dSDimitry Andric     if (config->thinLTOIndexOnly)
2006bdd1243dSDimitry Andric       return errorCount() == 0;
2007bdd1243dSDimitry Andric 
2008bdd1243dSDimitry Andric     // LTO may emit a non-hidden (extern) object file symbol even if the
2009bdd1243dSDimitry Andric     // corresponding bitcode symbol is hidden. In particular, this happens for
2010bdd1243dSDimitry Andric     // cross-module references to hidden symbols under ThinLTO. Thus, if we
2011bdd1243dSDimitry Andric     // compiled any bitcode files, we must redo the symbol hiding.
2012bdd1243dSDimitry Andric     if (didCompileBitcodeFiles)
2013bdd1243dSDimitry Andric       handleExplicitExports();
2014e8d8bef9SDimitry Andric     replaceCommonSymbols();
2015e8d8bef9SDimitry Andric 
20165ffd83dbSDimitry Andric     StringRef orderFile = args.getLastArgValue(OPT_order_file);
201781ad6265SDimitry Andric     if (!orderFile.empty())
201881ad6265SDimitry Andric       priorityBuilder.parseOrderFile(orderFile);
20195ffd83dbSDimitry Andric 
2020fe6060f1SDimitry Andric     referenceStubBinder();
2021fe6060f1SDimitry Andric 
2022fe6060f1SDimitry Andric     // FIXME: should terminate the link early based on errors encountered so
2023fe6060f1SDimitry Andric     // far?
20245ffd83dbSDimitry Andric 
2025fe6060f1SDimitry Andric     for (const Arg *arg : args.filtered(OPT_sectcreate)) {
2026e8d8bef9SDimitry Andric       StringRef segName = arg->getValue(0);
2027e8d8bef9SDimitry Andric       StringRef sectName = arg->getValue(1);
2028e8d8bef9SDimitry Andric       StringRef fileName = arg->getValue(2);
2029bdd1243dSDimitry Andric       std::optional<MemoryBufferRef> buffer = readFile(fileName);
2030e8d8bef9SDimitry Andric       if (buffer)
2031e8d8bef9SDimitry Andric         inputFiles.insert(make<OpaqueFile>(*buffer, segName, sectName));
2032e8d8bef9SDimitry Andric     }
20335ffd83dbSDimitry Andric 
20341fd87a68SDimitry Andric     for (const Arg *arg : args.filtered(OPT_add_empty_section)) {
20351fd87a68SDimitry Andric       StringRef segName = arg->getValue(0);
20361fd87a68SDimitry Andric       StringRef sectName = arg->getValue(1);
20371fd87a68SDimitry Andric       inputFiles.insert(make<OpaqueFile>(MemoryBufferRef(), segName, sectName));
20381fd87a68SDimitry Andric     }
20391fd87a68SDimitry Andric 
2040fe6060f1SDimitry Andric     gatherInputSections();
204104eeddc0SDimitry Andric     if (config->callGraphProfileSort)
204281ad6265SDimitry Andric       priorityBuilder.extractCallGraphProfile();
2043fe6060f1SDimitry Andric 
2044fe6060f1SDimitry Andric     if (config->deadStrip)
2045fe6060f1SDimitry Andric       markLive();
2046fe6060f1SDimitry Andric 
2047*0fca6ea1SDimitry Andric     // Ensure that no symbols point inside __mod_init_func sections if they are
2048*0fca6ea1SDimitry Andric     // removed due to -init_offsets. This must run after dead stripping.
2049*0fca6ea1SDimitry Andric     if (config->emitInitOffsets)
2050*0fca6ea1SDimitry Andric       eraseInitializerSymbols();
2051*0fca6ea1SDimitry Andric 
2052*0fca6ea1SDimitry Andric     // Categories are not subject to dead-strip. The __objc_catlist section is
2053*0fca6ea1SDimitry Andric     // marked as NO_DEAD_STRIP and that propagates into all category data.
205406c3fb27SDimitry Andric     if (args.hasArg(OPT_check_category_conflicts))
205506c3fb27SDimitry Andric       objc::checkCategories();
205606c3fb27SDimitry Andric 
2057*0fca6ea1SDimitry Andric     // Category merging uses "->live = false" to erase old category data, so
2058*0fca6ea1SDimitry Andric     // it has to run after dead-stripping (markLive).
2059*0fca6ea1SDimitry Andric     if (args.hasFlag(OPT_objc_category_merging, OPT_no_objc_category_merging,
2060*0fca6ea1SDimitry Andric                      false))
2061*0fca6ea1SDimitry Andric       objc::mergeCategories();
2062*0fca6ea1SDimitry Andric 
2063fe6060f1SDimitry Andric     // ICF assumes that all literals have been folded already, so we must run
2064fe6060f1SDimitry Andric     // foldIdenticalLiterals before foldIdenticalSections.
2065fe6060f1SDimitry Andric     foldIdenticalLiterals();
206681ad6265SDimitry Andric     if (config->icfLevel != ICFLevel::none) {
206781ad6265SDimitry Andric       if (config->icfLevel == ICFLevel::safe)
206881ad6265SDimitry Andric         markAddrSigSymbols();
2069fcaf7f86SDimitry Andric       foldIdenticalSections(/*onlyCfStrings=*/false);
2070bdd1243dSDimitry Andric     } else if (config->dedupStrings) {
2071fcaf7f86SDimitry Andric       foldIdenticalSections(/*onlyCfStrings=*/true);
207281ad6265SDimitry Andric     }
20735ffd83dbSDimitry Andric 
20745ffd83dbSDimitry Andric     // Write to an output file.
2075fe6060f1SDimitry Andric     if (target->wordSize == 8)
2076fe6060f1SDimitry Andric       writeResult<LP64>();
2077fe6060f1SDimitry Andric     else
2078fe6060f1SDimitry Andric       writeResult<ILP32>();
2079fe6060f1SDimitry Andric 
2080fe6060f1SDimitry Andric     depTracker->write(getLLDVersion(), inputFiles, config->outputFile);
2081fe6060f1SDimitry Andric   }
2082fe6060f1SDimitry Andric 
2083fe6060f1SDimitry Andric   if (config->timeTraceEnabled) {
2084349cc55cSDimitry Andric     checkError(timeTraceProfilerWrite(
208581ad6265SDimitry Andric         args.getLastArgValue(OPT_time_trace_eq).str(), config->outputFile));
2086fe6060f1SDimitry Andric 
2087fe6060f1SDimitry Andric     timeTraceProfilerCleanup();
2088fe6060f1SDimitry Andric   }
2089bdd1243dSDimitry Andric 
2090bdd1243dSDimitry Andric   if (errorCount() != 0 || config->strictAutoLink)
2091bdd1243dSDimitry Andric     for (const auto &warning : missingAutolinkWarnings)
2092bdd1243dSDimitry Andric       warn(warning);
2093bdd1243dSDimitry Andric 
209404eeddc0SDimitry Andric   return errorCount() == 0;
20955ffd83dbSDimitry Andric }
209606c3fb27SDimitry Andric } // namespace macho
209706c3fb27SDimitry Andric } // namespace lld
2098