xref: /aosp_15_r20/external/clang/lib/Sema/SemaExprObjC.cpp (revision 67e74705e28f6214e480b399dd47ea732279e315)
1*67e74705SXin Li //===--- SemaExprObjC.cpp - Semantic Analysis for ObjC Expressions --------===//
2*67e74705SXin Li //
3*67e74705SXin Li //                     The LLVM Compiler Infrastructure
4*67e74705SXin Li //
5*67e74705SXin Li // This file is distributed under the University of Illinois Open Source
6*67e74705SXin Li // License. See LICENSE.TXT for details.
7*67e74705SXin Li //
8*67e74705SXin Li //===----------------------------------------------------------------------===//
9*67e74705SXin Li //
10*67e74705SXin Li //  This file implements semantic analysis for Objective-C expressions.
11*67e74705SXin Li //
12*67e74705SXin Li //===----------------------------------------------------------------------===//
13*67e74705SXin Li 
14*67e74705SXin Li #include "clang/Sema/SemaInternal.h"
15*67e74705SXin Li #include "clang/AST/ASTContext.h"
16*67e74705SXin Li #include "clang/AST/DeclObjC.h"
17*67e74705SXin Li #include "clang/AST/ExprObjC.h"
18*67e74705SXin Li #include "clang/AST/StmtVisitor.h"
19*67e74705SXin Li #include "clang/AST/TypeLoc.h"
20*67e74705SXin Li #include "clang/Analysis/DomainSpecific/CocoaConventions.h"
21*67e74705SXin Li #include "clang/Edit/Commit.h"
22*67e74705SXin Li #include "clang/Edit/Rewriters.h"
23*67e74705SXin Li #include "clang/Lex/Preprocessor.h"
24*67e74705SXin Li #include "clang/Sema/Initialization.h"
25*67e74705SXin Li #include "clang/Sema/Lookup.h"
26*67e74705SXin Li #include "clang/Sema/Scope.h"
27*67e74705SXin Li #include "clang/Sema/ScopeInfo.h"
28*67e74705SXin Li #include "llvm/ADT/SmallString.h"
29*67e74705SXin Li 
30*67e74705SXin Li using namespace clang;
31*67e74705SXin Li using namespace sema;
32*67e74705SXin Li using llvm::makeArrayRef;
33*67e74705SXin Li 
ParseObjCStringLiteral(SourceLocation * AtLocs,ArrayRef<Expr * > Strings)34*67e74705SXin Li ExprResult Sema::ParseObjCStringLiteral(SourceLocation *AtLocs,
35*67e74705SXin Li                                         ArrayRef<Expr *> Strings) {
36*67e74705SXin Li   // Most ObjC strings are formed out of a single piece.  However, we *can*
37*67e74705SXin Li   // have strings formed out of multiple @ strings with multiple pptokens in
38*67e74705SXin Li   // each one, e.g. @"foo" "bar" @"baz" "qux"   which need to be turned into one
39*67e74705SXin Li   // StringLiteral for ObjCStringLiteral to hold onto.
40*67e74705SXin Li   StringLiteral *S = cast<StringLiteral>(Strings[0]);
41*67e74705SXin Li 
42*67e74705SXin Li   // If we have a multi-part string, merge it all together.
43*67e74705SXin Li   if (Strings.size() != 1) {
44*67e74705SXin Li     // Concatenate objc strings.
45*67e74705SXin Li     SmallString<128> StrBuf;
46*67e74705SXin Li     SmallVector<SourceLocation, 8> StrLocs;
47*67e74705SXin Li 
48*67e74705SXin Li     for (Expr *E : Strings) {
49*67e74705SXin Li       S = cast<StringLiteral>(E);
50*67e74705SXin Li 
51*67e74705SXin Li       // ObjC strings can't be wide or UTF.
52*67e74705SXin Li       if (!S->isAscii()) {
53*67e74705SXin Li         Diag(S->getLocStart(), diag::err_cfstring_literal_not_string_constant)
54*67e74705SXin Li           << S->getSourceRange();
55*67e74705SXin Li         return true;
56*67e74705SXin Li       }
57*67e74705SXin Li 
58*67e74705SXin Li       // Append the string.
59*67e74705SXin Li       StrBuf += S->getString();
60*67e74705SXin Li 
61*67e74705SXin Li       // Get the locations of the string tokens.
62*67e74705SXin Li       StrLocs.append(S->tokloc_begin(), S->tokloc_end());
63*67e74705SXin Li     }
64*67e74705SXin Li 
65*67e74705SXin Li     // Create the aggregate string with the appropriate content and location
66*67e74705SXin Li     // information.
67*67e74705SXin Li     const ConstantArrayType *CAT = Context.getAsConstantArrayType(S->getType());
68*67e74705SXin Li     assert(CAT && "String literal not of constant array type!");
69*67e74705SXin Li     QualType StrTy = Context.getConstantArrayType(
70*67e74705SXin Li         CAT->getElementType(), llvm::APInt(32, StrBuf.size() + 1),
71*67e74705SXin Li         CAT->getSizeModifier(), CAT->getIndexTypeCVRQualifiers());
72*67e74705SXin Li     S = StringLiteral::Create(Context, StrBuf, StringLiteral::Ascii,
73*67e74705SXin Li                               /*Pascal=*/false, StrTy, &StrLocs[0],
74*67e74705SXin Li                               StrLocs.size());
75*67e74705SXin Li   }
76*67e74705SXin Li 
77*67e74705SXin Li   return BuildObjCStringLiteral(AtLocs[0], S);
78*67e74705SXin Li }
79*67e74705SXin Li 
BuildObjCStringLiteral(SourceLocation AtLoc,StringLiteral * S)80*67e74705SXin Li ExprResult Sema::BuildObjCStringLiteral(SourceLocation AtLoc, StringLiteral *S){
81*67e74705SXin Li   // Verify that this composite string is acceptable for ObjC strings.
82*67e74705SXin Li   if (CheckObjCString(S))
83*67e74705SXin Li     return true;
84*67e74705SXin Li 
85*67e74705SXin Li   // Initialize the constant string interface lazily. This assumes
86*67e74705SXin Li   // the NSString interface is seen in this translation unit. Note: We
87*67e74705SXin Li   // don't use NSConstantString, since the runtime team considers this
88*67e74705SXin Li   // interface private (even though it appears in the header files).
89*67e74705SXin Li   QualType Ty = Context.getObjCConstantStringInterface();
90*67e74705SXin Li   if (!Ty.isNull()) {
91*67e74705SXin Li     Ty = Context.getObjCObjectPointerType(Ty);
92*67e74705SXin Li   } else if (getLangOpts().NoConstantCFStrings) {
93*67e74705SXin Li     IdentifierInfo *NSIdent=nullptr;
94*67e74705SXin Li     std::string StringClass(getLangOpts().ObjCConstantStringClass);
95*67e74705SXin Li 
96*67e74705SXin Li     if (StringClass.empty())
97*67e74705SXin Li       NSIdent = &Context.Idents.get("NSConstantString");
98*67e74705SXin Li     else
99*67e74705SXin Li       NSIdent = &Context.Idents.get(StringClass);
100*67e74705SXin Li 
101*67e74705SXin Li     NamedDecl *IF = LookupSingleName(TUScope, NSIdent, AtLoc,
102*67e74705SXin Li                                      LookupOrdinaryName);
103*67e74705SXin Li     if (ObjCInterfaceDecl *StrIF = dyn_cast_or_null<ObjCInterfaceDecl>(IF)) {
104*67e74705SXin Li       Context.setObjCConstantStringInterface(StrIF);
105*67e74705SXin Li       Ty = Context.getObjCConstantStringInterface();
106*67e74705SXin Li       Ty = Context.getObjCObjectPointerType(Ty);
107*67e74705SXin Li     } else {
108*67e74705SXin Li       // If there is no NSConstantString interface defined then treat this
109*67e74705SXin Li       // as error and recover from it.
110*67e74705SXin Li       Diag(S->getLocStart(), diag::err_no_nsconstant_string_class) << NSIdent
111*67e74705SXin Li         << S->getSourceRange();
112*67e74705SXin Li       Ty = Context.getObjCIdType();
113*67e74705SXin Li     }
114*67e74705SXin Li   } else {
115*67e74705SXin Li     IdentifierInfo *NSIdent = NSAPIObj->getNSClassId(NSAPI::ClassId_NSString);
116*67e74705SXin Li     NamedDecl *IF = LookupSingleName(TUScope, NSIdent, AtLoc,
117*67e74705SXin Li                                      LookupOrdinaryName);
118*67e74705SXin Li     if (ObjCInterfaceDecl *StrIF = dyn_cast_or_null<ObjCInterfaceDecl>(IF)) {
119*67e74705SXin Li       Context.setObjCConstantStringInterface(StrIF);
120*67e74705SXin Li       Ty = Context.getObjCConstantStringInterface();
121*67e74705SXin Li       Ty = Context.getObjCObjectPointerType(Ty);
122*67e74705SXin Li     } else {
123*67e74705SXin Li       // If there is no NSString interface defined, implicitly declare
124*67e74705SXin Li       // a @class NSString; and use that instead. This is to make sure
125*67e74705SXin Li       // type of an NSString literal is represented correctly, instead of
126*67e74705SXin Li       // being an 'id' type.
127*67e74705SXin Li       Ty = Context.getObjCNSStringType();
128*67e74705SXin Li       if (Ty.isNull()) {
129*67e74705SXin Li         ObjCInterfaceDecl *NSStringIDecl =
130*67e74705SXin Li           ObjCInterfaceDecl::Create (Context,
131*67e74705SXin Li                                      Context.getTranslationUnitDecl(),
132*67e74705SXin Li                                      SourceLocation(), NSIdent,
133*67e74705SXin Li                                      nullptr, nullptr, SourceLocation());
134*67e74705SXin Li         Ty = Context.getObjCInterfaceType(NSStringIDecl);
135*67e74705SXin Li         Context.setObjCNSStringType(Ty);
136*67e74705SXin Li       }
137*67e74705SXin Li       Ty = Context.getObjCObjectPointerType(Ty);
138*67e74705SXin Li     }
139*67e74705SXin Li   }
140*67e74705SXin Li 
141*67e74705SXin Li   return new (Context) ObjCStringLiteral(S, Ty, AtLoc);
142*67e74705SXin Li }
143*67e74705SXin Li 
144*67e74705SXin Li /// \brief Emits an error if the given method does not exist, or if the return
145*67e74705SXin Li /// type is not an Objective-C object.
validateBoxingMethod(Sema & S,SourceLocation Loc,const ObjCInterfaceDecl * Class,Selector Sel,const ObjCMethodDecl * Method)146*67e74705SXin Li static bool validateBoxingMethod(Sema &S, SourceLocation Loc,
147*67e74705SXin Li                                  const ObjCInterfaceDecl *Class,
148*67e74705SXin Li                                  Selector Sel, const ObjCMethodDecl *Method) {
149*67e74705SXin Li   if (!Method) {
150*67e74705SXin Li     // FIXME: Is there a better way to avoid quotes than using getName()?
151*67e74705SXin Li     S.Diag(Loc, diag::err_undeclared_boxing_method) << Sel << Class->getName();
152*67e74705SXin Li     return false;
153*67e74705SXin Li   }
154*67e74705SXin Li 
155*67e74705SXin Li   // Make sure the return type is reasonable.
156*67e74705SXin Li   QualType ReturnType = Method->getReturnType();
157*67e74705SXin Li   if (!ReturnType->isObjCObjectPointerType()) {
158*67e74705SXin Li     S.Diag(Loc, diag::err_objc_literal_method_sig)
159*67e74705SXin Li       << Sel;
160*67e74705SXin Li     S.Diag(Method->getLocation(), diag::note_objc_literal_method_return)
161*67e74705SXin Li       << ReturnType;
162*67e74705SXin Li     return false;
163*67e74705SXin Li   }
164*67e74705SXin Li 
165*67e74705SXin Li   return true;
166*67e74705SXin Li }
167*67e74705SXin Li 
168*67e74705SXin Li /// \brief Maps ObjCLiteralKind to NSClassIdKindKind
ClassKindFromLiteralKind(Sema::ObjCLiteralKind LiteralKind)169*67e74705SXin Li static NSAPI::NSClassIdKindKind ClassKindFromLiteralKind(
170*67e74705SXin Li                                             Sema::ObjCLiteralKind LiteralKind) {
171*67e74705SXin Li   switch (LiteralKind) {
172*67e74705SXin Li     case Sema::LK_Array:
173*67e74705SXin Li       return NSAPI::ClassId_NSArray;
174*67e74705SXin Li     case Sema::LK_Dictionary:
175*67e74705SXin Li       return NSAPI::ClassId_NSDictionary;
176*67e74705SXin Li     case Sema::LK_Numeric:
177*67e74705SXin Li       return NSAPI::ClassId_NSNumber;
178*67e74705SXin Li     case Sema::LK_String:
179*67e74705SXin Li       return NSAPI::ClassId_NSString;
180*67e74705SXin Li     case Sema::LK_Boxed:
181*67e74705SXin Li       return NSAPI::ClassId_NSValue;
182*67e74705SXin Li 
183*67e74705SXin Li     // there is no corresponding matching
184*67e74705SXin Li     // between LK_None/LK_Block and NSClassIdKindKind
185*67e74705SXin Li     case Sema::LK_Block:
186*67e74705SXin Li     case Sema::LK_None:
187*67e74705SXin Li       break;
188*67e74705SXin Li   }
189*67e74705SXin Li   llvm_unreachable("LiteralKind can't be converted into a ClassKind");
190*67e74705SXin Li }
191*67e74705SXin Li 
192*67e74705SXin Li /// \brief Validates ObjCInterfaceDecl availability.
193*67e74705SXin Li /// ObjCInterfaceDecl, used to create ObjC literals, should be defined
194*67e74705SXin Li /// if clang not in a debugger mode.
ValidateObjCLiteralInterfaceDecl(Sema & S,ObjCInterfaceDecl * Decl,SourceLocation Loc,Sema::ObjCLiteralKind LiteralKind)195*67e74705SXin Li static bool ValidateObjCLiteralInterfaceDecl(Sema &S, ObjCInterfaceDecl *Decl,
196*67e74705SXin Li                                             SourceLocation Loc,
197*67e74705SXin Li                                             Sema::ObjCLiteralKind LiteralKind) {
198*67e74705SXin Li   if (!Decl) {
199*67e74705SXin Li     NSAPI::NSClassIdKindKind Kind = ClassKindFromLiteralKind(LiteralKind);
200*67e74705SXin Li     IdentifierInfo *II = S.NSAPIObj->getNSClassId(Kind);
201*67e74705SXin Li     S.Diag(Loc, diag::err_undeclared_objc_literal_class)
202*67e74705SXin Li       << II->getName() << LiteralKind;
203*67e74705SXin Li     return false;
204*67e74705SXin Li   } else if (!Decl->hasDefinition() && !S.getLangOpts().DebuggerObjCLiteral) {
205*67e74705SXin Li     S.Diag(Loc, diag::err_undeclared_objc_literal_class)
206*67e74705SXin Li       << Decl->getName() << LiteralKind;
207*67e74705SXin Li     S.Diag(Decl->getLocation(), diag::note_forward_class);
208*67e74705SXin Li     return false;
209*67e74705SXin Li   }
210*67e74705SXin Li 
211*67e74705SXin Li   return true;
212*67e74705SXin Li }
213*67e74705SXin Li 
214*67e74705SXin Li /// \brief Looks up ObjCInterfaceDecl of a given NSClassIdKindKind.
215*67e74705SXin Li /// Used to create ObjC literals, such as NSDictionary (@{}),
216*67e74705SXin Li /// NSArray (@[]) and Boxed Expressions (@())
LookupObjCInterfaceDeclForLiteral(Sema & S,SourceLocation Loc,Sema::ObjCLiteralKind LiteralKind)217*67e74705SXin Li static ObjCInterfaceDecl *LookupObjCInterfaceDeclForLiteral(Sema &S,
218*67e74705SXin Li                                             SourceLocation Loc,
219*67e74705SXin Li                                             Sema::ObjCLiteralKind LiteralKind) {
220*67e74705SXin Li   NSAPI::NSClassIdKindKind ClassKind = ClassKindFromLiteralKind(LiteralKind);
221*67e74705SXin Li   IdentifierInfo *II = S.NSAPIObj->getNSClassId(ClassKind);
222*67e74705SXin Li   NamedDecl *IF = S.LookupSingleName(S.TUScope, II, Loc,
223*67e74705SXin Li                                      Sema::LookupOrdinaryName);
224*67e74705SXin Li   ObjCInterfaceDecl *ID = dyn_cast_or_null<ObjCInterfaceDecl>(IF);
225*67e74705SXin Li   if (!ID && S.getLangOpts().DebuggerObjCLiteral) {
226*67e74705SXin Li     ASTContext &Context = S.Context;
227*67e74705SXin Li     TranslationUnitDecl *TU = Context.getTranslationUnitDecl();
228*67e74705SXin Li     ID = ObjCInterfaceDecl::Create (Context, TU, SourceLocation(), II,
229*67e74705SXin Li                                     nullptr, nullptr, SourceLocation());
230*67e74705SXin Li   }
231*67e74705SXin Li 
232*67e74705SXin Li   if (!ValidateObjCLiteralInterfaceDecl(S, ID, Loc, LiteralKind)) {
233*67e74705SXin Li     ID = nullptr;
234*67e74705SXin Li   }
235*67e74705SXin Li 
236*67e74705SXin Li   return ID;
237*67e74705SXin Li }
238*67e74705SXin Li 
239*67e74705SXin Li /// \brief Retrieve the NSNumber factory method that should be used to create
240*67e74705SXin Li /// an Objective-C literal for the given type.
getNSNumberFactoryMethod(Sema & S,SourceLocation Loc,QualType NumberType,bool isLiteral=false,SourceRange R=SourceRange ())241*67e74705SXin Li static ObjCMethodDecl *getNSNumberFactoryMethod(Sema &S, SourceLocation Loc,
242*67e74705SXin Li                                                 QualType NumberType,
243*67e74705SXin Li                                                 bool isLiteral = false,
244*67e74705SXin Li                                                 SourceRange R = SourceRange()) {
245*67e74705SXin Li   Optional<NSAPI::NSNumberLiteralMethodKind> Kind =
246*67e74705SXin Li       S.NSAPIObj->getNSNumberFactoryMethodKind(NumberType);
247*67e74705SXin Li 
248*67e74705SXin Li   if (!Kind) {
249*67e74705SXin Li     if (isLiteral) {
250*67e74705SXin Li       S.Diag(Loc, diag::err_invalid_nsnumber_type)
251*67e74705SXin Li         << NumberType << R;
252*67e74705SXin Li     }
253*67e74705SXin Li     return nullptr;
254*67e74705SXin Li   }
255*67e74705SXin Li 
256*67e74705SXin Li   // If we already looked up this method, we're done.
257*67e74705SXin Li   if (S.NSNumberLiteralMethods[*Kind])
258*67e74705SXin Li     return S.NSNumberLiteralMethods[*Kind];
259*67e74705SXin Li 
260*67e74705SXin Li   Selector Sel = S.NSAPIObj->getNSNumberLiteralSelector(*Kind,
261*67e74705SXin Li                                                         /*Instance=*/false);
262*67e74705SXin Li 
263*67e74705SXin Li   ASTContext &CX = S.Context;
264*67e74705SXin Li 
265*67e74705SXin Li   // Look up the NSNumber class, if we haven't done so already. It's cached
266*67e74705SXin Li   // in the Sema instance.
267*67e74705SXin Li   if (!S.NSNumberDecl) {
268*67e74705SXin Li     S.NSNumberDecl = LookupObjCInterfaceDeclForLiteral(S, Loc,
269*67e74705SXin Li                                                        Sema::LK_Numeric);
270*67e74705SXin Li     if (!S.NSNumberDecl) {
271*67e74705SXin Li       return nullptr;
272*67e74705SXin Li     }
273*67e74705SXin Li   }
274*67e74705SXin Li 
275*67e74705SXin Li   if (S.NSNumberPointer.isNull()) {
276*67e74705SXin Li     // generate the pointer to NSNumber type.
277*67e74705SXin Li     QualType NSNumberObject = CX.getObjCInterfaceType(S.NSNumberDecl);
278*67e74705SXin Li     S.NSNumberPointer = CX.getObjCObjectPointerType(NSNumberObject);
279*67e74705SXin Li   }
280*67e74705SXin Li 
281*67e74705SXin Li   // Look for the appropriate method within NSNumber.
282*67e74705SXin Li   ObjCMethodDecl *Method = S.NSNumberDecl->lookupClassMethod(Sel);
283*67e74705SXin Li   if (!Method && S.getLangOpts().DebuggerObjCLiteral) {
284*67e74705SXin Li     // create a stub definition this NSNumber factory method.
285*67e74705SXin Li     TypeSourceInfo *ReturnTInfo = nullptr;
286*67e74705SXin Li     Method =
287*67e74705SXin Li         ObjCMethodDecl::Create(CX, SourceLocation(), SourceLocation(), Sel,
288*67e74705SXin Li                                S.NSNumberPointer, ReturnTInfo, S.NSNumberDecl,
289*67e74705SXin Li                                /*isInstance=*/false, /*isVariadic=*/false,
290*67e74705SXin Li                                /*isPropertyAccessor=*/false,
291*67e74705SXin Li                                /*isImplicitlyDeclared=*/true,
292*67e74705SXin Li                                /*isDefined=*/false, ObjCMethodDecl::Required,
293*67e74705SXin Li                                /*HasRelatedResultType=*/false);
294*67e74705SXin Li     ParmVarDecl *value = ParmVarDecl::Create(S.Context, Method,
295*67e74705SXin Li                                              SourceLocation(), SourceLocation(),
296*67e74705SXin Li                                              &CX.Idents.get("value"),
297*67e74705SXin Li                                              NumberType, /*TInfo=*/nullptr,
298*67e74705SXin Li                                              SC_None, nullptr);
299*67e74705SXin Li     Method->setMethodParams(S.Context, value, None);
300*67e74705SXin Li   }
301*67e74705SXin Li 
302*67e74705SXin Li   if (!validateBoxingMethod(S, Loc, S.NSNumberDecl, Sel, Method))
303*67e74705SXin Li     return nullptr;
304*67e74705SXin Li 
305*67e74705SXin Li   // Note: if the parameter type is out-of-line, we'll catch it later in the
306*67e74705SXin Li   // implicit conversion.
307*67e74705SXin Li 
308*67e74705SXin Li   S.NSNumberLiteralMethods[*Kind] = Method;
309*67e74705SXin Li   return Method;
310*67e74705SXin Li }
311*67e74705SXin Li 
312*67e74705SXin Li /// BuildObjCNumericLiteral - builds an ObjCBoxedExpr AST node for the
313*67e74705SXin Li /// numeric literal expression. Type of the expression will be "NSNumber *".
BuildObjCNumericLiteral(SourceLocation AtLoc,Expr * Number)314*67e74705SXin Li ExprResult Sema::BuildObjCNumericLiteral(SourceLocation AtLoc, Expr *Number) {
315*67e74705SXin Li   // Determine the type of the literal.
316*67e74705SXin Li   QualType NumberType = Number->getType();
317*67e74705SXin Li   if (CharacterLiteral *Char = dyn_cast<CharacterLiteral>(Number)) {
318*67e74705SXin Li     // In C, character literals have type 'int'. That's not the type we want
319*67e74705SXin Li     // to use to determine the Objective-c literal kind.
320*67e74705SXin Li     switch (Char->getKind()) {
321*67e74705SXin Li     case CharacterLiteral::Ascii:
322*67e74705SXin Li     case CharacterLiteral::UTF8:
323*67e74705SXin Li       NumberType = Context.CharTy;
324*67e74705SXin Li       break;
325*67e74705SXin Li 
326*67e74705SXin Li     case CharacterLiteral::Wide:
327*67e74705SXin Li       NumberType = Context.getWideCharType();
328*67e74705SXin Li       break;
329*67e74705SXin Li 
330*67e74705SXin Li     case CharacterLiteral::UTF16:
331*67e74705SXin Li       NumberType = Context.Char16Ty;
332*67e74705SXin Li       break;
333*67e74705SXin Li 
334*67e74705SXin Li     case CharacterLiteral::UTF32:
335*67e74705SXin Li       NumberType = Context.Char32Ty;
336*67e74705SXin Li       break;
337*67e74705SXin Li     }
338*67e74705SXin Li   }
339*67e74705SXin Li 
340*67e74705SXin Li   // Look for the appropriate method within NSNumber.
341*67e74705SXin Li   // Construct the literal.
342*67e74705SXin Li   SourceRange NR(Number->getSourceRange());
343*67e74705SXin Li   ObjCMethodDecl *Method = getNSNumberFactoryMethod(*this, AtLoc, NumberType,
344*67e74705SXin Li                                                     true, NR);
345*67e74705SXin Li   if (!Method)
346*67e74705SXin Li     return ExprError();
347*67e74705SXin Li 
348*67e74705SXin Li   // Convert the number to the type that the parameter expects.
349*67e74705SXin Li   ParmVarDecl *ParamDecl = Method->parameters()[0];
350*67e74705SXin Li   InitializedEntity Entity = InitializedEntity::InitializeParameter(Context,
351*67e74705SXin Li                                                                     ParamDecl);
352*67e74705SXin Li   ExprResult ConvertedNumber = PerformCopyInitialization(Entity,
353*67e74705SXin Li                                                          SourceLocation(),
354*67e74705SXin Li                                                          Number);
355*67e74705SXin Li   if (ConvertedNumber.isInvalid())
356*67e74705SXin Li     return ExprError();
357*67e74705SXin Li   Number = ConvertedNumber.get();
358*67e74705SXin Li 
359*67e74705SXin Li   // Use the effective source range of the literal, including the leading '@'.
360*67e74705SXin Li   return MaybeBindToTemporary(
361*67e74705SXin Li            new (Context) ObjCBoxedExpr(Number, NSNumberPointer, Method,
362*67e74705SXin Li                                        SourceRange(AtLoc, NR.getEnd())));
363*67e74705SXin Li }
364*67e74705SXin Li 
ActOnObjCBoolLiteral(SourceLocation AtLoc,SourceLocation ValueLoc,bool Value)365*67e74705SXin Li ExprResult Sema::ActOnObjCBoolLiteral(SourceLocation AtLoc,
366*67e74705SXin Li                                       SourceLocation ValueLoc,
367*67e74705SXin Li                                       bool Value) {
368*67e74705SXin Li   ExprResult Inner;
369*67e74705SXin Li   if (getLangOpts().CPlusPlus) {
370*67e74705SXin Li     Inner = ActOnCXXBoolLiteral(ValueLoc, Value? tok::kw_true : tok::kw_false);
371*67e74705SXin Li   } else {
372*67e74705SXin Li     // C doesn't actually have a way to represent literal values of type
373*67e74705SXin Li     // _Bool. So, we'll use 0/1 and implicit cast to _Bool.
374*67e74705SXin Li     Inner = ActOnIntegerConstant(ValueLoc, Value? 1 : 0);
375*67e74705SXin Li     Inner = ImpCastExprToType(Inner.get(), Context.BoolTy,
376*67e74705SXin Li                               CK_IntegralToBoolean);
377*67e74705SXin Li   }
378*67e74705SXin Li 
379*67e74705SXin Li   return BuildObjCNumericLiteral(AtLoc, Inner.get());
380*67e74705SXin Li }
381*67e74705SXin Li 
382*67e74705SXin Li /// \brief Check that the given expression is a valid element of an Objective-C
383*67e74705SXin Li /// collection literal.
CheckObjCCollectionLiteralElement(Sema & S,Expr * Element,QualType T,bool ArrayLiteral=false)384*67e74705SXin Li static ExprResult CheckObjCCollectionLiteralElement(Sema &S, Expr *Element,
385*67e74705SXin Li                                                     QualType T,
386*67e74705SXin Li                                                     bool ArrayLiteral = false) {
387*67e74705SXin Li   // If the expression is type-dependent, there's nothing for us to do.
388*67e74705SXin Li   if (Element->isTypeDependent())
389*67e74705SXin Li     return Element;
390*67e74705SXin Li 
391*67e74705SXin Li   ExprResult Result = S.CheckPlaceholderExpr(Element);
392*67e74705SXin Li   if (Result.isInvalid())
393*67e74705SXin Li     return ExprError();
394*67e74705SXin Li   Element = Result.get();
395*67e74705SXin Li 
396*67e74705SXin Li   // In C++, check for an implicit conversion to an Objective-C object pointer
397*67e74705SXin Li   // type.
398*67e74705SXin Li   if (S.getLangOpts().CPlusPlus && Element->getType()->isRecordType()) {
399*67e74705SXin Li     InitializedEntity Entity
400*67e74705SXin Li       = InitializedEntity::InitializeParameter(S.Context, T,
401*67e74705SXin Li                                                /*Consumed=*/false);
402*67e74705SXin Li     InitializationKind Kind
403*67e74705SXin Li       = InitializationKind::CreateCopy(Element->getLocStart(),
404*67e74705SXin Li                                        SourceLocation());
405*67e74705SXin Li     InitializationSequence Seq(S, Entity, Kind, Element);
406*67e74705SXin Li     if (!Seq.Failed())
407*67e74705SXin Li       return Seq.Perform(S, Entity, Kind, Element);
408*67e74705SXin Li   }
409*67e74705SXin Li 
410*67e74705SXin Li   Expr *OrigElement = Element;
411*67e74705SXin Li 
412*67e74705SXin Li   // Perform lvalue-to-rvalue conversion.
413*67e74705SXin Li   Result = S.DefaultLvalueConversion(Element);
414*67e74705SXin Li   if (Result.isInvalid())
415*67e74705SXin Li     return ExprError();
416*67e74705SXin Li   Element = Result.get();
417*67e74705SXin Li 
418*67e74705SXin Li   // Make sure that we have an Objective-C pointer type or block.
419*67e74705SXin Li   if (!Element->getType()->isObjCObjectPointerType() &&
420*67e74705SXin Li       !Element->getType()->isBlockPointerType()) {
421*67e74705SXin Li     bool Recovered = false;
422*67e74705SXin Li 
423*67e74705SXin Li     // If this is potentially an Objective-C numeric literal, add the '@'.
424*67e74705SXin Li     if (isa<IntegerLiteral>(OrigElement) ||
425*67e74705SXin Li         isa<CharacterLiteral>(OrigElement) ||
426*67e74705SXin Li         isa<FloatingLiteral>(OrigElement) ||
427*67e74705SXin Li         isa<ObjCBoolLiteralExpr>(OrigElement) ||
428*67e74705SXin Li         isa<CXXBoolLiteralExpr>(OrigElement)) {
429*67e74705SXin Li       if (S.NSAPIObj->getNSNumberFactoryMethodKind(OrigElement->getType())) {
430*67e74705SXin Li         int Which = isa<CharacterLiteral>(OrigElement) ? 1
431*67e74705SXin Li                   : (isa<CXXBoolLiteralExpr>(OrigElement) ||
432*67e74705SXin Li                      isa<ObjCBoolLiteralExpr>(OrigElement)) ? 2
433*67e74705SXin Li                   : 3;
434*67e74705SXin Li 
435*67e74705SXin Li         S.Diag(OrigElement->getLocStart(), diag::err_box_literal_collection)
436*67e74705SXin Li           << Which << OrigElement->getSourceRange()
437*67e74705SXin Li           << FixItHint::CreateInsertion(OrigElement->getLocStart(), "@");
438*67e74705SXin Li 
439*67e74705SXin Li         Result = S.BuildObjCNumericLiteral(OrigElement->getLocStart(),
440*67e74705SXin Li                                            OrigElement);
441*67e74705SXin Li         if (Result.isInvalid())
442*67e74705SXin Li           return ExprError();
443*67e74705SXin Li 
444*67e74705SXin Li         Element = Result.get();
445*67e74705SXin Li         Recovered = true;
446*67e74705SXin Li       }
447*67e74705SXin Li     }
448*67e74705SXin Li     // If this is potentially an Objective-C string literal, add the '@'.
449*67e74705SXin Li     else if (StringLiteral *String = dyn_cast<StringLiteral>(OrigElement)) {
450*67e74705SXin Li       if (String->isAscii()) {
451*67e74705SXin Li         S.Diag(OrigElement->getLocStart(), diag::err_box_literal_collection)
452*67e74705SXin Li           << 0 << OrigElement->getSourceRange()
453*67e74705SXin Li           << FixItHint::CreateInsertion(OrigElement->getLocStart(), "@");
454*67e74705SXin Li 
455*67e74705SXin Li         Result = S.BuildObjCStringLiteral(OrigElement->getLocStart(), String);
456*67e74705SXin Li         if (Result.isInvalid())
457*67e74705SXin Li           return ExprError();
458*67e74705SXin Li 
459*67e74705SXin Li         Element = Result.get();
460*67e74705SXin Li         Recovered = true;
461*67e74705SXin Li       }
462*67e74705SXin Li     }
463*67e74705SXin Li 
464*67e74705SXin Li     if (!Recovered) {
465*67e74705SXin Li       S.Diag(Element->getLocStart(), diag::err_invalid_collection_element)
466*67e74705SXin Li         << Element->getType();
467*67e74705SXin Li       return ExprError();
468*67e74705SXin Li     }
469*67e74705SXin Li   }
470*67e74705SXin Li   if (ArrayLiteral)
471*67e74705SXin Li     if (ObjCStringLiteral *getString =
472*67e74705SXin Li           dyn_cast<ObjCStringLiteral>(OrigElement)) {
473*67e74705SXin Li       if (StringLiteral *SL = getString->getString()) {
474*67e74705SXin Li         unsigned numConcat = SL->getNumConcatenated();
475*67e74705SXin Li         if (numConcat > 1) {
476*67e74705SXin Li           // Only warn if the concatenated string doesn't come from a macro.
477*67e74705SXin Li           bool hasMacro = false;
478*67e74705SXin Li           for (unsigned i = 0; i < numConcat ; ++i)
479*67e74705SXin Li             if (SL->getStrTokenLoc(i).isMacroID()) {
480*67e74705SXin Li               hasMacro = true;
481*67e74705SXin Li               break;
482*67e74705SXin Li             }
483*67e74705SXin Li           if (!hasMacro)
484*67e74705SXin Li             S.Diag(Element->getLocStart(),
485*67e74705SXin Li                    diag::warn_concatenated_nsarray_literal)
486*67e74705SXin Li               << Element->getType();
487*67e74705SXin Li         }
488*67e74705SXin Li       }
489*67e74705SXin Li     }
490*67e74705SXin Li 
491*67e74705SXin Li   // Make sure that the element has the type that the container factory
492*67e74705SXin Li   // function expects.
493*67e74705SXin Li   return S.PerformCopyInitialization(
494*67e74705SXin Li            InitializedEntity::InitializeParameter(S.Context, T,
495*67e74705SXin Li                                                   /*Consumed=*/false),
496*67e74705SXin Li            Element->getLocStart(), Element);
497*67e74705SXin Li }
498*67e74705SXin Li 
BuildObjCBoxedExpr(SourceRange SR,Expr * ValueExpr)499*67e74705SXin Li ExprResult Sema::BuildObjCBoxedExpr(SourceRange SR, Expr *ValueExpr) {
500*67e74705SXin Li   if (ValueExpr->isTypeDependent()) {
501*67e74705SXin Li     ObjCBoxedExpr *BoxedExpr =
502*67e74705SXin Li       new (Context) ObjCBoxedExpr(ValueExpr, Context.DependentTy, nullptr, SR);
503*67e74705SXin Li     return BoxedExpr;
504*67e74705SXin Li   }
505*67e74705SXin Li   ObjCMethodDecl *BoxingMethod = nullptr;
506*67e74705SXin Li   QualType BoxedType;
507*67e74705SXin Li   // Convert the expression to an RValue, so we can check for pointer types...
508*67e74705SXin Li   ExprResult RValue = DefaultFunctionArrayLvalueConversion(ValueExpr);
509*67e74705SXin Li   if (RValue.isInvalid()) {
510*67e74705SXin Li     return ExprError();
511*67e74705SXin Li   }
512*67e74705SXin Li   SourceLocation Loc = SR.getBegin();
513*67e74705SXin Li   ValueExpr = RValue.get();
514*67e74705SXin Li   QualType ValueType(ValueExpr->getType());
515*67e74705SXin Li   if (const PointerType *PT = ValueType->getAs<PointerType>()) {
516*67e74705SXin Li     QualType PointeeType = PT->getPointeeType();
517*67e74705SXin Li     if (Context.hasSameUnqualifiedType(PointeeType, Context.CharTy)) {
518*67e74705SXin Li 
519*67e74705SXin Li       if (!NSStringDecl) {
520*67e74705SXin Li         NSStringDecl = LookupObjCInterfaceDeclForLiteral(*this, Loc,
521*67e74705SXin Li                                                          Sema::LK_String);
522*67e74705SXin Li         if (!NSStringDecl) {
523*67e74705SXin Li           return ExprError();
524*67e74705SXin Li         }
525*67e74705SXin Li         QualType NSStringObject = Context.getObjCInterfaceType(NSStringDecl);
526*67e74705SXin Li         NSStringPointer = Context.getObjCObjectPointerType(NSStringObject);
527*67e74705SXin Li       }
528*67e74705SXin Li 
529*67e74705SXin Li       if (!StringWithUTF8StringMethod) {
530*67e74705SXin Li         IdentifierInfo *II = &Context.Idents.get("stringWithUTF8String");
531*67e74705SXin Li         Selector stringWithUTF8String = Context.Selectors.getUnarySelector(II);
532*67e74705SXin Li 
533*67e74705SXin Li         // Look for the appropriate method within NSString.
534*67e74705SXin Li         BoxingMethod = NSStringDecl->lookupClassMethod(stringWithUTF8String);
535*67e74705SXin Li         if (!BoxingMethod && getLangOpts().DebuggerObjCLiteral) {
536*67e74705SXin Li           // Debugger needs to work even if NSString hasn't been defined.
537*67e74705SXin Li           TypeSourceInfo *ReturnTInfo = nullptr;
538*67e74705SXin Li           ObjCMethodDecl *M = ObjCMethodDecl::Create(
539*67e74705SXin Li               Context, SourceLocation(), SourceLocation(), stringWithUTF8String,
540*67e74705SXin Li               NSStringPointer, ReturnTInfo, NSStringDecl,
541*67e74705SXin Li               /*isInstance=*/false, /*isVariadic=*/false,
542*67e74705SXin Li               /*isPropertyAccessor=*/false,
543*67e74705SXin Li               /*isImplicitlyDeclared=*/true,
544*67e74705SXin Li               /*isDefined=*/false, ObjCMethodDecl::Required,
545*67e74705SXin Li               /*HasRelatedResultType=*/false);
546*67e74705SXin Li           QualType ConstCharType = Context.CharTy.withConst();
547*67e74705SXin Li           ParmVarDecl *value =
548*67e74705SXin Li             ParmVarDecl::Create(Context, M,
549*67e74705SXin Li                                 SourceLocation(), SourceLocation(),
550*67e74705SXin Li                                 &Context.Idents.get("value"),
551*67e74705SXin Li                                 Context.getPointerType(ConstCharType),
552*67e74705SXin Li                                 /*TInfo=*/nullptr,
553*67e74705SXin Li                                 SC_None, nullptr);
554*67e74705SXin Li           M->setMethodParams(Context, value, None);
555*67e74705SXin Li           BoxingMethod = M;
556*67e74705SXin Li         }
557*67e74705SXin Li 
558*67e74705SXin Li         if (!validateBoxingMethod(*this, Loc, NSStringDecl,
559*67e74705SXin Li                                   stringWithUTF8String, BoxingMethod))
560*67e74705SXin Li            return ExprError();
561*67e74705SXin Li 
562*67e74705SXin Li         StringWithUTF8StringMethod = BoxingMethod;
563*67e74705SXin Li       }
564*67e74705SXin Li 
565*67e74705SXin Li       BoxingMethod = StringWithUTF8StringMethod;
566*67e74705SXin Li       BoxedType = NSStringPointer;
567*67e74705SXin Li     }
568*67e74705SXin Li   } else if (ValueType->isBuiltinType()) {
569*67e74705SXin Li     // The other types we support are numeric, char and BOOL/bool. We could also
570*67e74705SXin Li     // provide limited support for structure types, such as NSRange, NSRect, and
571*67e74705SXin Li     // NSSize. See NSValue (NSValueGeometryExtensions) in <Foundation/NSGeometry.h>
572*67e74705SXin Li     // for more details.
573*67e74705SXin Li 
574*67e74705SXin Li     // Check for a top-level character literal.
575*67e74705SXin Li     if (const CharacterLiteral *Char =
576*67e74705SXin Li         dyn_cast<CharacterLiteral>(ValueExpr->IgnoreParens())) {
577*67e74705SXin Li       // In C, character literals have type 'int'. That's not the type we want
578*67e74705SXin Li       // to use to determine the Objective-c literal kind.
579*67e74705SXin Li       switch (Char->getKind()) {
580*67e74705SXin Li       case CharacterLiteral::Ascii:
581*67e74705SXin Li       case CharacterLiteral::UTF8:
582*67e74705SXin Li         ValueType = Context.CharTy;
583*67e74705SXin Li         break;
584*67e74705SXin Li 
585*67e74705SXin Li       case CharacterLiteral::Wide:
586*67e74705SXin Li         ValueType = Context.getWideCharType();
587*67e74705SXin Li         break;
588*67e74705SXin Li 
589*67e74705SXin Li       case CharacterLiteral::UTF16:
590*67e74705SXin Li         ValueType = Context.Char16Ty;
591*67e74705SXin Li         break;
592*67e74705SXin Li 
593*67e74705SXin Li       case CharacterLiteral::UTF32:
594*67e74705SXin Li         ValueType = Context.Char32Ty;
595*67e74705SXin Li         break;
596*67e74705SXin Li       }
597*67e74705SXin Li     }
598*67e74705SXin Li     CheckForIntOverflow(ValueExpr);
599*67e74705SXin Li     // FIXME:  Do I need to do anything special with BoolTy expressions?
600*67e74705SXin Li 
601*67e74705SXin Li     // Look for the appropriate method within NSNumber.
602*67e74705SXin Li     BoxingMethod = getNSNumberFactoryMethod(*this, Loc, ValueType);
603*67e74705SXin Li     BoxedType = NSNumberPointer;
604*67e74705SXin Li   } else if (const EnumType *ET = ValueType->getAs<EnumType>()) {
605*67e74705SXin Li     if (!ET->getDecl()->isComplete()) {
606*67e74705SXin Li       Diag(Loc, diag::err_objc_incomplete_boxed_expression_type)
607*67e74705SXin Li         << ValueType << ValueExpr->getSourceRange();
608*67e74705SXin Li       return ExprError();
609*67e74705SXin Li     }
610*67e74705SXin Li 
611*67e74705SXin Li     BoxingMethod = getNSNumberFactoryMethod(*this, Loc,
612*67e74705SXin Li                                             ET->getDecl()->getIntegerType());
613*67e74705SXin Li     BoxedType = NSNumberPointer;
614*67e74705SXin Li   } else if (ValueType->isObjCBoxableRecordType()) {
615*67e74705SXin Li     // Support for structure types, that marked as objc_boxable
616*67e74705SXin Li     // struct __attribute__((objc_boxable)) s { ... };
617*67e74705SXin Li 
618*67e74705SXin Li     // Look up the NSValue class, if we haven't done so already. It's cached
619*67e74705SXin Li     // in the Sema instance.
620*67e74705SXin Li     if (!NSValueDecl) {
621*67e74705SXin Li       NSValueDecl = LookupObjCInterfaceDeclForLiteral(*this, Loc,
622*67e74705SXin Li                                                       Sema::LK_Boxed);
623*67e74705SXin Li       if (!NSValueDecl) {
624*67e74705SXin Li         return ExprError();
625*67e74705SXin Li       }
626*67e74705SXin Li 
627*67e74705SXin Li       // generate the pointer to NSValue type.
628*67e74705SXin Li       QualType NSValueObject = Context.getObjCInterfaceType(NSValueDecl);
629*67e74705SXin Li       NSValuePointer = Context.getObjCObjectPointerType(NSValueObject);
630*67e74705SXin Li     }
631*67e74705SXin Li 
632*67e74705SXin Li     if (!ValueWithBytesObjCTypeMethod) {
633*67e74705SXin Li       IdentifierInfo *II[] = {
634*67e74705SXin Li         &Context.Idents.get("valueWithBytes"),
635*67e74705SXin Li         &Context.Idents.get("objCType")
636*67e74705SXin Li       };
637*67e74705SXin Li       Selector ValueWithBytesObjCType = Context.Selectors.getSelector(2, II);
638*67e74705SXin Li 
639*67e74705SXin Li       // Look for the appropriate method within NSValue.
640*67e74705SXin Li       BoxingMethod = NSValueDecl->lookupClassMethod(ValueWithBytesObjCType);
641*67e74705SXin Li       if (!BoxingMethod && getLangOpts().DebuggerObjCLiteral) {
642*67e74705SXin Li         // Debugger needs to work even if NSValue hasn't been defined.
643*67e74705SXin Li         TypeSourceInfo *ReturnTInfo = nullptr;
644*67e74705SXin Li         ObjCMethodDecl *M = ObjCMethodDecl::Create(
645*67e74705SXin Li                                                Context,
646*67e74705SXin Li                                                SourceLocation(),
647*67e74705SXin Li                                                SourceLocation(),
648*67e74705SXin Li                                                ValueWithBytesObjCType,
649*67e74705SXin Li                                                NSValuePointer,
650*67e74705SXin Li                                                ReturnTInfo,
651*67e74705SXin Li                                                NSValueDecl,
652*67e74705SXin Li                                                /*isInstance=*/false,
653*67e74705SXin Li                                                /*isVariadic=*/false,
654*67e74705SXin Li                                                /*isPropertyAccessor=*/false,
655*67e74705SXin Li                                                /*isImplicitlyDeclared=*/true,
656*67e74705SXin Li                                                /*isDefined=*/false,
657*67e74705SXin Li                                                ObjCMethodDecl::Required,
658*67e74705SXin Li                                                /*HasRelatedResultType=*/false);
659*67e74705SXin Li 
660*67e74705SXin Li         SmallVector<ParmVarDecl *, 2> Params;
661*67e74705SXin Li 
662*67e74705SXin Li         ParmVarDecl *bytes =
663*67e74705SXin Li         ParmVarDecl::Create(Context, M,
664*67e74705SXin Li                             SourceLocation(), SourceLocation(),
665*67e74705SXin Li                             &Context.Idents.get("bytes"),
666*67e74705SXin Li                             Context.VoidPtrTy.withConst(),
667*67e74705SXin Li                             /*TInfo=*/nullptr,
668*67e74705SXin Li                             SC_None, nullptr);
669*67e74705SXin Li         Params.push_back(bytes);
670*67e74705SXin Li 
671*67e74705SXin Li         QualType ConstCharType = Context.CharTy.withConst();
672*67e74705SXin Li         ParmVarDecl *type =
673*67e74705SXin Li         ParmVarDecl::Create(Context, M,
674*67e74705SXin Li                             SourceLocation(), SourceLocation(),
675*67e74705SXin Li                             &Context.Idents.get("type"),
676*67e74705SXin Li                             Context.getPointerType(ConstCharType),
677*67e74705SXin Li                             /*TInfo=*/nullptr,
678*67e74705SXin Li                             SC_None, nullptr);
679*67e74705SXin Li         Params.push_back(type);
680*67e74705SXin Li 
681*67e74705SXin Li         M->setMethodParams(Context, Params, None);
682*67e74705SXin Li         BoxingMethod = M;
683*67e74705SXin Li       }
684*67e74705SXin Li 
685*67e74705SXin Li       if (!validateBoxingMethod(*this, Loc, NSValueDecl,
686*67e74705SXin Li                                 ValueWithBytesObjCType, BoxingMethod))
687*67e74705SXin Li         return ExprError();
688*67e74705SXin Li 
689*67e74705SXin Li       ValueWithBytesObjCTypeMethod = BoxingMethod;
690*67e74705SXin Li     }
691*67e74705SXin Li 
692*67e74705SXin Li     if (!ValueType.isTriviallyCopyableType(Context)) {
693*67e74705SXin Li       Diag(Loc, diag::err_objc_non_trivially_copyable_boxed_expression_type)
694*67e74705SXin Li         << ValueType << ValueExpr->getSourceRange();
695*67e74705SXin Li       return ExprError();
696*67e74705SXin Li     }
697*67e74705SXin Li 
698*67e74705SXin Li     BoxingMethod = ValueWithBytesObjCTypeMethod;
699*67e74705SXin Li     BoxedType = NSValuePointer;
700*67e74705SXin Li   }
701*67e74705SXin Li 
702*67e74705SXin Li   if (!BoxingMethod) {
703*67e74705SXin Li     Diag(Loc, diag::err_objc_illegal_boxed_expression_type)
704*67e74705SXin Li       << ValueType << ValueExpr->getSourceRange();
705*67e74705SXin Li     return ExprError();
706*67e74705SXin Li   }
707*67e74705SXin Li 
708*67e74705SXin Li   DiagnoseUseOfDecl(BoxingMethod, Loc);
709*67e74705SXin Li 
710*67e74705SXin Li   ExprResult ConvertedValueExpr;
711*67e74705SXin Li   if (ValueType->isObjCBoxableRecordType()) {
712*67e74705SXin Li     InitializedEntity IE = InitializedEntity::InitializeTemporary(ValueType);
713*67e74705SXin Li     ConvertedValueExpr = PerformCopyInitialization(IE, ValueExpr->getExprLoc(),
714*67e74705SXin Li                                                    ValueExpr);
715*67e74705SXin Li   } else {
716*67e74705SXin Li     // Convert the expression to the type that the parameter requires.
717*67e74705SXin Li     ParmVarDecl *ParamDecl = BoxingMethod->parameters()[0];
718*67e74705SXin Li     InitializedEntity IE = InitializedEntity::InitializeParameter(Context,
719*67e74705SXin Li                                                                   ParamDecl);
720*67e74705SXin Li     ConvertedValueExpr = PerformCopyInitialization(IE, SourceLocation(),
721*67e74705SXin Li                                                    ValueExpr);
722*67e74705SXin Li   }
723*67e74705SXin Li 
724*67e74705SXin Li   if (ConvertedValueExpr.isInvalid())
725*67e74705SXin Li     return ExprError();
726*67e74705SXin Li   ValueExpr = ConvertedValueExpr.get();
727*67e74705SXin Li 
728*67e74705SXin Li   ObjCBoxedExpr *BoxedExpr =
729*67e74705SXin Li     new (Context) ObjCBoxedExpr(ValueExpr, BoxedType,
730*67e74705SXin Li                                       BoxingMethod, SR);
731*67e74705SXin Li   return MaybeBindToTemporary(BoxedExpr);
732*67e74705SXin Li }
733*67e74705SXin Li 
734*67e74705SXin Li /// Build an ObjC subscript pseudo-object expression, given that
735*67e74705SXin Li /// that's supported by the runtime.
BuildObjCSubscriptExpression(SourceLocation RB,Expr * BaseExpr,Expr * IndexExpr,ObjCMethodDecl * getterMethod,ObjCMethodDecl * setterMethod)736*67e74705SXin Li ExprResult Sema::BuildObjCSubscriptExpression(SourceLocation RB, Expr *BaseExpr,
737*67e74705SXin Li                                         Expr *IndexExpr,
738*67e74705SXin Li                                         ObjCMethodDecl *getterMethod,
739*67e74705SXin Li                                         ObjCMethodDecl *setterMethod) {
740*67e74705SXin Li   assert(!LangOpts.isSubscriptPointerArithmetic());
741*67e74705SXin Li 
742*67e74705SXin Li   // We can't get dependent types here; our callers should have
743*67e74705SXin Li   // filtered them out.
744*67e74705SXin Li   assert((!BaseExpr->isTypeDependent() && !IndexExpr->isTypeDependent()) &&
745*67e74705SXin Li          "base or index cannot have dependent type here");
746*67e74705SXin Li 
747*67e74705SXin Li   // Filter out placeholders in the index.  In theory, overloads could
748*67e74705SXin Li   // be preserved here, although that might not actually work correctly.
749*67e74705SXin Li   ExprResult Result = CheckPlaceholderExpr(IndexExpr);
750*67e74705SXin Li   if (Result.isInvalid())
751*67e74705SXin Li     return ExprError();
752*67e74705SXin Li   IndexExpr = Result.get();
753*67e74705SXin Li 
754*67e74705SXin Li   // Perform lvalue-to-rvalue conversion on the base.
755*67e74705SXin Li   Result = DefaultLvalueConversion(BaseExpr);
756*67e74705SXin Li   if (Result.isInvalid())
757*67e74705SXin Li     return ExprError();
758*67e74705SXin Li   BaseExpr = Result.get();
759*67e74705SXin Li 
760*67e74705SXin Li   // Build the pseudo-object expression.
761*67e74705SXin Li   return new (Context) ObjCSubscriptRefExpr(
762*67e74705SXin Li       BaseExpr, IndexExpr, Context.PseudoObjectTy, VK_LValue, OK_ObjCSubscript,
763*67e74705SXin Li       getterMethod, setterMethod, RB);
764*67e74705SXin Li }
765*67e74705SXin Li 
BuildObjCArrayLiteral(SourceRange SR,MultiExprArg Elements)766*67e74705SXin Li ExprResult Sema::BuildObjCArrayLiteral(SourceRange SR, MultiExprArg Elements) {
767*67e74705SXin Li   SourceLocation Loc = SR.getBegin();
768*67e74705SXin Li 
769*67e74705SXin Li   if (!NSArrayDecl) {
770*67e74705SXin Li     NSArrayDecl = LookupObjCInterfaceDeclForLiteral(*this, Loc,
771*67e74705SXin Li                                                     Sema::LK_Array);
772*67e74705SXin Li     if (!NSArrayDecl) {
773*67e74705SXin Li       return ExprError();
774*67e74705SXin Li     }
775*67e74705SXin Li   }
776*67e74705SXin Li 
777*67e74705SXin Li   // Find the arrayWithObjects:count: method, if we haven't done so already.
778*67e74705SXin Li   QualType IdT = Context.getObjCIdType();
779*67e74705SXin Li   if (!ArrayWithObjectsMethod) {
780*67e74705SXin Li     Selector
781*67e74705SXin Li       Sel = NSAPIObj->getNSArraySelector(NSAPI::NSArr_arrayWithObjectsCount);
782*67e74705SXin Li     ObjCMethodDecl *Method = NSArrayDecl->lookupClassMethod(Sel);
783*67e74705SXin Li     if (!Method && getLangOpts().DebuggerObjCLiteral) {
784*67e74705SXin Li       TypeSourceInfo *ReturnTInfo = nullptr;
785*67e74705SXin Li       Method = ObjCMethodDecl::Create(
786*67e74705SXin Li           Context, SourceLocation(), SourceLocation(), Sel, IdT, ReturnTInfo,
787*67e74705SXin Li           Context.getTranslationUnitDecl(), false /*Instance*/,
788*67e74705SXin Li           false /*isVariadic*/,
789*67e74705SXin Li           /*isPropertyAccessor=*/false,
790*67e74705SXin Li           /*isImplicitlyDeclared=*/true, /*isDefined=*/false,
791*67e74705SXin Li           ObjCMethodDecl::Required, false);
792*67e74705SXin Li       SmallVector<ParmVarDecl *, 2> Params;
793*67e74705SXin Li       ParmVarDecl *objects = ParmVarDecl::Create(Context, Method,
794*67e74705SXin Li                                                  SourceLocation(),
795*67e74705SXin Li                                                  SourceLocation(),
796*67e74705SXin Li                                                  &Context.Idents.get("objects"),
797*67e74705SXin Li                                                  Context.getPointerType(IdT),
798*67e74705SXin Li                                                  /*TInfo=*/nullptr,
799*67e74705SXin Li                                                  SC_None, nullptr);
800*67e74705SXin Li       Params.push_back(objects);
801*67e74705SXin Li       ParmVarDecl *cnt = ParmVarDecl::Create(Context, Method,
802*67e74705SXin Li                                              SourceLocation(),
803*67e74705SXin Li                                              SourceLocation(),
804*67e74705SXin Li                                              &Context.Idents.get("cnt"),
805*67e74705SXin Li                                              Context.UnsignedLongTy,
806*67e74705SXin Li                                              /*TInfo=*/nullptr, SC_None,
807*67e74705SXin Li                                              nullptr);
808*67e74705SXin Li       Params.push_back(cnt);
809*67e74705SXin Li       Method->setMethodParams(Context, Params, None);
810*67e74705SXin Li     }
811*67e74705SXin Li 
812*67e74705SXin Li     if (!validateBoxingMethod(*this, Loc, NSArrayDecl, Sel, Method))
813*67e74705SXin Li       return ExprError();
814*67e74705SXin Li 
815*67e74705SXin Li     // Dig out the type that all elements should be converted to.
816*67e74705SXin Li     QualType T = Method->parameters()[0]->getType();
817*67e74705SXin Li     const PointerType *PtrT = T->getAs<PointerType>();
818*67e74705SXin Li     if (!PtrT ||
819*67e74705SXin Li         !Context.hasSameUnqualifiedType(PtrT->getPointeeType(), IdT)) {
820*67e74705SXin Li       Diag(SR.getBegin(), diag::err_objc_literal_method_sig)
821*67e74705SXin Li         << Sel;
822*67e74705SXin Li       Diag(Method->parameters()[0]->getLocation(),
823*67e74705SXin Li            diag::note_objc_literal_method_param)
824*67e74705SXin Li         << 0 << T
825*67e74705SXin Li         << Context.getPointerType(IdT.withConst());
826*67e74705SXin Li       return ExprError();
827*67e74705SXin Li     }
828*67e74705SXin Li 
829*67e74705SXin Li     // Check that the 'count' parameter is integral.
830*67e74705SXin Li     if (!Method->parameters()[1]->getType()->isIntegerType()) {
831*67e74705SXin Li       Diag(SR.getBegin(), diag::err_objc_literal_method_sig)
832*67e74705SXin Li         << Sel;
833*67e74705SXin Li       Diag(Method->parameters()[1]->getLocation(),
834*67e74705SXin Li            diag::note_objc_literal_method_param)
835*67e74705SXin Li         << 1
836*67e74705SXin Li         << Method->parameters()[1]->getType()
837*67e74705SXin Li         << "integral";
838*67e74705SXin Li       return ExprError();
839*67e74705SXin Li     }
840*67e74705SXin Li 
841*67e74705SXin Li     // We've found a good +arrayWithObjects:count: method. Save it!
842*67e74705SXin Li     ArrayWithObjectsMethod = Method;
843*67e74705SXin Li   }
844*67e74705SXin Li 
845*67e74705SXin Li   QualType ObjectsType = ArrayWithObjectsMethod->parameters()[0]->getType();
846*67e74705SXin Li   QualType RequiredType = ObjectsType->castAs<PointerType>()->getPointeeType();
847*67e74705SXin Li 
848*67e74705SXin Li   // Check that each of the elements provided is valid in a collection literal,
849*67e74705SXin Li   // performing conversions as necessary.
850*67e74705SXin Li   Expr **ElementsBuffer = Elements.data();
851*67e74705SXin Li   for (unsigned I = 0, N = Elements.size(); I != N; ++I) {
852*67e74705SXin Li     ExprResult Converted = CheckObjCCollectionLiteralElement(*this,
853*67e74705SXin Li                                                              ElementsBuffer[I],
854*67e74705SXin Li                                                              RequiredType, true);
855*67e74705SXin Li     if (Converted.isInvalid())
856*67e74705SXin Li       return ExprError();
857*67e74705SXin Li 
858*67e74705SXin Li     ElementsBuffer[I] = Converted.get();
859*67e74705SXin Li   }
860*67e74705SXin Li 
861*67e74705SXin Li   QualType Ty
862*67e74705SXin Li     = Context.getObjCObjectPointerType(
863*67e74705SXin Li                                     Context.getObjCInterfaceType(NSArrayDecl));
864*67e74705SXin Li 
865*67e74705SXin Li   return MaybeBindToTemporary(
866*67e74705SXin Li            ObjCArrayLiteral::Create(Context, Elements, Ty,
867*67e74705SXin Li                                     ArrayWithObjectsMethod, SR));
868*67e74705SXin Li }
869*67e74705SXin Li 
BuildObjCDictionaryLiteral(SourceRange SR,MutableArrayRef<ObjCDictionaryElement> Elements)870*67e74705SXin Li ExprResult Sema::BuildObjCDictionaryLiteral(SourceRange SR,
871*67e74705SXin Li                               MutableArrayRef<ObjCDictionaryElement> Elements) {
872*67e74705SXin Li   SourceLocation Loc = SR.getBegin();
873*67e74705SXin Li 
874*67e74705SXin Li   if (!NSDictionaryDecl) {
875*67e74705SXin Li     NSDictionaryDecl = LookupObjCInterfaceDeclForLiteral(*this, Loc,
876*67e74705SXin Li                                                          Sema::LK_Dictionary);
877*67e74705SXin Li     if (!NSDictionaryDecl) {
878*67e74705SXin Li       return ExprError();
879*67e74705SXin Li     }
880*67e74705SXin Li   }
881*67e74705SXin Li 
882*67e74705SXin Li   // Find the dictionaryWithObjects:forKeys:count: method, if we haven't done
883*67e74705SXin Li   // so already.
884*67e74705SXin Li   QualType IdT = Context.getObjCIdType();
885*67e74705SXin Li   if (!DictionaryWithObjectsMethod) {
886*67e74705SXin Li     Selector Sel = NSAPIObj->getNSDictionarySelector(
887*67e74705SXin Li                                NSAPI::NSDict_dictionaryWithObjectsForKeysCount);
888*67e74705SXin Li     ObjCMethodDecl *Method = NSDictionaryDecl->lookupClassMethod(Sel);
889*67e74705SXin Li     if (!Method && getLangOpts().DebuggerObjCLiteral) {
890*67e74705SXin Li       Method = ObjCMethodDecl::Create(Context,
891*67e74705SXin Li                            SourceLocation(), SourceLocation(), Sel,
892*67e74705SXin Li                            IdT,
893*67e74705SXin Li                            nullptr /*TypeSourceInfo */,
894*67e74705SXin Li                            Context.getTranslationUnitDecl(),
895*67e74705SXin Li                            false /*Instance*/, false/*isVariadic*/,
896*67e74705SXin Li                            /*isPropertyAccessor=*/false,
897*67e74705SXin Li                            /*isImplicitlyDeclared=*/true, /*isDefined=*/false,
898*67e74705SXin Li                            ObjCMethodDecl::Required,
899*67e74705SXin Li                            false);
900*67e74705SXin Li       SmallVector<ParmVarDecl *, 3> Params;
901*67e74705SXin Li       ParmVarDecl *objects = ParmVarDecl::Create(Context, Method,
902*67e74705SXin Li                                                  SourceLocation(),
903*67e74705SXin Li                                                  SourceLocation(),
904*67e74705SXin Li                                                  &Context.Idents.get("objects"),
905*67e74705SXin Li                                                  Context.getPointerType(IdT),
906*67e74705SXin Li                                                  /*TInfo=*/nullptr, SC_None,
907*67e74705SXin Li                                                  nullptr);
908*67e74705SXin Li       Params.push_back(objects);
909*67e74705SXin Li       ParmVarDecl *keys = ParmVarDecl::Create(Context, Method,
910*67e74705SXin Li                                               SourceLocation(),
911*67e74705SXin Li                                               SourceLocation(),
912*67e74705SXin Li                                               &Context.Idents.get("keys"),
913*67e74705SXin Li                                               Context.getPointerType(IdT),
914*67e74705SXin Li                                               /*TInfo=*/nullptr, SC_None,
915*67e74705SXin Li                                               nullptr);
916*67e74705SXin Li       Params.push_back(keys);
917*67e74705SXin Li       ParmVarDecl *cnt = ParmVarDecl::Create(Context, Method,
918*67e74705SXin Li                                              SourceLocation(),
919*67e74705SXin Li                                              SourceLocation(),
920*67e74705SXin Li                                              &Context.Idents.get("cnt"),
921*67e74705SXin Li                                              Context.UnsignedLongTy,
922*67e74705SXin Li                                              /*TInfo=*/nullptr, SC_None,
923*67e74705SXin Li                                              nullptr);
924*67e74705SXin Li       Params.push_back(cnt);
925*67e74705SXin Li       Method->setMethodParams(Context, Params, None);
926*67e74705SXin Li     }
927*67e74705SXin Li 
928*67e74705SXin Li     if (!validateBoxingMethod(*this, SR.getBegin(), NSDictionaryDecl, Sel,
929*67e74705SXin Li                               Method))
930*67e74705SXin Li        return ExprError();
931*67e74705SXin Li 
932*67e74705SXin Li     // Dig out the type that all values should be converted to.
933*67e74705SXin Li     QualType ValueT = Method->parameters()[0]->getType();
934*67e74705SXin Li     const PointerType *PtrValue = ValueT->getAs<PointerType>();
935*67e74705SXin Li     if (!PtrValue ||
936*67e74705SXin Li         !Context.hasSameUnqualifiedType(PtrValue->getPointeeType(), IdT)) {
937*67e74705SXin Li       Diag(SR.getBegin(), diag::err_objc_literal_method_sig)
938*67e74705SXin Li         << Sel;
939*67e74705SXin Li       Diag(Method->parameters()[0]->getLocation(),
940*67e74705SXin Li            diag::note_objc_literal_method_param)
941*67e74705SXin Li         << 0 << ValueT
942*67e74705SXin Li         << Context.getPointerType(IdT.withConst());
943*67e74705SXin Li       return ExprError();
944*67e74705SXin Li     }
945*67e74705SXin Li 
946*67e74705SXin Li     // Dig out the type that all keys should be converted to.
947*67e74705SXin Li     QualType KeyT = Method->parameters()[1]->getType();
948*67e74705SXin Li     const PointerType *PtrKey = KeyT->getAs<PointerType>();
949*67e74705SXin Li     if (!PtrKey ||
950*67e74705SXin Li         !Context.hasSameUnqualifiedType(PtrKey->getPointeeType(),
951*67e74705SXin Li                                         IdT)) {
952*67e74705SXin Li       bool err = true;
953*67e74705SXin Li       if (PtrKey) {
954*67e74705SXin Li         if (QIDNSCopying.isNull()) {
955*67e74705SXin Li           // key argument of selector is id<NSCopying>?
956*67e74705SXin Li           if (ObjCProtocolDecl *NSCopyingPDecl =
957*67e74705SXin Li               LookupProtocol(&Context.Idents.get("NSCopying"), SR.getBegin())) {
958*67e74705SXin Li             ObjCProtocolDecl *PQ[] = {NSCopyingPDecl};
959*67e74705SXin Li             QIDNSCopying =
960*67e74705SXin Li               Context.getObjCObjectType(Context.ObjCBuiltinIdTy, { },
961*67e74705SXin Li                                         llvm::makeArrayRef(
962*67e74705SXin Li                                           (ObjCProtocolDecl**) PQ,
963*67e74705SXin Li                                           1),
964*67e74705SXin Li                                         false);
965*67e74705SXin Li             QIDNSCopying = Context.getObjCObjectPointerType(QIDNSCopying);
966*67e74705SXin Li           }
967*67e74705SXin Li         }
968*67e74705SXin Li         if (!QIDNSCopying.isNull())
969*67e74705SXin Li           err = !Context.hasSameUnqualifiedType(PtrKey->getPointeeType(),
970*67e74705SXin Li                                                 QIDNSCopying);
971*67e74705SXin Li       }
972*67e74705SXin Li 
973*67e74705SXin Li       if (err) {
974*67e74705SXin Li         Diag(SR.getBegin(), diag::err_objc_literal_method_sig)
975*67e74705SXin Li           << Sel;
976*67e74705SXin Li         Diag(Method->parameters()[1]->getLocation(),
977*67e74705SXin Li              diag::note_objc_literal_method_param)
978*67e74705SXin Li           << 1 << KeyT
979*67e74705SXin Li           << Context.getPointerType(IdT.withConst());
980*67e74705SXin Li         return ExprError();
981*67e74705SXin Li       }
982*67e74705SXin Li     }
983*67e74705SXin Li 
984*67e74705SXin Li     // Check that the 'count' parameter is integral.
985*67e74705SXin Li     QualType CountType = Method->parameters()[2]->getType();
986*67e74705SXin Li     if (!CountType->isIntegerType()) {
987*67e74705SXin Li       Diag(SR.getBegin(), diag::err_objc_literal_method_sig)
988*67e74705SXin Li         << Sel;
989*67e74705SXin Li       Diag(Method->parameters()[2]->getLocation(),
990*67e74705SXin Li            diag::note_objc_literal_method_param)
991*67e74705SXin Li         << 2 << CountType
992*67e74705SXin Li         << "integral";
993*67e74705SXin Li       return ExprError();
994*67e74705SXin Li     }
995*67e74705SXin Li 
996*67e74705SXin Li     // We've found a good +dictionaryWithObjects:keys:count: method; save it!
997*67e74705SXin Li     DictionaryWithObjectsMethod = Method;
998*67e74705SXin Li   }
999*67e74705SXin Li 
1000*67e74705SXin Li   QualType ValuesT = DictionaryWithObjectsMethod->parameters()[0]->getType();
1001*67e74705SXin Li   QualType ValueT = ValuesT->castAs<PointerType>()->getPointeeType();
1002*67e74705SXin Li   QualType KeysT = DictionaryWithObjectsMethod->parameters()[1]->getType();
1003*67e74705SXin Li   QualType KeyT = KeysT->castAs<PointerType>()->getPointeeType();
1004*67e74705SXin Li 
1005*67e74705SXin Li   // Check that each of the keys and values provided is valid in a collection
1006*67e74705SXin Li   // literal, performing conversions as necessary.
1007*67e74705SXin Li   bool HasPackExpansions = false;
1008*67e74705SXin Li   for (ObjCDictionaryElement &Element : Elements) {
1009*67e74705SXin Li     // Check the key.
1010*67e74705SXin Li     ExprResult Key = CheckObjCCollectionLiteralElement(*this, Element.Key,
1011*67e74705SXin Li                                                        KeyT);
1012*67e74705SXin Li     if (Key.isInvalid())
1013*67e74705SXin Li       return ExprError();
1014*67e74705SXin Li 
1015*67e74705SXin Li     // Check the value.
1016*67e74705SXin Li     ExprResult Value
1017*67e74705SXin Li       = CheckObjCCollectionLiteralElement(*this, Element.Value, ValueT);
1018*67e74705SXin Li     if (Value.isInvalid())
1019*67e74705SXin Li       return ExprError();
1020*67e74705SXin Li 
1021*67e74705SXin Li     Element.Key = Key.get();
1022*67e74705SXin Li     Element.Value = Value.get();
1023*67e74705SXin Li 
1024*67e74705SXin Li     if (Element.EllipsisLoc.isInvalid())
1025*67e74705SXin Li       continue;
1026*67e74705SXin Li 
1027*67e74705SXin Li     if (!Element.Key->containsUnexpandedParameterPack() &&
1028*67e74705SXin Li         !Element.Value->containsUnexpandedParameterPack()) {
1029*67e74705SXin Li       Diag(Element.EllipsisLoc,
1030*67e74705SXin Li            diag::err_pack_expansion_without_parameter_packs)
1031*67e74705SXin Li         << SourceRange(Element.Key->getLocStart(),
1032*67e74705SXin Li                        Element.Value->getLocEnd());
1033*67e74705SXin Li       return ExprError();
1034*67e74705SXin Li     }
1035*67e74705SXin Li 
1036*67e74705SXin Li     HasPackExpansions = true;
1037*67e74705SXin Li   }
1038*67e74705SXin Li 
1039*67e74705SXin Li   QualType Ty
1040*67e74705SXin Li     = Context.getObjCObjectPointerType(
1041*67e74705SXin Li                                 Context.getObjCInterfaceType(NSDictionaryDecl));
1042*67e74705SXin Li   return MaybeBindToTemporary(ObjCDictionaryLiteral::Create(
1043*67e74705SXin Li       Context, Elements, HasPackExpansions, Ty,
1044*67e74705SXin Li       DictionaryWithObjectsMethod, SR));
1045*67e74705SXin Li }
1046*67e74705SXin Li 
BuildObjCEncodeExpression(SourceLocation AtLoc,TypeSourceInfo * EncodedTypeInfo,SourceLocation RParenLoc)1047*67e74705SXin Li ExprResult Sema::BuildObjCEncodeExpression(SourceLocation AtLoc,
1048*67e74705SXin Li                                       TypeSourceInfo *EncodedTypeInfo,
1049*67e74705SXin Li                                       SourceLocation RParenLoc) {
1050*67e74705SXin Li   QualType EncodedType = EncodedTypeInfo->getType();
1051*67e74705SXin Li   QualType StrTy;
1052*67e74705SXin Li   if (EncodedType->isDependentType())
1053*67e74705SXin Li     StrTy = Context.DependentTy;
1054*67e74705SXin Li   else {
1055*67e74705SXin Li     if (!EncodedType->getAsArrayTypeUnsafe() && //// Incomplete array is handled.
1056*67e74705SXin Li         !EncodedType->isVoidType()) // void is handled too.
1057*67e74705SXin Li       if (RequireCompleteType(AtLoc, EncodedType,
1058*67e74705SXin Li                               diag::err_incomplete_type_objc_at_encode,
1059*67e74705SXin Li                               EncodedTypeInfo->getTypeLoc()))
1060*67e74705SXin Li         return ExprError();
1061*67e74705SXin Li 
1062*67e74705SXin Li     std::string Str;
1063*67e74705SXin Li     QualType NotEncodedT;
1064*67e74705SXin Li     Context.getObjCEncodingForType(EncodedType, Str, nullptr, &NotEncodedT);
1065*67e74705SXin Li     if (!NotEncodedT.isNull())
1066*67e74705SXin Li       Diag(AtLoc, diag::warn_incomplete_encoded_type)
1067*67e74705SXin Li         << EncodedType << NotEncodedT;
1068*67e74705SXin Li 
1069*67e74705SXin Li     // The type of @encode is the same as the type of the corresponding string,
1070*67e74705SXin Li     // which is an array type.
1071*67e74705SXin Li     StrTy = Context.CharTy;
1072*67e74705SXin Li     // A C++ string literal has a const-qualified element type (C++ 2.13.4p1).
1073*67e74705SXin Li     if (getLangOpts().CPlusPlus || getLangOpts().ConstStrings)
1074*67e74705SXin Li       StrTy.addConst();
1075*67e74705SXin Li     StrTy = Context.getConstantArrayType(StrTy, llvm::APInt(32, Str.size()+1),
1076*67e74705SXin Li                                          ArrayType::Normal, 0);
1077*67e74705SXin Li   }
1078*67e74705SXin Li 
1079*67e74705SXin Li   return new (Context) ObjCEncodeExpr(StrTy, EncodedTypeInfo, AtLoc, RParenLoc);
1080*67e74705SXin Li }
1081*67e74705SXin Li 
ParseObjCEncodeExpression(SourceLocation AtLoc,SourceLocation EncodeLoc,SourceLocation LParenLoc,ParsedType ty,SourceLocation RParenLoc)1082*67e74705SXin Li ExprResult Sema::ParseObjCEncodeExpression(SourceLocation AtLoc,
1083*67e74705SXin Li                                            SourceLocation EncodeLoc,
1084*67e74705SXin Li                                            SourceLocation LParenLoc,
1085*67e74705SXin Li                                            ParsedType ty,
1086*67e74705SXin Li                                            SourceLocation RParenLoc) {
1087*67e74705SXin Li   // FIXME: Preserve type source info ?
1088*67e74705SXin Li   TypeSourceInfo *TInfo;
1089*67e74705SXin Li   QualType EncodedType = GetTypeFromParser(ty, &TInfo);
1090*67e74705SXin Li   if (!TInfo)
1091*67e74705SXin Li     TInfo = Context.getTrivialTypeSourceInfo(EncodedType,
1092*67e74705SXin Li                                              getLocForEndOfToken(LParenLoc));
1093*67e74705SXin Li 
1094*67e74705SXin Li   return BuildObjCEncodeExpression(AtLoc, TInfo, RParenLoc);
1095*67e74705SXin Li }
1096*67e74705SXin Li 
HelperToDiagnoseMismatchedMethodsInGlobalPool(Sema & S,SourceLocation AtLoc,SourceLocation LParenLoc,SourceLocation RParenLoc,ObjCMethodDecl * Method,ObjCMethodList & MethList)1097*67e74705SXin Li static bool HelperToDiagnoseMismatchedMethodsInGlobalPool(Sema &S,
1098*67e74705SXin Li                                                SourceLocation AtLoc,
1099*67e74705SXin Li                                                SourceLocation LParenLoc,
1100*67e74705SXin Li                                                SourceLocation RParenLoc,
1101*67e74705SXin Li                                                ObjCMethodDecl *Method,
1102*67e74705SXin Li                                                ObjCMethodList &MethList) {
1103*67e74705SXin Li   ObjCMethodList *M = &MethList;
1104*67e74705SXin Li   bool Warned = false;
1105*67e74705SXin Li   for (M = M->getNext(); M; M=M->getNext()) {
1106*67e74705SXin Li     ObjCMethodDecl *MatchingMethodDecl = M->getMethod();
1107*67e74705SXin Li     if (MatchingMethodDecl == Method ||
1108*67e74705SXin Li         isa<ObjCImplDecl>(MatchingMethodDecl->getDeclContext()) ||
1109*67e74705SXin Li         MatchingMethodDecl->getSelector() != Method->getSelector())
1110*67e74705SXin Li       continue;
1111*67e74705SXin Li     if (!S.MatchTwoMethodDeclarations(Method,
1112*67e74705SXin Li                                       MatchingMethodDecl, Sema::MMS_loose)) {
1113*67e74705SXin Li       if (!Warned) {
1114*67e74705SXin Li         Warned = true;
1115*67e74705SXin Li         S.Diag(AtLoc, diag::warning_multiple_selectors)
1116*67e74705SXin Li           << Method->getSelector() << FixItHint::CreateInsertion(LParenLoc, "(")
1117*67e74705SXin Li           << FixItHint::CreateInsertion(RParenLoc, ")");
1118*67e74705SXin Li         S.Diag(Method->getLocation(), diag::note_method_declared_at)
1119*67e74705SXin Li           << Method->getDeclName();
1120*67e74705SXin Li       }
1121*67e74705SXin Li       S.Diag(MatchingMethodDecl->getLocation(), diag::note_method_declared_at)
1122*67e74705SXin Li         << MatchingMethodDecl->getDeclName();
1123*67e74705SXin Li     }
1124*67e74705SXin Li   }
1125*67e74705SXin Li   return Warned;
1126*67e74705SXin Li }
1127*67e74705SXin Li 
DiagnoseMismatchedSelectors(Sema & S,SourceLocation AtLoc,ObjCMethodDecl * Method,SourceLocation LParenLoc,SourceLocation RParenLoc,bool WarnMultipleSelectors)1128*67e74705SXin Li static void DiagnoseMismatchedSelectors(Sema &S, SourceLocation AtLoc,
1129*67e74705SXin Li                                         ObjCMethodDecl *Method,
1130*67e74705SXin Li                                         SourceLocation LParenLoc,
1131*67e74705SXin Li                                         SourceLocation RParenLoc,
1132*67e74705SXin Li                                         bool WarnMultipleSelectors) {
1133*67e74705SXin Li   if (!WarnMultipleSelectors ||
1134*67e74705SXin Li       S.Diags.isIgnored(diag::warning_multiple_selectors, SourceLocation()))
1135*67e74705SXin Li     return;
1136*67e74705SXin Li   bool Warned = false;
1137*67e74705SXin Li   for (Sema::GlobalMethodPool::iterator b = S.MethodPool.begin(),
1138*67e74705SXin Li        e = S.MethodPool.end(); b != e; b++) {
1139*67e74705SXin Li     // first, instance methods
1140*67e74705SXin Li     ObjCMethodList &InstMethList = b->second.first;
1141*67e74705SXin Li     if (HelperToDiagnoseMismatchedMethodsInGlobalPool(S, AtLoc, LParenLoc, RParenLoc,
1142*67e74705SXin Li                                                       Method, InstMethList))
1143*67e74705SXin Li       Warned = true;
1144*67e74705SXin Li 
1145*67e74705SXin Li     // second, class methods
1146*67e74705SXin Li     ObjCMethodList &ClsMethList = b->second.second;
1147*67e74705SXin Li     if (HelperToDiagnoseMismatchedMethodsInGlobalPool(S, AtLoc, LParenLoc, RParenLoc,
1148*67e74705SXin Li                                                       Method, ClsMethList) || Warned)
1149*67e74705SXin Li       return;
1150*67e74705SXin Li   }
1151*67e74705SXin Li }
1152*67e74705SXin Li 
ParseObjCSelectorExpression(Selector Sel,SourceLocation AtLoc,SourceLocation SelLoc,SourceLocation LParenLoc,SourceLocation RParenLoc,bool WarnMultipleSelectors)1153*67e74705SXin Li ExprResult Sema::ParseObjCSelectorExpression(Selector Sel,
1154*67e74705SXin Li                                              SourceLocation AtLoc,
1155*67e74705SXin Li                                              SourceLocation SelLoc,
1156*67e74705SXin Li                                              SourceLocation LParenLoc,
1157*67e74705SXin Li                                              SourceLocation RParenLoc,
1158*67e74705SXin Li                                              bool WarnMultipleSelectors) {
1159*67e74705SXin Li   ObjCMethodDecl *Method = LookupInstanceMethodInGlobalPool(Sel,
1160*67e74705SXin Li                              SourceRange(LParenLoc, RParenLoc));
1161*67e74705SXin Li   if (!Method)
1162*67e74705SXin Li     Method = LookupFactoryMethodInGlobalPool(Sel,
1163*67e74705SXin Li                                           SourceRange(LParenLoc, RParenLoc));
1164*67e74705SXin Li   if (!Method) {
1165*67e74705SXin Li     if (const ObjCMethodDecl *OM = SelectorsForTypoCorrection(Sel)) {
1166*67e74705SXin Li       Selector MatchedSel = OM->getSelector();
1167*67e74705SXin Li       SourceRange SelectorRange(LParenLoc.getLocWithOffset(1),
1168*67e74705SXin Li                                 RParenLoc.getLocWithOffset(-1));
1169*67e74705SXin Li       Diag(SelLoc, diag::warn_undeclared_selector_with_typo)
1170*67e74705SXin Li         << Sel << MatchedSel
1171*67e74705SXin Li         << FixItHint::CreateReplacement(SelectorRange, MatchedSel.getAsString());
1172*67e74705SXin Li 
1173*67e74705SXin Li     } else
1174*67e74705SXin Li         Diag(SelLoc, diag::warn_undeclared_selector) << Sel;
1175*67e74705SXin Li   } else
1176*67e74705SXin Li     DiagnoseMismatchedSelectors(*this, AtLoc, Method, LParenLoc, RParenLoc,
1177*67e74705SXin Li                                 WarnMultipleSelectors);
1178*67e74705SXin Li 
1179*67e74705SXin Li   if (Method &&
1180*67e74705SXin Li       Method->getImplementationControl() != ObjCMethodDecl::Optional &&
1181*67e74705SXin Li       !getSourceManager().isInSystemHeader(Method->getLocation()))
1182*67e74705SXin Li     ReferencedSelectors.insert(std::make_pair(Sel, AtLoc));
1183*67e74705SXin Li 
1184*67e74705SXin Li   // In ARC, forbid the user from using @selector for
1185*67e74705SXin Li   // retain/release/autorelease/dealloc/retainCount.
1186*67e74705SXin Li   if (getLangOpts().ObjCAutoRefCount) {
1187*67e74705SXin Li     switch (Sel.getMethodFamily()) {
1188*67e74705SXin Li     case OMF_retain:
1189*67e74705SXin Li     case OMF_release:
1190*67e74705SXin Li     case OMF_autorelease:
1191*67e74705SXin Li     case OMF_retainCount:
1192*67e74705SXin Li     case OMF_dealloc:
1193*67e74705SXin Li       Diag(AtLoc, diag::err_arc_illegal_selector) <<
1194*67e74705SXin Li         Sel << SourceRange(LParenLoc, RParenLoc);
1195*67e74705SXin Li       break;
1196*67e74705SXin Li 
1197*67e74705SXin Li     case OMF_None:
1198*67e74705SXin Li     case OMF_alloc:
1199*67e74705SXin Li     case OMF_copy:
1200*67e74705SXin Li     case OMF_finalize:
1201*67e74705SXin Li     case OMF_init:
1202*67e74705SXin Li     case OMF_mutableCopy:
1203*67e74705SXin Li     case OMF_new:
1204*67e74705SXin Li     case OMF_self:
1205*67e74705SXin Li     case OMF_initialize:
1206*67e74705SXin Li     case OMF_performSelector:
1207*67e74705SXin Li       break;
1208*67e74705SXin Li     }
1209*67e74705SXin Li   }
1210*67e74705SXin Li   QualType Ty = Context.getObjCSelType();
1211*67e74705SXin Li   return new (Context) ObjCSelectorExpr(Ty, Sel, AtLoc, RParenLoc);
1212*67e74705SXin Li }
1213*67e74705SXin Li 
ParseObjCProtocolExpression(IdentifierInfo * ProtocolId,SourceLocation AtLoc,SourceLocation ProtoLoc,SourceLocation LParenLoc,SourceLocation ProtoIdLoc,SourceLocation RParenLoc)1214*67e74705SXin Li ExprResult Sema::ParseObjCProtocolExpression(IdentifierInfo *ProtocolId,
1215*67e74705SXin Li                                              SourceLocation AtLoc,
1216*67e74705SXin Li                                              SourceLocation ProtoLoc,
1217*67e74705SXin Li                                              SourceLocation LParenLoc,
1218*67e74705SXin Li                                              SourceLocation ProtoIdLoc,
1219*67e74705SXin Li                                              SourceLocation RParenLoc) {
1220*67e74705SXin Li   ObjCProtocolDecl* PDecl = LookupProtocol(ProtocolId, ProtoIdLoc);
1221*67e74705SXin Li   if (!PDecl) {
1222*67e74705SXin Li     Diag(ProtoLoc, diag::err_undeclared_protocol) << ProtocolId;
1223*67e74705SXin Li     return true;
1224*67e74705SXin Li   }
1225*67e74705SXin Li   if (PDecl->hasDefinition())
1226*67e74705SXin Li     PDecl = PDecl->getDefinition();
1227*67e74705SXin Li 
1228*67e74705SXin Li   QualType Ty = Context.getObjCProtoType();
1229*67e74705SXin Li   if (Ty.isNull())
1230*67e74705SXin Li     return true;
1231*67e74705SXin Li   Ty = Context.getObjCObjectPointerType(Ty);
1232*67e74705SXin Li   return new (Context) ObjCProtocolExpr(Ty, PDecl, AtLoc, ProtoIdLoc, RParenLoc);
1233*67e74705SXin Li }
1234*67e74705SXin Li 
1235*67e74705SXin Li /// Try to capture an implicit reference to 'self'.
tryCaptureObjCSelf(SourceLocation Loc)1236*67e74705SXin Li ObjCMethodDecl *Sema::tryCaptureObjCSelf(SourceLocation Loc) {
1237*67e74705SXin Li   DeclContext *DC = getFunctionLevelDeclContext();
1238*67e74705SXin Li 
1239*67e74705SXin Li   // If we're not in an ObjC method, error out.  Note that, unlike the
1240*67e74705SXin Li   // C++ case, we don't require an instance method --- class methods
1241*67e74705SXin Li   // still have a 'self', and we really do still need to capture it!
1242*67e74705SXin Li   ObjCMethodDecl *method = dyn_cast<ObjCMethodDecl>(DC);
1243*67e74705SXin Li   if (!method)
1244*67e74705SXin Li     return nullptr;
1245*67e74705SXin Li 
1246*67e74705SXin Li   tryCaptureVariable(method->getSelfDecl(), Loc);
1247*67e74705SXin Li 
1248*67e74705SXin Li   return method;
1249*67e74705SXin Li }
1250*67e74705SXin Li 
stripObjCInstanceType(ASTContext & Context,QualType T)1251*67e74705SXin Li static QualType stripObjCInstanceType(ASTContext &Context, QualType T) {
1252*67e74705SXin Li   QualType origType = T;
1253*67e74705SXin Li   if (auto nullability = AttributedType::stripOuterNullability(T)) {
1254*67e74705SXin Li     if (T == Context.getObjCInstanceType()) {
1255*67e74705SXin Li       return Context.getAttributedType(
1256*67e74705SXin Li                AttributedType::getNullabilityAttrKind(*nullability),
1257*67e74705SXin Li                Context.getObjCIdType(),
1258*67e74705SXin Li                Context.getObjCIdType());
1259*67e74705SXin Li     }
1260*67e74705SXin Li 
1261*67e74705SXin Li     return origType;
1262*67e74705SXin Li   }
1263*67e74705SXin Li 
1264*67e74705SXin Li   if (T == Context.getObjCInstanceType())
1265*67e74705SXin Li     return Context.getObjCIdType();
1266*67e74705SXin Li 
1267*67e74705SXin Li   return origType;
1268*67e74705SXin Li }
1269*67e74705SXin Li 
1270*67e74705SXin Li /// Determine the result type of a message send based on the receiver type,
1271*67e74705SXin Li /// method, and the kind of message send.
1272*67e74705SXin Li ///
1273*67e74705SXin Li /// This is the "base" result type, which will still need to be adjusted
1274*67e74705SXin Li /// to account for nullability.
getBaseMessageSendResultType(Sema & S,QualType ReceiverType,ObjCMethodDecl * Method,bool isClassMessage,bool isSuperMessage)1275*67e74705SXin Li static QualType getBaseMessageSendResultType(Sema &S,
1276*67e74705SXin Li                                              QualType ReceiverType,
1277*67e74705SXin Li                                              ObjCMethodDecl *Method,
1278*67e74705SXin Li                                              bool isClassMessage,
1279*67e74705SXin Li                                              bool isSuperMessage) {
1280*67e74705SXin Li   assert(Method && "Must have a method");
1281*67e74705SXin Li   if (!Method->hasRelatedResultType())
1282*67e74705SXin Li     return Method->getSendResultType(ReceiverType);
1283*67e74705SXin Li 
1284*67e74705SXin Li   ASTContext &Context = S.Context;
1285*67e74705SXin Li 
1286*67e74705SXin Li   // Local function that transfers the nullability of the method's
1287*67e74705SXin Li   // result type to the returned result.
1288*67e74705SXin Li   auto transferNullability = [&](QualType type) -> QualType {
1289*67e74705SXin Li     // If the method's result type has nullability, extract it.
1290*67e74705SXin Li     if (auto nullability = Method->getSendResultType(ReceiverType)
1291*67e74705SXin Li                              ->getNullability(Context)){
1292*67e74705SXin Li       // Strip off any outer nullability sugar from the provided type.
1293*67e74705SXin Li       (void)AttributedType::stripOuterNullability(type);
1294*67e74705SXin Li 
1295*67e74705SXin Li       // Form a new attributed type using the method result type's nullability.
1296*67e74705SXin Li       return Context.getAttributedType(
1297*67e74705SXin Li                AttributedType::getNullabilityAttrKind(*nullability),
1298*67e74705SXin Li                type,
1299*67e74705SXin Li                type);
1300*67e74705SXin Li     }
1301*67e74705SXin Li 
1302*67e74705SXin Li     return type;
1303*67e74705SXin Li   };
1304*67e74705SXin Li 
1305*67e74705SXin Li   // If a method has a related return type:
1306*67e74705SXin Li   //   - if the method found is an instance method, but the message send
1307*67e74705SXin Li   //     was a class message send, T is the declared return type of the method
1308*67e74705SXin Li   //     found
1309*67e74705SXin Li   if (Method->isInstanceMethod() && isClassMessage)
1310*67e74705SXin Li     return stripObjCInstanceType(Context,
1311*67e74705SXin Li                                  Method->getSendResultType(ReceiverType));
1312*67e74705SXin Li 
1313*67e74705SXin Li   //   - if the receiver is super, T is a pointer to the class of the
1314*67e74705SXin Li   //     enclosing method definition
1315*67e74705SXin Li   if (isSuperMessage) {
1316*67e74705SXin Li     if (ObjCMethodDecl *CurMethod = S.getCurMethodDecl())
1317*67e74705SXin Li       if (ObjCInterfaceDecl *Class = CurMethod->getClassInterface()) {
1318*67e74705SXin Li         return transferNullability(
1319*67e74705SXin Li                  Context.getObjCObjectPointerType(
1320*67e74705SXin Li                    Context.getObjCInterfaceType(Class)));
1321*67e74705SXin Li       }
1322*67e74705SXin Li   }
1323*67e74705SXin Li 
1324*67e74705SXin Li   //   - if the receiver is the name of a class U, T is a pointer to U
1325*67e74705SXin Li   if (ReceiverType->getAsObjCInterfaceType())
1326*67e74705SXin Li     return transferNullability(Context.getObjCObjectPointerType(ReceiverType));
1327*67e74705SXin Li   //   - if the receiver is of type Class or qualified Class type,
1328*67e74705SXin Li   //     T is the declared return type of the method.
1329*67e74705SXin Li   if (ReceiverType->isObjCClassType() ||
1330*67e74705SXin Li       ReceiverType->isObjCQualifiedClassType())
1331*67e74705SXin Li     return stripObjCInstanceType(Context,
1332*67e74705SXin Li                                  Method->getSendResultType(ReceiverType));
1333*67e74705SXin Li 
1334*67e74705SXin Li   //   - if the receiver is id, qualified id, Class, or qualified Class, T
1335*67e74705SXin Li   //     is the receiver type, otherwise
1336*67e74705SXin Li   //   - T is the type of the receiver expression.
1337*67e74705SXin Li   return transferNullability(ReceiverType);
1338*67e74705SXin Li }
1339*67e74705SXin Li 
getMessageSendResultType(QualType ReceiverType,ObjCMethodDecl * Method,bool isClassMessage,bool isSuperMessage)1340*67e74705SXin Li QualType Sema::getMessageSendResultType(QualType ReceiverType,
1341*67e74705SXin Li                                         ObjCMethodDecl *Method,
1342*67e74705SXin Li                                         bool isClassMessage,
1343*67e74705SXin Li                                         bool isSuperMessage) {
1344*67e74705SXin Li   // Produce the result type.
1345*67e74705SXin Li   QualType resultType = getBaseMessageSendResultType(*this, ReceiverType,
1346*67e74705SXin Li                                                      Method,
1347*67e74705SXin Li                                                      isClassMessage,
1348*67e74705SXin Li                                                      isSuperMessage);
1349*67e74705SXin Li 
1350*67e74705SXin Li   // If this is a class message, ignore the nullability of the receiver.
1351*67e74705SXin Li   if (isClassMessage)
1352*67e74705SXin Li     return resultType;
1353*67e74705SXin Li 
1354*67e74705SXin Li   // Map the nullability of the result into a table index.
1355*67e74705SXin Li   unsigned receiverNullabilityIdx = 0;
1356*67e74705SXin Li   if (auto nullability = ReceiverType->getNullability(Context))
1357*67e74705SXin Li     receiverNullabilityIdx = 1 + static_cast<unsigned>(*nullability);
1358*67e74705SXin Li 
1359*67e74705SXin Li   unsigned resultNullabilityIdx = 0;
1360*67e74705SXin Li   if (auto nullability = resultType->getNullability(Context))
1361*67e74705SXin Li     resultNullabilityIdx = 1 + static_cast<unsigned>(*nullability);
1362*67e74705SXin Li 
1363*67e74705SXin Li   // The table of nullability mappings, indexed by the receiver's nullability
1364*67e74705SXin Li   // and then the result type's nullability.
1365*67e74705SXin Li   static const uint8_t None = 0;
1366*67e74705SXin Li   static const uint8_t NonNull = 1;
1367*67e74705SXin Li   static const uint8_t Nullable = 2;
1368*67e74705SXin Li   static const uint8_t Unspecified = 3;
1369*67e74705SXin Li   static const uint8_t nullabilityMap[4][4] = {
1370*67e74705SXin Li     //                  None        NonNull       Nullable    Unspecified
1371*67e74705SXin Li     /* None */        { None,       None,         Nullable,   None },
1372*67e74705SXin Li     /* NonNull */     { None,       NonNull,      Nullable,   Unspecified },
1373*67e74705SXin Li     /* Nullable */    { Nullable,   Nullable,     Nullable,   Nullable },
1374*67e74705SXin Li     /* Unspecified */ { None,       Unspecified,  Nullable,   Unspecified }
1375*67e74705SXin Li   };
1376*67e74705SXin Li 
1377*67e74705SXin Li   unsigned newResultNullabilityIdx
1378*67e74705SXin Li     = nullabilityMap[receiverNullabilityIdx][resultNullabilityIdx];
1379*67e74705SXin Li   if (newResultNullabilityIdx == resultNullabilityIdx)
1380*67e74705SXin Li     return resultType;
1381*67e74705SXin Li 
1382*67e74705SXin Li   // Strip off the existing nullability. This removes as little type sugar as
1383*67e74705SXin Li   // possible.
1384*67e74705SXin Li   do {
1385*67e74705SXin Li     if (auto attributed = dyn_cast<AttributedType>(resultType.getTypePtr())) {
1386*67e74705SXin Li       resultType = attributed->getModifiedType();
1387*67e74705SXin Li     } else {
1388*67e74705SXin Li       resultType = resultType.getDesugaredType(Context);
1389*67e74705SXin Li     }
1390*67e74705SXin Li   } while (resultType->getNullability(Context));
1391*67e74705SXin Li 
1392*67e74705SXin Li   // Add nullability back if needed.
1393*67e74705SXin Li   if (newResultNullabilityIdx > 0) {
1394*67e74705SXin Li     auto newNullability
1395*67e74705SXin Li       = static_cast<NullabilityKind>(newResultNullabilityIdx-1);
1396*67e74705SXin Li     return Context.getAttributedType(
1397*67e74705SXin Li              AttributedType::getNullabilityAttrKind(newNullability),
1398*67e74705SXin Li              resultType, resultType);
1399*67e74705SXin Li   }
1400*67e74705SXin Li 
1401*67e74705SXin Li   return resultType;
1402*67e74705SXin Li }
1403*67e74705SXin Li 
1404*67e74705SXin Li /// Look for an ObjC method whose result type exactly matches the given type.
1405*67e74705SXin Li static const ObjCMethodDecl *
findExplicitInstancetypeDeclarer(const ObjCMethodDecl * MD,QualType instancetype)1406*67e74705SXin Li findExplicitInstancetypeDeclarer(const ObjCMethodDecl *MD,
1407*67e74705SXin Li                                  QualType instancetype) {
1408*67e74705SXin Li   if (MD->getReturnType() == instancetype)
1409*67e74705SXin Li     return MD;
1410*67e74705SXin Li 
1411*67e74705SXin Li   // For these purposes, a method in an @implementation overrides a
1412*67e74705SXin Li   // declaration in the @interface.
1413*67e74705SXin Li   if (const ObjCImplDecl *impl =
1414*67e74705SXin Li         dyn_cast<ObjCImplDecl>(MD->getDeclContext())) {
1415*67e74705SXin Li     const ObjCContainerDecl *iface;
1416*67e74705SXin Li     if (const ObjCCategoryImplDecl *catImpl =
1417*67e74705SXin Li           dyn_cast<ObjCCategoryImplDecl>(impl)) {
1418*67e74705SXin Li       iface = catImpl->getCategoryDecl();
1419*67e74705SXin Li     } else {
1420*67e74705SXin Li       iface = impl->getClassInterface();
1421*67e74705SXin Li     }
1422*67e74705SXin Li 
1423*67e74705SXin Li     const ObjCMethodDecl *ifaceMD =
1424*67e74705SXin Li       iface->getMethod(MD->getSelector(), MD->isInstanceMethod());
1425*67e74705SXin Li     if (ifaceMD) return findExplicitInstancetypeDeclarer(ifaceMD, instancetype);
1426*67e74705SXin Li   }
1427*67e74705SXin Li 
1428*67e74705SXin Li   SmallVector<const ObjCMethodDecl *, 4> overrides;
1429*67e74705SXin Li   MD->getOverriddenMethods(overrides);
1430*67e74705SXin Li   for (unsigned i = 0, e = overrides.size(); i != e; ++i) {
1431*67e74705SXin Li     if (const ObjCMethodDecl *result =
1432*67e74705SXin Li           findExplicitInstancetypeDeclarer(overrides[i], instancetype))
1433*67e74705SXin Li       return result;
1434*67e74705SXin Li   }
1435*67e74705SXin Li 
1436*67e74705SXin Li   return nullptr;
1437*67e74705SXin Li }
1438*67e74705SXin Li 
EmitRelatedResultTypeNoteForReturn(QualType destType)1439*67e74705SXin Li void Sema::EmitRelatedResultTypeNoteForReturn(QualType destType) {
1440*67e74705SXin Li   // Only complain if we're in an ObjC method and the required return
1441*67e74705SXin Li   // type doesn't match the method's declared return type.
1442*67e74705SXin Li   ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(CurContext);
1443*67e74705SXin Li   if (!MD || !MD->hasRelatedResultType() ||
1444*67e74705SXin Li       Context.hasSameUnqualifiedType(destType, MD->getReturnType()))
1445*67e74705SXin Li     return;
1446*67e74705SXin Li 
1447*67e74705SXin Li   // Look for a method overridden by this method which explicitly uses
1448*67e74705SXin Li   // 'instancetype'.
1449*67e74705SXin Li   if (const ObjCMethodDecl *overridden =
1450*67e74705SXin Li         findExplicitInstancetypeDeclarer(MD, Context.getObjCInstanceType())) {
1451*67e74705SXin Li     SourceRange range = overridden->getReturnTypeSourceRange();
1452*67e74705SXin Li     SourceLocation loc = range.getBegin();
1453*67e74705SXin Li     if (loc.isInvalid())
1454*67e74705SXin Li       loc = overridden->getLocation();
1455*67e74705SXin Li     Diag(loc, diag::note_related_result_type_explicit)
1456*67e74705SXin Li       << /*current method*/ 1 << range;
1457*67e74705SXin Li     return;
1458*67e74705SXin Li   }
1459*67e74705SXin Li 
1460*67e74705SXin Li   // Otherwise, if we have an interesting method family, note that.
1461*67e74705SXin Li   // This should always trigger if the above didn't.
1462*67e74705SXin Li   if (ObjCMethodFamily family = MD->getMethodFamily())
1463*67e74705SXin Li     Diag(MD->getLocation(), diag::note_related_result_type_family)
1464*67e74705SXin Li       << /*current method*/ 1
1465*67e74705SXin Li       << family;
1466*67e74705SXin Li }
1467*67e74705SXin Li 
EmitRelatedResultTypeNote(const Expr * E)1468*67e74705SXin Li void Sema::EmitRelatedResultTypeNote(const Expr *E) {
1469*67e74705SXin Li   E = E->IgnoreParenImpCasts();
1470*67e74705SXin Li   const ObjCMessageExpr *MsgSend = dyn_cast<ObjCMessageExpr>(E);
1471*67e74705SXin Li   if (!MsgSend)
1472*67e74705SXin Li     return;
1473*67e74705SXin Li 
1474*67e74705SXin Li   const ObjCMethodDecl *Method = MsgSend->getMethodDecl();
1475*67e74705SXin Li   if (!Method)
1476*67e74705SXin Li     return;
1477*67e74705SXin Li 
1478*67e74705SXin Li   if (!Method->hasRelatedResultType())
1479*67e74705SXin Li     return;
1480*67e74705SXin Li 
1481*67e74705SXin Li   if (Context.hasSameUnqualifiedType(
1482*67e74705SXin Li           Method->getReturnType().getNonReferenceType(), MsgSend->getType()))
1483*67e74705SXin Li     return;
1484*67e74705SXin Li 
1485*67e74705SXin Li   if (!Context.hasSameUnqualifiedType(Method->getReturnType(),
1486*67e74705SXin Li                                       Context.getObjCInstanceType()))
1487*67e74705SXin Li     return;
1488*67e74705SXin Li 
1489*67e74705SXin Li   Diag(Method->getLocation(), diag::note_related_result_type_inferred)
1490*67e74705SXin Li     << Method->isInstanceMethod() << Method->getSelector()
1491*67e74705SXin Li     << MsgSend->getType();
1492*67e74705SXin Li }
1493*67e74705SXin Li 
CheckMessageArgumentTypes(QualType ReceiverType,MultiExprArg Args,Selector Sel,ArrayRef<SourceLocation> SelectorLocs,ObjCMethodDecl * Method,bool isClassMessage,bool isSuperMessage,SourceLocation lbrac,SourceLocation rbrac,SourceRange RecRange,QualType & ReturnType,ExprValueKind & VK)1494*67e74705SXin Li bool Sema::CheckMessageArgumentTypes(QualType ReceiverType,
1495*67e74705SXin Li                                      MultiExprArg Args,
1496*67e74705SXin Li                                      Selector Sel,
1497*67e74705SXin Li                                      ArrayRef<SourceLocation> SelectorLocs,
1498*67e74705SXin Li                                      ObjCMethodDecl *Method,
1499*67e74705SXin Li                                      bool isClassMessage, bool isSuperMessage,
1500*67e74705SXin Li                                      SourceLocation lbrac, SourceLocation rbrac,
1501*67e74705SXin Li                                      SourceRange RecRange,
1502*67e74705SXin Li                                      QualType &ReturnType, ExprValueKind &VK) {
1503*67e74705SXin Li   SourceLocation SelLoc;
1504*67e74705SXin Li   if (!SelectorLocs.empty() && SelectorLocs.front().isValid())
1505*67e74705SXin Li     SelLoc = SelectorLocs.front();
1506*67e74705SXin Li   else
1507*67e74705SXin Li     SelLoc = lbrac;
1508*67e74705SXin Li 
1509*67e74705SXin Li   if (!Method) {
1510*67e74705SXin Li     // Apply default argument promotion as for (C99 6.5.2.2p6).
1511*67e74705SXin Li     for (unsigned i = 0, e = Args.size(); i != e; i++) {
1512*67e74705SXin Li       if (Args[i]->isTypeDependent())
1513*67e74705SXin Li         continue;
1514*67e74705SXin Li 
1515*67e74705SXin Li       ExprResult result;
1516*67e74705SXin Li       if (getLangOpts().DebuggerSupport) {
1517*67e74705SXin Li         QualType paramTy; // ignored
1518*67e74705SXin Li         result = checkUnknownAnyArg(SelLoc, Args[i], paramTy);
1519*67e74705SXin Li       } else {
1520*67e74705SXin Li         result = DefaultArgumentPromotion(Args[i]);
1521*67e74705SXin Li       }
1522*67e74705SXin Li       if (result.isInvalid())
1523*67e74705SXin Li         return true;
1524*67e74705SXin Li       Args[i] = result.get();
1525*67e74705SXin Li     }
1526*67e74705SXin Li 
1527*67e74705SXin Li     unsigned DiagID;
1528*67e74705SXin Li     if (getLangOpts().ObjCAutoRefCount)
1529*67e74705SXin Li       DiagID = diag::err_arc_method_not_found;
1530*67e74705SXin Li     else
1531*67e74705SXin Li       DiagID = isClassMessage ? diag::warn_class_method_not_found
1532*67e74705SXin Li                               : diag::warn_inst_method_not_found;
1533*67e74705SXin Li     if (!getLangOpts().DebuggerSupport) {
1534*67e74705SXin Li       const ObjCMethodDecl *OMD = SelectorsForTypoCorrection(Sel, ReceiverType);
1535*67e74705SXin Li       if (OMD && !OMD->isInvalidDecl()) {
1536*67e74705SXin Li         if (getLangOpts().ObjCAutoRefCount)
1537*67e74705SXin Li           DiagID = diag::error_method_not_found_with_typo;
1538*67e74705SXin Li         else
1539*67e74705SXin Li           DiagID = isClassMessage ? diag::warn_class_method_not_found_with_typo
1540*67e74705SXin Li                                   : diag::warn_instance_method_not_found_with_typo;
1541*67e74705SXin Li         Selector MatchedSel = OMD->getSelector();
1542*67e74705SXin Li         SourceRange SelectorRange(SelectorLocs.front(), SelectorLocs.back());
1543*67e74705SXin Li         if (MatchedSel.isUnarySelector())
1544*67e74705SXin Li           Diag(SelLoc, DiagID)
1545*67e74705SXin Li             << Sel<< isClassMessage << MatchedSel
1546*67e74705SXin Li             << FixItHint::CreateReplacement(SelectorRange, MatchedSel.getAsString());
1547*67e74705SXin Li         else
1548*67e74705SXin Li           Diag(SelLoc, DiagID) << Sel<< isClassMessage << MatchedSel;
1549*67e74705SXin Li       }
1550*67e74705SXin Li       else
1551*67e74705SXin Li         Diag(SelLoc, DiagID)
1552*67e74705SXin Li           << Sel << isClassMessage << SourceRange(SelectorLocs.front(),
1553*67e74705SXin Li                                                 SelectorLocs.back());
1554*67e74705SXin Li       // Find the class to which we are sending this message.
1555*67e74705SXin Li       if (ReceiverType->isObjCObjectPointerType()) {
1556*67e74705SXin Li         if (ObjCInterfaceDecl *ThisClass =
1557*67e74705SXin Li             ReceiverType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()) {
1558*67e74705SXin Li           Diag(ThisClass->getLocation(), diag::note_receiver_class_declared);
1559*67e74705SXin Li           if (!RecRange.isInvalid())
1560*67e74705SXin Li             if (ThisClass->lookupClassMethod(Sel))
1561*67e74705SXin Li               Diag(RecRange.getBegin(),diag::note_receiver_expr_here)
1562*67e74705SXin Li                 << FixItHint::CreateReplacement(RecRange,
1563*67e74705SXin Li                                                 ThisClass->getNameAsString());
1564*67e74705SXin Li         }
1565*67e74705SXin Li       }
1566*67e74705SXin Li     }
1567*67e74705SXin Li 
1568*67e74705SXin Li     // In debuggers, we want to use __unknown_anytype for these
1569*67e74705SXin Li     // results so that clients can cast them.
1570*67e74705SXin Li     if (getLangOpts().DebuggerSupport) {
1571*67e74705SXin Li       ReturnType = Context.UnknownAnyTy;
1572*67e74705SXin Li     } else {
1573*67e74705SXin Li       ReturnType = Context.getObjCIdType();
1574*67e74705SXin Li     }
1575*67e74705SXin Li     VK = VK_RValue;
1576*67e74705SXin Li     return false;
1577*67e74705SXin Li   }
1578*67e74705SXin Li 
1579*67e74705SXin Li   ReturnType = getMessageSendResultType(ReceiverType, Method, isClassMessage,
1580*67e74705SXin Li                                         isSuperMessage);
1581*67e74705SXin Li   VK = Expr::getValueKindForType(Method->getReturnType());
1582*67e74705SXin Li 
1583*67e74705SXin Li   unsigned NumNamedArgs = Sel.getNumArgs();
1584*67e74705SXin Li   // Method might have more arguments than selector indicates. This is due
1585*67e74705SXin Li   // to addition of c-style arguments in method.
1586*67e74705SXin Li   if (Method->param_size() > Sel.getNumArgs())
1587*67e74705SXin Li     NumNamedArgs = Method->param_size();
1588*67e74705SXin Li   // FIXME. This need be cleaned up.
1589*67e74705SXin Li   if (Args.size() < NumNamedArgs) {
1590*67e74705SXin Li     Diag(SelLoc, diag::err_typecheck_call_too_few_args)
1591*67e74705SXin Li       << 2 << NumNamedArgs << static_cast<unsigned>(Args.size());
1592*67e74705SXin Li     return false;
1593*67e74705SXin Li   }
1594*67e74705SXin Li 
1595*67e74705SXin Li   // Compute the set of type arguments to be substituted into each parameter
1596*67e74705SXin Li   // type.
1597*67e74705SXin Li   Optional<ArrayRef<QualType>> typeArgs
1598*67e74705SXin Li     = ReceiverType->getObjCSubstitutions(Method->getDeclContext());
1599*67e74705SXin Li   bool IsError = false;
1600*67e74705SXin Li   for (unsigned i = 0; i < NumNamedArgs; i++) {
1601*67e74705SXin Li     // We can't do any type-checking on a type-dependent argument.
1602*67e74705SXin Li     if (Args[i]->isTypeDependent())
1603*67e74705SXin Li       continue;
1604*67e74705SXin Li 
1605*67e74705SXin Li     Expr *argExpr = Args[i];
1606*67e74705SXin Li 
1607*67e74705SXin Li     ParmVarDecl *param = Method->parameters()[i];
1608*67e74705SXin Li     assert(argExpr && "CheckMessageArgumentTypes(): missing expression");
1609*67e74705SXin Li 
1610*67e74705SXin Li     // Strip the unbridged-cast placeholder expression off unless it's
1611*67e74705SXin Li     // a consumed argument.
1612*67e74705SXin Li     if (argExpr->hasPlaceholderType(BuiltinType::ARCUnbridgedCast) &&
1613*67e74705SXin Li         !param->hasAttr<CFConsumedAttr>())
1614*67e74705SXin Li       argExpr = stripARCUnbridgedCast(argExpr);
1615*67e74705SXin Li 
1616*67e74705SXin Li     // If the parameter is __unknown_anytype, infer its type
1617*67e74705SXin Li     // from the argument.
1618*67e74705SXin Li     if (param->getType() == Context.UnknownAnyTy) {
1619*67e74705SXin Li       QualType paramType;
1620*67e74705SXin Li       ExprResult argE = checkUnknownAnyArg(SelLoc, argExpr, paramType);
1621*67e74705SXin Li       if (argE.isInvalid()) {
1622*67e74705SXin Li         IsError = true;
1623*67e74705SXin Li       } else {
1624*67e74705SXin Li         Args[i] = argE.get();
1625*67e74705SXin Li 
1626*67e74705SXin Li         // Update the parameter type in-place.
1627*67e74705SXin Li         param->setType(paramType);
1628*67e74705SXin Li       }
1629*67e74705SXin Li       continue;
1630*67e74705SXin Li     }
1631*67e74705SXin Li 
1632*67e74705SXin Li     QualType origParamType = param->getType();
1633*67e74705SXin Li     QualType paramType = param->getType();
1634*67e74705SXin Li     if (typeArgs)
1635*67e74705SXin Li       paramType = paramType.substObjCTypeArgs(
1636*67e74705SXin Li                     Context,
1637*67e74705SXin Li                     *typeArgs,
1638*67e74705SXin Li                     ObjCSubstitutionContext::Parameter);
1639*67e74705SXin Li 
1640*67e74705SXin Li     if (RequireCompleteType(argExpr->getSourceRange().getBegin(),
1641*67e74705SXin Li                             paramType,
1642*67e74705SXin Li                             diag::err_call_incomplete_argument, argExpr))
1643*67e74705SXin Li       return true;
1644*67e74705SXin Li 
1645*67e74705SXin Li     InitializedEntity Entity
1646*67e74705SXin Li       = InitializedEntity::InitializeParameter(Context, param, paramType);
1647*67e74705SXin Li     ExprResult ArgE = PerformCopyInitialization(Entity, SourceLocation(), argExpr);
1648*67e74705SXin Li     if (ArgE.isInvalid())
1649*67e74705SXin Li       IsError = true;
1650*67e74705SXin Li     else {
1651*67e74705SXin Li       Args[i] = ArgE.getAs<Expr>();
1652*67e74705SXin Li 
1653*67e74705SXin Li       // If we are type-erasing a block to a block-compatible
1654*67e74705SXin Li       // Objective-C pointer type, we may need to extend the lifetime
1655*67e74705SXin Li       // of the block object.
1656*67e74705SXin Li       if (typeArgs && Args[i]->isRValue() && paramType->isBlockPointerType() &&
1657*67e74705SXin Li           Args[i]->getType()->isBlockPointerType() &&
1658*67e74705SXin Li           origParamType->isObjCObjectPointerType()) {
1659*67e74705SXin Li         ExprResult arg = Args[i];
1660*67e74705SXin Li         maybeExtendBlockObject(arg);
1661*67e74705SXin Li         Args[i] = arg.get();
1662*67e74705SXin Li       }
1663*67e74705SXin Li     }
1664*67e74705SXin Li   }
1665*67e74705SXin Li 
1666*67e74705SXin Li   // Promote additional arguments to variadic methods.
1667*67e74705SXin Li   if (Method->isVariadic()) {
1668*67e74705SXin Li     for (unsigned i = NumNamedArgs, e = Args.size(); i < e; ++i) {
1669*67e74705SXin Li       if (Args[i]->isTypeDependent())
1670*67e74705SXin Li         continue;
1671*67e74705SXin Li 
1672*67e74705SXin Li       ExprResult Arg = DefaultVariadicArgumentPromotion(Args[i], VariadicMethod,
1673*67e74705SXin Li                                                         nullptr);
1674*67e74705SXin Li       IsError |= Arg.isInvalid();
1675*67e74705SXin Li       Args[i] = Arg.get();
1676*67e74705SXin Li     }
1677*67e74705SXin Li   } else {
1678*67e74705SXin Li     // Check for extra arguments to non-variadic methods.
1679*67e74705SXin Li     if (Args.size() != NumNamedArgs) {
1680*67e74705SXin Li       Diag(Args[NumNamedArgs]->getLocStart(),
1681*67e74705SXin Li            diag::err_typecheck_call_too_many_args)
1682*67e74705SXin Li         << 2 /*method*/ << NumNamedArgs << static_cast<unsigned>(Args.size())
1683*67e74705SXin Li         << Method->getSourceRange()
1684*67e74705SXin Li         << SourceRange(Args[NumNamedArgs]->getLocStart(),
1685*67e74705SXin Li                        Args.back()->getLocEnd());
1686*67e74705SXin Li     }
1687*67e74705SXin Li   }
1688*67e74705SXin Li 
1689*67e74705SXin Li   DiagnoseSentinelCalls(Method, SelLoc, Args);
1690*67e74705SXin Li 
1691*67e74705SXin Li   // Do additional checkings on method.
1692*67e74705SXin Li   IsError |= CheckObjCMethodCall(
1693*67e74705SXin Li       Method, SelLoc, makeArrayRef(Args.data(), Args.size()));
1694*67e74705SXin Li 
1695*67e74705SXin Li   return IsError;
1696*67e74705SXin Li }
1697*67e74705SXin Li 
isSelfExpr(Expr * RExpr)1698*67e74705SXin Li bool Sema::isSelfExpr(Expr *RExpr) {
1699*67e74705SXin Li   // 'self' is objc 'self' in an objc method only.
1700*67e74705SXin Li   ObjCMethodDecl *Method =
1701*67e74705SXin Li       dyn_cast_or_null<ObjCMethodDecl>(CurContext->getNonClosureAncestor());
1702*67e74705SXin Li   return isSelfExpr(RExpr, Method);
1703*67e74705SXin Li }
1704*67e74705SXin Li 
isSelfExpr(Expr * receiver,const ObjCMethodDecl * method)1705*67e74705SXin Li bool Sema::isSelfExpr(Expr *receiver, const ObjCMethodDecl *method) {
1706*67e74705SXin Li   if (!method) return false;
1707*67e74705SXin Li 
1708*67e74705SXin Li   receiver = receiver->IgnoreParenLValueCasts();
1709*67e74705SXin Li   if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(receiver))
1710*67e74705SXin Li     if (DRE->getDecl() == method->getSelfDecl())
1711*67e74705SXin Li       return true;
1712*67e74705SXin Li   return false;
1713*67e74705SXin Li }
1714*67e74705SXin Li 
1715*67e74705SXin Li /// LookupMethodInType - Look up a method in an ObjCObjectType.
LookupMethodInObjectType(Selector sel,QualType type,bool isInstance)1716*67e74705SXin Li ObjCMethodDecl *Sema::LookupMethodInObjectType(Selector sel, QualType type,
1717*67e74705SXin Li                                                bool isInstance) {
1718*67e74705SXin Li   const ObjCObjectType *objType = type->castAs<ObjCObjectType>();
1719*67e74705SXin Li   if (ObjCInterfaceDecl *iface = objType->getInterface()) {
1720*67e74705SXin Li     // Look it up in the main interface (and categories, etc.)
1721*67e74705SXin Li     if (ObjCMethodDecl *method = iface->lookupMethod(sel, isInstance))
1722*67e74705SXin Li       return method;
1723*67e74705SXin Li 
1724*67e74705SXin Li     // Okay, look for "private" methods declared in any
1725*67e74705SXin Li     // @implementations we've seen.
1726*67e74705SXin Li     if (ObjCMethodDecl *method = iface->lookupPrivateMethod(sel, isInstance))
1727*67e74705SXin Li       return method;
1728*67e74705SXin Li   }
1729*67e74705SXin Li 
1730*67e74705SXin Li   // Check qualifiers.
1731*67e74705SXin Li   for (const auto *I : objType->quals())
1732*67e74705SXin Li     if (ObjCMethodDecl *method = I->lookupMethod(sel, isInstance))
1733*67e74705SXin Li       return method;
1734*67e74705SXin Li 
1735*67e74705SXin Li   return nullptr;
1736*67e74705SXin Li }
1737*67e74705SXin Li 
1738*67e74705SXin Li /// LookupMethodInQualifiedType - Lookups up a method in protocol qualifier
1739*67e74705SXin Li /// list of a qualified objective pointer type.
LookupMethodInQualifiedType(Selector Sel,const ObjCObjectPointerType * OPT,bool Instance)1740*67e74705SXin Li ObjCMethodDecl *Sema::LookupMethodInQualifiedType(Selector Sel,
1741*67e74705SXin Li                                               const ObjCObjectPointerType *OPT,
1742*67e74705SXin Li                                               bool Instance)
1743*67e74705SXin Li {
1744*67e74705SXin Li   ObjCMethodDecl *MD = nullptr;
1745*67e74705SXin Li   for (const auto *PROTO : OPT->quals()) {
1746*67e74705SXin Li     if ((MD = PROTO->lookupMethod(Sel, Instance))) {
1747*67e74705SXin Li       return MD;
1748*67e74705SXin Li     }
1749*67e74705SXin Li   }
1750*67e74705SXin Li   return nullptr;
1751*67e74705SXin Li }
1752*67e74705SXin Li 
1753*67e74705SXin Li /// HandleExprPropertyRefExpr - Handle foo.bar where foo is a pointer to an
1754*67e74705SXin Li /// objective C interface.  This is a property reference expression.
1755*67e74705SXin Li ExprResult Sema::
HandleExprPropertyRefExpr(const ObjCObjectPointerType * OPT,Expr * BaseExpr,SourceLocation OpLoc,DeclarationName MemberName,SourceLocation MemberLoc,SourceLocation SuperLoc,QualType SuperType,bool Super)1756*67e74705SXin Li HandleExprPropertyRefExpr(const ObjCObjectPointerType *OPT,
1757*67e74705SXin Li                           Expr *BaseExpr, SourceLocation OpLoc,
1758*67e74705SXin Li                           DeclarationName MemberName,
1759*67e74705SXin Li                           SourceLocation MemberLoc,
1760*67e74705SXin Li                           SourceLocation SuperLoc, QualType SuperType,
1761*67e74705SXin Li                           bool Super) {
1762*67e74705SXin Li   const ObjCInterfaceType *IFaceT = OPT->getInterfaceType();
1763*67e74705SXin Li   ObjCInterfaceDecl *IFace = IFaceT->getDecl();
1764*67e74705SXin Li 
1765*67e74705SXin Li   if (!MemberName.isIdentifier()) {
1766*67e74705SXin Li     Diag(MemberLoc, diag::err_invalid_property_name)
1767*67e74705SXin Li       << MemberName << QualType(OPT, 0);
1768*67e74705SXin Li     return ExprError();
1769*67e74705SXin Li   }
1770*67e74705SXin Li 
1771*67e74705SXin Li   IdentifierInfo *Member = MemberName.getAsIdentifierInfo();
1772*67e74705SXin Li 
1773*67e74705SXin Li   SourceRange BaseRange = Super? SourceRange(SuperLoc)
1774*67e74705SXin Li                                : BaseExpr->getSourceRange();
1775*67e74705SXin Li   if (RequireCompleteType(MemberLoc, OPT->getPointeeType(),
1776*67e74705SXin Li                           diag::err_property_not_found_forward_class,
1777*67e74705SXin Li                           MemberName, BaseRange))
1778*67e74705SXin Li     return ExprError();
1779*67e74705SXin Li 
1780*67e74705SXin Li   if (ObjCPropertyDecl *PD = IFace->FindPropertyDeclaration(
1781*67e74705SXin Li           Member, ObjCPropertyQueryKind::OBJC_PR_query_instance)) {
1782*67e74705SXin Li     // Check whether we can reference this property.
1783*67e74705SXin Li     if (DiagnoseUseOfDecl(PD, MemberLoc))
1784*67e74705SXin Li       return ExprError();
1785*67e74705SXin Li     if (Super)
1786*67e74705SXin Li       return new (Context)
1787*67e74705SXin Li           ObjCPropertyRefExpr(PD, Context.PseudoObjectTy, VK_LValue,
1788*67e74705SXin Li                               OK_ObjCProperty, MemberLoc, SuperLoc, SuperType);
1789*67e74705SXin Li     else
1790*67e74705SXin Li       return new (Context)
1791*67e74705SXin Li           ObjCPropertyRefExpr(PD, Context.PseudoObjectTy, VK_LValue,
1792*67e74705SXin Li                               OK_ObjCProperty, MemberLoc, BaseExpr);
1793*67e74705SXin Li   }
1794*67e74705SXin Li   // Check protocols on qualified interfaces.
1795*67e74705SXin Li   for (const auto *I : OPT->quals())
1796*67e74705SXin Li     if (ObjCPropertyDecl *PD = I->FindPropertyDeclaration(
1797*67e74705SXin Li             Member, ObjCPropertyQueryKind::OBJC_PR_query_instance)) {
1798*67e74705SXin Li       // Check whether we can reference this property.
1799*67e74705SXin Li       if (DiagnoseUseOfDecl(PD, MemberLoc))
1800*67e74705SXin Li         return ExprError();
1801*67e74705SXin Li 
1802*67e74705SXin Li       if (Super)
1803*67e74705SXin Li         return new (Context) ObjCPropertyRefExpr(
1804*67e74705SXin Li             PD, Context.PseudoObjectTy, VK_LValue, OK_ObjCProperty, MemberLoc,
1805*67e74705SXin Li             SuperLoc, SuperType);
1806*67e74705SXin Li       else
1807*67e74705SXin Li         return new (Context)
1808*67e74705SXin Li             ObjCPropertyRefExpr(PD, Context.PseudoObjectTy, VK_LValue,
1809*67e74705SXin Li                                 OK_ObjCProperty, MemberLoc, BaseExpr);
1810*67e74705SXin Li     }
1811*67e74705SXin Li   // If that failed, look for an "implicit" property by seeing if the nullary
1812*67e74705SXin Li   // selector is implemented.
1813*67e74705SXin Li 
1814*67e74705SXin Li   // FIXME: The logic for looking up nullary and unary selectors should be
1815*67e74705SXin Li   // shared with the code in ActOnInstanceMessage.
1816*67e74705SXin Li 
1817*67e74705SXin Li   Selector Sel = PP.getSelectorTable().getNullarySelector(Member);
1818*67e74705SXin Li   ObjCMethodDecl *Getter = IFace->lookupInstanceMethod(Sel);
1819*67e74705SXin Li 
1820*67e74705SXin Li   // May be found in property's qualified list.
1821*67e74705SXin Li   if (!Getter)
1822*67e74705SXin Li     Getter = LookupMethodInQualifiedType(Sel, OPT, true);
1823*67e74705SXin Li 
1824*67e74705SXin Li   // If this reference is in an @implementation, check for 'private' methods.
1825*67e74705SXin Li   if (!Getter)
1826*67e74705SXin Li     Getter = IFace->lookupPrivateMethod(Sel);
1827*67e74705SXin Li 
1828*67e74705SXin Li   if (Getter) {
1829*67e74705SXin Li     // Check if we can reference this property.
1830*67e74705SXin Li     if (DiagnoseUseOfDecl(Getter, MemberLoc))
1831*67e74705SXin Li       return ExprError();
1832*67e74705SXin Li   }
1833*67e74705SXin Li   // If we found a getter then this may be a valid dot-reference, we
1834*67e74705SXin Li   // will look for the matching setter, in case it is needed.
1835*67e74705SXin Li   Selector SetterSel =
1836*67e74705SXin Li     SelectorTable::constructSetterSelector(PP.getIdentifierTable(),
1837*67e74705SXin Li                                            PP.getSelectorTable(), Member);
1838*67e74705SXin Li   ObjCMethodDecl *Setter = IFace->lookupInstanceMethod(SetterSel);
1839*67e74705SXin Li 
1840*67e74705SXin Li   // May be found in property's qualified list.
1841*67e74705SXin Li   if (!Setter)
1842*67e74705SXin Li     Setter = LookupMethodInQualifiedType(SetterSel, OPT, true);
1843*67e74705SXin Li 
1844*67e74705SXin Li   if (!Setter) {
1845*67e74705SXin Li     // If this reference is in an @implementation, also check for 'private'
1846*67e74705SXin Li     // methods.
1847*67e74705SXin Li     Setter = IFace->lookupPrivateMethod(SetterSel);
1848*67e74705SXin Li   }
1849*67e74705SXin Li 
1850*67e74705SXin Li   if (Setter && DiagnoseUseOfDecl(Setter, MemberLoc))
1851*67e74705SXin Li     return ExprError();
1852*67e74705SXin Li 
1853*67e74705SXin Li   // Special warning if member name used in a property-dot for a setter accessor
1854*67e74705SXin Li   // does not use a property with same name; e.g. obj.X = ... for a property with
1855*67e74705SXin Li   // name 'x'.
1856*67e74705SXin Li   if (Setter && Setter->isImplicit() && Setter->isPropertyAccessor() &&
1857*67e74705SXin Li       !IFace->FindPropertyDeclaration(
1858*67e74705SXin Li           Member, ObjCPropertyQueryKind::OBJC_PR_query_instance)) {
1859*67e74705SXin Li       if (const ObjCPropertyDecl *PDecl = Setter->findPropertyDecl()) {
1860*67e74705SXin Li         // Do not warn if user is using property-dot syntax to make call to
1861*67e74705SXin Li         // user named setter.
1862*67e74705SXin Li         if (!(PDecl->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_setter))
1863*67e74705SXin Li           Diag(MemberLoc,
1864*67e74705SXin Li                diag::warn_property_access_suggest)
1865*67e74705SXin Li           << MemberName << QualType(OPT, 0) << PDecl->getName()
1866*67e74705SXin Li           << FixItHint::CreateReplacement(MemberLoc, PDecl->getName());
1867*67e74705SXin Li       }
1868*67e74705SXin Li   }
1869*67e74705SXin Li 
1870*67e74705SXin Li   if (Getter || Setter) {
1871*67e74705SXin Li     if (Super)
1872*67e74705SXin Li       return new (Context)
1873*67e74705SXin Li           ObjCPropertyRefExpr(Getter, Setter, Context.PseudoObjectTy, VK_LValue,
1874*67e74705SXin Li                               OK_ObjCProperty, MemberLoc, SuperLoc, SuperType);
1875*67e74705SXin Li     else
1876*67e74705SXin Li       return new (Context)
1877*67e74705SXin Li           ObjCPropertyRefExpr(Getter, Setter, Context.PseudoObjectTy, VK_LValue,
1878*67e74705SXin Li                               OK_ObjCProperty, MemberLoc, BaseExpr);
1879*67e74705SXin Li 
1880*67e74705SXin Li   }
1881*67e74705SXin Li 
1882*67e74705SXin Li   // Attempt to correct for typos in property names.
1883*67e74705SXin Li   if (TypoCorrection Corrected =
1884*67e74705SXin Li           CorrectTypo(DeclarationNameInfo(MemberName, MemberLoc),
1885*67e74705SXin Li                       LookupOrdinaryName, nullptr, nullptr,
1886*67e74705SXin Li                       llvm::make_unique<DeclFilterCCC<ObjCPropertyDecl>>(),
1887*67e74705SXin Li                       CTK_ErrorRecovery, IFace, false, OPT)) {
1888*67e74705SXin Li     DeclarationName TypoResult = Corrected.getCorrection();
1889*67e74705SXin Li     if (TypoResult.isIdentifier() &&
1890*67e74705SXin Li         TypoResult.getAsIdentifierInfo() == Member) {
1891*67e74705SXin Li       // There is no need to try the correction if it is the same.
1892*67e74705SXin Li       NamedDecl *ChosenDecl =
1893*67e74705SXin Li         Corrected.isKeyword() ? nullptr : Corrected.getFoundDecl();
1894*67e74705SXin Li       if (ChosenDecl && isa<ObjCPropertyDecl>(ChosenDecl))
1895*67e74705SXin Li         if (cast<ObjCPropertyDecl>(ChosenDecl)->isClassProperty()) {
1896*67e74705SXin Li           // This is a class property, we should not use the instance to
1897*67e74705SXin Li           // access it.
1898*67e74705SXin Li           Diag(MemberLoc, diag::err_class_property_found) << MemberName
1899*67e74705SXin Li           << OPT->getInterfaceDecl()->getName()
1900*67e74705SXin Li           << FixItHint::CreateReplacement(BaseExpr->getSourceRange(),
1901*67e74705SXin Li                                           OPT->getInterfaceDecl()->getName());
1902*67e74705SXin Li           return ExprError();
1903*67e74705SXin Li         }
1904*67e74705SXin Li     } else {
1905*67e74705SXin Li       diagnoseTypo(Corrected, PDiag(diag::err_property_not_found_suggest)
1906*67e74705SXin Li                                 << MemberName << QualType(OPT, 0));
1907*67e74705SXin Li       return HandleExprPropertyRefExpr(OPT, BaseExpr, OpLoc,
1908*67e74705SXin Li                                        TypoResult, MemberLoc,
1909*67e74705SXin Li                                        SuperLoc, SuperType, Super);
1910*67e74705SXin Li     }
1911*67e74705SXin Li   }
1912*67e74705SXin Li   ObjCInterfaceDecl *ClassDeclared;
1913*67e74705SXin Li   if (ObjCIvarDecl *Ivar =
1914*67e74705SXin Li       IFace->lookupInstanceVariable(Member, ClassDeclared)) {
1915*67e74705SXin Li     QualType T = Ivar->getType();
1916*67e74705SXin Li     if (const ObjCObjectPointerType * OBJPT =
1917*67e74705SXin Li         T->getAsObjCInterfacePointerType()) {
1918*67e74705SXin Li       if (RequireCompleteType(MemberLoc, OBJPT->getPointeeType(),
1919*67e74705SXin Li                               diag::err_property_not_as_forward_class,
1920*67e74705SXin Li                               MemberName, BaseExpr))
1921*67e74705SXin Li         return ExprError();
1922*67e74705SXin Li     }
1923*67e74705SXin Li     Diag(MemberLoc,
1924*67e74705SXin Li          diag::err_ivar_access_using_property_syntax_suggest)
1925*67e74705SXin Li     << MemberName << QualType(OPT, 0) << Ivar->getDeclName()
1926*67e74705SXin Li     << FixItHint::CreateReplacement(OpLoc, "->");
1927*67e74705SXin Li     return ExprError();
1928*67e74705SXin Li   }
1929*67e74705SXin Li 
1930*67e74705SXin Li   Diag(MemberLoc, diag::err_property_not_found)
1931*67e74705SXin Li     << MemberName << QualType(OPT, 0);
1932*67e74705SXin Li   if (Setter)
1933*67e74705SXin Li     Diag(Setter->getLocation(), diag::note_getter_unavailable)
1934*67e74705SXin Li           << MemberName << BaseExpr->getSourceRange();
1935*67e74705SXin Li   return ExprError();
1936*67e74705SXin Li }
1937*67e74705SXin Li 
1938*67e74705SXin Li ExprResult Sema::
ActOnClassPropertyRefExpr(IdentifierInfo & receiverName,IdentifierInfo & propertyName,SourceLocation receiverNameLoc,SourceLocation propertyNameLoc)1939*67e74705SXin Li ActOnClassPropertyRefExpr(IdentifierInfo &receiverName,
1940*67e74705SXin Li                           IdentifierInfo &propertyName,
1941*67e74705SXin Li                           SourceLocation receiverNameLoc,
1942*67e74705SXin Li                           SourceLocation propertyNameLoc) {
1943*67e74705SXin Li 
1944*67e74705SXin Li   IdentifierInfo *receiverNamePtr = &receiverName;
1945*67e74705SXin Li   ObjCInterfaceDecl *IFace = getObjCInterfaceDecl(receiverNamePtr,
1946*67e74705SXin Li                                                   receiverNameLoc);
1947*67e74705SXin Li 
1948*67e74705SXin Li   QualType SuperType;
1949*67e74705SXin Li   if (!IFace) {
1950*67e74705SXin Li     // If the "receiver" is 'super' in a method, handle it as an expression-like
1951*67e74705SXin Li     // property reference.
1952*67e74705SXin Li     if (receiverNamePtr->isStr("super")) {
1953*67e74705SXin Li       if (ObjCMethodDecl *CurMethod = tryCaptureObjCSelf(receiverNameLoc)) {
1954*67e74705SXin Li         if (auto classDecl = CurMethod->getClassInterface()) {
1955*67e74705SXin Li           SuperType = QualType(classDecl->getSuperClassType(), 0);
1956*67e74705SXin Li           if (CurMethod->isInstanceMethod()) {
1957*67e74705SXin Li             if (SuperType.isNull()) {
1958*67e74705SXin Li               // The current class does not have a superclass.
1959*67e74705SXin Li               Diag(receiverNameLoc, diag::error_root_class_cannot_use_super)
1960*67e74705SXin Li                 << CurMethod->getClassInterface()->getIdentifier();
1961*67e74705SXin Li               return ExprError();
1962*67e74705SXin Li             }
1963*67e74705SXin Li             QualType T = Context.getObjCObjectPointerType(SuperType);
1964*67e74705SXin Li 
1965*67e74705SXin Li             return HandleExprPropertyRefExpr(T->castAs<ObjCObjectPointerType>(),
1966*67e74705SXin Li                                              /*BaseExpr*/nullptr,
1967*67e74705SXin Li                                              SourceLocation()/*OpLoc*/,
1968*67e74705SXin Li                                              &propertyName,
1969*67e74705SXin Li                                              propertyNameLoc,
1970*67e74705SXin Li                                              receiverNameLoc, T, true);
1971*67e74705SXin Li           }
1972*67e74705SXin Li 
1973*67e74705SXin Li           // Otherwise, if this is a class method, try dispatching to our
1974*67e74705SXin Li           // superclass.
1975*67e74705SXin Li           IFace = CurMethod->getClassInterface()->getSuperClass();
1976*67e74705SXin Li         }
1977*67e74705SXin Li       }
1978*67e74705SXin Li     }
1979*67e74705SXin Li 
1980*67e74705SXin Li     if (!IFace) {
1981*67e74705SXin Li       Diag(receiverNameLoc, diag::err_expected_either) << tok::identifier
1982*67e74705SXin Li                                                        << tok::l_paren;
1983*67e74705SXin Li       return ExprError();
1984*67e74705SXin Li     }
1985*67e74705SXin Li   }
1986*67e74705SXin Li 
1987*67e74705SXin Li   // Search for a declared property first.
1988*67e74705SXin Li   Selector Sel = PP.getSelectorTable().getNullarySelector(&propertyName);
1989*67e74705SXin Li   ObjCMethodDecl *Getter = IFace->lookupClassMethod(Sel);
1990*67e74705SXin Li 
1991*67e74705SXin Li   // If this reference is in an @implementation, check for 'private' methods.
1992*67e74705SXin Li   if (!Getter)
1993*67e74705SXin Li     Getter = IFace->lookupPrivateClassMethod(Sel);
1994*67e74705SXin Li 
1995*67e74705SXin Li   if (Getter) {
1996*67e74705SXin Li     // FIXME: refactor/share with ActOnMemberReference().
1997*67e74705SXin Li     // Check if we can reference this property.
1998*67e74705SXin Li     if (DiagnoseUseOfDecl(Getter, propertyNameLoc))
1999*67e74705SXin Li       return ExprError();
2000*67e74705SXin Li   }
2001*67e74705SXin Li 
2002*67e74705SXin Li   // Look for the matching setter, in case it is needed.
2003*67e74705SXin Li   Selector SetterSel =
2004*67e74705SXin Li     SelectorTable::constructSetterSelector(PP.getIdentifierTable(),
2005*67e74705SXin Li                                             PP.getSelectorTable(),
2006*67e74705SXin Li                                            &propertyName);
2007*67e74705SXin Li 
2008*67e74705SXin Li   ObjCMethodDecl *Setter = IFace->lookupClassMethod(SetterSel);
2009*67e74705SXin Li   if (!Setter) {
2010*67e74705SXin Li     // If this reference is in an @implementation, also check for 'private'
2011*67e74705SXin Li     // methods.
2012*67e74705SXin Li     Setter = IFace->lookupPrivateClassMethod(SetterSel);
2013*67e74705SXin Li   }
2014*67e74705SXin Li   // Look through local category implementations associated with the class.
2015*67e74705SXin Li   if (!Setter)
2016*67e74705SXin Li     Setter = IFace->getCategoryClassMethod(SetterSel);
2017*67e74705SXin Li 
2018*67e74705SXin Li   if (Setter && DiagnoseUseOfDecl(Setter, propertyNameLoc))
2019*67e74705SXin Li     return ExprError();
2020*67e74705SXin Li 
2021*67e74705SXin Li   if (Getter || Setter) {
2022*67e74705SXin Li     if (!SuperType.isNull())
2023*67e74705SXin Li       return new (Context)
2024*67e74705SXin Li           ObjCPropertyRefExpr(Getter, Setter, Context.PseudoObjectTy, VK_LValue,
2025*67e74705SXin Li                               OK_ObjCProperty, propertyNameLoc, receiverNameLoc,
2026*67e74705SXin Li                               SuperType);
2027*67e74705SXin Li 
2028*67e74705SXin Li     return new (Context) ObjCPropertyRefExpr(
2029*67e74705SXin Li         Getter, Setter, Context.PseudoObjectTy, VK_LValue, OK_ObjCProperty,
2030*67e74705SXin Li         propertyNameLoc, receiverNameLoc, IFace);
2031*67e74705SXin Li   }
2032*67e74705SXin Li   return ExprError(Diag(propertyNameLoc, diag::err_property_not_found)
2033*67e74705SXin Li                      << &propertyName << Context.getObjCInterfaceType(IFace));
2034*67e74705SXin Li }
2035*67e74705SXin Li 
2036*67e74705SXin Li namespace {
2037*67e74705SXin Li 
2038*67e74705SXin Li class ObjCInterfaceOrSuperCCC : public CorrectionCandidateCallback {
2039*67e74705SXin Li  public:
ObjCInterfaceOrSuperCCC(ObjCMethodDecl * Method)2040*67e74705SXin Li   ObjCInterfaceOrSuperCCC(ObjCMethodDecl *Method) {
2041*67e74705SXin Li     // Determine whether "super" is acceptable in the current context.
2042*67e74705SXin Li     if (Method && Method->getClassInterface())
2043*67e74705SXin Li       WantObjCSuper = Method->getClassInterface()->getSuperClass();
2044*67e74705SXin Li   }
2045*67e74705SXin Li 
ValidateCandidate(const TypoCorrection & candidate)2046*67e74705SXin Li   bool ValidateCandidate(const TypoCorrection &candidate) override {
2047*67e74705SXin Li     return candidate.getCorrectionDeclAs<ObjCInterfaceDecl>() ||
2048*67e74705SXin Li         candidate.isKeyword("super");
2049*67e74705SXin Li   }
2050*67e74705SXin Li };
2051*67e74705SXin Li 
2052*67e74705SXin Li } // end anonymous namespace
2053*67e74705SXin Li 
getObjCMessageKind(Scope * S,IdentifierInfo * Name,SourceLocation NameLoc,bool IsSuper,bool HasTrailingDot,ParsedType & ReceiverType)2054*67e74705SXin Li Sema::ObjCMessageKind Sema::getObjCMessageKind(Scope *S,
2055*67e74705SXin Li                                                IdentifierInfo *Name,
2056*67e74705SXin Li                                                SourceLocation NameLoc,
2057*67e74705SXin Li                                                bool IsSuper,
2058*67e74705SXin Li                                                bool HasTrailingDot,
2059*67e74705SXin Li                                                ParsedType &ReceiverType) {
2060*67e74705SXin Li   ReceiverType = nullptr;
2061*67e74705SXin Li 
2062*67e74705SXin Li   // If the identifier is "super" and there is no trailing dot, we're
2063*67e74705SXin Li   // messaging super. If the identifier is "super" and there is a
2064*67e74705SXin Li   // trailing dot, it's an instance message.
2065*67e74705SXin Li   if (IsSuper && S->isInObjcMethodScope())
2066*67e74705SXin Li     return HasTrailingDot? ObjCInstanceMessage : ObjCSuperMessage;
2067*67e74705SXin Li 
2068*67e74705SXin Li   LookupResult Result(*this, Name, NameLoc, LookupOrdinaryName);
2069*67e74705SXin Li   LookupName(Result, S);
2070*67e74705SXin Li 
2071*67e74705SXin Li   switch (Result.getResultKind()) {
2072*67e74705SXin Li   case LookupResult::NotFound:
2073*67e74705SXin Li     // Normal name lookup didn't find anything. If we're in an
2074*67e74705SXin Li     // Objective-C method, look for ivars. If we find one, we're done!
2075*67e74705SXin Li     // FIXME: This is a hack. Ivar lookup should be part of normal
2076*67e74705SXin Li     // lookup.
2077*67e74705SXin Li     if (ObjCMethodDecl *Method = getCurMethodDecl()) {
2078*67e74705SXin Li       if (!Method->getClassInterface()) {
2079*67e74705SXin Li         // Fall back: let the parser try to parse it as an instance message.
2080*67e74705SXin Li         return ObjCInstanceMessage;
2081*67e74705SXin Li       }
2082*67e74705SXin Li 
2083*67e74705SXin Li       ObjCInterfaceDecl *ClassDeclared;
2084*67e74705SXin Li       if (Method->getClassInterface()->lookupInstanceVariable(Name,
2085*67e74705SXin Li                                                               ClassDeclared))
2086*67e74705SXin Li         return ObjCInstanceMessage;
2087*67e74705SXin Li     }
2088*67e74705SXin Li 
2089*67e74705SXin Li     // Break out; we'll perform typo correction below.
2090*67e74705SXin Li     break;
2091*67e74705SXin Li 
2092*67e74705SXin Li   case LookupResult::NotFoundInCurrentInstantiation:
2093*67e74705SXin Li   case LookupResult::FoundOverloaded:
2094*67e74705SXin Li   case LookupResult::FoundUnresolvedValue:
2095*67e74705SXin Li   case LookupResult::Ambiguous:
2096*67e74705SXin Li     Result.suppressDiagnostics();
2097*67e74705SXin Li     return ObjCInstanceMessage;
2098*67e74705SXin Li 
2099*67e74705SXin Li   case LookupResult::Found: {
2100*67e74705SXin Li     // If the identifier is a class or not, and there is a trailing dot,
2101*67e74705SXin Li     // it's an instance message.
2102*67e74705SXin Li     if (HasTrailingDot)
2103*67e74705SXin Li       return ObjCInstanceMessage;
2104*67e74705SXin Li     // We found something. If it's a type, then we have a class
2105*67e74705SXin Li     // message. Otherwise, it's an instance message.
2106*67e74705SXin Li     NamedDecl *ND = Result.getFoundDecl();
2107*67e74705SXin Li     QualType T;
2108*67e74705SXin Li     if (ObjCInterfaceDecl *Class = dyn_cast<ObjCInterfaceDecl>(ND))
2109*67e74705SXin Li       T = Context.getObjCInterfaceType(Class);
2110*67e74705SXin Li     else if (TypeDecl *Type = dyn_cast<TypeDecl>(ND)) {
2111*67e74705SXin Li       T = Context.getTypeDeclType(Type);
2112*67e74705SXin Li       DiagnoseUseOfDecl(Type, NameLoc);
2113*67e74705SXin Li     }
2114*67e74705SXin Li     else
2115*67e74705SXin Li       return ObjCInstanceMessage;
2116*67e74705SXin Li 
2117*67e74705SXin Li     //  We have a class message, and T is the type we're
2118*67e74705SXin Li     //  messaging. Build source-location information for it.
2119*67e74705SXin Li     TypeSourceInfo *TSInfo = Context.getTrivialTypeSourceInfo(T, NameLoc);
2120*67e74705SXin Li     ReceiverType = CreateParsedType(T, TSInfo);
2121*67e74705SXin Li     return ObjCClassMessage;
2122*67e74705SXin Li   }
2123*67e74705SXin Li   }
2124*67e74705SXin Li 
2125*67e74705SXin Li   if (TypoCorrection Corrected = CorrectTypo(
2126*67e74705SXin Li           Result.getLookupNameInfo(), Result.getLookupKind(), S, nullptr,
2127*67e74705SXin Li           llvm::make_unique<ObjCInterfaceOrSuperCCC>(getCurMethodDecl()),
2128*67e74705SXin Li           CTK_ErrorRecovery, nullptr, false, nullptr, false)) {
2129*67e74705SXin Li     if (Corrected.isKeyword()) {
2130*67e74705SXin Li       // If we've found the keyword "super" (the only keyword that would be
2131*67e74705SXin Li       // returned by CorrectTypo), this is a send to super.
2132*67e74705SXin Li       diagnoseTypo(Corrected,
2133*67e74705SXin Li                    PDiag(diag::err_unknown_receiver_suggest) << Name);
2134*67e74705SXin Li       return ObjCSuperMessage;
2135*67e74705SXin Li     } else if (ObjCInterfaceDecl *Class =
2136*67e74705SXin Li                    Corrected.getCorrectionDeclAs<ObjCInterfaceDecl>()) {
2137*67e74705SXin Li       // If we found a declaration, correct when it refers to an Objective-C
2138*67e74705SXin Li       // class.
2139*67e74705SXin Li       diagnoseTypo(Corrected,
2140*67e74705SXin Li                    PDiag(diag::err_unknown_receiver_suggest) << Name);
2141*67e74705SXin Li       QualType T = Context.getObjCInterfaceType(Class);
2142*67e74705SXin Li       TypeSourceInfo *TSInfo = Context.getTrivialTypeSourceInfo(T, NameLoc);
2143*67e74705SXin Li       ReceiverType = CreateParsedType(T, TSInfo);
2144*67e74705SXin Li       return ObjCClassMessage;
2145*67e74705SXin Li     }
2146*67e74705SXin Li   }
2147*67e74705SXin Li 
2148*67e74705SXin Li   // Fall back: let the parser try to parse it as an instance message.
2149*67e74705SXin Li   return ObjCInstanceMessage;
2150*67e74705SXin Li }
2151*67e74705SXin Li 
ActOnSuperMessage(Scope * S,SourceLocation SuperLoc,Selector Sel,SourceLocation LBracLoc,ArrayRef<SourceLocation> SelectorLocs,SourceLocation RBracLoc,MultiExprArg Args)2152*67e74705SXin Li ExprResult Sema::ActOnSuperMessage(Scope *S,
2153*67e74705SXin Li                                    SourceLocation SuperLoc,
2154*67e74705SXin Li                                    Selector Sel,
2155*67e74705SXin Li                                    SourceLocation LBracLoc,
2156*67e74705SXin Li                                    ArrayRef<SourceLocation> SelectorLocs,
2157*67e74705SXin Li                                    SourceLocation RBracLoc,
2158*67e74705SXin Li                                    MultiExprArg Args) {
2159*67e74705SXin Li   // Determine whether we are inside a method or not.
2160*67e74705SXin Li   ObjCMethodDecl *Method = tryCaptureObjCSelf(SuperLoc);
2161*67e74705SXin Li   if (!Method) {
2162*67e74705SXin Li     Diag(SuperLoc, diag::err_invalid_receiver_to_message_super);
2163*67e74705SXin Li     return ExprError();
2164*67e74705SXin Li   }
2165*67e74705SXin Li 
2166*67e74705SXin Li   ObjCInterfaceDecl *Class = Method->getClassInterface();
2167*67e74705SXin Li   if (!Class) {
2168*67e74705SXin Li     Diag(SuperLoc, diag::error_no_super_class_message)
2169*67e74705SXin Li       << Method->getDeclName();
2170*67e74705SXin Li     return ExprError();
2171*67e74705SXin Li   }
2172*67e74705SXin Li 
2173*67e74705SXin Li   QualType SuperTy(Class->getSuperClassType(), 0);
2174*67e74705SXin Li   if (SuperTy.isNull()) {
2175*67e74705SXin Li     // The current class does not have a superclass.
2176*67e74705SXin Li     Diag(SuperLoc, diag::error_root_class_cannot_use_super)
2177*67e74705SXin Li       << Class->getIdentifier();
2178*67e74705SXin Li     return ExprError();
2179*67e74705SXin Li   }
2180*67e74705SXin Li 
2181*67e74705SXin Li   // We are in a method whose class has a superclass, so 'super'
2182*67e74705SXin Li   // is acting as a keyword.
2183*67e74705SXin Li   if (Method->getSelector() == Sel)
2184*67e74705SXin Li     getCurFunction()->ObjCShouldCallSuper = false;
2185*67e74705SXin Li 
2186*67e74705SXin Li   if (Method->isInstanceMethod()) {
2187*67e74705SXin Li     // Since we are in an instance method, this is an instance
2188*67e74705SXin Li     // message to the superclass instance.
2189*67e74705SXin Li     SuperTy = Context.getObjCObjectPointerType(SuperTy);
2190*67e74705SXin Li     return BuildInstanceMessage(nullptr, SuperTy, SuperLoc,
2191*67e74705SXin Li                                 Sel, /*Method=*/nullptr,
2192*67e74705SXin Li                                 LBracLoc, SelectorLocs, RBracLoc, Args);
2193*67e74705SXin Li   }
2194*67e74705SXin Li 
2195*67e74705SXin Li   // Since we are in a class method, this is a class message to
2196*67e74705SXin Li   // the superclass.
2197*67e74705SXin Li   return BuildClassMessage(/*ReceiverTypeInfo=*/nullptr,
2198*67e74705SXin Li                            SuperTy,
2199*67e74705SXin Li                            SuperLoc, Sel, /*Method=*/nullptr,
2200*67e74705SXin Li                            LBracLoc, SelectorLocs, RBracLoc, Args);
2201*67e74705SXin Li }
2202*67e74705SXin Li 
BuildClassMessageImplicit(QualType ReceiverType,bool isSuperReceiver,SourceLocation Loc,Selector Sel,ObjCMethodDecl * Method,MultiExprArg Args)2203*67e74705SXin Li ExprResult Sema::BuildClassMessageImplicit(QualType ReceiverType,
2204*67e74705SXin Li                                            bool isSuperReceiver,
2205*67e74705SXin Li                                            SourceLocation Loc,
2206*67e74705SXin Li                                            Selector Sel,
2207*67e74705SXin Li                                            ObjCMethodDecl *Method,
2208*67e74705SXin Li                                            MultiExprArg Args) {
2209*67e74705SXin Li   TypeSourceInfo *receiverTypeInfo = nullptr;
2210*67e74705SXin Li   if (!ReceiverType.isNull())
2211*67e74705SXin Li     receiverTypeInfo = Context.getTrivialTypeSourceInfo(ReceiverType);
2212*67e74705SXin Li 
2213*67e74705SXin Li   return BuildClassMessage(receiverTypeInfo, ReceiverType,
2214*67e74705SXin Li                           /*SuperLoc=*/isSuperReceiver ? Loc : SourceLocation(),
2215*67e74705SXin Li                            Sel, Method, Loc, Loc, Loc, Args,
2216*67e74705SXin Li                            /*isImplicit=*/true);
2217*67e74705SXin Li }
2218*67e74705SXin Li 
applyCocoaAPICheck(Sema & S,const ObjCMessageExpr * Msg,unsigned DiagID,bool (* refactor)(const ObjCMessageExpr *,const NSAPI &,edit::Commit &))2219*67e74705SXin Li static void applyCocoaAPICheck(Sema &S, const ObjCMessageExpr *Msg,
2220*67e74705SXin Li                                unsigned DiagID,
2221*67e74705SXin Li                                bool (*refactor)(const ObjCMessageExpr *,
2222*67e74705SXin Li                                               const NSAPI &, edit::Commit &)) {
2223*67e74705SXin Li   SourceLocation MsgLoc = Msg->getExprLoc();
2224*67e74705SXin Li   if (S.Diags.isIgnored(DiagID, MsgLoc))
2225*67e74705SXin Li     return;
2226*67e74705SXin Li 
2227*67e74705SXin Li   SourceManager &SM = S.SourceMgr;
2228*67e74705SXin Li   edit::Commit ECommit(SM, S.LangOpts);
2229*67e74705SXin Li   if (refactor(Msg,*S.NSAPIObj, ECommit)) {
2230*67e74705SXin Li     DiagnosticBuilder Builder = S.Diag(MsgLoc, DiagID)
2231*67e74705SXin Li                         << Msg->getSelector() << Msg->getSourceRange();
2232*67e74705SXin Li     // FIXME: Don't emit diagnostic at all if fixits are non-commitable.
2233*67e74705SXin Li     if (!ECommit.isCommitable())
2234*67e74705SXin Li       return;
2235*67e74705SXin Li     for (edit::Commit::edit_iterator
2236*67e74705SXin Li            I = ECommit.edit_begin(), E = ECommit.edit_end(); I != E; ++I) {
2237*67e74705SXin Li       const edit::Commit::Edit &Edit = *I;
2238*67e74705SXin Li       switch (Edit.Kind) {
2239*67e74705SXin Li       case edit::Commit::Act_Insert:
2240*67e74705SXin Li         Builder.AddFixItHint(FixItHint::CreateInsertion(Edit.OrigLoc,
2241*67e74705SXin Li                                                         Edit.Text,
2242*67e74705SXin Li                                                         Edit.BeforePrev));
2243*67e74705SXin Li         break;
2244*67e74705SXin Li       case edit::Commit::Act_InsertFromRange:
2245*67e74705SXin Li         Builder.AddFixItHint(
2246*67e74705SXin Li             FixItHint::CreateInsertionFromRange(Edit.OrigLoc,
2247*67e74705SXin Li                                                 Edit.getInsertFromRange(SM),
2248*67e74705SXin Li                                                 Edit.BeforePrev));
2249*67e74705SXin Li         break;
2250*67e74705SXin Li       case edit::Commit::Act_Remove:
2251*67e74705SXin Li         Builder.AddFixItHint(FixItHint::CreateRemoval(Edit.getFileRange(SM)));
2252*67e74705SXin Li         break;
2253*67e74705SXin Li       }
2254*67e74705SXin Li     }
2255*67e74705SXin Li   }
2256*67e74705SXin Li }
2257*67e74705SXin Li 
checkCocoaAPI(Sema & S,const ObjCMessageExpr * Msg)2258*67e74705SXin Li static void checkCocoaAPI(Sema &S, const ObjCMessageExpr *Msg) {
2259*67e74705SXin Li   applyCocoaAPICheck(S, Msg, diag::warn_objc_redundant_literal_use,
2260*67e74705SXin Li                      edit::rewriteObjCRedundantCallWithLiteral);
2261*67e74705SXin Li }
2262*67e74705SXin Li 
2263*67e74705SXin Li /// \brief Diagnose use of %s directive in an NSString which is being passed
2264*67e74705SXin Li /// as formatting string to formatting method.
2265*67e74705SXin Li static void
DiagnoseCStringFormatDirectiveInObjCAPI(Sema & S,ObjCMethodDecl * Method,Selector Sel,Expr ** Args,unsigned NumArgs)2266*67e74705SXin Li DiagnoseCStringFormatDirectiveInObjCAPI(Sema &S,
2267*67e74705SXin Li                                         ObjCMethodDecl *Method,
2268*67e74705SXin Li                                         Selector Sel,
2269*67e74705SXin Li                                         Expr **Args, unsigned NumArgs) {
2270*67e74705SXin Li   unsigned Idx = 0;
2271*67e74705SXin Li   bool Format = false;
2272*67e74705SXin Li   ObjCStringFormatFamily SFFamily = Sel.getStringFormatFamily();
2273*67e74705SXin Li   if (SFFamily == ObjCStringFormatFamily::SFF_NSString) {
2274*67e74705SXin Li     Idx = 0;
2275*67e74705SXin Li     Format = true;
2276*67e74705SXin Li   }
2277*67e74705SXin Li   else if (Method) {
2278*67e74705SXin Li     for (const auto *I : Method->specific_attrs<FormatAttr>()) {
2279*67e74705SXin Li       if (S.GetFormatNSStringIdx(I, Idx)) {
2280*67e74705SXin Li         Format = true;
2281*67e74705SXin Li         break;
2282*67e74705SXin Li       }
2283*67e74705SXin Li     }
2284*67e74705SXin Li   }
2285*67e74705SXin Li   if (!Format || NumArgs <= Idx)
2286*67e74705SXin Li     return;
2287*67e74705SXin Li 
2288*67e74705SXin Li   Expr *FormatExpr = Args[Idx];
2289*67e74705SXin Li   if (ObjCStringLiteral *OSL =
2290*67e74705SXin Li       dyn_cast<ObjCStringLiteral>(FormatExpr->IgnoreParenImpCasts())) {
2291*67e74705SXin Li     StringLiteral *FormatString = OSL->getString();
2292*67e74705SXin Li     if (S.FormatStringHasSArg(FormatString)) {
2293*67e74705SXin Li       S.Diag(FormatExpr->getExprLoc(), diag::warn_objc_cdirective_format_string)
2294*67e74705SXin Li         << "%s" << 0 << 0;
2295*67e74705SXin Li       if (Method)
2296*67e74705SXin Li         S.Diag(Method->getLocation(), diag::note_method_declared_at)
2297*67e74705SXin Li           << Method->getDeclName();
2298*67e74705SXin Li     }
2299*67e74705SXin Li   }
2300*67e74705SXin Li }
2301*67e74705SXin Li 
2302*67e74705SXin Li /// \brief Build an Objective-C class message expression.
2303*67e74705SXin Li ///
2304*67e74705SXin Li /// This routine takes care of both normal class messages and
2305*67e74705SXin Li /// class messages to the superclass.
2306*67e74705SXin Li ///
2307*67e74705SXin Li /// \param ReceiverTypeInfo Type source information that describes the
2308*67e74705SXin Li /// receiver of this message. This may be NULL, in which case we are
2309*67e74705SXin Li /// sending to the superclass and \p SuperLoc must be a valid source
2310*67e74705SXin Li /// location.
2311*67e74705SXin Li 
2312*67e74705SXin Li /// \param ReceiverType The type of the object receiving the
2313*67e74705SXin Li /// message. When \p ReceiverTypeInfo is non-NULL, this is the same
2314*67e74705SXin Li /// type as that refers to. For a superclass send, this is the type of
2315*67e74705SXin Li /// the superclass.
2316*67e74705SXin Li ///
2317*67e74705SXin Li /// \param SuperLoc The location of the "super" keyword in a
2318*67e74705SXin Li /// superclass message.
2319*67e74705SXin Li ///
2320*67e74705SXin Li /// \param Sel The selector to which the message is being sent.
2321*67e74705SXin Li ///
2322*67e74705SXin Li /// \param Method The method that this class message is invoking, if
2323*67e74705SXin Li /// already known.
2324*67e74705SXin Li ///
2325*67e74705SXin Li /// \param LBracLoc The location of the opening square bracket ']'.
2326*67e74705SXin Li ///
2327*67e74705SXin Li /// \param RBracLoc The location of the closing square bracket ']'.
2328*67e74705SXin Li ///
2329*67e74705SXin Li /// \param ArgsIn The message arguments.
BuildClassMessage(TypeSourceInfo * ReceiverTypeInfo,QualType ReceiverType,SourceLocation SuperLoc,Selector Sel,ObjCMethodDecl * Method,SourceLocation LBracLoc,ArrayRef<SourceLocation> SelectorLocs,SourceLocation RBracLoc,MultiExprArg ArgsIn,bool isImplicit)2330*67e74705SXin Li ExprResult Sema::BuildClassMessage(TypeSourceInfo *ReceiverTypeInfo,
2331*67e74705SXin Li                                    QualType ReceiverType,
2332*67e74705SXin Li                                    SourceLocation SuperLoc,
2333*67e74705SXin Li                                    Selector Sel,
2334*67e74705SXin Li                                    ObjCMethodDecl *Method,
2335*67e74705SXin Li                                    SourceLocation LBracLoc,
2336*67e74705SXin Li                                    ArrayRef<SourceLocation> SelectorLocs,
2337*67e74705SXin Li                                    SourceLocation RBracLoc,
2338*67e74705SXin Li                                    MultiExprArg ArgsIn,
2339*67e74705SXin Li                                    bool isImplicit) {
2340*67e74705SXin Li   SourceLocation Loc = SuperLoc.isValid()? SuperLoc
2341*67e74705SXin Li     : ReceiverTypeInfo->getTypeLoc().getSourceRange().getBegin();
2342*67e74705SXin Li   if (LBracLoc.isInvalid()) {
2343*67e74705SXin Li     Diag(Loc, diag::err_missing_open_square_message_send)
2344*67e74705SXin Li       << FixItHint::CreateInsertion(Loc, "[");
2345*67e74705SXin Li     LBracLoc = Loc;
2346*67e74705SXin Li   }
2347*67e74705SXin Li   SourceLocation SelLoc;
2348*67e74705SXin Li   if (!SelectorLocs.empty() && SelectorLocs.front().isValid())
2349*67e74705SXin Li     SelLoc = SelectorLocs.front();
2350*67e74705SXin Li   else
2351*67e74705SXin Li     SelLoc = Loc;
2352*67e74705SXin Li 
2353*67e74705SXin Li   if (ReceiverType->isDependentType()) {
2354*67e74705SXin Li     // If the receiver type is dependent, we can't type-check anything
2355*67e74705SXin Li     // at this point. Build a dependent expression.
2356*67e74705SXin Li     unsigned NumArgs = ArgsIn.size();
2357*67e74705SXin Li     Expr **Args = ArgsIn.data();
2358*67e74705SXin Li     assert(SuperLoc.isInvalid() && "Message to super with dependent type");
2359*67e74705SXin Li     return ObjCMessageExpr::Create(
2360*67e74705SXin Li         Context, ReceiverType, VK_RValue, LBracLoc, ReceiverTypeInfo, Sel,
2361*67e74705SXin Li         SelectorLocs, /*Method=*/nullptr, makeArrayRef(Args, NumArgs), RBracLoc,
2362*67e74705SXin Li         isImplicit);
2363*67e74705SXin Li   }
2364*67e74705SXin Li 
2365*67e74705SXin Li   // Find the class to which we are sending this message.
2366*67e74705SXin Li   ObjCInterfaceDecl *Class = nullptr;
2367*67e74705SXin Li   const ObjCObjectType *ClassType = ReceiverType->getAs<ObjCObjectType>();
2368*67e74705SXin Li   if (!ClassType || !(Class = ClassType->getInterface())) {
2369*67e74705SXin Li     Diag(Loc, diag::err_invalid_receiver_class_message)
2370*67e74705SXin Li       << ReceiverType;
2371*67e74705SXin Li     return ExprError();
2372*67e74705SXin Li   }
2373*67e74705SXin Li   assert(Class && "We don't know which class we're messaging?");
2374*67e74705SXin Li   // objc++ diagnoses during typename annotation.
2375*67e74705SXin Li   if (!getLangOpts().CPlusPlus)
2376*67e74705SXin Li     (void)DiagnoseUseOfDecl(Class, SelLoc);
2377*67e74705SXin Li   // Find the method we are messaging.
2378*67e74705SXin Li   if (!Method) {
2379*67e74705SXin Li     SourceRange TypeRange
2380*67e74705SXin Li       = SuperLoc.isValid()? SourceRange(SuperLoc)
2381*67e74705SXin Li                           : ReceiverTypeInfo->getTypeLoc().getSourceRange();
2382*67e74705SXin Li     if (RequireCompleteType(Loc, Context.getObjCInterfaceType(Class),
2383*67e74705SXin Li                             (getLangOpts().ObjCAutoRefCount
2384*67e74705SXin Li                                ? diag::err_arc_receiver_forward_class
2385*67e74705SXin Li                                : diag::warn_receiver_forward_class),
2386*67e74705SXin Li                             TypeRange)) {
2387*67e74705SXin Li       // A forward class used in messaging is treated as a 'Class'
2388*67e74705SXin Li       Method = LookupFactoryMethodInGlobalPool(Sel,
2389*67e74705SXin Li                                                SourceRange(LBracLoc, RBracLoc));
2390*67e74705SXin Li       if (Method && !getLangOpts().ObjCAutoRefCount)
2391*67e74705SXin Li         Diag(Method->getLocation(), diag::note_method_sent_forward_class)
2392*67e74705SXin Li           << Method->getDeclName();
2393*67e74705SXin Li     }
2394*67e74705SXin Li     if (!Method)
2395*67e74705SXin Li       Method = Class->lookupClassMethod(Sel);
2396*67e74705SXin Li 
2397*67e74705SXin Li     // If we have an implementation in scope, check "private" methods.
2398*67e74705SXin Li     if (!Method)
2399*67e74705SXin Li       Method = Class->lookupPrivateClassMethod(Sel);
2400*67e74705SXin Li 
2401*67e74705SXin Li     if (Method && DiagnoseUseOfDecl(Method, SelLoc))
2402*67e74705SXin Li       return ExprError();
2403*67e74705SXin Li   }
2404*67e74705SXin Li 
2405*67e74705SXin Li   // Check the argument types and determine the result type.
2406*67e74705SXin Li   QualType ReturnType;
2407*67e74705SXin Li   ExprValueKind VK = VK_RValue;
2408*67e74705SXin Li 
2409*67e74705SXin Li   unsigned NumArgs = ArgsIn.size();
2410*67e74705SXin Li   Expr **Args = ArgsIn.data();
2411*67e74705SXin Li   if (CheckMessageArgumentTypes(ReceiverType, MultiExprArg(Args, NumArgs),
2412*67e74705SXin Li                                 Sel, SelectorLocs,
2413*67e74705SXin Li                                 Method, true,
2414*67e74705SXin Li                                 SuperLoc.isValid(), LBracLoc, RBracLoc,
2415*67e74705SXin Li                                 SourceRange(),
2416*67e74705SXin Li                                 ReturnType, VK))
2417*67e74705SXin Li     return ExprError();
2418*67e74705SXin Li 
2419*67e74705SXin Li   if (Method && !Method->getReturnType()->isVoidType() &&
2420*67e74705SXin Li       RequireCompleteType(LBracLoc, Method->getReturnType(),
2421*67e74705SXin Li                           diag::err_illegal_message_expr_incomplete_type))
2422*67e74705SXin Li     return ExprError();
2423*67e74705SXin Li 
2424*67e74705SXin Li   // Warn about explicit call of +initialize on its own class. But not on 'super'.
2425*67e74705SXin Li   if (Method && Method->getMethodFamily() == OMF_initialize) {
2426*67e74705SXin Li     if (!SuperLoc.isValid()) {
2427*67e74705SXin Li       const ObjCInterfaceDecl *ID =
2428*67e74705SXin Li         dyn_cast<ObjCInterfaceDecl>(Method->getDeclContext());
2429*67e74705SXin Li       if (ID == Class) {
2430*67e74705SXin Li         Diag(Loc, diag::warn_direct_initialize_call);
2431*67e74705SXin Li         Diag(Method->getLocation(), diag::note_method_declared_at)
2432*67e74705SXin Li           << Method->getDeclName();
2433*67e74705SXin Li       }
2434*67e74705SXin Li     }
2435*67e74705SXin Li     else if (ObjCMethodDecl *CurMeth = getCurMethodDecl()) {
2436*67e74705SXin Li       // [super initialize] is allowed only within an +initialize implementation
2437*67e74705SXin Li       if (CurMeth->getMethodFamily() != OMF_initialize) {
2438*67e74705SXin Li         Diag(Loc, diag::warn_direct_super_initialize_call);
2439*67e74705SXin Li         Diag(Method->getLocation(), diag::note_method_declared_at)
2440*67e74705SXin Li           << Method->getDeclName();
2441*67e74705SXin Li         Diag(CurMeth->getLocation(), diag::note_method_declared_at)
2442*67e74705SXin Li         << CurMeth->getDeclName();
2443*67e74705SXin Li       }
2444*67e74705SXin Li     }
2445*67e74705SXin Li   }
2446*67e74705SXin Li 
2447*67e74705SXin Li   DiagnoseCStringFormatDirectiveInObjCAPI(*this, Method, Sel, Args, NumArgs);
2448*67e74705SXin Li 
2449*67e74705SXin Li   // Construct the appropriate ObjCMessageExpr.
2450*67e74705SXin Li   ObjCMessageExpr *Result;
2451*67e74705SXin Li   if (SuperLoc.isValid())
2452*67e74705SXin Li     Result = ObjCMessageExpr::Create(Context, ReturnType, VK, LBracLoc,
2453*67e74705SXin Li                                      SuperLoc, /*IsInstanceSuper=*/false,
2454*67e74705SXin Li                                      ReceiverType, Sel, SelectorLocs,
2455*67e74705SXin Li                                      Method, makeArrayRef(Args, NumArgs),
2456*67e74705SXin Li                                      RBracLoc, isImplicit);
2457*67e74705SXin Li   else {
2458*67e74705SXin Li     Result = ObjCMessageExpr::Create(Context, ReturnType, VK, LBracLoc,
2459*67e74705SXin Li                                      ReceiverTypeInfo, Sel, SelectorLocs,
2460*67e74705SXin Li                                      Method, makeArrayRef(Args, NumArgs),
2461*67e74705SXin Li                                      RBracLoc, isImplicit);
2462*67e74705SXin Li     if (!isImplicit)
2463*67e74705SXin Li       checkCocoaAPI(*this, Result);
2464*67e74705SXin Li   }
2465*67e74705SXin Li   return MaybeBindToTemporary(Result);
2466*67e74705SXin Li }
2467*67e74705SXin Li 
2468*67e74705SXin Li // ActOnClassMessage - used for both unary and keyword messages.
2469*67e74705SXin Li // ArgExprs is optional - if it is present, the number of expressions
2470*67e74705SXin Li // is obtained from Sel.getNumArgs().
ActOnClassMessage(Scope * S,ParsedType Receiver,Selector Sel,SourceLocation LBracLoc,ArrayRef<SourceLocation> SelectorLocs,SourceLocation RBracLoc,MultiExprArg Args)2471*67e74705SXin Li ExprResult Sema::ActOnClassMessage(Scope *S,
2472*67e74705SXin Li                                    ParsedType Receiver,
2473*67e74705SXin Li                                    Selector Sel,
2474*67e74705SXin Li                                    SourceLocation LBracLoc,
2475*67e74705SXin Li                                    ArrayRef<SourceLocation> SelectorLocs,
2476*67e74705SXin Li                                    SourceLocation RBracLoc,
2477*67e74705SXin Li                                    MultiExprArg Args) {
2478*67e74705SXin Li   TypeSourceInfo *ReceiverTypeInfo;
2479*67e74705SXin Li   QualType ReceiverType = GetTypeFromParser(Receiver, &ReceiverTypeInfo);
2480*67e74705SXin Li   if (ReceiverType.isNull())
2481*67e74705SXin Li     return ExprError();
2482*67e74705SXin Li 
2483*67e74705SXin Li   if (!ReceiverTypeInfo)
2484*67e74705SXin Li     ReceiverTypeInfo = Context.getTrivialTypeSourceInfo(ReceiverType, LBracLoc);
2485*67e74705SXin Li 
2486*67e74705SXin Li   return BuildClassMessage(ReceiverTypeInfo, ReceiverType,
2487*67e74705SXin Li                            /*SuperLoc=*/SourceLocation(), Sel,
2488*67e74705SXin Li                            /*Method=*/nullptr, LBracLoc, SelectorLocs, RBracLoc,
2489*67e74705SXin Li                            Args);
2490*67e74705SXin Li }
2491*67e74705SXin Li 
BuildInstanceMessageImplicit(Expr * Receiver,QualType ReceiverType,SourceLocation Loc,Selector Sel,ObjCMethodDecl * Method,MultiExprArg Args)2492*67e74705SXin Li ExprResult Sema::BuildInstanceMessageImplicit(Expr *Receiver,
2493*67e74705SXin Li                                               QualType ReceiverType,
2494*67e74705SXin Li                                               SourceLocation Loc,
2495*67e74705SXin Li                                               Selector Sel,
2496*67e74705SXin Li                                               ObjCMethodDecl *Method,
2497*67e74705SXin Li                                               MultiExprArg Args) {
2498*67e74705SXin Li   return BuildInstanceMessage(Receiver, ReceiverType,
2499*67e74705SXin Li                               /*SuperLoc=*/!Receiver ? Loc : SourceLocation(),
2500*67e74705SXin Li                               Sel, Method, Loc, Loc, Loc, Args,
2501*67e74705SXin Li                               /*isImplicit=*/true);
2502*67e74705SXin Li }
2503*67e74705SXin Li 
2504*67e74705SXin Li /// \brief Build an Objective-C instance message expression.
2505*67e74705SXin Li ///
2506*67e74705SXin Li /// This routine takes care of both normal instance messages and
2507*67e74705SXin Li /// instance messages to the superclass instance.
2508*67e74705SXin Li ///
2509*67e74705SXin Li /// \param Receiver The expression that computes the object that will
2510*67e74705SXin Li /// receive this message. This may be empty, in which case we are
2511*67e74705SXin Li /// sending to the superclass instance and \p SuperLoc must be a valid
2512*67e74705SXin Li /// source location.
2513*67e74705SXin Li ///
2514*67e74705SXin Li /// \param ReceiverType The (static) type of the object receiving the
2515*67e74705SXin Li /// message. When a \p Receiver expression is provided, this is the
2516*67e74705SXin Li /// same type as that expression. For a superclass instance send, this
2517*67e74705SXin Li /// is a pointer to the type of the superclass.
2518*67e74705SXin Li ///
2519*67e74705SXin Li /// \param SuperLoc The location of the "super" keyword in a
2520*67e74705SXin Li /// superclass instance message.
2521*67e74705SXin Li ///
2522*67e74705SXin Li /// \param Sel The selector to which the message is being sent.
2523*67e74705SXin Li ///
2524*67e74705SXin Li /// \param Method The method that this instance message is invoking, if
2525*67e74705SXin Li /// already known.
2526*67e74705SXin Li ///
2527*67e74705SXin Li /// \param LBracLoc The location of the opening square bracket ']'.
2528*67e74705SXin Li ///
2529*67e74705SXin Li /// \param RBracLoc The location of the closing square bracket ']'.
2530*67e74705SXin Li ///
2531*67e74705SXin Li /// \param ArgsIn The message arguments.
BuildInstanceMessage(Expr * Receiver,QualType ReceiverType,SourceLocation SuperLoc,Selector Sel,ObjCMethodDecl * Method,SourceLocation LBracLoc,ArrayRef<SourceLocation> SelectorLocs,SourceLocation RBracLoc,MultiExprArg ArgsIn,bool isImplicit)2532*67e74705SXin Li ExprResult Sema::BuildInstanceMessage(Expr *Receiver,
2533*67e74705SXin Li                                       QualType ReceiverType,
2534*67e74705SXin Li                                       SourceLocation SuperLoc,
2535*67e74705SXin Li                                       Selector Sel,
2536*67e74705SXin Li                                       ObjCMethodDecl *Method,
2537*67e74705SXin Li                                       SourceLocation LBracLoc,
2538*67e74705SXin Li                                       ArrayRef<SourceLocation> SelectorLocs,
2539*67e74705SXin Li                                       SourceLocation RBracLoc,
2540*67e74705SXin Li                                       MultiExprArg ArgsIn,
2541*67e74705SXin Li                                       bool isImplicit) {
2542*67e74705SXin Li   // The location of the receiver.
2543*67e74705SXin Li   SourceLocation Loc = SuperLoc.isValid()? SuperLoc : Receiver->getLocStart();
2544*67e74705SXin Li   SourceRange RecRange =
2545*67e74705SXin Li       SuperLoc.isValid()? SuperLoc : Receiver->getSourceRange();
2546*67e74705SXin Li   SourceLocation SelLoc;
2547*67e74705SXin Li   if (!SelectorLocs.empty() && SelectorLocs.front().isValid())
2548*67e74705SXin Li     SelLoc = SelectorLocs.front();
2549*67e74705SXin Li   else
2550*67e74705SXin Li     SelLoc = Loc;
2551*67e74705SXin Li 
2552*67e74705SXin Li   if (LBracLoc.isInvalid()) {
2553*67e74705SXin Li     Diag(Loc, diag::err_missing_open_square_message_send)
2554*67e74705SXin Li       << FixItHint::CreateInsertion(Loc, "[");
2555*67e74705SXin Li     LBracLoc = Loc;
2556*67e74705SXin Li   }
2557*67e74705SXin Li 
2558*67e74705SXin Li   // If we have a receiver expression, perform appropriate promotions
2559*67e74705SXin Li   // and determine receiver type.
2560*67e74705SXin Li   if (Receiver) {
2561*67e74705SXin Li     if (Receiver->hasPlaceholderType()) {
2562*67e74705SXin Li       ExprResult Result;
2563*67e74705SXin Li       if (Receiver->getType() == Context.UnknownAnyTy)
2564*67e74705SXin Li         Result = forceUnknownAnyToType(Receiver, Context.getObjCIdType());
2565*67e74705SXin Li       else
2566*67e74705SXin Li         Result = CheckPlaceholderExpr(Receiver);
2567*67e74705SXin Li       if (Result.isInvalid()) return ExprError();
2568*67e74705SXin Li       Receiver = Result.get();
2569*67e74705SXin Li     }
2570*67e74705SXin Li 
2571*67e74705SXin Li     if (Receiver->isTypeDependent()) {
2572*67e74705SXin Li       // If the receiver is type-dependent, we can't type-check anything
2573*67e74705SXin Li       // at this point. Build a dependent expression.
2574*67e74705SXin Li       unsigned NumArgs = ArgsIn.size();
2575*67e74705SXin Li       Expr **Args = ArgsIn.data();
2576*67e74705SXin Li       assert(SuperLoc.isInvalid() && "Message to super with dependent type");
2577*67e74705SXin Li       return ObjCMessageExpr::Create(
2578*67e74705SXin Li           Context, Context.DependentTy, VK_RValue, LBracLoc, Receiver, Sel,
2579*67e74705SXin Li           SelectorLocs, /*Method=*/nullptr, makeArrayRef(Args, NumArgs),
2580*67e74705SXin Li           RBracLoc, isImplicit);
2581*67e74705SXin Li     }
2582*67e74705SXin Li 
2583*67e74705SXin Li     // If necessary, apply function/array conversion to the receiver.
2584*67e74705SXin Li     // C99 6.7.5.3p[7,8].
2585*67e74705SXin Li     ExprResult Result = DefaultFunctionArrayLvalueConversion(Receiver);
2586*67e74705SXin Li     if (Result.isInvalid())
2587*67e74705SXin Li       return ExprError();
2588*67e74705SXin Li     Receiver = Result.get();
2589*67e74705SXin Li     ReceiverType = Receiver->getType();
2590*67e74705SXin Li 
2591*67e74705SXin Li     // If the receiver is an ObjC pointer, a block pointer, or an
2592*67e74705SXin Li     // __attribute__((NSObject)) pointer, we don't need to do any
2593*67e74705SXin Li     // special conversion in order to look up a receiver.
2594*67e74705SXin Li     if (ReceiverType->isObjCRetainableType()) {
2595*67e74705SXin Li       // do nothing
2596*67e74705SXin Li     } else if (!getLangOpts().ObjCAutoRefCount &&
2597*67e74705SXin Li                !Context.getObjCIdType().isNull() &&
2598*67e74705SXin Li                (ReceiverType->isPointerType() ||
2599*67e74705SXin Li                 ReceiverType->isIntegerType())) {
2600*67e74705SXin Li       // Implicitly convert integers and pointers to 'id' but emit a warning.
2601*67e74705SXin Li       // But not in ARC.
2602*67e74705SXin Li       Diag(Loc, diag::warn_bad_receiver_type)
2603*67e74705SXin Li         << ReceiverType
2604*67e74705SXin Li         << Receiver->getSourceRange();
2605*67e74705SXin Li       if (ReceiverType->isPointerType()) {
2606*67e74705SXin Li         Receiver = ImpCastExprToType(Receiver, Context.getObjCIdType(),
2607*67e74705SXin Li                                      CK_CPointerToObjCPointerCast).get();
2608*67e74705SXin Li       } else {
2609*67e74705SXin Li         // TODO: specialized warning on null receivers?
2610*67e74705SXin Li         bool IsNull = Receiver->isNullPointerConstant(Context,
2611*67e74705SXin Li                                               Expr::NPC_ValueDependentIsNull);
2612*67e74705SXin Li         CastKind Kind = IsNull ? CK_NullToPointer : CK_IntegralToPointer;
2613*67e74705SXin Li         Receiver = ImpCastExprToType(Receiver, Context.getObjCIdType(),
2614*67e74705SXin Li                                      Kind).get();
2615*67e74705SXin Li       }
2616*67e74705SXin Li       ReceiverType = Receiver->getType();
2617*67e74705SXin Li     } else if (getLangOpts().CPlusPlus) {
2618*67e74705SXin Li       // The receiver must be a complete type.
2619*67e74705SXin Li       if (RequireCompleteType(Loc, Receiver->getType(),
2620*67e74705SXin Li                               diag::err_incomplete_receiver_type))
2621*67e74705SXin Li         return ExprError();
2622*67e74705SXin Li 
2623*67e74705SXin Li       ExprResult result = PerformContextuallyConvertToObjCPointer(Receiver);
2624*67e74705SXin Li       if (result.isUsable()) {
2625*67e74705SXin Li         Receiver = result.get();
2626*67e74705SXin Li         ReceiverType = Receiver->getType();
2627*67e74705SXin Li       }
2628*67e74705SXin Li     }
2629*67e74705SXin Li   }
2630*67e74705SXin Li 
2631*67e74705SXin Li   // There's a somewhat weird interaction here where we assume that we
2632*67e74705SXin Li   // won't actually have a method unless we also don't need to do some
2633*67e74705SXin Li   // of the more detailed type-checking on the receiver.
2634*67e74705SXin Li 
2635*67e74705SXin Li   if (!Method) {
2636*67e74705SXin Li     // Handle messages to id and __kindof types (where we use the
2637*67e74705SXin Li     // global method pool).
2638*67e74705SXin Li     const ObjCObjectType *typeBound = nullptr;
2639*67e74705SXin Li     bool receiverIsIdLike = ReceiverType->isObjCIdOrObjectKindOfType(Context,
2640*67e74705SXin Li                                                                      typeBound);
2641*67e74705SXin Li     if (receiverIsIdLike || ReceiverType->isBlockPointerType() ||
2642*67e74705SXin Li         (Receiver && Context.isObjCNSObjectType(Receiver->getType()))) {
2643*67e74705SXin Li       SmallVector<ObjCMethodDecl*, 4> Methods;
2644*67e74705SXin Li       // If we have a type bound, further filter the methods.
2645*67e74705SXin Li       CollectMultipleMethodsInGlobalPool(Sel, Methods, true/*InstanceFirst*/,
2646*67e74705SXin Li                                          true/*CheckTheOther*/, typeBound);
2647*67e74705SXin Li       if (!Methods.empty()) {
2648*67e74705SXin Li         // We chose the first method as the initial condidate, then try to
2649*67e74705SXin Li         // select a better one.
2650*67e74705SXin Li         Method = Methods[0];
2651*67e74705SXin Li 
2652*67e74705SXin Li         if (ObjCMethodDecl *BestMethod =
2653*67e74705SXin Li             SelectBestMethod(Sel, ArgsIn, Method->isInstanceMethod(), Methods))
2654*67e74705SXin Li           Method = BestMethod;
2655*67e74705SXin Li 
2656*67e74705SXin Li         if (!AreMultipleMethodsInGlobalPool(Sel, Method,
2657*67e74705SXin Li                                             SourceRange(LBracLoc, RBracLoc),
2658*67e74705SXin Li                                             receiverIsIdLike, Methods))
2659*67e74705SXin Li            DiagnoseUseOfDecl(Method, SelLoc);
2660*67e74705SXin Li       }
2661*67e74705SXin Li     } else if (ReceiverType->isObjCClassOrClassKindOfType() ||
2662*67e74705SXin Li                ReceiverType->isObjCQualifiedClassType()) {
2663*67e74705SXin Li       // Handle messages to Class.
2664*67e74705SXin Li       // We allow sending a message to a qualified Class ("Class<foo>"), which
2665*67e74705SXin Li       // is ok as long as one of the protocols implements the selector (if not,
2666*67e74705SXin Li       // warn).
2667*67e74705SXin Li       if (!ReceiverType->isObjCClassOrClassKindOfType()) {
2668*67e74705SXin Li         const ObjCObjectPointerType *QClassTy
2669*67e74705SXin Li           = ReceiverType->getAsObjCQualifiedClassType();
2670*67e74705SXin Li         // Search protocols for class methods.
2671*67e74705SXin Li         Method = LookupMethodInQualifiedType(Sel, QClassTy, false);
2672*67e74705SXin Li         if (!Method) {
2673*67e74705SXin Li           Method = LookupMethodInQualifiedType(Sel, QClassTy, true);
2674*67e74705SXin Li           // warn if instance method found for a Class message.
2675*67e74705SXin Li           if (Method) {
2676*67e74705SXin Li             Diag(SelLoc, diag::warn_instance_method_on_class_found)
2677*67e74705SXin Li               << Method->getSelector() << Sel;
2678*67e74705SXin Li             Diag(Method->getLocation(), diag::note_method_declared_at)
2679*67e74705SXin Li               << Method->getDeclName();
2680*67e74705SXin Li           }
2681*67e74705SXin Li         }
2682*67e74705SXin Li       } else {
2683*67e74705SXin Li         if (ObjCMethodDecl *CurMeth = getCurMethodDecl()) {
2684*67e74705SXin Li           if (ObjCInterfaceDecl *ClassDecl = CurMeth->getClassInterface()) {
2685*67e74705SXin Li             // First check the public methods in the class interface.
2686*67e74705SXin Li             Method = ClassDecl->lookupClassMethod(Sel);
2687*67e74705SXin Li 
2688*67e74705SXin Li             if (!Method)
2689*67e74705SXin Li               Method = ClassDecl->lookupPrivateClassMethod(Sel);
2690*67e74705SXin Li           }
2691*67e74705SXin Li           if (Method && DiagnoseUseOfDecl(Method, SelLoc))
2692*67e74705SXin Li             return ExprError();
2693*67e74705SXin Li         }
2694*67e74705SXin Li         if (!Method) {
2695*67e74705SXin Li           // If not messaging 'self', look for any factory method named 'Sel'.
2696*67e74705SXin Li           if (!Receiver || !isSelfExpr(Receiver)) {
2697*67e74705SXin Li             // If no class (factory) method was found, check if an _instance_
2698*67e74705SXin Li             // method of the same name exists in the root class only.
2699*67e74705SXin Li             SmallVector<ObjCMethodDecl*, 4> Methods;
2700*67e74705SXin Li             CollectMultipleMethodsInGlobalPool(Sel, Methods,
2701*67e74705SXin Li                                                false/*InstanceFirst*/,
2702*67e74705SXin Li                                                true/*CheckTheOther*/);
2703*67e74705SXin Li             if (!Methods.empty()) {
2704*67e74705SXin Li               // We chose the first method as the initial condidate, then try
2705*67e74705SXin Li               // to select a better one.
2706*67e74705SXin Li               Method = Methods[0];
2707*67e74705SXin Li 
2708*67e74705SXin Li               // If we find an instance method, emit waring.
2709*67e74705SXin Li               if (Method->isInstanceMethod()) {
2710*67e74705SXin Li                 if (const ObjCInterfaceDecl *ID =
2711*67e74705SXin Li                     dyn_cast<ObjCInterfaceDecl>(Method->getDeclContext())) {
2712*67e74705SXin Li                   if (ID->getSuperClass())
2713*67e74705SXin Li                     Diag(SelLoc, diag::warn_root_inst_method_not_found)
2714*67e74705SXin Li                         << Sel << SourceRange(LBracLoc, RBracLoc);
2715*67e74705SXin Li                 }
2716*67e74705SXin Li               }
2717*67e74705SXin Li 
2718*67e74705SXin Li              if (ObjCMethodDecl *BestMethod =
2719*67e74705SXin Li                  SelectBestMethod(Sel, ArgsIn, Method->isInstanceMethod(),
2720*67e74705SXin Li                                   Methods))
2721*67e74705SXin Li                Method = BestMethod;
2722*67e74705SXin Li             }
2723*67e74705SXin Li           }
2724*67e74705SXin Li         }
2725*67e74705SXin Li       }
2726*67e74705SXin Li     } else {
2727*67e74705SXin Li       ObjCInterfaceDecl *ClassDecl = nullptr;
2728*67e74705SXin Li 
2729*67e74705SXin Li       // We allow sending a message to a qualified ID ("id<foo>"), which is ok as
2730*67e74705SXin Li       // long as one of the protocols implements the selector (if not, warn).
2731*67e74705SXin Li       // And as long as message is not deprecated/unavailable (warn if it is).
2732*67e74705SXin Li       if (const ObjCObjectPointerType *QIdTy
2733*67e74705SXin Li                                    = ReceiverType->getAsObjCQualifiedIdType()) {
2734*67e74705SXin Li         // Search protocols for instance methods.
2735*67e74705SXin Li         Method = LookupMethodInQualifiedType(Sel, QIdTy, true);
2736*67e74705SXin Li         if (!Method)
2737*67e74705SXin Li           Method = LookupMethodInQualifiedType(Sel, QIdTy, false);
2738*67e74705SXin Li         if (Method && DiagnoseUseOfDecl(Method, SelLoc))
2739*67e74705SXin Li           return ExprError();
2740*67e74705SXin Li       } else if (const ObjCObjectPointerType *OCIType
2741*67e74705SXin Li                    = ReceiverType->getAsObjCInterfacePointerType()) {
2742*67e74705SXin Li         // We allow sending a message to a pointer to an interface (an object).
2743*67e74705SXin Li         ClassDecl = OCIType->getInterfaceDecl();
2744*67e74705SXin Li 
2745*67e74705SXin Li         // Try to complete the type. Under ARC, this is a hard error from which
2746*67e74705SXin Li         // we don't try to recover.
2747*67e74705SXin Li         // FIXME: In the non-ARC case, this will still be a hard error if the
2748*67e74705SXin Li         // definition is found in a module that's not visible.
2749*67e74705SXin Li         const ObjCInterfaceDecl *forwardClass = nullptr;
2750*67e74705SXin Li         if (RequireCompleteType(Loc, OCIType->getPointeeType(),
2751*67e74705SXin Li               getLangOpts().ObjCAutoRefCount
2752*67e74705SXin Li                 ? diag::err_arc_receiver_forward_instance
2753*67e74705SXin Li                 : diag::warn_receiver_forward_instance,
2754*67e74705SXin Li                                 Receiver? Receiver->getSourceRange()
2755*67e74705SXin Li                                         : SourceRange(SuperLoc))) {
2756*67e74705SXin Li           if (getLangOpts().ObjCAutoRefCount)
2757*67e74705SXin Li             return ExprError();
2758*67e74705SXin Li 
2759*67e74705SXin Li           forwardClass = OCIType->getInterfaceDecl();
2760*67e74705SXin Li           Diag(Receiver ? Receiver->getLocStart()
2761*67e74705SXin Li                         : SuperLoc, diag::note_receiver_is_id);
2762*67e74705SXin Li           Method = nullptr;
2763*67e74705SXin Li         } else {
2764*67e74705SXin Li           Method = ClassDecl->lookupInstanceMethod(Sel);
2765*67e74705SXin Li         }
2766*67e74705SXin Li 
2767*67e74705SXin Li         if (!Method)
2768*67e74705SXin Li           // Search protocol qualifiers.
2769*67e74705SXin Li           Method = LookupMethodInQualifiedType(Sel, OCIType, true);
2770*67e74705SXin Li 
2771*67e74705SXin Li         if (!Method) {
2772*67e74705SXin Li           // If we have implementations in scope, check "private" methods.
2773*67e74705SXin Li           Method = ClassDecl->lookupPrivateMethod(Sel);
2774*67e74705SXin Li 
2775*67e74705SXin Li           if (!Method && getLangOpts().ObjCAutoRefCount) {
2776*67e74705SXin Li             Diag(SelLoc, diag::err_arc_may_not_respond)
2777*67e74705SXin Li               << OCIType->getPointeeType() << Sel << RecRange
2778*67e74705SXin Li               << SourceRange(SelectorLocs.front(), SelectorLocs.back());
2779*67e74705SXin Li             return ExprError();
2780*67e74705SXin Li           }
2781*67e74705SXin Li 
2782*67e74705SXin Li           if (!Method && (!Receiver || !isSelfExpr(Receiver))) {
2783*67e74705SXin Li             // If we still haven't found a method, look in the global pool. This
2784*67e74705SXin Li             // behavior isn't very desirable, however we need it for GCC
2785*67e74705SXin Li             // compatibility. FIXME: should we deviate??
2786*67e74705SXin Li             if (OCIType->qual_empty()) {
2787*67e74705SXin Li               SmallVector<ObjCMethodDecl*, 4> Methods;
2788*67e74705SXin Li               CollectMultipleMethodsInGlobalPool(Sel, Methods,
2789*67e74705SXin Li                                                  true/*InstanceFirst*/,
2790*67e74705SXin Li                                                  false/*CheckTheOther*/);
2791*67e74705SXin Li               if (!Methods.empty()) {
2792*67e74705SXin Li                 // We chose the first method as the initial condidate, then try
2793*67e74705SXin Li                 // to select a better one.
2794*67e74705SXin Li                 Method = Methods[0];
2795*67e74705SXin Li 
2796*67e74705SXin Li                 if (ObjCMethodDecl *BestMethod =
2797*67e74705SXin Li                     SelectBestMethod(Sel, ArgsIn, Method->isInstanceMethod(),
2798*67e74705SXin Li                                      Methods))
2799*67e74705SXin Li                   Method = BestMethod;
2800*67e74705SXin Li 
2801*67e74705SXin Li                 AreMultipleMethodsInGlobalPool(Sel, Method,
2802*67e74705SXin Li                                                SourceRange(LBracLoc, RBracLoc),
2803*67e74705SXin Li                                                true/*receiverIdOrClass*/,
2804*67e74705SXin Li                                                Methods);
2805*67e74705SXin Li               }
2806*67e74705SXin Li               if (Method && !forwardClass)
2807*67e74705SXin Li                 Diag(SelLoc, diag::warn_maynot_respond)
2808*67e74705SXin Li                   << OCIType->getInterfaceDecl()->getIdentifier()
2809*67e74705SXin Li                   << Sel << RecRange;
2810*67e74705SXin Li             }
2811*67e74705SXin Li           }
2812*67e74705SXin Li         }
2813*67e74705SXin Li         if (Method && DiagnoseUseOfDecl(Method, SelLoc, forwardClass))
2814*67e74705SXin Li           return ExprError();
2815*67e74705SXin Li       } else {
2816*67e74705SXin Li         // Reject other random receiver types (e.g. structs).
2817*67e74705SXin Li         Diag(Loc, diag::err_bad_receiver_type)
2818*67e74705SXin Li           << ReceiverType << Receiver->getSourceRange();
2819*67e74705SXin Li         return ExprError();
2820*67e74705SXin Li       }
2821*67e74705SXin Li     }
2822*67e74705SXin Li   }
2823*67e74705SXin Li 
2824*67e74705SXin Li   FunctionScopeInfo *DIFunctionScopeInfo =
2825*67e74705SXin Li     (Method && Method->getMethodFamily() == OMF_init)
2826*67e74705SXin Li       ? getEnclosingFunction() : nullptr;
2827*67e74705SXin Li 
2828*67e74705SXin Li   if (DIFunctionScopeInfo &&
2829*67e74705SXin Li       DIFunctionScopeInfo->ObjCIsDesignatedInit &&
2830*67e74705SXin Li       (SuperLoc.isValid() || isSelfExpr(Receiver))) {
2831*67e74705SXin Li     bool isDesignatedInitChain = false;
2832*67e74705SXin Li     if (SuperLoc.isValid()) {
2833*67e74705SXin Li       if (const ObjCObjectPointerType *
2834*67e74705SXin Li             OCIType = ReceiverType->getAsObjCInterfacePointerType()) {
2835*67e74705SXin Li         if (const ObjCInterfaceDecl *ID = OCIType->getInterfaceDecl()) {
2836*67e74705SXin Li           // Either we know this is a designated initializer or we
2837*67e74705SXin Li           // conservatively assume it because we don't know for sure.
2838*67e74705SXin Li           if (!ID->declaresOrInheritsDesignatedInitializers() ||
2839*67e74705SXin Li               ID->isDesignatedInitializer(Sel)) {
2840*67e74705SXin Li             isDesignatedInitChain = true;
2841*67e74705SXin Li             DIFunctionScopeInfo->ObjCWarnForNoDesignatedInitChain = false;
2842*67e74705SXin Li           }
2843*67e74705SXin Li         }
2844*67e74705SXin Li       }
2845*67e74705SXin Li     }
2846*67e74705SXin Li     if (!isDesignatedInitChain) {
2847*67e74705SXin Li       const ObjCMethodDecl *InitMethod = nullptr;
2848*67e74705SXin Li       bool isDesignated =
2849*67e74705SXin Li         getCurMethodDecl()->isDesignatedInitializerForTheInterface(&InitMethod);
2850*67e74705SXin Li       assert(isDesignated && InitMethod);
2851*67e74705SXin Li       (void)isDesignated;
2852*67e74705SXin Li       Diag(SelLoc, SuperLoc.isValid() ?
2853*67e74705SXin Li              diag::warn_objc_designated_init_non_designated_init_call :
2854*67e74705SXin Li              diag::warn_objc_designated_init_non_super_designated_init_call);
2855*67e74705SXin Li       Diag(InitMethod->getLocation(),
2856*67e74705SXin Li            diag::note_objc_designated_init_marked_here);
2857*67e74705SXin Li     }
2858*67e74705SXin Li   }
2859*67e74705SXin Li 
2860*67e74705SXin Li   if (DIFunctionScopeInfo &&
2861*67e74705SXin Li       DIFunctionScopeInfo->ObjCIsSecondaryInit &&
2862*67e74705SXin Li       (SuperLoc.isValid() || isSelfExpr(Receiver))) {
2863*67e74705SXin Li     if (SuperLoc.isValid()) {
2864*67e74705SXin Li       Diag(SelLoc, diag::warn_objc_secondary_init_super_init_call);
2865*67e74705SXin Li     } else {
2866*67e74705SXin Li       DIFunctionScopeInfo->ObjCWarnForNoInitDelegation = false;
2867*67e74705SXin Li     }
2868*67e74705SXin Li   }
2869*67e74705SXin Li 
2870*67e74705SXin Li   // Check the message arguments.
2871*67e74705SXin Li   unsigned NumArgs = ArgsIn.size();
2872*67e74705SXin Li   Expr **Args = ArgsIn.data();
2873*67e74705SXin Li   QualType ReturnType;
2874*67e74705SXin Li   ExprValueKind VK = VK_RValue;
2875*67e74705SXin Li   bool ClassMessage = (ReceiverType->isObjCClassType() ||
2876*67e74705SXin Li                        ReceiverType->isObjCQualifiedClassType());
2877*67e74705SXin Li   if (CheckMessageArgumentTypes(ReceiverType, MultiExprArg(Args, NumArgs),
2878*67e74705SXin Li                                 Sel, SelectorLocs, Method,
2879*67e74705SXin Li                                 ClassMessage, SuperLoc.isValid(),
2880*67e74705SXin Li                                 LBracLoc, RBracLoc, RecRange, ReturnType, VK))
2881*67e74705SXin Li     return ExprError();
2882*67e74705SXin Li 
2883*67e74705SXin Li   if (Method && !Method->getReturnType()->isVoidType() &&
2884*67e74705SXin Li       RequireCompleteType(LBracLoc, Method->getReturnType(),
2885*67e74705SXin Li                           diag::err_illegal_message_expr_incomplete_type))
2886*67e74705SXin Li     return ExprError();
2887*67e74705SXin Li 
2888*67e74705SXin Li   // In ARC, forbid the user from sending messages to
2889*67e74705SXin Li   // retain/release/autorelease/dealloc/retainCount explicitly.
2890*67e74705SXin Li   if (getLangOpts().ObjCAutoRefCount) {
2891*67e74705SXin Li     ObjCMethodFamily family =
2892*67e74705SXin Li       (Method ? Method->getMethodFamily() : Sel.getMethodFamily());
2893*67e74705SXin Li     switch (family) {
2894*67e74705SXin Li     case OMF_init:
2895*67e74705SXin Li       if (Method)
2896*67e74705SXin Li         checkInitMethod(Method, ReceiverType);
2897*67e74705SXin Li 
2898*67e74705SXin Li     case OMF_None:
2899*67e74705SXin Li     case OMF_alloc:
2900*67e74705SXin Li     case OMF_copy:
2901*67e74705SXin Li     case OMF_finalize:
2902*67e74705SXin Li     case OMF_mutableCopy:
2903*67e74705SXin Li     case OMF_new:
2904*67e74705SXin Li     case OMF_self:
2905*67e74705SXin Li     case OMF_initialize:
2906*67e74705SXin Li       break;
2907*67e74705SXin Li 
2908*67e74705SXin Li     case OMF_dealloc:
2909*67e74705SXin Li     case OMF_retain:
2910*67e74705SXin Li     case OMF_release:
2911*67e74705SXin Li     case OMF_autorelease:
2912*67e74705SXin Li     case OMF_retainCount:
2913*67e74705SXin Li       Diag(SelLoc, diag::err_arc_illegal_explicit_message)
2914*67e74705SXin Li         << Sel << RecRange;
2915*67e74705SXin Li       break;
2916*67e74705SXin Li 
2917*67e74705SXin Li     case OMF_performSelector:
2918*67e74705SXin Li       if (Method && NumArgs >= 1) {
2919*67e74705SXin Li         if (ObjCSelectorExpr *SelExp = dyn_cast<ObjCSelectorExpr>(Args[0])) {
2920*67e74705SXin Li           Selector ArgSel = SelExp->getSelector();
2921*67e74705SXin Li           ObjCMethodDecl *SelMethod =
2922*67e74705SXin Li             LookupInstanceMethodInGlobalPool(ArgSel,
2923*67e74705SXin Li                                              SelExp->getSourceRange());
2924*67e74705SXin Li           if (!SelMethod)
2925*67e74705SXin Li             SelMethod =
2926*67e74705SXin Li               LookupFactoryMethodInGlobalPool(ArgSel,
2927*67e74705SXin Li                                               SelExp->getSourceRange());
2928*67e74705SXin Li           if (SelMethod) {
2929*67e74705SXin Li             ObjCMethodFamily SelFamily = SelMethod->getMethodFamily();
2930*67e74705SXin Li             switch (SelFamily) {
2931*67e74705SXin Li               case OMF_alloc:
2932*67e74705SXin Li               case OMF_copy:
2933*67e74705SXin Li               case OMF_mutableCopy:
2934*67e74705SXin Li               case OMF_new:
2935*67e74705SXin Li               case OMF_self:
2936*67e74705SXin Li               case OMF_init:
2937*67e74705SXin Li                 // Issue error, unless ns_returns_not_retained.
2938*67e74705SXin Li                 if (!SelMethod->hasAttr<NSReturnsNotRetainedAttr>()) {
2939*67e74705SXin Li                   // selector names a +1 method
2940*67e74705SXin Li                   Diag(SelLoc,
2941*67e74705SXin Li                        diag::err_arc_perform_selector_retains);
2942*67e74705SXin Li                   Diag(SelMethod->getLocation(), diag::note_method_declared_at)
2943*67e74705SXin Li                     << SelMethod->getDeclName();
2944*67e74705SXin Li                 }
2945*67e74705SXin Li                 break;
2946*67e74705SXin Li               default:
2947*67e74705SXin Li                 // +0 call. OK. unless ns_returns_retained.
2948*67e74705SXin Li                 if (SelMethod->hasAttr<NSReturnsRetainedAttr>()) {
2949*67e74705SXin Li                   // selector names a +1 method
2950*67e74705SXin Li                   Diag(SelLoc,
2951*67e74705SXin Li                        diag::err_arc_perform_selector_retains);
2952*67e74705SXin Li                   Diag(SelMethod->getLocation(), diag::note_method_declared_at)
2953*67e74705SXin Li                     << SelMethod->getDeclName();
2954*67e74705SXin Li                 }
2955*67e74705SXin Li                 break;
2956*67e74705SXin Li             }
2957*67e74705SXin Li           }
2958*67e74705SXin Li         } else {
2959*67e74705SXin Li           // error (may leak).
2960*67e74705SXin Li           Diag(SelLoc, diag::warn_arc_perform_selector_leaks);
2961*67e74705SXin Li           Diag(Args[0]->getExprLoc(), diag::note_used_here);
2962*67e74705SXin Li         }
2963*67e74705SXin Li       }
2964*67e74705SXin Li       break;
2965*67e74705SXin Li     }
2966*67e74705SXin Li   }
2967*67e74705SXin Li 
2968*67e74705SXin Li   DiagnoseCStringFormatDirectiveInObjCAPI(*this, Method, Sel, Args, NumArgs);
2969*67e74705SXin Li 
2970*67e74705SXin Li   // Construct the appropriate ObjCMessageExpr instance.
2971*67e74705SXin Li   ObjCMessageExpr *Result;
2972*67e74705SXin Li   if (SuperLoc.isValid())
2973*67e74705SXin Li     Result = ObjCMessageExpr::Create(Context, ReturnType, VK, LBracLoc,
2974*67e74705SXin Li                                      SuperLoc,  /*IsInstanceSuper=*/true,
2975*67e74705SXin Li                                      ReceiverType, Sel, SelectorLocs, Method,
2976*67e74705SXin Li                                      makeArrayRef(Args, NumArgs), RBracLoc,
2977*67e74705SXin Li                                      isImplicit);
2978*67e74705SXin Li   else {
2979*67e74705SXin Li     Result = ObjCMessageExpr::Create(Context, ReturnType, VK, LBracLoc,
2980*67e74705SXin Li                                      Receiver, Sel, SelectorLocs, Method,
2981*67e74705SXin Li                                      makeArrayRef(Args, NumArgs), RBracLoc,
2982*67e74705SXin Li                                      isImplicit);
2983*67e74705SXin Li     if (!isImplicit)
2984*67e74705SXin Li       checkCocoaAPI(*this, Result);
2985*67e74705SXin Li   }
2986*67e74705SXin Li 
2987*67e74705SXin Li   if (getLangOpts().ObjCAutoRefCount) {
2988*67e74705SXin Li     // In ARC, annotate delegate init calls.
2989*67e74705SXin Li     if (Result->getMethodFamily() == OMF_init &&
2990*67e74705SXin Li         (SuperLoc.isValid() || isSelfExpr(Receiver))) {
2991*67e74705SXin Li       // Only consider init calls *directly* in init implementations,
2992*67e74705SXin Li       // not within blocks.
2993*67e74705SXin Li       ObjCMethodDecl *method = dyn_cast<ObjCMethodDecl>(CurContext);
2994*67e74705SXin Li       if (method && method->getMethodFamily() == OMF_init) {
2995*67e74705SXin Li         // The implicit assignment to self means we also don't want to
2996*67e74705SXin Li         // consume the result.
2997*67e74705SXin Li         Result->setDelegateInitCall(true);
2998*67e74705SXin Li         return Result;
2999*67e74705SXin Li       }
3000*67e74705SXin Li     }
3001*67e74705SXin Li 
3002*67e74705SXin Li     // In ARC, check for message sends which are likely to introduce
3003*67e74705SXin Li     // retain cycles.
3004*67e74705SXin Li     checkRetainCycles(Result);
3005*67e74705SXin Li 
3006*67e74705SXin Li     if (!isImplicit && Method) {
3007*67e74705SXin Li       if (const ObjCPropertyDecl *Prop = Method->findPropertyDecl()) {
3008*67e74705SXin Li         bool IsWeak =
3009*67e74705SXin Li           Prop->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_weak;
3010*67e74705SXin Li         if (!IsWeak && Sel.isUnarySelector())
3011*67e74705SXin Li           IsWeak = ReturnType.getObjCLifetime() & Qualifiers::OCL_Weak;
3012*67e74705SXin Li         if (IsWeak &&
3013*67e74705SXin Li             !Diags.isIgnored(diag::warn_arc_repeated_use_of_weak, LBracLoc))
3014*67e74705SXin Li           getCurFunction()->recordUseOfWeak(Result, Prop);
3015*67e74705SXin Li       }
3016*67e74705SXin Li     }
3017*67e74705SXin Li   }
3018*67e74705SXin Li 
3019*67e74705SXin Li   CheckObjCCircularContainer(Result);
3020*67e74705SXin Li 
3021*67e74705SXin Li   return MaybeBindToTemporary(Result);
3022*67e74705SXin Li }
3023*67e74705SXin Li 
RemoveSelectorFromWarningCache(Sema & S,Expr * Arg)3024*67e74705SXin Li static void RemoveSelectorFromWarningCache(Sema &S, Expr* Arg) {
3025*67e74705SXin Li   if (ObjCSelectorExpr *OSE =
3026*67e74705SXin Li       dyn_cast<ObjCSelectorExpr>(Arg->IgnoreParenCasts())) {
3027*67e74705SXin Li     Selector Sel = OSE->getSelector();
3028*67e74705SXin Li     SourceLocation Loc = OSE->getAtLoc();
3029*67e74705SXin Li     auto Pos = S.ReferencedSelectors.find(Sel);
3030*67e74705SXin Li     if (Pos != S.ReferencedSelectors.end() && Pos->second == Loc)
3031*67e74705SXin Li       S.ReferencedSelectors.erase(Pos);
3032*67e74705SXin Li   }
3033*67e74705SXin Li }
3034*67e74705SXin Li 
3035*67e74705SXin Li // ActOnInstanceMessage - used for both unary and keyword messages.
3036*67e74705SXin Li // ArgExprs is optional - if it is present, the number of expressions
3037*67e74705SXin Li // is obtained from Sel.getNumArgs().
ActOnInstanceMessage(Scope * S,Expr * Receiver,Selector Sel,SourceLocation LBracLoc,ArrayRef<SourceLocation> SelectorLocs,SourceLocation RBracLoc,MultiExprArg Args)3038*67e74705SXin Li ExprResult Sema::ActOnInstanceMessage(Scope *S,
3039*67e74705SXin Li                                       Expr *Receiver,
3040*67e74705SXin Li                                       Selector Sel,
3041*67e74705SXin Li                                       SourceLocation LBracLoc,
3042*67e74705SXin Li                                       ArrayRef<SourceLocation> SelectorLocs,
3043*67e74705SXin Li                                       SourceLocation RBracLoc,
3044*67e74705SXin Li                                       MultiExprArg Args) {
3045*67e74705SXin Li   if (!Receiver)
3046*67e74705SXin Li     return ExprError();
3047*67e74705SXin Li 
3048*67e74705SXin Li   // A ParenListExpr can show up while doing error recovery with invalid code.
3049*67e74705SXin Li   if (isa<ParenListExpr>(Receiver)) {
3050*67e74705SXin Li     ExprResult Result = MaybeConvertParenListExprToParenExpr(S, Receiver);
3051*67e74705SXin Li     if (Result.isInvalid()) return ExprError();
3052*67e74705SXin Li     Receiver = Result.get();
3053*67e74705SXin Li   }
3054*67e74705SXin Li 
3055*67e74705SXin Li   if (RespondsToSelectorSel.isNull()) {
3056*67e74705SXin Li     IdentifierInfo *SelectorId = &Context.Idents.get("respondsToSelector");
3057*67e74705SXin Li     RespondsToSelectorSel = Context.Selectors.getUnarySelector(SelectorId);
3058*67e74705SXin Li   }
3059*67e74705SXin Li   if (Sel == RespondsToSelectorSel)
3060*67e74705SXin Li     RemoveSelectorFromWarningCache(*this, Args[0]);
3061*67e74705SXin Li 
3062*67e74705SXin Li   return BuildInstanceMessage(Receiver, Receiver->getType(),
3063*67e74705SXin Li                               /*SuperLoc=*/SourceLocation(), Sel,
3064*67e74705SXin Li                               /*Method=*/nullptr, LBracLoc, SelectorLocs,
3065*67e74705SXin Li                               RBracLoc, Args);
3066*67e74705SXin Li }
3067*67e74705SXin Li 
3068*67e74705SXin Li enum ARCConversionTypeClass {
3069*67e74705SXin Li   /// int, void, struct A
3070*67e74705SXin Li   ACTC_none,
3071*67e74705SXin Li 
3072*67e74705SXin Li   /// id, void (^)()
3073*67e74705SXin Li   ACTC_retainable,
3074*67e74705SXin Li 
3075*67e74705SXin Li   /// id*, id***, void (^*)(),
3076*67e74705SXin Li   ACTC_indirectRetainable,
3077*67e74705SXin Li 
3078*67e74705SXin Li   /// void* might be a normal C type, or it might a CF type.
3079*67e74705SXin Li   ACTC_voidPtr,
3080*67e74705SXin Li 
3081*67e74705SXin Li   /// struct A*
3082*67e74705SXin Li   ACTC_coreFoundation
3083*67e74705SXin Li };
3084*67e74705SXin Li 
isAnyRetainable(ARCConversionTypeClass ACTC)3085*67e74705SXin Li static bool isAnyRetainable(ARCConversionTypeClass ACTC) {
3086*67e74705SXin Li   return (ACTC == ACTC_retainable ||
3087*67e74705SXin Li           ACTC == ACTC_coreFoundation ||
3088*67e74705SXin Li           ACTC == ACTC_voidPtr);
3089*67e74705SXin Li }
3090*67e74705SXin Li 
isAnyCLike(ARCConversionTypeClass ACTC)3091*67e74705SXin Li static bool isAnyCLike(ARCConversionTypeClass ACTC) {
3092*67e74705SXin Li   return ACTC == ACTC_none ||
3093*67e74705SXin Li          ACTC == ACTC_voidPtr ||
3094*67e74705SXin Li          ACTC == ACTC_coreFoundation;
3095*67e74705SXin Li }
3096*67e74705SXin Li 
classifyTypeForARCConversion(QualType type)3097*67e74705SXin Li static ARCConversionTypeClass classifyTypeForARCConversion(QualType type) {
3098*67e74705SXin Li   bool isIndirect = false;
3099*67e74705SXin Li 
3100*67e74705SXin Li   // Ignore an outermost reference type.
3101*67e74705SXin Li   if (const ReferenceType *ref = type->getAs<ReferenceType>()) {
3102*67e74705SXin Li     type = ref->getPointeeType();
3103*67e74705SXin Li     isIndirect = true;
3104*67e74705SXin Li   }
3105*67e74705SXin Li 
3106*67e74705SXin Li   // Drill through pointers and arrays recursively.
3107*67e74705SXin Li   while (true) {
3108*67e74705SXin Li     if (const PointerType *ptr = type->getAs<PointerType>()) {
3109*67e74705SXin Li       type = ptr->getPointeeType();
3110*67e74705SXin Li 
3111*67e74705SXin Li       // The first level of pointer may be the innermost pointer on a CF type.
3112*67e74705SXin Li       if (!isIndirect) {
3113*67e74705SXin Li         if (type->isVoidType()) return ACTC_voidPtr;
3114*67e74705SXin Li         if (type->isRecordType()) return ACTC_coreFoundation;
3115*67e74705SXin Li       }
3116*67e74705SXin Li     } else if (const ArrayType *array = type->getAsArrayTypeUnsafe()) {
3117*67e74705SXin Li       type = QualType(array->getElementType()->getBaseElementTypeUnsafe(), 0);
3118*67e74705SXin Li     } else {
3119*67e74705SXin Li       break;
3120*67e74705SXin Li     }
3121*67e74705SXin Li     isIndirect = true;
3122*67e74705SXin Li   }
3123*67e74705SXin Li 
3124*67e74705SXin Li   if (isIndirect) {
3125*67e74705SXin Li     if (type->isObjCARCBridgableType())
3126*67e74705SXin Li       return ACTC_indirectRetainable;
3127*67e74705SXin Li     return ACTC_none;
3128*67e74705SXin Li   }
3129*67e74705SXin Li 
3130*67e74705SXin Li   if (type->isObjCARCBridgableType())
3131*67e74705SXin Li     return ACTC_retainable;
3132*67e74705SXin Li 
3133*67e74705SXin Li   return ACTC_none;
3134*67e74705SXin Li }
3135*67e74705SXin Li 
3136*67e74705SXin Li namespace {
3137*67e74705SXin Li   /// A result from the cast checker.
3138*67e74705SXin Li   enum ACCResult {
3139*67e74705SXin Li     /// Cannot be casted.
3140*67e74705SXin Li     ACC_invalid,
3141*67e74705SXin Li 
3142*67e74705SXin Li     /// Can be safely retained or not retained.
3143*67e74705SXin Li     ACC_bottom,
3144*67e74705SXin Li 
3145*67e74705SXin Li     /// Can be casted at +0.
3146*67e74705SXin Li     ACC_plusZero,
3147*67e74705SXin Li 
3148*67e74705SXin Li     /// Can be casted at +1.
3149*67e74705SXin Li     ACC_plusOne
3150*67e74705SXin Li   };
merge(ACCResult left,ACCResult right)3151*67e74705SXin Li   ACCResult merge(ACCResult left, ACCResult right) {
3152*67e74705SXin Li     if (left == right) return left;
3153*67e74705SXin Li     if (left == ACC_bottom) return right;
3154*67e74705SXin Li     if (right == ACC_bottom) return left;
3155*67e74705SXin Li     return ACC_invalid;
3156*67e74705SXin Li   }
3157*67e74705SXin Li 
3158*67e74705SXin Li   /// A checker which white-lists certain expressions whose conversion
3159*67e74705SXin Li   /// to or from retainable type would otherwise be forbidden in ARC.
3160*67e74705SXin Li   class ARCCastChecker : public StmtVisitor<ARCCastChecker, ACCResult> {
3161*67e74705SXin Li     typedef StmtVisitor<ARCCastChecker, ACCResult> super;
3162*67e74705SXin Li 
3163*67e74705SXin Li     ASTContext &Context;
3164*67e74705SXin Li     ARCConversionTypeClass SourceClass;
3165*67e74705SXin Li     ARCConversionTypeClass TargetClass;
3166*67e74705SXin Li     bool Diagnose;
3167*67e74705SXin Li 
isCFType(QualType type)3168*67e74705SXin Li     static bool isCFType(QualType type) {
3169*67e74705SXin Li       // Someday this can use ns_bridged.  For now, it has to do this.
3170*67e74705SXin Li       return type->isCARCBridgableType();
3171*67e74705SXin Li     }
3172*67e74705SXin Li 
3173*67e74705SXin Li   public:
ARCCastChecker(ASTContext & Context,ARCConversionTypeClass source,ARCConversionTypeClass target,bool diagnose)3174*67e74705SXin Li     ARCCastChecker(ASTContext &Context, ARCConversionTypeClass source,
3175*67e74705SXin Li                    ARCConversionTypeClass target, bool diagnose)
3176*67e74705SXin Li       : Context(Context), SourceClass(source), TargetClass(target),
3177*67e74705SXin Li         Diagnose(diagnose) {}
3178*67e74705SXin Li 
3179*67e74705SXin Li     using super::Visit;
Visit(Expr * e)3180*67e74705SXin Li     ACCResult Visit(Expr *e) {
3181*67e74705SXin Li       return super::Visit(e->IgnoreParens());
3182*67e74705SXin Li     }
3183*67e74705SXin Li 
VisitStmt(Stmt * s)3184*67e74705SXin Li     ACCResult VisitStmt(Stmt *s) {
3185*67e74705SXin Li       return ACC_invalid;
3186*67e74705SXin Li     }
3187*67e74705SXin Li 
3188*67e74705SXin Li     /// Null pointer constants can be casted however you please.
VisitExpr(Expr * e)3189*67e74705SXin Li     ACCResult VisitExpr(Expr *e) {
3190*67e74705SXin Li       if (e->isNullPointerConstant(Context, Expr::NPC_ValueDependentIsNotNull))
3191*67e74705SXin Li         return ACC_bottom;
3192*67e74705SXin Li       return ACC_invalid;
3193*67e74705SXin Li     }
3194*67e74705SXin Li 
3195*67e74705SXin Li     /// Objective-C string literals can be safely casted.
VisitObjCStringLiteral(ObjCStringLiteral * e)3196*67e74705SXin Li     ACCResult VisitObjCStringLiteral(ObjCStringLiteral *e) {
3197*67e74705SXin Li       // If we're casting to any retainable type, go ahead.  Global
3198*67e74705SXin Li       // strings are immune to retains, so this is bottom.
3199*67e74705SXin Li       if (isAnyRetainable(TargetClass)) return ACC_bottom;
3200*67e74705SXin Li 
3201*67e74705SXin Li       return ACC_invalid;
3202*67e74705SXin Li     }
3203*67e74705SXin Li 
3204*67e74705SXin Li     /// Look through certain implicit and explicit casts.
VisitCastExpr(CastExpr * e)3205*67e74705SXin Li     ACCResult VisitCastExpr(CastExpr *e) {
3206*67e74705SXin Li       switch (e->getCastKind()) {
3207*67e74705SXin Li         case CK_NullToPointer:
3208*67e74705SXin Li           return ACC_bottom;
3209*67e74705SXin Li 
3210*67e74705SXin Li         case CK_NoOp:
3211*67e74705SXin Li         case CK_LValueToRValue:
3212*67e74705SXin Li         case CK_BitCast:
3213*67e74705SXin Li         case CK_CPointerToObjCPointerCast:
3214*67e74705SXin Li         case CK_BlockPointerToObjCPointerCast:
3215*67e74705SXin Li         case CK_AnyPointerToBlockPointerCast:
3216*67e74705SXin Li           return Visit(e->getSubExpr());
3217*67e74705SXin Li 
3218*67e74705SXin Li         default:
3219*67e74705SXin Li           return ACC_invalid;
3220*67e74705SXin Li       }
3221*67e74705SXin Li     }
3222*67e74705SXin Li 
3223*67e74705SXin Li     /// Look through unary extension.
VisitUnaryExtension(UnaryOperator * e)3224*67e74705SXin Li     ACCResult VisitUnaryExtension(UnaryOperator *e) {
3225*67e74705SXin Li       return Visit(e->getSubExpr());
3226*67e74705SXin Li     }
3227*67e74705SXin Li 
3228*67e74705SXin Li     /// Ignore the LHS of a comma operator.
VisitBinComma(BinaryOperator * e)3229*67e74705SXin Li     ACCResult VisitBinComma(BinaryOperator *e) {
3230*67e74705SXin Li       return Visit(e->getRHS());
3231*67e74705SXin Li     }
3232*67e74705SXin Li 
3233*67e74705SXin Li     /// Conditional operators are okay if both sides are okay.
VisitConditionalOperator(ConditionalOperator * e)3234*67e74705SXin Li     ACCResult VisitConditionalOperator(ConditionalOperator *e) {
3235*67e74705SXin Li       ACCResult left = Visit(e->getTrueExpr());
3236*67e74705SXin Li       if (left == ACC_invalid) return ACC_invalid;
3237*67e74705SXin Li       return merge(left, Visit(e->getFalseExpr()));
3238*67e74705SXin Li     }
3239*67e74705SXin Li 
3240*67e74705SXin Li     /// Look through pseudo-objects.
VisitPseudoObjectExpr(PseudoObjectExpr * e)3241*67e74705SXin Li     ACCResult VisitPseudoObjectExpr(PseudoObjectExpr *e) {
3242*67e74705SXin Li       // If we're getting here, we should always have a result.
3243*67e74705SXin Li       return Visit(e->getResultExpr());
3244*67e74705SXin Li     }
3245*67e74705SXin Li 
3246*67e74705SXin Li     /// Statement expressions are okay if their result expression is okay.
VisitStmtExpr(StmtExpr * e)3247*67e74705SXin Li     ACCResult VisitStmtExpr(StmtExpr *e) {
3248*67e74705SXin Li       return Visit(e->getSubStmt()->body_back());
3249*67e74705SXin Li     }
3250*67e74705SXin Li 
3251*67e74705SXin Li     /// Some declaration references are okay.
VisitDeclRefExpr(DeclRefExpr * e)3252*67e74705SXin Li     ACCResult VisitDeclRefExpr(DeclRefExpr *e) {
3253*67e74705SXin Li       VarDecl *var = dyn_cast<VarDecl>(e->getDecl());
3254*67e74705SXin Li       // References to global constants are okay.
3255*67e74705SXin Li       if (isAnyRetainable(TargetClass) &&
3256*67e74705SXin Li           isAnyRetainable(SourceClass) &&
3257*67e74705SXin Li           var &&
3258*67e74705SXin Li           var->getStorageClass() == SC_Extern &&
3259*67e74705SXin Li           var->getType().isConstQualified()) {
3260*67e74705SXin Li 
3261*67e74705SXin Li         // In system headers, they can also be assumed to be immune to retains.
3262*67e74705SXin Li         // These are things like 'kCFStringTransformToLatin'.
3263*67e74705SXin Li         if (Context.getSourceManager().isInSystemHeader(var->getLocation()))
3264*67e74705SXin Li           return ACC_bottom;
3265*67e74705SXin Li 
3266*67e74705SXin Li         return ACC_plusZero;
3267*67e74705SXin Li       }
3268*67e74705SXin Li 
3269*67e74705SXin Li       // Nothing else.
3270*67e74705SXin Li       return ACC_invalid;
3271*67e74705SXin Li     }
3272*67e74705SXin Li 
3273*67e74705SXin Li     /// Some calls are okay.
VisitCallExpr(CallExpr * e)3274*67e74705SXin Li     ACCResult VisitCallExpr(CallExpr *e) {
3275*67e74705SXin Li       if (FunctionDecl *fn = e->getDirectCallee())
3276*67e74705SXin Li         if (ACCResult result = checkCallToFunction(fn))
3277*67e74705SXin Li           return result;
3278*67e74705SXin Li 
3279*67e74705SXin Li       return super::VisitCallExpr(e);
3280*67e74705SXin Li     }
3281*67e74705SXin Li 
checkCallToFunction(FunctionDecl * fn)3282*67e74705SXin Li     ACCResult checkCallToFunction(FunctionDecl *fn) {
3283*67e74705SXin Li       // Require a CF*Ref return type.
3284*67e74705SXin Li       if (!isCFType(fn->getReturnType()))
3285*67e74705SXin Li         return ACC_invalid;
3286*67e74705SXin Li 
3287*67e74705SXin Li       if (!isAnyRetainable(TargetClass))
3288*67e74705SXin Li         return ACC_invalid;
3289*67e74705SXin Li 
3290*67e74705SXin Li       // Honor an explicit 'not retained' attribute.
3291*67e74705SXin Li       if (fn->hasAttr<CFReturnsNotRetainedAttr>())
3292*67e74705SXin Li         return ACC_plusZero;
3293*67e74705SXin Li 
3294*67e74705SXin Li       // Honor an explicit 'retained' attribute, except that for
3295*67e74705SXin Li       // now we're not going to permit implicit handling of +1 results,
3296*67e74705SXin Li       // because it's a bit frightening.
3297*67e74705SXin Li       if (fn->hasAttr<CFReturnsRetainedAttr>())
3298*67e74705SXin Li         return Diagnose ? ACC_plusOne
3299*67e74705SXin Li                         : ACC_invalid; // ACC_plusOne if we start accepting this
3300*67e74705SXin Li 
3301*67e74705SXin Li       // Recognize this specific builtin function, which is used by CFSTR.
3302*67e74705SXin Li       unsigned builtinID = fn->getBuiltinID();
3303*67e74705SXin Li       if (builtinID == Builtin::BI__builtin___CFStringMakeConstantString)
3304*67e74705SXin Li         return ACC_bottom;
3305*67e74705SXin Li 
3306*67e74705SXin Li       // Otherwise, don't do anything implicit with an unaudited function.
3307*67e74705SXin Li       if (!fn->hasAttr<CFAuditedTransferAttr>())
3308*67e74705SXin Li         return ACC_invalid;
3309*67e74705SXin Li 
3310*67e74705SXin Li       // Otherwise, it's +0 unless it follows the create convention.
3311*67e74705SXin Li       if (ento::coreFoundation::followsCreateRule(fn))
3312*67e74705SXin Li         return Diagnose ? ACC_plusOne
3313*67e74705SXin Li                         : ACC_invalid; // ACC_plusOne if we start accepting this
3314*67e74705SXin Li 
3315*67e74705SXin Li       return ACC_plusZero;
3316*67e74705SXin Li     }
3317*67e74705SXin Li 
VisitObjCMessageExpr(ObjCMessageExpr * e)3318*67e74705SXin Li     ACCResult VisitObjCMessageExpr(ObjCMessageExpr *e) {
3319*67e74705SXin Li       return checkCallToMethod(e->getMethodDecl());
3320*67e74705SXin Li     }
3321*67e74705SXin Li 
VisitObjCPropertyRefExpr(ObjCPropertyRefExpr * e)3322*67e74705SXin Li     ACCResult VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *e) {
3323*67e74705SXin Li       ObjCMethodDecl *method;
3324*67e74705SXin Li       if (e->isExplicitProperty())
3325*67e74705SXin Li         method = e->getExplicitProperty()->getGetterMethodDecl();
3326*67e74705SXin Li       else
3327*67e74705SXin Li         method = e->getImplicitPropertyGetter();
3328*67e74705SXin Li       return checkCallToMethod(method);
3329*67e74705SXin Li     }
3330*67e74705SXin Li 
checkCallToMethod(ObjCMethodDecl * method)3331*67e74705SXin Li     ACCResult checkCallToMethod(ObjCMethodDecl *method) {
3332*67e74705SXin Li       if (!method) return ACC_invalid;
3333*67e74705SXin Li 
3334*67e74705SXin Li       // Check for message sends to functions returning CF types.  We
3335*67e74705SXin Li       // just obey the Cocoa conventions with these, even though the
3336*67e74705SXin Li       // return type is CF.
3337*67e74705SXin Li       if (!isAnyRetainable(TargetClass) || !isCFType(method->getReturnType()))
3338*67e74705SXin Li         return ACC_invalid;
3339*67e74705SXin Li 
3340*67e74705SXin Li       // If the method is explicitly marked not-retained, it's +0.
3341*67e74705SXin Li       if (method->hasAttr<CFReturnsNotRetainedAttr>())
3342*67e74705SXin Li         return ACC_plusZero;
3343*67e74705SXin Li 
3344*67e74705SXin Li       // If the method is explicitly marked as returning retained, or its
3345*67e74705SXin Li       // selector follows a +1 Cocoa convention, treat it as +1.
3346*67e74705SXin Li       if (method->hasAttr<CFReturnsRetainedAttr>())
3347*67e74705SXin Li         return ACC_plusOne;
3348*67e74705SXin Li 
3349*67e74705SXin Li       switch (method->getSelector().getMethodFamily()) {
3350*67e74705SXin Li       case OMF_alloc:
3351*67e74705SXin Li       case OMF_copy:
3352*67e74705SXin Li       case OMF_mutableCopy:
3353*67e74705SXin Li       case OMF_new:
3354*67e74705SXin Li         return ACC_plusOne;
3355*67e74705SXin Li 
3356*67e74705SXin Li       default:
3357*67e74705SXin Li         // Otherwise, treat it as +0.
3358*67e74705SXin Li         return ACC_plusZero;
3359*67e74705SXin Li       }
3360*67e74705SXin Li     }
3361*67e74705SXin Li   };
3362*67e74705SXin Li } // end anonymous namespace
3363*67e74705SXin Li 
isKnownName(StringRef name)3364*67e74705SXin Li bool Sema::isKnownName(StringRef name) {
3365*67e74705SXin Li   if (name.empty())
3366*67e74705SXin Li     return false;
3367*67e74705SXin Li   LookupResult R(*this, &Context.Idents.get(name), SourceLocation(),
3368*67e74705SXin Li                  Sema::LookupOrdinaryName);
3369*67e74705SXin Li   return LookupName(R, TUScope, false);
3370*67e74705SXin Li }
3371*67e74705SXin Li 
addFixitForObjCARCConversion(Sema & S,DiagnosticBuilder & DiagB,Sema::CheckedConversionKind CCK,SourceLocation afterLParen,QualType castType,Expr * castExpr,Expr * realCast,const char * bridgeKeyword,const char * CFBridgeName)3372*67e74705SXin Li static void addFixitForObjCARCConversion(Sema &S,
3373*67e74705SXin Li                                          DiagnosticBuilder &DiagB,
3374*67e74705SXin Li                                          Sema::CheckedConversionKind CCK,
3375*67e74705SXin Li                                          SourceLocation afterLParen,
3376*67e74705SXin Li                                          QualType castType,
3377*67e74705SXin Li                                          Expr *castExpr,
3378*67e74705SXin Li                                          Expr *realCast,
3379*67e74705SXin Li                                          const char *bridgeKeyword,
3380*67e74705SXin Li                                          const char *CFBridgeName) {
3381*67e74705SXin Li   // We handle C-style and implicit casts here.
3382*67e74705SXin Li   switch (CCK) {
3383*67e74705SXin Li   case Sema::CCK_ImplicitConversion:
3384*67e74705SXin Li   case Sema::CCK_CStyleCast:
3385*67e74705SXin Li   case Sema::CCK_OtherCast:
3386*67e74705SXin Li     break;
3387*67e74705SXin Li   case Sema::CCK_FunctionalCast:
3388*67e74705SXin Li     return;
3389*67e74705SXin Li   }
3390*67e74705SXin Li 
3391*67e74705SXin Li   if (CFBridgeName) {
3392*67e74705SXin Li     if (CCK == Sema::CCK_OtherCast) {
3393*67e74705SXin Li       if (const CXXNamedCastExpr *NCE = dyn_cast<CXXNamedCastExpr>(realCast)) {
3394*67e74705SXin Li         SourceRange range(NCE->getOperatorLoc(),
3395*67e74705SXin Li                           NCE->getAngleBrackets().getEnd());
3396*67e74705SXin Li         SmallString<32> BridgeCall;
3397*67e74705SXin Li 
3398*67e74705SXin Li         SourceManager &SM = S.getSourceManager();
3399*67e74705SXin Li         char PrevChar = *SM.getCharacterData(range.getBegin().getLocWithOffset(-1));
3400*67e74705SXin Li         if (Lexer::isIdentifierBodyChar(PrevChar, S.getLangOpts()))
3401*67e74705SXin Li           BridgeCall += ' ';
3402*67e74705SXin Li 
3403*67e74705SXin Li         BridgeCall += CFBridgeName;
3404*67e74705SXin Li         DiagB.AddFixItHint(FixItHint::CreateReplacement(range, BridgeCall));
3405*67e74705SXin Li       }
3406*67e74705SXin Li       return;
3407*67e74705SXin Li     }
3408*67e74705SXin Li     Expr *castedE = castExpr;
3409*67e74705SXin Li     if (CStyleCastExpr *CCE = dyn_cast<CStyleCastExpr>(castedE))
3410*67e74705SXin Li       castedE = CCE->getSubExpr();
3411*67e74705SXin Li     castedE = castedE->IgnoreImpCasts();
3412*67e74705SXin Li     SourceRange range = castedE->getSourceRange();
3413*67e74705SXin Li 
3414*67e74705SXin Li     SmallString<32> BridgeCall;
3415*67e74705SXin Li 
3416*67e74705SXin Li     SourceManager &SM = S.getSourceManager();
3417*67e74705SXin Li     char PrevChar = *SM.getCharacterData(range.getBegin().getLocWithOffset(-1));
3418*67e74705SXin Li     if (Lexer::isIdentifierBodyChar(PrevChar, S.getLangOpts()))
3419*67e74705SXin Li       BridgeCall += ' ';
3420*67e74705SXin Li 
3421*67e74705SXin Li     BridgeCall += CFBridgeName;
3422*67e74705SXin Li 
3423*67e74705SXin Li     if (isa<ParenExpr>(castedE)) {
3424*67e74705SXin Li       DiagB.AddFixItHint(FixItHint::CreateInsertion(range.getBegin(),
3425*67e74705SXin Li                          BridgeCall));
3426*67e74705SXin Li     } else {
3427*67e74705SXin Li       BridgeCall += '(';
3428*67e74705SXin Li       DiagB.AddFixItHint(FixItHint::CreateInsertion(range.getBegin(),
3429*67e74705SXin Li                                                     BridgeCall));
3430*67e74705SXin Li       DiagB.AddFixItHint(FixItHint::CreateInsertion(
3431*67e74705SXin Li                                        S.getLocForEndOfToken(range.getEnd()),
3432*67e74705SXin Li                                        ")"));
3433*67e74705SXin Li     }
3434*67e74705SXin Li     return;
3435*67e74705SXin Li   }
3436*67e74705SXin Li 
3437*67e74705SXin Li   if (CCK == Sema::CCK_CStyleCast) {
3438*67e74705SXin Li     DiagB.AddFixItHint(FixItHint::CreateInsertion(afterLParen, bridgeKeyword));
3439*67e74705SXin Li   } else if (CCK == Sema::CCK_OtherCast) {
3440*67e74705SXin Li     if (const CXXNamedCastExpr *NCE = dyn_cast<CXXNamedCastExpr>(realCast)) {
3441*67e74705SXin Li       std::string castCode = "(";
3442*67e74705SXin Li       castCode += bridgeKeyword;
3443*67e74705SXin Li       castCode += castType.getAsString();
3444*67e74705SXin Li       castCode += ")";
3445*67e74705SXin Li       SourceRange Range(NCE->getOperatorLoc(),
3446*67e74705SXin Li                         NCE->getAngleBrackets().getEnd());
3447*67e74705SXin Li       DiagB.AddFixItHint(FixItHint::CreateReplacement(Range, castCode));
3448*67e74705SXin Li     }
3449*67e74705SXin Li   } else {
3450*67e74705SXin Li     std::string castCode = "(";
3451*67e74705SXin Li     castCode += bridgeKeyword;
3452*67e74705SXin Li     castCode += castType.getAsString();
3453*67e74705SXin Li     castCode += ")";
3454*67e74705SXin Li     Expr *castedE = castExpr->IgnoreImpCasts();
3455*67e74705SXin Li     SourceRange range = castedE->getSourceRange();
3456*67e74705SXin Li     if (isa<ParenExpr>(castedE)) {
3457*67e74705SXin Li       DiagB.AddFixItHint(FixItHint::CreateInsertion(range.getBegin(),
3458*67e74705SXin Li                          castCode));
3459*67e74705SXin Li     } else {
3460*67e74705SXin Li       castCode += "(";
3461*67e74705SXin Li       DiagB.AddFixItHint(FixItHint::CreateInsertion(range.getBegin(),
3462*67e74705SXin Li                                                     castCode));
3463*67e74705SXin Li       DiagB.AddFixItHint(FixItHint::CreateInsertion(
3464*67e74705SXin Li                                        S.getLocForEndOfToken(range.getEnd()),
3465*67e74705SXin Li                                        ")"));
3466*67e74705SXin Li     }
3467*67e74705SXin Li   }
3468*67e74705SXin Li }
3469*67e74705SXin Li 
3470*67e74705SXin Li template <typename T>
getObjCBridgeAttr(const TypedefType * TD)3471*67e74705SXin Li static inline T *getObjCBridgeAttr(const TypedefType *TD) {
3472*67e74705SXin Li   TypedefNameDecl *TDNDecl = TD->getDecl();
3473*67e74705SXin Li   QualType QT = TDNDecl->getUnderlyingType();
3474*67e74705SXin Li   if (QT->isPointerType()) {
3475*67e74705SXin Li     QT = QT->getPointeeType();
3476*67e74705SXin Li     if (const RecordType *RT = QT->getAs<RecordType>())
3477*67e74705SXin Li       if (RecordDecl *RD = RT->getDecl()->getMostRecentDecl())
3478*67e74705SXin Li         return RD->getAttr<T>();
3479*67e74705SXin Li   }
3480*67e74705SXin Li   return nullptr;
3481*67e74705SXin Li }
3482*67e74705SXin Li 
ObjCBridgeRelatedAttrFromType(QualType T,TypedefNameDecl * & TDNDecl)3483*67e74705SXin Li static ObjCBridgeRelatedAttr *ObjCBridgeRelatedAttrFromType(QualType T,
3484*67e74705SXin Li                                                             TypedefNameDecl *&TDNDecl) {
3485*67e74705SXin Li   while (const TypedefType *TD = dyn_cast<TypedefType>(T.getTypePtr())) {
3486*67e74705SXin Li     TDNDecl = TD->getDecl();
3487*67e74705SXin Li     if (ObjCBridgeRelatedAttr *ObjCBAttr =
3488*67e74705SXin Li         getObjCBridgeAttr<ObjCBridgeRelatedAttr>(TD))
3489*67e74705SXin Li       return ObjCBAttr;
3490*67e74705SXin Li     T = TDNDecl->getUnderlyingType();
3491*67e74705SXin Li   }
3492*67e74705SXin Li   return nullptr;
3493*67e74705SXin Li }
3494*67e74705SXin Li 
3495*67e74705SXin Li static void
diagnoseObjCARCConversion(Sema & S,SourceRange castRange,QualType castType,ARCConversionTypeClass castACTC,Expr * castExpr,Expr * realCast,ARCConversionTypeClass exprACTC,Sema::CheckedConversionKind CCK)3496*67e74705SXin Li diagnoseObjCARCConversion(Sema &S, SourceRange castRange,
3497*67e74705SXin Li                           QualType castType, ARCConversionTypeClass castACTC,
3498*67e74705SXin Li                           Expr *castExpr, Expr *realCast,
3499*67e74705SXin Li                           ARCConversionTypeClass exprACTC,
3500*67e74705SXin Li                           Sema::CheckedConversionKind CCK) {
3501*67e74705SXin Li   SourceLocation loc =
3502*67e74705SXin Li     (castRange.isValid() ? castRange.getBegin() : castExpr->getExprLoc());
3503*67e74705SXin Li 
3504*67e74705SXin Li   if (S.makeUnavailableInSystemHeader(loc,
3505*67e74705SXin Li                                  UnavailableAttr::IR_ARCForbiddenConversion))
3506*67e74705SXin Li     return;
3507*67e74705SXin Li 
3508*67e74705SXin Li   QualType castExprType = castExpr->getType();
3509*67e74705SXin Li   // Defer emitting a diagnostic for bridge-related casts; that will be
3510*67e74705SXin Li   // handled by CheckObjCBridgeRelatedConversions.
3511*67e74705SXin Li   TypedefNameDecl *TDNDecl = nullptr;
3512*67e74705SXin Li   if ((castACTC == ACTC_coreFoundation &&  exprACTC == ACTC_retainable &&
3513*67e74705SXin Li        ObjCBridgeRelatedAttrFromType(castType, TDNDecl)) ||
3514*67e74705SXin Li       (exprACTC == ACTC_coreFoundation && castACTC == ACTC_retainable &&
3515*67e74705SXin Li        ObjCBridgeRelatedAttrFromType(castExprType, TDNDecl)))
3516*67e74705SXin Li     return;
3517*67e74705SXin Li 
3518*67e74705SXin Li   unsigned srcKind = 0;
3519*67e74705SXin Li   switch (exprACTC) {
3520*67e74705SXin Li   case ACTC_none:
3521*67e74705SXin Li   case ACTC_coreFoundation:
3522*67e74705SXin Li   case ACTC_voidPtr:
3523*67e74705SXin Li     srcKind = (castExprType->isPointerType() ? 1 : 0);
3524*67e74705SXin Li     break;
3525*67e74705SXin Li   case ACTC_retainable:
3526*67e74705SXin Li     srcKind = (castExprType->isBlockPointerType() ? 2 : 3);
3527*67e74705SXin Li     break;
3528*67e74705SXin Li   case ACTC_indirectRetainable:
3529*67e74705SXin Li     srcKind = 4;
3530*67e74705SXin Li     break;
3531*67e74705SXin Li   }
3532*67e74705SXin Li 
3533*67e74705SXin Li   // Check whether this could be fixed with a bridge cast.
3534*67e74705SXin Li   SourceLocation afterLParen = S.getLocForEndOfToken(castRange.getBegin());
3535*67e74705SXin Li   SourceLocation noteLoc = afterLParen.isValid() ? afterLParen : loc;
3536*67e74705SXin Li 
3537*67e74705SXin Li   // Bridge from an ARC type to a CF type.
3538*67e74705SXin Li   if (castACTC == ACTC_retainable && isAnyRetainable(exprACTC)) {
3539*67e74705SXin Li 
3540*67e74705SXin Li     S.Diag(loc, diag::err_arc_cast_requires_bridge)
3541*67e74705SXin Li       << unsigned(CCK == Sema::CCK_ImplicitConversion) // cast|implicit
3542*67e74705SXin Li       << 2 // of C pointer type
3543*67e74705SXin Li       << castExprType
3544*67e74705SXin Li       << unsigned(castType->isBlockPointerType()) // to ObjC|block type
3545*67e74705SXin Li       << castType
3546*67e74705SXin Li       << castRange
3547*67e74705SXin Li       << castExpr->getSourceRange();
3548*67e74705SXin Li     bool br = S.isKnownName("CFBridgingRelease");
3549*67e74705SXin Li     ACCResult CreateRule =
3550*67e74705SXin Li       ARCCastChecker(S.Context, exprACTC, castACTC, true).Visit(castExpr);
3551*67e74705SXin Li     assert(CreateRule != ACC_bottom && "This cast should already be accepted.");
3552*67e74705SXin Li     if (CreateRule != ACC_plusOne)
3553*67e74705SXin Li     {
3554*67e74705SXin Li       DiagnosticBuilder DiagB =
3555*67e74705SXin Li         (CCK != Sema::CCK_OtherCast) ? S.Diag(noteLoc, diag::note_arc_bridge)
3556*67e74705SXin Li                               : S.Diag(noteLoc, diag::note_arc_cstyle_bridge);
3557*67e74705SXin Li 
3558*67e74705SXin Li       addFixitForObjCARCConversion(S, DiagB, CCK, afterLParen,
3559*67e74705SXin Li                                    castType, castExpr, realCast, "__bridge ",
3560*67e74705SXin Li                                    nullptr);
3561*67e74705SXin Li     }
3562*67e74705SXin Li     if (CreateRule != ACC_plusZero)
3563*67e74705SXin Li     {
3564*67e74705SXin Li       DiagnosticBuilder DiagB =
3565*67e74705SXin Li         (CCK == Sema::CCK_OtherCast && !br) ?
3566*67e74705SXin Li           S.Diag(noteLoc, diag::note_arc_cstyle_bridge_transfer) << castExprType :
3567*67e74705SXin Li           S.Diag(br ? castExpr->getExprLoc() : noteLoc,
3568*67e74705SXin Li                  diag::note_arc_bridge_transfer)
3569*67e74705SXin Li             << castExprType << br;
3570*67e74705SXin Li 
3571*67e74705SXin Li       addFixitForObjCARCConversion(S, DiagB, CCK, afterLParen,
3572*67e74705SXin Li                                    castType, castExpr, realCast, "__bridge_transfer ",
3573*67e74705SXin Li                                    br ? "CFBridgingRelease" : nullptr);
3574*67e74705SXin Li     }
3575*67e74705SXin Li 
3576*67e74705SXin Li     return;
3577*67e74705SXin Li   }
3578*67e74705SXin Li 
3579*67e74705SXin Li   // Bridge from a CF type to an ARC type.
3580*67e74705SXin Li   if (exprACTC == ACTC_retainable && isAnyRetainable(castACTC)) {
3581*67e74705SXin Li     bool br = S.isKnownName("CFBridgingRetain");
3582*67e74705SXin Li     S.Diag(loc, diag::err_arc_cast_requires_bridge)
3583*67e74705SXin Li       << unsigned(CCK == Sema::CCK_ImplicitConversion) // cast|implicit
3584*67e74705SXin Li       << unsigned(castExprType->isBlockPointerType()) // of ObjC|block type
3585*67e74705SXin Li       << castExprType
3586*67e74705SXin Li       << 2 // to C pointer type
3587*67e74705SXin Li       << castType
3588*67e74705SXin Li       << castRange
3589*67e74705SXin Li       << castExpr->getSourceRange();
3590*67e74705SXin Li     ACCResult CreateRule =
3591*67e74705SXin Li       ARCCastChecker(S.Context, exprACTC, castACTC, true).Visit(castExpr);
3592*67e74705SXin Li     assert(CreateRule != ACC_bottom && "This cast should already be accepted.");
3593*67e74705SXin Li     if (CreateRule != ACC_plusOne)
3594*67e74705SXin Li     {
3595*67e74705SXin Li       DiagnosticBuilder DiagB =
3596*67e74705SXin Li       (CCK != Sema::CCK_OtherCast) ? S.Diag(noteLoc, diag::note_arc_bridge)
3597*67e74705SXin Li                                : S.Diag(noteLoc, diag::note_arc_cstyle_bridge);
3598*67e74705SXin Li       addFixitForObjCARCConversion(S, DiagB, CCK, afterLParen,
3599*67e74705SXin Li                                    castType, castExpr, realCast, "__bridge ",
3600*67e74705SXin Li                                    nullptr);
3601*67e74705SXin Li     }
3602*67e74705SXin Li     if (CreateRule != ACC_plusZero)
3603*67e74705SXin Li     {
3604*67e74705SXin Li       DiagnosticBuilder DiagB =
3605*67e74705SXin Li         (CCK == Sema::CCK_OtherCast && !br) ?
3606*67e74705SXin Li           S.Diag(noteLoc, diag::note_arc_cstyle_bridge_retained) << castType :
3607*67e74705SXin Li           S.Diag(br ? castExpr->getExprLoc() : noteLoc,
3608*67e74705SXin Li                  diag::note_arc_bridge_retained)
3609*67e74705SXin Li             << castType << br;
3610*67e74705SXin Li 
3611*67e74705SXin Li       addFixitForObjCARCConversion(S, DiagB, CCK, afterLParen,
3612*67e74705SXin Li                                    castType, castExpr, realCast, "__bridge_retained ",
3613*67e74705SXin Li                                    br ? "CFBridgingRetain" : nullptr);
3614*67e74705SXin Li     }
3615*67e74705SXin Li 
3616*67e74705SXin Li     return;
3617*67e74705SXin Li   }
3618*67e74705SXin Li 
3619*67e74705SXin Li   S.Diag(loc, diag::err_arc_mismatched_cast)
3620*67e74705SXin Li     << (CCK != Sema::CCK_ImplicitConversion)
3621*67e74705SXin Li     << srcKind << castExprType << castType
3622*67e74705SXin Li     << castRange << castExpr->getSourceRange();
3623*67e74705SXin Li }
3624*67e74705SXin Li 
3625*67e74705SXin Li template <typename TB>
CheckObjCBridgeNSCast(Sema & S,QualType castType,Expr * castExpr,bool & HadTheAttribute,bool warn)3626*67e74705SXin Li static bool CheckObjCBridgeNSCast(Sema &S, QualType castType, Expr *castExpr,
3627*67e74705SXin Li                                   bool &HadTheAttribute, bool warn) {
3628*67e74705SXin Li   QualType T = castExpr->getType();
3629*67e74705SXin Li   HadTheAttribute = false;
3630*67e74705SXin Li   while (const TypedefType *TD = dyn_cast<TypedefType>(T.getTypePtr())) {
3631*67e74705SXin Li     TypedefNameDecl *TDNDecl = TD->getDecl();
3632*67e74705SXin Li     if (TB *ObjCBAttr = getObjCBridgeAttr<TB>(TD)) {
3633*67e74705SXin Li       if (IdentifierInfo *Parm = ObjCBAttr->getBridgedType()) {
3634*67e74705SXin Li         HadTheAttribute = true;
3635*67e74705SXin Li         if (Parm->isStr("id"))
3636*67e74705SXin Li           return true;
3637*67e74705SXin Li 
3638*67e74705SXin Li         NamedDecl *Target = nullptr;
3639*67e74705SXin Li         // Check for an existing type with this name.
3640*67e74705SXin Li         LookupResult R(S, DeclarationName(Parm), SourceLocation(),
3641*67e74705SXin Li                        Sema::LookupOrdinaryName);
3642*67e74705SXin Li         if (S.LookupName(R, S.TUScope)) {
3643*67e74705SXin Li           Target = R.getFoundDecl();
3644*67e74705SXin Li           if (Target && isa<ObjCInterfaceDecl>(Target)) {
3645*67e74705SXin Li             ObjCInterfaceDecl *ExprClass = cast<ObjCInterfaceDecl>(Target);
3646*67e74705SXin Li             if (const ObjCObjectPointerType *InterfacePointerType =
3647*67e74705SXin Li                   castType->getAsObjCInterfacePointerType()) {
3648*67e74705SXin Li               ObjCInterfaceDecl *CastClass
3649*67e74705SXin Li                 = InterfacePointerType->getObjectType()->getInterface();
3650*67e74705SXin Li               if ((CastClass == ExprClass) ||
3651*67e74705SXin Li                   (CastClass && CastClass->isSuperClassOf(ExprClass)))
3652*67e74705SXin Li                 return true;
3653*67e74705SXin Li               if (warn)
3654*67e74705SXin Li                 S.Diag(castExpr->getLocStart(), diag::warn_objc_invalid_bridge)
3655*67e74705SXin Li                   << T << Target->getName() << castType->getPointeeType();
3656*67e74705SXin Li               return false;
3657*67e74705SXin Li             } else if (castType->isObjCIdType() ||
3658*67e74705SXin Li                        (S.Context.ObjCObjectAdoptsQTypeProtocols(
3659*67e74705SXin Li                           castType, ExprClass)))
3660*67e74705SXin Li               // ok to cast to 'id'.
3661*67e74705SXin Li               // casting to id<p-list> is ok if bridge type adopts all of
3662*67e74705SXin Li               // p-list protocols.
3663*67e74705SXin Li               return true;
3664*67e74705SXin Li             else {
3665*67e74705SXin Li               if (warn) {
3666*67e74705SXin Li                 S.Diag(castExpr->getLocStart(), diag::warn_objc_invalid_bridge)
3667*67e74705SXin Li                   << T << Target->getName() << castType;
3668*67e74705SXin Li                 S.Diag(TDNDecl->getLocStart(), diag::note_declared_at);
3669*67e74705SXin Li                 S.Diag(Target->getLocStart(), diag::note_declared_at);
3670*67e74705SXin Li               }
3671*67e74705SXin Li               return false;
3672*67e74705SXin Li            }
3673*67e74705SXin Li           }
3674*67e74705SXin Li         } else if (!castType->isObjCIdType()) {
3675*67e74705SXin Li           S.Diag(castExpr->getLocStart(), diag::err_objc_cf_bridged_not_interface)
3676*67e74705SXin Li             << castExpr->getType() << Parm;
3677*67e74705SXin Li           S.Diag(TDNDecl->getLocStart(), diag::note_declared_at);
3678*67e74705SXin Li           if (Target)
3679*67e74705SXin Li             S.Diag(Target->getLocStart(), diag::note_declared_at);
3680*67e74705SXin Li         }
3681*67e74705SXin Li         return true;
3682*67e74705SXin Li       }
3683*67e74705SXin Li       return false;
3684*67e74705SXin Li     }
3685*67e74705SXin Li     T = TDNDecl->getUnderlyingType();
3686*67e74705SXin Li   }
3687*67e74705SXin Li   return true;
3688*67e74705SXin Li }
3689*67e74705SXin Li 
3690*67e74705SXin Li template <typename TB>
CheckObjCBridgeCFCast(Sema & S,QualType castType,Expr * castExpr,bool & HadTheAttribute,bool warn)3691*67e74705SXin Li static bool CheckObjCBridgeCFCast(Sema &S, QualType castType, Expr *castExpr,
3692*67e74705SXin Li                                   bool &HadTheAttribute, bool warn) {
3693*67e74705SXin Li   QualType T = castType;
3694*67e74705SXin Li   HadTheAttribute = false;
3695*67e74705SXin Li   while (const TypedefType *TD = dyn_cast<TypedefType>(T.getTypePtr())) {
3696*67e74705SXin Li     TypedefNameDecl *TDNDecl = TD->getDecl();
3697*67e74705SXin Li     if (TB *ObjCBAttr = getObjCBridgeAttr<TB>(TD)) {
3698*67e74705SXin Li       if (IdentifierInfo *Parm = ObjCBAttr->getBridgedType()) {
3699*67e74705SXin Li         HadTheAttribute = true;
3700*67e74705SXin Li         if (Parm->isStr("id"))
3701*67e74705SXin Li           return true;
3702*67e74705SXin Li 
3703*67e74705SXin Li         NamedDecl *Target = nullptr;
3704*67e74705SXin Li         // Check for an existing type with this name.
3705*67e74705SXin Li         LookupResult R(S, DeclarationName(Parm), SourceLocation(),
3706*67e74705SXin Li                        Sema::LookupOrdinaryName);
3707*67e74705SXin Li         if (S.LookupName(R, S.TUScope)) {
3708*67e74705SXin Li           Target = R.getFoundDecl();
3709*67e74705SXin Li           if (Target && isa<ObjCInterfaceDecl>(Target)) {
3710*67e74705SXin Li             ObjCInterfaceDecl *CastClass = cast<ObjCInterfaceDecl>(Target);
3711*67e74705SXin Li             if (const ObjCObjectPointerType *InterfacePointerType =
3712*67e74705SXin Li                   castExpr->getType()->getAsObjCInterfacePointerType()) {
3713*67e74705SXin Li               ObjCInterfaceDecl *ExprClass
3714*67e74705SXin Li                 = InterfacePointerType->getObjectType()->getInterface();
3715*67e74705SXin Li               if ((CastClass == ExprClass) ||
3716*67e74705SXin Li                   (ExprClass && CastClass->isSuperClassOf(ExprClass)))
3717*67e74705SXin Li                 return true;
3718*67e74705SXin Li               if (warn) {
3719*67e74705SXin Li                 S.Diag(castExpr->getLocStart(), diag::warn_objc_invalid_bridge_to_cf)
3720*67e74705SXin Li                   << castExpr->getType()->getPointeeType() << T;
3721*67e74705SXin Li                 S.Diag(TDNDecl->getLocStart(), diag::note_declared_at);
3722*67e74705SXin Li               }
3723*67e74705SXin Li               return false;
3724*67e74705SXin Li             } else if (castExpr->getType()->isObjCIdType() ||
3725*67e74705SXin Li                        (S.Context.QIdProtocolsAdoptObjCObjectProtocols(
3726*67e74705SXin Li                           castExpr->getType(), CastClass)))
3727*67e74705SXin Li               // ok to cast an 'id' expression to a CFtype.
3728*67e74705SXin Li               // ok to cast an 'id<plist>' expression to CFtype provided plist
3729*67e74705SXin Li               // adopts all of CFtype's ObjetiveC's class plist.
3730*67e74705SXin Li               return true;
3731*67e74705SXin Li             else {
3732*67e74705SXin Li               if (warn) {
3733*67e74705SXin Li                 S.Diag(castExpr->getLocStart(), diag::warn_objc_invalid_bridge_to_cf)
3734*67e74705SXin Li                   << castExpr->getType() << castType;
3735*67e74705SXin Li                 S.Diag(TDNDecl->getLocStart(), diag::note_declared_at);
3736*67e74705SXin Li                 S.Diag(Target->getLocStart(), diag::note_declared_at);
3737*67e74705SXin Li               }
3738*67e74705SXin Li               return false;
3739*67e74705SXin Li             }
3740*67e74705SXin Li           }
3741*67e74705SXin Li         }
3742*67e74705SXin Li         S.Diag(castExpr->getLocStart(), diag::err_objc_ns_bridged_invalid_cfobject)
3743*67e74705SXin Li         << castExpr->getType() << castType;
3744*67e74705SXin Li         S.Diag(TDNDecl->getLocStart(), diag::note_declared_at);
3745*67e74705SXin Li         if (Target)
3746*67e74705SXin Li           S.Diag(Target->getLocStart(), diag::note_declared_at);
3747*67e74705SXin Li         return true;
3748*67e74705SXin Li       }
3749*67e74705SXin Li       return false;
3750*67e74705SXin Li     }
3751*67e74705SXin Li     T = TDNDecl->getUnderlyingType();
3752*67e74705SXin Li   }
3753*67e74705SXin Li   return true;
3754*67e74705SXin Li }
3755*67e74705SXin Li 
CheckTollFreeBridgeCast(QualType castType,Expr * castExpr)3756*67e74705SXin Li void Sema::CheckTollFreeBridgeCast(QualType castType, Expr *castExpr) {
3757*67e74705SXin Li   if (!getLangOpts().ObjC1)
3758*67e74705SXin Li     return;
3759*67e74705SXin Li   // warn in presence of __bridge casting to or from a toll free bridge cast.
3760*67e74705SXin Li   ARCConversionTypeClass exprACTC = classifyTypeForARCConversion(castExpr->getType());
3761*67e74705SXin Li   ARCConversionTypeClass castACTC = classifyTypeForARCConversion(castType);
3762*67e74705SXin Li   if (castACTC == ACTC_retainable && exprACTC == ACTC_coreFoundation) {
3763*67e74705SXin Li     bool HasObjCBridgeAttr;
3764*67e74705SXin Li     bool ObjCBridgeAttrWillNotWarn =
3765*67e74705SXin Li       CheckObjCBridgeNSCast<ObjCBridgeAttr>(*this, castType, castExpr, HasObjCBridgeAttr,
3766*67e74705SXin Li                                             false);
3767*67e74705SXin Li     if (ObjCBridgeAttrWillNotWarn && HasObjCBridgeAttr)
3768*67e74705SXin Li       return;
3769*67e74705SXin Li     bool HasObjCBridgeMutableAttr;
3770*67e74705SXin Li     bool ObjCBridgeMutableAttrWillNotWarn =
3771*67e74705SXin Li       CheckObjCBridgeNSCast<ObjCBridgeMutableAttr>(*this, castType, castExpr,
3772*67e74705SXin Li                                                    HasObjCBridgeMutableAttr, false);
3773*67e74705SXin Li     if (ObjCBridgeMutableAttrWillNotWarn && HasObjCBridgeMutableAttr)
3774*67e74705SXin Li       return;
3775*67e74705SXin Li 
3776*67e74705SXin Li     if (HasObjCBridgeAttr)
3777*67e74705SXin Li       CheckObjCBridgeNSCast<ObjCBridgeAttr>(*this, castType, castExpr, HasObjCBridgeAttr,
3778*67e74705SXin Li                                             true);
3779*67e74705SXin Li     else if (HasObjCBridgeMutableAttr)
3780*67e74705SXin Li       CheckObjCBridgeNSCast<ObjCBridgeMutableAttr>(*this, castType, castExpr,
3781*67e74705SXin Li                                                    HasObjCBridgeMutableAttr, true);
3782*67e74705SXin Li   }
3783*67e74705SXin Li   else if (castACTC == ACTC_coreFoundation && exprACTC == ACTC_retainable) {
3784*67e74705SXin Li     bool HasObjCBridgeAttr;
3785*67e74705SXin Li     bool ObjCBridgeAttrWillNotWarn =
3786*67e74705SXin Li       CheckObjCBridgeCFCast<ObjCBridgeAttr>(*this, castType, castExpr, HasObjCBridgeAttr,
3787*67e74705SXin Li                                             false);
3788*67e74705SXin Li     if (ObjCBridgeAttrWillNotWarn && HasObjCBridgeAttr)
3789*67e74705SXin Li       return;
3790*67e74705SXin Li     bool HasObjCBridgeMutableAttr;
3791*67e74705SXin Li     bool ObjCBridgeMutableAttrWillNotWarn =
3792*67e74705SXin Li       CheckObjCBridgeCFCast<ObjCBridgeMutableAttr>(*this, castType, castExpr,
3793*67e74705SXin Li                                                    HasObjCBridgeMutableAttr, false);
3794*67e74705SXin Li     if (ObjCBridgeMutableAttrWillNotWarn && HasObjCBridgeMutableAttr)
3795*67e74705SXin Li       return;
3796*67e74705SXin Li 
3797*67e74705SXin Li     if (HasObjCBridgeAttr)
3798*67e74705SXin Li       CheckObjCBridgeCFCast<ObjCBridgeAttr>(*this, castType, castExpr, HasObjCBridgeAttr,
3799*67e74705SXin Li                                             true);
3800*67e74705SXin Li     else if (HasObjCBridgeMutableAttr)
3801*67e74705SXin Li       CheckObjCBridgeCFCast<ObjCBridgeMutableAttr>(*this, castType, castExpr,
3802*67e74705SXin Li                                                    HasObjCBridgeMutableAttr, true);
3803*67e74705SXin Li   }
3804*67e74705SXin Li }
3805*67e74705SXin Li 
CheckObjCBridgeRelatedCast(QualType castType,Expr * castExpr)3806*67e74705SXin Li void Sema::CheckObjCBridgeRelatedCast(QualType castType, Expr *castExpr) {
3807*67e74705SXin Li   QualType SrcType = castExpr->getType();
3808*67e74705SXin Li   if (ObjCPropertyRefExpr *PRE = dyn_cast<ObjCPropertyRefExpr>(castExpr)) {
3809*67e74705SXin Li     if (PRE->isExplicitProperty()) {
3810*67e74705SXin Li       if (ObjCPropertyDecl *PDecl = PRE->getExplicitProperty())
3811*67e74705SXin Li         SrcType = PDecl->getType();
3812*67e74705SXin Li     }
3813*67e74705SXin Li     else if (PRE->isImplicitProperty()) {
3814*67e74705SXin Li       if (ObjCMethodDecl *Getter = PRE->getImplicitPropertyGetter())
3815*67e74705SXin Li         SrcType = Getter->getReturnType();
3816*67e74705SXin Li     }
3817*67e74705SXin Li   }
3818*67e74705SXin Li 
3819*67e74705SXin Li   ARCConversionTypeClass srcExprACTC = classifyTypeForARCConversion(SrcType);
3820*67e74705SXin Li   ARCConversionTypeClass castExprACTC = classifyTypeForARCConversion(castType);
3821*67e74705SXin Li   if (srcExprACTC != ACTC_retainable || castExprACTC != ACTC_coreFoundation)
3822*67e74705SXin Li     return;
3823*67e74705SXin Li   CheckObjCBridgeRelatedConversions(castExpr->getLocStart(),
3824*67e74705SXin Li                                     castType, SrcType, castExpr);
3825*67e74705SXin Li }
3826*67e74705SXin Li 
CheckTollFreeBridgeStaticCast(QualType castType,Expr * castExpr,CastKind & Kind)3827*67e74705SXin Li bool Sema::CheckTollFreeBridgeStaticCast(QualType castType, Expr *castExpr,
3828*67e74705SXin Li                                          CastKind &Kind) {
3829*67e74705SXin Li   if (!getLangOpts().ObjC1)
3830*67e74705SXin Li     return false;
3831*67e74705SXin Li   ARCConversionTypeClass exprACTC =
3832*67e74705SXin Li     classifyTypeForARCConversion(castExpr->getType());
3833*67e74705SXin Li   ARCConversionTypeClass castACTC = classifyTypeForARCConversion(castType);
3834*67e74705SXin Li   if ((castACTC == ACTC_retainable && exprACTC == ACTC_coreFoundation) ||
3835*67e74705SXin Li       (castACTC == ACTC_coreFoundation && exprACTC == ACTC_retainable)) {
3836*67e74705SXin Li     CheckTollFreeBridgeCast(castType, castExpr);
3837*67e74705SXin Li     Kind = (castACTC == ACTC_coreFoundation) ? CK_BitCast
3838*67e74705SXin Li                                              : CK_CPointerToObjCPointerCast;
3839*67e74705SXin Li     return true;
3840*67e74705SXin Li   }
3841*67e74705SXin Li   return false;
3842*67e74705SXin Li }
3843*67e74705SXin Li 
checkObjCBridgeRelatedComponents(SourceLocation Loc,QualType DestType,QualType SrcType,ObjCInterfaceDecl * & RelatedClass,ObjCMethodDecl * & ClassMethod,ObjCMethodDecl * & InstanceMethod,TypedefNameDecl * & TDNDecl,bool CfToNs,bool Diagnose)3844*67e74705SXin Li bool Sema::checkObjCBridgeRelatedComponents(SourceLocation Loc,
3845*67e74705SXin Li                                             QualType DestType, QualType SrcType,
3846*67e74705SXin Li                                             ObjCInterfaceDecl *&RelatedClass,
3847*67e74705SXin Li                                             ObjCMethodDecl *&ClassMethod,
3848*67e74705SXin Li                                             ObjCMethodDecl *&InstanceMethod,
3849*67e74705SXin Li                                             TypedefNameDecl *&TDNDecl,
3850*67e74705SXin Li                                             bool CfToNs, bool Diagnose) {
3851*67e74705SXin Li   QualType T = CfToNs ? SrcType : DestType;
3852*67e74705SXin Li   ObjCBridgeRelatedAttr *ObjCBAttr = ObjCBridgeRelatedAttrFromType(T, TDNDecl);
3853*67e74705SXin Li   if (!ObjCBAttr)
3854*67e74705SXin Li     return false;
3855*67e74705SXin Li 
3856*67e74705SXin Li   IdentifierInfo *RCId = ObjCBAttr->getRelatedClass();
3857*67e74705SXin Li   IdentifierInfo *CMId = ObjCBAttr->getClassMethod();
3858*67e74705SXin Li   IdentifierInfo *IMId = ObjCBAttr->getInstanceMethod();
3859*67e74705SXin Li   if (!RCId)
3860*67e74705SXin Li     return false;
3861*67e74705SXin Li   NamedDecl *Target = nullptr;
3862*67e74705SXin Li   // Check for an existing type with this name.
3863*67e74705SXin Li   LookupResult R(*this, DeclarationName(RCId), SourceLocation(),
3864*67e74705SXin Li                  Sema::LookupOrdinaryName);
3865*67e74705SXin Li   if (!LookupName(R, TUScope)) {
3866*67e74705SXin Li     if (Diagnose) {
3867*67e74705SXin Li       Diag(Loc, diag::err_objc_bridged_related_invalid_class) << RCId
3868*67e74705SXin Li             << SrcType << DestType;
3869*67e74705SXin Li       Diag(TDNDecl->getLocStart(), diag::note_declared_at);
3870*67e74705SXin Li     }
3871*67e74705SXin Li     return false;
3872*67e74705SXin Li   }
3873*67e74705SXin Li   Target = R.getFoundDecl();
3874*67e74705SXin Li   if (Target && isa<ObjCInterfaceDecl>(Target))
3875*67e74705SXin Li     RelatedClass = cast<ObjCInterfaceDecl>(Target);
3876*67e74705SXin Li   else {
3877*67e74705SXin Li     if (Diagnose) {
3878*67e74705SXin Li       Diag(Loc, diag::err_objc_bridged_related_invalid_class_name) << RCId
3879*67e74705SXin Li             << SrcType << DestType;
3880*67e74705SXin Li       Diag(TDNDecl->getLocStart(), diag::note_declared_at);
3881*67e74705SXin Li       if (Target)
3882*67e74705SXin Li         Diag(Target->getLocStart(), diag::note_declared_at);
3883*67e74705SXin Li     }
3884*67e74705SXin Li     return false;
3885*67e74705SXin Li   }
3886*67e74705SXin Li 
3887*67e74705SXin Li   // Check for an existing class method with the given selector name.
3888*67e74705SXin Li   if (CfToNs && CMId) {
3889*67e74705SXin Li     Selector Sel = Context.Selectors.getUnarySelector(CMId);
3890*67e74705SXin Li     ClassMethod = RelatedClass->lookupMethod(Sel, false);
3891*67e74705SXin Li     if (!ClassMethod) {
3892*67e74705SXin Li       if (Diagnose) {
3893*67e74705SXin Li         Diag(Loc, diag::err_objc_bridged_related_known_method)
3894*67e74705SXin Li               << SrcType << DestType << Sel << false;
3895*67e74705SXin Li         Diag(TDNDecl->getLocStart(), diag::note_declared_at);
3896*67e74705SXin Li       }
3897*67e74705SXin Li       return false;
3898*67e74705SXin Li     }
3899*67e74705SXin Li   }
3900*67e74705SXin Li 
3901*67e74705SXin Li   // Check for an existing instance method with the given selector name.
3902*67e74705SXin Li   if (!CfToNs && IMId) {
3903*67e74705SXin Li     Selector Sel = Context.Selectors.getNullarySelector(IMId);
3904*67e74705SXin Li     InstanceMethod = RelatedClass->lookupMethod(Sel, true);
3905*67e74705SXin Li     if (!InstanceMethod) {
3906*67e74705SXin Li       if (Diagnose) {
3907*67e74705SXin Li         Diag(Loc, diag::err_objc_bridged_related_known_method)
3908*67e74705SXin Li               << SrcType << DestType << Sel << true;
3909*67e74705SXin Li         Diag(TDNDecl->getLocStart(), diag::note_declared_at);
3910*67e74705SXin Li       }
3911*67e74705SXin Li       return false;
3912*67e74705SXin Li     }
3913*67e74705SXin Li   }
3914*67e74705SXin Li   return true;
3915*67e74705SXin Li }
3916*67e74705SXin Li 
3917*67e74705SXin Li bool
CheckObjCBridgeRelatedConversions(SourceLocation Loc,QualType DestType,QualType SrcType,Expr * & SrcExpr,bool Diagnose)3918*67e74705SXin Li Sema::CheckObjCBridgeRelatedConversions(SourceLocation Loc,
3919*67e74705SXin Li                                         QualType DestType, QualType SrcType,
3920*67e74705SXin Li                                         Expr *&SrcExpr, bool Diagnose) {
3921*67e74705SXin Li   ARCConversionTypeClass rhsExprACTC = classifyTypeForARCConversion(SrcType);
3922*67e74705SXin Li   ARCConversionTypeClass lhsExprACTC = classifyTypeForARCConversion(DestType);
3923*67e74705SXin Li   bool CfToNs = (rhsExprACTC == ACTC_coreFoundation && lhsExprACTC == ACTC_retainable);
3924*67e74705SXin Li   bool NsToCf = (rhsExprACTC == ACTC_retainable && lhsExprACTC == ACTC_coreFoundation);
3925*67e74705SXin Li   if (!CfToNs && !NsToCf)
3926*67e74705SXin Li     return false;
3927*67e74705SXin Li 
3928*67e74705SXin Li   ObjCInterfaceDecl *RelatedClass;
3929*67e74705SXin Li   ObjCMethodDecl *ClassMethod = nullptr;
3930*67e74705SXin Li   ObjCMethodDecl *InstanceMethod = nullptr;
3931*67e74705SXin Li   TypedefNameDecl *TDNDecl = nullptr;
3932*67e74705SXin Li   if (!checkObjCBridgeRelatedComponents(Loc, DestType, SrcType, RelatedClass,
3933*67e74705SXin Li                                         ClassMethod, InstanceMethod, TDNDecl,
3934*67e74705SXin Li                                         CfToNs, Diagnose))
3935*67e74705SXin Li     return false;
3936*67e74705SXin Li 
3937*67e74705SXin Li   if (CfToNs) {
3938*67e74705SXin Li     // Implicit conversion from CF to ObjC object is needed.
3939*67e74705SXin Li     if (ClassMethod) {
3940*67e74705SXin Li       if (Diagnose) {
3941*67e74705SXin Li         std::string ExpressionString = "[";
3942*67e74705SXin Li         ExpressionString += RelatedClass->getNameAsString();
3943*67e74705SXin Li         ExpressionString += " ";
3944*67e74705SXin Li         ExpressionString += ClassMethod->getSelector().getAsString();
3945*67e74705SXin Li         SourceLocation SrcExprEndLoc = getLocForEndOfToken(SrcExpr->getLocEnd());
3946*67e74705SXin Li         // Provide a fixit: [RelatedClass ClassMethod SrcExpr]
3947*67e74705SXin Li         Diag(Loc, diag::err_objc_bridged_related_known_method)
3948*67e74705SXin Li           << SrcType << DestType << ClassMethod->getSelector() << false
3949*67e74705SXin Li           << FixItHint::CreateInsertion(SrcExpr->getLocStart(), ExpressionString)
3950*67e74705SXin Li           << FixItHint::CreateInsertion(SrcExprEndLoc, "]");
3951*67e74705SXin Li         Diag(RelatedClass->getLocStart(), diag::note_declared_at);
3952*67e74705SXin Li         Diag(TDNDecl->getLocStart(), diag::note_declared_at);
3953*67e74705SXin Li 
3954*67e74705SXin Li         QualType receiverType = Context.getObjCInterfaceType(RelatedClass);
3955*67e74705SXin Li         // Argument.
3956*67e74705SXin Li         Expr *args[] = { SrcExpr };
3957*67e74705SXin Li         ExprResult msg = BuildClassMessageImplicit(receiverType, false,
3958*67e74705SXin Li                                       ClassMethod->getLocation(),
3959*67e74705SXin Li                                       ClassMethod->getSelector(), ClassMethod,
3960*67e74705SXin Li                                       MultiExprArg(args, 1));
3961*67e74705SXin Li         SrcExpr = msg.get();
3962*67e74705SXin Li       }
3963*67e74705SXin Li       return true;
3964*67e74705SXin Li     }
3965*67e74705SXin Li   }
3966*67e74705SXin Li   else {
3967*67e74705SXin Li     // Implicit conversion from ObjC type to CF object is needed.
3968*67e74705SXin Li     if (InstanceMethod) {
3969*67e74705SXin Li       if (Diagnose) {
3970*67e74705SXin Li         std::string ExpressionString;
3971*67e74705SXin Li         SourceLocation SrcExprEndLoc =
3972*67e74705SXin Li             getLocForEndOfToken(SrcExpr->getLocEnd());
3973*67e74705SXin Li         if (InstanceMethod->isPropertyAccessor())
3974*67e74705SXin Li           if (const ObjCPropertyDecl *PDecl =
3975*67e74705SXin Li                   InstanceMethod->findPropertyDecl()) {
3976*67e74705SXin Li             // fixit: ObjectExpr.propertyname when it is  aproperty accessor.
3977*67e74705SXin Li             ExpressionString = ".";
3978*67e74705SXin Li             ExpressionString += PDecl->getNameAsString();
3979*67e74705SXin Li             Diag(Loc, diag::err_objc_bridged_related_known_method)
3980*67e74705SXin Li                 << SrcType << DestType << InstanceMethod->getSelector() << true
3981*67e74705SXin Li                 << FixItHint::CreateInsertion(SrcExprEndLoc, ExpressionString);
3982*67e74705SXin Li           }
3983*67e74705SXin Li         if (ExpressionString.empty()) {
3984*67e74705SXin Li           // Provide a fixit: [ObjectExpr InstanceMethod]
3985*67e74705SXin Li           ExpressionString = " ";
3986*67e74705SXin Li           ExpressionString += InstanceMethod->getSelector().getAsString();
3987*67e74705SXin Li           ExpressionString += "]";
3988*67e74705SXin Li 
3989*67e74705SXin Li           Diag(Loc, diag::err_objc_bridged_related_known_method)
3990*67e74705SXin Li               << SrcType << DestType << InstanceMethod->getSelector() << true
3991*67e74705SXin Li               << FixItHint::CreateInsertion(SrcExpr->getLocStart(), "[")
3992*67e74705SXin Li               << FixItHint::CreateInsertion(SrcExprEndLoc, ExpressionString);
3993*67e74705SXin Li         }
3994*67e74705SXin Li         Diag(RelatedClass->getLocStart(), diag::note_declared_at);
3995*67e74705SXin Li         Diag(TDNDecl->getLocStart(), diag::note_declared_at);
3996*67e74705SXin Li 
3997*67e74705SXin Li         ExprResult msg =
3998*67e74705SXin Li           BuildInstanceMessageImplicit(SrcExpr, SrcType,
3999*67e74705SXin Li                                        InstanceMethod->getLocation(),
4000*67e74705SXin Li                                        InstanceMethod->getSelector(),
4001*67e74705SXin Li                                        InstanceMethod, None);
4002*67e74705SXin Li         SrcExpr = msg.get();
4003*67e74705SXin Li       }
4004*67e74705SXin Li       return true;
4005*67e74705SXin Li     }
4006*67e74705SXin Li   }
4007*67e74705SXin Li   return false;
4008*67e74705SXin Li }
4009*67e74705SXin Li 
4010*67e74705SXin Li Sema::ARCConversionResult
CheckObjCARCConversion(SourceRange castRange,QualType castType,Expr * & castExpr,CheckedConversionKind CCK,bool Diagnose,bool DiagnoseCFAudited,BinaryOperatorKind Opc)4011*67e74705SXin Li Sema::CheckObjCARCConversion(SourceRange castRange, QualType castType,
4012*67e74705SXin Li                              Expr *&castExpr, CheckedConversionKind CCK,
4013*67e74705SXin Li                              bool Diagnose,
4014*67e74705SXin Li                              bool DiagnoseCFAudited,
4015*67e74705SXin Li                              BinaryOperatorKind Opc) {
4016*67e74705SXin Li   QualType castExprType = castExpr->getType();
4017*67e74705SXin Li 
4018*67e74705SXin Li   // For the purposes of the classification, we assume reference types
4019*67e74705SXin Li   // will bind to temporaries.
4020*67e74705SXin Li   QualType effCastType = castType;
4021*67e74705SXin Li   if (const ReferenceType *ref = castType->getAs<ReferenceType>())
4022*67e74705SXin Li     effCastType = ref->getPointeeType();
4023*67e74705SXin Li 
4024*67e74705SXin Li   ARCConversionTypeClass exprACTC = classifyTypeForARCConversion(castExprType);
4025*67e74705SXin Li   ARCConversionTypeClass castACTC = classifyTypeForARCConversion(effCastType);
4026*67e74705SXin Li   if (exprACTC == castACTC) {
4027*67e74705SXin Li     // Check for viability and report error if casting an rvalue to a
4028*67e74705SXin Li     // life-time qualifier.
4029*67e74705SXin Li     if (castACTC == ACTC_retainable &&
4030*67e74705SXin Li         (CCK == CCK_CStyleCast || CCK == CCK_OtherCast) &&
4031*67e74705SXin Li         castType != castExprType) {
4032*67e74705SXin Li       const Type *DT = castType.getTypePtr();
4033*67e74705SXin Li       QualType QDT = castType;
4034*67e74705SXin Li       // We desugar some types but not others. We ignore those
4035*67e74705SXin Li       // that cannot happen in a cast; i.e. auto, and those which
4036*67e74705SXin Li       // should not be de-sugared; i.e typedef.
4037*67e74705SXin Li       if (const ParenType *PT = dyn_cast<ParenType>(DT))
4038*67e74705SXin Li         QDT = PT->desugar();
4039*67e74705SXin Li       else if (const TypeOfType *TP = dyn_cast<TypeOfType>(DT))
4040*67e74705SXin Li         QDT = TP->desugar();
4041*67e74705SXin Li       else if (const AttributedType *AT = dyn_cast<AttributedType>(DT))
4042*67e74705SXin Li         QDT = AT->desugar();
4043*67e74705SXin Li       if (QDT != castType &&
4044*67e74705SXin Li           QDT.getObjCLifetime() !=  Qualifiers::OCL_None) {
4045*67e74705SXin Li         if (Diagnose) {
4046*67e74705SXin Li           SourceLocation loc = (castRange.isValid() ? castRange.getBegin()
4047*67e74705SXin Li                                                     : castExpr->getExprLoc());
4048*67e74705SXin Li           Diag(loc, diag::err_arc_nolifetime_behavior);
4049*67e74705SXin Li         }
4050*67e74705SXin Li         return ACR_error;
4051*67e74705SXin Li       }
4052*67e74705SXin Li     }
4053*67e74705SXin Li     return ACR_okay;
4054*67e74705SXin Li   }
4055*67e74705SXin Li 
4056*67e74705SXin Li   if (isAnyCLike(exprACTC) && isAnyCLike(castACTC)) return ACR_okay;
4057*67e74705SXin Li 
4058*67e74705SXin Li   // Allow all of these types to be cast to integer types (but not
4059*67e74705SXin Li   // vice-versa).
4060*67e74705SXin Li   if (castACTC == ACTC_none && castType->isIntegralType(Context))
4061*67e74705SXin Li     return ACR_okay;
4062*67e74705SXin Li 
4063*67e74705SXin Li   // Allow casts between pointers to lifetime types (e.g., __strong id*)
4064*67e74705SXin Li   // and pointers to void (e.g., cv void *). Casting from void* to lifetime*
4065*67e74705SXin Li   // must be explicit.
4066*67e74705SXin Li   if (exprACTC == ACTC_indirectRetainable && castACTC == ACTC_voidPtr)
4067*67e74705SXin Li     return ACR_okay;
4068*67e74705SXin Li   if (castACTC == ACTC_indirectRetainable && exprACTC == ACTC_voidPtr &&
4069*67e74705SXin Li       CCK != CCK_ImplicitConversion)
4070*67e74705SXin Li     return ACR_okay;
4071*67e74705SXin Li 
4072*67e74705SXin Li   switch (ARCCastChecker(Context, exprACTC, castACTC, false).Visit(castExpr)) {
4073*67e74705SXin Li   // For invalid casts, fall through.
4074*67e74705SXin Li   case ACC_invalid:
4075*67e74705SXin Li     break;
4076*67e74705SXin Li 
4077*67e74705SXin Li   // Do nothing for both bottom and +0.
4078*67e74705SXin Li   case ACC_bottom:
4079*67e74705SXin Li   case ACC_plusZero:
4080*67e74705SXin Li     return ACR_okay;
4081*67e74705SXin Li 
4082*67e74705SXin Li   // If the result is +1, consume it here.
4083*67e74705SXin Li   case ACC_plusOne:
4084*67e74705SXin Li     castExpr = ImplicitCastExpr::Create(Context, castExpr->getType(),
4085*67e74705SXin Li                                         CK_ARCConsumeObject, castExpr,
4086*67e74705SXin Li                                         nullptr, VK_RValue);
4087*67e74705SXin Li     Cleanup.setExprNeedsCleanups(true);
4088*67e74705SXin Li     return ACR_okay;
4089*67e74705SXin Li   }
4090*67e74705SXin Li 
4091*67e74705SXin Li   // If this is a non-implicit cast from id or block type to a
4092*67e74705SXin Li   // CoreFoundation type, delay complaining in case the cast is used
4093*67e74705SXin Li   // in an acceptable context.
4094*67e74705SXin Li   if (exprACTC == ACTC_retainable && isAnyRetainable(castACTC) &&
4095*67e74705SXin Li       CCK != CCK_ImplicitConversion)
4096*67e74705SXin Li     return ACR_unbridged;
4097*67e74705SXin Li 
4098*67e74705SXin Li   // Issue a diagnostic about a missing @-sign when implicit casting a cstring
4099*67e74705SXin Li   // to 'NSString *', instead of falling through to report a "bridge cast"
4100*67e74705SXin Li   // diagnostic.
4101*67e74705SXin Li   if (castACTC == ACTC_retainable && exprACTC == ACTC_none &&
4102*67e74705SXin Li       ConversionToObjCStringLiteralCheck(castType, castExpr, Diagnose))
4103*67e74705SXin Li     return ACR_error;
4104*67e74705SXin Li 
4105*67e74705SXin Li   // Do not issue "bridge cast" diagnostic when implicit casting
4106*67e74705SXin Li   // a retainable object to a CF type parameter belonging to an audited
4107*67e74705SXin Li   // CF API function. Let caller issue a normal type mismatched diagnostic
4108*67e74705SXin Li   // instead.
4109*67e74705SXin Li   if ((!DiagnoseCFAudited || exprACTC != ACTC_retainable ||
4110*67e74705SXin Li        castACTC != ACTC_coreFoundation) &&
4111*67e74705SXin Li       !(exprACTC == ACTC_voidPtr && castACTC == ACTC_retainable &&
4112*67e74705SXin Li         (Opc == BO_NE || Opc == BO_EQ))) {
4113*67e74705SXin Li     if (Diagnose)
4114*67e74705SXin Li       diagnoseObjCARCConversion(*this, castRange, castType, castACTC, castExpr,
4115*67e74705SXin Li                                 castExpr, exprACTC, CCK);
4116*67e74705SXin Li     return ACR_error;
4117*67e74705SXin Li   }
4118*67e74705SXin Li   return ACR_okay;
4119*67e74705SXin Li }
4120*67e74705SXin Li 
4121*67e74705SXin Li /// Given that we saw an expression with the ARCUnbridgedCastTy
4122*67e74705SXin Li /// placeholder type, complain bitterly.
diagnoseARCUnbridgedCast(Expr * e)4123*67e74705SXin Li void Sema::diagnoseARCUnbridgedCast(Expr *e) {
4124*67e74705SXin Li   // We expect the spurious ImplicitCastExpr to already have been stripped.
4125*67e74705SXin Li   assert(!e->hasPlaceholderType(BuiltinType::ARCUnbridgedCast));
4126*67e74705SXin Li   CastExpr *realCast = cast<CastExpr>(e->IgnoreParens());
4127*67e74705SXin Li 
4128*67e74705SXin Li   SourceRange castRange;
4129*67e74705SXin Li   QualType castType;
4130*67e74705SXin Li   CheckedConversionKind CCK;
4131*67e74705SXin Li 
4132*67e74705SXin Li   if (CStyleCastExpr *cast = dyn_cast<CStyleCastExpr>(realCast)) {
4133*67e74705SXin Li     castRange = SourceRange(cast->getLParenLoc(), cast->getRParenLoc());
4134*67e74705SXin Li     castType = cast->getTypeAsWritten();
4135*67e74705SXin Li     CCK = CCK_CStyleCast;
4136*67e74705SXin Li   } else if (ExplicitCastExpr *cast = dyn_cast<ExplicitCastExpr>(realCast)) {
4137*67e74705SXin Li     castRange = cast->getTypeInfoAsWritten()->getTypeLoc().getSourceRange();
4138*67e74705SXin Li     castType = cast->getTypeAsWritten();
4139*67e74705SXin Li     CCK = CCK_OtherCast;
4140*67e74705SXin Li   } else {
4141*67e74705SXin Li     castType = cast->getType();
4142*67e74705SXin Li     CCK = CCK_ImplicitConversion;
4143*67e74705SXin Li   }
4144*67e74705SXin Li 
4145*67e74705SXin Li   ARCConversionTypeClass castACTC =
4146*67e74705SXin Li     classifyTypeForARCConversion(castType.getNonReferenceType());
4147*67e74705SXin Li 
4148*67e74705SXin Li   Expr *castExpr = realCast->getSubExpr();
4149*67e74705SXin Li   assert(classifyTypeForARCConversion(castExpr->getType()) == ACTC_retainable);
4150*67e74705SXin Li 
4151*67e74705SXin Li   diagnoseObjCARCConversion(*this, castRange, castType, castACTC,
4152*67e74705SXin Li                             castExpr, realCast, ACTC_retainable, CCK);
4153*67e74705SXin Li }
4154*67e74705SXin Li 
4155*67e74705SXin Li /// stripARCUnbridgedCast - Given an expression of ARCUnbridgedCast
4156*67e74705SXin Li /// type, remove the placeholder cast.
stripARCUnbridgedCast(Expr * e)4157*67e74705SXin Li Expr *Sema::stripARCUnbridgedCast(Expr *e) {
4158*67e74705SXin Li   assert(e->hasPlaceholderType(BuiltinType::ARCUnbridgedCast));
4159*67e74705SXin Li 
4160*67e74705SXin Li   if (ParenExpr *pe = dyn_cast<ParenExpr>(e)) {
4161*67e74705SXin Li     Expr *sub = stripARCUnbridgedCast(pe->getSubExpr());
4162*67e74705SXin Li     return new (Context) ParenExpr(pe->getLParen(), pe->getRParen(), sub);
4163*67e74705SXin Li   } else if (UnaryOperator *uo = dyn_cast<UnaryOperator>(e)) {
4164*67e74705SXin Li     assert(uo->getOpcode() == UO_Extension);
4165*67e74705SXin Li     Expr *sub = stripARCUnbridgedCast(uo->getSubExpr());
4166*67e74705SXin Li     return new (Context) UnaryOperator(sub, UO_Extension, sub->getType(),
4167*67e74705SXin Li                                    sub->getValueKind(), sub->getObjectKind(),
4168*67e74705SXin Li                                        uo->getOperatorLoc());
4169*67e74705SXin Li   } else if (GenericSelectionExpr *gse = dyn_cast<GenericSelectionExpr>(e)) {
4170*67e74705SXin Li     assert(!gse->isResultDependent());
4171*67e74705SXin Li 
4172*67e74705SXin Li     unsigned n = gse->getNumAssocs();
4173*67e74705SXin Li     SmallVector<Expr*, 4> subExprs(n);
4174*67e74705SXin Li     SmallVector<TypeSourceInfo*, 4> subTypes(n);
4175*67e74705SXin Li     for (unsigned i = 0; i != n; ++i) {
4176*67e74705SXin Li       subTypes[i] = gse->getAssocTypeSourceInfo(i);
4177*67e74705SXin Li       Expr *sub = gse->getAssocExpr(i);
4178*67e74705SXin Li       if (i == gse->getResultIndex())
4179*67e74705SXin Li         sub = stripARCUnbridgedCast(sub);
4180*67e74705SXin Li       subExprs[i] = sub;
4181*67e74705SXin Li     }
4182*67e74705SXin Li 
4183*67e74705SXin Li     return new (Context) GenericSelectionExpr(Context, gse->getGenericLoc(),
4184*67e74705SXin Li                                               gse->getControllingExpr(),
4185*67e74705SXin Li                                               subTypes, subExprs,
4186*67e74705SXin Li                                               gse->getDefaultLoc(),
4187*67e74705SXin Li                                               gse->getRParenLoc(),
4188*67e74705SXin Li                                        gse->containsUnexpandedParameterPack(),
4189*67e74705SXin Li                                               gse->getResultIndex());
4190*67e74705SXin Li   } else {
4191*67e74705SXin Li     assert(isa<ImplicitCastExpr>(e) && "bad form of unbridged cast!");
4192*67e74705SXin Li     return cast<ImplicitCastExpr>(e)->getSubExpr();
4193*67e74705SXin Li   }
4194*67e74705SXin Li }
4195*67e74705SXin Li 
CheckObjCARCUnavailableWeakConversion(QualType castType,QualType exprType)4196*67e74705SXin Li bool Sema::CheckObjCARCUnavailableWeakConversion(QualType castType,
4197*67e74705SXin Li                                                  QualType exprType) {
4198*67e74705SXin Li   QualType canCastType =
4199*67e74705SXin Li     Context.getCanonicalType(castType).getUnqualifiedType();
4200*67e74705SXin Li   QualType canExprType =
4201*67e74705SXin Li     Context.getCanonicalType(exprType).getUnqualifiedType();
4202*67e74705SXin Li   if (isa<ObjCObjectPointerType>(canCastType) &&
4203*67e74705SXin Li       castType.getObjCLifetime() == Qualifiers::OCL_Weak &&
4204*67e74705SXin Li       canExprType->isObjCObjectPointerType()) {
4205*67e74705SXin Li     if (const ObjCObjectPointerType *ObjT =
4206*67e74705SXin Li         canExprType->getAs<ObjCObjectPointerType>())
4207*67e74705SXin Li       if (const ObjCInterfaceDecl *ObjI = ObjT->getInterfaceDecl())
4208*67e74705SXin Li         return !ObjI->isArcWeakrefUnavailable();
4209*67e74705SXin Li   }
4210*67e74705SXin Li   return true;
4211*67e74705SXin Li }
4212*67e74705SXin Li 
4213*67e74705SXin Li /// Look for an ObjCReclaimReturnedObject cast and destroy it.
maybeUndoReclaimObject(Expr * e)4214*67e74705SXin Li static Expr *maybeUndoReclaimObject(Expr *e) {
4215*67e74705SXin Li   // For now, we just undo operands that are *immediately* reclaim
4216*67e74705SXin Li   // expressions, which prevents the vast majority of potential
4217*67e74705SXin Li   // problems here.  To catch them all, we'd need to rebuild arbitrary
4218*67e74705SXin Li   // value-propagating subexpressions --- we can't reliably rebuild
4219*67e74705SXin Li   // in-place because of expression sharing.
4220*67e74705SXin Li   if (ImplicitCastExpr *ice = dyn_cast<ImplicitCastExpr>(e))
4221*67e74705SXin Li     if (ice->getCastKind() == CK_ARCReclaimReturnedObject)
4222*67e74705SXin Li       return ice->getSubExpr();
4223*67e74705SXin Li 
4224*67e74705SXin Li   return e;
4225*67e74705SXin Li }
4226*67e74705SXin Li 
BuildObjCBridgedCast(SourceLocation LParenLoc,ObjCBridgeCastKind Kind,SourceLocation BridgeKeywordLoc,TypeSourceInfo * TSInfo,Expr * SubExpr)4227*67e74705SXin Li ExprResult Sema::BuildObjCBridgedCast(SourceLocation LParenLoc,
4228*67e74705SXin Li                                       ObjCBridgeCastKind Kind,
4229*67e74705SXin Li                                       SourceLocation BridgeKeywordLoc,
4230*67e74705SXin Li                                       TypeSourceInfo *TSInfo,
4231*67e74705SXin Li                                       Expr *SubExpr) {
4232*67e74705SXin Li   ExprResult SubResult = UsualUnaryConversions(SubExpr);
4233*67e74705SXin Li   if (SubResult.isInvalid()) return ExprError();
4234*67e74705SXin Li   SubExpr = SubResult.get();
4235*67e74705SXin Li 
4236*67e74705SXin Li   QualType T = TSInfo->getType();
4237*67e74705SXin Li   QualType FromType = SubExpr->getType();
4238*67e74705SXin Li 
4239*67e74705SXin Li   CastKind CK;
4240*67e74705SXin Li 
4241*67e74705SXin Li   bool MustConsume = false;
4242*67e74705SXin Li   if (T->isDependentType() || SubExpr->isTypeDependent()) {
4243*67e74705SXin Li     // Okay: we'll build a dependent expression type.
4244*67e74705SXin Li     CK = CK_Dependent;
4245*67e74705SXin Li   } else if (T->isObjCARCBridgableType() && FromType->isCARCBridgableType()) {
4246*67e74705SXin Li     // Casting CF -> id
4247*67e74705SXin Li     CK = (T->isBlockPointerType() ? CK_AnyPointerToBlockPointerCast
4248*67e74705SXin Li                                   : CK_CPointerToObjCPointerCast);
4249*67e74705SXin Li     switch (Kind) {
4250*67e74705SXin Li     case OBC_Bridge:
4251*67e74705SXin Li       break;
4252*67e74705SXin Li 
4253*67e74705SXin Li     case OBC_BridgeRetained: {
4254*67e74705SXin Li       bool br = isKnownName("CFBridgingRelease");
4255*67e74705SXin Li       Diag(BridgeKeywordLoc, diag::err_arc_bridge_cast_wrong_kind)
4256*67e74705SXin Li         << 2
4257*67e74705SXin Li         << FromType
4258*67e74705SXin Li         << (T->isBlockPointerType()? 1 : 0)
4259*67e74705SXin Li         << T
4260*67e74705SXin Li         << SubExpr->getSourceRange()
4261*67e74705SXin Li         << Kind;
4262*67e74705SXin Li       Diag(BridgeKeywordLoc, diag::note_arc_bridge)
4263*67e74705SXin Li         << FixItHint::CreateReplacement(BridgeKeywordLoc, "__bridge");
4264*67e74705SXin Li       Diag(BridgeKeywordLoc, diag::note_arc_bridge_transfer)
4265*67e74705SXin Li         << FromType << br
4266*67e74705SXin Li         << FixItHint::CreateReplacement(BridgeKeywordLoc,
4267*67e74705SXin Li                                         br ? "CFBridgingRelease "
4268*67e74705SXin Li                                            : "__bridge_transfer ");
4269*67e74705SXin Li 
4270*67e74705SXin Li       Kind = OBC_Bridge;
4271*67e74705SXin Li       break;
4272*67e74705SXin Li     }
4273*67e74705SXin Li 
4274*67e74705SXin Li     case OBC_BridgeTransfer:
4275*67e74705SXin Li       // We must consume the Objective-C object produced by the cast.
4276*67e74705SXin Li       MustConsume = true;
4277*67e74705SXin Li       break;
4278*67e74705SXin Li     }
4279*67e74705SXin Li   } else if (T->isCARCBridgableType() && FromType->isObjCARCBridgableType()) {
4280*67e74705SXin Li     // Okay: id -> CF
4281*67e74705SXin Li     CK = CK_BitCast;
4282*67e74705SXin Li     switch (Kind) {
4283*67e74705SXin Li     case OBC_Bridge:
4284*67e74705SXin Li       // Reclaiming a value that's going to be __bridge-casted to CF
4285*67e74705SXin Li       // is very dangerous, so we don't do it.
4286*67e74705SXin Li       SubExpr = maybeUndoReclaimObject(SubExpr);
4287*67e74705SXin Li       break;
4288*67e74705SXin Li 
4289*67e74705SXin Li     case OBC_BridgeRetained:
4290*67e74705SXin Li       // Produce the object before casting it.
4291*67e74705SXin Li       SubExpr = ImplicitCastExpr::Create(Context, FromType,
4292*67e74705SXin Li                                          CK_ARCProduceObject,
4293*67e74705SXin Li                                          SubExpr, nullptr, VK_RValue);
4294*67e74705SXin Li       break;
4295*67e74705SXin Li 
4296*67e74705SXin Li     case OBC_BridgeTransfer: {
4297*67e74705SXin Li       bool br = isKnownName("CFBridgingRetain");
4298*67e74705SXin Li       Diag(BridgeKeywordLoc, diag::err_arc_bridge_cast_wrong_kind)
4299*67e74705SXin Li         << (FromType->isBlockPointerType()? 1 : 0)
4300*67e74705SXin Li         << FromType
4301*67e74705SXin Li         << 2
4302*67e74705SXin Li         << T
4303*67e74705SXin Li         << SubExpr->getSourceRange()
4304*67e74705SXin Li         << Kind;
4305*67e74705SXin Li 
4306*67e74705SXin Li       Diag(BridgeKeywordLoc, diag::note_arc_bridge)
4307*67e74705SXin Li         << FixItHint::CreateReplacement(BridgeKeywordLoc, "__bridge ");
4308*67e74705SXin Li       Diag(BridgeKeywordLoc, diag::note_arc_bridge_retained)
4309*67e74705SXin Li         << T << br
4310*67e74705SXin Li         << FixItHint::CreateReplacement(BridgeKeywordLoc,
4311*67e74705SXin Li                           br ? "CFBridgingRetain " : "__bridge_retained");
4312*67e74705SXin Li 
4313*67e74705SXin Li       Kind = OBC_Bridge;
4314*67e74705SXin Li       break;
4315*67e74705SXin Li     }
4316*67e74705SXin Li     }
4317*67e74705SXin Li   } else {
4318*67e74705SXin Li     Diag(LParenLoc, diag::err_arc_bridge_cast_incompatible)
4319*67e74705SXin Li       << FromType << T << Kind
4320*67e74705SXin Li       << SubExpr->getSourceRange()
4321*67e74705SXin Li       << TSInfo->getTypeLoc().getSourceRange();
4322*67e74705SXin Li     return ExprError();
4323*67e74705SXin Li   }
4324*67e74705SXin Li 
4325*67e74705SXin Li   Expr *Result = new (Context) ObjCBridgedCastExpr(LParenLoc, Kind, CK,
4326*67e74705SXin Li                                                    BridgeKeywordLoc,
4327*67e74705SXin Li                                                    TSInfo, SubExpr);
4328*67e74705SXin Li 
4329*67e74705SXin Li   if (MustConsume) {
4330*67e74705SXin Li     Cleanup.setExprNeedsCleanups(true);
4331*67e74705SXin Li     Result = ImplicitCastExpr::Create(Context, T, CK_ARCConsumeObject, Result,
4332*67e74705SXin Li                                       nullptr, VK_RValue);
4333*67e74705SXin Li   }
4334*67e74705SXin Li 
4335*67e74705SXin Li   return Result;
4336*67e74705SXin Li }
4337*67e74705SXin Li 
ActOnObjCBridgedCast(Scope * S,SourceLocation LParenLoc,ObjCBridgeCastKind Kind,SourceLocation BridgeKeywordLoc,ParsedType Type,SourceLocation RParenLoc,Expr * SubExpr)4338*67e74705SXin Li ExprResult Sema::ActOnObjCBridgedCast(Scope *S,
4339*67e74705SXin Li                                       SourceLocation LParenLoc,
4340*67e74705SXin Li                                       ObjCBridgeCastKind Kind,
4341*67e74705SXin Li                                       SourceLocation BridgeKeywordLoc,
4342*67e74705SXin Li                                       ParsedType Type,
4343*67e74705SXin Li                                       SourceLocation RParenLoc,
4344*67e74705SXin Li                                       Expr *SubExpr) {
4345*67e74705SXin Li   TypeSourceInfo *TSInfo = nullptr;
4346*67e74705SXin Li   QualType T = GetTypeFromParser(Type, &TSInfo);
4347*67e74705SXin Li   if (Kind == OBC_Bridge)
4348*67e74705SXin Li     CheckTollFreeBridgeCast(T, SubExpr);
4349*67e74705SXin Li   if (!TSInfo)
4350*67e74705SXin Li     TSInfo = Context.getTrivialTypeSourceInfo(T, LParenLoc);
4351*67e74705SXin Li   return BuildObjCBridgedCast(LParenLoc, Kind, BridgeKeywordLoc, TSInfo,
4352*67e74705SXin Li                               SubExpr);
4353*67e74705SXin Li }
4354