//===- Demangle.cpp -------------------------------------------------------===//
//
//                     The MCLinker Project
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#include <mcld/Config/Config.h>
#include <mcld/Support/CXADemangle.tcc>
#include <mcld/Support/Demangle.h>

#ifdef HAVE_CXXABI_H
#include <cxxabi.h>
#endif

namespace mcld {

std::string demangleName(const std::string& pName) {
#ifdef HAVE_CXXABI_H
  // Spoil names of symbols with C linkage, so use an heuristic approach to
  // check if the name should be demangled.
  if (pName.substr(0, 2) != "_Z")
    return pName;
  // __cxa_demangle needs manually handle the memory release, so we wrap
  // it into this helper function.
  size_t output_leng;
  int status;
  char* buffer = abi::__cxa_demangle(pName.c_str(), /*buffer=*/0,
                                     &output_leng, &status);
  if (status != 0) { // Failed
    return pName;
  }
  std::string result(buffer);
  free(buffer);

  return result;
#else
  return pName;
#endif
}

bool isCtorOrDtor(const char* pName, size_t pLength)
{
  arena<bs> a;
  Db db(a);
  db.cv = 0;
  db.ref = 0;
  db.encoding_depth = 0;
  db.parsed_ctor_dtor_cv = false;
  db.tag_templates = true;
  db.template_param.emplace_back(a);
  db.fix_forward_references = false;
  db.try_to_parse_template_args = true;
  int internal_status = success;
  demangle(pName, pName + pLength, db, internal_status);
  if (internal_status == success &&
      db.fix_forward_references &&
      !db.template_param.empty() &&
      !db.template_param.front().empty()) {
    db.fix_forward_references = false;
    db.tag_templates = false;
    db.names.clear();
    db.subs.clear();
    demangle(pName, pName + pLength, db, internal_status);
    if (db.fix_forward_references)
      internal_status = invalid_mangled_name;
  }

  if (internal_status != success) {
    db.parsed_ctor_dtor_cv = false;
  }
  return db.parsed_ctor_dtor_cv;
}

} // namespace mcld