1 //===- MCSymbolCOFF.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 
9 #ifndef LLVM_MC_MCSYMBOLCOFF_H
10 #define LLVM_MC_MCSYMBOLCOFF_H
11 
12 #include "llvm/BinaryFormat/COFF.h"
13 #include "llvm/MC/MCSymbol.h"
14 #include <cstdint>
15 
16 namespace llvm {
17 
18 class MCSymbolCOFF : public MCSymbol {
19   /// This corresponds to the e_type field of the COFF symbol.
20   mutable uint16_t Type = 0;
21 
22   enum SymbolFlags : uint16_t {
23     SF_ClassMask = 0x00FF,
24     SF_ClassShift = 0,
25 
26     SF_SafeSEH = 0x0100,
27     SF_WeakExternalCharacteristicsMask = 0x0E00,
28     SF_WeakExternalCharacteristicsShift = 9,
29   };
30 
31 public:
32   MCSymbolCOFF(const StringMapEntry<bool> *Name, bool isTemporary)
33       : MCSymbol(SymbolKindCOFF, Name, isTemporary) {}
34 
35   uint16_t getType() const {
36     return Type;
37   }
38   void setType(uint16_t Ty) const {
39     Type = Ty;
40   }
41 
42   uint16_t getClass() const {
43     return (getFlags() & SF_ClassMask) >> SF_ClassShift;
44   }
45   void setClass(uint16_t StorageClass) const {
46     modifyFlags(StorageClass << SF_ClassShift, SF_ClassMask);
47   }
48 
49   COFF::WeakExternalCharacteristics getWeakExternalCharacteristics() const {
50     return static_cast<COFF::WeakExternalCharacteristics>((getFlags() & SF_WeakExternalCharacteristicsMask) >>
51            SF_WeakExternalCharacteristicsShift);
52   }
53   void setWeakExternalCharacteristics(COFF::WeakExternalCharacteristics Characteristics) const {
54     modifyFlags(Characteristics << SF_WeakExternalCharacteristicsShift,
55                 SF_WeakExternalCharacteristicsMask);
56   }
57   void setIsWeakExternal(bool WeakExt) const {
58     IsWeakExternal = WeakExt;
59   }
60 
61   bool isSafeSEH() const {
62     return getFlags() & SF_SafeSEH;
63   }
64   void setIsSafeSEH() const {
65     modifyFlags(SF_SafeSEH, SF_SafeSEH);
66   }
67 
68   static bool classof(const MCSymbol *S) { return S->isCOFF(); }
69 };
70 
71 } // end namespace llvm
72 
73 #endif // LLVM_MC_MCSYMBOLCOFF_H
74