xref: /freebsd/contrib/llvm-project/llvm/tools/llvm-nm/llvm-nm.cpp (revision fe6060f10f634930ff71b7c50291ddc610da2475)
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"
190b57cec5SDimitry Andric #include "llvm/BinaryFormat/COFF.h"
200b57cec5SDimitry Andric #include "llvm/Demangle/Demangle.h"
210b57cec5SDimitry Andric #include "llvm/IR/Function.h"
220b57cec5SDimitry Andric #include "llvm/IR/LLVMContext.h"
230b57cec5SDimitry Andric #include "llvm/Object/Archive.h"
240b57cec5SDimitry Andric #include "llvm/Object/COFF.h"
250b57cec5SDimitry Andric #include "llvm/Object/COFFImportFile.h"
260b57cec5SDimitry Andric #include "llvm/Object/ELFObjectFile.h"
270b57cec5SDimitry Andric #include "llvm/Object/IRObjectFile.h"
280b57cec5SDimitry Andric #include "llvm/Object/MachO.h"
290b57cec5SDimitry Andric #include "llvm/Object/MachOUniversal.h"
300b57cec5SDimitry Andric #include "llvm/Object/ObjectFile.h"
315ffd83dbSDimitry Andric #include "llvm/Object/TapiFile.h"
325ffd83dbSDimitry Andric #include "llvm/Object/TapiUniversal.h"
330b57cec5SDimitry Andric #include "llvm/Object/Wasm.h"
34*fe6060f1SDimitry Andric #include "llvm/Option/Arg.h"
35*fe6060f1SDimitry Andric #include "llvm/Option/ArgList.h"
36*fe6060f1SDimitry Andric #include "llvm/Option/Option.h"
370b57cec5SDimitry Andric #include "llvm/Support/CommandLine.h"
380b57cec5SDimitry Andric #include "llvm/Support/FileSystem.h"
390b57cec5SDimitry Andric #include "llvm/Support/Format.h"
400b57cec5SDimitry Andric #include "llvm/Support/InitLLVM.h"
410b57cec5SDimitry Andric #include "llvm/Support/MemoryBuffer.h"
420b57cec5SDimitry Andric #include "llvm/Support/Program.h"
430b57cec5SDimitry Andric #include "llvm/Support/Signals.h"
440b57cec5SDimitry Andric #include "llvm/Support/TargetSelect.h"
450b57cec5SDimitry Andric #include "llvm/Support/WithColor.h"
460b57cec5SDimitry Andric #include "llvm/Support/raw_ostream.h"
470b57cec5SDimitry Andric #include <vector>
480b57cec5SDimitry Andric 
490b57cec5SDimitry Andric using namespace llvm;
500b57cec5SDimitry Andric using namespace object;
510b57cec5SDimitry Andric 
520b57cec5SDimitry Andric namespace {
53*fe6060f1SDimitry Andric using namespace llvm::opt; // for HelpHidden in Opts.inc
54*fe6060f1SDimitry Andric enum ID {
55*fe6060f1SDimitry Andric   OPT_INVALID = 0, // This is not an option ID.
56*fe6060f1SDimitry Andric #define OPTION(PREFIX, NAME, ID, KIND, GROUP, ALIAS, ALIASARGS, FLAGS, PARAM,  \
57*fe6060f1SDimitry Andric                HELPTEXT, METAVAR, VALUES)                                      \
58*fe6060f1SDimitry Andric   OPT_##ID,
59*fe6060f1SDimitry Andric #include "Opts.inc"
60*fe6060f1SDimitry Andric #undef OPTION
61*fe6060f1SDimitry Andric };
620b57cec5SDimitry Andric 
63*fe6060f1SDimitry Andric #define PREFIX(NAME, VALUE) const char *const NAME[] = VALUE;
64*fe6060f1SDimitry Andric #include "Opts.inc"
65*fe6060f1SDimitry Andric #undef PREFIX
660b57cec5SDimitry Andric 
67*fe6060f1SDimitry Andric static const opt::OptTable::Info InfoTable[] = {
68*fe6060f1SDimitry Andric #define OPTION(PREFIX, NAME, ID, KIND, GROUP, ALIAS, ALIASARGS, FLAGS, PARAM,  \
69*fe6060f1SDimitry Andric                HELPTEXT, METAVAR, VALUES)                                      \
70*fe6060f1SDimitry Andric   {                                                                            \
71*fe6060f1SDimitry Andric       PREFIX,      NAME,      HELPTEXT,                                        \
72*fe6060f1SDimitry Andric       METAVAR,     OPT_##ID,  opt::Option::KIND##Class,                        \
73*fe6060f1SDimitry Andric       PARAM,       FLAGS,     OPT_##GROUP,                                     \
74*fe6060f1SDimitry Andric       OPT_##ALIAS, ALIASARGS, VALUES},
75*fe6060f1SDimitry Andric #include "Opts.inc"
76*fe6060f1SDimitry Andric #undef OPTION
77*fe6060f1SDimitry Andric };
780b57cec5SDimitry Andric 
79*fe6060f1SDimitry Andric class NmOptTable : public opt::OptTable {
80*fe6060f1SDimitry Andric public:
81*fe6060f1SDimitry Andric   NmOptTable() : OptTable(InfoTable) { setGroupedShortOptions(true); }
82*fe6060f1SDimitry Andric };
830b57cec5SDimitry Andric 
84*fe6060f1SDimitry Andric enum OutputFormatTy { bsd, sysv, posix, darwin, just_symbols };
85*fe6060f1SDimitry Andric } // namespace
860b57cec5SDimitry Andric 
87*fe6060f1SDimitry Andric static bool ArchiveMap;
88*fe6060f1SDimitry Andric static bool DebugSyms;
89*fe6060f1SDimitry Andric static bool DefinedOnly;
90*fe6060f1SDimitry Andric static bool Demangle;
91*fe6060f1SDimitry Andric static bool DynamicSyms;
92*fe6060f1SDimitry Andric static bool ExternalOnly;
93*fe6060f1SDimitry Andric static OutputFormatTy OutputFormat;
94*fe6060f1SDimitry Andric static bool NoLLVMBitcode;
95*fe6060f1SDimitry Andric static bool NoSort;
96*fe6060f1SDimitry Andric static bool NoWeakSymbols;
97*fe6060f1SDimitry Andric static bool NumericSort;
98*fe6060f1SDimitry Andric static bool PrintFileName;
99*fe6060f1SDimitry Andric static bool PrintSize;
100*fe6060f1SDimitry Andric static bool Quiet;
101*fe6060f1SDimitry Andric static bool ReverseSort;
102*fe6060f1SDimitry Andric static bool SpecialSyms;
103*fe6060f1SDimitry Andric static bool SizeSort;
104*fe6060f1SDimitry Andric static bool UndefinedOnly;
105*fe6060f1SDimitry Andric static bool WithoutAliases;
1060b57cec5SDimitry Andric 
107*fe6060f1SDimitry Andric namespace {
1080b57cec5SDimitry Andric enum Radix { d, o, x };
109*fe6060f1SDimitry Andric } // namespace
110*fe6060f1SDimitry Andric static Radix AddressRadix;
1110b57cec5SDimitry Andric 
112*fe6060f1SDimitry Andric // Mach-O specific options.
113*fe6060f1SDimitry Andric static bool ArchAll = false;
114*fe6060f1SDimitry Andric static std::vector<StringRef> ArchFlags;
115*fe6060f1SDimitry Andric static bool AddDyldInfo;
116*fe6060f1SDimitry Andric static bool AddInlinedInfo;
117*fe6060f1SDimitry Andric static bool DyldInfoOnly;
118*fe6060f1SDimitry Andric static bool FormatMachOasHex;
119*fe6060f1SDimitry Andric static bool NoDyldInfo;
120*fe6060f1SDimitry Andric static std::vector<StringRef> SegSect;
121*fe6060f1SDimitry Andric static bool MachOPrintSizeWarning = false;
1220b57cec5SDimitry Andric 
123*fe6060f1SDimitry Andric // Miscellaneous states.
124*fe6060f1SDimitry Andric static bool PrintAddress = true;
125*fe6060f1SDimitry Andric static bool MultipleFiles = false;
126*fe6060f1SDimitry Andric static bool HadError = false;
1270b57cec5SDimitry Andric 
128*fe6060f1SDimitry Andric static StringRef ToolName;
1290b57cec5SDimitry Andric 
1300b57cec5SDimitry Andric static void error(Twine Message, Twine Path = Twine()) {
1310b57cec5SDimitry Andric   HadError = true;
132*fe6060f1SDimitry Andric   WithColor::error(errs(), ToolName) << Path << ": " << Message << "\n";
1330b57cec5SDimitry Andric }
1340b57cec5SDimitry Andric 
1350b57cec5SDimitry Andric static bool error(std::error_code EC, Twine Path = Twine()) {
1360b57cec5SDimitry Andric   if (EC) {
1370b57cec5SDimitry Andric     error(EC.message(), Path);
1380b57cec5SDimitry Andric     return true;
1390b57cec5SDimitry Andric   }
1400b57cec5SDimitry Andric   return false;
1410b57cec5SDimitry Andric }
1420b57cec5SDimitry Andric 
1430b57cec5SDimitry Andric // This version of error() prints the archive name and member name, for example:
1440b57cec5SDimitry Andric // "libx.a(foo.o)" after the ToolName before the error message.  It sets
1450b57cec5SDimitry Andric // HadError but returns allowing the code to move on to other archive members.
1460b57cec5SDimitry Andric static void error(llvm::Error E, StringRef FileName, const Archive::Child &C,
1470b57cec5SDimitry Andric                   StringRef ArchitectureName = StringRef()) {
1480b57cec5SDimitry Andric   HadError = true;
1490b57cec5SDimitry Andric   WithColor::error(errs(), ToolName) << FileName;
1500b57cec5SDimitry Andric 
1510b57cec5SDimitry Andric   Expected<StringRef> NameOrErr = C.getName();
1520b57cec5SDimitry Andric   // TODO: if we have a error getting the name then it would be nice to print
1530b57cec5SDimitry Andric   // the index of which archive member this is and or its offset in the
1540b57cec5SDimitry Andric   // archive instead of "???" as the name.
1550b57cec5SDimitry Andric   if (!NameOrErr) {
1560b57cec5SDimitry Andric     consumeError(NameOrErr.takeError());
1570b57cec5SDimitry Andric     errs() << "(" << "???" << ")";
1580b57cec5SDimitry Andric   } else
1590b57cec5SDimitry Andric     errs() << "(" << NameOrErr.get() << ")";
1600b57cec5SDimitry Andric 
1610b57cec5SDimitry Andric   if (!ArchitectureName.empty())
1620b57cec5SDimitry Andric     errs() << " (for architecture " << ArchitectureName << ")";
1630b57cec5SDimitry Andric 
1640b57cec5SDimitry Andric   std::string Buf;
1650b57cec5SDimitry Andric   raw_string_ostream OS(Buf);
1660b57cec5SDimitry Andric   logAllUnhandledErrors(std::move(E), OS);
1670b57cec5SDimitry Andric   OS.flush();
168*fe6060f1SDimitry Andric   errs() << ": " << Buf << "\n";
1690b57cec5SDimitry Andric }
1700b57cec5SDimitry Andric 
1710b57cec5SDimitry Andric // This version of error() prints the file name and which architecture slice it
1720b57cec5SDimitry Andric // is from, for example: "foo.o (for architecture i386)" after the ToolName
1730b57cec5SDimitry Andric // before the error message.  It sets HadError but returns allowing the code to
1740b57cec5SDimitry Andric // move on to other architecture slices.
1750b57cec5SDimitry Andric static void error(llvm::Error E, StringRef FileName,
1760b57cec5SDimitry Andric                   StringRef ArchitectureName = StringRef()) {
1770b57cec5SDimitry Andric   HadError = true;
1780b57cec5SDimitry Andric   WithColor::error(errs(), ToolName) << FileName;
1790b57cec5SDimitry Andric 
1800b57cec5SDimitry Andric   if (!ArchitectureName.empty())
1810b57cec5SDimitry Andric     errs() << " (for architecture " << ArchitectureName << ")";
1820b57cec5SDimitry Andric 
1830b57cec5SDimitry Andric   std::string Buf;
1840b57cec5SDimitry Andric   raw_string_ostream OS(Buf);
1850b57cec5SDimitry Andric   logAllUnhandledErrors(std::move(E), OS);
1860b57cec5SDimitry Andric   OS.flush();
187*fe6060f1SDimitry Andric   errs() << ": " << Buf << "\n";
1880b57cec5SDimitry Andric }
1890b57cec5SDimitry Andric 
1900b57cec5SDimitry Andric namespace {
1910b57cec5SDimitry Andric struct NMSymbol {
1920b57cec5SDimitry Andric   uint64_t Address;
1930b57cec5SDimitry Andric   uint64_t Size;
1940b57cec5SDimitry Andric   char TypeChar;
195e8d8bef9SDimitry Andric   std::string Name;
1960b57cec5SDimitry Andric   StringRef SectionName;
1970b57cec5SDimitry Andric   StringRef TypeName;
1980b57cec5SDimitry Andric   BasicSymbolRef Sym;
1990b57cec5SDimitry Andric   // The Sym field above points to the native symbol in the object file,
2000b57cec5SDimitry Andric   // for Mach-O when we are creating symbols from the dyld info the above
2010b57cec5SDimitry Andric   // pointer is null as there is no native symbol.  In these cases the fields
2020b57cec5SDimitry Andric   // below are filled in to represent what would have been a Mach-O nlist
2030b57cec5SDimitry Andric   // native symbol.
2040b57cec5SDimitry Andric   uint32_t SymFlags;
2050b57cec5SDimitry Andric   SectionRef Section;
2060b57cec5SDimitry Andric   uint8_t NType;
2070b57cec5SDimitry Andric   uint8_t NSect;
2080b57cec5SDimitry Andric   uint16_t NDesc;
209e8d8bef9SDimitry Andric   std::string IndirectName;
2100b57cec5SDimitry Andric };
2110b57cec5SDimitry Andric } // anonymous namespace
2120b57cec5SDimitry Andric 
2130b57cec5SDimitry Andric static bool compareSymbolAddress(const NMSymbol &A, const NMSymbol &B) {
2140b57cec5SDimitry Andric   bool ADefined;
2155ffd83dbSDimitry Andric   // Symbol flags have been checked in the caller.
216e8d8bef9SDimitry Andric   if (A.Sym.getRawDataRefImpl().p) {
2175ffd83dbSDimitry Andric     uint32_t AFlags = cantFail(A.Sym.getFlags());
2185ffd83dbSDimitry Andric     ADefined = !(AFlags & SymbolRef::SF_Undefined);
219e8d8bef9SDimitry Andric   } else {
2200b57cec5SDimitry Andric     ADefined = A.TypeChar != 'U';
221e8d8bef9SDimitry Andric   }
2220b57cec5SDimitry Andric   bool BDefined;
2235ffd83dbSDimitry Andric   // Symbol flags have been checked in the caller.
224e8d8bef9SDimitry Andric   if (B.Sym.getRawDataRefImpl().p) {
2255ffd83dbSDimitry Andric     uint32_t BFlags = cantFail(B.Sym.getFlags());
2265ffd83dbSDimitry Andric     BDefined = !(BFlags & SymbolRef::SF_Undefined);
227e8d8bef9SDimitry Andric   } else {
2280b57cec5SDimitry Andric     BDefined = B.TypeChar != 'U';
229e8d8bef9SDimitry Andric   }
2300b57cec5SDimitry Andric   return std::make_tuple(ADefined, A.Address, A.Name, A.Size) <
2310b57cec5SDimitry Andric          std::make_tuple(BDefined, B.Address, B.Name, B.Size);
2320b57cec5SDimitry Andric }
2330b57cec5SDimitry Andric 
2340b57cec5SDimitry Andric static bool compareSymbolSize(const NMSymbol &A, const NMSymbol &B) {
2350b57cec5SDimitry Andric   return std::make_tuple(A.Size, A.Name, A.Address) <
2360b57cec5SDimitry Andric          std::make_tuple(B.Size, B.Name, B.Address);
2370b57cec5SDimitry Andric }
2380b57cec5SDimitry Andric 
2390b57cec5SDimitry Andric static bool compareSymbolName(const NMSymbol &A, const NMSymbol &B) {
2400b57cec5SDimitry Andric   return std::make_tuple(A.Name, A.Size, A.Address) <
2410b57cec5SDimitry Andric          std::make_tuple(B.Name, B.Size, B.Address);
2420b57cec5SDimitry Andric }
2430b57cec5SDimitry Andric 
2440b57cec5SDimitry Andric static char isSymbolList64Bit(SymbolicFile &Obj) {
2450b57cec5SDimitry Andric   if (auto *IRObj = dyn_cast<IRObjectFile>(&Obj))
2460b57cec5SDimitry Andric     return Triple(IRObj->getTargetTriple()).isArch64Bit();
2470b57cec5SDimitry Andric   if (isa<COFFObjectFile>(Obj) || isa<COFFImportFile>(Obj))
2480b57cec5SDimitry Andric     return false;
2490b57cec5SDimitry Andric   if (isa<WasmObjectFile>(Obj))
2500b57cec5SDimitry Andric     return false;
2515ffd83dbSDimitry Andric   if (TapiFile *Tapi = dyn_cast<TapiFile>(&Obj))
2525ffd83dbSDimitry Andric     return Tapi->is64Bit();
2530b57cec5SDimitry Andric   if (MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(&Obj))
2540b57cec5SDimitry Andric     return MachO->is64Bit();
2550b57cec5SDimitry Andric   return cast<ELFObjectFileBase>(Obj).getBytesInAddress() == 8;
2560b57cec5SDimitry Andric }
2570b57cec5SDimitry Andric 
2580b57cec5SDimitry Andric static StringRef CurrentFilename;
2590b57cec5SDimitry Andric static std::vector<NMSymbol> SymbolList;
2600b57cec5SDimitry Andric 
2610b57cec5SDimitry Andric static char getSymbolNMTypeChar(IRObjectFile &Obj, basic_symbol_iterator I);
2620b57cec5SDimitry Andric 
2630b57cec5SDimitry Andric // darwinPrintSymbol() is used to print a symbol from a Mach-O file when the
2640b57cec5SDimitry Andric // the OutputFormat is darwin or we are printing Mach-O symbols in hex.  For
2650b57cec5SDimitry Andric // the darwin format it produces the same output as darwin's nm(1) -m output
2660b57cec5SDimitry Andric // and when printing Mach-O symbols in hex it produces the same output as
2670b57cec5SDimitry Andric // darwin's nm(1) -x format.
2680b57cec5SDimitry Andric static void darwinPrintSymbol(SymbolicFile &Obj, const NMSymbol &S,
2690b57cec5SDimitry Andric                               char *SymbolAddrStr, const char *printBlanks,
2700b57cec5SDimitry Andric                               const char *printDashes,
2710b57cec5SDimitry Andric                               const char *printFormat) {
2720b57cec5SDimitry Andric   MachO::mach_header H;
2730b57cec5SDimitry Andric   MachO::mach_header_64 H_64;
2740b57cec5SDimitry Andric   uint32_t Filetype = MachO::MH_OBJECT;
2750b57cec5SDimitry Andric   uint32_t Flags = 0;
2760b57cec5SDimitry Andric   uint8_t NType = 0;
2770b57cec5SDimitry Andric   uint8_t NSect = 0;
2780b57cec5SDimitry Andric   uint16_t NDesc = 0;
2790b57cec5SDimitry Andric   uint32_t NStrx = 0;
2800b57cec5SDimitry Andric   uint64_t NValue = 0;
2810b57cec5SDimitry Andric   MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(&Obj);
2820b57cec5SDimitry Andric   if (Obj.isIR()) {
2835ffd83dbSDimitry Andric     uint32_t SymFlags = cantFail(S.Sym.getFlags());
2840b57cec5SDimitry Andric     if (SymFlags & SymbolRef::SF_Global)
2850b57cec5SDimitry Andric       NType |= MachO::N_EXT;
2860b57cec5SDimitry Andric     if (SymFlags & SymbolRef::SF_Hidden)
2870b57cec5SDimitry Andric       NType |= MachO::N_PEXT;
2880b57cec5SDimitry Andric     if (SymFlags & SymbolRef::SF_Undefined)
2890b57cec5SDimitry Andric       NType |= MachO::N_EXT | MachO::N_UNDF;
2900b57cec5SDimitry Andric     else {
2910b57cec5SDimitry Andric       // Here we have a symbol definition.  So to fake out a section name we
2920b57cec5SDimitry Andric       // use 1, 2 and 3 for section numbers.  See below where they are used to
2930b57cec5SDimitry Andric       // print out fake section names.
2940b57cec5SDimitry Andric       NType |= MachO::N_SECT;
2950b57cec5SDimitry Andric       if (SymFlags & SymbolRef::SF_Const)
2960b57cec5SDimitry Andric         NSect = 3;
2970b57cec5SDimitry Andric       else if (SymFlags & SymbolRef::SF_Executable)
2980b57cec5SDimitry Andric         NSect = 1;
2990b57cec5SDimitry Andric       else
3000b57cec5SDimitry Andric         NSect = 2;
3010b57cec5SDimitry Andric     }
3020b57cec5SDimitry Andric     if (SymFlags & SymbolRef::SF_Weak)
3030b57cec5SDimitry Andric       NDesc |= MachO::N_WEAK_DEF;
3040b57cec5SDimitry Andric   } else {
3050b57cec5SDimitry Andric     DataRefImpl SymDRI = S.Sym.getRawDataRefImpl();
3060b57cec5SDimitry Andric     if (MachO->is64Bit()) {
3070b57cec5SDimitry Andric       H_64 = MachO->MachOObjectFile::getHeader64();
3080b57cec5SDimitry Andric       Filetype = H_64.filetype;
3090b57cec5SDimitry Andric       Flags = H_64.flags;
3100b57cec5SDimitry Andric       if (SymDRI.p){
3110b57cec5SDimitry Andric         MachO::nlist_64 STE_64 = MachO->getSymbol64TableEntry(SymDRI);
3120b57cec5SDimitry Andric         NType = STE_64.n_type;
3130b57cec5SDimitry Andric         NSect = STE_64.n_sect;
3140b57cec5SDimitry Andric         NDesc = STE_64.n_desc;
3150b57cec5SDimitry Andric         NStrx = STE_64.n_strx;
3160b57cec5SDimitry Andric         NValue = STE_64.n_value;
3170b57cec5SDimitry Andric       } else {
3180b57cec5SDimitry Andric         NType = S.NType;
3190b57cec5SDimitry Andric         NSect = S.NSect;
3200b57cec5SDimitry Andric         NDesc = S.NDesc;
3210b57cec5SDimitry Andric         NStrx = 0;
3220b57cec5SDimitry Andric         NValue = S.Address;
3230b57cec5SDimitry Andric       }
3240b57cec5SDimitry Andric     } else {
3250b57cec5SDimitry Andric       H = MachO->MachOObjectFile::getHeader();
3260b57cec5SDimitry Andric       Filetype = H.filetype;
3270b57cec5SDimitry Andric       Flags = H.flags;
3280b57cec5SDimitry Andric       if (SymDRI.p){
3290b57cec5SDimitry Andric         MachO::nlist STE = MachO->getSymbolTableEntry(SymDRI);
3300b57cec5SDimitry Andric         NType = STE.n_type;
3310b57cec5SDimitry Andric         NSect = STE.n_sect;
3320b57cec5SDimitry Andric         NDesc = STE.n_desc;
3330b57cec5SDimitry Andric         NStrx = STE.n_strx;
3340b57cec5SDimitry Andric         NValue = STE.n_value;
3350b57cec5SDimitry Andric       } else {
3360b57cec5SDimitry Andric         NType = S.NType;
3370b57cec5SDimitry Andric         NSect = S.NSect;
3380b57cec5SDimitry Andric         NDesc = S.NDesc;
3390b57cec5SDimitry Andric         NStrx = 0;
3400b57cec5SDimitry Andric         NValue = S.Address;
3410b57cec5SDimitry Andric       }
3420b57cec5SDimitry Andric     }
3430b57cec5SDimitry Andric   }
3440b57cec5SDimitry Andric 
3450b57cec5SDimitry Andric   // If we are printing Mach-O symbols in hex do that and return.
3460b57cec5SDimitry Andric   if (FormatMachOasHex) {
3470b57cec5SDimitry Andric     outs() << format(printFormat, NValue) << ' '
3480b57cec5SDimitry Andric            << format("%02x %02x %04x %08x", NType, NSect, NDesc, NStrx) << ' '
3490b57cec5SDimitry Andric            << S.Name;
3500b57cec5SDimitry Andric     if ((NType & MachO::N_TYPE) == MachO::N_INDR) {
3510b57cec5SDimitry Andric       outs() << " (indirect for ";
3520b57cec5SDimitry Andric       outs() << format(printFormat, NValue) << ' ';
3530b57cec5SDimitry Andric       StringRef IndirectName;
3540b57cec5SDimitry Andric       if (S.Sym.getRawDataRefImpl().p) {
3550b57cec5SDimitry Andric         if (MachO->getIndirectName(S.Sym.getRawDataRefImpl(), IndirectName))
3560b57cec5SDimitry Andric           outs() << "?)";
3570b57cec5SDimitry Andric         else
3580b57cec5SDimitry Andric           outs() << IndirectName << ")";
3590b57cec5SDimitry Andric       } else
3600b57cec5SDimitry Andric         outs() << S.IndirectName << ")";
3610b57cec5SDimitry Andric     }
3620b57cec5SDimitry Andric     outs() << "\n";
3630b57cec5SDimitry Andric     return;
3640b57cec5SDimitry Andric   }
3650b57cec5SDimitry Andric 
3660b57cec5SDimitry Andric   if (PrintAddress) {
3670b57cec5SDimitry Andric     if ((NType & MachO::N_TYPE) == MachO::N_INDR)
3680b57cec5SDimitry Andric       strcpy(SymbolAddrStr, printBlanks);
3690b57cec5SDimitry Andric     if (Obj.isIR() && (NType & MachO::N_TYPE) == MachO::N_TYPE)
3700b57cec5SDimitry Andric       strcpy(SymbolAddrStr, printDashes);
3710b57cec5SDimitry Andric     outs() << SymbolAddrStr << ' ';
3720b57cec5SDimitry Andric   }
3730b57cec5SDimitry Andric 
3740b57cec5SDimitry Andric   switch (NType & MachO::N_TYPE) {
3750b57cec5SDimitry Andric   case MachO::N_UNDF:
3760b57cec5SDimitry Andric     if (NValue != 0) {
3770b57cec5SDimitry Andric       outs() << "(common) ";
3780b57cec5SDimitry Andric       if (MachO::GET_COMM_ALIGN(NDesc) != 0)
3790b57cec5SDimitry Andric         outs() << "(alignment 2^" << (int)MachO::GET_COMM_ALIGN(NDesc) << ") ";
3800b57cec5SDimitry Andric     } else {
3810b57cec5SDimitry Andric       if ((NType & MachO::N_TYPE) == MachO::N_PBUD)
3820b57cec5SDimitry Andric         outs() << "(prebound ";
3830b57cec5SDimitry Andric       else
3840b57cec5SDimitry Andric         outs() << "(";
3850b57cec5SDimitry Andric       if ((NDesc & MachO::REFERENCE_TYPE) ==
3860b57cec5SDimitry Andric           MachO::REFERENCE_FLAG_UNDEFINED_LAZY)
3870b57cec5SDimitry Andric         outs() << "undefined [lazy bound]) ";
3880b57cec5SDimitry Andric       else if ((NDesc & MachO::REFERENCE_TYPE) ==
3890b57cec5SDimitry Andric                MachO::REFERENCE_FLAG_PRIVATE_UNDEFINED_LAZY)
3900b57cec5SDimitry Andric         outs() << "undefined [private lazy bound]) ";
3910b57cec5SDimitry Andric       else if ((NDesc & MachO::REFERENCE_TYPE) ==
3920b57cec5SDimitry Andric                MachO::REFERENCE_FLAG_PRIVATE_UNDEFINED_NON_LAZY)
3930b57cec5SDimitry Andric         outs() << "undefined [private]) ";
3940b57cec5SDimitry Andric       else
3950b57cec5SDimitry Andric         outs() << "undefined) ";
3960b57cec5SDimitry Andric     }
3970b57cec5SDimitry Andric     break;
3980b57cec5SDimitry Andric   case MachO::N_ABS:
3990b57cec5SDimitry Andric     outs() << "(absolute) ";
4000b57cec5SDimitry Andric     break;
4010b57cec5SDimitry Andric   case MachO::N_INDR:
4020b57cec5SDimitry Andric     outs() << "(indirect) ";
4030b57cec5SDimitry Andric     break;
4040b57cec5SDimitry Andric   case MachO::N_SECT: {
4050b57cec5SDimitry Andric     if (Obj.isIR()) {
4060b57cec5SDimitry Andric       // For llvm bitcode files print out a fake section name using the values
4070b57cec5SDimitry Andric       // use 1, 2 and 3 for section numbers as set above.
4080b57cec5SDimitry Andric       if (NSect == 1)
4090b57cec5SDimitry Andric         outs() << "(LTO,CODE) ";
4100b57cec5SDimitry Andric       else if (NSect == 2)
4110b57cec5SDimitry Andric         outs() << "(LTO,DATA) ";
4120b57cec5SDimitry Andric       else if (NSect == 3)
4130b57cec5SDimitry Andric         outs() << "(LTO,RODATA) ";
4140b57cec5SDimitry Andric       else
4150b57cec5SDimitry Andric         outs() << "(?,?) ";
4160b57cec5SDimitry Andric       break;
4170b57cec5SDimitry Andric     }
4180b57cec5SDimitry Andric     section_iterator Sec = SectionRef();
4190b57cec5SDimitry Andric     if (S.Sym.getRawDataRefImpl().p) {
4200b57cec5SDimitry Andric       Expected<section_iterator> SecOrErr =
4210b57cec5SDimitry Andric           MachO->getSymbolSection(S.Sym.getRawDataRefImpl());
4220b57cec5SDimitry Andric       if (!SecOrErr) {
4230b57cec5SDimitry Andric         consumeError(SecOrErr.takeError());
4240b57cec5SDimitry Andric         outs() << "(?,?) ";
4250b57cec5SDimitry Andric         break;
4260b57cec5SDimitry Andric       }
4270b57cec5SDimitry Andric       Sec = *SecOrErr;
4280b57cec5SDimitry Andric       if (Sec == MachO->section_end()) {
4290b57cec5SDimitry Andric         outs() << "(?,?) ";
4300b57cec5SDimitry Andric         break;
4310b57cec5SDimitry Andric       }
4320b57cec5SDimitry Andric     } else {
4330b57cec5SDimitry Andric       Sec = S.Section;
4340b57cec5SDimitry Andric     }
4350b57cec5SDimitry Andric     DataRefImpl Ref = Sec->getRawDataRefImpl();
4360b57cec5SDimitry Andric     StringRef SectionName;
4370b57cec5SDimitry Andric     if (Expected<StringRef> NameOrErr = MachO->getSectionName(Ref))
4380b57cec5SDimitry Andric       SectionName = *NameOrErr;
4390b57cec5SDimitry Andric     StringRef SegmentName = MachO->getSectionFinalSegmentName(Ref);
4400b57cec5SDimitry Andric     outs() << "(" << SegmentName << "," << SectionName << ") ";
4410b57cec5SDimitry Andric     break;
4420b57cec5SDimitry Andric   }
4430b57cec5SDimitry Andric   default:
4440b57cec5SDimitry Andric     outs() << "(?) ";
4450b57cec5SDimitry Andric     break;
4460b57cec5SDimitry Andric   }
4470b57cec5SDimitry Andric 
4480b57cec5SDimitry Andric   if (NType & MachO::N_EXT) {
4490b57cec5SDimitry Andric     if (NDesc & MachO::REFERENCED_DYNAMICALLY)
4500b57cec5SDimitry Andric       outs() << "[referenced dynamically] ";
4510b57cec5SDimitry Andric     if (NType & MachO::N_PEXT) {
4520b57cec5SDimitry Andric       if ((NDesc & MachO::N_WEAK_DEF) == MachO::N_WEAK_DEF)
4530b57cec5SDimitry Andric         outs() << "weak private external ";
4540b57cec5SDimitry Andric       else
4550b57cec5SDimitry Andric         outs() << "private external ";
4560b57cec5SDimitry Andric     } else {
4570b57cec5SDimitry Andric       if ((NDesc & MachO::N_WEAK_REF) == MachO::N_WEAK_REF ||
4580b57cec5SDimitry Andric           (NDesc & MachO::N_WEAK_DEF) == MachO::N_WEAK_DEF) {
4590b57cec5SDimitry Andric         if ((NDesc & (MachO::N_WEAK_REF | MachO::N_WEAK_DEF)) ==
4600b57cec5SDimitry Andric             (MachO::N_WEAK_REF | MachO::N_WEAK_DEF))
4610b57cec5SDimitry Andric           outs() << "weak external automatically hidden ";
4620b57cec5SDimitry Andric         else
4630b57cec5SDimitry Andric           outs() << "weak external ";
4640b57cec5SDimitry Andric       } else
4650b57cec5SDimitry Andric         outs() << "external ";
4660b57cec5SDimitry Andric     }
4670b57cec5SDimitry Andric   } else {
4680b57cec5SDimitry Andric     if (NType & MachO::N_PEXT)
4690b57cec5SDimitry Andric       outs() << "non-external (was a private external) ";
4700b57cec5SDimitry Andric     else
4710b57cec5SDimitry Andric       outs() << "non-external ";
4720b57cec5SDimitry Andric   }
4730b57cec5SDimitry Andric 
4740b57cec5SDimitry Andric   if (Filetype == MachO::MH_OBJECT) {
4750b57cec5SDimitry Andric     if (NDesc & MachO::N_NO_DEAD_STRIP)
4760b57cec5SDimitry Andric       outs() << "[no dead strip] ";
4770b57cec5SDimitry Andric     if ((NType & MachO::N_TYPE) != MachO::N_UNDF &&
4780b57cec5SDimitry Andric         NDesc & MachO::N_SYMBOL_RESOLVER)
4790b57cec5SDimitry Andric       outs() << "[symbol resolver] ";
4800b57cec5SDimitry Andric     if ((NType & MachO::N_TYPE) != MachO::N_UNDF && NDesc & MachO::N_ALT_ENTRY)
4810b57cec5SDimitry Andric       outs() << "[alt entry] ";
4820b57cec5SDimitry Andric     if ((NType & MachO::N_TYPE) != MachO::N_UNDF && NDesc & MachO::N_COLD_FUNC)
4830b57cec5SDimitry Andric       outs() << "[cold func] ";
4840b57cec5SDimitry Andric   }
4850b57cec5SDimitry Andric 
4860b57cec5SDimitry Andric   if ((NDesc & MachO::N_ARM_THUMB_DEF) == MachO::N_ARM_THUMB_DEF)
4870b57cec5SDimitry Andric     outs() << "[Thumb] ";
4880b57cec5SDimitry Andric 
4890b57cec5SDimitry Andric   if ((NType & MachO::N_TYPE) == MachO::N_INDR) {
4900b57cec5SDimitry Andric     outs() << S.Name << " (for ";
4910b57cec5SDimitry Andric     StringRef IndirectName;
4920b57cec5SDimitry Andric     if (MachO) {
4930b57cec5SDimitry Andric       if (S.Sym.getRawDataRefImpl().p) {
4940b57cec5SDimitry Andric         if (MachO->getIndirectName(S.Sym.getRawDataRefImpl(), IndirectName))
4950b57cec5SDimitry Andric           outs() << "?)";
4960b57cec5SDimitry Andric         else
4970b57cec5SDimitry Andric           outs() << IndirectName << ")";
4980b57cec5SDimitry Andric       } else
4990b57cec5SDimitry Andric         outs() << S.IndirectName << ")";
5000b57cec5SDimitry Andric     } else
5010b57cec5SDimitry Andric       outs() << "?)";
5020b57cec5SDimitry Andric   } else
5030b57cec5SDimitry Andric     outs() << S.Name;
5040b57cec5SDimitry Andric 
5050b57cec5SDimitry Andric   if ((Flags & MachO::MH_TWOLEVEL) == MachO::MH_TWOLEVEL &&
5060b57cec5SDimitry Andric       (((NType & MachO::N_TYPE) == MachO::N_UNDF && NValue == 0) ||
5070b57cec5SDimitry Andric        (NType & MachO::N_TYPE) == MachO::N_PBUD)) {
5080b57cec5SDimitry Andric     uint32_t LibraryOrdinal = MachO::GET_LIBRARY_ORDINAL(NDesc);
5090b57cec5SDimitry Andric     if (LibraryOrdinal != 0) {
5100b57cec5SDimitry Andric       if (LibraryOrdinal == MachO::EXECUTABLE_ORDINAL)
5110b57cec5SDimitry Andric         outs() << " (from executable)";
5120b57cec5SDimitry Andric       else if (LibraryOrdinal == MachO::DYNAMIC_LOOKUP_ORDINAL)
5130b57cec5SDimitry Andric         outs() << " (dynamically looked up)";
5140b57cec5SDimitry Andric       else {
5150b57cec5SDimitry Andric         StringRef LibraryName;
5160b57cec5SDimitry Andric         if (!MachO ||
5170b57cec5SDimitry Andric             MachO->getLibraryShortNameByIndex(LibraryOrdinal - 1, LibraryName))
5180b57cec5SDimitry Andric           outs() << " (from bad library ordinal " << LibraryOrdinal << ")";
5190b57cec5SDimitry Andric         else
5200b57cec5SDimitry Andric           outs() << " (from " << LibraryName << ")";
5210b57cec5SDimitry Andric       }
5220b57cec5SDimitry Andric     }
5230b57cec5SDimitry Andric   }
5240b57cec5SDimitry Andric 
5250b57cec5SDimitry Andric   outs() << "\n";
5260b57cec5SDimitry Andric }
5270b57cec5SDimitry Andric 
5280b57cec5SDimitry Andric // Table that maps Darwin's Mach-O stab constants to strings to allow printing.
5290b57cec5SDimitry Andric struct DarwinStabName {
5300b57cec5SDimitry Andric   uint8_t NType;
5310b57cec5SDimitry Andric   const char *Name;
5320b57cec5SDimitry Andric };
5330b57cec5SDimitry Andric static const struct DarwinStabName DarwinStabNames[] = {
5340b57cec5SDimitry Andric     {MachO::N_GSYM, "GSYM"},
5350b57cec5SDimitry Andric     {MachO::N_FNAME, "FNAME"},
5360b57cec5SDimitry Andric     {MachO::N_FUN, "FUN"},
5370b57cec5SDimitry Andric     {MachO::N_STSYM, "STSYM"},
5380b57cec5SDimitry Andric     {MachO::N_LCSYM, "LCSYM"},
5390b57cec5SDimitry Andric     {MachO::N_BNSYM, "BNSYM"},
5400b57cec5SDimitry Andric     {MachO::N_PC, "PC"},
5410b57cec5SDimitry Andric     {MachO::N_AST, "AST"},
5420b57cec5SDimitry Andric     {MachO::N_OPT, "OPT"},
5430b57cec5SDimitry Andric     {MachO::N_RSYM, "RSYM"},
5440b57cec5SDimitry Andric     {MachO::N_SLINE, "SLINE"},
5450b57cec5SDimitry Andric     {MachO::N_ENSYM, "ENSYM"},
5460b57cec5SDimitry Andric     {MachO::N_SSYM, "SSYM"},
5470b57cec5SDimitry Andric     {MachO::N_SO, "SO"},
5480b57cec5SDimitry Andric     {MachO::N_OSO, "OSO"},
5490b57cec5SDimitry Andric     {MachO::N_LSYM, "LSYM"},
5500b57cec5SDimitry Andric     {MachO::N_BINCL, "BINCL"},
5510b57cec5SDimitry Andric     {MachO::N_SOL, "SOL"},
5520b57cec5SDimitry Andric     {MachO::N_PARAMS, "PARAM"},
5530b57cec5SDimitry Andric     {MachO::N_VERSION, "VERS"},
5540b57cec5SDimitry Andric     {MachO::N_OLEVEL, "OLEV"},
5550b57cec5SDimitry Andric     {MachO::N_PSYM, "PSYM"},
5560b57cec5SDimitry Andric     {MachO::N_EINCL, "EINCL"},
5570b57cec5SDimitry Andric     {MachO::N_ENTRY, "ENTRY"},
5580b57cec5SDimitry Andric     {MachO::N_LBRAC, "LBRAC"},
5590b57cec5SDimitry Andric     {MachO::N_EXCL, "EXCL"},
5600b57cec5SDimitry Andric     {MachO::N_RBRAC, "RBRAC"},
5610b57cec5SDimitry Andric     {MachO::N_BCOMM, "BCOMM"},
5620b57cec5SDimitry Andric     {MachO::N_ECOMM, "ECOMM"},
5630b57cec5SDimitry Andric     {MachO::N_ECOML, "ECOML"},
5640b57cec5SDimitry Andric     {MachO::N_LENG, "LENG"},
5650b57cec5SDimitry Andric };
5660b57cec5SDimitry Andric 
5670b57cec5SDimitry Andric static const char *getDarwinStabString(uint8_t NType) {
5680b57cec5SDimitry Andric   for (auto I : makeArrayRef(DarwinStabNames))
5690b57cec5SDimitry Andric     if (I.NType == NType)
5700b57cec5SDimitry Andric       return I.Name;
5710b57cec5SDimitry Andric   return nullptr;
5720b57cec5SDimitry Andric }
5730b57cec5SDimitry Andric 
5740b57cec5SDimitry Andric // darwinPrintStab() prints the n_sect, n_desc along with a symbolic name of
5750b57cec5SDimitry Andric // a stab n_type value in a Mach-O file.
5760b57cec5SDimitry Andric static void darwinPrintStab(MachOObjectFile *MachO, const NMSymbol &S) {
5770b57cec5SDimitry Andric   MachO::nlist_64 STE_64;
5780b57cec5SDimitry Andric   MachO::nlist STE;
5790b57cec5SDimitry Andric   uint8_t NType;
5800b57cec5SDimitry Andric   uint8_t NSect;
5810b57cec5SDimitry Andric   uint16_t NDesc;
5820b57cec5SDimitry Andric   DataRefImpl SymDRI = S.Sym.getRawDataRefImpl();
5830b57cec5SDimitry Andric   if (MachO->is64Bit()) {
5840b57cec5SDimitry Andric     STE_64 = MachO->getSymbol64TableEntry(SymDRI);
5850b57cec5SDimitry Andric     NType = STE_64.n_type;
5860b57cec5SDimitry Andric     NSect = STE_64.n_sect;
5870b57cec5SDimitry Andric     NDesc = STE_64.n_desc;
5880b57cec5SDimitry Andric   } else {
5890b57cec5SDimitry Andric     STE = MachO->getSymbolTableEntry(SymDRI);
5900b57cec5SDimitry Andric     NType = STE.n_type;
5910b57cec5SDimitry Andric     NSect = STE.n_sect;
5920b57cec5SDimitry Andric     NDesc = STE.n_desc;
5930b57cec5SDimitry Andric   }
5940b57cec5SDimitry Andric 
5950b57cec5SDimitry Andric   outs() << format(" %02x %04x ", NSect, NDesc);
5960b57cec5SDimitry Andric   if (const char *stabString = getDarwinStabString(NType))
5970b57cec5SDimitry Andric     outs() << format("%5.5s", stabString);
5980b57cec5SDimitry Andric   else
5990b57cec5SDimitry Andric     outs() << format("   %02x", NType);
6000b57cec5SDimitry Andric }
6010b57cec5SDimitry Andric 
6020b57cec5SDimitry Andric static Optional<std::string> demangle(StringRef Name, bool StripUnderscore) {
6030b57cec5SDimitry Andric   if (StripUnderscore && !Name.empty() && Name[0] == '_')
6040b57cec5SDimitry Andric     Name = Name.substr(1);
6050b57cec5SDimitry Andric 
6060b57cec5SDimitry Andric   if (!Name.startswith("_Z"))
6070b57cec5SDimitry Andric     return None;
6080b57cec5SDimitry Andric 
6090b57cec5SDimitry Andric   int Status;
6100b57cec5SDimitry Andric   char *Undecorated =
6110b57cec5SDimitry Andric       itaniumDemangle(Name.str().c_str(), nullptr, nullptr, &Status);
6120b57cec5SDimitry Andric   if (Status != 0)
6130b57cec5SDimitry Andric     return None;
6140b57cec5SDimitry Andric 
6150b57cec5SDimitry Andric   std::string S(Undecorated);
6160b57cec5SDimitry Andric   free(Undecorated);
6170b57cec5SDimitry Andric   return S;
6180b57cec5SDimitry Andric }
6190b57cec5SDimitry Andric 
6200b57cec5SDimitry Andric static bool symbolIsDefined(const NMSymbol &Sym) {
6210b57cec5SDimitry Andric   return Sym.TypeChar != 'U' && Sym.TypeChar != 'w' && Sym.TypeChar != 'v';
6220b57cec5SDimitry Andric }
6230b57cec5SDimitry Andric 
6245ffd83dbSDimitry Andric static void writeFileName(raw_ostream &S, StringRef ArchiveName,
6255ffd83dbSDimitry Andric                           StringRef ArchitectureName) {
6265ffd83dbSDimitry Andric   if (!ArchitectureName.empty())
6275ffd83dbSDimitry Andric     S << "(for architecture " << ArchitectureName << "):";
6285ffd83dbSDimitry Andric   if (OutputFormat == posix && !ArchiveName.empty())
6295ffd83dbSDimitry Andric     S << ArchiveName << "[" << CurrentFilename << "]: ";
6305ffd83dbSDimitry Andric   else {
6315ffd83dbSDimitry Andric     if (!ArchiveName.empty())
6325ffd83dbSDimitry Andric       S << ArchiveName << ":";
6335ffd83dbSDimitry Andric     S << CurrentFilename << ": ";
6345ffd83dbSDimitry Andric   }
6355ffd83dbSDimitry Andric }
6365ffd83dbSDimitry Andric 
6370b57cec5SDimitry Andric static void sortAndPrintSymbolList(SymbolicFile &Obj, bool printName,
6385ffd83dbSDimitry Andric                                    StringRef ArchiveName,
6395ffd83dbSDimitry Andric                                    StringRef ArchitectureName) {
6400b57cec5SDimitry Andric   if (!NoSort) {
6418bcb0991SDimitry Andric     using Comparator = bool (*)(const NMSymbol &, const NMSymbol &);
6428bcb0991SDimitry Andric     Comparator Cmp;
6430b57cec5SDimitry Andric     if (NumericSort)
6448bcb0991SDimitry Andric       Cmp = &compareSymbolAddress;
6450b57cec5SDimitry Andric     else if (SizeSort)
6468bcb0991SDimitry Andric       Cmp = &compareSymbolSize;
6470b57cec5SDimitry Andric     else
6488bcb0991SDimitry Andric       Cmp = &compareSymbolName;
6490b57cec5SDimitry Andric 
6500b57cec5SDimitry Andric     if (ReverseSort)
6518bcb0991SDimitry Andric       llvm::sort(SymbolList, [=](const NMSymbol &A, const NMSymbol &B) -> bool {
6528bcb0991SDimitry Andric         return Cmp(B, A);
6538bcb0991SDimitry Andric       });
6548bcb0991SDimitry Andric     else
6550b57cec5SDimitry Andric       llvm::sort(SymbolList, Cmp);
6560b57cec5SDimitry Andric   }
6570b57cec5SDimitry Andric 
6580b57cec5SDimitry Andric   if (!PrintFileName) {
659*fe6060f1SDimitry Andric     if ((OutputFormat == bsd || OutputFormat == posix ||
660*fe6060f1SDimitry Andric          OutputFormat == just_symbols) &&
661*fe6060f1SDimitry Andric         MultipleFiles && printName) {
6620b57cec5SDimitry Andric       outs() << '\n' << CurrentFilename << ":\n";
6630b57cec5SDimitry Andric     } else if (OutputFormat == sysv) {
6640b57cec5SDimitry Andric       outs() << "\n\nSymbols from " << CurrentFilename << ":\n\n";
6650b57cec5SDimitry Andric       if (isSymbolList64Bit(Obj))
6660b57cec5SDimitry Andric         outs() << "Name                  Value           Class        Type"
6670b57cec5SDimitry Andric                << "         Size             Line  Section\n";
6680b57cec5SDimitry Andric       else
6690b57cec5SDimitry Andric         outs() << "Name                  Value   Class        Type"
6700b57cec5SDimitry Andric                << "         Size     Line  Section\n";
6710b57cec5SDimitry Andric     }
6720b57cec5SDimitry Andric   }
6730b57cec5SDimitry Andric 
6740b57cec5SDimitry Andric   const char *printBlanks, *printDashes, *printFormat;
6750b57cec5SDimitry Andric   if (isSymbolList64Bit(Obj)) {
6760b57cec5SDimitry Andric     printBlanks = "                ";
6770b57cec5SDimitry Andric     printDashes = "----------------";
6780b57cec5SDimitry Andric     switch (AddressRadix) {
6790b57cec5SDimitry Andric     case Radix::o:
6800b57cec5SDimitry Andric       printFormat = OutputFormat == posix ? "%" PRIo64 : "%016" PRIo64;
6810b57cec5SDimitry Andric       break;
6820b57cec5SDimitry Andric     case Radix::x:
6830b57cec5SDimitry Andric       printFormat = OutputFormat == posix ? "%" PRIx64 : "%016" PRIx64;
6840b57cec5SDimitry Andric       break;
6850b57cec5SDimitry Andric     default:
6860b57cec5SDimitry Andric       printFormat = OutputFormat == posix ? "%" PRId64 : "%016" PRId64;
6870b57cec5SDimitry Andric     }
6880b57cec5SDimitry Andric   } else {
6890b57cec5SDimitry Andric     printBlanks = "        ";
6900b57cec5SDimitry Andric     printDashes = "--------";
6910b57cec5SDimitry Andric     switch (AddressRadix) {
6920b57cec5SDimitry Andric     case Radix::o:
6930b57cec5SDimitry Andric       printFormat = OutputFormat == posix ? "%" PRIo64 : "%08" PRIo64;
6940b57cec5SDimitry Andric       break;
6950b57cec5SDimitry Andric     case Radix::x:
6960b57cec5SDimitry Andric       printFormat = OutputFormat == posix ? "%" PRIx64 : "%08" PRIx64;
6970b57cec5SDimitry Andric       break;
6980b57cec5SDimitry Andric     default:
6990b57cec5SDimitry Andric       printFormat = OutputFormat == posix ? "%" PRId64 : "%08" PRId64;
7000b57cec5SDimitry Andric     }
7010b57cec5SDimitry Andric   }
7020b57cec5SDimitry Andric 
7030b57cec5SDimitry Andric   for (const NMSymbol &S : SymbolList) {
7040b57cec5SDimitry Andric     uint32_t SymFlags;
705e8d8bef9SDimitry Andric     std::string Name = S.Name;
7060b57cec5SDimitry Andric     MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(&Obj);
7070b57cec5SDimitry Andric     if (Demangle) {
7080b57cec5SDimitry Andric       if (Optional<std::string> Opt = demangle(S.Name, MachO))
7090b57cec5SDimitry Andric         Name = *Opt;
7100b57cec5SDimitry Andric     }
7115ffd83dbSDimitry Andric     if (S.Sym.getRawDataRefImpl().p) {
7125ffd83dbSDimitry Andric       Expected<uint32_t> SymFlagsOrErr = S.Sym.getFlags();
7135ffd83dbSDimitry Andric       if (!SymFlagsOrErr) {
7145ffd83dbSDimitry Andric         // TODO: Test this error.
7155ffd83dbSDimitry Andric         error(SymFlagsOrErr.takeError(), Obj.getFileName());
7165ffd83dbSDimitry Andric         return;
7175ffd83dbSDimitry Andric       }
7185ffd83dbSDimitry Andric       SymFlags = *SymFlagsOrErr;
7195ffd83dbSDimitry Andric     } else
7200b57cec5SDimitry Andric       SymFlags = S.SymFlags;
7210b57cec5SDimitry Andric 
7220b57cec5SDimitry Andric     bool Undefined = SymFlags & SymbolRef::SF_Undefined;
7230b57cec5SDimitry Andric     bool Global = SymFlags & SymbolRef::SF_Global;
7240b57cec5SDimitry Andric     bool Weak = SymFlags & SymbolRef::SF_Weak;
725*fe6060f1SDimitry Andric     bool FormatSpecific = SymFlags & SymbolRef::SF_FormatSpecific;
7260b57cec5SDimitry Andric     if ((!Undefined && UndefinedOnly) || (Undefined && DefinedOnly) ||
7275ffd83dbSDimitry Andric         (!Global && ExternalOnly) || (Weak && NoWeakSymbols) ||
728*fe6060f1SDimitry Andric         (FormatSpecific && !(SpecialSyms || DebugSyms)))
7290b57cec5SDimitry Andric       continue;
7300b57cec5SDimitry Andric     if (PrintFileName)
7315ffd83dbSDimitry Andric       writeFileName(outs(), ArchiveName, ArchitectureName);
732*fe6060f1SDimitry Andric     if ((OutputFormat == just_symbols ||
7330b57cec5SDimitry Andric          (UndefinedOnly && MachO && OutputFormat != darwin)) &&
7340b57cec5SDimitry Andric         OutputFormat != posix) {
7350b57cec5SDimitry Andric       outs() << Name << "\n";
7360b57cec5SDimitry Andric       continue;
7370b57cec5SDimitry Andric     }
7380b57cec5SDimitry Andric 
7390b57cec5SDimitry Andric     char SymbolAddrStr[23], SymbolSizeStr[23];
7400b57cec5SDimitry Andric 
7410b57cec5SDimitry Andric     // If the format is SysV or the symbol isn't defined, then print spaces.
7420b57cec5SDimitry Andric     if (OutputFormat == sysv || !symbolIsDefined(S)) {
7430b57cec5SDimitry Andric       if (OutputFormat == posix) {
7440b57cec5SDimitry Andric         format(printFormat, S.Address)
7450b57cec5SDimitry Andric             .print(SymbolAddrStr, sizeof(SymbolAddrStr));
7460b57cec5SDimitry Andric         format(printFormat, S.Size).print(SymbolSizeStr, sizeof(SymbolSizeStr));
7470b57cec5SDimitry Andric       } else {
7480b57cec5SDimitry Andric         strcpy(SymbolAddrStr, printBlanks);
7490b57cec5SDimitry Andric         strcpy(SymbolSizeStr, printBlanks);
7500b57cec5SDimitry Andric       }
7510b57cec5SDimitry Andric     }
7520b57cec5SDimitry Andric 
7530b57cec5SDimitry Andric     if (symbolIsDefined(S)) {
7540b57cec5SDimitry Andric       // Otherwise, print the symbol address and size.
7550b57cec5SDimitry Andric       if (Obj.isIR())
7560b57cec5SDimitry Andric         strcpy(SymbolAddrStr, printDashes);
7570b57cec5SDimitry Andric       else if (MachO && S.TypeChar == 'I')
7580b57cec5SDimitry Andric         strcpy(SymbolAddrStr, printBlanks);
7590b57cec5SDimitry Andric       else
7600b57cec5SDimitry Andric         format(printFormat, S.Address)
7610b57cec5SDimitry Andric             .print(SymbolAddrStr, sizeof(SymbolAddrStr));
7620b57cec5SDimitry Andric       format(printFormat, S.Size).print(SymbolSizeStr, sizeof(SymbolSizeStr));
7630b57cec5SDimitry Andric     }
7640b57cec5SDimitry Andric 
7650b57cec5SDimitry Andric     // If OutputFormat is darwin or we are printing Mach-O symbols in hex and
7660b57cec5SDimitry Andric     // we have a MachOObjectFile, call darwinPrintSymbol to print as darwin's
7670b57cec5SDimitry Andric     // nm(1) -m output or hex, else if OutputFormat is darwin or we are
7680b57cec5SDimitry Andric     // printing Mach-O symbols in hex and not a Mach-O object fall back to
7690b57cec5SDimitry Andric     // OutputFormat bsd (see below).
7700b57cec5SDimitry Andric     if ((OutputFormat == darwin || FormatMachOasHex) && (MachO || Obj.isIR())) {
7710b57cec5SDimitry Andric       darwinPrintSymbol(Obj, S, SymbolAddrStr, printBlanks, printDashes,
7720b57cec5SDimitry Andric                         printFormat);
7730b57cec5SDimitry Andric     } else if (OutputFormat == posix) {
7740b57cec5SDimitry Andric       outs() << Name << " " << S.TypeChar << " " << SymbolAddrStr << " "
7750b57cec5SDimitry Andric              << (MachO ? "0" : SymbolSizeStr) << "\n";
7760b57cec5SDimitry Andric     } else if (OutputFormat == bsd || (OutputFormat == darwin && !MachO)) {
7770b57cec5SDimitry Andric       if (PrintAddress)
7780b57cec5SDimitry Andric         outs() << SymbolAddrStr << ' ';
7790b57cec5SDimitry Andric       if (PrintSize)
7800b57cec5SDimitry Andric         outs() << SymbolSizeStr << ' ';
7810b57cec5SDimitry Andric       outs() << S.TypeChar;
7820b57cec5SDimitry Andric       if (S.TypeChar == '-' && MachO)
7830b57cec5SDimitry Andric         darwinPrintStab(MachO, S);
7840b57cec5SDimitry Andric       outs() << " " << Name;
7850b57cec5SDimitry Andric       if (S.TypeChar == 'I' && MachO) {
7860b57cec5SDimitry Andric         outs() << " (indirect for ";
7870b57cec5SDimitry Andric         if (S.Sym.getRawDataRefImpl().p) {
7880b57cec5SDimitry Andric           StringRef IndirectName;
7890b57cec5SDimitry Andric           if (MachO->getIndirectName(S.Sym.getRawDataRefImpl(), IndirectName))
7900b57cec5SDimitry Andric             outs() << "?)";
7910b57cec5SDimitry Andric           else
7920b57cec5SDimitry Andric             outs() << IndirectName << ")";
7930b57cec5SDimitry Andric         } else
7940b57cec5SDimitry Andric           outs() << S.IndirectName << ")";
7950b57cec5SDimitry Andric       }
7960b57cec5SDimitry Andric       outs() << "\n";
7970b57cec5SDimitry Andric     } else if (OutputFormat == sysv) {
7980b57cec5SDimitry Andric       outs() << left_justify(Name, 20) << "|" << SymbolAddrStr << "|   "
7990b57cec5SDimitry Andric              << S.TypeChar << "  |" << right_justify(S.TypeName, 18) << "|"
8000b57cec5SDimitry Andric              << SymbolSizeStr << "|     |" << S.SectionName << "\n";
8010b57cec5SDimitry Andric     }
8020b57cec5SDimitry Andric   }
8030b57cec5SDimitry Andric 
8040b57cec5SDimitry Andric   SymbolList.clear();
8050b57cec5SDimitry Andric }
8060b57cec5SDimitry Andric 
8070b57cec5SDimitry Andric static char getSymbolNMTypeChar(ELFObjectFileBase &Obj,
8080b57cec5SDimitry Andric                                 basic_symbol_iterator I) {
8090b57cec5SDimitry Andric   // OK, this is ELF
8100b57cec5SDimitry Andric   elf_symbol_iterator SymI(I);
8110b57cec5SDimitry Andric 
8120b57cec5SDimitry Andric   Expected<elf_section_iterator> SecIOrErr = SymI->getSection();
8130b57cec5SDimitry Andric   if (!SecIOrErr) {
8140b57cec5SDimitry Andric     consumeError(SecIOrErr.takeError());
8150b57cec5SDimitry Andric     return '?';
8160b57cec5SDimitry Andric   }
8170b57cec5SDimitry Andric 
8180b57cec5SDimitry Andric   uint8_t Binding = SymI->getBinding();
8190b57cec5SDimitry Andric   if (Binding == ELF::STB_GNU_UNIQUE)
8200b57cec5SDimitry Andric     return 'u';
8210b57cec5SDimitry Andric 
8220b57cec5SDimitry Andric   assert(Binding != ELF::STB_WEAK && "STB_WEAK not tested in calling function");
8230b57cec5SDimitry Andric   if (Binding != ELF::STB_GLOBAL && Binding != ELF::STB_LOCAL)
8240b57cec5SDimitry Andric     return '?';
8250b57cec5SDimitry Andric 
8260b57cec5SDimitry Andric   elf_section_iterator SecI = *SecIOrErr;
8270b57cec5SDimitry Andric   if (SecI != Obj.section_end()) {
8280b57cec5SDimitry Andric     uint32_t Type = SecI->getType();
8290b57cec5SDimitry Andric     uint64_t Flags = SecI->getFlags();
8300b57cec5SDimitry Andric     if (Flags & ELF::SHF_EXECINSTR)
8310b57cec5SDimitry Andric       return 't';
8320b57cec5SDimitry Andric     if (Type == ELF::SHT_NOBITS)
8330b57cec5SDimitry Andric       return 'b';
8340b57cec5SDimitry Andric     if (Flags & ELF::SHF_ALLOC)
8350b57cec5SDimitry Andric       return Flags & ELF::SHF_WRITE ? 'd' : 'r';
8360b57cec5SDimitry Andric 
8378bcb0991SDimitry Andric     auto NameOrErr = SecI->getName();
8388bcb0991SDimitry Andric     if (!NameOrErr) {
8398bcb0991SDimitry Andric       consumeError(NameOrErr.takeError());
8400b57cec5SDimitry Andric       return '?';
8418bcb0991SDimitry Andric     }
8428bcb0991SDimitry Andric     if ((*NameOrErr).startswith(".debug"))
8430b57cec5SDimitry Andric       return 'N';
8440b57cec5SDimitry Andric     if (!(Flags & ELF::SHF_WRITE))
8450b57cec5SDimitry Andric       return 'n';
8460b57cec5SDimitry Andric   }
8470b57cec5SDimitry Andric 
8480b57cec5SDimitry Andric   return '?';
8490b57cec5SDimitry Andric }
8500b57cec5SDimitry Andric 
8510b57cec5SDimitry Andric static char getSymbolNMTypeChar(COFFObjectFile &Obj, symbol_iterator I) {
8520b57cec5SDimitry Andric   COFFSymbolRef Symb = Obj.getCOFFSymbol(*I);
8530b57cec5SDimitry Andric   // OK, this is COFF.
8540b57cec5SDimitry Andric   symbol_iterator SymI(I);
8550b57cec5SDimitry Andric 
8560b57cec5SDimitry Andric   Expected<StringRef> Name = SymI->getName();
8570b57cec5SDimitry Andric   if (!Name) {
8580b57cec5SDimitry Andric     consumeError(Name.takeError());
8590b57cec5SDimitry Andric     return '?';
8600b57cec5SDimitry Andric   }
8610b57cec5SDimitry Andric 
8620b57cec5SDimitry Andric   char Ret = StringSwitch<char>(*Name)
8630b57cec5SDimitry Andric                  .StartsWith(".debug", 'N')
8640b57cec5SDimitry Andric                  .StartsWith(".sxdata", 'N')
8650b57cec5SDimitry Andric                  .Default('?');
8660b57cec5SDimitry Andric 
8670b57cec5SDimitry Andric   if (Ret != '?')
8680b57cec5SDimitry Andric     return Ret;
8690b57cec5SDimitry Andric 
8700b57cec5SDimitry Andric   uint32_t Characteristics = 0;
8710b57cec5SDimitry Andric   if (!COFF::isReservedSectionNumber(Symb.getSectionNumber())) {
8720b57cec5SDimitry Andric     Expected<section_iterator> SecIOrErr = SymI->getSection();
8730b57cec5SDimitry Andric     if (!SecIOrErr) {
8740b57cec5SDimitry Andric       consumeError(SecIOrErr.takeError());
8750b57cec5SDimitry Andric       return '?';
8760b57cec5SDimitry Andric     }
8770b57cec5SDimitry Andric     section_iterator SecI = *SecIOrErr;
8780b57cec5SDimitry Andric     const coff_section *Section = Obj.getCOFFSection(*SecI);
8790b57cec5SDimitry Andric     Characteristics = Section->Characteristics;
8800b57cec5SDimitry Andric     if (Expected<StringRef> NameOrErr = Obj.getSectionName(Section))
8810b57cec5SDimitry Andric       if (NameOrErr->startswith(".idata"))
8820b57cec5SDimitry Andric         return 'i';
8830b57cec5SDimitry Andric   }
8840b57cec5SDimitry Andric 
8850b57cec5SDimitry Andric   switch (Symb.getSectionNumber()) {
8860b57cec5SDimitry Andric   case COFF::IMAGE_SYM_DEBUG:
8870b57cec5SDimitry Andric     return 'n';
8880b57cec5SDimitry Andric   default:
8890b57cec5SDimitry Andric     // Check section type.
8900b57cec5SDimitry Andric     if (Characteristics & COFF::IMAGE_SCN_CNT_CODE)
8910b57cec5SDimitry Andric       return 't';
8920b57cec5SDimitry Andric     if (Characteristics & COFF::IMAGE_SCN_CNT_INITIALIZED_DATA)
8930b57cec5SDimitry Andric       return Characteristics & COFF::IMAGE_SCN_MEM_WRITE ? 'd' : 'r';
8940b57cec5SDimitry Andric     if (Characteristics & COFF::IMAGE_SCN_CNT_UNINITIALIZED_DATA)
8950b57cec5SDimitry Andric       return 'b';
8960b57cec5SDimitry Andric     if (Characteristics & COFF::IMAGE_SCN_LNK_INFO)
8970b57cec5SDimitry Andric       return 'i';
8980b57cec5SDimitry Andric     // Check for section symbol.
8990b57cec5SDimitry Andric     if (Symb.isSectionDefinition())
9000b57cec5SDimitry Andric       return 's';
9010b57cec5SDimitry Andric   }
9020b57cec5SDimitry Andric 
9030b57cec5SDimitry Andric   return '?';
9040b57cec5SDimitry Andric }
9050b57cec5SDimitry Andric 
9060b57cec5SDimitry Andric static char getSymbolNMTypeChar(COFFImportFile &Obj) {
9070b57cec5SDimitry Andric   switch (Obj.getCOFFImportHeader()->getType()) {
9080b57cec5SDimitry Andric   case COFF::IMPORT_CODE:
9090b57cec5SDimitry Andric     return 't';
9100b57cec5SDimitry Andric   case COFF::IMPORT_DATA:
9110b57cec5SDimitry Andric     return 'd';
9120b57cec5SDimitry Andric   case COFF::IMPORT_CONST:
9130b57cec5SDimitry Andric     return 'r';
9140b57cec5SDimitry Andric   }
9150b57cec5SDimitry Andric   return '?';
9160b57cec5SDimitry Andric }
9170b57cec5SDimitry Andric 
9180b57cec5SDimitry Andric static char getSymbolNMTypeChar(MachOObjectFile &Obj, basic_symbol_iterator I) {
9190b57cec5SDimitry Andric   DataRefImpl Symb = I->getRawDataRefImpl();
9200b57cec5SDimitry Andric   uint8_t NType = Obj.is64Bit() ? Obj.getSymbol64TableEntry(Symb).n_type
9210b57cec5SDimitry Andric                                 : Obj.getSymbolTableEntry(Symb).n_type;
9220b57cec5SDimitry Andric 
9230b57cec5SDimitry Andric   if (NType & MachO::N_STAB)
9240b57cec5SDimitry Andric     return '-';
9250b57cec5SDimitry Andric 
9260b57cec5SDimitry Andric   switch (NType & MachO::N_TYPE) {
9270b57cec5SDimitry Andric   case MachO::N_ABS:
9280b57cec5SDimitry Andric     return 's';
9290b57cec5SDimitry Andric   case MachO::N_INDR:
9300b57cec5SDimitry Andric     return 'i';
9310b57cec5SDimitry Andric   case MachO::N_SECT: {
9320b57cec5SDimitry Andric     Expected<section_iterator> SecOrErr = Obj.getSymbolSection(Symb);
9330b57cec5SDimitry Andric     if (!SecOrErr) {
9340b57cec5SDimitry Andric       consumeError(SecOrErr.takeError());
9350b57cec5SDimitry Andric       return 's';
9360b57cec5SDimitry Andric     }
9370b57cec5SDimitry Andric     section_iterator Sec = *SecOrErr;
9380b57cec5SDimitry Andric     if (Sec == Obj.section_end())
9390b57cec5SDimitry Andric       return 's';
9400b57cec5SDimitry Andric     DataRefImpl Ref = Sec->getRawDataRefImpl();
9410b57cec5SDimitry Andric     StringRef SectionName;
9420b57cec5SDimitry Andric     if (Expected<StringRef> NameOrErr = Obj.getSectionName(Ref))
9430b57cec5SDimitry Andric       SectionName = *NameOrErr;
9440b57cec5SDimitry Andric     StringRef SegmentName = Obj.getSectionFinalSegmentName(Ref);
9450b57cec5SDimitry Andric     if (Obj.is64Bit() && Obj.getHeader64().filetype == MachO::MH_KEXT_BUNDLE &&
9460b57cec5SDimitry Andric         SegmentName == "__TEXT_EXEC" && SectionName == "__text")
9470b57cec5SDimitry Andric       return 't';
9480b57cec5SDimitry Andric     if (SegmentName == "__TEXT" && SectionName == "__text")
9490b57cec5SDimitry Andric       return 't';
9500b57cec5SDimitry Andric     if (SegmentName == "__DATA" && SectionName == "__data")
9510b57cec5SDimitry Andric       return 'd';
9520b57cec5SDimitry Andric     if (SegmentName == "__DATA" && SectionName == "__bss")
9530b57cec5SDimitry Andric       return 'b';
9540b57cec5SDimitry Andric     return 's';
9550b57cec5SDimitry Andric   }
9560b57cec5SDimitry Andric   }
9570b57cec5SDimitry Andric 
9580b57cec5SDimitry Andric   return '?';
9590b57cec5SDimitry Andric }
9600b57cec5SDimitry Andric 
9615ffd83dbSDimitry Andric static char getSymbolNMTypeChar(TapiFile &Obj, basic_symbol_iterator I) {
9625ffd83dbSDimitry Andric   return 's';
9635ffd83dbSDimitry Andric }
9645ffd83dbSDimitry Andric 
9650b57cec5SDimitry Andric static char getSymbolNMTypeChar(WasmObjectFile &Obj, basic_symbol_iterator I) {
9665ffd83dbSDimitry Andric   uint32_t Flags = cantFail(I->getFlags());
9670b57cec5SDimitry Andric   if (Flags & SymbolRef::SF_Executable)
9680b57cec5SDimitry Andric     return 't';
9690b57cec5SDimitry Andric   return 'd';
9700b57cec5SDimitry Andric }
9710b57cec5SDimitry Andric 
9720b57cec5SDimitry Andric static char getSymbolNMTypeChar(IRObjectFile &Obj, basic_symbol_iterator I) {
9735ffd83dbSDimitry Andric   uint32_t Flags = cantFail(I->getFlags());
9740b57cec5SDimitry Andric   // FIXME: should we print 'b'? At the IR level we cannot be sure if this
9750b57cec5SDimitry Andric   // will be in bss or not, but we could approximate.
9760b57cec5SDimitry Andric   if (Flags & SymbolRef::SF_Executable)
9770b57cec5SDimitry Andric     return 't';
9780b57cec5SDimitry Andric   else if (Triple(Obj.getTargetTriple()).isOSDarwin() &&
9790b57cec5SDimitry Andric            (Flags & SymbolRef::SF_Const))
9800b57cec5SDimitry Andric     return 's';
9810b57cec5SDimitry Andric   else
9820b57cec5SDimitry Andric     return 'd';
9830b57cec5SDimitry Andric }
9840b57cec5SDimitry Andric 
9850b57cec5SDimitry Andric static bool isObject(SymbolicFile &Obj, basic_symbol_iterator I) {
986e8d8bef9SDimitry Andric   return isa<ELFObjectFileBase>(&Obj) &&
987e8d8bef9SDimitry Andric          elf_symbol_iterator(I)->getELFType() == ELF::STT_OBJECT;
9880b57cec5SDimitry Andric }
9890b57cec5SDimitry Andric 
9900b57cec5SDimitry Andric // For ELF object files, Set TypeName to the symbol typename, to be printed
9910b57cec5SDimitry Andric // in the 'Type' column of the SYSV format output.
9920b57cec5SDimitry Andric static StringRef getNMTypeName(SymbolicFile &Obj, basic_symbol_iterator I) {
9930b57cec5SDimitry Andric   if (isa<ELFObjectFileBase>(&Obj)) {
9940b57cec5SDimitry Andric     elf_symbol_iterator SymI(I);
9950b57cec5SDimitry Andric     return SymI->getELFTypeName();
9960b57cec5SDimitry Andric   }
9970b57cec5SDimitry Andric   return "";
9980b57cec5SDimitry Andric }
9990b57cec5SDimitry Andric 
10000b57cec5SDimitry Andric // Return Posix nm class type tag (single letter), but also set SecName and
10010b57cec5SDimitry Andric // section and name, to be used in format=sysv output.
10020b57cec5SDimitry Andric static char getNMSectionTagAndName(SymbolicFile &Obj, basic_symbol_iterator I,
10030b57cec5SDimitry Andric                                    StringRef &SecName) {
10045ffd83dbSDimitry Andric   // Symbol Flags have been checked in the caller.
10055ffd83dbSDimitry Andric   uint32_t Symflags = cantFail(I->getFlags());
10068bcb0991SDimitry Andric   if (ELFObjectFileBase *ELFObj = dyn_cast<ELFObjectFileBase>(&Obj)) {
10070b57cec5SDimitry Andric     if (Symflags & object::SymbolRef::SF_Absolute)
10080b57cec5SDimitry Andric       SecName = "*ABS*";
10090b57cec5SDimitry Andric     else if (Symflags & object::SymbolRef::SF_Common)
10100b57cec5SDimitry Andric       SecName = "*COM*";
10110b57cec5SDimitry Andric     else if (Symflags & object::SymbolRef::SF_Undefined)
10120b57cec5SDimitry Andric       SecName = "*UND*";
10130b57cec5SDimitry Andric     else {
10140b57cec5SDimitry Andric       elf_symbol_iterator SymI(I);
10150b57cec5SDimitry Andric       Expected<elf_section_iterator> SecIOrErr = SymI->getSection();
10160b57cec5SDimitry Andric       if (!SecIOrErr) {
10170b57cec5SDimitry Andric         consumeError(SecIOrErr.takeError());
10180b57cec5SDimitry Andric         return '?';
10190b57cec5SDimitry Andric       }
10208bcb0991SDimitry Andric 
10218bcb0991SDimitry Andric       if (*SecIOrErr == ELFObj->section_end())
10228bcb0991SDimitry Andric         return '?';
10238bcb0991SDimitry Andric 
10248bcb0991SDimitry Andric       Expected<StringRef> NameOrErr = (*SecIOrErr)->getName();
10258bcb0991SDimitry Andric       if (!NameOrErr) {
10268bcb0991SDimitry Andric         consumeError(NameOrErr.takeError());
10278bcb0991SDimitry Andric         return '?';
10288bcb0991SDimitry Andric       }
10298bcb0991SDimitry Andric       SecName = *NameOrErr;
10300b57cec5SDimitry Andric     }
10310b57cec5SDimitry Andric   }
10320b57cec5SDimitry Andric 
1033*fe6060f1SDimitry Andric   if (Symflags & object::SymbolRef::SF_Undefined) {
1034*fe6060f1SDimitry Andric     if (isa<MachOObjectFile>(Obj) || !(Symflags & object::SymbolRef::SF_Weak))
10350b57cec5SDimitry Andric       return 'U';
1036*fe6060f1SDimitry Andric     return isObject(Obj, I) ? 'v' : 'w';
1037*fe6060f1SDimitry Andric   }
1038*fe6060f1SDimitry Andric   if (isa<ELFObjectFileBase>(&Obj))
1039*fe6060f1SDimitry Andric     if (ELFSymbolRef(*I).getELFType() == ELF::STT_GNU_IFUNC)
1040*fe6060f1SDimitry Andric       return 'i';
1041*fe6060f1SDimitry Andric   if (!isa<MachOObjectFile>(Obj) && (Symflags & object::SymbolRef::SF_Weak))
1042*fe6060f1SDimitry Andric     return isObject(Obj, I) ? 'V' : 'W';
10430b57cec5SDimitry Andric 
10440b57cec5SDimitry Andric   if (Symflags & object::SymbolRef::SF_Common)
10450b57cec5SDimitry Andric     return 'C';
10460b57cec5SDimitry Andric 
10470b57cec5SDimitry Andric   char Ret = '?';
10480b57cec5SDimitry Andric   if (Symflags & object::SymbolRef::SF_Absolute)
10490b57cec5SDimitry Andric     Ret = 'a';
10500b57cec5SDimitry Andric   else if (IRObjectFile *IR = dyn_cast<IRObjectFile>(&Obj))
10510b57cec5SDimitry Andric     Ret = getSymbolNMTypeChar(*IR, I);
10520b57cec5SDimitry Andric   else if (COFFObjectFile *COFF = dyn_cast<COFFObjectFile>(&Obj))
10530b57cec5SDimitry Andric     Ret = getSymbolNMTypeChar(*COFF, I);
10540b57cec5SDimitry Andric   else if (COFFImportFile *COFFImport = dyn_cast<COFFImportFile>(&Obj))
10550b57cec5SDimitry Andric     Ret = getSymbolNMTypeChar(*COFFImport);
10560b57cec5SDimitry Andric   else if (MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(&Obj))
10570b57cec5SDimitry Andric     Ret = getSymbolNMTypeChar(*MachO, I);
10580b57cec5SDimitry Andric   else if (WasmObjectFile *Wasm = dyn_cast<WasmObjectFile>(&Obj))
10590b57cec5SDimitry Andric     Ret = getSymbolNMTypeChar(*Wasm, I);
10605ffd83dbSDimitry Andric   else if (TapiFile *Tapi = dyn_cast<TapiFile>(&Obj))
10615ffd83dbSDimitry Andric     Ret = getSymbolNMTypeChar(*Tapi, I);
1062480093f4SDimitry Andric   else if (ELFObjectFileBase *ELF = dyn_cast<ELFObjectFileBase>(&Obj)) {
1063480093f4SDimitry Andric     Ret = getSymbolNMTypeChar(*ELF, I);
1064480093f4SDimitry Andric     if (ELFSymbolRef(*I).getBinding() == ELF::STB_GNU_UNIQUE)
1065480093f4SDimitry Andric       return Ret;
1066480093f4SDimitry Andric   } else
1067480093f4SDimitry Andric     llvm_unreachable("unknown binary format");
10680b57cec5SDimitry Andric 
10690b57cec5SDimitry Andric   if (!(Symflags & object::SymbolRef::SF_Global))
10700b57cec5SDimitry Andric     return Ret;
10710b57cec5SDimitry Andric 
10720b57cec5SDimitry Andric   return toupper(Ret);
10730b57cec5SDimitry Andric }
10740b57cec5SDimitry Andric 
10750b57cec5SDimitry Andric // getNsectForSegSect() is used to implement the Mach-O "-s segname sectname"
10760b57cec5SDimitry Andric // option to dump only those symbols from that section in a Mach-O file.
10770b57cec5SDimitry Andric // It is called once for each Mach-O file from dumpSymbolNamesFromObject()
10780b57cec5SDimitry Andric // to get the section number for that named section from the command line
10790b57cec5SDimitry Andric // arguments. It returns the section number for that section in the Mach-O
10800b57cec5SDimitry Andric // file or zero it is not present.
10810b57cec5SDimitry Andric static unsigned getNsectForSegSect(MachOObjectFile *Obj) {
10820b57cec5SDimitry Andric   unsigned Nsect = 1;
10830b57cec5SDimitry Andric   for (auto &S : Obj->sections()) {
10840b57cec5SDimitry Andric     DataRefImpl Ref = S.getRawDataRefImpl();
10850b57cec5SDimitry Andric     StringRef SectionName;
10860b57cec5SDimitry Andric     if (Expected<StringRef> NameOrErr = Obj->getSectionName(Ref))
10870b57cec5SDimitry Andric       SectionName = *NameOrErr;
10880b57cec5SDimitry Andric     StringRef SegmentName = Obj->getSectionFinalSegmentName(Ref);
10890b57cec5SDimitry Andric     if (SegmentName == SegSect[0] && SectionName == SegSect[1])
10900b57cec5SDimitry Andric       return Nsect;
10910b57cec5SDimitry Andric     Nsect++;
10920b57cec5SDimitry Andric   }
10930b57cec5SDimitry Andric   return 0;
10940b57cec5SDimitry Andric }
10950b57cec5SDimitry Andric 
10960b57cec5SDimitry Andric // getNsectInMachO() is used to implement the Mach-O "-s segname sectname"
10970b57cec5SDimitry Andric // option to dump only those symbols from that section in a Mach-O file.
10980b57cec5SDimitry Andric // It is called once for each symbol in a Mach-O file from
10990b57cec5SDimitry Andric // dumpSymbolNamesFromObject() and returns the section number for that symbol
11000b57cec5SDimitry Andric // if it is in a section, else it returns 0.
11010b57cec5SDimitry Andric static unsigned getNsectInMachO(MachOObjectFile &Obj, BasicSymbolRef Sym) {
11020b57cec5SDimitry Andric   DataRefImpl Symb = Sym.getRawDataRefImpl();
11030b57cec5SDimitry Andric   if (Obj.is64Bit()) {
11040b57cec5SDimitry Andric     MachO::nlist_64 STE = Obj.getSymbol64TableEntry(Symb);
11050b57cec5SDimitry Andric     return (STE.n_type & MachO::N_TYPE) == MachO::N_SECT ? STE.n_sect : 0;
11060b57cec5SDimitry Andric   }
11070b57cec5SDimitry Andric   MachO::nlist STE = Obj.getSymbolTableEntry(Symb);
11080b57cec5SDimitry Andric   return (STE.n_type & MachO::N_TYPE) == MachO::N_SECT ? STE.n_sect : 0;
11090b57cec5SDimitry Andric }
11100b57cec5SDimitry Andric 
1111e8d8bef9SDimitry Andric static void dumpSymbolsFromDLInfoMachO(MachOObjectFile &MachO) {
1112e8d8bef9SDimitry Andric   size_t I = SymbolList.size();
11130b57cec5SDimitry Andric   std::string ExportsNameBuffer;
11140b57cec5SDimitry Andric   raw_string_ostream EOS(ExportsNameBuffer);
11150b57cec5SDimitry Andric   std::string BindsNameBuffer;
11160b57cec5SDimitry Andric   raw_string_ostream BOS(BindsNameBuffer);
11170b57cec5SDimitry Andric   std::string LazysNameBuffer;
11180b57cec5SDimitry Andric   raw_string_ostream LOS(LazysNameBuffer);
11190b57cec5SDimitry Andric   std::string WeaksNameBuffer;
11200b57cec5SDimitry Andric   raw_string_ostream WOS(WeaksNameBuffer);
11210b57cec5SDimitry Andric   std::string FunctionStartsNameBuffer;
11220b57cec5SDimitry Andric   raw_string_ostream FOS(FunctionStartsNameBuffer);
1123e8d8bef9SDimitry Andric 
11240b57cec5SDimitry Andric   MachO::mach_header H;
11250b57cec5SDimitry Andric   MachO::mach_header_64 H_64;
11260b57cec5SDimitry Andric   uint32_t HFlags = 0;
1127e8d8bef9SDimitry Andric   if (MachO.is64Bit()) {
1128e8d8bef9SDimitry Andric     H_64 = MachO.MachOObjectFile::getHeader64();
11290b57cec5SDimitry Andric     HFlags = H_64.flags;
11300b57cec5SDimitry Andric   } else {
1131e8d8bef9SDimitry Andric     H = MachO.MachOObjectFile::getHeader();
11320b57cec5SDimitry Andric     HFlags = H.flags;
11330b57cec5SDimitry Andric   }
11340b57cec5SDimitry Andric   uint64_t BaseSegmentAddress = 0;
1135e8d8bef9SDimitry Andric   for (const auto &Command : MachO.load_commands()) {
11360b57cec5SDimitry Andric     if (Command.C.cmd == MachO::LC_SEGMENT) {
1137e8d8bef9SDimitry Andric       MachO::segment_command Seg = MachO.getSegmentLoadCommand(Command);
11380b57cec5SDimitry Andric       if (Seg.fileoff == 0 && Seg.filesize != 0) {
11390b57cec5SDimitry Andric         BaseSegmentAddress = Seg.vmaddr;
11400b57cec5SDimitry Andric         break;
11410b57cec5SDimitry Andric       }
11420b57cec5SDimitry Andric     } else if (Command.C.cmd == MachO::LC_SEGMENT_64) {
1143e8d8bef9SDimitry Andric       MachO::segment_command_64 Seg = MachO.getSegment64LoadCommand(Command);
11440b57cec5SDimitry Andric       if (Seg.fileoff == 0 && Seg.filesize != 0) {
11450b57cec5SDimitry Andric         BaseSegmentAddress = Seg.vmaddr;
11460b57cec5SDimitry Andric         break;
11470b57cec5SDimitry Andric       }
11480b57cec5SDimitry Andric     }
11490b57cec5SDimitry Andric   }
11500b57cec5SDimitry Andric   if (DyldInfoOnly || AddDyldInfo ||
11510b57cec5SDimitry Andric       HFlags & MachO::MH_NLIST_OUTOFSYNC_WITH_DYLDINFO) {
11520b57cec5SDimitry Andric     unsigned ExportsAdded = 0;
11530b57cec5SDimitry Andric     Error Err = Error::success();
1154e8d8bef9SDimitry Andric     for (const llvm::object::ExportEntry &Entry : MachO.exports(Err)) {
11550b57cec5SDimitry Andric       bool found = false;
11560b57cec5SDimitry Andric       bool ReExport = false;
11570b57cec5SDimitry Andric       if (!DyldInfoOnly) {
11580b57cec5SDimitry Andric         for (const NMSymbol &S : SymbolList)
11590b57cec5SDimitry Andric           if (S.Address == Entry.address() + BaseSegmentAddress &&
11600b57cec5SDimitry Andric               S.Name == Entry.name()) {
11610b57cec5SDimitry Andric             found = true;
11620b57cec5SDimitry Andric             break;
11630b57cec5SDimitry Andric           }
11640b57cec5SDimitry Andric       }
11650b57cec5SDimitry Andric       if (!found) {
11660b57cec5SDimitry Andric         NMSymbol S = {};
11670b57cec5SDimitry Andric         S.Address = Entry.address() + BaseSegmentAddress;
11680b57cec5SDimitry Andric         S.Size = 0;
11690b57cec5SDimitry Andric         S.TypeChar = '\0';
1170e8d8bef9SDimitry Andric         S.Name = Entry.name().str();
11710b57cec5SDimitry Andric         // There is no symbol in the nlist symbol table for this so we set
11720b57cec5SDimitry Andric         // Sym effectivly to null and the rest of code in here must test for
11730b57cec5SDimitry Andric         // it and not do things like Sym.getFlags() for it.
11740b57cec5SDimitry Andric         S.Sym = BasicSymbolRef();
11750b57cec5SDimitry Andric         S.SymFlags = SymbolRef::SF_Global;
11760b57cec5SDimitry Andric         S.Section = SectionRef();
11770b57cec5SDimitry Andric         S.NType = 0;
11780b57cec5SDimitry Andric         S.NSect = 0;
11790b57cec5SDimitry Andric         S.NDesc = 0;
11800b57cec5SDimitry Andric 
11810b57cec5SDimitry Andric         uint64_t EFlags = Entry.flags();
11820b57cec5SDimitry Andric         bool Abs = ((EFlags & MachO::EXPORT_SYMBOL_FLAGS_KIND_MASK) ==
11830b57cec5SDimitry Andric                     MachO::EXPORT_SYMBOL_FLAGS_KIND_ABSOLUTE);
1184e8d8bef9SDimitry Andric         bool Resolver = (EFlags & MachO::EXPORT_SYMBOL_FLAGS_STUB_AND_RESOLVER);
11850b57cec5SDimitry Andric         ReExport = (EFlags & MachO::EXPORT_SYMBOL_FLAGS_REEXPORT);
11860b57cec5SDimitry Andric         bool WeakDef = (EFlags & MachO::EXPORT_SYMBOL_FLAGS_WEAK_DEFINITION);
11870b57cec5SDimitry Andric         if (WeakDef)
11880b57cec5SDimitry Andric           S.NDesc |= MachO::N_WEAK_DEF;
11890b57cec5SDimitry Andric         if (Abs) {
11900b57cec5SDimitry Andric           S.NType = MachO::N_EXT | MachO::N_ABS;
11910b57cec5SDimitry Andric           S.TypeChar = 'A';
11920b57cec5SDimitry Andric         } else if (ReExport) {
11930b57cec5SDimitry Andric           S.NType = MachO::N_EXT | MachO::N_INDR;
11940b57cec5SDimitry Andric           S.TypeChar = 'I';
11950b57cec5SDimitry Andric         } else {
11960b57cec5SDimitry Andric           S.NType = MachO::N_EXT | MachO::N_SECT;
11970b57cec5SDimitry Andric           if (Resolver) {
11980b57cec5SDimitry Andric             S.Address = Entry.other() + BaseSegmentAddress;
1199e8d8bef9SDimitry Andric             if ((S.Address & 1) != 0 && !MachO.is64Bit() &&
1200e8d8bef9SDimitry Andric                 H.cputype == MachO::CPU_TYPE_ARM) {
12010b57cec5SDimitry Andric               S.Address &= ~1LL;
12020b57cec5SDimitry Andric               S.NDesc |= MachO::N_ARM_THUMB_DEF;
12030b57cec5SDimitry Andric             }
12040b57cec5SDimitry Andric           } else {
12050b57cec5SDimitry Andric             S.Address = Entry.address() + BaseSegmentAddress;
12060b57cec5SDimitry Andric           }
12070b57cec5SDimitry Andric           StringRef SegmentName = StringRef();
12080b57cec5SDimitry Andric           StringRef SectionName = StringRef();
1209e8d8bef9SDimitry Andric           for (const SectionRef &Section : MachO.sections()) {
12100b57cec5SDimitry Andric             S.NSect++;
12118bcb0991SDimitry Andric 
12128bcb0991SDimitry Andric             if (Expected<StringRef> NameOrErr = Section.getName())
12138bcb0991SDimitry Andric               SectionName = *NameOrErr;
12148bcb0991SDimitry Andric             else
12158bcb0991SDimitry Andric               consumeError(NameOrErr.takeError());
12168bcb0991SDimitry Andric 
1217e8d8bef9SDimitry Andric             SegmentName =
1218e8d8bef9SDimitry Andric                 MachO.getSectionFinalSegmentName(Section.getRawDataRefImpl());
12190b57cec5SDimitry Andric             if (S.Address >= Section.getAddress() &&
12200b57cec5SDimitry Andric                 S.Address < Section.getAddress() + Section.getSize()) {
12210b57cec5SDimitry Andric               S.Section = Section;
12220b57cec5SDimitry Andric               break;
12230b57cec5SDimitry Andric             } else if (Entry.name() == "__mh_execute_header" &&
12240b57cec5SDimitry Andric                        SegmentName == "__TEXT" && SectionName == "__text") {
12250b57cec5SDimitry Andric               S.Section = Section;
12260b57cec5SDimitry Andric               S.NDesc |= MachO::REFERENCED_DYNAMICALLY;
12270b57cec5SDimitry Andric               break;
12280b57cec5SDimitry Andric             }
12290b57cec5SDimitry Andric           }
12300b57cec5SDimitry Andric           if (SegmentName == "__TEXT" && SectionName == "__text")
12310b57cec5SDimitry Andric             S.TypeChar = 'T';
12320b57cec5SDimitry Andric           else if (SegmentName == "__DATA" && SectionName == "__data")
12330b57cec5SDimitry Andric             S.TypeChar = 'D';
12340b57cec5SDimitry Andric           else if (SegmentName == "__DATA" && SectionName == "__bss")
12350b57cec5SDimitry Andric             S.TypeChar = 'B';
12360b57cec5SDimitry Andric           else
12370b57cec5SDimitry Andric             S.TypeChar = 'S';
12380b57cec5SDimitry Andric         }
12390b57cec5SDimitry Andric         SymbolList.push_back(S);
12400b57cec5SDimitry Andric 
12410b57cec5SDimitry Andric         EOS << Entry.name();
12420b57cec5SDimitry Andric         EOS << '\0';
12430b57cec5SDimitry Andric         ExportsAdded++;
12440b57cec5SDimitry Andric 
12450b57cec5SDimitry Andric         // For ReExports there are a two more things to do, first add the
12460b57cec5SDimitry Andric         // indirect name and second create the undefined symbol using the
12470b57cec5SDimitry Andric         // referened dynamic library.
12480b57cec5SDimitry Andric         if (ReExport) {
12490b57cec5SDimitry Andric 
12500b57cec5SDimitry Andric           // Add the indirect name.
12510b57cec5SDimitry Andric           if (Entry.otherName().empty())
12520b57cec5SDimitry Andric             EOS << Entry.name();
12530b57cec5SDimitry Andric           else
12540b57cec5SDimitry Andric             EOS << Entry.otherName();
12550b57cec5SDimitry Andric           EOS << '\0';
12560b57cec5SDimitry Andric 
12570b57cec5SDimitry Andric           // Now create the undefined symbol using the referened dynamic
12580b57cec5SDimitry Andric           // library.
12590b57cec5SDimitry Andric           NMSymbol U = {};
12600b57cec5SDimitry Andric           U.Address = 0;
12610b57cec5SDimitry Andric           U.Size = 0;
12620b57cec5SDimitry Andric           U.TypeChar = 'U';
12630b57cec5SDimitry Andric           if (Entry.otherName().empty())
1264e8d8bef9SDimitry Andric             U.Name = Entry.name().str();
12650b57cec5SDimitry Andric           else
1266e8d8bef9SDimitry Andric             U.Name = Entry.otherName().str();
12670b57cec5SDimitry Andric           // Again there is no symbol in the nlist symbol table for this so
12680b57cec5SDimitry Andric           // we set Sym effectivly to null and the rest of code in here must
12690b57cec5SDimitry Andric           // test for it and not do things like Sym.getFlags() for it.
12700b57cec5SDimitry Andric           U.Sym = BasicSymbolRef();
12710b57cec5SDimitry Andric           U.SymFlags = SymbolRef::SF_Global | SymbolRef::SF_Undefined;
12720b57cec5SDimitry Andric           U.Section = SectionRef();
12730b57cec5SDimitry Andric           U.NType = MachO::N_EXT | MachO::N_UNDF;
12740b57cec5SDimitry Andric           U.NSect = 0;
12750b57cec5SDimitry Andric           U.NDesc = 0;
12760b57cec5SDimitry Andric           // The library ordinal for this undefined symbol is in the export
12770b57cec5SDimitry Andric           // trie Entry.other().
12780b57cec5SDimitry Andric           MachO::SET_LIBRARY_ORDINAL(U.NDesc, Entry.other());
12790b57cec5SDimitry Andric           SymbolList.push_back(U);
12800b57cec5SDimitry Andric 
12810b57cec5SDimitry Andric           // Finally add the undefined symbol's name.
12820b57cec5SDimitry Andric           if (Entry.otherName().empty())
12830b57cec5SDimitry Andric             EOS << Entry.name();
12840b57cec5SDimitry Andric           else
12850b57cec5SDimitry Andric             EOS << Entry.otherName();
12860b57cec5SDimitry Andric           EOS << '\0';
12870b57cec5SDimitry Andric           ExportsAdded++;
12880b57cec5SDimitry Andric         }
12890b57cec5SDimitry Andric       }
12900b57cec5SDimitry Andric     }
12910b57cec5SDimitry Andric     if (Err)
1292e8d8bef9SDimitry Andric       error(std::move(Err), MachO.getFileName());
12930b57cec5SDimitry Andric     // Set the symbol names and indirect names for the added symbols.
12940b57cec5SDimitry Andric     if (ExportsAdded) {
12950b57cec5SDimitry Andric       EOS.flush();
12960b57cec5SDimitry Andric       const char *Q = ExportsNameBuffer.c_str();
12970b57cec5SDimitry Andric       for (unsigned K = 0; K < ExportsAdded; K++) {
12980b57cec5SDimitry Andric         SymbolList[I].Name = Q;
12990b57cec5SDimitry Andric         Q += strlen(Q) + 1;
13000b57cec5SDimitry Andric         if (SymbolList[I].TypeChar == 'I') {
13010b57cec5SDimitry Andric           SymbolList[I].IndirectName = Q;
13020b57cec5SDimitry Andric           Q += strlen(Q) + 1;
13030b57cec5SDimitry Andric         }
13040b57cec5SDimitry Andric         I++;
13050b57cec5SDimitry Andric       }
13060b57cec5SDimitry Andric     }
13070b57cec5SDimitry Andric 
13080b57cec5SDimitry Andric     // Add the undefined symbols from the bind entries.
13090b57cec5SDimitry Andric     unsigned BindsAdded = 0;
13100b57cec5SDimitry Andric     Error BErr = Error::success();
13110b57cec5SDimitry Andric     StringRef LastSymbolName = StringRef();
1312e8d8bef9SDimitry Andric     for (const llvm::object::MachOBindEntry &Entry : MachO.bindTable(BErr)) {
13130b57cec5SDimitry Andric       bool found = false;
13140b57cec5SDimitry Andric       if (LastSymbolName == Entry.symbolName())
13150b57cec5SDimitry Andric         found = true;
13160b57cec5SDimitry Andric       else if (!DyldInfoOnly) {
13170b57cec5SDimitry Andric         for (unsigned J = 0; J < SymbolList.size() && !found; ++J) {
13180b57cec5SDimitry Andric           if (SymbolList[J].Name == Entry.symbolName())
13190b57cec5SDimitry Andric             found = true;
13200b57cec5SDimitry Andric         }
13210b57cec5SDimitry Andric       }
13220b57cec5SDimitry Andric       if (!found) {
13230b57cec5SDimitry Andric         LastSymbolName = Entry.symbolName();
13240b57cec5SDimitry Andric         NMSymbol B = {};
13250b57cec5SDimitry Andric         B.Address = 0;
13260b57cec5SDimitry Andric         B.Size = 0;
13270b57cec5SDimitry Andric         B.TypeChar = 'U';
13280b57cec5SDimitry Andric         // There is no symbol in the nlist symbol table for this so we set
13290b57cec5SDimitry Andric         // Sym effectivly to null and the rest of code in here must test for
13300b57cec5SDimitry Andric         // it and not do things like Sym.getFlags() for it.
13310b57cec5SDimitry Andric         B.Sym = BasicSymbolRef();
13320b57cec5SDimitry Andric         B.SymFlags = SymbolRef::SF_Global | SymbolRef::SF_Undefined;
13330b57cec5SDimitry Andric         B.NType = MachO::N_EXT | MachO::N_UNDF;
13340b57cec5SDimitry Andric         B.NSect = 0;
13350b57cec5SDimitry Andric         B.NDesc = 0;
13360b57cec5SDimitry Andric         MachO::SET_LIBRARY_ORDINAL(B.NDesc, Entry.ordinal());
1337e8d8bef9SDimitry Andric         B.Name = Entry.symbolName().str();
13380b57cec5SDimitry Andric         SymbolList.push_back(B);
13390b57cec5SDimitry Andric         BOS << Entry.symbolName();
13400b57cec5SDimitry Andric         BOS << '\0';
13410b57cec5SDimitry Andric         BindsAdded++;
13420b57cec5SDimitry Andric       }
13430b57cec5SDimitry Andric     }
13440b57cec5SDimitry Andric     if (BErr)
1345e8d8bef9SDimitry Andric       error(std::move(BErr), MachO.getFileName());
13460b57cec5SDimitry Andric     // Set the symbol names and indirect names for the added symbols.
13470b57cec5SDimitry Andric     if (BindsAdded) {
13480b57cec5SDimitry Andric       BOS.flush();
13490b57cec5SDimitry Andric       const char *Q = BindsNameBuffer.c_str();
13500b57cec5SDimitry Andric       for (unsigned K = 0; K < BindsAdded; K++) {
13510b57cec5SDimitry Andric         SymbolList[I].Name = Q;
13520b57cec5SDimitry Andric         Q += strlen(Q) + 1;
13530b57cec5SDimitry Andric         if (SymbolList[I].TypeChar == 'I') {
13540b57cec5SDimitry Andric           SymbolList[I].IndirectName = Q;
13550b57cec5SDimitry Andric           Q += strlen(Q) + 1;
13560b57cec5SDimitry Andric         }
13570b57cec5SDimitry Andric         I++;
13580b57cec5SDimitry Andric       }
13590b57cec5SDimitry Andric     }
13600b57cec5SDimitry Andric 
13610b57cec5SDimitry Andric     // Add the undefined symbols from the lazy bind entries.
13620b57cec5SDimitry Andric     unsigned LazysAdded = 0;
13630b57cec5SDimitry Andric     Error LErr = Error::success();
13640b57cec5SDimitry Andric     LastSymbolName = StringRef();
13650b57cec5SDimitry Andric     for (const llvm::object::MachOBindEntry &Entry :
1366e8d8bef9SDimitry Andric          MachO.lazyBindTable(LErr)) {
13670b57cec5SDimitry Andric       bool found = false;
13680b57cec5SDimitry Andric       if (LastSymbolName == Entry.symbolName())
13690b57cec5SDimitry Andric         found = true;
13700b57cec5SDimitry Andric       else {
13710b57cec5SDimitry Andric         // Here we must check to see it this symbol is already in the
13720b57cec5SDimitry Andric         // SymbolList as it might have already have been added above via a
13730b57cec5SDimitry Andric         // non-lazy (bind) entry.
13740b57cec5SDimitry Andric         for (unsigned J = 0; J < SymbolList.size() && !found; ++J) {
13750b57cec5SDimitry Andric           if (SymbolList[J].Name == Entry.symbolName())
13760b57cec5SDimitry Andric             found = true;
13770b57cec5SDimitry Andric         }
13780b57cec5SDimitry Andric       }
13790b57cec5SDimitry Andric       if (!found) {
13800b57cec5SDimitry Andric         LastSymbolName = Entry.symbolName();
13810b57cec5SDimitry Andric         NMSymbol L = {};
1382e8d8bef9SDimitry Andric         L.Name = Entry.symbolName().str();
13830b57cec5SDimitry Andric         L.Address = 0;
13840b57cec5SDimitry Andric         L.Size = 0;
13850b57cec5SDimitry Andric         L.TypeChar = 'U';
13860b57cec5SDimitry Andric         // There is no symbol in the nlist symbol table for this so we set
13870b57cec5SDimitry Andric         // Sym effectivly to null and the rest of code in here must test for
13880b57cec5SDimitry Andric         // it and not do things like Sym.getFlags() for it.
13890b57cec5SDimitry Andric         L.Sym = BasicSymbolRef();
13900b57cec5SDimitry Andric         L.SymFlags = SymbolRef::SF_Global | SymbolRef::SF_Undefined;
13910b57cec5SDimitry Andric         L.NType = MachO::N_EXT | MachO::N_UNDF;
13920b57cec5SDimitry Andric         L.NSect = 0;
13930b57cec5SDimitry Andric         // The REFERENCE_FLAG_UNDEFINED_LAZY is no longer used but here it
13940b57cec5SDimitry Andric         // makes sence since we are creating this from a lazy bind entry.
13950b57cec5SDimitry Andric         L.NDesc = MachO::REFERENCE_FLAG_UNDEFINED_LAZY;
13960b57cec5SDimitry Andric         MachO::SET_LIBRARY_ORDINAL(L.NDesc, Entry.ordinal());
13970b57cec5SDimitry Andric         SymbolList.push_back(L);
13980b57cec5SDimitry Andric         LOS << Entry.symbolName();
13990b57cec5SDimitry Andric         LOS << '\0';
14000b57cec5SDimitry Andric         LazysAdded++;
14010b57cec5SDimitry Andric       }
14020b57cec5SDimitry Andric     }
14030b57cec5SDimitry Andric     if (LErr)
1404e8d8bef9SDimitry Andric       error(std::move(LErr), MachO.getFileName());
14050b57cec5SDimitry Andric     // Set the symbol names and indirect names for the added symbols.
14060b57cec5SDimitry Andric     if (LazysAdded) {
14070b57cec5SDimitry Andric       LOS.flush();
14080b57cec5SDimitry Andric       const char *Q = LazysNameBuffer.c_str();
14090b57cec5SDimitry Andric       for (unsigned K = 0; K < LazysAdded; K++) {
14100b57cec5SDimitry Andric         SymbolList[I].Name = Q;
14110b57cec5SDimitry Andric         Q += strlen(Q) + 1;
14120b57cec5SDimitry Andric         if (SymbolList[I].TypeChar == 'I') {
14130b57cec5SDimitry Andric           SymbolList[I].IndirectName = Q;
14140b57cec5SDimitry Andric           Q += strlen(Q) + 1;
14150b57cec5SDimitry Andric         }
14160b57cec5SDimitry Andric         I++;
14170b57cec5SDimitry Andric       }
14180b57cec5SDimitry Andric     }
14190b57cec5SDimitry Andric 
14200b57cec5SDimitry Andric     // Add the undefineds symbol from the weak bind entries which are not
14210b57cec5SDimitry Andric     // strong symbols.
14220b57cec5SDimitry Andric     unsigned WeaksAdded = 0;
14230b57cec5SDimitry Andric     Error WErr = Error::success();
14240b57cec5SDimitry Andric     LastSymbolName = StringRef();
14250b57cec5SDimitry Andric     for (const llvm::object::MachOBindEntry &Entry :
1426e8d8bef9SDimitry Andric          MachO.weakBindTable(WErr)) {
14270b57cec5SDimitry Andric       bool found = false;
14280b57cec5SDimitry Andric       unsigned J = 0;
14290b57cec5SDimitry Andric       if (LastSymbolName == Entry.symbolName() ||
14300b57cec5SDimitry Andric           Entry.flags() & MachO::BIND_SYMBOL_FLAGS_NON_WEAK_DEFINITION) {
14310b57cec5SDimitry Andric         found = true;
14320b57cec5SDimitry Andric       } else {
14330b57cec5SDimitry Andric         for (J = 0; J < SymbolList.size() && !found; ++J) {
14340b57cec5SDimitry Andric           if (SymbolList[J].Name == Entry.symbolName()) {
14350b57cec5SDimitry Andric             found = true;
14360b57cec5SDimitry Andric             break;
14370b57cec5SDimitry Andric           }
14380b57cec5SDimitry Andric         }
14390b57cec5SDimitry Andric       }
14400b57cec5SDimitry Andric       if (!found) {
14410b57cec5SDimitry Andric         LastSymbolName = Entry.symbolName();
1442e8d8bef9SDimitry Andric         NMSymbol W = {};
1443e8d8bef9SDimitry Andric         W.Name = Entry.symbolName().str();
14440b57cec5SDimitry Andric         W.Address = 0;
14450b57cec5SDimitry Andric         W.Size = 0;
14460b57cec5SDimitry Andric         W.TypeChar = 'U';
14470b57cec5SDimitry Andric         // There is no symbol in the nlist symbol table for this so we set
14480b57cec5SDimitry Andric         // Sym effectivly to null and the rest of code in here must test for
14490b57cec5SDimitry Andric         // it and not do things like Sym.getFlags() for it.
14500b57cec5SDimitry Andric         W.Sym = BasicSymbolRef();
14510b57cec5SDimitry Andric         W.SymFlags = SymbolRef::SF_Global | SymbolRef::SF_Undefined;
14520b57cec5SDimitry Andric         W.NType = MachO::N_EXT | MachO::N_UNDF;
14530b57cec5SDimitry Andric         W.NSect = 0;
14540b57cec5SDimitry Andric         // Odd that we are using N_WEAK_DEF on an undefined symbol but that is
14550b57cec5SDimitry Andric         // what is created in this case by the linker when there are real
14560b57cec5SDimitry Andric         // symbols in the nlist structs.
14570b57cec5SDimitry Andric         W.NDesc = MachO::N_WEAK_DEF;
14580b57cec5SDimitry Andric         SymbolList.push_back(W);
14590b57cec5SDimitry Andric         WOS << Entry.symbolName();
14600b57cec5SDimitry Andric         WOS << '\0';
14610b57cec5SDimitry Andric         WeaksAdded++;
14620b57cec5SDimitry Andric       } else {
14630b57cec5SDimitry Andric         // This is the case the symbol was previously been found and it could
14640b57cec5SDimitry Andric         // have been added from a bind or lazy bind symbol.  If so and not
14650b57cec5SDimitry Andric         // a definition also mark it as weak.
14660b57cec5SDimitry Andric         if (SymbolList[J].TypeChar == 'U')
14670b57cec5SDimitry Andric           // See comment above about N_WEAK_DEF.
14680b57cec5SDimitry Andric           SymbolList[J].NDesc |= MachO::N_WEAK_DEF;
14690b57cec5SDimitry Andric       }
14700b57cec5SDimitry Andric     }
14710b57cec5SDimitry Andric     if (WErr)
1472e8d8bef9SDimitry Andric       error(std::move(WErr), MachO.getFileName());
14730b57cec5SDimitry Andric     // Set the symbol names and indirect names for the added symbols.
14740b57cec5SDimitry Andric     if (WeaksAdded) {
14750b57cec5SDimitry Andric       WOS.flush();
14760b57cec5SDimitry Andric       const char *Q = WeaksNameBuffer.c_str();
14770b57cec5SDimitry Andric       for (unsigned K = 0; K < WeaksAdded; K++) {
14780b57cec5SDimitry Andric         SymbolList[I].Name = Q;
14790b57cec5SDimitry Andric         Q += strlen(Q) + 1;
14800b57cec5SDimitry Andric         if (SymbolList[I].TypeChar == 'I') {
14810b57cec5SDimitry Andric           SymbolList[I].IndirectName = Q;
14820b57cec5SDimitry Andric           Q += strlen(Q) + 1;
14830b57cec5SDimitry Andric         }
14840b57cec5SDimitry Andric         I++;
14850b57cec5SDimitry Andric       }
14860b57cec5SDimitry Andric     }
14870b57cec5SDimitry Andric 
14880b57cec5SDimitry Andric     // Trying adding symbol from the function starts table and LC_MAIN entry
14890b57cec5SDimitry Andric     // point.
14900b57cec5SDimitry Andric     SmallVector<uint64_t, 8> FoundFns;
14910b57cec5SDimitry Andric     uint64_t lc_main_offset = UINT64_MAX;
1492e8d8bef9SDimitry Andric     for (const auto &Command : MachO.load_commands()) {
14930b57cec5SDimitry Andric       if (Command.C.cmd == MachO::LC_FUNCTION_STARTS) {
14940b57cec5SDimitry Andric         // We found a function starts segment, parse the addresses for
14950b57cec5SDimitry Andric         // consumption.
14960b57cec5SDimitry Andric         MachO::linkedit_data_command LLC =
1497e8d8bef9SDimitry Andric             MachO.getLinkeditDataLoadCommand(Command);
14980b57cec5SDimitry Andric 
1499e8d8bef9SDimitry Andric         MachO.ReadULEB128s(LLC.dataoff, FoundFns);
15000b57cec5SDimitry Andric       } else if (Command.C.cmd == MachO::LC_MAIN) {
1501e8d8bef9SDimitry Andric         MachO::entry_point_command LCmain = MachO.getEntryPointCommand(Command);
15020b57cec5SDimitry Andric         lc_main_offset = LCmain.entryoff;
15030b57cec5SDimitry Andric       }
15040b57cec5SDimitry Andric     }
15050b57cec5SDimitry Andric     // See if these addresses are already in the symbol table.
15060b57cec5SDimitry Andric     unsigned FunctionStartsAdded = 0;
15070b57cec5SDimitry Andric     for (uint64_t f = 0; f < FoundFns.size(); f++) {
15080b57cec5SDimitry Andric       bool found = false;
15090b57cec5SDimitry Andric       for (unsigned J = 0; J < SymbolList.size() && !found; ++J) {
15100b57cec5SDimitry Andric         if (SymbolList[J].Address == FoundFns[f] + BaseSegmentAddress)
15110b57cec5SDimitry Andric           found = true;
15120b57cec5SDimitry Andric       }
15130b57cec5SDimitry Andric       // See this address is not already in the symbol table fake up an
15140b57cec5SDimitry Andric       // nlist for it.
15150b57cec5SDimitry Andric       if (!found) {
15160b57cec5SDimitry Andric         NMSymbol F = {};
15170b57cec5SDimitry Andric         F.Name = "<redacted function X>";
15180b57cec5SDimitry Andric         F.Address = FoundFns[f] + BaseSegmentAddress;
15190b57cec5SDimitry Andric         F.Size = 0;
15200b57cec5SDimitry Andric         // There is no symbol in the nlist symbol table for this so we set
15210b57cec5SDimitry Andric         // Sym effectivly to null and the rest of code in here must test for
15220b57cec5SDimitry Andric         // it and not do things like Sym.getFlags() for it.
15230b57cec5SDimitry Andric         F.Sym = BasicSymbolRef();
15240b57cec5SDimitry Andric         F.SymFlags = 0;
15250b57cec5SDimitry Andric         F.NType = MachO::N_SECT;
15260b57cec5SDimitry Andric         F.NSect = 0;
15270b57cec5SDimitry Andric         StringRef SegmentName = StringRef();
15280b57cec5SDimitry Andric         StringRef SectionName = StringRef();
1529e8d8bef9SDimitry Andric         for (const SectionRef &Section : MachO.sections()) {
15308bcb0991SDimitry Andric           if (Expected<StringRef> NameOrErr = Section.getName())
15318bcb0991SDimitry Andric             SectionName = *NameOrErr;
15328bcb0991SDimitry Andric           else
15338bcb0991SDimitry Andric             consumeError(NameOrErr.takeError());
15348bcb0991SDimitry Andric 
1535e8d8bef9SDimitry Andric           SegmentName =
1536e8d8bef9SDimitry Andric               MachO.getSectionFinalSegmentName(Section.getRawDataRefImpl());
15370b57cec5SDimitry Andric           F.NSect++;
15380b57cec5SDimitry Andric           if (F.Address >= Section.getAddress() &&
15390b57cec5SDimitry Andric               F.Address < Section.getAddress() + Section.getSize()) {
15400b57cec5SDimitry Andric             F.Section = Section;
15410b57cec5SDimitry Andric             break;
15420b57cec5SDimitry Andric           }
15430b57cec5SDimitry Andric         }
15440b57cec5SDimitry Andric         if (SegmentName == "__TEXT" && SectionName == "__text")
15450b57cec5SDimitry Andric           F.TypeChar = 't';
15460b57cec5SDimitry Andric         else if (SegmentName == "__DATA" && SectionName == "__data")
15470b57cec5SDimitry Andric           F.TypeChar = 'd';
15480b57cec5SDimitry Andric         else if (SegmentName == "__DATA" && SectionName == "__bss")
15490b57cec5SDimitry Andric           F.TypeChar = 'b';
15500b57cec5SDimitry Andric         else
15510b57cec5SDimitry Andric           F.TypeChar = 's';
15520b57cec5SDimitry Andric         F.NDesc = 0;
15530b57cec5SDimitry Andric         SymbolList.push_back(F);
15540b57cec5SDimitry Andric         if (FoundFns[f] == lc_main_offset)
15550b57cec5SDimitry Andric           FOS << "<redacted LC_MAIN>";
15560b57cec5SDimitry Andric         else
15570b57cec5SDimitry Andric           FOS << "<redacted function " << f << ">";
15580b57cec5SDimitry Andric         FOS << '\0';
15590b57cec5SDimitry Andric         FunctionStartsAdded++;
15600b57cec5SDimitry Andric       }
15610b57cec5SDimitry Andric     }
15620b57cec5SDimitry Andric     if (FunctionStartsAdded) {
15630b57cec5SDimitry Andric       FOS.flush();
15640b57cec5SDimitry Andric       const char *Q = FunctionStartsNameBuffer.c_str();
15650b57cec5SDimitry Andric       for (unsigned K = 0; K < FunctionStartsAdded; 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 }
15770b57cec5SDimitry Andric 
1578e8d8bef9SDimitry Andric namespace {
1579e8d8bef9SDimitry Andric struct SymbolVersion {
1580e8d8bef9SDimitry Andric   std::string Name;
1581e8d8bef9SDimitry Andric   bool IsDefault;
1582e8d8bef9SDimitry Andric };
1583e8d8bef9SDimitry Andric } // namespace
1584e8d8bef9SDimitry Andric 
1585e8d8bef9SDimitry Andric template <class ELFT>
1586e8d8bef9SDimitry Andric static Expected<std::vector<SymbolVersion>>
1587e8d8bef9SDimitry Andric readSymbolVersionsELF(const ELFFile<ELFT> &Obj, StringRef FileName,
1588e8d8bef9SDimitry Andric                       ELFObjectFileBase::elf_symbol_iterator_range Symbols) {
1589e8d8bef9SDimitry Andric   using Elf_Shdr = typename ELFT::Shdr;
1590e8d8bef9SDimitry Andric 
1591e8d8bef9SDimitry Andric   // We called sections() earlier, so can't fail here.
1592e8d8bef9SDimitry Andric   typename ELFT::ShdrRange SectionsOrErr = cantFail(Obj.sections());
1593e8d8bef9SDimitry Andric   const Elf_Shdr *SymVerSec = nullptr;
1594e8d8bef9SDimitry Andric   const Elf_Shdr *SymVerNeedSec = nullptr;
1595e8d8bef9SDimitry Andric   const Elf_Shdr *SymVerDefSec = nullptr;
1596e8d8bef9SDimitry Andric   for (const Elf_Shdr &Sec : SectionsOrErr) {
1597e8d8bef9SDimitry Andric     if (Sec.sh_type == ELF::SHT_GNU_versym)
1598e8d8bef9SDimitry Andric       SymVerSec = &Sec;
1599e8d8bef9SDimitry Andric     else if (Sec.sh_type == ELF::SHT_GNU_verdef)
1600e8d8bef9SDimitry Andric       SymVerDefSec = &Sec;
1601e8d8bef9SDimitry Andric     else if (Sec.sh_type == ELF::SHT_GNU_verneed)
1602e8d8bef9SDimitry Andric       SymVerNeedSec = &Sec;
1603e8d8bef9SDimitry Andric   }
1604e8d8bef9SDimitry Andric 
1605e8d8bef9SDimitry Andric   if (!SymVerSec)
1606e8d8bef9SDimitry Andric     return std::vector<SymbolVersion>{};
1607e8d8bef9SDimitry Andric 
1608e8d8bef9SDimitry Andric   Expected<SmallVector<Optional<VersionEntry>, 0>> MapOrErr =
1609e8d8bef9SDimitry Andric       Obj.loadVersionMap(SymVerNeedSec, SymVerDefSec);
1610e8d8bef9SDimitry Andric   if (!MapOrErr)
1611e8d8bef9SDimitry Andric     return MapOrErr.takeError();
1612e8d8bef9SDimitry Andric 
1613e8d8bef9SDimitry Andric   std::vector<SymbolVersion> Ret;
1614e8d8bef9SDimitry Andric   size_t I = 0;
1615e8d8bef9SDimitry Andric   for (auto It = Symbols.begin(), E = Symbols.end(); It != E; ++It) {
1616e8d8bef9SDimitry Andric     ++I;
1617e8d8bef9SDimitry Andric     Expected<const typename ELFT::Versym *> VerEntryOrErr =
1618e8d8bef9SDimitry Andric         Obj.template getEntry<typename ELFT::Versym>(*SymVerSec, I);
1619e8d8bef9SDimitry Andric     if (!VerEntryOrErr)
1620e8d8bef9SDimitry Andric       return createError("unable to read an entry with index " + Twine(I) +
1621e8d8bef9SDimitry Andric                          " from " + describe(Obj, *SymVerSec) + ": " +
1622e8d8bef9SDimitry Andric                          toString(VerEntryOrErr.takeError()));
1623e8d8bef9SDimitry Andric 
1624e8d8bef9SDimitry Andric     Expected<uint32_t> FlagsOrErr = It->getFlags();
1625e8d8bef9SDimitry Andric     if (!FlagsOrErr)
1626e8d8bef9SDimitry Andric       return createError("unable to read flags for symbol with index " +
1627e8d8bef9SDimitry Andric                          Twine(I) + ": " + toString(FlagsOrErr.takeError()));
1628e8d8bef9SDimitry Andric 
1629e8d8bef9SDimitry Andric     bool IsDefault;
1630e8d8bef9SDimitry Andric     Expected<StringRef> VerOrErr = Obj.getSymbolVersionByIndex(
1631e8d8bef9SDimitry Andric         (*VerEntryOrErr)->vs_index, IsDefault, *MapOrErr,
1632e8d8bef9SDimitry Andric         (*FlagsOrErr) & SymbolRef::SF_Undefined);
1633e8d8bef9SDimitry Andric     if (!VerOrErr)
1634e8d8bef9SDimitry Andric       return createError("unable to get a version for entry " + Twine(I) +
1635e8d8bef9SDimitry Andric                          " of " + describe(Obj, *SymVerSec) + ": " +
1636e8d8bef9SDimitry Andric                          toString(VerOrErr.takeError()));
1637e8d8bef9SDimitry Andric 
1638e8d8bef9SDimitry Andric     Ret.push_back({(*VerOrErr).str(), IsDefault});
1639e8d8bef9SDimitry Andric   }
1640e8d8bef9SDimitry Andric 
1641e8d8bef9SDimitry Andric   return Ret;
1642e8d8bef9SDimitry Andric }
1643e8d8bef9SDimitry Andric 
1644e8d8bef9SDimitry Andric static Expected<std::vector<SymbolVersion>>
1645e8d8bef9SDimitry Andric readSymbolVersionsELF(const ELFObjectFileBase &Obj,
1646e8d8bef9SDimitry Andric                       ELFObjectFileBase::elf_symbol_iterator_range Symbols) {
1647e8d8bef9SDimitry Andric   if (const auto *ELF = dyn_cast<ELF32LEObjectFile>(&Obj))
1648e8d8bef9SDimitry Andric     return readSymbolVersionsELF(ELF->getELFFile(), Obj.getFileName(), Symbols);
1649e8d8bef9SDimitry Andric   else if (const auto *ELF = dyn_cast<ELF32BEObjectFile>(&Obj))
1650e8d8bef9SDimitry Andric     return readSymbolVersionsELF(ELF->getELFFile(), Obj.getFileName(), Symbols);
1651e8d8bef9SDimitry Andric   else if (const auto *ELF = dyn_cast<ELF64LEObjectFile>(&Obj))
1652e8d8bef9SDimitry Andric     return readSymbolVersionsELF(ELF->getELFFile(), Obj.getFileName(), Symbols);
1653e8d8bef9SDimitry Andric   return readSymbolVersionsELF(cast<ELF64BEObjectFile>(&Obj)->getELFFile(),
1654e8d8bef9SDimitry Andric                                Obj.getFileName(), Symbols);
1655e8d8bef9SDimitry Andric }
1656e8d8bef9SDimitry Andric 
1657e8d8bef9SDimitry Andric static void dumpSymbolNamesFromObject(SymbolicFile &Obj, bool printName,
1658e8d8bef9SDimitry Andric                                       StringRef ArchiveName = {},
1659e8d8bef9SDimitry Andric                                       StringRef ArchitectureName = {}) {
1660e8d8bef9SDimitry Andric   auto Symbols = Obj.symbols();
1661e8d8bef9SDimitry Andric   std::vector<SymbolVersion> SymbolVersions;
1662e8d8bef9SDimitry Andric   if (DynamicSyms) {
1663e8d8bef9SDimitry Andric     const auto *E = dyn_cast<ELFObjectFileBase>(&Obj);
1664e8d8bef9SDimitry Andric     if (!E) {
1665e8d8bef9SDimitry Andric       error("File format has no dynamic symbol table", Obj.getFileName());
1666e8d8bef9SDimitry Andric       return;
1667e8d8bef9SDimitry Andric     }
1668e8d8bef9SDimitry Andric     Symbols = E->getDynamicSymbolIterators();
1669e8d8bef9SDimitry Andric 
1670e8d8bef9SDimitry Andric     if (Expected<std::vector<SymbolVersion>> VersionsOrErr =
1671e8d8bef9SDimitry Andric             readSymbolVersionsELF(*E, Symbols))
1672e8d8bef9SDimitry Andric       SymbolVersions = std::move(*VersionsOrErr);
1673e8d8bef9SDimitry Andric     else
1674e8d8bef9SDimitry Andric       WithColor::warning(errs(), ToolName)
1675e8d8bef9SDimitry Andric           << "unable to read symbol versions: "
1676e8d8bef9SDimitry Andric           << toString(VersionsOrErr.takeError()) << "\n";
1677e8d8bef9SDimitry Andric   }
1678e8d8bef9SDimitry Andric 
1679e8d8bef9SDimitry Andric   // If a "-s segname sectname" option was specified and this is a Mach-O
1680e8d8bef9SDimitry Andric   // file get the section number for that section in this object file.
1681e8d8bef9SDimitry Andric   unsigned int Nsect = 0;
1682e8d8bef9SDimitry Andric   MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(&Obj);
1683e8d8bef9SDimitry Andric   if (!SegSect.empty() && MachO) {
1684e8d8bef9SDimitry Andric     Nsect = getNsectForSegSect(MachO);
1685e8d8bef9SDimitry Andric     // If this section is not in the object file no symbols are printed.
1686e8d8bef9SDimitry Andric     if (Nsect == 0)
1687e8d8bef9SDimitry Andric       return;
1688e8d8bef9SDimitry Andric   }
1689e8d8bef9SDimitry Andric   if (!(MachO && DyldInfoOnly)) {
1690e8d8bef9SDimitry Andric     size_t I = -1;
1691e8d8bef9SDimitry Andric     for (BasicSymbolRef Sym : Symbols) {
1692e8d8bef9SDimitry Andric       ++I;
1693e8d8bef9SDimitry Andric       Expected<uint32_t> SymFlagsOrErr = Sym.getFlags();
1694e8d8bef9SDimitry Andric       if (!SymFlagsOrErr) {
1695e8d8bef9SDimitry Andric         error(SymFlagsOrErr.takeError(), Obj.getFileName());
1696e8d8bef9SDimitry Andric         return;
1697e8d8bef9SDimitry Andric       }
1698*fe6060f1SDimitry Andric 
1699*fe6060f1SDimitry Andric       // Don't drop format specifc symbols for ARM and AArch64 ELF targets, they
1700*fe6060f1SDimitry Andric       // are used to repesent mapping symbols and needed to honor the
1701*fe6060f1SDimitry Andric       // --special-syms option.
1702*fe6060f1SDimitry Andric       auto *ELFObj = dyn_cast<ELFObjectFileBase>(&Obj);
1703*fe6060f1SDimitry Andric       if ((!ELFObj || (ELFObj->getEMachine() != ELF::EM_ARM &&
1704*fe6060f1SDimitry Andric                        ELFObj->getEMachine() != ELF::EM_AARCH64)) &&
1705*fe6060f1SDimitry Andric           !DebugSyms && (*SymFlagsOrErr & SymbolRef::SF_FormatSpecific))
1706e8d8bef9SDimitry Andric         continue;
1707e8d8bef9SDimitry Andric       if (WithoutAliases && (*SymFlagsOrErr & SymbolRef::SF_Indirect))
1708e8d8bef9SDimitry Andric         continue;
1709e8d8bef9SDimitry Andric       // If a "-s segname sectname" option was specified and this is a Mach-O
1710e8d8bef9SDimitry Andric       // file and this section appears in this file, Nsect will be non-zero then
1711e8d8bef9SDimitry Andric       // see if this symbol is a symbol from that section and if not skip it.
1712e8d8bef9SDimitry Andric       if (Nsect && Nsect != getNsectInMachO(*MachO, Sym))
1713e8d8bef9SDimitry Andric         continue;
1714e8d8bef9SDimitry Andric       NMSymbol S = {};
1715e8d8bef9SDimitry Andric       S.Size = 0;
1716e8d8bef9SDimitry Andric       S.Address = 0;
1717e8d8bef9SDimitry Andric       if (isa<ELFObjectFileBase>(&Obj))
1718e8d8bef9SDimitry Andric         S.Size = ELFSymbolRef(Sym).getSize();
1719e8d8bef9SDimitry Andric       if (PrintAddress && isa<ObjectFile>(Obj)) {
1720e8d8bef9SDimitry Andric         SymbolRef SymRef(Sym);
1721e8d8bef9SDimitry Andric         Expected<uint64_t> AddressOrErr = SymRef.getAddress();
1722e8d8bef9SDimitry Andric         if (!AddressOrErr) {
1723e8d8bef9SDimitry Andric           consumeError(AddressOrErr.takeError());
1724e8d8bef9SDimitry Andric           break;
1725e8d8bef9SDimitry Andric         }
1726e8d8bef9SDimitry Andric         S.Address = *AddressOrErr;
1727e8d8bef9SDimitry Andric       }
1728e8d8bef9SDimitry Andric       S.TypeName = getNMTypeName(Obj, Sym);
1729e8d8bef9SDimitry Andric       S.TypeChar = getNMSectionTagAndName(Obj, Sym, S.SectionName);
1730e8d8bef9SDimitry Andric 
1731e8d8bef9SDimitry Andric       raw_string_ostream OS(S.Name);
1732e8d8bef9SDimitry Andric       if (Error E = Sym.printName(OS)) {
1733e8d8bef9SDimitry Andric         if (MachO) {
1734e8d8bef9SDimitry Andric           OS << "bad string index";
1735e8d8bef9SDimitry Andric           consumeError(std::move(E));
1736e8d8bef9SDimitry Andric         } else
1737e8d8bef9SDimitry Andric           error(std::move(E), Obj.getFileName());
1738e8d8bef9SDimitry Andric       }
1739e8d8bef9SDimitry Andric       if (!SymbolVersions.empty() && !SymbolVersions[I].Name.empty())
1740e8d8bef9SDimitry Andric         S.Name +=
1741e8d8bef9SDimitry Andric             (SymbolVersions[I].IsDefault ? "@@" : "@") + SymbolVersions[I].Name;
1742e8d8bef9SDimitry Andric 
1743e8d8bef9SDimitry Andric       S.Sym = Sym;
1744e8d8bef9SDimitry Andric       SymbolList.push_back(S);
1745e8d8bef9SDimitry Andric     }
1746e8d8bef9SDimitry Andric   }
1747e8d8bef9SDimitry Andric 
1748e8d8bef9SDimitry Andric   // If this is a Mach-O file where the nlist symbol table is out of sync
1749e8d8bef9SDimitry Andric   // with the dyld export trie then look through exports and fake up symbols
1750e8d8bef9SDimitry Andric   // for the ones that are missing (also done with the -add-dyldinfo flag).
1751e8d8bef9SDimitry Andric   // This is needed if strip(1) -T is run on a binary containing swift
1752e8d8bef9SDimitry Andric   // language symbols for example.  The option -only-dyldinfo will fake up
1753e8d8bef9SDimitry Andric   // all symbols from the dyld export trie as well as the bind info.
1754e8d8bef9SDimitry Andric   if (MachO && !NoDyldInfo)
1755e8d8bef9SDimitry Andric     dumpSymbolsFromDLInfoMachO(*MachO);
1756e8d8bef9SDimitry Andric 
17570b57cec5SDimitry Andric   CurrentFilename = Obj.getFileName();
17585ffd83dbSDimitry Andric 
1759*fe6060f1SDimitry Andric   if (Symbols.empty() && SymbolList.empty() && !Quiet) {
17605ffd83dbSDimitry Andric     writeFileName(errs(), ArchiveName, ArchitectureName);
17615ffd83dbSDimitry Andric     errs() << "no symbols\n";
17625ffd83dbSDimitry Andric   }
17635ffd83dbSDimitry Andric 
17640b57cec5SDimitry Andric   sortAndPrintSymbolList(Obj, printName, ArchiveName, ArchitectureName);
17650b57cec5SDimitry Andric }
17660b57cec5SDimitry Andric 
17670b57cec5SDimitry Andric // checkMachOAndArchFlags() checks to see if the SymbolicFile is a Mach-O file
17680b57cec5SDimitry Andric // and if it is and there is a list of architecture flags is specified then
17690b57cec5SDimitry Andric // check to make sure this Mach-O file is one of those architectures or all
17700b57cec5SDimitry Andric // architectures was specificed.  If not then an error is generated and this
17710b57cec5SDimitry Andric // routine returns false.  Else it returns true.
17720b57cec5SDimitry Andric static bool checkMachOAndArchFlags(SymbolicFile *O, std::string &Filename) {
17730b57cec5SDimitry Andric   auto *MachO = dyn_cast<MachOObjectFile>(O);
17740b57cec5SDimitry Andric 
17750b57cec5SDimitry Andric   if (!MachO || ArchAll || ArchFlags.empty())
17760b57cec5SDimitry Andric     return true;
17770b57cec5SDimitry Andric 
17780b57cec5SDimitry Andric   MachO::mach_header H;
17790b57cec5SDimitry Andric   MachO::mach_header_64 H_64;
17800b57cec5SDimitry Andric   Triple T;
17810b57cec5SDimitry Andric   const char *McpuDefault, *ArchFlag;
17820b57cec5SDimitry Andric   if (MachO->is64Bit()) {
17830b57cec5SDimitry Andric     H_64 = MachO->MachOObjectFile::getHeader64();
17840b57cec5SDimitry Andric     T = MachOObjectFile::getArchTriple(H_64.cputype, H_64.cpusubtype,
17850b57cec5SDimitry Andric                                        &McpuDefault, &ArchFlag);
17860b57cec5SDimitry Andric   } else {
17870b57cec5SDimitry Andric     H = MachO->MachOObjectFile::getHeader();
17880b57cec5SDimitry Andric     T = MachOObjectFile::getArchTriple(H.cputype, H.cpusubtype,
17890b57cec5SDimitry Andric                                        &McpuDefault, &ArchFlag);
17900b57cec5SDimitry Andric   }
17910b57cec5SDimitry Andric   const std::string ArchFlagName(ArchFlag);
1792e8d8bef9SDimitry Andric   if (!llvm::is_contained(ArchFlags, ArchFlagName)) {
17930b57cec5SDimitry Andric     error("No architecture specified", Filename);
17940b57cec5SDimitry Andric     return false;
17950b57cec5SDimitry Andric   }
17960b57cec5SDimitry Andric   return true;
17970b57cec5SDimitry Andric }
17980b57cec5SDimitry Andric 
17990b57cec5SDimitry Andric static void dumpSymbolNamesFromFile(std::string &Filename) {
18000b57cec5SDimitry Andric   ErrorOr<std::unique_ptr<MemoryBuffer>> BufferOrErr =
18010b57cec5SDimitry Andric       MemoryBuffer::getFileOrSTDIN(Filename);
18020b57cec5SDimitry Andric   if (error(BufferOrErr.getError(), Filename))
18030b57cec5SDimitry Andric     return;
18040b57cec5SDimitry Andric 
18050b57cec5SDimitry Andric   LLVMContext Context;
18060b57cec5SDimitry Andric   LLVMContext *ContextPtr = NoLLVMBitcode ? nullptr : &Context;
18070b57cec5SDimitry Andric   Expected<std::unique_ptr<Binary>> BinaryOrErr =
18080b57cec5SDimitry Andric       createBinary(BufferOrErr.get()->getMemBufferRef(), ContextPtr);
18090b57cec5SDimitry Andric   if (!BinaryOrErr) {
18100b57cec5SDimitry Andric     error(BinaryOrErr.takeError(), Filename);
18110b57cec5SDimitry Andric     return;
18120b57cec5SDimitry Andric   }
18130b57cec5SDimitry Andric   Binary &Bin = *BinaryOrErr.get();
18140b57cec5SDimitry Andric 
18150b57cec5SDimitry Andric   if (Archive *A = dyn_cast<Archive>(&Bin)) {
18160b57cec5SDimitry Andric     if (ArchiveMap) {
18170b57cec5SDimitry Andric       Archive::symbol_iterator I = A->symbol_begin();
18180b57cec5SDimitry Andric       Archive::symbol_iterator E = A->symbol_end();
18190b57cec5SDimitry Andric       if (I != E) {
18200b57cec5SDimitry Andric         outs() << "Archive map\n";
18210b57cec5SDimitry Andric         for (; I != E; ++I) {
18220b57cec5SDimitry Andric           Expected<Archive::Child> C = I->getMember();
18230b57cec5SDimitry Andric           if (!C) {
18240b57cec5SDimitry Andric             error(C.takeError(), Filename);
18250b57cec5SDimitry Andric             break;
18260b57cec5SDimitry Andric           }
18270b57cec5SDimitry Andric           Expected<StringRef> FileNameOrErr = C->getName();
18280b57cec5SDimitry Andric           if (!FileNameOrErr) {
18290b57cec5SDimitry Andric             error(FileNameOrErr.takeError(), Filename);
18300b57cec5SDimitry Andric             break;
18310b57cec5SDimitry Andric           }
18320b57cec5SDimitry Andric           StringRef SymName = I->getName();
18330b57cec5SDimitry Andric           outs() << SymName << " in " << FileNameOrErr.get() << "\n";
18340b57cec5SDimitry Andric         }
18350b57cec5SDimitry Andric         outs() << "\n";
18360b57cec5SDimitry Andric       }
18370b57cec5SDimitry Andric     }
18380b57cec5SDimitry Andric 
18390b57cec5SDimitry Andric     {
18400b57cec5SDimitry Andric       Error Err = Error::success();
18410b57cec5SDimitry Andric       for (auto &C : A->children(Err)) {
18420b57cec5SDimitry Andric         Expected<std::unique_ptr<Binary>> ChildOrErr =
18430b57cec5SDimitry Andric             C.getAsBinary(ContextPtr);
18440b57cec5SDimitry Andric         if (!ChildOrErr) {
18450b57cec5SDimitry Andric           if (auto E = isNotObjectErrorInvalidFileType(ChildOrErr.takeError()))
18460b57cec5SDimitry Andric             error(std::move(E), Filename, C);
18470b57cec5SDimitry Andric           continue;
18480b57cec5SDimitry Andric         }
18490b57cec5SDimitry Andric         if (SymbolicFile *O = dyn_cast<SymbolicFile>(&*ChildOrErr.get())) {
18500b57cec5SDimitry Andric           if (!MachOPrintSizeWarning && PrintSize &&  isa<MachOObjectFile>(O)) {
18510b57cec5SDimitry Andric             WithColor::warning(errs(), ToolName)
18520b57cec5SDimitry Andric                 << "sizes with -print-size for Mach-O files are always zero.\n";
18530b57cec5SDimitry Andric             MachOPrintSizeWarning = true;
18540b57cec5SDimitry Andric           }
18550b57cec5SDimitry Andric           if (!checkMachOAndArchFlags(O, Filename))
18560b57cec5SDimitry Andric             return;
18570b57cec5SDimitry Andric           if (!PrintFileName) {
18580b57cec5SDimitry Andric             outs() << "\n";
18590b57cec5SDimitry Andric             if (isa<MachOObjectFile>(O)) {
18600b57cec5SDimitry Andric               outs() << Filename << "(" << O->getFileName() << ")";
18610b57cec5SDimitry Andric             } else
18620b57cec5SDimitry Andric               outs() << O->getFileName();
18630b57cec5SDimitry Andric             outs() << ":\n";
18640b57cec5SDimitry Andric           }
18650b57cec5SDimitry Andric           dumpSymbolNamesFromObject(*O, false, Filename);
18660b57cec5SDimitry Andric         }
18670b57cec5SDimitry Andric       }
18680b57cec5SDimitry Andric       if (Err)
18690b57cec5SDimitry Andric         error(std::move(Err), A->getFileName());
18700b57cec5SDimitry Andric     }
18710b57cec5SDimitry Andric     return;
18720b57cec5SDimitry Andric   }
18730b57cec5SDimitry Andric   if (MachOUniversalBinary *UB = dyn_cast<MachOUniversalBinary>(&Bin)) {
18740b57cec5SDimitry Andric     // If we have a list of architecture flags specified dump only those.
18750b57cec5SDimitry Andric     if (!ArchAll && !ArchFlags.empty()) {
18760b57cec5SDimitry Andric       // Look for a slice in the universal binary that matches each ArchFlag.
18770b57cec5SDimitry Andric       bool ArchFound;
18780b57cec5SDimitry Andric       for (unsigned i = 0; i < ArchFlags.size(); ++i) {
18790b57cec5SDimitry Andric         ArchFound = false;
18800b57cec5SDimitry Andric         for (MachOUniversalBinary::object_iterator I = UB->begin_objects(),
18810b57cec5SDimitry Andric                                                    E = UB->end_objects();
18820b57cec5SDimitry Andric              I != E; ++I) {
18830b57cec5SDimitry Andric           if (ArchFlags[i] == I->getArchFlagName()) {
18840b57cec5SDimitry Andric             ArchFound = true;
18850b57cec5SDimitry Andric             Expected<std::unique_ptr<ObjectFile>> ObjOrErr =
18860b57cec5SDimitry Andric                 I->getAsObjectFile();
18870b57cec5SDimitry Andric             std::string ArchiveName;
18880b57cec5SDimitry Andric             std::string ArchitectureName;
18890b57cec5SDimitry Andric             ArchiveName.clear();
18900b57cec5SDimitry Andric             ArchitectureName.clear();
18910b57cec5SDimitry Andric             if (ObjOrErr) {
18920b57cec5SDimitry Andric               ObjectFile &Obj = *ObjOrErr.get();
18930b57cec5SDimitry Andric               if (ArchFlags.size() > 1) {
18940b57cec5SDimitry Andric                 if (PrintFileName)
18950b57cec5SDimitry Andric                   ArchitectureName = I->getArchFlagName();
18960b57cec5SDimitry Andric                 else
18970b57cec5SDimitry Andric                   outs() << "\n" << Obj.getFileName() << " (for architecture "
18980b57cec5SDimitry Andric                          << I->getArchFlagName() << ")"
18990b57cec5SDimitry Andric                          << ":\n";
19000b57cec5SDimitry Andric               }
19010b57cec5SDimitry Andric               dumpSymbolNamesFromObject(Obj, false, ArchiveName,
19020b57cec5SDimitry Andric                                         ArchitectureName);
19030b57cec5SDimitry Andric             } else if (auto E = isNotObjectErrorInvalidFileType(
19040b57cec5SDimitry Andric                        ObjOrErr.takeError())) {
19050b57cec5SDimitry Andric               error(std::move(E), Filename, ArchFlags.size() > 1 ?
19060b57cec5SDimitry Andric                     StringRef(I->getArchFlagName()) : StringRef());
19070b57cec5SDimitry Andric               continue;
19080b57cec5SDimitry Andric             } else if (Expected<std::unique_ptr<Archive>> AOrErr =
19090b57cec5SDimitry Andric                            I->getAsArchive()) {
19100b57cec5SDimitry Andric               std::unique_ptr<Archive> &A = *AOrErr;
19110b57cec5SDimitry Andric               Error Err = Error::success();
19120b57cec5SDimitry Andric               for (auto &C : A->children(Err)) {
19130b57cec5SDimitry Andric                 Expected<std::unique_ptr<Binary>> ChildOrErr =
19140b57cec5SDimitry Andric                     C.getAsBinary(ContextPtr);
19150b57cec5SDimitry Andric                 if (!ChildOrErr) {
19160b57cec5SDimitry Andric                   if (auto E = isNotObjectErrorInvalidFileType(
19170b57cec5SDimitry Andric                                        ChildOrErr.takeError())) {
19180b57cec5SDimitry Andric                     error(std::move(E), Filename, C, ArchFlags.size() > 1 ?
19190b57cec5SDimitry Andric                           StringRef(I->getArchFlagName()) : StringRef());
19200b57cec5SDimitry Andric                   }
19210b57cec5SDimitry Andric                   continue;
19220b57cec5SDimitry Andric                 }
19230b57cec5SDimitry Andric                 if (SymbolicFile *O =
19240b57cec5SDimitry Andric                         dyn_cast<SymbolicFile>(&*ChildOrErr.get())) {
19250b57cec5SDimitry Andric                   if (PrintFileName) {
19265ffd83dbSDimitry Andric                     ArchiveName = std::string(A->getFileName());
19270b57cec5SDimitry Andric                     if (ArchFlags.size() > 1)
19280b57cec5SDimitry Andric                       ArchitectureName = I->getArchFlagName();
19290b57cec5SDimitry Andric                   } else {
19300b57cec5SDimitry Andric                     outs() << "\n" << A->getFileName();
19310b57cec5SDimitry Andric                     outs() << "(" << O->getFileName() << ")";
19320b57cec5SDimitry Andric                     if (ArchFlags.size() > 1) {
19330b57cec5SDimitry Andric                       outs() << " (for architecture " << I->getArchFlagName()
19340b57cec5SDimitry Andric                              << ")";
19350b57cec5SDimitry Andric                     }
19360b57cec5SDimitry Andric                     outs() << ":\n";
19370b57cec5SDimitry Andric                   }
19380b57cec5SDimitry Andric                   dumpSymbolNamesFromObject(*O, false, ArchiveName,
19390b57cec5SDimitry Andric                                             ArchitectureName);
19400b57cec5SDimitry Andric                 }
19410b57cec5SDimitry Andric               }
19420b57cec5SDimitry Andric               if (Err)
19430b57cec5SDimitry Andric                 error(std::move(Err), A->getFileName());
19440b57cec5SDimitry Andric             } else {
19450b57cec5SDimitry Andric               consumeError(AOrErr.takeError());
19460b57cec5SDimitry Andric               error(Filename + " for architecture " +
19470b57cec5SDimitry Andric                     StringRef(I->getArchFlagName()) +
19480b57cec5SDimitry Andric                     " is not a Mach-O file or an archive file",
19490b57cec5SDimitry Andric                     "Mach-O universal file");
19500b57cec5SDimitry Andric             }
19510b57cec5SDimitry Andric           }
19520b57cec5SDimitry Andric         }
19530b57cec5SDimitry Andric         if (!ArchFound) {
19540b57cec5SDimitry Andric           error(ArchFlags[i],
19550b57cec5SDimitry Andric                 "file: " + Filename + " does not contain architecture");
19560b57cec5SDimitry Andric           return;
19570b57cec5SDimitry Andric         }
19580b57cec5SDimitry Andric       }
19590b57cec5SDimitry Andric       return;
19600b57cec5SDimitry Andric     }
19610b57cec5SDimitry Andric     // No architecture flags were specified so if this contains a slice that
19620b57cec5SDimitry Andric     // matches the host architecture dump only that.
19630b57cec5SDimitry Andric     if (!ArchAll) {
19640b57cec5SDimitry Andric       Triple HostTriple = MachOObjectFile::getHostArch();
19650b57cec5SDimitry Andric       StringRef HostArchName = HostTriple.getArchName();
19660b57cec5SDimitry Andric       for (MachOUniversalBinary::object_iterator I = UB->begin_objects(),
19670b57cec5SDimitry Andric                                                  E = UB->end_objects();
19680b57cec5SDimitry Andric            I != E; ++I) {
19690b57cec5SDimitry Andric         if (HostArchName == I->getArchFlagName()) {
19700b57cec5SDimitry Andric           Expected<std::unique_ptr<ObjectFile>> ObjOrErr = I->getAsObjectFile();
19710b57cec5SDimitry Andric           std::string ArchiveName;
19720b57cec5SDimitry Andric           if (ObjOrErr) {
19730b57cec5SDimitry Andric             ObjectFile &Obj = *ObjOrErr.get();
19740b57cec5SDimitry Andric             dumpSymbolNamesFromObject(Obj, false);
19750b57cec5SDimitry Andric           } else if (auto E = isNotObjectErrorInvalidFileType(
19760b57cec5SDimitry Andric                      ObjOrErr.takeError())) {
19770b57cec5SDimitry Andric             error(std::move(E), Filename);
19780b57cec5SDimitry Andric             return;
19790b57cec5SDimitry Andric           } else if (Expected<std::unique_ptr<Archive>> AOrErr =
19800b57cec5SDimitry Andric                          I->getAsArchive()) {
19810b57cec5SDimitry Andric             std::unique_ptr<Archive> &A = *AOrErr;
19820b57cec5SDimitry Andric             Error Err = Error::success();
19830b57cec5SDimitry Andric             for (auto &C : A->children(Err)) {
19840b57cec5SDimitry Andric               Expected<std::unique_ptr<Binary>> ChildOrErr =
19850b57cec5SDimitry Andric                   C.getAsBinary(ContextPtr);
19860b57cec5SDimitry Andric               if (!ChildOrErr) {
19870b57cec5SDimitry Andric                 if (auto E = isNotObjectErrorInvalidFileType(
19880b57cec5SDimitry Andric                                      ChildOrErr.takeError()))
19890b57cec5SDimitry Andric                   error(std::move(E), Filename, C);
19900b57cec5SDimitry Andric                 continue;
19910b57cec5SDimitry Andric               }
19920b57cec5SDimitry Andric               if (SymbolicFile *O =
19930b57cec5SDimitry Andric                       dyn_cast<SymbolicFile>(&*ChildOrErr.get())) {
19940b57cec5SDimitry Andric                 if (PrintFileName)
19955ffd83dbSDimitry Andric                   ArchiveName = std::string(A->getFileName());
19960b57cec5SDimitry Andric                 else
19970b57cec5SDimitry Andric                   outs() << "\n" << A->getFileName() << "(" << O->getFileName()
19980b57cec5SDimitry Andric                          << ")"
19990b57cec5SDimitry Andric                          << ":\n";
20000b57cec5SDimitry Andric                 dumpSymbolNamesFromObject(*O, false, ArchiveName);
20010b57cec5SDimitry Andric               }
20020b57cec5SDimitry Andric             }
20030b57cec5SDimitry Andric             if (Err)
20040b57cec5SDimitry Andric               error(std::move(Err), A->getFileName());
20050b57cec5SDimitry Andric           } else {
20060b57cec5SDimitry Andric             consumeError(AOrErr.takeError());
20070b57cec5SDimitry Andric             error(Filename + " for architecture " +
20080b57cec5SDimitry Andric                   StringRef(I->getArchFlagName()) +
20090b57cec5SDimitry Andric                   " is not a Mach-O file or an archive file",
20100b57cec5SDimitry Andric                   "Mach-O universal file");
20110b57cec5SDimitry Andric           }
20120b57cec5SDimitry Andric           return;
20130b57cec5SDimitry Andric         }
20140b57cec5SDimitry Andric       }
20150b57cec5SDimitry Andric     }
20160b57cec5SDimitry Andric     // Either all architectures have been specified or none have been specified
20170b57cec5SDimitry Andric     // and this does not contain the host architecture so dump all the slices.
20180b57cec5SDimitry Andric     bool moreThanOneArch = UB->getNumberOfObjects() > 1;
20190b57cec5SDimitry Andric     for (const MachOUniversalBinary::ObjectForArch &O : UB->objects()) {
20200b57cec5SDimitry Andric       Expected<std::unique_ptr<ObjectFile>> ObjOrErr = O.getAsObjectFile();
20210b57cec5SDimitry Andric       std::string ArchiveName;
20220b57cec5SDimitry Andric       std::string ArchitectureName;
20230b57cec5SDimitry Andric       ArchiveName.clear();
20240b57cec5SDimitry Andric       ArchitectureName.clear();
20250b57cec5SDimitry Andric       if (ObjOrErr) {
20260b57cec5SDimitry Andric         ObjectFile &Obj = *ObjOrErr.get();
20270b57cec5SDimitry Andric         if (PrintFileName) {
20280b57cec5SDimitry Andric           if (isa<MachOObjectFile>(Obj) && moreThanOneArch)
20290b57cec5SDimitry Andric             ArchitectureName = O.getArchFlagName();
20300b57cec5SDimitry Andric         } else {
20310b57cec5SDimitry Andric           if (moreThanOneArch)
20320b57cec5SDimitry Andric             outs() << "\n";
20330b57cec5SDimitry Andric           outs() << Obj.getFileName();
20340b57cec5SDimitry Andric           if (isa<MachOObjectFile>(Obj) && moreThanOneArch)
20350b57cec5SDimitry Andric             outs() << " (for architecture " << O.getArchFlagName() << ")";
20360b57cec5SDimitry Andric           outs() << ":\n";
20370b57cec5SDimitry Andric         }
20380b57cec5SDimitry Andric         dumpSymbolNamesFromObject(Obj, false, ArchiveName, ArchitectureName);
20390b57cec5SDimitry Andric       } else if (auto E = isNotObjectErrorInvalidFileType(
20400b57cec5SDimitry Andric                  ObjOrErr.takeError())) {
20410b57cec5SDimitry Andric         error(std::move(E), Filename, moreThanOneArch ?
20420b57cec5SDimitry Andric               StringRef(O.getArchFlagName()) : StringRef());
20430b57cec5SDimitry Andric         continue;
20440b57cec5SDimitry Andric       } else if (Expected<std::unique_ptr<Archive>> AOrErr =
20450b57cec5SDimitry Andric                   O.getAsArchive()) {
20460b57cec5SDimitry Andric         std::unique_ptr<Archive> &A = *AOrErr;
20470b57cec5SDimitry Andric         Error Err = Error::success();
20480b57cec5SDimitry Andric         for (auto &C : A->children(Err)) {
20490b57cec5SDimitry Andric           Expected<std::unique_ptr<Binary>> ChildOrErr =
20500b57cec5SDimitry Andric             C.getAsBinary(ContextPtr);
20510b57cec5SDimitry Andric           if (!ChildOrErr) {
20520b57cec5SDimitry Andric             if (auto E = isNotObjectErrorInvalidFileType(
20530b57cec5SDimitry Andric                                  ChildOrErr.takeError()))
20540b57cec5SDimitry Andric               error(std::move(E), Filename, C, moreThanOneArch ?
20550b57cec5SDimitry Andric                     StringRef(ArchitectureName) : StringRef());
20560b57cec5SDimitry Andric             continue;
20570b57cec5SDimitry Andric           }
20580b57cec5SDimitry Andric           if (SymbolicFile *F = dyn_cast<SymbolicFile>(&*ChildOrErr.get())) {
20590b57cec5SDimitry Andric             if (PrintFileName) {
20605ffd83dbSDimitry Andric               ArchiveName = std::string(A->getFileName());
20610b57cec5SDimitry Andric               if (isa<MachOObjectFile>(F) && moreThanOneArch)
20620b57cec5SDimitry Andric                 ArchitectureName = O.getArchFlagName();
20630b57cec5SDimitry Andric             } else {
20640b57cec5SDimitry Andric               outs() << "\n" << A->getFileName();
20650b57cec5SDimitry Andric               if (isa<MachOObjectFile>(F)) {
20660b57cec5SDimitry Andric                 outs() << "(" << F->getFileName() << ")";
20670b57cec5SDimitry Andric                 if (moreThanOneArch)
20680b57cec5SDimitry Andric                   outs() << " (for architecture " << O.getArchFlagName()
20690b57cec5SDimitry Andric                          << ")";
20700b57cec5SDimitry Andric               } else
20710b57cec5SDimitry Andric                 outs() << ":" << F->getFileName();
20720b57cec5SDimitry Andric               outs() << ":\n";
20730b57cec5SDimitry Andric             }
20740b57cec5SDimitry Andric             dumpSymbolNamesFromObject(*F, false, ArchiveName, ArchitectureName);
20750b57cec5SDimitry Andric           }
20760b57cec5SDimitry Andric         }
20770b57cec5SDimitry Andric         if (Err)
20780b57cec5SDimitry Andric           error(std::move(Err), A->getFileName());
20790b57cec5SDimitry Andric       } else {
20800b57cec5SDimitry Andric         consumeError(AOrErr.takeError());
20810b57cec5SDimitry Andric         error(Filename + " for architecture " +
20820b57cec5SDimitry Andric               StringRef(O.getArchFlagName()) +
20830b57cec5SDimitry Andric               " is not a Mach-O file or an archive file",
20840b57cec5SDimitry Andric               "Mach-O universal file");
20850b57cec5SDimitry Andric       }
20860b57cec5SDimitry Andric     }
20870b57cec5SDimitry Andric     return;
20880b57cec5SDimitry Andric   }
20895ffd83dbSDimitry Andric 
20905ffd83dbSDimitry Andric   if (TapiUniversal *TU = dyn_cast<TapiUniversal>(&Bin)) {
20915ffd83dbSDimitry Andric     for (const TapiUniversal::ObjectForArch &I : TU->objects()) {
20925ffd83dbSDimitry Andric       StringRef ArchName = I.getArchFlagName();
20935ffd83dbSDimitry Andric       const bool ShowArch =
2094e8d8bef9SDimitry Andric           ArchFlags.empty() || llvm::is_contained(ArchFlags, ArchName);
20955ffd83dbSDimitry Andric       if (!ShowArch)
20965ffd83dbSDimitry Andric         continue;
20975ffd83dbSDimitry Andric       if (!AddInlinedInfo && !I.isTopLevelLib())
20985ffd83dbSDimitry Andric         continue;
20995ffd83dbSDimitry Andric       if (auto ObjOrErr = I.getAsObjectFile()) {
21005ffd83dbSDimitry Andric         outs() << "\n"
21015ffd83dbSDimitry Andric                << I.getInstallName() << " (for architecture " << ArchName << ")"
21025ffd83dbSDimitry Andric                << ":\n";
21035ffd83dbSDimitry Andric         dumpSymbolNamesFromObject(*ObjOrErr.get(), false, {}, ArchName);
21045ffd83dbSDimitry Andric       } else if (Error E =
21055ffd83dbSDimitry Andric                      isNotObjectErrorInvalidFileType(ObjOrErr.takeError())) {
21065ffd83dbSDimitry Andric         error(std::move(E), Filename, ArchName);
21075ffd83dbSDimitry Andric       }
21085ffd83dbSDimitry Andric     }
21095ffd83dbSDimitry Andric 
21105ffd83dbSDimitry Andric     return;
21115ffd83dbSDimitry Andric   }
21125ffd83dbSDimitry Andric 
21130b57cec5SDimitry Andric   if (SymbolicFile *O = dyn_cast<SymbolicFile>(&Bin)) {
21140b57cec5SDimitry Andric     if (!MachOPrintSizeWarning && PrintSize &&  isa<MachOObjectFile>(O)) {
21150b57cec5SDimitry Andric       WithColor::warning(errs(), ToolName)
21160b57cec5SDimitry Andric           << "sizes with --print-size for Mach-O files are always zero.\n";
21170b57cec5SDimitry Andric       MachOPrintSizeWarning = true;
21180b57cec5SDimitry Andric     }
21190b57cec5SDimitry Andric     if (!checkMachOAndArchFlags(O, Filename))
21200b57cec5SDimitry Andric       return;
21210b57cec5SDimitry Andric     dumpSymbolNamesFromObject(*O, true);
21220b57cec5SDimitry Andric   }
21230b57cec5SDimitry Andric }
21240b57cec5SDimitry Andric 
21250b57cec5SDimitry Andric int main(int argc, char **argv) {
21260b57cec5SDimitry Andric   InitLLVM X(argc, argv);
2127*fe6060f1SDimitry Andric   BumpPtrAllocator A;
2128*fe6060f1SDimitry Andric   StringSaver Saver(A);
2129*fe6060f1SDimitry Andric   NmOptTable Tbl;
2130*fe6060f1SDimitry Andric   ToolName = argv[0];
2131*fe6060f1SDimitry Andric   opt::InputArgList Args =
2132*fe6060f1SDimitry Andric       Tbl.parseArgs(argc, argv, OPT_UNKNOWN, Saver, [&](StringRef Msg) {
2133*fe6060f1SDimitry Andric         error(Msg);
2134*fe6060f1SDimitry Andric         exit(1);
2135*fe6060f1SDimitry Andric       });
2136*fe6060f1SDimitry Andric   if (Args.hasArg(OPT_help)) {
2137*fe6060f1SDimitry Andric     Tbl.printHelp(
2138*fe6060f1SDimitry Andric         outs(),
2139*fe6060f1SDimitry Andric         (Twine(ToolName) + " [options] <input object files>").str().c_str(),
2140*fe6060f1SDimitry Andric         "LLVM symbol table dumper");
2141*fe6060f1SDimitry Andric     // TODO Replace this with OptTable API once it adds extrahelp support.
2142*fe6060f1SDimitry Andric     outs() << "\nPass @FILE as argument to read options from FILE.\n";
2143*fe6060f1SDimitry Andric     return 0;
2144*fe6060f1SDimitry Andric   }
2145*fe6060f1SDimitry Andric   if (Args.hasArg(OPT_version)) {
2146*fe6060f1SDimitry Andric     // This needs to contain the word "GNU", libtool looks for that string.
2147*fe6060f1SDimitry Andric     outs() << "llvm-nm, compatible with GNU nm" << '\n';
2148*fe6060f1SDimitry Andric     cl::PrintVersionMessage();
2149*fe6060f1SDimitry Andric     return 0;
2150*fe6060f1SDimitry Andric   }
2151*fe6060f1SDimitry Andric 
2152*fe6060f1SDimitry Andric   DebugSyms = Args.hasArg(OPT_debug_syms);
2153*fe6060f1SDimitry Andric   DefinedOnly = Args.hasArg(OPT_defined_only);
2154*fe6060f1SDimitry Andric   Demangle = Args.hasFlag(OPT_demangle, OPT_no_demangle, false);
2155*fe6060f1SDimitry Andric   DynamicSyms = Args.hasArg(OPT_dynamic);
2156*fe6060f1SDimitry Andric   ExternalOnly = Args.hasArg(OPT_extern_only);
2157*fe6060f1SDimitry Andric   StringRef V = Args.getLastArgValue(OPT_format_EQ, "bsd");
2158*fe6060f1SDimitry Andric   if (V == "bsd")
2159*fe6060f1SDimitry Andric     OutputFormat = bsd;
2160*fe6060f1SDimitry Andric   else if (V == "posix")
2161*fe6060f1SDimitry Andric     OutputFormat = posix;
2162*fe6060f1SDimitry Andric   else if (V == "sysv")
2163*fe6060f1SDimitry Andric     OutputFormat = sysv;
2164*fe6060f1SDimitry Andric   else if (V == "darwin")
2165*fe6060f1SDimitry Andric     OutputFormat = darwin;
2166*fe6060f1SDimitry Andric   else if (V == "just-symbols")
2167*fe6060f1SDimitry Andric     OutputFormat = just_symbols;
2168*fe6060f1SDimitry Andric   else
2169*fe6060f1SDimitry Andric     error("--format value should be one of: bsd, posix, sysv, darwin, "
2170*fe6060f1SDimitry Andric           "just-symbols");
2171*fe6060f1SDimitry Andric   NoLLVMBitcode = Args.hasArg(OPT_no_llvm_bc);
2172*fe6060f1SDimitry Andric   NoSort = Args.hasArg(OPT_no_sort);
2173*fe6060f1SDimitry Andric   NoWeakSymbols = Args.hasArg(OPT_no_weak);
2174*fe6060f1SDimitry Andric   NumericSort = Args.hasArg(OPT_numeric_sort);
2175*fe6060f1SDimitry Andric   ArchiveMap = Args.hasArg(OPT_print_armap);
2176*fe6060f1SDimitry Andric   PrintFileName = Args.hasArg(OPT_print_file_name);
2177*fe6060f1SDimitry Andric   PrintSize = Args.hasArg(OPT_print_size);
2178*fe6060f1SDimitry Andric   ReverseSort = Args.hasArg(OPT_reverse_sort);
2179*fe6060f1SDimitry Andric   Quiet = Args.hasArg(OPT_quiet);
2180*fe6060f1SDimitry Andric   V = Args.getLastArgValue(OPT_radix_EQ, "x");
2181*fe6060f1SDimitry Andric   if (V == "o")
2182*fe6060f1SDimitry Andric     AddressRadix = Radix::o;
2183*fe6060f1SDimitry Andric   else if (V == "d")
2184*fe6060f1SDimitry Andric     AddressRadix = Radix::d;
2185*fe6060f1SDimitry Andric   else if (V == "x")
2186*fe6060f1SDimitry Andric     AddressRadix = Radix::x;
2187*fe6060f1SDimitry Andric   else
2188*fe6060f1SDimitry Andric     error("--radix value should be one of: 'o' (octal), 'd' (decimal), 'x' "
2189*fe6060f1SDimitry Andric           "(hexadecimal)");
2190*fe6060f1SDimitry Andric   SizeSort = Args.hasArg(OPT_size_sort);
2191*fe6060f1SDimitry Andric   SpecialSyms = Args.hasArg(OPT_special_syms);
2192*fe6060f1SDimitry Andric   UndefinedOnly = Args.hasArg(OPT_undefined_only);
2193*fe6060f1SDimitry Andric   WithoutAliases = Args.hasArg(OPT_without_aliases);
2194*fe6060f1SDimitry Andric 
2195*fe6060f1SDimitry Andric   // Mach-O specific options.
2196*fe6060f1SDimitry Andric   FormatMachOasHex = Args.hasArg(OPT_x);
2197*fe6060f1SDimitry Andric   AddDyldInfo = Args.hasArg(OPT_add_dyldinfo);
2198*fe6060f1SDimitry Andric   AddInlinedInfo = Args.hasArg(OPT_add_inlinedinfo);
2199*fe6060f1SDimitry Andric   DyldInfoOnly = Args.hasArg(OPT_dyldinfo_only);
2200*fe6060f1SDimitry Andric   NoDyldInfo = Args.hasArg(OPT_no_dyldinfo);
22010b57cec5SDimitry Andric 
22020b57cec5SDimitry Andric   // llvm-nm only reads binary files.
22030b57cec5SDimitry Andric   if (error(sys::ChangeStdinToBinary()))
22040b57cec5SDimitry Andric     return 1;
22050b57cec5SDimitry Andric 
22060b57cec5SDimitry Andric   // These calls are needed so that we can read bitcode correctly.
22070b57cec5SDimitry Andric   llvm::InitializeAllTargetInfos();
22080b57cec5SDimitry Andric   llvm::InitializeAllTargetMCs();
22090b57cec5SDimitry Andric   llvm::InitializeAllAsmParsers();
22100b57cec5SDimitry Andric 
22110b57cec5SDimitry Andric   // The relative order of these is important. If you pass --size-sort it should
22120b57cec5SDimitry Andric   // only print out the size. However, if you pass -S --size-sort, it should
22130b57cec5SDimitry Andric   // print out both the size and address.
22140b57cec5SDimitry Andric   if (SizeSort && !PrintSize)
22150b57cec5SDimitry Andric     PrintAddress = false;
22160b57cec5SDimitry Andric   if (OutputFormat == sysv || SizeSort)
22170b57cec5SDimitry Andric     PrintSize = true;
22180b57cec5SDimitry Andric 
2219*fe6060f1SDimitry Andric   for (const auto *A : Args.filtered(OPT_arch_EQ)) {
2220*fe6060f1SDimitry Andric     SmallVector<StringRef, 2> Values;
2221*fe6060f1SDimitry Andric     llvm::SplitString(A->getValue(), Values, ",");
2222*fe6060f1SDimitry Andric     for (StringRef V : Values) {
2223*fe6060f1SDimitry Andric       if (V == "all")
22240b57cec5SDimitry Andric         ArchAll = true;
2225*fe6060f1SDimitry Andric       else if (MachOObjectFile::isValidArch(V))
2226*fe6060f1SDimitry Andric         ArchFlags.push_back(V);
2227*fe6060f1SDimitry Andric       else
2228*fe6060f1SDimitry Andric         error("Unknown architecture named '" + V + "'",
22290b57cec5SDimitry Andric               "for the --arch option");
22300b57cec5SDimitry Andric     }
22310b57cec5SDimitry Andric   }
22320b57cec5SDimitry Andric 
2233*fe6060f1SDimitry Andric   // Mach-O takes -s to accept two arguments. We emulate this by iterating over
2234*fe6060f1SDimitry Andric   // both OPT_s and OPT_INPUT.
2235*fe6060f1SDimitry Andric   std::vector<std::string> InputFilenames;
2236*fe6060f1SDimitry Andric   int SegSectArgs = 0;
2237*fe6060f1SDimitry Andric   for (opt::Arg *A : Args.filtered(OPT_s, OPT_INPUT)) {
2238*fe6060f1SDimitry Andric     if (SegSectArgs > 0) {
2239*fe6060f1SDimitry Andric       --SegSectArgs;
2240*fe6060f1SDimitry Andric       SegSect.push_back(A->getValue());
2241*fe6060f1SDimitry Andric     } else if (A->getOption().matches(OPT_s)) {
2242*fe6060f1SDimitry Andric       SegSectArgs = 2;
2243*fe6060f1SDimitry Andric     } else {
2244*fe6060f1SDimitry Andric       InputFilenames.push_back(A->getValue());
2245*fe6060f1SDimitry Andric     }
2246*fe6060f1SDimitry Andric   }
22470b57cec5SDimitry Andric   if (!SegSect.empty() && SegSect.size() != 2)
22480b57cec5SDimitry Andric     error("bad number of arguments (must be two arguments)",
22490b57cec5SDimitry Andric           "for the -s option");
22500b57cec5SDimitry Andric 
2251*fe6060f1SDimitry Andric   if (InputFilenames.empty())
2252*fe6060f1SDimitry Andric     InputFilenames.push_back("a.out");
2253*fe6060f1SDimitry Andric   if (InputFilenames.size() > 1)
2254*fe6060f1SDimitry Andric     MultipleFiles = true;
2255*fe6060f1SDimitry Andric 
22560b57cec5SDimitry Andric   if (NoDyldInfo && (AddDyldInfo || DyldInfoOnly))
22570b57cec5SDimitry Andric     error("--no-dyldinfo can't be used with --add-dyldinfo or --dyldinfo-only");
22580b57cec5SDimitry Andric 
22590b57cec5SDimitry Andric   llvm::for_each(InputFilenames, dumpSymbolNamesFromFile);
22600b57cec5SDimitry Andric 
22610b57cec5SDimitry Andric   if (HadError)
22620b57cec5SDimitry Andric     return 1;
22630b57cec5SDimitry Andric }
2264