1 //===--- Token.h - Token interface ------------------------------*- 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 the Token interface.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #ifndef LLVM_CLANG_LEX_TOKEN_H
15 #define LLVM_CLANG_LEX_TOKEN_H
16 
17 #include "clang/Basic/OperatorKinds.h"
18 #include "clang/Basic/SourceLocation.h"
19 #include "clang/Basic/TemplateKinds.h"
20 #include "clang/Basic/TokenKinds.h"
21 #include "llvm/ADT/StringRef.h"
22 #include <cstdlib>
23 
24 namespace clang {
25 
26 class IdentifierInfo;
27 
28 /// Token - This structure provides full information about a lexed token.
29 /// It is not intended to be space efficient, it is intended to return as much
30 /// information as possible about each returned token.  This is expected to be
31 /// compressed into a smaller form if memory footprint is important.
32 ///
33 /// The parser can create a special "annotation token" representing a stream of
34 /// tokens that were parsed and semantically resolved, e.g.: "foo::MyClass<int>"
35 /// can be represented by a single typename annotation token that carries
36 /// information about the SourceRange of the tokens and the type object.
37 class Token {
38   /// The location of the token. This is actually a SourceLocation.
39   unsigned Loc;
40 
41   // Conceptually these next two fields could be in a union.  However, this
42   // causes gcc 4.2 to pessimize LexTokenInternal, a very performance critical
43   // routine. Keeping as separate members with casts until a more beautiful fix
44   // presents itself.
45 
46   /// UintData - This holds either the length of the token text, when
47   /// a normal token, or the end of the SourceRange when an annotation
48   /// token.
49   unsigned UintData;
50 
51   /// PtrData - This is a union of four different pointer types, which depends
52   /// on what type of token this is:
53   ///  Identifiers, keywords, etc:
54   ///    This is an IdentifierInfo*, which contains the uniqued identifier
55   ///    spelling.
56   ///  Literals:  isLiteral() returns true.
57   ///    This is a pointer to the start of the token in a text buffer, which
58   ///    may be dirty (have trigraphs / escaped newlines).
59   ///  Annotations (resolved type names, C++ scopes, etc): isAnnotation().
60   ///    This is a pointer to sema-specific data for the annotation token.
61   ///  Eof:
62   //     This is a pointer to a Decl.
63   ///  Other:
64   ///    This is null.
65   void *PtrData;
66 
67   /// Kind - The actual flavor of token this is.
68   tok::TokenKind Kind;
69 
70   /// Flags - Bits we track about this token, members of the TokenFlags enum.
71   unsigned short Flags;
72 public:
73 
74   // Various flags set per token:
75   enum TokenFlags {
76     StartOfLine   = 0x01,  // At start of line or only after whitespace
77                            // (considering the line after macro expansion).
78     LeadingSpace  = 0x02,  // Whitespace exists before this token (considering
79                            // whitespace after macro expansion).
80     DisableExpand = 0x04,  // This identifier may never be macro expanded.
81     NeedsCleaning = 0x08,  // Contained an escaped newline or trigraph.
82     LeadingEmptyMacro = 0x10, // Empty macro exists before this token.
83     HasUDSuffix = 0x20,    // This string or character literal has a ud-suffix.
84     HasUCN = 0x40,         // This identifier contains a UCN.
85     IgnoredComma = 0x80,   // This comma is not a macro argument separator (MS).
86     StringifiedInMacro = 0x100, // This string or character literal is formed by
87                                 // macro stringizing or charizing operator.
88   };
89 
getKind()90   tok::TokenKind getKind() const { return Kind; }
setKind(tok::TokenKind K)91   void setKind(tok::TokenKind K) { Kind = K; }
92 
93   /// is/isNot - Predicates to check if this token is a specific kind, as in
94   /// "if (Tok.is(tok::l_brace)) {...}".
is(tok::TokenKind K)95   bool is(tok::TokenKind K) const { return Kind == K; }
isNot(tok::TokenKind K)96   bool isNot(tok::TokenKind K) const { return Kind != K; }
isOneOf(tok::TokenKind K1,tok::TokenKind K2)97   bool isOneOf(tok::TokenKind K1, tok::TokenKind K2) const {
98     return is(K1) || is(K2);
99   }
100   template <typename... Ts>
isOneOf(tok::TokenKind K1,tok::TokenKind K2,Ts...Ks)101   bool isOneOf(tok::TokenKind K1, tok::TokenKind K2, Ts... Ks) const {
102     return is(K1) || isOneOf(K2, Ks...);
103   }
104 
105   /// \brief Return true if this is a raw identifier (when lexing
106   /// in raw mode) or a non-keyword identifier (when lexing in non-raw mode).
isAnyIdentifier()107   bool isAnyIdentifier() const {
108     return tok::isAnyIdentifier(getKind());
109   }
110 
111   /// \brief Return true if this is a "literal", like a numeric
112   /// constant, string, etc.
isLiteral()113   bool isLiteral() const {
114     return tok::isLiteral(getKind());
115   }
116 
117   /// \brief Return true if this is any of tok::annot_* kind tokens.
isAnnotation()118   bool isAnnotation() const {
119     return tok::isAnnotation(getKind());
120   }
121 
122   /// \brief Return a source location identifier for the specified
123   /// offset in the current file.
getLocation()124   SourceLocation getLocation() const {
125     return SourceLocation::getFromRawEncoding(Loc);
126   }
getLength()127   unsigned getLength() const {
128     assert(!isAnnotation() && "Annotation tokens have no length field");
129     return UintData;
130   }
131 
setLocation(SourceLocation L)132   void setLocation(SourceLocation L) { Loc = L.getRawEncoding(); }
setLength(unsigned Len)133   void setLength(unsigned Len) {
134     assert(!isAnnotation() && "Annotation tokens have no length field");
135     UintData = Len;
136   }
137 
getAnnotationEndLoc()138   SourceLocation getAnnotationEndLoc() const {
139     assert(isAnnotation() && "Used AnnotEndLocID on non-annotation token");
140     return SourceLocation::getFromRawEncoding(UintData ? UintData : Loc);
141   }
setAnnotationEndLoc(SourceLocation L)142   void setAnnotationEndLoc(SourceLocation L) {
143     assert(isAnnotation() && "Used AnnotEndLocID on non-annotation token");
144     UintData = L.getRawEncoding();
145   }
146 
getLastLoc()147   SourceLocation getLastLoc() const {
148     return isAnnotation() ? getAnnotationEndLoc() : getLocation();
149   }
150 
getEndLoc()151   SourceLocation getEndLoc() const {
152     return isAnnotation() ? getAnnotationEndLoc()
153                           : getLocation().getLocWithOffset(getLength());
154   }
155 
156   /// \brief SourceRange of the group of tokens that this annotation token
157   /// represents.
getAnnotationRange()158   SourceRange getAnnotationRange() const {
159     return SourceRange(getLocation(), getAnnotationEndLoc());
160   }
setAnnotationRange(SourceRange R)161   void setAnnotationRange(SourceRange R) {
162     setLocation(R.getBegin());
163     setAnnotationEndLoc(R.getEnd());
164   }
165 
getName()166   const char *getName() const { return tok::getTokenName(Kind); }
167 
168   /// \brief Reset all flags to cleared.
startToken()169   void startToken() {
170     Kind = tok::unknown;
171     Flags = 0;
172     PtrData = nullptr;
173     UintData = 0;
174     Loc = SourceLocation().getRawEncoding();
175   }
176 
getIdentifierInfo()177   IdentifierInfo *getIdentifierInfo() const {
178     assert(isNot(tok::raw_identifier) &&
179            "getIdentifierInfo() on a tok::raw_identifier token!");
180     assert(!isAnnotation() &&
181            "getIdentifierInfo() on an annotation token!");
182     if (isLiteral()) return nullptr;
183     if (is(tok::eof)) return nullptr;
184     return (IdentifierInfo*) PtrData;
185   }
setIdentifierInfo(IdentifierInfo * II)186   void setIdentifierInfo(IdentifierInfo *II) {
187     PtrData = (void*) II;
188   }
189 
getEofData()190   const void *getEofData() const {
191     assert(is(tok::eof));
192     return reinterpret_cast<const void *>(PtrData);
193   }
setEofData(const void * D)194   void setEofData(const void *D) {
195     assert(is(tok::eof));
196     assert(!PtrData);
197     PtrData = const_cast<void *>(D);
198   }
199 
200   /// getRawIdentifier - For a raw identifier token (i.e., an identifier
201   /// lexed in raw mode), returns a reference to the text substring in the
202   /// buffer if known.
getRawIdentifier()203   StringRef getRawIdentifier() const {
204     assert(is(tok::raw_identifier));
205     return StringRef(reinterpret_cast<const char *>(PtrData), getLength());
206   }
setRawIdentifierData(const char * Ptr)207   void setRawIdentifierData(const char *Ptr) {
208     assert(is(tok::raw_identifier));
209     PtrData = const_cast<char*>(Ptr);
210   }
211 
212   /// getLiteralData - For a literal token (numeric constant, string, etc), this
213   /// returns a pointer to the start of it in the text buffer if known, null
214   /// otherwise.
getLiteralData()215   const char *getLiteralData() const {
216     assert(isLiteral() && "Cannot get literal data of non-literal");
217     return reinterpret_cast<const char*>(PtrData);
218   }
setLiteralData(const char * Ptr)219   void setLiteralData(const char *Ptr) {
220     assert(isLiteral() && "Cannot set literal data of non-literal");
221     PtrData = const_cast<char*>(Ptr);
222   }
223 
getAnnotationValue()224   void *getAnnotationValue() const {
225     assert(isAnnotation() && "Used AnnotVal on non-annotation token");
226     return PtrData;
227   }
setAnnotationValue(void * val)228   void setAnnotationValue(void *val) {
229     assert(isAnnotation() && "Used AnnotVal on non-annotation token");
230     PtrData = val;
231   }
232 
233   /// \brief Set the specified flag.
setFlag(TokenFlags Flag)234   void setFlag(TokenFlags Flag) {
235     Flags |= Flag;
236   }
237 
238   /// \brief Unset the specified flag.
clearFlag(TokenFlags Flag)239   void clearFlag(TokenFlags Flag) {
240     Flags &= ~Flag;
241   }
242 
243   /// \brief Return the internal represtation of the flags.
244   ///
245   /// This is only intended for low-level operations such as writing tokens to
246   /// disk.
getFlags()247   unsigned getFlags() const {
248     return Flags;
249   }
250 
251   /// \brief Set a flag to either true or false.
setFlagValue(TokenFlags Flag,bool Val)252   void setFlagValue(TokenFlags Flag, bool Val) {
253     if (Val)
254       setFlag(Flag);
255     else
256       clearFlag(Flag);
257   }
258 
259   /// isAtStartOfLine - Return true if this token is at the start of a line.
260   ///
isAtStartOfLine()261   bool isAtStartOfLine() const { return (Flags & StartOfLine) ? true : false; }
262 
263   /// \brief Return true if this token has whitespace before it.
264   ///
hasLeadingSpace()265   bool hasLeadingSpace() const { return (Flags & LeadingSpace) ? true : false; }
266 
267   /// \brief Return true if this identifier token should never
268   /// be expanded in the future, due to C99 6.10.3.4p2.
isExpandDisabled()269   bool isExpandDisabled() const {
270     return (Flags & DisableExpand) ? true : false;
271   }
272 
273   /// \brief Return true if we have an ObjC keyword identifier.
274   bool isObjCAtKeyword(tok::ObjCKeywordKind objcKey) const;
275 
276   /// \brief Return the ObjC keyword kind.
277   tok::ObjCKeywordKind getObjCKeywordID() const;
278 
279   /// \brief Return true if this token has trigraphs or escaped newlines in it.
needsCleaning()280   bool needsCleaning() const { return (Flags & NeedsCleaning) ? true : false; }
281 
282   /// \brief Return true if this token has an empty macro before it.
283   ///
hasLeadingEmptyMacro()284   bool hasLeadingEmptyMacro() const {
285     return (Flags & LeadingEmptyMacro) ? true : false;
286   }
287 
288   /// \brief Return true if this token is a string or character literal which
289   /// has a ud-suffix.
hasUDSuffix()290   bool hasUDSuffix() const { return (Flags & HasUDSuffix) ? true : false; }
291 
292   /// Returns true if this token contains a universal character name.
hasUCN()293   bool hasUCN() const { return (Flags & HasUCN) ? true : false; }
294 
295   /// Returns true if this token is formed by macro by stringizing or charizing
296   /// operator.
stringifiedInMacro()297   bool stringifiedInMacro() const {
298     return (Flags & StringifiedInMacro) ? true : false;
299   }
300 };
301 
302 /// \brief Information about the conditional stack (\#if directives)
303 /// currently active.
304 struct PPConditionalInfo {
305   /// \brief Location where the conditional started.
306   SourceLocation IfLoc;
307 
308   /// \brief True if this was contained in a skipping directive, e.g.,
309   /// in a "\#if 0" block.
310   bool WasSkipping;
311 
312   /// \brief True if we have emitted tokens already, and now we're in
313   /// an \#else block or something.  Only useful in Skipping blocks.
314   bool FoundNonSkip;
315 
316   /// \brief True if we've seen a \#else in this block.  If so,
317   /// \#elif/\#else directives are not allowed.
318   bool FoundElse;
319 };
320 
321 }  // end namespace clang
322 
323 namespace llvm {
324   template <>
325   struct isPodLike<clang::Token> { static const bool value = true; };
326 }  // end namespace llvm
327 
328 #endif
329