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 | |
Ted Kremenek | a705b04 | 2009-01-15 18:47:46 +0000 | [diff] [blame] | 82 | IdentifierTable::IdentifierTable(const LangOptions &LangOpts, |
| 83 | IdentifierInfoLookup* externalLookup) |
| 84 | : HashTable(8192), // Start with space for 8K identifiers. |
| 85 | ExternalLookup(externalLookup) { |
Chris Lattner | f2e3ac3 | 2006-10-27 03:59:10 +0000 | [diff] [blame] | 86 | // Populate the identifier table with info about keywords for the current |
| 87 | // language. |
Chris Lattner | 25e0d54 | 2006-10-18 06:07:05 +0000 | [diff] [blame] | 88 | AddKeywords(LangOpts); |
Ted Kremenek | 9466603 | 2012-03-01 22:53:32 +0000 | [diff] [blame] | 89 | |
| 90 | // Add the '_experimental_modules_import' contextual keyword. |
Douglas Gregor | c50d492 | 2012-12-11 22:11:52 +0000 | [diff] [blame] | 91 | get("import").setModulesImport(true); |
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); |
Chris Lattner | 25e0d54 | 2006-10-18 06:07:05 +0000 | [diff] [blame] | 240 | } |
| 241 | |
Serge Pavlov | 83cf078 | 2014-12-11 12:18:08 +0000 | [diff] [blame] | 242 | /// \brief Checks if the specified token kind represents a keyword in the |
| 243 | /// specified language. |
| 244 | /// \returns Status of the keyword in the language. |
| 245 | static KeywordStatus getTokenKwStatus(const LangOptions &LangOpts, |
| 246 | tok::TokenKind K) { |
| 247 | switch (K) { |
| 248 | #define KEYWORD(NAME, FLAGS) \ |
| 249 | case tok::kw_##NAME: return getKeywordStatus(LangOpts, FLAGS); |
| 250 | #include "clang/Basic/TokenKinds.def" |
| 251 | default: return KS_Disabled; |
| 252 | } |
| 253 | } |
| 254 | |
| 255 | /// \brief Returns true if the identifier represents a keyword in the |
| 256 | /// specified language. |
Alex Lorenz | f127821 | 2017-04-11 15:01:53 +0000 | [diff] [blame] | 257 | bool IdentifierInfo::isKeyword(const LangOptions &LangOpts) const { |
Serge Pavlov | 83cf078 | 2014-12-11 12:18:08 +0000 | [diff] [blame] | 258 | switch (getTokenKwStatus(LangOpts, getTokenID())) { |
| 259 | case KS_Enabled: |
| 260 | case KS_Extension: |
| 261 | return true; |
| 262 | default: |
| 263 | return false; |
| 264 | } |
| 265 | } |
| 266 | |
Alex Lorenz | f127821 | 2017-04-11 15:01:53 +0000 | [diff] [blame] | 267 | /// \brief Returns true if the identifier represents a C++ keyword in the |
| 268 | /// specified language. |
| 269 | bool IdentifierInfo::isCPlusPlusKeyword(const LangOptions &LangOpts) const { |
| 270 | if (!LangOpts.CPlusPlus || !isKeyword(LangOpts)) |
| 271 | return false; |
| 272 | // This is a C++ keyword if this identifier is not a keyword when checked |
| 273 | // using LangOptions without C++ support. |
| 274 | LangOptions LangOptsNoCPP = LangOpts; |
| 275 | LangOptsNoCPP.CPlusPlus = false; |
| 276 | LangOptsNoCPP.CPlusPlus11 = false; |
Richard Smith | 6c74e32 | 2017-08-13 21:32:33 +0000 | [diff] [blame] | 277 | LangOptsNoCPP.CPlusPlus2a = false; |
Alex Lorenz | f127821 | 2017-04-11 15:01:53 +0000 | [diff] [blame] | 278 | return !isKeyword(LangOptsNoCPP); |
| 279 | } |
| 280 | |
Chris Lattner | ff067ce | 2007-10-07 07:52:34 +0000 | [diff] [blame] | 281 | tok::PPKeywordKind IdentifierInfo::getPPKeywordID() const { |
| 282 | // We use a perfect hash function here involving the length of the keyword, |
| 283 | // the first and third character. For preprocessor ID's there are no |
| 284 | // collisions (if there were, the switch below would complain about duplicate |
| 285 | // case values). Note that this depends on 'if' being null terminated. |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 286 | |
Chris Lattner | ff067ce | 2007-10-07 07:52:34 +0000 | [diff] [blame] | 287 | #define HASH(LEN, FIRST, THIRD) \ |
| 288 | (LEN << 5) + (((FIRST-'a') + (THIRD-'a')) & 31) |
| 289 | #define CASE(LEN, FIRST, THIRD, NAME) \ |
| 290 | case HASH(LEN, FIRST, THIRD): \ |
| 291 | return memcmp(Name, #NAME, LEN) ? tok::pp_not_keyword : tok::pp_ ## NAME |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 292 | |
Chris Lattner | ff067ce | 2007-10-07 07:52:34 +0000 | [diff] [blame] | 293 | unsigned Len = getLength(); |
Chris Lattner | d2b8ce4 | 2007-10-10 20:59:57 +0000 | [diff] [blame] | 294 | if (Len < 2) return tok::pp_not_keyword; |
Daniel Dunbar | 1c0761d | 2009-10-17 18:13:02 +0000 | [diff] [blame] | 295 | const char *Name = getNameStart(); |
Chris Lattner | ff067ce | 2007-10-07 07:52:34 +0000 | [diff] [blame] | 296 | switch (HASH(Len, Name[0], Name[2])) { |
| 297 | default: return tok::pp_not_keyword; |
| 298 | CASE( 2, 'i', '\0', if); |
| 299 | CASE( 4, 'e', 'i', elif); |
| 300 | CASE( 4, 'e', 's', else); |
| 301 | CASE( 4, 'l', 'n', line); |
| 302 | CASE( 4, 's', 'c', sccs); |
| 303 | CASE( 5, 'e', 'd', endif); |
| 304 | CASE( 5, 'e', 'r', error); |
| 305 | CASE( 5, 'i', 'e', ident); |
| 306 | CASE( 5, 'i', 'd', ifdef); |
| 307 | CASE( 5, 'u', 'd', undef); |
| 308 | |
| 309 | CASE( 6, 'a', 's', assert); |
| 310 | CASE( 6, 'd', 'f', define); |
| 311 | CASE( 6, 'i', 'n', ifndef); |
| 312 | CASE( 6, 'i', 'p', import); |
| 313 | CASE( 6, 'p', 'a', pragma); |
Douglas Gregor | 0bf886d | 2012-01-03 18:24:14 +0000 | [diff] [blame] | 314 | |
Chris Lattner | ff067ce | 2007-10-07 07:52:34 +0000 | [diff] [blame] | 315 | CASE( 7, 'd', 'f', defined); |
| 316 | CASE( 7, 'i', 'c', include); |
| 317 | CASE( 7, 'w', 'r', warning); |
| 318 | |
| 319 | CASE( 8, 'u', 'a', unassert); |
| 320 | CASE(12, 'i', 'c', include_next); |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 321 | |
Douglas Gregor | 663b48f | 2012-01-03 19:48:16 +0000 | [diff] [blame] | 322 | CASE(14, '_', 'p', __public_macro); |
| 323 | |
| 324 | CASE(15, '_', 'p', __private_macro); |
| 325 | |
Chris Lattner | 14a7f39 | 2009-04-08 18:24:34 +0000 | [diff] [blame] | 326 | CASE(16, '_', 'i', __include_macros); |
Chris Lattner | ff067ce | 2007-10-07 07:52:34 +0000 | [diff] [blame] | 327 | #undef CASE |
| 328 | #undef HASH |
| 329 | } |
| 330 | } |
Chris Lattner | 25e0d54 | 2006-10-18 06:07:05 +0000 | [diff] [blame] | 331 | |
| 332 | //===----------------------------------------------------------------------===// |
| 333 | // Stats Implementation |
| 334 | //===----------------------------------------------------------------------===// |
| 335 | |
Chris Lattner | 22eb972 | 2006-06-18 05:43:12 +0000 | [diff] [blame] | 336 | /// PrintStats - Print statistics about how well the identifier table is doing |
| 337 | /// at hashing identifiers. |
| 338 | void IdentifierTable::PrintStats() const { |
Chris Lattner | 2b9e19b | 2006-10-29 23:43:13 +0000 | [diff] [blame] | 339 | unsigned NumBuckets = HashTable.getNumBuckets(); |
| 340 | unsigned NumIdentifiers = HashTable.getNumItems(); |
| 341 | unsigned NumEmptyBuckets = NumBuckets-NumIdentifiers; |
Chris Lattner | 22eb972 | 2006-06-18 05:43:12 +0000 | [diff] [blame] | 342 | unsigned AverageIdentifierSize = 0; |
| 343 | unsigned MaxIdentifierLength = 0; |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 344 | |
Chris Lattner | 2b9e19b | 2006-10-29 23:43:13 +0000 | [diff] [blame] | 345 | // TODO: Figure out maximum times an identifier had to probe for -stats. |
Ted Kremenek | 52f73ca | 2009-01-20 23:28:34 +0000 | [diff] [blame] | 346 | for (llvm::StringMap<IdentifierInfo*, llvm::BumpPtrAllocator>::const_iterator |
Chris Lattner | b055f2d | 2007-02-11 08:19:57 +0000 | [diff] [blame] | 347 | I = HashTable.begin(), E = HashTable.end(); I != E; ++I) { |
| 348 | unsigned IdLen = I->getKeyLength(); |
| 349 | AverageIdentifierSize += IdLen; |
| 350 | if (MaxIdentifierLength < IdLen) |
| 351 | MaxIdentifierLength = IdLen; |
| 352 | } |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 353 | |
Chris Lattner | 23b7eb6 | 2007-06-15 23:05:46 +0000 | [diff] [blame] | 354 | fprintf(stderr, "\n*** Identifier Table Stats:\n"); |
| 355 | fprintf(stderr, "# Identifiers: %d\n", NumIdentifiers); |
| 356 | fprintf(stderr, "# Empty Buckets: %d\n", NumEmptyBuckets); |
| 357 | fprintf(stderr, "Hash density (#identifiers per bucket): %f\n", |
| 358 | NumIdentifiers/(double)NumBuckets); |
| 359 | fprintf(stderr, "Ave identifier length: %f\n", |
| 360 | (AverageIdentifierSize/(double)NumIdentifiers)); |
| 361 | fprintf(stderr, "Max identifier length: %d\n", MaxIdentifierLength); |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 362 | |
Chris Lattner | 22eb972 | 2006-06-18 05:43:12 +0000 | [diff] [blame] | 363 | // Compute statistics about the memory allocated for identifiers. |
Chris Lattner | 2b9e19b | 2006-10-29 23:43:13 +0000 | [diff] [blame] | 364 | HashTable.getAllocator().PrintStats(); |
Chris Lattner | 22eb972 | 2006-06-18 05:43:12 +0000 | [diff] [blame] | 365 | } |
Steve Naroff | f73590d | 2007-09-27 14:38:14 +0000 | [diff] [blame] | 366 | |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 367 | //===----------------------------------------------------------------------===// |
| 368 | // SelectorTable Implementation |
| 369 | //===----------------------------------------------------------------------===// |
| 370 | |
Chris Lattner | dadc762 | 2007-10-05 20:15:24 +0000 | [diff] [blame] | 371 | unsigned llvm::DenseMapInfo<clang::Selector>::getHashValue(clang::Selector S) { |
| 372 | return DenseMapInfo<void*>::getHashValue(S.getAsOpaquePtr()); |
| 373 | } |
| 374 | |
Douglas Gregor | 77324f3 | 2008-11-17 14:58:09 +0000 | [diff] [blame] | 375 | namespace clang { |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 376 | |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 377 | /// MultiKeywordSelector - One of these variable length records is kept for each |
| 378 | /// selector containing more than one keyword. We use a folding set |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 379 | /// to unique aggregate names (keyword selectors in ObjC parlance). Access to |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 380 | /// this class is provided strictly through Selector. |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 381 | class MultiKeywordSelector |
Douglas Gregor | 77324f3 | 2008-11-17 14:58:09 +0000 | [diff] [blame] | 382 | : public DeclarationNameExtra, public llvm::FoldingSetNode { |
Douglas Gregor | 77324f3 | 2008-11-17 14:58:09 +0000 | [diff] [blame] | 383 | MultiKeywordSelector(unsigned nKeys) { |
| 384 | ExtraKindOrNumArgs = NUM_EXTRA_KINDS + nKeys; |
| 385 | } |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 386 | |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 387 | public: |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 388 | // Constructor for keyword selectors. |
| 389 | MultiKeywordSelector(unsigned nKeys, IdentifierInfo **IIV) { |
| 390 | assert((nKeys > 1) && "not a multi-keyword selector"); |
Douglas Gregor | 77324f3 | 2008-11-17 14:58:09 +0000 | [diff] [blame] | 391 | ExtraKindOrNumArgs = NUM_EXTRA_KINDS + nKeys; |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 392 | |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 393 | // Fill in the trailing keyword array. |
| 394 | IdentifierInfo **KeyInfo = reinterpret_cast<IdentifierInfo **>(this+1); |
| 395 | for (unsigned i = 0; i != nKeys; ++i) |
| 396 | KeyInfo[i] = IIV[i]; |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 397 | } |
| 398 | |
Chris Lattner | f7f34d0 | 2007-10-07 01:33:16 +0000 | [diff] [blame] | 399 | // getName - Derive the full selector name and return it. |
| 400 | std::string getName() const; |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 401 | |
Douglas Gregor | 77324f3 | 2008-11-17 14:58:09 +0000 | [diff] [blame] | 402 | unsigned getNumArgs() const { return ExtraKindOrNumArgs - NUM_EXTRA_KINDS; } |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 403 | |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 404 | using keyword_iterator = IdentifierInfo *const *; |
| 405 | |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 406 | keyword_iterator keyword_begin() const { |
| 407 | return reinterpret_cast<keyword_iterator>(this+1); |
| 408 | } |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 409 | |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 410 | keyword_iterator keyword_end() const { |
| 411 | return keyword_begin()+getNumArgs(); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 412 | } |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 413 | |
Chris Lattner | f7f34d0 | 2007-10-07 01:33:16 +0000 | [diff] [blame] | 414 | IdentifierInfo *getIdentifierInfoForSlot(unsigned i) const { |
Douglas Gregor | 77324f3 | 2008-11-17 14:58:09 +0000 | [diff] [blame] | 415 | assert(i < getNumArgs() && "getIdentifierInfoForSlot(): illegal index"); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 416 | return keyword_begin()[i]; |
| 417 | } |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 418 | |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 419 | static void Profile(llvm::FoldingSetNodeID &ID, |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 420 | keyword_iterator ArgTys, unsigned NumArgs) { |
| 421 | ID.AddInteger(NumArgs); |
Chris Lattner | f7f34d0 | 2007-10-07 01:33:16 +0000 | [diff] [blame] | 422 | for (unsigned i = 0; i != NumArgs; ++i) |
| 423 | ID.AddPointer(ArgTys[i]); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 424 | } |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 425 | |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 426 | void Profile(llvm::FoldingSetNodeID &ID) { |
Douglas Gregor | 77324f3 | 2008-11-17 14:58:09 +0000 | [diff] [blame] | 427 | Profile(ID, keyword_begin(), getNumArgs()); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 428 | } |
| 429 | }; |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 430 | |
| 431 | } // namespace clang. |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 432 | |
| 433 | unsigned Selector::getNumArgs() const { |
| 434 | unsigned IIF = getIdentifierInfoFlag(); |
Douglas Gregor | 93a586f | 2012-05-04 18:24:37 +0000 | [diff] [blame] | 435 | if (IIF <= ZeroArg) |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 436 | return 0; |
| 437 | if (IIF == OneArg) |
| 438 | return 1; |
Douglas Gregor | 93a586f | 2012-05-04 18:24:37 +0000 | [diff] [blame] | 439 | // We point to a MultiKeywordSelector. |
| 440 | MultiKeywordSelector *SI = getMultiKeywordSelector(); |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 441 | return SI->getNumArgs(); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 442 | } |
| 443 | |
Chris Lattner | f7f34d0 | 2007-10-07 01:33:16 +0000 | [diff] [blame] | 444 | IdentifierInfo *Selector::getIdentifierInfoForSlot(unsigned argIndex) const { |
Douglas Gregor | 93a586f | 2012-05-04 18:24:37 +0000 | [diff] [blame] | 445 | if (getIdentifierInfoFlag() < MultiArg) { |
Chris Lattner | f7f34d0 | 2007-10-07 01:33:16 +0000 | [diff] [blame] | 446 | assert(argIndex == 0 && "illegal keyword index"); |
Douglas Gregor | 70091b8 | 2009-04-26 22:20:50 +0000 | [diff] [blame] | 447 | return getAsIdentifierInfo(); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 448 | } |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 449 | |
Douglas Gregor | 93a586f | 2012-05-04 18:24:37 +0000 | [diff] [blame] | 450 | // We point to a MultiKeywordSelector. |
| 451 | MultiKeywordSelector *SI = getMultiKeywordSelector(); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 452 | return SI->getIdentifierInfoForSlot(argIndex); |
| 453 | } |
| 454 | |
Chris Lattner | 0e62c1c | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 455 | StringRef Selector::getNameForSlot(unsigned int argIndex) const { |
Douglas Gregor | af2a6ae | 2011-02-18 22:29:55 +0000 | [diff] [blame] | 456 | IdentifierInfo *II = getIdentifierInfoForSlot(argIndex); |
Chris Lattner | 0e62c1c | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 457 | return II? II->getName() : StringRef(); |
Douglas Gregor | af2a6ae | 2011-02-18 22:29:55 +0000 | [diff] [blame] | 458 | } |
| 459 | |
Chris Lattner | f7f34d0 | 2007-10-07 01:33:16 +0000 | [diff] [blame] | 460 | std::string MultiKeywordSelector::getName() const { |
Dylan Noblesmith | 2c1dd27 | 2012-02-05 02:13:05 +0000 | [diff] [blame] | 461 | SmallString<256> Str; |
Daniel Dunbar | 1c0761d | 2009-10-17 18:13:02 +0000 | [diff] [blame] | 462 | llvm::raw_svector_ostream OS(Str); |
Chris Lattner | f7f34d0 | 2007-10-07 01:33:16 +0000 | [diff] [blame] | 463 | for (keyword_iterator I = keyword_begin(), E = keyword_end(); I != E; ++I) { |
| 464 | if (*I) |
Daniel Dunbar | 07d0785 | 2009-10-18 21:17:35 +0000 | [diff] [blame] | 465 | OS << (*I)->getName(); |
Daniel Dunbar | 1c0761d | 2009-10-17 18:13:02 +0000 | [diff] [blame] | 466 | OS << ':'; |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 467 | } |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 468 | |
Daniel Dunbar | 1c0761d | 2009-10-17 18:13:02 +0000 | [diff] [blame] | 469 | return OS.str(); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 470 | } |
| 471 | |
Chris Lattner | e4b9569 | 2008-11-24 03:33:13 +0000 | [diff] [blame] | 472 | std::string Selector::getAsString() const { |
Douglas Gregor | 70091b8 | 2009-04-26 22:20:50 +0000 | [diff] [blame] | 473 | if (InfoPtr == 0) |
| 474 | return "<null selector>"; |
| 475 | |
Douglas Gregor | 93a586f | 2012-05-04 18:24:37 +0000 | [diff] [blame] | 476 | if (getIdentifierInfoFlag() < MultiArg) { |
Ted Kremenek | 5f080b4 | 2009-03-06 23:36:28 +0000 | [diff] [blame] | 477 | IdentifierInfo *II = getAsIdentifierInfo(); |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 478 | |
Chandler Carruth | 5966677 | 2016-11-04 06:32:57 +0000 | [diff] [blame] | 479 | if (getNumArgs() == 0) { |
| 480 | assert(II && "If the number of arguments is 0 then II is guaranteed to " |
| 481 | "not be null."); |
Daniel Dunbar | 07d0785 | 2009-10-18 21:17:35 +0000 | [diff] [blame] | 482 | return II->getName(); |
Chandler Carruth | 5966677 | 2016-11-04 06:32:57 +0000 | [diff] [blame] | 483 | } |
Ted Kremenek | 5f080b4 | 2009-03-06 23:36:28 +0000 | [diff] [blame] | 484 | |
Daniel Dunbar | 1c0761d | 2009-10-17 18:13:02 +0000 | [diff] [blame] | 485 | if (!II) |
| 486 | return ":"; |
| 487 | |
Daniel Dunbar | 07d0785 | 2009-10-18 21:17:35 +0000 | [diff] [blame] | 488 | return II->getName().str() + ":"; |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 489 | } |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 490 | |
Douglas Gregor | 93a586f | 2012-05-04 18:24:37 +0000 | [diff] [blame] | 491 | // We have a multiple keyword selector. |
| 492 | return getMultiKeywordSelector()->getName(); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 493 | } |
| 494 | |
Aaron Ballman | b190f97 | 2014-01-03 17:59:55 +0000 | [diff] [blame] | 495 | void Selector::print(llvm::raw_ostream &OS) const { |
| 496 | OS << getAsString(); |
| 497 | } |
| 498 | |
John McCall | b452625 | 2011-03-02 01:50:55 +0000 | [diff] [blame] | 499 | /// Interpreting the given string using the normal CamelCase |
| 500 | /// conventions, determine whether the given string starts with the |
| 501 | /// given "word", which is assumed to end in a lowercase letter. |
Chris Lattner | 0e62c1c | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 502 | static bool startsWithWord(StringRef name, StringRef word) { |
John McCall | b452625 | 2011-03-02 01:50:55 +0000 | [diff] [blame] | 503 | if (name.size() < word.size()) return false; |
Jordan Rose | a7d0384 | 2013-02-08 22:30:41 +0000 | [diff] [blame] | 504 | return ((name.size() == word.size() || !isLowercase(name[word.size()])) && |
| 505 | name.startswith(word)); |
John McCall | b452625 | 2011-03-02 01:50:55 +0000 | [diff] [blame] | 506 | } |
| 507 | |
| 508 | ObjCMethodFamily Selector::getMethodFamilyImpl(Selector sel) { |
| 509 | IdentifierInfo *first = sel.getIdentifierInfoForSlot(0); |
| 510 | if (!first) return OMF_None; |
| 511 | |
Chris Lattner | 0e62c1c | 2011-07-23 10:55:15 +0000 | [diff] [blame] | 512 | StringRef name = first->getName(); |
John McCall | b452625 | 2011-03-02 01:50:55 +0000 | [diff] [blame] | 513 | if (sel.isUnarySelector()) { |
| 514 | if (name == "autorelease") return OMF_autorelease; |
| 515 | if (name == "dealloc") return OMF_dealloc; |
Nico Weber | 1fb8266 | 2011-08-28 22:35:17 +0000 | [diff] [blame] | 516 | if (name == "finalize") return OMF_finalize; |
John McCall | b452625 | 2011-03-02 01:50:55 +0000 | [diff] [blame] | 517 | if (name == "release") return OMF_release; |
| 518 | if (name == "retain") return OMF_retain; |
| 519 | if (name == "retainCount") return OMF_retainCount; |
Douglas Gregor | 3382372 | 2011-06-11 01:09:30 +0000 | [diff] [blame] | 520 | if (name == "self") return OMF_self; |
Fariborz Jahanian | 78e9deb | 2014-08-22 16:57:26 +0000 | [diff] [blame] | 521 | if (name == "initialize") return OMF_initialize; |
John McCall | b452625 | 2011-03-02 01:50:55 +0000 | [diff] [blame] | 522 | } |
Alex Lorenz | 0e23c61 | 2017-03-06 15:58:34 +0000 | [diff] [blame] | 523 | |
| 524 | if (name == "performSelector" || name == "performSelectorInBackground" || |
| 525 | name == "performSelectorOnMainThread") |
| 526 | return OMF_performSelector; |
John McCall | b452625 | 2011-03-02 01:50:55 +0000 | [diff] [blame] | 527 | |
| 528 | // The other method families may begin with a prefix of underscores. |
| 529 | while (!name.empty() && name.front() == '_') |
| 530 | name = name.substr(1); |
| 531 | |
| 532 | if (name.empty()) return OMF_None; |
| 533 | switch (name.front()) { |
| 534 | case 'a': |
| 535 | if (startsWithWord(name, "alloc")) return OMF_alloc; |
| 536 | break; |
| 537 | case 'c': |
| 538 | if (startsWithWord(name, "copy")) return OMF_copy; |
| 539 | break; |
| 540 | case 'i': |
| 541 | if (startsWithWord(name, "init")) return OMF_init; |
| 542 | break; |
| 543 | case 'm': |
| 544 | if (startsWithWord(name, "mutableCopy")) return OMF_mutableCopy; |
| 545 | break; |
| 546 | case 'n': |
| 547 | if (startsWithWord(name, "new")) return OMF_new; |
| 548 | break; |
| 549 | default: |
| 550 | break; |
| 551 | } |
| 552 | |
| 553 | return OMF_None; |
| 554 | } |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 555 | |
Fariborz Jahanian | 7122135 | 2013-07-23 22:42:28 +0000 | [diff] [blame] | 556 | ObjCInstanceTypeFamily Selector::getInstTypeMethodFamily(Selector sel) { |
Fariborz Jahanian | 4f3a64f | 2013-07-23 19:31:17 +0000 | [diff] [blame] | 557 | IdentifierInfo *first = sel.getIdentifierInfoForSlot(0); |
| 558 | if (!first) return OIT_None; |
| 559 | |
| 560 | StringRef name = first->getName(); |
| 561 | |
| 562 | if (name.empty()) return OIT_None; |
| 563 | switch (name.front()) { |
| 564 | case 'a': |
Fariborz Jahanian | 4ccdc73 | 2013-08-29 16:22:26 +0000 | [diff] [blame] | 565 | if (startsWithWord(name, "array")) return OIT_Array; |
Fariborz Jahanian | 4f3a64f | 2013-07-23 19:31:17 +0000 | [diff] [blame] | 566 | break; |
| 567 | case 'd': |
Fariborz Jahanian | 7c87b43 | 2013-10-10 18:23:13 +0000 | [diff] [blame] | 568 | if (startsWithWord(name, "default")) return OIT_ReturnsSelf; |
Fariborz Jahanian | 4f3a64f | 2013-07-23 19:31:17 +0000 | [diff] [blame] | 569 | if (startsWithWord(name, "dictionary")) return OIT_Dictionary; |
| 570 | break; |
Fariborz Jahanian | 9275c68 | 2013-08-02 20:54:18 +0000 | [diff] [blame] | 571 | case 's': |
Fariborz Jahanian | 7c87b43 | 2013-10-10 18:23:13 +0000 | [diff] [blame] | 572 | if (startsWithWord(name, "shared")) return OIT_ReturnsSelf; |
| 573 | if (startsWithWord(name, "standard")) return OIT_Singleton; |
Galina Kistanova | ddcd281 | 2017-06-03 06:40:10 +0000 | [diff] [blame] | 574 | break; |
Fariborz Jahanian | 1c900bc | 2013-09-18 20:35:47 +0000 | [diff] [blame] | 575 | case 'i': |
| 576 | if (startsWithWord(name, "init")) return OIT_Init; |
Fariborz Jahanian | 4f3a64f | 2013-07-23 19:31:17 +0000 | [diff] [blame] | 577 | default: |
| 578 | break; |
| 579 | } |
| 580 | return OIT_None; |
| 581 | } |
| 582 | |
Fariborz Jahanian | 6485fe4 | 2014-09-09 23:10:54 +0000 | [diff] [blame] | 583 | ObjCStringFormatFamily Selector::getStringFormatFamilyImpl(Selector sel) { |
| 584 | IdentifierInfo *first = sel.getIdentifierInfoForSlot(0); |
| 585 | if (!first) return SFF_None; |
| 586 | |
| 587 | StringRef name = first->getName(); |
| 588 | |
| 589 | switch (name.front()) { |
| 590 | case 'a': |
| 591 | if (name == "appendFormat") return SFF_NSString; |
| 592 | break; |
| 593 | |
| 594 | case 'i': |
| 595 | if (name == "initWithFormat") return SFF_NSString; |
| 596 | break; |
| 597 | |
| 598 | case 'l': |
| 599 | if (name == "localizedStringWithFormat") return SFF_NSString; |
| 600 | break; |
| 601 | |
| 602 | case 's': |
| 603 | if (name == "stringByAppendingFormat" || |
| 604 | name == "stringWithFormat") return SFF_NSString; |
| 605 | break; |
| 606 | } |
| 607 | return SFF_None; |
| 608 | } |
| 609 | |
Chris Lattner | 1a84994 | 2009-03-04 05:35:38 +0000 | [diff] [blame] | 610 | namespace { |
Eugene Zelenko | 918e0ca | 2017-11-03 22:35:27 +0000 | [diff] [blame] | 611 | |
| 612 | struct SelectorTableImpl { |
| 613 | llvm::FoldingSet<MultiKeywordSelector> Table; |
| 614 | llvm::BumpPtrAllocator Allocator; |
| 615 | }; |
| 616 | |
| 617 | } // namespace |
Chris Lattner | 1a84994 | 2009-03-04 05:35:38 +0000 | [diff] [blame] | 618 | |
| 619 | static SelectorTableImpl &getSelectorTableImpl(void *P) { |
| 620 | return *static_cast<SelectorTableImpl*>(P); |
| 621 | } |
| 622 | |
Adrian Prantl | 6e77c96 | 2013-06-10 21:36:55 +0000 | [diff] [blame] | 623 | SmallString<64> |
Adrian Prantl | a4ce906 | 2013-06-07 22:29:12 +0000 | [diff] [blame] | 624 | SelectorTable::constructSetterName(StringRef Name) { |
Adrian Prantl | 6e77c96 | 2013-06-10 21:36:55 +0000 | [diff] [blame] | 625 | SmallString<64> SetterName("set"); |
| 626 | SetterName += Name; |
| 627 | SetterName[3] = toUppercase(SetterName[3]); |
| 628 | return SetterName; |
Adrian Prantl | a4ce906 | 2013-06-07 22:29:12 +0000 | [diff] [blame] | 629 | } |
| 630 | |
Adrian Prantl | 6e77c96 | 2013-06-10 21:36:55 +0000 | [diff] [blame] | 631 | Selector |
Adrian Prantl | a4ce906 | 2013-06-07 22:29:12 +0000 | [diff] [blame] | 632 | SelectorTable::constructSetterSelector(IdentifierTable &Idents, |
| 633 | SelectorTable &SelTable, |
| 634 | const IdentifierInfo *Name) { |
| 635 | IdentifierInfo *SetterName = |
| 636 | &Idents.get(constructSetterName(Name->getName())); |
Benjamin Kramer | 4903802 | 2012-02-04 13:45:25 +0000 | [diff] [blame] | 637 | return SelTable.getUnarySelector(SetterName); |
| 638 | } |
| 639 | |
Ted Kremenek | 1c2239e | 2011-04-18 22:47:04 +0000 | [diff] [blame] | 640 | size_t SelectorTable::getTotalMemory() const { |
| 641 | SelectorTableImpl &SelTabImpl = getSelectorTableImpl(Impl); |
| 642 | return SelTabImpl.Allocator.getTotalMemory(); |
| 643 | } |
Chris Lattner | 1a84994 | 2009-03-04 05:35:38 +0000 | [diff] [blame] | 644 | |
Chris Lattner | 5700fab | 2007-10-07 02:00:24 +0000 | [diff] [blame] | 645 | Selector SelectorTable::getSelector(unsigned nKeys, IdentifierInfo **IIV) { |
| 646 | if (nKeys < 2) |
| 647 | return Selector(IIV[0], nKeys); |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 648 | |
Chris Lattner | 1a84994 | 2009-03-04 05:35:38 +0000 | [diff] [blame] | 649 | SelectorTableImpl &SelTabImpl = getSelectorTableImpl(Impl); |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 650 | |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 651 | // Unique selector, to guarantee there is one per name. |
| 652 | llvm::FoldingSetNodeID ID; |
| 653 | MultiKeywordSelector::Profile(ID, IIV, nKeys); |
| 654 | |
Craig Topper | f1186c5 | 2014-05-08 06:41:40 +0000 | [diff] [blame] | 655 | void *InsertPos = nullptr; |
Chris Lattner | 1a84994 | 2009-03-04 05:35:38 +0000 | [diff] [blame] | 656 | if (MultiKeywordSelector *SI = |
| 657 | SelTabImpl.Table.FindNodeOrInsertPos(ID, InsertPos)) |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 658 | return Selector(SI); |
Mike Stump | 11289f4 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 659 | |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 660 | // MultiKeywordSelector objects are not allocated with new because they have a |
| 661 | // variable size array (for parameter types) at the end of them. |
Chris Lattner | 1a84994 | 2009-03-04 05:35:38 +0000 | [diff] [blame] | 662 | unsigned Size = sizeof(MultiKeywordSelector) + nKeys*sizeof(IdentifierInfo *); |
| 663 | MultiKeywordSelector *SI = |
Benjamin Kramer | c3f8925 | 2016-10-20 14:27:22 +0000 | [diff] [blame] | 664 | (MultiKeywordSelector *)SelTabImpl.Allocator.Allocate( |
| 665 | Size, alignof(MultiKeywordSelector)); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 666 | new (SI) MultiKeywordSelector(nKeys, IIV); |
Chris Lattner | 1a84994 | 2009-03-04 05:35:38 +0000 | [diff] [blame] | 667 | SelTabImpl.Table.InsertNode(SI, InsertPos); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 668 | return Selector(SI); |
| 669 | } |
| 670 | |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 671 | SelectorTable::SelectorTable() { |
Chris Lattner | 1a84994 | 2009-03-04 05:35:38 +0000 | [diff] [blame] | 672 | Impl = new SelectorTableImpl(); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 673 | } |
| 674 | |
| 675 | SelectorTable::~SelectorTable() { |
Chris Lattner | 1a84994 | 2009-03-04 05:35:38 +0000 | [diff] [blame] | 676 | delete &getSelectorTableImpl(Impl); |
Steve Naroff | e61bfa8 | 2007-10-05 18:42:47 +0000 | [diff] [blame] | 677 | } |
| 678 | |
Douglas Gregor | 71395fa | 2009-11-04 00:56:37 +0000 | [diff] [blame] | 679 | const char *clang::getOperatorSpelling(OverloadedOperatorKind Operator) { |
| 680 | switch (Operator) { |
| 681 | case OO_None: |
| 682 | case NUM_OVERLOADED_OPERATORS: |
Craig Topper | f1186c5 | 2014-05-08 06:41:40 +0000 | [diff] [blame] | 683 | return nullptr; |
Kovarththanan Rajaratnam | 00682a4 | 2010-03-12 11:27:37 +0000 | [diff] [blame] | 684 | |
Douglas Gregor | 71395fa | 2009-11-04 00:56:37 +0000 | [diff] [blame] | 685 | #define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \ |
| 686 | case OO_##Name: return Spelling; |
| 687 | #include "clang/Basic/OperatorKinds.def" |
| 688 | } |
Kovarththanan Rajaratnam | 00682a4 | 2010-03-12 11:27:37 +0000 | [diff] [blame] | 689 | |
David Blaikie | 8a40f70 | 2012-01-17 06:56:22 +0000 | [diff] [blame] | 690 | llvm_unreachable("Invalid OverloadedOperatorKind!"); |
Douglas Gregor | 71395fa | 2009-11-04 00:56:37 +0000 | [diff] [blame] | 691 | } |
Douglas Gregor | 813a066 | 2015-06-19 18:14:38 +0000 | [diff] [blame] | 692 | |
Douglas Gregor | aea7afd | 2015-06-24 22:02:08 +0000 | [diff] [blame] | 693 | StringRef clang::getNullabilitySpelling(NullabilityKind kind, |
| 694 | bool isContextSensitive) { |
Douglas Gregor | 813a066 | 2015-06-19 18:14:38 +0000 | [diff] [blame] | 695 | switch (kind) { |
| 696 | case NullabilityKind::NonNull: |
Douglas Gregor | aea7afd | 2015-06-24 22:02:08 +0000 | [diff] [blame] | 697 | return isContextSensitive ? "nonnull" : "_Nonnull"; |
Douglas Gregor | 813a066 | 2015-06-19 18:14:38 +0000 | [diff] [blame] | 698 | |
| 699 | case NullabilityKind::Nullable: |
Douglas Gregor | aea7afd | 2015-06-24 22:02:08 +0000 | [diff] [blame] | 700 | return isContextSensitive ? "nullable" : "_Nullable"; |
Douglas Gregor | 813a066 | 2015-06-19 18:14:38 +0000 | [diff] [blame] | 701 | |
| 702 | case NullabilityKind::Unspecified: |
Douglas Gregor | aea7afd | 2015-06-24 22:02:08 +0000 | [diff] [blame] | 703 | return isContextSensitive ? "null_unspecified" : "_Null_unspecified"; |
Douglas Gregor | 813a066 | 2015-06-19 18:14:38 +0000 | [diff] [blame] | 704 | } |
Nico Weber | 708aabd | 2015-06-20 00:06:30 +0000 | [diff] [blame] | 705 | llvm_unreachable("Unknown nullability kind."); |
Douglas Gregor | 813a066 | 2015-06-19 18:14:38 +0000 | [diff] [blame] | 706 | } |