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