diff --git a/llvm/test/tools/llvm-readobj/MachO/stabs-sorted.yaml b/llvm/test/tools/llvm-readobj/MachO/stabs-sorted.yaml new file mode 100644 --- /dev/null +++ b/llvm/test/tools/llvm-readobj/MachO/stabs-sorted.yaml @@ -0,0 +1,127 @@ +## Verify that llvm-readobj can dump files with stab symbols in a sorted order, + +# RUN: yaml2obj %s -o %t +# RUN: llvm-readobj --syms --sort-symbols %t | FileCheck %s --check-prefix=SORTED + +# SORTED: Symbols [ +# SORTED-NEXT: Symbol { +# SORTED-NEXT: Name: _foo (44) +# SORTED-NEXT: Type: SymDebugTable (0x24) +# SORTED-NEXT: Section: (0x1) +# SORTED-NEXT: RefType: UndefinedNonLazy (0x0) +# SORTED-NEXT: Flags [ (0x0) +# SORTED-NEXT: ] +# SORTED-NEXT: Value: 0x100000FA0 +# SORTED-NEXT: } +# SORTED-NEXT: Symbol { +# SORTED-NEXT: Name: (1) +# SORTED-NEXT: Type: SymDebugTable (0x2E) +# SORTED-NEXT: Section: (0x1) +# SORTED-NEXT: RefType: UndefinedNonLazy (0x0) +# SORTED-NEXT: Flags [ (0x0) +# SORTED-NEXT: ] +# SORTED-NEXT: Value: 0x100000FA0 +# SORTED-NEXT: } +# SORTED-NEXT: Symbol { +# SORTED-NEXT: Name: (1) +# SORTED-NEXT: Type: SymDebugTable (0x4E) +# SORTED-NEXT: Section: (0x1) +# SORTED-NEXT: RefType: UndefinedNonLazy (0x0) +# SORTED-NEXT: Flags [ (0x0) +# SORTED-NEXT: ] +# SORTED-NEXT: Value: 0xF +# SORTED-NEXT: } +# SORTED-NEXT: Symbol { +# SORTED-NEXT: Name: /Volumes/Sandbox/ (2) +# SORTED-NEXT: Type: SymDebugTable (0x64) +# SORTED-NEXT: Section: (0x0) +# SORTED-NEXT: RefType: UndefinedNonLazy (0x0) +# SORTED-NEXT: Flags [ (0x0) +# SORTED-NEXT: ] +# SORTED-NEXT: Value: 0x0 +# SORTED-NEXT: } +# SORTED-NEXT: Symbol { +# SORTED-NEXT: Name: /Volumes/Sandbox/test.o (20) +# SORTED-NEXT: Type: SymDebugTable (0x66) +# SORTED-NEXT: Section: (0x3) +# SORTED-NEXT: RefType: ReferenceFlagUndefinedLazy (0x1) +# SORTED-NEXT: Flags [ (0x0) +# SORTED-NEXT: ] +# SORTED-NEXT: Value: 0x5F72D5E2 +# SORTED-NEXT: } +# SORTED-NEXT: ] + +--- !mach-o +FileHeader: + magic: 0xFEEDFACF + cputype: 0x01000007 + cpusubtype: 0x00000003 + filetype: 0x00000002 + ncmds: 2 + sizeofcmds: 744 + flags: 0x00200085 + reserved: 0x00000000 +LoadCommands: + - cmd: LC_SEGMENT_64 + cmdsize: 232 + segname: __TEXT + vmaddr: 4294967296 + vmsize: 4096 + fileoff: 0 + filesize: 4096 + maxprot: 5 + initprot: 5 + nsects: 1 + flags: 0 + Sections: + - sectname: __text + segname: __TEXT + addr: 0x00000000 + size: 0 + offset: 0x00000000 + align: 4 + reloff: 0x00000000 + nreloc: 0 + flags: 0x80000400 + reserved1: 0x00000000 + reserved2: 0x00000000 + reserved3: 0x00000000 + - cmd: LC_SYMTAB + cmdsize: 24 + symoff: 4152 + nsyms: 5 + stroff: 4328 + strsize: 49 +LinkEditData: + NameList: + - n_strx: 2 + n_type: 0x64 ## N_SO + n_sect: 0 + n_desc: 0 + n_value: 0 + - n_strx: 20 + n_type: 0x66 ## N_OSO + n_sect: 3 + n_desc: 1 + n_value: 1601361378 + - n_strx: 1 + n_type: 0x2E ## N_BNSYM + n_sect: 1 + n_desc: 0 + n_value: 4294971296 + - n_strx: 44 + n_type: 0x24 ## N_FUN + n_sect: 1 + n_desc: 0 + n_value: 4294971296 + - n_strx: 1 + n_type: 0x4E ## N_ENSYM + n_sect: 1 + n_desc: 0 + n_value: 15 + StringTable: + - ' ' + - '/Volumes/Sandbox/' + - '/Volumes/Sandbox/test.o' + - _foo +... diff --git a/llvm/tools/llvm-readobj/MachODumper.cpp b/llvm/tools/llvm-readobj/MachODumper.cpp --- a/llvm/tools/llvm-readobj/MachODumper.cpp +++ b/llvm/tools/llvm-readobj/MachODumper.cpp @@ -17,7 +17,12 @@ #include "llvm/ADT/StringExtras.h" #include "llvm/Object/MachO.h" #include "llvm/Support/Casting.h" +#include "llvm/Support/FormattedStream.h" #include "llvm/Support/ScopedPrinter.h" +#include "llvm/Support/raw_ostream.h" + +#include +#include using namespace llvm; using namespace object; @@ -49,8 +54,9 @@ template void printFileHeaders(const MachHeader &Header); - void printSymbols() override; - void printDynamicSymbols() override; + void printSymbols(bool SortSymbols) override; + void printDynamicSymbols(bool SortSymbols) override; + void printSymbol(const SymbolRef &Symbol, ScopedPrinter &W); void printSymbol(const SymbolRef &Symbol); void printRelocation(const RelocationRef &Reloc); @@ -601,19 +607,64 @@ } } -void MachODumper::printSymbols() { +void MachODumper::printSymbols(bool SortSymbols) { ListScope Group(W, "Symbols"); - for (const SymbolRef &Symbol : Obj->symbols()) { - printSymbol(Symbol); + if (SortSymbols) { + // The references returned by calling Obj->symbols() are temporary + // and we can't hold onto them after the loop. In order to sort the symbols, + // we have to print their representation to string and sort them later. + + assert(opts::Output != opts::JSON && "not expecting JSON format yet."); + // Tuple of + // Sort by type (local/external/undefined), then name. + std::vector> SortedSymbols; + for (SymbolRef Symbol : Obj->symbols()) { + std::string SymRep; + raw_string_ostream StringOs(SymRep); + formatted_raw_ostream FormattedOs(StringOs); + ScopedPrinter Printer(FormattedOs); + Printer.indent(W.getIndentLevel()); + printSymbol(Symbol, Printer); + + const DataRefImpl &DRI = Symbol.getRawDataRefImpl(); + SortedSymbols.emplace_back(Obj->getSymbol64TableEntry(DRI).n_type, + Symbol.getName()->str(), std::move(SymRep)); + } + + llvm::stable_sort( + SortedSymbols, + [](const std::tuple &LHS, + const std::tuple &RHS) { + uint8_t LType = std::get<0>(LHS); + uint8_t RType = std::get<0>(RHS); + + if (LType == RType) + return std::get<1>(LHS).compare(std::get<1>(RHS)) < 0; + return LType < RType; + // TODO:: Maybe use Symbol's addresses as tie-breaker if + // names are the same. + }); + + // Unindent one level because the Symbols were already padded. + W.unindent(); + for (auto &Tup : SortedSymbols) + W.startLine() << std::get<2>(Tup); + } else { + for (SymbolRef Symbol : Obj->symbols()) + printSymbol(Symbol); } } -void MachODumper::printDynamicSymbols() { +void MachODumper::printDynamicSymbols(bool SortSymbols) { ListScope Group(W, "DynamicSymbols"); } void MachODumper::printSymbol(const SymbolRef &Symbol) { + printSymbol(Symbol, W); +} + +void MachODumper::printSymbol(const SymbolRef &Symbol, ScopedPrinter &W) { StringRef SymbolName; Expected SymbolNameOrErr = Symbol.getName(); if (!SymbolNameOrErr) { diff --git a/llvm/tools/llvm-readobj/ObjDumper.h b/llvm/tools/llvm-readobj/ObjDumper.h --- a/llvm/tools/llvm-readobj/ObjDumper.h +++ b/llvm/tools/llvm-readobj/ObjDumper.h @@ -47,10 +47,14 @@ virtual void printSectionHeaders() = 0; virtual void printRelocations() = 0; virtual void printSymbols(bool PrintSymbols, bool PrintDynamicSymbols) { + printSymbols(PrintSymbols, PrintDynamicSymbols, false); + } + virtual void printSymbols(bool PrintSymbols, bool PrintDynamicSymbols, + bool SortSymbols) { if (PrintSymbols) - printSymbols(); + printSymbols(SortSymbols); if (PrintDynamicSymbols) - printDynamicSymbols(); + printDynamicSymbols(SortSymbols); } virtual void printProgramHeaders(bool PrintProgramHeaders, cl::boolOrDefault PrintSectionMapping) { @@ -132,8 +136,10 @@ ScopedPrinter &W; private: - virtual void printSymbols() {} - virtual void printDynamicSymbols() {} + virtual void printSymbols() { printSymbols(false); } + virtual void printSymbols(bool SortSymbols) {} + virtual void printDynamicSymbols() { printDynamicSymbols(false); } + virtual void printDynamicSymbols(bool SortSymbols) {} virtual void printProgramHeaders() {} virtual void printSectionMapping() {} diff --git a/llvm/tools/llvm-readobj/Opts.td b/llvm/tools/llvm-readobj/Opts.td --- a/llvm/tools/llvm-readobj/Opts.td +++ b/llvm/tools/llvm-readobj/Opts.td @@ -43,6 +43,7 @@ def string_table : FF<"string-table", "Display the string table (only for XCOFF now)">; def symbols : FF<"symbols", "Display the symbol table. Also display the dynamic symbol table when using GNU output style for ELF">; def unwind : FF<"unwind", "Display unwind information">; +def sort_symbols : FF<"sort-symbols", "Sort symbol before outputing symtab">; // ELF specific options. def grp_elf : OptionGroup<"kind">, HelpText<"OPTIONS (ELF specific)">; diff --git a/llvm/tools/llvm-readobj/llvm-readobj.cpp b/llvm/tools/llvm-readobj/llvm-readobj.cpp --- a/llvm/tools/llvm-readobj/llvm-readobj.cpp +++ b/llvm/tools/llvm-readobj/llvm-readobj.cpp @@ -113,6 +113,7 @@ static bool Symbols; static bool UnwindInfo; static cl::boolOrDefault SectionMapping; +static bool SortSymbols; // ELF specific options. static bool DynamicTable; @@ -253,6 +254,7 @@ opts::ProgramHeaders = Args.hasArg(OPT_program_headers); opts::RawRelr = Args.hasArg(OPT_raw_relr); opts::SectionGroups = Args.hasArg(OPT_section_groups); + opts::SortSymbols = Args.hasArg(OPT_sort_symbols); opts::VersionInfo = Args.hasArg(OPT_version_info); // Mach-O specific options. @@ -374,7 +376,8 @@ if (opts::UnwindInfo) Dumper->printUnwindInfo(); if (opts::Symbols || opts::DynamicSymbols) - Dumper->printSymbols(opts::Symbols, opts::DynamicSymbols); + Dumper->printSymbols(opts::Symbols, opts::DynamicSymbols, + opts::SortSymbols); if (!opts::StringDump.empty()) Dumper->printSectionsAsString(Obj, opts::StringDump); if (!opts::HexDump.empty())