10b57cec5SDimitry Andric //===-- llvm-nm.cpp - Symbol table dumping utility for llvm ---------------===// 20b57cec5SDimitry Andric // 30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 60b57cec5SDimitry Andric // 70b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 80b57cec5SDimitry Andric // 90b57cec5SDimitry Andric // This program is a utility that works like traditional Unix "nm", that is, it 100b57cec5SDimitry Andric // prints out the names of symbols in a bitcode or object file, along with some 110b57cec5SDimitry Andric // information about each symbol. 120b57cec5SDimitry Andric // 130b57cec5SDimitry Andric // This "nm" supports many of the features of GNU "nm", including its different 140b57cec5SDimitry Andric // output formats. 150b57cec5SDimitry Andric // 160b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 170b57cec5SDimitry Andric 180b57cec5SDimitry Andric #include "llvm/ADT/StringSwitch.h" 190b57cec5SDimitry Andric #include "llvm/BinaryFormat/COFF.h" 2081ad6265SDimitry Andric #include "llvm/BinaryFormat/XCOFF.h" 210b57cec5SDimitry Andric #include "llvm/Demangle/Demangle.h" 220b57cec5SDimitry Andric #include "llvm/IR/Function.h" 230b57cec5SDimitry Andric #include "llvm/IR/LLVMContext.h" 240b57cec5SDimitry Andric #include "llvm/Object/Archive.h" 250b57cec5SDimitry Andric #include "llvm/Object/COFF.h" 260b57cec5SDimitry Andric #include "llvm/Object/COFFImportFile.h" 270b57cec5SDimitry Andric #include "llvm/Object/ELFObjectFile.h" 280b57cec5SDimitry Andric #include "llvm/Object/IRObjectFile.h" 290b57cec5SDimitry Andric #include "llvm/Object/MachO.h" 300b57cec5SDimitry Andric #include "llvm/Object/MachOUniversal.h" 310b57cec5SDimitry Andric #include "llvm/Object/ObjectFile.h" 325ffd83dbSDimitry Andric #include "llvm/Object/TapiFile.h" 335ffd83dbSDimitry Andric #include "llvm/Object/TapiUniversal.h" 340b57cec5SDimitry Andric #include "llvm/Object/Wasm.h" 3504eeddc0SDimitry Andric #include "llvm/Object/XCOFFObjectFile.h" 36fe6060f1SDimitry Andric #include "llvm/Option/Arg.h" 37fe6060f1SDimitry Andric #include "llvm/Option/ArgList.h" 38fe6060f1SDimitry Andric #include "llvm/Option/Option.h" 390b57cec5SDimitry Andric #include "llvm/Support/CommandLine.h" 400b57cec5SDimitry Andric #include "llvm/Support/FileSystem.h" 410b57cec5SDimitry Andric #include "llvm/Support/Format.h" 420b57cec5SDimitry Andric #include "llvm/Support/InitLLVM.h" 43*06c3fb27SDimitry Andric #include "llvm/Support/LLVMDriver.h" 440b57cec5SDimitry Andric #include "llvm/Support/MemoryBuffer.h" 450b57cec5SDimitry Andric #include "llvm/Support/Program.h" 460b57cec5SDimitry Andric #include "llvm/Support/Signals.h" 470b57cec5SDimitry Andric #include "llvm/Support/TargetSelect.h" 480b57cec5SDimitry Andric #include "llvm/Support/WithColor.h" 490b57cec5SDimitry Andric #include "llvm/Support/raw_ostream.h" 50*06c3fb27SDimitry Andric #include "llvm/TargetParser/Host.h" 51*06c3fb27SDimitry Andric #include "llvm/TargetParser/Triple.h" 520b57cec5SDimitry Andric #include <vector> 530b57cec5SDimitry Andric 540b57cec5SDimitry Andric using namespace llvm; 550b57cec5SDimitry Andric using namespace object; 560b57cec5SDimitry Andric 570b57cec5SDimitry Andric namespace { 58fe6060f1SDimitry Andric using namespace llvm::opt; // for HelpHidden in Opts.inc 59fe6060f1SDimitry Andric enum ID { 60fe6060f1SDimitry Andric OPT_INVALID = 0, // This is not an option ID. 61fe6060f1SDimitry Andric #define OPTION(PREFIX, NAME, ID, KIND, GROUP, ALIAS, ALIASARGS, FLAGS, PARAM, \ 62fe6060f1SDimitry Andric HELPTEXT, METAVAR, VALUES) \ 63fe6060f1SDimitry Andric OPT_##ID, 64fe6060f1SDimitry Andric #include "Opts.inc" 65fe6060f1SDimitry Andric #undef OPTION 66fe6060f1SDimitry Andric }; 670b57cec5SDimitry Andric 68bdd1243dSDimitry Andric #define PREFIX(NAME, VALUE) \ 69bdd1243dSDimitry Andric static constexpr StringLiteral NAME##_init[] = VALUE; \ 70bdd1243dSDimitry Andric static constexpr ArrayRef<StringLiteral> NAME(NAME##_init, \ 71bdd1243dSDimitry Andric std::size(NAME##_init) - 1); 72fe6060f1SDimitry Andric #include "Opts.inc" 73fe6060f1SDimitry Andric #undef PREFIX 740b57cec5SDimitry Andric 75bdd1243dSDimitry Andric static constexpr opt::OptTable::Info InfoTable[] = { 76fe6060f1SDimitry Andric #define OPTION(PREFIX, NAME, ID, KIND, GROUP, ALIAS, ALIASARGS, FLAGS, PARAM, \ 77fe6060f1SDimitry Andric HELPTEXT, METAVAR, VALUES) \ 78fe6060f1SDimitry Andric { \ 79fe6060f1SDimitry Andric PREFIX, NAME, HELPTEXT, \ 80fe6060f1SDimitry Andric METAVAR, OPT_##ID, opt::Option::KIND##Class, \ 81fe6060f1SDimitry Andric PARAM, FLAGS, OPT_##GROUP, \ 82fe6060f1SDimitry Andric OPT_##ALIAS, ALIASARGS, VALUES}, 83fe6060f1SDimitry Andric #include "Opts.inc" 84fe6060f1SDimitry Andric #undef OPTION 85fe6060f1SDimitry Andric }; 860b57cec5SDimitry Andric 87bdd1243dSDimitry Andric class NmOptTable : public opt::GenericOptTable { 88fe6060f1SDimitry Andric public: 89bdd1243dSDimitry Andric NmOptTable() : opt::GenericOptTable(InfoTable) { 90bdd1243dSDimitry Andric setGroupedShortOptions(true); 91bdd1243dSDimitry Andric } 92fe6060f1SDimitry Andric }; 930b57cec5SDimitry Andric 94fe6060f1SDimitry Andric enum OutputFormatTy { bsd, sysv, posix, darwin, just_symbols }; 9581ad6265SDimitry Andric enum class BitModeTy { Bit32, Bit64, Bit32_64, Any }; 96fe6060f1SDimitry Andric } // namespace 970b57cec5SDimitry Andric 98fe6060f1SDimitry Andric static bool ArchiveMap; 9981ad6265SDimitry Andric static BitModeTy BitMode; 100fe6060f1SDimitry Andric static bool DebugSyms; 101fe6060f1SDimitry Andric static bool DefinedOnly; 102fe6060f1SDimitry Andric static bool Demangle; 103fe6060f1SDimitry Andric static bool DynamicSyms; 10481ad6265SDimitry Andric static bool ExportSymbols; 105fe6060f1SDimitry Andric static bool ExternalOnly; 106fe6060f1SDimitry Andric static OutputFormatTy OutputFormat; 107fe6060f1SDimitry Andric static bool NoLLVMBitcode; 108fe6060f1SDimitry Andric static bool NoSort; 109fe6060f1SDimitry Andric static bool NoWeakSymbols; 110fe6060f1SDimitry Andric static bool NumericSort; 111fe6060f1SDimitry Andric static bool PrintFileName; 112fe6060f1SDimitry Andric static bool PrintSize; 113fe6060f1SDimitry Andric static bool Quiet; 114fe6060f1SDimitry Andric static bool ReverseSort; 115fe6060f1SDimitry Andric static bool SpecialSyms; 116fe6060f1SDimitry Andric static bool SizeSort; 117fe6060f1SDimitry Andric static bool UndefinedOnly; 118fe6060f1SDimitry Andric static bool WithoutAliases; 1190b57cec5SDimitry Andric 12081ad6265SDimitry Andric // XCOFF-specific options. 12181ad6265SDimitry Andric static bool NoRsrc; 12281ad6265SDimitry Andric 123fe6060f1SDimitry Andric namespace { 1240b57cec5SDimitry Andric enum Radix { d, o, x }; 125fe6060f1SDimitry Andric } // namespace 126fe6060f1SDimitry Andric static Radix AddressRadix; 1270b57cec5SDimitry Andric 128fe6060f1SDimitry Andric // Mach-O specific options. 129fe6060f1SDimitry Andric static bool ArchAll = false; 130fe6060f1SDimitry Andric static std::vector<StringRef> ArchFlags; 131fe6060f1SDimitry Andric static bool AddDyldInfo; 132fe6060f1SDimitry Andric static bool AddInlinedInfo; 133fe6060f1SDimitry Andric static bool DyldInfoOnly; 134fe6060f1SDimitry Andric static bool FormatMachOasHex; 135fe6060f1SDimitry Andric static bool NoDyldInfo; 136fe6060f1SDimitry Andric static std::vector<StringRef> SegSect; 137fe6060f1SDimitry Andric static bool MachOPrintSizeWarning = false; 1380b57cec5SDimitry Andric 139fe6060f1SDimitry Andric // Miscellaneous states. 140fe6060f1SDimitry Andric static bool PrintAddress = true; 141fe6060f1SDimitry Andric static bool MultipleFiles = false; 142fe6060f1SDimitry Andric static bool HadError = false; 1430b57cec5SDimitry Andric 144fe6060f1SDimitry Andric static StringRef ToolName; 1450b57cec5SDimitry Andric 14681ad6265SDimitry Andric static void warn(Error Err, Twine FileName, Twine Context = Twine(), 14781ad6265SDimitry Andric Twine Archive = Twine()) { 14804eeddc0SDimitry Andric assert(Err); 14904eeddc0SDimitry Andric 15004eeddc0SDimitry Andric // Flush the standard output so that the warning isn't interleaved with other 15104eeddc0SDimitry Andric // output if stdout and stderr are writing to the same place. 15204eeddc0SDimitry Andric outs().flush(); 15304eeddc0SDimitry Andric 15404eeddc0SDimitry Andric handleAllErrors(std::move(Err), [&](const ErrorInfoBase &EI) { 15504eeddc0SDimitry Andric WithColor::warning(errs(), ToolName) 15681ad6265SDimitry Andric << (Archive.str().empty() ? FileName : Archive + "(" + FileName + ")") 15781ad6265SDimitry Andric << ": " << (Context.str().empty() ? "" : Context + ": ") << EI.message() 15881ad6265SDimitry Andric << "\n"; 15904eeddc0SDimitry Andric }); 16004eeddc0SDimitry Andric } 16104eeddc0SDimitry Andric 1620b57cec5SDimitry Andric static void error(Twine Message, Twine Path = Twine()) { 1630b57cec5SDimitry Andric HadError = true; 164fe6060f1SDimitry Andric WithColor::error(errs(), ToolName) << Path << ": " << Message << "\n"; 1650b57cec5SDimitry Andric } 1660b57cec5SDimitry Andric 1670b57cec5SDimitry Andric static bool error(std::error_code EC, Twine Path = Twine()) { 1680b57cec5SDimitry Andric if (EC) { 1690b57cec5SDimitry Andric error(EC.message(), Path); 1700b57cec5SDimitry Andric return true; 1710b57cec5SDimitry Andric } 1720b57cec5SDimitry Andric return false; 1730b57cec5SDimitry Andric } 1740b57cec5SDimitry Andric 1750b57cec5SDimitry Andric // This version of error() prints the archive name and member name, for example: 1760b57cec5SDimitry Andric // "libx.a(foo.o)" after the ToolName before the error message. It sets 1770b57cec5SDimitry Andric // HadError but returns allowing the code to move on to other archive members. 1780b57cec5SDimitry Andric static void error(llvm::Error E, StringRef FileName, const Archive::Child &C, 1790b57cec5SDimitry Andric StringRef ArchitectureName = StringRef()) { 1800b57cec5SDimitry Andric HadError = true; 1810b57cec5SDimitry Andric WithColor::error(errs(), ToolName) << FileName; 1820b57cec5SDimitry Andric 1830b57cec5SDimitry Andric Expected<StringRef> NameOrErr = C.getName(); 1840b57cec5SDimitry Andric // TODO: if we have a error getting the name then it would be nice to print 1850b57cec5SDimitry Andric // the index of which archive member this is and or its offset in the 1860b57cec5SDimitry Andric // archive instead of "???" as the name. 1870b57cec5SDimitry Andric if (!NameOrErr) { 1880b57cec5SDimitry Andric consumeError(NameOrErr.takeError()); 1890b57cec5SDimitry Andric errs() << "(" << "???" << ")"; 1900b57cec5SDimitry Andric } else 1910b57cec5SDimitry Andric errs() << "(" << NameOrErr.get() << ")"; 1920b57cec5SDimitry Andric 1930b57cec5SDimitry Andric if (!ArchitectureName.empty()) 1940b57cec5SDimitry Andric errs() << " (for architecture " << ArchitectureName << ")"; 1950b57cec5SDimitry Andric 1960b57cec5SDimitry Andric std::string Buf; 1970b57cec5SDimitry Andric raw_string_ostream OS(Buf); 1980b57cec5SDimitry Andric logAllUnhandledErrors(std::move(E), OS); 1990b57cec5SDimitry Andric OS.flush(); 200fe6060f1SDimitry Andric errs() << ": " << Buf << "\n"; 2010b57cec5SDimitry Andric } 2020b57cec5SDimitry Andric 2030b57cec5SDimitry Andric // This version of error() prints the file name and which architecture slice it 2040b57cec5SDimitry Andric // is from, for example: "foo.o (for architecture i386)" after the ToolName 2050b57cec5SDimitry Andric // before the error message. It sets HadError but returns allowing the code to 2060b57cec5SDimitry Andric // move on to other architecture slices. 2070b57cec5SDimitry Andric static void error(llvm::Error E, StringRef FileName, 2080b57cec5SDimitry Andric StringRef ArchitectureName = StringRef()) { 2090b57cec5SDimitry Andric HadError = true; 2100b57cec5SDimitry Andric WithColor::error(errs(), ToolName) << FileName; 2110b57cec5SDimitry Andric 2120b57cec5SDimitry Andric if (!ArchitectureName.empty()) 2130b57cec5SDimitry Andric errs() << " (for architecture " << ArchitectureName << ")"; 2140b57cec5SDimitry Andric 2150b57cec5SDimitry Andric std::string Buf; 2160b57cec5SDimitry Andric raw_string_ostream OS(Buf); 2170b57cec5SDimitry Andric logAllUnhandledErrors(std::move(E), OS); 2180b57cec5SDimitry Andric OS.flush(); 219fe6060f1SDimitry Andric errs() << ": " << Buf << "\n"; 2200b57cec5SDimitry Andric } 2210b57cec5SDimitry Andric 2220b57cec5SDimitry Andric namespace { 2230b57cec5SDimitry Andric struct NMSymbol { 2240b57cec5SDimitry Andric uint64_t Address; 2250b57cec5SDimitry Andric uint64_t Size; 2260b57cec5SDimitry Andric char TypeChar; 227e8d8bef9SDimitry Andric std::string Name; 2280b57cec5SDimitry Andric StringRef SectionName; 2290b57cec5SDimitry Andric StringRef TypeName; 2300b57cec5SDimitry Andric BasicSymbolRef Sym; 23181ad6265SDimitry Andric StringRef Visibility; 23281ad6265SDimitry Andric 2330b57cec5SDimitry Andric // The Sym field above points to the native symbol in the object file, 2340b57cec5SDimitry Andric // for Mach-O when we are creating symbols from the dyld info the above 2350b57cec5SDimitry Andric // pointer is null as there is no native symbol. In these cases the fields 2360b57cec5SDimitry Andric // below are filled in to represent what would have been a Mach-O nlist 2370b57cec5SDimitry Andric // native symbol. 2380b57cec5SDimitry Andric uint32_t SymFlags; 2390b57cec5SDimitry Andric SectionRef Section; 2400b57cec5SDimitry Andric uint8_t NType; 2410b57cec5SDimitry Andric uint8_t NSect; 2420b57cec5SDimitry Andric uint16_t NDesc; 243e8d8bef9SDimitry Andric std::string IndirectName; 24481ad6265SDimitry Andric 24581ad6265SDimitry Andric bool isDefined() const { 24681ad6265SDimitry Andric if (Sym.getRawDataRefImpl().p) { 24781ad6265SDimitry Andric uint32_t Flags = cantFail(Sym.getFlags()); 24881ad6265SDimitry Andric return !(Flags & SymbolRef::SF_Undefined); 24981ad6265SDimitry Andric } 25081ad6265SDimitry Andric return TypeChar != 'U'; 25181ad6265SDimitry Andric } 25281ad6265SDimitry Andric 25381ad6265SDimitry Andric bool initializeFlags(const SymbolicFile &Obj) { 25481ad6265SDimitry Andric Expected<uint32_t> SymFlagsOrErr = Sym.getFlags(); 25581ad6265SDimitry Andric if (!SymFlagsOrErr) { 25681ad6265SDimitry Andric // TODO: Test this error. 25781ad6265SDimitry Andric error(SymFlagsOrErr.takeError(), Obj.getFileName()); 25881ad6265SDimitry Andric return false; 25981ad6265SDimitry Andric } 26081ad6265SDimitry Andric SymFlags = *SymFlagsOrErr; 26181ad6265SDimitry Andric return true; 26281ad6265SDimitry Andric } 26381ad6265SDimitry Andric 26481ad6265SDimitry Andric bool shouldPrint() const { 26581ad6265SDimitry Andric bool Undefined = SymFlags & SymbolRef::SF_Undefined; 26681ad6265SDimitry Andric bool Global = SymFlags & SymbolRef::SF_Global; 26781ad6265SDimitry Andric bool Weak = SymFlags & SymbolRef::SF_Weak; 26881ad6265SDimitry Andric bool FormatSpecific = SymFlags & SymbolRef::SF_FormatSpecific; 26981ad6265SDimitry Andric if ((!Undefined && UndefinedOnly) || (Undefined && DefinedOnly) || 27081ad6265SDimitry Andric (!Global && ExternalOnly) || (Weak && NoWeakSymbols) || 27181ad6265SDimitry Andric (FormatSpecific && !(SpecialSyms || DebugSyms))) 27281ad6265SDimitry Andric return false; 27381ad6265SDimitry Andric return true; 27481ad6265SDimitry Andric } 2750b57cec5SDimitry Andric }; 2760b57cec5SDimitry Andric 27781ad6265SDimitry Andric bool operator<(const NMSymbol &A, const NMSymbol &B) { 27881ad6265SDimitry Andric if (NumericSort) 27981ad6265SDimitry Andric return std::make_tuple(A.isDefined(), A.Address, A.Name, A.Size) < 28081ad6265SDimitry Andric std::make_tuple(B.isDefined(), B.Address, B.Name, B.Size); 28181ad6265SDimitry Andric if (SizeSort) 2820b57cec5SDimitry Andric return std::make_tuple(A.Size, A.Name, A.Address) < 2830b57cec5SDimitry Andric std::make_tuple(B.Size, B.Name, B.Address); 28481ad6265SDimitry Andric if (ExportSymbols) 28581ad6265SDimitry Andric return std::make_tuple(A.Name, A.Visibility) < 28681ad6265SDimitry Andric std::make_tuple(B.Name, B.Visibility); 2870b57cec5SDimitry Andric return std::make_tuple(A.Name, A.Size, A.Address) < 2880b57cec5SDimitry Andric std::make_tuple(B.Name, B.Size, B.Address); 2890b57cec5SDimitry Andric } 2900b57cec5SDimitry Andric 29181ad6265SDimitry Andric bool operator>(const NMSymbol &A, const NMSymbol &B) { return B < A; } 29281ad6265SDimitry Andric bool operator==(const NMSymbol &A, const NMSymbol &B) { 29381ad6265SDimitry Andric return !(A < B) && !(B < A); 29481ad6265SDimitry Andric } 29581ad6265SDimitry Andric } // anonymous namespace 29681ad6265SDimitry Andric 2970b57cec5SDimitry Andric static StringRef CurrentFilename; 2980b57cec5SDimitry Andric 2990b57cec5SDimitry Andric static char getSymbolNMTypeChar(IRObjectFile &Obj, basic_symbol_iterator I); 3000b57cec5SDimitry Andric 3010b57cec5SDimitry Andric // darwinPrintSymbol() is used to print a symbol from a Mach-O file when the 3020b57cec5SDimitry Andric // the OutputFormat is darwin or we are printing Mach-O symbols in hex. For 3030b57cec5SDimitry Andric // the darwin format it produces the same output as darwin's nm(1) -m output 3040b57cec5SDimitry Andric // and when printing Mach-O symbols in hex it produces the same output as 3050b57cec5SDimitry Andric // darwin's nm(1) -x format. 3060b57cec5SDimitry Andric static void darwinPrintSymbol(SymbolicFile &Obj, const NMSymbol &S, 3070b57cec5SDimitry Andric char *SymbolAddrStr, const char *printBlanks, 3080b57cec5SDimitry Andric const char *printDashes, 3090b57cec5SDimitry Andric const char *printFormat) { 3100b57cec5SDimitry Andric MachO::mach_header H; 3110b57cec5SDimitry Andric MachO::mach_header_64 H_64; 3120b57cec5SDimitry Andric uint32_t Filetype = MachO::MH_OBJECT; 3130b57cec5SDimitry Andric uint32_t Flags = 0; 3140b57cec5SDimitry Andric uint8_t NType = 0; 3150b57cec5SDimitry Andric uint8_t NSect = 0; 3160b57cec5SDimitry Andric uint16_t NDesc = 0; 3170b57cec5SDimitry Andric uint32_t NStrx = 0; 3180b57cec5SDimitry Andric uint64_t NValue = 0; 3190b57cec5SDimitry Andric MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(&Obj); 3200b57cec5SDimitry Andric if (Obj.isIR()) { 3215ffd83dbSDimitry Andric uint32_t SymFlags = cantFail(S.Sym.getFlags()); 3220b57cec5SDimitry Andric if (SymFlags & SymbolRef::SF_Global) 3230b57cec5SDimitry Andric NType |= MachO::N_EXT; 3240b57cec5SDimitry Andric if (SymFlags & SymbolRef::SF_Hidden) 3250b57cec5SDimitry Andric NType |= MachO::N_PEXT; 3260b57cec5SDimitry Andric if (SymFlags & SymbolRef::SF_Undefined) 3270b57cec5SDimitry Andric NType |= MachO::N_EXT | MachO::N_UNDF; 3280b57cec5SDimitry Andric else { 3290b57cec5SDimitry Andric // Here we have a symbol definition. So to fake out a section name we 3300b57cec5SDimitry Andric // use 1, 2 and 3 for section numbers. See below where they are used to 3310b57cec5SDimitry Andric // print out fake section names. 3320b57cec5SDimitry Andric NType |= MachO::N_SECT; 3330b57cec5SDimitry Andric if (SymFlags & SymbolRef::SF_Const) 3340b57cec5SDimitry Andric NSect = 3; 3350b57cec5SDimitry Andric else if (SymFlags & SymbolRef::SF_Executable) 3360b57cec5SDimitry Andric NSect = 1; 3370b57cec5SDimitry Andric else 3380b57cec5SDimitry Andric NSect = 2; 3390b57cec5SDimitry Andric } 3400b57cec5SDimitry Andric if (SymFlags & SymbolRef::SF_Weak) 3410b57cec5SDimitry Andric NDesc |= MachO::N_WEAK_DEF; 3420b57cec5SDimitry Andric } else { 3430b57cec5SDimitry Andric DataRefImpl SymDRI = S.Sym.getRawDataRefImpl(); 3440b57cec5SDimitry Andric if (MachO->is64Bit()) { 3450b57cec5SDimitry Andric H_64 = MachO->MachOObjectFile::getHeader64(); 3460b57cec5SDimitry Andric Filetype = H_64.filetype; 3470b57cec5SDimitry Andric Flags = H_64.flags; 3480b57cec5SDimitry Andric if (SymDRI.p){ 3490b57cec5SDimitry Andric MachO::nlist_64 STE_64 = MachO->getSymbol64TableEntry(SymDRI); 3500b57cec5SDimitry Andric NType = STE_64.n_type; 3510b57cec5SDimitry Andric NSect = STE_64.n_sect; 3520b57cec5SDimitry Andric NDesc = STE_64.n_desc; 3530b57cec5SDimitry Andric NStrx = STE_64.n_strx; 3540b57cec5SDimitry Andric NValue = STE_64.n_value; 3550b57cec5SDimitry Andric } else { 3560b57cec5SDimitry Andric NType = S.NType; 3570b57cec5SDimitry Andric NSect = S.NSect; 3580b57cec5SDimitry Andric NDesc = S.NDesc; 3590b57cec5SDimitry Andric NStrx = 0; 3600b57cec5SDimitry Andric NValue = S.Address; 3610b57cec5SDimitry Andric } 3620b57cec5SDimitry Andric } else { 3630b57cec5SDimitry Andric H = MachO->MachOObjectFile::getHeader(); 3640b57cec5SDimitry Andric Filetype = H.filetype; 3650b57cec5SDimitry Andric Flags = H.flags; 3660b57cec5SDimitry Andric if (SymDRI.p){ 3670b57cec5SDimitry Andric MachO::nlist STE = MachO->getSymbolTableEntry(SymDRI); 3680b57cec5SDimitry Andric NType = STE.n_type; 3690b57cec5SDimitry Andric NSect = STE.n_sect; 3700b57cec5SDimitry Andric NDesc = STE.n_desc; 3710b57cec5SDimitry Andric NStrx = STE.n_strx; 3720b57cec5SDimitry Andric NValue = STE.n_value; 3730b57cec5SDimitry Andric } else { 3740b57cec5SDimitry Andric NType = S.NType; 3750b57cec5SDimitry Andric NSect = S.NSect; 3760b57cec5SDimitry Andric NDesc = S.NDesc; 3770b57cec5SDimitry Andric NStrx = 0; 3780b57cec5SDimitry Andric NValue = S.Address; 3790b57cec5SDimitry Andric } 3800b57cec5SDimitry Andric } 3810b57cec5SDimitry Andric } 3820b57cec5SDimitry Andric 3830b57cec5SDimitry Andric // If we are printing Mach-O symbols in hex do that and return. 3840b57cec5SDimitry Andric if (FormatMachOasHex) { 3850b57cec5SDimitry Andric outs() << format(printFormat, NValue) << ' ' 3860b57cec5SDimitry Andric << format("%02x %02x %04x %08x", NType, NSect, NDesc, NStrx) << ' ' 3870b57cec5SDimitry Andric << S.Name; 3880b57cec5SDimitry Andric if ((NType & MachO::N_TYPE) == MachO::N_INDR) { 3890b57cec5SDimitry Andric outs() << " (indirect for "; 3900b57cec5SDimitry Andric outs() << format(printFormat, NValue) << ' '; 3910b57cec5SDimitry Andric StringRef IndirectName; 3920b57cec5SDimitry Andric if (S.Sym.getRawDataRefImpl().p) { 3930b57cec5SDimitry Andric if (MachO->getIndirectName(S.Sym.getRawDataRefImpl(), IndirectName)) 3940b57cec5SDimitry Andric outs() << "?)"; 3950b57cec5SDimitry Andric else 3960b57cec5SDimitry Andric outs() << IndirectName << ")"; 3970b57cec5SDimitry Andric } else 3980b57cec5SDimitry Andric outs() << S.IndirectName << ")"; 3990b57cec5SDimitry Andric } 4000b57cec5SDimitry Andric outs() << "\n"; 4010b57cec5SDimitry Andric return; 4020b57cec5SDimitry Andric } 4030b57cec5SDimitry Andric 4040b57cec5SDimitry Andric if (PrintAddress) { 4050b57cec5SDimitry Andric if ((NType & MachO::N_TYPE) == MachO::N_INDR) 4060b57cec5SDimitry Andric strcpy(SymbolAddrStr, printBlanks); 4070b57cec5SDimitry Andric if (Obj.isIR() && (NType & MachO::N_TYPE) == MachO::N_TYPE) 4080b57cec5SDimitry Andric strcpy(SymbolAddrStr, printDashes); 4090b57cec5SDimitry Andric outs() << SymbolAddrStr << ' '; 4100b57cec5SDimitry Andric } 4110b57cec5SDimitry Andric 4120b57cec5SDimitry Andric switch (NType & MachO::N_TYPE) { 4130b57cec5SDimitry Andric case MachO::N_UNDF: 4140b57cec5SDimitry Andric if (NValue != 0) { 4150b57cec5SDimitry Andric outs() << "(common) "; 4160b57cec5SDimitry Andric if (MachO::GET_COMM_ALIGN(NDesc) != 0) 4170b57cec5SDimitry Andric outs() << "(alignment 2^" << (int)MachO::GET_COMM_ALIGN(NDesc) << ") "; 4180b57cec5SDimitry Andric } else { 4190b57cec5SDimitry Andric if ((NType & MachO::N_TYPE) == MachO::N_PBUD) 4200b57cec5SDimitry Andric outs() << "(prebound "; 4210b57cec5SDimitry Andric else 4220b57cec5SDimitry Andric outs() << "("; 4230b57cec5SDimitry Andric if ((NDesc & MachO::REFERENCE_TYPE) == 4240b57cec5SDimitry Andric MachO::REFERENCE_FLAG_UNDEFINED_LAZY) 4250b57cec5SDimitry Andric outs() << "undefined [lazy bound]) "; 4260b57cec5SDimitry Andric else if ((NDesc & MachO::REFERENCE_TYPE) == 4270b57cec5SDimitry Andric MachO::REFERENCE_FLAG_PRIVATE_UNDEFINED_LAZY) 4280b57cec5SDimitry Andric outs() << "undefined [private lazy bound]) "; 4290b57cec5SDimitry Andric else if ((NDesc & MachO::REFERENCE_TYPE) == 4300b57cec5SDimitry Andric MachO::REFERENCE_FLAG_PRIVATE_UNDEFINED_NON_LAZY) 4310b57cec5SDimitry Andric outs() << "undefined [private]) "; 4320b57cec5SDimitry Andric else 4330b57cec5SDimitry Andric outs() << "undefined) "; 4340b57cec5SDimitry Andric } 4350b57cec5SDimitry Andric break; 4360b57cec5SDimitry Andric case MachO::N_ABS: 4370b57cec5SDimitry Andric outs() << "(absolute) "; 4380b57cec5SDimitry Andric break; 4390b57cec5SDimitry Andric case MachO::N_INDR: 4400b57cec5SDimitry Andric outs() << "(indirect) "; 4410b57cec5SDimitry Andric break; 4420b57cec5SDimitry Andric case MachO::N_SECT: { 4430b57cec5SDimitry Andric if (Obj.isIR()) { 4440b57cec5SDimitry Andric // For llvm bitcode files print out a fake section name using the values 4450b57cec5SDimitry Andric // use 1, 2 and 3 for section numbers as set above. 4460b57cec5SDimitry Andric if (NSect == 1) 4470b57cec5SDimitry Andric outs() << "(LTO,CODE) "; 4480b57cec5SDimitry Andric else if (NSect == 2) 4490b57cec5SDimitry Andric outs() << "(LTO,DATA) "; 4500b57cec5SDimitry Andric else if (NSect == 3) 4510b57cec5SDimitry Andric outs() << "(LTO,RODATA) "; 4520b57cec5SDimitry Andric else 4530b57cec5SDimitry Andric outs() << "(?,?) "; 4540b57cec5SDimitry Andric break; 4550b57cec5SDimitry Andric } 4560b57cec5SDimitry Andric section_iterator Sec = SectionRef(); 4570b57cec5SDimitry Andric if (S.Sym.getRawDataRefImpl().p) { 4580b57cec5SDimitry Andric Expected<section_iterator> SecOrErr = 4590b57cec5SDimitry Andric MachO->getSymbolSection(S.Sym.getRawDataRefImpl()); 4600b57cec5SDimitry Andric if (!SecOrErr) { 4610b57cec5SDimitry Andric consumeError(SecOrErr.takeError()); 4620b57cec5SDimitry Andric outs() << "(?,?) "; 4630b57cec5SDimitry Andric break; 4640b57cec5SDimitry Andric } 4650b57cec5SDimitry Andric Sec = *SecOrErr; 4660b57cec5SDimitry Andric if (Sec == MachO->section_end()) { 4670b57cec5SDimitry Andric outs() << "(?,?) "; 4680b57cec5SDimitry Andric break; 4690b57cec5SDimitry Andric } 4700b57cec5SDimitry Andric } else { 4710b57cec5SDimitry Andric Sec = S.Section; 4720b57cec5SDimitry Andric } 4730b57cec5SDimitry Andric DataRefImpl Ref = Sec->getRawDataRefImpl(); 4740b57cec5SDimitry Andric StringRef SectionName; 4750b57cec5SDimitry Andric if (Expected<StringRef> NameOrErr = MachO->getSectionName(Ref)) 4760b57cec5SDimitry Andric SectionName = *NameOrErr; 4770b57cec5SDimitry Andric StringRef SegmentName = MachO->getSectionFinalSegmentName(Ref); 4780b57cec5SDimitry Andric outs() << "(" << SegmentName << "," << SectionName << ") "; 4790b57cec5SDimitry Andric break; 4800b57cec5SDimitry Andric } 4810b57cec5SDimitry Andric default: 4820b57cec5SDimitry Andric outs() << "(?) "; 4830b57cec5SDimitry Andric break; 4840b57cec5SDimitry Andric } 4850b57cec5SDimitry Andric 4860b57cec5SDimitry Andric if (NType & MachO::N_EXT) { 4870b57cec5SDimitry Andric if (NDesc & MachO::REFERENCED_DYNAMICALLY) 4880b57cec5SDimitry Andric outs() << "[referenced dynamically] "; 4890b57cec5SDimitry Andric if (NType & MachO::N_PEXT) { 4900b57cec5SDimitry Andric if ((NDesc & MachO::N_WEAK_DEF) == MachO::N_WEAK_DEF) 4910b57cec5SDimitry Andric outs() << "weak private external "; 4920b57cec5SDimitry Andric else 4930b57cec5SDimitry Andric outs() << "private external "; 4940b57cec5SDimitry Andric } else { 4950b57cec5SDimitry Andric if ((NDesc & MachO::N_WEAK_REF) == MachO::N_WEAK_REF || 4960b57cec5SDimitry Andric (NDesc & MachO::N_WEAK_DEF) == MachO::N_WEAK_DEF) { 4970b57cec5SDimitry Andric if ((NDesc & (MachO::N_WEAK_REF | MachO::N_WEAK_DEF)) == 4980b57cec5SDimitry Andric (MachO::N_WEAK_REF | MachO::N_WEAK_DEF)) 4990b57cec5SDimitry Andric outs() << "weak external automatically hidden "; 5000b57cec5SDimitry Andric else 5010b57cec5SDimitry Andric outs() << "weak external "; 5020b57cec5SDimitry Andric } else 5030b57cec5SDimitry Andric outs() << "external "; 5040b57cec5SDimitry Andric } 5050b57cec5SDimitry Andric } else { 5060b57cec5SDimitry Andric if (NType & MachO::N_PEXT) 5070b57cec5SDimitry Andric outs() << "non-external (was a private external) "; 5080b57cec5SDimitry Andric else 5090b57cec5SDimitry Andric outs() << "non-external "; 5100b57cec5SDimitry Andric } 5110b57cec5SDimitry Andric 5120b57cec5SDimitry Andric if (Filetype == MachO::MH_OBJECT) { 5130b57cec5SDimitry Andric if (NDesc & MachO::N_NO_DEAD_STRIP) 5140b57cec5SDimitry Andric outs() << "[no dead strip] "; 5150b57cec5SDimitry Andric if ((NType & MachO::N_TYPE) != MachO::N_UNDF && 5160b57cec5SDimitry Andric NDesc & MachO::N_SYMBOL_RESOLVER) 5170b57cec5SDimitry Andric outs() << "[symbol resolver] "; 5180b57cec5SDimitry Andric if ((NType & MachO::N_TYPE) != MachO::N_UNDF && NDesc & MachO::N_ALT_ENTRY) 5190b57cec5SDimitry Andric outs() << "[alt entry] "; 5200b57cec5SDimitry Andric if ((NType & MachO::N_TYPE) != MachO::N_UNDF && NDesc & MachO::N_COLD_FUNC) 5210b57cec5SDimitry Andric outs() << "[cold func] "; 5220b57cec5SDimitry Andric } 5230b57cec5SDimitry Andric 5240b57cec5SDimitry Andric if ((NDesc & MachO::N_ARM_THUMB_DEF) == MachO::N_ARM_THUMB_DEF) 5250b57cec5SDimitry Andric outs() << "[Thumb] "; 5260b57cec5SDimitry Andric 5270b57cec5SDimitry Andric if ((NType & MachO::N_TYPE) == MachO::N_INDR) { 5280b57cec5SDimitry Andric outs() << S.Name << " (for "; 5290b57cec5SDimitry Andric StringRef IndirectName; 5300b57cec5SDimitry Andric if (MachO) { 5310b57cec5SDimitry Andric if (S.Sym.getRawDataRefImpl().p) { 5320b57cec5SDimitry Andric if (MachO->getIndirectName(S.Sym.getRawDataRefImpl(), IndirectName)) 5330b57cec5SDimitry Andric outs() << "?)"; 5340b57cec5SDimitry Andric else 5350b57cec5SDimitry Andric outs() << IndirectName << ")"; 5360b57cec5SDimitry Andric } else 5370b57cec5SDimitry Andric outs() << S.IndirectName << ")"; 5380b57cec5SDimitry Andric } else 5390b57cec5SDimitry Andric outs() << "?)"; 5400b57cec5SDimitry Andric } else 5410b57cec5SDimitry Andric outs() << S.Name; 5420b57cec5SDimitry Andric 5430b57cec5SDimitry Andric if ((Flags & MachO::MH_TWOLEVEL) == MachO::MH_TWOLEVEL && 5440b57cec5SDimitry Andric (((NType & MachO::N_TYPE) == MachO::N_UNDF && NValue == 0) || 5450b57cec5SDimitry Andric (NType & MachO::N_TYPE) == MachO::N_PBUD)) { 5460b57cec5SDimitry Andric uint32_t LibraryOrdinal = MachO::GET_LIBRARY_ORDINAL(NDesc); 5470b57cec5SDimitry Andric if (LibraryOrdinal != 0) { 5480b57cec5SDimitry Andric if (LibraryOrdinal == MachO::EXECUTABLE_ORDINAL) 5490b57cec5SDimitry Andric outs() << " (from executable)"; 5500b57cec5SDimitry Andric else if (LibraryOrdinal == MachO::DYNAMIC_LOOKUP_ORDINAL) 5510b57cec5SDimitry Andric outs() << " (dynamically looked up)"; 5520b57cec5SDimitry Andric else { 5530b57cec5SDimitry Andric StringRef LibraryName; 5540b57cec5SDimitry Andric if (!MachO || 5550b57cec5SDimitry Andric MachO->getLibraryShortNameByIndex(LibraryOrdinal - 1, LibraryName)) 5560b57cec5SDimitry Andric outs() << " (from bad library ordinal " << LibraryOrdinal << ")"; 5570b57cec5SDimitry Andric else 5580b57cec5SDimitry Andric outs() << " (from " << LibraryName << ")"; 5590b57cec5SDimitry Andric } 5600b57cec5SDimitry Andric } 5610b57cec5SDimitry Andric } 5620b57cec5SDimitry Andric 5630b57cec5SDimitry Andric outs() << "\n"; 5640b57cec5SDimitry Andric } 5650b57cec5SDimitry Andric 5660b57cec5SDimitry Andric // Table that maps Darwin's Mach-O stab constants to strings to allow printing. 5670b57cec5SDimitry Andric struct DarwinStabName { 5680b57cec5SDimitry Andric uint8_t NType; 5690b57cec5SDimitry Andric const char *Name; 5700b57cec5SDimitry Andric }; 571349cc55cSDimitry Andric const struct DarwinStabName DarwinStabNames[] = { 5720b57cec5SDimitry Andric {MachO::N_GSYM, "GSYM"}, 5730b57cec5SDimitry Andric {MachO::N_FNAME, "FNAME"}, 5740b57cec5SDimitry Andric {MachO::N_FUN, "FUN"}, 5750b57cec5SDimitry Andric {MachO::N_STSYM, "STSYM"}, 5760b57cec5SDimitry Andric {MachO::N_LCSYM, "LCSYM"}, 5770b57cec5SDimitry Andric {MachO::N_BNSYM, "BNSYM"}, 5780b57cec5SDimitry Andric {MachO::N_PC, "PC"}, 5790b57cec5SDimitry Andric {MachO::N_AST, "AST"}, 5800b57cec5SDimitry Andric {MachO::N_OPT, "OPT"}, 5810b57cec5SDimitry Andric {MachO::N_RSYM, "RSYM"}, 5820b57cec5SDimitry Andric {MachO::N_SLINE, "SLINE"}, 5830b57cec5SDimitry Andric {MachO::N_ENSYM, "ENSYM"}, 5840b57cec5SDimitry Andric {MachO::N_SSYM, "SSYM"}, 5850b57cec5SDimitry Andric {MachO::N_SO, "SO"}, 5860b57cec5SDimitry Andric {MachO::N_OSO, "OSO"}, 5870b57cec5SDimitry Andric {MachO::N_LSYM, "LSYM"}, 5880b57cec5SDimitry Andric {MachO::N_BINCL, "BINCL"}, 5890b57cec5SDimitry Andric {MachO::N_SOL, "SOL"}, 5900b57cec5SDimitry Andric {MachO::N_PARAMS, "PARAM"}, 5910b57cec5SDimitry Andric {MachO::N_VERSION, "VERS"}, 5920b57cec5SDimitry Andric {MachO::N_OLEVEL, "OLEV"}, 5930b57cec5SDimitry Andric {MachO::N_PSYM, "PSYM"}, 5940b57cec5SDimitry Andric {MachO::N_EINCL, "EINCL"}, 5950b57cec5SDimitry Andric {MachO::N_ENTRY, "ENTRY"}, 5960b57cec5SDimitry Andric {MachO::N_LBRAC, "LBRAC"}, 5970b57cec5SDimitry Andric {MachO::N_EXCL, "EXCL"}, 5980b57cec5SDimitry Andric {MachO::N_RBRAC, "RBRAC"}, 5990b57cec5SDimitry Andric {MachO::N_BCOMM, "BCOMM"}, 6000b57cec5SDimitry Andric {MachO::N_ECOMM, "ECOMM"}, 6010b57cec5SDimitry Andric {MachO::N_ECOML, "ECOML"}, 6020b57cec5SDimitry Andric {MachO::N_LENG, "LENG"}, 6030b57cec5SDimitry Andric }; 6040b57cec5SDimitry Andric 6050b57cec5SDimitry Andric static const char *getDarwinStabString(uint8_t NType) { 606bdd1243dSDimitry Andric for (auto I : ArrayRef(DarwinStabNames)) 6070b57cec5SDimitry Andric if (I.NType == NType) 6080b57cec5SDimitry Andric return I.Name; 6090b57cec5SDimitry Andric return nullptr; 6100b57cec5SDimitry Andric } 6110b57cec5SDimitry Andric 6120b57cec5SDimitry Andric // darwinPrintStab() prints the n_sect, n_desc along with a symbolic name of 6130b57cec5SDimitry Andric // a stab n_type value in a Mach-O file. 6140b57cec5SDimitry Andric static void darwinPrintStab(MachOObjectFile *MachO, const NMSymbol &S) { 6150b57cec5SDimitry Andric MachO::nlist_64 STE_64; 6160b57cec5SDimitry Andric MachO::nlist STE; 6170b57cec5SDimitry Andric uint8_t NType; 6180b57cec5SDimitry Andric uint8_t NSect; 6190b57cec5SDimitry Andric uint16_t NDesc; 6200b57cec5SDimitry Andric DataRefImpl SymDRI = S.Sym.getRawDataRefImpl(); 6210b57cec5SDimitry Andric if (MachO->is64Bit()) { 6220b57cec5SDimitry Andric STE_64 = MachO->getSymbol64TableEntry(SymDRI); 6230b57cec5SDimitry Andric NType = STE_64.n_type; 6240b57cec5SDimitry Andric NSect = STE_64.n_sect; 6250b57cec5SDimitry Andric NDesc = STE_64.n_desc; 6260b57cec5SDimitry Andric } else { 6270b57cec5SDimitry Andric STE = MachO->getSymbolTableEntry(SymDRI); 6280b57cec5SDimitry Andric NType = STE.n_type; 6290b57cec5SDimitry Andric NSect = STE.n_sect; 6300b57cec5SDimitry Andric NDesc = STE.n_desc; 6310b57cec5SDimitry Andric } 6320b57cec5SDimitry Andric 6330b57cec5SDimitry Andric outs() << format(" %02x %04x ", NSect, NDesc); 6340b57cec5SDimitry Andric if (const char *stabString = getDarwinStabString(NType)) 6350b57cec5SDimitry Andric outs() << format("%5.5s", stabString); 6360b57cec5SDimitry Andric else 6370b57cec5SDimitry Andric outs() << format(" %02x", NType); 6380b57cec5SDimitry Andric } 6390b57cec5SDimitry Andric 640bdd1243dSDimitry Andric static std::optional<std::string> demangle(StringRef Name) { 641349cc55cSDimitry Andric std::string Demangled; 642*06c3fb27SDimitry Andric if (nonMicrosoftDemangle(Name, Demangled)) 643349cc55cSDimitry Andric return Demangled; 644bdd1243dSDimitry Andric return std::nullopt; 6450b57cec5SDimitry Andric } 6460b57cec5SDimitry Andric 647bdd1243dSDimitry Andric static std::optional<std::string> demangleXCOFF(StringRef Name) { 64804eeddc0SDimitry Andric if (Name.empty() || Name[0] != '.') 64904eeddc0SDimitry Andric return demangle(Name); 65004eeddc0SDimitry Andric 65104eeddc0SDimitry Andric Name = Name.drop_front(); 652bdd1243dSDimitry Andric std::optional<std::string> DemangledName = demangle(Name); 65304eeddc0SDimitry Andric if (DemangledName) 65404eeddc0SDimitry Andric return "." + *DemangledName; 655bdd1243dSDimitry Andric return std::nullopt; 65604eeddc0SDimitry Andric } 65704eeddc0SDimitry Andric 658bdd1243dSDimitry Andric static std::optional<std::string> demangleMachO(StringRef Name) { 65904eeddc0SDimitry Andric if (!Name.empty() && Name[0] == '_') 66004eeddc0SDimitry Andric Name = Name.drop_front(); 66104eeddc0SDimitry Andric return demangle(Name); 66204eeddc0SDimitry Andric } 66304eeddc0SDimitry Andric 6640b57cec5SDimitry Andric static bool symbolIsDefined(const NMSymbol &Sym) { 6650b57cec5SDimitry Andric return Sym.TypeChar != 'U' && Sym.TypeChar != 'w' && Sym.TypeChar != 'v'; 6660b57cec5SDimitry Andric } 6670b57cec5SDimitry Andric 6685ffd83dbSDimitry Andric static void writeFileName(raw_ostream &S, StringRef ArchiveName, 6695ffd83dbSDimitry Andric StringRef ArchitectureName) { 6705ffd83dbSDimitry Andric if (!ArchitectureName.empty()) 6715ffd83dbSDimitry Andric S << "(for architecture " << ArchitectureName << "):"; 6725ffd83dbSDimitry Andric if (OutputFormat == posix && !ArchiveName.empty()) 6735ffd83dbSDimitry Andric S << ArchiveName << "[" << CurrentFilename << "]: "; 6745ffd83dbSDimitry Andric else { 6755ffd83dbSDimitry Andric if (!ArchiveName.empty()) 6765ffd83dbSDimitry Andric S << ArchiveName << ":"; 6775ffd83dbSDimitry Andric S << CurrentFilename << ": "; 6785ffd83dbSDimitry Andric } 6795ffd83dbSDimitry Andric } 6805ffd83dbSDimitry Andric 68181ad6265SDimitry Andric static void sortSymbolList(std::vector<NMSymbol> &SymbolList) { 68281ad6265SDimitry Andric if (NoSort) 68381ad6265SDimitry Andric return; 6840b57cec5SDimitry Andric 6850b57cec5SDimitry Andric if (ReverseSort) 68681ad6265SDimitry Andric llvm::sort(SymbolList, std::greater<>()); 6878bcb0991SDimitry Andric else 68881ad6265SDimitry Andric llvm::sort(SymbolList); 6890b57cec5SDimitry Andric } 6900b57cec5SDimitry Andric 69181ad6265SDimitry Andric static void printExportSymbolList(const std::vector<NMSymbol> &SymbolList) { 69281ad6265SDimitry Andric for (const NMSymbol &Sym : SymbolList) { 69381ad6265SDimitry Andric outs() << Sym.Name; 69481ad6265SDimitry Andric if (!Sym.Visibility.empty()) 69581ad6265SDimitry Andric outs() << ' ' << Sym.Visibility; 69681ad6265SDimitry Andric outs() << '\n'; 69781ad6265SDimitry Andric } 69881ad6265SDimitry Andric } 69981ad6265SDimitry Andric 70081ad6265SDimitry Andric static void printSymbolList(SymbolicFile &Obj, 70181ad6265SDimitry Andric std::vector<NMSymbol> &SymbolList, bool printName, 70281ad6265SDimitry Andric StringRef ArchiveName, StringRef ArchitectureName) { 7030b57cec5SDimitry Andric if (!PrintFileName) { 704fe6060f1SDimitry Andric if ((OutputFormat == bsd || OutputFormat == posix || 705fe6060f1SDimitry Andric OutputFormat == just_symbols) && 706fe6060f1SDimitry Andric MultipleFiles && printName) { 7070b57cec5SDimitry Andric outs() << '\n' << CurrentFilename << ":\n"; 7080b57cec5SDimitry Andric } else if (OutputFormat == sysv) { 7090b57cec5SDimitry Andric outs() << "\n\nSymbols from " << CurrentFilename << ":\n\n"; 710*06c3fb27SDimitry Andric if (Obj.is64Bit()) 7110b57cec5SDimitry Andric outs() << "Name Value Class Type" 7120b57cec5SDimitry Andric << " Size Line Section\n"; 7130b57cec5SDimitry Andric else 7140b57cec5SDimitry Andric outs() << "Name Value Class Type" 7150b57cec5SDimitry Andric << " Size Line Section\n"; 7160b57cec5SDimitry Andric } 7170b57cec5SDimitry Andric } 7180b57cec5SDimitry Andric 7190b57cec5SDimitry Andric const char *printBlanks, *printDashes, *printFormat; 720*06c3fb27SDimitry Andric if (Obj.is64Bit()) { 7210b57cec5SDimitry Andric printBlanks = " "; 7220b57cec5SDimitry Andric printDashes = "----------------"; 7230b57cec5SDimitry Andric switch (AddressRadix) { 7240b57cec5SDimitry Andric case Radix::o: 7250b57cec5SDimitry Andric printFormat = OutputFormat == posix ? "%" PRIo64 : "%016" PRIo64; 7260b57cec5SDimitry Andric break; 7270b57cec5SDimitry Andric case Radix::x: 7280b57cec5SDimitry Andric printFormat = OutputFormat == posix ? "%" PRIx64 : "%016" PRIx64; 7290b57cec5SDimitry Andric break; 7300b57cec5SDimitry Andric default: 7310b57cec5SDimitry Andric printFormat = OutputFormat == posix ? "%" PRId64 : "%016" PRId64; 7320b57cec5SDimitry Andric } 7330b57cec5SDimitry Andric } else { 7340b57cec5SDimitry Andric printBlanks = " "; 7350b57cec5SDimitry Andric printDashes = "--------"; 7360b57cec5SDimitry Andric switch (AddressRadix) { 7370b57cec5SDimitry Andric case Radix::o: 7380b57cec5SDimitry Andric printFormat = OutputFormat == posix ? "%" PRIo64 : "%08" PRIo64; 7390b57cec5SDimitry Andric break; 7400b57cec5SDimitry Andric case Radix::x: 7410b57cec5SDimitry Andric printFormat = OutputFormat == posix ? "%" PRIx64 : "%08" PRIx64; 7420b57cec5SDimitry Andric break; 7430b57cec5SDimitry Andric default: 7440b57cec5SDimitry Andric printFormat = OutputFormat == posix ? "%" PRId64 : "%08" PRId64; 7450b57cec5SDimitry Andric } 7460b57cec5SDimitry Andric } 7470b57cec5SDimitry Andric 7480b57cec5SDimitry Andric for (const NMSymbol &S : SymbolList) { 74981ad6265SDimitry Andric if (!S.shouldPrint()) 75081ad6265SDimitry Andric continue; 75181ad6265SDimitry Andric 752e8d8bef9SDimitry Andric std::string Name = S.Name; 7530b57cec5SDimitry Andric MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(&Obj); 7540b57cec5SDimitry Andric if (Demangle) { 755bdd1243dSDimitry Andric function_ref<std::optional<std::string>(StringRef)> Fn = ::demangle; 75604eeddc0SDimitry Andric if (Obj.isXCOFF()) 75704eeddc0SDimitry Andric Fn = demangleXCOFF; 75804eeddc0SDimitry Andric if (Obj.isMachO()) 75904eeddc0SDimitry Andric Fn = demangleMachO; 760bdd1243dSDimitry Andric if (std::optional<std::string> Opt = Fn(S.Name)) 7610b57cec5SDimitry Andric Name = *Opt; 7620b57cec5SDimitry Andric } 7630b57cec5SDimitry Andric 7640b57cec5SDimitry Andric if (PrintFileName) 7655ffd83dbSDimitry Andric writeFileName(outs(), ArchiveName, ArchitectureName); 766fe6060f1SDimitry Andric if ((OutputFormat == just_symbols || 7670b57cec5SDimitry Andric (UndefinedOnly && MachO && OutputFormat != darwin)) && 7680b57cec5SDimitry Andric OutputFormat != posix) { 7690b57cec5SDimitry Andric outs() << Name << "\n"; 7700b57cec5SDimitry Andric continue; 7710b57cec5SDimitry Andric } 7720b57cec5SDimitry Andric 7730b57cec5SDimitry Andric char SymbolAddrStr[23], SymbolSizeStr[23]; 7740b57cec5SDimitry Andric 7750b57cec5SDimitry Andric // If the format is SysV or the symbol isn't defined, then print spaces. 7760b57cec5SDimitry Andric if (OutputFormat == sysv || !symbolIsDefined(S)) { 7770b57cec5SDimitry Andric if (OutputFormat == posix) { 7780b57cec5SDimitry Andric format(printFormat, S.Address) 7790b57cec5SDimitry Andric .print(SymbolAddrStr, sizeof(SymbolAddrStr)); 7800b57cec5SDimitry Andric format(printFormat, S.Size).print(SymbolSizeStr, sizeof(SymbolSizeStr)); 7810b57cec5SDimitry Andric } else { 7820b57cec5SDimitry Andric strcpy(SymbolAddrStr, printBlanks); 7830b57cec5SDimitry Andric strcpy(SymbolSizeStr, printBlanks); 7840b57cec5SDimitry Andric } 7850b57cec5SDimitry Andric } 7860b57cec5SDimitry Andric 7870b57cec5SDimitry Andric if (symbolIsDefined(S)) { 7880b57cec5SDimitry Andric // Otherwise, print the symbol address and size. 7890b57cec5SDimitry Andric if (Obj.isIR()) 7900b57cec5SDimitry Andric strcpy(SymbolAddrStr, printDashes); 7910b57cec5SDimitry Andric else if (MachO && S.TypeChar == 'I') 7920b57cec5SDimitry Andric strcpy(SymbolAddrStr, printBlanks); 7930b57cec5SDimitry Andric else 7940b57cec5SDimitry Andric format(printFormat, S.Address) 7950b57cec5SDimitry Andric .print(SymbolAddrStr, sizeof(SymbolAddrStr)); 7960b57cec5SDimitry Andric format(printFormat, S.Size).print(SymbolSizeStr, sizeof(SymbolSizeStr)); 7970b57cec5SDimitry Andric } 7980b57cec5SDimitry Andric 7990b57cec5SDimitry Andric // If OutputFormat is darwin or we are printing Mach-O symbols in hex and 8000b57cec5SDimitry Andric // we have a MachOObjectFile, call darwinPrintSymbol to print as darwin's 8010b57cec5SDimitry Andric // nm(1) -m output or hex, else if OutputFormat is darwin or we are 8020b57cec5SDimitry Andric // printing Mach-O symbols in hex and not a Mach-O object fall back to 8030b57cec5SDimitry Andric // OutputFormat bsd (see below). 8040b57cec5SDimitry Andric if ((OutputFormat == darwin || FormatMachOasHex) && (MachO || Obj.isIR())) { 8050b57cec5SDimitry Andric darwinPrintSymbol(Obj, S, SymbolAddrStr, printBlanks, printDashes, 8060b57cec5SDimitry Andric printFormat); 8070b57cec5SDimitry Andric } else if (OutputFormat == posix) { 8080b57cec5SDimitry Andric outs() << Name << " " << S.TypeChar << " " << SymbolAddrStr << " " 8090b57cec5SDimitry Andric << (MachO ? "0" : SymbolSizeStr) << "\n"; 8100b57cec5SDimitry Andric } else if (OutputFormat == bsd || (OutputFormat == darwin && !MachO)) { 8110b57cec5SDimitry Andric if (PrintAddress) 8120b57cec5SDimitry Andric outs() << SymbolAddrStr << ' '; 8130b57cec5SDimitry Andric if (PrintSize) 8140b57cec5SDimitry Andric outs() << SymbolSizeStr << ' '; 8150b57cec5SDimitry Andric outs() << S.TypeChar; 8160b57cec5SDimitry Andric if (S.TypeChar == '-' && MachO) 8170b57cec5SDimitry Andric darwinPrintStab(MachO, S); 8180b57cec5SDimitry Andric outs() << " " << Name; 8190b57cec5SDimitry Andric if (S.TypeChar == 'I' && MachO) { 8200b57cec5SDimitry Andric outs() << " (indirect for "; 8210b57cec5SDimitry Andric if (S.Sym.getRawDataRefImpl().p) { 8220b57cec5SDimitry Andric StringRef IndirectName; 8230b57cec5SDimitry Andric if (MachO->getIndirectName(S.Sym.getRawDataRefImpl(), IndirectName)) 8240b57cec5SDimitry Andric outs() << "?)"; 8250b57cec5SDimitry Andric else 8260b57cec5SDimitry Andric outs() << IndirectName << ")"; 8270b57cec5SDimitry Andric } else 8280b57cec5SDimitry Andric outs() << S.IndirectName << ")"; 8290b57cec5SDimitry Andric } 8300b57cec5SDimitry Andric outs() << "\n"; 8310b57cec5SDimitry Andric } else if (OutputFormat == sysv) { 8320b57cec5SDimitry Andric outs() << left_justify(Name, 20) << "|" << SymbolAddrStr << "| " 8330b57cec5SDimitry Andric << S.TypeChar << " |" << right_justify(S.TypeName, 18) << "|" 8340b57cec5SDimitry Andric << SymbolSizeStr << "| |" << S.SectionName << "\n"; 8350b57cec5SDimitry Andric } 8360b57cec5SDimitry Andric } 8370b57cec5SDimitry Andric 8380b57cec5SDimitry Andric SymbolList.clear(); 8390b57cec5SDimitry Andric } 8400b57cec5SDimitry Andric 8410b57cec5SDimitry Andric static char getSymbolNMTypeChar(ELFObjectFileBase &Obj, 8420b57cec5SDimitry Andric basic_symbol_iterator I) { 8430b57cec5SDimitry Andric // OK, this is ELF 8440b57cec5SDimitry Andric elf_symbol_iterator SymI(I); 8450b57cec5SDimitry Andric 8460b57cec5SDimitry Andric Expected<elf_section_iterator> SecIOrErr = SymI->getSection(); 8470b57cec5SDimitry Andric if (!SecIOrErr) { 8480b57cec5SDimitry Andric consumeError(SecIOrErr.takeError()); 8490b57cec5SDimitry Andric return '?'; 8500b57cec5SDimitry Andric } 8510b57cec5SDimitry Andric 8520b57cec5SDimitry Andric uint8_t Binding = SymI->getBinding(); 8530b57cec5SDimitry Andric if (Binding == ELF::STB_GNU_UNIQUE) 8540b57cec5SDimitry Andric return 'u'; 8550b57cec5SDimitry Andric 8560b57cec5SDimitry Andric assert(Binding != ELF::STB_WEAK && "STB_WEAK not tested in calling function"); 8570b57cec5SDimitry Andric if (Binding != ELF::STB_GLOBAL && Binding != ELF::STB_LOCAL) 8580b57cec5SDimitry Andric return '?'; 8590b57cec5SDimitry Andric 8600b57cec5SDimitry Andric elf_section_iterator SecI = *SecIOrErr; 8610b57cec5SDimitry Andric if (SecI != Obj.section_end()) { 8620b57cec5SDimitry Andric uint32_t Type = SecI->getType(); 8630b57cec5SDimitry Andric uint64_t Flags = SecI->getFlags(); 8640b57cec5SDimitry Andric if (Flags & ELF::SHF_EXECINSTR) 8650b57cec5SDimitry Andric return 't'; 8660b57cec5SDimitry Andric if (Type == ELF::SHT_NOBITS) 8670b57cec5SDimitry Andric return 'b'; 8680b57cec5SDimitry Andric if (Flags & ELF::SHF_ALLOC) 8690b57cec5SDimitry Andric return Flags & ELF::SHF_WRITE ? 'd' : 'r'; 8700b57cec5SDimitry Andric 8718bcb0991SDimitry Andric auto NameOrErr = SecI->getName(); 8728bcb0991SDimitry Andric if (!NameOrErr) { 8738bcb0991SDimitry Andric consumeError(NameOrErr.takeError()); 8740b57cec5SDimitry Andric return '?'; 8758bcb0991SDimitry Andric } 8768bcb0991SDimitry Andric if ((*NameOrErr).startswith(".debug")) 8770b57cec5SDimitry Andric return 'N'; 8780b57cec5SDimitry Andric if (!(Flags & ELF::SHF_WRITE)) 8790b57cec5SDimitry Andric return 'n'; 8800b57cec5SDimitry Andric } 8810b57cec5SDimitry Andric 8820b57cec5SDimitry Andric return '?'; 8830b57cec5SDimitry Andric } 8840b57cec5SDimitry Andric 8850b57cec5SDimitry Andric static char getSymbolNMTypeChar(COFFObjectFile &Obj, symbol_iterator I) { 8860b57cec5SDimitry Andric COFFSymbolRef Symb = Obj.getCOFFSymbol(*I); 8870b57cec5SDimitry Andric // OK, this is COFF. 8880b57cec5SDimitry Andric symbol_iterator SymI(I); 8890b57cec5SDimitry Andric 8900b57cec5SDimitry Andric Expected<StringRef> Name = SymI->getName(); 8910b57cec5SDimitry Andric if (!Name) { 8920b57cec5SDimitry Andric consumeError(Name.takeError()); 8930b57cec5SDimitry Andric return '?'; 8940b57cec5SDimitry Andric } 8950b57cec5SDimitry Andric 8960b57cec5SDimitry Andric char Ret = StringSwitch<char>(*Name) 8970b57cec5SDimitry Andric .StartsWith(".debug", 'N') 8980b57cec5SDimitry Andric .StartsWith(".sxdata", 'N') 8990b57cec5SDimitry Andric .Default('?'); 9000b57cec5SDimitry Andric 9010b57cec5SDimitry Andric if (Ret != '?') 9020b57cec5SDimitry Andric return Ret; 9030b57cec5SDimitry Andric 9040b57cec5SDimitry Andric uint32_t Characteristics = 0; 9050b57cec5SDimitry Andric if (!COFF::isReservedSectionNumber(Symb.getSectionNumber())) { 9060b57cec5SDimitry Andric Expected<section_iterator> SecIOrErr = SymI->getSection(); 9070b57cec5SDimitry Andric if (!SecIOrErr) { 9080b57cec5SDimitry Andric consumeError(SecIOrErr.takeError()); 9090b57cec5SDimitry Andric return '?'; 9100b57cec5SDimitry Andric } 9110b57cec5SDimitry Andric section_iterator SecI = *SecIOrErr; 9120b57cec5SDimitry Andric const coff_section *Section = Obj.getCOFFSection(*SecI); 9130b57cec5SDimitry Andric Characteristics = Section->Characteristics; 9140b57cec5SDimitry Andric if (Expected<StringRef> NameOrErr = Obj.getSectionName(Section)) 9150b57cec5SDimitry Andric if (NameOrErr->startswith(".idata")) 9160b57cec5SDimitry Andric return 'i'; 9170b57cec5SDimitry Andric } 9180b57cec5SDimitry Andric 9190b57cec5SDimitry Andric switch (Symb.getSectionNumber()) { 9200b57cec5SDimitry Andric case COFF::IMAGE_SYM_DEBUG: 9210b57cec5SDimitry Andric return 'n'; 9220b57cec5SDimitry Andric default: 9230b57cec5SDimitry Andric // Check section type. 9240b57cec5SDimitry Andric if (Characteristics & COFF::IMAGE_SCN_CNT_CODE) 9250b57cec5SDimitry Andric return 't'; 9260b57cec5SDimitry Andric if (Characteristics & COFF::IMAGE_SCN_CNT_INITIALIZED_DATA) 9270b57cec5SDimitry Andric return Characteristics & COFF::IMAGE_SCN_MEM_WRITE ? 'd' : 'r'; 9280b57cec5SDimitry Andric if (Characteristics & COFF::IMAGE_SCN_CNT_UNINITIALIZED_DATA) 9290b57cec5SDimitry Andric return 'b'; 9300b57cec5SDimitry Andric if (Characteristics & COFF::IMAGE_SCN_LNK_INFO) 9310b57cec5SDimitry Andric return 'i'; 9320b57cec5SDimitry Andric // Check for section symbol. 9330b57cec5SDimitry Andric if (Symb.isSectionDefinition()) 9340b57cec5SDimitry Andric return 's'; 9350b57cec5SDimitry Andric } 9360b57cec5SDimitry Andric 9370b57cec5SDimitry Andric return '?'; 9380b57cec5SDimitry Andric } 9390b57cec5SDimitry Andric 94004eeddc0SDimitry Andric static char getSymbolNMTypeChar(XCOFFObjectFile &Obj, symbol_iterator I) { 94104eeddc0SDimitry Andric Expected<uint32_t> TypeOrErr = I->getType(); 94204eeddc0SDimitry Andric if (!TypeOrErr) { 94304eeddc0SDimitry Andric warn(TypeOrErr.takeError(), Obj.getFileName(), 94404eeddc0SDimitry Andric "for symbol with index " + 94504eeddc0SDimitry Andric Twine(Obj.getSymbolIndex(I->getRawDataRefImpl().p))); 94604eeddc0SDimitry Andric return '?'; 94704eeddc0SDimitry Andric } 94804eeddc0SDimitry Andric 94904eeddc0SDimitry Andric uint32_t SymType = *TypeOrErr; 95004eeddc0SDimitry Andric 95104eeddc0SDimitry Andric if (SymType == SymbolRef::ST_File) 95204eeddc0SDimitry Andric return 'f'; 95304eeddc0SDimitry Andric 95404eeddc0SDimitry Andric // If the I->getSection() call would return an error, the earlier I->getType() 95504eeddc0SDimitry Andric // call will already have returned the same error first. 95604eeddc0SDimitry Andric section_iterator SecIter = cantFail(I->getSection()); 95704eeddc0SDimitry Andric 95804eeddc0SDimitry Andric if (SecIter == Obj.section_end()) 95904eeddc0SDimitry Andric return '?'; 96004eeddc0SDimitry Andric 96104eeddc0SDimitry Andric if (Obj.isDebugSection(SecIter->getRawDataRefImpl())) 96204eeddc0SDimitry Andric return 'N'; 96304eeddc0SDimitry Andric 96404eeddc0SDimitry Andric if (SecIter->isText()) 96504eeddc0SDimitry Andric return 't'; 96604eeddc0SDimitry Andric 96704eeddc0SDimitry Andric if (SecIter->isData()) 96804eeddc0SDimitry Andric return 'd'; 96904eeddc0SDimitry Andric 97004eeddc0SDimitry Andric if (SecIter->isBSS()) 97104eeddc0SDimitry Andric return 'b'; 97204eeddc0SDimitry Andric 97304eeddc0SDimitry Andric return '?'; 97404eeddc0SDimitry Andric } 97504eeddc0SDimitry Andric 9760b57cec5SDimitry Andric static char getSymbolNMTypeChar(COFFImportFile &Obj) { 9770b57cec5SDimitry Andric switch (Obj.getCOFFImportHeader()->getType()) { 9780b57cec5SDimitry Andric case COFF::IMPORT_CODE: 9790b57cec5SDimitry Andric return 't'; 9800b57cec5SDimitry Andric case COFF::IMPORT_DATA: 9810b57cec5SDimitry Andric return 'd'; 9820b57cec5SDimitry Andric case COFF::IMPORT_CONST: 9830b57cec5SDimitry Andric return 'r'; 9840b57cec5SDimitry Andric } 9850b57cec5SDimitry Andric return '?'; 9860b57cec5SDimitry Andric } 9870b57cec5SDimitry Andric 9880b57cec5SDimitry Andric static char getSymbolNMTypeChar(MachOObjectFile &Obj, basic_symbol_iterator I) { 9890b57cec5SDimitry Andric DataRefImpl Symb = I->getRawDataRefImpl(); 9900b57cec5SDimitry Andric uint8_t NType = Obj.is64Bit() ? Obj.getSymbol64TableEntry(Symb).n_type 9910b57cec5SDimitry Andric : Obj.getSymbolTableEntry(Symb).n_type; 9920b57cec5SDimitry Andric 9930b57cec5SDimitry Andric if (NType & MachO::N_STAB) 9940b57cec5SDimitry Andric return '-'; 9950b57cec5SDimitry Andric 9960b57cec5SDimitry Andric switch (NType & MachO::N_TYPE) { 9970b57cec5SDimitry Andric case MachO::N_ABS: 9980b57cec5SDimitry Andric return 's'; 9990b57cec5SDimitry Andric case MachO::N_INDR: 10000b57cec5SDimitry Andric return 'i'; 10010b57cec5SDimitry Andric case MachO::N_SECT: { 10020b57cec5SDimitry Andric Expected<section_iterator> SecOrErr = Obj.getSymbolSection(Symb); 10030b57cec5SDimitry Andric if (!SecOrErr) { 10040b57cec5SDimitry Andric consumeError(SecOrErr.takeError()); 10050b57cec5SDimitry Andric return 's'; 10060b57cec5SDimitry Andric } 10070b57cec5SDimitry Andric section_iterator Sec = *SecOrErr; 10080b57cec5SDimitry Andric if (Sec == Obj.section_end()) 10090b57cec5SDimitry Andric return 's'; 10100b57cec5SDimitry Andric DataRefImpl Ref = Sec->getRawDataRefImpl(); 10110b57cec5SDimitry Andric StringRef SectionName; 10120b57cec5SDimitry Andric if (Expected<StringRef> NameOrErr = Obj.getSectionName(Ref)) 10130b57cec5SDimitry Andric SectionName = *NameOrErr; 10140b57cec5SDimitry Andric StringRef SegmentName = Obj.getSectionFinalSegmentName(Ref); 10150b57cec5SDimitry Andric if (Obj.is64Bit() && Obj.getHeader64().filetype == MachO::MH_KEXT_BUNDLE && 10160b57cec5SDimitry Andric SegmentName == "__TEXT_EXEC" && SectionName == "__text") 10170b57cec5SDimitry Andric return 't'; 10180b57cec5SDimitry Andric if (SegmentName == "__TEXT" && SectionName == "__text") 10190b57cec5SDimitry Andric return 't'; 10200b57cec5SDimitry Andric if (SegmentName == "__DATA" && SectionName == "__data") 10210b57cec5SDimitry Andric return 'd'; 10220b57cec5SDimitry Andric if (SegmentName == "__DATA" && SectionName == "__bss") 10230b57cec5SDimitry Andric return 'b'; 10240b57cec5SDimitry Andric return 's'; 10250b57cec5SDimitry Andric } 10260b57cec5SDimitry Andric } 10270b57cec5SDimitry Andric 10280b57cec5SDimitry Andric return '?'; 10290b57cec5SDimitry Andric } 10300b57cec5SDimitry Andric 10315ffd83dbSDimitry Andric static char getSymbolNMTypeChar(TapiFile &Obj, basic_symbol_iterator I) { 1032*06c3fb27SDimitry Andric auto Type = cantFail(Obj.getSymbolType(I->getRawDataRefImpl())); 1033*06c3fb27SDimitry Andric switch (Type) { 1034*06c3fb27SDimitry Andric case SymbolRef::ST_Data: 1035*06c3fb27SDimitry Andric return 'd'; 1036*06c3fb27SDimitry Andric case SymbolRef::ST_Function: 1037*06c3fb27SDimitry Andric return 't'; 1038*06c3fb27SDimitry Andric default: 10395ffd83dbSDimitry Andric return 's'; 10405ffd83dbSDimitry Andric } 1041*06c3fb27SDimitry Andric } 10425ffd83dbSDimitry Andric 10430b57cec5SDimitry Andric static char getSymbolNMTypeChar(WasmObjectFile &Obj, basic_symbol_iterator I) { 10445ffd83dbSDimitry Andric uint32_t Flags = cantFail(I->getFlags()); 10450b57cec5SDimitry Andric if (Flags & SymbolRef::SF_Executable) 10460b57cec5SDimitry Andric return 't'; 10470b57cec5SDimitry Andric return 'd'; 10480b57cec5SDimitry Andric } 10490b57cec5SDimitry Andric 10500b57cec5SDimitry Andric static char getSymbolNMTypeChar(IRObjectFile &Obj, basic_symbol_iterator I) { 10515ffd83dbSDimitry Andric uint32_t Flags = cantFail(I->getFlags()); 10520b57cec5SDimitry Andric // FIXME: should we print 'b'? At the IR level we cannot be sure if this 10530b57cec5SDimitry Andric // will be in bss or not, but we could approximate. 10540b57cec5SDimitry Andric if (Flags & SymbolRef::SF_Executable) 10550b57cec5SDimitry Andric return 't'; 10560b57cec5SDimitry Andric else if (Triple(Obj.getTargetTriple()).isOSDarwin() && 10570b57cec5SDimitry Andric (Flags & SymbolRef::SF_Const)) 10580b57cec5SDimitry Andric return 's'; 10590b57cec5SDimitry Andric else 10600b57cec5SDimitry Andric return 'd'; 10610b57cec5SDimitry Andric } 10620b57cec5SDimitry Andric 10630b57cec5SDimitry Andric static bool isObject(SymbolicFile &Obj, basic_symbol_iterator I) { 1064e8d8bef9SDimitry Andric return isa<ELFObjectFileBase>(&Obj) && 1065e8d8bef9SDimitry Andric elf_symbol_iterator(I)->getELFType() == ELF::STT_OBJECT; 10660b57cec5SDimitry Andric } 10670b57cec5SDimitry Andric 10680b57cec5SDimitry Andric // For ELF object files, Set TypeName to the symbol typename, to be printed 10690b57cec5SDimitry Andric // in the 'Type' column of the SYSV format output. 10700b57cec5SDimitry Andric static StringRef getNMTypeName(SymbolicFile &Obj, basic_symbol_iterator I) { 10710b57cec5SDimitry Andric if (isa<ELFObjectFileBase>(&Obj)) { 10720b57cec5SDimitry Andric elf_symbol_iterator SymI(I); 10730b57cec5SDimitry Andric return SymI->getELFTypeName(); 10740b57cec5SDimitry Andric } 10750b57cec5SDimitry Andric return ""; 10760b57cec5SDimitry Andric } 10770b57cec5SDimitry Andric 10780b57cec5SDimitry Andric // Return Posix nm class type tag (single letter), but also set SecName and 10790b57cec5SDimitry Andric // section and name, to be used in format=sysv output. 10800b57cec5SDimitry Andric static char getNMSectionTagAndName(SymbolicFile &Obj, basic_symbol_iterator I, 10810b57cec5SDimitry Andric StringRef &SecName) { 10825ffd83dbSDimitry Andric // Symbol Flags have been checked in the caller. 10835ffd83dbSDimitry Andric uint32_t Symflags = cantFail(I->getFlags()); 10848bcb0991SDimitry Andric if (ELFObjectFileBase *ELFObj = dyn_cast<ELFObjectFileBase>(&Obj)) { 10850b57cec5SDimitry Andric if (Symflags & object::SymbolRef::SF_Absolute) 10860b57cec5SDimitry Andric SecName = "*ABS*"; 10870b57cec5SDimitry Andric else if (Symflags & object::SymbolRef::SF_Common) 10880b57cec5SDimitry Andric SecName = "*COM*"; 10890b57cec5SDimitry Andric else if (Symflags & object::SymbolRef::SF_Undefined) 10900b57cec5SDimitry Andric SecName = "*UND*"; 10910b57cec5SDimitry Andric else { 10920b57cec5SDimitry Andric elf_symbol_iterator SymI(I); 10930b57cec5SDimitry Andric Expected<elf_section_iterator> SecIOrErr = SymI->getSection(); 10940b57cec5SDimitry Andric if (!SecIOrErr) { 10950b57cec5SDimitry Andric consumeError(SecIOrErr.takeError()); 10960b57cec5SDimitry Andric return '?'; 10970b57cec5SDimitry Andric } 10988bcb0991SDimitry Andric 10998bcb0991SDimitry Andric if (*SecIOrErr == ELFObj->section_end()) 11008bcb0991SDimitry Andric return '?'; 11018bcb0991SDimitry Andric 11028bcb0991SDimitry Andric Expected<StringRef> NameOrErr = (*SecIOrErr)->getName(); 11038bcb0991SDimitry Andric if (!NameOrErr) { 11048bcb0991SDimitry Andric consumeError(NameOrErr.takeError()); 11058bcb0991SDimitry Andric return '?'; 11068bcb0991SDimitry Andric } 11078bcb0991SDimitry Andric SecName = *NameOrErr; 11080b57cec5SDimitry Andric } 11090b57cec5SDimitry Andric } 11100b57cec5SDimitry Andric 1111fe6060f1SDimitry Andric if (Symflags & object::SymbolRef::SF_Undefined) { 1112fe6060f1SDimitry Andric if (isa<MachOObjectFile>(Obj) || !(Symflags & object::SymbolRef::SF_Weak)) 11130b57cec5SDimitry Andric return 'U'; 1114fe6060f1SDimitry Andric return isObject(Obj, I) ? 'v' : 'w'; 1115fe6060f1SDimitry Andric } 1116fe6060f1SDimitry Andric if (isa<ELFObjectFileBase>(&Obj)) 1117fe6060f1SDimitry Andric if (ELFSymbolRef(*I).getELFType() == ELF::STT_GNU_IFUNC) 1118fe6060f1SDimitry Andric return 'i'; 1119fe6060f1SDimitry Andric if (!isa<MachOObjectFile>(Obj) && (Symflags & object::SymbolRef::SF_Weak)) 1120fe6060f1SDimitry Andric return isObject(Obj, I) ? 'V' : 'W'; 11210b57cec5SDimitry Andric 11220b57cec5SDimitry Andric if (Symflags & object::SymbolRef::SF_Common) 11230b57cec5SDimitry Andric return 'C'; 11240b57cec5SDimitry Andric 11250b57cec5SDimitry Andric char Ret = '?'; 11260b57cec5SDimitry Andric if (Symflags & object::SymbolRef::SF_Absolute) 11270b57cec5SDimitry Andric Ret = 'a'; 11280b57cec5SDimitry Andric else if (IRObjectFile *IR = dyn_cast<IRObjectFile>(&Obj)) 11290b57cec5SDimitry Andric Ret = getSymbolNMTypeChar(*IR, I); 11300b57cec5SDimitry Andric else if (COFFObjectFile *COFF = dyn_cast<COFFObjectFile>(&Obj)) 11310b57cec5SDimitry Andric Ret = getSymbolNMTypeChar(*COFF, I); 113204eeddc0SDimitry Andric else if (XCOFFObjectFile *XCOFF = dyn_cast<XCOFFObjectFile>(&Obj)) 113304eeddc0SDimitry Andric Ret = getSymbolNMTypeChar(*XCOFF, I); 11340b57cec5SDimitry Andric else if (COFFImportFile *COFFImport = dyn_cast<COFFImportFile>(&Obj)) 11350b57cec5SDimitry Andric Ret = getSymbolNMTypeChar(*COFFImport); 11360b57cec5SDimitry Andric else if (MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(&Obj)) 11370b57cec5SDimitry Andric Ret = getSymbolNMTypeChar(*MachO, I); 11380b57cec5SDimitry Andric else if (WasmObjectFile *Wasm = dyn_cast<WasmObjectFile>(&Obj)) 11390b57cec5SDimitry Andric Ret = getSymbolNMTypeChar(*Wasm, I); 11405ffd83dbSDimitry Andric else if (TapiFile *Tapi = dyn_cast<TapiFile>(&Obj)) 11415ffd83dbSDimitry Andric Ret = getSymbolNMTypeChar(*Tapi, I); 1142480093f4SDimitry Andric else if (ELFObjectFileBase *ELF = dyn_cast<ELFObjectFileBase>(&Obj)) { 1143480093f4SDimitry Andric Ret = getSymbolNMTypeChar(*ELF, I); 1144480093f4SDimitry Andric if (ELFSymbolRef(*I).getBinding() == ELF::STB_GNU_UNIQUE) 1145480093f4SDimitry Andric return Ret; 1146480093f4SDimitry Andric } else 1147480093f4SDimitry Andric llvm_unreachable("unknown binary format"); 11480b57cec5SDimitry Andric 11490b57cec5SDimitry Andric if (!(Symflags & object::SymbolRef::SF_Global)) 11500b57cec5SDimitry Andric return Ret; 11510b57cec5SDimitry Andric 11520b57cec5SDimitry Andric return toupper(Ret); 11530b57cec5SDimitry Andric } 11540b57cec5SDimitry Andric 11550b57cec5SDimitry Andric // getNsectForSegSect() is used to implement the Mach-O "-s segname sectname" 11560b57cec5SDimitry Andric // option to dump only those symbols from that section in a Mach-O file. 115781ad6265SDimitry Andric // It is called once for each Mach-O file from getSymbolNamesFromObject() 11580b57cec5SDimitry Andric // to get the section number for that named section from the command line 11590b57cec5SDimitry Andric // arguments. It returns the section number for that section in the Mach-O 11600b57cec5SDimitry Andric // file or zero it is not present. 11610b57cec5SDimitry Andric static unsigned getNsectForSegSect(MachOObjectFile *Obj) { 11620b57cec5SDimitry Andric unsigned Nsect = 1; 11630b57cec5SDimitry Andric for (auto &S : Obj->sections()) { 11640b57cec5SDimitry Andric DataRefImpl Ref = S.getRawDataRefImpl(); 11650b57cec5SDimitry Andric StringRef SectionName; 11660b57cec5SDimitry Andric if (Expected<StringRef> NameOrErr = Obj->getSectionName(Ref)) 11670b57cec5SDimitry Andric SectionName = *NameOrErr; 11680b57cec5SDimitry Andric StringRef SegmentName = Obj->getSectionFinalSegmentName(Ref); 11690b57cec5SDimitry Andric if (SegmentName == SegSect[0] && SectionName == SegSect[1]) 11700b57cec5SDimitry Andric return Nsect; 11710b57cec5SDimitry Andric Nsect++; 11720b57cec5SDimitry Andric } 11730b57cec5SDimitry Andric return 0; 11740b57cec5SDimitry Andric } 11750b57cec5SDimitry Andric 11760b57cec5SDimitry Andric // getNsectInMachO() is used to implement the Mach-O "-s segname sectname" 11770b57cec5SDimitry Andric // option to dump only those symbols from that section in a Mach-O file. 11780b57cec5SDimitry Andric // It is called once for each symbol in a Mach-O file from 117981ad6265SDimitry Andric // getSymbolNamesFromObject() and returns the section number for that symbol 11800b57cec5SDimitry Andric // if it is in a section, else it returns 0. 11810b57cec5SDimitry Andric static unsigned getNsectInMachO(MachOObjectFile &Obj, BasicSymbolRef Sym) { 11820b57cec5SDimitry Andric DataRefImpl Symb = Sym.getRawDataRefImpl(); 11830b57cec5SDimitry Andric if (Obj.is64Bit()) { 11840b57cec5SDimitry Andric MachO::nlist_64 STE = Obj.getSymbol64TableEntry(Symb); 11850b57cec5SDimitry Andric return (STE.n_type & MachO::N_TYPE) == MachO::N_SECT ? STE.n_sect : 0; 11860b57cec5SDimitry Andric } 11870b57cec5SDimitry Andric MachO::nlist STE = Obj.getSymbolTableEntry(Symb); 11880b57cec5SDimitry Andric return (STE.n_type & MachO::N_TYPE) == MachO::N_SECT ? STE.n_sect : 0; 11890b57cec5SDimitry Andric } 11900b57cec5SDimitry Andric 119181ad6265SDimitry Andric static void dumpSymbolsFromDLInfoMachO(MachOObjectFile &MachO, 119281ad6265SDimitry Andric std::vector<NMSymbol> &SymbolList) { 1193e8d8bef9SDimitry Andric size_t I = SymbolList.size(); 11940b57cec5SDimitry Andric std::string ExportsNameBuffer; 11950b57cec5SDimitry Andric raw_string_ostream EOS(ExportsNameBuffer); 11960b57cec5SDimitry Andric std::string BindsNameBuffer; 11970b57cec5SDimitry Andric raw_string_ostream BOS(BindsNameBuffer); 11980b57cec5SDimitry Andric std::string LazysNameBuffer; 11990b57cec5SDimitry Andric raw_string_ostream LOS(LazysNameBuffer); 12000b57cec5SDimitry Andric std::string WeaksNameBuffer; 12010b57cec5SDimitry Andric raw_string_ostream WOS(WeaksNameBuffer); 12020b57cec5SDimitry Andric std::string FunctionStartsNameBuffer; 12030b57cec5SDimitry Andric raw_string_ostream FOS(FunctionStartsNameBuffer); 1204e8d8bef9SDimitry Andric 12050b57cec5SDimitry Andric MachO::mach_header H; 12060b57cec5SDimitry Andric MachO::mach_header_64 H_64; 12070b57cec5SDimitry Andric uint32_t HFlags = 0; 1208e8d8bef9SDimitry Andric if (MachO.is64Bit()) { 1209e8d8bef9SDimitry Andric H_64 = MachO.MachOObjectFile::getHeader64(); 12100b57cec5SDimitry Andric HFlags = H_64.flags; 12110b57cec5SDimitry Andric } else { 1212e8d8bef9SDimitry Andric H = MachO.MachOObjectFile::getHeader(); 12130b57cec5SDimitry Andric HFlags = H.flags; 12140b57cec5SDimitry Andric } 12150b57cec5SDimitry Andric uint64_t BaseSegmentAddress = 0; 1216e8d8bef9SDimitry Andric for (const auto &Command : MachO.load_commands()) { 12170b57cec5SDimitry Andric if (Command.C.cmd == MachO::LC_SEGMENT) { 1218e8d8bef9SDimitry Andric MachO::segment_command Seg = MachO.getSegmentLoadCommand(Command); 12190b57cec5SDimitry Andric if (Seg.fileoff == 0 && Seg.filesize != 0) { 12200b57cec5SDimitry Andric BaseSegmentAddress = Seg.vmaddr; 12210b57cec5SDimitry Andric break; 12220b57cec5SDimitry Andric } 12230b57cec5SDimitry Andric } else if (Command.C.cmd == MachO::LC_SEGMENT_64) { 1224e8d8bef9SDimitry Andric MachO::segment_command_64 Seg = MachO.getSegment64LoadCommand(Command); 12250b57cec5SDimitry Andric if (Seg.fileoff == 0 && Seg.filesize != 0) { 12260b57cec5SDimitry Andric BaseSegmentAddress = Seg.vmaddr; 12270b57cec5SDimitry Andric break; 12280b57cec5SDimitry Andric } 12290b57cec5SDimitry Andric } 12300b57cec5SDimitry Andric } 12310b57cec5SDimitry Andric if (DyldInfoOnly || AddDyldInfo || 12320b57cec5SDimitry Andric HFlags & MachO::MH_NLIST_OUTOFSYNC_WITH_DYLDINFO) { 12330b57cec5SDimitry Andric unsigned ExportsAdded = 0; 12340b57cec5SDimitry Andric Error Err = Error::success(); 1235e8d8bef9SDimitry Andric for (const llvm::object::ExportEntry &Entry : MachO.exports(Err)) { 12360b57cec5SDimitry Andric bool found = false; 12370b57cec5SDimitry Andric bool ReExport = false; 12380b57cec5SDimitry Andric if (!DyldInfoOnly) { 12390b57cec5SDimitry Andric for (const NMSymbol &S : SymbolList) 12400b57cec5SDimitry Andric if (S.Address == Entry.address() + BaseSegmentAddress && 12410b57cec5SDimitry Andric S.Name == Entry.name()) { 12420b57cec5SDimitry Andric found = true; 12430b57cec5SDimitry Andric break; 12440b57cec5SDimitry Andric } 12450b57cec5SDimitry Andric } 12460b57cec5SDimitry Andric if (!found) { 12470b57cec5SDimitry Andric NMSymbol S = {}; 12480b57cec5SDimitry Andric S.Address = Entry.address() + BaseSegmentAddress; 12490b57cec5SDimitry Andric S.Size = 0; 12500b57cec5SDimitry Andric S.TypeChar = '\0'; 1251e8d8bef9SDimitry Andric S.Name = Entry.name().str(); 12520b57cec5SDimitry Andric // There is no symbol in the nlist symbol table for this so we set 12530b57cec5SDimitry Andric // Sym effectivly to null and the rest of code in here must test for 12540b57cec5SDimitry Andric // it and not do things like Sym.getFlags() for it. 12550b57cec5SDimitry Andric S.Sym = BasicSymbolRef(); 12560b57cec5SDimitry Andric S.SymFlags = SymbolRef::SF_Global; 12570b57cec5SDimitry Andric S.Section = SectionRef(); 12580b57cec5SDimitry Andric S.NType = 0; 12590b57cec5SDimitry Andric S.NSect = 0; 12600b57cec5SDimitry Andric S.NDesc = 0; 12610b57cec5SDimitry Andric 12620b57cec5SDimitry Andric uint64_t EFlags = Entry.flags(); 12630b57cec5SDimitry Andric bool Abs = ((EFlags & MachO::EXPORT_SYMBOL_FLAGS_KIND_MASK) == 12640b57cec5SDimitry Andric MachO::EXPORT_SYMBOL_FLAGS_KIND_ABSOLUTE); 1265e8d8bef9SDimitry Andric bool Resolver = (EFlags & MachO::EXPORT_SYMBOL_FLAGS_STUB_AND_RESOLVER); 12660b57cec5SDimitry Andric ReExport = (EFlags & MachO::EXPORT_SYMBOL_FLAGS_REEXPORT); 12670b57cec5SDimitry Andric bool WeakDef = (EFlags & MachO::EXPORT_SYMBOL_FLAGS_WEAK_DEFINITION); 12680b57cec5SDimitry Andric if (WeakDef) 12690b57cec5SDimitry Andric S.NDesc |= MachO::N_WEAK_DEF; 12700b57cec5SDimitry Andric if (Abs) { 12710b57cec5SDimitry Andric S.NType = MachO::N_EXT | MachO::N_ABS; 12720b57cec5SDimitry Andric S.TypeChar = 'A'; 12730b57cec5SDimitry Andric } else if (ReExport) { 12740b57cec5SDimitry Andric S.NType = MachO::N_EXT | MachO::N_INDR; 12750b57cec5SDimitry Andric S.TypeChar = 'I'; 12760b57cec5SDimitry Andric } else { 12770b57cec5SDimitry Andric S.NType = MachO::N_EXT | MachO::N_SECT; 12780b57cec5SDimitry Andric if (Resolver) { 12790b57cec5SDimitry Andric S.Address = Entry.other() + BaseSegmentAddress; 1280e8d8bef9SDimitry Andric if ((S.Address & 1) != 0 && !MachO.is64Bit() && 1281e8d8bef9SDimitry Andric H.cputype == MachO::CPU_TYPE_ARM) { 12820b57cec5SDimitry Andric S.Address &= ~1LL; 12830b57cec5SDimitry Andric S.NDesc |= MachO::N_ARM_THUMB_DEF; 12840b57cec5SDimitry Andric } 12850b57cec5SDimitry Andric } else { 12860b57cec5SDimitry Andric S.Address = Entry.address() + BaseSegmentAddress; 12870b57cec5SDimitry Andric } 12880b57cec5SDimitry Andric StringRef SegmentName = StringRef(); 12890b57cec5SDimitry Andric StringRef SectionName = StringRef(); 1290e8d8bef9SDimitry Andric for (const SectionRef &Section : MachO.sections()) { 12910b57cec5SDimitry Andric S.NSect++; 12928bcb0991SDimitry Andric 12938bcb0991SDimitry Andric if (Expected<StringRef> NameOrErr = Section.getName()) 12948bcb0991SDimitry Andric SectionName = *NameOrErr; 12958bcb0991SDimitry Andric else 12968bcb0991SDimitry Andric consumeError(NameOrErr.takeError()); 12978bcb0991SDimitry Andric 1298e8d8bef9SDimitry Andric SegmentName = 1299e8d8bef9SDimitry Andric MachO.getSectionFinalSegmentName(Section.getRawDataRefImpl()); 13000b57cec5SDimitry Andric if (S.Address >= Section.getAddress() && 13010b57cec5SDimitry Andric S.Address < Section.getAddress() + Section.getSize()) { 13020b57cec5SDimitry Andric S.Section = Section; 13030b57cec5SDimitry Andric break; 13040b57cec5SDimitry Andric } else if (Entry.name() == "__mh_execute_header" && 13050b57cec5SDimitry Andric SegmentName == "__TEXT" && SectionName == "__text") { 13060b57cec5SDimitry Andric S.Section = Section; 13070b57cec5SDimitry Andric S.NDesc |= MachO::REFERENCED_DYNAMICALLY; 13080b57cec5SDimitry Andric break; 13090b57cec5SDimitry Andric } 13100b57cec5SDimitry Andric } 13110b57cec5SDimitry Andric if (SegmentName == "__TEXT" && SectionName == "__text") 13120b57cec5SDimitry Andric S.TypeChar = 'T'; 13130b57cec5SDimitry Andric else if (SegmentName == "__DATA" && SectionName == "__data") 13140b57cec5SDimitry Andric S.TypeChar = 'D'; 13150b57cec5SDimitry Andric else if (SegmentName == "__DATA" && SectionName == "__bss") 13160b57cec5SDimitry Andric S.TypeChar = 'B'; 13170b57cec5SDimitry Andric else 13180b57cec5SDimitry Andric S.TypeChar = 'S'; 13190b57cec5SDimitry Andric } 13200b57cec5SDimitry Andric SymbolList.push_back(S); 13210b57cec5SDimitry Andric 13220b57cec5SDimitry Andric EOS << Entry.name(); 13230b57cec5SDimitry Andric EOS << '\0'; 13240b57cec5SDimitry Andric ExportsAdded++; 13250b57cec5SDimitry Andric 13260b57cec5SDimitry Andric // For ReExports there are a two more things to do, first add the 13270b57cec5SDimitry Andric // indirect name and second create the undefined symbol using the 13280b57cec5SDimitry Andric // referened dynamic library. 13290b57cec5SDimitry Andric if (ReExport) { 13300b57cec5SDimitry Andric 13310b57cec5SDimitry Andric // Add the indirect name. 13320b57cec5SDimitry Andric if (Entry.otherName().empty()) 13330b57cec5SDimitry Andric EOS << Entry.name(); 13340b57cec5SDimitry Andric else 13350b57cec5SDimitry Andric EOS << Entry.otherName(); 13360b57cec5SDimitry Andric EOS << '\0'; 13370b57cec5SDimitry Andric 13380b57cec5SDimitry Andric // Now create the undefined symbol using the referened dynamic 13390b57cec5SDimitry Andric // library. 13400b57cec5SDimitry Andric NMSymbol U = {}; 13410b57cec5SDimitry Andric U.Address = 0; 13420b57cec5SDimitry Andric U.Size = 0; 13430b57cec5SDimitry Andric U.TypeChar = 'U'; 13440b57cec5SDimitry Andric if (Entry.otherName().empty()) 1345e8d8bef9SDimitry Andric U.Name = Entry.name().str(); 13460b57cec5SDimitry Andric else 1347e8d8bef9SDimitry Andric U.Name = Entry.otherName().str(); 13480b57cec5SDimitry Andric // Again there is no symbol in the nlist symbol table for this so 13490b57cec5SDimitry Andric // we set Sym effectivly to null and the rest of code in here must 13500b57cec5SDimitry Andric // test for it and not do things like Sym.getFlags() for it. 13510b57cec5SDimitry Andric U.Sym = BasicSymbolRef(); 13520b57cec5SDimitry Andric U.SymFlags = SymbolRef::SF_Global | SymbolRef::SF_Undefined; 13530b57cec5SDimitry Andric U.Section = SectionRef(); 13540b57cec5SDimitry Andric U.NType = MachO::N_EXT | MachO::N_UNDF; 13550b57cec5SDimitry Andric U.NSect = 0; 13560b57cec5SDimitry Andric U.NDesc = 0; 13570b57cec5SDimitry Andric // The library ordinal for this undefined symbol is in the export 13580b57cec5SDimitry Andric // trie Entry.other(). 13590b57cec5SDimitry Andric MachO::SET_LIBRARY_ORDINAL(U.NDesc, Entry.other()); 13600b57cec5SDimitry Andric SymbolList.push_back(U); 13610b57cec5SDimitry Andric 13620b57cec5SDimitry Andric // Finally add the undefined symbol's name. 13630b57cec5SDimitry Andric if (Entry.otherName().empty()) 13640b57cec5SDimitry Andric EOS << Entry.name(); 13650b57cec5SDimitry Andric else 13660b57cec5SDimitry Andric EOS << Entry.otherName(); 13670b57cec5SDimitry Andric EOS << '\0'; 13680b57cec5SDimitry Andric ExportsAdded++; 13690b57cec5SDimitry Andric } 13700b57cec5SDimitry Andric } 13710b57cec5SDimitry Andric } 13720b57cec5SDimitry Andric if (Err) 1373e8d8bef9SDimitry Andric error(std::move(Err), MachO.getFileName()); 13740b57cec5SDimitry Andric // Set the symbol names and indirect names for the added symbols. 13750b57cec5SDimitry Andric if (ExportsAdded) { 13760b57cec5SDimitry Andric EOS.flush(); 13770b57cec5SDimitry Andric const char *Q = ExportsNameBuffer.c_str(); 13780b57cec5SDimitry Andric for (unsigned K = 0; K < ExportsAdded; K++) { 13790b57cec5SDimitry Andric SymbolList[I].Name = Q; 13800b57cec5SDimitry Andric Q += strlen(Q) + 1; 13810b57cec5SDimitry Andric if (SymbolList[I].TypeChar == 'I') { 13820b57cec5SDimitry Andric SymbolList[I].IndirectName = Q; 13830b57cec5SDimitry Andric Q += strlen(Q) + 1; 13840b57cec5SDimitry Andric } 13850b57cec5SDimitry Andric I++; 13860b57cec5SDimitry Andric } 13870b57cec5SDimitry Andric } 13880b57cec5SDimitry Andric 13890b57cec5SDimitry Andric // Add the undefined symbols from the bind entries. 13900b57cec5SDimitry Andric unsigned BindsAdded = 0; 13910b57cec5SDimitry Andric Error BErr = Error::success(); 13920b57cec5SDimitry Andric StringRef LastSymbolName = StringRef(); 1393e8d8bef9SDimitry Andric for (const llvm::object::MachOBindEntry &Entry : MachO.bindTable(BErr)) { 13940b57cec5SDimitry Andric bool found = false; 13950b57cec5SDimitry Andric if (LastSymbolName == Entry.symbolName()) 13960b57cec5SDimitry Andric found = true; 13970b57cec5SDimitry Andric else if (!DyldInfoOnly) { 13980b57cec5SDimitry Andric for (unsigned J = 0; J < SymbolList.size() && !found; ++J) { 13990b57cec5SDimitry Andric if (SymbolList[J].Name == Entry.symbolName()) 14000b57cec5SDimitry Andric found = true; 14010b57cec5SDimitry Andric } 14020b57cec5SDimitry Andric } 14030b57cec5SDimitry Andric if (!found) { 14040b57cec5SDimitry Andric LastSymbolName = Entry.symbolName(); 14050b57cec5SDimitry Andric NMSymbol B = {}; 14060b57cec5SDimitry Andric B.Address = 0; 14070b57cec5SDimitry Andric B.Size = 0; 14080b57cec5SDimitry Andric B.TypeChar = 'U'; 14090b57cec5SDimitry Andric // There is no symbol in the nlist symbol table for this so we set 14100b57cec5SDimitry Andric // Sym effectivly to null and the rest of code in here must test for 14110b57cec5SDimitry Andric // it and not do things like Sym.getFlags() for it. 14120b57cec5SDimitry Andric B.Sym = BasicSymbolRef(); 14130b57cec5SDimitry Andric B.SymFlags = SymbolRef::SF_Global | SymbolRef::SF_Undefined; 14140b57cec5SDimitry Andric B.NType = MachO::N_EXT | MachO::N_UNDF; 14150b57cec5SDimitry Andric B.NSect = 0; 14160b57cec5SDimitry Andric B.NDesc = 0; 14170b57cec5SDimitry Andric MachO::SET_LIBRARY_ORDINAL(B.NDesc, Entry.ordinal()); 1418e8d8bef9SDimitry Andric B.Name = Entry.symbolName().str(); 14190b57cec5SDimitry Andric SymbolList.push_back(B); 14200b57cec5SDimitry Andric BOS << Entry.symbolName(); 14210b57cec5SDimitry Andric BOS << '\0'; 14220b57cec5SDimitry Andric BindsAdded++; 14230b57cec5SDimitry Andric } 14240b57cec5SDimitry Andric } 14250b57cec5SDimitry Andric if (BErr) 1426e8d8bef9SDimitry Andric error(std::move(BErr), MachO.getFileName()); 14270b57cec5SDimitry Andric // Set the symbol names and indirect names for the added symbols. 14280b57cec5SDimitry Andric if (BindsAdded) { 14290b57cec5SDimitry Andric BOS.flush(); 14300b57cec5SDimitry Andric const char *Q = BindsNameBuffer.c_str(); 14310b57cec5SDimitry Andric for (unsigned K = 0; K < BindsAdded; K++) { 14320b57cec5SDimitry Andric SymbolList[I].Name = Q; 14330b57cec5SDimitry Andric Q += strlen(Q) + 1; 14340b57cec5SDimitry Andric if (SymbolList[I].TypeChar == 'I') { 14350b57cec5SDimitry Andric SymbolList[I].IndirectName = Q; 14360b57cec5SDimitry Andric Q += strlen(Q) + 1; 14370b57cec5SDimitry Andric } 14380b57cec5SDimitry Andric I++; 14390b57cec5SDimitry Andric } 14400b57cec5SDimitry Andric } 14410b57cec5SDimitry Andric 14420b57cec5SDimitry Andric // Add the undefined symbols from the lazy bind entries. 14430b57cec5SDimitry Andric unsigned LazysAdded = 0; 14440b57cec5SDimitry Andric Error LErr = Error::success(); 14450b57cec5SDimitry Andric LastSymbolName = StringRef(); 14460b57cec5SDimitry Andric for (const llvm::object::MachOBindEntry &Entry : 1447e8d8bef9SDimitry Andric MachO.lazyBindTable(LErr)) { 14480b57cec5SDimitry Andric bool found = false; 14490b57cec5SDimitry Andric if (LastSymbolName == Entry.symbolName()) 14500b57cec5SDimitry Andric found = true; 14510b57cec5SDimitry Andric else { 14520b57cec5SDimitry Andric // Here we must check to see it this symbol is already in the 14530b57cec5SDimitry Andric // SymbolList as it might have already have been added above via a 14540b57cec5SDimitry Andric // non-lazy (bind) entry. 14550b57cec5SDimitry Andric for (unsigned J = 0; J < SymbolList.size() && !found; ++J) { 14560b57cec5SDimitry Andric if (SymbolList[J].Name == Entry.symbolName()) 14570b57cec5SDimitry Andric found = true; 14580b57cec5SDimitry Andric } 14590b57cec5SDimitry Andric } 14600b57cec5SDimitry Andric if (!found) { 14610b57cec5SDimitry Andric LastSymbolName = Entry.symbolName(); 14620b57cec5SDimitry Andric NMSymbol L = {}; 1463e8d8bef9SDimitry Andric L.Name = Entry.symbolName().str(); 14640b57cec5SDimitry Andric L.Address = 0; 14650b57cec5SDimitry Andric L.Size = 0; 14660b57cec5SDimitry Andric L.TypeChar = 'U'; 14670b57cec5SDimitry Andric // There is no symbol in the nlist symbol table for this so we set 14680b57cec5SDimitry Andric // Sym effectivly to null and the rest of code in here must test for 14690b57cec5SDimitry Andric // it and not do things like Sym.getFlags() for it. 14700b57cec5SDimitry Andric L.Sym = BasicSymbolRef(); 14710b57cec5SDimitry Andric L.SymFlags = SymbolRef::SF_Global | SymbolRef::SF_Undefined; 14720b57cec5SDimitry Andric L.NType = MachO::N_EXT | MachO::N_UNDF; 14730b57cec5SDimitry Andric L.NSect = 0; 14740b57cec5SDimitry Andric // The REFERENCE_FLAG_UNDEFINED_LAZY is no longer used but here it 14750b57cec5SDimitry Andric // makes sence since we are creating this from a lazy bind entry. 14760b57cec5SDimitry Andric L.NDesc = MachO::REFERENCE_FLAG_UNDEFINED_LAZY; 14770b57cec5SDimitry Andric MachO::SET_LIBRARY_ORDINAL(L.NDesc, Entry.ordinal()); 14780b57cec5SDimitry Andric SymbolList.push_back(L); 14790b57cec5SDimitry Andric LOS << Entry.symbolName(); 14800b57cec5SDimitry Andric LOS << '\0'; 14810b57cec5SDimitry Andric LazysAdded++; 14820b57cec5SDimitry Andric } 14830b57cec5SDimitry Andric } 14840b57cec5SDimitry Andric if (LErr) 1485e8d8bef9SDimitry Andric error(std::move(LErr), MachO.getFileName()); 14860b57cec5SDimitry Andric // Set the symbol names and indirect names for the added symbols. 14870b57cec5SDimitry Andric if (LazysAdded) { 14880b57cec5SDimitry Andric LOS.flush(); 14890b57cec5SDimitry Andric const char *Q = LazysNameBuffer.c_str(); 14900b57cec5SDimitry Andric for (unsigned K = 0; K < LazysAdded; K++) { 14910b57cec5SDimitry Andric SymbolList[I].Name = Q; 14920b57cec5SDimitry Andric Q += strlen(Q) + 1; 14930b57cec5SDimitry Andric if (SymbolList[I].TypeChar == 'I') { 14940b57cec5SDimitry Andric SymbolList[I].IndirectName = Q; 14950b57cec5SDimitry Andric Q += strlen(Q) + 1; 14960b57cec5SDimitry Andric } 14970b57cec5SDimitry Andric I++; 14980b57cec5SDimitry Andric } 14990b57cec5SDimitry Andric } 15000b57cec5SDimitry Andric 15010b57cec5SDimitry Andric // Add the undefineds symbol from the weak bind entries which are not 15020b57cec5SDimitry Andric // strong symbols. 15030b57cec5SDimitry Andric unsigned WeaksAdded = 0; 15040b57cec5SDimitry Andric Error WErr = Error::success(); 15050b57cec5SDimitry Andric LastSymbolName = StringRef(); 15060b57cec5SDimitry Andric for (const llvm::object::MachOBindEntry &Entry : 1507e8d8bef9SDimitry Andric MachO.weakBindTable(WErr)) { 15080b57cec5SDimitry Andric bool found = false; 15090b57cec5SDimitry Andric unsigned J = 0; 15100b57cec5SDimitry Andric if (LastSymbolName == Entry.symbolName() || 15110b57cec5SDimitry Andric Entry.flags() & MachO::BIND_SYMBOL_FLAGS_NON_WEAK_DEFINITION) { 15120b57cec5SDimitry Andric found = true; 15130b57cec5SDimitry Andric } else { 15140b57cec5SDimitry Andric for (J = 0; J < SymbolList.size() && !found; ++J) { 15150b57cec5SDimitry Andric if (SymbolList[J].Name == Entry.symbolName()) { 15160b57cec5SDimitry Andric found = true; 15170b57cec5SDimitry Andric break; 15180b57cec5SDimitry Andric } 15190b57cec5SDimitry Andric } 15200b57cec5SDimitry Andric } 15210b57cec5SDimitry Andric if (!found) { 15220b57cec5SDimitry Andric LastSymbolName = Entry.symbolName(); 1523e8d8bef9SDimitry Andric NMSymbol W = {}; 1524e8d8bef9SDimitry Andric W.Name = Entry.symbolName().str(); 15250b57cec5SDimitry Andric W.Address = 0; 15260b57cec5SDimitry Andric W.Size = 0; 15270b57cec5SDimitry Andric W.TypeChar = 'U'; 15280b57cec5SDimitry Andric // There is no symbol in the nlist symbol table for this so we set 15290b57cec5SDimitry Andric // Sym effectivly to null and the rest of code in here must test for 15300b57cec5SDimitry Andric // it and not do things like Sym.getFlags() for it. 15310b57cec5SDimitry Andric W.Sym = BasicSymbolRef(); 15320b57cec5SDimitry Andric W.SymFlags = SymbolRef::SF_Global | SymbolRef::SF_Undefined; 15330b57cec5SDimitry Andric W.NType = MachO::N_EXT | MachO::N_UNDF; 15340b57cec5SDimitry Andric W.NSect = 0; 15350b57cec5SDimitry Andric // Odd that we are using N_WEAK_DEF on an undefined symbol but that is 15360b57cec5SDimitry Andric // what is created in this case by the linker when there are real 15370b57cec5SDimitry Andric // symbols in the nlist structs. 15380b57cec5SDimitry Andric W.NDesc = MachO::N_WEAK_DEF; 15390b57cec5SDimitry Andric SymbolList.push_back(W); 15400b57cec5SDimitry Andric WOS << Entry.symbolName(); 15410b57cec5SDimitry Andric WOS << '\0'; 15420b57cec5SDimitry Andric WeaksAdded++; 15430b57cec5SDimitry Andric } else { 15440b57cec5SDimitry Andric // This is the case the symbol was previously been found and it could 15450b57cec5SDimitry Andric // have been added from a bind or lazy bind symbol. If so and not 15460b57cec5SDimitry Andric // a definition also mark it as weak. 15470b57cec5SDimitry Andric if (SymbolList[J].TypeChar == 'U') 15480b57cec5SDimitry Andric // See comment above about N_WEAK_DEF. 15490b57cec5SDimitry Andric SymbolList[J].NDesc |= MachO::N_WEAK_DEF; 15500b57cec5SDimitry Andric } 15510b57cec5SDimitry Andric } 15520b57cec5SDimitry Andric if (WErr) 1553e8d8bef9SDimitry Andric error(std::move(WErr), MachO.getFileName()); 15540b57cec5SDimitry Andric // Set the symbol names and indirect names for the added symbols. 15550b57cec5SDimitry Andric if (WeaksAdded) { 15560b57cec5SDimitry Andric WOS.flush(); 15570b57cec5SDimitry Andric const char *Q = WeaksNameBuffer.c_str(); 15580b57cec5SDimitry Andric for (unsigned K = 0; K < WeaksAdded; K++) { 15590b57cec5SDimitry Andric SymbolList[I].Name = Q; 15600b57cec5SDimitry Andric Q += strlen(Q) + 1; 15610b57cec5SDimitry Andric if (SymbolList[I].TypeChar == 'I') { 15620b57cec5SDimitry Andric SymbolList[I].IndirectName = Q; 15630b57cec5SDimitry Andric Q += strlen(Q) + 1; 15640b57cec5SDimitry Andric } 15650b57cec5SDimitry Andric I++; 15660b57cec5SDimitry Andric } 15670b57cec5SDimitry Andric } 15680b57cec5SDimitry Andric 15690b57cec5SDimitry Andric // Trying adding symbol from the function starts table and LC_MAIN entry 15700b57cec5SDimitry Andric // point. 15710b57cec5SDimitry Andric SmallVector<uint64_t, 8> FoundFns; 15720b57cec5SDimitry Andric uint64_t lc_main_offset = UINT64_MAX; 1573e8d8bef9SDimitry Andric for (const auto &Command : MachO.load_commands()) { 15740b57cec5SDimitry Andric if (Command.C.cmd == MachO::LC_FUNCTION_STARTS) { 15750b57cec5SDimitry Andric // We found a function starts segment, parse the addresses for 15760b57cec5SDimitry Andric // consumption. 15770b57cec5SDimitry Andric MachO::linkedit_data_command LLC = 1578e8d8bef9SDimitry Andric MachO.getLinkeditDataLoadCommand(Command); 15790b57cec5SDimitry Andric 1580e8d8bef9SDimitry Andric MachO.ReadULEB128s(LLC.dataoff, FoundFns); 15810b57cec5SDimitry Andric } else if (Command.C.cmd == MachO::LC_MAIN) { 1582e8d8bef9SDimitry Andric MachO::entry_point_command LCmain = MachO.getEntryPointCommand(Command); 15830b57cec5SDimitry Andric lc_main_offset = LCmain.entryoff; 15840b57cec5SDimitry Andric } 15850b57cec5SDimitry Andric } 15860b57cec5SDimitry Andric // See if these addresses are already in the symbol table. 15870b57cec5SDimitry Andric unsigned FunctionStartsAdded = 0; 15880b57cec5SDimitry Andric for (uint64_t f = 0; f < FoundFns.size(); f++) { 15890b57cec5SDimitry Andric bool found = false; 15900b57cec5SDimitry Andric for (unsigned J = 0; J < SymbolList.size() && !found; ++J) { 15910b57cec5SDimitry Andric if (SymbolList[J].Address == FoundFns[f] + BaseSegmentAddress) 15920b57cec5SDimitry Andric found = true; 15930b57cec5SDimitry Andric } 15940b57cec5SDimitry Andric // See this address is not already in the symbol table fake up an 15950b57cec5SDimitry Andric // nlist for it. 15960b57cec5SDimitry Andric if (!found) { 15970b57cec5SDimitry Andric NMSymbol F = {}; 15980b57cec5SDimitry Andric F.Name = "<redacted function X>"; 15990b57cec5SDimitry Andric F.Address = FoundFns[f] + BaseSegmentAddress; 16000b57cec5SDimitry Andric F.Size = 0; 16010b57cec5SDimitry Andric // There is no symbol in the nlist symbol table for this so we set 16020b57cec5SDimitry Andric // Sym effectivly to null and the rest of code in here must test for 16030b57cec5SDimitry Andric // it and not do things like Sym.getFlags() for it. 16040b57cec5SDimitry Andric F.Sym = BasicSymbolRef(); 16050b57cec5SDimitry Andric F.SymFlags = 0; 16060b57cec5SDimitry Andric F.NType = MachO::N_SECT; 16070b57cec5SDimitry Andric F.NSect = 0; 16080b57cec5SDimitry Andric StringRef SegmentName = StringRef(); 16090b57cec5SDimitry Andric StringRef SectionName = StringRef(); 1610e8d8bef9SDimitry Andric for (const SectionRef &Section : MachO.sections()) { 16118bcb0991SDimitry Andric if (Expected<StringRef> NameOrErr = Section.getName()) 16128bcb0991SDimitry Andric SectionName = *NameOrErr; 16138bcb0991SDimitry Andric else 16148bcb0991SDimitry Andric consumeError(NameOrErr.takeError()); 16158bcb0991SDimitry Andric 1616e8d8bef9SDimitry Andric SegmentName = 1617e8d8bef9SDimitry Andric MachO.getSectionFinalSegmentName(Section.getRawDataRefImpl()); 16180b57cec5SDimitry Andric F.NSect++; 16190b57cec5SDimitry Andric if (F.Address >= Section.getAddress() && 16200b57cec5SDimitry Andric F.Address < Section.getAddress() + Section.getSize()) { 16210b57cec5SDimitry Andric F.Section = Section; 16220b57cec5SDimitry Andric break; 16230b57cec5SDimitry Andric } 16240b57cec5SDimitry Andric } 16250b57cec5SDimitry Andric if (SegmentName == "__TEXT" && SectionName == "__text") 16260b57cec5SDimitry Andric F.TypeChar = 't'; 16270b57cec5SDimitry Andric else if (SegmentName == "__DATA" && SectionName == "__data") 16280b57cec5SDimitry Andric F.TypeChar = 'd'; 16290b57cec5SDimitry Andric else if (SegmentName == "__DATA" && SectionName == "__bss") 16300b57cec5SDimitry Andric F.TypeChar = 'b'; 16310b57cec5SDimitry Andric else 16320b57cec5SDimitry Andric F.TypeChar = 's'; 16330b57cec5SDimitry Andric F.NDesc = 0; 16340b57cec5SDimitry Andric SymbolList.push_back(F); 16350b57cec5SDimitry Andric if (FoundFns[f] == lc_main_offset) 16360b57cec5SDimitry Andric FOS << "<redacted LC_MAIN>"; 16370b57cec5SDimitry Andric else 16380b57cec5SDimitry Andric FOS << "<redacted function " << f << ">"; 16390b57cec5SDimitry Andric FOS << '\0'; 16400b57cec5SDimitry Andric FunctionStartsAdded++; 16410b57cec5SDimitry Andric } 16420b57cec5SDimitry Andric } 16430b57cec5SDimitry Andric if (FunctionStartsAdded) { 16440b57cec5SDimitry Andric FOS.flush(); 16450b57cec5SDimitry Andric const char *Q = FunctionStartsNameBuffer.c_str(); 16460b57cec5SDimitry Andric for (unsigned K = 0; K < FunctionStartsAdded; K++) { 16470b57cec5SDimitry Andric SymbolList[I].Name = Q; 16480b57cec5SDimitry Andric Q += strlen(Q) + 1; 16490b57cec5SDimitry Andric if (SymbolList[I].TypeChar == 'I') { 16500b57cec5SDimitry Andric SymbolList[I].IndirectName = Q; 16510b57cec5SDimitry Andric Q += strlen(Q) + 1; 16520b57cec5SDimitry Andric } 16530b57cec5SDimitry Andric I++; 16540b57cec5SDimitry Andric } 16550b57cec5SDimitry Andric } 16560b57cec5SDimitry Andric } 16570b57cec5SDimitry Andric } 16580b57cec5SDimitry Andric 165981ad6265SDimitry Andric static bool shouldDump(SymbolicFile &Obj) { 166081ad6265SDimitry Andric // The -X option is currently only implemented for XCOFF, ELF, and IR object 166181ad6265SDimitry Andric // files. The option isn't fundamentally impossible with other formats, just 166281ad6265SDimitry Andric // isn't implemented. 166381ad6265SDimitry Andric if (!isa<XCOFFObjectFile>(Obj) && !isa<ELFObjectFileBase>(Obj) && 166481ad6265SDimitry Andric !isa<IRObjectFile>(Obj)) 166581ad6265SDimitry Andric return true; 166681ad6265SDimitry Andric 1667*06c3fb27SDimitry Andric return Obj.is64Bit() ? BitMode != BitModeTy::Bit32 166881ad6265SDimitry Andric : BitMode != BitModeTy::Bit64; 166981ad6265SDimitry Andric } 167081ad6265SDimitry Andric 167181ad6265SDimitry Andric static void getXCOFFExports(XCOFFObjectFile *XCOFFObj, 167281ad6265SDimitry Andric std::vector<NMSymbol> &SymbolList, 167381ad6265SDimitry Andric StringRef ArchiveName) { 167481ad6265SDimitry Andric // Skip Shared object file. 167581ad6265SDimitry Andric if (XCOFFObj->getFlags() & XCOFF::F_SHROBJ) 167681ad6265SDimitry Andric return; 167781ad6265SDimitry Andric 167881ad6265SDimitry Andric for (SymbolRef Sym : XCOFFObj->symbols()) { 167981ad6265SDimitry Andric // There is no visibility in old 32 bit XCOFF object file interpret. 168081ad6265SDimitry Andric bool HasVisibilityAttr = 168181ad6265SDimitry Andric XCOFFObj->is64Bit() || (XCOFFObj->auxiliaryHeader32() && 168281ad6265SDimitry Andric (XCOFFObj->auxiliaryHeader32()->getVersion() == 168381ad6265SDimitry Andric XCOFF::NEW_XCOFF_INTERPRET)); 168481ad6265SDimitry Andric 168581ad6265SDimitry Andric if (HasVisibilityAttr) { 168681ad6265SDimitry Andric XCOFFSymbolRef XCOFFSym = XCOFFObj->toSymbolRef(Sym.getRawDataRefImpl()); 168781ad6265SDimitry Andric uint16_t SymType = XCOFFSym.getSymbolType(); 168881ad6265SDimitry Andric if ((SymType & XCOFF::VISIBILITY_MASK) == XCOFF::SYM_V_INTERNAL) 168981ad6265SDimitry Andric continue; 169081ad6265SDimitry Andric if ((SymType & XCOFF::VISIBILITY_MASK) == XCOFF::SYM_V_HIDDEN) 169181ad6265SDimitry Andric continue; 169281ad6265SDimitry Andric } 169381ad6265SDimitry Andric 169481ad6265SDimitry Andric Expected<section_iterator> SymSecOrErr = Sym.getSection(); 169581ad6265SDimitry Andric if (!SymSecOrErr) { 169681ad6265SDimitry Andric warn(SymSecOrErr.takeError(), XCOFFObj->getFileName(), 169781ad6265SDimitry Andric "for symbol with index " + 169881ad6265SDimitry Andric Twine(XCOFFObj->getSymbolIndex(Sym.getRawDataRefImpl().p)), 169981ad6265SDimitry Andric ArchiveName); 170081ad6265SDimitry Andric continue; 170181ad6265SDimitry Andric } 170281ad6265SDimitry Andric section_iterator SecIter = *SymSecOrErr; 170381ad6265SDimitry Andric // If the symbol is not in a text or data section, it is not exported. 170481ad6265SDimitry Andric if (SecIter == XCOFFObj->section_end()) 170581ad6265SDimitry Andric continue; 170681ad6265SDimitry Andric if (!(SecIter->isText() || SecIter->isData() || SecIter->isBSS())) 170781ad6265SDimitry Andric continue; 170881ad6265SDimitry Andric 170981ad6265SDimitry Andric StringRef SymName = cantFail(Sym.getName()); 171081ad6265SDimitry Andric if (SymName.empty()) 171181ad6265SDimitry Andric continue; 171281ad6265SDimitry Andric if (SymName.startswith("__sinit") || SymName.startswith("__sterm") || 171381ad6265SDimitry Andric SymName.front() == '.' || SymName.front() == '(') 171481ad6265SDimitry Andric continue; 171581ad6265SDimitry Andric 171681ad6265SDimitry Andric // Check the SymName regex matching with "^__[0-9]+__". 171781ad6265SDimitry Andric if (SymName.size() > 4 && SymName.startswith("__") && 171881ad6265SDimitry Andric SymName.endswith("__")) { 171981ad6265SDimitry Andric if (std::all_of(SymName.begin() + 2, SymName.end() - 2, isDigit)) 172081ad6265SDimitry Andric continue; 172181ad6265SDimitry Andric } 172281ad6265SDimitry Andric 172381ad6265SDimitry Andric if (SymName == "__rsrc" && NoRsrc) 172481ad6265SDimitry Andric continue; 172581ad6265SDimitry Andric 172681ad6265SDimitry Andric if (SymName.startswith("__tf1")) 172781ad6265SDimitry Andric SymName = SymName.substr(6); 172881ad6265SDimitry Andric else if (SymName.startswith("__tf9")) 172981ad6265SDimitry Andric SymName = SymName.substr(14); 173081ad6265SDimitry Andric 173181ad6265SDimitry Andric NMSymbol S = {}; 173281ad6265SDimitry Andric S.Name = SymName.str(); 173381ad6265SDimitry Andric S.Sym = Sym; 173481ad6265SDimitry Andric 173581ad6265SDimitry Andric if (HasVisibilityAttr) { 173681ad6265SDimitry Andric XCOFFSymbolRef XCOFFSym = XCOFFObj->toSymbolRef(Sym.getRawDataRefImpl()); 173781ad6265SDimitry Andric uint16_t SymType = XCOFFSym.getSymbolType(); 173881ad6265SDimitry Andric if ((SymType & XCOFF::VISIBILITY_MASK) == XCOFF::SYM_V_PROTECTED) 173981ad6265SDimitry Andric S.Visibility = "protected"; 174081ad6265SDimitry Andric else if ((SymType & XCOFF::VISIBILITY_MASK) == XCOFF::SYM_V_EXPORTED) 174181ad6265SDimitry Andric S.Visibility = "export"; 174281ad6265SDimitry Andric } 174381ad6265SDimitry Andric if (S.initializeFlags(*XCOFFObj)) 174481ad6265SDimitry Andric SymbolList.push_back(S); 174581ad6265SDimitry Andric } 174681ad6265SDimitry Andric } 174781ad6265SDimitry Andric 174881ad6265SDimitry Andric static Expected<SymbolicFile::basic_symbol_iterator_range> 174981ad6265SDimitry Andric getDynamicSyms(SymbolicFile &Obj) { 175081ad6265SDimitry Andric const auto *E = dyn_cast<ELFObjectFileBase>(&Obj); 175181ad6265SDimitry Andric if (!E) 175281ad6265SDimitry Andric return createError("File format has no dynamic symbol table"); 175381ad6265SDimitry Andric return E->getDynamicSymbolIterators(); 175481ad6265SDimitry Andric } 175581ad6265SDimitry Andric 175681ad6265SDimitry Andric // Returns false if there is error found or true otherwise. 175781ad6265SDimitry Andric static bool getSymbolNamesFromObject(SymbolicFile &Obj, 175881ad6265SDimitry Andric std::vector<NMSymbol> &SymbolList) { 1759e8d8bef9SDimitry Andric auto Symbols = Obj.symbols(); 1760349cc55cSDimitry Andric std::vector<VersionEntry> SymbolVersions; 1761e8d8bef9SDimitry Andric 176281ad6265SDimitry Andric if (DynamicSyms) { 176381ad6265SDimitry Andric Expected<SymbolicFile::basic_symbol_iterator_range> SymbolsOrErr = 176481ad6265SDimitry Andric getDynamicSyms(Obj); 176581ad6265SDimitry Andric if (!SymbolsOrErr) { 176681ad6265SDimitry Andric error(SymbolsOrErr.takeError(), Obj.getFileName()); 176781ad6265SDimitry Andric return false; 176881ad6265SDimitry Andric } 176981ad6265SDimitry Andric Symbols = *SymbolsOrErr; 177081ad6265SDimitry Andric if (const auto *E = dyn_cast<ELFObjectFileBase>(&Obj)) { 1771349cc55cSDimitry Andric if (Expected<std::vector<VersionEntry>> VersionsOrErr = 1772349cc55cSDimitry Andric E->readDynsymVersions()) 1773e8d8bef9SDimitry Andric SymbolVersions = std::move(*VersionsOrErr); 1774e8d8bef9SDimitry Andric else 1775e8d8bef9SDimitry Andric WithColor::warning(errs(), ToolName) 1776e8d8bef9SDimitry Andric << "unable to read symbol versions: " 1777e8d8bef9SDimitry Andric << toString(VersionsOrErr.takeError()) << "\n"; 1778e8d8bef9SDimitry Andric } 177981ad6265SDimitry Andric } 1780e8d8bef9SDimitry Andric // If a "-s segname sectname" option was specified and this is a Mach-O 1781e8d8bef9SDimitry Andric // file get the section number for that section in this object file. 1782e8d8bef9SDimitry Andric unsigned int Nsect = 0; 1783e8d8bef9SDimitry Andric MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(&Obj); 1784e8d8bef9SDimitry Andric if (!SegSect.empty() && MachO) { 1785e8d8bef9SDimitry Andric Nsect = getNsectForSegSect(MachO); 1786e8d8bef9SDimitry Andric // If this section is not in the object file no symbols are printed. 1787e8d8bef9SDimitry Andric if (Nsect == 0) 178881ad6265SDimitry Andric return false; 1789e8d8bef9SDimitry Andric } 179081ad6265SDimitry Andric 1791e8d8bef9SDimitry Andric if (!(MachO && DyldInfoOnly)) { 1792e8d8bef9SDimitry Andric size_t I = -1; 1793e8d8bef9SDimitry Andric for (BasicSymbolRef Sym : Symbols) { 1794e8d8bef9SDimitry Andric ++I; 1795e8d8bef9SDimitry Andric Expected<uint32_t> SymFlagsOrErr = Sym.getFlags(); 1796e8d8bef9SDimitry Andric if (!SymFlagsOrErr) { 1797e8d8bef9SDimitry Andric error(SymFlagsOrErr.takeError(), Obj.getFileName()); 179881ad6265SDimitry Andric return false; 1799e8d8bef9SDimitry Andric } 1800fe6060f1SDimitry Andric 1801fe6060f1SDimitry Andric // Don't drop format specifc symbols for ARM and AArch64 ELF targets, they 1802fe6060f1SDimitry Andric // are used to repesent mapping symbols and needed to honor the 1803fe6060f1SDimitry Andric // --special-syms option. 1804fe6060f1SDimitry Andric auto *ELFObj = dyn_cast<ELFObjectFileBase>(&Obj); 1805fe6060f1SDimitry Andric if ((!ELFObj || (ELFObj->getEMachine() != ELF::EM_ARM && 1806fe6060f1SDimitry Andric ELFObj->getEMachine() != ELF::EM_AARCH64)) && 1807fe6060f1SDimitry Andric !DebugSyms && (*SymFlagsOrErr & SymbolRef::SF_FormatSpecific)) 1808e8d8bef9SDimitry Andric continue; 1809e8d8bef9SDimitry Andric if (WithoutAliases && (*SymFlagsOrErr & SymbolRef::SF_Indirect)) 1810e8d8bef9SDimitry Andric continue; 1811e8d8bef9SDimitry Andric // If a "-s segname sectname" option was specified and this is a Mach-O 1812e8d8bef9SDimitry Andric // file and this section appears in this file, Nsect will be non-zero then 1813e8d8bef9SDimitry Andric // see if this symbol is a symbol from that section and if not skip it. 1814e8d8bef9SDimitry Andric if (Nsect && Nsect != getNsectInMachO(*MachO, Sym)) 1815e8d8bef9SDimitry Andric continue; 1816e8d8bef9SDimitry Andric NMSymbol S = {}; 1817e8d8bef9SDimitry Andric S.Size = 0; 1818e8d8bef9SDimitry Andric S.Address = 0; 1819e8d8bef9SDimitry Andric if (isa<ELFObjectFileBase>(&Obj)) 1820e8d8bef9SDimitry Andric S.Size = ELFSymbolRef(Sym).getSize(); 182104eeddc0SDimitry Andric 182204eeddc0SDimitry Andric if (const XCOFFObjectFile *XCOFFObj = 182304eeddc0SDimitry Andric dyn_cast<const XCOFFObjectFile>(&Obj)) 182404eeddc0SDimitry Andric S.Size = XCOFFObj->getSymbolSize(Sym.getRawDataRefImpl()); 182504eeddc0SDimitry Andric 1826e8d8bef9SDimitry Andric if (PrintAddress && isa<ObjectFile>(Obj)) { 1827e8d8bef9SDimitry Andric SymbolRef SymRef(Sym); 1828e8d8bef9SDimitry Andric Expected<uint64_t> AddressOrErr = SymRef.getAddress(); 1829e8d8bef9SDimitry Andric if (!AddressOrErr) { 1830e8d8bef9SDimitry Andric consumeError(AddressOrErr.takeError()); 1831e8d8bef9SDimitry Andric break; 1832e8d8bef9SDimitry Andric } 1833e8d8bef9SDimitry Andric S.Address = *AddressOrErr; 1834e8d8bef9SDimitry Andric } 1835e8d8bef9SDimitry Andric S.TypeName = getNMTypeName(Obj, Sym); 1836e8d8bef9SDimitry Andric S.TypeChar = getNMSectionTagAndName(Obj, Sym, S.SectionName); 1837e8d8bef9SDimitry Andric 1838e8d8bef9SDimitry Andric raw_string_ostream OS(S.Name); 1839e8d8bef9SDimitry Andric if (Error E = Sym.printName(OS)) { 1840e8d8bef9SDimitry Andric if (MachO) { 1841e8d8bef9SDimitry Andric OS << "bad string index"; 1842e8d8bef9SDimitry Andric consumeError(std::move(E)); 1843e8d8bef9SDimitry Andric } else 1844e8d8bef9SDimitry Andric error(std::move(E), Obj.getFileName()); 1845e8d8bef9SDimitry Andric } 1846e8d8bef9SDimitry Andric if (!SymbolVersions.empty() && !SymbolVersions[I].Name.empty()) 1847e8d8bef9SDimitry Andric S.Name += 1848349cc55cSDimitry Andric (SymbolVersions[I].IsVerDef ? "@@" : "@") + SymbolVersions[I].Name; 1849e8d8bef9SDimitry Andric 1850e8d8bef9SDimitry Andric S.Sym = Sym; 185181ad6265SDimitry Andric if (S.initializeFlags(Obj)) 1852e8d8bef9SDimitry Andric SymbolList.push_back(S); 1853e8d8bef9SDimitry Andric } 1854e8d8bef9SDimitry Andric } 1855e8d8bef9SDimitry Andric 1856e8d8bef9SDimitry Andric // If this is a Mach-O file where the nlist symbol table is out of sync 1857e8d8bef9SDimitry Andric // with the dyld export trie then look through exports and fake up symbols 1858e8d8bef9SDimitry Andric // for the ones that are missing (also done with the -add-dyldinfo flag). 1859e8d8bef9SDimitry Andric // This is needed if strip(1) -T is run on a binary containing swift 1860e8d8bef9SDimitry Andric // language symbols for example. The option -only-dyldinfo will fake up 1861e8d8bef9SDimitry Andric // all symbols from the dyld export trie as well as the bind info. 1862e8d8bef9SDimitry Andric if (MachO && !NoDyldInfo) 186381ad6265SDimitry Andric dumpSymbolsFromDLInfoMachO(*MachO, SymbolList); 1864e8d8bef9SDimitry Andric 186581ad6265SDimitry Andric return true; 186681ad6265SDimitry Andric } 186781ad6265SDimitry Andric 186881ad6265SDimitry Andric static void printObjectLabel(bool PrintArchiveName, StringRef ArchiveName, 186981ad6265SDimitry Andric StringRef ArchitectureName, 187081ad6265SDimitry Andric StringRef ObjectFileName) { 187181ad6265SDimitry Andric outs() << "\n"; 187281ad6265SDimitry Andric if (ArchiveName.empty() || !PrintArchiveName) 187381ad6265SDimitry Andric outs() << ObjectFileName; 187481ad6265SDimitry Andric else 187581ad6265SDimitry Andric outs() << ArchiveName << "(" << ObjectFileName << ")"; 187681ad6265SDimitry Andric if (!ArchitectureName.empty()) 187781ad6265SDimitry Andric outs() << " (for architecture " << ArchitectureName << ")"; 187881ad6265SDimitry Andric outs() << ":\n"; 187981ad6265SDimitry Andric } 188081ad6265SDimitry Andric 188181ad6265SDimitry Andric static Expected<bool> hasSymbols(SymbolicFile &Obj) { 188281ad6265SDimitry Andric if (DynamicSyms) { 188381ad6265SDimitry Andric Expected<SymbolicFile::basic_symbol_iterator_range> DynamicSymsOrErr = 188481ad6265SDimitry Andric getDynamicSyms(Obj); 188581ad6265SDimitry Andric if (!DynamicSymsOrErr) 188681ad6265SDimitry Andric return DynamicSymsOrErr.takeError(); 188781ad6265SDimitry Andric return !DynamicSymsOrErr->empty(); 188881ad6265SDimitry Andric } 188981ad6265SDimitry Andric return !Obj.symbols().empty(); 189081ad6265SDimitry Andric } 189181ad6265SDimitry Andric 189281ad6265SDimitry Andric static void dumpSymbolNamesFromObject( 189381ad6265SDimitry Andric SymbolicFile &Obj, std::vector<NMSymbol> &SymbolList, 189481ad6265SDimitry Andric bool PrintSymbolObject, bool PrintObjectLabel, StringRef ArchiveName = {}, 189581ad6265SDimitry Andric StringRef ArchitectureName = {}, StringRef ObjectName = {}, 189681ad6265SDimitry Andric bool PrintArchiveName = true) { 189781ad6265SDimitry Andric if (!shouldDump(Obj)) 189881ad6265SDimitry Andric return; 189981ad6265SDimitry Andric 190081ad6265SDimitry Andric if (ExportSymbols && Obj.isXCOFF()) { 190181ad6265SDimitry Andric XCOFFObjectFile *XCOFFObj = cast<XCOFFObjectFile>(&Obj); 190281ad6265SDimitry Andric getXCOFFExports(XCOFFObj, SymbolList, ArchiveName); 190381ad6265SDimitry Andric return; 190481ad6265SDimitry Andric } 190581ad6265SDimitry Andric 190681ad6265SDimitry Andric if (PrintObjectLabel && !ExportSymbols) 190781ad6265SDimitry Andric printObjectLabel(PrintArchiveName, ArchiveName, ArchitectureName, 190881ad6265SDimitry Andric ObjectName.empty() ? Obj.getFileName() : ObjectName); 190981ad6265SDimitry Andric if (!getSymbolNamesFromObject(Obj, SymbolList) || ExportSymbols) 191081ad6265SDimitry Andric return; 19110b57cec5SDimitry Andric CurrentFilename = Obj.getFileName(); 19125ffd83dbSDimitry Andric 191381ad6265SDimitry Andric // If there is an error in hasSymbols(), the error should be encountered in 191481ad6265SDimitry Andric // function getSymbolNamesFromObject first. 191581ad6265SDimitry Andric if (!cantFail(hasSymbols(Obj)) && SymbolList.empty() && !Quiet) { 19165ffd83dbSDimitry Andric writeFileName(errs(), ArchiveName, ArchitectureName); 19175ffd83dbSDimitry Andric errs() << "no symbols\n"; 19185ffd83dbSDimitry Andric } 19195ffd83dbSDimitry Andric 192081ad6265SDimitry Andric sortSymbolList(SymbolList); 192181ad6265SDimitry Andric printSymbolList(Obj, SymbolList, PrintSymbolObject, ArchiveName, 192281ad6265SDimitry Andric ArchitectureName); 19230b57cec5SDimitry Andric } 19240b57cec5SDimitry Andric 19250b57cec5SDimitry Andric // checkMachOAndArchFlags() checks to see if the SymbolicFile is a Mach-O file 19260b57cec5SDimitry Andric // and if it is and there is a list of architecture flags is specified then 19270b57cec5SDimitry Andric // check to make sure this Mach-O file is one of those architectures or all 19280b57cec5SDimitry Andric // architectures was specificed. If not then an error is generated and this 19290b57cec5SDimitry Andric // routine returns false. Else it returns true. 193081ad6265SDimitry Andric static bool checkMachOAndArchFlags(SymbolicFile *O, StringRef Filename) { 19310b57cec5SDimitry Andric auto *MachO = dyn_cast<MachOObjectFile>(O); 19320b57cec5SDimitry Andric 19330b57cec5SDimitry Andric if (!MachO || ArchAll || ArchFlags.empty()) 19340b57cec5SDimitry Andric return true; 19350b57cec5SDimitry Andric 19360b57cec5SDimitry Andric MachO::mach_header H; 19370b57cec5SDimitry Andric MachO::mach_header_64 H_64; 19380b57cec5SDimitry Andric Triple T; 19390b57cec5SDimitry Andric const char *McpuDefault, *ArchFlag; 19400b57cec5SDimitry Andric if (MachO->is64Bit()) { 19410b57cec5SDimitry Andric H_64 = MachO->MachOObjectFile::getHeader64(); 19420b57cec5SDimitry Andric T = MachOObjectFile::getArchTriple(H_64.cputype, H_64.cpusubtype, 19430b57cec5SDimitry Andric &McpuDefault, &ArchFlag); 19440b57cec5SDimitry Andric } else { 19450b57cec5SDimitry Andric H = MachO->MachOObjectFile::getHeader(); 19460b57cec5SDimitry Andric T = MachOObjectFile::getArchTriple(H.cputype, H.cpusubtype, 19470b57cec5SDimitry Andric &McpuDefault, &ArchFlag); 19480b57cec5SDimitry Andric } 19490b57cec5SDimitry Andric const std::string ArchFlagName(ArchFlag); 1950e8d8bef9SDimitry Andric if (!llvm::is_contained(ArchFlags, ArchFlagName)) { 19510b57cec5SDimitry Andric error("No architecture specified", Filename); 19520b57cec5SDimitry Andric return false; 19530b57cec5SDimitry Andric } 19540b57cec5SDimitry Andric return true; 19550b57cec5SDimitry Andric } 19560b57cec5SDimitry Andric 1957*06c3fb27SDimitry Andric static void printArchiveMap(iterator_range<Archive::symbol_iterator> &map, 1958*06c3fb27SDimitry Andric StringRef Filename) { 1959*06c3fb27SDimitry Andric for (auto I : map) { 1960*06c3fb27SDimitry Andric Expected<Archive::Child> C = I.getMember(); 19610b57cec5SDimitry Andric if (!C) { 19620b57cec5SDimitry Andric error(C.takeError(), Filename); 19630b57cec5SDimitry Andric break; 19640b57cec5SDimitry Andric } 19650b57cec5SDimitry Andric Expected<StringRef> FileNameOrErr = C->getName(); 19660b57cec5SDimitry Andric if (!FileNameOrErr) { 19670b57cec5SDimitry Andric error(FileNameOrErr.takeError(), Filename); 19680b57cec5SDimitry Andric break; 19690b57cec5SDimitry Andric } 1970*06c3fb27SDimitry Andric StringRef SymName = I.getName(); 19710b57cec5SDimitry Andric outs() << SymName << " in " << FileNameOrErr.get() << "\n"; 19720b57cec5SDimitry Andric } 1973*06c3fb27SDimitry Andric 19740b57cec5SDimitry Andric outs() << "\n"; 19750b57cec5SDimitry Andric } 1976*06c3fb27SDimitry Andric 1977*06c3fb27SDimitry Andric static void dumpArchiveMap(Archive *A, StringRef Filename) { 1978*06c3fb27SDimitry Andric auto Map = A->symbols(); 1979*06c3fb27SDimitry Andric if (!Map.empty()) { 1980*06c3fb27SDimitry Andric outs() << "Archive map\n"; 1981*06c3fb27SDimitry Andric printArchiveMap(Map, Filename); 1982*06c3fb27SDimitry Andric } 1983*06c3fb27SDimitry Andric 1984*06c3fb27SDimitry Andric auto ECMap = A->ec_symbols(); 1985*06c3fb27SDimitry Andric if (!ECMap) { 1986*06c3fb27SDimitry Andric warn(ECMap.takeError(), Filename); 1987*06c3fb27SDimitry Andric } else if (!ECMap->empty()) { 1988*06c3fb27SDimitry Andric outs() << "Archive EC map\n"; 1989*06c3fb27SDimitry Andric printArchiveMap(*ECMap, Filename); 1990*06c3fb27SDimitry Andric } 19910b57cec5SDimitry Andric } 19920b57cec5SDimitry Andric 199381ad6265SDimitry Andric static void dumpArchive(Archive *A, std::vector<NMSymbol> &SymbolList, 199481ad6265SDimitry Andric StringRef Filename, LLVMContext *ContextPtr) { 199581ad6265SDimitry Andric if (ArchiveMap) 199681ad6265SDimitry Andric dumpArchiveMap(A, Filename); 199781ad6265SDimitry Andric 19980b57cec5SDimitry Andric Error Err = Error::success(); 19990b57cec5SDimitry Andric for (auto &C : A->children(Err)) { 200081ad6265SDimitry Andric Expected<std::unique_ptr<Binary>> ChildOrErr = C.getAsBinary(ContextPtr); 20010b57cec5SDimitry Andric if (!ChildOrErr) { 20020b57cec5SDimitry Andric if (auto E = isNotObjectErrorInvalidFileType(ChildOrErr.takeError())) 20030b57cec5SDimitry Andric error(std::move(E), Filename, C); 20040b57cec5SDimitry Andric continue; 20050b57cec5SDimitry Andric } 20060b57cec5SDimitry Andric if (SymbolicFile *O = dyn_cast<SymbolicFile>(&*ChildOrErr.get())) { 20070b57cec5SDimitry Andric if (!MachOPrintSizeWarning && PrintSize && isa<MachOObjectFile>(O)) { 20080b57cec5SDimitry Andric WithColor::warning(errs(), ToolName) 20090b57cec5SDimitry Andric << "sizes with -print-size for Mach-O files are always zero.\n"; 20100b57cec5SDimitry Andric MachOPrintSizeWarning = true; 20110b57cec5SDimitry Andric } 20120b57cec5SDimitry Andric if (!checkMachOAndArchFlags(O, Filename)) 20130b57cec5SDimitry Andric return; 201481ad6265SDimitry Andric dumpSymbolNamesFromObject(*O, SymbolList, /*PrintSymbolObject=*/false, 201581ad6265SDimitry Andric !PrintFileName, Filename, 201681ad6265SDimitry Andric /*ArchitectureName=*/{}, O->getFileName(), 201781ad6265SDimitry Andric /*PrintArchiveName=*/false); 20180b57cec5SDimitry Andric } 20190b57cec5SDimitry Andric } 20200b57cec5SDimitry Andric if (Err) 20210b57cec5SDimitry Andric error(std::move(Err), A->getFileName()); 20220b57cec5SDimitry Andric } 202381ad6265SDimitry Andric 202481ad6265SDimitry Andric static void dumpMachOUniversalBinaryMatchArchFlags( 202581ad6265SDimitry Andric MachOUniversalBinary *UB, std::vector<NMSymbol> &SymbolList, 202681ad6265SDimitry Andric StringRef Filename, LLVMContext *ContextPtr) { 20270b57cec5SDimitry Andric // Look for a slice in the universal binary that matches each ArchFlag. 20280b57cec5SDimitry Andric bool ArchFound; 20290b57cec5SDimitry Andric for (unsigned i = 0; i < ArchFlags.size(); ++i) { 20300b57cec5SDimitry Andric ArchFound = false; 20310b57cec5SDimitry Andric for (MachOUniversalBinary::object_iterator I = UB->begin_objects(), 20320b57cec5SDimitry Andric E = UB->end_objects(); 20330b57cec5SDimitry Andric I != E; ++I) { 20340b57cec5SDimitry Andric if (ArchFlags[i] == I->getArchFlagName()) { 20350b57cec5SDimitry Andric ArchFound = true; 203681ad6265SDimitry Andric Expected<std::unique_ptr<ObjectFile>> ObjOrErr = I->getAsObjectFile(); 20370b57cec5SDimitry Andric std::string ArchiveName; 20380b57cec5SDimitry Andric std::string ArchitectureName; 20390b57cec5SDimitry Andric ArchiveName.clear(); 20400b57cec5SDimitry Andric ArchitectureName.clear(); 20410b57cec5SDimitry Andric if (ObjOrErr) { 20420b57cec5SDimitry Andric ObjectFile &Obj = *ObjOrErr.get(); 204381ad6265SDimitry Andric if (ArchFlags.size() > 1) 20440b57cec5SDimitry Andric ArchitectureName = I->getArchFlagName(); 204581ad6265SDimitry Andric dumpSymbolNamesFromObject(Obj, SymbolList, 204681ad6265SDimitry Andric /*PrintSymbolObject=*/false, 204781ad6265SDimitry Andric (ArchFlags.size() > 1) && !PrintFileName, 204881ad6265SDimitry Andric ArchiveName, ArchitectureName); 204981ad6265SDimitry Andric } else if (auto E = 205081ad6265SDimitry Andric isNotObjectErrorInvalidFileType(ObjOrErr.takeError())) { 205181ad6265SDimitry Andric error(std::move(E), Filename, 205281ad6265SDimitry Andric ArchFlags.size() > 1 ? StringRef(I->getArchFlagName()) 205381ad6265SDimitry Andric : StringRef()); 20540b57cec5SDimitry Andric continue; 20550b57cec5SDimitry Andric } else if (Expected<std::unique_ptr<Archive>> AOrErr = 20560b57cec5SDimitry Andric I->getAsArchive()) { 20570b57cec5SDimitry Andric std::unique_ptr<Archive> &A = *AOrErr; 20580b57cec5SDimitry Andric Error Err = Error::success(); 20590b57cec5SDimitry Andric for (auto &C : A->children(Err)) { 20600b57cec5SDimitry Andric Expected<std::unique_ptr<Binary>> ChildOrErr = 20610b57cec5SDimitry Andric C.getAsBinary(ContextPtr); 20620b57cec5SDimitry Andric if (!ChildOrErr) { 206381ad6265SDimitry Andric if (auto E = 206481ad6265SDimitry Andric isNotObjectErrorInvalidFileType(ChildOrErr.takeError())) { 206581ad6265SDimitry Andric error(std::move(E), Filename, C, 206681ad6265SDimitry Andric ArchFlags.size() > 1 ? StringRef(I->getArchFlagName()) 206781ad6265SDimitry Andric : StringRef()); 20680b57cec5SDimitry Andric } 20690b57cec5SDimitry Andric continue; 20700b57cec5SDimitry Andric } 207181ad6265SDimitry Andric if (SymbolicFile *O = dyn_cast<SymbolicFile>(&*ChildOrErr.get())) { 20725ffd83dbSDimitry Andric ArchiveName = std::string(A->getFileName()); 20730b57cec5SDimitry Andric if (ArchFlags.size() > 1) 20740b57cec5SDimitry Andric ArchitectureName = I->getArchFlagName(); 207581ad6265SDimitry Andric dumpSymbolNamesFromObject( 207681ad6265SDimitry Andric *O, SymbolList, /*PrintSymbolObject=*/false, !PrintFileName, 207781ad6265SDimitry Andric ArchiveName, ArchitectureName); 20780b57cec5SDimitry Andric } 20790b57cec5SDimitry Andric } 20800b57cec5SDimitry Andric if (Err) 20810b57cec5SDimitry Andric error(std::move(Err), A->getFileName()); 20820b57cec5SDimitry Andric } else { 20830b57cec5SDimitry Andric consumeError(AOrErr.takeError()); 20840b57cec5SDimitry Andric error(Filename + " for architecture " + 20850b57cec5SDimitry Andric StringRef(I->getArchFlagName()) + 20860b57cec5SDimitry Andric " is not a Mach-O file or an archive file", 20870b57cec5SDimitry Andric "Mach-O universal file"); 20880b57cec5SDimitry Andric } 20890b57cec5SDimitry Andric } 20900b57cec5SDimitry Andric } 20910b57cec5SDimitry Andric if (!ArchFound) { 20920b57cec5SDimitry Andric error(ArchFlags[i], 20930b57cec5SDimitry Andric "file: " + Filename + " does not contain architecture"); 20940b57cec5SDimitry Andric return; 20950b57cec5SDimitry Andric } 20960b57cec5SDimitry Andric } 20970b57cec5SDimitry Andric } 209881ad6265SDimitry Andric 209981ad6265SDimitry Andric // Returns true If the binary contains a slice that matches the host 210081ad6265SDimitry Andric // architecture, or false otherwise. 210181ad6265SDimitry Andric static bool dumpMachOUniversalBinaryMatchHost(MachOUniversalBinary *UB, 210281ad6265SDimitry Andric std::vector<NMSymbol> &SymbolList, 210381ad6265SDimitry Andric StringRef Filename, 210481ad6265SDimitry Andric LLVMContext *ContextPtr) { 21050b57cec5SDimitry Andric Triple HostTriple = MachOObjectFile::getHostArch(); 21060b57cec5SDimitry Andric StringRef HostArchName = HostTriple.getArchName(); 21070b57cec5SDimitry Andric for (MachOUniversalBinary::object_iterator I = UB->begin_objects(), 21080b57cec5SDimitry Andric E = UB->end_objects(); 21090b57cec5SDimitry Andric I != E; ++I) { 21100b57cec5SDimitry Andric if (HostArchName == I->getArchFlagName()) { 21110b57cec5SDimitry Andric Expected<std::unique_ptr<ObjectFile>> ObjOrErr = I->getAsObjectFile(); 21120b57cec5SDimitry Andric std::string ArchiveName; 21130b57cec5SDimitry Andric if (ObjOrErr) { 21140b57cec5SDimitry Andric ObjectFile &Obj = *ObjOrErr.get(); 211581ad6265SDimitry Andric dumpSymbolNamesFromObject(Obj, SymbolList, /*PrintSymbolObject=*/false, 211681ad6265SDimitry Andric /*PrintObjectLabel=*/false); 211781ad6265SDimitry Andric } else if (auto E = isNotObjectErrorInvalidFileType(ObjOrErr.takeError())) 21180b57cec5SDimitry Andric error(std::move(E), Filename); 211981ad6265SDimitry Andric else if (Expected<std::unique_ptr<Archive>> AOrErr = I->getAsArchive()) { 21200b57cec5SDimitry Andric std::unique_ptr<Archive> &A = *AOrErr; 21210b57cec5SDimitry Andric Error Err = Error::success(); 21220b57cec5SDimitry Andric for (auto &C : A->children(Err)) { 21230b57cec5SDimitry Andric Expected<std::unique_ptr<Binary>> ChildOrErr = 21240b57cec5SDimitry Andric C.getAsBinary(ContextPtr); 21250b57cec5SDimitry Andric if (!ChildOrErr) { 212681ad6265SDimitry Andric if (auto E = 212781ad6265SDimitry Andric isNotObjectErrorInvalidFileType(ChildOrErr.takeError())) 21280b57cec5SDimitry Andric error(std::move(E), Filename, C); 21290b57cec5SDimitry Andric continue; 21300b57cec5SDimitry Andric } 213181ad6265SDimitry Andric if (SymbolicFile *O = dyn_cast<SymbolicFile>(&*ChildOrErr.get())) { 21325ffd83dbSDimitry Andric ArchiveName = std::string(A->getFileName()); 213381ad6265SDimitry Andric dumpSymbolNamesFromObject(*O, SymbolList, 213481ad6265SDimitry Andric /*PrintSymbolObject=*/false, 213581ad6265SDimitry Andric !PrintFileName, ArchiveName); 21360b57cec5SDimitry Andric } 21370b57cec5SDimitry Andric } 21380b57cec5SDimitry Andric if (Err) 21390b57cec5SDimitry Andric error(std::move(Err), A->getFileName()); 21400b57cec5SDimitry Andric } else { 21410b57cec5SDimitry Andric consumeError(AOrErr.takeError()); 21420b57cec5SDimitry Andric error(Filename + " for architecture " + 21430b57cec5SDimitry Andric StringRef(I->getArchFlagName()) + 21440b57cec5SDimitry Andric " is not a Mach-O file or an archive file", 21450b57cec5SDimitry Andric "Mach-O universal file"); 21460b57cec5SDimitry Andric } 214781ad6265SDimitry Andric return true; 21480b57cec5SDimitry Andric } 21490b57cec5SDimitry Andric } 215081ad6265SDimitry Andric return false; 21510b57cec5SDimitry Andric } 215281ad6265SDimitry Andric 215381ad6265SDimitry Andric static void dumpMachOUniversalBinaryArchAll(MachOUniversalBinary *UB, 215481ad6265SDimitry Andric std::vector<NMSymbol> &SymbolList, 215581ad6265SDimitry Andric StringRef Filename, 215681ad6265SDimitry Andric LLVMContext *ContextPtr) { 21570b57cec5SDimitry Andric bool moreThanOneArch = UB->getNumberOfObjects() > 1; 21580b57cec5SDimitry Andric for (const MachOUniversalBinary::ObjectForArch &O : UB->objects()) { 21590b57cec5SDimitry Andric Expected<std::unique_ptr<ObjectFile>> ObjOrErr = O.getAsObjectFile(); 21600b57cec5SDimitry Andric std::string ArchiveName; 21610b57cec5SDimitry Andric std::string ArchitectureName; 21620b57cec5SDimitry Andric ArchiveName.clear(); 21630b57cec5SDimitry Andric ArchitectureName.clear(); 21640b57cec5SDimitry Andric if (ObjOrErr) { 21650b57cec5SDimitry Andric ObjectFile &Obj = *ObjOrErr.get(); 21660b57cec5SDimitry Andric if (isa<MachOObjectFile>(Obj) && moreThanOneArch) 21670b57cec5SDimitry Andric ArchitectureName = O.getArchFlagName(); 216881ad6265SDimitry Andric dumpSymbolNamesFromObject(Obj, SymbolList, /*PrintSymbolObject=*/false, 216981ad6265SDimitry Andric !PrintFileName, ArchiveName, ArchitectureName); 217081ad6265SDimitry Andric } else if (auto E = isNotObjectErrorInvalidFileType(ObjOrErr.takeError())) { 217181ad6265SDimitry Andric error(std::move(E), Filename, 217281ad6265SDimitry Andric moreThanOneArch ? StringRef(O.getArchFlagName()) : StringRef()); 21730b57cec5SDimitry Andric continue; 217481ad6265SDimitry Andric } else if (Expected<std::unique_ptr<Archive>> AOrErr = O.getAsArchive()) { 21750b57cec5SDimitry Andric std::unique_ptr<Archive> &A = *AOrErr; 21760b57cec5SDimitry Andric Error Err = Error::success(); 21770b57cec5SDimitry Andric for (auto &C : A->children(Err)) { 21780b57cec5SDimitry Andric Expected<std::unique_ptr<Binary>> ChildOrErr = 21790b57cec5SDimitry Andric C.getAsBinary(ContextPtr); 21800b57cec5SDimitry Andric if (!ChildOrErr) { 218181ad6265SDimitry Andric if (auto E = isNotObjectErrorInvalidFileType(ChildOrErr.takeError())) 218281ad6265SDimitry Andric error(std::move(E), Filename, C, 218381ad6265SDimitry Andric moreThanOneArch ? StringRef(ArchitectureName) : StringRef()); 21840b57cec5SDimitry Andric continue; 21850b57cec5SDimitry Andric } 21860b57cec5SDimitry Andric if (SymbolicFile *F = dyn_cast<SymbolicFile>(&*ChildOrErr.get())) { 21875ffd83dbSDimitry Andric ArchiveName = std::string(A->getFileName()); 21880b57cec5SDimitry Andric if (isa<MachOObjectFile>(F) && moreThanOneArch) 21890b57cec5SDimitry Andric ArchitectureName = O.getArchFlagName(); 219081ad6265SDimitry Andric dumpSymbolNamesFromObject(*F, SymbolList, /*PrintSymbolObject=*/false, 219181ad6265SDimitry Andric !PrintFileName, ArchiveName, 219281ad6265SDimitry Andric ArchitectureName); 21930b57cec5SDimitry Andric } 21940b57cec5SDimitry Andric } 21950b57cec5SDimitry Andric if (Err) 21960b57cec5SDimitry Andric error(std::move(Err), A->getFileName()); 21970b57cec5SDimitry Andric } else { 21980b57cec5SDimitry Andric consumeError(AOrErr.takeError()); 219981ad6265SDimitry Andric error(Filename + " for architecture " + StringRef(O.getArchFlagName()) + 22000b57cec5SDimitry Andric " is not a Mach-O file or an archive file", 22010b57cec5SDimitry Andric "Mach-O universal file"); 22020b57cec5SDimitry Andric } 22030b57cec5SDimitry Andric } 220481ad6265SDimitry Andric } 220581ad6265SDimitry Andric 220681ad6265SDimitry Andric static void dumpMachOUniversalBinary(MachOUniversalBinary *UB, 220781ad6265SDimitry Andric std::vector<NMSymbol> &SymbolList, 220881ad6265SDimitry Andric StringRef Filename, 220981ad6265SDimitry Andric LLVMContext *ContextPtr) { 221081ad6265SDimitry Andric // If we have a list of architecture flags specified dump only those. 221181ad6265SDimitry Andric if (!ArchAll && !ArchFlags.empty()) { 221281ad6265SDimitry Andric dumpMachOUniversalBinaryMatchArchFlags(UB, SymbolList, Filename, 221381ad6265SDimitry Andric ContextPtr); 22140b57cec5SDimitry Andric return; 22150b57cec5SDimitry Andric } 22165ffd83dbSDimitry Andric 221781ad6265SDimitry Andric // No architecture flags were specified so if this contains a slice that 221881ad6265SDimitry Andric // matches the host architecture dump only that. 221981ad6265SDimitry Andric if (!ArchAll && 222081ad6265SDimitry Andric dumpMachOUniversalBinaryMatchHost(UB, SymbolList, Filename, ContextPtr)) 222181ad6265SDimitry Andric return; 222281ad6265SDimitry Andric 222381ad6265SDimitry Andric // Either all architectures have been specified or none have been specified 222481ad6265SDimitry Andric // and this does not contain the host architecture so dump all the slices. 222581ad6265SDimitry Andric dumpMachOUniversalBinaryArchAll(UB, SymbolList, Filename, ContextPtr); 222681ad6265SDimitry Andric } 222781ad6265SDimitry Andric 222881ad6265SDimitry Andric static void dumpTapiUniversal(TapiUniversal *TU, 222981ad6265SDimitry Andric std::vector<NMSymbol> &SymbolList, 223081ad6265SDimitry Andric StringRef Filename) { 22315ffd83dbSDimitry Andric for (const TapiUniversal::ObjectForArch &I : TU->objects()) { 22325ffd83dbSDimitry Andric StringRef ArchName = I.getArchFlagName(); 22335ffd83dbSDimitry Andric const bool ShowArch = 2234e8d8bef9SDimitry Andric ArchFlags.empty() || llvm::is_contained(ArchFlags, ArchName); 22355ffd83dbSDimitry Andric if (!ShowArch) 22365ffd83dbSDimitry Andric continue; 22375ffd83dbSDimitry Andric if (!AddInlinedInfo && !I.isTopLevelLib()) 22385ffd83dbSDimitry Andric continue; 223981ad6265SDimitry Andric if (auto ObjOrErr = I.getAsObjectFile()) 224081ad6265SDimitry Andric dumpSymbolNamesFromObject( 224181ad6265SDimitry Andric *ObjOrErr.get(), SymbolList, /*PrintSymbolObject=*/false, 224281ad6265SDimitry Andric /*PrintObjectLabel=*/true, 224381ad6265SDimitry Andric /*ArchiveName=*/{}, ArchName, I.getInstallName()); 224481ad6265SDimitry Andric else if (Error E = isNotObjectErrorInvalidFileType(ObjOrErr.takeError())) { 22455ffd83dbSDimitry Andric error(std::move(E), Filename, ArchName); 22465ffd83dbSDimitry Andric } 22475ffd83dbSDimitry Andric } 22485ffd83dbSDimitry Andric } 22495ffd83dbSDimitry Andric 225081ad6265SDimitry Andric static void dumpSymbolicFile(SymbolicFile *O, std::vector<NMSymbol> &SymbolList, 225181ad6265SDimitry Andric StringRef Filename) { 22520b57cec5SDimitry Andric if (!MachOPrintSizeWarning && PrintSize && isa<MachOObjectFile>(O)) { 22530b57cec5SDimitry Andric WithColor::warning(errs(), ToolName) 22540b57cec5SDimitry Andric << "sizes with --print-size for Mach-O files are always zero.\n"; 22550b57cec5SDimitry Andric MachOPrintSizeWarning = true; 22560b57cec5SDimitry Andric } 22570b57cec5SDimitry Andric if (!checkMachOAndArchFlags(O, Filename)) 22580b57cec5SDimitry Andric return; 225981ad6265SDimitry Andric dumpSymbolNamesFromObject(*O, SymbolList, /*PrintSymbolObject=*/true, 226081ad6265SDimitry Andric /*PrintObjectLabel=*/false); 22610b57cec5SDimitry Andric } 226281ad6265SDimitry Andric 226381ad6265SDimitry Andric static std::vector<NMSymbol> dumpSymbolNamesFromFile(StringRef Filename) { 226481ad6265SDimitry Andric std::vector<NMSymbol> SymbolList; 226581ad6265SDimitry Andric ErrorOr<std::unique_ptr<MemoryBuffer>> BufferOrErr = 226681ad6265SDimitry Andric MemoryBuffer::getFileOrSTDIN(Filename); 226781ad6265SDimitry Andric if (error(BufferOrErr.getError(), Filename)) 226881ad6265SDimitry Andric return SymbolList; 226981ad6265SDimitry Andric 227081ad6265SDimitry Andric LLVMContext Context; 227181ad6265SDimitry Andric LLVMContext *ContextPtr = NoLLVMBitcode ? nullptr : &Context; 227281ad6265SDimitry Andric Expected<std::unique_ptr<Binary>> BinaryOrErr = 227381ad6265SDimitry Andric createBinary(BufferOrErr.get()->getMemBufferRef(), ContextPtr); 227481ad6265SDimitry Andric if (!BinaryOrErr) { 227581ad6265SDimitry Andric error(BinaryOrErr.takeError(), Filename); 227681ad6265SDimitry Andric return SymbolList; 227781ad6265SDimitry Andric } 227881ad6265SDimitry Andric Binary &Bin = *BinaryOrErr.get(); 227981ad6265SDimitry Andric if (Archive *A = dyn_cast<Archive>(&Bin)) 228081ad6265SDimitry Andric dumpArchive(A, SymbolList, Filename, ContextPtr); 228181ad6265SDimitry Andric else if (MachOUniversalBinary *UB = dyn_cast<MachOUniversalBinary>(&Bin)) 228281ad6265SDimitry Andric dumpMachOUniversalBinary(UB, SymbolList, Filename, ContextPtr); 228381ad6265SDimitry Andric else if (TapiUniversal *TU = dyn_cast<TapiUniversal>(&Bin)) 228481ad6265SDimitry Andric dumpTapiUniversal(TU, SymbolList, Filename); 228581ad6265SDimitry Andric else if (SymbolicFile *O = dyn_cast<SymbolicFile>(&Bin)) 228681ad6265SDimitry Andric dumpSymbolicFile(O, SymbolList, Filename); 228781ad6265SDimitry Andric return SymbolList; 228881ad6265SDimitry Andric } 228981ad6265SDimitry Andric 229081ad6265SDimitry Andric static void 229181ad6265SDimitry Andric exportSymbolNamesFromFiles(const std::vector<std::string> &InputFilenames) { 229281ad6265SDimitry Andric std::vector<NMSymbol> SymbolList; 229381ad6265SDimitry Andric for (const auto &FileName : InputFilenames) { 229481ad6265SDimitry Andric std::vector<NMSymbol> FileSymList = dumpSymbolNamesFromFile(FileName); 229581ad6265SDimitry Andric SymbolList.insert(SymbolList.end(), FileSymList.begin(), FileSymList.end()); 229681ad6265SDimitry Andric } 229781ad6265SDimitry Andric 229881ad6265SDimitry Andric // Delete symbols which should not be printed from SymolList. 2299bdd1243dSDimitry Andric llvm::erase_if(SymbolList, 2300bdd1243dSDimitry Andric [](const NMSymbol &s) { return !s.shouldPrint(); }); 230181ad6265SDimitry Andric sortSymbolList(SymbolList); 230281ad6265SDimitry Andric SymbolList.erase(std::unique(SymbolList.begin(), SymbolList.end()), 230381ad6265SDimitry Andric SymbolList.end()); 230481ad6265SDimitry Andric printExportSymbolList(SymbolList); 23050b57cec5SDimitry Andric } 23060b57cec5SDimitry Andric 2307*06c3fb27SDimitry Andric int llvm_nm_main(int argc, char **argv, const llvm::ToolContext &) { 23080b57cec5SDimitry Andric InitLLVM X(argc, argv); 2309fe6060f1SDimitry Andric BumpPtrAllocator A; 2310fe6060f1SDimitry Andric StringSaver Saver(A); 2311fe6060f1SDimitry Andric NmOptTable Tbl; 2312fe6060f1SDimitry Andric ToolName = argv[0]; 2313fe6060f1SDimitry Andric opt::InputArgList Args = 2314fe6060f1SDimitry Andric Tbl.parseArgs(argc, argv, OPT_UNKNOWN, Saver, [&](StringRef Msg) { 2315fe6060f1SDimitry Andric error(Msg); 2316fe6060f1SDimitry Andric exit(1); 2317fe6060f1SDimitry Andric }); 2318fe6060f1SDimitry Andric if (Args.hasArg(OPT_help)) { 2319fe6060f1SDimitry Andric Tbl.printHelp( 2320fe6060f1SDimitry Andric outs(), 2321fe6060f1SDimitry Andric (Twine(ToolName) + " [options] <input object files>").str().c_str(), 2322fe6060f1SDimitry Andric "LLVM symbol table dumper"); 2323fe6060f1SDimitry Andric // TODO Replace this with OptTable API once it adds extrahelp support. 2324fe6060f1SDimitry Andric outs() << "\nPass @FILE as argument to read options from FILE.\n"; 2325fe6060f1SDimitry Andric return 0; 2326fe6060f1SDimitry Andric } 2327fe6060f1SDimitry Andric if (Args.hasArg(OPT_version)) { 2328fe6060f1SDimitry Andric // This needs to contain the word "GNU", libtool looks for that string. 2329fe6060f1SDimitry Andric outs() << "llvm-nm, compatible with GNU nm" << '\n'; 2330fe6060f1SDimitry Andric cl::PrintVersionMessage(); 2331fe6060f1SDimitry Andric return 0; 2332fe6060f1SDimitry Andric } 2333fe6060f1SDimitry Andric 2334fe6060f1SDimitry Andric DebugSyms = Args.hasArg(OPT_debug_syms); 2335fe6060f1SDimitry Andric DefinedOnly = Args.hasArg(OPT_defined_only); 2336fe6060f1SDimitry Andric Demangle = Args.hasFlag(OPT_demangle, OPT_no_demangle, false); 2337fe6060f1SDimitry Andric DynamicSyms = Args.hasArg(OPT_dynamic); 2338fe6060f1SDimitry Andric ExternalOnly = Args.hasArg(OPT_extern_only); 2339fe6060f1SDimitry Andric StringRef V = Args.getLastArgValue(OPT_format_EQ, "bsd"); 2340fe6060f1SDimitry Andric if (V == "bsd") 2341fe6060f1SDimitry Andric OutputFormat = bsd; 2342fe6060f1SDimitry Andric else if (V == "posix") 2343fe6060f1SDimitry Andric OutputFormat = posix; 2344fe6060f1SDimitry Andric else if (V == "sysv") 2345fe6060f1SDimitry Andric OutputFormat = sysv; 2346fe6060f1SDimitry Andric else if (V == "darwin") 2347fe6060f1SDimitry Andric OutputFormat = darwin; 2348fe6060f1SDimitry Andric else if (V == "just-symbols") 2349fe6060f1SDimitry Andric OutputFormat = just_symbols; 2350fe6060f1SDimitry Andric else 2351fe6060f1SDimitry Andric error("--format value should be one of: bsd, posix, sysv, darwin, " 2352fe6060f1SDimitry Andric "just-symbols"); 2353fe6060f1SDimitry Andric NoLLVMBitcode = Args.hasArg(OPT_no_llvm_bc); 2354fe6060f1SDimitry Andric NoSort = Args.hasArg(OPT_no_sort); 2355fe6060f1SDimitry Andric NoWeakSymbols = Args.hasArg(OPT_no_weak); 2356fe6060f1SDimitry Andric NumericSort = Args.hasArg(OPT_numeric_sort); 2357fe6060f1SDimitry Andric ArchiveMap = Args.hasArg(OPT_print_armap); 2358fe6060f1SDimitry Andric PrintFileName = Args.hasArg(OPT_print_file_name); 2359fe6060f1SDimitry Andric PrintSize = Args.hasArg(OPT_print_size); 2360fe6060f1SDimitry Andric ReverseSort = Args.hasArg(OPT_reverse_sort); 236181ad6265SDimitry Andric ExportSymbols = Args.hasArg(OPT_export_symbols); 236281ad6265SDimitry Andric if (ExportSymbols) { 236381ad6265SDimitry Andric ExternalOnly = true; 236481ad6265SDimitry Andric DefinedOnly = true; 236581ad6265SDimitry Andric } 236681ad6265SDimitry Andric 2367fe6060f1SDimitry Andric Quiet = Args.hasArg(OPT_quiet); 2368fe6060f1SDimitry Andric V = Args.getLastArgValue(OPT_radix_EQ, "x"); 2369fe6060f1SDimitry Andric if (V == "o") 2370fe6060f1SDimitry Andric AddressRadix = Radix::o; 2371fe6060f1SDimitry Andric else if (V == "d") 2372fe6060f1SDimitry Andric AddressRadix = Radix::d; 2373fe6060f1SDimitry Andric else if (V == "x") 2374fe6060f1SDimitry Andric AddressRadix = Radix::x; 2375fe6060f1SDimitry Andric else 2376fe6060f1SDimitry Andric error("--radix value should be one of: 'o' (octal), 'd' (decimal), 'x' " 2377fe6060f1SDimitry Andric "(hexadecimal)"); 2378fe6060f1SDimitry Andric SizeSort = Args.hasArg(OPT_size_sort); 2379fe6060f1SDimitry Andric SpecialSyms = Args.hasArg(OPT_special_syms); 2380fe6060f1SDimitry Andric UndefinedOnly = Args.hasArg(OPT_undefined_only); 2381fe6060f1SDimitry Andric WithoutAliases = Args.hasArg(OPT_without_aliases); 2382fe6060f1SDimitry Andric 2383bdd1243dSDimitry Andric // Get BitMode from enviornment variable "OBJECT_MODE" for AIX OS, if 2384bdd1243dSDimitry Andric // specified. 2385bdd1243dSDimitry Andric Triple HostTriple(sys::getProcessTriple()); 2386bdd1243dSDimitry Andric if (HostTriple.isOSAIX()) { 2387bdd1243dSDimitry Andric BitMode = StringSwitch<BitModeTy>(getenv("OBJECT_MODE")) 2388bdd1243dSDimitry Andric .Case("32", BitModeTy::Bit32) 2389bdd1243dSDimitry Andric .Case("64", BitModeTy::Bit64) 2390bdd1243dSDimitry Andric .Case("32_64", BitModeTy::Bit32_64) 2391bdd1243dSDimitry Andric .Case("any", BitModeTy::Any) 2392bdd1243dSDimitry Andric .Default(BitModeTy::Bit32); 2393bdd1243dSDimitry Andric } else 2394bdd1243dSDimitry Andric BitMode = BitModeTy::Any; 2395bdd1243dSDimitry Andric 2396bdd1243dSDimitry Andric if (Arg *A = Args.getLastArg(OPT_X)) { 2397bdd1243dSDimitry Andric StringRef Mode = A->getValue(); 239881ad6265SDimitry Andric if (Mode == "32") 239981ad6265SDimitry Andric BitMode = BitModeTy::Bit32; 240081ad6265SDimitry Andric else if (Mode == "64") 240181ad6265SDimitry Andric BitMode = BitModeTy::Bit64; 240281ad6265SDimitry Andric else if (Mode == "32_64") 240381ad6265SDimitry Andric BitMode = BitModeTy::Bit32_64; 240481ad6265SDimitry Andric else if (Mode == "any") 240581ad6265SDimitry Andric BitMode = BitModeTy::Any; 240681ad6265SDimitry Andric else 240781ad6265SDimitry Andric error("-X value should be one of: 32, 64, 32_64, (default) any"); 2408bdd1243dSDimitry Andric } 240981ad6265SDimitry Andric 2410fe6060f1SDimitry Andric // Mach-O specific options. 2411fe6060f1SDimitry Andric FormatMachOasHex = Args.hasArg(OPT_x); 2412fe6060f1SDimitry Andric AddDyldInfo = Args.hasArg(OPT_add_dyldinfo); 2413fe6060f1SDimitry Andric AddInlinedInfo = Args.hasArg(OPT_add_inlinedinfo); 2414fe6060f1SDimitry Andric DyldInfoOnly = Args.hasArg(OPT_dyldinfo_only); 2415fe6060f1SDimitry Andric NoDyldInfo = Args.hasArg(OPT_no_dyldinfo); 24160b57cec5SDimitry Andric 241781ad6265SDimitry Andric // XCOFF specific options. 241881ad6265SDimitry Andric NoRsrc = Args.hasArg(OPT_no_rsrc); 241981ad6265SDimitry Andric 24200b57cec5SDimitry Andric // llvm-nm only reads binary files. 24210b57cec5SDimitry Andric if (error(sys::ChangeStdinToBinary())) 24220b57cec5SDimitry Andric return 1; 24230b57cec5SDimitry Andric 24240b57cec5SDimitry Andric // These calls are needed so that we can read bitcode correctly. 24250b57cec5SDimitry Andric llvm::InitializeAllTargetInfos(); 24260b57cec5SDimitry Andric llvm::InitializeAllTargetMCs(); 24270b57cec5SDimitry Andric llvm::InitializeAllAsmParsers(); 24280b57cec5SDimitry Andric 24290b57cec5SDimitry Andric // The relative order of these is important. If you pass --size-sort it should 24300b57cec5SDimitry Andric // only print out the size. However, if you pass -S --size-sort, it should 24310b57cec5SDimitry Andric // print out both the size and address. 24320b57cec5SDimitry Andric if (SizeSort && !PrintSize) 24330b57cec5SDimitry Andric PrintAddress = false; 24340b57cec5SDimitry Andric if (OutputFormat == sysv || SizeSort) 24350b57cec5SDimitry Andric PrintSize = true; 24360b57cec5SDimitry Andric 2437fe6060f1SDimitry Andric for (const auto *A : Args.filtered(OPT_arch_EQ)) { 2438fe6060f1SDimitry Andric SmallVector<StringRef, 2> Values; 2439fe6060f1SDimitry Andric llvm::SplitString(A->getValue(), Values, ","); 2440fe6060f1SDimitry Andric for (StringRef V : Values) { 2441fe6060f1SDimitry Andric if (V == "all") 24420b57cec5SDimitry Andric ArchAll = true; 2443fe6060f1SDimitry Andric else if (MachOObjectFile::isValidArch(V)) 2444fe6060f1SDimitry Andric ArchFlags.push_back(V); 2445fe6060f1SDimitry Andric else 2446fe6060f1SDimitry Andric error("Unknown architecture named '" + V + "'", 24470b57cec5SDimitry Andric "for the --arch option"); 24480b57cec5SDimitry Andric } 24490b57cec5SDimitry Andric } 24500b57cec5SDimitry Andric 2451fe6060f1SDimitry Andric // Mach-O takes -s to accept two arguments. We emulate this by iterating over 2452fe6060f1SDimitry Andric // both OPT_s and OPT_INPUT. 2453fe6060f1SDimitry Andric std::vector<std::string> InputFilenames; 2454fe6060f1SDimitry Andric int SegSectArgs = 0; 2455fe6060f1SDimitry Andric for (opt::Arg *A : Args.filtered(OPT_s, OPT_INPUT)) { 2456fe6060f1SDimitry Andric if (SegSectArgs > 0) { 2457fe6060f1SDimitry Andric --SegSectArgs; 2458fe6060f1SDimitry Andric SegSect.push_back(A->getValue()); 2459fe6060f1SDimitry Andric } else if (A->getOption().matches(OPT_s)) { 2460fe6060f1SDimitry Andric SegSectArgs = 2; 2461fe6060f1SDimitry Andric } else { 2462fe6060f1SDimitry Andric InputFilenames.push_back(A->getValue()); 2463fe6060f1SDimitry Andric } 2464fe6060f1SDimitry Andric } 24650b57cec5SDimitry Andric if (!SegSect.empty() && SegSect.size() != 2) 24660b57cec5SDimitry Andric error("bad number of arguments (must be two arguments)", 24670b57cec5SDimitry Andric "for the -s option"); 24680b57cec5SDimitry Andric 2469fe6060f1SDimitry Andric if (InputFilenames.empty()) 2470fe6060f1SDimitry Andric InputFilenames.push_back("a.out"); 2471fe6060f1SDimitry Andric if (InputFilenames.size() > 1) 2472fe6060f1SDimitry Andric MultipleFiles = true; 2473fe6060f1SDimitry Andric 24740b57cec5SDimitry Andric if (NoDyldInfo && (AddDyldInfo || DyldInfoOnly)) 24750b57cec5SDimitry Andric error("--no-dyldinfo can't be used with --add-dyldinfo or --dyldinfo-only"); 24760b57cec5SDimitry Andric 247781ad6265SDimitry Andric if (ExportSymbols) 247881ad6265SDimitry Andric exportSymbolNamesFromFiles(InputFilenames); 247981ad6265SDimitry Andric else 24800b57cec5SDimitry Andric llvm::for_each(InputFilenames, dumpSymbolNamesFromFile); 24810b57cec5SDimitry Andric 24820b57cec5SDimitry Andric if (HadError) 24830b57cec5SDimitry Andric return 1; 2484bdd1243dSDimitry Andric return 0; 24850b57cec5SDimitry Andric } 2486