blob: 07dfb5d299776933d9328e6d35a12194b4c59cb7 [file] [log] [blame] [edit]
//===- MCSymbolXCOFF.h - ----------------------------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_MC_MCSYMBOLXCOFF_H
#define LLVM_MC_MCSYMBOLXCOFF_H
#include "llvm/ADT/Optional.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/BinaryFormat/XCOFF.h"
#include "llvm/MC/MCSymbol.h"
namespace llvm {
class MCSectionXCOFF;
class MCSymbolXCOFF : public MCSymbol {
public:
MCSymbolXCOFF(const StringMapEntry<bool> *Name, bool isTemporary)
: MCSymbol(SymbolKindXCOFF, Name, isTemporary) {}
static bool classof(const MCSymbol *S) { return S->isXCOFF(); }
void setStorageClass(XCOFF::StorageClass SC) {
assert((!StorageClass.hasValue() || StorageClass.getValue() == SC) &&
"Redefining StorageClass of XCOFF MCSymbol.");
StorageClass = SC;
};
XCOFF::StorageClass getStorageClass() const {
assert(StorageClass.hasValue() &&
"StorageClass not set on XCOFF MCSymbol.");
return StorageClass.getValue();
}
void setContainingCsect(MCSectionXCOFF *C) {
assert((!ContainingCsect || ContainingCsect == C) &&
"Trying to set a containing csect that doesn't match the one that"
"this symbol is already mapped to.");
ContainingCsect = C;
}
MCSectionXCOFF *getContainingCsect() const {
assert(ContainingCsect &&
"Trying to get containing csect but none was set.");
return ContainingCsect;
}
bool hasContainingCsect() const { return ContainingCsect != nullptr; }
StringRef getUnqualifiedName() const {
const StringRef name = getName();
if (name.back() == ']') {
StringRef lhs, rhs;
std::tie(lhs, rhs) = name.rsplit('[');
assert(!rhs.empty() && "Invalid SMC format in XCOFF symbol.");
return lhs;
}
return name;
}
private:
Optional<XCOFF::StorageClass> StorageClass;
MCSectionXCOFF *ContainingCsect = nullptr;
};
} // end namespace llvm
#endif // LLVM_MC_MCSYMBOLXCOFF_H