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