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