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