mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
a0d3214e4c
To get this to work, delay some "cleanup" work in the demangler. For example, we now preserve in the tree whether something was mangled as an allocating initializer, and we only special-case the class vs. non-class cases in the pretty printer. Also fixes a number of remangling bugs, of course. Swift SVN r24534
130 lines
4.3 KiB
C++
130 lines
4.3 KiB
C++
//===-- swift-demangle.cpp - Swift Demangler app ---------------------------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2015 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See http://swift.org/LICENSE.txt for license information
|
|
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This is the entry point.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "swift/Basic/DemangleWrappers.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
#include "llvm/Support/PrettyStackTrace.h"
|
|
#include "llvm/Support/Regex.h"
|
|
#include "llvm/Support/Signals.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
#include <string>
|
|
|
|
static llvm::cl::opt<bool>
|
|
ExpandMode("expand",
|
|
llvm::cl::desc("Expand mode (show node structure of the demangling)"));
|
|
|
|
static llvm::cl::opt<bool>
|
|
CompactMode("compact",
|
|
llvm::cl::desc("Compact mode (only emit the demangled names)"));
|
|
|
|
static llvm::cl::opt<bool>
|
|
TreeOnly("tree-only",
|
|
llvm::cl::desc("Tree-only mode (do not show the demangled string)"));
|
|
|
|
static llvm::cl::opt<bool>
|
|
RemangleMode("test-remangle",
|
|
llvm::cl::desc("Remangle test mode (show the remangled string)"));
|
|
|
|
static llvm::cl::opt<bool>
|
|
DisableSugar("no-sugar",
|
|
llvm::cl::desc("No sugar mode (disable common language idioms such as ? and [] from the output)"));
|
|
|
|
static llvm::cl::list<std::string>
|
|
InputNames(llvm::cl::Positional, llvm::cl::desc("[mangled name...]"),
|
|
llvm::cl::ZeroOrMore);
|
|
|
|
static void demangle(llvm::raw_ostream &os, llvm::StringRef name,
|
|
const swift::Demangle::DemangleOptions &options) {
|
|
bool hadLeadingUnderscore = false;
|
|
if (name.startswith("__")) {
|
|
hadLeadingUnderscore = true;
|
|
name = name.substr(1);
|
|
}
|
|
swift::Demangle::NodePointer pointer =
|
|
swift::demangle_wrappers::demangleSymbolAsNode(name);
|
|
if (ExpandMode || TreeOnly) {
|
|
llvm::outs() << "Demangling for " << name << '\n';
|
|
swift::demangle_wrappers::NodeDumper(pointer).print(llvm::outs());
|
|
}
|
|
if (RemangleMode) {
|
|
if (hadLeadingUnderscore) llvm::outs() << '_';
|
|
// Just reprint the original mangled name if it didn't demangle.
|
|
// This makes it easier to share the same database between the
|
|
// mangling and demangling tests.
|
|
if (!pointer) {
|
|
llvm::outs() << name;
|
|
} else {
|
|
llvm::outs() << swift::Demangle::mangleNode(pointer);
|
|
}
|
|
return;
|
|
}
|
|
if (!TreeOnly) {
|
|
std::string string = swift::Demangle::nodeToString(pointer, options);
|
|
if (!CompactMode)
|
|
llvm::outs() << name << " ---> ";
|
|
llvm::outs() << (string.empty() ? name : llvm::StringRef(string));
|
|
}
|
|
}
|
|
|
|
static llvm::StringRef substrBefore(llvm::StringRef whole,
|
|
llvm::StringRef part) {
|
|
return whole.slice(0, part.data() - whole.data());
|
|
}
|
|
|
|
static llvm::StringRef substrAfter(llvm::StringRef whole,
|
|
llvm::StringRef part) {
|
|
return whole.substr((part.data() - whole.data()) + part.size());
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
llvm::sys::PrintStackTraceOnErrorSignal();
|
|
llvm::PrettyStackTraceProgram X(argc, argv);
|
|
llvm::cl::ParseCommandLineOptions(argc, argv);
|
|
|
|
swift::Demangle::DemangleOptions options;
|
|
options.SynthesizeSugarOnTypes = !DisableSugar;
|
|
|
|
if (InputNames.empty()) {
|
|
CompactMode = true;
|
|
auto input = llvm::MemoryBuffer::getSTDIN();
|
|
if (!input) {
|
|
llvm::errs() << input.getError().message() << '\n';
|
|
return EXIT_FAILURE;
|
|
}
|
|
llvm::StringRef inputContents = input.get()->getBuffer();
|
|
|
|
// This doesn't handle Unicode symbols, but maybe that's okay.
|
|
llvm::Regex maybeSymbol("_T[_a-zA-Z0-9]+");
|
|
llvm::SmallVector<llvm::StringRef, 1> matches;
|
|
while (maybeSymbol.match(inputContents, &matches)) {
|
|
llvm::outs() << substrBefore(inputContents, matches.front());
|
|
demangle(llvm::outs(), matches.front(), options);
|
|
inputContents = substrAfter(inputContents, matches.front());
|
|
}
|
|
llvm::outs() << inputContents;
|
|
|
|
} else {
|
|
for (llvm::StringRef name : InputNames) {
|
|
demangle(llvm::outs(), name, options);
|
|
llvm::outs() << '\n';
|
|
}
|
|
}
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|