6bf7fd693b55375256b5fcdd551e16ca3f76a24e
[oota-llvm.git] / lib / Target / ELFTargetAsmInfo.cpp
1 //===-- ELFTargetAsmInfo.cpp - ELF asm properties ---------------*- 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 defines target asm properties related what form asm statements
11 // should take in general on ELF-based targets
12 //
13 //===----------------------------------------------------------------------===//
14
15 #include "llvm/Constants.h"
16 #include "llvm/DerivedTypes.h"
17 #include "llvm/Function.h"
18 #include "llvm/GlobalVariable.h"
19 #include "llvm/ADT/StringExtras.h"
20 #include "llvm/CodeGen/MachineConstantPool.h"
21 #include "llvm/Target/ELFTargetAsmInfo.h"
22 #include "llvm/Target/TargetMachine.h"
23 #include "llvm/Target/TargetData.h"
24
25 using namespace llvm;
26
27 ELFTargetAsmInfo::ELFTargetAsmInfo(const TargetMachine &TM)
28   : TargetAsmInfo(TM) {
29
30   BSSSection_  = getUnnamedSection("\t.bss",
31                                    SectionFlags::Writeable | SectionFlags::BSS);
32   ReadOnlySection = getNamedSection("\t.rodata", SectionFlags::None);
33   TLSDataSection = getNamedSection("\t.tdata",
34                                    SectionFlags::Writeable | SectionFlags::TLS);
35   TLSBSSSection = getNamedSection("\t.tbss",
36                 SectionFlags::Writeable | SectionFlags::TLS | SectionFlags::BSS);
37
38   DataRelSection = getNamedSection("\t.data.rel", SectionFlags::Writeable);
39   DataRelLocalSection = getNamedSection("\t.data.rel.local",
40                                         SectionFlags::Writeable);
41   DataRelROSection = getNamedSection("\t.data.rel.ro",
42                                      SectionFlags::Writeable);
43   DataRelROLocalSection = getNamedSection("\t.data.rel.ro.local",
44                                           SectionFlags::Writeable);
45 }
46
47 SectionKind::Kind
48 ELFTargetAsmInfo::SectionKindForGlobal(const GlobalValue *GV) const {
49   SectionKind::Kind Kind = TargetAsmInfo::SectionKindForGlobal(GV);
50
51   if (Kind != SectionKind::Data)
52     return Kind;
53
54   // Decide, whether we need data.rel stuff
55   const GlobalVariable* GVar = dyn_cast<GlobalVariable>(GV);
56   if (GVar->hasInitializer()) {
57     Constant *C = GVar->getInitializer();
58     bool isConstant = GVar->isConstant();
59     unsigned Reloc = RelocBehaviour();
60     if (Reloc != Reloc::None && C->ContainsRelocations(Reloc))
61       return (C->ContainsRelocations(Reloc::Global) ?
62               (isConstant ?
63                SectionKind::DataRelRO : SectionKind::DataRel) :
64               (isConstant ?
65                SectionKind::DataRelROLocal : SectionKind::DataRelLocal));
66   }
67
68   return Kind;
69 }
70
71 const Section*
72 ELFTargetAsmInfo::SelectSectionForGlobal(const GlobalValue *GV) const {
73   SectionKind::Kind Kind = SectionKindForGlobal(GV);
74
75   if (const Function *F = dyn_cast<Function>(GV)) {
76     switch (F->getLinkage()) {
77      default: assert(0 && "Unknown linkage type!");
78      case Function::PrivateLinkage:
79      case Function::InternalLinkage:
80      case Function::DLLExportLinkage:
81      case Function::ExternalLinkage:
82       return TextSection;
83      case Function::WeakAnyLinkage:
84      case Function::WeakODRLinkage:
85      case Function::LinkOnceAnyLinkage:
86      case Function::LinkOnceODRLinkage:
87       std::string Name = UniqueSectionForGlobal(GV, Kind);
88       unsigned Flags = SectionFlagsForGlobal(GV, Name.c_str());
89       return getNamedSection(Name.c_str(), Flags);
90     }
91   } else if (const GlobalVariable *GVar = dyn_cast<GlobalVariable>(GV)) {
92     if (GVar->isWeakForLinker()) {
93       std::string Name = UniqueSectionForGlobal(GVar, Kind);
94       unsigned Flags = SectionFlagsForGlobal(GVar, Name.c_str());
95       return getNamedSection(Name.c_str(), Flags);
96     } else {
97       switch (Kind) {
98        case SectionKind::Data:
99        case SectionKind::SmallData:
100         return DataSection;
101        case SectionKind::DataRel:
102         return DataRelSection;
103        case SectionKind::DataRelLocal:
104         return DataRelLocalSection;
105        case SectionKind::DataRelRO:
106         return DataRelROSection;
107        case SectionKind::DataRelROLocal:
108         return DataRelROLocalSection;
109        case SectionKind::BSS:
110        case SectionKind::SmallBSS:
111         // ELF targets usually have BSS sections
112         return getBSSSection_();
113        case SectionKind::ROData:
114        case SectionKind::SmallROData:
115         return getReadOnlySection();
116        case SectionKind::RODataMergeStr:
117         return MergeableStringSection(GVar);
118        case SectionKind::RODataMergeConst:
119         return MergeableConstSection(GVar);
120        case SectionKind::ThreadData:
121         // ELF targets usually support TLS stuff
122         return TLSDataSection;
123        case SectionKind::ThreadBSS:
124         return TLSBSSSection;
125        default:
126         assert(0 && "Unsuported section kind for global");
127       }
128     }
129   } else
130     assert(0 && "Unsupported global");
131
132   return NULL;
133 }
134
135 const Section*
136 ELFTargetAsmInfo::SelectSectionForMachineConst(const Type *Ty) const {
137   // FIXME: Support data.rel stuff someday
138   return MergeableConstSection(Ty);
139 }
140
141 const Section*
142 ELFTargetAsmInfo::MergeableConstSection(const GlobalVariable *GV) const {
143   Constant *C = GV->getInitializer();
144   return MergeableConstSection(C->getType());
145 }
146
147 inline const Section*
148 ELFTargetAsmInfo::MergeableConstSection(const Type *Ty) const {
149   const TargetData *TD = TM.getTargetData();
150
151   // FIXME: string here is temporary, until stuff will fully land in.
152   // We cannot use {Four,Eight,Sixteen}ByteConstantSection here, since it's
153   // currently directly used by asmprinter.
154   unsigned Size = TD->getTypePaddedSize(Ty);
155   if (Size == 4 || Size == 8 || Size == 16) {
156     std::string Name =  ".rodata.cst" + utostr(Size);
157
158     return getNamedSection(Name.c_str(),
159                            SectionFlags::setEntitySize(SectionFlags::Mergeable,
160                                                        Size));
161   }
162
163   return getReadOnlySection();
164 }
165
166 const Section*
167 ELFTargetAsmInfo::MergeableStringSection(const GlobalVariable *GV) const {
168   const TargetData *TD = TM.getTargetData();
169   Constant *C = cast<GlobalVariable>(GV)->getInitializer();
170   const Type *Ty = cast<ArrayType>(C->getType())->getElementType();
171
172   unsigned Size = TD->getTypePaddedSize(Ty);
173   if (Size <= 16) {
174     assert(getCStringSection() && "Should have string section prefix");
175
176     // We also need alignment here
177     unsigned Align = TD->getPrefTypeAlignment(Ty);
178     if (Align < Size)
179       Align = Size;
180
181     std::string Name = getCStringSection() + utostr(Size) + '.' + utostr(Align);
182     unsigned Flags = SectionFlags::setEntitySize(SectionFlags::Mergeable |
183                                                  SectionFlags::Strings,
184                                                  Size);
185     return getNamedSection(Name.c_str(), Flags);
186   }
187
188   return getReadOnlySection();
189 }
190
191 std::string ELFTargetAsmInfo::printSectionFlags(unsigned flags) const {
192   std::string Flags = ",\"";
193
194   if (!(flags & SectionFlags::Debug))
195     Flags += 'a';
196   if (flags & SectionFlags::Code)
197     Flags += 'x';
198   if (flags & SectionFlags::Writeable)
199     Flags += 'w';
200   if (flags & SectionFlags::Mergeable)
201     Flags += 'M';
202   if (flags & SectionFlags::Strings)
203     Flags += 'S';
204   if (flags & SectionFlags::TLS)
205     Flags += 'T';
206   if (flags & SectionFlags::Small)
207     Flags += 's';
208
209   Flags += "\",";
210
211   // If comment string is '@', e.g. as on ARM - use '%' instead
212   if (strcmp(CommentString, "@") == 0)
213     Flags += '%';
214   else
215     Flags += '@';
216
217   // FIXME: There can be exceptions here
218   if (flags & SectionFlags::BSS)
219     Flags += "nobits";
220   else
221     Flags += "progbits";
222
223   if (unsigned entitySize = SectionFlags::getEntitySize(flags))
224     Flags += "," + utostr(entitySize);
225
226   return Flags;
227 }