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" 310b57cec5SDimitry Andric #include "llvm/Object/Wasm.h" 320b57cec5SDimitry Andric #include "llvm/Support/CommandLine.h" 330b57cec5SDimitry Andric #include "llvm/Support/FileSystem.h" 340b57cec5SDimitry Andric #include "llvm/Support/Format.h" 350b57cec5SDimitry Andric #include "llvm/Support/InitLLVM.h" 360b57cec5SDimitry Andric #include "llvm/Support/MemoryBuffer.h" 370b57cec5SDimitry Andric #include "llvm/Support/Program.h" 380b57cec5SDimitry Andric #include "llvm/Support/Signals.h" 390b57cec5SDimitry Andric #include "llvm/Support/TargetSelect.h" 400b57cec5SDimitry Andric #include "llvm/Support/WithColor.h" 410b57cec5SDimitry Andric #include "llvm/Support/raw_ostream.h" 420b57cec5SDimitry Andric #include <vector> 430b57cec5SDimitry Andric 440b57cec5SDimitry Andric using namespace llvm; 450b57cec5SDimitry Andric using namespace object; 460b57cec5SDimitry Andric 470b57cec5SDimitry Andric namespace { 480b57cec5SDimitry Andric enum OutputFormatTy { bsd, sysv, posix, darwin }; 490b57cec5SDimitry Andric 500b57cec5SDimitry Andric cl::OptionCategory NMCat("llvm-nm Options"); 510b57cec5SDimitry Andric 520b57cec5SDimitry Andric cl::opt<OutputFormatTy> OutputFormat( 530b57cec5SDimitry Andric "format", cl::desc("Specify output format"), 540b57cec5SDimitry Andric cl::values(clEnumVal(bsd, "BSD format"), clEnumVal(sysv, "System V format"), 550b57cec5SDimitry Andric clEnumVal(posix, "POSIX.2 format"), 560b57cec5SDimitry Andric clEnumVal(darwin, "Darwin -m format")), 570b57cec5SDimitry Andric cl::init(bsd), cl::cat(NMCat)); 580b57cec5SDimitry Andric cl::alias OutputFormat2("f", cl::desc("Alias for --format"), 590b57cec5SDimitry Andric cl::aliasopt(OutputFormat)); 600b57cec5SDimitry Andric 610b57cec5SDimitry Andric cl::list<std::string> InputFilenames(cl::Positional, cl::desc("<input files>"), 620b57cec5SDimitry Andric cl::ZeroOrMore); 630b57cec5SDimitry Andric 640b57cec5SDimitry Andric cl::opt<bool> UndefinedOnly("undefined-only", 650b57cec5SDimitry Andric cl::desc("Show only undefined symbols"), 660b57cec5SDimitry Andric cl::cat(NMCat)); 670b57cec5SDimitry Andric cl::alias UndefinedOnly2("u", cl::desc("Alias for --undefined-only"), 680b57cec5SDimitry Andric cl::aliasopt(UndefinedOnly), cl::Grouping); 690b57cec5SDimitry Andric 700b57cec5SDimitry Andric cl::opt<bool> DynamicSyms("dynamic", 710b57cec5SDimitry Andric cl::desc("Display the dynamic symbols instead " 720b57cec5SDimitry Andric "of normal symbols."), 730b57cec5SDimitry Andric cl::cat(NMCat)); 740b57cec5SDimitry Andric cl::alias DynamicSyms2("D", cl::desc("Alias for --dynamic"), 750b57cec5SDimitry Andric cl::aliasopt(DynamicSyms), cl::Grouping); 760b57cec5SDimitry Andric 770b57cec5SDimitry Andric cl::opt<bool> DefinedOnly("defined-only", cl::desc("Show only defined symbols"), 780b57cec5SDimitry Andric cl::cat(NMCat)); 790b57cec5SDimitry Andric cl::alias DefinedOnly2("U", cl::desc("Alias for --defined-only"), 800b57cec5SDimitry Andric cl::aliasopt(DefinedOnly), cl::Grouping); 810b57cec5SDimitry Andric 820b57cec5SDimitry Andric cl::opt<bool> ExternalOnly("extern-only", 830b57cec5SDimitry Andric cl::desc("Show only external symbols"), 840b57cec5SDimitry Andric cl::ZeroOrMore, cl::cat(NMCat)); 850b57cec5SDimitry Andric cl::alias ExternalOnly2("g", cl::desc("Alias for --extern-only"), 860b57cec5SDimitry Andric cl::aliasopt(ExternalOnly), cl::Grouping, 870b57cec5SDimitry Andric cl::ZeroOrMore); 880b57cec5SDimitry Andric 890b57cec5SDimitry Andric cl::opt<bool> NoWeakSymbols("no-weak", cl::desc("Show only non-weak symbols"), 900b57cec5SDimitry Andric cl::cat(NMCat)); 910b57cec5SDimitry Andric cl::alias NoWeakSymbols2("W", cl::desc("Alias for --no-weak"), 920b57cec5SDimitry Andric cl::aliasopt(NoWeakSymbols), cl::Grouping); 930b57cec5SDimitry Andric 940b57cec5SDimitry Andric cl::opt<bool> BSDFormat("B", cl::desc("Alias for --format=bsd"), cl::Grouping, 950b57cec5SDimitry Andric cl::cat(NMCat)); 960b57cec5SDimitry Andric cl::opt<bool> POSIXFormat("P", cl::desc("Alias for --format=posix"), 970b57cec5SDimitry Andric cl::Grouping, cl::cat(NMCat)); 980b57cec5SDimitry Andric cl::alias Portability("portability", cl::desc("Alias for --format=posix"), 990b57cec5SDimitry Andric cl::aliasopt(POSIXFormat), cl::NotHidden); 1000b57cec5SDimitry Andric cl::opt<bool> DarwinFormat("m", cl::desc("Alias for --format=darwin"), 1010b57cec5SDimitry Andric cl::Grouping, cl::cat(NMCat)); 1020b57cec5SDimitry Andric 1030b57cec5SDimitry Andric static cl::list<std::string> 1040b57cec5SDimitry Andric ArchFlags("arch", cl::desc("architecture(s) from a Mach-O file to dump"), 1050b57cec5SDimitry Andric cl::ZeroOrMore, cl::cat(NMCat)); 1060b57cec5SDimitry Andric bool ArchAll = false; 1070b57cec5SDimitry Andric 1080b57cec5SDimitry Andric cl::opt<bool> PrintFileName( 1090b57cec5SDimitry Andric "print-file-name", 1100b57cec5SDimitry Andric cl::desc("Precede each symbol with the object file it came from"), 1110b57cec5SDimitry Andric cl::cat(NMCat)); 1120b57cec5SDimitry Andric 1130b57cec5SDimitry Andric cl::alias PrintFileNameA("A", cl::desc("Alias for --print-file-name"), 1140b57cec5SDimitry Andric cl::aliasopt(PrintFileName), cl::Grouping); 1150b57cec5SDimitry Andric cl::alias PrintFileNameo("o", cl::desc("Alias for --print-file-name"), 1160b57cec5SDimitry Andric cl::aliasopt(PrintFileName), cl::Grouping); 1170b57cec5SDimitry Andric 1180b57cec5SDimitry Andric cl::opt<bool> DebugSyms("debug-syms", 1190b57cec5SDimitry Andric cl::desc("Show all symbols, even debugger only"), 1200b57cec5SDimitry Andric cl::cat(NMCat)); 1210b57cec5SDimitry Andric cl::alias DebugSymsa("a", cl::desc("Alias for --debug-syms"), 1220b57cec5SDimitry Andric cl::aliasopt(DebugSyms), cl::Grouping); 1230b57cec5SDimitry Andric 1240b57cec5SDimitry Andric cl::opt<bool> NumericSort("numeric-sort", cl::desc("Sort symbols by address"), 1250b57cec5SDimitry Andric cl::cat(NMCat)); 1260b57cec5SDimitry Andric cl::alias NumericSortn("n", cl::desc("Alias for --numeric-sort"), 1270b57cec5SDimitry Andric cl::aliasopt(NumericSort), cl::Grouping); 1280b57cec5SDimitry Andric cl::alias NumericSortv("v", cl::desc("Alias for --numeric-sort"), 1290b57cec5SDimitry Andric cl::aliasopt(NumericSort), cl::Grouping); 1300b57cec5SDimitry Andric 1310b57cec5SDimitry Andric cl::opt<bool> NoSort("no-sort", cl::desc("Show symbols in order encountered"), 1320b57cec5SDimitry Andric cl::cat(NMCat)); 1330b57cec5SDimitry Andric cl::alias NoSortp("p", cl::desc("Alias for --no-sort"), cl::aliasopt(NoSort), 1340b57cec5SDimitry Andric cl::Grouping); 1350b57cec5SDimitry Andric 1360b57cec5SDimitry Andric cl::opt<bool> Demangle("demangle", cl::ZeroOrMore, 1370b57cec5SDimitry Andric cl::desc("Demangle C++ symbol names"), cl::cat(NMCat)); 1380b57cec5SDimitry Andric cl::alias DemangleC("C", cl::desc("Alias for --demangle"), 1390b57cec5SDimitry Andric cl::aliasopt(Demangle), cl::Grouping); 1400b57cec5SDimitry Andric cl::opt<bool> NoDemangle("no-demangle", cl::init(false), cl::ZeroOrMore, 1410b57cec5SDimitry Andric cl::desc("Don't demangle symbol names"), 1420b57cec5SDimitry Andric cl::cat(NMCat)); 1430b57cec5SDimitry Andric 1440b57cec5SDimitry Andric cl::opt<bool> ReverseSort("reverse-sort", cl::desc("Sort in reverse order"), 1450b57cec5SDimitry Andric cl::cat(NMCat)); 1460b57cec5SDimitry Andric cl::alias ReverseSortr("r", cl::desc("Alias for --reverse-sort"), 1470b57cec5SDimitry Andric cl::aliasopt(ReverseSort), cl::Grouping); 1480b57cec5SDimitry Andric 1490b57cec5SDimitry Andric cl::opt<bool> PrintSize("print-size", 1500b57cec5SDimitry Andric cl::desc("Show symbol size as well as address"), 1510b57cec5SDimitry Andric cl::cat(NMCat)); 1520b57cec5SDimitry Andric cl::alias PrintSizeS("S", cl::desc("Alias for --print-size"), 1530b57cec5SDimitry Andric cl::aliasopt(PrintSize), cl::Grouping); 1540b57cec5SDimitry Andric bool MachOPrintSizeWarning = false; 1550b57cec5SDimitry Andric 1560b57cec5SDimitry Andric cl::opt<bool> SizeSort("size-sort", cl::desc("Sort symbols by size"), 1570b57cec5SDimitry Andric cl::cat(NMCat)); 1580b57cec5SDimitry Andric 1590b57cec5SDimitry Andric cl::opt<bool> WithoutAliases("without-aliases", cl::Hidden, 1600b57cec5SDimitry Andric cl::desc("Exclude aliases from output"), 1610b57cec5SDimitry Andric cl::cat(NMCat)); 1620b57cec5SDimitry Andric 1630b57cec5SDimitry Andric cl::opt<bool> ArchiveMap("print-armap", cl::desc("Print the archive map"), 1640b57cec5SDimitry Andric cl::cat(NMCat)); 1650b57cec5SDimitry Andric cl::alias ArchiveMaps("M", cl::desc("Alias for --print-armap"), 1660b57cec5SDimitry Andric cl::aliasopt(ArchiveMap), cl::Grouping); 1670b57cec5SDimitry Andric 1680b57cec5SDimitry Andric enum Radix { d, o, x }; 1690b57cec5SDimitry Andric cl::opt<Radix> 1700b57cec5SDimitry Andric AddressRadix("radix", cl::desc("Radix (o/d/x) for printing symbol Values"), 1710b57cec5SDimitry Andric cl::values(clEnumVal(d, "decimal"), clEnumVal(o, "octal"), 1720b57cec5SDimitry Andric clEnumVal(x, "hexadecimal")), 1730b57cec5SDimitry Andric cl::init(x), cl::cat(NMCat)); 1740b57cec5SDimitry Andric cl::alias RadixAlias("t", cl::desc("Alias for --radix"), 1750b57cec5SDimitry Andric cl::aliasopt(AddressRadix)); 1760b57cec5SDimitry Andric 1770b57cec5SDimitry Andric cl::opt<bool> JustSymbolName("just-symbol-name", 1780b57cec5SDimitry Andric cl::desc("Print just the symbol's name"), 1790b57cec5SDimitry Andric cl::cat(NMCat)); 1800b57cec5SDimitry Andric cl::alias JustSymbolNames("j", cl::desc("Alias for --just-symbol-name"), 1810b57cec5SDimitry Andric cl::aliasopt(JustSymbolName), cl::Grouping); 1820b57cec5SDimitry Andric 1830b57cec5SDimitry Andric cl::opt<bool> SpecialSyms("special-syms", 1840b57cec5SDimitry Andric cl::desc("No-op. Used for GNU compatibility only")); 1850b57cec5SDimitry Andric 1860b57cec5SDimitry Andric cl::list<std::string> SegSect("s", cl::multi_val(2), cl::ZeroOrMore, 1870b57cec5SDimitry Andric cl::value_desc("segment section"), cl::Hidden, 1880b57cec5SDimitry Andric cl::desc("Dump only symbols from this segment " 1890b57cec5SDimitry Andric "and section name, Mach-O only"), 1900b57cec5SDimitry Andric cl::cat(NMCat)); 1910b57cec5SDimitry Andric 1920b57cec5SDimitry Andric cl::opt<bool> FormatMachOasHex("x", 1930b57cec5SDimitry Andric cl::desc("Print symbol entry in hex, " 1940b57cec5SDimitry Andric "Mach-O only"), 1950b57cec5SDimitry Andric cl::Grouping, cl::cat(NMCat)); 1960b57cec5SDimitry Andric cl::opt<bool> AddDyldInfo("add-dyldinfo", 1970b57cec5SDimitry Andric cl::desc("Add symbols from the dyldinfo not already " 1980b57cec5SDimitry Andric "in the symbol table, Mach-O only"), 1990b57cec5SDimitry Andric cl::cat(NMCat)); 2000b57cec5SDimitry Andric cl::opt<bool> NoDyldInfo("no-dyldinfo", 2010b57cec5SDimitry Andric cl::desc("Don't add any symbols from the dyldinfo, " 2020b57cec5SDimitry Andric "Mach-O only"), 2030b57cec5SDimitry Andric cl::cat(NMCat)); 2040b57cec5SDimitry Andric cl::opt<bool> DyldInfoOnly("dyldinfo-only", 2050b57cec5SDimitry Andric cl::desc("Show only symbols from the dyldinfo, " 2060b57cec5SDimitry Andric "Mach-O only"), 2070b57cec5SDimitry Andric cl::cat(NMCat)); 2080b57cec5SDimitry Andric 2090b57cec5SDimitry Andric cl::opt<bool> NoLLVMBitcode("no-llvm-bc", 2100b57cec5SDimitry Andric cl::desc("Disable LLVM bitcode reader"), 2110b57cec5SDimitry Andric cl::cat(NMCat)); 2120b57cec5SDimitry Andric 2130b57cec5SDimitry Andric cl::extrahelp HelpResponse("\nPass @FILE as argument to read options from FILE.\n"); 2140b57cec5SDimitry Andric 2150b57cec5SDimitry Andric bool PrintAddress = true; 2160b57cec5SDimitry Andric 2170b57cec5SDimitry Andric bool MultipleFiles = false; 2180b57cec5SDimitry Andric 2190b57cec5SDimitry Andric bool HadError = false; 2200b57cec5SDimitry Andric 2210b57cec5SDimitry Andric std::string ToolName; 2220b57cec5SDimitry Andric } // anonymous namespace 2230b57cec5SDimitry Andric 2240b57cec5SDimitry Andric static void error(Twine Message, Twine Path = Twine()) { 2250b57cec5SDimitry Andric HadError = true; 2260b57cec5SDimitry Andric WithColor::error(errs(), ToolName) << Path << ": " << Message << ".\n"; 2270b57cec5SDimitry Andric } 2280b57cec5SDimitry Andric 2290b57cec5SDimitry Andric static bool error(std::error_code EC, Twine Path = Twine()) { 2300b57cec5SDimitry Andric if (EC) { 2310b57cec5SDimitry Andric error(EC.message(), Path); 2320b57cec5SDimitry Andric return true; 2330b57cec5SDimitry Andric } 2340b57cec5SDimitry Andric return false; 2350b57cec5SDimitry Andric } 2360b57cec5SDimitry Andric 2370b57cec5SDimitry Andric // This version of error() prints the archive name and member name, for example: 2380b57cec5SDimitry Andric // "libx.a(foo.o)" after the ToolName before the error message. It sets 2390b57cec5SDimitry Andric // HadError but returns allowing the code to move on to other archive members. 2400b57cec5SDimitry Andric static void error(llvm::Error E, StringRef FileName, const Archive::Child &C, 2410b57cec5SDimitry Andric StringRef ArchitectureName = StringRef()) { 2420b57cec5SDimitry Andric HadError = true; 2430b57cec5SDimitry Andric WithColor::error(errs(), ToolName) << FileName; 2440b57cec5SDimitry Andric 2450b57cec5SDimitry Andric Expected<StringRef> NameOrErr = C.getName(); 2460b57cec5SDimitry Andric // TODO: if we have a error getting the name then it would be nice to print 2470b57cec5SDimitry Andric // the index of which archive member this is and or its offset in the 2480b57cec5SDimitry Andric // archive instead of "???" as the name. 2490b57cec5SDimitry Andric if (!NameOrErr) { 2500b57cec5SDimitry Andric consumeError(NameOrErr.takeError()); 2510b57cec5SDimitry Andric errs() << "(" << "???" << ")"; 2520b57cec5SDimitry Andric } else 2530b57cec5SDimitry Andric errs() << "(" << NameOrErr.get() << ")"; 2540b57cec5SDimitry Andric 2550b57cec5SDimitry Andric if (!ArchitectureName.empty()) 2560b57cec5SDimitry Andric errs() << " (for architecture " << ArchitectureName << ") "; 2570b57cec5SDimitry Andric 2580b57cec5SDimitry Andric std::string Buf; 2590b57cec5SDimitry Andric raw_string_ostream OS(Buf); 2600b57cec5SDimitry Andric logAllUnhandledErrors(std::move(E), OS); 2610b57cec5SDimitry Andric OS.flush(); 2620b57cec5SDimitry Andric errs() << " " << Buf << "\n"; 2630b57cec5SDimitry Andric } 2640b57cec5SDimitry Andric 2650b57cec5SDimitry Andric // This version of error() prints the file name and which architecture slice it 2660b57cec5SDimitry Andric // is from, for example: "foo.o (for architecture i386)" after the ToolName 2670b57cec5SDimitry Andric // before the error message. It sets HadError but returns allowing the code to 2680b57cec5SDimitry Andric // move on to other architecture slices. 2690b57cec5SDimitry Andric static void error(llvm::Error E, StringRef FileName, 2700b57cec5SDimitry Andric StringRef ArchitectureName = StringRef()) { 2710b57cec5SDimitry Andric HadError = true; 2720b57cec5SDimitry Andric WithColor::error(errs(), ToolName) << FileName; 2730b57cec5SDimitry Andric 2740b57cec5SDimitry Andric if (!ArchitectureName.empty()) 2750b57cec5SDimitry Andric errs() << " (for architecture " << ArchitectureName << ") "; 2760b57cec5SDimitry Andric 2770b57cec5SDimitry Andric std::string Buf; 2780b57cec5SDimitry Andric raw_string_ostream OS(Buf); 2790b57cec5SDimitry Andric logAllUnhandledErrors(std::move(E), OS); 2800b57cec5SDimitry Andric OS.flush(); 2810b57cec5SDimitry Andric errs() << " " << Buf << "\n"; 2820b57cec5SDimitry Andric } 2830b57cec5SDimitry Andric 2840b57cec5SDimitry Andric namespace { 2850b57cec5SDimitry Andric struct NMSymbol { 2860b57cec5SDimitry Andric uint64_t Address; 2870b57cec5SDimitry Andric uint64_t Size; 2880b57cec5SDimitry Andric char TypeChar; 2890b57cec5SDimitry Andric StringRef Name; 2900b57cec5SDimitry Andric StringRef SectionName; 2910b57cec5SDimitry Andric StringRef TypeName; 2920b57cec5SDimitry Andric BasicSymbolRef Sym; 2930b57cec5SDimitry Andric // The Sym field above points to the native symbol in the object file, 2940b57cec5SDimitry Andric // for Mach-O when we are creating symbols from the dyld info the above 2950b57cec5SDimitry Andric // pointer is null as there is no native symbol. In these cases the fields 2960b57cec5SDimitry Andric // below are filled in to represent what would have been a Mach-O nlist 2970b57cec5SDimitry Andric // native symbol. 2980b57cec5SDimitry Andric uint32_t SymFlags; 2990b57cec5SDimitry Andric SectionRef Section; 3000b57cec5SDimitry Andric uint8_t NType; 3010b57cec5SDimitry Andric uint8_t NSect; 3020b57cec5SDimitry Andric uint16_t NDesc; 3030b57cec5SDimitry Andric StringRef IndirectName; 3040b57cec5SDimitry Andric }; 3050b57cec5SDimitry Andric } // anonymous namespace 3060b57cec5SDimitry Andric 3070b57cec5SDimitry Andric static bool compareSymbolAddress(const NMSymbol &A, const NMSymbol &B) { 3080b57cec5SDimitry Andric bool ADefined; 3090b57cec5SDimitry Andric if (A.Sym.getRawDataRefImpl().p) 3100b57cec5SDimitry Andric ADefined = !(A.Sym.getFlags() & SymbolRef::SF_Undefined); 3110b57cec5SDimitry Andric else 3120b57cec5SDimitry Andric ADefined = A.TypeChar != 'U'; 3130b57cec5SDimitry Andric bool BDefined; 3140b57cec5SDimitry Andric if (B.Sym.getRawDataRefImpl().p) 3150b57cec5SDimitry Andric BDefined = !(B.Sym.getFlags() & SymbolRef::SF_Undefined); 3160b57cec5SDimitry Andric else 3170b57cec5SDimitry Andric BDefined = B.TypeChar != 'U'; 3180b57cec5SDimitry Andric return std::make_tuple(ADefined, A.Address, A.Name, A.Size) < 3190b57cec5SDimitry Andric std::make_tuple(BDefined, B.Address, B.Name, B.Size); 3200b57cec5SDimitry Andric } 3210b57cec5SDimitry Andric 3220b57cec5SDimitry Andric static bool compareSymbolSize(const NMSymbol &A, const NMSymbol &B) { 3230b57cec5SDimitry Andric return std::make_tuple(A.Size, A.Name, A.Address) < 3240b57cec5SDimitry Andric std::make_tuple(B.Size, B.Name, B.Address); 3250b57cec5SDimitry Andric } 3260b57cec5SDimitry Andric 3270b57cec5SDimitry Andric static bool compareSymbolName(const NMSymbol &A, const NMSymbol &B) { 3280b57cec5SDimitry Andric return std::make_tuple(A.Name, A.Size, A.Address) < 3290b57cec5SDimitry Andric std::make_tuple(B.Name, B.Size, B.Address); 3300b57cec5SDimitry Andric } 3310b57cec5SDimitry Andric 3320b57cec5SDimitry Andric static char isSymbolList64Bit(SymbolicFile &Obj) { 3330b57cec5SDimitry Andric if (auto *IRObj = dyn_cast<IRObjectFile>(&Obj)) 3340b57cec5SDimitry Andric return Triple(IRObj->getTargetTriple()).isArch64Bit(); 3350b57cec5SDimitry Andric if (isa<COFFObjectFile>(Obj) || isa<COFFImportFile>(Obj)) 3360b57cec5SDimitry Andric return false; 3370b57cec5SDimitry Andric if (isa<WasmObjectFile>(Obj)) 3380b57cec5SDimitry Andric return false; 3390b57cec5SDimitry Andric if (MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(&Obj)) 3400b57cec5SDimitry Andric return MachO->is64Bit(); 3410b57cec5SDimitry Andric return cast<ELFObjectFileBase>(Obj).getBytesInAddress() == 8; 3420b57cec5SDimitry Andric } 3430b57cec5SDimitry Andric 3440b57cec5SDimitry Andric static StringRef CurrentFilename; 3450b57cec5SDimitry Andric static std::vector<NMSymbol> SymbolList; 3460b57cec5SDimitry Andric 3470b57cec5SDimitry Andric static char getSymbolNMTypeChar(IRObjectFile &Obj, basic_symbol_iterator I); 3480b57cec5SDimitry Andric 3490b57cec5SDimitry Andric // darwinPrintSymbol() is used to print a symbol from a Mach-O file when the 3500b57cec5SDimitry Andric // the OutputFormat is darwin or we are printing Mach-O symbols in hex. For 3510b57cec5SDimitry Andric // the darwin format it produces the same output as darwin's nm(1) -m output 3520b57cec5SDimitry Andric // and when printing Mach-O symbols in hex it produces the same output as 3530b57cec5SDimitry Andric // darwin's nm(1) -x format. 3540b57cec5SDimitry Andric static void darwinPrintSymbol(SymbolicFile &Obj, const NMSymbol &S, 3550b57cec5SDimitry Andric char *SymbolAddrStr, const char *printBlanks, 3560b57cec5SDimitry Andric const char *printDashes, 3570b57cec5SDimitry Andric const char *printFormat) { 3580b57cec5SDimitry Andric MachO::mach_header H; 3590b57cec5SDimitry Andric MachO::mach_header_64 H_64; 3600b57cec5SDimitry Andric uint32_t Filetype = MachO::MH_OBJECT; 3610b57cec5SDimitry Andric uint32_t Flags = 0; 3620b57cec5SDimitry Andric uint8_t NType = 0; 3630b57cec5SDimitry Andric uint8_t NSect = 0; 3640b57cec5SDimitry Andric uint16_t NDesc = 0; 3650b57cec5SDimitry Andric uint32_t NStrx = 0; 3660b57cec5SDimitry Andric uint64_t NValue = 0; 3670b57cec5SDimitry Andric MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(&Obj); 3680b57cec5SDimitry Andric if (Obj.isIR()) { 3690b57cec5SDimitry Andric uint32_t SymFlags = S.Sym.getFlags(); 3700b57cec5SDimitry Andric if (SymFlags & SymbolRef::SF_Global) 3710b57cec5SDimitry Andric NType |= MachO::N_EXT; 3720b57cec5SDimitry Andric if (SymFlags & SymbolRef::SF_Hidden) 3730b57cec5SDimitry Andric NType |= MachO::N_PEXT; 3740b57cec5SDimitry Andric if (SymFlags & SymbolRef::SF_Undefined) 3750b57cec5SDimitry Andric NType |= MachO::N_EXT | MachO::N_UNDF; 3760b57cec5SDimitry Andric else { 3770b57cec5SDimitry Andric // Here we have a symbol definition. So to fake out a section name we 3780b57cec5SDimitry Andric // use 1, 2 and 3 for section numbers. See below where they are used to 3790b57cec5SDimitry Andric // print out fake section names. 3800b57cec5SDimitry Andric NType |= MachO::N_SECT; 3810b57cec5SDimitry Andric if (SymFlags & SymbolRef::SF_Const) 3820b57cec5SDimitry Andric NSect = 3; 3830b57cec5SDimitry Andric else if (SymFlags & SymbolRef::SF_Executable) 3840b57cec5SDimitry Andric NSect = 1; 3850b57cec5SDimitry Andric else 3860b57cec5SDimitry Andric NSect = 2; 3870b57cec5SDimitry Andric } 3880b57cec5SDimitry Andric if (SymFlags & SymbolRef::SF_Weak) 3890b57cec5SDimitry Andric NDesc |= MachO::N_WEAK_DEF; 3900b57cec5SDimitry Andric } else { 3910b57cec5SDimitry Andric DataRefImpl SymDRI = S.Sym.getRawDataRefImpl(); 3920b57cec5SDimitry Andric if (MachO->is64Bit()) { 3930b57cec5SDimitry Andric H_64 = MachO->MachOObjectFile::getHeader64(); 3940b57cec5SDimitry Andric Filetype = H_64.filetype; 3950b57cec5SDimitry Andric Flags = H_64.flags; 3960b57cec5SDimitry Andric if (SymDRI.p){ 3970b57cec5SDimitry Andric MachO::nlist_64 STE_64 = MachO->getSymbol64TableEntry(SymDRI); 3980b57cec5SDimitry Andric NType = STE_64.n_type; 3990b57cec5SDimitry Andric NSect = STE_64.n_sect; 4000b57cec5SDimitry Andric NDesc = STE_64.n_desc; 4010b57cec5SDimitry Andric NStrx = STE_64.n_strx; 4020b57cec5SDimitry Andric NValue = STE_64.n_value; 4030b57cec5SDimitry Andric } else { 4040b57cec5SDimitry Andric NType = S.NType; 4050b57cec5SDimitry Andric NSect = S.NSect; 4060b57cec5SDimitry Andric NDesc = S.NDesc; 4070b57cec5SDimitry Andric NStrx = 0; 4080b57cec5SDimitry Andric NValue = S.Address; 4090b57cec5SDimitry Andric } 4100b57cec5SDimitry Andric } else { 4110b57cec5SDimitry Andric H = MachO->MachOObjectFile::getHeader(); 4120b57cec5SDimitry Andric Filetype = H.filetype; 4130b57cec5SDimitry Andric Flags = H.flags; 4140b57cec5SDimitry Andric if (SymDRI.p){ 4150b57cec5SDimitry Andric MachO::nlist STE = MachO->getSymbolTableEntry(SymDRI); 4160b57cec5SDimitry Andric NType = STE.n_type; 4170b57cec5SDimitry Andric NSect = STE.n_sect; 4180b57cec5SDimitry Andric NDesc = STE.n_desc; 4190b57cec5SDimitry Andric NStrx = STE.n_strx; 4200b57cec5SDimitry Andric NValue = STE.n_value; 4210b57cec5SDimitry Andric } else { 4220b57cec5SDimitry Andric NType = S.NType; 4230b57cec5SDimitry Andric NSect = S.NSect; 4240b57cec5SDimitry Andric NDesc = S.NDesc; 4250b57cec5SDimitry Andric NStrx = 0; 4260b57cec5SDimitry Andric NValue = S.Address; 4270b57cec5SDimitry Andric } 4280b57cec5SDimitry Andric } 4290b57cec5SDimitry Andric } 4300b57cec5SDimitry Andric 4310b57cec5SDimitry Andric // If we are printing Mach-O symbols in hex do that and return. 4320b57cec5SDimitry Andric if (FormatMachOasHex) { 4330b57cec5SDimitry Andric outs() << format(printFormat, NValue) << ' ' 4340b57cec5SDimitry Andric << format("%02x %02x %04x %08x", NType, NSect, NDesc, NStrx) << ' ' 4350b57cec5SDimitry Andric << S.Name; 4360b57cec5SDimitry Andric if ((NType & MachO::N_TYPE) == MachO::N_INDR) { 4370b57cec5SDimitry Andric outs() << " (indirect for "; 4380b57cec5SDimitry Andric outs() << format(printFormat, NValue) << ' '; 4390b57cec5SDimitry Andric StringRef IndirectName; 4400b57cec5SDimitry Andric if (S.Sym.getRawDataRefImpl().p) { 4410b57cec5SDimitry Andric if (MachO->getIndirectName(S.Sym.getRawDataRefImpl(), IndirectName)) 4420b57cec5SDimitry Andric outs() << "?)"; 4430b57cec5SDimitry Andric else 4440b57cec5SDimitry Andric outs() << IndirectName << ")"; 4450b57cec5SDimitry Andric } else 4460b57cec5SDimitry Andric outs() << S.IndirectName << ")"; 4470b57cec5SDimitry Andric } 4480b57cec5SDimitry Andric outs() << "\n"; 4490b57cec5SDimitry Andric return; 4500b57cec5SDimitry Andric } 4510b57cec5SDimitry Andric 4520b57cec5SDimitry Andric if (PrintAddress) { 4530b57cec5SDimitry Andric if ((NType & MachO::N_TYPE) == MachO::N_INDR) 4540b57cec5SDimitry Andric strcpy(SymbolAddrStr, printBlanks); 4550b57cec5SDimitry Andric if (Obj.isIR() && (NType & MachO::N_TYPE) == MachO::N_TYPE) 4560b57cec5SDimitry Andric strcpy(SymbolAddrStr, printDashes); 4570b57cec5SDimitry Andric outs() << SymbolAddrStr << ' '; 4580b57cec5SDimitry Andric } 4590b57cec5SDimitry Andric 4600b57cec5SDimitry Andric switch (NType & MachO::N_TYPE) { 4610b57cec5SDimitry Andric case MachO::N_UNDF: 4620b57cec5SDimitry Andric if (NValue != 0) { 4630b57cec5SDimitry Andric outs() << "(common) "; 4640b57cec5SDimitry Andric if (MachO::GET_COMM_ALIGN(NDesc) != 0) 4650b57cec5SDimitry Andric outs() << "(alignment 2^" << (int)MachO::GET_COMM_ALIGN(NDesc) << ") "; 4660b57cec5SDimitry Andric } else { 4670b57cec5SDimitry Andric if ((NType & MachO::N_TYPE) == MachO::N_PBUD) 4680b57cec5SDimitry Andric outs() << "(prebound "; 4690b57cec5SDimitry Andric else 4700b57cec5SDimitry Andric outs() << "("; 4710b57cec5SDimitry Andric if ((NDesc & MachO::REFERENCE_TYPE) == 4720b57cec5SDimitry Andric MachO::REFERENCE_FLAG_UNDEFINED_LAZY) 4730b57cec5SDimitry Andric outs() << "undefined [lazy bound]) "; 4740b57cec5SDimitry Andric else if ((NDesc & MachO::REFERENCE_TYPE) == 4750b57cec5SDimitry Andric MachO::REFERENCE_FLAG_PRIVATE_UNDEFINED_LAZY) 4760b57cec5SDimitry Andric outs() << "undefined [private lazy bound]) "; 4770b57cec5SDimitry Andric else if ((NDesc & MachO::REFERENCE_TYPE) == 4780b57cec5SDimitry Andric MachO::REFERENCE_FLAG_PRIVATE_UNDEFINED_NON_LAZY) 4790b57cec5SDimitry Andric outs() << "undefined [private]) "; 4800b57cec5SDimitry Andric else 4810b57cec5SDimitry Andric outs() << "undefined) "; 4820b57cec5SDimitry Andric } 4830b57cec5SDimitry Andric break; 4840b57cec5SDimitry Andric case MachO::N_ABS: 4850b57cec5SDimitry Andric outs() << "(absolute) "; 4860b57cec5SDimitry Andric break; 4870b57cec5SDimitry Andric case MachO::N_INDR: 4880b57cec5SDimitry Andric outs() << "(indirect) "; 4890b57cec5SDimitry Andric break; 4900b57cec5SDimitry Andric case MachO::N_SECT: { 4910b57cec5SDimitry Andric if (Obj.isIR()) { 4920b57cec5SDimitry Andric // For llvm bitcode files print out a fake section name using the values 4930b57cec5SDimitry Andric // use 1, 2 and 3 for section numbers as set above. 4940b57cec5SDimitry Andric if (NSect == 1) 4950b57cec5SDimitry Andric outs() << "(LTO,CODE) "; 4960b57cec5SDimitry Andric else if (NSect == 2) 4970b57cec5SDimitry Andric outs() << "(LTO,DATA) "; 4980b57cec5SDimitry Andric else if (NSect == 3) 4990b57cec5SDimitry Andric outs() << "(LTO,RODATA) "; 5000b57cec5SDimitry Andric else 5010b57cec5SDimitry Andric outs() << "(?,?) "; 5020b57cec5SDimitry Andric break; 5030b57cec5SDimitry Andric } 5040b57cec5SDimitry Andric section_iterator Sec = SectionRef(); 5050b57cec5SDimitry Andric if (S.Sym.getRawDataRefImpl().p) { 5060b57cec5SDimitry Andric Expected<section_iterator> SecOrErr = 5070b57cec5SDimitry Andric MachO->getSymbolSection(S.Sym.getRawDataRefImpl()); 5080b57cec5SDimitry Andric if (!SecOrErr) { 5090b57cec5SDimitry Andric consumeError(SecOrErr.takeError()); 5100b57cec5SDimitry Andric outs() << "(?,?) "; 5110b57cec5SDimitry Andric break; 5120b57cec5SDimitry Andric } 5130b57cec5SDimitry Andric Sec = *SecOrErr; 5140b57cec5SDimitry Andric if (Sec == MachO->section_end()) { 5150b57cec5SDimitry Andric outs() << "(?,?) "; 5160b57cec5SDimitry Andric break; 5170b57cec5SDimitry Andric } 5180b57cec5SDimitry Andric } else { 5190b57cec5SDimitry Andric Sec = S.Section; 5200b57cec5SDimitry Andric } 5210b57cec5SDimitry Andric DataRefImpl Ref = Sec->getRawDataRefImpl(); 5220b57cec5SDimitry Andric StringRef SectionName; 5230b57cec5SDimitry Andric if (Expected<StringRef> NameOrErr = MachO->getSectionName(Ref)) 5240b57cec5SDimitry Andric SectionName = *NameOrErr; 5250b57cec5SDimitry Andric StringRef SegmentName = MachO->getSectionFinalSegmentName(Ref); 5260b57cec5SDimitry Andric outs() << "(" << SegmentName << "," << SectionName << ") "; 5270b57cec5SDimitry Andric break; 5280b57cec5SDimitry Andric } 5290b57cec5SDimitry Andric default: 5300b57cec5SDimitry Andric outs() << "(?) "; 5310b57cec5SDimitry Andric break; 5320b57cec5SDimitry Andric } 5330b57cec5SDimitry Andric 5340b57cec5SDimitry Andric if (NType & MachO::N_EXT) { 5350b57cec5SDimitry Andric if (NDesc & MachO::REFERENCED_DYNAMICALLY) 5360b57cec5SDimitry Andric outs() << "[referenced dynamically] "; 5370b57cec5SDimitry Andric if (NType & MachO::N_PEXT) { 5380b57cec5SDimitry Andric if ((NDesc & MachO::N_WEAK_DEF) == MachO::N_WEAK_DEF) 5390b57cec5SDimitry Andric outs() << "weak private external "; 5400b57cec5SDimitry Andric else 5410b57cec5SDimitry Andric outs() << "private external "; 5420b57cec5SDimitry Andric } else { 5430b57cec5SDimitry Andric if ((NDesc & MachO::N_WEAK_REF) == MachO::N_WEAK_REF || 5440b57cec5SDimitry Andric (NDesc & MachO::N_WEAK_DEF) == MachO::N_WEAK_DEF) { 5450b57cec5SDimitry Andric if ((NDesc & (MachO::N_WEAK_REF | MachO::N_WEAK_DEF)) == 5460b57cec5SDimitry Andric (MachO::N_WEAK_REF | MachO::N_WEAK_DEF)) 5470b57cec5SDimitry Andric outs() << "weak external automatically hidden "; 5480b57cec5SDimitry Andric else 5490b57cec5SDimitry Andric outs() << "weak external "; 5500b57cec5SDimitry Andric } else 5510b57cec5SDimitry Andric outs() << "external "; 5520b57cec5SDimitry Andric } 5530b57cec5SDimitry Andric } else { 5540b57cec5SDimitry Andric if (NType & MachO::N_PEXT) 5550b57cec5SDimitry Andric outs() << "non-external (was a private external) "; 5560b57cec5SDimitry Andric else 5570b57cec5SDimitry Andric outs() << "non-external "; 5580b57cec5SDimitry Andric } 5590b57cec5SDimitry Andric 5600b57cec5SDimitry Andric if (Filetype == MachO::MH_OBJECT) { 5610b57cec5SDimitry Andric if (NDesc & MachO::N_NO_DEAD_STRIP) 5620b57cec5SDimitry Andric outs() << "[no dead strip] "; 5630b57cec5SDimitry Andric if ((NType & MachO::N_TYPE) != MachO::N_UNDF && 5640b57cec5SDimitry Andric NDesc & MachO::N_SYMBOL_RESOLVER) 5650b57cec5SDimitry Andric outs() << "[symbol resolver] "; 5660b57cec5SDimitry Andric if ((NType & MachO::N_TYPE) != MachO::N_UNDF && NDesc & MachO::N_ALT_ENTRY) 5670b57cec5SDimitry Andric outs() << "[alt entry] "; 5680b57cec5SDimitry Andric if ((NType & MachO::N_TYPE) != MachO::N_UNDF && NDesc & MachO::N_COLD_FUNC) 5690b57cec5SDimitry Andric outs() << "[cold func] "; 5700b57cec5SDimitry Andric } 5710b57cec5SDimitry Andric 5720b57cec5SDimitry Andric if ((NDesc & MachO::N_ARM_THUMB_DEF) == MachO::N_ARM_THUMB_DEF) 5730b57cec5SDimitry Andric outs() << "[Thumb] "; 5740b57cec5SDimitry Andric 5750b57cec5SDimitry Andric if ((NType & MachO::N_TYPE) == MachO::N_INDR) { 5760b57cec5SDimitry Andric outs() << S.Name << " (for "; 5770b57cec5SDimitry Andric StringRef IndirectName; 5780b57cec5SDimitry Andric if (MachO) { 5790b57cec5SDimitry Andric if (S.Sym.getRawDataRefImpl().p) { 5800b57cec5SDimitry Andric if (MachO->getIndirectName(S.Sym.getRawDataRefImpl(), IndirectName)) 5810b57cec5SDimitry Andric outs() << "?)"; 5820b57cec5SDimitry Andric else 5830b57cec5SDimitry Andric outs() << IndirectName << ")"; 5840b57cec5SDimitry Andric } else 5850b57cec5SDimitry Andric outs() << S.IndirectName << ")"; 5860b57cec5SDimitry Andric } else 5870b57cec5SDimitry Andric outs() << "?)"; 5880b57cec5SDimitry Andric } else 5890b57cec5SDimitry Andric outs() << S.Name; 5900b57cec5SDimitry Andric 5910b57cec5SDimitry Andric if ((Flags & MachO::MH_TWOLEVEL) == MachO::MH_TWOLEVEL && 5920b57cec5SDimitry Andric (((NType & MachO::N_TYPE) == MachO::N_UNDF && NValue == 0) || 5930b57cec5SDimitry Andric (NType & MachO::N_TYPE) == MachO::N_PBUD)) { 5940b57cec5SDimitry Andric uint32_t LibraryOrdinal = MachO::GET_LIBRARY_ORDINAL(NDesc); 5950b57cec5SDimitry Andric if (LibraryOrdinal != 0) { 5960b57cec5SDimitry Andric if (LibraryOrdinal == MachO::EXECUTABLE_ORDINAL) 5970b57cec5SDimitry Andric outs() << " (from executable)"; 5980b57cec5SDimitry Andric else if (LibraryOrdinal == MachO::DYNAMIC_LOOKUP_ORDINAL) 5990b57cec5SDimitry Andric outs() << " (dynamically looked up)"; 6000b57cec5SDimitry Andric else { 6010b57cec5SDimitry Andric StringRef LibraryName; 6020b57cec5SDimitry Andric if (!MachO || 6030b57cec5SDimitry Andric MachO->getLibraryShortNameByIndex(LibraryOrdinal - 1, LibraryName)) 6040b57cec5SDimitry Andric outs() << " (from bad library ordinal " << LibraryOrdinal << ")"; 6050b57cec5SDimitry Andric else 6060b57cec5SDimitry Andric outs() << " (from " << LibraryName << ")"; 6070b57cec5SDimitry Andric } 6080b57cec5SDimitry Andric } 6090b57cec5SDimitry Andric } 6100b57cec5SDimitry Andric 6110b57cec5SDimitry Andric outs() << "\n"; 6120b57cec5SDimitry Andric } 6130b57cec5SDimitry Andric 6140b57cec5SDimitry Andric // Table that maps Darwin's Mach-O stab constants to strings to allow printing. 6150b57cec5SDimitry Andric struct DarwinStabName { 6160b57cec5SDimitry Andric uint8_t NType; 6170b57cec5SDimitry Andric const char *Name; 6180b57cec5SDimitry Andric }; 6190b57cec5SDimitry Andric static const struct DarwinStabName DarwinStabNames[] = { 6200b57cec5SDimitry Andric {MachO::N_GSYM, "GSYM"}, 6210b57cec5SDimitry Andric {MachO::N_FNAME, "FNAME"}, 6220b57cec5SDimitry Andric {MachO::N_FUN, "FUN"}, 6230b57cec5SDimitry Andric {MachO::N_STSYM, "STSYM"}, 6240b57cec5SDimitry Andric {MachO::N_LCSYM, "LCSYM"}, 6250b57cec5SDimitry Andric {MachO::N_BNSYM, "BNSYM"}, 6260b57cec5SDimitry Andric {MachO::N_PC, "PC"}, 6270b57cec5SDimitry Andric {MachO::N_AST, "AST"}, 6280b57cec5SDimitry Andric {MachO::N_OPT, "OPT"}, 6290b57cec5SDimitry Andric {MachO::N_RSYM, "RSYM"}, 6300b57cec5SDimitry Andric {MachO::N_SLINE, "SLINE"}, 6310b57cec5SDimitry Andric {MachO::N_ENSYM, "ENSYM"}, 6320b57cec5SDimitry Andric {MachO::N_SSYM, "SSYM"}, 6330b57cec5SDimitry Andric {MachO::N_SO, "SO"}, 6340b57cec5SDimitry Andric {MachO::N_OSO, "OSO"}, 6350b57cec5SDimitry Andric {MachO::N_LSYM, "LSYM"}, 6360b57cec5SDimitry Andric {MachO::N_BINCL, "BINCL"}, 6370b57cec5SDimitry Andric {MachO::N_SOL, "SOL"}, 6380b57cec5SDimitry Andric {MachO::N_PARAMS, "PARAM"}, 6390b57cec5SDimitry Andric {MachO::N_VERSION, "VERS"}, 6400b57cec5SDimitry Andric {MachO::N_OLEVEL, "OLEV"}, 6410b57cec5SDimitry Andric {MachO::N_PSYM, "PSYM"}, 6420b57cec5SDimitry Andric {MachO::N_EINCL, "EINCL"}, 6430b57cec5SDimitry Andric {MachO::N_ENTRY, "ENTRY"}, 6440b57cec5SDimitry Andric {MachO::N_LBRAC, "LBRAC"}, 6450b57cec5SDimitry Andric {MachO::N_EXCL, "EXCL"}, 6460b57cec5SDimitry Andric {MachO::N_RBRAC, "RBRAC"}, 6470b57cec5SDimitry Andric {MachO::N_BCOMM, "BCOMM"}, 6480b57cec5SDimitry Andric {MachO::N_ECOMM, "ECOMM"}, 6490b57cec5SDimitry Andric {MachO::N_ECOML, "ECOML"}, 6500b57cec5SDimitry Andric {MachO::N_LENG, "LENG"}, 6510b57cec5SDimitry Andric }; 6520b57cec5SDimitry Andric 6530b57cec5SDimitry Andric static const char *getDarwinStabString(uint8_t NType) { 6540b57cec5SDimitry Andric for (auto I : makeArrayRef(DarwinStabNames)) 6550b57cec5SDimitry Andric if (I.NType == NType) 6560b57cec5SDimitry Andric return I.Name; 6570b57cec5SDimitry Andric return nullptr; 6580b57cec5SDimitry Andric } 6590b57cec5SDimitry Andric 6600b57cec5SDimitry Andric // darwinPrintStab() prints the n_sect, n_desc along with a symbolic name of 6610b57cec5SDimitry Andric // a stab n_type value in a Mach-O file. 6620b57cec5SDimitry Andric static void darwinPrintStab(MachOObjectFile *MachO, const NMSymbol &S) { 6630b57cec5SDimitry Andric MachO::nlist_64 STE_64; 6640b57cec5SDimitry Andric MachO::nlist STE; 6650b57cec5SDimitry Andric uint8_t NType; 6660b57cec5SDimitry Andric uint8_t NSect; 6670b57cec5SDimitry Andric uint16_t NDesc; 6680b57cec5SDimitry Andric DataRefImpl SymDRI = S.Sym.getRawDataRefImpl(); 6690b57cec5SDimitry Andric if (MachO->is64Bit()) { 6700b57cec5SDimitry Andric STE_64 = MachO->getSymbol64TableEntry(SymDRI); 6710b57cec5SDimitry Andric NType = STE_64.n_type; 6720b57cec5SDimitry Andric NSect = STE_64.n_sect; 6730b57cec5SDimitry Andric NDesc = STE_64.n_desc; 6740b57cec5SDimitry Andric } else { 6750b57cec5SDimitry Andric STE = MachO->getSymbolTableEntry(SymDRI); 6760b57cec5SDimitry Andric NType = STE.n_type; 6770b57cec5SDimitry Andric NSect = STE.n_sect; 6780b57cec5SDimitry Andric NDesc = STE.n_desc; 6790b57cec5SDimitry Andric } 6800b57cec5SDimitry Andric 6810b57cec5SDimitry Andric outs() << format(" %02x %04x ", NSect, NDesc); 6820b57cec5SDimitry Andric if (const char *stabString = getDarwinStabString(NType)) 6830b57cec5SDimitry Andric outs() << format("%5.5s", stabString); 6840b57cec5SDimitry Andric else 6850b57cec5SDimitry Andric outs() << format(" %02x", NType); 6860b57cec5SDimitry Andric } 6870b57cec5SDimitry Andric 6880b57cec5SDimitry Andric static Optional<std::string> demangle(StringRef Name, bool StripUnderscore) { 6890b57cec5SDimitry Andric if (StripUnderscore && !Name.empty() && Name[0] == '_') 6900b57cec5SDimitry Andric Name = Name.substr(1); 6910b57cec5SDimitry Andric 6920b57cec5SDimitry Andric if (!Name.startswith("_Z")) 6930b57cec5SDimitry Andric return None; 6940b57cec5SDimitry Andric 6950b57cec5SDimitry Andric int Status; 6960b57cec5SDimitry Andric char *Undecorated = 6970b57cec5SDimitry Andric itaniumDemangle(Name.str().c_str(), nullptr, nullptr, &Status); 6980b57cec5SDimitry Andric if (Status != 0) 6990b57cec5SDimitry Andric return None; 7000b57cec5SDimitry Andric 7010b57cec5SDimitry Andric std::string S(Undecorated); 7020b57cec5SDimitry Andric free(Undecorated); 7030b57cec5SDimitry Andric return S; 7040b57cec5SDimitry Andric } 7050b57cec5SDimitry Andric 7060b57cec5SDimitry Andric static bool symbolIsDefined(const NMSymbol &Sym) { 7070b57cec5SDimitry Andric return Sym.TypeChar != 'U' && Sym.TypeChar != 'w' && Sym.TypeChar != 'v'; 7080b57cec5SDimitry Andric } 7090b57cec5SDimitry Andric 7100b57cec5SDimitry Andric static void sortAndPrintSymbolList(SymbolicFile &Obj, bool printName, 7110b57cec5SDimitry Andric const std::string &ArchiveName, 7120b57cec5SDimitry Andric const std::string &ArchitectureName) { 7130b57cec5SDimitry Andric if (!NoSort) { 7148bcb0991SDimitry Andric using Comparator = bool (*)(const NMSymbol &, const NMSymbol &); 7158bcb0991SDimitry Andric Comparator Cmp; 7160b57cec5SDimitry Andric if (NumericSort) 7178bcb0991SDimitry Andric Cmp = &compareSymbolAddress; 7180b57cec5SDimitry Andric else if (SizeSort) 7198bcb0991SDimitry Andric Cmp = &compareSymbolSize; 7200b57cec5SDimitry Andric else 7218bcb0991SDimitry Andric Cmp = &compareSymbolName; 7220b57cec5SDimitry Andric 7230b57cec5SDimitry Andric if (ReverseSort) 7248bcb0991SDimitry Andric llvm::sort(SymbolList, [=](const NMSymbol &A, const NMSymbol &B) -> bool { 7258bcb0991SDimitry Andric return Cmp(B, A); 7268bcb0991SDimitry Andric }); 7278bcb0991SDimitry Andric else 7280b57cec5SDimitry Andric llvm::sort(SymbolList, Cmp); 7290b57cec5SDimitry Andric } 7300b57cec5SDimitry Andric 7310b57cec5SDimitry Andric if (!PrintFileName) { 7320b57cec5SDimitry Andric if (OutputFormat == posix && MultipleFiles && printName) { 7330b57cec5SDimitry Andric outs() << '\n' << CurrentFilename << ":\n"; 7340b57cec5SDimitry Andric } else if (OutputFormat == bsd && MultipleFiles && printName) { 7350b57cec5SDimitry Andric outs() << "\n" << CurrentFilename << ":\n"; 7360b57cec5SDimitry Andric } else if (OutputFormat == sysv) { 7370b57cec5SDimitry Andric outs() << "\n\nSymbols from " << CurrentFilename << ":\n\n"; 7380b57cec5SDimitry Andric if (isSymbolList64Bit(Obj)) 7390b57cec5SDimitry Andric outs() << "Name Value Class Type" 7400b57cec5SDimitry Andric << " Size Line Section\n"; 7410b57cec5SDimitry Andric else 7420b57cec5SDimitry Andric outs() << "Name Value Class Type" 7430b57cec5SDimitry Andric << " Size Line Section\n"; 7440b57cec5SDimitry Andric } 7450b57cec5SDimitry Andric } 7460b57cec5SDimitry Andric 7470b57cec5SDimitry Andric const char *printBlanks, *printDashes, *printFormat; 7480b57cec5SDimitry Andric if (isSymbolList64Bit(Obj)) { 7490b57cec5SDimitry Andric printBlanks = " "; 7500b57cec5SDimitry Andric printDashes = "----------------"; 7510b57cec5SDimitry Andric switch (AddressRadix) { 7520b57cec5SDimitry Andric case Radix::o: 7530b57cec5SDimitry Andric printFormat = OutputFormat == posix ? "%" PRIo64 : "%016" PRIo64; 7540b57cec5SDimitry Andric break; 7550b57cec5SDimitry Andric case Radix::x: 7560b57cec5SDimitry Andric printFormat = OutputFormat == posix ? "%" PRIx64 : "%016" PRIx64; 7570b57cec5SDimitry Andric break; 7580b57cec5SDimitry Andric default: 7590b57cec5SDimitry Andric printFormat = OutputFormat == posix ? "%" PRId64 : "%016" PRId64; 7600b57cec5SDimitry Andric } 7610b57cec5SDimitry Andric } else { 7620b57cec5SDimitry Andric printBlanks = " "; 7630b57cec5SDimitry Andric printDashes = "--------"; 7640b57cec5SDimitry Andric switch (AddressRadix) { 7650b57cec5SDimitry Andric case Radix::o: 7660b57cec5SDimitry Andric printFormat = OutputFormat == posix ? "%" PRIo64 : "%08" PRIo64; 7670b57cec5SDimitry Andric break; 7680b57cec5SDimitry Andric case Radix::x: 7690b57cec5SDimitry Andric printFormat = OutputFormat == posix ? "%" PRIx64 : "%08" PRIx64; 7700b57cec5SDimitry Andric break; 7710b57cec5SDimitry Andric default: 7720b57cec5SDimitry Andric printFormat = OutputFormat == posix ? "%" PRId64 : "%08" PRId64; 7730b57cec5SDimitry Andric } 7740b57cec5SDimitry Andric } 7750b57cec5SDimitry Andric 7760b57cec5SDimitry Andric auto writeFileName = [&](raw_ostream &S) { 7770b57cec5SDimitry Andric if (!ArchitectureName.empty()) 7780b57cec5SDimitry Andric S << "(for architecture " << ArchitectureName << "):"; 7790b57cec5SDimitry Andric if (OutputFormat == posix && !ArchiveName.empty()) 7800b57cec5SDimitry Andric S << ArchiveName << "[" << CurrentFilename << "]: "; 7810b57cec5SDimitry Andric else { 7820b57cec5SDimitry Andric if (!ArchiveName.empty()) 7830b57cec5SDimitry Andric S << ArchiveName << ":"; 7840b57cec5SDimitry Andric S << CurrentFilename << ": "; 7850b57cec5SDimitry Andric } 7860b57cec5SDimitry Andric }; 7870b57cec5SDimitry Andric 7880b57cec5SDimitry Andric if (SymbolList.empty()) { 7890b57cec5SDimitry Andric if (PrintFileName) 7900b57cec5SDimitry Andric writeFileName(errs()); 7910b57cec5SDimitry Andric errs() << "no symbols\n"; 7920b57cec5SDimitry Andric } 7930b57cec5SDimitry Andric 7940b57cec5SDimitry Andric for (const NMSymbol &S : SymbolList) { 7950b57cec5SDimitry Andric uint32_t SymFlags; 7960b57cec5SDimitry Andric std::string Name = S.Name.str(); 7970b57cec5SDimitry Andric MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(&Obj); 7980b57cec5SDimitry Andric if (Demangle) { 7990b57cec5SDimitry Andric if (Optional<std::string> Opt = demangle(S.Name, MachO)) 8000b57cec5SDimitry Andric Name = *Opt; 8010b57cec5SDimitry Andric } 8020b57cec5SDimitry Andric if (S.Sym.getRawDataRefImpl().p) 8030b57cec5SDimitry Andric SymFlags = S.Sym.getFlags(); 8040b57cec5SDimitry Andric else 8050b57cec5SDimitry Andric SymFlags = S.SymFlags; 8060b57cec5SDimitry Andric 8070b57cec5SDimitry Andric bool Undefined = SymFlags & SymbolRef::SF_Undefined; 8080b57cec5SDimitry Andric bool Global = SymFlags & SymbolRef::SF_Global; 8090b57cec5SDimitry Andric bool Weak = SymFlags & SymbolRef::SF_Weak; 8100b57cec5SDimitry Andric if ((!Undefined && UndefinedOnly) || (Undefined && DefinedOnly) || 8110b57cec5SDimitry Andric (!Global && ExternalOnly) || (Weak && NoWeakSymbols)) 8120b57cec5SDimitry Andric continue; 8130b57cec5SDimitry Andric if (PrintFileName) 8140b57cec5SDimitry Andric writeFileName(outs()); 8150b57cec5SDimitry Andric if ((JustSymbolName || 8160b57cec5SDimitry Andric (UndefinedOnly && MachO && OutputFormat != darwin)) && 8170b57cec5SDimitry Andric OutputFormat != posix) { 8180b57cec5SDimitry Andric outs() << Name << "\n"; 8190b57cec5SDimitry Andric continue; 8200b57cec5SDimitry Andric } 8210b57cec5SDimitry Andric 8220b57cec5SDimitry Andric char SymbolAddrStr[23], SymbolSizeStr[23]; 8230b57cec5SDimitry Andric 8240b57cec5SDimitry Andric // If the format is SysV or the symbol isn't defined, then print spaces. 8250b57cec5SDimitry Andric if (OutputFormat == sysv || !symbolIsDefined(S)) { 8260b57cec5SDimitry Andric if (OutputFormat == posix) { 8270b57cec5SDimitry Andric format(printFormat, S.Address) 8280b57cec5SDimitry Andric .print(SymbolAddrStr, sizeof(SymbolAddrStr)); 8290b57cec5SDimitry Andric format(printFormat, S.Size).print(SymbolSizeStr, sizeof(SymbolSizeStr)); 8300b57cec5SDimitry Andric } else { 8310b57cec5SDimitry Andric strcpy(SymbolAddrStr, printBlanks); 8320b57cec5SDimitry Andric strcpy(SymbolSizeStr, printBlanks); 8330b57cec5SDimitry Andric } 8340b57cec5SDimitry Andric } 8350b57cec5SDimitry Andric 8360b57cec5SDimitry Andric if (symbolIsDefined(S)) { 8370b57cec5SDimitry Andric // Otherwise, print the symbol address and size. 8380b57cec5SDimitry Andric if (Obj.isIR()) 8390b57cec5SDimitry Andric strcpy(SymbolAddrStr, printDashes); 8400b57cec5SDimitry Andric else if (MachO && S.TypeChar == 'I') 8410b57cec5SDimitry Andric strcpy(SymbolAddrStr, printBlanks); 8420b57cec5SDimitry Andric else 8430b57cec5SDimitry Andric format(printFormat, S.Address) 8440b57cec5SDimitry Andric .print(SymbolAddrStr, sizeof(SymbolAddrStr)); 8450b57cec5SDimitry Andric format(printFormat, S.Size).print(SymbolSizeStr, sizeof(SymbolSizeStr)); 8460b57cec5SDimitry Andric } 8470b57cec5SDimitry Andric 8480b57cec5SDimitry Andric // If OutputFormat is darwin or we are printing Mach-O symbols in hex and 8490b57cec5SDimitry Andric // we have a MachOObjectFile, call darwinPrintSymbol to print as darwin's 8500b57cec5SDimitry Andric // nm(1) -m output or hex, else if OutputFormat is darwin or we are 8510b57cec5SDimitry Andric // printing Mach-O symbols in hex and not a Mach-O object fall back to 8520b57cec5SDimitry Andric // OutputFormat bsd (see below). 8530b57cec5SDimitry Andric if ((OutputFormat == darwin || FormatMachOasHex) && (MachO || Obj.isIR())) { 8540b57cec5SDimitry Andric darwinPrintSymbol(Obj, S, SymbolAddrStr, printBlanks, printDashes, 8550b57cec5SDimitry Andric printFormat); 8560b57cec5SDimitry Andric } else if (OutputFormat == posix) { 8570b57cec5SDimitry Andric outs() << Name << " " << S.TypeChar << " " << SymbolAddrStr << " " 8580b57cec5SDimitry Andric << (MachO ? "0" : SymbolSizeStr) << "\n"; 8590b57cec5SDimitry Andric } else if (OutputFormat == bsd || (OutputFormat == darwin && !MachO)) { 8600b57cec5SDimitry Andric if (PrintAddress) 8610b57cec5SDimitry Andric outs() << SymbolAddrStr << ' '; 8620b57cec5SDimitry Andric if (PrintSize) 8630b57cec5SDimitry Andric outs() << SymbolSizeStr << ' '; 8640b57cec5SDimitry Andric outs() << S.TypeChar; 8650b57cec5SDimitry Andric if (S.TypeChar == '-' && MachO) 8660b57cec5SDimitry Andric darwinPrintStab(MachO, S); 8670b57cec5SDimitry Andric outs() << " " << Name; 8680b57cec5SDimitry Andric if (S.TypeChar == 'I' && MachO) { 8690b57cec5SDimitry Andric outs() << " (indirect for "; 8700b57cec5SDimitry Andric if (S.Sym.getRawDataRefImpl().p) { 8710b57cec5SDimitry Andric StringRef IndirectName; 8720b57cec5SDimitry Andric if (MachO->getIndirectName(S.Sym.getRawDataRefImpl(), IndirectName)) 8730b57cec5SDimitry Andric outs() << "?)"; 8740b57cec5SDimitry Andric else 8750b57cec5SDimitry Andric outs() << IndirectName << ")"; 8760b57cec5SDimitry Andric } else 8770b57cec5SDimitry Andric outs() << S.IndirectName << ")"; 8780b57cec5SDimitry Andric } 8790b57cec5SDimitry Andric outs() << "\n"; 8800b57cec5SDimitry Andric } else if (OutputFormat == sysv) { 8810b57cec5SDimitry Andric outs() << left_justify(Name, 20) << "|" << SymbolAddrStr << "| " 8820b57cec5SDimitry Andric << S.TypeChar << " |" << right_justify(S.TypeName, 18) << "|" 8830b57cec5SDimitry Andric << SymbolSizeStr << "| |" << S.SectionName << "\n"; 8840b57cec5SDimitry Andric } 8850b57cec5SDimitry Andric } 8860b57cec5SDimitry Andric 8870b57cec5SDimitry Andric SymbolList.clear(); 8880b57cec5SDimitry Andric } 8890b57cec5SDimitry Andric 8900b57cec5SDimitry Andric static char getSymbolNMTypeChar(ELFObjectFileBase &Obj, 8910b57cec5SDimitry Andric basic_symbol_iterator I) { 8920b57cec5SDimitry Andric // OK, this is ELF 8930b57cec5SDimitry Andric elf_symbol_iterator SymI(I); 8940b57cec5SDimitry Andric 8950b57cec5SDimitry Andric Expected<elf_section_iterator> SecIOrErr = SymI->getSection(); 8960b57cec5SDimitry Andric if (!SecIOrErr) { 8970b57cec5SDimitry Andric consumeError(SecIOrErr.takeError()); 8980b57cec5SDimitry Andric return '?'; 8990b57cec5SDimitry Andric } 9000b57cec5SDimitry Andric 9010b57cec5SDimitry Andric uint8_t Binding = SymI->getBinding(); 9020b57cec5SDimitry Andric if (Binding == ELF::STB_GNU_UNIQUE) 9030b57cec5SDimitry Andric return 'u'; 9040b57cec5SDimitry Andric 9050b57cec5SDimitry Andric assert(Binding != ELF::STB_WEAK && "STB_WEAK not tested in calling function"); 9060b57cec5SDimitry Andric if (Binding != ELF::STB_GLOBAL && Binding != ELF::STB_LOCAL) 9070b57cec5SDimitry Andric return '?'; 9080b57cec5SDimitry Andric 9090b57cec5SDimitry Andric elf_section_iterator SecI = *SecIOrErr; 9100b57cec5SDimitry Andric if (SecI != Obj.section_end()) { 9110b57cec5SDimitry Andric uint32_t Type = SecI->getType(); 9120b57cec5SDimitry Andric uint64_t Flags = SecI->getFlags(); 9130b57cec5SDimitry Andric if (Flags & ELF::SHF_EXECINSTR) 9140b57cec5SDimitry Andric return 't'; 9150b57cec5SDimitry Andric if (Type == ELF::SHT_NOBITS) 9160b57cec5SDimitry Andric return 'b'; 9170b57cec5SDimitry Andric if (Flags & ELF::SHF_ALLOC) 9180b57cec5SDimitry Andric return Flags & ELF::SHF_WRITE ? 'd' : 'r'; 9190b57cec5SDimitry Andric 9208bcb0991SDimitry Andric auto NameOrErr = SecI->getName(); 9218bcb0991SDimitry Andric if (!NameOrErr) { 9228bcb0991SDimitry Andric consumeError(NameOrErr.takeError()); 9230b57cec5SDimitry Andric return '?'; 9248bcb0991SDimitry Andric } 9258bcb0991SDimitry Andric if ((*NameOrErr).startswith(".debug")) 9260b57cec5SDimitry Andric return 'N'; 9270b57cec5SDimitry Andric if (!(Flags & ELF::SHF_WRITE)) 9280b57cec5SDimitry Andric return 'n'; 9290b57cec5SDimitry Andric } 9300b57cec5SDimitry Andric 9310b57cec5SDimitry Andric return '?'; 9320b57cec5SDimitry Andric } 9330b57cec5SDimitry Andric 9340b57cec5SDimitry Andric static char getSymbolNMTypeChar(COFFObjectFile &Obj, symbol_iterator I) { 9350b57cec5SDimitry Andric COFFSymbolRef Symb = Obj.getCOFFSymbol(*I); 9360b57cec5SDimitry Andric // OK, this is COFF. 9370b57cec5SDimitry Andric symbol_iterator SymI(I); 9380b57cec5SDimitry Andric 9390b57cec5SDimitry Andric Expected<StringRef> Name = SymI->getName(); 9400b57cec5SDimitry Andric if (!Name) { 9410b57cec5SDimitry Andric consumeError(Name.takeError()); 9420b57cec5SDimitry Andric return '?'; 9430b57cec5SDimitry Andric } 9440b57cec5SDimitry Andric 9450b57cec5SDimitry Andric char Ret = StringSwitch<char>(*Name) 9460b57cec5SDimitry Andric .StartsWith(".debug", 'N') 9470b57cec5SDimitry Andric .StartsWith(".sxdata", 'N') 9480b57cec5SDimitry Andric .Default('?'); 9490b57cec5SDimitry Andric 9500b57cec5SDimitry Andric if (Ret != '?') 9510b57cec5SDimitry Andric return Ret; 9520b57cec5SDimitry Andric 9530b57cec5SDimitry Andric uint32_t Characteristics = 0; 9540b57cec5SDimitry Andric if (!COFF::isReservedSectionNumber(Symb.getSectionNumber())) { 9550b57cec5SDimitry Andric Expected<section_iterator> SecIOrErr = SymI->getSection(); 9560b57cec5SDimitry Andric if (!SecIOrErr) { 9570b57cec5SDimitry Andric consumeError(SecIOrErr.takeError()); 9580b57cec5SDimitry Andric return '?'; 9590b57cec5SDimitry Andric } 9600b57cec5SDimitry Andric section_iterator SecI = *SecIOrErr; 9610b57cec5SDimitry Andric const coff_section *Section = Obj.getCOFFSection(*SecI); 9620b57cec5SDimitry Andric Characteristics = Section->Characteristics; 9630b57cec5SDimitry Andric if (Expected<StringRef> NameOrErr = Obj.getSectionName(Section)) 9640b57cec5SDimitry Andric if (NameOrErr->startswith(".idata")) 9650b57cec5SDimitry Andric return 'i'; 9660b57cec5SDimitry Andric } 9670b57cec5SDimitry Andric 9680b57cec5SDimitry Andric switch (Symb.getSectionNumber()) { 9690b57cec5SDimitry Andric case COFF::IMAGE_SYM_DEBUG: 9700b57cec5SDimitry Andric return 'n'; 9710b57cec5SDimitry Andric default: 9720b57cec5SDimitry Andric // Check section type. 9730b57cec5SDimitry Andric if (Characteristics & COFF::IMAGE_SCN_CNT_CODE) 9740b57cec5SDimitry Andric return 't'; 9750b57cec5SDimitry Andric if (Characteristics & COFF::IMAGE_SCN_CNT_INITIALIZED_DATA) 9760b57cec5SDimitry Andric return Characteristics & COFF::IMAGE_SCN_MEM_WRITE ? 'd' : 'r'; 9770b57cec5SDimitry Andric if (Characteristics & COFF::IMAGE_SCN_CNT_UNINITIALIZED_DATA) 9780b57cec5SDimitry Andric return 'b'; 9790b57cec5SDimitry Andric if (Characteristics & COFF::IMAGE_SCN_LNK_INFO) 9800b57cec5SDimitry Andric return 'i'; 9810b57cec5SDimitry Andric // Check for section symbol. 9820b57cec5SDimitry Andric if (Symb.isSectionDefinition()) 9830b57cec5SDimitry Andric return 's'; 9840b57cec5SDimitry Andric } 9850b57cec5SDimitry Andric 9860b57cec5SDimitry Andric return '?'; 9870b57cec5SDimitry Andric } 9880b57cec5SDimitry Andric 9890b57cec5SDimitry Andric static char getSymbolNMTypeChar(COFFImportFile &Obj) { 9900b57cec5SDimitry Andric switch (Obj.getCOFFImportHeader()->getType()) { 9910b57cec5SDimitry Andric case COFF::IMPORT_CODE: 9920b57cec5SDimitry Andric return 't'; 9930b57cec5SDimitry Andric case COFF::IMPORT_DATA: 9940b57cec5SDimitry Andric return 'd'; 9950b57cec5SDimitry Andric case COFF::IMPORT_CONST: 9960b57cec5SDimitry Andric return 'r'; 9970b57cec5SDimitry Andric } 9980b57cec5SDimitry Andric return '?'; 9990b57cec5SDimitry Andric } 10000b57cec5SDimitry Andric 10010b57cec5SDimitry Andric static char getSymbolNMTypeChar(MachOObjectFile &Obj, basic_symbol_iterator I) { 10020b57cec5SDimitry Andric DataRefImpl Symb = I->getRawDataRefImpl(); 10030b57cec5SDimitry Andric uint8_t NType = Obj.is64Bit() ? Obj.getSymbol64TableEntry(Symb).n_type 10040b57cec5SDimitry Andric : Obj.getSymbolTableEntry(Symb).n_type; 10050b57cec5SDimitry Andric 10060b57cec5SDimitry Andric if (NType & MachO::N_STAB) 10070b57cec5SDimitry Andric return '-'; 10080b57cec5SDimitry Andric 10090b57cec5SDimitry Andric switch (NType & MachO::N_TYPE) { 10100b57cec5SDimitry Andric case MachO::N_ABS: 10110b57cec5SDimitry Andric return 's'; 10120b57cec5SDimitry Andric case MachO::N_INDR: 10130b57cec5SDimitry Andric return 'i'; 10140b57cec5SDimitry Andric case MachO::N_SECT: { 10150b57cec5SDimitry Andric Expected<section_iterator> SecOrErr = Obj.getSymbolSection(Symb); 10160b57cec5SDimitry Andric if (!SecOrErr) { 10170b57cec5SDimitry Andric consumeError(SecOrErr.takeError()); 10180b57cec5SDimitry Andric return 's'; 10190b57cec5SDimitry Andric } 10200b57cec5SDimitry Andric section_iterator Sec = *SecOrErr; 10210b57cec5SDimitry Andric if (Sec == Obj.section_end()) 10220b57cec5SDimitry Andric return 's'; 10230b57cec5SDimitry Andric DataRefImpl Ref = Sec->getRawDataRefImpl(); 10240b57cec5SDimitry Andric StringRef SectionName; 10250b57cec5SDimitry Andric if (Expected<StringRef> NameOrErr = Obj.getSectionName(Ref)) 10260b57cec5SDimitry Andric SectionName = *NameOrErr; 10270b57cec5SDimitry Andric StringRef SegmentName = Obj.getSectionFinalSegmentName(Ref); 10280b57cec5SDimitry Andric if (Obj.is64Bit() && Obj.getHeader64().filetype == MachO::MH_KEXT_BUNDLE && 10290b57cec5SDimitry Andric SegmentName == "__TEXT_EXEC" && SectionName == "__text") 10300b57cec5SDimitry Andric return 't'; 10310b57cec5SDimitry Andric if (SegmentName == "__TEXT" && SectionName == "__text") 10320b57cec5SDimitry Andric return 't'; 10330b57cec5SDimitry Andric if (SegmentName == "__DATA" && SectionName == "__data") 10340b57cec5SDimitry Andric return 'd'; 10350b57cec5SDimitry Andric if (SegmentName == "__DATA" && SectionName == "__bss") 10360b57cec5SDimitry Andric return 'b'; 10370b57cec5SDimitry Andric return 's'; 10380b57cec5SDimitry Andric } 10390b57cec5SDimitry Andric } 10400b57cec5SDimitry Andric 10410b57cec5SDimitry Andric return '?'; 10420b57cec5SDimitry Andric } 10430b57cec5SDimitry Andric 10440b57cec5SDimitry Andric static char getSymbolNMTypeChar(WasmObjectFile &Obj, basic_symbol_iterator I) { 10450b57cec5SDimitry Andric uint32_t Flags = I->getFlags(); 10460b57cec5SDimitry Andric if (Flags & SymbolRef::SF_Executable) 10470b57cec5SDimitry Andric return 't'; 10480b57cec5SDimitry Andric return 'd'; 10490b57cec5SDimitry Andric } 10500b57cec5SDimitry Andric 10510b57cec5SDimitry Andric static char getSymbolNMTypeChar(IRObjectFile &Obj, basic_symbol_iterator I) { 10520b57cec5SDimitry Andric uint32_t Flags = I->getFlags(); 10530b57cec5SDimitry Andric // FIXME: should we print 'b'? At the IR level we cannot be sure if this 10540b57cec5SDimitry Andric // will be in bss or not, but we could approximate. 10550b57cec5SDimitry Andric if (Flags & SymbolRef::SF_Executable) 10560b57cec5SDimitry Andric return 't'; 10570b57cec5SDimitry Andric else if (Triple(Obj.getTargetTriple()).isOSDarwin() && 10580b57cec5SDimitry Andric (Flags & SymbolRef::SF_Const)) 10590b57cec5SDimitry Andric return 's'; 10600b57cec5SDimitry Andric else 10610b57cec5SDimitry Andric return 'd'; 10620b57cec5SDimitry Andric } 10630b57cec5SDimitry Andric 10640b57cec5SDimitry Andric static bool isObject(SymbolicFile &Obj, basic_symbol_iterator I) { 10650b57cec5SDimitry Andric return !dyn_cast<ELFObjectFileBase>(&Obj) 10660b57cec5SDimitry Andric ? false 10670b57cec5SDimitry Andric : elf_symbol_iterator(I)->getELFType() == ELF::STT_OBJECT; 10680b57cec5SDimitry Andric } 10690b57cec5SDimitry Andric 10700b57cec5SDimitry Andric // For ELF object files, Set TypeName to the symbol typename, to be printed 10710b57cec5SDimitry Andric // in the 'Type' column of the SYSV format output. 10720b57cec5SDimitry Andric static StringRef getNMTypeName(SymbolicFile &Obj, basic_symbol_iterator I) { 10730b57cec5SDimitry Andric if (isa<ELFObjectFileBase>(&Obj)) { 10740b57cec5SDimitry Andric elf_symbol_iterator SymI(I); 10750b57cec5SDimitry Andric return SymI->getELFTypeName(); 10760b57cec5SDimitry Andric } 10770b57cec5SDimitry Andric return ""; 10780b57cec5SDimitry Andric } 10790b57cec5SDimitry Andric 10800b57cec5SDimitry Andric // Return Posix nm class type tag (single letter), but also set SecName and 10810b57cec5SDimitry Andric // section and name, to be used in format=sysv output. 10820b57cec5SDimitry Andric static char getNMSectionTagAndName(SymbolicFile &Obj, basic_symbol_iterator I, 10830b57cec5SDimitry Andric StringRef &SecName) { 10840b57cec5SDimitry Andric uint32_t Symflags = I->getFlags(); 10858bcb0991SDimitry Andric if (ELFObjectFileBase *ELFObj = dyn_cast<ELFObjectFileBase>(&Obj)) { 10860b57cec5SDimitry Andric if (Symflags & object::SymbolRef::SF_Absolute) 10870b57cec5SDimitry Andric SecName = "*ABS*"; 10880b57cec5SDimitry Andric else if (Symflags & object::SymbolRef::SF_Common) 10890b57cec5SDimitry Andric SecName = "*COM*"; 10900b57cec5SDimitry Andric else if (Symflags & object::SymbolRef::SF_Undefined) 10910b57cec5SDimitry Andric SecName = "*UND*"; 10920b57cec5SDimitry Andric else { 10930b57cec5SDimitry Andric elf_symbol_iterator SymI(I); 10940b57cec5SDimitry Andric Expected<elf_section_iterator> SecIOrErr = SymI->getSection(); 10950b57cec5SDimitry Andric if (!SecIOrErr) { 10960b57cec5SDimitry Andric consumeError(SecIOrErr.takeError()); 10970b57cec5SDimitry Andric return '?'; 10980b57cec5SDimitry Andric } 10998bcb0991SDimitry Andric 11008bcb0991SDimitry Andric if (*SecIOrErr == ELFObj->section_end()) 11018bcb0991SDimitry Andric return '?'; 11028bcb0991SDimitry Andric 11038bcb0991SDimitry Andric Expected<StringRef> NameOrErr = (*SecIOrErr)->getName(); 11048bcb0991SDimitry Andric if (!NameOrErr) { 11058bcb0991SDimitry Andric consumeError(NameOrErr.takeError()); 11068bcb0991SDimitry Andric return '?'; 11078bcb0991SDimitry Andric } 11088bcb0991SDimitry Andric SecName = *NameOrErr; 11090b57cec5SDimitry Andric } 11100b57cec5SDimitry Andric } 11110b57cec5SDimitry Andric 11120b57cec5SDimitry Andric if ((Symflags & object::SymbolRef::SF_Weak) && !isa<MachOObjectFile>(Obj)) { 11130b57cec5SDimitry Andric char Ret = isObject(Obj, I) ? 'v' : 'w'; 11140b57cec5SDimitry Andric return (!(Symflags & object::SymbolRef::SF_Undefined)) ? toupper(Ret) : Ret; 11150b57cec5SDimitry Andric } 11160b57cec5SDimitry Andric 11170b57cec5SDimitry Andric if (Symflags & object::SymbolRef::SF_Undefined) 11180b57cec5SDimitry Andric return 'U'; 11190b57cec5SDimitry Andric 11200b57cec5SDimitry Andric if (Symflags & object::SymbolRef::SF_Common) 11210b57cec5SDimitry Andric return 'C'; 11220b57cec5SDimitry Andric 11230b57cec5SDimitry Andric char Ret = '?'; 11240b57cec5SDimitry Andric if (Symflags & object::SymbolRef::SF_Absolute) 11250b57cec5SDimitry Andric Ret = 'a'; 11260b57cec5SDimitry Andric else if (IRObjectFile *IR = dyn_cast<IRObjectFile>(&Obj)) 11270b57cec5SDimitry Andric Ret = getSymbolNMTypeChar(*IR, I); 11280b57cec5SDimitry Andric else if (COFFObjectFile *COFF = dyn_cast<COFFObjectFile>(&Obj)) 11290b57cec5SDimitry Andric Ret = getSymbolNMTypeChar(*COFF, I); 11300b57cec5SDimitry Andric else if (COFFImportFile *COFFImport = dyn_cast<COFFImportFile>(&Obj)) 11310b57cec5SDimitry Andric Ret = getSymbolNMTypeChar(*COFFImport); 11320b57cec5SDimitry Andric else if (MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(&Obj)) 11330b57cec5SDimitry Andric Ret = getSymbolNMTypeChar(*MachO, I); 11340b57cec5SDimitry Andric else if (WasmObjectFile *Wasm = dyn_cast<WasmObjectFile>(&Obj)) 11350b57cec5SDimitry Andric Ret = getSymbolNMTypeChar(*Wasm, I); 1136*480093f4SDimitry Andric else if (ELFObjectFileBase *ELF = dyn_cast<ELFObjectFileBase>(&Obj)) { 1137*480093f4SDimitry Andric if (ELFSymbolRef(*I).getELFType() == ELF::STT_GNU_IFUNC) 1138*480093f4SDimitry Andric return 'i'; 1139*480093f4SDimitry Andric Ret = getSymbolNMTypeChar(*ELF, I); 1140*480093f4SDimitry Andric if (ELFSymbolRef(*I).getBinding() == ELF::STB_GNU_UNIQUE) 1141*480093f4SDimitry Andric return Ret; 1142*480093f4SDimitry Andric } else 1143*480093f4SDimitry Andric llvm_unreachable("unknown binary format"); 11440b57cec5SDimitry Andric 11450b57cec5SDimitry Andric if (!(Symflags & object::SymbolRef::SF_Global)) 11460b57cec5SDimitry Andric return Ret; 11470b57cec5SDimitry Andric 11480b57cec5SDimitry Andric return toupper(Ret); 11490b57cec5SDimitry Andric } 11500b57cec5SDimitry Andric 11510b57cec5SDimitry Andric // getNsectForSegSect() is used to implement the Mach-O "-s segname sectname" 11520b57cec5SDimitry Andric // option to dump only those symbols from that section in a Mach-O file. 11530b57cec5SDimitry Andric // It is called once for each Mach-O file from dumpSymbolNamesFromObject() 11540b57cec5SDimitry Andric // to get the section number for that named section from the command line 11550b57cec5SDimitry Andric // arguments. It returns the section number for that section in the Mach-O 11560b57cec5SDimitry Andric // file or zero it is not present. 11570b57cec5SDimitry Andric static unsigned getNsectForSegSect(MachOObjectFile *Obj) { 11580b57cec5SDimitry Andric unsigned Nsect = 1; 11590b57cec5SDimitry Andric for (auto &S : Obj->sections()) { 11600b57cec5SDimitry Andric DataRefImpl Ref = S.getRawDataRefImpl(); 11610b57cec5SDimitry Andric StringRef SectionName; 11620b57cec5SDimitry Andric if (Expected<StringRef> NameOrErr = Obj->getSectionName(Ref)) 11630b57cec5SDimitry Andric SectionName = *NameOrErr; 11640b57cec5SDimitry Andric StringRef SegmentName = Obj->getSectionFinalSegmentName(Ref); 11650b57cec5SDimitry Andric if (SegmentName == SegSect[0] && SectionName == SegSect[1]) 11660b57cec5SDimitry Andric return Nsect; 11670b57cec5SDimitry Andric Nsect++; 11680b57cec5SDimitry Andric } 11690b57cec5SDimitry Andric return 0; 11700b57cec5SDimitry Andric } 11710b57cec5SDimitry Andric 11720b57cec5SDimitry Andric // getNsectInMachO() is used to implement the Mach-O "-s segname sectname" 11730b57cec5SDimitry Andric // option to dump only those symbols from that section in a Mach-O file. 11740b57cec5SDimitry Andric // It is called once for each symbol in a Mach-O file from 11750b57cec5SDimitry Andric // dumpSymbolNamesFromObject() and returns the section number for that symbol 11760b57cec5SDimitry Andric // if it is in a section, else it returns 0. 11770b57cec5SDimitry Andric static unsigned getNsectInMachO(MachOObjectFile &Obj, BasicSymbolRef Sym) { 11780b57cec5SDimitry Andric DataRefImpl Symb = Sym.getRawDataRefImpl(); 11790b57cec5SDimitry Andric if (Obj.is64Bit()) { 11800b57cec5SDimitry Andric MachO::nlist_64 STE = Obj.getSymbol64TableEntry(Symb); 11810b57cec5SDimitry Andric return (STE.n_type & MachO::N_TYPE) == MachO::N_SECT ? STE.n_sect : 0; 11820b57cec5SDimitry Andric } 11830b57cec5SDimitry Andric MachO::nlist STE = Obj.getSymbolTableEntry(Symb); 11840b57cec5SDimitry Andric return (STE.n_type & MachO::N_TYPE) == MachO::N_SECT ? STE.n_sect : 0; 11850b57cec5SDimitry Andric } 11860b57cec5SDimitry Andric 11870b57cec5SDimitry Andric static void 11880b57cec5SDimitry Andric dumpSymbolNamesFromObject(SymbolicFile &Obj, bool printName, 11890b57cec5SDimitry Andric const std::string &ArchiveName = std::string(), 11900b57cec5SDimitry Andric const std::string &ArchitectureName = std::string()) { 11910b57cec5SDimitry Andric auto Symbols = Obj.symbols(); 11920b57cec5SDimitry Andric if (DynamicSyms) { 11930b57cec5SDimitry Andric const auto *E = dyn_cast<ELFObjectFileBase>(&Obj); 11940b57cec5SDimitry Andric if (!E) { 11950b57cec5SDimitry Andric error("File format has no dynamic symbol table", Obj.getFileName()); 11960b57cec5SDimitry Andric return; 11970b57cec5SDimitry Andric } 11980b57cec5SDimitry Andric Symbols = E->getDynamicSymbolIterators(); 11990b57cec5SDimitry Andric } 12000b57cec5SDimitry Andric std::string NameBuffer; 12010b57cec5SDimitry Andric raw_string_ostream OS(NameBuffer); 12020b57cec5SDimitry Andric // If a "-s segname sectname" option was specified and this is a Mach-O 12030b57cec5SDimitry Andric // file get the section number for that section in this object file. 12040b57cec5SDimitry Andric unsigned int Nsect = 0; 12050b57cec5SDimitry Andric MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(&Obj); 12060b57cec5SDimitry Andric if (!SegSect.empty() && MachO) { 12070b57cec5SDimitry Andric Nsect = getNsectForSegSect(MachO); 12080b57cec5SDimitry Andric // If this section is not in the object file no symbols are printed. 12090b57cec5SDimitry Andric if (Nsect == 0) 12100b57cec5SDimitry Andric return; 12110b57cec5SDimitry Andric } 12120b57cec5SDimitry Andric if (!MachO || !DyldInfoOnly) { 12130b57cec5SDimitry Andric for (BasicSymbolRef Sym : Symbols) { 12140b57cec5SDimitry Andric uint32_t SymFlags = Sym.getFlags(); 12150b57cec5SDimitry Andric if (!DebugSyms && (SymFlags & SymbolRef::SF_FormatSpecific)) 12160b57cec5SDimitry Andric continue; 12170b57cec5SDimitry Andric if (WithoutAliases && (SymFlags & SymbolRef::SF_Indirect)) 12180b57cec5SDimitry Andric continue; 12190b57cec5SDimitry Andric // If a "-s segname sectname" option was specified and this is a Mach-O 12200b57cec5SDimitry Andric // file and this section appears in this file, Nsect will be non-zero then 12210b57cec5SDimitry Andric // see if this symbol is a symbol from that section and if not skip it. 12220b57cec5SDimitry Andric if (Nsect && Nsect != getNsectInMachO(*MachO, Sym)) 12230b57cec5SDimitry Andric continue; 12240b57cec5SDimitry Andric NMSymbol S = {}; 12250b57cec5SDimitry Andric S.Size = 0; 12260b57cec5SDimitry Andric S.Address = 0; 12270b57cec5SDimitry Andric if (isa<ELFObjectFileBase>(&Obj)) 12280b57cec5SDimitry Andric S.Size = ELFSymbolRef(Sym).getSize(); 12290b57cec5SDimitry Andric if (PrintAddress && isa<ObjectFile>(Obj)) { 12300b57cec5SDimitry Andric SymbolRef SymRef(Sym); 12310b57cec5SDimitry Andric Expected<uint64_t> AddressOrErr = SymRef.getAddress(); 12320b57cec5SDimitry Andric if (!AddressOrErr) { 12330b57cec5SDimitry Andric consumeError(AddressOrErr.takeError()); 12340b57cec5SDimitry Andric break; 12350b57cec5SDimitry Andric } 12360b57cec5SDimitry Andric S.Address = *AddressOrErr; 12370b57cec5SDimitry Andric } 12380b57cec5SDimitry Andric S.TypeName = getNMTypeName(Obj, Sym); 12390b57cec5SDimitry Andric S.TypeChar = getNMSectionTagAndName(Obj, Sym, S.SectionName); 12400b57cec5SDimitry Andric if (Error E = Sym.printName(OS)) { 12410b57cec5SDimitry Andric if (MachO) { 12420b57cec5SDimitry Andric OS << "bad string index"; 12430b57cec5SDimitry Andric consumeError(std::move(E)); 12440b57cec5SDimitry Andric } else 12450b57cec5SDimitry Andric error(std::move(E), Obj.getFileName()); 12460b57cec5SDimitry Andric } 12470b57cec5SDimitry Andric OS << '\0'; 12480b57cec5SDimitry Andric S.Sym = Sym; 12490b57cec5SDimitry Andric SymbolList.push_back(S); 12500b57cec5SDimitry Andric } 12510b57cec5SDimitry Andric } 12520b57cec5SDimitry Andric 12530b57cec5SDimitry Andric OS.flush(); 12540b57cec5SDimitry Andric const char *P = NameBuffer.c_str(); 12550b57cec5SDimitry Andric unsigned I; 12560b57cec5SDimitry Andric for (I = 0; I < SymbolList.size(); ++I) { 12570b57cec5SDimitry Andric SymbolList[I].Name = P; 12580b57cec5SDimitry Andric P += strlen(P) + 1; 12590b57cec5SDimitry Andric } 12600b57cec5SDimitry Andric 12610b57cec5SDimitry Andric // If this is a Mach-O file where the nlist symbol table is out of sync 12620b57cec5SDimitry Andric // with the dyld export trie then look through exports and fake up symbols 12630b57cec5SDimitry Andric // for the ones that are missing (also done with the -add-dyldinfo flag). 12640b57cec5SDimitry Andric // This is needed if strip(1) -T is run on a binary containing swift 12650b57cec5SDimitry Andric // language symbols for example. The option -only-dyldinfo will fake up 12660b57cec5SDimitry Andric // all symbols from the dyld export trie as well as the bind info. 12670b57cec5SDimitry Andric std::string ExportsNameBuffer; 12680b57cec5SDimitry Andric raw_string_ostream EOS(ExportsNameBuffer); 12690b57cec5SDimitry Andric std::string BindsNameBuffer; 12700b57cec5SDimitry Andric raw_string_ostream BOS(BindsNameBuffer); 12710b57cec5SDimitry Andric std::string LazysNameBuffer; 12720b57cec5SDimitry Andric raw_string_ostream LOS(LazysNameBuffer); 12730b57cec5SDimitry Andric std::string WeaksNameBuffer; 12740b57cec5SDimitry Andric raw_string_ostream WOS(WeaksNameBuffer); 12750b57cec5SDimitry Andric std::string FunctionStartsNameBuffer; 12760b57cec5SDimitry Andric raw_string_ostream FOS(FunctionStartsNameBuffer); 12770b57cec5SDimitry Andric if (MachO && !NoDyldInfo) { 12780b57cec5SDimitry Andric MachO::mach_header H; 12790b57cec5SDimitry Andric MachO::mach_header_64 H_64; 12800b57cec5SDimitry Andric uint32_t HFlags = 0; 12810b57cec5SDimitry Andric if (MachO->is64Bit()) { 12820b57cec5SDimitry Andric H_64 = MachO->MachOObjectFile::getHeader64(); 12830b57cec5SDimitry Andric HFlags = H_64.flags; 12840b57cec5SDimitry Andric } else { 12850b57cec5SDimitry Andric H = MachO->MachOObjectFile::getHeader(); 12860b57cec5SDimitry Andric HFlags = H.flags; 12870b57cec5SDimitry Andric } 12880b57cec5SDimitry Andric uint64_t BaseSegmentAddress = 0; 12890b57cec5SDimitry Andric for (const auto &Command : MachO->load_commands()) { 12900b57cec5SDimitry Andric if (Command.C.cmd == MachO::LC_SEGMENT) { 12910b57cec5SDimitry Andric MachO::segment_command Seg = MachO->getSegmentLoadCommand(Command); 12920b57cec5SDimitry Andric if (Seg.fileoff == 0 && Seg.filesize != 0) { 12930b57cec5SDimitry Andric BaseSegmentAddress = Seg.vmaddr; 12940b57cec5SDimitry Andric break; 12950b57cec5SDimitry Andric } 12960b57cec5SDimitry Andric } else if (Command.C.cmd == MachO::LC_SEGMENT_64) { 12970b57cec5SDimitry Andric MachO::segment_command_64 Seg = MachO->getSegment64LoadCommand(Command); 12980b57cec5SDimitry Andric if (Seg.fileoff == 0 && Seg.filesize != 0) { 12990b57cec5SDimitry Andric BaseSegmentAddress = Seg.vmaddr; 13000b57cec5SDimitry Andric break; 13010b57cec5SDimitry Andric } 13020b57cec5SDimitry Andric } 13030b57cec5SDimitry Andric } 13040b57cec5SDimitry Andric if (DyldInfoOnly || AddDyldInfo || 13050b57cec5SDimitry Andric HFlags & MachO::MH_NLIST_OUTOFSYNC_WITH_DYLDINFO) { 13060b57cec5SDimitry Andric unsigned ExportsAdded = 0; 13070b57cec5SDimitry Andric Error Err = Error::success(); 13080b57cec5SDimitry Andric for (const llvm::object::ExportEntry &Entry : MachO->exports(Err)) { 13090b57cec5SDimitry Andric bool found = false; 13100b57cec5SDimitry Andric bool ReExport = false; 13110b57cec5SDimitry Andric if (!DyldInfoOnly) { 13120b57cec5SDimitry Andric for (const NMSymbol &S : SymbolList) 13130b57cec5SDimitry Andric if (S.Address == Entry.address() + BaseSegmentAddress && 13140b57cec5SDimitry Andric S.Name == Entry.name()) { 13150b57cec5SDimitry Andric found = true; 13160b57cec5SDimitry Andric break; 13170b57cec5SDimitry Andric } 13180b57cec5SDimitry Andric } 13190b57cec5SDimitry Andric if (!found) { 13200b57cec5SDimitry Andric NMSymbol S = {}; 13210b57cec5SDimitry Andric S.Address = Entry.address() + BaseSegmentAddress; 13220b57cec5SDimitry Andric S.Size = 0; 13230b57cec5SDimitry Andric S.TypeChar = '\0'; 13240b57cec5SDimitry Andric S.Name = Entry.name(); 13250b57cec5SDimitry Andric // There is no symbol in the nlist symbol table for this so we set 13260b57cec5SDimitry Andric // Sym effectivly to null and the rest of code in here must test for 13270b57cec5SDimitry Andric // it and not do things like Sym.getFlags() for it. 13280b57cec5SDimitry Andric S.Sym = BasicSymbolRef(); 13290b57cec5SDimitry Andric S.SymFlags = SymbolRef::SF_Global; 13300b57cec5SDimitry Andric S.Section = SectionRef(); 13310b57cec5SDimitry Andric S.NType = 0; 13320b57cec5SDimitry Andric S.NSect = 0; 13330b57cec5SDimitry Andric S.NDesc = 0; 13340b57cec5SDimitry Andric S.IndirectName = StringRef(); 13350b57cec5SDimitry Andric 13360b57cec5SDimitry Andric uint64_t EFlags = Entry.flags(); 13370b57cec5SDimitry Andric bool Abs = ((EFlags & MachO::EXPORT_SYMBOL_FLAGS_KIND_MASK) == 13380b57cec5SDimitry Andric MachO::EXPORT_SYMBOL_FLAGS_KIND_ABSOLUTE); 13390b57cec5SDimitry Andric bool Resolver = (EFlags & 13400b57cec5SDimitry Andric MachO::EXPORT_SYMBOL_FLAGS_STUB_AND_RESOLVER); 13410b57cec5SDimitry Andric ReExport = (EFlags & MachO::EXPORT_SYMBOL_FLAGS_REEXPORT); 13420b57cec5SDimitry Andric bool WeakDef = (EFlags & MachO::EXPORT_SYMBOL_FLAGS_WEAK_DEFINITION); 13430b57cec5SDimitry Andric if (WeakDef) 13440b57cec5SDimitry Andric S.NDesc |= MachO::N_WEAK_DEF; 13450b57cec5SDimitry Andric if (Abs) { 13460b57cec5SDimitry Andric S.NType = MachO::N_EXT | MachO::N_ABS; 13470b57cec5SDimitry Andric S.TypeChar = 'A'; 13480b57cec5SDimitry Andric } else if (ReExport) { 13490b57cec5SDimitry Andric S.NType = MachO::N_EXT | MachO::N_INDR; 13500b57cec5SDimitry Andric S.TypeChar = 'I'; 13510b57cec5SDimitry Andric } else { 13520b57cec5SDimitry Andric S.NType = MachO::N_EXT | MachO::N_SECT; 13530b57cec5SDimitry Andric if (Resolver) { 13540b57cec5SDimitry Andric S.Address = Entry.other() + BaseSegmentAddress; 13550b57cec5SDimitry Andric if ((S.Address & 1) != 0 && 13560b57cec5SDimitry Andric !MachO->is64Bit() && H.cputype == MachO::CPU_TYPE_ARM){ 13570b57cec5SDimitry Andric S.Address &= ~1LL; 13580b57cec5SDimitry Andric S.NDesc |= MachO::N_ARM_THUMB_DEF; 13590b57cec5SDimitry Andric } 13600b57cec5SDimitry Andric } else { 13610b57cec5SDimitry Andric S.Address = Entry.address() + BaseSegmentAddress; 13620b57cec5SDimitry Andric } 13630b57cec5SDimitry Andric StringRef SegmentName = StringRef(); 13640b57cec5SDimitry Andric StringRef SectionName = StringRef(); 13650b57cec5SDimitry Andric for (const SectionRef &Section : MachO->sections()) { 13660b57cec5SDimitry Andric S.NSect++; 13678bcb0991SDimitry Andric 13688bcb0991SDimitry Andric if (Expected<StringRef> NameOrErr = Section.getName()) 13698bcb0991SDimitry Andric SectionName = *NameOrErr; 13708bcb0991SDimitry Andric else 13718bcb0991SDimitry Andric consumeError(NameOrErr.takeError()); 13728bcb0991SDimitry Andric 13730b57cec5SDimitry Andric SegmentName = MachO->getSectionFinalSegmentName( 13740b57cec5SDimitry Andric Section.getRawDataRefImpl()); 13750b57cec5SDimitry Andric if (S.Address >= Section.getAddress() && 13760b57cec5SDimitry Andric S.Address < Section.getAddress() + Section.getSize()) { 13770b57cec5SDimitry Andric S.Section = Section; 13780b57cec5SDimitry Andric break; 13790b57cec5SDimitry Andric } else if (Entry.name() == "__mh_execute_header" && 13800b57cec5SDimitry Andric SegmentName == "__TEXT" && SectionName == "__text") { 13810b57cec5SDimitry Andric S.Section = Section; 13820b57cec5SDimitry Andric S.NDesc |= MachO::REFERENCED_DYNAMICALLY; 13830b57cec5SDimitry Andric break; 13840b57cec5SDimitry Andric } 13850b57cec5SDimitry Andric } 13860b57cec5SDimitry Andric if (SegmentName == "__TEXT" && SectionName == "__text") 13870b57cec5SDimitry Andric S.TypeChar = 'T'; 13880b57cec5SDimitry Andric else if (SegmentName == "__DATA" && SectionName == "__data") 13890b57cec5SDimitry Andric S.TypeChar = 'D'; 13900b57cec5SDimitry Andric else if (SegmentName == "__DATA" && SectionName == "__bss") 13910b57cec5SDimitry Andric S.TypeChar = 'B'; 13920b57cec5SDimitry Andric else 13930b57cec5SDimitry Andric S.TypeChar = 'S'; 13940b57cec5SDimitry Andric } 13950b57cec5SDimitry Andric SymbolList.push_back(S); 13960b57cec5SDimitry Andric 13970b57cec5SDimitry Andric EOS << Entry.name(); 13980b57cec5SDimitry Andric EOS << '\0'; 13990b57cec5SDimitry Andric ExportsAdded++; 14000b57cec5SDimitry Andric 14010b57cec5SDimitry Andric // For ReExports there are a two more things to do, first add the 14020b57cec5SDimitry Andric // indirect name and second create the undefined symbol using the 14030b57cec5SDimitry Andric // referened dynamic library. 14040b57cec5SDimitry Andric if (ReExport) { 14050b57cec5SDimitry Andric 14060b57cec5SDimitry Andric // Add the indirect name. 14070b57cec5SDimitry Andric if (Entry.otherName().empty()) 14080b57cec5SDimitry Andric EOS << Entry.name(); 14090b57cec5SDimitry Andric else 14100b57cec5SDimitry Andric EOS << Entry.otherName(); 14110b57cec5SDimitry Andric EOS << '\0'; 14120b57cec5SDimitry Andric 14130b57cec5SDimitry Andric // Now create the undefined symbol using the referened dynamic 14140b57cec5SDimitry Andric // library. 14150b57cec5SDimitry Andric NMSymbol U = {}; 14160b57cec5SDimitry Andric U.Address = 0; 14170b57cec5SDimitry Andric U.Size = 0; 14180b57cec5SDimitry Andric U.TypeChar = 'U'; 14190b57cec5SDimitry Andric if (Entry.otherName().empty()) 14200b57cec5SDimitry Andric U.Name = Entry.name(); 14210b57cec5SDimitry Andric else 14220b57cec5SDimitry Andric U.Name = Entry.otherName(); 14230b57cec5SDimitry Andric // Again there is no symbol in the nlist symbol table for this so 14240b57cec5SDimitry Andric // we set Sym effectivly to null and the rest of code in here must 14250b57cec5SDimitry Andric // test for it and not do things like Sym.getFlags() for it. 14260b57cec5SDimitry Andric U.Sym = BasicSymbolRef(); 14270b57cec5SDimitry Andric U.SymFlags = SymbolRef::SF_Global | SymbolRef::SF_Undefined; 14280b57cec5SDimitry Andric U.Section = SectionRef(); 14290b57cec5SDimitry Andric U.NType = MachO::N_EXT | MachO::N_UNDF; 14300b57cec5SDimitry Andric U.NSect = 0; 14310b57cec5SDimitry Andric U.NDesc = 0; 14320b57cec5SDimitry Andric // The library ordinal for this undefined symbol is in the export 14330b57cec5SDimitry Andric // trie Entry.other(). 14340b57cec5SDimitry Andric MachO::SET_LIBRARY_ORDINAL(U.NDesc, Entry.other()); 14350b57cec5SDimitry Andric U.IndirectName = StringRef(); 14360b57cec5SDimitry Andric SymbolList.push_back(U); 14370b57cec5SDimitry Andric 14380b57cec5SDimitry Andric // Finally add the undefined symbol's name. 14390b57cec5SDimitry Andric if (Entry.otherName().empty()) 14400b57cec5SDimitry Andric EOS << Entry.name(); 14410b57cec5SDimitry Andric else 14420b57cec5SDimitry Andric EOS << Entry.otherName(); 14430b57cec5SDimitry Andric EOS << '\0'; 14440b57cec5SDimitry Andric ExportsAdded++; 14450b57cec5SDimitry Andric } 14460b57cec5SDimitry Andric } 14470b57cec5SDimitry Andric } 14480b57cec5SDimitry Andric if (Err) 14490b57cec5SDimitry Andric error(std::move(Err), MachO->getFileName()); 14500b57cec5SDimitry Andric // Set the symbol names and indirect names for the added symbols. 14510b57cec5SDimitry Andric if (ExportsAdded) { 14520b57cec5SDimitry Andric EOS.flush(); 14530b57cec5SDimitry Andric const char *Q = ExportsNameBuffer.c_str(); 14540b57cec5SDimitry Andric for (unsigned K = 0; K < ExportsAdded; K++) { 14550b57cec5SDimitry Andric SymbolList[I].Name = Q; 14560b57cec5SDimitry Andric Q += strlen(Q) + 1; 14570b57cec5SDimitry Andric if (SymbolList[I].TypeChar == 'I') { 14580b57cec5SDimitry Andric SymbolList[I].IndirectName = Q; 14590b57cec5SDimitry Andric Q += strlen(Q) + 1; 14600b57cec5SDimitry Andric } 14610b57cec5SDimitry Andric I++; 14620b57cec5SDimitry Andric } 14630b57cec5SDimitry Andric } 14640b57cec5SDimitry Andric 14650b57cec5SDimitry Andric // Add the undefined symbols from the bind entries. 14660b57cec5SDimitry Andric unsigned BindsAdded = 0; 14670b57cec5SDimitry Andric Error BErr = Error::success(); 14680b57cec5SDimitry Andric StringRef LastSymbolName = StringRef(); 14690b57cec5SDimitry Andric for (const llvm::object::MachOBindEntry &Entry : MachO->bindTable(BErr)) { 14700b57cec5SDimitry Andric bool found = false; 14710b57cec5SDimitry Andric if (LastSymbolName == Entry.symbolName()) 14720b57cec5SDimitry Andric found = true; 14730b57cec5SDimitry Andric else if(!DyldInfoOnly) { 14740b57cec5SDimitry Andric for (unsigned J = 0; J < SymbolList.size() && !found; ++J) { 14750b57cec5SDimitry Andric if (SymbolList[J].Name == Entry.symbolName()) 14760b57cec5SDimitry Andric found = true; 14770b57cec5SDimitry Andric } 14780b57cec5SDimitry Andric } 14790b57cec5SDimitry Andric if (!found) { 14800b57cec5SDimitry Andric LastSymbolName = Entry.symbolName(); 14810b57cec5SDimitry Andric NMSymbol B = {}; 14820b57cec5SDimitry Andric B.Address = 0; 14830b57cec5SDimitry Andric B.Size = 0; 14840b57cec5SDimitry Andric B.TypeChar = 'U'; 14850b57cec5SDimitry Andric // There is no symbol in the nlist symbol table for this so we set 14860b57cec5SDimitry Andric // Sym effectivly to null and the rest of code in here must test for 14870b57cec5SDimitry Andric // it and not do things like Sym.getFlags() for it. 14880b57cec5SDimitry Andric B.Sym = BasicSymbolRef(); 14890b57cec5SDimitry Andric B.SymFlags = SymbolRef::SF_Global | SymbolRef::SF_Undefined; 14900b57cec5SDimitry Andric B.NType = MachO::N_EXT | MachO::N_UNDF; 14910b57cec5SDimitry Andric B.NSect = 0; 14920b57cec5SDimitry Andric B.NDesc = 0; 14930b57cec5SDimitry Andric MachO::SET_LIBRARY_ORDINAL(B.NDesc, Entry.ordinal()); 14940b57cec5SDimitry Andric B.IndirectName = StringRef(); 14950b57cec5SDimitry Andric B.Name = Entry.symbolName(); 14960b57cec5SDimitry Andric SymbolList.push_back(B); 14970b57cec5SDimitry Andric BOS << Entry.symbolName(); 14980b57cec5SDimitry Andric BOS << '\0'; 14990b57cec5SDimitry Andric BindsAdded++; 15000b57cec5SDimitry Andric } 15010b57cec5SDimitry Andric } 15020b57cec5SDimitry Andric if (BErr) 15030b57cec5SDimitry Andric error(std::move(BErr), MachO->getFileName()); 15040b57cec5SDimitry Andric // Set the symbol names and indirect names for the added symbols. 15050b57cec5SDimitry Andric if (BindsAdded) { 15060b57cec5SDimitry Andric BOS.flush(); 15070b57cec5SDimitry Andric const char *Q = BindsNameBuffer.c_str(); 15080b57cec5SDimitry Andric for (unsigned K = 0; K < BindsAdded; K++) { 15090b57cec5SDimitry Andric SymbolList[I].Name = Q; 15100b57cec5SDimitry Andric Q += strlen(Q) + 1; 15110b57cec5SDimitry Andric if (SymbolList[I].TypeChar == 'I') { 15120b57cec5SDimitry Andric SymbolList[I].IndirectName = Q; 15130b57cec5SDimitry Andric Q += strlen(Q) + 1; 15140b57cec5SDimitry Andric } 15150b57cec5SDimitry Andric I++; 15160b57cec5SDimitry Andric } 15170b57cec5SDimitry Andric } 15180b57cec5SDimitry Andric 15190b57cec5SDimitry Andric // Add the undefined symbols from the lazy bind entries. 15200b57cec5SDimitry Andric unsigned LazysAdded = 0; 15210b57cec5SDimitry Andric Error LErr = Error::success(); 15220b57cec5SDimitry Andric LastSymbolName = StringRef(); 15230b57cec5SDimitry Andric for (const llvm::object::MachOBindEntry &Entry : 15240b57cec5SDimitry Andric MachO->lazyBindTable(LErr)) { 15250b57cec5SDimitry Andric bool found = false; 15260b57cec5SDimitry Andric if (LastSymbolName == Entry.symbolName()) 15270b57cec5SDimitry Andric found = true; 15280b57cec5SDimitry Andric else { 15290b57cec5SDimitry Andric // Here we must check to see it this symbol is already in the 15300b57cec5SDimitry Andric // SymbolList as it might have already have been added above via a 15310b57cec5SDimitry Andric // non-lazy (bind) entry. 15320b57cec5SDimitry Andric for (unsigned J = 0; J < SymbolList.size() && !found; ++J) { 15330b57cec5SDimitry Andric if (SymbolList[J].Name == Entry.symbolName()) 15340b57cec5SDimitry Andric found = true; 15350b57cec5SDimitry Andric } 15360b57cec5SDimitry Andric } 15370b57cec5SDimitry Andric if (!found) { 15380b57cec5SDimitry Andric LastSymbolName = Entry.symbolName(); 15390b57cec5SDimitry Andric NMSymbol L = {}; 15400b57cec5SDimitry Andric L.Name = Entry.symbolName(); 15410b57cec5SDimitry Andric L.Address = 0; 15420b57cec5SDimitry Andric L.Size = 0; 15430b57cec5SDimitry Andric L.TypeChar = 'U'; 15440b57cec5SDimitry Andric // There is no symbol in the nlist symbol table for this so we set 15450b57cec5SDimitry Andric // Sym effectivly to null and the rest of code in here must test for 15460b57cec5SDimitry Andric // it and not do things like Sym.getFlags() for it. 15470b57cec5SDimitry Andric L.Sym = BasicSymbolRef(); 15480b57cec5SDimitry Andric L.SymFlags = SymbolRef::SF_Global | SymbolRef::SF_Undefined; 15490b57cec5SDimitry Andric L.NType = MachO::N_EXT | MachO::N_UNDF; 15500b57cec5SDimitry Andric L.NSect = 0; 15510b57cec5SDimitry Andric // The REFERENCE_FLAG_UNDEFINED_LAZY is no longer used but here it 15520b57cec5SDimitry Andric // makes sence since we are creating this from a lazy bind entry. 15530b57cec5SDimitry Andric L.NDesc = MachO::REFERENCE_FLAG_UNDEFINED_LAZY; 15540b57cec5SDimitry Andric MachO::SET_LIBRARY_ORDINAL(L.NDesc, Entry.ordinal()); 15550b57cec5SDimitry Andric L.IndirectName = StringRef(); 15560b57cec5SDimitry Andric SymbolList.push_back(L); 15570b57cec5SDimitry Andric LOS << Entry.symbolName(); 15580b57cec5SDimitry Andric LOS << '\0'; 15590b57cec5SDimitry Andric LazysAdded++; 15600b57cec5SDimitry Andric } 15610b57cec5SDimitry Andric } 15620b57cec5SDimitry Andric if (LErr) 15630b57cec5SDimitry Andric error(std::move(LErr), MachO->getFileName()); 15640b57cec5SDimitry Andric // Set the symbol names and indirect names for the added symbols. 15650b57cec5SDimitry Andric if (LazysAdded) { 15660b57cec5SDimitry Andric LOS.flush(); 15670b57cec5SDimitry Andric const char *Q = LazysNameBuffer.c_str(); 15680b57cec5SDimitry Andric for (unsigned K = 0; K < LazysAdded; K++) { 15690b57cec5SDimitry Andric SymbolList[I].Name = Q; 15700b57cec5SDimitry Andric Q += strlen(Q) + 1; 15710b57cec5SDimitry Andric if (SymbolList[I].TypeChar == 'I') { 15720b57cec5SDimitry Andric SymbolList[I].IndirectName = Q; 15730b57cec5SDimitry Andric Q += strlen(Q) + 1; 15740b57cec5SDimitry Andric } 15750b57cec5SDimitry Andric I++; 15760b57cec5SDimitry Andric } 15770b57cec5SDimitry Andric } 15780b57cec5SDimitry Andric 15790b57cec5SDimitry Andric // Add the undefineds symbol from the weak bind entries which are not 15800b57cec5SDimitry Andric // strong symbols. 15810b57cec5SDimitry Andric unsigned WeaksAdded = 0; 15820b57cec5SDimitry Andric Error WErr = Error::success(); 15830b57cec5SDimitry Andric LastSymbolName = StringRef(); 15840b57cec5SDimitry Andric for (const llvm::object::MachOBindEntry &Entry : 15850b57cec5SDimitry Andric MachO->weakBindTable(WErr)) { 15860b57cec5SDimitry Andric bool found = false; 15870b57cec5SDimitry Andric unsigned J = 0; 15880b57cec5SDimitry Andric if (LastSymbolName == Entry.symbolName() || 15890b57cec5SDimitry Andric Entry.flags() & MachO::BIND_SYMBOL_FLAGS_NON_WEAK_DEFINITION) { 15900b57cec5SDimitry Andric found = true; 15910b57cec5SDimitry Andric } else { 15920b57cec5SDimitry Andric for (J = 0; J < SymbolList.size() && !found; ++J) { 15930b57cec5SDimitry Andric if (SymbolList[J].Name == Entry.symbolName()) { 15940b57cec5SDimitry Andric found = true; 15950b57cec5SDimitry Andric break; 15960b57cec5SDimitry Andric } 15970b57cec5SDimitry Andric } 15980b57cec5SDimitry Andric } 15990b57cec5SDimitry Andric if (!found) { 16000b57cec5SDimitry Andric LastSymbolName = Entry.symbolName(); 16010b57cec5SDimitry Andric NMSymbol W; 16020b57cec5SDimitry Andric memset(&W, '\0', sizeof(NMSymbol)); 16030b57cec5SDimitry Andric W.Name = Entry.symbolName(); 16040b57cec5SDimitry Andric W.Address = 0; 16050b57cec5SDimitry Andric W.Size = 0; 16060b57cec5SDimitry Andric W.TypeChar = 'U'; 16070b57cec5SDimitry Andric // There is no symbol in the nlist symbol table for this so we set 16080b57cec5SDimitry Andric // Sym effectivly to null and the rest of code in here must test for 16090b57cec5SDimitry Andric // it and not do things like Sym.getFlags() for it. 16100b57cec5SDimitry Andric W.Sym = BasicSymbolRef(); 16110b57cec5SDimitry Andric W.SymFlags = SymbolRef::SF_Global | SymbolRef::SF_Undefined; 16120b57cec5SDimitry Andric W.NType = MachO::N_EXT | MachO::N_UNDF; 16130b57cec5SDimitry Andric W.NSect = 0; 16140b57cec5SDimitry Andric // Odd that we are using N_WEAK_DEF on an undefined symbol but that is 16150b57cec5SDimitry Andric // what is created in this case by the linker when there are real 16160b57cec5SDimitry Andric // symbols in the nlist structs. 16170b57cec5SDimitry Andric W.NDesc = MachO::N_WEAK_DEF; 16180b57cec5SDimitry Andric W.IndirectName = StringRef(); 16190b57cec5SDimitry Andric SymbolList.push_back(W); 16200b57cec5SDimitry Andric WOS << Entry.symbolName(); 16210b57cec5SDimitry Andric WOS << '\0'; 16220b57cec5SDimitry Andric WeaksAdded++; 16230b57cec5SDimitry Andric } else { 16240b57cec5SDimitry Andric // This is the case the symbol was previously been found and it could 16250b57cec5SDimitry Andric // have been added from a bind or lazy bind symbol. If so and not 16260b57cec5SDimitry Andric // a definition also mark it as weak. 16270b57cec5SDimitry Andric if (SymbolList[J].TypeChar == 'U') 16280b57cec5SDimitry Andric // See comment above about N_WEAK_DEF. 16290b57cec5SDimitry Andric SymbolList[J].NDesc |= MachO::N_WEAK_DEF; 16300b57cec5SDimitry Andric } 16310b57cec5SDimitry Andric } 16320b57cec5SDimitry Andric if (WErr) 16330b57cec5SDimitry Andric error(std::move(WErr), MachO->getFileName()); 16340b57cec5SDimitry Andric // Set the symbol names and indirect names for the added symbols. 16350b57cec5SDimitry Andric if (WeaksAdded) { 16360b57cec5SDimitry Andric WOS.flush(); 16370b57cec5SDimitry Andric const char *Q = WeaksNameBuffer.c_str(); 16380b57cec5SDimitry Andric for (unsigned K = 0; K < WeaksAdded; K++) { 16390b57cec5SDimitry Andric SymbolList[I].Name = Q; 16400b57cec5SDimitry Andric Q += strlen(Q) + 1; 16410b57cec5SDimitry Andric if (SymbolList[I].TypeChar == 'I') { 16420b57cec5SDimitry Andric SymbolList[I].IndirectName = Q; 16430b57cec5SDimitry Andric Q += strlen(Q) + 1; 16440b57cec5SDimitry Andric } 16450b57cec5SDimitry Andric I++; 16460b57cec5SDimitry Andric } 16470b57cec5SDimitry Andric } 16480b57cec5SDimitry Andric 16490b57cec5SDimitry Andric // Trying adding symbol from the function starts table and LC_MAIN entry 16500b57cec5SDimitry Andric // point. 16510b57cec5SDimitry Andric SmallVector<uint64_t, 8> FoundFns; 16520b57cec5SDimitry Andric uint64_t lc_main_offset = UINT64_MAX; 16530b57cec5SDimitry Andric for (const auto &Command : MachO->load_commands()) { 16540b57cec5SDimitry Andric if (Command.C.cmd == MachO::LC_FUNCTION_STARTS) { 16550b57cec5SDimitry Andric // We found a function starts segment, parse the addresses for 16560b57cec5SDimitry Andric // consumption. 16570b57cec5SDimitry Andric MachO::linkedit_data_command LLC = 16580b57cec5SDimitry Andric MachO->getLinkeditDataLoadCommand(Command); 16590b57cec5SDimitry Andric 16600b57cec5SDimitry Andric MachO->ReadULEB128s(LLC.dataoff, FoundFns); 16610b57cec5SDimitry Andric } else if (Command.C.cmd == MachO::LC_MAIN) { 16620b57cec5SDimitry Andric MachO::entry_point_command LCmain = 16630b57cec5SDimitry Andric MachO->getEntryPointCommand(Command); 16640b57cec5SDimitry Andric lc_main_offset = LCmain.entryoff; 16650b57cec5SDimitry Andric } 16660b57cec5SDimitry Andric } 16670b57cec5SDimitry Andric // See if these addresses are already in the symbol table. 16680b57cec5SDimitry Andric unsigned FunctionStartsAdded = 0; 16690b57cec5SDimitry Andric for (uint64_t f = 0; f < FoundFns.size(); f++) { 16700b57cec5SDimitry Andric bool found = false; 16710b57cec5SDimitry Andric for (unsigned J = 0; J < SymbolList.size() && !found; ++J) { 16720b57cec5SDimitry Andric if (SymbolList[J].Address == FoundFns[f] + BaseSegmentAddress) 16730b57cec5SDimitry Andric found = true; 16740b57cec5SDimitry Andric } 16750b57cec5SDimitry Andric // See this address is not already in the symbol table fake up an 16760b57cec5SDimitry Andric // nlist for it. 16770b57cec5SDimitry Andric if (!found) { 16780b57cec5SDimitry Andric NMSymbol F = {}; 16790b57cec5SDimitry Andric F.Name = "<redacted function X>"; 16800b57cec5SDimitry Andric F.Address = FoundFns[f] + BaseSegmentAddress; 16810b57cec5SDimitry Andric F.Size = 0; 16820b57cec5SDimitry Andric // There is no symbol in the nlist symbol table for this so we set 16830b57cec5SDimitry Andric // Sym effectivly to null and the rest of code in here must test for 16840b57cec5SDimitry Andric // it and not do things like Sym.getFlags() for it. 16850b57cec5SDimitry Andric F.Sym = BasicSymbolRef(); 16860b57cec5SDimitry Andric F.SymFlags = 0; 16870b57cec5SDimitry Andric F.NType = MachO::N_SECT; 16880b57cec5SDimitry Andric F.NSect = 0; 16890b57cec5SDimitry Andric StringRef SegmentName = StringRef(); 16900b57cec5SDimitry Andric StringRef SectionName = StringRef(); 16910b57cec5SDimitry Andric for (const SectionRef &Section : MachO->sections()) { 16928bcb0991SDimitry Andric if (Expected<StringRef> NameOrErr = Section.getName()) 16938bcb0991SDimitry Andric SectionName = *NameOrErr; 16948bcb0991SDimitry Andric else 16958bcb0991SDimitry Andric consumeError(NameOrErr.takeError()); 16968bcb0991SDimitry Andric 16970b57cec5SDimitry Andric SegmentName = MachO->getSectionFinalSegmentName( 16980b57cec5SDimitry Andric Section.getRawDataRefImpl()); 16990b57cec5SDimitry Andric F.NSect++; 17000b57cec5SDimitry Andric if (F.Address >= Section.getAddress() && 17010b57cec5SDimitry Andric F.Address < Section.getAddress() + Section.getSize()) { 17020b57cec5SDimitry Andric F.Section = Section; 17030b57cec5SDimitry Andric break; 17040b57cec5SDimitry Andric } 17050b57cec5SDimitry Andric } 17060b57cec5SDimitry Andric if (SegmentName == "__TEXT" && SectionName == "__text") 17070b57cec5SDimitry Andric F.TypeChar = 't'; 17080b57cec5SDimitry Andric else if (SegmentName == "__DATA" && SectionName == "__data") 17090b57cec5SDimitry Andric F.TypeChar = 'd'; 17100b57cec5SDimitry Andric else if (SegmentName == "__DATA" && SectionName == "__bss") 17110b57cec5SDimitry Andric F.TypeChar = 'b'; 17120b57cec5SDimitry Andric else 17130b57cec5SDimitry Andric F.TypeChar = 's'; 17140b57cec5SDimitry Andric F.NDesc = 0; 17150b57cec5SDimitry Andric F.IndirectName = StringRef(); 17160b57cec5SDimitry Andric SymbolList.push_back(F); 17170b57cec5SDimitry Andric if (FoundFns[f] == lc_main_offset) 17180b57cec5SDimitry Andric FOS << "<redacted LC_MAIN>"; 17190b57cec5SDimitry Andric else 17200b57cec5SDimitry Andric FOS << "<redacted function " << f << ">"; 17210b57cec5SDimitry Andric FOS << '\0'; 17220b57cec5SDimitry Andric FunctionStartsAdded++; 17230b57cec5SDimitry Andric } 17240b57cec5SDimitry Andric } 17250b57cec5SDimitry Andric if (FunctionStartsAdded) { 17260b57cec5SDimitry Andric FOS.flush(); 17270b57cec5SDimitry Andric const char *Q = FunctionStartsNameBuffer.c_str(); 17280b57cec5SDimitry Andric for (unsigned K = 0; K < FunctionStartsAdded; K++) { 17290b57cec5SDimitry Andric SymbolList[I].Name = Q; 17300b57cec5SDimitry Andric Q += strlen(Q) + 1; 17310b57cec5SDimitry Andric if (SymbolList[I].TypeChar == 'I') { 17320b57cec5SDimitry Andric SymbolList[I].IndirectName = Q; 17330b57cec5SDimitry Andric Q += strlen(Q) + 1; 17340b57cec5SDimitry Andric } 17350b57cec5SDimitry Andric I++; 17360b57cec5SDimitry Andric } 17370b57cec5SDimitry Andric } 17380b57cec5SDimitry Andric } 17390b57cec5SDimitry Andric } 17400b57cec5SDimitry Andric 17410b57cec5SDimitry Andric CurrentFilename = Obj.getFileName(); 17420b57cec5SDimitry Andric sortAndPrintSymbolList(Obj, printName, ArchiveName, ArchitectureName); 17430b57cec5SDimitry Andric } 17440b57cec5SDimitry Andric 17450b57cec5SDimitry Andric // checkMachOAndArchFlags() checks to see if the SymbolicFile is a Mach-O file 17460b57cec5SDimitry Andric // and if it is and there is a list of architecture flags is specified then 17470b57cec5SDimitry Andric // check to make sure this Mach-O file is one of those architectures or all 17480b57cec5SDimitry Andric // architectures was specificed. If not then an error is generated and this 17490b57cec5SDimitry Andric // routine returns false. Else it returns true. 17500b57cec5SDimitry Andric static bool checkMachOAndArchFlags(SymbolicFile *O, std::string &Filename) { 17510b57cec5SDimitry Andric auto *MachO = dyn_cast<MachOObjectFile>(O); 17520b57cec5SDimitry Andric 17530b57cec5SDimitry Andric if (!MachO || ArchAll || ArchFlags.empty()) 17540b57cec5SDimitry Andric return true; 17550b57cec5SDimitry Andric 17560b57cec5SDimitry Andric MachO::mach_header H; 17570b57cec5SDimitry Andric MachO::mach_header_64 H_64; 17580b57cec5SDimitry Andric Triple T; 17590b57cec5SDimitry Andric const char *McpuDefault, *ArchFlag; 17600b57cec5SDimitry Andric if (MachO->is64Bit()) { 17610b57cec5SDimitry Andric H_64 = MachO->MachOObjectFile::getHeader64(); 17620b57cec5SDimitry Andric T = MachOObjectFile::getArchTriple(H_64.cputype, H_64.cpusubtype, 17630b57cec5SDimitry Andric &McpuDefault, &ArchFlag); 17640b57cec5SDimitry Andric } else { 17650b57cec5SDimitry Andric H = MachO->MachOObjectFile::getHeader(); 17660b57cec5SDimitry Andric T = MachOObjectFile::getArchTriple(H.cputype, H.cpusubtype, 17670b57cec5SDimitry Andric &McpuDefault, &ArchFlag); 17680b57cec5SDimitry Andric } 17690b57cec5SDimitry Andric const std::string ArchFlagName(ArchFlag); 17700b57cec5SDimitry Andric if (none_of(ArchFlags, [&](const std::string &Name) { 17710b57cec5SDimitry Andric return Name == ArchFlagName; 17720b57cec5SDimitry Andric })) { 17730b57cec5SDimitry Andric error("No architecture specified", Filename); 17740b57cec5SDimitry Andric return false; 17750b57cec5SDimitry Andric } 17760b57cec5SDimitry Andric return true; 17770b57cec5SDimitry Andric } 17780b57cec5SDimitry Andric 17790b57cec5SDimitry Andric static void dumpSymbolNamesFromFile(std::string &Filename) { 17800b57cec5SDimitry Andric ErrorOr<std::unique_ptr<MemoryBuffer>> BufferOrErr = 17810b57cec5SDimitry Andric MemoryBuffer::getFileOrSTDIN(Filename); 17820b57cec5SDimitry Andric if (error(BufferOrErr.getError(), Filename)) 17830b57cec5SDimitry Andric return; 17840b57cec5SDimitry Andric 17850b57cec5SDimitry Andric LLVMContext Context; 17860b57cec5SDimitry Andric LLVMContext *ContextPtr = NoLLVMBitcode ? nullptr : &Context; 17870b57cec5SDimitry Andric Expected<std::unique_ptr<Binary>> BinaryOrErr = 17880b57cec5SDimitry Andric createBinary(BufferOrErr.get()->getMemBufferRef(), ContextPtr); 17890b57cec5SDimitry Andric if (!BinaryOrErr) { 17900b57cec5SDimitry Andric error(BinaryOrErr.takeError(), Filename); 17910b57cec5SDimitry Andric return; 17920b57cec5SDimitry Andric } 17930b57cec5SDimitry Andric Binary &Bin = *BinaryOrErr.get(); 17940b57cec5SDimitry Andric 17950b57cec5SDimitry Andric if (Archive *A = dyn_cast<Archive>(&Bin)) { 17960b57cec5SDimitry Andric if (ArchiveMap) { 17970b57cec5SDimitry Andric Archive::symbol_iterator I = A->symbol_begin(); 17980b57cec5SDimitry Andric Archive::symbol_iterator E = A->symbol_end(); 17990b57cec5SDimitry Andric if (I != E) { 18000b57cec5SDimitry Andric outs() << "Archive map\n"; 18010b57cec5SDimitry Andric for (; I != E; ++I) { 18020b57cec5SDimitry Andric Expected<Archive::Child> C = I->getMember(); 18030b57cec5SDimitry Andric if (!C) { 18040b57cec5SDimitry Andric error(C.takeError(), Filename); 18050b57cec5SDimitry Andric break; 18060b57cec5SDimitry Andric } 18070b57cec5SDimitry Andric Expected<StringRef> FileNameOrErr = C->getName(); 18080b57cec5SDimitry Andric if (!FileNameOrErr) { 18090b57cec5SDimitry Andric error(FileNameOrErr.takeError(), Filename); 18100b57cec5SDimitry Andric break; 18110b57cec5SDimitry Andric } 18120b57cec5SDimitry Andric StringRef SymName = I->getName(); 18130b57cec5SDimitry Andric outs() << SymName << " in " << FileNameOrErr.get() << "\n"; 18140b57cec5SDimitry Andric } 18150b57cec5SDimitry Andric outs() << "\n"; 18160b57cec5SDimitry Andric } 18170b57cec5SDimitry Andric } 18180b57cec5SDimitry Andric 18190b57cec5SDimitry Andric { 18200b57cec5SDimitry Andric Error Err = Error::success(); 18210b57cec5SDimitry Andric for (auto &C : A->children(Err)) { 18220b57cec5SDimitry Andric Expected<std::unique_ptr<Binary>> ChildOrErr = 18230b57cec5SDimitry Andric C.getAsBinary(ContextPtr); 18240b57cec5SDimitry Andric if (!ChildOrErr) { 18250b57cec5SDimitry Andric if (auto E = isNotObjectErrorInvalidFileType(ChildOrErr.takeError())) 18260b57cec5SDimitry Andric error(std::move(E), Filename, C); 18270b57cec5SDimitry Andric continue; 18280b57cec5SDimitry Andric } 18290b57cec5SDimitry Andric if (SymbolicFile *O = dyn_cast<SymbolicFile>(&*ChildOrErr.get())) { 18300b57cec5SDimitry Andric if (!MachOPrintSizeWarning && PrintSize && isa<MachOObjectFile>(O)) { 18310b57cec5SDimitry Andric WithColor::warning(errs(), ToolName) 18320b57cec5SDimitry Andric << "sizes with -print-size for Mach-O files are always zero.\n"; 18330b57cec5SDimitry Andric MachOPrintSizeWarning = true; 18340b57cec5SDimitry Andric } 18350b57cec5SDimitry Andric if (!checkMachOAndArchFlags(O, Filename)) 18360b57cec5SDimitry Andric return; 18370b57cec5SDimitry Andric if (!PrintFileName) { 18380b57cec5SDimitry Andric outs() << "\n"; 18390b57cec5SDimitry Andric if (isa<MachOObjectFile>(O)) { 18400b57cec5SDimitry Andric outs() << Filename << "(" << O->getFileName() << ")"; 18410b57cec5SDimitry Andric } else 18420b57cec5SDimitry Andric outs() << O->getFileName(); 18430b57cec5SDimitry Andric outs() << ":\n"; 18440b57cec5SDimitry Andric } 18450b57cec5SDimitry Andric dumpSymbolNamesFromObject(*O, false, Filename); 18460b57cec5SDimitry Andric } 18470b57cec5SDimitry Andric } 18480b57cec5SDimitry Andric if (Err) 18490b57cec5SDimitry Andric error(std::move(Err), A->getFileName()); 18500b57cec5SDimitry Andric } 18510b57cec5SDimitry Andric return; 18520b57cec5SDimitry Andric } 18530b57cec5SDimitry Andric if (MachOUniversalBinary *UB = dyn_cast<MachOUniversalBinary>(&Bin)) { 18540b57cec5SDimitry Andric // If we have a list of architecture flags specified dump only those. 18550b57cec5SDimitry Andric if (!ArchAll && !ArchFlags.empty()) { 18560b57cec5SDimitry Andric // Look for a slice in the universal binary that matches each ArchFlag. 18570b57cec5SDimitry Andric bool ArchFound; 18580b57cec5SDimitry Andric for (unsigned i = 0; i < ArchFlags.size(); ++i) { 18590b57cec5SDimitry Andric ArchFound = false; 18600b57cec5SDimitry Andric for (MachOUniversalBinary::object_iterator I = UB->begin_objects(), 18610b57cec5SDimitry Andric E = UB->end_objects(); 18620b57cec5SDimitry Andric I != E; ++I) { 18630b57cec5SDimitry Andric if (ArchFlags[i] == I->getArchFlagName()) { 18640b57cec5SDimitry Andric ArchFound = true; 18650b57cec5SDimitry Andric Expected<std::unique_ptr<ObjectFile>> ObjOrErr = 18660b57cec5SDimitry Andric I->getAsObjectFile(); 18670b57cec5SDimitry Andric std::string ArchiveName; 18680b57cec5SDimitry Andric std::string ArchitectureName; 18690b57cec5SDimitry Andric ArchiveName.clear(); 18700b57cec5SDimitry Andric ArchitectureName.clear(); 18710b57cec5SDimitry Andric if (ObjOrErr) { 18720b57cec5SDimitry Andric ObjectFile &Obj = *ObjOrErr.get(); 18730b57cec5SDimitry Andric if (ArchFlags.size() > 1) { 18740b57cec5SDimitry Andric if (PrintFileName) 18750b57cec5SDimitry Andric ArchitectureName = I->getArchFlagName(); 18760b57cec5SDimitry Andric else 18770b57cec5SDimitry Andric outs() << "\n" << Obj.getFileName() << " (for architecture " 18780b57cec5SDimitry Andric << I->getArchFlagName() << ")" 18790b57cec5SDimitry Andric << ":\n"; 18800b57cec5SDimitry Andric } 18810b57cec5SDimitry Andric dumpSymbolNamesFromObject(Obj, false, ArchiveName, 18820b57cec5SDimitry Andric ArchitectureName); 18830b57cec5SDimitry Andric } else if (auto E = isNotObjectErrorInvalidFileType( 18840b57cec5SDimitry Andric ObjOrErr.takeError())) { 18850b57cec5SDimitry Andric error(std::move(E), Filename, ArchFlags.size() > 1 ? 18860b57cec5SDimitry Andric StringRef(I->getArchFlagName()) : StringRef()); 18870b57cec5SDimitry Andric continue; 18880b57cec5SDimitry Andric } else if (Expected<std::unique_ptr<Archive>> AOrErr = 18890b57cec5SDimitry Andric I->getAsArchive()) { 18900b57cec5SDimitry Andric std::unique_ptr<Archive> &A = *AOrErr; 18910b57cec5SDimitry Andric Error Err = Error::success(); 18920b57cec5SDimitry Andric for (auto &C : A->children(Err)) { 18930b57cec5SDimitry Andric Expected<std::unique_ptr<Binary>> ChildOrErr = 18940b57cec5SDimitry Andric C.getAsBinary(ContextPtr); 18950b57cec5SDimitry Andric if (!ChildOrErr) { 18960b57cec5SDimitry Andric if (auto E = isNotObjectErrorInvalidFileType( 18970b57cec5SDimitry Andric ChildOrErr.takeError())) { 18980b57cec5SDimitry Andric error(std::move(E), Filename, C, ArchFlags.size() > 1 ? 18990b57cec5SDimitry Andric StringRef(I->getArchFlagName()) : StringRef()); 19000b57cec5SDimitry Andric } 19010b57cec5SDimitry Andric continue; 19020b57cec5SDimitry Andric } 19030b57cec5SDimitry Andric if (SymbolicFile *O = 19040b57cec5SDimitry Andric dyn_cast<SymbolicFile>(&*ChildOrErr.get())) { 19050b57cec5SDimitry Andric if (PrintFileName) { 19060b57cec5SDimitry Andric ArchiveName = A->getFileName(); 19070b57cec5SDimitry Andric if (ArchFlags.size() > 1) 19080b57cec5SDimitry Andric ArchitectureName = I->getArchFlagName(); 19090b57cec5SDimitry Andric } else { 19100b57cec5SDimitry Andric outs() << "\n" << A->getFileName(); 19110b57cec5SDimitry Andric outs() << "(" << O->getFileName() << ")"; 19120b57cec5SDimitry Andric if (ArchFlags.size() > 1) { 19130b57cec5SDimitry Andric outs() << " (for architecture " << I->getArchFlagName() 19140b57cec5SDimitry Andric << ")"; 19150b57cec5SDimitry Andric } 19160b57cec5SDimitry Andric outs() << ":\n"; 19170b57cec5SDimitry Andric } 19180b57cec5SDimitry Andric dumpSymbolNamesFromObject(*O, false, ArchiveName, 19190b57cec5SDimitry Andric ArchitectureName); 19200b57cec5SDimitry Andric } 19210b57cec5SDimitry Andric } 19220b57cec5SDimitry Andric if (Err) 19230b57cec5SDimitry Andric error(std::move(Err), A->getFileName()); 19240b57cec5SDimitry Andric } else { 19250b57cec5SDimitry Andric consumeError(AOrErr.takeError()); 19260b57cec5SDimitry Andric error(Filename + " for architecture " + 19270b57cec5SDimitry Andric StringRef(I->getArchFlagName()) + 19280b57cec5SDimitry Andric " is not a Mach-O file or an archive file", 19290b57cec5SDimitry Andric "Mach-O universal file"); 19300b57cec5SDimitry Andric } 19310b57cec5SDimitry Andric } 19320b57cec5SDimitry Andric } 19330b57cec5SDimitry Andric if (!ArchFound) { 19340b57cec5SDimitry Andric error(ArchFlags[i], 19350b57cec5SDimitry Andric "file: " + Filename + " does not contain architecture"); 19360b57cec5SDimitry Andric return; 19370b57cec5SDimitry Andric } 19380b57cec5SDimitry Andric } 19390b57cec5SDimitry Andric return; 19400b57cec5SDimitry Andric } 19410b57cec5SDimitry Andric // No architecture flags were specified so if this contains a slice that 19420b57cec5SDimitry Andric // matches the host architecture dump only that. 19430b57cec5SDimitry Andric if (!ArchAll) { 19440b57cec5SDimitry Andric Triple HostTriple = MachOObjectFile::getHostArch(); 19450b57cec5SDimitry Andric StringRef HostArchName = HostTriple.getArchName(); 19460b57cec5SDimitry Andric for (MachOUniversalBinary::object_iterator I = UB->begin_objects(), 19470b57cec5SDimitry Andric E = UB->end_objects(); 19480b57cec5SDimitry Andric I != E; ++I) { 19490b57cec5SDimitry Andric if (HostArchName == I->getArchFlagName()) { 19500b57cec5SDimitry Andric Expected<std::unique_ptr<ObjectFile>> ObjOrErr = I->getAsObjectFile(); 19510b57cec5SDimitry Andric std::string ArchiveName; 19520b57cec5SDimitry Andric if (ObjOrErr) { 19530b57cec5SDimitry Andric ObjectFile &Obj = *ObjOrErr.get(); 19540b57cec5SDimitry Andric dumpSymbolNamesFromObject(Obj, false); 19550b57cec5SDimitry Andric } else if (auto E = isNotObjectErrorInvalidFileType( 19560b57cec5SDimitry Andric ObjOrErr.takeError())) { 19570b57cec5SDimitry Andric error(std::move(E), Filename); 19580b57cec5SDimitry Andric return; 19590b57cec5SDimitry Andric } else if (Expected<std::unique_ptr<Archive>> AOrErr = 19600b57cec5SDimitry Andric I->getAsArchive()) { 19610b57cec5SDimitry Andric std::unique_ptr<Archive> &A = *AOrErr; 19620b57cec5SDimitry Andric Error Err = Error::success(); 19630b57cec5SDimitry Andric for (auto &C : A->children(Err)) { 19640b57cec5SDimitry Andric Expected<std::unique_ptr<Binary>> ChildOrErr = 19650b57cec5SDimitry Andric C.getAsBinary(ContextPtr); 19660b57cec5SDimitry Andric if (!ChildOrErr) { 19670b57cec5SDimitry Andric if (auto E = isNotObjectErrorInvalidFileType( 19680b57cec5SDimitry Andric ChildOrErr.takeError())) 19690b57cec5SDimitry Andric error(std::move(E), Filename, C); 19700b57cec5SDimitry Andric continue; 19710b57cec5SDimitry Andric } 19720b57cec5SDimitry Andric if (SymbolicFile *O = 19730b57cec5SDimitry Andric dyn_cast<SymbolicFile>(&*ChildOrErr.get())) { 19740b57cec5SDimitry Andric if (PrintFileName) 19750b57cec5SDimitry Andric ArchiveName = A->getFileName(); 19760b57cec5SDimitry Andric else 19770b57cec5SDimitry Andric outs() << "\n" << A->getFileName() << "(" << O->getFileName() 19780b57cec5SDimitry Andric << ")" 19790b57cec5SDimitry Andric << ":\n"; 19800b57cec5SDimitry Andric dumpSymbolNamesFromObject(*O, false, ArchiveName); 19810b57cec5SDimitry Andric } 19820b57cec5SDimitry Andric } 19830b57cec5SDimitry Andric if (Err) 19840b57cec5SDimitry Andric error(std::move(Err), A->getFileName()); 19850b57cec5SDimitry Andric } else { 19860b57cec5SDimitry Andric consumeError(AOrErr.takeError()); 19870b57cec5SDimitry Andric error(Filename + " for architecture " + 19880b57cec5SDimitry Andric StringRef(I->getArchFlagName()) + 19890b57cec5SDimitry Andric " is not a Mach-O file or an archive file", 19900b57cec5SDimitry Andric "Mach-O universal file"); 19910b57cec5SDimitry Andric } 19920b57cec5SDimitry Andric return; 19930b57cec5SDimitry Andric } 19940b57cec5SDimitry Andric } 19950b57cec5SDimitry Andric } 19960b57cec5SDimitry Andric // Either all architectures have been specified or none have been specified 19970b57cec5SDimitry Andric // and this does not contain the host architecture so dump all the slices. 19980b57cec5SDimitry Andric bool moreThanOneArch = UB->getNumberOfObjects() > 1; 19990b57cec5SDimitry Andric for (const MachOUniversalBinary::ObjectForArch &O : UB->objects()) { 20000b57cec5SDimitry Andric Expected<std::unique_ptr<ObjectFile>> ObjOrErr = O.getAsObjectFile(); 20010b57cec5SDimitry Andric std::string ArchiveName; 20020b57cec5SDimitry Andric std::string ArchitectureName; 20030b57cec5SDimitry Andric ArchiveName.clear(); 20040b57cec5SDimitry Andric ArchitectureName.clear(); 20050b57cec5SDimitry Andric if (ObjOrErr) { 20060b57cec5SDimitry Andric ObjectFile &Obj = *ObjOrErr.get(); 20070b57cec5SDimitry Andric if (PrintFileName) { 20080b57cec5SDimitry Andric if (isa<MachOObjectFile>(Obj) && moreThanOneArch) 20090b57cec5SDimitry Andric ArchitectureName = O.getArchFlagName(); 20100b57cec5SDimitry Andric } else { 20110b57cec5SDimitry Andric if (moreThanOneArch) 20120b57cec5SDimitry Andric outs() << "\n"; 20130b57cec5SDimitry Andric outs() << Obj.getFileName(); 20140b57cec5SDimitry Andric if (isa<MachOObjectFile>(Obj) && moreThanOneArch) 20150b57cec5SDimitry Andric outs() << " (for architecture " << O.getArchFlagName() << ")"; 20160b57cec5SDimitry Andric outs() << ":\n"; 20170b57cec5SDimitry Andric } 20180b57cec5SDimitry Andric dumpSymbolNamesFromObject(Obj, false, ArchiveName, ArchitectureName); 20190b57cec5SDimitry Andric } else if (auto E = isNotObjectErrorInvalidFileType( 20200b57cec5SDimitry Andric ObjOrErr.takeError())) { 20210b57cec5SDimitry Andric error(std::move(E), Filename, moreThanOneArch ? 20220b57cec5SDimitry Andric StringRef(O.getArchFlagName()) : StringRef()); 20230b57cec5SDimitry Andric continue; 20240b57cec5SDimitry Andric } else if (Expected<std::unique_ptr<Archive>> AOrErr = 20250b57cec5SDimitry Andric O.getAsArchive()) { 20260b57cec5SDimitry Andric std::unique_ptr<Archive> &A = *AOrErr; 20270b57cec5SDimitry Andric Error Err = Error::success(); 20280b57cec5SDimitry Andric for (auto &C : A->children(Err)) { 20290b57cec5SDimitry Andric Expected<std::unique_ptr<Binary>> ChildOrErr = 20300b57cec5SDimitry Andric C.getAsBinary(ContextPtr); 20310b57cec5SDimitry Andric if (!ChildOrErr) { 20320b57cec5SDimitry Andric if (auto E = isNotObjectErrorInvalidFileType( 20330b57cec5SDimitry Andric ChildOrErr.takeError())) 20340b57cec5SDimitry Andric error(std::move(E), Filename, C, moreThanOneArch ? 20350b57cec5SDimitry Andric StringRef(ArchitectureName) : StringRef()); 20360b57cec5SDimitry Andric continue; 20370b57cec5SDimitry Andric } 20380b57cec5SDimitry Andric if (SymbolicFile *F = dyn_cast<SymbolicFile>(&*ChildOrErr.get())) { 20390b57cec5SDimitry Andric if (PrintFileName) { 20400b57cec5SDimitry Andric ArchiveName = A->getFileName(); 20410b57cec5SDimitry Andric if (isa<MachOObjectFile>(F) && moreThanOneArch) 20420b57cec5SDimitry Andric ArchitectureName = O.getArchFlagName(); 20430b57cec5SDimitry Andric } else { 20440b57cec5SDimitry Andric outs() << "\n" << A->getFileName(); 20450b57cec5SDimitry Andric if (isa<MachOObjectFile>(F)) { 20460b57cec5SDimitry Andric outs() << "(" << F->getFileName() << ")"; 20470b57cec5SDimitry Andric if (moreThanOneArch) 20480b57cec5SDimitry Andric outs() << " (for architecture " << O.getArchFlagName() 20490b57cec5SDimitry Andric << ")"; 20500b57cec5SDimitry Andric } else 20510b57cec5SDimitry Andric outs() << ":" << F->getFileName(); 20520b57cec5SDimitry Andric outs() << ":\n"; 20530b57cec5SDimitry Andric } 20540b57cec5SDimitry Andric dumpSymbolNamesFromObject(*F, false, ArchiveName, ArchitectureName); 20550b57cec5SDimitry Andric } 20560b57cec5SDimitry Andric } 20570b57cec5SDimitry Andric if (Err) 20580b57cec5SDimitry Andric error(std::move(Err), A->getFileName()); 20590b57cec5SDimitry Andric } else { 20600b57cec5SDimitry Andric consumeError(AOrErr.takeError()); 20610b57cec5SDimitry Andric error(Filename + " for architecture " + 20620b57cec5SDimitry Andric StringRef(O.getArchFlagName()) + 20630b57cec5SDimitry Andric " is not a Mach-O file or an archive file", 20640b57cec5SDimitry Andric "Mach-O universal file"); 20650b57cec5SDimitry Andric } 20660b57cec5SDimitry Andric } 20670b57cec5SDimitry Andric return; 20680b57cec5SDimitry Andric } 20690b57cec5SDimitry Andric if (SymbolicFile *O = dyn_cast<SymbolicFile>(&Bin)) { 20700b57cec5SDimitry Andric if (!MachOPrintSizeWarning && PrintSize && isa<MachOObjectFile>(O)) { 20710b57cec5SDimitry Andric WithColor::warning(errs(), ToolName) 20720b57cec5SDimitry Andric << "sizes with --print-size for Mach-O files are always zero.\n"; 20730b57cec5SDimitry Andric MachOPrintSizeWarning = true; 20740b57cec5SDimitry Andric } 20750b57cec5SDimitry Andric if (!checkMachOAndArchFlags(O, Filename)) 20760b57cec5SDimitry Andric return; 20770b57cec5SDimitry Andric dumpSymbolNamesFromObject(*O, true); 20780b57cec5SDimitry Andric } 20790b57cec5SDimitry Andric } 20800b57cec5SDimitry Andric 20810b57cec5SDimitry Andric int main(int argc, char **argv) { 20820b57cec5SDimitry Andric InitLLVM X(argc, argv); 20830b57cec5SDimitry Andric cl::HideUnrelatedOptions(NMCat); 20840b57cec5SDimitry Andric cl::ParseCommandLineOptions(argc, argv, "llvm symbol table dumper\n"); 20850b57cec5SDimitry Andric 20860b57cec5SDimitry Andric // llvm-nm only reads binary files. 20870b57cec5SDimitry Andric if (error(sys::ChangeStdinToBinary())) 20880b57cec5SDimitry Andric return 1; 20890b57cec5SDimitry Andric 20900b57cec5SDimitry Andric // These calls are needed so that we can read bitcode correctly. 20910b57cec5SDimitry Andric llvm::InitializeAllTargetInfos(); 20920b57cec5SDimitry Andric llvm::InitializeAllTargetMCs(); 20930b57cec5SDimitry Andric llvm::InitializeAllAsmParsers(); 20940b57cec5SDimitry Andric 20950b57cec5SDimitry Andric ToolName = argv[0]; 20960b57cec5SDimitry Andric if (BSDFormat) 20970b57cec5SDimitry Andric OutputFormat = bsd; 20980b57cec5SDimitry Andric if (POSIXFormat) 20990b57cec5SDimitry Andric OutputFormat = posix; 21000b57cec5SDimitry Andric if (DarwinFormat) 21010b57cec5SDimitry Andric OutputFormat = darwin; 21020b57cec5SDimitry Andric 21030b57cec5SDimitry Andric // The relative order of these is important. If you pass --size-sort it should 21040b57cec5SDimitry Andric // only print out the size. However, if you pass -S --size-sort, it should 21050b57cec5SDimitry Andric // print out both the size and address. 21060b57cec5SDimitry Andric if (SizeSort && !PrintSize) 21070b57cec5SDimitry Andric PrintAddress = false; 21080b57cec5SDimitry Andric if (OutputFormat == sysv || SizeSort) 21090b57cec5SDimitry Andric PrintSize = true; 21100b57cec5SDimitry Andric if (InputFilenames.empty()) 21110b57cec5SDimitry Andric InputFilenames.push_back("a.out"); 21120b57cec5SDimitry Andric if (InputFilenames.size() > 1) 21130b57cec5SDimitry Andric MultipleFiles = true; 21140b57cec5SDimitry Andric 21150b57cec5SDimitry Andric // If both --demangle and --no-demangle are specified then pick the last one. 21160b57cec5SDimitry Andric if (NoDemangle.getPosition() > Demangle.getPosition()) 21170b57cec5SDimitry Andric Demangle = !NoDemangle; 21180b57cec5SDimitry Andric 21190b57cec5SDimitry Andric for (unsigned i = 0; i < ArchFlags.size(); ++i) { 21200b57cec5SDimitry Andric if (ArchFlags[i] == "all") { 21210b57cec5SDimitry Andric ArchAll = true; 21220b57cec5SDimitry Andric } else { 21230b57cec5SDimitry Andric if (!MachOObjectFile::isValidArch(ArchFlags[i])) 21240b57cec5SDimitry Andric error("Unknown architecture named '" + ArchFlags[i] + "'", 21250b57cec5SDimitry Andric "for the --arch option"); 21260b57cec5SDimitry Andric } 21270b57cec5SDimitry Andric } 21280b57cec5SDimitry Andric 21290b57cec5SDimitry Andric if (!SegSect.empty() && SegSect.size() != 2) 21300b57cec5SDimitry Andric error("bad number of arguments (must be two arguments)", 21310b57cec5SDimitry Andric "for the -s option"); 21320b57cec5SDimitry Andric 21330b57cec5SDimitry Andric if (NoDyldInfo && (AddDyldInfo || DyldInfoOnly)) 21340b57cec5SDimitry Andric error("--no-dyldinfo can't be used with --add-dyldinfo or --dyldinfo-only"); 21350b57cec5SDimitry Andric 21360b57cec5SDimitry Andric llvm::for_each(InputFilenames, dumpSymbolNamesFromFile); 21370b57cec5SDimitry Andric 21380b57cec5SDimitry Andric if (HadError) 21390b57cec5SDimitry Andric return 1; 21400b57cec5SDimitry Andric } 2141