xref: /freebsd/contrib/llvm-project/clang/lib/AST/NSAPI.cpp (revision bdd1243df58e60e85101c09001d9812a789b6bc4)
10b57cec5SDimitry Andric //===--- NSAPI.cpp - NSFoundation APIs ------------------------------------===//
20b57cec5SDimitry Andric //
30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
60b57cec5SDimitry Andric //
70b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
80b57cec5SDimitry Andric 
90b57cec5SDimitry Andric #include "clang/AST/NSAPI.h"
100b57cec5SDimitry Andric #include "clang/AST/ASTContext.h"
110b57cec5SDimitry Andric #include "clang/AST/DeclObjC.h"
120b57cec5SDimitry Andric #include "clang/AST/Expr.h"
130b57cec5SDimitry Andric #include "llvm/ADT/StringSwitch.h"
14*bdd1243dSDimitry Andric #include <optional>
150b57cec5SDimitry Andric 
160b57cec5SDimitry Andric using namespace clang;
170b57cec5SDimitry Andric 
180b57cec5SDimitry Andric NSAPI::NSAPI(ASTContext &ctx)
190b57cec5SDimitry Andric   : Ctx(ctx), ClassIds(), BOOLId(nullptr), NSIntegerId(nullptr),
200b57cec5SDimitry Andric     NSUIntegerId(nullptr), NSASCIIStringEncodingId(nullptr),
210b57cec5SDimitry Andric     NSUTF8StringEncodingId(nullptr) {}
220b57cec5SDimitry Andric 
230b57cec5SDimitry Andric IdentifierInfo *NSAPI::getNSClassId(NSClassIdKindKind K) const {
240b57cec5SDimitry Andric   static const char *ClassName[NumClassIds] = {
250b57cec5SDimitry Andric     "NSObject",
260b57cec5SDimitry Andric     "NSString",
270b57cec5SDimitry Andric     "NSArray",
280b57cec5SDimitry Andric     "NSMutableArray",
290b57cec5SDimitry Andric     "NSDictionary",
300b57cec5SDimitry Andric     "NSMutableDictionary",
310b57cec5SDimitry Andric     "NSNumber",
320b57cec5SDimitry Andric     "NSMutableSet",
330b57cec5SDimitry Andric     "NSMutableOrderedSet",
340b57cec5SDimitry Andric     "NSValue"
350b57cec5SDimitry Andric   };
360b57cec5SDimitry Andric 
370b57cec5SDimitry Andric   if (!ClassIds[K])
380b57cec5SDimitry Andric     return (ClassIds[K] = &Ctx.Idents.get(ClassName[K]));
390b57cec5SDimitry Andric 
400b57cec5SDimitry Andric   return ClassIds[K];
410b57cec5SDimitry Andric }
420b57cec5SDimitry Andric 
430b57cec5SDimitry Andric Selector NSAPI::getNSStringSelector(NSStringMethodKind MK) const {
440b57cec5SDimitry Andric   if (NSStringSelectors[MK].isNull()) {
450b57cec5SDimitry Andric     Selector Sel;
460b57cec5SDimitry Andric     switch (MK) {
470b57cec5SDimitry Andric     case NSStr_stringWithString:
480b57cec5SDimitry Andric       Sel = Ctx.Selectors.getUnarySelector(&Ctx.Idents.get("stringWithString"));
490b57cec5SDimitry Andric       break;
500b57cec5SDimitry Andric     case NSStr_stringWithUTF8String:
510b57cec5SDimitry Andric       Sel = Ctx.Selectors.getUnarySelector(
520b57cec5SDimitry Andric                                        &Ctx.Idents.get("stringWithUTF8String"));
530b57cec5SDimitry Andric       break;
540b57cec5SDimitry Andric     case NSStr_initWithUTF8String:
550b57cec5SDimitry Andric       Sel = Ctx.Selectors.getUnarySelector(
560b57cec5SDimitry Andric                                        &Ctx.Idents.get("initWithUTF8String"));
570b57cec5SDimitry Andric       break;
580b57cec5SDimitry Andric     case NSStr_stringWithCStringEncoding: {
590b57cec5SDimitry Andric       IdentifierInfo *KeyIdents[] = {
600b57cec5SDimitry Andric         &Ctx.Idents.get("stringWithCString"),
610b57cec5SDimitry Andric         &Ctx.Idents.get("encoding")
620b57cec5SDimitry Andric       };
630b57cec5SDimitry Andric       Sel = Ctx.Selectors.getSelector(2, KeyIdents);
640b57cec5SDimitry Andric       break;
650b57cec5SDimitry Andric     }
660b57cec5SDimitry Andric     case NSStr_stringWithCString:
670b57cec5SDimitry Andric       Sel= Ctx.Selectors.getUnarySelector(&Ctx.Idents.get("stringWithCString"));
680b57cec5SDimitry Andric       break;
690b57cec5SDimitry Andric     case NSStr_initWithString:
700b57cec5SDimitry Andric       Sel = Ctx.Selectors.getUnarySelector(&Ctx.Idents.get("initWithString"));
710b57cec5SDimitry Andric       break;
720b57cec5SDimitry Andric     }
730b57cec5SDimitry Andric     return (NSStringSelectors[MK] = Sel);
740b57cec5SDimitry Andric   }
750b57cec5SDimitry Andric 
760b57cec5SDimitry Andric   return NSStringSelectors[MK];
770b57cec5SDimitry Andric }
780b57cec5SDimitry Andric 
790b57cec5SDimitry Andric Selector NSAPI::getNSArraySelector(NSArrayMethodKind MK) const {
800b57cec5SDimitry Andric   if (NSArraySelectors[MK].isNull()) {
810b57cec5SDimitry Andric     Selector Sel;
820b57cec5SDimitry Andric     switch (MK) {
830b57cec5SDimitry Andric     case NSArr_array:
840b57cec5SDimitry Andric       Sel = Ctx.Selectors.getNullarySelector(&Ctx.Idents.get("array"));
850b57cec5SDimitry Andric       break;
860b57cec5SDimitry Andric     case NSArr_arrayWithArray:
870b57cec5SDimitry Andric       Sel = Ctx.Selectors.getUnarySelector(&Ctx.Idents.get("arrayWithArray"));
880b57cec5SDimitry Andric       break;
890b57cec5SDimitry Andric     case NSArr_arrayWithObject:
900b57cec5SDimitry Andric       Sel = Ctx.Selectors.getUnarySelector(&Ctx.Idents.get("arrayWithObject"));
910b57cec5SDimitry Andric       break;
920b57cec5SDimitry Andric     case NSArr_arrayWithObjects:
930b57cec5SDimitry Andric       Sel = Ctx.Selectors.getUnarySelector(&Ctx.Idents.get("arrayWithObjects"));
940b57cec5SDimitry Andric       break;
950b57cec5SDimitry Andric     case NSArr_arrayWithObjectsCount: {
960b57cec5SDimitry Andric       IdentifierInfo *KeyIdents[] = {
970b57cec5SDimitry Andric         &Ctx.Idents.get("arrayWithObjects"),
980b57cec5SDimitry Andric         &Ctx.Idents.get("count")
990b57cec5SDimitry Andric       };
1000b57cec5SDimitry Andric       Sel = Ctx.Selectors.getSelector(2, KeyIdents);
1010b57cec5SDimitry Andric       break;
1020b57cec5SDimitry Andric     }
1030b57cec5SDimitry Andric     case NSArr_initWithArray:
1040b57cec5SDimitry Andric       Sel = Ctx.Selectors.getUnarySelector(&Ctx.Idents.get("initWithArray"));
1050b57cec5SDimitry Andric       break;
1060b57cec5SDimitry Andric     case NSArr_initWithObjects:
1070b57cec5SDimitry Andric       Sel = Ctx.Selectors.getUnarySelector(&Ctx.Idents.get("initWithObjects"));
1080b57cec5SDimitry Andric       break;
1090b57cec5SDimitry Andric     case NSArr_objectAtIndex:
1100b57cec5SDimitry Andric       Sel = Ctx.Selectors.getUnarySelector(&Ctx.Idents.get("objectAtIndex"));
1110b57cec5SDimitry Andric       break;
1120b57cec5SDimitry Andric     case NSMutableArr_replaceObjectAtIndex: {
1130b57cec5SDimitry Andric       IdentifierInfo *KeyIdents[] = {
1140b57cec5SDimitry Andric         &Ctx.Idents.get("replaceObjectAtIndex"),
1150b57cec5SDimitry Andric         &Ctx.Idents.get("withObject")
1160b57cec5SDimitry Andric       };
1170b57cec5SDimitry Andric       Sel = Ctx.Selectors.getSelector(2, KeyIdents);
1180b57cec5SDimitry Andric       break;
1190b57cec5SDimitry Andric     }
1200b57cec5SDimitry Andric     case NSMutableArr_addObject:
1210b57cec5SDimitry Andric       Sel = Ctx.Selectors.getUnarySelector(&Ctx.Idents.get("addObject"));
1220b57cec5SDimitry Andric       break;
1230b57cec5SDimitry Andric     case NSMutableArr_insertObjectAtIndex: {
1240b57cec5SDimitry Andric       IdentifierInfo *KeyIdents[] = {
1250b57cec5SDimitry Andric         &Ctx.Idents.get("insertObject"),
1260b57cec5SDimitry Andric         &Ctx.Idents.get("atIndex")
1270b57cec5SDimitry Andric       };
1280b57cec5SDimitry Andric       Sel = Ctx.Selectors.getSelector(2, KeyIdents);
1290b57cec5SDimitry Andric       break;
1300b57cec5SDimitry Andric     }
1310b57cec5SDimitry Andric     case NSMutableArr_setObjectAtIndexedSubscript: {
1320b57cec5SDimitry Andric       IdentifierInfo *KeyIdents[] = {
1330b57cec5SDimitry Andric         &Ctx.Idents.get("setObject"),
1340b57cec5SDimitry Andric         &Ctx.Idents.get("atIndexedSubscript")
1350b57cec5SDimitry Andric       };
1360b57cec5SDimitry Andric       Sel = Ctx.Selectors.getSelector(2, KeyIdents);
1370b57cec5SDimitry Andric       break;
1380b57cec5SDimitry Andric     }
1390b57cec5SDimitry Andric     }
1400b57cec5SDimitry Andric     return (NSArraySelectors[MK] = Sel);
1410b57cec5SDimitry Andric   }
1420b57cec5SDimitry Andric 
1430b57cec5SDimitry Andric   return NSArraySelectors[MK];
1440b57cec5SDimitry Andric }
1450b57cec5SDimitry Andric 
146*bdd1243dSDimitry Andric std::optional<NSAPI::NSArrayMethodKind>
147*bdd1243dSDimitry Andric NSAPI::getNSArrayMethodKind(Selector Sel) {
1480b57cec5SDimitry Andric   for (unsigned i = 0; i != NumNSArrayMethods; ++i) {
1490b57cec5SDimitry Andric     NSArrayMethodKind MK = NSArrayMethodKind(i);
1500b57cec5SDimitry Andric     if (Sel == getNSArraySelector(MK))
1510b57cec5SDimitry Andric       return MK;
1520b57cec5SDimitry Andric   }
1530b57cec5SDimitry Andric 
154*bdd1243dSDimitry Andric   return std::nullopt;
1550b57cec5SDimitry Andric }
1560b57cec5SDimitry Andric 
1570b57cec5SDimitry Andric Selector NSAPI::getNSDictionarySelector(
1580b57cec5SDimitry Andric                                        NSDictionaryMethodKind MK) const {
1590b57cec5SDimitry Andric   if (NSDictionarySelectors[MK].isNull()) {
1600b57cec5SDimitry Andric     Selector Sel;
1610b57cec5SDimitry Andric     switch (MK) {
1620b57cec5SDimitry Andric     case NSDict_dictionary:
1630b57cec5SDimitry Andric       Sel = Ctx.Selectors.getNullarySelector(&Ctx.Idents.get("dictionary"));
1640b57cec5SDimitry Andric       break;
1650b57cec5SDimitry Andric     case NSDict_dictionaryWithDictionary:
1660b57cec5SDimitry Andric       Sel = Ctx.Selectors.getUnarySelector(
1670b57cec5SDimitry Andric                                    &Ctx.Idents.get("dictionaryWithDictionary"));
1680b57cec5SDimitry Andric       break;
1690b57cec5SDimitry Andric     case NSDict_dictionaryWithObjectForKey: {
1700b57cec5SDimitry Andric       IdentifierInfo *KeyIdents[] = {
1710b57cec5SDimitry Andric         &Ctx.Idents.get("dictionaryWithObject"),
1720b57cec5SDimitry Andric         &Ctx.Idents.get("forKey")
1730b57cec5SDimitry Andric       };
1740b57cec5SDimitry Andric       Sel = Ctx.Selectors.getSelector(2, KeyIdents);
1750b57cec5SDimitry Andric       break;
1760b57cec5SDimitry Andric     }
1770b57cec5SDimitry Andric     case NSDict_dictionaryWithObjectsForKeys: {
1780b57cec5SDimitry Andric       IdentifierInfo *KeyIdents[] = {
1790b57cec5SDimitry Andric         &Ctx.Idents.get("dictionaryWithObjects"),
1800b57cec5SDimitry Andric         &Ctx.Idents.get("forKeys")
1810b57cec5SDimitry Andric       };
1820b57cec5SDimitry Andric       Sel = Ctx.Selectors.getSelector(2, KeyIdents);
1830b57cec5SDimitry Andric       break;
1840b57cec5SDimitry Andric     }
1850b57cec5SDimitry Andric     case NSDict_dictionaryWithObjectsForKeysCount: {
1860b57cec5SDimitry Andric       IdentifierInfo *KeyIdents[] = {
1870b57cec5SDimitry Andric         &Ctx.Idents.get("dictionaryWithObjects"),
1880b57cec5SDimitry Andric         &Ctx.Idents.get("forKeys"),
1890b57cec5SDimitry Andric         &Ctx.Idents.get("count")
1900b57cec5SDimitry Andric       };
1910b57cec5SDimitry Andric       Sel = Ctx.Selectors.getSelector(3, KeyIdents);
1920b57cec5SDimitry Andric       break;
1930b57cec5SDimitry Andric     }
1940b57cec5SDimitry Andric     case NSDict_dictionaryWithObjectsAndKeys:
1950b57cec5SDimitry Andric       Sel = Ctx.Selectors.getUnarySelector(
1960b57cec5SDimitry Andric                                &Ctx.Idents.get("dictionaryWithObjectsAndKeys"));
1970b57cec5SDimitry Andric       break;
1980b57cec5SDimitry Andric     case NSDict_initWithDictionary:
1990b57cec5SDimitry Andric       Sel = Ctx.Selectors.getUnarySelector(
2000b57cec5SDimitry Andric                                          &Ctx.Idents.get("initWithDictionary"));
2010b57cec5SDimitry Andric       break;
2020b57cec5SDimitry Andric     case NSDict_initWithObjectsAndKeys:
2030b57cec5SDimitry Andric       Sel = Ctx.Selectors.getUnarySelector(
2040b57cec5SDimitry Andric                                      &Ctx.Idents.get("initWithObjectsAndKeys"));
2050b57cec5SDimitry Andric       break;
2060b57cec5SDimitry Andric     case NSDict_initWithObjectsForKeys: {
2070b57cec5SDimitry Andric       IdentifierInfo *KeyIdents[] = {
2080b57cec5SDimitry Andric         &Ctx.Idents.get("initWithObjects"),
2090b57cec5SDimitry Andric         &Ctx.Idents.get("forKeys")
2100b57cec5SDimitry Andric       };
2110b57cec5SDimitry Andric       Sel = Ctx.Selectors.getSelector(2, KeyIdents);
2120b57cec5SDimitry Andric       break;
2130b57cec5SDimitry Andric     }
2140b57cec5SDimitry Andric     case NSDict_objectForKey:
2150b57cec5SDimitry Andric       Sel = Ctx.Selectors.getUnarySelector(&Ctx.Idents.get("objectForKey"));
2160b57cec5SDimitry Andric       break;
2170b57cec5SDimitry Andric     case NSMutableDict_setObjectForKey: {
2180b57cec5SDimitry Andric       IdentifierInfo *KeyIdents[] = {
2190b57cec5SDimitry Andric         &Ctx.Idents.get("setObject"),
2200b57cec5SDimitry Andric         &Ctx.Idents.get("forKey")
2210b57cec5SDimitry Andric       };
2220b57cec5SDimitry Andric       Sel = Ctx.Selectors.getSelector(2, KeyIdents);
2230b57cec5SDimitry Andric       break;
2240b57cec5SDimitry Andric     }
2250b57cec5SDimitry Andric     case NSMutableDict_setObjectForKeyedSubscript: {
2260b57cec5SDimitry Andric       IdentifierInfo *KeyIdents[] = {
2270b57cec5SDimitry Andric         &Ctx.Idents.get("setObject"),
2280b57cec5SDimitry Andric         &Ctx.Idents.get("forKeyedSubscript")
2290b57cec5SDimitry Andric       };
2300b57cec5SDimitry Andric       Sel = Ctx.Selectors.getSelector(2, KeyIdents);
2310b57cec5SDimitry Andric       break;
2320b57cec5SDimitry Andric     }
2330b57cec5SDimitry Andric     case NSMutableDict_setValueForKey: {
2340b57cec5SDimitry Andric       IdentifierInfo *KeyIdents[] = {
2350b57cec5SDimitry Andric         &Ctx.Idents.get("setValue"),
2360b57cec5SDimitry Andric         &Ctx.Idents.get("forKey")
2370b57cec5SDimitry Andric       };
2380b57cec5SDimitry Andric       Sel = Ctx.Selectors.getSelector(2, KeyIdents);
2390b57cec5SDimitry Andric       break;
2400b57cec5SDimitry Andric     }
2410b57cec5SDimitry Andric     }
2420b57cec5SDimitry Andric     return (NSDictionarySelectors[MK] = Sel);
2430b57cec5SDimitry Andric   }
2440b57cec5SDimitry Andric 
2450b57cec5SDimitry Andric   return NSDictionarySelectors[MK];
2460b57cec5SDimitry Andric }
2470b57cec5SDimitry Andric 
248*bdd1243dSDimitry Andric std::optional<NSAPI::NSDictionaryMethodKind>
2490b57cec5SDimitry Andric NSAPI::getNSDictionaryMethodKind(Selector Sel) {
2500b57cec5SDimitry Andric   for (unsigned i = 0; i != NumNSDictionaryMethods; ++i) {
2510b57cec5SDimitry Andric     NSDictionaryMethodKind MK = NSDictionaryMethodKind(i);
2520b57cec5SDimitry Andric     if (Sel == getNSDictionarySelector(MK))
2530b57cec5SDimitry Andric       return MK;
2540b57cec5SDimitry Andric   }
2550b57cec5SDimitry Andric 
256*bdd1243dSDimitry Andric   return std::nullopt;
2570b57cec5SDimitry Andric }
2580b57cec5SDimitry Andric 
2590b57cec5SDimitry Andric Selector NSAPI::getNSSetSelector(NSSetMethodKind MK) const {
2600b57cec5SDimitry Andric   if (NSSetSelectors[MK].isNull()) {
2610b57cec5SDimitry Andric     Selector Sel;
2620b57cec5SDimitry Andric     switch (MK) {
2630b57cec5SDimitry Andric     case NSMutableSet_addObject:
2640b57cec5SDimitry Andric       Sel = Ctx.Selectors.getUnarySelector(&Ctx.Idents.get("addObject"));
2650b57cec5SDimitry Andric       break;
2660b57cec5SDimitry Andric     case NSOrderedSet_insertObjectAtIndex: {
2670b57cec5SDimitry Andric       IdentifierInfo *KeyIdents[] = {
2680b57cec5SDimitry Andric         &Ctx.Idents.get("insertObject"),
2690b57cec5SDimitry Andric         &Ctx.Idents.get("atIndex")
2700b57cec5SDimitry Andric       };
2710b57cec5SDimitry Andric       Sel = Ctx.Selectors.getSelector(2, KeyIdents);
2720b57cec5SDimitry Andric       break;
2730b57cec5SDimitry Andric     }
2740b57cec5SDimitry Andric     case NSOrderedSet_setObjectAtIndex: {
2750b57cec5SDimitry Andric       IdentifierInfo *KeyIdents[] = {
2760b57cec5SDimitry Andric         &Ctx.Idents.get("setObject"),
2770b57cec5SDimitry Andric         &Ctx.Idents.get("atIndex")
2780b57cec5SDimitry Andric       };
2790b57cec5SDimitry Andric       Sel = Ctx.Selectors.getSelector(2, KeyIdents);
2800b57cec5SDimitry Andric       break;
2810b57cec5SDimitry Andric     }
2820b57cec5SDimitry Andric     case NSOrderedSet_setObjectAtIndexedSubscript: {
2830b57cec5SDimitry Andric       IdentifierInfo *KeyIdents[] = {
2840b57cec5SDimitry Andric         &Ctx.Idents.get("setObject"),
2850b57cec5SDimitry Andric         &Ctx.Idents.get("atIndexedSubscript")
2860b57cec5SDimitry Andric       };
2870b57cec5SDimitry Andric       Sel = Ctx.Selectors.getSelector(2, KeyIdents);
2880b57cec5SDimitry Andric       break;
2890b57cec5SDimitry Andric     }
2900b57cec5SDimitry Andric     case NSOrderedSet_replaceObjectAtIndexWithObject: {
2910b57cec5SDimitry Andric       IdentifierInfo *KeyIdents[] = {
2920b57cec5SDimitry Andric         &Ctx.Idents.get("replaceObjectAtIndex"),
2930b57cec5SDimitry Andric         &Ctx.Idents.get("withObject")
2940b57cec5SDimitry Andric       };
2950b57cec5SDimitry Andric       Sel = Ctx.Selectors.getSelector(2, KeyIdents);
2960b57cec5SDimitry Andric       break;
2970b57cec5SDimitry Andric     }
2980b57cec5SDimitry Andric     }
2990b57cec5SDimitry Andric     return (NSSetSelectors[MK] = Sel);
3000b57cec5SDimitry Andric   }
3010b57cec5SDimitry Andric 
3020b57cec5SDimitry Andric   return NSSetSelectors[MK];
3030b57cec5SDimitry Andric }
3040b57cec5SDimitry Andric 
305*bdd1243dSDimitry Andric std::optional<NSAPI::NSSetMethodKind> NSAPI::getNSSetMethodKind(Selector Sel) {
3060b57cec5SDimitry Andric   for (unsigned i = 0; i != NumNSSetMethods; ++i) {
3070b57cec5SDimitry Andric     NSSetMethodKind MK = NSSetMethodKind(i);
3080b57cec5SDimitry Andric     if (Sel == getNSSetSelector(MK))
3090b57cec5SDimitry Andric       return MK;
3100b57cec5SDimitry Andric   }
3110b57cec5SDimitry Andric 
312*bdd1243dSDimitry Andric   return std::nullopt;
3130b57cec5SDimitry Andric }
3140b57cec5SDimitry Andric 
3150b57cec5SDimitry Andric Selector NSAPI::getNSNumberLiteralSelector(NSNumberLiteralMethodKind MK,
3160b57cec5SDimitry Andric                                            bool Instance) const {
3170b57cec5SDimitry Andric   static const char *ClassSelectorName[NumNSNumberLiteralMethods] = {
3180b57cec5SDimitry Andric     "numberWithChar",
3190b57cec5SDimitry Andric     "numberWithUnsignedChar",
3200b57cec5SDimitry Andric     "numberWithShort",
3210b57cec5SDimitry Andric     "numberWithUnsignedShort",
3220b57cec5SDimitry Andric     "numberWithInt",
3230b57cec5SDimitry Andric     "numberWithUnsignedInt",
3240b57cec5SDimitry Andric     "numberWithLong",
3250b57cec5SDimitry Andric     "numberWithUnsignedLong",
3260b57cec5SDimitry Andric     "numberWithLongLong",
3270b57cec5SDimitry Andric     "numberWithUnsignedLongLong",
3280b57cec5SDimitry Andric     "numberWithFloat",
3290b57cec5SDimitry Andric     "numberWithDouble",
3300b57cec5SDimitry Andric     "numberWithBool",
3310b57cec5SDimitry Andric     "numberWithInteger",
3320b57cec5SDimitry Andric     "numberWithUnsignedInteger"
3330b57cec5SDimitry Andric   };
3340b57cec5SDimitry Andric   static const char *InstanceSelectorName[NumNSNumberLiteralMethods] = {
3350b57cec5SDimitry Andric     "initWithChar",
3360b57cec5SDimitry Andric     "initWithUnsignedChar",
3370b57cec5SDimitry Andric     "initWithShort",
3380b57cec5SDimitry Andric     "initWithUnsignedShort",
3390b57cec5SDimitry Andric     "initWithInt",
3400b57cec5SDimitry Andric     "initWithUnsignedInt",
3410b57cec5SDimitry Andric     "initWithLong",
3420b57cec5SDimitry Andric     "initWithUnsignedLong",
3430b57cec5SDimitry Andric     "initWithLongLong",
3440b57cec5SDimitry Andric     "initWithUnsignedLongLong",
3450b57cec5SDimitry Andric     "initWithFloat",
3460b57cec5SDimitry Andric     "initWithDouble",
3470b57cec5SDimitry Andric     "initWithBool",
3480b57cec5SDimitry Andric     "initWithInteger",
3490b57cec5SDimitry Andric     "initWithUnsignedInteger"
3500b57cec5SDimitry Andric   };
3510b57cec5SDimitry Andric 
3520b57cec5SDimitry Andric   Selector *Sels;
3530b57cec5SDimitry Andric   const char **Names;
3540b57cec5SDimitry Andric   if (Instance) {
3550b57cec5SDimitry Andric     Sels = NSNumberInstanceSelectors;
3560b57cec5SDimitry Andric     Names = InstanceSelectorName;
3570b57cec5SDimitry Andric   } else {
3580b57cec5SDimitry Andric     Sels = NSNumberClassSelectors;
3590b57cec5SDimitry Andric     Names = ClassSelectorName;
3600b57cec5SDimitry Andric   }
3610b57cec5SDimitry Andric 
3620b57cec5SDimitry Andric   if (Sels[MK].isNull())
3630b57cec5SDimitry Andric     Sels[MK] = Ctx.Selectors.getUnarySelector(&Ctx.Idents.get(Names[MK]));
3640b57cec5SDimitry Andric   return Sels[MK];
3650b57cec5SDimitry Andric }
3660b57cec5SDimitry Andric 
367*bdd1243dSDimitry Andric std::optional<NSAPI::NSNumberLiteralMethodKind>
3680b57cec5SDimitry Andric NSAPI::getNSNumberLiteralMethodKind(Selector Sel) const {
3690b57cec5SDimitry Andric   for (unsigned i = 0; i != NumNSNumberLiteralMethods; ++i) {
3700b57cec5SDimitry Andric     NSNumberLiteralMethodKind MK = NSNumberLiteralMethodKind(i);
3710b57cec5SDimitry Andric     if (isNSNumberLiteralSelector(MK, Sel))
3720b57cec5SDimitry Andric       return MK;
3730b57cec5SDimitry Andric   }
3740b57cec5SDimitry Andric 
375*bdd1243dSDimitry Andric   return std::nullopt;
3760b57cec5SDimitry Andric }
3770b57cec5SDimitry Andric 
378*bdd1243dSDimitry Andric std::optional<NSAPI::NSNumberLiteralMethodKind>
3790b57cec5SDimitry Andric NSAPI::getNSNumberFactoryMethodKind(QualType T) const {
3800b57cec5SDimitry Andric   const BuiltinType *BT = T->getAs<BuiltinType>();
3810b57cec5SDimitry Andric   if (!BT)
382*bdd1243dSDimitry Andric     return std::nullopt;
3830b57cec5SDimitry Andric 
3840b57cec5SDimitry Andric   const TypedefType *TDT = T->getAs<TypedefType>();
3850b57cec5SDimitry Andric   if (TDT) {
3860b57cec5SDimitry Andric     QualType TDTTy = QualType(TDT, 0);
3870b57cec5SDimitry Andric     if (isObjCBOOLType(TDTTy))
3880b57cec5SDimitry Andric       return NSAPI::NSNumberWithBool;
3890b57cec5SDimitry Andric     if (isObjCNSIntegerType(TDTTy))
3900b57cec5SDimitry Andric       return NSAPI::NSNumberWithInteger;
3910b57cec5SDimitry Andric     if (isObjCNSUIntegerType(TDTTy))
3920b57cec5SDimitry Andric       return NSAPI::NSNumberWithUnsignedInteger;
3930b57cec5SDimitry Andric   }
3940b57cec5SDimitry Andric 
3950b57cec5SDimitry Andric   switch (BT->getKind()) {
3960b57cec5SDimitry Andric   case BuiltinType::Char_S:
3970b57cec5SDimitry Andric   case BuiltinType::SChar:
3980b57cec5SDimitry Andric     return NSAPI::NSNumberWithChar;
3990b57cec5SDimitry Andric   case BuiltinType::Char_U:
4000b57cec5SDimitry Andric   case BuiltinType::UChar:
4010b57cec5SDimitry Andric     return NSAPI::NSNumberWithUnsignedChar;
4020b57cec5SDimitry Andric   case BuiltinType::Short:
4030b57cec5SDimitry Andric     return NSAPI::NSNumberWithShort;
4040b57cec5SDimitry Andric   case BuiltinType::UShort:
4050b57cec5SDimitry Andric     return NSAPI::NSNumberWithUnsignedShort;
4060b57cec5SDimitry Andric   case BuiltinType::Int:
4070b57cec5SDimitry Andric     return NSAPI::NSNumberWithInt;
4080b57cec5SDimitry Andric   case BuiltinType::UInt:
4090b57cec5SDimitry Andric     return NSAPI::NSNumberWithUnsignedInt;
4100b57cec5SDimitry Andric   case BuiltinType::Long:
4110b57cec5SDimitry Andric     return NSAPI::NSNumberWithLong;
4120b57cec5SDimitry Andric   case BuiltinType::ULong:
4130b57cec5SDimitry Andric     return NSAPI::NSNumberWithUnsignedLong;
4140b57cec5SDimitry Andric   case BuiltinType::LongLong:
4150b57cec5SDimitry Andric     return NSAPI::NSNumberWithLongLong;
4160b57cec5SDimitry Andric   case BuiltinType::ULongLong:
4170b57cec5SDimitry Andric     return NSAPI::NSNumberWithUnsignedLongLong;
4180b57cec5SDimitry Andric   case BuiltinType::Float:
4190b57cec5SDimitry Andric     return NSAPI::NSNumberWithFloat;
4200b57cec5SDimitry Andric   case BuiltinType::Double:
4210b57cec5SDimitry Andric     return NSAPI::NSNumberWithDouble;
4220b57cec5SDimitry Andric   case BuiltinType::Bool:
4230b57cec5SDimitry Andric     return NSAPI::NSNumberWithBool;
4240b57cec5SDimitry Andric 
4250b57cec5SDimitry Andric   case BuiltinType::Void:
4260b57cec5SDimitry Andric   case BuiltinType::WChar_U:
4270b57cec5SDimitry Andric   case BuiltinType::WChar_S:
4280b57cec5SDimitry Andric   case BuiltinType::Char8:
4290b57cec5SDimitry Andric   case BuiltinType::Char16:
4300b57cec5SDimitry Andric   case BuiltinType::Char32:
4310b57cec5SDimitry Andric   case BuiltinType::Int128:
4320b57cec5SDimitry Andric   case BuiltinType::LongDouble:
4330b57cec5SDimitry Andric   case BuiltinType::ShortAccum:
4340b57cec5SDimitry Andric   case BuiltinType::Accum:
4350b57cec5SDimitry Andric   case BuiltinType::LongAccum:
4360b57cec5SDimitry Andric   case BuiltinType::UShortAccum:
4370b57cec5SDimitry Andric   case BuiltinType::UAccum:
4380b57cec5SDimitry Andric   case BuiltinType::ULongAccum:
4390b57cec5SDimitry Andric   case BuiltinType::ShortFract:
4400b57cec5SDimitry Andric   case BuiltinType::Fract:
4410b57cec5SDimitry Andric   case BuiltinType::LongFract:
4420b57cec5SDimitry Andric   case BuiltinType::UShortFract:
4430b57cec5SDimitry Andric   case BuiltinType::UFract:
4440b57cec5SDimitry Andric   case BuiltinType::ULongFract:
4450b57cec5SDimitry Andric   case BuiltinType::SatShortAccum:
4460b57cec5SDimitry Andric   case BuiltinType::SatAccum:
4470b57cec5SDimitry Andric   case BuiltinType::SatLongAccum:
4480b57cec5SDimitry Andric   case BuiltinType::SatUShortAccum:
4490b57cec5SDimitry Andric   case BuiltinType::SatUAccum:
4500b57cec5SDimitry Andric   case BuiltinType::SatULongAccum:
4510b57cec5SDimitry Andric   case BuiltinType::SatShortFract:
4520b57cec5SDimitry Andric   case BuiltinType::SatFract:
4530b57cec5SDimitry Andric   case BuiltinType::SatLongFract:
4540b57cec5SDimitry Andric   case BuiltinType::SatUShortFract:
4550b57cec5SDimitry Andric   case BuiltinType::SatUFract:
4560b57cec5SDimitry Andric   case BuiltinType::SatULongFract:
4570b57cec5SDimitry Andric   case BuiltinType::UInt128:
4580b57cec5SDimitry Andric   case BuiltinType::Float16:
4590b57cec5SDimitry Andric   case BuiltinType::Float128:
460349cc55cSDimitry Andric   case BuiltinType::Ibm128:
4610b57cec5SDimitry Andric   case BuiltinType::NullPtr:
4620b57cec5SDimitry Andric   case BuiltinType::ObjCClass:
4630b57cec5SDimitry Andric   case BuiltinType::ObjCId:
4640b57cec5SDimitry Andric   case BuiltinType::ObjCSel:
4650b57cec5SDimitry Andric #define IMAGE_TYPE(ImgType, Id, SingletonId, Access, Suffix) \
4660b57cec5SDimitry Andric   case BuiltinType::Id:
4670b57cec5SDimitry Andric #include "clang/Basic/OpenCLImageTypes.def"
4680b57cec5SDimitry Andric #define EXT_OPAQUE_TYPE(ExtType, Id, Ext) \
4690b57cec5SDimitry Andric   case BuiltinType::Id:
4700b57cec5SDimitry Andric #include "clang/Basic/OpenCLExtensionTypes.def"
4710b57cec5SDimitry Andric   case BuiltinType::OCLSampler:
4720b57cec5SDimitry Andric   case BuiltinType::OCLEvent:
4730b57cec5SDimitry Andric   case BuiltinType::OCLClkEvent:
4740b57cec5SDimitry Andric   case BuiltinType::OCLQueue:
4750b57cec5SDimitry Andric   case BuiltinType::OCLReserveID:
476a7dea167SDimitry Andric #define SVE_TYPE(Name, Id, SingletonId) \
477a7dea167SDimitry Andric   case BuiltinType::Id:
478a7dea167SDimitry Andric #include "clang/Basic/AArch64SVEACLETypes.def"
479e8d8bef9SDimitry Andric #define PPC_VECTOR_TYPE(Name, Id, Size) \
480e8d8bef9SDimitry Andric   case BuiltinType::Id:
481e8d8bef9SDimitry Andric #include "clang/Basic/PPCTypes.def"
482fe6060f1SDimitry Andric #define RVV_TYPE(Name, Id, SingletonId) case BuiltinType::Id:
483fe6060f1SDimitry Andric #include "clang/Basic/RISCVVTypes.def"
4840b57cec5SDimitry Andric   case BuiltinType::BoundMember:
4850b57cec5SDimitry Andric   case BuiltinType::Dependent:
4860b57cec5SDimitry Andric   case BuiltinType::Overload:
4870b57cec5SDimitry Andric   case BuiltinType::UnknownAny:
4880b57cec5SDimitry Andric   case BuiltinType::ARCUnbridgedCast:
4890b57cec5SDimitry Andric   case BuiltinType::Half:
4900b57cec5SDimitry Andric   case BuiltinType::PseudoObject:
4910b57cec5SDimitry Andric   case BuiltinType::BuiltinFn:
4925ffd83dbSDimitry Andric   case BuiltinType::IncompleteMatrixIdx:
4930b57cec5SDimitry Andric   case BuiltinType::OMPArraySection:
4945ffd83dbSDimitry Andric   case BuiltinType::OMPArrayShaping:
4955ffd83dbSDimitry Andric   case BuiltinType::OMPIterator:
4965ffd83dbSDimitry Andric   case BuiltinType::BFloat16:
4970b57cec5SDimitry Andric     break;
4980b57cec5SDimitry Andric   }
4990b57cec5SDimitry Andric 
500*bdd1243dSDimitry Andric   return std::nullopt;
5010b57cec5SDimitry Andric }
5020b57cec5SDimitry Andric 
5030b57cec5SDimitry Andric /// Returns true if \param T is a typedef of "BOOL" in objective-c.
5040b57cec5SDimitry Andric bool NSAPI::isObjCBOOLType(QualType T) const {
5050b57cec5SDimitry Andric   return isObjCTypedef(T, "BOOL", BOOLId);
5060b57cec5SDimitry Andric }
5070b57cec5SDimitry Andric /// Returns true if \param T is a typedef of "NSInteger" in objective-c.
5080b57cec5SDimitry Andric bool NSAPI::isObjCNSIntegerType(QualType T) const {
5090b57cec5SDimitry Andric   return isObjCTypedef(T, "NSInteger", NSIntegerId);
5100b57cec5SDimitry Andric }
5110b57cec5SDimitry Andric /// Returns true if \param T is a typedef of "NSUInteger" in objective-c.
5120b57cec5SDimitry Andric bool NSAPI::isObjCNSUIntegerType(QualType T) const {
5130b57cec5SDimitry Andric   return isObjCTypedef(T, "NSUInteger", NSUIntegerId);
5140b57cec5SDimitry Andric }
5150b57cec5SDimitry Andric 
5160b57cec5SDimitry Andric StringRef NSAPI::GetNSIntegralKind(QualType T) const {
5170b57cec5SDimitry Andric   if (!Ctx.getLangOpts().ObjC || T.isNull())
5180b57cec5SDimitry Andric     return StringRef();
5190b57cec5SDimitry Andric 
5200b57cec5SDimitry Andric   while (const TypedefType *TDT = T->getAs<TypedefType>()) {
5210b57cec5SDimitry Andric     StringRef NSIntegralResust =
5220b57cec5SDimitry Andric       llvm::StringSwitch<StringRef>(
5230b57cec5SDimitry Andric         TDT->getDecl()->getDeclName().getAsIdentifierInfo()->getName())
5240b57cec5SDimitry Andric     .Case("int8_t", "int8_t")
5250b57cec5SDimitry Andric     .Case("int16_t", "int16_t")
5260b57cec5SDimitry Andric     .Case("int32_t", "int32_t")
5270b57cec5SDimitry Andric     .Case("NSInteger", "NSInteger")
5280b57cec5SDimitry Andric     .Case("int64_t", "int64_t")
5290b57cec5SDimitry Andric     .Case("uint8_t", "uint8_t")
5300b57cec5SDimitry Andric     .Case("uint16_t", "uint16_t")
5310b57cec5SDimitry Andric     .Case("uint32_t", "uint32_t")
5320b57cec5SDimitry Andric     .Case("NSUInteger", "NSUInteger")
5330b57cec5SDimitry Andric     .Case("uint64_t", "uint64_t")
5340b57cec5SDimitry Andric     .Default(StringRef());
5350b57cec5SDimitry Andric     if (!NSIntegralResust.empty())
5360b57cec5SDimitry Andric       return NSIntegralResust;
5370b57cec5SDimitry Andric     T = TDT->desugar();
5380b57cec5SDimitry Andric   }
5390b57cec5SDimitry Andric   return StringRef();
5400b57cec5SDimitry Andric }
5410b57cec5SDimitry Andric 
5420b57cec5SDimitry Andric bool NSAPI::isMacroDefined(StringRef Id) const {
5430b57cec5SDimitry Andric   // FIXME: Check whether the relevant module macros are visible.
5440b57cec5SDimitry Andric   return Ctx.Idents.get(Id).hasMacroDefinition();
5450b57cec5SDimitry Andric }
5460b57cec5SDimitry Andric 
5470b57cec5SDimitry Andric bool NSAPI::isSubclassOfNSClass(ObjCInterfaceDecl *InterfaceDecl,
5480b57cec5SDimitry Andric                                 NSClassIdKindKind NSClassKind) const {
5490b57cec5SDimitry Andric   if (!InterfaceDecl) {
5500b57cec5SDimitry Andric     return false;
5510b57cec5SDimitry Andric   }
5520b57cec5SDimitry Andric 
5530b57cec5SDimitry Andric   IdentifierInfo *NSClassID = getNSClassId(NSClassKind);
5540b57cec5SDimitry Andric 
5550b57cec5SDimitry Andric   bool IsSubclass = false;
5560b57cec5SDimitry Andric   do {
5570b57cec5SDimitry Andric     IsSubclass = NSClassID == InterfaceDecl->getIdentifier();
5580b57cec5SDimitry Andric 
5590b57cec5SDimitry Andric     if (IsSubclass) {
5600b57cec5SDimitry Andric       break;
5610b57cec5SDimitry Andric     }
5620b57cec5SDimitry Andric   } while ((InterfaceDecl = InterfaceDecl->getSuperClass()));
5630b57cec5SDimitry Andric 
5640b57cec5SDimitry Andric   return IsSubclass;
5650b57cec5SDimitry Andric }
5660b57cec5SDimitry Andric 
5670b57cec5SDimitry Andric bool NSAPI::isObjCTypedef(QualType T,
5680b57cec5SDimitry Andric                           StringRef name, IdentifierInfo *&II) const {
5690b57cec5SDimitry Andric   if (!Ctx.getLangOpts().ObjC)
5700b57cec5SDimitry Andric     return false;
5710b57cec5SDimitry Andric   if (T.isNull())
5720b57cec5SDimitry Andric     return false;
5730b57cec5SDimitry Andric 
5740b57cec5SDimitry Andric   if (!II)
5750b57cec5SDimitry Andric     II = &Ctx.Idents.get(name);
5760b57cec5SDimitry Andric 
5770b57cec5SDimitry Andric   while (const TypedefType *TDT = T->getAs<TypedefType>()) {
5780b57cec5SDimitry Andric     if (TDT->getDecl()->getDeclName().getAsIdentifierInfo() == II)
5790b57cec5SDimitry Andric       return true;
5800b57cec5SDimitry Andric     T = TDT->desugar();
5810b57cec5SDimitry Andric   }
5820b57cec5SDimitry Andric 
5830b57cec5SDimitry Andric   return false;
5840b57cec5SDimitry Andric }
5850b57cec5SDimitry Andric 
5860b57cec5SDimitry Andric bool NSAPI::isObjCEnumerator(const Expr *E,
5870b57cec5SDimitry Andric                              StringRef name, IdentifierInfo *&II) const {
5880b57cec5SDimitry Andric   if (!Ctx.getLangOpts().ObjC)
5890b57cec5SDimitry Andric     return false;
5900b57cec5SDimitry Andric   if (!E)
5910b57cec5SDimitry Andric     return false;
5920b57cec5SDimitry Andric 
5930b57cec5SDimitry Andric   if (!II)
5940b57cec5SDimitry Andric     II = &Ctx.Idents.get(name);
5950b57cec5SDimitry Andric 
5960b57cec5SDimitry Andric   if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E->IgnoreParenImpCasts()))
5970b57cec5SDimitry Andric     if (const EnumConstantDecl *
5980b57cec5SDimitry Andric           EnumD = dyn_cast_or_null<EnumConstantDecl>(DRE->getDecl()))
5990b57cec5SDimitry Andric       return EnumD->getIdentifier() == II;
6000b57cec5SDimitry Andric 
6010b57cec5SDimitry Andric   return false;
6020b57cec5SDimitry Andric }
6030b57cec5SDimitry Andric 
6040b57cec5SDimitry Andric Selector NSAPI::getOrInitSelector(ArrayRef<StringRef> Ids,
6050b57cec5SDimitry Andric                                   Selector &Sel) const {
6060b57cec5SDimitry Andric   if (Sel.isNull()) {
6070b57cec5SDimitry Andric     SmallVector<IdentifierInfo *, 4> Idents;
6080b57cec5SDimitry Andric     for (ArrayRef<StringRef>::const_iterator
6090b57cec5SDimitry Andric            I = Ids.begin(), E = Ids.end(); I != E; ++I)
6100b57cec5SDimitry Andric       Idents.push_back(&Ctx.Idents.get(*I));
6110b57cec5SDimitry Andric     Sel = Ctx.Selectors.getSelector(Idents.size(), Idents.data());
6120b57cec5SDimitry Andric   }
6130b57cec5SDimitry Andric   return Sel;
6140b57cec5SDimitry Andric }
6150b57cec5SDimitry Andric 
6160b57cec5SDimitry Andric Selector NSAPI::getOrInitNullarySelector(StringRef Id, Selector &Sel) const {
6170b57cec5SDimitry Andric   if (Sel.isNull()) {
6180b57cec5SDimitry Andric     IdentifierInfo *Ident = &Ctx.Idents.get(Id);
6190b57cec5SDimitry Andric     Sel = Ctx.Selectors.getSelector(0, &Ident);
6200b57cec5SDimitry Andric   }
6210b57cec5SDimitry Andric   return Sel;
6220b57cec5SDimitry Andric }
623