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