llvm/lib/Demangle/Demangle.cpp

   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 <cstdlib>
  15 #include <cstring>
  16
  17 static bool isItaniumEncoding(const char *S) {
  18   // Itanium encoding requires 1 or 3 leading underscores, followed by 'Z'.
  19   return std::strncmp(S, "_Z", 2) == 0 || std::strncmp(S, "___Z", 4) == 0;
  20 }
  21
  22 static bool isRustEncoding(const char *S) { return S[0] == '_' && S[1] == 'R'; }
  23
  24 static bool isDLangEncoding(const std::string &MangledName) {
  25   return MangledName.size() >= 2 && MangledName[0] == '_' &&
  26          MangledName[1] == 'D';
  27 }
  28
  29 std::string llvm::demangle(const std::string &MangledName) {
  30   std::string Result;
  31   const char *S = MangledName.c_str();
  32
  33   if (nonMicrosoftDemangle(S, Result))
  34     return Result;
  35
  36   if (S[0] == '_' && nonMicrosoftDemangle(S + 1, Result))
  37     return Result;
  38
  39   if (char *Demangled =
  40           microsoftDemangle(S, nullptr, nullptr, nullptr, nullptr)) {
  41     Result = Demangled;
  42     std::free(Demangled);
  43     return Result;
  44   }
  45
  46   return MangledName;
  47 }
  48
  49 bool llvm::nonMicrosoftDemangle(const char *MangledName, std::string &Result) {
  50   char *Demangled = nullptr;
  51   if (isItaniumEncoding(MangledName))
  52     Demangled = itaniumDemangle(MangledName, nullptr, nullptr, nullptr);
  53   else if (isRustEncoding(MangledName))
  54     Demangled = rustDemangle(MangledName, nullptr, nullptr, nullptr);
  55   else if (isDLangEncoding(MangledName))
  56     Demangled = dlangDemangle(MangledName);
  57
  58   if (!Demangled)
  59     return false;
  60
  61   Result = Demangled;
  62   std::free(Demangled);
  63   return true;
  64 }