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