Index: include/llvm/DebugInfo/Symbolize/DIPrinter.h =================================================================== --- include/llvm/DebugInfo/Symbolize/DIPrinter.h +++ include/llvm/DebugInfo/Symbolize/DIPrinter.h @@ -28,13 +28,16 @@ raw_ostream &OS; bool PrintFunctionNames; bool PrintPretty; - void printName(const DILineInfo &Info, bool Inlined); + int PrintSourceContext; + + void print(const DILineInfo &Info, bool Inlined); + void printContext(std::string FileName, uint32_t Line); public: DIPrinter(raw_ostream &OS, bool PrintFunctionNames = true, - bool PrintPretty = false) + bool PrintPretty = false, int PrintSourceContext = 0) : OS(OS), PrintFunctionNames(PrintFunctionNames), - PrintPretty(PrintPretty) {} + PrintPretty(PrintPretty), PrintSourceContext(PrintSourceContext) {} DIPrinter &operator<<(const DILineInfo &Info); DIPrinter &operator<<(const DIInliningInfo &Info); Index: lib/DebugInfo/Symbolize/DIPrinter.cpp =================================================================== --- lib/DebugInfo/Symbolize/DIPrinter.cpp +++ lib/DebugInfo/Symbolize/DIPrinter.cpp @@ -24,7 +24,40 @@ static const char kDILineInfoBadString[] = ""; static const char kBadString[] = "??"; -void DIPrinter::printName(const DILineInfo &Info, bool Inlined) { +// Prints source code around in the FileName the Line. +void DIPrinter::printContext(std::string FileName, uint32_t Line) { + if (PrintSourceContext <= 0) + return; + + ErrorOr> BufOrErr = + MemoryBuffer::getFile(FileName); + if (!BufOrErr) + return; + + std::unique_ptr Buf = std::move(BufOrErr.get()); + StringRef FileContent = Buf->getBuffer(); + + size_t FirstLine = std::max(1u, Line - PrintSourceContext / 2); + size_t LastLine = FirstLine + PrintSourceContext; + size_t MaxLineNumberWidth = std::ceil(std::log10(LastLine)); + + size_t LineStart = 0; + for (uint32_t I = 0; LineStart != StringRef::npos && I <= LastLine; ++I) { + size_t LineEnd = FileContent.find_first_of("\n", LineStart); + if (I + 1 >= FirstLine && I < LastLine) { + OS << format_decimal(I + 1, MaxLineNumberWidth); + if (I + 1 == Line) { + OS << " >: "; + } else { + OS << " : "; + } + OS << FileContent.substr(LineStart, LineEnd - LineStart) << "\n"; + } + LineStart = FileContent.find_first_not_of("\r", LineEnd + 1); + } +} + +void DIPrinter::print(const DILineInfo &Info, bool Inlined) { if (PrintFunctionNames) { std::string FunctionName = Info.FunctionName; if (FunctionName == kDILineInfoBadString) @@ -38,21 +71,22 @@ if (Filename == kDILineInfoBadString) Filename = kBadString; OS << Filename << ":" << Info.Line << ":" << Info.Column << "\n"; + printContext(Filename, Info.Line); } DIPrinter &DIPrinter::operator<<(const DILineInfo &Info) { - printName(Info, false); + print(Info, false); return *this; } DIPrinter &DIPrinter::operator<<(const DIInliningInfo &Info) { uint32_t FramesNum = Info.getNumberOfFrames(); if (FramesNum == 0) { - printName(DILineInfo(), false); + print(DILineInfo(), false); return *this; } for (uint32_t i = 0; i < FramesNum; i++) - printName(Info.getFrame(i), i > 0); + print(Info.getFrame(i), i > 0); return *this; } Index: test/lit.cfg =================================================================== --- test/lit.cfg +++ test/lit.cfg @@ -194,6 +194,7 @@ config.substitutions.append( ('%shlibext', config.llvm_shlib_ext) ) config.substitutions.append( ('%exeext', config.llvm_exe_ext) ) config.substitutions.append( ('%python', config.python_executable) ) +config.substitutions.append( ('%host_cc', config.host_cc) ) # OCaml substitutions. # Support tests for both native and bytecode builds. @@ -276,6 +277,7 @@ r"\bllvm-split\b", r"\bllvm-tblgen\b", r"\bllvm-c-test\b", + NOJUNK + r"\bllvm-symbolizer\b", NOJUNK + r"\bopt\b", r"\bFileCheck\b", r"\bobj2yaml\b", Index: test/tools/llvm-symbolizer/print_context.c =================================================================== --- /dev/null +++ test/tools/llvm-symbolizer/print_context.c @@ -0,0 +1,22 @@ +// REQUIRES: x86_64-linux +// RUN: %host_cc -O0 -g %s -o %t 2>&1 +// RUN: %t 2>&1 | llvm-symbolizer -print-source-context-lines=5 -obj=%t | FileCheck %s --check-prefix=CHECK + +#include + +int inc(int a) { + return a + 1; +} + +int main() { + printf("%p\n", inc); + return 0; +} + +// CHECK: inc +// CHECK: print_context.c:7 +// CHECK: 5 : #include +// CHECK: 6 : +// CHECK: 7 >: int inc +// CHECK: 8 : return +// CHECK: 9 : } Index: tools/llvm-symbolizer/llvm-symbolizer.cpp =================================================================== --- tools/llvm-symbolizer/llvm-symbolizer.cpp +++ tools/llvm-symbolizer/llvm-symbolizer.cpp @@ -82,6 +82,10 @@ ClPrettyPrint("pretty-print", cl::init(false), cl::desc("Make the output more human friendly")); +static cl::opt ClPrintSourceContextLines( + "print-source-context-lines", cl::init(0), + cl::desc("Print N number of source file context")); + static bool error(std::error_code ec) { if (!ec) return false; @@ -155,7 +159,7 @@ LLVMSymbolizer Symbolizer(Opts); DIPrinter Printer(outs(), ClPrintFunctions != FunctionNameKind::None, - ClPrettyPrint); + ClPrettyPrint, ClPrintSourceContextLines); const int kMaxInputStringLength = 1024; char InputString[kMaxInputStringLength];