SymbolRemappingReader.h 4.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132
  1. //===- SymbolRemappingReader.h - Read symbol remapping file -----*- C++ -*-===//
  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. // This file contains definitions needed for reading and applying symbol
  10. // remapping files.
  11. //
  12. // Support is provided only for the Itanium C++ name mangling scheme for now.
  13. //
  14. // NOTE: If you are making changes to this file format, please remember
  15. // to document them in the Clang documentation at
  16. // tools/clang/docs/UsersManual.rst.
  17. //
  18. // File format
  19. // -----------
  20. //
  21. // The symbol remappings are written as an ASCII text file. Blank lines and
  22. // lines starting with a # are ignored. All other lines specify a kind of
  23. // mangled name fragment, along with two fragments of that kind that should
  24. // be treated as equivalent, separated by spaces.
  25. //
  26. // See http://itanium-cxx-abi.github.io/cxx-abi/abi.html#mangling for a
  27. // description of the Itanium name mangling scheme.
  28. //
  29. // The accepted fragment kinds are:
  30. //
  31. // * name A <name>, such as 6foobar or St3__1
  32. // * type A <type>, such as Ss or N4llvm9StringRefE
  33. // * encoding An <encoding> (a complete mangling without the leading _Z)
  34. //
  35. // For example:
  36. //
  37. // # Ignore int / long differences to treat symbols from 32-bit and 64-bit
  38. // # builds with differing size_t / ptrdiff_t / intptr_t as equivalent.
  39. // type i l
  40. // type j m
  41. //
  42. // # Ignore differences between libc++ and libstdc++, and between libstdc++'s
  43. // # C++98 and C++11 ABIs.
  44. // name 3std St3__1
  45. // name 3std St7__cxx11
  46. //
  47. // # Remap a function overload to a specialization of a template (including
  48. // # any local symbols declared within it).
  49. // encoding N2NS1fEi N2NS1fIiEEvT_
  50. //
  51. // # Substitutions must be remapped separately from namespace 'std' for now.
  52. // name Sa NSt3__19allocatorE
  53. // name Sb NSt3__112basic_stringE
  54. // type Ss NSt3__112basic_stringIcSt11char_traitsIcESaE
  55. // # ...
  56. //
  57. //===----------------------------------------------------------------------===//
  58. #ifndef LLVM_SUPPORT_SYMBOLREMAPPINGREADER_H
  59. #define LLVM_SUPPORT_SYMBOLREMAPPINGREADER_H
  60. #include "llvm/ADT/StringRef.h"
  61. #include "llvm/Support/Error.h"
  62. #include "llvm/Support/ItaniumManglingCanonicalizer.h"
  63. #include "llvm/Support/MemoryBuffer.h"
  64. namespace llvm {
  65. class SymbolRemappingParseError : public ErrorInfo<SymbolRemappingParseError> {
  66. public:
  67. SymbolRemappingParseError(StringRef File, int64_t Line, const Twine &Message)
  68. : File(File), Line(Line), Message(Message.str()) {}
  69. void log(llvm::raw_ostream &OS) const override {
  70. OS << File << ':' << Line << ": " << Message;
  71. }
  72. std::error_code convertToErrorCode() const override {
  73. return llvm::inconvertibleErrorCode();
  74. }
  75. StringRef getFileName() const { return File; }
  76. int64_t getLineNum() const { return Line; }
  77. StringRef getMessage() const { return Message; }
  78. static char ID;
  79. private:
  80. std::string File;
  81. int64_t Line;
  82. std::string Message;
  83. };
  84. /// Reader for symbol remapping files.
  85. ///
  86. /// Remaps the symbol names in profile data to match those in the program
  87. /// according to a set of rules specified in a given file.
  88. class SymbolRemappingReader {
  89. public:
  90. /// Read remappings from the given buffer, which must live as long as
  91. /// the remapper.
  92. Error read(MemoryBuffer &B);
  93. /// A Key represents an equivalence class of symbol names.
  94. using Key = uintptr_t;
  95. /// Construct a key for the given symbol, or return an existing one if an
  96. /// equivalent name has already been inserted. The symbol name must live
  97. /// as long as the remapper.
  98. ///
  99. /// The result will be Key() if the name cannot be remapped (typically
  100. /// because it is not a valid mangled name).
  101. Key insert(StringRef FunctionName) {
  102. return Canonicalizer.canonicalize(FunctionName);
  103. }
  104. /// Map the given symbol name into the key for the corresponding equivalence
  105. /// class.
  106. ///
  107. /// The result will typically be Key() if no equivalent symbol has been
  108. /// inserted, but this is not guaranteed: a Key different from all keys ever
  109. /// returned by \c insert may be returned instead.
  110. Key lookup(StringRef FunctionName) {
  111. return Canonicalizer.lookup(FunctionName);
  112. }
  113. private:
  114. ItaniumManglingCanonicalizer Canonicalizer;
  115. };
  116. } // end namespace llvm
  117. #endif // LLVM_SUPPORT_SYMBOLREMAPPINGREADER_H