1//===-- Demangle.cpp - Common demangling functions ------------------------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8///
9/// \file This file contains definitions of common demangling functions.
10///
11//===----------------------------------------------------------------------===//
12
13#include "llvm/Demangle/Demangle.h"
14#include "llvm/Demangle/StringViewExtras.h"
15#include <cstdlib>
16#include <string_view>
17
18using llvm::itanium_demangle::starts_with;
19
20std::string llvm::demangle(std::string_view MangledName) {
21 std::string Result;
22
23 if (nonMicrosoftDemangle(MangledName, Result))
24 return Result;
25
26 if (starts_with(self: MangledName, C: '_') &&
27 nonMicrosoftDemangle(MangledName: MangledName.substr(pos: 1), Result,
28 /*CanHaveLeadingDot=*/false))
29 return Result;
30
31 if (char *Demangled = microsoftDemangle(mangled_name: MangledName, n_read: nullptr, status: nullptr)) {
32 Result = Demangled;
33 std::free(ptr: Demangled);
34 } else {
35 Result = MangledName;
36 }
37 return Result;
38}
39
40static bool isItaniumEncoding(std::string_view S) {
41 // Itanium encoding requires 1 or 3 leading underscores, followed by 'Z'.
42 return starts_with(haystack: S, needle: "_Z") || starts_with(haystack: S, needle: "___Z");
43}
44
45static bool isRustEncoding(std::string_view S) { return starts_with(haystack: S, needle: "_R"); }
46
47static bool isDLangEncoding(std::string_view S) { return starts_with(haystack: S, needle: "_D"); }
48
49bool llvm::nonMicrosoftDemangle(std::string_view MangledName,
50 std::string &Result, bool CanHaveLeadingDot,
51 bool ParseParams) {
52 char *Demangled = nullptr;
53
54 // Do not consider the dot prefix as part of the demangled symbol name.
55 if (CanHaveLeadingDot && MangledName.size() > 0 && MangledName[0] == '.') {
56 MangledName.remove_prefix(n: 1);
57 Result = ".";
58 }
59
60 if (isItaniumEncoding(S: MangledName))
61 Demangled = itaniumDemangle(mangled_name: MangledName, ParseParams);
62 else if (isRustEncoding(S: MangledName))
63 Demangled = rustDemangle(MangledName);
64 else if (isDLangEncoding(S: MangledName))
65 Demangled = dlangDemangle(MangledName);
66
67 if (!Demangled)
68 return false;
69
70 Result += Demangled;
71 std::free(ptr: Demangled);
72 return true;
73}
74