xref: /freebsd/contrib/llvm-project/llvm/tools/llvm-nm/llvm-nm.cpp (revision bdd1243df58e60e85101c09001d9812a789b6bc4)
10b57cec5SDimitry Andric //===-- llvm-nm.cpp - Symbol table dumping utility for llvm ---------------===//
20b57cec5SDimitry Andric //
30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
60b57cec5SDimitry Andric //
70b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
80b57cec5SDimitry Andric //
90b57cec5SDimitry Andric // This program is a utility that works like traditional Unix "nm", that is, it
100b57cec5SDimitry Andric // prints out the names of symbols in a bitcode or object file, along with some
110b57cec5SDimitry Andric // information about each symbol.
120b57cec5SDimitry Andric //
130b57cec5SDimitry Andric // This "nm" supports many of the features of GNU "nm", including its different
140b57cec5SDimitry Andric // output formats.
150b57cec5SDimitry Andric //
160b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
170b57cec5SDimitry Andric 
180b57cec5SDimitry Andric #include "llvm/ADT/StringSwitch.h"
19*bdd1243dSDimitry Andric #include "llvm/ADT/Triple.h"
200b57cec5SDimitry Andric #include "llvm/BinaryFormat/COFF.h"
2181ad6265SDimitry Andric #include "llvm/BinaryFormat/XCOFF.h"
220b57cec5SDimitry Andric #include "llvm/Demangle/Demangle.h"
230b57cec5SDimitry Andric #include "llvm/IR/Function.h"
240b57cec5SDimitry Andric #include "llvm/IR/LLVMContext.h"
250b57cec5SDimitry Andric #include "llvm/Object/Archive.h"
260b57cec5SDimitry Andric #include "llvm/Object/COFF.h"
270b57cec5SDimitry Andric #include "llvm/Object/COFFImportFile.h"
280b57cec5SDimitry Andric #include "llvm/Object/ELFObjectFile.h"
290b57cec5SDimitry Andric #include "llvm/Object/IRObjectFile.h"
300b57cec5SDimitry Andric #include "llvm/Object/MachO.h"
310b57cec5SDimitry Andric #include "llvm/Object/MachOUniversal.h"
320b57cec5SDimitry Andric #include "llvm/Object/ObjectFile.h"
335ffd83dbSDimitry Andric #include "llvm/Object/TapiFile.h"
345ffd83dbSDimitry Andric #include "llvm/Object/TapiUniversal.h"
350b57cec5SDimitry Andric #include "llvm/Object/Wasm.h"
3604eeddc0SDimitry Andric #include "llvm/Object/XCOFFObjectFile.h"
37fe6060f1SDimitry Andric #include "llvm/Option/Arg.h"
38fe6060f1SDimitry Andric #include "llvm/Option/ArgList.h"
39fe6060f1SDimitry Andric #include "llvm/Option/Option.h"
400b57cec5SDimitry Andric #include "llvm/Support/CommandLine.h"
410b57cec5SDimitry Andric #include "llvm/Support/FileSystem.h"
420b57cec5SDimitry Andric #include "llvm/Support/Format.h"
43*bdd1243dSDimitry Andric #include "llvm/Support/Host.h"
440b57cec5SDimitry Andric #include "llvm/Support/InitLLVM.h"
450b57cec5SDimitry Andric #include "llvm/Support/MemoryBuffer.h"
460b57cec5SDimitry Andric #include "llvm/Support/Program.h"
470b57cec5SDimitry Andric #include "llvm/Support/Signals.h"
480b57cec5SDimitry Andric #include "llvm/Support/TargetSelect.h"
490b57cec5SDimitry Andric #include "llvm/Support/WithColor.h"
500b57cec5SDimitry Andric #include "llvm/Support/raw_ostream.h"
510b57cec5SDimitry Andric #include <vector>
520b57cec5SDimitry Andric 
530b57cec5SDimitry Andric using namespace llvm;
540b57cec5SDimitry Andric using namespace object;
550b57cec5SDimitry Andric 
560b57cec5SDimitry Andric namespace {
57fe6060f1SDimitry Andric using namespace llvm::opt; // for HelpHidden in Opts.inc
58fe6060f1SDimitry Andric enum ID {
59fe6060f1SDimitry Andric   OPT_INVALID = 0, // This is not an option ID.
60fe6060f1SDimitry Andric #define OPTION(PREFIX, NAME, ID, KIND, GROUP, ALIAS, ALIASARGS, FLAGS, PARAM,  \
61fe6060f1SDimitry Andric                HELPTEXT, METAVAR, VALUES)                                      \
62fe6060f1SDimitry Andric   OPT_##ID,
63fe6060f1SDimitry Andric #include "Opts.inc"
64fe6060f1SDimitry Andric #undef OPTION
65fe6060f1SDimitry Andric };
660b57cec5SDimitry Andric 
67*bdd1243dSDimitry Andric #define PREFIX(NAME, VALUE)                                                    \
68*bdd1243dSDimitry Andric   static constexpr StringLiteral NAME##_init[] = VALUE;                        \
69*bdd1243dSDimitry Andric   static constexpr ArrayRef<StringLiteral> NAME(NAME##_init,                   \
70*bdd1243dSDimitry Andric                                                 std::size(NAME##_init) - 1);
71fe6060f1SDimitry Andric #include "Opts.inc"
72fe6060f1SDimitry Andric #undef PREFIX
730b57cec5SDimitry Andric 
74*bdd1243dSDimitry Andric static constexpr opt::OptTable::Info InfoTable[] = {
75fe6060f1SDimitry Andric #define OPTION(PREFIX, NAME, ID, KIND, GROUP, ALIAS, ALIASARGS, FLAGS, PARAM,  \
76fe6060f1SDimitry Andric                HELPTEXT, METAVAR, VALUES)                                      \
77fe6060f1SDimitry Andric   {                                                                            \
78fe6060f1SDimitry Andric       PREFIX,      NAME,      HELPTEXT,                                        \
79fe6060f1SDimitry Andric       METAVAR,     OPT_##ID,  opt::Option::KIND##Class,                        \
80fe6060f1SDimitry Andric       PARAM,       FLAGS,     OPT_##GROUP,                                     \
81fe6060f1SDimitry Andric       OPT_##ALIAS, ALIASARGS, VALUES},
82fe6060f1SDimitry Andric #include "Opts.inc"
83fe6060f1SDimitry Andric #undef OPTION
84fe6060f1SDimitry Andric };
850b57cec5SDimitry Andric 
86*bdd1243dSDimitry Andric class NmOptTable : public opt::GenericOptTable {
87fe6060f1SDimitry Andric public:
88*bdd1243dSDimitry Andric   NmOptTable() : opt::GenericOptTable(InfoTable) {
89*bdd1243dSDimitry Andric     setGroupedShortOptions(true);
90*bdd1243dSDimitry Andric   }
91fe6060f1SDimitry Andric };
920b57cec5SDimitry Andric 
93fe6060f1SDimitry Andric enum OutputFormatTy { bsd, sysv, posix, darwin, just_symbols };
9481ad6265SDimitry Andric enum class BitModeTy { Bit32, Bit64, Bit32_64, Any };
95fe6060f1SDimitry Andric } // namespace
960b57cec5SDimitry Andric 
97fe6060f1SDimitry Andric static bool ArchiveMap;
9881ad6265SDimitry Andric static BitModeTy BitMode;
99fe6060f1SDimitry Andric static bool DebugSyms;
100fe6060f1SDimitry Andric static bool DefinedOnly;
101fe6060f1SDimitry Andric static bool Demangle;
102fe6060f1SDimitry Andric static bool DynamicSyms;
10381ad6265SDimitry Andric static bool ExportSymbols;
104fe6060f1SDimitry Andric static bool ExternalOnly;
105fe6060f1SDimitry Andric static OutputFormatTy OutputFormat;
106fe6060f1SDimitry Andric static bool NoLLVMBitcode;
107fe6060f1SDimitry Andric static bool NoSort;
108fe6060f1SDimitry Andric static bool NoWeakSymbols;
109fe6060f1SDimitry Andric static bool NumericSort;
110fe6060f1SDimitry Andric static bool PrintFileName;
111fe6060f1SDimitry Andric static bool PrintSize;
112fe6060f1SDimitry Andric static bool Quiet;
113fe6060f1SDimitry Andric static bool ReverseSort;
114fe6060f1SDimitry Andric static bool SpecialSyms;
115fe6060f1SDimitry Andric static bool SizeSort;
116fe6060f1SDimitry Andric static bool UndefinedOnly;
117fe6060f1SDimitry Andric static bool WithoutAliases;
1180b57cec5SDimitry Andric 
11981ad6265SDimitry Andric // XCOFF-specific options.
12081ad6265SDimitry Andric static bool NoRsrc;
12181ad6265SDimitry Andric 
122fe6060f1SDimitry Andric namespace {
1230b57cec5SDimitry Andric enum Radix { d, o, x };
124fe6060f1SDimitry Andric } // namespace
125fe6060f1SDimitry Andric static Radix AddressRadix;
1260b57cec5SDimitry Andric 
127fe6060f1SDimitry Andric // Mach-O specific options.
128fe6060f1SDimitry Andric static bool ArchAll = false;
129fe6060f1SDimitry Andric static std::vector<StringRef> ArchFlags;
130fe6060f1SDimitry Andric static bool AddDyldInfo;
131fe6060f1SDimitry Andric static bool AddInlinedInfo;
132fe6060f1SDimitry Andric static bool DyldInfoOnly;
133fe6060f1SDimitry Andric static bool FormatMachOasHex;
134fe6060f1SDimitry Andric static bool NoDyldInfo;
135fe6060f1SDimitry Andric static std::vector<StringRef> SegSect;
136fe6060f1SDimitry Andric static bool MachOPrintSizeWarning = false;
1370b57cec5SDimitry Andric 
138fe6060f1SDimitry Andric // Miscellaneous states.
139fe6060f1SDimitry Andric static bool PrintAddress = true;
140fe6060f1SDimitry Andric static bool MultipleFiles = false;
141fe6060f1SDimitry Andric static bool HadError = false;
1420b57cec5SDimitry Andric 
143fe6060f1SDimitry Andric static StringRef ToolName;
1440b57cec5SDimitry Andric 
14581ad6265SDimitry Andric static void warn(Error Err, Twine FileName, Twine Context = Twine(),
14681ad6265SDimitry Andric                  Twine Archive = Twine()) {
14704eeddc0SDimitry Andric   assert(Err);
14804eeddc0SDimitry Andric 
14904eeddc0SDimitry Andric   // Flush the standard output so that the warning isn't interleaved with other
15004eeddc0SDimitry Andric   // output if stdout and stderr are writing to the same place.
15104eeddc0SDimitry Andric   outs().flush();
15204eeddc0SDimitry Andric 
15304eeddc0SDimitry Andric   handleAllErrors(std::move(Err), [&](const ErrorInfoBase &EI) {
15404eeddc0SDimitry Andric     WithColor::warning(errs(), ToolName)
15581ad6265SDimitry Andric         << (Archive.str().empty() ? FileName : Archive + "(" + FileName + ")")
15681ad6265SDimitry Andric         << ": " << (Context.str().empty() ? "" : Context + ": ") << EI.message()
15781ad6265SDimitry Andric         << "\n";
15804eeddc0SDimitry Andric   });
15904eeddc0SDimitry Andric }
16004eeddc0SDimitry Andric 
1610b57cec5SDimitry Andric static void error(Twine Message, Twine Path = Twine()) {
1620b57cec5SDimitry Andric   HadError = true;
163fe6060f1SDimitry Andric   WithColor::error(errs(), ToolName) << Path << ": " << Message << "\n";
1640b57cec5SDimitry Andric }
1650b57cec5SDimitry Andric 
1660b57cec5SDimitry Andric static bool error(std::error_code EC, Twine Path = Twine()) {
1670b57cec5SDimitry Andric   if (EC) {
1680b57cec5SDimitry Andric     error(EC.message(), Path);
1690b57cec5SDimitry Andric     return true;
1700b57cec5SDimitry Andric   }
1710b57cec5SDimitry Andric   return false;
1720b57cec5SDimitry Andric }
1730b57cec5SDimitry Andric 
1740b57cec5SDimitry Andric // This version of error() prints the archive name and member name, for example:
1750b57cec5SDimitry Andric // "libx.a(foo.o)" after the ToolName before the error message.  It sets
1760b57cec5SDimitry Andric // HadError but returns allowing the code to move on to other archive members.
1770b57cec5SDimitry Andric static void error(llvm::Error E, StringRef FileName, const Archive::Child &C,
1780b57cec5SDimitry Andric                   StringRef ArchitectureName = StringRef()) {
1790b57cec5SDimitry Andric   HadError = true;
1800b57cec5SDimitry Andric   WithColor::error(errs(), ToolName) << FileName;
1810b57cec5SDimitry Andric 
1820b57cec5SDimitry Andric   Expected<StringRef> NameOrErr = C.getName();
1830b57cec5SDimitry Andric   // TODO: if we have a error getting the name then it would be nice to print
1840b57cec5SDimitry Andric   // the index of which archive member this is and or its offset in the
1850b57cec5SDimitry Andric   // archive instead of "???" as the name.
1860b57cec5SDimitry Andric   if (!NameOrErr) {
1870b57cec5SDimitry Andric     consumeError(NameOrErr.takeError());
1880b57cec5SDimitry Andric     errs() << "(" << "???" << ")";
1890b57cec5SDimitry Andric   } else
1900b57cec5SDimitry Andric     errs() << "(" << NameOrErr.get() << ")";
1910b57cec5SDimitry Andric 
1920b57cec5SDimitry Andric   if (!ArchitectureName.empty())
1930b57cec5SDimitry Andric     errs() << " (for architecture " << ArchitectureName << ")";
1940b57cec5SDimitry Andric 
1950b57cec5SDimitry Andric   std::string Buf;
1960b57cec5SDimitry Andric   raw_string_ostream OS(Buf);
1970b57cec5SDimitry Andric   logAllUnhandledErrors(std::move(E), OS);
1980b57cec5SDimitry Andric   OS.flush();
199fe6060f1SDimitry Andric   errs() << ": " << Buf << "\n";
2000b57cec5SDimitry Andric }
2010b57cec5SDimitry Andric 
2020b57cec5SDimitry Andric // This version of error() prints the file name and which architecture slice it
2030b57cec5SDimitry Andric // is from, for example: "foo.o (for architecture i386)" after the ToolName
2040b57cec5SDimitry Andric // before the error message.  It sets HadError but returns allowing the code to
2050b57cec5SDimitry Andric // move on to other architecture slices.
2060b57cec5SDimitry Andric static void error(llvm::Error E, StringRef FileName,
2070b57cec5SDimitry Andric                   StringRef ArchitectureName = StringRef()) {
2080b57cec5SDimitry Andric   HadError = true;
2090b57cec5SDimitry Andric   WithColor::error(errs(), ToolName) << FileName;
2100b57cec5SDimitry Andric 
2110b57cec5SDimitry Andric   if (!ArchitectureName.empty())
2120b57cec5SDimitry Andric     errs() << " (for architecture " << ArchitectureName << ")";
2130b57cec5SDimitry Andric 
2140b57cec5SDimitry Andric   std::string Buf;
2150b57cec5SDimitry Andric   raw_string_ostream OS(Buf);
2160b57cec5SDimitry Andric   logAllUnhandledErrors(std::move(E), OS);
2170b57cec5SDimitry Andric   OS.flush();
218fe6060f1SDimitry Andric   errs() << ": " << Buf << "\n";
2190b57cec5SDimitry Andric }
2200b57cec5SDimitry Andric 
2210b57cec5SDimitry Andric namespace {
2220b57cec5SDimitry Andric struct NMSymbol {
2230b57cec5SDimitry Andric   uint64_t Address;
2240b57cec5SDimitry Andric   uint64_t Size;
2250b57cec5SDimitry Andric   char TypeChar;
226e8d8bef9SDimitry Andric   std::string Name;
2270b57cec5SDimitry Andric   StringRef SectionName;
2280b57cec5SDimitry Andric   StringRef TypeName;
2290b57cec5SDimitry Andric   BasicSymbolRef Sym;
23081ad6265SDimitry Andric   StringRef Visibility;
23181ad6265SDimitry Andric 
2320b57cec5SDimitry Andric   // The Sym field above points to the native symbol in the object file,
2330b57cec5SDimitry Andric   // for Mach-O when we are creating symbols from the dyld info the above
2340b57cec5SDimitry Andric   // pointer is null as there is no native symbol.  In these cases the fields
2350b57cec5SDimitry Andric   // below are filled in to represent what would have been a Mach-O nlist
2360b57cec5SDimitry Andric   // native symbol.
2370b57cec5SDimitry Andric   uint32_t SymFlags;
2380b57cec5SDimitry Andric   SectionRef Section;
2390b57cec5SDimitry Andric   uint8_t NType;
2400b57cec5SDimitry Andric   uint8_t NSect;
2410b57cec5SDimitry Andric   uint16_t NDesc;
242e8d8bef9SDimitry Andric   std::string IndirectName;
24381ad6265SDimitry Andric 
24481ad6265SDimitry Andric   bool isDefined() const {
24581ad6265SDimitry Andric     if (Sym.getRawDataRefImpl().p) {
24681ad6265SDimitry Andric       uint32_t Flags = cantFail(Sym.getFlags());
24781ad6265SDimitry Andric       return !(Flags & SymbolRef::SF_Undefined);
24881ad6265SDimitry Andric     }
24981ad6265SDimitry Andric     return TypeChar != 'U';
25081ad6265SDimitry Andric   }
25181ad6265SDimitry Andric 
25281ad6265SDimitry Andric   bool initializeFlags(const SymbolicFile &Obj) {
25381ad6265SDimitry Andric     Expected<uint32_t> SymFlagsOrErr = Sym.getFlags();
25481ad6265SDimitry Andric     if (!SymFlagsOrErr) {
25581ad6265SDimitry Andric       // TODO: Test this error.
25681ad6265SDimitry Andric       error(SymFlagsOrErr.takeError(), Obj.getFileName());
25781ad6265SDimitry Andric       return false;
25881ad6265SDimitry Andric     }
25981ad6265SDimitry Andric     SymFlags = *SymFlagsOrErr;
26081ad6265SDimitry Andric     return true;
26181ad6265SDimitry Andric   }
26281ad6265SDimitry Andric 
26381ad6265SDimitry Andric   bool shouldPrint() const {
26481ad6265SDimitry Andric     bool Undefined = SymFlags & SymbolRef::SF_Undefined;
26581ad6265SDimitry Andric     bool Global = SymFlags & SymbolRef::SF_Global;
26681ad6265SDimitry Andric     bool Weak = SymFlags & SymbolRef::SF_Weak;
26781ad6265SDimitry Andric     bool FormatSpecific = SymFlags & SymbolRef::SF_FormatSpecific;
26881ad6265SDimitry Andric     if ((!Undefined && UndefinedOnly) || (Undefined && DefinedOnly) ||
26981ad6265SDimitry Andric         (!Global && ExternalOnly) || (Weak && NoWeakSymbols) ||
27081ad6265SDimitry Andric         (FormatSpecific && !(SpecialSyms || DebugSyms)))
27181ad6265SDimitry Andric       return false;
27281ad6265SDimitry Andric     return true;
27381ad6265SDimitry Andric   }
2740b57cec5SDimitry Andric };
2750b57cec5SDimitry Andric 
27681ad6265SDimitry Andric bool operator<(const NMSymbol &A, const NMSymbol &B) {
27781ad6265SDimitry Andric   if (NumericSort)
27881ad6265SDimitry Andric     return std::make_tuple(A.isDefined(), A.Address, A.Name, A.Size) <
27981ad6265SDimitry Andric            std::make_tuple(B.isDefined(), B.Address, B.Name, B.Size);
28081ad6265SDimitry Andric   if (SizeSort)
2810b57cec5SDimitry Andric     return std::make_tuple(A.Size, A.Name, A.Address) <
2820b57cec5SDimitry Andric            std::make_tuple(B.Size, B.Name, B.Address);
28381ad6265SDimitry Andric   if (ExportSymbols)
28481ad6265SDimitry Andric     return std::make_tuple(A.Name, A.Visibility) <
28581ad6265SDimitry Andric            std::make_tuple(B.Name, B.Visibility);
2860b57cec5SDimitry Andric   return std::make_tuple(A.Name, A.Size, A.Address) <
2870b57cec5SDimitry Andric          std::make_tuple(B.Name, B.Size, B.Address);
2880b57cec5SDimitry Andric }
2890b57cec5SDimitry Andric 
29081ad6265SDimitry Andric bool operator>(const NMSymbol &A, const NMSymbol &B) { return B < A; }
29181ad6265SDimitry Andric bool operator==(const NMSymbol &A, const NMSymbol &B) {
29281ad6265SDimitry Andric   return !(A < B) && !(B < A);
29381ad6265SDimitry Andric }
29481ad6265SDimitry Andric } // anonymous namespace
29581ad6265SDimitry Andric 
2960b57cec5SDimitry Andric static char isSymbolList64Bit(SymbolicFile &Obj) {
2970b57cec5SDimitry Andric   if (auto *IRObj = dyn_cast<IRObjectFile>(&Obj))
2980b57cec5SDimitry Andric     return Triple(IRObj->getTargetTriple()).isArch64Bit();
2990b57cec5SDimitry Andric   if (isa<COFFObjectFile>(Obj) || isa<COFFImportFile>(Obj))
3000b57cec5SDimitry Andric     return false;
30104eeddc0SDimitry Andric   if (XCOFFObjectFile *XCOFFObj = dyn_cast<XCOFFObjectFile>(&Obj))
30204eeddc0SDimitry Andric     return XCOFFObj->is64Bit();
3030b57cec5SDimitry Andric   if (isa<WasmObjectFile>(Obj))
3040b57cec5SDimitry Andric     return false;
3055ffd83dbSDimitry Andric   if (TapiFile *Tapi = dyn_cast<TapiFile>(&Obj))
3065ffd83dbSDimitry Andric     return Tapi->is64Bit();
3070b57cec5SDimitry Andric   if (MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(&Obj))
3080b57cec5SDimitry Andric     return MachO->is64Bit();
3090b57cec5SDimitry Andric   return cast<ELFObjectFileBase>(Obj).getBytesInAddress() == 8;
3100b57cec5SDimitry Andric }
3110b57cec5SDimitry Andric 
3120b57cec5SDimitry Andric static StringRef CurrentFilename;
3130b57cec5SDimitry Andric 
3140b57cec5SDimitry Andric static char getSymbolNMTypeChar(IRObjectFile &Obj, basic_symbol_iterator I);
3150b57cec5SDimitry Andric 
3160b57cec5SDimitry Andric // darwinPrintSymbol() is used to print a symbol from a Mach-O file when the
3170b57cec5SDimitry Andric // the OutputFormat is darwin or we are printing Mach-O symbols in hex.  For
3180b57cec5SDimitry Andric // the darwin format it produces the same output as darwin's nm(1) -m output
3190b57cec5SDimitry Andric // and when printing Mach-O symbols in hex it produces the same output as
3200b57cec5SDimitry Andric // darwin's nm(1) -x format.
3210b57cec5SDimitry Andric static void darwinPrintSymbol(SymbolicFile &Obj, const NMSymbol &S,
3220b57cec5SDimitry Andric                               char *SymbolAddrStr, const char *printBlanks,
3230b57cec5SDimitry Andric                               const char *printDashes,
3240b57cec5SDimitry Andric                               const char *printFormat) {
3250b57cec5SDimitry Andric   MachO::mach_header H;
3260b57cec5SDimitry Andric   MachO::mach_header_64 H_64;
3270b57cec5SDimitry Andric   uint32_t Filetype = MachO::MH_OBJECT;
3280b57cec5SDimitry Andric   uint32_t Flags = 0;
3290b57cec5SDimitry Andric   uint8_t NType = 0;
3300b57cec5SDimitry Andric   uint8_t NSect = 0;
3310b57cec5SDimitry Andric   uint16_t NDesc = 0;
3320b57cec5SDimitry Andric   uint32_t NStrx = 0;
3330b57cec5SDimitry Andric   uint64_t NValue = 0;
3340b57cec5SDimitry Andric   MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(&Obj);
3350b57cec5SDimitry Andric   if (Obj.isIR()) {
3365ffd83dbSDimitry Andric     uint32_t SymFlags = cantFail(S.Sym.getFlags());
3370b57cec5SDimitry Andric     if (SymFlags & SymbolRef::SF_Global)
3380b57cec5SDimitry Andric       NType |= MachO::N_EXT;
3390b57cec5SDimitry Andric     if (SymFlags & SymbolRef::SF_Hidden)
3400b57cec5SDimitry Andric       NType |= MachO::N_PEXT;
3410b57cec5SDimitry Andric     if (SymFlags & SymbolRef::SF_Undefined)
3420b57cec5SDimitry Andric       NType |= MachO::N_EXT | MachO::N_UNDF;
3430b57cec5SDimitry Andric     else {
3440b57cec5SDimitry Andric       // Here we have a symbol definition.  So to fake out a section name we
3450b57cec5SDimitry Andric       // use 1, 2 and 3 for section numbers.  See below where they are used to
3460b57cec5SDimitry Andric       // print out fake section names.
3470b57cec5SDimitry Andric       NType |= MachO::N_SECT;
3480b57cec5SDimitry Andric       if (SymFlags & SymbolRef::SF_Const)
3490b57cec5SDimitry Andric         NSect = 3;
3500b57cec5SDimitry Andric       else if (SymFlags & SymbolRef::SF_Executable)
3510b57cec5SDimitry Andric         NSect = 1;
3520b57cec5SDimitry Andric       else
3530b57cec5SDimitry Andric         NSect = 2;
3540b57cec5SDimitry Andric     }
3550b57cec5SDimitry Andric     if (SymFlags & SymbolRef::SF_Weak)
3560b57cec5SDimitry Andric       NDesc |= MachO::N_WEAK_DEF;
3570b57cec5SDimitry Andric   } else {
3580b57cec5SDimitry Andric     DataRefImpl SymDRI = S.Sym.getRawDataRefImpl();
3590b57cec5SDimitry Andric     if (MachO->is64Bit()) {
3600b57cec5SDimitry Andric       H_64 = MachO->MachOObjectFile::getHeader64();
3610b57cec5SDimitry Andric       Filetype = H_64.filetype;
3620b57cec5SDimitry Andric       Flags = H_64.flags;
3630b57cec5SDimitry Andric       if (SymDRI.p){
3640b57cec5SDimitry Andric         MachO::nlist_64 STE_64 = MachO->getSymbol64TableEntry(SymDRI);
3650b57cec5SDimitry Andric         NType = STE_64.n_type;
3660b57cec5SDimitry Andric         NSect = STE_64.n_sect;
3670b57cec5SDimitry Andric         NDesc = STE_64.n_desc;
3680b57cec5SDimitry Andric         NStrx = STE_64.n_strx;
3690b57cec5SDimitry Andric         NValue = STE_64.n_value;
3700b57cec5SDimitry Andric       } else {
3710b57cec5SDimitry Andric         NType = S.NType;
3720b57cec5SDimitry Andric         NSect = S.NSect;
3730b57cec5SDimitry Andric         NDesc = S.NDesc;
3740b57cec5SDimitry Andric         NStrx = 0;
3750b57cec5SDimitry Andric         NValue = S.Address;
3760b57cec5SDimitry Andric       }
3770b57cec5SDimitry Andric     } else {
3780b57cec5SDimitry Andric       H = MachO->MachOObjectFile::getHeader();
3790b57cec5SDimitry Andric       Filetype = H.filetype;
3800b57cec5SDimitry Andric       Flags = H.flags;
3810b57cec5SDimitry Andric       if (SymDRI.p){
3820b57cec5SDimitry Andric         MachO::nlist STE = MachO->getSymbolTableEntry(SymDRI);
3830b57cec5SDimitry Andric         NType = STE.n_type;
3840b57cec5SDimitry Andric         NSect = STE.n_sect;
3850b57cec5SDimitry Andric         NDesc = STE.n_desc;
3860b57cec5SDimitry Andric         NStrx = STE.n_strx;
3870b57cec5SDimitry Andric         NValue = STE.n_value;
3880b57cec5SDimitry Andric       } else {
3890b57cec5SDimitry Andric         NType = S.NType;
3900b57cec5SDimitry Andric         NSect = S.NSect;
3910b57cec5SDimitry Andric         NDesc = S.NDesc;
3920b57cec5SDimitry Andric         NStrx = 0;
3930b57cec5SDimitry Andric         NValue = S.Address;
3940b57cec5SDimitry Andric       }
3950b57cec5SDimitry Andric     }
3960b57cec5SDimitry Andric   }
3970b57cec5SDimitry Andric 
3980b57cec5SDimitry Andric   // If we are printing Mach-O symbols in hex do that and return.
3990b57cec5SDimitry Andric   if (FormatMachOasHex) {
4000b57cec5SDimitry Andric     outs() << format(printFormat, NValue) << ' '
4010b57cec5SDimitry Andric            << format("%02x %02x %04x %08x", NType, NSect, NDesc, NStrx) << ' '
4020b57cec5SDimitry Andric            << S.Name;
4030b57cec5SDimitry Andric     if ((NType & MachO::N_TYPE) == MachO::N_INDR) {
4040b57cec5SDimitry Andric       outs() << " (indirect for ";
4050b57cec5SDimitry Andric       outs() << format(printFormat, NValue) << ' ';
4060b57cec5SDimitry Andric       StringRef IndirectName;
4070b57cec5SDimitry Andric       if (S.Sym.getRawDataRefImpl().p) {
4080b57cec5SDimitry Andric         if (MachO->getIndirectName(S.Sym.getRawDataRefImpl(), IndirectName))
4090b57cec5SDimitry Andric           outs() << "?)";
4100b57cec5SDimitry Andric         else
4110b57cec5SDimitry Andric           outs() << IndirectName << ")";
4120b57cec5SDimitry Andric       } else
4130b57cec5SDimitry Andric         outs() << S.IndirectName << ")";
4140b57cec5SDimitry Andric     }
4150b57cec5SDimitry Andric     outs() << "\n";
4160b57cec5SDimitry Andric     return;
4170b57cec5SDimitry Andric   }
4180b57cec5SDimitry Andric 
4190b57cec5SDimitry Andric   if (PrintAddress) {
4200b57cec5SDimitry Andric     if ((NType & MachO::N_TYPE) == MachO::N_INDR)
4210b57cec5SDimitry Andric       strcpy(SymbolAddrStr, printBlanks);
4220b57cec5SDimitry Andric     if (Obj.isIR() && (NType & MachO::N_TYPE) == MachO::N_TYPE)
4230b57cec5SDimitry Andric       strcpy(SymbolAddrStr, printDashes);
4240b57cec5SDimitry Andric     outs() << SymbolAddrStr << ' ';
4250b57cec5SDimitry Andric   }
4260b57cec5SDimitry Andric 
4270b57cec5SDimitry Andric   switch (NType & MachO::N_TYPE) {
4280b57cec5SDimitry Andric   case MachO::N_UNDF:
4290b57cec5SDimitry Andric     if (NValue != 0) {
4300b57cec5SDimitry Andric       outs() << "(common) ";
4310b57cec5SDimitry Andric       if (MachO::GET_COMM_ALIGN(NDesc) != 0)
4320b57cec5SDimitry Andric         outs() << "(alignment 2^" << (int)MachO::GET_COMM_ALIGN(NDesc) << ") ";
4330b57cec5SDimitry Andric     } else {
4340b57cec5SDimitry Andric       if ((NType & MachO::N_TYPE) == MachO::N_PBUD)
4350b57cec5SDimitry Andric         outs() << "(prebound ";
4360b57cec5SDimitry Andric       else
4370b57cec5SDimitry Andric         outs() << "(";
4380b57cec5SDimitry Andric       if ((NDesc & MachO::REFERENCE_TYPE) ==
4390b57cec5SDimitry Andric           MachO::REFERENCE_FLAG_UNDEFINED_LAZY)
4400b57cec5SDimitry Andric         outs() << "undefined [lazy bound]) ";
4410b57cec5SDimitry Andric       else if ((NDesc & MachO::REFERENCE_TYPE) ==
4420b57cec5SDimitry Andric                MachO::REFERENCE_FLAG_PRIVATE_UNDEFINED_LAZY)
4430b57cec5SDimitry Andric         outs() << "undefined [private lazy bound]) ";
4440b57cec5SDimitry Andric       else if ((NDesc & MachO::REFERENCE_TYPE) ==
4450b57cec5SDimitry Andric                MachO::REFERENCE_FLAG_PRIVATE_UNDEFINED_NON_LAZY)
4460b57cec5SDimitry Andric         outs() << "undefined [private]) ";
4470b57cec5SDimitry Andric       else
4480b57cec5SDimitry Andric         outs() << "undefined) ";
4490b57cec5SDimitry Andric     }
4500b57cec5SDimitry Andric     break;
4510b57cec5SDimitry Andric   case MachO::N_ABS:
4520b57cec5SDimitry Andric     outs() << "(absolute) ";
4530b57cec5SDimitry Andric     break;
4540b57cec5SDimitry Andric   case MachO::N_INDR:
4550b57cec5SDimitry Andric     outs() << "(indirect) ";
4560b57cec5SDimitry Andric     break;
4570b57cec5SDimitry Andric   case MachO::N_SECT: {
4580b57cec5SDimitry Andric     if (Obj.isIR()) {
4590b57cec5SDimitry Andric       // For llvm bitcode files print out a fake section name using the values
4600b57cec5SDimitry Andric       // use 1, 2 and 3 for section numbers as set above.
4610b57cec5SDimitry Andric       if (NSect == 1)
4620b57cec5SDimitry Andric         outs() << "(LTO,CODE) ";
4630b57cec5SDimitry Andric       else if (NSect == 2)
4640b57cec5SDimitry Andric         outs() << "(LTO,DATA) ";
4650b57cec5SDimitry Andric       else if (NSect == 3)
4660b57cec5SDimitry Andric         outs() << "(LTO,RODATA) ";
4670b57cec5SDimitry Andric       else
4680b57cec5SDimitry Andric         outs() << "(?,?) ";
4690b57cec5SDimitry Andric       break;
4700b57cec5SDimitry Andric     }
4710b57cec5SDimitry Andric     section_iterator Sec = SectionRef();
4720b57cec5SDimitry Andric     if (S.Sym.getRawDataRefImpl().p) {
4730b57cec5SDimitry Andric       Expected<section_iterator> SecOrErr =
4740b57cec5SDimitry Andric           MachO->getSymbolSection(S.Sym.getRawDataRefImpl());
4750b57cec5SDimitry Andric       if (!SecOrErr) {
4760b57cec5SDimitry Andric         consumeError(SecOrErr.takeError());
4770b57cec5SDimitry Andric         outs() << "(?,?) ";
4780b57cec5SDimitry Andric         break;
4790b57cec5SDimitry Andric       }
4800b57cec5SDimitry Andric       Sec = *SecOrErr;
4810b57cec5SDimitry Andric       if (Sec == MachO->section_end()) {
4820b57cec5SDimitry Andric         outs() << "(?,?) ";
4830b57cec5SDimitry Andric         break;
4840b57cec5SDimitry Andric       }
4850b57cec5SDimitry Andric     } else {
4860b57cec5SDimitry Andric       Sec = S.Section;
4870b57cec5SDimitry Andric     }
4880b57cec5SDimitry Andric     DataRefImpl Ref = Sec->getRawDataRefImpl();
4890b57cec5SDimitry Andric     StringRef SectionName;
4900b57cec5SDimitry Andric     if (Expected<StringRef> NameOrErr = MachO->getSectionName(Ref))
4910b57cec5SDimitry Andric       SectionName = *NameOrErr;
4920b57cec5SDimitry Andric     StringRef SegmentName = MachO->getSectionFinalSegmentName(Ref);
4930b57cec5SDimitry Andric     outs() << "(" << SegmentName << "," << SectionName << ") ";
4940b57cec5SDimitry Andric     break;
4950b57cec5SDimitry Andric   }
4960b57cec5SDimitry Andric   default:
4970b57cec5SDimitry Andric     outs() << "(?) ";
4980b57cec5SDimitry Andric     break;
4990b57cec5SDimitry Andric   }
5000b57cec5SDimitry Andric 
5010b57cec5SDimitry Andric   if (NType & MachO::N_EXT) {
5020b57cec5SDimitry Andric     if (NDesc & MachO::REFERENCED_DYNAMICALLY)
5030b57cec5SDimitry Andric       outs() << "[referenced dynamically] ";
5040b57cec5SDimitry Andric     if (NType & MachO::N_PEXT) {
5050b57cec5SDimitry Andric       if ((NDesc & MachO::N_WEAK_DEF) == MachO::N_WEAK_DEF)
5060b57cec5SDimitry Andric         outs() << "weak private external ";
5070b57cec5SDimitry Andric       else
5080b57cec5SDimitry Andric         outs() << "private external ";
5090b57cec5SDimitry Andric     } else {
5100b57cec5SDimitry Andric       if ((NDesc & MachO::N_WEAK_REF) == MachO::N_WEAK_REF ||
5110b57cec5SDimitry Andric           (NDesc & MachO::N_WEAK_DEF) == MachO::N_WEAK_DEF) {
5120b57cec5SDimitry Andric         if ((NDesc & (MachO::N_WEAK_REF | MachO::N_WEAK_DEF)) ==
5130b57cec5SDimitry Andric             (MachO::N_WEAK_REF | MachO::N_WEAK_DEF))
5140b57cec5SDimitry Andric           outs() << "weak external automatically hidden ";
5150b57cec5SDimitry Andric         else
5160b57cec5SDimitry Andric           outs() << "weak external ";
5170b57cec5SDimitry Andric       } else
5180b57cec5SDimitry Andric         outs() << "external ";
5190b57cec5SDimitry Andric     }
5200b57cec5SDimitry Andric   } else {
5210b57cec5SDimitry Andric     if (NType & MachO::N_PEXT)
5220b57cec5SDimitry Andric       outs() << "non-external (was a private external) ";
5230b57cec5SDimitry Andric     else
5240b57cec5SDimitry Andric       outs() << "non-external ";
5250b57cec5SDimitry Andric   }
5260b57cec5SDimitry Andric 
5270b57cec5SDimitry Andric   if (Filetype == MachO::MH_OBJECT) {
5280b57cec5SDimitry Andric     if (NDesc & MachO::N_NO_DEAD_STRIP)
5290b57cec5SDimitry Andric       outs() << "[no dead strip] ";
5300b57cec5SDimitry Andric     if ((NType & MachO::N_TYPE) != MachO::N_UNDF &&
5310b57cec5SDimitry Andric         NDesc & MachO::N_SYMBOL_RESOLVER)
5320b57cec5SDimitry Andric       outs() << "[symbol resolver] ";
5330b57cec5SDimitry Andric     if ((NType & MachO::N_TYPE) != MachO::N_UNDF && NDesc & MachO::N_ALT_ENTRY)
5340b57cec5SDimitry Andric       outs() << "[alt entry] ";
5350b57cec5SDimitry Andric     if ((NType & MachO::N_TYPE) != MachO::N_UNDF && NDesc & MachO::N_COLD_FUNC)
5360b57cec5SDimitry Andric       outs() << "[cold func] ";
5370b57cec5SDimitry Andric   }
5380b57cec5SDimitry Andric 
5390b57cec5SDimitry Andric   if ((NDesc & MachO::N_ARM_THUMB_DEF) == MachO::N_ARM_THUMB_DEF)
5400b57cec5SDimitry Andric     outs() << "[Thumb] ";
5410b57cec5SDimitry Andric 
5420b57cec5SDimitry Andric   if ((NType & MachO::N_TYPE) == MachO::N_INDR) {
5430b57cec5SDimitry Andric     outs() << S.Name << " (for ";
5440b57cec5SDimitry Andric     StringRef IndirectName;
5450b57cec5SDimitry Andric     if (MachO) {
5460b57cec5SDimitry Andric       if (S.Sym.getRawDataRefImpl().p) {
5470b57cec5SDimitry Andric         if (MachO->getIndirectName(S.Sym.getRawDataRefImpl(), IndirectName))
5480b57cec5SDimitry Andric           outs() << "?)";
5490b57cec5SDimitry Andric         else
5500b57cec5SDimitry Andric           outs() << IndirectName << ")";
5510b57cec5SDimitry Andric       } else
5520b57cec5SDimitry Andric         outs() << S.IndirectName << ")";
5530b57cec5SDimitry Andric     } else
5540b57cec5SDimitry Andric       outs() << "?)";
5550b57cec5SDimitry Andric   } else
5560b57cec5SDimitry Andric     outs() << S.Name;
5570b57cec5SDimitry Andric 
5580b57cec5SDimitry Andric   if ((Flags & MachO::MH_TWOLEVEL) == MachO::MH_TWOLEVEL &&
5590b57cec5SDimitry Andric       (((NType & MachO::N_TYPE) == MachO::N_UNDF && NValue == 0) ||
5600b57cec5SDimitry Andric        (NType & MachO::N_TYPE) == MachO::N_PBUD)) {
5610b57cec5SDimitry Andric     uint32_t LibraryOrdinal = MachO::GET_LIBRARY_ORDINAL(NDesc);
5620b57cec5SDimitry Andric     if (LibraryOrdinal != 0) {
5630b57cec5SDimitry Andric       if (LibraryOrdinal == MachO::EXECUTABLE_ORDINAL)
5640b57cec5SDimitry Andric         outs() << " (from executable)";
5650b57cec5SDimitry Andric       else if (LibraryOrdinal == MachO::DYNAMIC_LOOKUP_ORDINAL)
5660b57cec5SDimitry Andric         outs() << " (dynamically looked up)";
5670b57cec5SDimitry Andric       else {
5680b57cec5SDimitry Andric         StringRef LibraryName;
5690b57cec5SDimitry Andric         if (!MachO ||
5700b57cec5SDimitry Andric             MachO->getLibraryShortNameByIndex(LibraryOrdinal - 1, LibraryName))
5710b57cec5SDimitry Andric           outs() << " (from bad library ordinal " << LibraryOrdinal << ")";
5720b57cec5SDimitry Andric         else
5730b57cec5SDimitry Andric           outs() << " (from " << LibraryName << ")";
5740b57cec5SDimitry Andric       }
5750b57cec5SDimitry Andric     }
5760b57cec5SDimitry Andric   }
5770b57cec5SDimitry Andric 
5780b57cec5SDimitry Andric   outs() << "\n";
5790b57cec5SDimitry Andric }
5800b57cec5SDimitry Andric 
5810b57cec5SDimitry Andric // Table that maps Darwin's Mach-O stab constants to strings to allow printing.
5820b57cec5SDimitry Andric struct DarwinStabName {
5830b57cec5SDimitry Andric   uint8_t NType;
5840b57cec5SDimitry Andric   const char *Name;
5850b57cec5SDimitry Andric };
586349cc55cSDimitry Andric const struct DarwinStabName DarwinStabNames[] = {
5870b57cec5SDimitry Andric     {MachO::N_GSYM, "GSYM"},
5880b57cec5SDimitry Andric     {MachO::N_FNAME, "FNAME"},
5890b57cec5SDimitry Andric     {MachO::N_FUN, "FUN"},
5900b57cec5SDimitry Andric     {MachO::N_STSYM, "STSYM"},
5910b57cec5SDimitry Andric     {MachO::N_LCSYM, "LCSYM"},
5920b57cec5SDimitry Andric     {MachO::N_BNSYM, "BNSYM"},
5930b57cec5SDimitry Andric     {MachO::N_PC, "PC"},
5940b57cec5SDimitry Andric     {MachO::N_AST, "AST"},
5950b57cec5SDimitry Andric     {MachO::N_OPT, "OPT"},
5960b57cec5SDimitry Andric     {MachO::N_RSYM, "RSYM"},
5970b57cec5SDimitry Andric     {MachO::N_SLINE, "SLINE"},
5980b57cec5SDimitry Andric     {MachO::N_ENSYM, "ENSYM"},
5990b57cec5SDimitry Andric     {MachO::N_SSYM, "SSYM"},
6000b57cec5SDimitry Andric     {MachO::N_SO, "SO"},
6010b57cec5SDimitry Andric     {MachO::N_OSO, "OSO"},
6020b57cec5SDimitry Andric     {MachO::N_LSYM, "LSYM"},
6030b57cec5SDimitry Andric     {MachO::N_BINCL, "BINCL"},
6040b57cec5SDimitry Andric     {MachO::N_SOL, "SOL"},
6050b57cec5SDimitry Andric     {MachO::N_PARAMS, "PARAM"},
6060b57cec5SDimitry Andric     {MachO::N_VERSION, "VERS"},
6070b57cec5SDimitry Andric     {MachO::N_OLEVEL, "OLEV"},
6080b57cec5SDimitry Andric     {MachO::N_PSYM, "PSYM"},
6090b57cec5SDimitry Andric     {MachO::N_EINCL, "EINCL"},
6100b57cec5SDimitry Andric     {MachO::N_ENTRY, "ENTRY"},
6110b57cec5SDimitry Andric     {MachO::N_LBRAC, "LBRAC"},
6120b57cec5SDimitry Andric     {MachO::N_EXCL, "EXCL"},
6130b57cec5SDimitry Andric     {MachO::N_RBRAC, "RBRAC"},
6140b57cec5SDimitry Andric     {MachO::N_BCOMM, "BCOMM"},
6150b57cec5SDimitry Andric     {MachO::N_ECOMM, "ECOMM"},
6160b57cec5SDimitry Andric     {MachO::N_ECOML, "ECOML"},
6170b57cec5SDimitry Andric     {MachO::N_LENG, "LENG"},
6180b57cec5SDimitry Andric };
6190b57cec5SDimitry Andric 
6200b57cec5SDimitry Andric static const char *getDarwinStabString(uint8_t NType) {
621*bdd1243dSDimitry Andric   for (auto I : ArrayRef(DarwinStabNames))
6220b57cec5SDimitry Andric     if (I.NType == NType)
6230b57cec5SDimitry Andric       return I.Name;
6240b57cec5SDimitry Andric   return nullptr;
6250b57cec5SDimitry Andric }
6260b57cec5SDimitry Andric 
6270b57cec5SDimitry Andric // darwinPrintStab() prints the n_sect, n_desc along with a symbolic name of
6280b57cec5SDimitry Andric // a stab n_type value in a Mach-O file.
6290b57cec5SDimitry Andric static void darwinPrintStab(MachOObjectFile *MachO, const NMSymbol &S) {
6300b57cec5SDimitry Andric   MachO::nlist_64 STE_64;
6310b57cec5SDimitry Andric   MachO::nlist STE;
6320b57cec5SDimitry Andric   uint8_t NType;
6330b57cec5SDimitry Andric   uint8_t NSect;
6340b57cec5SDimitry Andric   uint16_t NDesc;
6350b57cec5SDimitry Andric   DataRefImpl SymDRI = S.Sym.getRawDataRefImpl();
6360b57cec5SDimitry Andric   if (MachO->is64Bit()) {
6370b57cec5SDimitry Andric     STE_64 = MachO->getSymbol64TableEntry(SymDRI);
6380b57cec5SDimitry Andric     NType = STE_64.n_type;
6390b57cec5SDimitry Andric     NSect = STE_64.n_sect;
6400b57cec5SDimitry Andric     NDesc = STE_64.n_desc;
6410b57cec5SDimitry Andric   } else {
6420b57cec5SDimitry Andric     STE = MachO->getSymbolTableEntry(SymDRI);
6430b57cec5SDimitry Andric     NType = STE.n_type;
6440b57cec5SDimitry Andric     NSect = STE.n_sect;
6450b57cec5SDimitry Andric     NDesc = STE.n_desc;
6460b57cec5SDimitry Andric   }
6470b57cec5SDimitry Andric 
6480b57cec5SDimitry Andric   outs() << format(" %02x %04x ", NSect, NDesc);
6490b57cec5SDimitry Andric   if (const char *stabString = getDarwinStabString(NType))
6500b57cec5SDimitry Andric     outs() << format("%5.5s", stabString);
6510b57cec5SDimitry Andric   else
6520b57cec5SDimitry Andric     outs() << format("   %02x", NType);
6530b57cec5SDimitry Andric }
6540b57cec5SDimitry Andric 
655*bdd1243dSDimitry Andric static std::optional<std::string> demangle(StringRef Name) {
656349cc55cSDimitry Andric   std::string Demangled;
65704eeddc0SDimitry Andric   if (nonMicrosoftDemangle(Name.str().c_str(), Demangled))
658349cc55cSDimitry Andric     return Demangled;
659*bdd1243dSDimitry Andric   return std::nullopt;
6600b57cec5SDimitry Andric }
6610b57cec5SDimitry Andric 
662*bdd1243dSDimitry Andric static std::optional<std::string> demangleXCOFF(StringRef Name) {
66304eeddc0SDimitry Andric   if (Name.empty() || Name[0] != '.')
66404eeddc0SDimitry Andric     return demangle(Name);
66504eeddc0SDimitry Andric 
66604eeddc0SDimitry Andric   Name = Name.drop_front();
667*bdd1243dSDimitry Andric   std::optional<std::string> DemangledName = demangle(Name);
66804eeddc0SDimitry Andric   if (DemangledName)
66904eeddc0SDimitry Andric     return "." + *DemangledName;
670*bdd1243dSDimitry Andric   return std::nullopt;
67104eeddc0SDimitry Andric }
67204eeddc0SDimitry Andric 
673*bdd1243dSDimitry Andric static std::optional<std::string> demangleMachO(StringRef Name) {
67404eeddc0SDimitry Andric   if (!Name.empty() && Name[0] == '_')
67504eeddc0SDimitry Andric     Name = Name.drop_front();
67604eeddc0SDimitry Andric   return demangle(Name);
67704eeddc0SDimitry Andric }
67804eeddc0SDimitry Andric 
6790b57cec5SDimitry Andric static bool symbolIsDefined(const NMSymbol &Sym) {
6800b57cec5SDimitry Andric   return Sym.TypeChar != 'U' && Sym.TypeChar != 'w' && Sym.TypeChar != 'v';
6810b57cec5SDimitry Andric }
6820b57cec5SDimitry Andric 
6835ffd83dbSDimitry Andric static void writeFileName(raw_ostream &S, StringRef ArchiveName,
6845ffd83dbSDimitry Andric                           StringRef ArchitectureName) {
6855ffd83dbSDimitry Andric   if (!ArchitectureName.empty())
6865ffd83dbSDimitry Andric     S << "(for architecture " << ArchitectureName << "):";
6875ffd83dbSDimitry Andric   if (OutputFormat == posix && !ArchiveName.empty())
6885ffd83dbSDimitry Andric     S << ArchiveName << "[" << CurrentFilename << "]: ";
6895ffd83dbSDimitry Andric   else {
6905ffd83dbSDimitry Andric     if (!ArchiveName.empty())
6915ffd83dbSDimitry Andric       S << ArchiveName << ":";
6925ffd83dbSDimitry Andric     S << CurrentFilename << ": ";
6935ffd83dbSDimitry Andric   }
6945ffd83dbSDimitry Andric }
6955ffd83dbSDimitry Andric 
69681ad6265SDimitry Andric static void sortSymbolList(std::vector<NMSymbol> &SymbolList) {
69781ad6265SDimitry Andric   if (NoSort)
69881ad6265SDimitry Andric     return;
6990b57cec5SDimitry Andric 
7000b57cec5SDimitry Andric   if (ReverseSort)
70181ad6265SDimitry Andric     llvm::sort(SymbolList, std::greater<>());
7028bcb0991SDimitry Andric   else
70381ad6265SDimitry Andric     llvm::sort(SymbolList);
7040b57cec5SDimitry Andric }
7050b57cec5SDimitry Andric 
70681ad6265SDimitry Andric static void printExportSymbolList(const std::vector<NMSymbol> &SymbolList) {
70781ad6265SDimitry Andric   for (const NMSymbol &Sym : SymbolList) {
70881ad6265SDimitry Andric     outs() << Sym.Name;
70981ad6265SDimitry Andric     if (!Sym.Visibility.empty())
71081ad6265SDimitry Andric       outs() << ' ' << Sym.Visibility;
71181ad6265SDimitry Andric     outs() << '\n';
71281ad6265SDimitry Andric   }
71381ad6265SDimitry Andric }
71481ad6265SDimitry Andric 
71581ad6265SDimitry Andric static void printSymbolList(SymbolicFile &Obj,
71681ad6265SDimitry Andric                             std::vector<NMSymbol> &SymbolList, bool printName,
71781ad6265SDimitry Andric                             StringRef ArchiveName, StringRef ArchitectureName) {
7180b57cec5SDimitry Andric   if (!PrintFileName) {
719fe6060f1SDimitry Andric     if ((OutputFormat == bsd || OutputFormat == posix ||
720fe6060f1SDimitry Andric          OutputFormat == just_symbols) &&
721fe6060f1SDimitry Andric         MultipleFiles && printName) {
7220b57cec5SDimitry Andric       outs() << '\n' << CurrentFilename << ":\n";
7230b57cec5SDimitry Andric     } else if (OutputFormat == sysv) {
7240b57cec5SDimitry Andric       outs() << "\n\nSymbols from " << CurrentFilename << ":\n\n";
7250b57cec5SDimitry Andric       if (isSymbolList64Bit(Obj))
7260b57cec5SDimitry Andric         outs() << "Name                  Value           Class        Type"
7270b57cec5SDimitry Andric                << "         Size             Line  Section\n";
7280b57cec5SDimitry Andric       else
7290b57cec5SDimitry Andric         outs() << "Name                  Value   Class        Type"
7300b57cec5SDimitry Andric                << "         Size     Line  Section\n";
7310b57cec5SDimitry Andric     }
7320b57cec5SDimitry Andric   }
7330b57cec5SDimitry Andric 
7340b57cec5SDimitry Andric   const char *printBlanks, *printDashes, *printFormat;
7350b57cec5SDimitry Andric   if (isSymbolList64Bit(Obj)) {
7360b57cec5SDimitry Andric     printBlanks = "                ";
7370b57cec5SDimitry Andric     printDashes = "----------------";
7380b57cec5SDimitry Andric     switch (AddressRadix) {
7390b57cec5SDimitry Andric     case Radix::o:
7400b57cec5SDimitry Andric       printFormat = OutputFormat == posix ? "%" PRIo64 : "%016" PRIo64;
7410b57cec5SDimitry Andric       break;
7420b57cec5SDimitry Andric     case Radix::x:
7430b57cec5SDimitry Andric       printFormat = OutputFormat == posix ? "%" PRIx64 : "%016" PRIx64;
7440b57cec5SDimitry Andric       break;
7450b57cec5SDimitry Andric     default:
7460b57cec5SDimitry Andric       printFormat = OutputFormat == posix ? "%" PRId64 : "%016" PRId64;
7470b57cec5SDimitry Andric     }
7480b57cec5SDimitry Andric   } else {
7490b57cec5SDimitry Andric     printBlanks = "        ";
7500b57cec5SDimitry Andric     printDashes = "--------";
7510b57cec5SDimitry Andric     switch (AddressRadix) {
7520b57cec5SDimitry Andric     case Radix::o:
7530b57cec5SDimitry Andric       printFormat = OutputFormat == posix ? "%" PRIo64 : "%08" PRIo64;
7540b57cec5SDimitry Andric       break;
7550b57cec5SDimitry Andric     case Radix::x:
7560b57cec5SDimitry Andric       printFormat = OutputFormat == posix ? "%" PRIx64 : "%08" PRIx64;
7570b57cec5SDimitry Andric       break;
7580b57cec5SDimitry Andric     default:
7590b57cec5SDimitry Andric       printFormat = OutputFormat == posix ? "%" PRId64 : "%08" PRId64;
7600b57cec5SDimitry Andric     }
7610b57cec5SDimitry Andric   }
7620b57cec5SDimitry Andric 
7630b57cec5SDimitry Andric   for (const NMSymbol &S : SymbolList) {
76481ad6265SDimitry Andric     if (!S.shouldPrint())
76581ad6265SDimitry Andric       continue;
76681ad6265SDimitry Andric 
767e8d8bef9SDimitry Andric     std::string Name = S.Name;
7680b57cec5SDimitry Andric     MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(&Obj);
7690b57cec5SDimitry Andric     if (Demangle) {
770*bdd1243dSDimitry Andric       function_ref<std::optional<std::string>(StringRef)> Fn = ::demangle;
77104eeddc0SDimitry Andric       if (Obj.isXCOFF())
77204eeddc0SDimitry Andric         Fn = demangleXCOFF;
77304eeddc0SDimitry Andric       if (Obj.isMachO())
77404eeddc0SDimitry Andric         Fn = demangleMachO;
775*bdd1243dSDimitry Andric       if (std::optional<std::string> Opt = Fn(S.Name))
7760b57cec5SDimitry Andric         Name = *Opt;
7770b57cec5SDimitry Andric     }
7780b57cec5SDimitry Andric 
7790b57cec5SDimitry Andric     if (PrintFileName)
7805ffd83dbSDimitry Andric       writeFileName(outs(), ArchiveName, ArchitectureName);
781fe6060f1SDimitry Andric     if ((OutputFormat == just_symbols ||
7820b57cec5SDimitry Andric          (UndefinedOnly && MachO && OutputFormat != darwin)) &&
7830b57cec5SDimitry Andric         OutputFormat != posix) {
7840b57cec5SDimitry Andric       outs() << Name << "\n";
7850b57cec5SDimitry Andric       continue;
7860b57cec5SDimitry Andric     }
7870b57cec5SDimitry Andric 
7880b57cec5SDimitry Andric     char SymbolAddrStr[23], SymbolSizeStr[23];
7890b57cec5SDimitry Andric 
7900b57cec5SDimitry Andric     // If the format is SysV or the symbol isn't defined, then print spaces.
7910b57cec5SDimitry Andric     if (OutputFormat == sysv || !symbolIsDefined(S)) {
7920b57cec5SDimitry Andric       if (OutputFormat == posix) {
7930b57cec5SDimitry Andric         format(printFormat, S.Address)
7940b57cec5SDimitry Andric             .print(SymbolAddrStr, sizeof(SymbolAddrStr));
7950b57cec5SDimitry Andric         format(printFormat, S.Size).print(SymbolSizeStr, sizeof(SymbolSizeStr));
7960b57cec5SDimitry Andric       } else {
7970b57cec5SDimitry Andric         strcpy(SymbolAddrStr, printBlanks);
7980b57cec5SDimitry Andric         strcpy(SymbolSizeStr, printBlanks);
7990b57cec5SDimitry Andric       }
8000b57cec5SDimitry Andric     }
8010b57cec5SDimitry Andric 
8020b57cec5SDimitry Andric     if (symbolIsDefined(S)) {
8030b57cec5SDimitry Andric       // Otherwise, print the symbol address and size.
8040b57cec5SDimitry Andric       if (Obj.isIR())
8050b57cec5SDimitry Andric         strcpy(SymbolAddrStr, printDashes);
8060b57cec5SDimitry Andric       else if (MachO && S.TypeChar == 'I')
8070b57cec5SDimitry Andric         strcpy(SymbolAddrStr, printBlanks);
8080b57cec5SDimitry Andric       else
8090b57cec5SDimitry Andric         format(printFormat, S.Address)
8100b57cec5SDimitry Andric             .print(SymbolAddrStr, sizeof(SymbolAddrStr));
8110b57cec5SDimitry Andric       format(printFormat, S.Size).print(SymbolSizeStr, sizeof(SymbolSizeStr));
8120b57cec5SDimitry Andric     }
8130b57cec5SDimitry Andric 
8140b57cec5SDimitry Andric     // If OutputFormat is darwin or we are printing Mach-O symbols in hex and
8150b57cec5SDimitry Andric     // we have a MachOObjectFile, call darwinPrintSymbol to print as darwin's
8160b57cec5SDimitry Andric     // nm(1) -m output or hex, else if OutputFormat is darwin or we are
8170b57cec5SDimitry Andric     // printing Mach-O symbols in hex and not a Mach-O object fall back to
8180b57cec5SDimitry Andric     // OutputFormat bsd (see below).
8190b57cec5SDimitry Andric     if ((OutputFormat == darwin || FormatMachOasHex) && (MachO || Obj.isIR())) {
8200b57cec5SDimitry Andric       darwinPrintSymbol(Obj, S, SymbolAddrStr, printBlanks, printDashes,
8210b57cec5SDimitry Andric                         printFormat);
8220b57cec5SDimitry Andric     } else if (OutputFormat == posix) {
8230b57cec5SDimitry Andric       outs() << Name << " " << S.TypeChar << " " << SymbolAddrStr << " "
8240b57cec5SDimitry Andric              << (MachO ? "0" : SymbolSizeStr) << "\n";
8250b57cec5SDimitry Andric     } else if (OutputFormat == bsd || (OutputFormat == darwin && !MachO)) {
8260b57cec5SDimitry Andric       if (PrintAddress)
8270b57cec5SDimitry Andric         outs() << SymbolAddrStr << ' ';
8280b57cec5SDimitry Andric       if (PrintSize)
8290b57cec5SDimitry Andric         outs() << SymbolSizeStr << ' ';
8300b57cec5SDimitry Andric       outs() << S.TypeChar;
8310b57cec5SDimitry Andric       if (S.TypeChar == '-' && MachO)
8320b57cec5SDimitry Andric         darwinPrintStab(MachO, S);
8330b57cec5SDimitry Andric       outs() << " " << Name;
8340b57cec5SDimitry Andric       if (S.TypeChar == 'I' && MachO) {
8350b57cec5SDimitry Andric         outs() << " (indirect for ";
8360b57cec5SDimitry Andric         if (S.Sym.getRawDataRefImpl().p) {
8370b57cec5SDimitry Andric           StringRef IndirectName;
8380b57cec5SDimitry Andric           if (MachO->getIndirectName(S.Sym.getRawDataRefImpl(), IndirectName))
8390b57cec5SDimitry Andric             outs() << "?)";
8400b57cec5SDimitry Andric           else
8410b57cec5SDimitry Andric             outs() << IndirectName << ")";
8420b57cec5SDimitry Andric         } else
8430b57cec5SDimitry Andric           outs() << S.IndirectName << ")";
8440b57cec5SDimitry Andric       }
8450b57cec5SDimitry Andric       outs() << "\n";
8460b57cec5SDimitry Andric     } else if (OutputFormat == sysv) {
8470b57cec5SDimitry Andric       outs() << left_justify(Name, 20) << "|" << SymbolAddrStr << "|   "
8480b57cec5SDimitry Andric              << S.TypeChar << "  |" << right_justify(S.TypeName, 18) << "|"
8490b57cec5SDimitry Andric              << SymbolSizeStr << "|     |" << S.SectionName << "\n";
8500b57cec5SDimitry Andric     }
8510b57cec5SDimitry Andric   }
8520b57cec5SDimitry Andric 
8530b57cec5SDimitry Andric   SymbolList.clear();
8540b57cec5SDimitry Andric }
8550b57cec5SDimitry Andric 
8560b57cec5SDimitry Andric static char getSymbolNMTypeChar(ELFObjectFileBase &Obj,
8570b57cec5SDimitry Andric                                 basic_symbol_iterator I) {
8580b57cec5SDimitry Andric   // OK, this is ELF
8590b57cec5SDimitry Andric   elf_symbol_iterator SymI(I);
8600b57cec5SDimitry Andric 
8610b57cec5SDimitry Andric   Expected<elf_section_iterator> SecIOrErr = SymI->getSection();
8620b57cec5SDimitry Andric   if (!SecIOrErr) {
8630b57cec5SDimitry Andric     consumeError(SecIOrErr.takeError());
8640b57cec5SDimitry Andric     return '?';
8650b57cec5SDimitry Andric   }
8660b57cec5SDimitry Andric 
8670b57cec5SDimitry Andric   uint8_t Binding = SymI->getBinding();
8680b57cec5SDimitry Andric   if (Binding == ELF::STB_GNU_UNIQUE)
8690b57cec5SDimitry Andric     return 'u';
8700b57cec5SDimitry Andric 
8710b57cec5SDimitry Andric   assert(Binding != ELF::STB_WEAK && "STB_WEAK not tested in calling function");
8720b57cec5SDimitry Andric   if (Binding != ELF::STB_GLOBAL && Binding != ELF::STB_LOCAL)
8730b57cec5SDimitry Andric     return '?';
8740b57cec5SDimitry Andric 
8750b57cec5SDimitry Andric   elf_section_iterator SecI = *SecIOrErr;
8760b57cec5SDimitry Andric   if (SecI != Obj.section_end()) {
8770b57cec5SDimitry Andric     uint32_t Type = SecI->getType();
8780b57cec5SDimitry Andric     uint64_t Flags = SecI->getFlags();
8790b57cec5SDimitry Andric     if (Flags & ELF::SHF_EXECINSTR)
8800b57cec5SDimitry Andric       return 't';
8810b57cec5SDimitry Andric     if (Type == ELF::SHT_NOBITS)
8820b57cec5SDimitry Andric       return 'b';
8830b57cec5SDimitry Andric     if (Flags & ELF::SHF_ALLOC)
8840b57cec5SDimitry Andric       return Flags & ELF::SHF_WRITE ? 'd' : 'r';
8850b57cec5SDimitry Andric 
8868bcb0991SDimitry Andric     auto NameOrErr = SecI->getName();
8878bcb0991SDimitry Andric     if (!NameOrErr) {
8888bcb0991SDimitry Andric       consumeError(NameOrErr.takeError());
8890b57cec5SDimitry Andric       return '?';
8908bcb0991SDimitry Andric     }
8918bcb0991SDimitry Andric     if ((*NameOrErr).startswith(".debug"))
8920b57cec5SDimitry Andric       return 'N';
8930b57cec5SDimitry Andric     if (!(Flags & ELF::SHF_WRITE))
8940b57cec5SDimitry Andric       return 'n';
8950b57cec5SDimitry Andric   }
8960b57cec5SDimitry Andric 
8970b57cec5SDimitry Andric   return '?';
8980b57cec5SDimitry Andric }
8990b57cec5SDimitry Andric 
9000b57cec5SDimitry Andric static char getSymbolNMTypeChar(COFFObjectFile &Obj, symbol_iterator I) {
9010b57cec5SDimitry Andric   COFFSymbolRef Symb = Obj.getCOFFSymbol(*I);
9020b57cec5SDimitry Andric   // OK, this is COFF.
9030b57cec5SDimitry Andric   symbol_iterator SymI(I);
9040b57cec5SDimitry Andric 
9050b57cec5SDimitry Andric   Expected<StringRef> Name = SymI->getName();
9060b57cec5SDimitry Andric   if (!Name) {
9070b57cec5SDimitry Andric     consumeError(Name.takeError());
9080b57cec5SDimitry Andric     return '?';
9090b57cec5SDimitry Andric   }
9100b57cec5SDimitry Andric 
9110b57cec5SDimitry Andric   char Ret = StringSwitch<char>(*Name)
9120b57cec5SDimitry Andric                  .StartsWith(".debug", 'N')
9130b57cec5SDimitry Andric                  .StartsWith(".sxdata", 'N')
9140b57cec5SDimitry Andric                  .Default('?');
9150b57cec5SDimitry Andric 
9160b57cec5SDimitry Andric   if (Ret != '?')
9170b57cec5SDimitry Andric     return Ret;
9180b57cec5SDimitry Andric 
9190b57cec5SDimitry Andric   uint32_t Characteristics = 0;
9200b57cec5SDimitry Andric   if (!COFF::isReservedSectionNumber(Symb.getSectionNumber())) {
9210b57cec5SDimitry Andric     Expected<section_iterator> SecIOrErr = SymI->getSection();
9220b57cec5SDimitry Andric     if (!SecIOrErr) {
9230b57cec5SDimitry Andric       consumeError(SecIOrErr.takeError());
9240b57cec5SDimitry Andric       return '?';
9250b57cec5SDimitry Andric     }
9260b57cec5SDimitry Andric     section_iterator SecI = *SecIOrErr;
9270b57cec5SDimitry Andric     const coff_section *Section = Obj.getCOFFSection(*SecI);
9280b57cec5SDimitry Andric     Characteristics = Section->Characteristics;
9290b57cec5SDimitry Andric     if (Expected<StringRef> NameOrErr = Obj.getSectionName(Section))
9300b57cec5SDimitry Andric       if (NameOrErr->startswith(".idata"))
9310b57cec5SDimitry Andric         return 'i';
9320b57cec5SDimitry Andric   }
9330b57cec5SDimitry Andric 
9340b57cec5SDimitry Andric   switch (Symb.getSectionNumber()) {
9350b57cec5SDimitry Andric   case COFF::IMAGE_SYM_DEBUG:
9360b57cec5SDimitry Andric     return 'n';
9370b57cec5SDimitry Andric   default:
9380b57cec5SDimitry Andric     // Check section type.
9390b57cec5SDimitry Andric     if (Characteristics & COFF::IMAGE_SCN_CNT_CODE)
9400b57cec5SDimitry Andric       return 't';
9410b57cec5SDimitry Andric     if (Characteristics & COFF::IMAGE_SCN_CNT_INITIALIZED_DATA)
9420b57cec5SDimitry Andric       return Characteristics & COFF::IMAGE_SCN_MEM_WRITE ? 'd' : 'r';
9430b57cec5SDimitry Andric     if (Characteristics & COFF::IMAGE_SCN_CNT_UNINITIALIZED_DATA)
9440b57cec5SDimitry Andric       return 'b';
9450b57cec5SDimitry Andric     if (Characteristics & COFF::IMAGE_SCN_LNK_INFO)
9460b57cec5SDimitry Andric       return 'i';
9470b57cec5SDimitry Andric     // Check for section symbol.
9480b57cec5SDimitry Andric     if (Symb.isSectionDefinition())
9490b57cec5SDimitry Andric       return 's';
9500b57cec5SDimitry Andric   }
9510b57cec5SDimitry Andric 
9520b57cec5SDimitry Andric   return '?';
9530b57cec5SDimitry Andric }
9540b57cec5SDimitry Andric 
95504eeddc0SDimitry Andric static char getSymbolNMTypeChar(XCOFFObjectFile &Obj, symbol_iterator I) {
95604eeddc0SDimitry Andric   Expected<uint32_t> TypeOrErr = I->getType();
95704eeddc0SDimitry Andric   if (!TypeOrErr) {
95804eeddc0SDimitry Andric     warn(TypeOrErr.takeError(), Obj.getFileName(),
95904eeddc0SDimitry Andric          "for symbol with index " +
96004eeddc0SDimitry Andric              Twine(Obj.getSymbolIndex(I->getRawDataRefImpl().p)));
96104eeddc0SDimitry Andric     return '?';
96204eeddc0SDimitry Andric   }
96304eeddc0SDimitry Andric 
96404eeddc0SDimitry Andric   uint32_t SymType = *TypeOrErr;
96504eeddc0SDimitry Andric 
96604eeddc0SDimitry Andric   if (SymType == SymbolRef::ST_File)
96704eeddc0SDimitry Andric     return 'f';
96804eeddc0SDimitry Andric 
96904eeddc0SDimitry Andric   // If the I->getSection() call would return an error, the earlier I->getType()
97004eeddc0SDimitry Andric   // call will already have returned the same error first.
97104eeddc0SDimitry Andric   section_iterator SecIter = cantFail(I->getSection());
97204eeddc0SDimitry Andric 
97304eeddc0SDimitry Andric   if (SecIter == Obj.section_end())
97404eeddc0SDimitry Andric     return '?';
97504eeddc0SDimitry Andric 
97604eeddc0SDimitry Andric   if (Obj.isDebugSection(SecIter->getRawDataRefImpl()))
97704eeddc0SDimitry Andric     return 'N';
97804eeddc0SDimitry Andric 
97904eeddc0SDimitry Andric   if (SecIter->isText())
98004eeddc0SDimitry Andric     return 't';
98104eeddc0SDimitry Andric 
98204eeddc0SDimitry Andric   if (SecIter->isData())
98304eeddc0SDimitry Andric     return 'd';
98404eeddc0SDimitry Andric 
98504eeddc0SDimitry Andric   if (SecIter->isBSS())
98604eeddc0SDimitry Andric     return 'b';
98704eeddc0SDimitry Andric 
98804eeddc0SDimitry Andric   return '?';
98904eeddc0SDimitry Andric }
99004eeddc0SDimitry Andric 
9910b57cec5SDimitry Andric static char getSymbolNMTypeChar(COFFImportFile &Obj) {
9920b57cec5SDimitry Andric   switch (Obj.getCOFFImportHeader()->getType()) {
9930b57cec5SDimitry Andric   case COFF::IMPORT_CODE:
9940b57cec5SDimitry Andric     return 't';
9950b57cec5SDimitry Andric   case COFF::IMPORT_DATA:
9960b57cec5SDimitry Andric     return 'd';
9970b57cec5SDimitry Andric   case COFF::IMPORT_CONST:
9980b57cec5SDimitry Andric     return 'r';
9990b57cec5SDimitry Andric   }
10000b57cec5SDimitry Andric   return '?';
10010b57cec5SDimitry Andric }
10020b57cec5SDimitry Andric 
10030b57cec5SDimitry Andric static char getSymbolNMTypeChar(MachOObjectFile &Obj, basic_symbol_iterator I) {
10040b57cec5SDimitry Andric   DataRefImpl Symb = I->getRawDataRefImpl();
10050b57cec5SDimitry Andric   uint8_t NType = Obj.is64Bit() ? Obj.getSymbol64TableEntry(Symb).n_type
10060b57cec5SDimitry Andric                                 : Obj.getSymbolTableEntry(Symb).n_type;
10070b57cec5SDimitry Andric 
10080b57cec5SDimitry Andric   if (NType & MachO::N_STAB)
10090b57cec5SDimitry Andric     return '-';
10100b57cec5SDimitry Andric 
10110b57cec5SDimitry Andric   switch (NType & MachO::N_TYPE) {
10120b57cec5SDimitry Andric   case MachO::N_ABS:
10130b57cec5SDimitry Andric     return 's';
10140b57cec5SDimitry Andric   case MachO::N_INDR:
10150b57cec5SDimitry Andric     return 'i';
10160b57cec5SDimitry Andric   case MachO::N_SECT: {
10170b57cec5SDimitry Andric     Expected<section_iterator> SecOrErr = Obj.getSymbolSection(Symb);
10180b57cec5SDimitry Andric     if (!SecOrErr) {
10190b57cec5SDimitry Andric       consumeError(SecOrErr.takeError());
10200b57cec5SDimitry Andric       return 's';
10210b57cec5SDimitry Andric     }
10220b57cec5SDimitry Andric     section_iterator Sec = *SecOrErr;
10230b57cec5SDimitry Andric     if (Sec == Obj.section_end())
10240b57cec5SDimitry Andric       return 's';
10250b57cec5SDimitry Andric     DataRefImpl Ref = Sec->getRawDataRefImpl();
10260b57cec5SDimitry Andric     StringRef SectionName;
10270b57cec5SDimitry Andric     if (Expected<StringRef> NameOrErr = Obj.getSectionName(Ref))
10280b57cec5SDimitry Andric       SectionName = *NameOrErr;
10290b57cec5SDimitry Andric     StringRef SegmentName = Obj.getSectionFinalSegmentName(Ref);
10300b57cec5SDimitry Andric     if (Obj.is64Bit() && Obj.getHeader64().filetype == MachO::MH_KEXT_BUNDLE &&
10310b57cec5SDimitry Andric         SegmentName == "__TEXT_EXEC" && SectionName == "__text")
10320b57cec5SDimitry Andric       return 't';
10330b57cec5SDimitry Andric     if (SegmentName == "__TEXT" && SectionName == "__text")
10340b57cec5SDimitry Andric       return 't';
10350b57cec5SDimitry Andric     if (SegmentName == "__DATA" && SectionName == "__data")
10360b57cec5SDimitry Andric       return 'd';
10370b57cec5SDimitry Andric     if (SegmentName == "__DATA" && SectionName == "__bss")
10380b57cec5SDimitry Andric       return 'b';
10390b57cec5SDimitry Andric     return 's';
10400b57cec5SDimitry Andric   }
10410b57cec5SDimitry Andric   }
10420b57cec5SDimitry Andric 
10430b57cec5SDimitry Andric   return '?';
10440b57cec5SDimitry Andric }
10450b57cec5SDimitry Andric 
10465ffd83dbSDimitry Andric static char getSymbolNMTypeChar(TapiFile &Obj, basic_symbol_iterator I) {
10475ffd83dbSDimitry Andric   return 's';
10485ffd83dbSDimitry Andric }
10495ffd83dbSDimitry Andric 
10500b57cec5SDimitry Andric static char getSymbolNMTypeChar(WasmObjectFile &Obj, basic_symbol_iterator I) {
10515ffd83dbSDimitry Andric   uint32_t Flags = cantFail(I->getFlags());
10520b57cec5SDimitry Andric   if (Flags & SymbolRef::SF_Executable)
10530b57cec5SDimitry Andric     return 't';
10540b57cec5SDimitry Andric   return 'd';
10550b57cec5SDimitry Andric }
10560b57cec5SDimitry Andric 
10570b57cec5SDimitry Andric static char getSymbolNMTypeChar(IRObjectFile &Obj, basic_symbol_iterator I) {
10585ffd83dbSDimitry Andric   uint32_t Flags = cantFail(I->getFlags());
10590b57cec5SDimitry Andric   // FIXME: should we print 'b'? At the IR level we cannot be sure if this
10600b57cec5SDimitry Andric   // will be in bss or not, but we could approximate.
10610b57cec5SDimitry Andric   if (Flags & SymbolRef::SF_Executable)
10620b57cec5SDimitry Andric     return 't';
10630b57cec5SDimitry Andric   else if (Triple(Obj.getTargetTriple()).isOSDarwin() &&
10640b57cec5SDimitry Andric            (Flags & SymbolRef::SF_Const))
10650b57cec5SDimitry Andric     return 's';
10660b57cec5SDimitry Andric   else
10670b57cec5SDimitry Andric     return 'd';
10680b57cec5SDimitry Andric }
10690b57cec5SDimitry Andric 
10700b57cec5SDimitry Andric static bool isObject(SymbolicFile &Obj, basic_symbol_iterator I) {
1071e8d8bef9SDimitry Andric   return isa<ELFObjectFileBase>(&Obj) &&
1072e8d8bef9SDimitry Andric          elf_symbol_iterator(I)->getELFType() == ELF::STT_OBJECT;
10730b57cec5SDimitry Andric }
10740b57cec5SDimitry Andric 
10750b57cec5SDimitry Andric // For ELF object files, Set TypeName to the symbol typename, to be printed
10760b57cec5SDimitry Andric // in the 'Type' column of the SYSV format output.
10770b57cec5SDimitry Andric static StringRef getNMTypeName(SymbolicFile &Obj, basic_symbol_iterator I) {
10780b57cec5SDimitry Andric   if (isa<ELFObjectFileBase>(&Obj)) {
10790b57cec5SDimitry Andric     elf_symbol_iterator SymI(I);
10800b57cec5SDimitry Andric     return SymI->getELFTypeName();
10810b57cec5SDimitry Andric   }
10820b57cec5SDimitry Andric   return "";
10830b57cec5SDimitry Andric }
10840b57cec5SDimitry Andric 
10850b57cec5SDimitry Andric // Return Posix nm class type tag (single letter), but also set SecName and
10860b57cec5SDimitry Andric // section and name, to be used in format=sysv output.
10870b57cec5SDimitry Andric static char getNMSectionTagAndName(SymbolicFile &Obj, basic_symbol_iterator I,
10880b57cec5SDimitry Andric                                    StringRef &SecName) {
10895ffd83dbSDimitry Andric   // Symbol Flags have been checked in the caller.
10905ffd83dbSDimitry Andric   uint32_t Symflags = cantFail(I->getFlags());
10918bcb0991SDimitry Andric   if (ELFObjectFileBase *ELFObj = dyn_cast<ELFObjectFileBase>(&Obj)) {
10920b57cec5SDimitry Andric     if (Symflags & object::SymbolRef::SF_Absolute)
10930b57cec5SDimitry Andric       SecName = "*ABS*";
10940b57cec5SDimitry Andric     else if (Symflags & object::SymbolRef::SF_Common)
10950b57cec5SDimitry Andric       SecName = "*COM*";
10960b57cec5SDimitry Andric     else if (Symflags & object::SymbolRef::SF_Undefined)
10970b57cec5SDimitry Andric       SecName = "*UND*";
10980b57cec5SDimitry Andric     else {
10990b57cec5SDimitry Andric       elf_symbol_iterator SymI(I);
11000b57cec5SDimitry Andric       Expected<elf_section_iterator> SecIOrErr = SymI->getSection();
11010b57cec5SDimitry Andric       if (!SecIOrErr) {
11020b57cec5SDimitry Andric         consumeError(SecIOrErr.takeError());
11030b57cec5SDimitry Andric         return '?';
11040b57cec5SDimitry Andric       }
11058bcb0991SDimitry Andric 
11068bcb0991SDimitry Andric       if (*SecIOrErr == ELFObj->section_end())
11078bcb0991SDimitry Andric         return '?';
11088bcb0991SDimitry Andric 
11098bcb0991SDimitry Andric       Expected<StringRef> NameOrErr = (*SecIOrErr)->getName();
11108bcb0991SDimitry Andric       if (!NameOrErr) {
11118bcb0991SDimitry Andric         consumeError(NameOrErr.takeError());
11128bcb0991SDimitry Andric         return '?';
11138bcb0991SDimitry Andric       }
11148bcb0991SDimitry Andric       SecName = *NameOrErr;
11150b57cec5SDimitry Andric     }
11160b57cec5SDimitry Andric   }
11170b57cec5SDimitry Andric 
1118fe6060f1SDimitry Andric   if (Symflags & object::SymbolRef::SF_Undefined) {
1119fe6060f1SDimitry Andric     if (isa<MachOObjectFile>(Obj) || !(Symflags & object::SymbolRef::SF_Weak))
11200b57cec5SDimitry Andric       return 'U';
1121fe6060f1SDimitry Andric     return isObject(Obj, I) ? 'v' : 'w';
1122fe6060f1SDimitry Andric   }
1123fe6060f1SDimitry Andric   if (isa<ELFObjectFileBase>(&Obj))
1124fe6060f1SDimitry Andric     if (ELFSymbolRef(*I).getELFType() == ELF::STT_GNU_IFUNC)
1125fe6060f1SDimitry Andric       return 'i';
1126fe6060f1SDimitry Andric   if (!isa<MachOObjectFile>(Obj) && (Symflags & object::SymbolRef::SF_Weak))
1127fe6060f1SDimitry Andric     return isObject(Obj, I) ? 'V' : 'W';
11280b57cec5SDimitry Andric 
11290b57cec5SDimitry Andric   if (Symflags & object::SymbolRef::SF_Common)
11300b57cec5SDimitry Andric     return 'C';
11310b57cec5SDimitry Andric 
11320b57cec5SDimitry Andric   char Ret = '?';
11330b57cec5SDimitry Andric   if (Symflags & object::SymbolRef::SF_Absolute)
11340b57cec5SDimitry Andric     Ret = 'a';
11350b57cec5SDimitry Andric   else if (IRObjectFile *IR = dyn_cast<IRObjectFile>(&Obj))
11360b57cec5SDimitry Andric     Ret = getSymbolNMTypeChar(*IR, I);
11370b57cec5SDimitry Andric   else if (COFFObjectFile *COFF = dyn_cast<COFFObjectFile>(&Obj))
11380b57cec5SDimitry Andric     Ret = getSymbolNMTypeChar(*COFF, I);
113904eeddc0SDimitry Andric   else if (XCOFFObjectFile *XCOFF = dyn_cast<XCOFFObjectFile>(&Obj))
114004eeddc0SDimitry Andric     Ret = getSymbolNMTypeChar(*XCOFF, I);
11410b57cec5SDimitry Andric   else if (COFFImportFile *COFFImport = dyn_cast<COFFImportFile>(&Obj))
11420b57cec5SDimitry Andric     Ret = getSymbolNMTypeChar(*COFFImport);
11430b57cec5SDimitry Andric   else if (MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(&Obj))
11440b57cec5SDimitry Andric     Ret = getSymbolNMTypeChar(*MachO, I);
11450b57cec5SDimitry Andric   else if (WasmObjectFile *Wasm = dyn_cast<WasmObjectFile>(&Obj))
11460b57cec5SDimitry Andric     Ret = getSymbolNMTypeChar(*Wasm, I);
11475ffd83dbSDimitry Andric   else if (TapiFile *Tapi = dyn_cast<TapiFile>(&Obj))
11485ffd83dbSDimitry Andric     Ret = getSymbolNMTypeChar(*Tapi, I);
1149480093f4SDimitry Andric   else if (ELFObjectFileBase *ELF = dyn_cast<ELFObjectFileBase>(&Obj)) {
1150480093f4SDimitry Andric     Ret = getSymbolNMTypeChar(*ELF, I);
1151480093f4SDimitry Andric     if (ELFSymbolRef(*I).getBinding() == ELF::STB_GNU_UNIQUE)
1152480093f4SDimitry Andric       return Ret;
1153480093f4SDimitry Andric   } else
1154480093f4SDimitry Andric     llvm_unreachable("unknown binary format");
11550b57cec5SDimitry Andric 
11560b57cec5SDimitry Andric   if (!(Symflags & object::SymbolRef::SF_Global))
11570b57cec5SDimitry Andric     return Ret;
11580b57cec5SDimitry Andric 
11590b57cec5SDimitry Andric   return toupper(Ret);
11600b57cec5SDimitry Andric }
11610b57cec5SDimitry Andric 
11620b57cec5SDimitry Andric // getNsectForSegSect() is used to implement the Mach-O "-s segname sectname"
11630b57cec5SDimitry Andric // option to dump only those symbols from that section in a Mach-O file.
116481ad6265SDimitry Andric // It is called once for each Mach-O file from getSymbolNamesFromObject()
11650b57cec5SDimitry Andric // to get the section number for that named section from the command line
11660b57cec5SDimitry Andric // arguments. It returns the section number for that section in the Mach-O
11670b57cec5SDimitry Andric // file or zero it is not present.
11680b57cec5SDimitry Andric static unsigned getNsectForSegSect(MachOObjectFile *Obj) {
11690b57cec5SDimitry Andric   unsigned Nsect = 1;
11700b57cec5SDimitry Andric   for (auto &S : Obj->sections()) {
11710b57cec5SDimitry Andric     DataRefImpl Ref = S.getRawDataRefImpl();
11720b57cec5SDimitry Andric     StringRef SectionName;
11730b57cec5SDimitry Andric     if (Expected<StringRef> NameOrErr = Obj->getSectionName(Ref))
11740b57cec5SDimitry Andric       SectionName = *NameOrErr;
11750b57cec5SDimitry Andric     StringRef SegmentName = Obj->getSectionFinalSegmentName(Ref);
11760b57cec5SDimitry Andric     if (SegmentName == SegSect[0] && SectionName == SegSect[1])
11770b57cec5SDimitry Andric       return Nsect;
11780b57cec5SDimitry Andric     Nsect++;
11790b57cec5SDimitry Andric   }
11800b57cec5SDimitry Andric   return 0;
11810b57cec5SDimitry Andric }
11820b57cec5SDimitry Andric 
11830b57cec5SDimitry Andric // getNsectInMachO() is used to implement the Mach-O "-s segname sectname"
11840b57cec5SDimitry Andric // option to dump only those symbols from that section in a Mach-O file.
11850b57cec5SDimitry Andric // It is called once for each symbol in a Mach-O file from
118681ad6265SDimitry Andric // getSymbolNamesFromObject() and returns the section number for that symbol
11870b57cec5SDimitry Andric // if it is in a section, else it returns 0.
11880b57cec5SDimitry Andric static unsigned getNsectInMachO(MachOObjectFile &Obj, BasicSymbolRef Sym) {
11890b57cec5SDimitry Andric   DataRefImpl Symb = Sym.getRawDataRefImpl();
11900b57cec5SDimitry Andric   if (Obj.is64Bit()) {
11910b57cec5SDimitry Andric     MachO::nlist_64 STE = Obj.getSymbol64TableEntry(Symb);
11920b57cec5SDimitry Andric     return (STE.n_type & MachO::N_TYPE) == MachO::N_SECT ? STE.n_sect : 0;
11930b57cec5SDimitry Andric   }
11940b57cec5SDimitry Andric   MachO::nlist STE = Obj.getSymbolTableEntry(Symb);
11950b57cec5SDimitry Andric   return (STE.n_type & MachO::N_TYPE) == MachO::N_SECT ? STE.n_sect : 0;
11960b57cec5SDimitry Andric }
11970b57cec5SDimitry Andric 
119881ad6265SDimitry Andric static void dumpSymbolsFromDLInfoMachO(MachOObjectFile &MachO,
119981ad6265SDimitry Andric                                        std::vector<NMSymbol> &SymbolList) {
1200e8d8bef9SDimitry Andric   size_t I = SymbolList.size();
12010b57cec5SDimitry Andric   std::string ExportsNameBuffer;
12020b57cec5SDimitry Andric   raw_string_ostream EOS(ExportsNameBuffer);
12030b57cec5SDimitry Andric   std::string BindsNameBuffer;
12040b57cec5SDimitry Andric   raw_string_ostream BOS(BindsNameBuffer);
12050b57cec5SDimitry Andric   std::string LazysNameBuffer;
12060b57cec5SDimitry Andric   raw_string_ostream LOS(LazysNameBuffer);
12070b57cec5SDimitry Andric   std::string WeaksNameBuffer;
12080b57cec5SDimitry Andric   raw_string_ostream WOS(WeaksNameBuffer);
12090b57cec5SDimitry Andric   std::string FunctionStartsNameBuffer;
12100b57cec5SDimitry Andric   raw_string_ostream FOS(FunctionStartsNameBuffer);
1211e8d8bef9SDimitry Andric 
12120b57cec5SDimitry Andric   MachO::mach_header H;
12130b57cec5SDimitry Andric   MachO::mach_header_64 H_64;
12140b57cec5SDimitry Andric   uint32_t HFlags = 0;
1215e8d8bef9SDimitry Andric   if (MachO.is64Bit()) {
1216e8d8bef9SDimitry Andric     H_64 = MachO.MachOObjectFile::getHeader64();
12170b57cec5SDimitry Andric     HFlags = H_64.flags;
12180b57cec5SDimitry Andric   } else {
1219e8d8bef9SDimitry Andric     H = MachO.MachOObjectFile::getHeader();
12200b57cec5SDimitry Andric     HFlags = H.flags;
12210b57cec5SDimitry Andric   }
12220b57cec5SDimitry Andric   uint64_t BaseSegmentAddress = 0;
1223e8d8bef9SDimitry Andric   for (const auto &Command : MachO.load_commands()) {
12240b57cec5SDimitry Andric     if (Command.C.cmd == MachO::LC_SEGMENT) {
1225e8d8bef9SDimitry Andric       MachO::segment_command Seg = MachO.getSegmentLoadCommand(Command);
12260b57cec5SDimitry Andric       if (Seg.fileoff == 0 && Seg.filesize != 0) {
12270b57cec5SDimitry Andric         BaseSegmentAddress = Seg.vmaddr;
12280b57cec5SDimitry Andric         break;
12290b57cec5SDimitry Andric       }
12300b57cec5SDimitry Andric     } else if (Command.C.cmd == MachO::LC_SEGMENT_64) {
1231e8d8bef9SDimitry Andric       MachO::segment_command_64 Seg = MachO.getSegment64LoadCommand(Command);
12320b57cec5SDimitry Andric       if (Seg.fileoff == 0 && Seg.filesize != 0) {
12330b57cec5SDimitry Andric         BaseSegmentAddress = Seg.vmaddr;
12340b57cec5SDimitry Andric         break;
12350b57cec5SDimitry Andric       }
12360b57cec5SDimitry Andric     }
12370b57cec5SDimitry Andric   }
12380b57cec5SDimitry Andric   if (DyldInfoOnly || AddDyldInfo ||
12390b57cec5SDimitry Andric       HFlags & MachO::MH_NLIST_OUTOFSYNC_WITH_DYLDINFO) {
12400b57cec5SDimitry Andric     unsigned ExportsAdded = 0;
12410b57cec5SDimitry Andric     Error Err = Error::success();
1242e8d8bef9SDimitry Andric     for (const llvm::object::ExportEntry &Entry : MachO.exports(Err)) {
12430b57cec5SDimitry Andric       bool found = false;
12440b57cec5SDimitry Andric       bool ReExport = false;
12450b57cec5SDimitry Andric       if (!DyldInfoOnly) {
12460b57cec5SDimitry Andric         for (const NMSymbol &S : SymbolList)
12470b57cec5SDimitry Andric           if (S.Address == Entry.address() + BaseSegmentAddress &&
12480b57cec5SDimitry Andric               S.Name == Entry.name()) {
12490b57cec5SDimitry Andric             found = true;
12500b57cec5SDimitry Andric             break;
12510b57cec5SDimitry Andric           }
12520b57cec5SDimitry Andric       }
12530b57cec5SDimitry Andric       if (!found) {
12540b57cec5SDimitry Andric         NMSymbol S = {};
12550b57cec5SDimitry Andric         S.Address = Entry.address() + BaseSegmentAddress;
12560b57cec5SDimitry Andric         S.Size = 0;
12570b57cec5SDimitry Andric         S.TypeChar = '\0';
1258e8d8bef9SDimitry Andric         S.Name = Entry.name().str();
12590b57cec5SDimitry Andric         // There is no symbol in the nlist symbol table for this so we set
12600b57cec5SDimitry Andric         // Sym effectivly to null and the rest of code in here must test for
12610b57cec5SDimitry Andric         // it and not do things like Sym.getFlags() for it.
12620b57cec5SDimitry Andric         S.Sym = BasicSymbolRef();
12630b57cec5SDimitry Andric         S.SymFlags = SymbolRef::SF_Global;
12640b57cec5SDimitry Andric         S.Section = SectionRef();
12650b57cec5SDimitry Andric         S.NType = 0;
12660b57cec5SDimitry Andric         S.NSect = 0;
12670b57cec5SDimitry Andric         S.NDesc = 0;
12680b57cec5SDimitry Andric 
12690b57cec5SDimitry Andric         uint64_t EFlags = Entry.flags();
12700b57cec5SDimitry Andric         bool Abs = ((EFlags & MachO::EXPORT_SYMBOL_FLAGS_KIND_MASK) ==
12710b57cec5SDimitry Andric                     MachO::EXPORT_SYMBOL_FLAGS_KIND_ABSOLUTE);
1272e8d8bef9SDimitry Andric         bool Resolver = (EFlags & MachO::EXPORT_SYMBOL_FLAGS_STUB_AND_RESOLVER);
12730b57cec5SDimitry Andric         ReExport = (EFlags & MachO::EXPORT_SYMBOL_FLAGS_REEXPORT);
12740b57cec5SDimitry Andric         bool WeakDef = (EFlags & MachO::EXPORT_SYMBOL_FLAGS_WEAK_DEFINITION);
12750b57cec5SDimitry Andric         if (WeakDef)
12760b57cec5SDimitry Andric           S.NDesc |= MachO::N_WEAK_DEF;
12770b57cec5SDimitry Andric         if (Abs) {
12780b57cec5SDimitry Andric           S.NType = MachO::N_EXT | MachO::N_ABS;
12790b57cec5SDimitry Andric           S.TypeChar = 'A';
12800b57cec5SDimitry Andric         } else if (ReExport) {
12810b57cec5SDimitry Andric           S.NType = MachO::N_EXT | MachO::N_INDR;
12820b57cec5SDimitry Andric           S.TypeChar = 'I';
12830b57cec5SDimitry Andric         } else {
12840b57cec5SDimitry Andric           S.NType = MachO::N_EXT | MachO::N_SECT;
12850b57cec5SDimitry Andric           if (Resolver) {
12860b57cec5SDimitry Andric             S.Address = Entry.other() + BaseSegmentAddress;
1287e8d8bef9SDimitry Andric             if ((S.Address & 1) != 0 && !MachO.is64Bit() &&
1288e8d8bef9SDimitry Andric                 H.cputype == MachO::CPU_TYPE_ARM) {
12890b57cec5SDimitry Andric               S.Address &= ~1LL;
12900b57cec5SDimitry Andric               S.NDesc |= MachO::N_ARM_THUMB_DEF;
12910b57cec5SDimitry Andric             }
12920b57cec5SDimitry Andric           } else {
12930b57cec5SDimitry Andric             S.Address = Entry.address() + BaseSegmentAddress;
12940b57cec5SDimitry Andric           }
12950b57cec5SDimitry Andric           StringRef SegmentName = StringRef();
12960b57cec5SDimitry Andric           StringRef SectionName = StringRef();
1297e8d8bef9SDimitry Andric           for (const SectionRef &Section : MachO.sections()) {
12980b57cec5SDimitry Andric             S.NSect++;
12998bcb0991SDimitry Andric 
13008bcb0991SDimitry Andric             if (Expected<StringRef> NameOrErr = Section.getName())
13018bcb0991SDimitry Andric               SectionName = *NameOrErr;
13028bcb0991SDimitry Andric             else
13038bcb0991SDimitry Andric               consumeError(NameOrErr.takeError());
13048bcb0991SDimitry Andric 
1305e8d8bef9SDimitry Andric             SegmentName =
1306e8d8bef9SDimitry Andric                 MachO.getSectionFinalSegmentName(Section.getRawDataRefImpl());
13070b57cec5SDimitry Andric             if (S.Address >= Section.getAddress() &&
13080b57cec5SDimitry Andric                 S.Address < Section.getAddress() + Section.getSize()) {
13090b57cec5SDimitry Andric               S.Section = Section;
13100b57cec5SDimitry Andric               break;
13110b57cec5SDimitry Andric             } else if (Entry.name() == "__mh_execute_header" &&
13120b57cec5SDimitry Andric                        SegmentName == "__TEXT" && SectionName == "__text") {
13130b57cec5SDimitry Andric               S.Section = Section;
13140b57cec5SDimitry Andric               S.NDesc |= MachO::REFERENCED_DYNAMICALLY;
13150b57cec5SDimitry Andric               break;
13160b57cec5SDimitry Andric             }
13170b57cec5SDimitry Andric           }
13180b57cec5SDimitry Andric           if (SegmentName == "__TEXT" && SectionName == "__text")
13190b57cec5SDimitry Andric             S.TypeChar = 'T';
13200b57cec5SDimitry Andric           else if (SegmentName == "__DATA" && SectionName == "__data")
13210b57cec5SDimitry Andric             S.TypeChar = 'D';
13220b57cec5SDimitry Andric           else if (SegmentName == "__DATA" && SectionName == "__bss")
13230b57cec5SDimitry Andric             S.TypeChar = 'B';
13240b57cec5SDimitry Andric           else
13250b57cec5SDimitry Andric             S.TypeChar = 'S';
13260b57cec5SDimitry Andric         }
13270b57cec5SDimitry Andric         SymbolList.push_back(S);
13280b57cec5SDimitry Andric 
13290b57cec5SDimitry Andric         EOS << Entry.name();
13300b57cec5SDimitry Andric         EOS << '\0';
13310b57cec5SDimitry Andric         ExportsAdded++;
13320b57cec5SDimitry Andric 
13330b57cec5SDimitry Andric         // For ReExports there are a two more things to do, first add the
13340b57cec5SDimitry Andric         // indirect name and second create the undefined symbol using the
13350b57cec5SDimitry Andric         // referened dynamic library.
13360b57cec5SDimitry Andric         if (ReExport) {
13370b57cec5SDimitry Andric 
13380b57cec5SDimitry Andric           // Add the indirect name.
13390b57cec5SDimitry Andric           if (Entry.otherName().empty())
13400b57cec5SDimitry Andric             EOS << Entry.name();
13410b57cec5SDimitry Andric           else
13420b57cec5SDimitry Andric             EOS << Entry.otherName();
13430b57cec5SDimitry Andric           EOS << '\0';
13440b57cec5SDimitry Andric 
13450b57cec5SDimitry Andric           // Now create the undefined symbol using the referened dynamic
13460b57cec5SDimitry Andric           // library.
13470b57cec5SDimitry Andric           NMSymbol U = {};
13480b57cec5SDimitry Andric           U.Address = 0;
13490b57cec5SDimitry Andric           U.Size = 0;
13500b57cec5SDimitry Andric           U.TypeChar = 'U';
13510b57cec5SDimitry Andric           if (Entry.otherName().empty())
1352e8d8bef9SDimitry Andric             U.Name = Entry.name().str();
13530b57cec5SDimitry Andric           else
1354e8d8bef9SDimitry Andric             U.Name = Entry.otherName().str();
13550b57cec5SDimitry Andric           // Again there is no symbol in the nlist symbol table for this so
13560b57cec5SDimitry Andric           // we set Sym effectivly to null and the rest of code in here must
13570b57cec5SDimitry Andric           // test for it and not do things like Sym.getFlags() for it.
13580b57cec5SDimitry Andric           U.Sym = BasicSymbolRef();
13590b57cec5SDimitry Andric           U.SymFlags = SymbolRef::SF_Global | SymbolRef::SF_Undefined;
13600b57cec5SDimitry Andric           U.Section = SectionRef();
13610b57cec5SDimitry Andric           U.NType = MachO::N_EXT | MachO::N_UNDF;
13620b57cec5SDimitry Andric           U.NSect = 0;
13630b57cec5SDimitry Andric           U.NDesc = 0;
13640b57cec5SDimitry Andric           // The library ordinal for this undefined symbol is in the export
13650b57cec5SDimitry Andric           // trie Entry.other().
13660b57cec5SDimitry Andric           MachO::SET_LIBRARY_ORDINAL(U.NDesc, Entry.other());
13670b57cec5SDimitry Andric           SymbolList.push_back(U);
13680b57cec5SDimitry Andric 
13690b57cec5SDimitry Andric           // Finally add the undefined symbol's name.
13700b57cec5SDimitry Andric           if (Entry.otherName().empty())
13710b57cec5SDimitry Andric             EOS << Entry.name();
13720b57cec5SDimitry Andric           else
13730b57cec5SDimitry Andric             EOS << Entry.otherName();
13740b57cec5SDimitry Andric           EOS << '\0';
13750b57cec5SDimitry Andric           ExportsAdded++;
13760b57cec5SDimitry Andric         }
13770b57cec5SDimitry Andric       }
13780b57cec5SDimitry Andric     }
13790b57cec5SDimitry Andric     if (Err)
1380e8d8bef9SDimitry Andric       error(std::move(Err), MachO.getFileName());
13810b57cec5SDimitry Andric     // Set the symbol names and indirect names for the added symbols.
13820b57cec5SDimitry Andric     if (ExportsAdded) {
13830b57cec5SDimitry Andric       EOS.flush();
13840b57cec5SDimitry Andric       const char *Q = ExportsNameBuffer.c_str();
13850b57cec5SDimitry Andric       for (unsigned K = 0; K < ExportsAdded; K++) {
13860b57cec5SDimitry Andric         SymbolList[I].Name = Q;
13870b57cec5SDimitry Andric         Q += strlen(Q) + 1;
13880b57cec5SDimitry Andric         if (SymbolList[I].TypeChar == 'I') {
13890b57cec5SDimitry Andric           SymbolList[I].IndirectName = Q;
13900b57cec5SDimitry Andric           Q += strlen(Q) + 1;
13910b57cec5SDimitry Andric         }
13920b57cec5SDimitry Andric         I++;
13930b57cec5SDimitry Andric       }
13940b57cec5SDimitry Andric     }
13950b57cec5SDimitry Andric 
13960b57cec5SDimitry Andric     // Add the undefined symbols from the bind entries.
13970b57cec5SDimitry Andric     unsigned BindsAdded = 0;
13980b57cec5SDimitry Andric     Error BErr = Error::success();
13990b57cec5SDimitry Andric     StringRef LastSymbolName = StringRef();
1400e8d8bef9SDimitry Andric     for (const llvm::object::MachOBindEntry &Entry : MachO.bindTable(BErr)) {
14010b57cec5SDimitry Andric       bool found = false;
14020b57cec5SDimitry Andric       if (LastSymbolName == Entry.symbolName())
14030b57cec5SDimitry Andric         found = true;
14040b57cec5SDimitry Andric       else if (!DyldInfoOnly) {
14050b57cec5SDimitry Andric         for (unsigned J = 0; J < SymbolList.size() && !found; ++J) {
14060b57cec5SDimitry Andric           if (SymbolList[J].Name == Entry.symbolName())
14070b57cec5SDimitry Andric             found = true;
14080b57cec5SDimitry Andric         }
14090b57cec5SDimitry Andric       }
14100b57cec5SDimitry Andric       if (!found) {
14110b57cec5SDimitry Andric         LastSymbolName = Entry.symbolName();
14120b57cec5SDimitry Andric         NMSymbol B = {};
14130b57cec5SDimitry Andric         B.Address = 0;
14140b57cec5SDimitry Andric         B.Size = 0;
14150b57cec5SDimitry Andric         B.TypeChar = 'U';
14160b57cec5SDimitry Andric         // There is no symbol in the nlist symbol table for this so we set
14170b57cec5SDimitry Andric         // Sym effectivly to null and the rest of code in here must test for
14180b57cec5SDimitry Andric         // it and not do things like Sym.getFlags() for it.
14190b57cec5SDimitry Andric         B.Sym = BasicSymbolRef();
14200b57cec5SDimitry Andric         B.SymFlags = SymbolRef::SF_Global | SymbolRef::SF_Undefined;
14210b57cec5SDimitry Andric         B.NType = MachO::N_EXT | MachO::N_UNDF;
14220b57cec5SDimitry Andric         B.NSect = 0;
14230b57cec5SDimitry Andric         B.NDesc = 0;
14240b57cec5SDimitry Andric         MachO::SET_LIBRARY_ORDINAL(B.NDesc, Entry.ordinal());
1425e8d8bef9SDimitry Andric         B.Name = Entry.symbolName().str();
14260b57cec5SDimitry Andric         SymbolList.push_back(B);
14270b57cec5SDimitry Andric         BOS << Entry.symbolName();
14280b57cec5SDimitry Andric         BOS << '\0';
14290b57cec5SDimitry Andric         BindsAdded++;
14300b57cec5SDimitry Andric       }
14310b57cec5SDimitry Andric     }
14320b57cec5SDimitry Andric     if (BErr)
1433e8d8bef9SDimitry Andric       error(std::move(BErr), MachO.getFileName());
14340b57cec5SDimitry Andric     // Set the symbol names and indirect names for the added symbols.
14350b57cec5SDimitry Andric     if (BindsAdded) {
14360b57cec5SDimitry Andric       BOS.flush();
14370b57cec5SDimitry Andric       const char *Q = BindsNameBuffer.c_str();
14380b57cec5SDimitry Andric       for (unsigned K = 0; K < BindsAdded; K++) {
14390b57cec5SDimitry Andric         SymbolList[I].Name = Q;
14400b57cec5SDimitry Andric         Q += strlen(Q) + 1;
14410b57cec5SDimitry Andric         if (SymbolList[I].TypeChar == 'I') {
14420b57cec5SDimitry Andric           SymbolList[I].IndirectName = Q;
14430b57cec5SDimitry Andric           Q += strlen(Q) + 1;
14440b57cec5SDimitry Andric         }
14450b57cec5SDimitry Andric         I++;
14460b57cec5SDimitry Andric       }
14470b57cec5SDimitry Andric     }
14480b57cec5SDimitry Andric 
14490b57cec5SDimitry Andric     // Add the undefined symbols from the lazy bind entries.
14500b57cec5SDimitry Andric     unsigned LazysAdded = 0;
14510b57cec5SDimitry Andric     Error LErr = Error::success();
14520b57cec5SDimitry Andric     LastSymbolName = StringRef();
14530b57cec5SDimitry Andric     for (const llvm::object::MachOBindEntry &Entry :
1454e8d8bef9SDimitry Andric          MachO.lazyBindTable(LErr)) {
14550b57cec5SDimitry Andric       bool found = false;
14560b57cec5SDimitry Andric       if (LastSymbolName == Entry.symbolName())
14570b57cec5SDimitry Andric         found = true;
14580b57cec5SDimitry Andric       else {
14590b57cec5SDimitry Andric         // Here we must check to see it this symbol is already in the
14600b57cec5SDimitry Andric         // SymbolList as it might have already have been added above via a
14610b57cec5SDimitry Andric         // non-lazy (bind) entry.
14620b57cec5SDimitry Andric         for (unsigned J = 0; J < SymbolList.size() && !found; ++J) {
14630b57cec5SDimitry Andric           if (SymbolList[J].Name == Entry.symbolName())
14640b57cec5SDimitry Andric             found = true;
14650b57cec5SDimitry Andric         }
14660b57cec5SDimitry Andric       }
14670b57cec5SDimitry Andric       if (!found) {
14680b57cec5SDimitry Andric         LastSymbolName = Entry.symbolName();
14690b57cec5SDimitry Andric         NMSymbol L = {};
1470e8d8bef9SDimitry Andric         L.Name = Entry.symbolName().str();
14710b57cec5SDimitry Andric         L.Address = 0;
14720b57cec5SDimitry Andric         L.Size = 0;
14730b57cec5SDimitry Andric         L.TypeChar = 'U';
14740b57cec5SDimitry Andric         // There is no symbol in the nlist symbol table for this so we set
14750b57cec5SDimitry Andric         // Sym effectivly to null and the rest of code in here must test for
14760b57cec5SDimitry Andric         // it and not do things like Sym.getFlags() for it.
14770b57cec5SDimitry Andric         L.Sym = BasicSymbolRef();
14780b57cec5SDimitry Andric         L.SymFlags = SymbolRef::SF_Global | SymbolRef::SF_Undefined;
14790b57cec5SDimitry Andric         L.NType = MachO::N_EXT | MachO::N_UNDF;
14800b57cec5SDimitry Andric         L.NSect = 0;
14810b57cec5SDimitry Andric         // The REFERENCE_FLAG_UNDEFINED_LAZY is no longer used but here it
14820b57cec5SDimitry Andric         // makes sence since we are creating this from a lazy bind entry.
14830b57cec5SDimitry Andric         L.NDesc = MachO::REFERENCE_FLAG_UNDEFINED_LAZY;
14840b57cec5SDimitry Andric         MachO::SET_LIBRARY_ORDINAL(L.NDesc, Entry.ordinal());
14850b57cec5SDimitry Andric         SymbolList.push_back(L);
14860b57cec5SDimitry Andric         LOS << Entry.symbolName();
14870b57cec5SDimitry Andric         LOS << '\0';
14880b57cec5SDimitry Andric         LazysAdded++;
14890b57cec5SDimitry Andric       }
14900b57cec5SDimitry Andric     }
14910b57cec5SDimitry Andric     if (LErr)
1492e8d8bef9SDimitry Andric       error(std::move(LErr), MachO.getFileName());
14930b57cec5SDimitry Andric     // Set the symbol names and indirect names for the added symbols.
14940b57cec5SDimitry Andric     if (LazysAdded) {
14950b57cec5SDimitry Andric       LOS.flush();
14960b57cec5SDimitry Andric       const char *Q = LazysNameBuffer.c_str();
14970b57cec5SDimitry Andric       for (unsigned K = 0; K < LazysAdded; K++) {
14980b57cec5SDimitry Andric         SymbolList[I].Name = Q;
14990b57cec5SDimitry Andric         Q += strlen(Q) + 1;
15000b57cec5SDimitry Andric         if (SymbolList[I].TypeChar == 'I') {
15010b57cec5SDimitry Andric           SymbolList[I].IndirectName = Q;
15020b57cec5SDimitry Andric           Q += strlen(Q) + 1;
15030b57cec5SDimitry Andric         }
15040b57cec5SDimitry Andric         I++;
15050b57cec5SDimitry Andric       }
15060b57cec5SDimitry Andric     }
15070b57cec5SDimitry Andric 
15080b57cec5SDimitry Andric     // Add the undefineds symbol from the weak bind entries which are not
15090b57cec5SDimitry Andric     // strong symbols.
15100b57cec5SDimitry Andric     unsigned WeaksAdded = 0;
15110b57cec5SDimitry Andric     Error WErr = Error::success();
15120b57cec5SDimitry Andric     LastSymbolName = StringRef();
15130b57cec5SDimitry Andric     for (const llvm::object::MachOBindEntry &Entry :
1514e8d8bef9SDimitry Andric          MachO.weakBindTable(WErr)) {
15150b57cec5SDimitry Andric       bool found = false;
15160b57cec5SDimitry Andric       unsigned J = 0;
15170b57cec5SDimitry Andric       if (LastSymbolName == Entry.symbolName() ||
15180b57cec5SDimitry Andric           Entry.flags() & MachO::BIND_SYMBOL_FLAGS_NON_WEAK_DEFINITION) {
15190b57cec5SDimitry Andric         found = true;
15200b57cec5SDimitry Andric       } else {
15210b57cec5SDimitry Andric         for (J = 0; J < SymbolList.size() && !found; ++J) {
15220b57cec5SDimitry Andric           if (SymbolList[J].Name == Entry.symbolName()) {
15230b57cec5SDimitry Andric             found = true;
15240b57cec5SDimitry Andric             break;
15250b57cec5SDimitry Andric           }
15260b57cec5SDimitry Andric         }
15270b57cec5SDimitry Andric       }
15280b57cec5SDimitry Andric       if (!found) {
15290b57cec5SDimitry Andric         LastSymbolName = Entry.symbolName();
1530e8d8bef9SDimitry Andric         NMSymbol W = {};
1531e8d8bef9SDimitry Andric         W.Name = Entry.symbolName().str();
15320b57cec5SDimitry Andric         W.Address = 0;
15330b57cec5SDimitry Andric         W.Size = 0;
15340b57cec5SDimitry Andric         W.TypeChar = 'U';
15350b57cec5SDimitry Andric         // There is no symbol in the nlist symbol table for this so we set
15360b57cec5SDimitry Andric         // Sym effectivly to null and the rest of code in here must test for
15370b57cec5SDimitry Andric         // it and not do things like Sym.getFlags() for it.
15380b57cec5SDimitry Andric         W.Sym = BasicSymbolRef();
15390b57cec5SDimitry Andric         W.SymFlags = SymbolRef::SF_Global | SymbolRef::SF_Undefined;
15400b57cec5SDimitry Andric         W.NType = MachO::N_EXT | MachO::N_UNDF;
15410b57cec5SDimitry Andric         W.NSect = 0;
15420b57cec5SDimitry Andric         // Odd that we are using N_WEAK_DEF on an undefined symbol but that is
15430b57cec5SDimitry Andric         // what is created in this case by the linker when there are real
15440b57cec5SDimitry Andric         // symbols in the nlist structs.
15450b57cec5SDimitry Andric         W.NDesc = MachO::N_WEAK_DEF;
15460b57cec5SDimitry Andric         SymbolList.push_back(W);
15470b57cec5SDimitry Andric         WOS << Entry.symbolName();
15480b57cec5SDimitry Andric         WOS << '\0';
15490b57cec5SDimitry Andric         WeaksAdded++;
15500b57cec5SDimitry Andric       } else {
15510b57cec5SDimitry Andric         // This is the case the symbol was previously been found and it could
15520b57cec5SDimitry Andric         // have been added from a bind or lazy bind symbol.  If so and not
15530b57cec5SDimitry Andric         // a definition also mark it as weak.
15540b57cec5SDimitry Andric         if (SymbolList[J].TypeChar == 'U')
15550b57cec5SDimitry Andric           // See comment above about N_WEAK_DEF.
15560b57cec5SDimitry Andric           SymbolList[J].NDesc |= MachO::N_WEAK_DEF;
15570b57cec5SDimitry Andric       }
15580b57cec5SDimitry Andric     }
15590b57cec5SDimitry Andric     if (WErr)
1560e8d8bef9SDimitry Andric       error(std::move(WErr), MachO.getFileName());
15610b57cec5SDimitry Andric     // Set the symbol names and indirect names for the added symbols.
15620b57cec5SDimitry Andric     if (WeaksAdded) {
15630b57cec5SDimitry Andric       WOS.flush();
15640b57cec5SDimitry Andric       const char *Q = WeaksNameBuffer.c_str();
15650b57cec5SDimitry Andric       for (unsigned K = 0; K < WeaksAdded; K++) {
15660b57cec5SDimitry Andric         SymbolList[I].Name = Q;
15670b57cec5SDimitry Andric         Q += strlen(Q) + 1;
15680b57cec5SDimitry Andric         if (SymbolList[I].TypeChar == 'I') {
15690b57cec5SDimitry Andric           SymbolList[I].IndirectName = Q;
15700b57cec5SDimitry Andric           Q += strlen(Q) + 1;
15710b57cec5SDimitry Andric         }
15720b57cec5SDimitry Andric         I++;
15730b57cec5SDimitry Andric       }
15740b57cec5SDimitry Andric     }
15750b57cec5SDimitry Andric 
15760b57cec5SDimitry Andric     // Trying adding symbol from the function starts table and LC_MAIN entry
15770b57cec5SDimitry Andric     // point.
15780b57cec5SDimitry Andric     SmallVector<uint64_t, 8> FoundFns;
15790b57cec5SDimitry Andric     uint64_t lc_main_offset = UINT64_MAX;
1580e8d8bef9SDimitry Andric     for (const auto &Command : MachO.load_commands()) {
15810b57cec5SDimitry Andric       if (Command.C.cmd == MachO::LC_FUNCTION_STARTS) {
15820b57cec5SDimitry Andric         // We found a function starts segment, parse the addresses for
15830b57cec5SDimitry Andric         // consumption.
15840b57cec5SDimitry Andric         MachO::linkedit_data_command LLC =
1585e8d8bef9SDimitry Andric             MachO.getLinkeditDataLoadCommand(Command);
15860b57cec5SDimitry Andric 
1587e8d8bef9SDimitry Andric         MachO.ReadULEB128s(LLC.dataoff, FoundFns);
15880b57cec5SDimitry Andric       } else if (Command.C.cmd == MachO::LC_MAIN) {
1589e8d8bef9SDimitry Andric         MachO::entry_point_command LCmain = MachO.getEntryPointCommand(Command);
15900b57cec5SDimitry Andric         lc_main_offset = LCmain.entryoff;
15910b57cec5SDimitry Andric       }
15920b57cec5SDimitry Andric     }
15930b57cec5SDimitry Andric     // See if these addresses are already in the symbol table.
15940b57cec5SDimitry Andric     unsigned FunctionStartsAdded = 0;
15950b57cec5SDimitry Andric     for (uint64_t f = 0; f < FoundFns.size(); f++) {
15960b57cec5SDimitry Andric       bool found = false;
15970b57cec5SDimitry Andric       for (unsigned J = 0; J < SymbolList.size() && !found; ++J) {
15980b57cec5SDimitry Andric         if (SymbolList[J].Address == FoundFns[f] + BaseSegmentAddress)
15990b57cec5SDimitry Andric           found = true;
16000b57cec5SDimitry Andric       }
16010b57cec5SDimitry Andric       // See this address is not already in the symbol table fake up an
16020b57cec5SDimitry Andric       // nlist for it.
16030b57cec5SDimitry Andric       if (!found) {
16040b57cec5SDimitry Andric         NMSymbol F = {};
16050b57cec5SDimitry Andric         F.Name = "<redacted function X>";
16060b57cec5SDimitry Andric         F.Address = FoundFns[f] + BaseSegmentAddress;
16070b57cec5SDimitry Andric         F.Size = 0;
16080b57cec5SDimitry Andric         // There is no symbol in the nlist symbol table for this so we set
16090b57cec5SDimitry Andric         // Sym effectivly to null and the rest of code in here must test for
16100b57cec5SDimitry Andric         // it and not do things like Sym.getFlags() for it.
16110b57cec5SDimitry Andric         F.Sym = BasicSymbolRef();
16120b57cec5SDimitry Andric         F.SymFlags = 0;
16130b57cec5SDimitry Andric         F.NType = MachO::N_SECT;
16140b57cec5SDimitry Andric         F.NSect = 0;
16150b57cec5SDimitry Andric         StringRef SegmentName = StringRef();
16160b57cec5SDimitry Andric         StringRef SectionName = StringRef();
1617e8d8bef9SDimitry Andric         for (const SectionRef &Section : MachO.sections()) {
16188bcb0991SDimitry Andric           if (Expected<StringRef> NameOrErr = Section.getName())
16198bcb0991SDimitry Andric             SectionName = *NameOrErr;
16208bcb0991SDimitry Andric           else
16218bcb0991SDimitry Andric             consumeError(NameOrErr.takeError());
16228bcb0991SDimitry Andric 
1623e8d8bef9SDimitry Andric           SegmentName =
1624e8d8bef9SDimitry Andric               MachO.getSectionFinalSegmentName(Section.getRawDataRefImpl());
16250b57cec5SDimitry Andric           F.NSect++;
16260b57cec5SDimitry Andric           if (F.Address >= Section.getAddress() &&
16270b57cec5SDimitry Andric               F.Address < Section.getAddress() + Section.getSize()) {
16280b57cec5SDimitry Andric             F.Section = Section;
16290b57cec5SDimitry Andric             break;
16300b57cec5SDimitry Andric           }
16310b57cec5SDimitry Andric         }
16320b57cec5SDimitry Andric         if (SegmentName == "__TEXT" && SectionName == "__text")
16330b57cec5SDimitry Andric           F.TypeChar = 't';
16340b57cec5SDimitry Andric         else if (SegmentName == "__DATA" && SectionName == "__data")
16350b57cec5SDimitry Andric           F.TypeChar = 'd';
16360b57cec5SDimitry Andric         else if (SegmentName == "__DATA" && SectionName == "__bss")
16370b57cec5SDimitry Andric           F.TypeChar = 'b';
16380b57cec5SDimitry Andric         else
16390b57cec5SDimitry Andric           F.TypeChar = 's';
16400b57cec5SDimitry Andric         F.NDesc = 0;
16410b57cec5SDimitry Andric         SymbolList.push_back(F);
16420b57cec5SDimitry Andric         if (FoundFns[f] == lc_main_offset)
16430b57cec5SDimitry Andric           FOS << "<redacted LC_MAIN>";
16440b57cec5SDimitry Andric         else
16450b57cec5SDimitry Andric           FOS << "<redacted function " << f << ">";
16460b57cec5SDimitry Andric         FOS << '\0';
16470b57cec5SDimitry Andric         FunctionStartsAdded++;
16480b57cec5SDimitry Andric       }
16490b57cec5SDimitry Andric     }
16500b57cec5SDimitry Andric     if (FunctionStartsAdded) {
16510b57cec5SDimitry Andric       FOS.flush();
16520b57cec5SDimitry Andric       const char *Q = FunctionStartsNameBuffer.c_str();
16530b57cec5SDimitry Andric       for (unsigned K = 0; K < FunctionStartsAdded; K++) {
16540b57cec5SDimitry Andric         SymbolList[I].Name = Q;
16550b57cec5SDimitry Andric         Q += strlen(Q) + 1;
16560b57cec5SDimitry Andric         if (SymbolList[I].TypeChar == 'I') {
16570b57cec5SDimitry Andric           SymbolList[I].IndirectName = Q;
16580b57cec5SDimitry Andric           Q += strlen(Q) + 1;
16590b57cec5SDimitry Andric         }
16600b57cec5SDimitry Andric         I++;
16610b57cec5SDimitry Andric       }
16620b57cec5SDimitry Andric     }
16630b57cec5SDimitry Andric   }
16640b57cec5SDimitry Andric }
16650b57cec5SDimitry Andric 
166681ad6265SDimitry Andric static bool shouldDump(SymbolicFile &Obj) {
166781ad6265SDimitry Andric   // The -X option is currently only implemented for XCOFF, ELF, and IR object
166881ad6265SDimitry Andric   // files. The option isn't fundamentally impossible with other formats, just
166981ad6265SDimitry Andric   // isn't implemented.
167081ad6265SDimitry Andric   if (!isa<XCOFFObjectFile>(Obj) && !isa<ELFObjectFileBase>(Obj) &&
167181ad6265SDimitry Andric       !isa<IRObjectFile>(Obj))
167281ad6265SDimitry Andric     return true;
167381ad6265SDimitry Andric 
167481ad6265SDimitry Andric   return isSymbolList64Bit(Obj) ? BitMode != BitModeTy::Bit32
167581ad6265SDimitry Andric                                 : BitMode != BitModeTy::Bit64;
167681ad6265SDimitry Andric }
167781ad6265SDimitry Andric 
167881ad6265SDimitry Andric static void getXCOFFExports(XCOFFObjectFile *XCOFFObj,
167981ad6265SDimitry Andric                             std::vector<NMSymbol> &SymbolList,
168081ad6265SDimitry Andric                             StringRef ArchiveName) {
168181ad6265SDimitry Andric   // Skip Shared object file.
168281ad6265SDimitry Andric   if (XCOFFObj->getFlags() & XCOFF::F_SHROBJ)
168381ad6265SDimitry Andric     return;
168481ad6265SDimitry Andric 
168581ad6265SDimitry Andric   for (SymbolRef Sym : XCOFFObj->symbols()) {
168681ad6265SDimitry Andric     // There is no visibility in old 32 bit XCOFF object file interpret.
168781ad6265SDimitry Andric     bool HasVisibilityAttr =
168881ad6265SDimitry Andric         XCOFFObj->is64Bit() || (XCOFFObj->auxiliaryHeader32() &&
168981ad6265SDimitry Andric                                 (XCOFFObj->auxiliaryHeader32()->getVersion() ==
169081ad6265SDimitry Andric                                  XCOFF::NEW_XCOFF_INTERPRET));
169181ad6265SDimitry Andric 
169281ad6265SDimitry Andric     if (HasVisibilityAttr) {
169381ad6265SDimitry Andric       XCOFFSymbolRef XCOFFSym = XCOFFObj->toSymbolRef(Sym.getRawDataRefImpl());
169481ad6265SDimitry Andric       uint16_t SymType = XCOFFSym.getSymbolType();
169581ad6265SDimitry Andric       if ((SymType & XCOFF::VISIBILITY_MASK) == XCOFF::SYM_V_INTERNAL)
169681ad6265SDimitry Andric         continue;
169781ad6265SDimitry Andric       if ((SymType & XCOFF::VISIBILITY_MASK) == XCOFF::SYM_V_HIDDEN)
169881ad6265SDimitry Andric         continue;
169981ad6265SDimitry Andric     }
170081ad6265SDimitry Andric 
170181ad6265SDimitry Andric     Expected<section_iterator> SymSecOrErr = Sym.getSection();
170281ad6265SDimitry Andric     if (!SymSecOrErr) {
170381ad6265SDimitry Andric       warn(SymSecOrErr.takeError(), XCOFFObj->getFileName(),
170481ad6265SDimitry Andric            "for symbol with index " +
170581ad6265SDimitry Andric                Twine(XCOFFObj->getSymbolIndex(Sym.getRawDataRefImpl().p)),
170681ad6265SDimitry Andric            ArchiveName);
170781ad6265SDimitry Andric       continue;
170881ad6265SDimitry Andric     }
170981ad6265SDimitry Andric     section_iterator SecIter = *SymSecOrErr;
171081ad6265SDimitry Andric     // If the symbol is not in a text or data section, it is not exported.
171181ad6265SDimitry Andric     if (SecIter == XCOFFObj->section_end())
171281ad6265SDimitry Andric       continue;
171381ad6265SDimitry Andric     if (!(SecIter->isText() || SecIter->isData() || SecIter->isBSS()))
171481ad6265SDimitry Andric       continue;
171581ad6265SDimitry Andric 
171681ad6265SDimitry Andric     StringRef SymName = cantFail(Sym.getName());
171781ad6265SDimitry Andric     if (SymName.empty())
171881ad6265SDimitry Andric       continue;
171981ad6265SDimitry Andric     if (SymName.startswith("__sinit") || SymName.startswith("__sterm") ||
172081ad6265SDimitry Andric         SymName.front() == '.' || SymName.front() == '(')
172181ad6265SDimitry Andric       continue;
172281ad6265SDimitry Andric 
172381ad6265SDimitry Andric     // Check the SymName regex matching with "^__[0-9]+__".
172481ad6265SDimitry Andric     if (SymName.size() > 4 && SymName.startswith("__") &&
172581ad6265SDimitry Andric         SymName.endswith("__")) {
172681ad6265SDimitry Andric       if (std::all_of(SymName.begin() + 2, SymName.end() - 2, isDigit))
172781ad6265SDimitry Andric         continue;
172881ad6265SDimitry Andric     }
172981ad6265SDimitry Andric 
173081ad6265SDimitry Andric     if (SymName == "__rsrc" && NoRsrc)
173181ad6265SDimitry Andric       continue;
173281ad6265SDimitry Andric 
173381ad6265SDimitry Andric     if (SymName.startswith("__tf1"))
173481ad6265SDimitry Andric       SymName = SymName.substr(6);
173581ad6265SDimitry Andric     else if (SymName.startswith("__tf9"))
173681ad6265SDimitry Andric       SymName = SymName.substr(14);
173781ad6265SDimitry Andric 
173881ad6265SDimitry Andric     NMSymbol S = {};
173981ad6265SDimitry Andric     S.Name = SymName.str();
174081ad6265SDimitry Andric     S.Sym = Sym;
174181ad6265SDimitry Andric 
174281ad6265SDimitry Andric     if (HasVisibilityAttr) {
174381ad6265SDimitry Andric       XCOFFSymbolRef XCOFFSym = XCOFFObj->toSymbolRef(Sym.getRawDataRefImpl());
174481ad6265SDimitry Andric       uint16_t SymType = XCOFFSym.getSymbolType();
174581ad6265SDimitry Andric       if ((SymType & XCOFF::VISIBILITY_MASK) == XCOFF::SYM_V_PROTECTED)
174681ad6265SDimitry Andric         S.Visibility = "protected";
174781ad6265SDimitry Andric       else if ((SymType & XCOFF::VISIBILITY_MASK) == XCOFF::SYM_V_EXPORTED)
174881ad6265SDimitry Andric         S.Visibility = "export";
174981ad6265SDimitry Andric     }
175081ad6265SDimitry Andric     if (S.initializeFlags(*XCOFFObj))
175181ad6265SDimitry Andric       SymbolList.push_back(S);
175281ad6265SDimitry Andric   }
175381ad6265SDimitry Andric }
175481ad6265SDimitry Andric 
175581ad6265SDimitry Andric static Expected<SymbolicFile::basic_symbol_iterator_range>
175681ad6265SDimitry Andric getDynamicSyms(SymbolicFile &Obj) {
175781ad6265SDimitry Andric   const auto *E = dyn_cast<ELFObjectFileBase>(&Obj);
175881ad6265SDimitry Andric   if (!E)
175981ad6265SDimitry Andric     return createError("File format has no dynamic symbol table");
176081ad6265SDimitry Andric   return E->getDynamicSymbolIterators();
176181ad6265SDimitry Andric }
176281ad6265SDimitry Andric 
176381ad6265SDimitry Andric // Returns false if there is error found or true otherwise.
176481ad6265SDimitry Andric static bool getSymbolNamesFromObject(SymbolicFile &Obj,
176581ad6265SDimitry Andric                                      std::vector<NMSymbol> &SymbolList) {
1766e8d8bef9SDimitry Andric   auto Symbols = Obj.symbols();
1767349cc55cSDimitry Andric   std::vector<VersionEntry> SymbolVersions;
1768e8d8bef9SDimitry Andric 
176981ad6265SDimitry Andric   if (DynamicSyms) {
177081ad6265SDimitry Andric     Expected<SymbolicFile::basic_symbol_iterator_range> SymbolsOrErr =
177181ad6265SDimitry Andric         getDynamicSyms(Obj);
177281ad6265SDimitry Andric     if (!SymbolsOrErr) {
177381ad6265SDimitry Andric       error(SymbolsOrErr.takeError(), Obj.getFileName());
177481ad6265SDimitry Andric       return false;
177581ad6265SDimitry Andric     }
177681ad6265SDimitry Andric     Symbols = *SymbolsOrErr;
177781ad6265SDimitry Andric     if (const auto *E = dyn_cast<ELFObjectFileBase>(&Obj)) {
1778349cc55cSDimitry Andric       if (Expected<std::vector<VersionEntry>> VersionsOrErr =
1779349cc55cSDimitry Andric               E->readDynsymVersions())
1780e8d8bef9SDimitry Andric         SymbolVersions = std::move(*VersionsOrErr);
1781e8d8bef9SDimitry Andric       else
1782e8d8bef9SDimitry Andric         WithColor::warning(errs(), ToolName)
1783e8d8bef9SDimitry Andric             << "unable to read symbol versions: "
1784e8d8bef9SDimitry Andric             << toString(VersionsOrErr.takeError()) << "\n";
1785e8d8bef9SDimitry Andric     }
178681ad6265SDimitry Andric   }
1787e8d8bef9SDimitry Andric   // If a "-s segname sectname" option was specified and this is a Mach-O
1788e8d8bef9SDimitry Andric   // file get the section number for that section in this object file.
1789e8d8bef9SDimitry Andric   unsigned int Nsect = 0;
1790e8d8bef9SDimitry Andric   MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(&Obj);
1791e8d8bef9SDimitry Andric   if (!SegSect.empty() && MachO) {
1792e8d8bef9SDimitry Andric     Nsect = getNsectForSegSect(MachO);
1793e8d8bef9SDimitry Andric     // If this section is not in the object file no symbols are printed.
1794e8d8bef9SDimitry Andric     if (Nsect == 0)
179581ad6265SDimitry Andric       return false;
1796e8d8bef9SDimitry Andric   }
179781ad6265SDimitry Andric 
1798e8d8bef9SDimitry Andric   if (!(MachO && DyldInfoOnly)) {
1799e8d8bef9SDimitry Andric     size_t I = -1;
1800e8d8bef9SDimitry Andric     for (BasicSymbolRef Sym : Symbols) {
1801e8d8bef9SDimitry Andric       ++I;
1802e8d8bef9SDimitry Andric       Expected<uint32_t> SymFlagsOrErr = Sym.getFlags();
1803e8d8bef9SDimitry Andric       if (!SymFlagsOrErr) {
1804e8d8bef9SDimitry Andric         error(SymFlagsOrErr.takeError(), Obj.getFileName());
180581ad6265SDimitry Andric         return false;
1806e8d8bef9SDimitry Andric       }
1807fe6060f1SDimitry Andric 
1808fe6060f1SDimitry Andric       // Don't drop format specifc symbols for ARM and AArch64 ELF targets, they
1809fe6060f1SDimitry Andric       // are used to repesent mapping symbols and needed to honor the
1810fe6060f1SDimitry Andric       // --special-syms option.
1811fe6060f1SDimitry Andric       auto *ELFObj = dyn_cast<ELFObjectFileBase>(&Obj);
1812fe6060f1SDimitry Andric       if ((!ELFObj || (ELFObj->getEMachine() != ELF::EM_ARM &&
1813fe6060f1SDimitry Andric                        ELFObj->getEMachine() != ELF::EM_AARCH64)) &&
1814fe6060f1SDimitry Andric           !DebugSyms && (*SymFlagsOrErr & SymbolRef::SF_FormatSpecific))
1815e8d8bef9SDimitry Andric         continue;
1816e8d8bef9SDimitry Andric       if (WithoutAliases && (*SymFlagsOrErr & SymbolRef::SF_Indirect))
1817e8d8bef9SDimitry Andric         continue;
1818e8d8bef9SDimitry Andric       // If a "-s segname sectname" option was specified and this is a Mach-O
1819e8d8bef9SDimitry Andric       // file and this section appears in this file, Nsect will be non-zero then
1820e8d8bef9SDimitry Andric       // see if this symbol is a symbol from that section and if not skip it.
1821e8d8bef9SDimitry Andric       if (Nsect && Nsect != getNsectInMachO(*MachO, Sym))
1822e8d8bef9SDimitry Andric         continue;
1823e8d8bef9SDimitry Andric       NMSymbol S = {};
1824e8d8bef9SDimitry Andric       S.Size = 0;
1825e8d8bef9SDimitry Andric       S.Address = 0;
1826e8d8bef9SDimitry Andric       if (isa<ELFObjectFileBase>(&Obj))
1827e8d8bef9SDimitry Andric         S.Size = ELFSymbolRef(Sym).getSize();
182804eeddc0SDimitry Andric 
182904eeddc0SDimitry Andric       if (const XCOFFObjectFile *XCOFFObj =
183004eeddc0SDimitry Andric               dyn_cast<const XCOFFObjectFile>(&Obj))
183104eeddc0SDimitry Andric         S.Size = XCOFFObj->getSymbolSize(Sym.getRawDataRefImpl());
183204eeddc0SDimitry Andric 
1833e8d8bef9SDimitry Andric       if (PrintAddress && isa<ObjectFile>(Obj)) {
1834e8d8bef9SDimitry Andric         SymbolRef SymRef(Sym);
1835e8d8bef9SDimitry Andric         Expected<uint64_t> AddressOrErr = SymRef.getAddress();
1836e8d8bef9SDimitry Andric         if (!AddressOrErr) {
1837e8d8bef9SDimitry Andric           consumeError(AddressOrErr.takeError());
1838e8d8bef9SDimitry Andric           break;
1839e8d8bef9SDimitry Andric         }
1840e8d8bef9SDimitry Andric         S.Address = *AddressOrErr;
1841e8d8bef9SDimitry Andric       }
1842e8d8bef9SDimitry Andric       S.TypeName = getNMTypeName(Obj, Sym);
1843e8d8bef9SDimitry Andric       S.TypeChar = getNMSectionTagAndName(Obj, Sym, S.SectionName);
1844e8d8bef9SDimitry Andric 
1845e8d8bef9SDimitry Andric       raw_string_ostream OS(S.Name);
1846e8d8bef9SDimitry Andric       if (Error E = Sym.printName(OS)) {
1847e8d8bef9SDimitry Andric         if (MachO) {
1848e8d8bef9SDimitry Andric           OS << "bad string index";
1849e8d8bef9SDimitry Andric           consumeError(std::move(E));
1850e8d8bef9SDimitry Andric         } else
1851e8d8bef9SDimitry Andric           error(std::move(E), Obj.getFileName());
1852e8d8bef9SDimitry Andric       }
1853e8d8bef9SDimitry Andric       if (!SymbolVersions.empty() && !SymbolVersions[I].Name.empty())
1854e8d8bef9SDimitry Andric         S.Name +=
1855349cc55cSDimitry Andric             (SymbolVersions[I].IsVerDef ? "@@" : "@") + SymbolVersions[I].Name;
1856e8d8bef9SDimitry Andric 
1857e8d8bef9SDimitry Andric       S.Sym = Sym;
185881ad6265SDimitry Andric       if (S.initializeFlags(Obj))
1859e8d8bef9SDimitry Andric         SymbolList.push_back(S);
1860e8d8bef9SDimitry Andric     }
1861e8d8bef9SDimitry Andric   }
1862e8d8bef9SDimitry Andric 
1863e8d8bef9SDimitry Andric   // If this is a Mach-O file where the nlist symbol table is out of sync
1864e8d8bef9SDimitry Andric   // with the dyld export trie then look through exports and fake up symbols
1865e8d8bef9SDimitry Andric   // for the ones that are missing (also done with the -add-dyldinfo flag).
1866e8d8bef9SDimitry Andric   // This is needed if strip(1) -T is run on a binary containing swift
1867e8d8bef9SDimitry Andric   // language symbols for example.  The option -only-dyldinfo will fake up
1868e8d8bef9SDimitry Andric   // all symbols from the dyld export trie as well as the bind info.
1869e8d8bef9SDimitry Andric   if (MachO && !NoDyldInfo)
187081ad6265SDimitry Andric     dumpSymbolsFromDLInfoMachO(*MachO, SymbolList);
1871e8d8bef9SDimitry Andric 
187281ad6265SDimitry Andric   return true;
187381ad6265SDimitry Andric }
187481ad6265SDimitry Andric 
187581ad6265SDimitry Andric static void printObjectLabel(bool PrintArchiveName, StringRef ArchiveName,
187681ad6265SDimitry Andric                              StringRef ArchitectureName,
187781ad6265SDimitry Andric                              StringRef ObjectFileName) {
187881ad6265SDimitry Andric   outs() << "\n";
187981ad6265SDimitry Andric   if (ArchiveName.empty() || !PrintArchiveName)
188081ad6265SDimitry Andric     outs() << ObjectFileName;
188181ad6265SDimitry Andric   else
188281ad6265SDimitry Andric     outs() << ArchiveName << "(" << ObjectFileName << ")";
188381ad6265SDimitry Andric   if (!ArchitectureName.empty())
188481ad6265SDimitry Andric     outs() << " (for architecture " << ArchitectureName << ")";
188581ad6265SDimitry Andric   outs() << ":\n";
188681ad6265SDimitry Andric }
188781ad6265SDimitry Andric 
188881ad6265SDimitry Andric static Expected<bool> hasSymbols(SymbolicFile &Obj) {
188981ad6265SDimitry Andric   if (DynamicSyms) {
189081ad6265SDimitry Andric     Expected<SymbolicFile::basic_symbol_iterator_range> DynamicSymsOrErr =
189181ad6265SDimitry Andric         getDynamicSyms(Obj);
189281ad6265SDimitry Andric     if (!DynamicSymsOrErr)
189381ad6265SDimitry Andric       return DynamicSymsOrErr.takeError();
189481ad6265SDimitry Andric     return !DynamicSymsOrErr->empty();
189581ad6265SDimitry Andric   }
189681ad6265SDimitry Andric   return !Obj.symbols().empty();
189781ad6265SDimitry Andric }
189881ad6265SDimitry Andric 
189981ad6265SDimitry Andric static void dumpSymbolNamesFromObject(
190081ad6265SDimitry Andric     SymbolicFile &Obj, std::vector<NMSymbol> &SymbolList,
190181ad6265SDimitry Andric     bool PrintSymbolObject, bool PrintObjectLabel, StringRef ArchiveName = {},
190281ad6265SDimitry Andric     StringRef ArchitectureName = {}, StringRef ObjectName = {},
190381ad6265SDimitry Andric     bool PrintArchiveName = true) {
190481ad6265SDimitry Andric   if (!shouldDump(Obj))
190581ad6265SDimitry Andric     return;
190681ad6265SDimitry Andric 
190781ad6265SDimitry Andric   if (ExportSymbols && Obj.isXCOFF()) {
190881ad6265SDimitry Andric     XCOFFObjectFile *XCOFFObj = cast<XCOFFObjectFile>(&Obj);
190981ad6265SDimitry Andric     getXCOFFExports(XCOFFObj, SymbolList, ArchiveName);
191081ad6265SDimitry Andric     return;
191181ad6265SDimitry Andric   }
191281ad6265SDimitry Andric 
191381ad6265SDimitry Andric   if (PrintObjectLabel && !ExportSymbols)
191481ad6265SDimitry Andric     printObjectLabel(PrintArchiveName, ArchiveName, ArchitectureName,
191581ad6265SDimitry Andric                      ObjectName.empty() ? Obj.getFileName() : ObjectName);
191681ad6265SDimitry Andric   if (!getSymbolNamesFromObject(Obj, SymbolList) || ExportSymbols)
191781ad6265SDimitry Andric     return;
19180b57cec5SDimitry Andric   CurrentFilename = Obj.getFileName();
19195ffd83dbSDimitry Andric 
192081ad6265SDimitry Andric   // If there is an error in hasSymbols(), the error should be encountered in
192181ad6265SDimitry Andric   // function getSymbolNamesFromObject first.
192281ad6265SDimitry Andric   if (!cantFail(hasSymbols(Obj)) && SymbolList.empty() && !Quiet) {
19235ffd83dbSDimitry Andric     writeFileName(errs(), ArchiveName, ArchitectureName);
19245ffd83dbSDimitry Andric     errs() << "no symbols\n";
19255ffd83dbSDimitry Andric   }
19265ffd83dbSDimitry Andric 
192781ad6265SDimitry Andric   sortSymbolList(SymbolList);
192881ad6265SDimitry Andric   printSymbolList(Obj, SymbolList, PrintSymbolObject, ArchiveName,
192981ad6265SDimitry Andric                   ArchitectureName);
19300b57cec5SDimitry Andric }
19310b57cec5SDimitry Andric 
19320b57cec5SDimitry Andric // checkMachOAndArchFlags() checks to see if the SymbolicFile is a Mach-O file
19330b57cec5SDimitry Andric // and if it is and there is a list of architecture flags is specified then
19340b57cec5SDimitry Andric // check to make sure this Mach-O file is one of those architectures or all
19350b57cec5SDimitry Andric // architectures was specificed.  If not then an error is generated and this
19360b57cec5SDimitry Andric // routine returns false.  Else it returns true.
193781ad6265SDimitry Andric static bool checkMachOAndArchFlags(SymbolicFile *O, StringRef Filename) {
19380b57cec5SDimitry Andric   auto *MachO = dyn_cast<MachOObjectFile>(O);
19390b57cec5SDimitry Andric 
19400b57cec5SDimitry Andric   if (!MachO || ArchAll || ArchFlags.empty())
19410b57cec5SDimitry Andric     return true;
19420b57cec5SDimitry Andric 
19430b57cec5SDimitry Andric   MachO::mach_header H;
19440b57cec5SDimitry Andric   MachO::mach_header_64 H_64;
19450b57cec5SDimitry Andric   Triple T;
19460b57cec5SDimitry Andric   const char *McpuDefault, *ArchFlag;
19470b57cec5SDimitry Andric   if (MachO->is64Bit()) {
19480b57cec5SDimitry Andric     H_64 = MachO->MachOObjectFile::getHeader64();
19490b57cec5SDimitry Andric     T = MachOObjectFile::getArchTriple(H_64.cputype, H_64.cpusubtype,
19500b57cec5SDimitry Andric                                        &McpuDefault, &ArchFlag);
19510b57cec5SDimitry Andric   } else {
19520b57cec5SDimitry Andric     H = MachO->MachOObjectFile::getHeader();
19530b57cec5SDimitry Andric     T = MachOObjectFile::getArchTriple(H.cputype, H.cpusubtype,
19540b57cec5SDimitry Andric                                        &McpuDefault, &ArchFlag);
19550b57cec5SDimitry Andric   }
19560b57cec5SDimitry Andric   const std::string ArchFlagName(ArchFlag);
1957e8d8bef9SDimitry Andric   if (!llvm::is_contained(ArchFlags, ArchFlagName)) {
19580b57cec5SDimitry Andric     error("No architecture specified", Filename);
19590b57cec5SDimitry Andric     return false;
19600b57cec5SDimitry Andric   }
19610b57cec5SDimitry Andric   return true;
19620b57cec5SDimitry Andric }
19630b57cec5SDimitry Andric 
196481ad6265SDimitry Andric static void dumpArchiveMap(Archive *A, StringRef Filename) {
19650b57cec5SDimitry Andric   Archive::symbol_iterator I = A->symbol_begin();
19660b57cec5SDimitry Andric   Archive::symbol_iterator E = A->symbol_end();
19670b57cec5SDimitry Andric   if (I != E) {
19680b57cec5SDimitry Andric     outs() << "Archive map\n";
19690b57cec5SDimitry Andric     for (; I != E; ++I) {
19700b57cec5SDimitry Andric       Expected<Archive::Child> C = I->getMember();
19710b57cec5SDimitry Andric       if (!C) {
19720b57cec5SDimitry Andric         error(C.takeError(), Filename);
19730b57cec5SDimitry Andric         break;
19740b57cec5SDimitry Andric       }
19750b57cec5SDimitry Andric       Expected<StringRef> FileNameOrErr = C->getName();
19760b57cec5SDimitry Andric       if (!FileNameOrErr) {
19770b57cec5SDimitry Andric         error(FileNameOrErr.takeError(), Filename);
19780b57cec5SDimitry Andric         break;
19790b57cec5SDimitry Andric       }
19800b57cec5SDimitry Andric       StringRef SymName = I->getName();
19810b57cec5SDimitry Andric       outs() << SymName << " in " << FileNameOrErr.get() << "\n";
19820b57cec5SDimitry Andric     }
19830b57cec5SDimitry Andric     outs() << "\n";
19840b57cec5SDimitry Andric   }
19850b57cec5SDimitry Andric }
19860b57cec5SDimitry Andric 
198781ad6265SDimitry Andric static void dumpArchive(Archive *A, std::vector<NMSymbol> &SymbolList,
198881ad6265SDimitry Andric                         StringRef Filename, LLVMContext *ContextPtr) {
198981ad6265SDimitry Andric   if (ArchiveMap)
199081ad6265SDimitry Andric     dumpArchiveMap(A, Filename);
199181ad6265SDimitry Andric 
19920b57cec5SDimitry Andric   Error Err = Error::success();
19930b57cec5SDimitry Andric   for (auto &C : A->children(Err)) {
199481ad6265SDimitry Andric     Expected<std::unique_ptr<Binary>> ChildOrErr = C.getAsBinary(ContextPtr);
19950b57cec5SDimitry Andric     if (!ChildOrErr) {
19960b57cec5SDimitry Andric       if (auto E = isNotObjectErrorInvalidFileType(ChildOrErr.takeError()))
19970b57cec5SDimitry Andric         error(std::move(E), Filename, C);
19980b57cec5SDimitry Andric       continue;
19990b57cec5SDimitry Andric     }
20000b57cec5SDimitry Andric     if (SymbolicFile *O = dyn_cast<SymbolicFile>(&*ChildOrErr.get())) {
20010b57cec5SDimitry Andric       if (!MachOPrintSizeWarning && PrintSize && isa<MachOObjectFile>(O)) {
20020b57cec5SDimitry Andric         WithColor::warning(errs(), ToolName)
20030b57cec5SDimitry Andric             << "sizes with -print-size for Mach-O files are always zero.\n";
20040b57cec5SDimitry Andric         MachOPrintSizeWarning = true;
20050b57cec5SDimitry Andric       }
20060b57cec5SDimitry Andric       if (!checkMachOAndArchFlags(O, Filename))
20070b57cec5SDimitry Andric         return;
200881ad6265SDimitry Andric       dumpSymbolNamesFromObject(*O, SymbolList, /*PrintSymbolObject=*/false,
200981ad6265SDimitry Andric                                 !PrintFileName, Filename,
201081ad6265SDimitry Andric                                 /*ArchitectureName=*/{}, O->getFileName(),
201181ad6265SDimitry Andric                                 /*PrintArchiveName=*/false);
20120b57cec5SDimitry Andric     }
20130b57cec5SDimitry Andric   }
20140b57cec5SDimitry Andric   if (Err)
20150b57cec5SDimitry Andric     error(std::move(Err), A->getFileName());
20160b57cec5SDimitry Andric }
201781ad6265SDimitry Andric 
201881ad6265SDimitry Andric static void dumpMachOUniversalBinaryMatchArchFlags(
201981ad6265SDimitry Andric     MachOUniversalBinary *UB, std::vector<NMSymbol> &SymbolList,
202081ad6265SDimitry Andric     StringRef Filename, LLVMContext *ContextPtr) {
20210b57cec5SDimitry Andric   // Look for a slice in the universal binary that matches each ArchFlag.
20220b57cec5SDimitry Andric   bool ArchFound;
20230b57cec5SDimitry Andric   for (unsigned i = 0; i < ArchFlags.size(); ++i) {
20240b57cec5SDimitry Andric     ArchFound = false;
20250b57cec5SDimitry Andric     for (MachOUniversalBinary::object_iterator I = UB->begin_objects(),
20260b57cec5SDimitry Andric                                                E = UB->end_objects();
20270b57cec5SDimitry Andric          I != E; ++I) {
20280b57cec5SDimitry Andric       if (ArchFlags[i] == I->getArchFlagName()) {
20290b57cec5SDimitry Andric         ArchFound = true;
203081ad6265SDimitry Andric         Expected<std::unique_ptr<ObjectFile>> ObjOrErr = I->getAsObjectFile();
20310b57cec5SDimitry Andric         std::string ArchiveName;
20320b57cec5SDimitry Andric         std::string ArchitectureName;
20330b57cec5SDimitry Andric         ArchiveName.clear();
20340b57cec5SDimitry Andric         ArchitectureName.clear();
20350b57cec5SDimitry Andric         if (ObjOrErr) {
20360b57cec5SDimitry Andric           ObjectFile &Obj = *ObjOrErr.get();
203781ad6265SDimitry Andric           if (ArchFlags.size() > 1)
20380b57cec5SDimitry Andric             ArchitectureName = I->getArchFlagName();
203981ad6265SDimitry Andric           dumpSymbolNamesFromObject(Obj, SymbolList,
204081ad6265SDimitry Andric                                     /*PrintSymbolObject=*/false,
204181ad6265SDimitry Andric                                     (ArchFlags.size() > 1) && !PrintFileName,
204281ad6265SDimitry Andric                                     ArchiveName, ArchitectureName);
204381ad6265SDimitry Andric         } else if (auto E =
204481ad6265SDimitry Andric                        isNotObjectErrorInvalidFileType(ObjOrErr.takeError())) {
204581ad6265SDimitry Andric           error(std::move(E), Filename,
204681ad6265SDimitry Andric                 ArchFlags.size() > 1 ? StringRef(I->getArchFlagName())
204781ad6265SDimitry Andric                                      : StringRef());
20480b57cec5SDimitry Andric           continue;
20490b57cec5SDimitry Andric         } else if (Expected<std::unique_ptr<Archive>> AOrErr =
20500b57cec5SDimitry Andric                        I->getAsArchive()) {
20510b57cec5SDimitry Andric           std::unique_ptr<Archive> &A = *AOrErr;
20520b57cec5SDimitry Andric           Error Err = Error::success();
20530b57cec5SDimitry Andric           for (auto &C : A->children(Err)) {
20540b57cec5SDimitry Andric             Expected<std::unique_ptr<Binary>> ChildOrErr =
20550b57cec5SDimitry Andric                 C.getAsBinary(ContextPtr);
20560b57cec5SDimitry Andric             if (!ChildOrErr) {
205781ad6265SDimitry Andric               if (auto E =
205881ad6265SDimitry Andric                       isNotObjectErrorInvalidFileType(ChildOrErr.takeError())) {
205981ad6265SDimitry Andric                 error(std::move(E), Filename, C,
206081ad6265SDimitry Andric                       ArchFlags.size() > 1 ? StringRef(I->getArchFlagName())
206181ad6265SDimitry Andric                                            : StringRef());
20620b57cec5SDimitry Andric               }
20630b57cec5SDimitry Andric               continue;
20640b57cec5SDimitry Andric             }
206581ad6265SDimitry Andric             if (SymbolicFile *O = dyn_cast<SymbolicFile>(&*ChildOrErr.get())) {
20665ffd83dbSDimitry Andric               ArchiveName = std::string(A->getFileName());
20670b57cec5SDimitry Andric               if (ArchFlags.size() > 1)
20680b57cec5SDimitry Andric                 ArchitectureName = I->getArchFlagName();
206981ad6265SDimitry Andric               dumpSymbolNamesFromObject(
207081ad6265SDimitry Andric                   *O, SymbolList, /*PrintSymbolObject=*/false, !PrintFileName,
207181ad6265SDimitry Andric                   ArchiveName, ArchitectureName);
20720b57cec5SDimitry Andric             }
20730b57cec5SDimitry Andric           }
20740b57cec5SDimitry Andric           if (Err)
20750b57cec5SDimitry Andric             error(std::move(Err), A->getFileName());
20760b57cec5SDimitry Andric         } else {
20770b57cec5SDimitry Andric           consumeError(AOrErr.takeError());
20780b57cec5SDimitry Andric           error(Filename + " for architecture " +
20790b57cec5SDimitry Andric                     StringRef(I->getArchFlagName()) +
20800b57cec5SDimitry Andric                     " is not a Mach-O file or an archive file",
20810b57cec5SDimitry Andric                 "Mach-O universal file");
20820b57cec5SDimitry Andric         }
20830b57cec5SDimitry Andric       }
20840b57cec5SDimitry Andric     }
20850b57cec5SDimitry Andric     if (!ArchFound) {
20860b57cec5SDimitry Andric       error(ArchFlags[i],
20870b57cec5SDimitry Andric             "file: " + Filename + " does not contain architecture");
20880b57cec5SDimitry Andric       return;
20890b57cec5SDimitry Andric     }
20900b57cec5SDimitry Andric   }
20910b57cec5SDimitry Andric }
209281ad6265SDimitry Andric 
209381ad6265SDimitry Andric // Returns true If the binary contains a slice that matches the host
209481ad6265SDimitry Andric // architecture, or false otherwise.
209581ad6265SDimitry Andric static bool dumpMachOUniversalBinaryMatchHost(MachOUniversalBinary *UB,
209681ad6265SDimitry Andric                                               std::vector<NMSymbol> &SymbolList,
209781ad6265SDimitry Andric                                               StringRef Filename,
209881ad6265SDimitry Andric                                               LLVMContext *ContextPtr) {
20990b57cec5SDimitry Andric   Triple HostTriple = MachOObjectFile::getHostArch();
21000b57cec5SDimitry Andric   StringRef HostArchName = HostTriple.getArchName();
21010b57cec5SDimitry Andric   for (MachOUniversalBinary::object_iterator I = UB->begin_objects(),
21020b57cec5SDimitry Andric                                              E = UB->end_objects();
21030b57cec5SDimitry Andric        I != E; ++I) {
21040b57cec5SDimitry Andric     if (HostArchName == I->getArchFlagName()) {
21050b57cec5SDimitry Andric       Expected<std::unique_ptr<ObjectFile>> ObjOrErr = I->getAsObjectFile();
21060b57cec5SDimitry Andric       std::string ArchiveName;
21070b57cec5SDimitry Andric       if (ObjOrErr) {
21080b57cec5SDimitry Andric         ObjectFile &Obj = *ObjOrErr.get();
210981ad6265SDimitry Andric         dumpSymbolNamesFromObject(Obj, SymbolList, /*PrintSymbolObject=*/false,
211081ad6265SDimitry Andric                                   /*PrintObjectLabel=*/false);
211181ad6265SDimitry Andric       } else if (auto E = isNotObjectErrorInvalidFileType(ObjOrErr.takeError()))
21120b57cec5SDimitry Andric         error(std::move(E), Filename);
211381ad6265SDimitry Andric       else if (Expected<std::unique_ptr<Archive>> AOrErr = I->getAsArchive()) {
21140b57cec5SDimitry Andric         std::unique_ptr<Archive> &A = *AOrErr;
21150b57cec5SDimitry Andric         Error Err = Error::success();
21160b57cec5SDimitry Andric         for (auto &C : A->children(Err)) {
21170b57cec5SDimitry Andric           Expected<std::unique_ptr<Binary>> ChildOrErr =
21180b57cec5SDimitry Andric               C.getAsBinary(ContextPtr);
21190b57cec5SDimitry Andric           if (!ChildOrErr) {
212081ad6265SDimitry Andric             if (auto E =
212181ad6265SDimitry Andric                     isNotObjectErrorInvalidFileType(ChildOrErr.takeError()))
21220b57cec5SDimitry Andric               error(std::move(E), Filename, C);
21230b57cec5SDimitry Andric             continue;
21240b57cec5SDimitry Andric           }
212581ad6265SDimitry Andric           if (SymbolicFile *O = dyn_cast<SymbolicFile>(&*ChildOrErr.get())) {
21265ffd83dbSDimitry Andric             ArchiveName = std::string(A->getFileName());
212781ad6265SDimitry Andric             dumpSymbolNamesFromObject(*O, SymbolList,
212881ad6265SDimitry Andric                                       /*PrintSymbolObject=*/false,
212981ad6265SDimitry Andric                                       !PrintFileName, ArchiveName);
21300b57cec5SDimitry Andric           }
21310b57cec5SDimitry Andric         }
21320b57cec5SDimitry Andric         if (Err)
21330b57cec5SDimitry Andric           error(std::move(Err), A->getFileName());
21340b57cec5SDimitry Andric       } else {
21350b57cec5SDimitry Andric         consumeError(AOrErr.takeError());
21360b57cec5SDimitry Andric         error(Filename + " for architecture " +
21370b57cec5SDimitry Andric                   StringRef(I->getArchFlagName()) +
21380b57cec5SDimitry Andric                   " is not a Mach-O file or an archive file",
21390b57cec5SDimitry Andric               "Mach-O universal file");
21400b57cec5SDimitry Andric       }
214181ad6265SDimitry Andric       return true;
21420b57cec5SDimitry Andric     }
21430b57cec5SDimitry Andric   }
214481ad6265SDimitry Andric   return false;
21450b57cec5SDimitry Andric }
214681ad6265SDimitry Andric 
214781ad6265SDimitry Andric static void dumpMachOUniversalBinaryArchAll(MachOUniversalBinary *UB,
214881ad6265SDimitry Andric                                             std::vector<NMSymbol> &SymbolList,
214981ad6265SDimitry Andric                                             StringRef Filename,
215081ad6265SDimitry Andric                                             LLVMContext *ContextPtr) {
21510b57cec5SDimitry Andric   bool moreThanOneArch = UB->getNumberOfObjects() > 1;
21520b57cec5SDimitry Andric   for (const MachOUniversalBinary::ObjectForArch &O : UB->objects()) {
21530b57cec5SDimitry Andric     Expected<std::unique_ptr<ObjectFile>> ObjOrErr = O.getAsObjectFile();
21540b57cec5SDimitry Andric     std::string ArchiveName;
21550b57cec5SDimitry Andric     std::string ArchitectureName;
21560b57cec5SDimitry Andric     ArchiveName.clear();
21570b57cec5SDimitry Andric     ArchitectureName.clear();
21580b57cec5SDimitry Andric     if (ObjOrErr) {
21590b57cec5SDimitry Andric       ObjectFile &Obj = *ObjOrErr.get();
21600b57cec5SDimitry Andric       if (isa<MachOObjectFile>(Obj) && moreThanOneArch)
21610b57cec5SDimitry Andric         ArchitectureName = O.getArchFlagName();
216281ad6265SDimitry Andric       dumpSymbolNamesFromObject(Obj, SymbolList, /*PrintSymbolObject=*/false,
216381ad6265SDimitry Andric                                 !PrintFileName, ArchiveName, ArchitectureName);
216481ad6265SDimitry Andric     } else if (auto E = isNotObjectErrorInvalidFileType(ObjOrErr.takeError())) {
216581ad6265SDimitry Andric       error(std::move(E), Filename,
216681ad6265SDimitry Andric             moreThanOneArch ? StringRef(O.getArchFlagName()) : StringRef());
21670b57cec5SDimitry Andric       continue;
216881ad6265SDimitry Andric     } else if (Expected<std::unique_ptr<Archive>> AOrErr = O.getAsArchive()) {
21690b57cec5SDimitry Andric       std::unique_ptr<Archive> &A = *AOrErr;
21700b57cec5SDimitry Andric       Error Err = Error::success();
21710b57cec5SDimitry Andric       for (auto &C : A->children(Err)) {
21720b57cec5SDimitry Andric         Expected<std::unique_ptr<Binary>> ChildOrErr =
21730b57cec5SDimitry Andric             C.getAsBinary(ContextPtr);
21740b57cec5SDimitry Andric         if (!ChildOrErr) {
217581ad6265SDimitry Andric           if (auto E = isNotObjectErrorInvalidFileType(ChildOrErr.takeError()))
217681ad6265SDimitry Andric             error(std::move(E), Filename, C,
217781ad6265SDimitry Andric                   moreThanOneArch ? StringRef(ArchitectureName) : StringRef());
21780b57cec5SDimitry Andric           continue;
21790b57cec5SDimitry Andric         }
21800b57cec5SDimitry Andric         if (SymbolicFile *F = dyn_cast<SymbolicFile>(&*ChildOrErr.get())) {
21815ffd83dbSDimitry Andric           ArchiveName = std::string(A->getFileName());
21820b57cec5SDimitry Andric           if (isa<MachOObjectFile>(F) && moreThanOneArch)
21830b57cec5SDimitry Andric             ArchitectureName = O.getArchFlagName();
218481ad6265SDimitry Andric           dumpSymbolNamesFromObject(*F, SymbolList, /*PrintSymbolObject=*/false,
218581ad6265SDimitry Andric                                     !PrintFileName, ArchiveName,
218681ad6265SDimitry Andric                                     ArchitectureName);
21870b57cec5SDimitry Andric         }
21880b57cec5SDimitry Andric       }
21890b57cec5SDimitry Andric       if (Err)
21900b57cec5SDimitry Andric         error(std::move(Err), A->getFileName());
21910b57cec5SDimitry Andric     } else {
21920b57cec5SDimitry Andric       consumeError(AOrErr.takeError());
219381ad6265SDimitry Andric       error(Filename + " for architecture " + StringRef(O.getArchFlagName()) +
21940b57cec5SDimitry Andric                 " is not a Mach-O file or an archive file",
21950b57cec5SDimitry Andric             "Mach-O universal file");
21960b57cec5SDimitry Andric     }
21970b57cec5SDimitry Andric   }
219881ad6265SDimitry Andric }
219981ad6265SDimitry Andric 
220081ad6265SDimitry Andric static void dumpMachOUniversalBinary(MachOUniversalBinary *UB,
220181ad6265SDimitry Andric                                      std::vector<NMSymbol> &SymbolList,
220281ad6265SDimitry Andric                                      StringRef Filename,
220381ad6265SDimitry Andric                                      LLVMContext *ContextPtr) {
220481ad6265SDimitry Andric   // If we have a list of architecture flags specified dump only those.
220581ad6265SDimitry Andric   if (!ArchAll && !ArchFlags.empty()) {
220681ad6265SDimitry Andric     dumpMachOUniversalBinaryMatchArchFlags(UB, SymbolList, Filename,
220781ad6265SDimitry Andric                                            ContextPtr);
22080b57cec5SDimitry Andric     return;
22090b57cec5SDimitry Andric   }
22105ffd83dbSDimitry Andric 
221181ad6265SDimitry Andric   // No architecture flags were specified so if this contains a slice that
221281ad6265SDimitry Andric   // matches the host architecture dump only that.
221381ad6265SDimitry Andric   if (!ArchAll &&
221481ad6265SDimitry Andric       dumpMachOUniversalBinaryMatchHost(UB, SymbolList, Filename, ContextPtr))
221581ad6265SDimitry Andric     return;
221681ad6265SDimitry Andric 
221781ad6265SDimitry Andric   // Either all architectures have been specified or none have been specified
221881ad6265SDimitry Andric   // and this does not contain the host architecture so dump all the slices.
221981ad6265SDimitry Andric   dumpMachOUniversalBinaryArchAll(UB, SymbolList, Filename, ContextPtr);
222081ad6265SDimitry Andric }
222181ad6265SDimitry Andric 
222281ad6265SDimitry Andric static void dumpTapiUniversal(TapiUniversal *TU,
222381ad6265SDimitry Andric                               std::vector<NMSymbol> &SymbolList,
222481ad6265SDimitry Andric                               StringRef Filename) {
22255ffd83dbSDimitry Andric   for (const TapiUniversal::ObjectForArch &I : TU->objects()) {
22265ffd83dbSDimitry Andric     StringRef ArchName = I.getArchFlagName();
22275ffd83dbSDimitry Andric     const bool ShowArch =
2228e8d8bef9SDimitry Andric         ArchFlags.empty() || llvm::is_contained(ArchFlags, ArchName);
22295ffd83dbSDimitry Andric     if (!ShowArch)
22305ffd83dbSDimitry Andric       continue;
22315ffd83dbSDimitry Andric     if (!AddInlinedInfo && !I.isTopLevelLib())
22325ffd83dbSDimitry Andric       continue;
223381ad6265SDimitry Andric     if (auto ObjOrErr = I.getAsObjectFile())
223481ad6265SDimitry Andric       dumpSymbolNamesFromObject(
223581ad6265SDimitry Andric           *ObjOrErr.get(), SymbolList, /*PrintSymbolObject=*/false,
223681ad6265SDimitry Andric           /*PrintObjectLabel=*/true,
223781ad6265SDimitry Andric           /*ArchiveName=*/{}, ArchName, I.getInstallName());
223881ad6265SDimitry Andric     else if (Error E = isNotObjectErrorInvalidFileType(ObjOrErr.takeError())) {
22395ffd83dbSDimitry Andric       error(std::move(E), Filename, ArchName);
22405ffd83dbSDimitry Andric     }
22415ffd83dbSDimitry Andric   }
22425ffd83dbSDimitry Andric }
22435ffd83dbSDimitry Andric 
224481ad6265SDimitry Andric static void dumpSymbolicFile(SymbolicFile *O, std::vector<NMSymbol> &SymbolList,
224581ad6265SDimitry Andric                              StringRef Filename) {
22460b57cec5SDimitry Andric   if (!MachOPrintSizeWarning && PrintSize && isa<MachOObjectFile>(O)) {
22470b57cec5SDimitry Andric     WithColor::warning(errs(), ToolName)
22480b57cec5SDimitry Andric         << "sizes with --print-size for Mach-O files are always zero.\n";
22490b57cec5SDimitry Andric     MachOPrintSizeWarning = true;
22500b57cec5SDimitry Andric   }
22510b57cec5SDimitry Andric   if (!checkMachOAndArchFlags(O, Filename))
22520b57cec5SDimitry Andric     return;
225381ad6265SDimitry Andric   dumpSymbolNamesFromObject(*O, SymbolList, /*PrintSymbolObject=*/true,
225481ad6265SDimitry Andric                             /*PrintObjectLabel=*/false);
22550b57cec5SDimitry Andric }
225681ad6265SDimitry Andric 
225781ad6265SDimitry Andric static std::vector<NMSymbol> dumpSymbolNamesFromFile(StringRef Filename) {
225881ad6265SDimitry Andric   std::vector<NMSymbol> SymbolList;
225981ad6265SDimitry Andric   ErrorOr<std::unique_ptr<MemoryBuffer>> BufferOrErr =
226081ad6265SDimitry Andric       MemoryBuffer::getFileOrSTDIN(Filename);
226181ad6265SDimitry Andric   if (error(BufferOrErr.getError(), Filename))
226281ad6265SDimitry Andric     return SymbolList;
226381ad6265SDimitry Andric 
226481ad6265SDimitry Andric   // Always enable opaque pointers, to handle archives with mixed typed and
226581ad6265SDimitry Andric   // opaque pointer bitcode files gracefully. As we're only reading symbols,
226681ad6265SDimitry Andric   // the used pointer types don't matter.
226781ad6265SDimitry Andric   LLVMContext Context;
226881ad6265SDimitry Andric   Context.setOpaquePointers(true);
226981ad6265SDimitry Andric   LLVMContext *ContextPtr = NoLLVMBitcode ? nullptr : &Context;
227081ad6265SDimitry Andric   Expected<std::unique_ptr<Binary>> BinaryOrErr =
227181ad6265SDimitry Andric       createBinary(BufferOrErr.get()->getMemBufferRef(), ContextPtr);
227281ad6265SDimitry Andric   if (!BinaryOrErr) {
227381ad6265SDimitry Andric     error(BinaryOrErr.takeError(), Filename);
227481ad6265SDimitry Andric     return SymbolList;
227581ad6265SDimitry Andric   }
227681ad6265SDimitry Andric   Binary &Bin = *BinaryOrErr.get();
227781ad6265SDimitry Andric   if (Archive *A = dyn_cast<Archive>(&Bin))
227881ad6265SDimitry Andric     dumpArchive(A, SymbolList, Filename, ContextPtr);
227981ad6265SDimitry Andric   else if (MachOUniversalBinary *UB = dyn_cast<MachOUniversalBinary>(&Bin))
228081ad6265SDimitry Andric     dumpMachOUniversalBinary(UB, SymbolList, Filename, ContextPtr);
228181ad6265SDimitry Andric   else if (TapiUniversal *TU = dyn_cast<TapiUniversal>(&Bin))
228281ad6265SDimitry Andric     dumpTapiUniversal(TU, SymbolList, Filename);
228381ad6265SDimitry Andric   else if (SymbolicFile *O = dyn_cast<SymbolicFile>(&Bin))
228481ad6265SDimitry Andric     dumpSymbolicFile(O, SymbolList, Filename);
228581ad6265SDimitry Andric   return SymbolList;
228681ad6265SDimitry Andric }
228781ad6265SDimitry Andric 
228881ad6265SDimitry Andric static void
228981ad6265SDimitry Andric exportSymbolNamesFromFiles(const std::vector<std::string> &InputFilenames) {
229081ad6265SDimitry Andric   std::vector<NMSymbol> SymbolList;
229181ad6265SDimitry Andric   for (const auto &FileName : InputFilenames) {
229281ad6265SDimitry Andric     std::vector<NMSymbol> FileSymList = dumpSymbolNamesFromFile(FileName);
229381ad6265SDimitry Andric     SymbolList.insert(SymbolList.end(), FileSymList.begin(), FileSymList.end());
229481ad6265SDimitry Andric   }
229581ad6265SDimitry Andric 
229681ad6265SDimitry Andric   // Delete symbols which should not be printed from SymolList.
2297*bdd1243dSDimitry Andric   llvm::erase_if(SymbolList,
2298*bdd1243dSDimitry Andric                  [](const NMSymbol &s) { return !s.shouldPrint(); });
229981ad6265SDimitry Andric   sortSymbolList(SymbolList);
230081ad6265SDimitry Andric   SymbolList.erase(std::unique(SymbolList.begin(), SymbolList.end()),
230181ad6265SDimitry Andric                    SymbolList.end());
230281ad6265SDimitry Andric   printExportSymbolList(SymbolList);
23030b57cec5SDimitry Andric }
23040b57cec5SDimitry Andric 
2305*bdd1243dSDimitry Andric int llvm_nm_main(int argc, char **argv) {
23060b57cec5SDimitry Andric   InitLLVM X(argc, argv);
2307fe6060f1SDimitry Andric   BumpPtrAllocator A;
2308fe6060f1SDimitry Andric   StringSaver Saver(A);
2309fe6060f1SDimitry Andric   NmOptTable Tbl;
2310fe6060f1SDimitry Andric   ToolName = argv[0];
2311fe6060f1SDimitry Andric   opt::InputArgList Args =
2312fe6060f1SDimitry Andric       Tbl.parseArgs(argc, argv, OPT_UNKNOWN, Saver, [&](StringRef Msg) {
2313fe6060f1SDimitry Andric         error(Msg);
2314fe6060f1SDimitry Andric         exit(1);
2315fe6060f1SDimitry Andric       });
2316fe6060f1SDimitry Andric   if (Args.hasArg(OPT_help)) {
2317fe6060f1SDimitry Andric     Tbl.printHelp(
2318fe6060f1SDimitry Andric         outs(),
2319fe6060f1SDimitry Andric         (Twine(ToolName) + " [options] <input object files>").str().c_str(),
2320fe6060f1SDimitry Andric         "LLVM symbol table dumper");
2321fe6060f1SDimitry Andric     // TODO Replace this with OptTable API once it adds extrahelp support.
2322fe6060f1SDimitry Andric     outs() << "\nPass @FILE as argument to read options from FILE.\n";
2323fe6060f1SDimitry Andric     return 0;
2324fe6060f1SDimitry Andric   }
2325fe6060f1SDimitry Andric   if (Args.hasArg(OPT_version)) {
2326fe6060f1SDimitry Andric     // This needs to contain the word "GNU", libtool looks for that string.
2327fe6060f1SDimitry Andric     outs() << "llvm-nm, compatible with GNU nm" << '\n';
2328fe6060f1SDimitry Andric     cl::PrintVersionMessage();
2329fe6060f1SDimitry Andric     return 0;
2330fe6060f1SDimitry Andric   }
2331fe6060f1SDimitry Andric 
2332fe6060f1SDimitry Andric   DebugSyms = Args.hasArg(OPT_debug_syms);
2333fe6060f1SDimitry Andric   DefinedOnly = Args.hasArg(OPT_defined_only);
2334fe6060f1SDimitry Andric   Demangle = Args.hasFlag(OPT_demangle, OPT_no_demangle, false);
2335fe6060f1SDimitry Andric   DynamicSyms = Args.hasArg(OPT_dynamic);
2336fe6060f1SDimitry Andric   ExternalOnly = Args.hasArg(OPT_extern_only);
2337fe6060f1SDimitry Andric   StringRef V = Args.getLastArgValue(OPT_format_EQ, "bsd");
2338fe6060f1SDimitry Andric   if (V == "bsd")
2339fe6060f1SDimitry Andric     OutputFormat = bsd;
2340fe6060f1SDimitry Andric   else if (V == "posix")
2341fe6060f1SDimitry Andric     OutputFormat = posix;
2342fe6060f1SDimitry Andric   else if (V == "sysv")
2343fe6060f1SDimitry Andric     OutputFormat = sysv;
2344fe6060f1SDimitry Andric   else if (V == "darwin")
2345fe6060f1SDimitry Andric     OutputFormat = darwin;
2346fe6060f1SDimitry Andric   else if (V == "just-symbols")
2347fe6060f1SDimitry Andric     OutputFormat = just_symbols;
2348fe6060f1SDimitry Andric   else
2349fe6060f1SDimitry Andric     error("--format value should be one of: bsd, posix, sysv, darwin, "
2350fe6060f1SDimitry Andric           "just-symbols");
2351fe6060f1SDimitry Andric   NoLLVMBitcode = Args.hasArg(OPT_no_llvm_bc);
2352fe6060f1SDimitry Andric   NoSort = Args.hasArg(OPT_no_sort);
2353fe6060f1SDimitry Andric   NoWeakSymbols = Args.hasArg(OPT_no_weak);
2354fe6060f1SDimitry Andric   NumericSort = Args.hasArg(OPT_numeric_sort);
2355fe6060f1SDimitry Andric   ArchiveMap = Args.hasArg(OPT_print_armap);
2356fe6060f1SDimitry Andric   PrintFileName = Args.hasArg(OPT_print_file_name);
2357fe6060f1SDimitry Andric   PrintSize = Args.hasArg(OPT_print_size);
2358fe6060f1SDimitry Andric   ReverseSort = Args.hasArg(OPT_reverse_sort);
235981ad6265SDimitry Andric   ExportSymbols = Args.hasArg(OPT_export_symbols);
236081ad6265SDimitry Andric   if (ExportSymbols) {
236181ad6265SDimitry Andric     ExternalOnly = true;
236281ad6265SDimitry Andric     DefinedOnly = true;
236381ad6265SDimitry Andric   }
236481ad6265SDimitry Andric 
2365fe6060f1SDimitry Andric   Quiet = Args.hasArg(OPT_quiet);
2366fe6060f1SDimitry Andric   V = Args.getLastArgValue(OPT_radix_EQ, "x");
2367fe6060f1SDimitry Andric   if (V == "o")
2368fe6060f1SDimitry Andric     AddressRadix = Radix::o;
2369fe6060f1SDimitry Andric   else if (V == "d")
2370fe6060f1SDimitry Andric     AddressRadix = Radix::d;
2371fe6060f1SDimitry Andric   else if (V == "x")
2372fe6060f1SDimitry Andric     AddressRadix = Radix::x;
2373fe6060f1SDimitry Andric   else
2374fe6060f1SDimitry Andric     error("--radix value should be one of: 'o' (octal), 'd' (decimal), 'x' "
2375fe6060f1SDimitry Andric           "(hexadecimal)");
2376fe6060f1SDimitry Andric   SizeSort = Args.hasArg(OPT_size_sort);
2377fe6060f1SDimitry Andric   SpecialSyms = Args.hasArg(OPT_special_syms);
2378fe6060f1SDimitry Andric   UndefinedOnly = Args.hasArg(OPT_undefined_only);
2379fe6060f1SDimitry Andric   WithoutAliases = Args.hasArg(OPT_without_aliases);
2380fe6060f1SDimitry Andric 
2381*bdd1243dSDimitry Andric   // Get BitMode from enviornment variable "OBJECT_MODE" for AIX OS, if
2382*bdd1243dSDimitry Andric   // specified.
2383*bdd1243dSDimitry Andric   Triple HostTriple(sys::getProcessTriple());
2384*bdd1243dSDimitry Andric   if (HostTriple.isOSAIX()) {
2385*bdd1243dSDimitry Andric     BitMode = StringSwitch<BitModeTy>(getenv("OBJECT_MODE"))
2386*bdd1243dSDimitry Andric                   .Case("32", BitModeTy::Bit32)
2387*bdd1243dSDimitry Andric                   .Case("64", BitModeTy::Bit64)
2388*bdd1243dSDimitry Andric                   .Case("32_64", BitModeTy::Bit32_64)
2389*bdd1243dSDimitry Andric                   .Case("any", BitModeTy::Any)
2390*bdd1243dSDimitry Andric                   .Default(BitModeTy::Bit32);
2391*bdd1243dSDimitry Andric   } else
2392*bdd1243dSDimitry Andric     BitMode = BitModeTy::Any;
2393*bdd1243dSDimitry Andric 
2394*bdd1243dSDimitry Andric   if (Arg *A = Args.getLastArg(OPT_X)) {
2395*bdd1243dSDimitry Andric     StringRef Mode = A->getValue();
239681ad6265SDimitry Andric     if (Mode == "32")
239781ad6265SDimitry Andric       BitMode = BitModeTy::Bit32;
239881ad6265SDimitry Andric     else if (Mode == "64")
239981ad6265SDimitry Andric       BitMode = BitModeTy::Bit64;
240081ad6265SDimitry Andric     else if (Mode == "32_64")
240181ad6265SDimitry Andric       BitMode = BitModeTy::Bit32_64;
240281ad6265SDimitry Andric     else if (Mode == "any")
240381ad6265SDimitry Andric       BitMode = BitModeTy::Any;
240481ad6265SDimitry Andric     else
240581ad6265SDimitry Andric       error("-X value should be one of: 32, 64, 32_64, (default) any");
2406*bdd1243dSDimitry Andric   }
240781ad6265SDimitry Andric 
2408fe6060f1SDimitry Andric   // Mach-O specific options.
2409fe6060f1SDimitry Andric   FormatMachOasHex = Args.hasArg(OPT_x);
2410fe6060f1SDimitry Andric   AddDyldInfo = Args.hasArg(OPT_add_dyldinfo);
2411fe6060f1SDimitry Andric   AddInlinedInfo = Args.hasArg(OPT_add_inlinedinfo);
2412fe6060f1SDimitry Andric   DyldInfoOnly = Args.hasArg(OPT_dyldinfo_only);
2413fe6060f1SDimitry Andric   NoDyldInfo = Args.hasArg(OPT_no_dyldinfo);
24140b57cec5SDimitry Andric 
241581ad6265SDimitry Andric   // XCOFF specific options.
241681ad6265SDimitry Andric   NoRsrc = Args.hasArg(OPT_no_rsrc);
241781ad6265SDimitry Andric 
24180b57cec5SDimitry Andric   // llvm-nm only reads binary files.
24190b57cec5SDimitry Andric   if (error(sys::ChangeStdinToBinary()))
24200b57cec5SDimitry Andric     return 1;
24210b57cec5SDimitry Andric 
24220b57cec5SDimitry Andric   // These calls are needed so that we can read bitcode correctly.
24230b57cec5SDimitry Andric   llvm::InitializeAllTargetInfos();
24240b57cec5SDimitry Andric   llvm::InitializeAllTargetMCs();
24250b57cec5SDimitry Andric   llvm::InitializeAllAsmParsers();
24260b57cec5SDimitry Andric 
24270b57cec5SDimitry Andric   // The relative order of these is important. If you pass --size-sort it should
24280b57cec5SDimitry Andric   // only print out the size. However, if you pass -S --size-sort, it should
24290b57cec5SDimitry Andric   // print out both the size and address.
24300b57cec5SDimitry Andric   if (SizeSort && !PrintSize)
24310b57cec5SDimitry Andric     PrintAddress = false;
24320b57cec5SDimitry Andric   if (OutputFormat == sysv || SizeSort)
24330b57cec5SDimitry Andric     PrintSize = true;
24340b57cec5SDimitry Andric 
2435fe6060f1SDimitry Andric   for (const auto *A : Args.filtered(OPT_arch_EQ)) {
2436fe6060f1SDimitry Andric     SmallVector<StringRef, 2> Values;
2437fe6060f1SDimitry Andric     llvm::SplitString(A->getValue(), Values, ",");
2438fe6060f1SDimitry Andric     for (StringRef V : Values) {
2439fe6060f1SDimitry Andric       if (V == "all")
24400b57cec5SDimitry Andric         ArchAll = true;
2441fe6060f1SDimitry Andric       else if (MachOObjectFile::isValidArch(V))
2442fe6060f1SDimitry Andric         ArchFlags.push_back(V);
2443fe6060f1SDimitry Andric       else
2444fe6060f1SDimitry Andric         error("Unknown architecture named '" + V + "'",
24450b57cec5SDimitry Andric               "for the --arch option");
24460b57cec5SDimitry Andric     }
24470b57cec5SDimitry Andric   }
24480b57cec5SDimitry Andric 
2449fe6060f1SDimitry Andric   // Mach-O takes -s to accept two arguments. We emulate this by iterating over
2450fe6060f1SDimitry Andric   // both OPT_s and OPT_INPUT.
2451fe6060f1SDimitry Andric   std::vector<std::string> InputFilenames;
2452fe6060f1SDimitry Andric   int SegSectArgs = 0;
2453fe6060f1SDimitry Andric   for (opt::Arg *A : Args.filtered(OPT_s, OPT_INPUT)) {
2454fe6060f1SDimitry Andric     if (SegSectArgs > 0) {
2455fe6060f1SDimitry Andric       --SegSectArgs;
2456fe6060f1SDimitry Andric       SegSect.push_back(A->getValue());
2457fe6060f1SDimitry Andric     } else if (A->getOption().matches(OPT_s)) {
2458fe6060f1SDimitry Andric       SegSectArgs = 2;
2459fe6060f1SDimitry Andric     } else {
2460fe6060f1SDimitry Andric       InputFilenames.push_back(A->getValue());
2461fe6060f1SDimitry Andric     }
2462fe6060f1SDimitry Andric   }
24630b57cec5SDimitry Andric   if (!SegSect.empty() && SegSect.size() != 2)
24640b57cec5SDimitry Andric     error("bad number of arguments (must be two arguments)",
24650b57cec5SDimitry Andric           "for the -s option");
24660b57cec5SDimitry Andric 
2467fe6060f1SDimitry Andric   if (InputFilenames.empty())
2468fe6060f1SDimitry Andric     InputFilenames.push_back("a.out");
2469fe6060f1SDimitry Andric   if (InputFilenames.size() > 1)
2470fe6060f1SDimitry Andric     MultipleFiles = true;
2471fe6060f1SDimitry Andric 
24720b57cec5SDimitry Andric   if (NoDyldInfo && (AddDyldInfo || DyldInfoOnly))
24730b57cec5SDimitry Andric     error("--no-dyldinfo can't be used with --add-dyldinfo or --dyldinfo-only");
24740b57cec5SDimitry Andric 
247581ad6265SDimitry Andric   if (ExportSymbols)
247681ad6265SDimitry Andric     exportSymbolNamesFromFiles(InputFilenames);
247781ad6265SDimitry Andric   else
24780b57cec5SDimitry Andric     llvm::for_each(InputFilenames, dumpSymbolNamesFromFile);
24790b57cec5SDimitry Andric 
24800b57cec5SDimitry Andric   if (HadError)
24810b57cec5SDimitry Andric     return 1;
2482*bdd1243dSDimitry Andric   return 0;
24830b57cec5SDimitry Andric }
2484