Fix pr24486.
[oota-llvm.git] / include / llvm / MC / MCSectionCOFF.h
1 //===- MCSectionCOFF.h - COFF Machine Code Sections -------------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file declares the MCSectionCOFF class.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #ifndef LLVM_MC_MCSECTIONCOFF_H
15 #define LLVM_MC_MCSECTIONCOFF_H
16
17 #include "llvm/ADT/StringRef.h"
18 #include "llvm/MC/MCSection.h"
19
20 namespace llvm {
21 class MCSymbol;
22
23 /// MCSectionCOFF - This represents a section on Windows
24   class MCSectionCOFF : public MCSection {
25     // The memory for this string is stored in the same MCContext as *this.
26     StringRef SectionName;
27
28     // FIXME: The following fields should not be mutable, but are for now so
29     // the asm parser can honor the .linkonce directive.
30
31     /// Characteristics - This is the Characteristics field of a section,
32     /// drawn from the enums below.
33     mutable unsigned Characteristics;
34
35     /// The COMDAT symbol of this section. Only valid if this is a COMDAT
36     /// section. Two COMDAT sections are merged if they have the same
37     /// COMDAT symbol.
38     MCSymbol *COMDATSymbol;
39
40     /// Selection - This is the Selection field for the section symbol, if
41     /// it is a COMDAT section (Characteristics & IMAGE_SCN_LNK_COMDAT) != 0
42     mutable int Selection;
43
44   private:
45     friend class MCContext;
46     MCSectionCOFF(StringRef Section, unsigned Characteristics,
47                   MCSymbol *COMDATSymbol, int Selection, SectionKind K,
48                   MCSymbol *Begin)
49         : MCSection(SV_COFF, K, Begin), SectionName(Section),
50           Characteristics(Characteristics), COMDATSymbol(COMDATSymbol),
51           Selection(Selection) {
52       assert ((Characteristics & 0x00F00000) == 0 &&
53         "alignment must not be set upon section creation");
54     }
55     ~MCSectionCOFF() override;
56
57   public:
58     /// ShouldOmitSectionDirective - Decides whether a '.section' directive
59     /// should be printed before the section name
60     bool ShouldOmitSectionDirective(StringRef Name, const MCAsmInfo &MAI) const;
61
62     StringRef getSectionName() const { return SectionName; }
63     unsigned getCharacteristics() const { return Characteristics; }
64     MCSymbol *getCOMDATSymbol() const { return COMDATSymbol; }
65     int getSelection() const { return Selection; }
66
67     void setSelection(int Selection) const;
68
69     void PrintSwitchToSection(const MCAsmInfo &MAI, raw_ostream &OS,
70                               const MCExpr *Subsection) const override;
71     bool UseCodeAlign() const override;
72     bool isVirtualSection() const override;
73
74     static bool classof(const MCSection *S) {
75       return S->getVariant() == SV_COFF;
76     }
77   };
78
79 } // end namespace llvm
80
81 #endif