Make the 'get*AlignmentFromAttr' functions into member functions within the Attribute...
[oota-llvm.git] / include / llvm / Attributes.h
1 //===-- llvm/Attributes.h - Container for Attributes ------------*- 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 contains the simple types necessary to represent the
11 // attributes associated with functions and their calls.
12 //
13 //===----------------------------------------------------------------------===//
14
15 #ifndef LLVM_ATTRIBUTES_H
16 #define LLVM_ATTRIBUTES_H
17
18 #include "llvm/Support/MathExtras.h"
19 #include "llvm/ADT/ArrayRef.h"
20 #include <cassert>
21 #include <string>
22
23 namespace llvm {
24
25 class Type;
26
27 namespace Attribute {
28
29 /// We use this proxy POD type to allow constructing Attributes constants using
30 /// initializer lists. Do not use this class directly.
31 struct AttrConst {
32   uint64_t v;
33   AttrConst operator | (const AttrConst Attrs) const {
34     AttrConst Res = {v | Attrs.v};
35     return Res;
36   }
37   AttrConst operator ~ () const {
38     AttrConst Res = {~v};
39     return Res;
40   }
41 };
42
43 /// Function parameters and results can have attributes to indicate how they
44 /// should be treated by optimizations and code generation. This enumeration
45 /// lists the attributes that can be associated with parameters, function
46 /// results or the function itself.
47 /// @brief Function attributes.
48
49 /// We declare AttrConst objects that will be used throughout the code and also
50 /// raw uint64_t objects with _i suffix to be used below for other constant
51 /// declarations. This is done to avoid static CTORs and at the same time to
52 /// keep type-safety of Attributes.
53 #define DECLARE_LLVM_ATTRIBUTE(name, value) \
54   const uint64_t name##_i = value; \
55   const AttrConst name = {value};
56
57 DECLARE_LLVM_ATTRIBUTE(None,0)    ///< No attributes have been set
58 DECLARE_LLVM_ATTRIBUTE(ZExt,1<<0) ///< Zero extended before/after call
59 DECLARE_LLVM_ATTRIBUTE(SExt,1<<1) ///< Sign extended before/after call
60 DECLARE_LLVM_ATTRIBUTE(NoReturn,1<<2) ///< Mark the function as not returning
61 DECLARE_LLVM_ATTRIBUTE(InReg,1<<3) ///< Force argument to be passed in register
62 DECLARE_LLVM_ATTRIBUTE(StructRet,1<<4) ///< Hidden pointer to structure to return
63 DECLARE_LLVM_ATTRIBUTE(NoUnwind,1<<5) ///< Function doesn't unwind stack
64 DECLARE_LLVM_ATTRIBUTE(NoAlias,1<<6) ///< Considered to not alias after call
65 DECLARE_LLVM_ATTRIBUTE(ByVal,1<<7) ///< Pass structure by value
66 DECLARE_LLVM_ATTRIBUTE(Nest,1<<8) ///< Nested function static chain
67 DECLARE_LLVM_ATTRIBUTE(ReadNone,1<<9) ///< Function does not access memory
68 DECLARE_LLVM_ATTRIBUTE(ReadOnly,1<<10) ///< Function only reads from memory
69 DECLARE_LLVM_ATTRIBUTE(NoInline,1<<11) ///< inline=never
70 DECLARE_LLVM_ATTRIBUTE(AlwaysInline,1<<12) ///< inline=always
71 DECLARE_LLVM_ATTRIBUTE(OptimizeForSize,1<<13) ///< opt_size
72 DECLARE_LLVM_ATTRIBUTE(StackProtect,1<<14) ///< Stack protection.
73 DECLARE_LLVM_ATTRIBUTE(StackProtectReq,1<<15) ///< Stack protection required.
74 DECLARE_LLVM_ATTRIBUTE(Alignment,31<<16) ///< Alignment of parameter (5 bits)
75                                      // stored as log2 of alignment with +1 bias
76                                      // 0 means unaligned different from align 1
77 DECLARE_LLVM_ATTRIBUTE(NoCapture,1<<21) ///< Function creates no aliases of pointer
78 DECLARE_LLVM_ATTRIBUTE(NoRedZone,1<<22) /// disable redzone
79 DECLARE_LLVM_ATTRIBUTE(NoImplicitFloat,1<<23) /// disable implicit floating point
80                                            /// instructions.
81 DECLARE_LLVM_ATTRIBUTE(Naked,1<<24) ///< Naked function
82 DECLARE_LLVM_ATTRIBUTE(InlineHint,1<<25) ///< source said inlining was
83                                            ///desirable
84 DECLARE_LLVM_ATTRIBUTE(StackAlignment,7<<26) ///< Alignment of stack for
85                                            ///function (3 bits) stored as log2
86                                            ///of alignment with +1 bias
87                                            ///0 means unaligned (different from
88                                            ///alignstack= {1))
89 DECLARE_LLVM_ATTRIBUTE(ReturnsTwice,1<<29) ///< Function can return twice
90 DECLARE_LLVM_ATTRIBUTE(UWTable,1<<30) ///< Function must be in a unwind
91                                            ///table
92 DECLARE_LLVM_ATTRIBUTE(NonLazyBind,1U<<31) ///< Function is called early and/or
93                                             /// often, so lazy binding isn't
94                                             /// worthwhile.
95 DECLARE_LLVM_ATTRIBUTE(AddressSafety,1ULL<<32) ///< Address safety checking is on.
96
97 #undef DECLARE_LLVM_ATTRIBUTE
98
99 }  // namespace Attribute
100
101 /// Attributes - A bitset of attributes.
102 class Attributes {
103   // Currently, we need less than 64 bits.
104   uint64_t Bits;
105 public:
106   Attributes() : Bits(0) { }
107   explicit Attributes(uint64_t Val) : Bits(Val) { }
108   /*implicit*/ Attributes(Attribute::AttrConst Val) : Bits(Val.v) { }
109
110   // Attribute query methods.
111   // FIXME: StackAlignment & Alignment attributes have no predicate methods.
112   bool hasAttributes() const {
113     return Bits != 0;
114   }
115   bool hasAttributes(const Attributes &A) const {
116     return Bits & A.Bits;
117   }
118
119   bool hasZExtAttr() const {
120     return Bits & Attribute::ZExt_i;
121   }
122   bool hasSExtAttr() const {
123     return Bits & Attribute::SExt_i;
124   }
125   bool hasNoReturnAttr() const {
126     return Bits & Attribute::NoReturn_i;
127   }
128   bool hasInRegAttr() const {
129     return Bits & Attribute::InReg_i;
130   }
131   bool hasStructRetAttr() const {
132     return Bits & Attribute::StructRet_i;
133   }
134   bool hasNoUnwindAttr() const {
135     return Bits & Attribute::NoUnwind_i;
136   }
137   bool hasNoAliasAttr() const {
138     return Bits & Attribute::NoAlias_i;
139   }
140   bool hasByValAttr() const {
141     return Bits & Attribute::ByVal_i;
142   }
143   bool hasNestAttr() const {
144     return Bits & Attribute::Nest_i;
145   }
146   bool hasReadNoneAttr() const {
147     return Bits & Attribute::ReadNone_i;
148   }
149   bool hasReadOnlyAttr() const {
150     return Bits & Attribute::ReadOnly_i;
151   }
152   bool hasNoInlineAttr() const {
153     return Bits & Attribute::NoInline_i;
154   }
155   bool hasAlwaysInlineAttr() const {
156     return Bits & Attribute::AlwaysInline_i;
157   }
158   bool hasOptimizeForSizeAttr() const {
159     return Bits & Attribute::OptimizeForSize_i;
160   }
161   bool hasStackProtectAttr() const {
162     return Bits & Attribute::StackProtect_i;
163   }
164   bool hasStackProtectReqAttr() const {
165     return Bits & Attribute::StackProtectReq_i;
166   }
167   bool hasAlignmentAttr() const {
168     return Bits & Attribute::Alignment_i;
169   }
170   bool hasNoCaptureAttr() const {
171     return Bits & Attribute::NoCapture_i;
172   }
173   bool hasNoRedZoneAttr() const {
174     return Bits & Attribute::NoRedZone_i;
175   }
176   bool hasNoImplicitFloatAttr() const {
177     return Bits & Attribute::NoImplicitFloat_i;
178   }
179   bool hasNakedAttr() const {
180     return Bits & Attribute::Naked_i;
181   }
182   bool hasInlineHintAttr() const {
183     return Bits & Attribute::InlineHint_i;
184   }
185   bool hasReturnsTwiceAttr() const {
186     return Bits & Attribute::ReturnsTwice_i;
187   }
188   bool hasStackAlignmentAttr() const {
189     return Bits & Attribute::StackAlignment_i;
190   }
191   bool hasUWTableAttr() const {
192     return Bits & Attribute::UWTable_i;
193   }
194   bool hasNonLazyBindAttr() const {
195     return Bits & Attribute::NonLazyBind_i;
196   }
197   bool hasAddressSafetyAttr() const {
198     return Bits & Attribute::AddressSafety_i;
199   }
200
201   uint64_t getRawAlignment() const {
202     return Bits & Attribute::Alignment_i;
203   }
204   uint64_t getRawStackAlignment() const {
205     return Bits & Attribute::StackAlignment_i;
206   }
207
208   /// This returns the alignment field of an attribute as a byte alignment
209   /// value.
210   unsigned getAlignment() const {
211     if (!hasAlignmentAttr())
212       return 0;
213
214     return 1U << ((getRawAlignment() >> 16) - 1);
215   }
216
217   /// This returns the stack alignment field of an attribute as a byte alignment
218   /// value.
219   unsigned getStackAlignment() const {
220     if (!hasStackAlignmentAttr())
221       return 0;
222
223     return 1U << ((getRawStackAlignment() >> 26) - 1);
224   }
225
226   // This is a "safe bool() operator".
227   operator const void *() const { return Bits ? this : 0; }
228   bool isEmptyOrSingleton() const { return (Bits & (Bits - 1)) == 0; }
229   bool operator == (const Attributes &Attrs) const {
230     return Bits == Attrs.Bits;
231   }
232   bool operator != (const Attributes &Attrs) const {
233     return Bits != Attrs.Bits;
234   }
235   Attributes operator | (const Attributes &Attrs) const {
236     return Attributes(Bits | Attrs.Bits);
237   }
238   Attributes operator & (const Attributes &Attrs) const {
239     return Attributes(Bits & Attrs.Bits);
240   }
241   Attributes operator ^ (const Attributes &Attrs) const {
242     return Attributes(Bits ^ Attrs.Bits);
243   }
244   Attributes &operator |= (const Attributes &Attrs) {
245     Bits |= Attrs.Bits;
246     return *this;
247   }
248   Attributes &operator &= (const Attributes &Attrs) {
249     Bits &= Attrs.Bits;
250     return *this;
251   }
252   Attributes operator ~ () const { return Attributes(~Bits); }
253   uint64_t Raw() const { return Bits; }
254
255   /// The set of Attributes set in Attributes is converted to a string of
256   /// equivalent mnemonics. This is, presumably, for writing out the mnemonics
257   /// for the assembly writer.
258   /// @brief Convert attribute bits to text
259   std::string getAsString() const;
260 };
261
262 namespace Attribute {
263
264 /// Note that uwtable is about the ABI or the user mandating an entry in the
265 /// unwind table. The nounwind attribute is about an exception passing by the
266 /// function.
267 /// In a theoretical system that uses tables for profiling and sjlj for
268 /// exceptions, they would be fully independent. In a normal system that
269 /// uses tables for both, the semantics are:
270 /// nil                = Needs an entry because an exception might pass by.
271 /// nounwind           = No need for an entry
272 /// uwtable            = Needs an entry because the ABI says so and because
273 ///                      an exception might pass by.
274 /// uwtable + nounwind = Needs an entry because the ABI says so.
275
276 /// @brief Attributes that only apply to function parameters.
277 const AttrConst ParameterOnly = {ByVal_i | Nest_i |
278     StructRet_i | NoCapture_i};
279
280 /// @brief Attributes that may be applied to the function itself.  These cannot
281 /// be used on return values or function parameters.
282 const AttrConst FunctionOnly = {NoReturn_i | NoUnwind_i | ReadNone_i |
283   ReadOnly_i | NoInline_i | AlwaysInline_i | OptimizeForSize_i |
284   StackProtect_i | StackProtectReq_i | NoRedZone_i | NoImplicitFloat_i |
285   Naked_i | InlineHint_i | StackAlignment_i |
286   UWTable_i | NonLazyBind_i | ReturnsTwice_i | AddressSafety_i};
287
288 /// @brief Parameter attributes that do not apply to vararg call arguments.
289 const AttrConst VarArgsIncompatible = {StructRet_i};
290
291 /// @brief Attributes that are mutually incompatible.
292 const AttrConst MutuallyIncompatible[5] = {
293   {ByVal_i | Nest_i | StructRet_i},
294   {ByVal_i | Nest_i | InReg_i },
295   {ZExt_i  | SExt_i},
296   {ReadNone_i | ReadOnly_i},
297   {NoInline_i | AlwaysInline_i}
298 };
299
300 /// @brief Which attributes cannot be applied to a type.
301 Attributes typeIncompatible(Type *Ty);
302
303 /// This turns an int alignment (a power of 2, normally) into the
304 /// form used internally in Attributes.
305 inline Attributes constructAlignmentFromInt(unsigned i) {
306   // Default alignment, allow the target to define how to align it.
307   if (i == 0)
308     return None;
309
310   assert(isPowerOf2_32(i) && "Alignment must be a power of two.");
311   assert(i <= 0x40000000 && "Alignment too large.");
312   return Attributes((Log2_32(i)+1) << 16);
313 }
314
315 /// This turns an int stack alignment (which must be a power of 2) into
316 /// the form used internally in Attributes.
317 inline Attributes constructStackAlignmentFromInt(unsigned i) {
318   // Default alignment, allow the target to define how to align it.
319   if (i == 0)
320     return None;
321
322   assert(isPowerOf2_32(i) && "Alignment must be a power of two.");
323   assert(i <= 0x100 && "Alignment too large.");
324   return Attributes((Log2_32(i)+1) << 26);
325 }
326
327 /// This returns an integer containing an encoding of all the
328 /// LLVM attributes found in the given attribute bitset.  Any
329 /// change to this encoding is a breaking change to bitcode
330 /// compatibility.
331 inline uint64_t encodeLLVMAttributesForBitcode(Attributes Attrs) {
332   // FIXME: It doesn't make sense to store the alignment information as an
333   // expanded out value, we should store it as a log2 value.  However, we can't
334   // just change that here without breaking bitcode compatibility.  If this ever
335   // becomes a problem in practice, we should introduce new tag numbers in the
336   // bitcode file and have those tags use a more efficiently encoded alignment
337   // field.
338
339   // Store the alignment in the bitcode as a 16-bit raw value instead of a
340   // 5-bit log2 encoded value. Shift the bits above the alignment up by
341   // 11 bits.
342
343   uint64_t EncodedAttrs = Attrs.Raw() & 0xffff;
344   if (Attrs.hasAlignmentAttr())
345     EncodedAttrs |= (1ull << 16) <<
346       ((Attrs.getRawAlignment() - 1) >> 16);
347   EncodedAttrs |= (Attrs.Raw() & (0xfffull << 21)) << 11;
348
349   return EncodedAttrs;
350 }
351
352 /// This returns an attribute bitset containing the LLVM attributes
353 /// that have been decoded from the given integer.  This function
354 /// must stay in sync with 'encodeLLVMAttributesForBitcode'.
355 inline Attributes decodeLLVMAttributesForBitcode(uint64_t EncodedAttrs) {
356   // The alignment is stored as a 16-bit raw value from bits 31--16.
357   // We shift the bits above 31 down by 11 bits.
358
359   unsigned Alignment = (EncodedAttrs & (0xffffull << 16)) >> 16;
360   assert((!Alignment || isPowerOf2_32(Alignment)) &&
361          "Alignment must be a power of two.");
362
363   Attributes Attrs(EncodedAttrs & 0xffff);
364   if (Alignment)
365     Attrs |= Attribute::constructAlignmentFromInt(Alignment);
366   Attrs |= Attributes((EncodedAttrs & (0xfffull << 32)) >> 11);
367
368   return Attrs;
369 }
370
371 } // end namespace Attribute
372
373 /// This is just a pair of values to associate a set of attributes
374 /// with an index.
375 struct AttributeWithIndex {
376   Attributes Attrs; ///< The attributes that are set, or'd together.
377   unsigned Index; ///< Index of the parameter for which the attributes apply.
378                   ///< Index 0 is used for return value attributes.
379                   ///< Index ~0U is used for function attributes.
380
381   static AttributeWithIndex get(unsigned Idx, Attributes Attrs) {
382     AttributeWithIndex P;
383     P.Index = Idx;
384     P.Attrs = Attrs;
385     return P;
386   }
387 };
388
389 //===----------------------------------------------------------------------===//
390 // AttrListPtr Smart Pointer
391 //===----------------------------------------------------------------------===//
392
393 class AttributeListImpl;
394
395 /// AttrListPtr - This class manages the ref count for the opaque
396 /// AttributeListImpl object and provides accessors for it.
397 class AttrListPtr {
398   /// AttrList - The attributes that we are managing.  This can be null
399   /// to represent the empty attributes list.
400   AttributeListImpl *AttrList;
401 public:
402   AttrListPtr() : AttrList(0) {}
403   AttrListPtr(const AttrListPtr &P);
404   const AttrListPtr &operator=(const AttrListPtr &RHS);
405   ~AttrListPtr();
406
407   //===--------------------------------------------------------------------===//
408   // Attribute List Construction and Mutation
409   //===--------------------------------------------------------------------===//
410
411   /// get - Return a Attributes list with the specified parameters in it.
412   static AttrListPtr get(ArrayRef<AttributeWithIndex> Attrs);
413
414   /// addAttr - Add the specified attribute at the specified index to this
415   /// attribute list.  Since attribute lists are immutable, this
416   /// returns the new list.
417   AttrListPtr addAttr(unsigned Idx, Attributes Attrs) const;
418
419   /// removeAttr - Remove the specified attribute at the specified index from
420   /// this attribute list.  Since attribute lists are immutable, this
421   /// returns the new list.
422   AttrListPtr removeAttr(unsigned Idx, Attributes Attrs) const;
423
424   //===--------------------------------------------------------------------===//
425   // Attribute List Accessors
426   //===--------------------------------------------------------------------===//
427   /// getParamAttributes - The attributes for the specified index are
428   /// returned.
429   Attributes getParamAttributes(unsigned Idx) const {
430     assert (Idx && Idx != ~0U && "Invalid parameter index!");
431     return getAttributes(Idx);
432   }
433
434   /// getRetAttributes - The attributes for the ret value are
435   /// returned.
436   Attributes getRetAttributes() const {
437     return getAttributes(0);
438   }
439
440   /// getFnAttributes - The function attributes are returned.
441   Attributes getFnAttributes() const {
442     return getAttributes(~0U);
443   }
444
445   /// paramHasAttr - Return true if the specified parameter index has the
446   /// specified attribute set.
447   bool paramHasAttr(unsigned Idx, Attributes Attr) const {
448     return getAttributes(Idx).hasAttributes(Attr);
449   }
450
451   /// getParamAlignment - Return the alignment for the specified function
452   /// parameter.
453   unsigned getParamAlignment(unsigned Idx) const {
454     return getAttributes(Idx).getAlignment();
455   }
456
457   /// hasAttrSomewhere - Return true if the specified attribute is set for at
458   /// least one parameter or for the return value.
459   bool hasAttrSomewhere(Attributes Attr) const;
460
461   /// operator==/!= - Provide equality predicates.
462   bool operator==(const AttrListPtr &RHS) const
463   { return AttrList == RHS.AttrList; }
464   bool operator!=(const AttrListPtr &RHS) const
465   { return AttrList != RHS.AttrList; }
466
467   void dump() const;
468
469   //===--------------------------------------------------------------------===//
470   // Attribute List Introspection
471   //===--------------------------------------------------------------------===//
472
473   /// getRawPointer - Return a raw pointer that uniquely identifies this
474   /// attribute list.
475   void *getRawPointer() const {
476     return AttrList;
477   }
478
479   // Attributes are stored as a dense set of slots, where there is one
480   // slot for each argument that has an attribute.  This allows walking over the
481   // dense set instead of walking the sparse list of attributes.
482
483   /// isEmpty - Return true if there are no attributes.
484   ///
485   bool isEmpty() const {
486     return AttrList == 0;
487   }
488
489   /// getNumSlots - Return the number of slots used in this attribute list.
490   /// This is the number of arguments that have an attribute set on them
491   /// (including the function itself).
492   unsigned getNumSlots() const;
493
494   /// getSlot - Return the AttributeWithIndex at the specified slot.  This
495   /// holds a index number plus a set of attributes.
496   const AttributeWithIndex &getSlot(unsigned Slot) const;
497
498 private:
499   explicit AttrListPtr(AttributeListImpl *L);
500
501   /// getAttributes - The attributes for the specified index are
502   /// returned.  Attributes for the result are denoted with Idx = 0.
503   Attributes getAttributes(unsigned Idx) const;
504
505 };
506
507 } // End llvm namespace
508
509 #endif