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