MCSymbolXCOFF.h 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687
  1. #pragma once
  2. #ifdef __GNUC__
  3. #pragma GCC diagnostic push
  4. #pragma GCC diagnostic ignored "-Wunused-parameter"
  5. #endif
  6. //===- MCSymbolXCOFF.h - ----------------------------------------*- C++ -*-===//
  7. //
  8. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
  9. // See https://llvm.org/LICENSE.txt for license information.
  10. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
  11. //
  12. //===----------------------------------------------------------------------===//
  13. #ifndef LLVM_MC_MCSYMBOLXCOFF_H
  14. #define LLVM_MC_MCSYMBOLXCOFF_H
  15. #include "llvm/ADT/Optional.h"
  16. #include "llvm/ADT/StringRef.h"
  17. #include "llvm/BinaryFormat/XCOFF.h"
  18. #include "llvm/MC/MCSymbol.h"
  19. namespace llvm {
  20. class MCSectionXCOFF;
  21. class MCSymbolXCOFF : public MCSymbol {
  22. public:
  23. MCSymbolXCOFF(const StringMapEntry<bool> *Name, bool isTemporary)
  24. : MCSymbol(SymbolKindXCOFF, Name, isTemporary) {}
  25. static bool classof(const MCSymbol *S) { return S->isXCOFF(); }
  26. static StringRef getUnqualifiedName(StringRef Name) {
  27. if (Name.back() == ']') {
  28. StringRef Lhs, Rhs;
  29. std::tie(Lhs, Rhs) = Name.rsplit('[');
  30. assert(!Rhs.empty() && "Invalid SMC format in XCOFF symbol.");
  31. return Lhs;
  32. }
  33. return Name;
  34. }
  35. void setStorageClass(XCOFF::StorageClass SC) {
  36. StorageClass = SC;
  37. };
  38. XCOFF::StorageClass getStorageClass() const {
  39. assert(StorageClass.hasValue() &&
  40. "StorageClass not set on XCOFF MCSymbol.");
  41. return StorageClass.getValue();
  42. }
  43. StringRef getUnqualifiedName() const { return getUnqualifiedName(getName()); }
  44. MCSectionXCOFF *getRepresentedCsect() const;
  45. void setRepresentedCsect(MCSectionXCOFF *C);
  46. void setVisibilityType(XCOFF::VisibilityType SVT) { VisibilityType = SVT; };
  47. XCOFF::VisibilityType getVisibilityType() const { return VisibilityType; }
  48. bool hasRename() const { return !SymbolTableName.empty(); }
  49. void setSymbolTableName(StringRef STN) { SymbolTableName = STN; }
  50. StringRef getSymbolTableName() const {
  51. if (hasRename())
  52. return SymbolTableName;
  53. return getUnqualifiedName();
  54. }
  55. private:
  56. Optional<XCOFF::StorageClass> StorageClass;
  57. MCSectionXCOFF *RepresentedCsect = nullptr;
  58. XCOFF::VisibilityType VisibilityType = XCOFF::SYM_V_UNSPECIFIED;
  59. StringRef SymbolTableName;
  60. };
  61. } // end namespace llvm
  62. #endif // LLVM_MC_MCSYMBOLXCOFF_H
  63. #ifdef __GNUC__
  64. #pragma GCC diagnostic pop
  65. #endif