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