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