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