Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 1 | //===- IdentifierTable.cpp - Hash table for identifier lookup -------------===// |
Chris Lattner | 22eb972 | 2006-06-18 05:43:12 +0000 | [diff] [blame] | 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
Chris Lattner | 5b12ab8 | 2007-12-29 19:59:25 +0000 | [diff] [blame] | 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
Chris Lattner | 22eb972 | 2006-06-18 05:43:12 +0000 | [diff] [blame] | 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
Chris Lattner | c79f6fb | 2006-07-04 17:53:21 +0000 | [diff] [blame] | 10 | // This file implements the IdentifierInfo, IdentifierVisitor, and |
Chris Lattner | 91cbf11 | 2006-07-03 04:28:52 +0000 | [diff] [blame] | 11 | // IdentifierTable interfaces. |
Chris Lattner | 22eb972 | 2006-06-18 05:43:12 +0000 | [diff] [blame] | 12 | // |
| 13 | //===----------------------------------------------------------------------===// |
| 14 | |
Benjamin Kramer | f3ca2698 | 2014-05-10 16:31:55 +0000 | [diff] [blame] | 15 | #include "clang/Basic/IdentifierTable.h" |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 16 | #include "clang/Basic/CharInfo.h" |
Chandler Carruth | 5553d0d | 2014-01-07 11:51:46 +0000 | [diff] [blame] | 17 | #include "clang/Basic/LangOptions.h" |
Benjamin Kramer | f3ca2698 | 2014-05-10 16:31:55 +0000 | [diff] [blame] | 18 | #include "clang/Basic/OperatorKinds.h" |
Douglas Gregor | 813a066 | 2015-06-19 18:14:38 +0000 | [diff] [blame] | 19 | #include "clang/Basic/Specifiers.h" |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 20 | #include "clang/Basic/TokenKinds.h" |
| 21 | #include "llvm/ADT/DenseMapInfo.h" |
Chandler Carruth | 3a02247 | 2012-12-04 09:13:33 +0000 | [diff] [blame] | 22 | #include "llvm/ADT/FoldingSet.h" |
Benjamin Kramer | f3ca2698 | 2014-05-10 16:31:55 +0000 | [diff] [blame] | 23 | #include "llvm/ADT/SmallString.h" |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 24 | #include "llvm/ADT/StringMap.h" |
| 25 | #include "llvm/ADT/StringRef.h" |
| 26 | #include "llvm/Support/Allocator.h" |
David Blaikie | 8a40f70 | 2012-01-17 06:56:22 +0000 | [diff] [blame] | 27 | #include "llvm/Support/ErrorHandling.h" |
Chandler Carruth | 3a02247 | 2012-12-04 09:13:33 +0000 | [diff] [blame] | 28 | #include "llvm/Support/raw_ostream.h" |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 29 | #include <cassert> |
Chris Lattner | c25d8a7 | 2009-03-02 22:20:04 +0000 | [diff] [blame] | 30 | #include <cstdio> |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 31 | #include <cstring> |
| 32 | #include <string> |
Ted Kremenek | f25f4a3 | 2007-10-23 22:18:37 +0000 | [diff] [blame] | 33 | |
Chris Lattner | 22eb972 | 2006-06-18 05:43:12 +0000 | [diff] [blame] | 34 | using namespace clang; |
| 35 | |
| 36 | //===----------------------------------------------------------------------===// |
Chris Lattner | c79f6fb | 2006-07-04 17:53:21 +0000 | [diff] [blame] | 37 | // IdentifierInfo Implementation |
Chris Lattner | 22eb972 | 2006-06-18 05:43:12 +0000 | [diff] [blame] | 38 | //===----------------------------------------------------------------------===// |
| 39 | |
Ted Kremenek | 52f73ca | 2009-01-20 23:28:34 +0000 | [diff] [blame] | 40 | IdentifierInfo::IdentifierInfo() { |
Chris Lattner | 3bc804e | 2006-10-28 23:46:24 +0000 | [diff] [blame] | 41 | TokenID = tok::identifier; |
Douglas Gregor | 2ad7ee9 | 2008-11-06 16:32:23 +0000 | [diff] [blame] | 42 | ObjCOrBuiltinID = 0; |
Chris Lattner | a441ca6 | 2007-10-07 07:09:52 +0000 | [diff] [blame] | 43 | HasMacro = false; |
Alexander Kornienko | 1d26c02 | 2012-09-25 17:18:14 +0000 | [diff] [blame] | 44 | HadMacro = false; |
Chris Lattner | 3bc804e | 2006-10-28 23:46:24 +0000 | [diff] [blame] | 45 | IsExtension = false; |
Richard Smith | 31d5184 | 2015-05-14 04:00:59 +0000 | [diff] [blame] | 46 | IsFutureCompatKeyword = false; |
Chris Lattner | 3bc804e | 2006-10-28 23:46:24 +0000 | [diff] [blame] | 47 | IsPoisoned = false; |
Chris Lattner | 5b9f489 | 2006-11-21 17:23:33 +0000 | [diff] [blame] | 48 | IsCPPOperatorKeyword = false; |
Chris Lattner | ad89ec0 | 2009-01-21 07:43:11 +0000 | [diff] [blame] | 49 | NeedsHandleIdentifier = false; |
Sebastian Redl | d44cd6a | 2010-08-18 23:57:06 +0000 | [diff] [blame] | 50 | IsFromAST = false; |
Douglas Gregor | 935bc7a2 | 2011-10-27 09:33:13 +0000 | [diff] [blame] | 51 | ChangedAfterLoad = false; |
Richard Smith | d79514e | 2016-02-05 19:03:40 +0000 | [diff] [blame] | 52 | FEChangedAfterLoad = false; |
Argyrios Kyrtzidis | 3084a61 | 2010-08-11 22:55:12 +0000 | [diff] [blame] | 53 | RevertedTokenID = false; |
Douglas Gregor | 935bc7a2 | 2011-10-27 09:33:13 +0000 | [diff] [blame] | 54 | OutOfDate = false; |
Ted Kremenek | c1e4dd0 | 2012-03-01 22:07:04 +0000 | [diff] [blame] | 55 | IsModulesImport = false; |
Chris Lattner | 3bc804e | 2006-10-28 23:46:24 +0000 | [diff] [blame] | 56 | } |
| 57 | |
Chris Lattner | 91cbf11 | 2006-07-03 04:28:52 +0000 | [diff] [blame] | 58 | //===----------------------------------------------------------------------===// |
Chris Lattner | 22eb972 | 2006-06-18 05:43:12 +0000 | [diff] [blame] | 59 | // IdentifierTable Implementation |
| 60 | //===----------------------------------------------------------------------===// |
| 61 | |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 62 | IdentifierIterator::~IdentifierIterator() = default; |
Douglas Gregor | 57756ea | 2010-10-14 22:11:03 +0000 | [diff] [blame] | 63 | |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 64 | IdentifierInfoLookup::~IdentifierInfoLookup() = default; |
Ted Kremenek | a705b04 | 2009-01-15 18:47:46 +0000 | [diff] [blame] | 65 | |
Douglas Gregor | 57756ea | 2010-10-14 22:11:03 +0000 | [diff] [blame] | 66 | namespace { |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 67 | |
| 68 | /// \brief A simple identifier lookup iterator that represents an |
| 69 | /// empty sequence of identifiers. |
| 70 | class EmptyLookupIterator : public IdentifierIterator |
| 71 | { |
| 72 | public: |
| 73 | StringRef Next() override { return StringRef(); } |
| 74 | }; |
| 75 | |
| 76 | } // namespace |
Douglas Gregor | 57756ea | 2010-10-14 22:11:03 +0000 | [diff] [blame] | 77 | |
Argyrios Kyrtzidis | 9aca3c6 | 2013-04-17 22:10:55 +0000 | [diff] [blame] | 78 | IdentifierIterator *IdentifierInfoLookup::getIdentifiers() { |
Douglas Gregor | 57756ea | 2010-10-14 22:11:03 +0000 | [diff] [blame] | 79 | return new EmptyLookupIterator(); |
| 80 | } |
| 81 | |
Aaron Ballman | d742dc2 | 2018-04-16 21:07:08 +0000 | [diff] [blame^] | 82 | IdentifierTable::IdentifierTable(IdentifierInfoLookup *ExternalLookup) |
| 83 | : HashTable(8192), // Start with space for 8K identifiers. |
| 84 | ExternalLookup(ExternalLookup) {} |
| 85 | |
Ted Kremenek | a705b04 | 2009-01-15 18:47:46 +0000 | [diff] [blame] | 86 | IdentifierTable::IdentifierTable(const LangOptions &LangOpts, |
Aaron Ballman | d742dc2 | 2018-04-16 21:07:08 +0000 | [diff] [blame^] | 87 | IdentifierInfoLookup *ExternalLookup) |
| 88 | : IdentifierTable(ExternalLookup) { |
Chris Lattner | f2e3ac3 | 2006-10-27 03:59:10 +0000 | [diff] [blame] | 89 | // Populate the identifier table with info about keywords for the current |
| 90 | // language. |
Chris Lattner | 25e0d54 | 2006-10-18 06:07:05 +0000 | [diff] [blame] | 91 | AddKeywords(LangOpts); |
Chris Lattner | 91cbf11 | 2006-07-03 04:28:52 +0000 | [diff] [blame] | 92 | } |
Chris Lattner | 22eb972 | 2006-06-18 05:43:12 +0000 | [diff] [blame] | 93 | |
Chris Lattner | 25e0d54 | 2006-10-18 06:07:05 +0000 | [diff] [blame] | 94 | //===----------------------------------------------------------------------===// |
| 95 | // Language Keyword Implementation |
| 96 | //===----------------------------------------------------------------------===// |
| 97 | |
Eli Friedman | 2b680b4 | 2009-04-28 03:13:54 +0000 | [diff] [blame] | 98 | // Constants for TokenKinds.def |
| 99 | namespace { |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 100 | |
Eli Friedman | 2b680b4 | 2009-04-28 03:13:54 +0000 | [diff] [blame] | 101 | enum { |
Dylan Noblesmith | 92c07c2 | 2011-04-09 13:34:05 +0000 | [diff] [blame] | 102 | KEYC99 = 0x1, |
| 103 | KEYCXX = 0x2, |
Richard Smith | 89645bc | 2013-01-02 12:01:23 +0000 | [diff] [blame] | 104 | KEYCXX11 = 0x4, |
Dylan Noblesmith | 92c07c2 | 2011-04-09 13:34:05 +0000 | [diff] [blame] | 105 | KEYGNU = 0x8, |
| 106 | KEYMS = 0x10, |
| 107 | BOOLSUPPORT = 0x20, |
| 108 | KEYALTIVEC = 0x40, |
| 109 | KEYNOCXX = 0x80, |
| 110 | KEYBORLAND = 0x100, |
| 111 | KEYOPENCL = 0x200, |
Benjamin Kramer | e56f393 | 2011-12-23 17:00:35 +0000 | [diff] [blame] | 112 | KEYC11 = 0x400, |
John McCall | 31168b0 | 2011-06-15 23:02:42 +0000 | [diff] [blame] | 113 | KEYARC = 0x800, |
David Majnemer | 28aae9c | 2015-03-18 04:15:23 +0000 | [diff] [blame] | 114 | KEYNOMS18 = 0x01000, |
Anastasia Stulova | b1152f1 | 2015-03-18 12:55:29 +0000 | [diff] [blame] | 115 | KEYNOOPENCL = 0x02000, |
| 116 | WCHARSUPPORT = 0x04000, |
| 117 | HALFSUPPORT = 0x08000, |
Hubert Tong | 6e04f98 | 2015-06-05 01:10:24 +0000 | [diff] [blame] | 118 | KEYCONCEPTS = 0x10000, |
Douglas Gregor | ab209d8 | 2015-07-07 03:58:42 +0000 | [diff] [blame] | 119 | KEYOBJC2 = 0x20000, |
Ulrich Weigand | 3c5038a | 2015-07-30 14:08:36 +0000 | [diff] [blame] | 120 | KEYZVECTOR = 0x40000, |
Richard Smith | eb7927e | 2015-10-22 03:52:15 +0000 | [diff] [blame] | 121 | KEYCOROUTINES = 0x80000, |
Richard Smith | c7bf380 | 2016-07-23 02:32:21 +0000 | [diff] [blame] | 122 | KEYMODULES = 0x100000, |
Richard Smith | 6c74e32 | 2017-08-13 21:32:33 +0000 | [diff] [blame] | 123 | KEYCXX2A = 0x200000, |
| 124 | KEYALLCXX = KEYCXX | KEYCXX11 | KEYCXX2A, |
| 125 | KEYALL = (0x3fffff & ~KEYNOMS18 & |
Anastasia Stulova | b1152f1 | 2015-03-18 12:55:29 +0000 | [diff] [blame] | 126 | ~KEYNOOPENCL) // KEYNOMS18 and KEYNOOPENCL are used to exclude. |
Eli Friedman | 2b680b4 | 2009-04-28 03:13:54 +0000 | [diff] [blame] | 127 | }; |
Serge Pavlov | 77af381 | 2014-10-29 10:59:18 +0000 | [diff] [blame] | 128 | |
| 129 | /// \brief How a keyword is treated in the selected standard. |
| 130 | enum KeywordStatus { |
| 131 | KS_Disabled, // Disabled |
| 132 | KS_Extension, // Is an extension |
| 133 | KS_Enabled, // Enabled |
| 134 | KS_Future // Is a keyword in future standard |
| 135 | }; |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 136 | |
| 137 | } // namespace |
Serge Pavlov | 77af381 | 2014-10-29 10:59:18 +0000 | [diff] [blame] | 138 | |
| 139 | /// \brief Translates flags as specified in TokenKinds.def into keyword status |
| 140 | /// in the given language standard. |
Serge Pavlov | 83cf078 | 2014-12-11 12:18:08 +0000 | [diff] [blame] | 141 | static KeywordStatus getKeywordStatus(const LangOptions &LangOpts, |
Serge Pavlov | 77af381 | 2014-10-29 10:59:18 +0000 | [diff] [blame] | 142 | unsigned Flags) { |
| 143 | if (Flags == KEYALL) return KS_Enabled; |
| 144 | if (LangOpts.CPlusPlus && (Flags & KEYCXX)) return KS_Enabled; |
| 145 | if (LangOpts.CPlusPlus11 && (Flags & KEYCXX11)) return KS_Enabled; |
Richard Smith | 6c74e32 | 2017-08-13 21:32:33 +0000 | [diff] [blame] | 146 | if (LangOpts.CPlusPlus2a && (Flags & KEYCXX2A)) return KS_Enabled; |
Serge Pavlov | 77af381 | 2014-10-29 10:59:18 +0000 | [diff] [blame] | 147 | if (LangOpts.C99 && (Flags & KEYC99)) return KS_Enabled; |
| 148 | if (LangOpts.GNUKeywords && (Flags & KEYGNU)) return KS_Extension; |
| 149 | if (LangOpts.MicrosoftExt && (Flags & KEYMS)) return KS_Extension; |
| 150 | if (LangOpts.Borland && (Flags & KEYBORLAND)) return KS_Extension; |
| 151 | if (LangOpts.Bool && (Flags & BOOLSUPPORT)) return KS_Enabled; |
| 152 | if (LangOpts.Half && (Flags & HALFSUPPORT)) return KS_Enabled; |
| 153 | if (LangOpts.WChar && (Flags & WCHARSUPPORT)) return KS_Enabled; |
| 154 | if (LangOpts.AltiVec && (Flags & KEYALTIVEC)) return KS_Enabled; |
| 155 | if (LangOpts.OpenCL && (Flags & KEYOPENCL)) return KS_Enabled; |
| 156 | if (!LangOpts.CPlusPlus && (Flags & KEYNOCXX)) return KS_Enabled; |
| 157 | if (LangOpts.C11 && (Flags & KEYC11)) return KS_Enabled; |
| 158 | // We treat bridge casts as objective-C keywords so we can warn on them |
| 159 | // in non-arc mode. |
| 160 | if (LangOpts.ObjC2 && (Flags & KEYARC)) return KS_Enabled; |
Douglas Gregor | ab209d8 | 2015-07-07 03:58:42 +0000 | [diff] [blame] | 161 | if (LangOpts.ObjC2 && (Flags & KEYOBJC2)) return KS_Enabled; |
Richard Smith | c7bf380 | 2016-07-23 02:32:21 +0000 | [diff] [blame] | 162 | if (LangOpts.ConceptsTS && (Flags & KEYCONCEPTS)) return KS_Enabled; |
Gor Nishanov | 4ffb434 | 2016-10-02 03:31:58 +0000 | [diff] [blame] | 163 | if (LangOpts.CoroutinesTS && (Flags & KEYCOROUTINES)) return KS_Enabled; |
Richard Smith | c7bf380 | 2016-07-23 02:32:21 +0000 | [diff] [blame] | 164 | if (LangOpts.ModulesTS && (Flags & KEYMODULES)) return KS_Enabled; |
Richard Smith | 6c74e32 | 2017-08-13 21:32:33 +0000 | [diff] [blame] | 165 | if (LangOpts.CPlusPlus && (Flags & KEYALLCXX)) return KS_Future; |
Serge Pavlov | 77af381 | 2014-10-29 10:59:18 +0000 | [diff] [blame] | 166 | return KS_Disabled; |
Eli Friedman | 2b680b4 | 2009-04-28 03:13:54 +0000 | [diff] [blame] | 167 | } |
| 168 | |
Chris Lattner | 25e0d54 | 2006-10-18 06:07:05 +0000 | [diff] [blame] | 169 | /// AddKeyword - This method is used to associate a token ID with specific |
| 170 | /// identifiers because they are language keywords. This causes the lexer to |
| 171 | /// automatically map matching identifiers to specialized token codes. |
Chris Lattner | 0e62c1c | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 172 | static void AddKeyword(StringRef Keyword, |
Eli Friedman | 2b680b4 | 2009-04-28 03:13:54 +0000 | [diff] [blame] | 173 | tok::TokenKind TokenCode, unsigned Flags, |
Chris Lattner | 25e0d54 | 2006-10-18 06:07:05 +0000 | [diff] [blame] | 174 | const LangOptions &LangOpts, IdentifierTable &Table) { |
Serge Pavlov | 83cf078 | 2014-12-11 12:18:08 +0000 | [diff] [blame] | 175 | KeywordStatus AddResult = getKeywordStatus(LangOpts, Flags); |
Richard Smith | 4dd85d6 | 2011-10-11 19:57:52 +0000 | [diff] [blame] | 176 | |
Alp Toker | bfa3934 | 2014-01-14 12:51:41 +0000 | [diff] [blame] | 177 | // Don't add this keyword under MSVCCompat. |
David Majnemer | 28aae9c | 2015-03-18 04:15:23 +0000 | [diff] [blame] | 178 | if (LangOpts.MSVCCompat && (Flags & KEYNOMS18) && |
David Majnemer | b710a93 | 2015-05-11 03:57:49 +0000 | [diff] [blame] | 179 | !LangOpts.isCompatibleWithMSVC(LangOptions::MSVC2015)) |
Anastasia Stulova | b1152f1 | 2015-03-18 12:55:29 +0000 | [diff] [blame] | 180 | return; |
| 181 | |
| 182 | // Don't add this keyword under OpenCL. |
| 183 | if (LangOpts.OpenCL && (Flags & KEYNOOPENCL)) |
| 184 | return; |
| 185 | |
Eli Friedman | 2b680b4 | 2009-04-28 03:13:54 +0000 | [diff] [blame] | 186 | // Don't add this keyword if disabled in this language. |
Serge Pavlov | 77af381 | 2014-10-29 10:59:18 +0000 | [diff] [blame] | 187 | if (AddResult == KS_Disabled) return; |
Eli Friedman | 2b680b4 | 2009-04-28 03:13:54 +0000 | [diff] [blame] | 188 | |
Richard Smith | 4dd85d6 | 2011-10-11 19:57:52 +0000 | [diff] [blame] | 189 | IdentifierInfo &Info = |
Serge Pavlov | 77af381 | 2014-10-29 10:59:18 +0000 | [diff] [blame] | 190 | Table.get(Keyword, AddResult == KS_Future ? tok::identifier : TokenCode); |
| 191 | Info.setIsExtensionToken(AddResult == KS_Extension); |
Richard Smith | 31d5184 | 2015-05-14 04:00:59 +0000 | [diff] [blame] | 192 | Info.setIsFutureCompatKeyword(AddResult == KS_Future); |
Chris Lattner | 25e0d54 | 2006-10-18 06:07:05 +0000 | [diff] [blame] | 193 | } |
| 194 | |
Chris Lattner | 5b9f489 | 2006-11-21 17:23:33 +0000 | [diff] [blame] | 195 | /// AddCXXOperatorKeyword - Register a C++ operator keyword alternative |
| 196 | /// representations. |
Chris Lattner | 0e62c1c | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 197 | static void AddCXXOperatorKeyword(StringRef Keyword, |
Chris Lattner | 5b9f489 | 2006-11-21 17:23:33 +0000 | [diff] [blame] | 198 | tok::TokenKind TokenCode, |
| 199 | IdentifierTable &Table) { |
Argyrios Kyrtzidis | 3084a61 | 2010-08-11 22:55:12 +0000 | [diff] [blame] | 200 | IdentifierInfo &Info = Table.get(Keyword, TokenCode); |
Ted Kremenek | f25f4a3 | 2007-10-23 22:18:37 +0000 | [diff] [blame] | 201 | Info.setIsCPlusPlusOperatorKeyword(); |
Chris Lattner | 5b9f489 | 2006-11-21 17:23:33 +0000 | [diff] [blame] | 202 | } |
| 203 | |
James Dennett | 0d8a3f8 | 2012-06-15 21:27:44 +0000 | [diff] [blame] | 204 | /// AddObjCKeyword - Register an Objective-C \@keyword like "class" "selector" |
| 205 | /// or "property". |
Chris Lattner | 0e62c1c | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 206 | static void AddObjCKeyword(StringRef Name, |
Kovarththanan Rajaratnam | 00682a4 | 2010-03-12 11:27:37 +0000 | [diff] [blame] | 207 | tok::ObjCKeywordKind ObjCID, |
Chris Lattner | 25e0d54 | 2006-10-18 06:07:05 +0000 | [diff] [blame] | 208 | IdentifierTable &Table) { |
Kovarththanan Rajaratnam | 00682a4 | 2010-03-12 11:27:37 +0000 | [diff] [blame] | 209 | Table.get(Name).setObjCKeywordID(ObjCID); |
Chris Lattner | 25e0d54 | 2006-10-18 06:07:05 +0000 | [diff] [blame] | 210 | } |
| 211 | |
| 212 | /// AddKeywords - Add all keywords to the symbol table. |
| 213 | /// |
| 214 | void IdentifierTable::AddKeywords(const LangOptions &LangOpts) { |
Chris Lattner | 25e0d54 | 2006-10-18 06:07:05 +0000 | [diff] [blame] | 215 | // Add keywords and tokens for the current language. |
| 216 | #define KEYWORD(NAME, FLAGS) \ |
Chris Lattner | 0e62c1c | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 217 | AddKeyword(StringRef(#NAME), tok::kw_ ## NAME, \ |
Eli Friedman | 2b680b4 | 2009-04-28 03:13:54 +0000 | [diff] [blame] | 218 | FLAGS, LangOpts, *this); |
| 219 | #define ALIAS(NAME, TOK, FLAGS) \ |
Chris Lattner | 0e62c1c | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 220 | AddKeyword(StringRef(NAME), tok::kw_ ## TOK, \ |
Eli Friedman | 2b680b4 | 2009-04-28 03:13:54 +0000 | [diff] [blame] | 221 | FLAGS, LangOpts, *this); |
Chris Lattner | 5b9f489 | 2006-11-21 17:23:33 +0000 | [diff] [blame] | 222 | #define CXX_KEYWORD_OPERATOR(NAME, ALIAS) \ |
Chris Lattner | 3e7592e | 2006-12-04 07:48:37 +0000 | [diff] [blame] | 223 | if (LangOpts.CXXOperatorNames) \ |
Chris Lattner | 0e62c1c | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 224 | AddCXXOperatorKeyword(StringRef(#NAME), tok::ALIAS, *this); |
Chris Lattner | 25e0d54 | 2006-10-18 06:07:05 +0000 | [diff] [blame] | 225 | #define OBJC1_AT_KEYWORD(NAME) \ |
| 226 | if (LangOpts.ObjC1) \ |
Chris Lattner | 0e62c1c | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 227 | AddObjCKeyword(StringRef(#NAME), tok::objc_##NAME, *this); |
Chris Lattner | 25e0d54 | 2006-10-18 06:07:05 +0000 | [diff] [blame] | 228 | #define OBJC2_AT_KEYWORD(NAME) \ |
| 229 | if (LangOpts.ObjC2) \ |
Chris Lattner | 0e62c1c | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 230 | AddObjCKeyword(StringRef(#NAME), tok::objc_##NAME, *this); |
John McCall | 3943973 | 2011-04-09 22:50:59 +0000 | [diff] [blame] | 231 | #define TESTING_KEYWORD(NAME, FLAGS) |
Chris Lattner | 25e0d54 | 2006-10-18 06:07:05 +0000 | [diff] [blame] | 232 | #include "clang/Basic/TokenKinds.def" |
John McCall | 3943973 | 2011-04-09 22:50:59 +0000 | [diff] [blame] | 233 | |
| 234 | if (LangOpts.ParseUnknownAnytype) |
| 235 | AddKeyword("__unknown_anytype", tok::kw___unknown_anytype, KEYALL, |
| 236 | LangOpts, *this); |
Aaron Ballman | 674cf26 | 2015-05-26 19:44:52 +0000 | [diff] [blame] | 237 | |
Saleem Abdulrasool | d170c4b | 2015-10-04 17:51:05 +0000 | [diff] [blame] | 238 | if (LangOpts.DeclSpecKeyword) |
Aaron Ballman | 674cf26 | 2015-05-26 19:44:52 +0000 | [diff] [blame] | 239 | AddKeyword("__declspec", tok::kw___declspec, KEYALL, LangOpts, *this); |
Aaron Ballman | d742dc2 | 2018-04-16 21:07:08 +0000 | [diff] [blame^] | 240 | |
| 241 | // Add the '_experimental_modules_import' contextual keyword. |
| 242 | get("import").setModulesImport(true); |
Chris Lattner | 25e0d54 | 2006-10-18 06:07:05 +0000 | [diff] [blame] | 243 | } |
| 244 | |
Serge Pavlov | 83cf078 | 2014-12-11 12:18:08 +0000 | [diff] [blame] | 245 | /// \brief Checks if the specified token kind represents a keyword in the |
| 246 | /// specified language. |
| 247 | /// \returns Status of the keyword in the language. |
| 248 | static KeywordStatus getTokenKwStatus(const LangOptions &LangOpts, |
| 249 | tok::TokenKind K) { |
| 250 | switch (K) { |
| 251 | #define KEYWORD(NAME, FLAGS) \ |
| 252 | case tok::kw_##NAME: return getKeywordStatus(LangOpts, FLAGS); |
| 253 | #include "clang/Basic/TokenKinds.def" |
| 254 | default: return KS_Disabled; |
| 255 | } |
| 256 | } |
| 257 | |
| 258 | /// \brief Returns true if the identifier represents a keyword in the |
| 259 | /// specified language. |
Alex Lorenz | f127821 | 2017-04-11 15:01:53 +0000 | [diff] [blame] | 260 | bool IdentifierInfo::isKeyword(const LangOptions &LangOpts) const { |
Serge Pavlov | 83cf078 | 2014-12-11 12:18:08 +0000 | [diff] [blame] | 261 | switch (getTokenKwStatus(LangOpts, getTokenID())) { |
| 262 | case KS_Enabled: |
| 263 | case KS_Extension: |
| 264 | return true; |
| 265 | default: |
| 266 | return false; |
| 267 | } |
| 268 | } |
| 269 | |
Alex Lorenz | f127821 | 2017-04-11 15:01:53 +0000 | [diff] [blame] | 270 | /// \brief Returns true if the identifier represents a C++ keyword in the |
| 271 | /// specified language. |
| 272 | bool IdentifierInfo::isCPlusPlusKeyword(const LangOptions &LangOpts) const { |
| 273 | if (!LangOpts.CPlusPlus || !isKeyword(LangOpts)) |
| 274 | return false; |
| 275 | // This is a C++ keyword if this identifier is not a keyword when checked |
| 276 | // using LangOptions without C++ support. |
| 277 | LangOptions LangOptsNoCPP = LangOpts; |
| 278 | LangOptsNoCPP.CPlusPlus = false; |
| 279 | LangOptsNoCPP.CPlusPlus11 = false; |
Richard Smith | 6c74e32 | 2017-08-13 21:32:33 +0000 | [diff] [blame] | 280 | LangOptsNoCPP.CPlusPlus2a = false; |
Alex Lorenz | f127821 | 2017-04-11 15:01:53 +0000 | [diff] [blame] | 281 | return !isKeyword(LangOptsNoCPP); |
| 282 | } |
| 283 | |
Chris Lattner | ff067ce | 2007-10-07 07:52:34 +0000 | [diff] [blame] | 284 | tok::PPKeywordKind IdentifierInfo::getPPKeywordID() const { |
| 285 | // We use a perfect hash function here involving the length of the keyword, |
| 286 | // the first and third character. For preprocessor ID's there are no |
| 287 | // collisions (if there were, the switch below would complain about duplicate |
| 288 | // case values). Note that this depends on 'if' being null terminated. |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 289 | |
Chris Lattner | ff067ce | 2007-10-07 07:52:34 +0000 | [diff] [blame] | 290 | #define HASH(LEN, FIRST, THIRD) \ |
| 291 | (LEN << 5) + (((FIRST-'a') + (THIRD-'a')) & 31) |
| 292 | #define CASE(LEN, FIRST, THIRD, NAME) \ |
| 293 | case HASH(LEN, FIRST, THIRD): \ |
| 294 | return memcmp(Name, #NAME, LEN) ? tok::pp_not_keyword : tok::pp_ ## NAME |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 295 | |
Chris Lattner | ff067ce | 2007-10-07 07:52:34 +0000 | [diff] [blame] | 296 | unsigned Len = getLength(); |
Chris Lattner | d2b8ce4 | 2007-10-10 20:59:57 +0000 | [diff] [blame] | 297 | if (Len < 2) return tok::pp_not_keyword; |
Daniel Dunbar | 1c0761d | 2009-10-17 18:13:02 +0000 | [diff] [blame] | 298 | const char *Name = getNameStart(); |
Chris Lattner | ff067ce | 2007-10-07 07:52:34 +0000 | [diff] [blame] | 299 | switch (HASH(Len, Name[0], Name[2])) { |
| 300 | default: return tok::pp_not_keyword; |
| 301 | CASE( 2, 'i', '\0', if); |
| 302 | CASE( 4, 'e', 'i', elif); |
| 303 | CASE( 4, 'e', 's', else); |
| 304 | CASE( 4, 'l', 'n', line); |
| 305 | CASE( 4, 's', 'c', sccs); |
| 306 | CASE( 5, 'e', 'd', endif); |
| 307 | CASE( 5, 'e', 'r', error); |
| 308 | CASE( 5, 'i', 'e', ident); |
| 309 | CASE( 5, 'i', 'd', ifdef); |
| 310 | CASE( 5, 'u', 'd', undef); |
| 311 | |
| 312 | CASE( 6, 'a', 's', assert); |
| 313 | CASE( 6, 'd', 'f', define); |
| 314 | CASE( 6, 'i', 'n', ifndef); |
| 315 | CASE( 6, 'i', 'p', import); |
| 316 | CASE( 6, 'p', 'a', pragma); |
Douglas Gregor | 0bf886d | 2012-01-03 18:24:14 +0000 | [diff] [blame] | 317 | |
Chris Lattner | ff067ce | 2007-10-07 07:52:34 +0000 | [diff] [blame] | 318 | CASE( 7, 'd', 'f', defined); |
| 319 | CASE( 7, 'i', 'c', include); |
| 320 | CASE( 7, 'w', 'r', warning); |
| 321 | |
| 322 | CASE( 8, 'u', 'a', unassert); |
| 323 | CASE(12, 'i', 'c', include_next); |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 324 | |
Douglas Gregor | 663b48f | 2012-01-03 19:48:16 +0000 | [diff] [blame] | 325 | CASE(14, '_', 'p', __public_macro); |
| 326 | |
| 327 | CASE(15, '_', 'p', __private_macro); |
| 328 | |
Chris Lattner | 14a7f39 | 2009-04-08 18:24:34 +0000 | [diff] [blame] | 329 | CASE(16, '_', 'i', __include_macros); |
Chris Lattner | ff067ce | 2007-10-07 07:52:34 +0000 | [diff] [blame] | 330 | #undef CASE |
| 331 | #undef HASH |
| 332 | } |
| 333 | } |
Chris Lattner | 25e0d54 | 2006-10-18 06:07:05 +0000 | [diff] [blame] | 334 | |
| 335 | //===----------------------------------------------------------------------===// |
| 336 | // Stats Implementation |
| 337 | //===----------------------------------------------------------------------===// |
| 338 | |
Chris Lattner | 22eb972 | 2006-06-18 05:43:12 +0000 | [diff] [blame] | 339 | /// PrintStats - Print statistics about how well the identifier table is doing |
| 340 | /// at hashing identifiers. |
| 341 | void IdentifierTable::PrintStats() const { |
Chris Lattner | 2b9e19b | 2006-10-29 23:43:13 +0000 | [diff] [blame] | 342 | unsigned NumBuckets = HashTable.getNumBuckets(); |
| 343 | unsigned NumIdentifiers = HashTable.getNumItems(); |
| 344 | unsigned NumEmptyBuckets = NumBuckets-NumIdentifiers; |
Chris Lattner | 22eb972 | 2006-06-18 05:43:12 +0000 | [diff] [blame] | 345 | unsigned AverageIdentifierSize = 0; |
| 346 | unsigned MaxIdentifierLength = 0; |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 347 | |
Chris Lattner | 2b9e19b | 2006-10-29 23:43:13 +0000 | [diff] [blame] | 348 | // TODO: Figure out maximum times an identifier had to probe for -stats. |
Ted Kremenek | 52f73ca | 2009-01-20 23:28:34 +0000 | [diff] [blame] | 349 | for (llvm::StringMap<IdentifierInfo*, llvm::BumpPtrAllocator>::const_iterator |
Chris Lattner | b055f2d | 2007-02-11 08:19:57 +0000 | [diff] [blame] | 350 | I = HashTable.begin(), E = HashTable.end(); I != E; ++I) { |
| 351 | unsigned IdLen = I->getKeyLength(); |
| 352 | AverageIdentifierSize += IdLen; |
| 353 | if (MaxIdentifierLength < IdLen) |
| 354 | MaxIdentifierLength = IdLen; |
| 355 | } |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 356 | |
Chris Lattner | 23b7eb6 | 2007-06-15 23:05:46 +0000 | [diff] [blame] | 357 | fprintf(stderr, "\n*** Identifier Table Stats:\n"); |
| 358 | fprintf(stderr, "# Identifiers: %d\n", NumIdentifiers); |
| 359 | fprintf(stderr, "# Empty Buckets: %d\n", NumEmptyBuckets); |
| 360 | fprintf(stderr, "Hash density (#identifiers per bucket): %f\n", |
| 361 | NumIdentifiers/(double)NumBuckets); |
| 362 | fprintf(stderr, "Ave identifier length: %f\n", |
| 363 | (AverageIdentifierSize/(double)NumIdentifiers)); |
| 364 | fprintf(stderr, "Max identifier length: %d\n", MaxIdentifierLength); |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 365 | |
Chris Lattner | 22eb972 | 2006-06-18 05:43:12 +0000 | [diff] [blame] | 366 | // Compute statistics about the memory allocated for identifiers. |
Chris Lattner | 2b9e19b | 2006-10-29 23:43:13 +0000 | [diff] [blame] | 367 | HashTable.getAllocator().PrintStats(); |
Chris Lattner | 22eb972 | 2006-06-18 05:43:12 +0000 | [diff] [blame] | 368 | } |
Steve Naroff | f73590d | 2007-09-27 14:38:14 +0000 | [diff] [blame] | 369 | |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 370 | //===----------------------------------------------------------------------===// |
| 371 | // SelectorTable Implementation |
| 372 | //===----------------------------------------------------------------------===// |
| 373 | |
Chris Lattner | dadc762 | 2007-10-05 20:15:24 +0000 | [diff] [blame] | 374 | unsigned llvm::DenseMapInfo<clang::Selector>::getHashValue(clang::Selector S) { |
| 375 | return DenseMapInfo<void*>::getHashValue(S.getAsOpaquePtr()); |
| 376 | } |
| 377 | |
Douglas Gregor | 77324f3 | 2008-11-17 14:58:09 +0000 | [diff] [blame] | 378 | namespace clang { |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 379 | |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 380 | /// MultiKeywordSelector - One of these variable length records is kept for each |
| 381 | /// selector containing more than one keyword. We use a folding set |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 382 | /// to unique aggregate names (keyword selectors in ObjC parlance). Access to |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 383 | /// this class is provided strictly through Selector. |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 384 | class MultiKeywordSelector |
Douglas Gregor | 77324f3 | 2008-11-17 14:58:09 +0000 | [diff] [blame] | 385 | : public DeclarationNameExtra, public llvm::FoldingSetNode { |
Douglas Gregor | 77324f3 | 2008-11-17 14:58:09 +0000 | [diff] [blame] | 386 | MultiKeywordSelector(unsigned nKeys) { |
| 387 | ExtraKindOrNumArgs = NUM_EXTRA_KINDS + nKeys; |
| 388 | } |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 389 | |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 390 | public: |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 391 | // Constructor for keyword selectors. |
| 392 | MultiKeywordSelector(unsigned nKeys, IdentifierInfo **IIV) { |
| 393 | assert((nKeys > 1) && "not a multi-keyword selector"); |
Douglas Gregor | 77324f3 | 2008-11-17 14:58:09 +0000 | [diff] [blame] | 394 | ExtraKindOrNumArgs = NUM_EXTRA_KINDS + nKeys; |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 395 | |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 396 | // Fill in the trailing keyword array. |
| 397 | IdentifierInfo **KeyInfo = reinterpret_cast<IdentifierInfo **>(this+1); |
| 398 | for (unsigned i = 0; i != nKeys; ++i) |
| 399 | KeyInfo[i] = IIV[i]; |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 400 | } |
| 401 | |
Chris Lattner | f7f34d0 | 2007-10-07 01:33:16 +0000 | [diff] [blame] | 402 | // getName - Derive the full selector name and return it. |
| 403 | std::string getName() const; |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 404 | |
Douglas Gregor | 77324f3 | 2008-11-17 14:58:09 +0000 | [diff] [blame] | 405 | unsigned getNumArgs() const { return ExtraKindOrNumArgs - NUM_EXTRA_KINDS; } |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 406 | |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 407 | using keyword_iterator = IdentifierInfo *const *; |
| 408 | |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 409 | keyword_iterator keyword_begin() const { |
| 410 | return reinterpret_cast<keyword_iterator>(this+1); |
| 411 | } |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 412 | |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 413 | keyword_iterator keyword_end() const { |
| 414 | return keyword_begin()+getNumArgs(); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 415 | } |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 416 | |
Chris Lattner | f7f34d0 | 2007-10-07 01:33:16 +0000 | [diff] [blame] | 417 | IdentifierInfo *getIdentifierInfoForSlot(unsigned i) const { |
Douglas Gregor | 77324f3 | 2008-11-17 14:58:09 +0000 | [diff] [blame] | 418 | assert(i < getNumArgs() && "getIdentifierInfoForSlot(): illegal index"); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 419 | return keyword_begin()[i]; |
| 420 | } |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 421 | |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 422 | static void Profile(llvm::FoldingSetNodeID &ID, |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 423 | keyword_iterator ArgTys, unsigned NumArgs) { |
| 424 | ID.AddInteger(NumArgs); |
Chris Lattner | f7f34d0 | 2007-10-07 01:33:16 +0000 | [diff] [blame] | 425 | for (unsigned i = 0; i != NumArgs; ++i) |
| 426 | ID.AddPointer(ArgTys[i]); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 427 | } |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 428 | |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 429 | void Profile(llvm::FoldingSetNodeID &ID) { |
Douglas Gregor | 77324f3 | 2008-11-17 14:58:09 +0000 | [diff] [blame] | 430 | Profile(ID, keyword_begin(), getNumArgs()); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 431 | } |
| 432 | }; |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 433 | |
| 434 | } // namespace clang. |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 435 | |
| 436 | unsigned Selector::getNumArgs() const { |
| 437 | unsigned IIF = getIdentifierInfoFlag(); |
Douglas Gregor | 93a586f | 2012-05-04 18:24:37 +0000 | [diff] [blame] | 438 | if (IIF <= ZeroArg) |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 439 | return 0; |
| 440 | if (IIF == OneArg) |
| 441 | return 1; |
Douglas Gregor | 93a586f | 2012-05-04 18:24:37 +0000 | [diff] [blame] | 442 | // We point to a MultiKeywordSelector. |
| 443 | MultiKeywordSelector *SI = getMultiKeywordSelector(); |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 444 | return SI->getNumArgs(); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 445 | } |
| 446 | |
Chris Lattner | f7f34d0 | 2007-10-07 01:33:16 +0000 | [diff] [blame] | 447 | IdentifierInfo *Selector::getIdentifierInfoForSlot(unsigned argIndex) const { |
Douglas Gregor | 93a586f | 2012-05-04 18:24:37 +0000 | [diff] [blame] | 448 | if (getIdentifierInfoFlag() < MultiArg) { |
Chris Lattner | f7f34d0 | 2007-10-07 01:33:16 +0000 | [diff] [blame] | 449 | assert(argIndex == 0 && "illegal keyword index"); |
Douglas Gregor | 70091b8 | 2009-04-26 22:20:50 +0000 | [diff] [blame] | 450 | return getAsIdentifierInfo(); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 451 | } |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 452 | |
Douglas Gregor | 93a586f | 2012-05-04 18:24:37 +0000 | [diff] [blame] | 453 | // We point to a MultiKeywordSelector. |
| 454 | MultiKeywordSelector *SI = getMultiKeywordSelector(); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 455 | return SI->getIdentifierInfoForSlot(argIndex); |
| 456 | } |
| 457 | |
Chris Lattner | 0e62c1c | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 458 | StringRef Selector::getNameForSlot(unsigned int argIndex) const { |
Douglas Gregor | af2a6ae | 2011-02-18 22:29:55 +0000 | [diff] [blame] | 459 | IdentifierInfo *II = getIdentifierInfoForSlot(argIndex); |
Chris Lattner | 0e62c1c | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 460 | return II? II->getName() : StringRef(); |
Douglas Gregor | af2a6ae | 2011-02-18 22:29:55 +0000 | [diff] [blame] | 461 | } |
| 462 | |
Chris Lattner | f7f34d0 | 2007-10-07 01:33:16 +0000 | [diff] [blame] | 463 | std::string MultiKeywordSelector::getName() const { |
Dylan Noblesmith | 2c1dd27 | 2012-02-05 02:13:05 +0000 | [diff] [blame] | 464 | SmallString<256> Str; |
Daniel Dunbar | 1c0761d | 2009-10-17 18:13:02 +0000 | [diff] [blame] | 465 | llvm::raw_svector_ostream OS(Str); |
Chris Lattner | f7f34d0 | 2007-10-07 01:33:16 +0000 | [diff] [blame] | 466 | for (keyword_iterator I = keyword_begin(), E = keyword_end(); I != E; ++I) { |
| 467 | if (*I) |
Daniel Dunbar | 07d0785 | 2009-10-18 21:17:35 +0000 | [diff] [blame] | 468 | OS << (*I)->getName(); |
Daniel Dunbar | 1c0761d | 2009-10-17 18:13:02 +0000 | [diff] [blame] | 469 | OS << ':'; |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 470 | } |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 471 | |
Daniel Dunbar | 1c0761d | 2009-10-17 18:13:02 +0000 | [diff] [blame] | 472 | return OS.str(); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 473 | } |
| 474 | |
Chris Lattner | e4b9569 | 2008-11-24 03:33:13 +0000 | [diff] [blame] | 475 | std::string Selector::getAsString() const { |
Douglas Gregor | 70091b8 | 2009-04-26 22:20:50 +0000 | [diff] [blame] | 476 | if (InfoPtr == 0) |
| 477 | return "<null selector>"; |
| 478 | |
Douglas Gregor | 93a586f | 2012-05-04 18:24:37 +0000 | [diff] [blame] | 479 | if (getIdentifierInfoFlag() < MultiArg) { |
Ted Kremenek | 5f080b4 | 2009-03-06 23:36:28 +0000 | [diff] [blame] | 480 | IdentifierInfo *II = getAsIdentifierInfo(); |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 481 | |
Chandler Carruth | 5966677 | 2016-11-04 06:32:57 +0000 | [diff] [blame] | 482 | if (getNumArgs() == 0) { |
| 483 | assert(II && "If the number of arguments is 0 then II is guaranteed to " |
| 484 | "not be null."); |
Daniel Dunbar | 07d0785 | 2009-10-18 21:17:35 +0000 | [diff] [blame] | 485 | return II->getName(); |
Chandler Carruth | 5966677 | 2016-11-04 06:32:57 +0000 | [diff] [blame] | 486 | } |
Ted Kremenek | 5f080b4 | 2009-03-06 23:36:28 +0000 | [diff] [blame] | 487 | |
Daniel Dunbar | 1c0761d | 2009-10-17 18:13:02 +0000 | [diff] [blame] | 488 | if (!II) |
| 489 | return ":"; |
| 490 | |
Daniel Dunbar | 07d0785 | 2009-10-18 21:17:35 +0000 | [diff] [blame] | 491 | return II->getName().str() + ":"; |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 492 | } |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 493 | |
Douglas Gregor | 93a586f | 2012-05-04 18:24:37 +0000 | [diff] [blame] | 494 | // We have a multiple keyword selector. |
| 495 | return getMultiKeywordSelector()->getName(); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 496 | } |
| 497 | |
Aaron Ballman | b190f97 | 2014-01-03 17:59:55 +0000 | [diff] [blame] | 498 | void Selector::print(llvm::raw_ostream &OS) const { |
| 499 | OS << getAsString(); |
| 500 | } |
| 501 | |
John McCall | b452625 | 2011-03-02 01:50:55 +0000 | [diff] [blame] | 502 | /// Interpreting the given string using the normal CamelCase |
| 503 | /// conventions, determine whether the given string starts with the |
| 504 | /// given "word", which is assumed to end in a lowercase letter. |
Chris Lattner | 0e62c1c | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 505 | static bool startsWithWord(StringRef name, StringRef word) { |
John McCall | b452625 | 2011-03-02 01:50:55 +0000 | [diff] [blame] | 506 | if (name.size() < word.size()) return false; |
Jordan Rose | a7d0384 | 2013-02-08 22:30:41 +0000 | [diff] [blame] | 507 | return ((name.size() == word.size() || !isLowercase(name[word.size()])) && |
| 508 | name.startswith(word)); |
John McCall | b452625 | 2011-03-02 01:50:55 +0000 | [diff] [blame] | 509 | } |
| 510 | |
| 511 | ObjCMethodFamily Selector::getMethodFamilyImpl(Selector sel) { |
| 512 | IdentifierInfo *first = sel.getIdentifierInfoForSlot(0); |
| 513 | if (!first) return OMF_None; |
| 514 | |
Chris Lattner | 0e62c1c | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 515 | StringRef name = first->getName(); |
John McCall | b452625 | 2011-03-02 01:50:55 +0000 | [diff] [blame] | 516 | if (sel.isUnarySelector()) { |
| 517 | if (name == "autorelease") return OMF_autorelease; |
| 518 | if (name == "dealloc") return OMF_dealloc; |
Nico Weber | 1fb8266 | 2011-08-28 22:35:17 +0000 | [diff] [blame] | 519 | if (name == "finalize") return OMF_finalize; |
John McCall | b452625 | 2011-03-02 01:50:55 +0000 | [diff] [blame] | 520 | if (name == "release") return OMF_release; |
| 521 | if (name == "retain") return OMF_retain; |
| 522 | if (name == "retainCount") return OMF_retainCount; |
Douglas Gregor | 3382372 | 2011-06-11 01:09:30 +0000 | [diff] [blame] | 523 | if (name == "self") return OMF_self; |
Fariborz Jahanian | 78e9deb | 2014-08-22 16:57:26 +0000 | [diff] [blame] | 524 | if (name == "initialize") return OMF_initialize; |
John McCall | b452625 | 2011-03-02 01:50:55 +0000 | [diff] [blame] | 525 | } |
Alex Lorenz | 0e23c61 | 2017-03-06 15:58:34 +0000 | [diff] [blame] | 526 | |
| 527 | if (name == "performSelector" || name == "performSelectorInBackground" || |
| 528 | name == "performSelectorOnMainThread") |
| 529 | return OMF_performSelector; |
John McCall | b452625 | 2011-03-02 01:50:55 +0000 | [diff] [blame] | 530 | |
| 531 | // The other method families may begin with a prefix of underscores. |
| 532 | while (!name.empty() && name.front() == '_') |
| 533 | name = name.substr(1); |
| 534 | |
| 535 | if (name.empty()) return OMF_None; |
| 536 | switch (name.front()) { |
| 537 | case 'a': |
| 538 | if (startsWithWord(name, "alloc")) return OMF_alloc; |
| 539 | break; |
| 540 | case 'c': |
| 541 | if (startsWithWord(name, "copy")) return OMF_copy; |
| 542 | break; |
| 543 | case 'i': |
| 544 | if (startsWithWord(name, "init")) return OMF_init; |
| 545 | break; |
| 546 | case 'm': |
| 547 | if (startsWithWord(name, "mutableCopy")) return OMF_mutableCopy; |
| 548 | break; |
| 549 | case 'n': |
| 550 | if (startsWithWord(name, "new")) return OMF_new; |
| 551 | break; |
| 552 | default: |
| 553 | break; |
| 554 | } |
| 555 | |
| 556 | return OMF_None; |
| 557 | } |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 558 | |
Fariborz Jahanian | 7122135 | 2013-07-23 22:42:28 +0000 | [diff] [blame] | 559 | ObjCInstanceTypeFamily Selector::getInstTypeMethodFamily(Selector sel) { |
Fariborz Jahanian | 4f3a64f | 2013-07-23 19:31:17 +0000 | [diff] [blame] | 560 | IdentifierInfo *first = sel.getIdentifierInfoForSlot(0); |
| 561 | if (!first) return OIT_None; |
| 562 | |
| 563 | StringRef name = first->getName(); |
| 564 | |
| 565 | if (name.empty()) return OIT_None; |
| 566 | switch (name.front()) { |
| 567 | case 'a': |
Fariborz Jahanian | 4ccdc73 | 2013-08-29 16:22:26 +0000 | [diff] [blame] | 568 | if (startsWithWord(name, "array")) return OIT_Array; |
Fariborz Jahanian | 4f3a64f | 2013-07-23 19:31:17 +0000 | [diff] [blame] | 569 | break; |
| 570 | case 'd': |
Fariborz Jahanian | 7c87b43 | 2013-10-10 18:23:13 +0000 | [diff] [blame] | 571 | if (startsWithWord(name, "default")) return OIT_ReturnsSelf; |
Fariborz Jahanian | 4f3a64f | 2013-07-23 19:31:17 +0000 | [diff] [blame] | 572 | if (startsWithWord(name, "dictionary")) return OIT_Dictionary; |
| 573 | break; |
Fariborz Jahanian | 9275c68 | 2013-08-02 20:54:18 +0000 | [diff] [blame] | 574 | case 's': |
Fariborz Jahanian | 7c87b43 | 2013-10-10 18:23:13 +0000 | [diff] [blame] | 575 | if (startsWithWord(name, "shared")) return OIT_ReturnsSelf; |
| 576 | if (startsWithWord(name, "standard")) return OIT_Singleton; |
Galina Kistanova | ddcd281 | 2017-06-03 06:40:10 +0000 | [diff] [blame] | 577 | break; |
Fariborz Jahanian | 1c900bc | 2013-09-18 20:35:47 +0000 | [diff] [blame] | 578 | case 'i': |
| 579 | if (startsWithWord(name, "init")) return OIT_Init; |
Fariborz Jahanian | 4f3a64f | 2013-07-23 19:31:17 +0000 | [diff] [blame] | 580 | default: |
| 581 | break; |
| 582 | } |
| 583 | return OIT_None; |
| 584 | } |
| 585 | |
Fariborz Jahanian | 6485fe4 | 2014-09-09 23:10:54 +0000 | [diff] [blame] | 586 | ObjCStringFormatFamily Selector::getStringFormatFamilyImpl(Selector sel) { |
| 587 | IdentifierInfo *first = sel.getIdentifierInfoForSlot(0); |
| 588 | if (!first) return SFF_None; |
| 589 | |
| 590 | StringRef name = first->getName(); |
| 591 | |
| 592 | switch (name.front()) { |
| 593 | case 'a': |
| 594 | if (name == "appendFormat") return SFF_NSString; |
| 595 | break; |
| 596 | |
| 597 | case 'i': |
| 598 | if (name == "initWithFormat") return SFF_NSString; |
| 599 | break; |
| 600 | |
| 601 | case 'l': |
| 602 | if (name == "localizedStringWithFormat") return SFF_NSString; |
| 603 | break; |
| 604 | |
| 605 | case 's': |
| 606 | if (name == "stringByAppendingFormat" || |
| 607 | name == "stringWithFormat") return SFF_NSString; |
| 608 | break; |
| 609 | } |
| 610 | return SFF_None; |
| 611 | } |
| 612 | |
Chris Lattner | 1a84994 | 2009-03-04 05:35:38 +0000 | [diff] [blame] | 613 | namespace { |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 614 | |
| 615 | struct SelectorTableImpl { |
| 616 | llvm::FoldingSet<MultiKeywordSelector> Table; |
| 617 | llvm::BumpPtrAllocator Allocator; |
| 618 | }; |
| 619 | |
| 620 | } // namespace |
Chris Lattner | 1a84994 | 2009-03-04 05:35:38 +0000 | [diff] [blame] | 621 | |
| 622 | static SelectorTableImpl &getSelectorTableImpl(void *P) { |
| 623 | return *static_cast<SelectorTableImpl*>(P); |
| 624 | } |
| 625 | |
Adrian Prantl | 6e77c96 | 2013-06-10 21:36:55 +0000 | [diff] [blame] | 626 | SmallString<64> |
Adrian Prantl | a4ce906 | 2013-06-07 22:29:12 +0000 | [diff] [blame] | 627 | SelectorTable::constructSetterName(StringRef Name) { |
Adrian Prantl | 6e77c96 | 2013-06-10 21:36:55 +0000 | [diff] [blame] | 628 | SmallString<64> SetterName("set"); |
| 629 | SetterName += Name; |
| 630 | SetterName[3] = toUppercase(SetterName[3]); |
| 631 | return SetterName; |
Adrian Prantl | a4ce906 | 2013-06-07 22:29:12 +0000 | [diff] [blame] | 632 | } |
| 633 | |
Adrian Prantl | 6e77c96 | 2013-06-10 21:36:55 +0000 | [diff] [blame] | 634 | Selector |
Adrian Prantl | a4ce906 | 2013-06-07 22:29:12 +0000 | [diff] [blame] | 635 | SelectorTable::constructSetterSelector(IdentifierTable &Idents, |
| 636 | SelectorTable &SelTable, |
| 637 | const IdentifierInfo *Name) { |
| 638 | IdentifierInfo *SetterName = |
| 639 | &Idents.get(constructSetterName(Name->getName())); |
Benjamin Kramer | 4903802 | 2012-02-04 13:45:25 +0000 | [diff] [blame] | 640 | return SelTable.getUnarySelector(SetterName); |
| 641 | } |
| 642 | |
Ted Kremenek | 1c2239e | 2011-04-18 22:47:04 +0000 | [diff] [blame] | 643 | size_t SelectorTable::getTotalMemory() const { |
| 644 | SelectorTableImpl &SelTabImpl = getSelectorTableImpl(Impl); |
| 645 | return SelTabImpl.Allocator.getTotalMemory(); |
| 646 | } |
Chris Lattner | 1a84994 | 2009-03-04 05:35:38 +0000 | [diff] [blame] | 647 | |
Chris Lattner | 5700fab | 2007-10-07 02:00:24 +0000 | [diff] [blame] | 648 | Selector SelectorTable::getSelector(unsigned nKeys, IdentifierInfo **IIV) { |
| 649 | if (nKeys < 2) |
| 650 | return Selector(IIV[0], nKeys); |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 651 | |
Chris Lattner | 1a84994 | 2009-03-04 05:35:38 +0000 | [diff] [blame] | 652 | SelectorTableImpl &SelTabImpl = getSelectorTableImpl(Impl); |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 653 | |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 654 | // Unique selector, to guarantee there is one per name. |
| 655 | llvm::FoldingSetNodeID ID; |
| 656 | MultiKeywordSelector::Profile(ID, IIV, nKeys); |
| 657 | |
Craig Topper | f1186c5 | 2014-05-08 06:41:40 +0000 | [diff] [blame] | 658 | void *InsertPos = nullptr; |
Chris Lattner | 1a84994 | 2009-03-04 05:35:38 +0000 | [diff] [blame] | 659 | if (MultiKeywordSelector *SI = |
| 660 | SelTabImpl.Table.FindNodeOrInsertPos(ID, InsertPos)) |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 661 | return Selector(SI); |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 662 | |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 663 | // MultiKeywordSelector objects are not allocated with new because they have a |
| 664 | // variable size array (for parameter types) at the end of them. |
Chris Lattner | 1a84994 | 2009-03-04 05:35:38 +0000 | [diff] [blame] | 665 | unsigned Size = sizeof(MultiKeywordSelector) + nKeys*sizeof(IdentifierInfo *); |
| 666 | MultiKeywordSelector *SI = |
Benjamin Kramer | c3f8925 | 2016-10-20 14:27:22 +0000 | [diff] [blame] | 667 | (MultiKeywordSelector *)SelTabImpl.Allocator.Allocate( |
| 668 | Size, alignof(MultiKeywordSelector)); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 669 | new (SI) MultiKeywordSelector(nKeys, IIV); |
Chris Lattner | 1a84994 | 2009-03-04 05:35:38 +0000 | [diff] [blame] | 670 | SelTabImpl.Table.InsertNode(SI, InsertPos); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 671 | return Selector(SI); |
| 672 | } |
| 673 | |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 674 | SelectorTable::SelectorTable() { |
Chris Lattner | 1a84994 | 2009-03-04 05:35:38 +0000 | [diff] [blame] | 675 | Impl = new SelectorTableImpl(); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 676 | } |
| 677 | |
| 678 | SelectorTable::~SelectorTable() { |
Chris Lattner | 1a84994 | 2009-03-04 05:35:38 +0000 | [diff] [blame] | 679 | delete &getSelectorTableImpl(Impl); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 680 | } |
| 681 | |
Douglas Gregor | 71395fa | 2009-11-04 00:56:37 +0000 | [diff] [blame] | 682 | const char *clang::getOperatorSpelling(OverloadedOperatorKind Operator) { |
| 683 | switch (Operator) { |
| 684 | case OO_None: |
| 685 | case NUM_OVERLOADED_OPERATORS: |
Craig Topper | f1186c5 | 2014-05-08 06:41:40 +0000 | [diff] [blame] | 686 | return nullptr; |
Kovarththanan Rajaratnam | 00682a4 | 2010-03-12 11:27:37 +0000 | [diff] [blame] | 687 | |
Douglas Gregor | 71395fa | 2009-11-04 00:56:37 +0000 | [diff] [blame] | 688 | #define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \ |
| 689 | case OO_##Name: return Spelling; |
| 690 | #include "clang/Basic/OperatorKinds.def" |
| 691 | } |
Kovarththanan Rajaratnam | 00682a4 | 2010-03-12 11:27:37 +0000 | [diff] [blame] | 692 | |
David Blaikie | 8a40f70 | 2012-01-17 06:56:22 +0000 | [diff] [blame] | 693 | llvm_unreachable("Invalid OverloadedOperatorKind!"); |
Douglas Gregor | 71395fa | 2009-11-04 00:56:37 +0000 | [diff] [blame] | 694 | } |
Douglas Gregor | 813a066 | 2015-06-19 18:14:38 +0000 | [diff] [blame] | 695 | |
Douglas Gregor | aea7afd | 2015-06-24 22:02:08 +0000 | [diff] [blame] | 696 | StringRef clang::getNullabilitySpelling(NullabilityKind kind, |
| 697 | bool isContextSensitive) { |
Douglas Gregor | 813a066 | 2015-06-19 18:14:38 +0000 | [diff] [blame] | 698 | switch (kind) { |
| 699 | case NullabilityKind::NonNull: |
Douglas Gregor | aea7afd | 2015-06-24 22:02:08 +0000 | [diff] [blame] | 700 | return isContextSensitive ? "nonnull" : "_Nonnull"; |
Douglas Gregor | 813a066 | 2015-06-19 18:14:38 +0000 | [diff] [blame] | 701 | |
| 702 | case NullabilityKind::Nullable: |
Douglas Gregor | aea7afd | 2015-06-24 22:02:08 +0000 | [diff] [blame] | 703 | return isContextSensitive ? "nullable" : "_Nullable"; |
Douglas Gregor | 813a066 | 2015-06-19 18:14:38 +0000 | [diff] [blame] | 704 | |
| 705 | case NullabilityKind::Unspecified: |
Douglas Gregor | aea7afd | 2015-06-24 22:02:08 +0000 | [diff] [blame] | 706 | return isContextSensitive ? "null_unspecified" : "_Null_unspecified"; |
Douglas Gregor | 813a066 | 2015-06-19 18:14:38 +0000 | [diff] [blame] | 707 | } |
Nico Weber | 708aabd | 2015-06-20 00:06:30 +0000 | [diff] [blame] | 708 | llvm_unreachable("Unknown nullability kind."); |
Douglas Gregor | 813a066 | 2015-06-19 18:14:38 +0000 | [diff] [blame] | 709 | } |