Home | History | Annotate | Line # | Download | only in MC
      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/MC/MCSymbol.h"
     13 #include <cstdint>
     14 
     15 namespace llvm {
     16 
     17 class MCSymbolCOFF : public MCSymbol {
     18   /// This corresponds to the e_type field of the COFF symbol.
     19   mutable uint16_t Type = 0;
     20 
     21   enum SymbolFlags : uint16_t {
     22     SF_ClassMask = 0x00FF,
     23     SF_ClassShift = 0,
     24 
     25     SF_WeakExternal = 0x0100,
     26     SF_SafeSEH = 0x0200,
     27   };
     28 
     29 public:
     30   MCSymbolCOFF(const StringMapEntry<bool> *Name, bool isTemporary)
     31       : MCSymbol(SymbolKindCOFF, Name, isTemporary) {}
     32 
     33   uint16_t getType() const {
     34     return Type;
     35   }
     36   void setType(uint16_t Ty) const {
     37     Type = Ty;
     38   }
     39 
     40   uint16_t getClass() const {
     41     return (getFlags() & SF_ClassMask) >> SF_ClassShift;
     42   }
     43   void setClass(uint16_t StorageClass) const {
     44     modifyFlags(StorageClass << SF_ClassShift, SF_ClassMask);
     45   }
     46 
     47   bool isWeakExternal() const {
     48     return getFlags() & SF_WeakExternal;
     49   }
     50   void setIsWeakExternal() const {
     51     modifyFlags(SF_WeakExternal, SF_WeakExternal);
     52   }
     53 
     54   bool isSafeSEH() const {
     55     return getFlags() & SF_SafeSEH;
     56   }
     57   void setIsSafeSEH() const {
     58     modifyFlags(SF_SafeSEH, SF_SafeSEH);
     59   }
     60 
     61   static bool classof(const MCSymbol *S) { return S->isCOFF(); }
     62 };
     63 
     64 } // end namespace llvm
     65 
     66 #endif // LLVM_MC_MCSYMBOLCOFF_H
     67