Index: llvm/trunk/test/tools/llvm-cxxfilt/simple.test =================================================================== --- llvm/trunk/test/tools/llvm-cxxfilt/simple.test +++ llvm/trunk/test/tools/llvm-cxxfilt/simple.test @@ -1,4 +1,10 @@ RUN: llvm-cxxfilt _Z1fi abc | FileCheck %s +RUN: echo "Mangled _Z1fi and _Z3foov in string." | llvm-cxxfilt \ +RUN: | FileCheck %s --check-prefix=CHECK-STRING +RUN: llvm-cxxfilt "CLI remains mangled _Z1fi" \ +RUN: | FileCheck %s --check-prefix=CHECK-MANGLED CHECK: f(int) CHECK-NEXT: abc +CHECK-STRING: Mangled f(int) and foo() in string. +CHECK-MANGLED: CLI remains mangled _Z1fi Index: llvm/trunk/test/tools/llvm-cxxfilt/types.test =================================================================== --- llvm/trunk/test/tools/llvm-cxxfilt/types.test +++ llvm/trunk/test/tools/llvm-cxxfilt/types.test @@ -1,5 +1,6 @@ RUN: llvm-cxxfilt -t f i | FileCheck %s +RUN: echo "f i" | llvm-cxxfilt -t | FileCheck %s --check-prefix="CHECK-STRING" CHECK: float CHECK-NEXT: int - +CHECK-STRING: float int Index: llvm/trunk/tools/llvm-cxxfilt/llvm-cxxfilt.cpp =================================================================== --- llvm/trunk/tools/llvm-cxxfilt/llvm-cxxfilt.cpp +++ llvm/trunk/tools/llvm-cxxfilt/llvm-cxxfilt.cpp @@ -6,6 +6,7 @@ // //===----------------------------------------------------------------------===// +#include "llvm/ADT/StringExtras.h" #include "llvm/Demangle/Demangle.h" #include "llvm/Support/CommandLine.h" #include "llvm/Support/InitLLVM.h" @@ -51,7 +52,7 @@ static cl::list Decorated(cl::Positional, cl::desc(""), cl::ZeroOrMore); -static void demangle(llvm::raw_ostream &OS, const std::string &Mangled) { +static std::string demangle(llvm::raw_ostream &OS, const std::string &Mangled) { int Status; const char *Decorated = Mangled.c_str(); @@ -72,10 +73,28 @@ Undecorated = itaniumDemangle(Decorated + 6, nullptr, nullptr, &Status); } - OS << (Undecorated ? Undecorated : Mangled) << '\n'; - OS.flush(); - + std::string Result(Undecorated ? Undecorated : Mangled); free(Undecorated); + return Result; +} + +// If 'Split' is true, then 'Mangled' is broken into individual words and each +// word is demangled. Otherwise, the entire string is treated as a single +// mangled item. The result is output to 'OS'. +static void demangleLine(llvm::raw_ostream &OS, StringRef Mangled, bool Split) { + std::string Result; + if (Split) { + SmallVector Words; + SplitString(Mangled, Words); + for (auto Word : Words) + Result += demangle(OS, Word) + ' '; + // Remove the trailing space character. + if (Result.back() == ' ') + Result.pop_back(); + } else + Result = demangle(OS, Mangled); + OS << Result << '\n'; + OS.flush(); } int main(int argc, char **argv) { @@ -85,10 +104,10 @@ if (Decorated.empty()) for (std::string Mangled; std::getline(std::cin, Mangled);) - demangle(llvm::outs(), Mangled); + demangleLine(llvm::outs(), Mangled, true); else for (const auto &Symbol : Decorated) - demangle(llvm::outs(), Symbol); + demangleLine(llvm::outs(), Symbol, false); return EXIT_SUCCESS; }