MCSymbolXCOFF.h 2.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. //===- MCSymbolXCOFF.h - ----------------------------------------*- 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. #ifndef LLVM_MC_MCSYMBOLXCOFF_H
  9. #define LLVM_MC_MCSYMBOLXCOFF_H
  10. #include "llvm/ADT/Optional.h"
  11. #include "llvm/ADT/StringRef.h"
  12. #include "llvm/BinaryFormat/XCOFF.h"
  13. #include "llvm/MC/MCSymbol.h"
  14. namespace llvm {
  15. class MCSectionXCOFF;
  16. class MCSymbolXCOFF : public MCSymbol {
  17. public:
  18. MCSymbolXCOFF(const StringMapEntry<bool> *Name, bool isTemporary)
  19. : MCSymbol(SymbolKindXCOFF, Name, isTemporary) {}
  20. static bool classof(const MCSymbol *S) { return S->isXCOFF(); }
  21. static StringRef getUnqualifiedName(StringRef Name) {
  22. if (Name.back() == ']') {
  23. StringRef Lhs, Rhs;
  24. std::tie(Lhs, Rhs) = Name.rsplit('[');
  25. assert(!Rhs.empty() && "Invalid SMC format in XCOFF symbol.");
  26. return Lhs;
  27. }
  28. return Name;
  29. }
  30. void setStorageClass(XCOFF::StorageClass SC) {
  31. StorageClass = SC;
  32. };
  33. XCOFF::StorageClass getStorageClass() const {
  34. assert(StorageClass.hasValue() &&
  35. "StorageClass not set on XCOFF MCSymbol.");
  36. return StorageClass.getValue();
  37. }
  38. StringRef getUnqualifiedName() const { return getUnqualifiedName(getName()); }
  39. MCSectionXCOFF *getRepresentedCsect() const;
  40. void setRepresentedCsect(MCSectionXCOFF *C);
  41. void setVisibilityType(XCOFF::VisibilityType SVT) { VisibilityType = SVT; };
  42. XCOFF::VisibilityType getVisibilityType() const { return VisibilityType; }
  43. bool hasRename() const { return !SymbolTableName.empty(); }
  44. void setSymbolTableName(StringRef STN) { SymbolTableName = STN; }
  45. StringRef getSymbolTableName() const {
  46. if (hasRename())
  47. return SymbolTableName;
  48. return getUnqualifiedName();
  49. }
  50. private:
  51. Optional<XCOFF::StorageClass> StorageClass;
  52. MCSectionXCOFF *RepresentedCsect = nullptr;
  53. XCOFF::VisibilityType VisibilityType = XCOFF::SYM_V_UNSPECIFIED;
  54. StringRef SymbolTableName;
  55. };
  56. } // end namespace llvm
  57. #endif // LLVM_MC_MCSYMBOLXCOFF_H