137 lines
4.9 KiB
C++
137 lines
4.9 KiB
C++
//===-- llvm-undname.cpp - Microsoft ABI name undecorator
|
|
//------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This utility works like the windows undname utility. It converts mangled
|
|
// Microsoft symbol names into pretty C/C++ human-readable names.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/Demangle/Demangle.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/ErrorOr.h"
|
|
#include "llvm/Support/InitLLVM.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
#include "llvm/Support/Process.h"
|
|
#include "llvm/Support/WithColor.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include <cstdio>
|
|
#include <cstring>
|
|
#include <iostream>
|
|
#include <string>
|
|
|
|
using namespace llvm;
|
|
|
|
cl::opt<bool> DumpBackReferences("backrefs", cl::Optional,
|
|
cl::desc("dump backreferences"), cl::Hidden,
|
|
cl::init(false));
|
|
cl::opt<bool> NoAccessSpecifier("no-access-specifier", cl::Optional,
|
|
cl::desc("skip access specifiers"), cl::Hidden,
|
|
cl::init(false));
|
|
cl::opt<bool> NoCallingConvention("no-calling-convention", cl::Optional,
|
|
cl::desc("skip calling convention"),
|
|
cl::Hidden, cl::init(false));
|
|
cl::opt<bool> NoReturnType("no-return-type", cl::Optional,
|
|
cl::desc("skip return types"), cl::Hidden,
|
|
cl::init(false));
|
|
cl::opt<bool> NoMemberType("no-member-type", cl::Optional,
|
|
cl::desc("skip member types"), cl::Hidden,
|
|
cl::init(false));
|
|
cl::opt<std::string> RawFile("raw-file", cl::Optional,
|
|
cl::desc("for fuzzer data"), cl::Hidden);
|
|
cl::opt<bool> WarnTrailing("warn-trailing", cl::Optional,
|
|
cl::desc("warn on trailing characters"), cl::Hidden,
|
|
cl::init(false));
|
|
cl::list<std::string> Symbols(cl::Positional, cl::desc("<input symbols>"),
|
|
cl::ZeroOrMore);
|
|
|
|
static bool msDemangle(const std::string &S) {
|
|
int Status;
|
|
MSDemangleFlags Flags = MSDF_None;
|
|
if (DumpBackReferences)
|
|
Flags = MSDemangleFlags(Flags | MSDF_DumpBackrefs);
|
|
if (NoAccessSpecifier)
|
|
Flags = MSDemangleFlags(Flags | MSDF_NoAccessSpecifier);
|
|
if (NoCallingConvention)
|
|
Flags = MSDemangleFlags(Flags | MSDF_NoCallingConvention);
|
|
if (NoReturnType)
|
|
Flags = MSDemangleFlags(Flags | MSDF_NoReturnType);
|
|
if (NoMemberType)
|
|
Flags = MSDemangleFlags(Flags | MSDF_NoMemberType);
|
|
|
|
size_t NRead;
|
|
char *ResultBuf =
|
|
microsoftDemangle(S.c_str(), &NRead, nullptr, nullptr, &Status, Flags);
|
|
if (Status == llvm::demangle_success) {
|
|
outs() << ResultBuf << "\n";
|
|
outs().flush();
|
|
if (WarnTrailing && NRead < S.size())
|
|
WithColor::warning() << "trailing characters: " << S.c_str() + NRead
|
|
<< "\n";
|
|
} else {
|
|
WithColor::error() << "Invalid mangled name\n";
|
|
}
|
|
std::free(ResultBuf);
|
|
return Status == llvm::demangle_success;
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
InitLLVM X(argc, argv);
|
|
|
|
cl::ParseCommandLineOptions(argc, argv, "llvm-undname\n");
|
|
|
|
if (!RawFile.empty()) {
|
|
ErrorOr<std::unique_ptr<MemoryBuffer>> FileOrErr =
|
|
MemoryBuffer::getFileOrSTDIN(RawFile);
|
|
if (std::error_code EC = FileOrErr.getError()) {
|
|
WithColor::error() << "Could not open input file \'" << RawFile
|
|
<< "\': " << EC.message() << '\n';
|
|
return 1;
|
|
}
|
|
return msDemangle(std::string(FileOrErr->get()->getBuffer())) ? 0 : 1;
|
|
}
|
|
|
|
bool Success = true;
|
|
if (Symbols.empty()) {
|
|
while (true) {
|
|
std::string LineStr;
|
|
std::getline(std::cin, LineStr);
|
|
if (std::cin.eof())
|
|
break;
|
|
|
|
StringRef Line(LineStr);
|
|
Line = Line.trim();
|
|
if (Line.empty() || Line.startswith("#") || Line.startswith(";"))
|
|
continue;
|
|
|
|
// If the user is manually typing in these decorated names, don't echo
|
|
// them to the terminal a second time. If they're coming from redirected
|
|
// input, however, then we should display the input line so that the
|
|
// mangled and demangled name can be easily correlated in the output.
|
|
if (!sys::Process::StandardInIsUserInput()) {
|
|
outs() << Line << "\n";
|
|
outs().flush();
|
|
}
|
|
if (!msDemangle(std::string(Line)))
|
|
Success = false;
|
|
outs() << "\n";
|
|
}
|
|
} else {
|
|
for (StringRef S : Symbols) {
|
|
outs() << S << "\n";
|
|
outs().flush();
|
|
if (!msDemangle(std::string(S)))
|
|
Success = false;
|
|
outs() << "\n";
|
|
}
|
|
}
|
|
|
|
return Success ? 0 : 1;
|
|
}
|